out of range fix

M3chD09 · M3chD09 · commit d4bb201bac91 · 2021-05-27T19:08:25.000+08:00
diff --git a/datautil/datautil.go b/datautil/datautil.go
@@ -54,8 +54,13 @@ func FindValidPart(data []byte) (int, int) {
 		return 0, 0
 	}
 
+	a := (n - r) % 20
+	if a == 0 {
+		a = 20
+	}
+
 	// 也许收获满满
-	r = n - (n-r)%20
+	r = n - a
 	return l, r + 1
 }
 
diff --git a/datautil/datautil_test.go b/datautil/datautil_test.go
@@ -115,6 +115,60 @@ func TestFindValidPart(t *testing.T) {
 			startIdx: 5,
 			endIdx:   25,
 		},
+		{
+			data: []byte{
+				0xa,
+
+				0x1, 0x2, 0x4,
+				0x30, 0x4, 0x0, 0x20,
+				0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0,
+				0x41, 0xe2, 0x0, 0x0,
+				0xa,
+
+				0x1, 0x2, 0x4,
+				0xa8, 0x4, 0x0, 0x20,
+				0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0,
+				0x41, 0xe2, 0x0, 0x0,
+				0xa,
+
+				0x1, 0x2, 0x4,
+				0xac, 0x1, 0x0, 0x20,
+				0x0, 0x0, 0xb4, 0xbe, 0x0, 0x0, 0x0, 0x0,
+				0x4b, 0xe2, 0x0, 0x0,
+				0xa,
+
+				0x1, 0x2, 0x4,
+				0x40, 0x3, 0x0, 0x20,
+				0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0,
+				0x4b, 0xe2, 0x0, 0x0,
+				0xa,
+
+				0x1, 0x2, 0x4,
+				0xb8, 0x3, 0x0, 0x20,
+				0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0,
+				0x4b, 0xe2, 0x0, 0x0,
+				0xa,
+
+				0x1, 0x2, 0x4,
+				0x30, 0x4, 0x0, 0x20,
+				0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0,
+				0x4b, 0xe2, 0x0, 0x0,
+				0xa,
+
+				0x1, 0x2, 0x4,
+				0xa8, 0x4, 0x0, 0x20,
+				0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0,
+				0x4b, 0xe2, 0x0, 0x0,
+				0xa,
+
+				0x1, 0x2, 0x4,
+				0xac, 0x1, 0x0, 0x20, 0x0, 0x80, 0x9d, 0xbe,
+				0x0, 0x0, 0x0, 0x0,
+				0x55, 0xe2, 0x0, 0x0,
+			},
+			startIdx: 1,
+			endIdx:   141,
+		},
 	}
 	for _, c := range cases {
 		s, n := FindValidPart(c.data)

Original file line number	Diff line number	Diff line change
`@@ -54,8 +54,13 @@ func FindValidPart(data []byte) (int, int) {`
`54`	`54`	`return 0, 0`
`55`	`55`	`}`
`56`	`56`
	`57`	`+ a := (n - r) % 20`
	`58`	`+ if a == 0 {`
	`59`	`+ a = 20`
	`60`	`+ }`
	`61`	`+`
`57`	`62`	`// 也许收获满满`
`58`		`- r = n - (n-r)%20`
	`63`	`+ r = n - a`
`59`	`64`	`return l, r + 1`
`60`	`65`	`}`
`61`	`66`