simple.go 18 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750
  1. // Copyright (c) 2012-2020 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. "math"
  6. "reflect"
  7. "time"
  8. )
  9. const (
  10. _ uint8 = iota
  11. simpleVdNil = 1
  12. simpleVdFalse = 2
  13. simpleVdTrue = 3
  14. simpleVdFloat32 = 4
  15. simpleVdFloat64 = 5
  16. // each lasts for 4 (ie n, n+1, n+2, n+3)
  17. simpleVdPosInt = 8
  18. simpleVdNegInt = 12
  19. simpleVdTime = 24
  20. // containers: each lasts for 4 (ie n, n+1, n+2, ... n+7)
  21. simpleVdString = 216
  22. simpleVdByteArray = 224
  23. simpleVdArray = 232
  24. simpleVdMap = 240
  25. simpleVdExt = 248
  26. )
  27. var simpledescNames = map[byte]string{
  28. simpleVdNil: "null",
  29. simpleVdFalse: "false",
  30. simpleVdTrue: "true",
  31. simpleVdFloat32: "float32",
  32. simpleVdFloat64: "float64",
  33. simpleVdPosInt: "+int",
  34. simpleVdNegInt: "-int",
  35. simpleVdTime: "time",
  36. simpleVdString: "string",
  37. simpleVdByteArray: "binary",
  38. simpleVdArray: "array",
  39. simpleVdMap: "map",
  40. simpleVdExt: "ext",
  41. }
  42. func simpledesc(bd byte) (s string) {
  43. s = simpledescNames[bd]
  44. if s == "" {
  45. s = "unknown"
  46. }
  47. return
  48. }
  49. type simpleEncDriver struct {
  50. noBuiltInTypes
  51. encDriverNoopContainerWriter
  52. encDriverNoState
  53. h *SimpleHandle
  54. // b [8]byte
  55. e Encoder
  56. }
  57. func (e *simpleEncDriver) encoder() *Encoder {
  58. return &e.e
  59. }
  60. func (e *simpleEncDriver) EncodeNil() {
  61. e.e.encWr.writen1(simpleVdNil)
  62. }
  63. func (e *simpleEncDriver) EncodeBool(b bool) {
  64. if e.h.EncZeroValuesAsNil && e.e.c != containerMapKey && !b {
  65. e.EncodeNil()
  66. return
  67. }
  68. if b {
  69. e.e.encWr.writen1(simpleVdTrue)
  70. } else {
  71. e.e.encWr.writen1(simpleVdFalse)
  72. }
  73. }
  74. func (e *simpleEncDriver) EncodeFloat32(f float32) {
  75. if e.h.EncZeroValuesAsNil && e.e.c != containerMapKey && f == 0.0 {
  76. e.EncodeNil()
  77. return
  78. }
  79. e.e.encWr.writen1(simpleVdFloat32)
  80. bigen.writeUint32(e.e.w(), math.Float32bits(f))
  81. }
  82. func (e *simpleEncDriver) EncodeFloat64(f float64) {
  83. if e.h.EncZeroValuesAsNil && e.e.c != containerMapKey && f == 0.0 {
  84. e.EncodeNil()
  85. return
  86. }
  87. e.e.encWr.writen1(simpleVdFloat64)
  88. bigen.writeUint64(e.e.w(), math.Float64bits(f))
  89. }
  90. func (e *simpleEncDriver) EncodeInt(v int64) {
  91. if v < 0 {
  92. e.encUint(uint64(-v), simpleVdNegInt)
  93. } else {
  94. e.encUint(uint64(v), simpleVdPosInt)
  95. }
  96. }
  97. func (e *simpleEncDriver) EncodeUint(v uint64) {
  98. e.encUint(v, simpleVdPosInt)
  99. }
  100. func (e *simpleEncDriver) encUint(v uint64, bd uint8) {
  101. if e.h.EncZeroValuesAsNil && e.e.c != containerMapKey && v == 0 {
  102. e.EncodeNil()
  103. return
  104. }
  105. if v <= math.MaxUint8 {
  106. e.e.encWr.writen2(bd, uint8(v))
  107. } else if v <= math.MaxUint16 {
  108. e.e.encWr.writen1(bd + 1)
  109. bigen.writeUint16(e.e.w(), uint16(v))
  110. } else if v <= math.MaxUint32 {
  111. e.e.encWr.writen1(bd + 2)
  112. bigen.writeUint32(e.e.w(), uint32(v))
  113. } else { // if v <= math.MaxUint64 {
  114. e.e.encWr.writen1(bd + 3)
  115. bigen.writeUint64(e.e.w(), v)
  116. }
  117. }
  118. func (e *simpleEncDriver) encLen(bd byte, length int) {
  119. if length == 0 {
  120. e.e.encWr.writen1(bd)
  121. } else if length <= math.MaxUint8 {
  122. e.e.encWr.writen1(bd + 1)
  123. e.e.encWr.writen1(uint8(length))
  124. } else if length <= math.MaxUint16 {
  125. e.e.encWr.writen1(bd + 2)
  126. bigen.writeUint16(e.e.w(), uint16(length))
  127. } else if int64(length) <= math.MaxUint32 {
  128. e.e.encWr.writen1(bd + 3)
  129. bigen.writeUint32(e.e.w(), uint32(length))
  130. } else {
  131. e.e.encWr.writen1(bd + 4)
  132. bigen.writeUint64(e.e.w(), uint64(length))
  133. }
  134. }
  135. func (e *simpleEncDriver) EncodeExt(v interface{}, basetype reflect.Type, xtag uint64, ext Ext) {
  136. var bs0, bs []byte
  137. if ext == SelfExt {
  138. bs0 = e.e.blist.get(1024)
  139. bs = bs0
  140. e.e.sideEncode(v, basetype, &bs)
  141. } else {
  142. bs = ext.WriteExt(v)
  143. }
  144. if bs == nil {
  145. e.EncodeNil()
  146. goto END
  147. }
  148. e.encodeExtPreamble(uint8(xtag), len(bs))
  149. e.e.encWr.writeb(bs)
  150. END:
  151. if ext == SelfExt {
  152. e.e.blist.put(bs)
  153. if !byteSliceSameData(bs0, bs) {
  154. e.e.blist.put(bs0)
  155. }
  156. }
  157. }
  158. func (e *simpleEncDriver) EncodeRawExt(re *RawExt) {
  159. e.encodeExtPreamble(uint8(re.Tag), len(re.Data))
  160. e.e.encWr.writeb(re.Data)
  161. }
  162. func (e *simpleEncDriver) encodeExtPreamble(xtag byte, length int) {
  163. e.encLen(simpleVdExt, length)
  164. e.e.encWr.writen1(xtag)
  165. }
  166. func (e *simpleEncDriver) WriteArrayStart(length int) {
  167. e.encLen(simpleVdArray, length)
  168. }
  169. func (e *simpleEncDriver) WriteMapStart(length int) {
  170. e.encLen(simpleVdMap, length)
  171. }
  172. func (e *simpleEncDriver) EncodeString(v string) {
  173. if e.h.EncZeroValuesAsNil && e.e.c != containerMapKey && v == "" {
  174. e.EncodeNil()
  175. return
  176. }
  177. if e.h.StringToRaw {
  178. e.encLen(simpleVdByteArray, len(v))
  179. } else {
  180. e.encLen(simpleVdString, len(v))
  181. }
  182. e.e.encWr.writestr(v)
  183. }
  184. func (e *simpleEncDriver) EncodeStringBytesRaw(v []byte) {
  185. // if e.h.EncZeroValuesAsNil && e.c != containerMapKey && v == nil {
  186. if v == nil {
  187. e.EncodeNil()
  188. return
  189. }
  190. e.encLen(simpleVdByteArray, len(v))
  191. e.e.encWr.writeb(v)
  192. }
  193. func (e *simpleEncDriver) EncodeTime(t time.Time) {
  194. // if e.h.EncZeroValuesAsNil && e.c != containerMapKey && t.IsZero() {
  195. if t.IsZero() {
  196. e.EncodeNil()
  197. return
  198. }
  199. v, err := t.MarshalBinary()
  200. e.e.onerror(err)
  201. e.e.encWr.writen2(simpleVdTime, uint8(len(v)))
  202. e.e.encWr.writeb(v)
  203. }
  204. //------------------------------------
  205. type simpleDecDriver struct {
  206. h *SimpleHandle
  207. bdAndBdread
  208. _ bool
  209. noBuiltInTypes
  210. decDriverNoopContainerReader
  211. decDriverNoopNumberHelper
  212. d Decoder
  213. }
  214. func (d *simpleDecDriver) decoder() *Decoder {
  215. return &d.d
  216. }
  217. func (d *simpleDecDriver) descBd() string {
  218. return sprintf("%v (%s)", d.bd, simpledesc(d.bd))
  219. }
  220. func (d *simpleDecDriver) readNextBd() {
  221. d.bd = d.d.decRd.readn1()
  222. d.bdRead = true
  223. }
  224. func (d *simpleDecDriver) advanceNil() (null bool) {
  225. if !d.bdRead {
  226. d.readNextBd()
  227. }
  228. if d.bd == simpleVdNil {
  229. d.bdRead = false
  230. return true // null = true
  231. }
  232. return
  233. }
  234. func (d *simpleDecDriver) ContainerType() (vt valueType) {
  235. if !d.bdRead {
  236. d.readNextBd()
  237. }
  238. switch d.bd {
  239. case simpleVdNil:
  240. d.bdRead = false
  241. return valueTypeNil
  242. case simpleVdByteArray, simpleVdByteArray + 1,
  243. simpleVdByteArray + 2, simpleVdByteArray + 3, simpleVdByteArray + 4:
  244. return valueTypeBytes
  245. case simpleVdString, simpleVdString + 1,
  246. simpleVdString + 2, simpleVdString + 3, simpleVdString + 4:
  247. return valueTypeString
  248. case simpleVdArray, simpleVdArray + 1,
  249. simpleVdArray + 2, simpleVdArray + 3, simpleVdArray + 4:
  250. return valueTypeArray
  251. case simpleVdMap, simpleVdMap + 1,
  252. simpleVdMap + 2, simpleVdMap + 3, simpleVdMap + 4:
  253. return valueTypeMap
  254. }
  255. return valueTypeUnset
  256. }
  257. func (d *simpleDecDriver) TryNil() bool {
  258. return d.advanceNil()
  259. }
  260. func (d *simpleDecDriver) decFloat() (f float64, ok bool) {
  261. ok = true
  262. switch d.bd {
  263. case simpleVdFloat32:
  264. f = float64(math.Float32frombits(bigen.Uint32(d.d.decRd.readn4())))
  265. case simpleVdFloat64:
  266. f = math.Float64frombits(bigen.Uint64(d.d.decRd.readn8()))
  267. default:
  268. ok = false
  269. }
  270. return
  271. }
  272. func (d *simpleDecDriver) decInteger() (ui uint64, neg, ok bool) {
  273. ok = true
  274. switch d.bd {
  275. case simpleVdPosInt:
  276. ui = uint64(d.d.decRd.readn1())
  277. case simpleVdPosInt + 1:
  278. ui = uint64(bigen.Uint16(d.d.decRd.readn2()))
  279. case simpleVdPosInt + 2:
  280. ui = uint64(bigen.Uint32(d.d.decRd.readn4()))
  281. case simpleVdPosInt + 3:
  282. ui = uint64(bigen.Uint64(d.d.decRd.readn8()))
  283. case simpleVdNegInt:
  284. ui = uint64(d.d.decRd.readn1())
  285. neg = true
  286. case simpleVdNegInt + 1:
  287. ui = uint64(bigen.Uint16(d.d.decRd.readn2()))
  288. neg = true
  289. case simpleVdNegInt + 2:
  290. ui = uint64(bigen.Uint32(d.d.decRd.readn4()))
  291. neg = true
  292. case simpleVdNegInt + 3:
  293. ui = uint64(bigen.Uint64(d.d.decRd.readn8()))
  294. neg = true
  295. default:
  296. ok = false
  297. // d.d.errorf("integer only valid from pos/neg integer1..8. Invalid descriptor: %v", d.bd)
  298. }
  299. // DO NOT do this check below, because callers may only want the unsigned value:
  300. //
  301. // if ui > math.MaxInt64 {
  302. // d.d.errorf("decIntAny: Integer out of range for signed int64: %v", ui)
  303. // return
  304. // }
  305. return
  306. }
  307. func (d *simpleDecDriver) DecodeInt64() (i int64) {
  308. if d.advanceNil() {
  309. return
  310. }
  311. i = decNegintPosintFloatNumberHelper{&d.d}.int64(d.decInteger())
  312. d.bdRead = false
  313. return
  314. }
  315. func (d *simpleDecDriver) DecodeUint64() (ui uint64) {
  316. if d.advanceNil() {
  317. return
  318. }
  319. ui = decNegintPosintFloatNumberHelper{&d.d}.uint64(d.decInteger())
  320. d.bdRead = false
  321. return
  322. }
  323. func (d *simpleDecDriver) DecodeFloat64() (f float64) {
  324. if d.advanceNil() {
  325. return
  326. }
  327. f = decNegintPosintFloatNumberHelper{&d.d}.float64(d.decFloat())
  328. d.bdRead = false
  329. return
  330. }
  331. // bool can be decoded from bool only (single byte).
  332. func (d *simpleDecDriver) DecodeBool() (b bool) {
  333. if d.advanceNil() {
  334. return
  335. }
  336. if d.bd == simpleVdFalse {
  337. } else if d.bd == simpleVdTrue {
  338. b = true
  339. } else {
  340. d.d.errorf("cannot decode bool - %s: %x", msgBadDesc, d.bd)
  341. }
  342. d.bdRead = false
  343. return
  344. }
  345. func (d *simpleDecDriver) ReadMapStart() (length int) {
  346. if d.advanceNil() {
  347. return containerLenNil
  348. }
  349. d.bdRead = false
  350. return d.decLen()
  351. }
  352. func (d *simpleDecDriver) ReadArrayStart() (length int) {
  353. if d.advanceNil() {
  354. return containerLenNil
  355. }
  356. d.bdRead = false
  357. return d.decLen()
  358. }
  359. func (d *simpleDecDriver) uint2Len(ui uint64) int {
  360. if chkOvf.Uint(ui, intBitsize) {
  361. d.d.errorf("overflow integer: %v", ui)
  362. }
  363. return int(ui)
  364. }
  365. func (d *simpleDecDriver) decLen() int {
  366. switch d.bd & 7 { // d.bd % 8 {
  367. case 0:
  368. return 0
  369. case 1:
  370. return int(d.d.decRd.readn1())
  371. case 2:
  372. return int(bigen.Uint16(d.d.decRd.readn2()))
  373. case 3:
  374. return d.uint2Len(uint64(bigen.Uint32(d.d.decRd.readn4())))
  375. case 4:
  376. return d.uint2Len(bigen.Uint64(d.d.decRd.readn8()))
  377. }
  378. d.d.errorf("cannot read length: bd%%8 must be in range 0..4. Got: %d", d.bd%8)
  379. return -1
  380. }
  381. func (d *simpleDecDriver) DecodeStringAsBytes() (s []byte) {
  382. return d.DecodeBytes(nil)
  383. }
  384. func (d *simpleDecDriver) DecodeBytes(bs []byte) (bsOut []byte) {
  385. d.d.decByteState = decByteStateNone
  386. if d.advanceNil() {
  387. return
  388. }
  389. // check if an "array" of uint8's (see ContainerType for how to infer if an array)
  390. if d.bd >= simpleVdArray && d.bd <= simpleVdMap+4 {
  391. if bs == nil {
  392. d.d.decByteState = decByteStateReuseBuf
  393. bs = d.d.b[:]
  394. }
  395. slen := d.ReadArrayStart()
  396. var changed bool
  397. if bs, changed = usableByteSlice(bs, slen); changed {
  398. d.d.decByteState = decByteStateNone
  399. }
  400. for i := 0; i < len(bs); i++ {
  401. bs[i] = uint8(chkOvf.UintV(d.DecodeUint64(), 8))
  402. }
  403. for i := len(bs); i < slen; i++ {
  404. bs = append(bs, uint8(chkOvf.UintV(d.DecodeUint64(), 8)))
  405. }
  406. return bs
  407. }
  408. clen := d.decLen()
  409. d.bdRead = false
  410. if d.d.zerocopy() {
  411. d.d.decByteState = decByteStateZerocopy
  412. return d.d.decRd.rb.readx(uint(clen))
  413. }
  414. if bs == nil {
  415. d.d.decByteState = decByteStateReuseBuf
  416. bs = d.d.b[:]
  417. }
  418. return decByteSlice(d.d.r(), clen, d.d.h.MaxInitLen, bs)
  419. }
  420. func (d *simpleDecDriver) DecodeTime() (t time.Time) {
  421. if d.advanceNil() {
  422. return
  423. }
  424. if d.bd != simpleVdTime {
  425. d.d.errorf("invalid descriptor for time.Time - expect 0x%x, received 0x%x", simpleVdTime, d.bd)
  426. }
  427. d.bdRead = false
  428. clen := uint(d.d.decRd.readn1())
  429. b := d.d.decRd.readx(clen)
  430. d.d.onerror((&t).UnmarshalBinary(b))
  431. return
  432. }
  433. func (d *simpleDecDriver) DecodeExt(rv interface{}, basetype reflect.Type, xtag uint64, ext Ext) {
  434. if xtag > 0xff {
  435. d.d.errorf("ext: tag must be <= 0xff; got: %v", xtag)
  436. }
  437. if d.advanceNil() {
  438. return
  439. }
  440. xbs, realxtag1, zerocopy := d.decodeExtV(ext != nil, uint8(xtag))
  441. realxtag := uint64(realxtag1)
  442. if ext == nil {
  443. re := rv.(*RawExt)
  444. re.Tag = realxtag
  445. re.setData(xbs, zerocopy)
  446. } else if ext == SelfExt {
  447. d.d.sideDecode(rv, basetype, xbs)
  448. } else {
  449. ext.ReadExt(rv, xbs)
  450. }
  451. }
  452. func (d *simpleDecDriver) decodeExtV(verifyTag bool, tag byte) (xbs []byte, xtag byte, zerocopy bool) {
  453. switch d.bd {
  454. case simpleVdExt, simpleVdExt + 1, simpleVdExt + 2, simpleVdExt + 3, simpleVdExt + 4:
  455. l := d.decLen()
  456. xtag = d.d.decRd.readn1()
  457. if verifyTag && xtag != tag {
  458. d.d.errorf("wrong extension tag. Got %b. Expecting: %v", xtag, tag)
  459. }
  460. if d.d.bytes {
  461. xbs = d.d.decRd.rb.readx(uint(l))
  462. zerocopy = true
  463. } else {
  464. xbs = decByteSlice(d.d.r(), l, d.d.h.MaxInitLen, d.d.b[:])
  465. }
  466. case simpleVdByteArray, simpleVdByteArray + 1,
  467. simpleVdByteArray + 2, simpleVdByteArray + 3, simpleVdByteArray + 4:
  468. xbs = d.DecodeBytes(nil)
  469. default:
  470. d.d.errorf("ext - %s - expecting extensions/bytearray, got: 0x%x", msgBadDesc, d.bd)
  471. }
  472. d.bdRead = false
  473. return
  474. }
  475. func (d *simpleDecDriver) DecodeNaked() {
  476. if !d.bdRead {
  477. d.readNextBd()
  478. }
  479. n := d.d.naked()
  480. var decodeFurther bool
  481. switch d.bd {
  482. case simpleVdNil:
  483. n.v = valueTypeNil
  484. case simpleVdFalse:
  485. n.v = valueTypeBool
  486. n.b = false
  487. case simpleVdTrue:
  488. n.v = valueTypeBool
  489. n.b = true
  490. case simpleVdPosInt, simpleVdPosInt + 1, simpleVdPosInt + 2, simpleVdPosInt + 3:
  491. if d.h.SignedInteger {
  492. n.v = valueTypeInt
  493. n.i = d.DecodeInt64()
  494. } else {
  495. n.v = valueTypeUint
  496. n.u = d.DecodeUint64()
  497. }
  498. case simpleVdNegInt, simpleVdNegInt + 1, simpleVdNegInt + 2, simpleVdNegInt + 3:
  499. n.v = valueTypeInt
  500. n.i = d.DecodeInt64()
  501. case simpleVdFloat32:
  502. n.v = valueTypeFloat
  503. n.f = d.DecodeFloat64()
  504. case simpleVdFloat64:
  505. n.v = valueTypeFloat
  506. n.f = d.DecodeFloat64()
  507. case simpleVdTime:
  508. n.v = valueTypeTime
  509. n.t = d.DecodeTime()
  510. case simpleVdString, simpleVdString + 1,
  511. simpleVdString + 2, simpleVdString + 3, simpleVdString + 4:
  512. n.v = valueTypeString
  513. n.s = d.d.stringZC(d.DecodeStringAsBytes())
  514. case simpleVdByteArray, simpleVdByteArray + 1,
  515. simpleVdByteArray + 2, simpleVdByteArray + 3, simpleVdByteArray + 4:
  516. d.d.fauxUnionReadRawBytes(false)
  517. case simpleVdExt, simpleVdExt + 1, simpleVdExt + 2, simpleVdExt + 3, simpleVdExt + 4:
  518. n.v = valueTypeExt
  519. l := d.decLen()
  520. n.u = uint64(d.d.decRd.readn1())
  521. if d.d.bytes {
  522. n.l = d.d.decRd.rb.readx(uint(l))
  523. } else {
  524. n.l = decByteSlice(d.d.r(), l, d.d.h.MaxInitLen, d.d.b[:])
  525. }
  526. case simpleVdArray, simpleVdArray + 1, simpleVdArray + 2,
  527. simpleVdArray + 3, simpleVdArray + 4:
  528. n.v = valueTypeArray
  529. decodeFurther = true
  530. case simpleVdMap, simpleVdMap + 1, simpleVdMap + 2, simpleVdMap + 3, simpleVdMap + 4:
  531. n.v = valueTypeMap
  532. decodeFurther = true
  533. default:
  534. d.d.errorf("cannot infer value - %s 0x%x", msgBadDesc, d.bd)
  535. }
  536. if !decodeFurther {
  537. d.bdRead = false
  538. }
  539. }
  540. func (d *simpleDecDriver) nextValueBytes(v0 []byte) (v []byte) {
  541. if !d.bdRead {
  542. d.readNextBd()
  543. }
  544. v = v0
  545. var h = decNextValueBytesHelper{d: &d.d}
  546. var cursor = d.d.rb.c - 1
  547. h.append1(&v, d.bd)
  548. v = d.nextValueBytesBdReadR(v)
  549. d.bdRead = false
  550. h.bytesRdV(&v, cursor)
  551. return
  552. }
  553. func (d *simpleDecDriver) nextValueBytesR(v0 []byte) (v []byte) {
  554. d.readNextBd()
  555. v = v0
  556. var h = decNextValueBytesHelper{d: &d.d}
  557. h.append1(&v, d.bd)
  558. return d.nextValueBytesBdReadR(v)
  559. }
  560. func (d *simpleDecDriver) nextValueBytesBdReadR(v0 []byte) (v []byte) {
  561. v = v0
  562. var h = decNextValueBytesHelper{d: &d.d}
  563. c := d.bd
  564. var length uint
  565. switch c {
  566. case simpleVdNil, simpleVdFalse, simpleVdTrue, simpleVdString, simpleVdByteArray:
  567. // pass
  568. case simpleVdPosInt, simpleVdNegInt:
  569. h.append1(&v, d.d.decRd.readn1())
  570. case simpleVdPosInt + 1, simpleVdNegInt + 1:
  571. h.appendN(&v, d.d.decRd.readx(2)...)
  572. case simpleVdPosInt + 2, simpleVdNegInt + 2, simpleVdFloat32:
  573. h.appendN(&v, d.d.decRd.readx(4)...)
  574. case simpleVdPosInt + 3, simpleVdNegInt + 3, simpleVdFloat64:
  575. h.appendN(&v, d.d.decRd.readx(8)...)
  576. case simpleVdTime:
  577. c = d.d.decRd.readn1()
  578. h.append1(&v, c)
  579. h.appendN(&v, d.d.decRd.readx(uint(c))...)
  580. default:
  581. switch c & 7 { // c % 8 {
  582. case 0:
  583. length = 0
  584. case 1:
  585. b := d.d.decRd.readn1()
  586. length = uint(b)
  587. h.append1(&v, b)
  588. case 2:
  589. x := d.d.decRd.readn2()
  590. length = uint(bigen.Uint16(x))
  591. h.appendN(&v, x[:]...)
  592. case 3:
  593. x := d.d.decRd.readn4()
  594. length = uint(bigen.Uint32(x))
  595. h.appendN(&v, x[:]...)
  596. case 4:
  597. x := d.d.decRd.readn8()
  598. length = uint(bigen.Uint64(x))
  599. h.appendN(&v, x[:]...)
  600. }
  601. bExt := c >= simpleVdExt && c <= simpleVdExt+7
  602. bStr := c >= simpleVdString && c <= simpleVdString+7
  603. bByteArray := c >= simpleVdByteArray && c <= simpleVdByteArray+7
  604. bArray := c >= simpleVdArray && c <= simpleVdArray+7
  605. bMap := c >= simpleVdMap && c <= simpleVdMap+7
  606. if !(bExt || bStr || bByteArray || bArray || bMap) {
  607. d.d.errorf("cannot infer value - %s 0x%x", msgBadDesc, c)
  608. }
  609. if bExt {
  610. h.append1(&v, d.d.decRd.readn1()) // tag
  611. }
  612. if length == 0 {
  613. break
  614. }
  615. if bArray {
  616. for i := uint(0); i < length; i++ {
  617. v = d.nextValueBytesR(v)
  618. }
  619. } else if bMap {
  620. for i := uint(0); i < length; i++ {
  621. v = d.nextValueBytesR(v)
  622. v = d.nextValueBytesR(v)
  623. }
  624. } else {
  625. h.appendN(&v, d.d.decRd.readx(length)...)
  626. }
  627. }
  628. return
  629. }
  630. //------------------------------------
  631. // SimpleHandle is a Handle for a very simple encoding format.
  632. //
  633. // simple is a simplistic codec similar to binc, but not as compact.
  634. // - Encoding of a value is always preceded by the descriptor byte (bd)
  635. // - True, false, nil are encoded fully in 1 byte (the descriptor)
  636. // - Integers (intXXX, uintXXX) are encoded in 1, 2, 4 or 8 bytes (plus a descriptor byte).
  637. // There are positive (uintXXX and intXXX >= 0) and negative (intXXX < 0) integers.
  638. // - Floats are encoded in 4 or 8 bytes (plus a descriptor byte)
  639. // - Length of containers (strings, bytes, array, map, extensions)
  640. // are encoded in 0, 1, 2, 4 or 8 bytes.
  641. // Zero-length containers have no length encoded.
  642. // For others, the number of bytes is given by pow(2, bd%3)
  643. // - maps are encoded as [bd] [length] [[key][value]]...
  644. // - arrays are encoded as [bd] [length] [value]...
  645. // - extensions are encoded as [bd] [length] [tag] [byte]...
  646. // - strings/bytearrays are encoded as [bd] [length] [byte]...
  647. // - time.Time are encoded as [bd] [length] [byte]...
  648. //
  649. // The full spec will be published soon.
  650. type SimpleHandle struct {
  651. binaryEncodingType
  652. BasicHandle
  653. // EncZeroValuesAsNil says to encode zero values for numbers, bool, string, etc as nil
  654. EncZeroValuesAsNil bool
  655. }
  656. // Name returns the name of the handle: simple
  657. func (h *SimpleHandle) Name() string { return "simple" }
  658. func (h *SimpleHandle) desc(bd byte) string { return simpledesc(bd) }
  659. func (h *SimpleHandle) newEncDriver() encDriver {
  660. var e = &simpleEncDriver{h: h}
  661. e.e.e = e
  662. e.e.init(h)
  663. e.reset()
  664. return e
  665. }
  666. func (h *SimpleHandle) newDecDriver() decDriver {
  667. d := &simpleDecDriver{h: h}
  668. d.d.d = d
  669. d.d.init(h)
  670. d.reset()
  671. return d
  672. }
  673. var _ decDriver = (*simpleDecDriver)(nil)
  674. var _ encDriver = (*simpleEncDriver)(nil)