mirror of
https://github.com/shadow1ng/fscan.git
synced 2026-09-22 03:10:42 +08:00
109 lines
2.8 KiB
Go
109 lines
2.8 KiB
Go
package lib
|
||
|
||
import (
|
||
"testing"
|
||
|
||
"github.com/google/cel-go/common/types"
|
||
)
|
||
|
||
func TestRegisterCryptoImplementations(t *testing.T) {
|
||
overloads := registerCryptoImplementations()
|
||
|
||
// 建立 operator → index 映射
|
||
idx := make(map[string]int, len(overloads))
|
||
for i, o := range overloads {
|
||
idx[o.Operator] = i
|
||
}
|
||
|
||
t.Run("md5_string", func(t *testing.T) {
|
||
i, ok := idx["md5_string"]
|
||
if !ok {
|
||
t.Fatal("overload md5_string not found")
|
||
}
|
||
unary := overloads[i].Unary
|
||
|
||
tests := []struct {
|
||
name string
|
||
input types.String
|
||
want types.String
|
||
wantErr bool
|
||
}{
|
||
{"hello", "hello", "5d41402abc4b2a76b9719d911017c592", false},
|
||
{"empty", "", "d41d8cd98f00b204e9800998ecf8427e", false},
|
||
{"abc", "abc", "900150983cd24fb0d6963f7d28e17f72", false},
|
||
}
|
||
|
||
for _, tc := range tests {
|
||
t.Run(tc.name, func(t *testing.T) {
|
||
result := unary(tc.input)
|
||
if types.IsError(result) {
|
||
t.Fatalf("unexpected error: %v", result)
|
||
}
|
||
got, ok := result.(types.String)
|
||
if !ok {
|
||
t.Fatalf("expected types.String, got %T", result)
|
||
}
|
||
if got != tc.want {
|
||
t.Errorf("md5(%q) = %q, want %q", tc.input, got, tc.want)
|
||
}
|
||
})
|
||
}
|
||
})
|
||
|
||
t.Run("md5_string_wrong_type", func(t *testing.T) {
|
||
i := idx["md5_string"]
|
||
result := overloads[i].Unary(types.Int(42))
|
||
if !types.IsError(result) {
|
||
t.Errorf("expected error for non-String input, got %v", result)
|
||
}
|
||
})
|
||
|
||
t.Run("shiro_key_valid", func(t *testing.T) {
|
||
i, ok := idx["shiro_key"]
|
||
if !ok {
|
||
t.Fatal("overload shiro_key not found")
|
||
}
|
||
binary := overloads[i].Binary
|
||
|
||
// kPH+bIxk5D2deZiIxcaaaA== 是常见 shiro 默认 key
|
||
result := binary(types.String("kPH+bIxk5D2deZiIxcaaaA=="), types.String("cbc"))
|
||
if types.IsError(result) {
|
||
t.Fatalf("unexpected error: %v", result)
|
||
}
|
||
got, ok := result.(types.String)
|
||
if !ok {
|
||
t.Fatalf("expected types.String, got %T", result)
|
||
}
|
||
if got == "" {
|
||
t.Error("shiro_key returned empty string")
|
||
}
|
||
})
|
||
|
||
t.Run("shiro_key_invalid_base64", func(t *testing.T) {
|
||
i := idx["shiro_key"]
|
||
binary := overloads[i].Binary
|
||
|
||
// 无效 base64，GetShrioCookie 会返回 ""，函数返回 NewErr
|
||
result := binary(types.String("!!!not_valid_base64!!!"), types.String("cbc"))
|
||
if !types.IsError(result) {
|
||
t.Errorf("expected error for invalid base64 key, got %v", result)
|
||
}
|
||
})
|
||
|
||
t.Run("shiro_key_wrong_key_type", func(t *testing.T) {
|
||
i := idx["shiro_key"]
|
||
result := overloads[i].Binary(types.Int(1), types.String("cbc"))
|
||
if !types.IsError(result) {
|
||
t.Error("expected error for non-String key")
|
||
}
|
||
})
|
||
|
||
t.Run("shiro_key_wrong_mode_type", func(t *testing.T) {
|
||
i := idx["shiro_key"]
|
||
result := overloads[i].Binary(types.String("kPH+bIxk5D2deZiIxcaaaA=="), types.Int(0))
|
||
if !types.IsError(result) {
|
||
t.Error("expected error for non-String mode")
|
||
}
|
||
})
|
||
}
|