57 lines
1.1 KiB
Go
57 lines
1.1 KiB
Go
package ssob
|
|
|
|
import (
|
|
"encoding/binary"
|
|
"fmt"
|
|
"io"
|
|
"reflect"
|
|
"sync"
|
|
)
|
|
|
|
type UnsafeDecoder struct {
|
|
mutex sync.Mutex
|
|
r io.Reader
|
|
}
|
|
|
|
func NewUnsafeDecoder(r io.Reader) *UnsafeDecoder {
|
|
dec := new(UnsafeDecoder)
|
|
dec.r = r
|
|
return dec
|
|
}
|
|
|
|
func (dec *UnsafeDecoder) Decode(e interface{}) (err error) {
|
|
return dec.DecodeValue(reflect.ValueOf(e))
|
|
}
|
|
|
|
func (dec *UnsafeDecoder) DecodeValue(value reflect.Value) (err error) {
|
|
if value.Kind() == reflect.Invalid {
|
|
return fmt.Errorf("ssob: Cannot decode nil: %w", ErrValueInvalid)
|
|
}
|
|
|
|
if value.Kind() == reflect.Ptr && value.IsNil() {
|
|
return fmt.Errorf("ssob: Cannot decode nil of type %s: %w", value.Type().String(), ErrValueInvalid)
|
|
}
|
|
|
|
dec.mutex.Lock()
|
|
defer dec.mutex.Unlock()
|
|
|
|
lb := make([]byte, 4)
|
|
err = binary.Read(dec.r, binary.BigEndian, lb)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
l, n, err := UnsafeUnmarshalInt32(lb)
|
|
if err != nil || n != 4 {
|
|
return err
|
|
}
|
|
|
|
bb := make([]byte, l)
|
|
err = binary.Read(dec.r, binary.BigEndian, bb)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
_, err = unsafeUnmarshal(value.Interface(), bb)
|
|
return err
|
|
}
|