Fix broken test

liamg · Dec 28, 2020 · ead7a2d · ead7a2d
1 parent dc23aea
commit ead7a2d
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 2 deletions.
diff --git a/internal/app/pax/exploit.go b/internal/app/pax/exploit.go
@@ -131,7 +131,7 @@ func (x *exploiter) decode(input []byte) ([]byte, error) {
 		decoded, err := url.QueryUnescape(string(input))
 		return []byte(decoded), err
 	case EncodingBase64:
-		var dst []byte
+		dst := make([]byte, base64.StdEncoding.DecodedLen(len(input)))
 		_, err := base64.StdEncoding.Decode(dst, input)
 		return dst, err
 	case EncodingBase64URL:

diff --git a/internal/app/pax/exploit_test.go b/internal/app/pax/exploit_test.go
@@ -53,7 +53,7 @@ func Test_DecryptViaCookie(t *testing.T) {
 	result, err := Decrypt(ts.URL, encodedCipherText, &ExploitOptions{
 		BlockSize: aes.BlockSize,
 		Cookies:   "ENC=" + encodedCipherText,
-		Encoding:  EncodingBase64URL,
+		Encoding:  EncodingBase64,
 	})
 	require.NoError(t, err)
 

diff --git a/internal/app/pax/server.go → internal/app/pax/server_test.go b/internal/app/pax/server.go → internal/app/pax/server_test.go