114 lines
3.0 KiB
Go
114 lines
3.0 KiB
Go
package decoder
|
|
|
|
import (
|
|
"encoding/base64"
|
|
"unsafe"
|
|
|
|
"github.com/goccy/go-json/internal/errors"
|
|
"github.com/goccy/go-json/internal/runtime"
|
|
)
|
|
|
|
type bytesDecoder struct {
|
|
typ *runtime.Type
|
|
sliceDecoder Decoder
|
|
stringDecoder *stringDecoder
|
|
structName string
|
|
fieldName string
|
|
}
|
|
|
|
func byteUnmarshalerSliceDecoder(typ *runtime.Type, structName string, fieldName string) Decoder {
|
|
var unmarshalDecoder Decoder
|
|
switch {
|
|
case runtime.PtrTo(typ).Implements(unmarshalJSONType):
|
|
unmarshalDecoder = newUnmarshalJSONDecoder(runtime.PtrTo(typ), structName, fieldName)
|
|
case runtime.PtrTo(typ).Implements(unmarshalTextType):
|
|
unmarshalDecoder = newUnmarshalTextDecoder(runtime.PtrTo(typ), structName, fieldName)
|
|
default:
|
|
unmarshalDecoder, _ = compileUint8(typ, structName, fieldName)
|
|
}
|
|
return newSliceDecoder(unmarshalDecoder, typ, 1, structName, fieldName)
|
|
}
|
|
|
|
func newBytesDecoder(typ *runtime.Type, structName string, fieldName string) *bytesDecoder {
|
|
return &bytesDecoder{
|
|
typ: typ,
|
|
sliceDecoder: byteUnmarshalerSliceDecoder(typ, structName, fieldName),
|
|
stringDecoder: newStringDecoder(structName, fieldName),
|
|
structName: structName,
|
|
fieldName: fieldName,
|
|
}
|
|
}
|
|
|
|
func (d *bytesDecoder) DecodeStream(s *Stream, depth int64, p unsafe.Pointer) error {
|
|
bytes, err := d.decodeStreamBinary(s, depth, p)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if bytes == nil {
|
|
s.reset()
|
|
return nil
|
|
}
|
|
decodedLen := base64.StdEncoding.DecodedLen(len(bytes))
|
|
buf := make([]byte, decodedLen)
|
|
n, err := base64.StdEncoding.Decode(buf, bytes)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
*(*[]byte)(p) = buf[:n]
|
|
s.reset()
|
|
return nil
|
|
}
|
|
|
|
func (d *bytesDecoder) Decode(ctx *RuntimeContext, cursor, depth int64, p unsafe.Pointer) (int64, error) {
|
|
bytes, c, err := d.decodeBinary(ctx, cursor, depth, p)
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
if bytes == nil {
|
|
return c, nil
|
|
}
|
|
cursor = c
|
|
decodedLen := base64.StdEncoding.DecodedLen(len(bytes))
|
|
b := make([]byte, decodedLen)
|
|
n, err := base64.StdEncoding.Decode(b, bytes)
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
*(*[]byte)(p) = b[:n]
|
|
return cursor, nil
|
|
}
|
|
|
|
func (d *bytesDecoder) decodeStreamBinary(s *Stream, depth int64, p unsafe.Pointer) ([]byte, error) {
|
|
c := s.skipWhiteSpace()
|
|
if c == '[' {
|
|
if d.sliceDecoder == nil {
|
|
return nil, &errors.UnmarshalTypeError{
|
|
Type: runtime.RType2Type(d.typ),
|
|
Offset: s.totalOffset(),
|
|
}
|
|
}
|
|
err := d.sliceDecoder.DecodeStream(s, depth, p)
|
|
return nil, err
|
|
}
|
|
return d.stringDecoder.decodeStreamByte(s)
|
|
}
|
|
|
|
func (d *bytesDecoder) decodeBinary(ctx *RuntimeContext, cursor, depth int64, p unsafe.Pointer) ([]byte, int64, error) {
|
|
buf := ctx.Buf
|
|
cursor = skipWhiteSpace(buf, cursor)
|
|
if buf[cursor] == '[' {
|
|
if d.sliceDecoder == nil {
|
|
return nil, 0, &errors.UnmarshalTypeError{
|
|
Type: runtime.RType2Type(d.typ),
|
|
Offset: cursor,
|
|
}
|
|
}
|
|
c, err := d.sliceDecoder.Decode(ctx, cursor, depth, p)
|
|
if err != nil {
|
|
return nil, 0, err
|
|
}
|
|
return nil, c, nil
|
|
}
|
|
return d.stringDecoder.decodeByte(buf, cursor)
|
|
}
|