binc.go 21 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946
  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. "math"
  6. "reflect"
  7. "time"
  8. )
  9. const bincDoPrune = true // No longer needed. Needed before as C lib did not support pruning.
  10. // vd as low 4 bits (there are 16 slots)
  11. const (
  12. bincVdSpecial byte = iota
  13. bincVdPosInt
  14. bincVdNegInt
  15. bincVdFloat
  16. bincVdString
  17. bincVdByteArray
  18. bincVdArray
  19. bincVdMap
  20. bincVdTimestamp
  21. bincVdSmallInt
  22. bincVdUnicodeOther
  23. bincVdSymbol
  24. bincVdDecimal
  25. _ // open slot
  26. _ // open slot
  27. bincVdCustomExt = 0x0f
  28. )
  29. const (
  30. bincSpNil byte = iota
  31. bincSpFalse
  32. bincSpTrue
  33. bincSpNan
  34. bincSpPosInf
  35. bincSpNegInf
  36. bincSpZeroFloat
  37. bincSpZero
  38. bincSpNegOne
  39. )
  40. const (
  41. bincFlBin16 byte = iota
  42. bincFlBin32
  43. _ // bincFlBin32e
  44. bincFlBin64
  45. _ // bincFlBin64e
  46. // others not currently supported
  47. )
  48. type bincEncDriver struct {
  49. e *Encoder
  50. w encWriter
  51. m map[string]uint16 // symbols
  52. b [scratchByteArrayLen]byte
  53. s uint16 // symbols sequencer
  54. // encNoSeparator
  55. encDriverNoopContainerWriter
  56. }
  57. func (e *bincEncDriver) IsBuiltinType(rt uintptr) bool {
  58. return rt == timeTypId
  59. }
  60. func (e *bincEncDriver) EncodeBuiltin(rt uintptr, v interface{}) {
  61. if rt == timeTypId {
  62. var bs []byte
  63. switch x := v.(type) {
  64. case time.Time:
  65. bs = encodeTime(x)
  66. case *time.Time:
  67. bs = encodeTime(*x)
  68. default:
  69. e.e.errorf("binc error encoding builtin: expect time.Time, received %T", v)
  70. }
  71. e.w.writen1(bincVdTimestamp<<4 | uint8(len(bs)))
  72. e.w.writeb(bs)
  73. }
  74. }
  75. func (e *bincEncDriver) EncodeNil() {
  76. e.w.writen1(bincVdSpecial<<4 | bincSpNil)
  77. }
  78. func (e *bincEncDriver) EncodeBool(b bool) {
  79. if b {
  80. e.w.writen1(bincVdSpecial<<4 | bincSpTrue)
  81. } else {
  82. e.w.writen1(bincVdSpecial<<4 | bincSpFalse)
  83. }
  84. }
  85. func (e *bincEncDriver) EncodeFloat32(f float32) {
  86. if f == 0 {
  87. e.w.writen1(bincVdSpecial<<4 | bincSpZeroFloat)
  88. return
  89. }
  90. e.w.writen1(bincVdFloat<<4 | bincFlBin32)
  91. bigenHelper{e.b[:4], e.w}.writeUint32(math.Float32bits(f))
  92. }
  93. func (e *bincEncDriver) EncodeFloat64(f float64) {
  94. if f == 0 {
  95. e.w.writen1(bincVdSpecial<<4 | bincSpZeroFloat)
  96. return
  97. }
  98. bigen.PutUint64(e.b[:8], math.Float64bits(f))
  99. if bincDoPrune {
  100. i := 7
  101. for ; i >= 0 && (e.b[i] == 0); i-- {
  102. }
  103. i++
  104. if i <= 6 {
  105. e.w.writen1(bincVdFloat<<4 | 0x8 | bincFlBin64)
  106. e.w.writen1(byte(i))
  107. e.w.writeb(e.b[:i])
  108. return
  109. }
  110. }
  111. e.w.writen1(bincVdFloat<<4 | bincFlBin64)
  112. e.w.writeb(e.b[:8])
  113. }
  114. func (e *bincEncDriver) encIntegerPrune(bd byte, pos bool, v uint64, lim uint8) {
  115. if lim == 4 {
  116. bigen.PutUint32(e.b[:lim], uint32(v))
  117. } else {
  118. bigen.PutUint64(e.b[:lim], v)
  119. }
  120. if bincDoPrune {
  121. i := pruneSignExt(e.b[:lim], pos)
  122. e.w.writen1(bd | lim - 1 - byte(i))
  123. e.w.writeb(e.b[i:lim])
  124. } else {
  125. e.w.writen1(bd | lim - 1)
  126. e.w.writeb(e.b[:lim])
  127. }
  128. }
  129. func (e *bincEncDriver) EncodeInt(v int64) {
  130. const nbd byte = bincVdNegInt << 4
  131. if v >= 0 {
  132. e.encUint(bincVdPosInt<<4, true, uint64(v))
  133. } else if v == -1 {
  134. e.w.writen1(bincVdSpecial<<4 | bincSpNegOne)
  135. } else {
  136. e.encUint(bincVdNegInt<<4, false, uint64(-v))
  137. }
  138. }
  139. func (e *bincEncDriver) EncodeUint(v uint64) {
  140. e.encUint(bincVdPosInt<<4, true, v)
  141. }
  142. func (e *bincEncDriver) encUint(bd byte, pos bool, v uint64) {
  143. if v == 0 {
  144. e.w.writen1(bincVdSpecial<<4 | bincSpZero)
  145. } else if pos && v >= 1 && v <= 16 {
  146. e.w.writen1(bincVdSmallInt<<4 | byte(v-1))
  147. } else if v <= math.MaxUint8 {
  148. e.w.writen2(bd|0x0, byte(v))
  149. } else if v <= math.MaxUint16 {
  150. e.w.writen1(bd | 0x01)
  151. bigenHelper{e.b[:2], e.w}.writeUint16(uint16(v))
  152. } else if v <= math.MaxUint32 {
  153. e.encIntegerPrune(bd, pos, v, 4)
  154. } else {
  155. e.encIntegerPrune(bd, pos, v, 8)
  156. }
  157. }
  158. func (e *bincEncDriver) EncodeExt(rv interface{}, xtag uint64, ext Ext, _ *Encoder) {
  159. bs := ext.WriteExt(rv)
  160. if bs == nil {
  161. e.EncodeNil()
  162. return
  163. }
  164. e.encodeExtPreamble(uint8(xtag), len(bs))
  165. e.w.writeb(bs)
  166. }
  167. func (e *bincEncDriver) EncodeRawExt(re *RawExt, _ *Encoder) {
  168. e.encodeExtPreamble(uint8(re.Tag), len(re.Data))
  169. e.w.writeb(re.Data)
  170. }
  171. func (e *bincEncDriver) encodeExtPreamble(xtag byte, length int) {
  172. e.encLen(bincVdCustomExt<<4, uint64(length))
  173. e.w.writen1(xtag)
  174. }
  175. func (e *bincEncDriver) WriteArrayStart(length int) {
  176. e.encLen(bincVdArray<<4, uint64(length))
  177. }
  178. func (e *bincEncDriver) WriteMapStart(length int) {
  179. e.encLen(bincVdMap<<4, uint64(length))
  180. }
  181. func (e *bincEncDriver) EncodeString(c charEncoding, v string) {
  182. l := uint64(len(v))
  183. e.encBytesLen(c, l)
  184. if l > 0 {
  185. e.w.writestr(v)
  186. }
  187. }
  188. func (e *bincEncDriver) EncodeSymbol(v string) {
  189. // if WriteSymbolsNoRefs {
  190. // e.encodeString(c_UTF8, v)
  191. // return
  192. // }
  193. //symbols only offer benefit when string length > 1.
  194. //This is because strings with length 1 take only 2 bytes to store
  195. //(bd with embedded length, and single byte for string val).
  196. l := len(v)
  197. if l == 0 {
  198. e.encBytesLen(c_UTF8, 0)
  199. return
  200. } else if l == 1 {
  201. e.encBytesLen(c_UTF8, 1)
  202. e.w.writen1(v[0])
  203. return
  204. }
  205. if e.m == nil {
  206. e.m = make(map[string]uint16, 16)
  207. }
  208. ui, ok := e.m[v]
  209. if ok {
  210. if ui <= math.MaxUint8 {
  211. e.w.writen2(bincVdSymbol<<4, byte(ui))
  212. } else {
  213. e.w.writen1(bincVdSymbol<<4 | 0x8)
  214. bigenHelper{e.b[:2], e.w}.writeUint16(ui)
  215. }
  216. } else {
  217. e.s++
  218. ui = e.s
  219. //ui = uint16(atomic.AddUint32(&e.s, 1))
  220. e.m[v] = ui
  221. var lenprec uint8
  222. if l <= math.MaxUint8 {
  223. // lenprec = 0
  224. } else if l <= math.MaxUint16 {
  225. lenprec = 1
  226. } else if int64(l) <= math.MaxUint32 {
  227. lenprec = 2
  228. } else {
  229. lenprec = 3
  230. }
  231. if ui <= math.MaxUint8 {
  232. e.w.writen2(bincVdSymbol<<4|0x0|0x4|lenprec, byte(ui))
  233. } else {
  234. e.w.writen1(bincVdSymbol<<4 | 0x8 | 0x4 | lenprec)
  235. bigenHelper{e.b[:2], e.w}.writeUint16(ui)
  236. }
  237. if lenprec == 0 {
  238. e.w.writen1(byte(l))
  239. } else if lenprec == 1 {
  240. bigenHelper{e.b[:2], e.w}.writeUint16(uint16(l))
  241. } else if lenprec == 2 {
  242. bigenHelper{e.b[:4], e.w}.writeUint32(uint32(l))
  243. } else {
  244. bigenHelper{e.b[:8], e.w}.writeUint64(uint64(l))
  245. }
  246. e.w.writestr(v)
  247. }
  248. }
  249. func (e *bincEncDriver) EncodeStringBytes(c charEncoding, v []byte) {
  250. l := uint64(len(v))
  251. e.encBytesLen(c, l)
  252. if l > 0 {
  253. e.w.writeb(v)
  254. }
  255. }
  256. func (e *bincEncDriver) encBytesLen(c charEncoding, length uint64) {
  257. //TODO: support bincUnicodeOther (for now, just use string or bytearray)
  258. if c == c_RAW {
  259. e.encLen(bincVdByteArray<<4, length)
  260. } else {
  261. e.encLen(bincVdString<<4, length)
  262. }
  263. }
  264. func (e *bincEncDriver) encLen(bd byte, l uint64) {
  265. if l < 12 {
  266. e.w.writen1(bd | uint8(l+4))
  267. } else {
  268. e.encLenNumber(bd, l)
  269. }
  270. }
  271. func (e *bincEncDriver) encLenNumber(bd byte, v uint64) {
  272. if v <= math.MaxUint8 {
  273. e.w.writen2(bd, byte(v))
  274. } else if v <= math.MaxUint16 {
  275. e.w.writen1(bd | 0x01)
  276. bigenHelper{e.b[:2], e.w}.writeUint16(uint16(v))
  277. } else if v <= math.MaxUint32 {
  278. e.w.writen1(bd | 0x02)
  279. bigenHelper{e.b[:4], e.w}.writeUint32(uint32(v))
  280. } else {
  281. e.w.writen1(bd | 0x03)
  282. bigenHelper{e.b[:8], e.w}.writeUint64(uint64(v))
  283. }
  284. }
  285. //------------------------------------
  286. type bincDecSymbol struct {
  287. s string
  288. b []byte
  289. i uint16
  290. }
  291. type bincDecDriver struct {
  292. d *Decoder
  293. h *BincHandle
  294. r decReader
  295. br bool // bytes reader
  296. bdRead bool
  297. bd byte
  298. vd byte
  299. vs byte
  300. // noStreamingCodec
  301. // decNoSeparator
  302. b [scratchByteArrayLen]byte
  303. // linear searching on this slice is ok,
  304. // because we typically expect < 32 symbols in each stream.
  305. s []bincDecSymbol
  306. decDriverNoopContainerReader
  307. }
  308. func (d *bincDecDriver) readNextBd() {
  309. d.bd = d.r.readn1()
  310. d.vd = d.bd >> 4
  311. d.vs = d.bd & 0x0f
  312. d.bdRead = true
  313. }
  314. func (d *bincDecDriver) uncacheRead() {
  315. if d.bdRead {
  316. d.r.unreadn1()
  317. d.bdRead = false
  318. }
  319. }
  320. func (d *bincDecDriver) ContainerType() (vt valueType) {
  321. if !d.bdRead {
  322. d.readNextBd()
  323. }
  324. if d.vd == bincVdSpecial && d.vs == bincSpNil {
  325. return valueTypeNil
  326. } else if d.vd == bincVdByteArray {
  327. return valueTypeBytes
  328. } else if d.vd == bincVdString {
  329. return valueTypeString
  330. } else if d.vd == bincVdArray {
  331. return valueTypeArray
  332. } else if d.vd == bincVdMap {
  333. return valueTypeMap
  334. } else {
  335. // d.d.errorf("isContainerType: unsupported parameter: %v", vt)
  336. }
  337. return valueTypeUnset
  338. }
  339. func (d *bincDecDriver) TryDecodeAsNil() bool {
  340. if !d.bdRead {
  341. d.readNextBd()
  342. }
  343. if d.bd == bincVdSpecial<<4|bincSpNil {
  344. d.bdRead = false
  345. return true
  346. }
  347. return false
  348. }
  349. func (d *bincDecDriver) IsBuiltinType(rt uintptr) bool {
  350. return rt == timeTypId
  351. }
  352. func (d *bincDecDriver) DecodeBuiltin(rt uintptr, v interface{}) {
  353. if !d.bdRead {
  354. d.readNextBd()
  355. }
  356. if rt == timeTypId {
  357. if d.vd != bincVdTimestamp {
  358. d.d.errorf("Invalid d.vd. Expecting 0x%x. Received: 0x%x", bincVdTimestamp, d.vd)
  359. return
  360. }
  361. tt, err := decodeTime(d.r.readx(int(d.vs)))
  362. if err != nil {
  363. panic(err)
  364. }
  365. var vt *time.Time = v.(*time.Time)
  366. *vt = tt
  367. d.bdRead = false
  368. }
  369. }
  370. func (d *bincDecDriver) decFloatPre(vs, defaultLen byte) {
  371. if vs&0x8 == 0 {
  372. d.r.readb(d.b[0:defaultLen])
  373. } else {
  374. l := d.r.readn1()
  375. if l > 8 {
  376. d.d.errorf("At most 8 bytes used to represent float. Received: %v bytes", l)
  377. return
  378. }
  379. for i := l; i < 8; i++ {
  380. d.b[i] = 0
  381. }
  382. d.r.readb(d.b[0:l])
  383. }
  384. }
  385. func (d *bincDecDriver) decFloat() (f float64) {
  386. //if true { f = math.Float64frombits(bigen.Uint64(d.r.readx(8))); break; }
  387. if x := d.vs & 0x7; x == bincFlBin32 {
  388. d.decFloatPre(d.vs, 4)
  389. f = float64(math.Float32frombits(bigen.Uint32(d.b[0:4])))
  390. } else if x == bincFlBin64 {
  391. d.decFloatPre(d.vs, 8)
  392. f = math.Float64frombits(bigen.Uint64(d.b[0:8]))
  393. } else {
  394. d.d.errorf("only float32 and float64 are supported. d.vd: 0x%x, d.vs: 0x%x", d.vd, d.vs)
  395. return
  396. }
  397. return
  398. }
  399. func (d *bincDecDriver) decUint() (v uint64) {
  400. // need to inline the code (interface conversion and type assertion expensive)
  401. switch d.vs {
  402. case 0:
  403. v = uint64(d.r.readn1())
  404. case 1:
  405. d.r.readb(d.b[6:8])
  406. v = uint64(bigen.Uint16(d.b[6:8]))
  407. case 2:
  408. d.b[4] = 0
  409. d.r.readb(d.b[5:8])
  410. v = uint64(bigen.Uint32(d.b[4:8]))
  411. case 3:
  412. d.r.readb(d.b[4:8])
  413. v = uint64(bigen.Uint32(d.b[4:8]))
  414. case 4, 5, 6:
  415. lim := int(7 - d.vs)
  416. d.r.readb(d.b[lim:8])
  417. for i := 0; i < lim; i++ {
  418. d.b[i] = 0
  419. }
  420. v = uint64(bigen.Uint64(d.b[:8]))
  421. case 7:
  422. d.r.readb(d.b[:8])
  423. v = uint64(bigen.Uint64(d.b[:8]))
  424. default:
  425. d.d.errorf("unsigned integers with greater than 64 bits of precision not supported")
  426. return
  427. }
  428. return
  429. }
  430. func (d *bincDecDriver) decCheckInteger() (ui uint64, neg bool) {
  431. if !d.bdRead {
  432. d.readNextBd()
  433. }
  434. vd, vs := d.vd, d.vs
  435. if vd == bincVdPosInt {
  436. ui = d.decUint()
  437. } else if vd == bincVdNegInt {
  438. ui = d.decUint()
  439. neg = true
  440. } else if vd == bincVdSmallInt {
  441. ui = uint64(d.vs) + 1
  442. } else if vd == bincVdSpecial {
  443. if vs == bincSpZero {
  444. //i = 0
  445. } else if vs == bincSpNegOne {
  446. neg = true
  447. ui = 1
  448. } else {
  449. d.d.errorf("numeric decode fails for special value: d.vs: 0x%x", d.vs)
  450. return
  451. }
  452. } else {
  453. d.d.errorf("number can only be decoded from uint or int values. d.bd: 0x%x, d.vd: 0x%x", d.bd, d.vd)
  454. return
  455. }
  456. return
  457. }
  458. func (d *bincDecDriver) DecodeInt(bitsize uint8) (i int64) {
  459. ui, neg := d.decCheckInteger()
  460. i, overflow := chkOvf.SignedInt(ui)
  461. if overflow {
  462. d.d.errorf("simple: overflow converting %v to signed integer", ui)
  463. return
  464. }
  465. if neg {
  466. i = -i
  467. }
  468. if chkOvf.Int(i, bitsize) {
  469. d.d.errorf("binc: overflow integer: %v for num bits: %v", i, bitsize)
  470. return
  471. }
  472. d.bdRead = false
  473. return
  474. }
  475. func (d *bincDecDriver) DecodeUint(bitsize uint8) (ui uint64) {
  476. ui, neg := d.decCheckInteger()
  477. if neg {
  478. d.d.errorf("Assigning negative signed value to unsigned type")
  479. return
  480. }
  481. if chkOvf.Uint(ui, bitsize) {
  482. d.d.errorf("binc: overflow integer: %v", ui)
  483. return
  484. }
  485. d.bdRead = false
  486. return
  487. }
  488. func (d *bincDecDriver) DecodeFloat(chkOverflow32 bool) (f float64) {
  489. if !d.bdRead {
  490. d.readNextBd()
  491. }
  492. vd, vs := d.vd, d.vs
  493. if vd == bincVdSpecial {
  494. d.bdRead = false
  495. if vs == bincSpNan {
  496. return math.NaN()
  497. } else if vs == bincSpPosInf {
  498. return math.Inf(1)
  499. } else if vs == bincSpZeroFloat || vs == bincSpZero {
  500. return
  501. } else if vs == bincSpNegInf {
  502. return math.Inf(-1)
  503. } else {
  504. d.d.errorf("Invalid d.vs decoding float where d.vd=bincVdSpecial: %v", d.vs)
  505. return
  506. }
  507. } else if vd == bincVdFloat {
  508. f = d.decFloat()
  509. } else {
  510. f = float64(d.DecodeInt(64))
  511. }
  512. if chkOverflow32 && chkOvf.Float32(f) {
  513. d.d.errorf("binc: float32 overflow: %v", f)
  514. return
  515. }
  516. d.bdRead = false
  517. return
  518. }
  519. // bool can be decoded from bool only (single byte).
  520. func (d *bincDecDriver) DecodeBool() (b bool) {
  521. if !d.bdRead {
  522. d.readNextBd()
  523. }
  524. if bd := d.bd; bd == (bincVdSpecial | bincSpFalse) {
  525. // b = false
  526. } else if bd == (bincVdSpecial | bincSpTrue) {
  527. b = true
  528. } else {
  529. d.d.errorf("Invalid single-byte value for bool: %s: %x", msgBadDesc, d.bd)
  530. return
  531. }
  532. d.bdRead = false
  533. return
  534. }
  535. func (d *bincDecDriver) ReadMapStart() (length int) {
  536. if !d.bdRead {
  537. d.readNextBd()
  538. }
  539. if d.vd != bincVdMap {
  540. d.d.errorf("Invalid d.vd for map. Expecting 0x%x. Got: 0x%x", bincVdMap, d.vd)
  541. return
  542. }
  543. length = d.decLen()
  544. d.bdRead = false
  545. return
  546. }
  547. func (d *bincDecDriver) ReadArrayStart() (length int) {
  548. if !d.bdRead {
  549. d.readNextBd()
  550. }
  551. if d.vd != bincVdArray {
  552. d.d.errorf("Invalid d.vd for array. Expecting 0x%x. Got: 0x%x", bincVdArray, d.vd)
  553. return
  554. }
  555. length = d.decLen()
  556. d.bdRead = false
  557. return
  558. }
  559. func (d *bincDecDriver) decLen() int {
  560. if d.vs > 3 {
  561. return int(d.vs - 4)
  562. }
  563. return int(d.decLenNumber())
  564. }
  565. func (d *bincDecDriver) decLenNumber() (v uint64) {
  566. if x := d.vs; x == 0 {
  567. v = uint64(d.r.readn1())
  568. } else if x == 1 {
  569. d.r.readb(d.b[6:8])
  570. v = uint64(bigen.Uint16(d.b[6:8]))
  571. } else if x == 2 {
  572. d.r.readb(d.b[4:8])
  573. v = uint64(bigen.Uint32(d.b[4:8]))
  574. } else {
  575. d.r.readb(d.b[:8])
  576. v = bigen.Uint64(d.b[:8])
  577. }
  578. return
  579. }
  580. func (d *bincDecDriver) decStringAndBytes(bs []byte, withString, zerocopy bool) (bs2 []byte, s string) {
  581. if !d.bdRead {
  582. d.readNextBd()
  583. }
  584. if d.bd == bincVdSpecial<<4|bincSpNil {
  585. d.bdRead = false
  586. return
  587. }
  588. var slen int = -1
  589. // var ok bool
  590. switch d.vd {
  591. case bincVdString, bincVdByteArray:
  592. slen = d.decLen()
  593. if zerocopy {
  594. if d.br {
  595. bs2 = d.r.readx(slen)
  596. } else if len(bs) == 0 {
  597. bs2 = decByteSlice(d.r, slen, d.d.h.MaxInitLen, d.b[:])
  598. } else {
  599. bs2 = decByteSlice(d.r, slen, d.d.h.MaxInitLen, bs)
  600. }
  601. } else {
  602. bs2 = decByteSlice(d.r, slen, d.d.h.MaxInitLen, bs)
  603. }
  604. if withString {
  605. s = string(bs2)
  606. }
  607. case bincVdSymbol:
  608. // zerocopy doesn't apply for symbols,
  609. // as the values must be stored in a table for later use.
  610. //
  611. //from vs: extract numSymbolBytes, containsStringVal, strLenPrecision,
  612. //extract symbol
  613. //if containsStringVal, read it and put in map
  614. //else look in map for string value
  615. var symbol uint16
  616. vs := d.vs
  617. if vs&0x8 == 0 {
  618. symbol = uint16(d.r.readn1())
  619. } else {
  620. symbol = uint16(bigen.Uint16(d.r.readx(2)))
  621. }
  622. if d.s == nil {
  623. d.s = make([]bincDecSymbol, 0, 16)
  624. }
  625. if vs&0x4 == 0 {
  626. for i := range d.s {
  627. j := &d.s[i]
  628. if j.i == symbol {
  629. bs2 = j.b
  630. if withString {
  631. if j.s == "" && bs2 != nil {
  632. j.s = string(bs2)
  633. }
  634. s = j.s
  635. }
  636. break
  637. }
  638. }
  639. } else {
  640. switch vs & 0x3 {
  641. case 0:
  642. slen = int(d.r.readn1())
  643. case 1:
  644. slen = int(bigen.Uint16(d.r.readx(2)))
  645. case 2:
  646. slen = int(bigen.Uint32(d.r.readx(4)))
  647. case 3:
  648. slen = int(bigen.Uint64(d.r.readx(8)))
  649. }
  650. // since using symbols, do not store any part of
  651. // the parameter bs in the map, as it might be a shared buffer.
  652. // bs2 = decByteSlice(d.r, slen, bs)
  653. bs2 = decByteSlice(d.r, slen, d.d.h.MaxInitLen, nil)
  654. if withString {
  655. s = string(bs2)
  656. }
  657. d.s = append(d.s, bincDecSymbol{i: symbol, s: s, b: bs2})
  658. }
  659. default:
  660. d.d.errorf("Invalid d.vd. Expecting string:0x%x, bytearray:0x%x or symbol: 0x%x. Got: 0x%x",
  661. bincVdString, bincVdByteArray, bincVdSymbol, d.vd)
  662. return
  663. }
  664. d.bdRead = false
  665. return
  666. }
  667. func (d *bincDecDriver) DecodeString() (s string) {
  668. // DecodeBytes does not accommodate symbols, whose impl stores string version in map.
  669. // Use decStringAndBytes directly.
  670. // return string(d.DecodeBytes(d.b[:], true, true))
  671. _, s = d.decStringAndBytes(d.b[:], true, true)
  672. return
  673. }
  674. func (d *bincDecDriver) DecodeStringAsBytes() (s []byte) {
  675. s, _ = d.decStringAndBytes(d.b[:], false, true)
  676. return
  677. }
  678. func (d *bincDecDriver) DecodeBytes(bs []byte, zerocopy bool) (bsOut []byte) {
  679. if !d.bdRead {
  680. d.readNextBd()
  681. }
  682. if d.bd == bincVdSpecial<<4|bincSpNil {
  683. d.bdRead = false
  684. return nil
  685. }
  686. var clen int
  687. if d.vd == bincVdString || d.vd == bincVdByteArray {
  688. clen = d.decLen()
  689. } else {
  690. d.d.errorf("Invalid d.vd for bytes. Expecting string:0x%x or bytearray:0x%x. Got: 0x%x",
  691. bincVdString, bincVdByteArray, d.vd)
  692. return
  693. }
  694. d.bdRead = false
  695. if zerocopy {
  696. if d.br {
  697. return d.r.readx(clen)
  698. } else if len(bs) == 0 {
  699. bs = d.b[:]
  700. }
  701. }
  702. return decByteSlice(d.r, clen, d.d.h.MaxInitLen, bs)
  703. }
  704. func (d *bincDecDriver) DecodeExt(rv interface{}, xtag uint64, ext Ext) (realxtag uint64) {
  705. if xtag > 0xff {
  706. d.d.errorf("decodeExt: tag must be <= 0xff; got: %v", xtag)
  707. return
  708. }
  709. realxtag1, xbs := d.decodeExtV(ext != nil, uint8(xtag))
  710. realxtag = uint64(realxtag1)
  711. if ext == nil {
  712. re := rv.(*RawExt)
  713. re.Tag = realxtag
  714. re.Data = detachZeroCopyBytes(d.br, re.Data, xbs)
  715. } else {
  716. ext.ReadExt(rv, xbs)
  717. }
  718. return
  719. }
  720. func (d *bincDecDriver) decodeExtV(verifyTag bool, tag byte) (xtag byte, xbs []byte) {
  721. if !d.bdRead {
  722. d.readNextBd()
  723. }
  724. if d.vd == bincVdCustomExt {
  725. l := d.decLen()
  726. xtag = d.r.readn1()
  727. if verifyTag && xtag != tag {
  728. d.d.errorf("Wrong extension tag. Got %b. Expecting: %v", xtag, tag)
  729. return
  730. }
  731. xbs = d.r.readx(l)
  732. } else if d.vd == bincVdByteArray {
  733. xbs = d.DecodeBytes(nil, true)
  734. } else {
  735. d.d.errorf("Invalid d.vd for extensions (Expecting extensions or byte array). Got: 0x%x", d.vd)
  736. return
  737. }
  738. d.bdRead = false
  739. return
  740. }
  741. func (d *bincDecDriver) DecodeNaked() {
  742. if !d.bdRead {
  743. d.readNextBd()
  744. }
  745. n := d.d.n
  746. var decodeFurther bool
  747. switch d.vd {
  748. case bincVdSpecial:
  749. switch d.vs {
  750. case bincSpNil:
  751. n.v = valueTypeNil
  752. case bincSpFalse:
  753. n.v = valueTypeBool
  754. n.b = false
  755. case bincSpTrue:
  756. n.v = valueTypeBool
  757. n.b = true
  758. case bincSpNan:
  759. n.v = valueTypeFloat
  760. n.f = math.NaN()
  761. case bincSpPosInf:
  762. n.v = valueTypeFloat
  763. n.f = math.Inf(1)
  764. case bincSpNegInf:
  765. n.v = valueTypeFloat
  766. n.f = math.Inf(-1)
  767. case bincSpZeroFloat:
  768. n.v = valueTypeFloat
  769. n.f = float64(0)
  770. case bincSpZero:
  771. n.v = valueTypeUint
  772. n.u = uint64(0) // int8(0)
  773. case bincSpNegOne:
  774. n.v = valueTypeInt
  775. n.i = int64(-1) // int8(-1)
  776. default:
  777. d.d.errorf("decodeNaked: Unrecognized special value 0x%x", d.vs)
  778. }
  779. case bincVdSmallInt:
  780. n.v = valueTypeUint
  781. n.u = uint64(int8(d.vs)) + 1 // int8(d.vs) + 1
  782. case bincVdPosInt:
  783. n.v = valueTypeUint
  784. n.u = d.decUint()
  785. case bincVdNegInt:
  786. n.v = valueTypeInt
  787. n.i = -(int64(d.decUint()))
  788. case bincVdFloat:
  789. n.v = valueTypeFloat
  790. n.f = d.decFloat()
  791. case bincVdSymbol:
  792. n.v = valueTypeSymbol
  793. n.s = d.DecodeString()
  794. case bincVdString:
  795. n.v = valueTypeString
  796. n.s = d.DecodeString()
  797. case bincVdByteArray:
  798. n.v = valueTypeBytes
  799. n.l = d.DecodeBytes(nil, false)
  800. case bincVdTimestamp:
  801. n.v = valueTypeTimestamp
  802. tt, err := decodeTime(d.r.readx(int(d.vs)))
  803. if err != nil {
  804. panic(err)
  805. }
  806. n.t = tt
  807. case bincVdCustomExt:
  808. n.v = valueTypeExt
  809. l := d.decLen()
  810. n.u = uint64(d.r.readn1())
  811. n.l = d.r.readx(l)
  812. case bincVdArray:
  813. n.v = valueTypeArray
  814. decodeFurther = true
  815. case bincVdMap:
  816. n.v = valueTypeMap
  817. decodeFurther = true
  818. default:
  819. d.d.errorf("decodeNaked: Unrecognized d.vd: 0x%x", d.vd)
  820. }
  821. if !decodeFurther {
  822. d.bdRead = false
  823. }
  824. if n.v == valueTypeUint && d.h.SignedInteger {
  825. n.v = valueTypeInt
  826. n.i = int64(n.u)
  827. }
  828. return
  829. }
  830. //------------------------------------
  831. //BincHandle is a Handle for the Binc Schema-Free Encoding Format
  832. //defined at https://github.com/ugorji/binc .
  833. //
  834. //BincHandle currently supports all Binc features with the following EXCEPTIONS:
  835. // - only integers up to 64 bits of precision are supported.
  836. // big integers are unsupported.
  837. // - Only IEEE 754 binary32 and binary64 floats are supported (ie Go float32 and float64 types).
  838. // extended precision and decimal IEEE 754 floats are unsupported.
  839. // - Only UTF-8 strings supported.
  840. // Unicode_Other Binc types (UTF16, UTF32) are currently unsupported.
  841. //
  842. //Note that these EXCEPTIONS are temporary and full support is possible and may happen soon.
  843. type BincHandle struct {
  844. BasicHandle
  845. binaryEncodingType
  846. noElemSeparators
  847. }
  848. func (h *BincHandle) SetBytesExt(rt reflect.Type, tag uint64, ext BytesExt) (err error) {
  849. return h.SetExt(rt, tag, &setExtWrapper{b: ext})
  850. }
  851. func (h *BincHandle) newEncDriver(e *Encoder) encDriver {
  852. return &bincEncDriver{e: e, w: e.w}
  853. }
  854. func (h *BincHandle) newDecDriver(d *Decoder) decDriver {
  855. return &bincDecDriver{d: d, h: h, r: d.r, br: d.bytes}
  856. }
  857. func (_ *BincHandle) IsBuiltinType(rt uintptr) bool {
  858. return rt == timeTypId
  859. }
  860. func (e *bincEncDriver) reset() {
  861. e.w = e.e.w
  862. e.s = 0
  863. e.m = nil
  864. }
  865. func (d *bincDecDriver) reset() {
  866. d.r, d.br = d.d.r, d.d.bytes
  867. d.s = nil
  868. d.bd, d.bdRead, d.vd, d.vs = 0, false, 0, 0
  869. }
  870. var _ decDriver = (*bincDecDriver)(nil)
  871. var _ encDriver = (*bincEncDriver)(nil)