update
This commit is contained in:
parent
3faf6fc167
commit
75a1bf1576
4 changed files with 90 additions and 0 deletions
12
errors.go
12
errors.go
|
@ -2,6 +2,18 @@ package securecookie
|
|||
|
||||
import "errors"
|
||||
|
||||
func NewUsageError(err error) *cookieError {
|
||||
return &cookieError{cause: err, typ: usageError}
|
||||
}
|
||||
|
||||
func NewInternalError(err error) *cookieError {
|
||||
return &cookieError{cause: err, typ: internalError}
|
||||
}
|
||||
|
||||
func NewDecodeError(err error) *cookieError {
|
||||
return &cookieError{cause: err, typ: decodeError}
|
||||
}
|
||||
|
||||
func IsValueTooLong(err error) bool {
|
||||
return errors.Is(err, errEncodedValueTooLong) || errors.Is(err, errValueToDecodeTooLong)
|
||||
}
|
||||
|
|
|
@ -49,6 +49,8 @@ func SetSerializer(codecs []Codec, sz Serializer) {
|
|||
for _, c := range codecs {
|
||||
if codec, ok := c.(*SecureCookie); ok {
|
||||
codec.SetSerializer(sz)
|
||||
} else if codec, ok := c.(*UnsafeCodec); ok {
|
||||
codec.SetSerializer(sz)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
|
56
unsafecodec.go
Normal file
56
unsafecodec.go
Normal file
|
@ -0,0 +1,56 @@
|
|||
package securecookie
|
||||
|
||||
import "fmt"
|
||||
|
||||
var _ Codec = (*UnsafeCodec)(nil)
|
||||
|
||||
func NewUnsafeCodec() *UnsafeCodec {
|
||||
s := &UnsafeCodec{
|
||||
sz: GobEncoder{},
|
||||
}
|
||||
return s
|
||||
}
|
||||
|
||||
// UnsafeCodec encodes and decodes
|
||||
type UnsafeCodec struct {
|
||||
err error
|
||||
sz Serializer
|
||||
}
|
||||
|
||||
// Encoding sets the encoding/serialization method for cookies.
|
||||
//
|
||||
// Default is encoding/gob. To encode special structures using encoding/gob,
|
||||
// they must be registered first using gob.Register().
|
||||
func (s *UnsafeCodec) SetSerializer(sz Serializer) *UnsafeCodec {
|
||||
s.sz = sz
|
||||
|
||||
return s
|
||||
}
|
||||
|
||||
// Encode encodes a value.
|
||||
func (s *UnsafeCodec) Encode(name string, value interface{}) (string, error) {
|
||||
if s.err != nil {
|
||||
return "", s.err
|
||||
}
|
||||
var err error
|
||||
var b []byte
|
||||
// Serialize.
|
||||
if b, err = s.sz.Serialize(value); err != nil {
|
||||
return "", cookieError{cause: fmt.Errorf(`%w: %s`, err, name), typ: usageError}
|
||||
}
|
||||
// Done.
|
||||
return string(b), nil
|
||||
}
|
||||
|
||||
// Decode decodes a value. The dst argument must be a pointer.
|
||||
func (s *UnsafeCodec) Decode(name, value string, dst interface{}, _ ...int) error {
|
||||
if s.err != nil {
|
||||
return s.err
|
||||
}
|
||||
// Deserialize.
|
||||
if err := s.sz.Deserialize([]byte(value), dst); err != nil {
|
||||
return cookieError{cause: fmt.Errorf(`%w: %s`, err, name), typ: decodeError}
|
||||
}
|
||||
// Done.
|
||||
return nil
|
||||
}
|
20
unsafecodec_test.go
Normal file
20
unsafecodec_test.go
Normal file
|
@ -0,0 +1,20 @@
|
|||
package securecookie
|
||||
|
||||
import "testing"
|
||||
|
||||
func TestUnsafeCodec(t *testing.T) {
|
||||
c := NewUnsafeCodec()
|
||||
str := `TestUnsafeCodec`
|
||||
encoded, err := c.Encode(`name`, str)
|
||||
if err != nil {
|
||||
t.Error(err)
|
||||
}
|
||||
var decoded string
|
||||
err = c.Decode(`name`, encoded, &decoded)
|
||||
if err != nil {
|
||||
t.Error(err)
|
||||
}
|
||||
if decoded != str {
|
||||
t.Fatalf("expected %s, got: %#v", str, decoded)
|
||||
}
|
||||
}
|
Loading…
Reference in a new issue