2019-10-04 12:28:45 +02:00
|
|
|
package ssob
|
|
|
|
|
|
|
|
import (
|
|
|
|
"encoding/binary"
|
|
|
|
"errors"
|
2021-08-21 22:29:18 +02:00
|
|
|
"fmt"
|
2019-10-04 12:28:45 +02:00
|
|
|
"io"
|
|
|
|
"reflect"
|
|
|
|
"sync"
|
2021-09-02 00:08:59 +02:00
|
|
|
|
|
|
|
"gitea.olznet.de/OlzNet/slog"
|
2019-10-04 12:28:45 +02:00
|
|
|
)
|
|
|
|
|
|
|
|
type Decoder struct {
|
|
|
|
mutex sync.Mutex
|
|
|
|
r io.Reader
|
|
|
|
}
|
|
|
|
|
|
|
|
func NewDecoder(r io.Reader) *Decoder {
|
|
|
|
dec := new(Decoder)
|
|
|
|
dec.r = r
|
|
|
|
return dec
|
|
|
|
}
|
|
|
|
|
|
|
|
func (dec *Decoder) Decode(e interface{}) (err error) {
|
2021-09-02 00:08:59 +02:00
|
|
|
t := reflect.TypeOf(e)
|
|
|
|
slog.LOG_DEBUGFLN("Dec: %v", t)
|
|
|
|
if t == reflect.TypeOf(errors.New("")) || t == reflect.TypeOf((*error)(nil)) {
|
2021-08-21 22:29:18 +02:00
|
|
|
var errstr string
|
|
|
|
if err = dec.Decode(&errstr); err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
if errstr != "" {
|
2021-09-02 00:08:59 +02:00
|
|
|
*e.(*error) = errors.New(errstr)
|
2021-08-21 22:29:18 +02:00
|
|
|
}
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2021-09-02 00:08:59 +02:00
|
|
|
return dec.DecodeValue(reflect.ValueOf(e))
|
|
|
|
}
|
|
|
|
|
|
|
|
func (dec *Decoder) DecodeValue(value reflect.Value) (err error) {
|
2019-10-04 12:28:45 +02:00
|
|
|
if value.Kind() == reflect.Invalid {
|
2021-08-21 22:29:18 +02:00
|
|
|
return fmt.Errorf("ssob: Cannot decode nil: %w", ErrValueInvalid)
|
2019-10-04 12:28:45 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
if value.Kind() == reflect.Ptr && value.IsNil() {
|
2021-08-21 22:29:18 +02:00
|
|
|
return fmt.Errorf("ssob: Cannot decode nil of type %s: %w", value.Type().String(), ErrValueInvalid)
|
2019-10-04 12:28:45 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
dec.mutex.Lock()
|
|
|
|
defer dec.mutex.Unlock()
|
|
|
|
|
2019-10-05 01:42:09 +02:00
|
|
|
lb := make([]byte, 4)
|
2019-10-04 12:28:45 +02:00
|
|
|
err = binary.Read(dec.r, binary.BigEndian, lb)
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
2019-10-05 01:42:09 +02:00
|
|
|
l, n, err := UnmarshalInt32(lb)
|
|
|
|
if err != nil || n != 4 {
|
2019-10-04 12:28:45 +02:00
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
bb := make([]byte, l)
|
|
|
|
err = binary.Read(dec.r, binary.BigEndian, bb)
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
_, err = unmarshal(value.Interface(), bb)
|
|
|
|
return err
|
|
|
|
}
|
2021-09-02 00:08:59 +02:00
|
|
|
|
|
|
|
func (dec *Decoder) NDecode() (ret interface{}, err error) {
|
|
|
|
st := make([]byte, ST_ID_SIZE)
|
|
|
|
|
|
|
|
dec.mutex.Lock()
|
|
|
|
defer dec.mutex.Unlock()
|
|
|
|
|
|
|
|
if err = binary.Read(dec.r, binary.BigEndian, st); err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
|
|
|
|
stype := binary.BigEndian.Uint32(st)
|
|
|
|
if t, ok := cacheTypes.types[stype]; !ok {
|
|
|
|
return nil, fmt.Errorf("ssob: Decode type '%d' not registered: %w", stype, ErrTypeUnknown)
|
|
|
|
} else {
|
|
|
|
return t.unmarshal(dec.r)
|
|
|
|
}
|
|
|
|
}
|