41 lines
869 B
Go
41 lines
869 B
Go
package encrypt
|
|
|
|
import (
|
|
"errors"
|
|
"reflect"
|
|
)
|
|
|
|
var methods = map[string]reflect.Type{
|
|
"raw": reflect.TypeOf(new(RawMethod)).Elem(),
|
|
"aes-256-cfb": reflect.TypeOf(new(AES256CFBMethod)).Elem(),
|
|
}
|
|
|
|
func NewMethodInstance(method string, key string, iv string) (MethodInterface, error) {
|
|
valueType, ok := methods[method]
|
|
if !ok {
|
|
return nil, errors.New("method '" + method + "' not found")
|
|
}
|
|
|
|
instance, ok := reflect.New(valueType).Interface().(MethodInterface)
|
|
if !ok {
|
|
return nil, errors.New("method '" + method + "' must implement MethodInterface")
|
|
}
|
|
|
|
err := instance.Init([]byte(key), []byte(iv))
|
|
return instance, err
|
|
}
|
|
|
|
func RecoverMethodPanic(err interface{}) error {
|
|
if err == nil {
|
|
return nil
|
|
}
|
|
|
|
if s, ok := err.(string); ok {
|
|
return errors.New(s)
|
|
}
|
|
if e, ok := err.(error); ok {
|
|
return e
|
|
}
|
|
return errors.New("unknown error")
|
|
}
|