cbor.go 20 KB

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