github-personal/databunker
1
0
Fork 0
mirror of https://github.com/outbackdingo/databunker.git synced 2026-01-27 18:18:43 +00:00
Code Issues Packages Projects Releases Wiki Activity

make sure to generate 6 digit random number

Browse Source
This commit is contained in:
root
2020-08-18 20:27:25 +00:00
parent 085b276098
commit 2f38724af0
2 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/users_db.go
View File

@@ -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

2
src/utils.go
View File

@@ -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))
}