cbor.go 20 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898
  1. // Copyright (c) 2012-2018 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. "time"
  7. )
  8. // major
  9. const (
  10. cborMajorUint byte = iota
  11. cborMajorNegInt
  12. cborMajorBytes
  13. cborMajorString
  14. cborMajorArray
  15. cborMajorMap
  16. cborMajorTag
  17. cborMajorSimpleOrFloat
  18. )
  19. // simple
  20. const (
  21. cborBdFalse byte = 0xf4 + iota
  22. cborBdTrue
  23. cborBdNil
  24. cborBdUndefined
  25. cborBdExt
  26. cborBdFloat16
  27. cborBdFloat32
  28. cborBdFloat64
  29. )
  30. // indefinite
  31. const (
  32. cborBdIndefiniteBytes byte = 0x5f
  33. cborBdIndefiniteString byte = 0x7f
  34. cborBdIndefiniteArray byte = 0x9f
  35. cborBdIndefiniteMap byte = 0xbf
  36. cborBdBreak byte = 0xff
  37. )
  38. // These define some in-stream descriptors for
  39. // manual encoding e.g. when doing explicit indefinite-length
  40. const (
  41. CborStreamBytes byte = 0x5f
  42. CborStreamString byte = 0x7f
  43. CborStreamArray byte = 0x9f
  44. CborStreamMap byte = 0xbf
  45. CborStreamBreak byte = 0xff
  46. )
  47. // base values
  48. const (
  49. cborBaseUint byte = 0x00
  50. cborBaseNegInt byte = 0x20
  51. cborBaseBytes byte = 0x40
  52. cborBaseString byte = 0x60
  53. cborBaseArray byte = 0x80
  54. cborBaseMap byte = 0xa0
  55. cborBaseTag byte = 0xc0
  56. cborBaseSimple byte = 0xe0
  57. )
  58. // const (
  59. // cborSelfDesrTag byte = 0xd9
  60. // cborSelfDesrTag2 byte = 0xd9
  61. // cborSelfDesrTag3 byte = 0xf7
  62. // )
  63. func cbordesc(bd byte) string {
  64. switch bd >> 5 {
  65. case cborMajorUint:
  66. return "(u)int"
  67. case cborMajorNegInt:
  68. return "int"
  69. case cborMajorBytes:
  70. return "bytes"
  71. case cborMajorString:
  72. return "string"
  73. case cborMajorArray:
  74. return "array"
  75. case cborMajorMap:
  76. return "map"
  77. case cborMajorTag:
  78. return "tag"
  79. case cborMajorSimpleOrFloat: // default
  80. switch bd {
  81. case cborBdNil:
  82. return "nil"
  83. case cborBdFalse:
  84. return "false"
  85. case cborBdTrue:
  86. return "true"
  87. case cborBdFloat16, cborBdFloat32, cborBdFloat64:
  88. return "float"
  89. case cborBdIndefiniteBytes:
  90. return "bytes*"
  91. case cborBdIndefiniteString:
  92. return "string*"
  93. case cborBdIndefiniteArray:
  94. return "array*"
  95. case cborBdIndefiniteMap:
  96. return "map*"
  97. default:
  98. return "unknown(simple)"
  99. }
  100. }
  101. return "unknown"
  102. }
  103. // -------------------
  104. type cborEncDriver struct {
  105. noBuiltInTypes
  106. encDriverNoopContainerWriter
  107. h *CborHandle
  108. x [8]byte
  109. _ [6]uint64 // padding
  110. e Encoder
  111. }
  112. func (e *cborEncDriver) encoder() *Encoder {
  113. return &e.e
  114. }
  115. func (e *cborEncDriver) EncodeNil() {
  116. e.e.encWr.writen1(cborBdNil)
  117. }
  118. func (e *cborEncDriver) EncodeBool(b bool) {
  119. if b {
  120. e.e.encWr.writen1(cborBdTrue)
  121. } else {
  122. e.e.encWr.writen1(cborBdFalse)
  123. }
  124. }
  125. func (e *cborEncDriver) EncodeFloat32(f float32) {
  126. e.e.encWr.writen1(cborBdFloat32)
  127. bigenHelper{e.x[:4], e.e.w()}.writeUint32(math.Float32bits(f))
  128. }
  129. func (e *cborEncDriver) EncodeFloat64(f float64) {
  130. e.e.encWr.writen1(cborBdFloat64)
  131. bigenHelper{e.x[:8], e.e.w()}.writeUint64(math.Float64bits(f))
  132. }
  133. func (e *cborEncDriver) encUint(v uint64, bd byte) {
  134. if v <= 0x17 {
  135. e.e.encWr.writen1(byte(v) + bd)
  136. } else if v <= math.MaxUint8 {
  137. e.e.encWr.writen2(bd+0x18, uint8(v))
  138. } else if v <= math.MaxUint16 {
  139. e.e.encWr.writen1(bd + 0x19)
  140. bigenHelper{e.x[:2], e.e.w()}.writeUint16(uint16(v))
  141. } else if v <= math.MaxUint32 {
  142. e.e.encWr.writen1(bd + 0x1a)
  143. bigenHelper{e.x[:4], e.e.w()}.writeUint32(uint32(v))
  144. } else { // if v <= math.MaxUint64 {
  145. e.e.encWr.writen1(bd + 0x1b)
  146. bigenHelper{e.x[:8], e.e.w()}.writeUint64(v)
  147. }
  148. }
  149. func (e *cborEncDriver) EncodeInt(v int64) {
  150. if v < 0 {
  151. e.encUint(uint64(-1-v), cborBaseNegInt)
  152. } else {
  153. e.encUint(uint64(v), cborBaseUint)
  154. }
  155. }
  156. func (e *cborEncDriver) EncodeUint(v uint64) {
  157. e.encUint(v, cborBaseUint)
  158. }
  159. func (e *cborEncDriver) encLen(bd byte, length int) {
  160. e.encUint(uint64(length), bd)
  161. }
  162. func (e *cborEncDriver) EncodeTime(t time.Time) {
  163. if t.IsZero() {
  164. e.EncodeNil()
  165. } else if e.h.TimeRFC3339 {
  166. e.encUint(0, cborBaseTag)
  167. e.EncodeStringEnc(cUTF8, t.Format(time.RFC3339Nano))
  168. } else {
  169. e.encUint(1, cborBaseTag)
  170. t = t.UTC().Round(time.Microsecond)
  171. sec, nsec := t.Unix(), uint64(t.Nanosecond())
  172. if nsec == 0 {
  173. e.EncodeInt(sec)
  174. } else {
  175. e.EncodeFloat64(float64(sec) + float64(nsec)/1e9)
  176. }
  177. }
  178. }
  179. func (e *cborEncDriver) EncodeExt(rv interface{}, xtag uint64, ext Ext) {
  180. e.encUint(uint64(xtag), cborBaseTag)
  181. if ext == SelfExt {
  182. rv2 := baseRV(rv)
  183. e.e.encodeValue(rv2, e.h.fnNoExt(rv2.Type()))
  184. } else if v := ext.ConvertExt(rv); v == nil {
  185. e.EncodeNil()
  186. } else {
  187. e.e.encode(v)
  188. }
  189. }
  190. func (e *cborEncDriver) EncodeRawExt(re *RawExt) {
  191. e.encUint(uint64(re.Tag), cborBaseTag)
  192. // only encodes re.Value (never re.Data)
  193. // if false && re.Data != nil {
  194. // en.encode(re.Data)
  195. // } else if re.Value != nil {
  196. if re.Value != nil {
  197. e.e.encode(re.Value)
  198. } else {
  199. e.EncodeNil()
  200. }
  201. }
  202. func (e *cborEncDriver) WriteArrayStart(length int) {
  203. if e.h.IndefiniteLength {
  204. e.e.encWr.writen1(cborBdIndefiniteArray)
  205. } else {
  206. e.encLen(cborBaseArray, length)
  207. }
  208. }
  209. func (e *cborEncDriver) WriteMapStart(length int) {
  210. if e.h.IndefiniteLength {
  211. e.e.encWr.writen1(cborBdIndefiniteMap)
  212. } else {
  213. e.encLen(cborBaseMap, length)
  214. }
  215. }
  216. func (e *cborEncDriver) WriteMapEnd() {
  217. if e.h.IndefiniteLength {
  218. e.e.encWr.writen1(cborBdBreak)
  219. }
  220. }
  221. func (e *cborEncDriver) WriteArrayEnd() {
  222. if e.h.IndefiniteLength {
  223. e.e.encWr.writen1(cborBdBreak)
  224. }
  225. }
  226. func (e *cborEncDriver) EncodeStringEnc(c charEncoding, v string) {
  227. e.encStringBytesS(cborBaseString, v)
  228. }
  229. func (e *cborEncDriver) EncodeStringBytesRaw(v []byte) {
  230. if v == nil {
  231. e.EncodeNil()
  232. } else {
  233. e.encStringBytesS(cborBaseBytes, stringView(v))
  234. }
  235. }
  236. func (e *cborEncDriver) encStringBytesS(bb byte, v string) {
  237. if e.h.IndefiniteLength {
  238. if bb == cborBaseBytes {
  239. e.e.encWr.writen1(cborBdIndefiniteBytes)
  240. } else {
  241. e.e.encWr.writen1(cborBdIndefiniteString)
  242. }
  243. var vlen uint = uint(len(v))
  244. blen := vlen / 4
  245. if blen == 0 {
  246. blen = 64
  247. } else if blen > 1024 {
  248. blen = 1024
  249. }
  250. for i := uint(0); i < vlen; {
  251. var v2 string
  252. i2 := i + blen
  253. if i2 >= i && i2 < vlen {
  254. v2 = v[i:i2]
  255. } else {
  256. v2 = v[i:]
  257. }
  258. e.encLen(bb, len(v2))
  259. e.e.encWr.writestr(v2)
  260. i = i2
  261. }
  262. e.e.encWr.writen1(cborBdBreak)
  263. } else {
  264. e.encLen(bb, len(v))
  265. e.e.encWr.writestr(v)
  266. }
  267. }
  268. // ----------------------
  269. type cborDecDriver struct {
  270. decDriverNoopContainerReader
  271. h *CborHandle
  272. bdRead bool
  273. bd byte
  274. st bool // skip tags
  275. fnil bool // found nil
  276. noBuiltInTypes
  277. // decNoSeparator
  278. _ [6]uint64 // padding cache-aligned
  279. d Decoder
  280. }
  281. // func (d *cborDecDriver) readNextBdSkipTags() {
  282. // d.bd = d.d.decRd.readn1()
  283. // if d.h.SkipUnexpectedTags {
  284. // for d.bd >= cborBaseTag && d.bd < cborBaseSimple {
  285. // d.decUint()
  286. // d.bd = d.d.decRd.readn1()
  287. // }
  288. // }
  289. // d.bdRead = true
  290. // }
  291. // func (d *cborDecDriver) readNextBd() {
  292. // d.bd = d.d.decRd.readn1()
  293. // if d.handleCborSelfDesc && d.bd == cborSelfDesrTag {
  294. // if x := d.readn1(); x == cborSelfDesrTag2 {
  295. // if x = d.readn1(); x != cborSelfDesrTag3 {
  296. // d.d.errorf("mishandled self desc: expected 0xd9d9f7, got: 0xd9d9%x", x)
  297. // }
  298. // } else {
  299. // d.unreadn1()
  300. // }
  301. // }
  302. // d.bdRead = true
  303. // }
  304. func (d *cborDecDriver) decoder() *Decoder {
  305. return &d.d
  306. }
  307. func (d *cborDecDriver) readNextBd() {
  308. d.bd = d.d.decRd.readn1()
  309. d.bdRead = true
  310. }
  311. func (d *cborDecDriver) advanceNil() (null bool) {
  312. d.fnil = false
  313. if !d.bdRead {
  314. d.readNextBd()
  315. }
  316. if d.bd == cborBdNil || d.bd == cborBdUndefined {
  317. d.bdRead = false
  318. d.fnil = true
  319. null = true
  320. }
  321. return
  322. }
  323. // skipTags is called to skip any tags in the stream.
  324. //
  325. // Since any value can be tagged, then we should call skipTags
  326. // before any value is decoded.
  327. //
  328. // By definition, skipTags should not be called before
  329. // checking for break, or nil or undefined.
  330. func (d *cborDecDriver) skipTags() {
  331. for d.bd>>5 == cborMajorTag {
  332. d.decUint()
  333. d.bd = d.d.decRd.readn1()
  334. }
  335. }
  336. func (d *cborDecDriver) uncacheRead() {
  337. if d.bdRead {
  338. d.d.decRd.unreadn1()
  339. d.bdRead = false
  340. }
  341. }
  342. func (d *cborDecDriver) ContainerType() (vt valueType) {
  343. d.fnil = false
  344. if !d.bdRead {
  345. d.readNextBd()
  346. }
  347. if d.st {
  348. d.skipTags()
  349. }
  350. if d.bd == cborBdNil {
  351. d.bdRead = false // always consume nil after seeing it in container type
  352. d.fnil = true
  353. return valueTypeNil
  354. } else if d.bd == cborBdIndefiniteBytes || (d.bd>>5 == cborMajorBytes) {
  355. return valueTypeBytes
  356. } else if d.bd == cborBdIndefiniteString || (d.bd>>5 == cborMajorString) {
  357. return valueTypeString
  358. } else if d.bd == cborBdIndefiniteArray || (d.bd>>5 == cborMajorArray) {
  359. return valueTypeArray
  360. } else if d.bd == cborBdIndefiniteMap || (d.bd>>5 == cborMajorMap) {
  361. return valueTypeMap
  362. }
  363. // else {
  364. // d.d.errorf("isContainerType: unsupported parameter: %v", vt)
  365. // }
  366. return valueTypeUnset
  367. }
  368. func (d *cborDecDriver) Nil() bool {
  369. return d.fnil
  370. }
  371. func (d *cborDecDriver) TryNil() bool {
  372. return d.advanceNil()
  373. }
  374. func (d *cborDecDriver) CheckBreak() (v bool) {
  375. if !d.bdRead {
  376. d.readNextBd()
  377. }
  378. if d.bd == cborBdBreak {
  379. d.bdRead = false
  380. v = true
  381. }
  382. return
  383. }
  384. func (d *cborDecDriver) decUint() (ui uint64) {
  385. v := d.bd & 0x1f
  386. if v <= 0x17 {
  387. ui = uint64(v)
  388. } else {
  389. if v == 0x18 {
  390. ui = uint64(d.d.decRd.readn1())
  391. } else if v == 0x19 {
  392. ui = uint64(bigen.Uint16(d.d.decRd.readx(2)))
  393. } else if v == 0x1a {
  394. ui = uint64(bigen.Uint32(d.d.decRd.readx(4)))
  395. } else if v == 0x1b {
  396. ui = uint64(bigen.Uint64(d.d.decRd.readx(8)))
  397. } else {
  398. d.d.errorf("invalid descriptor decoding uint: %x/%s", d.bd, cbordesc(d.bd))
  399. return
  400. }
  401. }
  402. return
  403. }
  404. func (d *cborDecDriver) decCheckInteger() (neg bool) {
  405. if d.st {
  406. d.skipTags()
  407. }
  408. major := d.bd >> 5
  409. if major == cborMajorUint {
  410. } else if major == cborMajorNegInt {
  411. neg = true
  412. } else {
  413. d.d.errorf("invalid integer; got major %v from descriptor %x/%s, expected %v or %v",
  414. major, d.bd, cbordesc(d.bd), cborMajorUint, cborMajorNegInt)
  415. }
  416. return
  417. }
  418. func cborDecInt64(ui uint64, neg bool) (i int64) {
  419. // check if this number can be converted to an int without overflow
  420. if neg {
  421. i = -(chkOvf.SignedIntV(ui + 1))
  422. } else {
  423. i = chkOvf.SignedIntV(ui)
  424. }
  425. return
  426. }
  427. func (d *cborDecDriver) decLen() int {
  428. return int(d.decUint())
  429. }
  430. func (d *cborDecDriver) decAppendIndefiniteBytes(bs []byte) []byte {
  431. d.bdRead = false
  432. for !d.CheckBreak() {
  433. if major := d.bd >> 5; major != cborMajorBytes && major != cborMajorString {
  434. d.d.errorf("invalid indefinite string/bytes; got major %v, expected %x/%s",
  435. major, d.bd, cbordesc(d.bd))
  436. }
  437. n := uint(d.decLen())
  438. oldLen := uint(len(bs))
  439. newLen := oldLen + n
  440. if newLen > uint(cap(bs)) {
  441. bs2 := make([]byte, newLen, 2*uint(cap(bs))+n)
  442. copy(bs2, bs)
  443. bs = bs2
  444. } else {
  445. bs = bs[:newLen]
  446. }
  447. d.d.decRd.readb(bs[oldLen:newLen])
  448. // bs = append(bs, d.d.decRd.readn()...)
  449. d.bdRead = false
  450. }
  451. d.bdRead = false
  452. return bs
  453. }
  454. func (d *cborDecDriver) DecodeInt64() (i int64) {
  455. if d.advanceNil() {
  456. return
  457. }
  458. neg := d.decCheckInteger()
  459. ui := d.decUint()
  460. d.bdRead = false
  461. return cborDecInt64(ui, neg)
  462. }
  463. func (d *cborDecDriver) DecodeUint64() (ui uint64) {
  464. if d.advanceNil() {
  465. return
  466. }
  467. if d.decCheckInteger() {
  468. d.d.errorf("cannot assign negative signed value to unsigned type")
  469. }
  470. ui = d.decUint()
  471. d.bdRead = false
  472. return
  473. }
  474. func (d *cborDecDriver) DecodeFloat64() (f float64) {
  475. if d.advanceNil() {
  476. return
  477. }
  478. if d.st {
  479. d.skipTags()
  480. }
  481. switch d.bd {
  482. case cborBdFloat16:
  483. f = float64(math.Float32frombits(halfFloatToFloatBits(bigen.Uint16(d.d.decRd.readx(2)))))
  484. case cborBdFloat32:
  485. f = float64(math.Float32frombits(bigen.Uint32(d.d.decRd.readx(4))))
  486. case cborBdFloat64:
  487. f = math.Float64frombits(bigen.Uint64(d.d.decRd.readx(8)))
  488. default:
  489. major := d.bd >> 5
  490. if major == cborMajorUint {
  491. f = float64(cborDecInt64(d.decUint(), false))
  492. } else if major == cborMajorNegInt {
  493. f = float64(cborDecInt64(d.decUint(), true))
  494. } else {
  495. d.d.errorf("invalid float descriptor; got %d/%s, expected float16/32/64 or (-)int",
  496. d.bd, cbordesc(d.bd))
  497. }
  498. }
  499. d.bdRead = false
  500. return
  501. }
  502. // bool can be decoded from bool only (single byte).
  503. func (d *cborDecDriver) DecodeBool() (b bool) {
  504. if d.advanceNil() {
  505. return
  506. }
  507. if d.st {
  508. d.skipTags()
  509. }
  510. if d.bd == cborBdTrue {
  511. b = true
  512. } else if d.bd == cborBdFalse {
  513. } else {
  514. d.d.errorf("not bool - %s %x/%s", msgBadDesc, d.bd, cbordesc(d.bd))
  515. return
  516. }
  517. d.bdRead = false
  518. return
  519. }
  520. func (d *cborDecDriver) ReadMapStart() (length int) {
  521. if d.advanceNil() {
  522. return decContainerLenNil
  523. }
  524. if d.st {
  525. d.skipTags()
  526. }
  527. d.bdRead = false
  528. if d.bd == cborBdIndefiniteMap {
  529. return decContainerLenUnknown
  530. }
  531. if d.bd>>5 != cborMajorMap {
  532. d.d.errorf("error reading map; got major type: %x, expected %x/%s",
  533. d.bd>>5, cborMajorMap, cbordesc(d.bd))
  534. }
  535. return d.decLen()
  536. }
  537. func (d *cborDecDriver) ReadArrayStart() (length int) {
  538. if d.advanceNil() {
  539. return decContainerLenNil
  540. }
  541. if d.st {
  542. d.skipTags()
  543. }
  544. d.bdRead = false
  545. if d.bd == cborBdIndefiniteArray {
  546. return decContainerLenUnknown
  547. }
  548. if d.bd>>5 != cborMajorArray {
  549. d.d.errorf("invalid array; got major type: %x, expect: %x/%s",
  550. d.bd>>5, cborMajorArray, cbordesc(d.bd))
  551. }
  552. return d.decLen()
  553. }
  554. func (d *cborDecDriver) DecodeBytes(bs []byte, zerocopy bool) (bsOut []byte) {
  555. if d.advanceNil() {
  556. return
  557. }
  558. if d.st {
  559. d.skipTags()
  560. }
  561. if d.bd == cborBdIndefiniteBytes || d.bd == cborBdIndefiniteString {
  562. d.bdRead = false
  563. if bs == nil {
  564. if zerocopy {
  565. return d.decAppendIndefiniteBytes(d.d.b[:0])
  566. }
  567. return d.decAppendIndefiniteBytes(zeroByteSlice)
  568. }
  569. return d.decAppendIndefiniteBytes(bs[:0])
  570. }
  571. // check if an "array" of uint8's (see ContainerType for how to infer if an array)
  572. // if d.bd == cborBdIndefiniteArray || (d.bd >> 5 == cborMajorArray) {
  573. // bsOut, _ = fastpathTV.DecSliceUint8V(bs, true, d.d)
  574. // return
  575. // }
  576. if d.bd == cborBdIndefiniteArray {
  577. d.bdRead = false
  578. if zerocopy && len(bs) == 0 {
  579. bs = d.d.b[:]
  580. }
  581. if bs == nil {
  582. bs = []byte{}
  583. } else {
  584. bs = bs[:0]
  585. }
  586. for !d.CheckBreak() {
  587. bs = append(bs, uint8(chkOvf.UintV(d.DecodeUint64(), 8)))
  588. }
  589. return bs
  590. }
  591. if d.bd>>5 == cborMajorArray {
  592. d.bdRead = false
  593. if zerocopy && len(bs) == 0 {
  594. bs = d.d.b[:]
  595. }
  596. slen := d.decLen()
  597. bs = usableByteSlice(bs, slen)
  598. for i := 0; i < len(bs); i++ {
  599. bs[i] = uint8(chkOvf.UintV(d.DecodeUint64(), 8))
  600. }
  601. return bs
  602. }
  603. clen := d.decLen()
  604. d.bdRead = false
  605. if zerocopy {
  606. if d.d.bytes {
  607. return d.d.decRd.readx(uint(clen))
  608. } else if len(bs) == 0 {
  609. bs = d.d.b[:]
  610. }
  611. }
  612. return decByteSlice(d.d.r(), clen, d.h.MaxInitLen, bs)
  613. }
  614. func (d *cborDecDriver) DecodeStringAsBytes() (s []byte) {
  615. return d.DecodeBytes(d.d.b[:], true)
  616. }
  617. func (d *cborDecDriver) DecodeTime() (t time.Time) {
  618. if d.advanceNil() {
  619. return
  620. }
  621. if d.bd>>5 != cborMajorTag {
  622. d.d.errorf("error reading tag; expected major type: %x, got: %x", cborMajorTag, d.bd>>5)
  623. }
  624. xtag := d.decUint()
  625. d.bdRead = false
  626. return d.decodeTime(xtag)
  627. }
  628. func (d *cborDecDriver) decodeTime(xtag uint64) (t time.Time) {
  629. switch xtag {
  630. case 0:
  631. var err error
  632. if t, err = time.Parse(time.RFC3339, stringView(d.DecodeStringAsBytes())); err != nil {
  633. d.d.errorv(err)
  634. }
  635. case 1:
  636. // if !d.bdRead {
  637. // d.readNextBd()
  638. // }
  639. // // decode an int64 or a float, and infer time.Time from there.
  640. // // for floats, round to microseconds, as that is what is guaranteed to fit well.
  641. // switch {
  642. // case d.bd == cborBdFloat16, d.bd == cborBdFloat32:
  643. // f1, f2 := math.Modf(d.DecodeFloat64())
  644. // t = time.Unix(int64(f1), int64(f2*1e9))
  645. // case d.bd == cborBdFloat64:
  646. // f1, f2 := math.Modf(d.DecodeFloat64())
  647. // t = time.Unix(int64(f1), int64(f2*1e9))
  648. // case d.bd >= cborBaseUint && d.bd < cborBaseBytes:
  649. // t = time.Unix(d.DecodeInt64(), 0)
  650. // default:
  651. // d.d.errorf("time.Time can only be decoded from a number (or RFC3339 string)")
  652. // }
  653. f1, f2 := math.Modf(d.DecodeFloat64())
  654. t = time.Unix(int64(f1), int64(f2*1e9))
  655. default:
  656. d.d.errorf("invalid tag for time.Time - expecting 0 or 1, got 0x%x", xtag)
  657. }
  658. t = t.UTC().Round(time.Microsecond)
  659. return
  660. }
  661. func (d *cborDecDriver) DecodeExt(rv interface{}, xtag uint64, ext Ext) {
  662. if d.advanceNil() {
  663. return
  664. }
  665. if d.bd>>5 != cborMajorTag {
  666. d.d.errorf("error reading tag; expected major type: %x, got: %x", cborMajorTag, d.bd>>5)
  667. }
  668. realxtag := d.decUint()
  669. d.bdRead = false
  670. if ext == nil {
  671. re := rv.(*RawExt)
  672. re.Tag = realxtag
  673. d.d.decode(&re.Value)
  674. } else if xtag != realxtag {
  675. d.d.errorf("Wrong extension tag. Got %b. Expecting: %v", realxtag, xtag)
  676. return
  677. } else if ext == SelfExt {
  678. rv2 := baseRV(rv)
  679. d.d.decodeValue(rv2, d.h.fnNoExt(rv2.Type()))
  680. } else {
  681. d.d.interfaceExtConvertAndDecode(rv, ext)
  682. }
  683. d.bdRead = false
  684. }
  685. func (d *cborDecDriver) DecodeNaked() {
  686. if !d.bdRead {
  687. d.readNextBd()
  688. }
  689. d.fnil = false
  690. n := d.d.naked()
  691. var decodeFurther bool
  692. switch d.bd >> 5 {
  693. case cborMajorUint:
  694. if d.h.SignedInteger {
  695. n.v = valueTypeInt
  696. n.i = d.DecodeInt64()
  697. } else {
  698. n.v = valueTypeUint
  699. n.u = d.DecodeUint64()
  700. }
  701. case cborMajorNegInt:
  702. n.v = valueTypeInt
  703. n.i = d.DecodeInt64()
  704. case cborMajorBytes:
  705. decNakedReadRawBytes(d, &d.d, n, d.h.RawToString)
  706. case cborMajorString:
  707. n.v = valueTypeString
  708. n.s = string(d.DecodeStringAsBytes())
  709. case cborMajorArray:
  710. n.v = valueTypeArray
  711. decodeFurther = true
  712. case cborMajorMap:
  713. n.v = valueTypeMap
  714. decodeFurther = true
  715. case cborMajorTag:
  716. n.v = valueTypeExt
  717. n.u = d.decUint()
  718. n.l = nil
  719. if n.u == 0 || n.u == 1 {
  720. d.bdRead = false
  721. n.v = valueTypeTime
  722. n.t = d.decodeTime(n.u)
  723. } else if d.st && d.h.getExtForTag(n.u) == nil {
  724. // d.skipTags() // no need to call this - tags already skipped
  725. d.bdRead = false
  726. d.DecodeNaked()
  727. return // return when done (as true recursive function)
  728. }
  729. // d.bdRead = false
  730. // d.d.decode(&re.Value) // handled by decode itself.
  731. // decodeFurther = true
  732. case cborMajorSimpleOrFloat:
  733. switch d.bd {
  734. case cborBdNil, cborBdUndefined:
  735. n.v = valueTypeNil
  736. d.fnil = true
  737. case cborBdFalse:
  738. n.v = valueTypeBool
  739. n.b = false
  740. case cborBdTrue:
  741. n.v = valueTypeBool
  742. n.b = true
  743. case cborBdFloat16, cborBdFloat32, cborBdFloat64:
  744. n.v = valueTypeFloat
  745. n.f = d.DecodeFloat64()
  746. case cborBdIndefiniteBytes:
  747. decNakedReadRawBytes(d, &d.d, n, d.h.RawToString)
  748. case cborBdIndefiniteString:
  749. n.v = valueTypeString
  750. n.s = string(d.DecodeStringAsBytes())
  751. case cborBdIndefiniteArray:
  752. n.v = valueTypeArray
  753. decodeFurther = true
  754. case cborBdIndefiniteMap:
  755. n.v = valueTypeMap
  756. decodeFurther = true
  757. default:
  758. d.d.errorf("decodeNaked: Unrecognized d.bd: 0x%x", d.bd)
  759. }
  760. default: // should never happen
  761. d.d.errorf("decodeNaked: Unrecognized d.bd: 0x%x", d.bd)
  762. }
  763. if !decodeFurther {
  764. d.bdRead = false
  765. }
  766. }
  767. // -------------------------
  768. // CborHandle is a Handle for the CBOR encoding format,
  769. // defined at http://tools.ietf.org/html/rfc7049 and documented further at http://cbor.io .
  770. //
  771. // CBOR is comprehensively supported, including support for:
  772. // - indefinite-length arrays/maps/bytes/strings
  773. // - (extension) tags in range 0..0xffff (0 .. 65535)
  774. // - half, single and double-precision floats
  775. // - all numbers (1, 2, 4 and 8-byte signed and unsigned integers)
  776. // - nil, true, false, ...
  777. // - arrays and maps, bytes and text strings
  778. //
  779. // None of the optional extensions (with tags) defined in the spec are supported out-of-the-box.
  780. // Users can implement them as needed (using SetExt), including spec-documented ones:
  781. // - timestamp, BigNum, BigFloat, Decimals,
  782. // - Encoded Text (e.g. URL, regexp, base64, MIME Message), etc.
  783. type CborHandle struct {
  784. binaryEncodingType
  785. // noElemSeparators
  786. BasicHandle
  787. // IndefiniteLength=true, means that we encode using indefinitelength
  788. IndefiniteLength bool
  789. // TimeRFC3339 says to encode time.Time using RFC3339 format.
  790. // If unset, we encode time.Time using seconds past epoch.
  791. TimeRFC3339 bool
  792. // SkipUnexpectedTags says to skip over any tags for which extensions are
  793. // not defined. This is in keeping with the cbor spec on "Optional Tagging of Items".
  794. //
  795. // Furthermore, this allows the skipping over of the Self Describing Tag 0xd9d9f7.
  796. SkipUnexpectedTags bool
  797. _ [7]uint64 // padding (cache-aligned)
  798. }
  799. // Name returns the name of the handle: cbor
  800. func (h *CborHandle) Name() string { return "cbor" }
  801. func (h *CborHandle) newEncDriver() encDriver {
  802. var e = &cborEncDriver{h: h}
  803. e.e.e = e
  804. e.e.init(h)
  805. e.reset()
  806. return e
  807. }
  808. func (h *CborHandle) newDecDriver() decDriver {
  809. d := &cborDecDriver{h: h, st: h.SkipUnexpectedTags}
  810. d.d.d = d
  811. d.d.init(h)
  812. d.reset()
  813. return d
  814. }
  815. func (e *cborEncDriver) reset() {
  816. }
  817. func (d *cborDecDriver) reset() {
  818. d.bd = 0
  819. d.bdRead = false
  820. d.fnil = false
  821. d.st = d.h.SkipUnexpectedTags
  822. }
  823. var _ decDriver = (*cborDecDriver)(nil)
  824. var _ encDriver = (*cborEncDriver)(nil)