fix CorruptInputError index error, #1

emmansun · Nov 2, 2023 · ca9c310 · ca9c310
1 parent 66abdc7
commit ca9c310
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/base64_asm.go b/base64_asm.go
@@ -42,10 +42,13 @@ func decode(enc *Encoding, dst, src []byte) (int, error) {
 			// decoded by SIMD
 			remain = srcLen - remain
 			src = src[remain:]
-			remain = (remain / 4) * 3
-			dst = dst[remain:]
+			dstStart := (remain / 4) * 3
+			dst = dst[dstStart:]
 			n, err := decodeGeneric(enc, dst, src)
-			return n + remain, err
+			if cerr, ok := err.(CorruptInputError); ok {
+				return n + dstStart, CorruptInputError(int(cerr) + remain)
+			}
+			return n + dstStart, err
 		}
 	}
 	return decodeGeneric(enc, dst, src)