From 2f38724af055c3d4cd2b2c6fd412ac1073ede363 Mon Sep 17 00:00:00 2001 From: root Date: Tue, 18 Aug 2020 20:27:25 +0000 Subject: [PATCH] make sure to generate 6 digit random number --- src/users_db.go | 2 +- src/utils.go | 2 ++ 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/src/users_db.go b/src/users_db.go index aa7dfbd..3e7ab37 100644 --- a/src/users_db.go +++ b/src/users_db.go @@ -65,7 +65,7 @@ func (dbobj dbcon) createUserRecord(parsedData userJSON, event *auditEvent) (str } func (dbobj dbcon) generateTempLoginCode(userTOKEN string) int32 { - rnd := randNum(4) + rnd := randNum(6) fmt.Printf("random: %d\n", rnd) bdoc := bson.M{} bdoc["tempcode"] = rnd diff --git a/src/utils.go b/src/utils.go index 9a013b4..8329403 100644 --- a/src/utils.go +++ b/src/utils.go @@ -539,6 +539,7 @@ func randSeq(n int) string { return string(b) } +var numbers0 = []rune("123456789") var numbers = []rune("0123456789") func randNum(n int) int32 { @@ -546,5 +547,6 @@ func randNum(n int) int32 { for i := range b { b[i] = numbers[rand.Intn(len(numbers))] } + b[0] = numbers0[rand.Intn(len(numbers0))] return atoi(string(b)) }