package nametransform import ( "bytes" "strings" "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() } } } func TestIsValidName(t *testing.T) { cases := []struct { in string want bool }{ {"", false}, {".", false}, {"..", false}, {"...", true}, {"asdasd/asdasd", false}, {"asdasd\000asdasd", false}, {"hello", true}, {strings.Repeat("x", 255), true}, {strings.Repeat("x", 256), false}, } for _, c := range cases { have := IsValidName(c.in) if (have == nil) != c.want { t.Errorf("IsValidName(%q): want %v have %v", c.in, c.want, have) } } } func TestIsValidXattrName(t *testing.T) { cases := []struct { in string want bool }{ {"", false}, {".", true}, {"..", true}, {"...", true}, {"asdasd/asdasd", true}, {"asdasd\000asdasd", false}, {"hello", true}, {strings.Repeat("x", 255), true}, {strings.Repeat("x", 256), true}, } for _, c := range cases { have := isValidXattrName(c.in) if (have == nil) != c.want { t.Errorf("isValidXattrName(%q): want %v have %v", c.in, c.want, have) } } }