123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867 |
- // +build !safe
- // +build !appengine
- // +build go1.7
- // Copyright (c) 2012-2018 Ugorji Nwoke. All rights reserved.
- // Use of this source code is governed by a MIT license found in the LICENSE file.
- package codec
- import (
- "reflect"
- "sync/atomic"
- "time"
- "unsafe"
- )
- // This file has unsafe variants of some helper methods.
- // NOTE: See helper_not_unsafe.go for the usage information.
- // For reflect.Value code, we decided to do the following:
- // - if we know the kind, we can elide conditional checks for
- // - SetXXX (Int, Uint, String, Bool, etc)
- // - SetLen
- //
- // We can also optimize
- // - IsNil
- const safeMode = false
- // keep in sync with GO_ROOT/src/reflect/value.go
- const (
- unsafeFlagIndir = 1 << 7
- unsafeFlagAddr = 1 << 8
- unsafeFlagKindMask = (1 << 5) - 1 // 5 bits for 27 kinds (up to 31)
- // unsafeTypeKindDirectIface = 1 << 5
- )
- type unsafeString struct {
- Data unsafe.Pointer
- Len int
- }
- type unsafeSlice struct {
- Data unsafe.Pointer
- Len int
- Cap int
- }
- type unsafeIntf struct {
- typ unsafe.Pointer
- word unsafe.Pointer
- }
- type unsafeReflectValue struct {
- typ unsafe.Pointer
- ptr unsafe.Pointer
- flag uintptr
- }
- func stringView(v []byte) string {
- if len(v) == 0 {
- return ""
- }
- bx := (*unsafeSlice)(unsafe.Pointer(&v))
- return *(*string)(unsafe.Pointer(&unsafeString{bx.Data, bx.Len}))
- }
- func bytesView(v string) []byte {
- if len(v) == 0 {
- return zeroByteSlice
- }
- sx := (*unsafeString)(unsafe.Pointer(&v))
- return *(*[]byte)(unsafe.Pointer(&unsafeSlice{sx.Data, sx.Len, sx.Len}))
- }
- // // isNilRef says whether the interface is a nil reference or not.
- // //
- // // A reference here is a pointer-sized reference i.e. map, ptr, chan, func, unsafepointer.
- // // It is optional to extend this to also check if slices or interfaces are nil also.
- // //
- // // NOTE: There is no global way of checking if an interface is nil.
- // // For true references (map, ptr, func, chan), you can just look
- // // at the word of the interface.
- // // However, for slices, you have to dereference
- // // the word, and get a pointer to the 3-word interface value.
- // func isNilRef(v interface{}) (rv reflect.Value, isnil bool) {
- // isnil = ((*unsafeIntf)(unsafe.Pointer(&v))).word == nil
- // return
- // }
- func isNil(v interface{}) (rv reflect.Value, isnil bool) {
- var ui = (*unsafeIntf)(unsafe.Pointer(&v))
- if ui.word == nil {
- isnil = true
- return
- }
- rv = rv4i(v) // reflect.value is cheap and inline'able
- tk := rv.Kind()
- isnil = (tk == reflect.Interface || tk == reflect.Slice) && *(*unsafe.Pointer)(ui.word) == nil
- return
- }
- func rv2ptr(urv *unsafeReflectValue) (ptr unsafe.Pointer) {
- // true references (map, func, chan, ptr - NOT slice) may be double-referenced? as flagIndir
- if refBitset.isset(byte(urv.flag&unsafeFlagKindMask)) && urv.flag&unsafeFlagIndir != 0 {
- ptr = *(*unsafe.Pointer)(urv.ptr)
- } else {
- ptr = urv.ptr
- }
- return
- }
- func rv4i(i interface{}) (rv reflect.Value) {
- // Unfortunately, we cannot get the "kind" of the interface directly here.
- // We need the 'rtype', whose structure changes in different go versions.
- // Finally, it's not clear that there is benefit to reimplementing it,
- // as the "escapes(i)" is not clearly expensive since we want i to exist on the heap.
- return reflect.ValueOf(i)
- }
- func rv2i(rv reflect.Value) interface{} {
- // We tap into implememtation details from
- // the source go stdlib reflect/value.go, and trims the implementation.
- urv := (*unsafeReflectValue)(unsafe.Pointer(&rv))
- return *(*interface{})(unsafe.Pointer(&unsafeIntf{typ: urv.typ, word: rv2ptr(urv)}))
- }
- func rvIsNil(rv reflect.Value) bool {
- urv := (*unsafeReflectValue)(unsafe.Pointer(&rv))
- if urv.flag&unsafeFlagIndir != 0 {
- return *(*unsafe.Pointer)(urv.ptr) == nil
- }
- return urv.ptr == nil
- }
- func rvSetSliceLen(rv reflect.Value, length int) {
- urv := (*unsafeReflectValue)(unsafe.Pointer(&rv))
- (*unsafeString)(urv.ptr).Len = length
- }
- func rvZeroAddrK(t reflect.Type, k reflect.Kind) (rv reflect.Value) {
- urv := (*unsafeReflectValue)(unsafe.Pointer(&rv))
- urv.flag = uintptr(k) | unsafeFlagIndir | unsafeFlagAddr
- urv.typ = ((*unsafeIntf)(unsafe.Pointer(&t))).word
- urv.ptr = unsafe_New(urv.typ)
- return
- }
- func rvConvert(v reflect.Value, t reflect.Type) (rv reflect.Value) {
- uv := (*unsafeReflectValue)(unsafe.Pointer(&v))
- urv := (*unsafeReflectValue)(unsafe.Pointer(&rv))
- *urv = *uv
- urv.typ = ((*unsafeIntf)(unsafe.Pointer(&t))).word
- return
- }
- func rt2id(rt reflect.Type) uintptr {
- return uintptr(((*unsafeIntf)(unsafe.Pointer(&rt))).word)
- }
- func i2rtid(i interface{}) uintptr {
- return uintptr(((*unsafeIntf)(unsafe.Pointer(&i))).typ)
- }
- // --------------------------
- func isEmptyValue(v reflect.Value, tinfos *TypeInfos, deref, checkStruct bool) bool {
- urv := (*unsafeReflectValue)(unsafe.Pointer(&v))
- if urv.flag == 0 {
- return true
- }
- switch v.Kind() {
- case reflect.Invalid:
- return true
- case reflect.String:
- return (*unsafeString)(urv.ptr).Len == 0
- case reflect.Slice:
- return (*unsafeSlice)(urv.ptr).Len == 0
- case reflect.Bool:
- return !*(*bool)(urv.ptr)
- case reflect.Int:
- return *(*int)(urv.ptr) == 0
- case reflect.Int8:
- return *(*int8)(urv.ptr) == 0
- case reflect.Int16:
- return *(*int16)(urv.ptr) == 0
- case reflect.Int32:
- return *(*int32)(urv.ptr) == 0
- case reflect.Int64:
- return *(*int64)(urv.ptr) == 0
- case reflect.Uint:
- return *(*uint)(urv.ptr) == 0
- case reflect.Uint8:
- return *(*uint8)(urv.ptr) == 0
- case reflect.Uint16:
- return *(*uint16)(urv.ptr) == 0
- case reflect.Uint32:
- return *(*uint32)(urv.ptr) == 0
- case reflect.Uint64:
- return *(*uint64)(urv.ptr) == 0
- case reflect.Uintptr:
- return *(*uintptr)(urv.ptr) == 0
- case reflect.Float32:
- return *(*float32)(urv.ptr) == 0
- case reflect.Float64:
- return *(*float64)(urv.ptr) == 0
- case reflect.Interface:
- isnil := urv.ptr == nil || *(*unsafe.Pointer)(urv.ptr) == nil
- if deref {
- if isnil {
- return true
- }
- return isEmptyValue(v.Elem(), tinfos, deref, checkStruct)
- }
- return isnil
- case reflect.Ptr:
- // isnil := urv.ptr == nil // (not sufficient, as a pointer value encodes the type)
- isnil := urv.ptr == nil || *(*unsafe.Pointer)(urv.ptr) == nil
- if deref {
- if isnil {
- return true
- }
- return isEmptyValue(v.Elem(), tinfos, deref, checkStruct)
- }
- return isnil
- case reflect.Struct:
- return isEmptyStruct(v, tinfos, deref, checkStruct)
- case reflect.Map, reflect.Array, reflect.Chan:
- return v.Len() == 0
- }
- return false
- }
- // --------------------------
- // atomicXXX is expected to be 2 words (for symmetry with atomic.Value)
- //
- // Note that we do not atomically load/store length and data pointer separately,
- // as this could lead to some races. Instead, we atomically load/store cappedSlice.
- //
- // Note: with atomic.(Load|Store)Pointer, we MUST work with an unsafe.Pointer directly.
- // ----------------------
- type atomicTypeInfoSlice struct {
- v unsafe.Pointer // *[]rtid2ti
- _ uint64 // padding (atomicXXX expected to be 2 words)
- }
- func (x *atomicTypeInfoSlice) load() (s []rtid2ti) {
- x2 := atomic.LoadPointer(&x.v)
- if x2 != nil {
- s = *(*[]rtid2ti)(x2)
- }
- return
- }
- func (x *atomicTypeInfoSlice) store(p []rtid2ti) {
- atomic.StorePointer(&x.v, unsafe.Pointer(&p))
- }
- // --------------------------
- type atomicRtidFnSlice struct {
- v unsafe.Pointer // *[]codecRtidFn
- _ uint64 // padding (atomicXXX expected to be 2 words) (make 1 word so JsonHandle fits)
- }
- func (x *atomicRtidFnSlice) load() (s []codecRtidFn) {
- x2 := atomic.LoadPointer(&x.v)
- if x2 != nil {
- s = *(*[]codecRtidFn)(x2)
- }
- return
- }
- func (x *atomicRtidFnSlice) store(p []codecRtidFn) {
- atomic.StorePointer(&x.v, unsafe.Pointer(&p))
- }
- // --------------------------
- type atomicClsErr struct {
- v unsafe.Pointer // *clsErr
- _ uint64 // padding (atomicXXX expected to be 2 words)
- }
- func (x *atomicClsErr) load() (e clsErr) {
- x2 := (*clsErr)(atomic.LoadPointer(&x.v))
- if x2 != nil {
- e = *x2
- }
- return
- }
- func (x *atomicClsErr) store(p clsErr) {
- atomic.StorePointer(&x.v, unsafe.Pointer(&p))
- }
- // --------------------------
- // to create a reflect.Value for each member field of fauxUnion,
- // we first create a global fauxUnion, and create reflect.Value
- // for them all.
- // This way, we have the flags and type in the reflect.Value.
- // Then, when a reflect.Value is called, we just copy it,
- // update the ptr to the fauxUnion's, and return it.
- type unsafeDecNakedWrapper struct {
- fauxUnion
- ru, ri, rf, rl, rs, rb, rt reflect.Value // mapping to the primitives above
- }
- func (n *unsafeDecNakedWrapper) init() {
- n.ru = rv4i(&n.u).Elem()
- n.ri = rv4i(&n.i).Elem()
- n.rf = rv4i(&n.f).Elem()
- n.rl = rv4i(&n.l).Elem()
- n.rs = rv4i(&n.s).Elem()
- n.rt = rv4i(&n.t).Elem()
- n.rb = rv4i(&n.b).Elem()
- // n.rr[] = rv4i(&n.)
- }
- var defUnsafeDecNakedWrapper unsafeDecNakedWrapper
- func init() {
- defUnsafeDecNakedWrapper.init()
- }
- func (n *fauxUnion) ru() (v reflect.Value) {
- v = defUnsafeDecNakedWrapper.ru
- ((*unsafeReflectValue)(unsafe.Pointer(&v))).ptr = unsafe.Pointer(&n.u)
- return
- }
- func (n *fauxUnion) ri() (v reflect.Value) {
- v = defUnsafeDecNakedWrapper.ri
- ((*unsafeReflectValue)(unsafe.Pointer(&v))).ptr = unsafe.Pointer(&n.i)
- return
- }
- func (n *fauxUnion) rf() (v reflect.Value) {
- v = defUnsafeDecNakedWrapper.rf
- ((*unsafeReflectValue)(unsafe.Pointer(&v))).ptr = unsafe.Pointer(&n.f)
- return
- }
- func (n *fauxUnion) rl() (v reflect.Value) {
- v = defUnsafeDecNakedWrapper.rl
- ((*unsafeReflectValue)(unsafe.Pointer(&v))).ptr = unsafe.Pointer(&n.l)
- return
- }
- func (n *fauxUnion) rs() (v reflect.Value) {
- v = defUnsafeDecNakedWrapper.rs
- ((*unsafeReflectValue)(unsafe.Pointer(&v))).ptr = unsafe.Pointer(&n.s)
- return
- }
- func (n *fauxUnion) rt() (v reflect.Value) {
- v = defUnsafeDecNakedWrapper.rt
- ((*unsafeReflectValue)(unsafe.Pointer(&v))).ptr = unsafe.Pointer(&n.t)
- return
- }
- func (n *fauxUnion) rb() (v reflect.Value) {
- v = defUnsafeDecNakedWrapper.rb
- ((*unsafeReflectValue)(unsafe.Pointer(&v))).ptr = unsafe.Pointer(&n.b)
- return
- }
- // --------------------------
- func rvSetBytes(rv reflect.Value, v []byte) {
- urv := (*unsafeReflectValue)(unsafe.Pointer(&rv))
- *(*[]byte)(urv.ptr) = v
- }
- func rvSetString(rv reflect.Value, v string) {
- urv := (*unsafeReflectValue)(unsafe.Pointer(&rv))
- *(*string)(urv.ptr) = v
- }
- func rvSetBool(rv reflect.Value, v bool) {
- urv := (*unsafeReflectValue)(unsafe.Pointer(&rv))
- *(*bool)(urv.ptr) = v
- }
- func rvSetTime(rv reflect.Value, v time.Time) {
- urv := (*unsafeReflectValue)(unsafe.Pointer(&rv))
- *(*time.Time)(urv.ptr) = v
- }
- func rvSetFloat32(rv reflect.Value, v float32) {
- urv := (*unsafeReflectValue)(unsafe.Pointer(&rv))
- *(*float32)(urv.ptr) = v
- }
- func rvSetFloat64(rv reflect.Value, v float64) {
- urv := (*unsafeReflectValue)(unsafe.Pointer(&rv))
- *(*float64)(urv.ptr) = v
- }
- func rvSetInt(rv reflect.Value, v int) {
- urv := (*unsafeReflectValue)(unsafe.Pointer(&rv))
- *(*int)(urv.ptr) = v
- }
- func rvSetInt8(rv reflect.Value, v int8) {
- urv := (*unsafeReflectValue)(unsafe.Pointer(&rv))
- *(*int8)(urv.ptr) = v
- }
- func rvSetInt16(rv reflect.Value, v int16) {
- urv := (*unsafeReflectValue)(unsafe.Pointer(&rv))
- *(*int16)(urv.ptr) = v
- }
- func rvSetInt32(rv reflect.Value, v int32) {
- urv := (*unsafeReflectValue)(unsafe.Pointer(&rv))
- *(*int32)(urv.ptr) = v
- }
- func rvSetInt64(rv reflect.Value, v int64) {
- urv := (*unsafeReflectValue)(unsafe.Pointer(&rv))
- *(*int64)(urv.ptr) = v
- }
- func rvSetUint(rv reflect.Value, v uint) {
- urv := (*unsafeReflectValue)(unsafe.Pointer(&rv))
- *(*uint)(urv.ptr) = v
- }
- func rvSetUintptr(rv reflect.Value, v uintptr) {
- urv := (*unsafeReflectValue)(unsafe.Pointer(&rv))
- *(*uintptr)(urv.ptr) = v
- }
- func rvSetUint8(rv reflect.Value, v uint8) {
- urv := (*unsafeReflectValue)(unsafe.Pointer(&rv))
- *(*uint8)(urv.ptr) = v
- }
- func rvSetUint16(rv reflect.Value, v uint16) {
- urv := (*unsafeReflectValue)(unsafe.Pointer(&rv))
- *(*uint16)(urv.ptr) = v
- }
- func rvSetUint32(rv reflect.Value, v uint32) {
- urv := (*unsafeReflectValue)(unsafe.Pointer(&rv))
- *(*uint32)(urv.ptr) = v
- }
- func rvSetUint64(rv reflect.Value, v uint64) {
- urv := (*unsafeReflectValue)(unsafe.Pointer(&rv))
- *(*uint64)(urv.ptr) = v
- }
- // ----------------
- // rvSetDirect is rv.Set for all kinds except reflect.Interface
- func rvSetDirect(rv reflect.Value, v reflect.Value) {
- urv := (*unsafeReflectValue)(unsafe.Pointer(&rv))
- uv := (*unsafeReflectValue)(unsafe.Pointer(&v))
- if uv.flag&unsafeFlagIndir == 0 {
- *(*unsafe.Pointer)(urv.ptr) = uv.ptr
- } else {
- typedmemmove(urv.typ, urv.ptr, uv.ptr)
- }
- }
- // rvSlice returns a slice of the slice of lenth
- func rvSlice(rv reflect.Value, length int) (v reflect.Value) {
- urv := (*unsafeReflectValue)(unsafe.Pointer(&rv))
- uv := (*unsafeReflectValue)(unsafe.Pointer(&v))
- *uv = *urv
- var x []unsafe.Pointer
- uv.ptr = unsafe.Pointer(&x)
- *(*unsafeSlice)(uv.ptr) = *(*unsafeSlice)(urv.ptr)
- (*unsafeSlice)(uv.ptr).Len = length
- return
- }
- // ------------
- func rvSliceIndex(rv reflect.Value, i int, ti *typeInfo) (v reflect.Value) {
- urv := (*unsafeReflectValue)(unsafe.Pointer(&rv))
- uv := (*unsafeReflectValue)(unsafe.Pointer(&v))
- uv.ptr = unsafe.Pointer(uintptr(((*unsafeSlice)(urv.ptr)).Data) + (ti.elemsize * uintptr(i)))
- uv.typ = ((*unsafeIntf)(unsafe.Pointer(&ti.elem))).word
- uv.flag = uintptr(ti.elemkind) | unsafeFlagIndir | unsafeFlagAddr
- return
- }
- func rvGetSliceLen(rv reflect.Value) int {
- urv := (*unsafeReflectValue)(unsafe.Pointer(&rv))
- return (*unsafeSlice)(urv.ptr).Len
- }
- func rvGetSliceCap(rv reflect.Value) int {
- urv := (*unsafeReflectValue)(unsafe.Pointer(&rv))
- return (*unsafeSlice)(urv.ptr).Cap
- }
- func rvGetArrayBytesRO(rv reflect.Value, scratch []byte) (bs []byte) {
- l := rv.Len()
- urv := (*unsafeReflectValue)(unsafe.Pointer(&rv))
- bx := (*unsafeSlice)(unsafe.Pointer(&bs))
- bx.Data = urv.ptr
- bx.Len, bx.Cap = l, l
- return
- }
- func rvGetArray4Slice(rv reflect.Value) (v reflect.Value) {
- // It is possible that this slice is based off an array with a larger
- // len that we want (where array len == slice cap).
- // However, it is ok to create an array type that is a subset of the full
- // e.g. full slice is based off a *[16]byte, but we can create a *[4]byte
- // off of it. That is ok.
- //
- // Consequently, we use rvGetSliceLen, not rvGetSliceCap.
- t := reflectArrayOf(rvGetSliceLen(rv), rv.Type().Elem())
- // v = rvZeroAddrK(t, reflect.Array)
- uv := (*unsafeReflectValue)(unsafe.Pointer(&v))
- uv.flag = uintptr(reflect.Array) | unsafeFlagIndir | unsafeFlagAddr
- uv.typ = ((*unsafeIntf)(unsafe.Pointer(&t))).word
- urv := (*unsafeReflectValue)(unsafe.Pointer(&rv))
- uv.ptr = *(*unsafe.Pointer)(urv.ptr) // slice rv has a ptr to the slice.
- return
- }
- func rvGetSlice4Array(rv reflect.Value, tslice reflect.Type) (v reflect.Value) {
- uv := (*unsafeReflectValue)(unsafe.Pointer(&v))
- var x []unsafe.Pointer
- uv.ptr = unsafe.Pointer(&x)
- uv.typ = ((*unsafeIntf)(unsafe.Pointer(&tslice))).word
- uv.flag = unsafeFlagIndir | uintptr(reflect.Slice)
- s := (*unsafeSlice)(uv.ptr)
- s.Data = ((*unsafeReflectValue)(unsafe.Pointer(&rv))).ptr
- s.Len = rv.Len()
- s.Cap = s.Len
- return
- }
- func rvCopySlice(dest, src reflect.Value) {
- t := dest.Type().Elem()
- urv := (*unsafeReflectValue)(unsafe.Pointer(&dest))
- destPtr := urv.ptr
- urv = (*unsafeReflectValue)(unsafe.Pointer(&src))
- typedslicecopy((*unsafeIntf)(unsafe.Pointer(&t)).word,
- *(*unsafeSlice)(destPtr), *(*unsafeSlice)(urv.ptr))
- }
- // ------------
- func rvGetBool(rv reflect.Value) bool {
- v := (*unsafeReflectValue)(unsafe.Pointer(&rv))
- return *(*bool)(v.ptr)
- }
- func rvGetBytes(rv reflect.Value) []byte {
- v := (*unsafeReflectValue)(unsafe.Pointer(&rv))
- return *(*[]byte)(v.ptr)
- }
- func rvGetTime(rv reflect.Value) time.Time {
- v := (*unsafeReflectValue)(unsafe.Pointer(&rv))
- return *(*time.Time)(v.ptr)
- }
- func rvGetString(rv reflect.Value) string {
- v := (*unsafeReflectValue)(unsafe.Pointer(&rv))
- return *(*string)(v.ptr)
- }
- func rvGetFloat64(rv reflect.Value) float64 {
- v := (*unsafeReflectValue)(unsafe.Pointer(&rv))
- return *(*float64)(v.ptr)
- }
- func rvGetFloat32(rv reflect.Value) float32 {
- v := (*unsafeReflectValue)(unsafe.Pointer(&rv))
- return *(*float32)(v.ptr)
- }
- func rvGetInt(rv reflect.Value) int {
- v := (*unsafeReflectValue)(unsafe.Pointer(&rv))
- return *(*int)(v.ptr)
- }
- func rvGetInt8(rv reflect.Value) int8 {
- v := (*unsafeReflectValue)(unsafe.Pointer(&rv))
- return *(*int8)(v.ptr)
- }
- func rvGetInt16(rv reflect.Value) int16 {
- v := (*unsafeReflectValue)(unsafe.Pointer(&rv))
- return *(*int16)(v.ptr)
- }
- func rvGetInt32(rv reflect.Value) int32 {
- v := (*unsafeReflectValue)(unsafe.Pointer(&rv))
- return *(*int32)(v.ptr)
- }
- func rvGetInt64(rv reflect.Value) int64 {
- v := (*unsafeReflectValue)(unsafe.Pointer(&rv))
- return *(*int64)(v.ptr)
- }
- func rvGetUint(rv reflect.Value) uint {
- v := (*unsafeReflectValue)(unsafe.Pointer(&rv))
- return *(*uint)(v.ptr)
- }
- func rvGetUint8(rv reflect.Value) uint8 {
- v := (*unsafeReflectValue)(unsafe.Pointer(&rv))
- return *(*uint8)(v.ptr)
- }
- func rvGetUint16(rv reflect.Value) uint16 {
- v := (*unsafeReflectValue)(unsafe.Pointer(&rv))
- return *(*uint16)(v.ptr)
- }
- func rvGetUint32(rv reflect.Value) uint32 {
- v := (*unsafeReflectValue)(unsafe.Pointer(&rv))
- return *(*uint32)(v.ptr)
- }
- func rvGetUint64(rv reflect.Value) uint64 {
- v := (*unsafeReflectValue)(unsafe.Pointer(&rv))
- return *(*uint64)(v.ptr)
- }
- func rvGetUintptr(rv reflect.Value) uintptr {
- v := (*unsafeReflectValue)(unsafe.Pointer(&rv))
- return *(*uintptr)(v.ptr)
- }
- // ------------ map range and map indexing ----------
- // regular calls to map via reflection: MapKeys, MapIndex, MapRange/MapIter etc
- // will always allocate for each map key or value.
- //
- // It is more performant to provide a value that the map entry is set into,
- // and that elides the allocation.
- // unsafeMapHashIter
- //
- // go 1.4+ has runtime/hashmap.go or runtime/map.go which has a
- // hIter struct with the first 2 values being key and value
- // of the current iteration.
- //
- // This *hIter is passed to mapiterinit, mapiternext, mapiterkey, mapiterelem.
- // We bypass the reflect wrapper functions and just use the *hIter directly.
- //
- // Though *hIter has many fields, we only care about the first 2.
- type unsafeMapHashIter struct {
- key, value unsafe.Pointer
- // other fields are ignored
- }
- type mapIter struct {
- unsafeMapIter
- }
- type unsafeMapIter struct {
- it *unsafeMapHashIter
- // k, v reflect.Value
- mtyp, ktyp, vtyp unsafe.Pointer
- mptr, kptr, vptr unsafe.Pointer
- kisref, visref bool
- mapvalues bool
- done bool
- started bool
- // _ [2]uint64 // padding (cache-aligned)
- }
- func (t *unsafeMapIter) ValidKV() (r bool) {
- return false
- }
- func (t *unsafeMapIter) Next() (r bool) {
- if t == nil || t.done {
- return
- }
- if t.started {
- mapiternext((unsafe.Pointer)(t.it))
- } else {
- t.started = true
- }
- t.done = t.it.key == nil
- if t.done {
- return
- }
- unsafeMapSet(t.kptr, t.ktyp, t.it.key, t.kisref)
- if t.mapvalues {
- unsafeMapSet(t.vptr, t.vtyp, t.it.value, t.visref)
- }
- return true
- }
- func (t *unsafeMapIter) Key() (r reflect.Value) {
- return
- }
- func (t *unsafeMapIter) Value() (r reflect.Value) {
- return
- }
- func (t *unsafeMapIter) Done() {
- }
- func unsafeMapSet(p, ptyp, p2 unsafe.Pointer, isref bool) {
- if isref {
- *(*unsafe.Pointer)(p) = *(*unsafe.Pointer)(p2) // p2
- } else {
- typedmemmove(ptyp, p, p2) // *(*unsafe.Pointer)(p2)) // p2)
- }
- }
- func unsafeMapKVPtr(urv *unsafeReflectValue) unsafe.Pointer {
- if urv.flag&unsafeFlagIndir == 0 {
- return unsafe.Pointer(&urv.ptr)
- }
- return urv.ptr
- }
- func mapRange(t *mapIter, m, k, v reflect.Value, mapvalues bool) {
- if rvIsNil(m) {
- t.done = true
- return
- }
- t.done = false
- t.started = false
- t.mapvalues = mapvalues
- var urv *unsafeReflectValue
- urv = (*unsafeReflectValue)(unsafe.Pointer(&m))
- t.mtyp = urv.typ
- t.mptr = rv2ptr(urv)
- t.it = (*unsafeMapHashIter)(mapiterinit(t.mtyp, t.mptr))
- urv = (*unsafeReflectValue)(unsafe.Pointer(&k))
- t.ktyp = urv.typ
- t.kptr = urv.ptr
- t.kisref = refBitset.isset(byte(k.Kind()))
- if mapvalues {
- urv = (*unsafeReflectValue)(unsafe.Pointer(&v))
- t.vtyp = urv.typ
- t.vptr = urv.ptr
- t.visref = refBitset.isset(byte(v.Kind()))
- } else {
- t.vtyp = nil
- t.vptr = nil
- }
- }
- func mapGet(m, k, v reflect.Value) (vv reflect.Value) {
- var urv = (*unsafeReflectValue)(unsafe.Pointer(&k))
- var kptr = unsafeMapKVPtr(urv)
- urv = (*unsafeReflectValue)(unsafe.Pointer(&m))
- vvptr := mapaccess(urv.typ, rv2ptr(urv), kptr)
- if vvptr == nil {
- return
- }
- // vvptr = *(*unsafe.Pointer)(vvptr)
- urv = (*unsafeReflectValue)(unsafe.Pointer(&v))
- unsafeMapSet(urv.ptr, urv.typ, vvptr, refBitset.isset(byte(v.Kind())))
- return v
- }
- func mapSet(m, k, v reflect.Value) {
- var urv = (*unsafeReflectValue)(unsafe.Pointer(&k))
- var kptr = unsafeMapKVPtr(urv)
- urv = (*unsafeReflectValue)(unsafe.Pointer(&v))
- var vptr = unsafeMapKVPtr(urv)
- urv = (*unsafeReflectValue)(unsafe.Pointer(&m))
- mapassign(urv.typ, rv2ptr(urv), kptr, vptr)
- }
- // func mapDelete(m, k reflect.Value) {
- // var urv = (*unsafeReflectValue)(unsafe.Pointer(&k))
- // var kptr = unsafeMapKVPtr(urv)
- // urv = (*unsafeReflectValue)(unsafe.Pointer(&m))
- // mapdelete(urv.typ, rv2ptr(urv), kptr)
- // }
- // return an addressable reflect value that can be used in mapRange and mapGet operations.
- //
- // all calls to mapGet or mapRange will call here to get an addressable reflect.Value.
- func mapAddressableRV(t reflect.Type, k reflect.Kind) (r reflect.Value) {
- // return reflect.New(t).Elem()
- return rvZeroAddrK(t, k)
- }
- //go:linkname mapiterinit reflect.mapiterinit
- //go:noescape
- func mapiterinit(typ unsafe.Pointer, it unsafe.Pointer) (key unsafe.Pointer)
- //go:linkname mapiternext reflect.mapiternext
- //go:noescape
- func mapiternext(it unsafe.Pointer) (key unsafe.Pointer)
- //go:linkname mapaccess reflect.mapaccess
- //go:noescape
- func mapaccess(typ unsafe.Pointer, m unsafe.Pointer, key unsafe.Pointer) (val unsafe.Pointer)
- //go:linkname mapassign reflect.mapassign
- //go:noescape
- func mapassign(typ unsafe.Pointer, m unsafe.Pointer, key, val unsafe.Pointer)
- //go:linkname mapdelete reflect.mapdelete
- //go:noescape
- func mapdelete(typ unsafe.Pointer, m unsafe.Pointer, key unsafe.Pointer)
- //go:linkname typedmemmove reflect.typedmemmove
- //go:noescape
- func typedmemmove(typ unsafe.Pointer, dst, src unsafe.Pointer)
- //go:linkname unsafe_New reflect.unsafe_New
- //go:noescape
- func unsafe_New(typ unsafe.Pointer) unsafe.Pointer
- //go:linkname typedslicecopy reflect.typedslicecopy
- //go:noescape
- func typedslicecopy(elemType unsafe.Pointer, dst, src unsafeSlice) int
- // ---------- ENCODER optimized ---------------
- func (e *Encoder) jsondriver() *jsonEncDriver {
- return (*jsonEncDriver)((*unsafeIntf)(unsafe.Pointer(&e.e)).word)
- }
- // ---------- DECODER optimized ---------------
- func (d *Decoder) checkBreak() bool {
- // jsonDecDriver.CheckBreak() CANNOT be inlined.
- // Consequently, there's no benefit in incurring the cost of this
- // wrapping function checkBreak.
- //
- // It is faster to just call the interface method directly.
- // if d.js {
- // return d.jsondriver().CheckBreak()
- // }
- // if d.cbor {
- // return d.cbordriver().CheckBreak()
- // }
- return d.d.CheckBreak()
- }
- func (d *Decoder) jsondriver() *jsonDecDriver {
- return (*jsonDecDriver)((*unsafeIntf)(unsafe.Pointer(&d.d)).word)
- }
- // func (d *Decoder) cbordriver() *cborDecDriver {
- // return (*cborDecDriver)((*unsafeIntf)(unsafe.Pointer(&d.d)).word)
- // }
|