decode.go 39 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544
  1. // Copyright (c) 2012-2015 Ugorji Nwoke. All rights reserved.
  2. // Use of this source code is governed by a MIT license found in the LICENSE file.
  3. package codec
  4. import (
  5. "encoding"
  6. "errors"
  7. "fmt"
  8. "io"
  9. "reflect"
  10. )
  11. // Some tagging information for error messages.
  12. const (
  13. msgBadDesc = "Unrecognized descriptor byte"
  14. msgDecCannotExpandArr = "cannot expand go array from %v to stream length: %v"
  15. )
  16. var (
  17. onlyMapOrArrayCanDecodeIntoStructErr = errors.New("only encoded map or array can be decoded into a struct")
  18. cannotDecodeIntoNilErr = errors.New("cannot decode into nil")
  19. )
  20. // decReader abstracts the reading source, allowing implementations that can
  21. // read from an io.Reader or directly off a byte slice with zero-copying.
  22. type decReader interface {
  23. unreadn1()
  24. // readx will use the implementation scratch buffer if possible i.e. n < len(scratchbuf), OR
  25. // just return a view of the []byte being decoded from.
  26. // Ensure you call detachZeroCopyBytes later if this needs to be sent outside codec control.
  27. readx(n int) []byte
  28. readb([]byte)
  29. readn1() uint8
  30. readn1eof() (v uint8, eof bool)
  31. numread() int // number of bytes read
  32. track()
  33. stopTrack() []byte
  34. }
  35. type decReaderByteScanner interface {
  36. io.Reader
  37. io.ByteScanner
  38. }
  39. type decDriver interface {
  40. // this will check if the next token is a break.
  41. CheckBreak() bool
  42. TryDecodeAsNil() bool
  43. // check if a container type: vt is one of: Bytes, String, Nil, Slice or Map.
  44. // if vt param == valueTypeNil, and nil is seen in stream, consume the nil.
  45. IsContainerType(vt valueType) bool
  46. IsBuiltinType(rt uintptr) bool
  47. DecodeBuiltin(rt uintptr, v interface{})
  48. //decodeNaked: Numbers are decoded as int64, uint64, float64 only (no smaller sized number types).
  49. //for extensions, decodeNaked must completely decode them as a *RawExt.
  50. //extensions should also use readx to decode them, for efficiency.
  51. //kInterface will extract the detached byte slice if it has to pass it outside its realm.
  52. DecodeNaked() (v interface{}, vt valueType, decodeFurther bool)
  53. DecodeInt(bitsize uint8) (i int64)
  54. DecodeUint(bitsize uint8) (ui uint64)
  55. DecodeFloat(chkOverflow32 bool) (f float64)
  56. DecodeBool() (b bool)
  57. // DecodeString can also decode symbols.
  58. // It looks redundant as DecodeBytes is available.
  59. // However, some codecs (e.g. binc) support symbols and can
  60. // return a pre-stored string value, meaning that it can bypass
  61. // the cost of []byte->string conversion.
  62. DecodeString() (s string)
  63. // DecodeBytes may be called directly, without going through reflection.
  64. // Consequently, it must be designed to handle possible nil.
  65. DecodeBytes(bs []byte, isstring, zerocopy bool) (bsOut []byte)
  66. // decodeExt will decode into a *RawExt or into an extension.
  67. DecodeExt(v interface{}, xtag uint64, ext Ext) (realxtag uint64)
  68. // decodeExt(verifyTag bool, tag byte) (xtag byte, xbs []byte)
  69. ReadMapStart() int
  70. ReadArrayStart() int
  71. // ReadEnd registers the end of a map or array.
  72. ReadEnd()
  73. }
  74. type decNoSeparator struct{}
  75. func (_ decNoSeparator) ReadEnd() {}
  76. type DecodeOptions struct {
  77. // MapType specifies type to use during schema-less decoding of a map in the stream.
  78. // If nil, we use map[interface{}]interface{}
  79. MapType reflect.Type
  80. // SliceType specifies type to use during schema-less decoding of an array in the stream.
  81. // If nil, we use []interface{}
  82. SliceType reflect.Type
  83. // If ErrorIfNoField, return an error when decoding a map
  84. // from a codec stream into a struct, and no matching struct field is found.
  85. ErrorIfNoField bool
  86. // If ErrorIfNoArrayExpand, return an error when decoding a slice/array that cannot be expanded.
  87. // For example, the stream contains an array of 8 items, but you are decoding into a [4]T array,
  88. // or you are decoding into a slice of length 4 which is non-addressable (and so cannot be set).
  89. ErrorIfNoArrayExpand bool
  90. // If SignedInteger, use the int64 during schema-less decoding of unsigned values (not uint64).
  91. SignedInteger bool
  92. }
  93. // ------------------------------------
  94. // ioDecByteScanner implements Read(), ReadByte(...), UnreadByte(...) methods
  95. // of io.Reader, io.ByteScanner.
  96. type ioDecByteScanner struct {
  97. r io.Reader
  98. l byte // last byte
  99. ls byte // last byte status. 0: init-canDoNothing, 1: canRead, 2: canUnread
  100. b [1]byte // tiny buffer for reading single bytes
  101. }
  102. func (z *ioDecByteScanner) Read(p []byte) (n int, err error) {
  103. var firstByte bool
  104. if z.ls == 1 {
  105. z.ls = 2
  106. p[0] = z.l
  107. if len(p) == 1 {
  108. n = 1
  109. return
  110. }
  111. firstByte = true
  112. p = p[1:]
  113. }
  114. n, err = z.r.Read(p)
  115. if n > 0 {
  116. if err == io.EOF && n == len(p) {
  117. err = nil // read was successful, so postpone EOF (till next time)
  118. }
  119. z.l = p[n-1]
  120. z.ls = 2
  121. }
  122. if firstByte {
  123. n++
  124. }
  125. return
  126. }
  127. func (z *ioDecByteScanner) ReadByte() (c byte, err error) {
  128. n, err := z.Read(z.b[:])
  129. if n == 1 {
  130. c = z.b[0]
  131. if err == io.EOF {
  132. err = nil // read was successful, so postpone EOF (till next time)
  133. }
  134. }
  135. return
  136. }
  137. func (z *ioDecByteScanner) UnreadByte() (err error) {
  138. x := z.ls
  139. if x == 0 {
  140. err = errors.New("cannot unread - nothing has been read")
  141. } else if x == 1 {
  142. err = errors.New("cannot unread - last byte has not been read")
  143. } else if x == 2 {
  144. z.ls = 1
  145. }
  146. return
  147. }
  148. // ioDecReader is a decReader that reads off an io.Reader
  149. type ioDecReader struct {
  150. br decReaderByteScanner
  151. // temp byte array re-used internally for efficiency during read.
  152. // shares buffer with Decoder, so we keep size of struct within 8 words.
  153. x *[scratchByteArrayLen]byte
  154. bs ioDecByteScanner
  155. n int // num read
  156. tr []byte // tracking bytes read
  157. trb bool
  158. }
  159. func (z *ioDecReader) numread() int {
  160. return z.n
  161. }
  162. func (z *ioDecReader) readx(n int) (bs []byte) {
  163. if n <= 0 {
  164. return
  165. }
  166. if n < len(z.x) {
  167. bs = z.x[:n]
  168. } else {
  169. bs = make([]byte, n)
  170. }
  171. if _, err := io.ReadAtLeast(z.br, bs, n); err != nil {
  172. panic(err)
  173. }
  174. z.n += len(bs)
  175. if z.trb {
  176. z.tr = append(z.tr, bs...)
  177. }
  178. return
  179. }
  180. func (z *ioDecReader) readb(bs []byte) {
  181. if len(bs) == 0 {
  182. return
  183. }
  184. n, err := io.ReadAtLeast(z.br, bs, len(bs))
  185. z.n += n
  186. if err != nil {
  187. panic(err)
  188. }
  189. if z.trb {
  190. z.tr = append(z.tr, bs...)
  191. }
  192. }
  193. func (z *ioDecReader) readn1() (b uint8) {
  194. b, err := z.br.ReadByte()
  195. if err != nil {
  196. panic(err)
  197. }
  198. z.n++
  199. if z.trb {
  200. z.tr = append(z.tr, b)
  201. }
  202. return b
  203. }
  204. func (z *ioDecReader) readn1eof() (b uint8, eof bool) {
  205. b, err := z.br.ReadByte()
  206. if err == nil {
  207. z.n++
  208. if z.trb {
  209. z.tr = append(z.tr, b)
  210. }
  211. } else if err == io.EOF {
  212. eof = true
  213. } else {
  214. panic(err)
  215. }
  216. return
  217. }
  218. func (z *ioDecReader) unreadn1() {
  219. err := z.br.UnreadByte()
  220. if err != nil {
  221. panic(err)
  222. }
  223. z.n--
  224. if z.trb {
  225. if l := len(z.tr) - 1; l >= 0 {
  226. z.tr = z.tr[:l]
  227. }
  228. }
  229. }
  230. func (z *ioDecReader) track() {
  231. if z.tr != nil {
  232. z.tr = z.tr[:0]
  233. }
  234. z.trb = true
  235. }
  236. func (z *ioDecReader) stopTrack() (bs []byte) {
  237. z.trb = false
  238. return z.tr
  239. }
  240. // ------------------------------------
  241. var bytesDecReaderCannotUnreadErr = errors.New("cannot unread last byte read")
  242. // bytesDecReader is a decReader that reads off a byte slice with zero copying
  243. type bytesDecReader struct {
  244. b []byte // data
  245. c int // cursor
  246. a int // available
  247. t int // track start
  248. }
  249. func (z *bytesDecReader) numread() int {
  250. return z.c
  251. }
  252. func (z *bytesDecReader) unreadn1() {
  253. if z.c == 0 || len(z.b) == 0 {
  254. panic(bytesDecReaderCannotUnreadErr)
  255. }
  256. z.c--
  257. z.a++
  258. return
  259. }
  260. func (z *bytesDecReader) readx(n int) (bs []byte) {
  261. // slicing from a non-constant start position is more expensive,
  262. // as more computation is required to decipher the pointer start position.
  263. // However, we do it only once, and it's better than reslicing both z.b and return value.
  264. if n <= 0 {
  265. } else if z.a == 0 {
  266. panic(io.EOF)
  267. } else if n > z.a {
  268. panic(io.ErrUnexpectedEOF)
  269. } else {
  270. c0 := z.c
  271. z.c = c0 + n
  272. z.a = z.a - n
  273. bs = z.b[c0:z.c]
  274. }
  275. return
  276. }
  277. func (z *bytesDecReader) readn1() (v uint8) {
  278. if z.a == 0 {
  279. panic(io.EOF)
  280. }
  281. v = z.b[z.c]
  282. z.c++
  283. z.a--
  284. return
  285. }
  286. func (z *bytesDecReader) readn1eof() (v uint8, eof bool) {
  287. if z.a == 0 {
  288. eof = true
  289. return
  290. }
  291. v = z.b[z.c]
  292. z.c++
  293. z.a--
  294. return
  295. }
  296. func (z *bytesDecReader) readb(bs []byte) {
  297. copy(bs, z.readx(len(bs)))
  298. }
  299. func (z *bytesDecReader) track() {
  300. z.t = z.c
  301. }
  302. func (z *bytesDecReader) stopTrack() (bs []byte) {
  303. return z.b[z.t:z.c]
  304. }
  305. // ------------------------------------
  306. type decFnInfoX struct {
  307. d *Decoder
  308. ti *typeInfo
  309. xfFn Ext
  310. xfTag uint64
  311. seq seqType
  312. }
  313. // decFnInfo has methods for handling decoding of a specific type
  314. // based on some characteristics (builtin, extension, reflect Kind, etc)
  315. type decFnInfo struct {
  316. // use decFnInfo as a value receiver.
  317. // keep most of it less-used variables accessible via a pointer (*decFnInfoX).
  318. // As sweet spot for value-receiver is 3 words, keep everything except
  319. // decDriver (which everyone needs) directly accessible.
  320. // ensure decFnInfoX is set for everyone who needs it i.e.
  321. // rawExt, ext, builtin, (selfer|binary|text)Marshal, kSlice, kStruct, kMap, kInterface, fastpath
  322. dd decDriver
  323. *decFnInfoX
  324. }
  325. // ----------------------------------------
  326. type decFn struct {
  327. i decFnInfo
  328. f func(decFnInfo, reflect.Value)
  329. }
  330. func (f decFnInfo) builtin(rv reflect.Value) {
  331. f.dd.DecodeBuiltin(f.ti.rtid, rv.Addr().Interface())
  332. }
  333. func (f decFnInfo) rawExt(rv reflect.Value) {
  334. f.dd.DecodeExt(rv.Addr().Interface(), 0, nil)
  335. }
  336. func (f decFnInfo) ext(rv reflect.Value) {
  337. f.dd.DecodeExt(rv.Addr().Interface(), f.xfTag, f.xfFn)
  338. }
  339. func (f decFnInfo) getValueForUnmarshalInterface(rv reflect.Value, indir int8) (v interface{}) {
  340. if indir == -1 {
  341. v = rv.Addr().Interface()
  342. } else if indir == 0 {
  343. v = rv.Interface()
  344. } else {
  345. for j := int8(0); j < indir; j++ {
  346. if rv.IsNil() {
  347. rv.Set(reflect.New(rv.Type().Elem()))
  348. }
  349. rv = rv.Elem()
  350. }
  351. v = rv.Interface()
  352. }
  353. return
  354. }
  355. func (f decFnInfo) selferUnmarshal(rv reflect.Value) {
  356. f.getValueForUnmarshalInterface(rv, f.ti.csIndir).(Selfer).CodecDecodeSelf(f.d)
  357. }
  358. func (f decFnInfo) binaryUnmarshal(rv reflect.Value) {
  359. bm := f.getValueForUnmarshalInterface(rv, f.ti.bunmIndir).(encoding.BinaryUnmarshaler)
  360. xbs := f.dd.DecodeBytes(nil, false, true)
  361. if fnerr := bm.UnmarshalBinary(xbs); fnerr != nil {
  362. panic(fnerr)
  363. }
  364. }
  365. func (f decFnInfo) textUnmarshal(rv reflect.Value) {
  366. tm := f.getValueForUnmarshalInterface(rv, f.ti.tunmIndir).(encoding.TextUnmarshaler)
  367. fnerr := tm.UnmarshalText(f.dd.DecodeBytes(f.d.b[:], true, true))
  368. if fnerr != nil {
  369. panic(fnerr)
  370. }
  371. }
  372. func (f decFnInfo) jsonUnmarshal(rv reflect.Value) {
  373. tm := f.getValueForUnmarshalInterface(rv, f.ti.junmIndir).(jsonUnmarshaler)
  374. // bs := f.dd.DecodeBytes(f.d.b[:], true, true)
  375. // grab the bytes to be read, as UnmarshalJSON wants the full JSON to unmarshal it itself.
  376. f.d.r.track()
  377. f.d.swallow()
  378. bs := f.d.r.stopTrack()
  379. // fmt.Printf(">>>>>> REFLECTION JSON: %s\n", bs)
  380. fnerr := tm.UnmarshalJSON(bs)
  381. if fnerr != nil {
  382. panic(fnerr)
  383. }
  384. }
  385. func (f decFnInfo) kErr(rv reflect.Value) {
  386. f.d.errorf("no decoding function defined for kind %v", rv.Kind())
  387. }
  388. func (f decFnInfo) kString(rv reflect.Value) {
  389. rv.SetString(f.dd.DecodeString())
  390. }
  391. func (f decFnInfo) kBool(rv reflect.Value) {
  392. rv.SetBool(f.dd.DecodeBool())
  393. }
  394. func (f decFnInfo) kInt(rv reflect.Value) {
  395. rv.SetInt(f.dd.DecodeInt(intBitsize))
  396. }
  397. func (f decFnInfo) kInt64(rv reflect.Value) {
  398. rv.SetInt(f.dd.DecodeInt(64))
  399. }
  400. func (f decFnInfo) kInt32(rv reflect.Value) {
  401. rv.SetInt(f.dd.DecodeInt(32))
  402. }
  403. func (f decFnInfo) kInt8(rv reflect.Value) {
  404. rv.SetInt(f.dd.DecodeInt(8))
  405. }
  406. func (f decFnInfo) kInt16(rv reflect.Value) {
  407. rv.SetInt(f.dd.DecodeInt(16))
  408. }
  409. func (f decFnInfo) kFloat32(rv reflect.Value) {
  410. rv.SetFloat(f.dd.DecodeFloat(true))
  411. }
  412. func (f decFnInfo) kFloat64(rv reflect.Value) {
  413. rv.SetFloat(f.dd.DecodeFloat(false))
  414. }
  415. func (f decFnInfo) kUint8(rv reflect.Value) {
  416. rv.SetUint(f.dd.DecodeUint(8))
  417. }
  418. func (f decFnInfo) kUint64(rv reflect.Value) {
  419. rv.SetUint(f.dd.DecodeUint(64))
  420. }
  421. func (f decFnInfo) kUint(rv reflect.Value) {
  422. rv.SetUint(f.dd.DecodeUint(uintBitsize))
  423. }
  424. func (f decFnInfo) kUintptr(rv reflect.Value) {
  425. rv.SetUint(f.dd.DecodeUint(uintBitsize))
  426. }
  427. func (f decFnInfo) kUint32(rv reflect.Value) {
  428. rv.SetUint(f.dd.DecodeUint(32))
  429. }
  430. func (f decFnInfo) kUint16(rv reflect.Value) {
  431. rv.SetUint(f.dd.DecodeUint(16))
  432. }
  433. // func (f decFnInfo) kPtr(rv reflect.Value) {
  434. // debugf(">>>>>>> ??? decode kPtr called - shouldn't get called")
  435. // if rv.IsNil() {
  436. // rv.Set(reflect.New(rv.Type().Elem()))
  437. // }
  438. // f.d.decodeValue(rv.Elem())
  439. // }
  440. // var kIntfCtr uint64
  441. func (f decFnInfo) kInterfaceNaked() (rvn reflect.Value) {
  442. // nil interface:
  443. // use some hieristics to decode it appropriately
  444. // based on the detected next value in the stream.
  445. v, vt, decodeFurther := f.dd.DecodeNaked()
  446. if vt == valueTypeNil {
  447. return
  448. }
  449. // We cannot decode non-nil stream value into nil interface with methods (e.g. io.Reader).
  450. if num := f.ti.rt.NumMethod(); num > 0 {
  451. f.d.errorf("cannot decode non-nil codec value into nil %v (%v methods)", f.ti.rt, num)
  452. return
  453. }
  454. var useRvn bool
  455. switch vt {
  456. case valueTypeMap:
  457. if f.d.h.MapType == nil {
  458. var m2 map[interface{}]interface{}
  459. v = &m2
  460. } else {
  461. rvn = reflect.New(f.d.h.MapType).Elem()
  462. useRvn = true
  463. }
  464. case valueTypeArray:
  465. if f.d.h.SliceType == nil {
  466. var m2 []interface{}
  467. v = &m2
  468. } else {
  469. rvn = reflect.New(f.d.h.SliceType).Elem()
  470. useRvn = true
  471. }
  472. case valueTypeExt:
  473. re := v.(*RawExt)
  474. bfn := f.d.h.getExtForTag(re.Tag)
  475. if bfn == nil {
  476. re.Data = detachZeroCopyBytes(f.d.bytes, nil, re.Data)
  477. rvn = reflect.ValueOf(*re)
  478. } else {
  479. rvnA := reflect.New(bfn.rt)
  480. rvn = rvnA.Elem()
  481. if re.Data != nil {
  482. bfn.ext.ReadExt(rvnA.Interface(), re.Data)
  483. } else {
  484. bfn.ext.UpdateExt(rvnA.Interface(), re.Value)
  485. }
  486. }
  487. return
  488. }
  489. if decodeFurther {
  490. if useRvn {
  491. f.d.decodeValue(rvn, decFn{})
  492. } else if v != nil {
  493. // this v is a pointer, so we need to dereference it when done
  494. f.d.decode(v)
  495. rvn = reflect.ValueOf(v).Elem()
  496. useRvn = true
  497. }
  498. }
  499. if !useRvn && v != nil {
  500. rvn = reflect.ValueOf(v)
  501. }
  502. return
  503. }
  504. func (f decFnInfo) kInterface(rv reflect.Value) {
  505. // debugf("\t===> kInterface")
  506. // Note:
  507. // A consequence of how kInterface works, is that
  508. // if an interface already contains something, we try
  509. // to decode into what was there before.
  510. // We do not replace with a generic value (as got from decodeNaked).
  511. if rv.IsNil() {
  512. rvn := f.kInterfaceNaked()
  513. if rvn.IsValid() {
  514. rv.Set(rvn)
  515. }
  516. } else {
  517. rve := rv.Elem()
  518. // Note: interface{} is settable, but underlying type may not be.
  519. // Consequently, we have to set the reflect.Value directly.
  520. // if underlying type is settable (e.g. ptr or interface),
  521. // we just decode into it.
  522. // Else we create a settable value, decode into it, and set on the interface.
  523. if rve.CanSet() {
  524. f.d.decodeValue(rve, decFn{})
  525. } else {
  526. rve2 := reflect.New(rve.Type()).Elem()
  527. rve2.Set(rve)
  528. f.d.decodeValue(rve2, decFn{})
  529. rv.Set(rve2)
  530. }
  531. }
  532. }
  533. func (f decFnInfo) kStruct(rv reflect.Value) {
  534. fti := f.ti
  535. d := f.d
  536. if f.dd.IsContainerType(valueTypeMap) {
  537. containerLen := f.dd.ReadMapStart()
  538. if containerLen == 0 {
  539. f.dd.ReadEnd()
  540. return
  541. }
  542. tisfi := fti.sfi
  543. hasLen := containerLen >= 0
  544. if hasLen {
  545. for j := 0; j < containerLen; j++ {
  546. // rvkencname := f.dd.DecodeString()
  547. rvkencname := stringView(f.dd.DecodeBytes(f.d.b[:], true, true))
  548. // rvksi := ti.getForEncName(rvkencname)
  549. if k := fti.indexForEncName(rvkencname); k > -1 {
  550. si := tisfi[k]
  551. if f.dd.TryDecodeAsNil() {
  552. si.setToZeroValue(rv)
  553. } else {
  554. d.decodeValue(si.field(rv, true), decFn{})
  555. }
  556. } else {
  557. d.structFieldNotFound(-1, rvkencname)
  558. }
  559. }
  560. } else {
  561. for j := 0; !f.dd.CheckBreak(); j++ {
  562. // rvkencname := f.dd.DecodeString()
  563. rvkencname := stringView(f.dd.DecodeBytes(f.d.b[:], true, true))
  564. // rvksi := ti.getForEncName(rvkencname)
  565. if k := fti.indexForEncName(rvkencname); k > -1 {
  566. si := tisfi[k]
  567. if f.dd.TryDecodeAsNil() {
  568. si.setToZeroValue(rv)
  569. } else {
  570. d.decodeValue(si.field(rv, true), decFn{})
  571. }
  572. } else {
  573. d.structFieldNotFound(-1, rvkencname)
  574. }
  575. }
  576. f.dd.ReadEnd()
  577. }
  578. } else if f.dd.IsContainerType(valueTypeArray) {
  579. containerLen := f.dd.ReadArrayStart()
  580. if containerLen == 0 {
  581. f.dd.ReadEnd()
  582. return
  583. }
  584. // Not much gain from doing it two ways for array.
  585. // Arrays are not used as much for structs.
  586. hasLen := containerLen >= 0
  587. for j, si := range fti.sfip {
  588. if hasLen {
  589. if j == containerLen {
  590. break
  591. }
  592. } else if f.dd.CheckBreak() {
  593. break
  594. }
  595. if f.dd.TryDecodeAsNil() {
  596. si.setToZeroValue(rv)
  597. } else {
  598. d.decodeValue(si.field(rv, true), decFn{})
  599. }
  600. }
  601. if containerLen > len(fti.sfip) {
  602. // read remaining values and throw away
  603. for j := len(fti.sfip); j < containerLen; j++ {
  604. d.structFieldNotFound(j, "")
  605. }
  606. }
  607. f.dd.ReadEnd()
  608. } else {
  609. f.d.error(onlyMapOrArrayCanDecodeIntoStructErr)
  610. return
  611. }
  612. }
  613. func (f decFnInfo) kSlice(rv reflect.Value) {
  614. // A slice can be set from a map or array in stream.
  615. // This way, the order can be kept (as order is lost with map).
  616. ti := f.ti
  617. d := f.d
  618. if f.dd.IsContainerType(valueTypeBytes) || f.dd.IsContainerType(valueTypeString) {
  619. if ti.rtid == uint8SliceTypId || ti.rt.Elem().Kind() == reflect.Uint8 {
  620. if f.seq == seqTypeChan {
  621. bs2 := f.dd.DecodeBytes(nil, false, true)
  622. ch := rv.Interface().(chan<- byte)
  623. for _, b := range bs2 {
  624. ch <- b
  625. }
  626. } else {
  627. rvbs := rv.Bytes()
  628. bs2 := f.dd.DecodeBytes(rvbs, false, false)
  629. if rvbs == nil && bs2 != nil || rvbs != nil && bs2 == nil || len(bs2) != len(rvbs) {
  630. if rv.CanSet() {
  631. rv.SetBytes(bs2)
  632. } else {
  633. copy(rvbs, bs2)
  634. }
  635. }
  636. }
  637. return
  638. }
  639. }
  640. // array := f.seq == seqTypeChan
  641. slh, containerLenS := d.decSliceHelperStart()
  642. // an array can never return a nil slice. so no need to check f.array here.
  643. if rv.IsNil() {
  644. // either chan or slice
  645. if f.seq == seqTypeSlice {
  646. if containerLenS <= 0 {
  647. rv.Set(reflect.MakeSlice(ti.rt, 0, 0))
  648. } else {
  649. rv.Set(reflect.MakeSlice(ti.rt, containerLenS, containerLenS))
  650. }
  651. } else if f.seq == seqTypeChan {
  652. if containerLenS <= 0 {
  653. rv.Set(reflect.MakeChan(ti.rt, 0))
  654. } else {
  655. rv.Set(reflect.MakeChan(ti.rt, containerLenS))
  656. }
  657. }
  658. }
  659. rvlen := rv.Len()
  660. if containerLenS == 0 {
  661. if f.seq == seqTypeSlice && rvlen != 0 {
  662. rv.SetLen(0)
  663. }
  664. // f.dd.ReadEnd()
  665. return
  666. }
  667. rtelem0 := ti.rt.Elem()
  668. rtelem := rtelem0
  669. for rtelem.Kind() == reflect.Ptr {
  670. rtelem = rtelem.Elem()
  671. }
  672. fn := d.getDecFn(rtelem, true, true)
  673. rv0 := rv
  674. rvChanged := false
  675. rvcap := rv.Cap()
  676. // for j := 0; j < containerLenS; j++ {
  677. hasLen := containerLenS >= 0
  678. if hasLen {
  679. if f.seq == seqTypeChan {
  680. // handle chan specially:
  681. for j := 0; j < containerLenS; j++ {
  682. rv0 := reflect.New(rtelem0).Elem()
  683. d.decodeValue(rv0, fn)
  684. rv.Send(rv0)
  685. }
  686. } else {
  687. numToRead := containerLenS
  688. if containerLenS > rvcap {
  689. if f.seq == seqTypeArray {
  690. d.arrayCannotExpand(rv.Len(), containerLenS)
  691. numToRead = rvlen
  692. } else {
  693. rv = reflect.MakeSlice(ti.rt, containerLenS, containerLenS)
  694. if rvlen > 0 && !isMutableKind(ti.rt.Kind()) {
  695. rv1 := rv0
  696. rv1.SetLen(rvcap)
  697. reflect.Copy(rv, rv1)
  698. }
  699. rvChanged = true
  700. rvlen = containerLenS
  701. }
  702. } else if containerLenS != rvlen {
  703. if f.seq == seqTypeSlice {
  704. rv.SetLen(containerLenS)
  705. rvlen = containerLenS
  706. }
  707. }
  708. j := 0
  709. for ; j < numToRead; j++ {
  710. d.decodeValue(rv.Index(j), fn)
  711. }
  712. if f.seq == seqTypeArray {
  713. for ; j < containerLenS; j++ {
  714. d.swallow()
  715. }
  716. }
  717. }
  718. } else {
  719. for j := 0; !f.dd.CheckBreak(); j++ {
  720. var decodeIntoBlank bool
  721. // if indefinite, etc, then expand the slice if necessary
  722. if j >= rvlen {
  723. if f.seq == seqTypeArray {
  724. d.arrayCannotExpand(rvlen, j+1)
  725. decodeIntoBlank = true
  726. } else if f.seq == seqTypeSlice {
  727. rv = reflect.Append(rv, reflect.Zero(rtelem0))
  728. rvlen++
  729. rvChanged = true
  730. }
  731. }
  732. if f.seq == seqTypeChan {
  733. rv0 := reflect.New(rtelem0).Elem()
  734. d.decodeValue(rv0, fn)
  735. rv.Send(rv0)
  736. } else if decodeIntoBlank {
  737. d.swallow()
  738. } else {
  739. d.decodeValue(rv.Index(j), fn)
  740. }
  741. }
  742. slh.End()
  743. }
  744. if rvChanged {
  745. rv0.Set(rv)
  746. }
  747. }
  748. func (f decFnInfo) kArray(rv reflect.Value) {
  749. // f.d.decodeValue(rv.Slice(0, rv.Len()))
  750. f.kSlice(rv.Slice(0, rv.Len()))
  751. }
  752. func (f decFnInfo) kMap(rv reflect.Value) {
  753. containerLen := f.dd.ReadMapStart()
  754. ti := f.ti
  755. if rv.IsNil() {
  756. rv.Set(reflect.MakeMap(ti.rt))
  757. }
  758. if containerLen == 0 {
  759. // It is not length-prefix style container. They have no End marker.
  760. // f.dd.ReadMapEnd()
  761. return
  762. }
  763. d := f.d
  764. ktype, vtype := ti.rt.Key(), ti.rt.Elem()
  765. ktypeId := reflect.ValueOf(ktype).Pointer()
  766. var keyFn, valFn decFn
  767. var xtyp reflect.Type
  768. for xtyp = ktype; xtyp.Kind() == reflect.Ptr; xtyp = xtyp.Elem() {
  769. }
  770. keyFn = d.getDecFn(xtyp, true, true)
  771. for xtyp = vtype; xtyp.Kind() == reflect.Ptr; xtyp = xtyp.Elem() {
  772. }
  773. valFn = d.getDecFn(xtyp, true, true)
  774. // for j := 0; j < containerLen; j++ {
  775. if containerLen > 0 {
  776. for j := 0; j < containerLen; j++ {
  777. rvk := reflect.New(ktype).Elem()
  778. d.decodeValue(rvk, keyFn)
  779. // special case if a byte array.
  780. if ktypeId == intfTypId {
  781. rvk = rvk.Elem()
  782. if rvk.Type() == uint8SliceTyp {
  783. rvk = reflect.ValueOf(string(rvk.Bytes()))
  784. }
  785. }
  786. rvv := rv.MapIndex(rvk)
  787. // TODO: is !IsValid check required?
  788. if !rvv.IsValid() {
  789. rvv = reflect.New(vtype).Elem()
  790. }
  791. d.decodeValue(rvv, valFn)
  792. rv.SetMapIndex(rvk, rvv)
  793. }
  794. } else {
  795. for j := 0; !f.dd.CheckBreak(); j++ {
  796. rvk := reflect.New(ktype).Elem()
  797. d.decodeValue(rvk, keyFn)
  798. // special case if a byte array.
  799. if ktypeId == intfTypId {
  800. rvk = rvk.Elem()
  801. if rvk.Type() == uint8SliceTyp {
  802. rvk = reflect.ValueOf(string(rvk.Bytes()))
  803. }
  804. }
  805. rvv := rv.MapIndex(rvk)
  806. if !rvv.IsValid() {
  807. rvv = reflect.New(vtype).Elem()
  808. }
  809. d.decodeValue(rvv, valFn)
  810. rv.SetMapIndex(rvk, rvv)
  811. }
  812. f.dd.ReadEnd()
  813. }
  814. }
  815. type rtidDecFn struct {
  816. rtid uintptr
  817. fn decFn
  818. }
  819. // A Decoder reads and decodes an object from an input stream in the codec format.
  820. type Decoder struct {
  821. // hopefully, reduce derefencing cost by laying the decReader inside the Decoder.
  822. // Try to put things that go together to fit within a cache line (8 words).
  823. d decDriver
  824. // NOTE: Decoder shouldn't call it's read methods,
  825. // as the handler MAY need to do some coordination.
  826. r decReader
  827. //sa [32]rtidDecFn
  828. s []rtidDecFn
  829. h *BasicHandle
  830. rb bytesDecReader
  831. hh Handle
  832. be bool // is binary encoding
  833. bytes bool // is bytes reader
  834. js bool // is json handle
  835. ri ioDecReader
  836. f map[uintptr]decFn
  837. _ uintptr // for alignment purposes, so next one starts from a cache line
  838. b [scratchByteArrayLen]byte
  839. }
  840. // NewDecoder returns a Decoder for decoding a stream of bytes from an io.Reader.
  841. //
  842. // For efficiency, Users are encouraged to pass in a memory buffered reader
  843. // (eg bufio.Reader, bytes.Buffer).
  844. func NewDecoder(r io.Reader, h Handle) (d *Decoder) {
  845. d = &Decoder{hh: h, h: h.getBasicHandle(), be: h.isBinary()}
  846. //d.s = d.sa[:0]
  847. d.ri.x = &d.b
  848. d.ri.bs.r = r
  849. var ok bool
  850. d.ri.br, ok = r.(decReaderByteScanner)
  851. if !ok {
  852. d.ri.br = &d.ri.bs
  853. }
  854. d.r = &d.ri
  855. _, d.js = h.(*JsonHandle)
  856. d.d = h.newDecDriver(d)
  857. return
  858. }
  859. // NewDecoderBytes returns a Decoder which efficiently decodes directly
  860. // from a byte slice with zero copying.
  861. func NewDecoderBytes(in []byte, h Handle) (d *Decoder) {
  862. d = &Decoder{hh: h, h: h.getBasicHandle(), be: h.isBinary(), bytes: true}
  863. //d.s = d.sa[:0]
  864. d.rb.b = in
  865. d.rb.a = len(in)
  866. d.r = &d.rb
  867. _, d.js = h.(*JsonHandle)
  868. d.d = h.newDecDriver(d)
  869. // d.d = h.newDecDriver(decReaderT{true, &d.rb, &d.ri})
  870. return
  871. }
  872. // Decode decodes the stream from reader and stores the result in the
  873. // value pointed to by v. v cannot be a nil pointer. v can also be
  874. // a reflect.Value of a pointer.
  875. //
  876. // Note that a pointer to a nil interface is not a nil pointer.
  877. // If you do not know what type of stream it is, pass in a pointer to a nil interface.
  878. // We will decode and store a value in that nil interface.
  879. //
  880. // Sample usages:
  881. // // Decoding into a non-nil typed value
  882. // var f float32
  883. // err = codec.NewDecoder(r, handle).Decode(&f)
  884. //
  885. // // Decoding into nil interface
  886. // var v interface{}
  887. // dec := codec.NewDecoder(r, handle)
  888. // err = dec.Decode(&v)
  889. //
  890. // When decoding into a nil interface{}, we will decode into an appropriate value based
  891. // on the contents of the stream:
  892. // - Numbers are decoded as float64, int64 or uint64.
  893. // - Other values are decoded appropriately depending on the type:
  894. // bool, string, []byte, time.Time, etc
  895. // - Extensions are decoded as RawExt (if no ext function registered for the tag)
  896. // Configurations exist on the Handle to override defaults
  897. // (e.g. for MapType, SliceType and how to decode raw bytes).
  898. //
  899. // When decoding into a non-nil interface{} value, the mode of encoding is based on the
  900. // type of the value. When a value is seen:
  901. // - If an extension is registered for it, call that extension function
  902. // - If it implements BinaryUnmarshaler, call its UnmarshalBinary(data []byte) error
  903. // - Else decode it based on its reflect.Kind
  904. //
  905. // There are some special rules when decoding into containers (slice/array/map/struct).
  906. // Decode will typically use the stream contents to UPDATE the container.
  907. // - A map can be decoded from a stream map, by updating matching keys.
  908. // - A slice can be decoded from a stream array,
  909. // by updating the first n elements, where n is length of the stream.
  910. // - A slice can be decoded from a stream map, by decoding as if
  911. // it contains a sequence of key-value pairs.
  912. // - A struct can be decoded from a stream map, by updating matching fields.
  913. // - A struct can be decoded from a stream array,
  914. // by updating fields as they occur in the struct (by index).
  915. //
  916. // When decoding a stream map or array with length of 0 into a nil map or slice,
  917. // we reset the destination map or slice to a zero-length value.
  918. //
  919. // However, when decoding a stream nil, we reset the destination container
  920. // to its "zero" value (e.g. nil for slice/map, etc).
  921. //
  922. func (d *Decoder) Decode(v interface{}) (err error) {
  923. defer panicToErr(&err)
  924. d.decode(v)
  925. return
  926. }
  927. // this is not a smart swallow, as it allocates objects and does unnecessary work.
  928. func (d *Decoder) swallowViaHammer() {
  929. var blank interface{}
  930. d.decodeValue(reflect.ValueOf(&blank).Elem(), decFn{})
  931. }
  932. func (d *Decoder) swallow() {
  933. // smarter decode that just swallows the content
  934. dd := d.d
  935. switch {
  936. case dd.TryDecodeAsNil():
  937. case dd.IsContainerType(valueTypeMap):
  938. containerLen := dd.ReadMapStart()
  939. clenGtEqualZero := containerLen >= 0
  940. for j := 0; ; j++ {
  941. if clenGtEqualZero {
  942. if j >= containerLen {
  943. break
  944. }
  945. } else if dd.CheckBreak() {
  946. break
  947. }
  948. d.swallow()
  949. d.swallow()
  950. }
  951. dd.ReadEnd()
  952. case dd.IsContainerType(valueTypeArray):
  953. containerLenS := dd.ReadArrayStart()
  954. clenGtEqualZero := containerLenS >= 0
  955. for j := 0; ; j++ {
  956. if clenGtEqualZero {
  957. if j >= containerLenS {
  958. break
  959. }
  960. } else if dd.CheckBreak() {
  961. break
  962. }
  963. d.swallow()
  964. }
  965. dd.ReadEnd()
  966. case dd.IsContainerType(valueTypeBytes):
  967. dd.DecodeBytes(d.b[:], false, true)
  968. case dd.IsContainerType(valueTypeString):
  969. dd.DecodeBytes(d.b[:], true, true)
  970. // dd.DecodeStringAsBytes(d.b[:])
  971. default:
  972. // these are all primitives, which we can get from decodeNaked
  973. dd.DecodeNaked()
  974. }
  975. }
  976. // MustDecode is like Decode, but panics if unable to Decode.
  977. // This provides insight to the code location that triggered the error.
  978. func (d *Decoder) MustDecode(v interface{}) {
  979. d.decode(v)
  980. }
  981. func (d *Decoder) decode(iv interface{}) {
  982. // if ics, ok := iv.(Selfer); ok {
  983. // ics.CodecDecodeSelf(d)
  984. // return
  985. // }
  986. if d.d.TryDecodeAsNil() {
  987. switch v := iv.(type) {
  988. case nil:
  989. case *string:
  990. *v = ""
  991. case *bool:
  992. *v = false
  993. case *int:
  994. *v = 0
  995. case *int8:
  996. *v = 0
  997. case *int16:
  998. *v = 0
  999. case *int32:
  1000. *v = 0
  1001. case *int64:
  1002. *v = 0
  1003. case *uint:
  1004. *v = 0
  1005. case *uint8:
  1006. *v = 0
  1007. case *uint16:
  1008. *v = 0
  1009. case *uint32:
  1010. *v = 0
  1011. case *uint64:
  1012. *v = 0
  1013. case *float32:
  1014. *v = 0
  1015. case *float64:
  1016. *v = 0
  1017. case *[]uint8:
  1018. *v = nil
  1019. case reflect.Value:
  1020. d.chkPtrValue(v)
  1021. v = v.Elem()
  1022. if v.IsValid() {
  1023. v.Set(reflect.Zero(v.Type()))
  1024. }
  1025. default:
  1026. rv := reflect.ValueOf(iv)
  1027. d.chkPtrValue(rv)
  1028. rv = rv.Elem()
  1029. if rv.IsValid() {
  1030. rv.Set(reflect.Zero(rv.Type()))
  1031. }
  1032. }
  1033. return
  1034. }
  1035. switch v := iv.(type) {
  1036. case nil:
  1037. d.error(cannotDecodeIntoNilErr)
  1038. return
  1039. case Selfer:
  1040. v.CodecDecodeSelf(d)
  1041. case reflect.Value:
  1042. d.chkPtrValue(v)
  1043. d.decodeValueNotNil(v.Elem(), decFn{})
  1044. case *string:
  1045. *v = d.d.DecodeString()
  1046. case *bool:
  1047. *v = d.d.DecodeBool()
  1048. case *int:
  1049. *v = int(d.d.DecodeInt(intBitsize))
  1050. case *int8:
  1051. *v = int8(d.d.DecodeInt(8))
  1052. case *int16:
  1053. *v = int16(d.d.DecodeInt(16))
  1054. case *int32:
  1055. *v = int32(d.d.DecodeInt(32))
  1056. case *int64:
  1057. *v = d.d.DecodeInt(64)
  1058. case *uint:
  1059. *v = uint(d.d.DecodeUint(uintBitsize))
  1060. case *uint8:
  1061. *v = uint8(d.d.DecodeUint(8))
  1062. case *uint16:
  1063. *v = uint16(d.d.DecodeUint(16))
  1064. case *uint32:
  1065. *v = uint32(d.d.DecodeUint(32))
  1066. case *uint64:
  1067. *v = d.d.DecodeUint(64)
  1068. case *float32:
  1069. *v = float32(d.d.DecodeFloat(true))
  1070. case *float64:
  1071. *v = d.d.DecodeFloat(false)
  1072. case *[]uint8:
  1073. *v = d.d.DecodeBytes(*v, false, false)
  1074. case *interface{}:
  1075. d.decodeValueNotNil(reflect.ValueOf(iv).Elem(), decFn{})
  1076. default:
  1077. if !fastpathDecodeTypeSwitch(iv, d) {
  1078. d.decodeI(iv, true, false, false, false)
  1079. }
  1080. }
  1081. }
  1082. func (d *Decoder) preDecodeValue(rv reflect.Value, tryNil bool) (rv2 reflect.Value, proceed bool) {
  1083. if tryNil && d.d.TryDecodeAsNil() {
  1084. // No need to check if a ptr, recursively, to determine
  1085. // whether to set value to nil.
  1086. // Just always set value to its zero type.
  1087. if rv.IsValid() { // rv.CanSet() // always settable, except it's invalid
  1088. rv.Set(reflect.Zero(rv.Type()))
  1089. }
  1090. return
  1091. }
  1092. // If stream is not containing a nil value, then we can deref to the base
  1093. // non-pointer value, and decode into that.
  1094. for rv.Kind() == reflect.Ptr {
  1095. if rv.IsNil() {
  1096. rv.Set(reflect.New(rv.Type().Elem()))
  1097. }
  1098. rv = rv.Elem()
  1099. }
  1100. return rv, true
  1101. }
  1102. func (d *Decoder) decodeI(iv interface{}, checkPtr, tryNil, checkFastpath, checkCodecSelfer bool) {
  1103. rv := reflect.ValueOf(iv)
  1104. if checkPtr {
  1105. d.chkPtrValue(rv)
  1106. }
  1107. rv, proceed := d.preDecodeValue(rv, tryNil)
  1108. if proceed {
  1109. fn := d.getDecFn(rv.Type(), checkFastpath, checkCodecSelfer)
  1110. fn.f(fn.i, rv)
  1111. }
  1112. }
  1113. func (d *Decoder) decodeValue(rv reflect.Value, fn decFn) {
  1114. if rv, proceed := d.preDecodeValue(rv, true); proceed {
  1115. if fn.f == nil {
  1116. fn = d.getDecFn(rv.Type(), true, true)
  1117. }
  1118. fn.f(fn.i, rv)
  1119. }
  1120. }
  1121. func (d *Decoder) decodeValueNotNil(rv reflect.Value, fn decFn) {
  1122. if rv, proceed := d.preDecodeValue(rv, false); proceed {
  1123. if fn.f == nil {
  1124. fn = d.getDecFn(rv.Type(), true, true)
  1125. }
  1126. fn.f(fn.i, rv)
  1127. }
  1128. }
  1129. func (d *Decoder) getDecFn(rt reflect.Type, checkFastpath, checkCodecSelfer bool) (fn decFn) {
  1130. rtid := reflect.ValueOf(rt).Pointer()
  1131. // retrieve or register a focus'ed function for this type
  1132. // to eliminate need to do the retrieval multiple times
  1133. // if d.f == nil && d.s == nil { debugf("---->Creating new dec f map for type: %v\n", rt) }
  1134. var ok bool
  1135. if useMapForCodecCache {
  1136. fn, ok = d.f[rtid]
  1137. } else {
  1138. for _, v := range d.s {
  1139. if v.rtid == rtid {
  1140. fn, ok = v.fn, true
  1141. break
  1142. }
  1143. }
  1144. }
  1145. if ok {
  1146. return
  1147. }
  1148. // debugf("\tCreating new dec fn for type: %v\n", rt)
  1149. ti := getTypeInfo(rtid, rt)
  1150. var fi decFnInfo
  1151. fi.dd = d.d
  1152. // fi.decFnInfoX = new(decFnInfoX)
  1153. // An extension can be registered for any type, regardless of the Kind
  1154. // (e.g. type BitSet int64, type MyStruct { / * unexported fields * / }, type X []int, etc.
  1155. //
  1156. // We can't check if it's an extension byte here first, because the user may have
  1157. // registered a pointer or non-pointer type, meaning we may have to recurse first
  1158. // before matching a mapped type, even though the extension byte is already detected.
  1159. //
  1160. // NOTE: if decoding into a nil interface{}, we return a non-nil
  1161. // value except even if the container registers a length of 0.
  1162. if checkCodecSelfer && ti.cs {
  1163. fi.decFnInfoX = &decFnInfoX{d: d, ti: ti}
  1164. fn.f = (decFnInfo).selferUnmarshal
  1165. } else if rtid == rawExtTypId {
  1166. fi.decFnInfoX = &decFnInfoX{d: d, ti: ti}
  1167. fn.f = (decFnInfo).rawExt
  1168. } else if d.d.IsBuiltinType(rtid) {
  1169. fi.decFnInfoX = &decFnInfoX{d: d, ti: ti}
  1170. fn.f = (decFnInfo).builtin
  1171. } else if xfFn := d.h.getExt(rtid); xfFn != nil {
  1172. // fi.decFnInfoX = &decFnInfoX{xfTag: xfFn.tag, xfFn: xfFn.ext}
  1173. fi.decFnInfoX = &decFnInfoX{d: d, ti: ti}
  1174. fi.xfTag, fi.xfFn = xfFn.tag, xfFn.ext
  1175. fn.f = (decFnInfo).ext
  1176. } else if supportMarshalInterfaces && d.be && ti.bunm {
  1177. fi.decFnInfoX = &decFnInfoX{d: d, ti: ti}
  1178. fn.f = (decFnInfo).binaryUnmarshal
  1179. } else if supportMarshalInterfaces && !d.be && d.js && ti.junm {
  1180. //If JSON, we should check JSONUnmarshal before textUnmarshal
  1181. fi.decFnInfoX = &decFnInfoX{d: d, ti: ti}
  1182. fn.f = (decFnInfo).jsonUnmarshal
  1183. } else if supportMarshalInterfaces && !d.be && ti.tunm {
  1184. fi.decFnInfoX = &decFnInfoX{d: d, ti: ti}
  1185. fn.f = (decFnInfo).textUnmarshal
  1186. } else {
  1187. rk := rt.Kind()
  1188. if fastpathEnabled && checkFastpath && (rk == reflect.Map || rk == reflect.Slice) {
  1189. if rt.PkgPath() == "" {
  1190. if idx := fastpathAV.index(rtid); idx != -1 {
  1191. fi.decFnInfoX = &decFnInfoX{d: d, ti: ti}
  1192. fn.f = fastpathAV[idx].decfn
  1193. }
  1194. } else {
  1195. // use mapping for underlying type if there
  1196. ok = false
  1197. var rtu reflect.Type
  1198. if rk == reflect.Map {
  1199. rtu = reflect.MapOf(rt.Key(), rt.Elem())
  1200. } else {
  1201. rtu = reflect.SliceOf(rt.Elem())
  1202. }
  1203. rtuid := reflect.ValueOf(rtu).Pointer()
  1204. if idx := fastpathAV.index(rtuid); idx != -1 {
  1205. xfnf := fastpathAV[idx].decfn
  1206. xrt := fastpathAV[idx].rt
  1207. fi.decFnInfoX = &decFnInfoX{d: d, ti: ti}
  1208. fn.f = func(xf decFnInfo, xrv reflect.Value) {
  1209. // xfnf(xf, xrv.Convert(xrt))
  1210. xfnf(xf, xrv.Addr().Convert(reflect.PtrTo(xrt)).Elem())
  1211. }
  1212. }
  1213. }
  1214. }
  1215. if fn.f == nil {
  1216. switch rk {
  1217. case reflect.String:
  1218. fn.f = (decFnInfo).kString
  1219. case reflect.Bool:
  1220. fn.f = (decFnInfo).kBool
  1221. case reflect.Int:
  1222. fn.f = (decFnInfo).kInt
  1223. case reflect.Int64:
  1224. fn.f = (decFnInfo).kInt64
  1225. case reflect.Int32:
  1226. fn.f = (decFnInfo).kInt32
  1227. case reflect.Int8:
  1228. fn.f = (decFnInfo).kInt8
  1229. case reflect.Int16:
  1230. fn.f = (decFnInfo).kInt16
  1231. case reflect.Float32:
  1232. fn.f = (decFnInfo).kFloat32
  1233. case reflect.Float64:
  1234. fn.f = (decFnInfo).kFloat64
  1235. case reflect.Uint8:
  1236. fn.f = (decFnInfo).kUint8
  1237. case reflect.Uint64:
  1238. fn.f = (decFnInfo).kUint64
  1239. case reflect.Uint:
  1240. fn.f = (decFnInfo).kUint
  1241. case reflect.Uint32:
  1242. fn.f = (decFnInfo).kUint32
  1243. case reflect.Uint16:
  1244. fn.f = (decFnInfo).kUint16
  1245. // case reflect.Ptr:
  1246. // fn.f = (decFnInfo).kPtr
  1247. case reflect.Uintptr:
  1248. fn.f = (decFnInfo).kUintptr
  1249. case reflect.Interface:
  1250. fi.decFnInfoX = &decFnInfoX{d: d, ti: ti}
  1251. fn.f = (decFnInfo).kInterface
  1252. case reflect.Struct:
  1253. fi.decFnInfoX = &decFnInfoX{d: d, ti: ti}
  1254. fn.f = (decFnInfo).kStruct
  1255. case reflect.Chan:
  1256. fi.decFnInfoX = &decFnInfoX{d: d, ti: ti, seq: seqTypeChan}
  1257. fn.f = (decFnInfo).kSlice
  1258. case reflect.Slice:
  1259. fi.decFnInfoX = &decFnInfoX{d: d, ti: ti, seq: seqTypeSlice}
  1260. fn.f = (decFnInfo).kSlice
  1261. case reflect.Array:
  1262. // fi.decFnInfoX = &decFnInfoX{array: true}
  1263. fi.decFnInfoX = &decFnInfoX{d: d, ti: ti, seq: seqTypeArray}
  1264. fn.f = (decFnInfo).kArray
  1265. case reflect.Map:
  1266. fi.decFnInfoX = &decFnInfoX{d: d, ti: ti}
  1267. fn.f = (decFnInfo).kMap
  1268. default:
  1269. fn.f = (decFnInfo).kErr
  1270. }
  1271. }
  1272. }
  1273. fn.i = fi
  1274. if useMapForCodecCache {
  1275. if d.f == nil {
  1276. d.f = make(map[uintptr]decFn, 32)
  1277. }
  1278. d.f[rtid] = fn
  1279. } else {
  1280. if d.s == nil {
  1281. d.s = make([]rtidDecFn, 0, 32)
  1282. }
  1283. d.s = append(d.s, rtidDecFn{rtid, fn})
  1284. }
  1285. return
  1286. }
  1287. func (d *Decoder) structFieldNotFound(index int, rvkencname string) {
  1288. if d.h.ErrorIfNoField {
  1289. if index >= 0 {
  1290. d.errorf("no matching struct field found when decoding stream array at index %v", index)
  1291. return
  1292. } else if rvkencname != "" {
  1293. d.errorf("no matching struct field found when decoding stream map with key %s", rvkencname)
  1294. return
  1295. }
  1296. }
  1297. d.swallow()
  1298. }
  1299. func (d *Decoder) arrayCannotExpand(sliceLen, streamLen int) {
  1300. if d.h.ErrorIfNoArrayExpand {
  1301. d.errorf("cannot expand array len during decode from %v to %v", sliceLen, streamLen)
  1302. }
  1303. }
  1304. func (d *Decoder) chkPtrValue(rv reflect.Value) {
  1305. // We can only decode into a non-nil pointer
  1306. if rv.Kind() == reflect.Ptr && !rv.IsNil() {
  1307. return
  1308. }
  1309. if !rv.IsValid() {
  1310. d.error(cannotDecodeIntoNilErr)
  1311. return
  1312. }
  1313. if !rv.CanInterface() {
  1314. d.errorf("cannot decode into a value without an interface: %v", rv)
  1315. return
  1316. }
  1317. rvi := rv.Interface()
  1318. d.errorf("cannot decode into non-pointer or nil pointer. Got: %v, %T, %v", rv.Kind(), rvi, rvi)
  1319. }
  1320. func (d *Decoder) error(err error) {
  1321. panic(err)
  1322. }
  1323. func (d *Decoder) errorf(format string, params ...interface{}) {
  1324. params2 := make([]interface{}, len(params)+1)
  1325. params2[0] = d.r.numread()
  1326. copy(params2[1:], params)
  1327. err := fmt.Errorf("[pos %d]: "+format, params2...)
  1328. panic(err)
  1329. }
  1330. // --------------------------------------------------
  1331. // decSliceHelper assists when decoding into a slice, from a map or an array in the stream.
  1332. // A slice can be set from a map or array in stream. This supports the MapBySlice interface.
  1333. type decSliceHelper struct {
  1334. dd decDriver
  1335. ct valueType
  1336. }
  1337. func (d *Decoder) decSliceHelperStart() (x decSliceHelper, clen int) {
  1338. x.dd = d.d
  1339. if x.dd.IsContainerType(valueTypeArray) {
  1340. x.ct = valueTypeArray
  1341. clen = x.dd.ReadArrayStart()
  1342. } else if x.dd.IsContainerType(valueTypeMap) {
  1343. x.ct = valueTypeMap
  1344. clen = x.dd.ReadMapStart() * 2
  1345. } else {
  1346. d.errorf("only encoded map or array can be decoded into a slice")
  1347. }
  1348. return
  1349. }
  1350. func (x decSliceHelper) End() {
  1351. x.dd.ReadEnd()
  1352. }
  1353. func decByteSlice(r decReader, clen int, bs []byte) (bsOut []byte) {
  1354. if clen == 0 {
  1355. return zeroByteSlice
  1356. }
  1357. if len(bs) == clen {
  1358. bsOut = bs
  1359. } else if cap(bs) >= clen {
  1360. bsOut = bs[:clen]
  1361. } else {
  1362. bsOut = make([]byte, clen)
  1363. }
  1364. r.readb(bsOut)
  1365. return
  1366. }
  1367. func detachZeroCopyBytes(isBytesReader bool, dest []byte, in []byte) (out []byte) {
  1368. if xlen := len(in); xlen > 0 {
  1369. if isBytesReader || xlen <= scratchByteArrayLen {
  1370. if cap(dest) >= xlen {
  1371. out = dest[:xlen]
  1372. } else {
  1373. out = make([]byte, xlen)
  1374. }
  1375. copy(out, in)
  1376. return
  1377. }
  1378. }
  1379. return in
  1380. }
  1381. // // implement overall decReader wrapping both, for possible use inline:
  1382. // type decReaderT struct {
  1383. // bytes bool
  1384. // rb *bytesDecReader
  1385. // ri *ioDecReader
  1386. // }
  1387. //
  1388. // // implement *Decoder as a decReader.
  1389. // // Using decReaderT (defined just above) caused performance degradation
  1390. // // possibly because of constant copying the value,
  1391. // // and some value->interface conversion causing allocation.
  1392. // func (d *Decoder) unreadn1() {
  1393. // if d.bytes {
  1394. // d.rb.unreadn1()
  1395. // } else {
  1396. // d.ri.unreadn1()
  1397. // }
  1398. // }
  1399. // ... for other methods of decReader.
  1400. // Testing showed that performance improvement was negligible.