Fix overflow bug

evalphobia · evalphobia · commit f9ec2ea40e64 · 2018-02-19T19:49:28.000+09:00
diff --git a/googletts/token.go b/googletts/token.go
@@ -42,11 +42,11 @@ func getTTSKeyFromHTML() (key1, key2 int64, err error) {
 	if len(matchList) != 4 {
 		return 0, 0, fmt.Errorf("HTML parse error")
 	}
-	part1, _ := strconv.Atoi(matchList[1])
-	part2, _ := strconv.Atoi(matchList[2])
-	part3, _ := strconv.Atoi(matchList[3])
+	part1, _ := strconv.ParseInt(matchList[1], 10, 64)
+	part2, _ := strconv.ParseInt(matchList[2], 10, 64)
+	part3, _ := strconv.ParseInt(matchList[3], 10, 64)
 
-	return int64(part3), int64(part1 + part2), nil
+	return part3, part1 + part2, nil
 }
 
 // CalculateToken caluculates token from tts text and seed keys.

Original file line number	Diff line number	Diff line change
`@@ -42,11 +42,11 @@ func getTTSKeyFromHTML() (key1, key2 int64, err error) {`
`42`	`42`	`if len(matchList) != 4 {`
`43`	`43`	`return 0, 0, fmt.Errorf("HTML parse error")`
`44`	`44`	`}`
`45`		`- part1, _ := strconv.Atoi(matchList[1])`
`46`		`- part2, _ := strconv.Atoi(matchList[2])`
`47`		`- part3, _ := strconv.Atoi(matchList[3])`
	`45`	`+ part1, _ := strconv.ParseInt(matchList[1], 10, 64)`
	`46`	`+ part2, _ := strconv.ParseInt(matchList[2], 10, 64)`
	`47`	`+ part3, _ := strconv.ParseInt(matchList[3], 10, 64)`
`48`	`48`
`49`		`- return int64(part3), int64(part1 + part2), nil`
	`49`	`+ return part3, part1 + part2, nil`
`50`	`50`	`}`
`51`	`51`
`52`	`52`	`// CalculateToken caluculates token from tts text and seed keys.`