f6b1c680b3
The comment still mentioned CBC, which has been removed a long time ago. The test definition can be rewritten using slice literals, saving sume stuttering.
54 lines
1.1 KiB
Go
54 lines
1.1 KiB
Go
package nametransform
|
|
|
|
import (
|
|
"bytes"
|
|
"testing"
|
|
)
|
|
|
|
func TestPad16(t *testing.T) {
|
|
s := [][]byte{
|
|
[]byte("foo"),
|
|
[]byte("12345678901234567"),
|
|
[]byte("12345678901234567abcdefg"),
|
|
}
|
|
|
|
for i := range s {
|
|
orig := s[i]
|
|
padded := pad16(orig)
|
|
if len(padded) <= len(orig) {
|
|
t.Errorf("Padded length not bigger than orig: %d", len(padded))
|
|
}
|
|
if len(padded)%16 != 0 {
|
|
t.Errorf("Length is not aligend: %d", len(padded))
|
|
}
|
|
unpadded, err := unPad16(padded)
|
|
if err != nil {
|
|
t.Error("unPad16 returned error:", err)
|
|
}
|
|
if len(unpadded) != len(orig) {
|
|
t.Errorf("Size mismatch: orig=%d unpadded=%d", len(s[i]), len(unpadded))
|
|
}
|
|
if !bytes.Equal(orig, unpadded) {
|
|
t.Error("Content mismatch orig vs unpadded")
|
|
}
|
|
}
|
|
}
|
|
|
|
// TestUnpad16Garbage - unPad16 should never crash on corrupt or malicious inputs
|
|
func TestUnpad16Garbage(t *testing.T) {
|
|
testCases := [][]byte{
|
|
make([]byte, 0),
|
|
make([]byte, 16),
|
|
make([]byte, 1),
|
|
make([]byte, 17),
|
|
bytes.Repeat([]byte{16}, 16),
|
|
bytes.Repeat([]byte{17}, 16),
|
|
}
|
|
for _, v := range testCases {
|
|
_, err := unPad16([]byte(v))
|
|
if err == nil {
|
|
t.Fail()
|
|
}
|
|
}
|
|
}
|