cbor.go 20 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903
  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.encStringBytesS(cborBaseString, 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) EncodeString(v string) {
  227. if e.h.StringToRaw {
  228. e.EncodeStringBytesRaw(bytesView(v))
  229. return
  230. }
  231. e.encStringBytesS(cborBaseString, v)
  232. }
  233. func (e *cborEncDriver) EncodeStringBytesRaw(v []byte) {
  234. if v == nil {
  235. e.EncodeNil()
  236. } else {
  237. e.encStringBytesS(cborBaseBytes, stringView(v))
  238. }
  239. }
  240. func (e *cborEncDriver) encStringBytesS(bb byte, v string) {
  241. if e.h.IndefiniteLength {
  242. if bb == cborBaseBytes {
  243. e.e.encWr.writen1(cborBdIndefiniteBytes)
  244. } else {
  245. e.e.encWr.writen1(cborBdIndefiniteString)
  246. }
  247. var vlen uint = uint(len(v))
  248. blen := vlen / 4
  249. if blen == 0 {
  250. blen = 64
  251. } else if blen > 1024 {
  252. blen = 1024
  253. }
  254. for i := uint(0); i < vlen; {
  255. var v2 string
  256. i2 := i + blen
  257. if i2 >= i && i2 < vlen {
  258. v2 = v[i:i2]
  259. } else {
  260. v2 = v[i:]
  261. }
  262. e.encLen(bb, len(v2))
  263. e.e.encWr.writestr(v2)
  264. i = i2
  265. }
  266. e.e.encWr.writen1(cborBdBreak)
  267. } else {
  268. e.encLen(bb, len(v))
  269. e.e.encWr.writestr(v)
  270. }
  271. }
  272. // ----------------------
  273. type cborDecDriver struct {
  274. decDriverNoopContainerReader
  275. h *CborHandle
  276. bdRead bool
  277. bd byte
  278. st bool // skip tags
  279. fnil bool // found nil
  280. noBuiltInTypes
  281. // decNoSeparator
  282. _ [6]uint64 // padding cache-aligned
  283. d Decoder
  284. }
  285. // func (d *cborDecDriver) readNextBdSkipTags() {
  286. // d.bd = d.d.decRd.readn1()
  287. // if d.h.SkipUnexpectedTags {
  288. // for d.bd >= cborBaseTag && d.bd < cborBaseSimple {
  289. // d.decUint()
  290. // d.bd = d.d.decRd.readn1()
  291. // }
  292. // }
  293. // d.bdRead = true
  294. // }
  295. // func (d *cborDecDriver) readNextBd() {
  296. // d.bd = d.d.decRd.readn1()
  297. // if d.handleCborSelfDesc && d.bd == cborSelfDesrTag {
  298. // if x := d.readn1(); x == cborSelfDesrTag2 {
  299. // if x = d.readn1(); x != cborSelfDesrTag3 {
  300. // d.d.errorf("mishandled self desc: expected 0xd9d9f7, got: 0xd9d9%x", x)
  301. // }
  302. // } else {
  303. // d.unreadn1()
  304. // }
  305. // }
  306. // d.bdRead = true
  307. // }
  308. func (d *cborDecDriver) decoder() *Decoder {
  309. return &d.d
  310. }
  311. func (d *cborDecDriver) readNextBd() {
  312. d.bd = d.d.decRd.readn1()
  313. d.bdRead = true
  314. }
  315. func (d *cborDecDriver) advanceNil() (null bool) {
  316. d.fnil = false
  317. if !d.bdRead {
  318. d.readNextBd()
  319. }
  320. if d.bd == cborBdNil || d.bd == cborBdUndefined {
  321. d.bdRead = false
  322. d.fnil = true
  323. null = true
  324. }
  325. return
  326. }
  327. // skipTags is called to skip any tags in the stream.
  328. //
  329. // Since any value can be tagged, then we should call skipTags
  330. // before any value is decoded.
  331. //
  332. // By definition, skipTags should not be called before
  333. // checking for break, or nil or undefined.
  334. func (d *cborDecDriver) skipTags() {
  335. for d.bd>>5 == cborMajorTag {
  336. d.decUint()
  337. d.bd = d.d.decRd.readn1()
  338. }
  339. }
  340. func (d *cborDecDriver) uncacheRead() {
  341. if d.bdRead {
  342. d.d.decRd.unreadn1()
  343. d.bdRead = false
  344. }
  345. }
  346. func (d *cborDecDriver) ContainerType() (vt valueType) {
  347. d.fnil = false
  348. if !d.bdRead {
  349. d.readNextBd()
  350. }
  351. if d.st {
  352. d.skipTags()
  353. }
  354. if d.bd == cborBdNil {
  355. d.bdRead = false // always consume nil after seeing it in container type
  356. d.fnil = true
  357. return valueTypeNil
  358. } else if d.bd == cborBdIndefiniteBytes || (d.bd>>5 == cborMajorBytes) {
  359. return valueTypeBytes
  360. } else if d.bd == cborBdIndefiniteString || (d.bd>>5 == cborMajorString) {
  361. return valueTypeString
  362. } else if d.bd == cborBdIndefiniteArray || (d.bd>>5 == cborMajorArray) {
  363. return valueTypeArray
  364. } else if d.bd == cborBdIndefiniteMap || (d.bd>>5 == cborMajorMap) {
  365. return valueTypeMap
  366. }
  367. // else {
  368. // d.d.errorf("isContainerType: unsupported parameter: %v", vt)
  369. // }
  370. return valueTypeUnset
  371. }
  372. func (d *cborDecDriver) Nil() bool {
  373. return d.fnil
  374. }
  375. func (d *cborDecDriver) TryNil() bool {
  376. return d.advanceNil()
  377. }
  378. func (d *cborDecDriver) CheckBreak() (v bool) {
  379. if !d.bdRead {
  380. d.readNextBd()
  381. }
  382. if d.bd == cborBdBreak {
  383. d.bdRead = false
  384. v = true
  385. }
  386. return
  387. }
  388. func (d *cborDecDriver) decUint() (ui uint64) {
  389. v := d.bd & 0x1f
  390. if v <= 0x17 {
  391. ui = uint64(v)
  392. } else {
  393. if v == 0x18 {
  394. ui = uint64(d.d.decRd.readn1())
  395. } else if v == 0x19 {
  396. ui = uint64(bigen.Uint16(d.d.decRd.readx(2)))
  397. } else if v == 0x1a {
  398. ui = uint64(bigen.Uint32(d.d.decRd.readx(4)))
  399. } else if v == 0x1b {
  400. ui = uint64(bigen.Uint64(d.d.decRd.readx(8)))
  401. } else {
  402. d.d.errorf("invalid descriptor decoding uint: %x/%s", d.bd, cbordesc(d.bd))
  403. return
  404. }
  405. }
  406. return
  407. }
  408. func (d *cborDecDriver) decCheckInteger() (neg bool) {
  409. if d.st {
  410. d.skipTags()
  411. }
  412. major := d.bd >> 5
  413. if major == cborMajorUint {
  414. } else if major == cborMajorNegInt {
  415. neg = true
  416. } else {
  417. d.d.errorf("invalid integer; got major %v from descriptor %x/%s, expected %v or %v",
  418. major, d.bd, cbordesc(d.bd), cborMajorUint, cborMajorNegInt)
  419. }
  420. return
  421. }
  422. func cborDecInt64(ui uint64, neg bool) (i int64) {
  423. // check if this number can be converted to an int without overflow
  424. if neg {
  425. i = -(chkOvf.SignedIntV(ui + 1))
  426. } else {
  427. i = chkOvf.SignedIntV(ui)
  428. }
  429. return
  430. }
  431. func (d *cborDecDriver) decLen() int {
  432. return int(d.decUint())
  433. }
  434. func (d *cborDecDriver) decAppendIndefiniteBytes(bs []byte) []byte {
  435. d.bdRead = false
  436. for !d.CheckBreak() {
  437. if major := d.bd >> 5; major != cborMajorBytes && major != cborMajorString {
  438. d.d.errorf("invalid indefinite string/bytes; got major %v, expected %x/%s",
  439. major, d.bd, cbordesc(d.bd))
  440. }
  441. n := uint(d.decLen())
  442. oldLen := uint(len(bs))
  443. newLen := oldLen + n
  444. if newLen > uint(cap(bs)) {
  445. bs2 := make([]byte, newLen, 2*uint(cap(bs))+n)
  446. copy(bs2, bs)
  447. bs = bs2
  448. } else {
  449. bs = bs[:newLen]
  450. }
  451. d.d.decRd.readb(bs[oldLen:newLen])
  452. // bs = append(bs, d.d.decRd.readn()...)
  453. d.bdRead = false
  454. }
  455. d.bdRead = false
  456. return bs
  457. }
  458. func (d *cborDecDriver) DecodeInt64() (i int64) {
  459. if d.advanceNil() {
  460. return
  461. }
  462. neg := d.decCheckInteger()
  463. ui := d.decUint()
  464. d.bdRead = false
  465. return cborDecInt64(ui, neg)
  466. }
  467. func (d *cborDecDriver) DecodeUint64() (ui uint64) {
  468. if d.advanceNil() {
  469. return
  470. }
  471. if d.decCheckInteger() {
  472. d.d.errorf("cannot assign negative signed value to unsigned type")
  473. }
  474. ui = d.decUint()
  475. d.bdRead = false
  476. return
  477. }
  478. func (d *cborDecDriver) DecodeFloat64() (f float64) {
  479. if d.advanceNil() {
  480. return
  481. }
  482. if d.st {
  483. d.skipTags()
  484. }
  485. switch d.bd {
  486. case cborBdFloat16:
  487. f = float64(math.Float32frombits(halfFloatToFloatBits(bigen.Uint16(d.d.decRd.readx(2)))))
  488. case cborBdFloat32:
  489. f = float64(math.Float32frombits(bigen.Uint32(d.d.decRd.readx(4))))
  490. case cborBdFloat64:
  491. f = math.Float64frombits(bigen.Uint64(d.d.decRd.readx(8)))
  492. default:
  493. major := d.bd >> 5
  494. if major == cborMajorUint {
  495. f = float64(cborDecInt64(d.decUint(), false))
  496. } else if major == cborMajorNegInt {
  497. f = float64(cborDecInt64(d.decUint(), true))
  498. } else {
  499. d.d.errorf("invalid float descriptor; got %d/%s, expected float16/32/64 or (-)int",
  500. d.bd, cbordesc(d.bd))
  501. }
  502. }
  503. d.bdRead = false
  504. return
  505. }
  506. // bool can be decoded from bool only (single byte).
  507. func (d *cborDecDriver) DecodeBool() (b bool) {
  508. if d.advanceNil() {
  509. return
  510. }
  511. if d.st {
  512. d.skipTags()
  513. }
  514. if d.bd == cborBdTrue {
  515. b = true
  516. } else if d.bd == cborBdFalse {
  517. } else {
  518. d.d.errorf("not bool - %s %x/%s", msgBadDesc, d.bd, cbordesc(d.bd))
  519. return
  520. }
  521. d.bdRead = false
  522. return
  523. }
  524. func (d *cborDecDriver) ReadMapStart() (length int) {
  525. if d.advanceNil() {
  526. return decContainerLenNil
  527. }
  528. if d.st {
  529. d.skipTags()
  530. }
  531. d.bdRead = false
  532. if d.bd == cborBdIndefiniteMap {
  533. return decContainerLenUnknown
  534. }
  535. if d.bd>>5 != cborMajorMap {
  536. d.d.errorf("error reading map; got major type: %x, expected %x/%s",
  537. d.bd>>5, cborMajorMap, cbordesc(d.bd))
  538. }
  539. return d.decLen()
  540. }
  541. func (d *cborDecDriver) ReadArrayStart() (length int) {
  542. if d.advanceNil() {
  543. return decContainerLenNil
  544. }
  545. if d.st {
  546. d.skipTags()
  547. }
  548. d.bdRead = false
  549. if d.bd == cborBdIndefiniteArray {
  550. return decContainerLenUnknown
  551. }
  552. if d.bd>>5 != cborMajorArray {
  553. d.d.errorf("invalid array; got major type: %x, expect: %x/%s",
  554. d.bd>>5, cborMajorArray, cbordesc(d.bd))
  555. }
  556. return d.decLen()
  557. }
  558. func (d *cborDecDriver) DecodeBytes(bs []byte, zerocopy bool) (bsOut []byte) {
  559. if d.advanceNil() {
  560. return
  561. }
  562. if d.st {
  563. d.skipTags()
  564. }
  565. if d.bd == cborBdIndefiniteBytes || d.bd == cborBdIndefiniteString {
  566. d.bdRead = false
  567. if bs == nil {
  568. if zerocopy {
  569. return d.decAppendIndefiniteBytes(d.d.b[:0])
  570. }
  571. return d.decAppendIndefiniteBytes(zeroByteSlice)
  572. }
  573. return d.decAppendIndefiniteBytes(bs[:0])
  574. }
  575. // check if an "array" of uint8's (see ContainerType for how to infer if an array)
  576. // if d.bd == cborBdIndefiniteArray || (d.bd >> 5 == cborMajorArray) {
  577. // bsOut, _ = fastpathTV.DecSliceUint8V(bs, true, d.d)
  578. // return
  579. // }
  580. if d.bd == cborBdIndefiniteArray {
  581. d.bdRead = false
  582. if zerocopy && len(bs) == 0 {
  583. bs = d.d.b[:]
  584. }
  585. if bs == nil {
  586. bs = []byte{}
  587. } else {
  588. bs = bs[:0]
  589. }
  590. for !d.CheckBreak() {
  591. bs = append(bs, uint8(chkOvf.UintV(d.DecodeUint64(), 8)))
  592. }
  593. return bs
  594. }
  595. if d.bd>>5 == cborMajorArray {
  596. d.bdRead = false
  597. if zerocopy && len(bs) == 0 {
  598. bs = d.d.b[:]
  599. }
  600. slen := d.decLen()
  601. bs = usableByteSlice(bs, slen)
  602. for i := 0; i < len(bs); i++ {
  603. bs[i] = uint8(chkOvf.UintV(d.DecodeUint64(), 8))
  604. }
  605. return bs
  606. }
  607. clen := d.decLen()
  608. d.bdRead = false
  609. if zerocopy {
  610. if d.d.bytes {
  611. return d.d.decRd.readx(uint(clen))
  612. } else if len(bs) == 0 {
  613. bs = d.d.b[:]
  614. }
  615. }
  616. return decByteSlice(d.d.r(), clen, d.h.MaxInitLen, bs)
  617. }
  618. func (d *cborDecDriver) DecodeStringAsBytes() (s []byte) {
  619. return d.DecodeBytes(d.d.b[:], true)
  620. }
  621. func (d *cborDecDriver) DecodeTime() (t time.Time) {
  622. if d.advanceNil() {
  623. return
  624. }
  625. if d.bd>>5 != cborMajorTag {
  626. d.d.errorf("error reading tag; expected major type: %x, got: %x", cborMajorTag, d.bd>>5)
  627. }
  628. xtag := d.decUint()
  629. d.bdRead = false
  630. return d.decodeTime(xtag)
  631. }
  632. func (d *cborDecDriver) decodeTime(xtag uint64) (t time.Time) {
  633. switch xtag {
  634. case 0:
  635. var err error
  636. if t, err = time.Parse(time.RFC3339, stringView(d.DecodeStringAsBytes())); err != nil {
  637. d.d.errorv(err)
  638. }
  639. case 1:
  640. // if !d.bdRead {
  641. // d.readNextBd()
  642. // }
  643. // // decode an int64 or a float, and infer time.Time from there.
  644. // // for floats, round to microseconds, as that is what is guaranteed to fit well.
  645. // switch {
  646. // case d.bd == cborBdFloat16, d.bd == cborBdFloat32:
  647. // f1, f2 := math.Modf(d.DecodeFloat64())
  648. // t = time.Unix(int64(f1), int64(f2*1e9))
  649. // case d.bd == cborBdFloat64:
  650. // f1, f2 := math.Modf(d.DecodeFloat64())
  651. // t = time.Unix(int64(f1), int64(f2*1e9))
  652. // case d.bd >= cborBaseUint && d.bd < cborBaseBytes:
  653. // t = time.Unix(d.DecodeInt64(), 0)
  654. // default:
  655. // d.d.errorf("time.Time can only be decoded from a number (or RFC3339 string)")
  656. // }
  657. f1, f2 := math.Modf(d.DecodeFloat64())
  658. t = time.Unix(int64(f1), int64(f2*1e9))
  659. default:
  660. d.d.errorf("invalid tag for time.Time - expecting 0 or 1, got 0x%x", xtag)
  661. }
  662. t = t.UTC().Round(time.Microsecond)
  663. return
  664. }
  665. func (d *cborDecDriver) DecodeExt(rv interface{}, xtag uint64, ext Ext) {
  666. if d.advanceNil() {
  667. return
  668. }
  669. if d.bd>>5 != cborMajorTag {
  670. d.d.errorf("error reading tag; expected major type: %x, got: %x", cborMajorTag, d.bd>>5)
  671. }
  672. realxtag := d.decUint()
  673. d.bdRead = false
  674. if ext == nil {
  675. re := rv.(*RawExt)
  676. re.Tag = realxtag
  677. d.d.decode(&re.Value)
  678. } else if xtag != realxtag {
  679. d.d.errorf("Wrong extension tag. Got %b. Expecting: %v", realxtag, xtag)
  680. return
  681. } else if ext == SelfExt {
  682. rv2 := baseRV(rv)
  683. d.d.decodeValue(rv2, d.h.fnNoExt(rv2.Type()))
  684. } else {
  685. d.d.interfaceExtConvertAndDecode(rv, ext)
  686. }
  687. d.bdRead = false
  688. }
  689. func (d *cborDecDriver) DecodeNaked() {
  690. if !d.bdRead {
  691. d.readNextBd()
  692. }
  693. d.fnil = false
  694. n := d.d.naked()
  695. var decodeFurther bool
  696. switch d.bd >> 5 {
  697. case cborMajorUint:
  698. if d.h.SignedInteger {
  699. n.v = valueTypeInt
  700. n.i = d.DecodeInt64()
  701. } else {
  702. n.v = valueTypeUint
  703. n.u = d.DecodeUint64()
  704. }
  705. case cborMajorNegInt:
  706. n.v = valueTypeInt
  707. n.i = d.DecodeInt64()
  708. case cborMajorBytes:
  709. decNakedReadRawBytes(d, &d.d, n, d.h.RawToString)
  710. case cborMajorString:
  711. n.v = valueTypeString
  712. n.s = string(d.DecodeStringAsBytes())
  713. case cborMajorArray:
  714. n.v = valueTypeArray
  715. decodeFurther = true
  716. case cborMajorMap:
  717. n.v = valueTypeMap
  718. decodeFurther = true
  719. case cborMajorTag:
  720. n.v = valueTypeExt
  721. n.u = d.decUint()
  722. n.l = nil
  723. if n.u == 0 || n.u == 1 {
  724. d.bdRead = false
  725. n.v = valueTypeTime
  726. n.t = d.decodeTime(n.u)
  727. } else if d.st && d.h.getExtForTag(n.u) == nil {
  728. // d.skipTags() // no need to call this - tags already skipped
  729. d.bdRead = false
  730. d.DecodeNaked()
  731. return // return when done (as true recursive function)
  732. }
  733. // d.bdRead = false
  734. // d.d.decode(&re.Value) // handled by decode itself.
  735. // decodeFurther = true
  736. case cborMajorSimpleOrFloat:
  737. switch d.bd {
  738. case cborBdNil, cborBdUndefined:
  739. n.v = valueTypeNil
  740. d.fnil = true
  741. case cborBdFalse:
  742. n.v = valueTypeBool
  743. n.b = false
  744. case cborBdTrue:
  745. n.v = valueTypeBool
  746. n.b = true
  747. case cborBdFloat16, cborBdFloat32, cborBdFloat64:
  748. n.v = valueTypeFloat
  749. n.f = d.DecodeFloat64()
  750. case cborBdIndefiniteBytes:
  751. decNakedReadRawBytes(d, &d.d, n, d.h.RawToString)
  752. case cborBdIndefiniteString:
  753. n.v = valueTypeString
  754. n.s = string(d.DecodeStringAsBytes())
  755. case cborBdIndefiniteArray:
  756. n.v = valueTypeArray
  757. decodeFurther = true
  758. case cborBdIndefiniteMap:
  759. n.v = valueTypeMap
  760. decodeFurther = true
  761. default:
  762. d.d.errorf("decodeNaked: Unrecognized d.bd: 0x%x", d.bd)
  763. }
  764. default: // should never happen
  765. d.d.errorf("decodeNaked: Unrecognized d.bd: 0x%x", d.bd)
  766. }
  767. if !decodeFurther {
  768. d.bdRead = false
  769. }
  770. }
  771. // -------------------------
  772. // CborHandle is a Handle for the CBOR encoding format,
  773. // defined at http://tools.ietf.org/html/rfc7049 and documented further at http://cbor.io .
  774. //
  775. // CBOR is comprehensively supported, including support for:
  776. // - indefinite-length arrays/maps/bytes/strings
  777. // - (extension) tags in range 0..0xffff (0 .. 65535)
  778. // - half, single and double-precision floats
  779. // - all numbers (1, 2, 4 and 8-byte signed and unsigned integers)
  780. // - nil, true, false, ...
  781. // - arrays and maps, bytes and text strings
  782. //
  783. // None of the optional extensions (with tags) defined in the spec are supported out-of-the-box.
  784. // Users can implement them as needed (using SetExt), including spec-documented ones:
  785. // - timestamp, BigNum, BigFloat, Decimals,
  786. // - Encoded Text (e.g. URL, regexp, base64, MIME Message), etc.
  787. type CborHandle struct {
  788. binaryEncodingType
  789. // noElemSeparators
  790. BasicHandle
  791. // IndefiniteLength=true, means that we encode using indefinitelength
  792. IndefiniteLength bool
  793. // TimeRFC3339 says to encode time.Time using RFC3339 format.
  794. // If unset, we encode time.Time using seconds past epoch.
  795. TimeRFC3339 bool
  796. // SkipUnexpectedTags says to skip over any tags for which extensions are
  797. // not defined. This is in keeping with the cbor spec on "Optional Tagging of Items".
  798. //
  799. // Furthermore, this allows the skipping over of the Self Describing Tag 0xd9d9f7.
  800. SkipUnexpectedTags bool
  801. _ [7]uint64 // padding (cache-aligned)
  802. }
  803. // Name returns the name of the handle: cbor
  804. func (h *CborHandle) Name() string { return "cbor" }
  805. func (h *CborHandle) newEncDriver() encDriver {
  806. var e = &cborEncDriver{h: h}
  807. e.e.e = e
  808. e.e.init(h)
  809. e.reset()
  810. return e
  811. }
  812. func (h *CborHandle) newDecDriver() decDriver {
  813. d := &cborDecDriver{h: h, st: h.SkipUnexpectedTags}
  814. d.d.d = d
  815. d.d.cbor = true
  816. d.d.init(h)
  817. d.reset()
  818. return d
  819. }
  820. func (e *cborEncDriver) reset() {
  821. }
  822. func (d *cborDecDriver) reset() {
  823. d.bd = 0
  824. d.bdRead = false
  825. d.fnil = false
  826. d.st = d.h.SkipUnexpectedTags
  827. }
  828. var _ decDriver = (*cborDecDriver)(nil)
  829. var _ encDriver = (*cborEncDriver)(nil)