57 lines
1.0 KiB
Go
57 lines
1.0 KiB
Go
|
package ssob
|
||
|
|
||
|
import (
|
||
|
"encoding/binary"
|
||
|
"errors"
|
||
|
"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 errors.New("ssob: Cannot decode nil value")
|
||
|
}
|
||
|
|
||
|
if value.Kind() == reflect.Ptr && value.IsNil() {
|
||
|
return errors.New("ssob: Cannot decode nil of type " + value.Type().String())
|
||
|
}
|
||
|
|
||
|
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
|
||
|
}
|