reader.go 24 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248
  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 "io"
  5. /*
  6. // decReader abstracts the reading source, allowing implementations that can
  7. // read from an io.Reader or directly off a byte slice with zero-copying.
  8. //
  9. // Deprecated: Use decReaderSwitch instead.
  10. type decReader interface {
  11. unreadn1()
  12. // readx will use the implementation scratch buffer if possible i.e. n < len(scratchbuf), OR
  13. // just return a view of the []byte being decoded from.
  14. // Ensure you call detachZeroCopyBytes later if this needs to be sent outside codec control.
  15. readx(n int) []byte
  16. readb([]byte)
  17. readn1() uint8
  18. numread() uint // number of bytes read
  19. track()
  20. stopTrack() []byte
  21. // skip will skip any byte that matches, and return the first non-matching byte
  22. skip(accept *bitset256) (token byte)
  23. // readTo will read any byte that matches, stopping once no-longer matching.
  24. readTo(accept *bitset256) (out []byte)
  25. // readUntil will read, only stopping once it matches the 'stop' byte.
  26. readUntil(stop byte) (out []byte)
  27. }
  28. */
  29. // ------------------------------------------------
  30. type unreadByteStatus uint8
  31. // unreadByteStatus goes from
  32. // undefined (when initialized) -- (read) --> canUnread -- (unread) --> canRead ...
  33. const (
  34. unreadByteUndefined unreadByteStatus = iota
  35. unreadByteCanRead
  36. unreadByteCanUnread
  37. )
  38. // func appendPool(bs []byte, b byte, bufp *bytesBufPooler) []byte {
  39. // if cap(bs)-len(bs) < 1 {
  40. // bs = bufp.ensureCap(len(bs)+1, bs)
  41. // }
  42. // bs = append(bs, b)
  43. // return bs
  44. // }
  45. // func appendPoolMany(bs []byte, b []byte, bufp *bytesBufPooler) []byte {
  46. // if cap(bs)-len(bs) < 1 {
  47. // bs = bufp.ensureCap(len(bs)+1, bs)
  48. // }
  49. // bs = append(bs, b...)
  50. // return bs
  51. // }
  52. // --------------------
  53. type ioDecReaderCommon struct {
  54. r io.Reader // the reader passed in
  55. n uint // num read
  56. l byte // last byte
  57. ls unreadByteStatus // last byte status
  58. trb bool // tracking bytes turned on
  59. _ bool
  60. b [4]byte // tiny buffer for reading single bytes
  61. tr bytesBufPoolerPlus // buffer for tracking bytes
  62. bufr bytesBufPoolerPlus // buffer for readTo/readUntil
  63. }
  64. func (z *ioDecReaderCommon) reset(r io.Reader) {
  65. z.r = r
  66. z.ls = unreadByteUndefined
  67. z.l, z.n = 0, 0
  68. z.trb = false
  69. z.tr.end()
  70. }
  71. func (z *ioDecReaderCommon) numread() uint {
  72. return z.n
  73. }
  74. func (z *ioDecReaderCommon) track() {
  75. z.tr.ensureCap(1)
  76. z.tr.buf = z.tr.buf[:0]
  77. z.trb = true
  78. }
  79. func (z *ioDecReaderCommon) stopTrack() (bs []byte) {
  80. z.trb = false
  81. return z.tr.buf
  82. }
  83. func (z *ioDecReaderCommon) resetBufr() {
  84. z.bufr.ensureCap(1)
  85. z.bufr.buf = z.bufr.buf[:0]
  86. }
  87. func (z *ioDecReaderCommon) release() {
  88. z.tr.end()
  89. z.bufr.end()
  90. }
  91. // ------------------------------------------
  92. // ioDecReader is a decReader that reads off an io.Reader.
  93. //
  94. // It also has a fallback implementation of ByteScanner if needed.
  95. type ioDecReader struct {
  96. ioDecReaderCommon
  97. rr io.Reader
  98. br io.ByteScanner
  99. x [scratchByteArrayLen]byte // for: get struct field name, swallow valueTypeBytes, etc
  100. // _ [1]uint64 // padding
  101. }
  102. func (z *ioDecReader) reset(r io.Reader) {
  103. z.ioDecReaderCommon.reset(r)
  104. var ok bool
  105. z.rr = r
  106. z.br, ok = r.(io.ByteScanner)
  107. if !ok {
  108. z.br = z
  109. z.rr = z
  110. }
  111. }
  112. func (z *ioDecReader) Read(p []byte) (n int, err error) {
  113. if len(p) == 0 {
  114. return
  115. }
  116. var firstByte bool
  117. if z.ls == unreadByteCanRead {
  118. z.ls = unreadByteCanUnread
  119. p[0] = z.l
  120. if len(p) == 1 {
  121. n = 1
  122. return
  123. }
  124. firstByte = true
  125. p = p[1:]
  126. }
  127. n, err = z.r.Read(p)
  128. if n > 0 {
  129. if err == io.EOF && n == len(p) {
  130. err = nil // read was successful, so postpone EOF (till next time)
  131. }
  132. z.l = p[n-1]
  133. z.ls = unreadByteCanUnread
  134. }
  135. if firstByte {
  136. n++
  137. }
  138. return
  139. }
  140. func (z *ioDecReader) ReadByte() (c byte, err error) {
  141. n, err := z.Read(z.b[:1])
  142. if n == 1 {
  143. c = z.b[0]
  144. if err == io.EOF {
  145. err = nil // read was successful, so postpone EOF (till next time)
  146. }
  147. }
  148. return
  149. }
  150. func (z *ioDecReader) UnreadByte() (err error) {
  151. switch z.ls {
  152. case unreadByteCanUnread:
  153. z.ls = unreadByteCanRead
  154. case unreadByteCanRead:
  155. err = errDecUnreadByteLastByteNotRead
  156. case unreadByteUndefined:
  157. err = errDecUnreadByteNothingToRead
  158. default:
  159. err = errDecUnreadByteUnknown
  160. }
  161. return
  162. }
  163. func (z *ioDecReader) readx(n uint) (bs []byte) {
  164. if n == 0 {
  165. return
  166. }
  167. if n < uint(len(z.x)) {
  168. bs = z.x[:n]
  169. } else {
  170. bs = make([]byte, n)
  171. }
  172. if _, err := decReadFull(z.rr, bs); err != nil {
  173. panic(err)
  174. }
  175. z.n += uint(len(bs))
  176. if z.trb {
  177. z.tr.appends(bs)
  178. }
  179. return
  180. }
  181. func (z *ioDecReader) readb(bs []byte) {
  182. if len(bs) == 0 {
  183. return
  184. }
  185. if _, err := decReadFull(z.rr, bs); err != nil {
  186. panic(err)
  187. }
  188. z.n += uint(len(bs))
  189. if z.trb {
  190. z.tr.appends(bs)
  191. }
  192. }
  193. func (z *ioDecReader) readn1eof() (b uint8, eof bool) {
  194. b, err := z.br.ReadByte()
  195. if err == nil {
  196. z.n++
  197. if z.trb {
  198. z.tr.append(b)
  199. }
  200. } else if err == io.EOF {
  201. eof = true
  202. } else {
  203. panic(err)
  204. }
  205. return
  206. }
  207. func (z *ioDecReader) readn1() (b uint8) {
  208. b, err := z.br.ReadByte()
  209. if err == nil {
  210. z.n++
  211. if z.trb {
  212. z.tr.append(b)
  213. }
  214. return
  215. }
  216. panic(err)
  217. }
  218. func (z *ioDecReader) skip(accept *bitset256) (token byte) {
  219. var eof bool
  220. // for {
  221. // token, eof = z.readn1eof()
  222. // if eof {
  223. // return
  224. // }
  225. // if accept.isset(token) {
  226. // continue
  227. // }
  228. // return
  229. // }
  230. LOOP:
  231. token, eof = z.readn1eof()
  232. if eof {
  233. return
  234. }
  235. if accept.isset(token) {
  236. goto LOOP
  237. }
  238. return
  239. }
  240. func (z *ioDecReader) readTo(accept *bitset256) []byte {
  241. // out = in
  242. // for {
  243. // token, eof := z.readn1eof()
  244. // if eof {
  245. // return
  246. // }
  247. // if accept.isset(token) {
  248. // out = append(out, token)
  249. // } else {
  250. // z.unreadn1()
  251. // return
  252. // }
  253. // }
  254. z.resetBufr()
  255. LOOP:
  256. token, eof := z.readn1eof()
  257. if eof {
  258. return z.bufr.buf
  259. }
  260. if accept.isset(token) {
  261. // out = append(out, token)
  262. z.bufr.append(token)
  263. goto LOOP
  264. }
  265. z.unreadn1()
  266. return z.bufr.buf
  267. }
  268. func (z *ioDecReader) readUntil(stop byte) []byte {
  269. // for {
  270. // token, eof := z.readn1eof()
  271. // if eof {
  272. // panic(io.EOF)
  273. // }
  274. // out = append(out, token)
  275. // if token == stop {
  276. // return
  277. // }
  278. // }
  279. z.resetBufr()
  280. LOOP:
  281. token, eof := z.readn1eof()
  282. if eof {
  283. panic(io.EOF)
  284. }
  285. z.bufr.append(token)
  286. if token == stop {
  287. return z.bufr.buf
  288. }
  289. goto LOOP
  290. }
  291. //go:noinline
  292. func (z *ioDecReader) unreadn1() {
  293. err := z.br.UnreadByte()
  294. if err != nil {
  295. panic(err)
  296. }
  297. z.n--
  298. if z.trb {
  299. if l := len(z.tr.buf) - 1; l >= 0 {
  300. z.tr.buf = z.tr.buf[:l]
  301. }
  302. }
  303. }
  304. // ------------------------------------
  305. type bufioDecReader struct {
  306. ioDecReaderCommon
  307. // _ uint64 // padding (cache-aligned)
  308. c uint // cursor
  309. // buf []byte
  310. bp bytesBufPoolerPlus
  311. // err error
  312. }
  313. func (z *bufioDecReader) reset(r io.Reader, bufsize int) {
  314. z.ioDecReaderCommon.reset(r)
  315. z.c = 0
  316. if cap(z.bp.buf) < bufsize {
  317. z.bp.get(bufsize)
  318. // z.buf = make([]byte, 0, bufsize)
  319. }
  320. z.bp.buf = z.bp.buf[:0]
  321. }
  322. func (z *bufioDecReader) release() {
  323. z.ioDecReaderCommon.release()
  324. z.bp.end()
  325. }
  326. func (z *bufioDecReader) readb(p []byte) {
  327. var n = uint(copy(p, z.bp.buf[z.c:]))
  328. z.n += n
  329. z.c += n
  330. if len(p) == int(n) {
  331. if z.trb {
  332. z.tr.appends(p)
  333. }
  334. } else {
  335. z.readbFill(p, n)
  336. }
  337. }
  338. //go:noinline - fallback when z.bp.buf is consumed
  339. func (z *bufioDecReader) readbFill(p0 []byte, n uint) {
  340. // at this point, there's nothing in z.bp.buf to read (z.bp.buf is fully consumed)
  341. p := p0[n:]
  342. var n2 uint
  343. var err error
  344. if len(p) > cap(z.bp.buf) {
  345. n2, err = decReadFull(z.r, p)
  346. if err != nil {
  347. panic(err)
  348. }
  349. n += n2
  350. z.n += n2
  351. // always keep last byte in z.bp.buf
  352. z.bp.buf = z.bp.buf[:1]
  353. z.bp.buf[0] = p[len(p)-1]
  354. z.c = 1
  355. if z.trb {
  356. z.tr.appends(p0[:n])
  357. }
  358. return
  359. }
  360. // z.c is now 0, and len(p) <= cap(z.bp.buf)
  361. LOOP:
  362. // for len(p) > 0 && z.err == nil {
  363. if len(p) > 0 {
  364. z.bp.buf = z.bp.buf[0:cap(z.bp.buf)]
  365. var n1 int
  366. n1, err = z.r.Read(z.bp.buf)
  367. n2 = uint(n1)
  368. if n2 == 0 && err != nil {
  369. panic(err)
  370. }
  371. z.bp.buf = z.bp.buf[:n2]
  372. n2 = uint(copy(p, z.bp.buf))
  373. z.c = n2
  374. n += n2
  375. z.n += n2
  376. p = p[n2:]
  377. goto LOOP
  378. }
  379. if z.c == 0 {
  380. z.bp.buf = z.bp.buf[:1]
  381. z.bp.buf[0] = p[len(p)-1]
  382. z.c = 1
  383. }
  384. if z.trb {
  385. z.tr.appends(p0[:n])
  386. }
  387. }
  388. func (z *bufioDecReader) readn1() (b byte) {
  389. // fast-path, so we elide calling into Read() most of the time
  390. if z.c < uint(len(z.bp.buf)) {
  391. b = z.bp.buf[z.c]
  392. z.c++
  393. z.n++
  394. if z.trb {
  395. z.tr.append(b)
  396. }
  397. } else { // meaning z.c == len(z.bp.buf) or greater ... so need to fill
  398. z.readbFill(z.b[:1], 0)
  399. b = z.b[0]
  400. }
  401. return
  402. }
  403. func (z *bufioDecReader) unreadn1() {
  404. if z.c == 0 {
  405. panic(errDecUnreadByteNothingToRead)
  406. }
  407. z.c--
  408. z.n--
  409. if z.trb {
  410. z.tr.buf = z.tr.buf[:len(z.tr.buf)-1]
  411. }
  412. }
  413. func (z *bufioDecReader) readx(n uint) (bs []byte) {
  414. if n == 0 {
  415. // return
  416. } else if z.c+n <= uint(len(z.bp.buf)) {
  417. bs = z.bp.buf[z.c : z.c+n]
  418. z.n += n
  419. z.c += n
  420. if z.trb {
  421. z.tr.appends(bs)
  422. }
  423. } else {
  424. bs = make([]byte, n)
  425. // n no longer used - can reuse
  426. n = uint(copy(bs, z.bp.buf[z.c:]))
  427. z.n += n
  428. z.c += n
  429. z.readbFill(bs, n)
  430. }
  431. return
  432. }
  433. // func (z *bufioDecReader) doTrack(y uint) {
  434. // z.tr = append(z.tr, z.bp.buf[z.c:y]...) // cost=14???
  435. // }
  436. // func (z *bufioDecReader) skipLoopFn(i uint) {
  437. // z.n += (i - z.c) - 1
  438. // i++
  439. // if z.trb {
  440. // // z.tr = append(z.tr, z.bp.buf[z.c:i]...)
  441. // z.doTrack(i)
  442. // }
  443. // z.c = i
  444. // }
  445. func (z *bufioDecReader) skip(accept *bitset256) (token byte) {
  446. // token, _ = z.search(nil, accept, 0, 1); return
  447. // for i := z.c; i < len(z.bp.buf); i++ {
  448. // if token = z.bp.buf[i]; !accept.isset(token) {
  449. // z.skipLoopFn(i)
  450. // return
  451. // }
  452. // }
  453. i := z.c
  454. LOOP:
  455. if i < uint(len(z.bp.buf)) {
  456. // inline z.skipLoopFn(i) and refactor, so cost is within inline budget
  457. token = z.bp.buf[i]
  458. i++
  459. if accept.isset(token) {
  460. goto LOOP
  461. }
  462. z.n += i - 2 - z.c
  463. if z.trb {
  464. z.tr.appends(z.bp.buf[z.c:i]) // z.doTrack(i)
  465. }
  466. z.c = i
  467. return
  468. }
  469. return z.skipFill(accept)
  470. }
  471. func (z *bufioDecReader) skipFill(accept *bitset256) (token byte) {
  472. // defer func() { xdebugf("skipFill '%c'", token) }()
  473. z.n += uint(len(z.bp.buf)) - z.c
  474. if z.trb {
  475. z.tr.appends(z.bp.buf[z.c:])
  476. }
  477. var i, n2 int
  478. var err error
  479. for {
  480. z.c = 0
  481. z.bp.buf = z.bp.buf[0:cap(z.bp.buf)]
  482. n2, err = z.r.Read(z.bp.buf)
  483. if n2 == 0 && err != nil {
  484. panic(err)
  485. }
  486. z.bp.buf = z.bp.buf[:n2]
  487. for i, token = range z.bp.buf {
  488. if !accept.isset(token) {
  489. z.n += (uint(i) - z.c) - 1
  490. z.loopFn(uint(i + 1))
  491. return
  492. }
  493. }
  494. // for i := 0; i < n2; i++ {
  495. // if token = z.bp.buf[i]; !accept.isset(token) {
  496. // z.skipLoopFn(i)
  497. // return
  498. // }
  499. // }
  500. z.n += uint(n2)
  501. if z.trb {
  502. z.tr.appends(z.bp.buf)
  503. }
  504. }
  505. }
  506. // func (z *bufioDecReader) readLoopFn(i uint, out0 []byte) (out []byte) {
  507. // out = appendPool(out0, z.bp.buf[z.c:i]...)
  508. // z.loopFn(i)
  509. // return
  510. // }
  511. func (z *bufioDecReader) loopFn(i uint) {
  512. if z.trb {
  513. z.tr.appends(z.bp.buf[z.c:i]) // z.doTrack(i)
  514. }
  515. z.c = i
  516. }
  517. // func (z *bufioDecReader) readToLoopFn(i uint, out0 []byte) (out []byte) {
  518. // // out0 is never nil
  519. // z.n += (i - z.c) - 1
  520. // return z.readLoopFn(i, out0)
  521. // }
  522. func (z *bufioDecReader) readTo(accept *bitset256) (out []byte) {
  523. // defer func() { xdebug2f("bufio: readTo: %s", out) }()
  524. // _, out = z.search(in, accept, 0, 2); return
  525. // for i := z.c; i < len(z.bp.buf); i++ {
  526. // if !accept.isset(z.bp.buf[i]) {
  527. // return z.readToLoopFn(i, nil)
  528. // }
  529. // }
  530. i := z.c
  531. LOOP:
  532. if i < uint(len(z.bp.buf)) {
  533. if !accept.isset(z.bp.buf[i]) {
  534. // return z.readToLoopFn(i, nil)
  535. // inline readToLoopFn here (for performance)
  536. z.n += (i - z.c) - 1
  537. out = z.bp.buf[z.c:i]
  538. if z.trb {
  539. z.tr.appends(z.bp.buf[z.c:i]) // z.doTrack(i)
  540. }
  541. z.c = i
  542. return
  543. }
  544. i++
  545. goto LOOP
  546. }
  547. return z.readToFill(accept)
  548. }
  549. func (z *bufioDecReader) readToFill(accept *bitset256) []byte {
  550. z.resetBufr()
  551. z.n += uint(len(z.bp.buf)) - z.c
  552. z.bufr.appends(z.bp.buf[z.c:])
  553. if z.trb {
  554. z.tr.appends(z.bp.buf[z.c:])
  555. }
  556. var n2 int
  557. var err error
  558. for {
  559. z.c = 0
  560. z.bp.buf = z.bp.buf[:cap(z.bp.buf)]
  561. n2, err = z.r.Read(z.bp.buf)
  562. if n2 == 0 && err != nil {
  563. if err == io.EOF {
  564. return z.bufr.buf // readTo should read until it matches or end is reached
  565. }
  566. panic(err)
  567. }
  568. z.bp.buf = z.bp.buf[:n2]
  569. for i, token := range z.bp.buf {
  570. if !accept.isset(token) {
  571. z.n += (uint(i) - z.c) - 1
  572. z.bufr.appends(z.bp.buf[z.c:i])
  573. z.loopFn(uint(i))
  574. return z.bufr.buf
  575. }
  576. }
  577. // for i := 0; i < n2; i++ {
  578. // if !accept.isset(z.bp.buf[i]) {
  579. // return z.readToLoopFn(i, out)
  580. // }
  581. // }
  582. z.bufr.appends(z.bp.buf)
  583. z.n += uint(n2)
  584. if z.trb {
  585. z.tr.appends(z.bp.buf)
  586. }
  587. }
  588. }
  589. // func (z *bufioDecReader) readUntilLoopFn(i uint, out0 []byte) (out []byte) {
  590. // z.n += (i - z.c) - 1
  591. // return z.readLoopFn(i+1, out0)
  592. // }
  593. func (z *bufioDecReader) readUntil(stop byte) (out []byte) {
  594. // defer func() { xdebug2f("bufio: readUntil: %s", out) }()
  595. // _, out = z.search(in, nil, stop, 4); return
  596. // for i := z.c; i < len(z.bp.buf); i++ {
  597. // if z.bp.buf[i] == stop {
  598. // return z.readUntilLoopFn(i, nil)
  599. // }
  600. // }
  601. i := z.c
  602. LOOP:
  603. if i < uint(len(z.bp.buf)) {
  604. if z.bp.buf[i] == stop {
  605. // inline readUntilLoopFn
  606. // return z.readUntilLoopFn(i, nil)
  607. z.n += (i - z.c) - 1
  608. i++
  609. out = z.bp.buf[z.c:i]
  610. if z.trb {
  611. z.tr.appends(z.bp.buf[z.c:i]) // z.doTrack(i)
  612. }
  613. z.c = i
  614. return
  615. }
  616. i++
  617. goto LOOP
  618. }
  619. return z.readUntilFill(stop)
  620. }
  621. func (z *bufioDecReader) readUntilFill(stop byte) []byte {
  622. z.resetBufr()
  623. z.n += uint(len(z.bp.buf)) - z.c
  624. z.bufr.appends(z.bp.buf[z.c:])
  625. if z.trb {
  626. z.tr.appends(z.bp.buf[z.c:])
  627. }
  628. for {
  629. z.c = 0
  630. z.bp.buf = z.bp.buf[0:cap(z.bp.buf)]
  631. n1, err := z.r.Read(z.bp.buf)
  632. if n1 == 0 && err != nil {
  633. panic(err)
  634. }
  635. n2 := uint(n1)
  636. z.bp.buf = z.bp.buf[:n2]
  637. for i, token := range z.bp.buf {
  638. if token == stop {
  639. z.n += (uint(i) - z.c) - 1
  640. z.bufr.appends(z.bp.buf[z.c : i+1])
  641. z.loopFn(uint(i + 1))
  642. return z.bufr.buf
  643. }
  644. }
  645. // for i := 0; i < n2; i++ {
  646. // if z.bp.buf[i] == stop {
  647. // return z.readUntilLoopFn(i, out)
  648. // }
  649. // }
  650. z.bufr.appends(z.bp.buf)
  651. z.n += n2
  652. if z.trb {
  653. z.tr.appends(z.bp.buf)
  654. }
  655. }
  656. }
  657. // ------------------------------------
  658. // bytesDecReader is a decReader that reads off a byte slice with zero copying
  659. type bytesDecReader struct {
  660. b []byte // data
  661. c uint // cursor
  662. t uint // track start
  663. // a int // available
  664. }
  665. func (z *bytesDecReader) reset(in []byte) {
  666. z.b = in
  667. // z.a = len(in)
  668. z.c = 0
  669. z.t = 0
  670. }
  671. func (z *bytesDecReader) numread() uint {
  672. return z.c
  673. }
  674. func (z *bytesDecReader) unreadn1() {
  675. if z.c == 0 || len(z.b) == 0 {
  676. panic(errBytesDecReaderCannotUnread)
  677. }
  678. z.c--
  679. // z.a++
  680. }
  681. func (z *bytesDecReader) readx(n uint) (bs []byte) {
  682. // slicing from a non-constant start position is more expensive,
  683. // as more computation is required to decipher the pointer start position.
  684. // However, we do it only once, and it's better than reslicing both z.b and return value.
  685. // if n <= 0 {
  686. // } else if z.a == 0 {
  687. // panic(io.EOF)
  688. // } else if n > z.a {
  689. // panic(io.ErrUnexpectedEOF)
  690. // } else {
  691. // c0 := z.c
  692. // z.c = c0 + n
  693. // z.a = z.a - n
  694. // bs = z.b[c0:z.c]
  695. // }
  696. // return
  697. if n != 0 {
  698. z.c += n
  699. if z.c > uint(len(z.b)) {
  700. z.c = uint(len(z.b))
  701. panic(io.EOF)
  702. }
  703. bs = z.b[z.c-n : z.c]
  704. }
  705. return
  706. // if n == 0 {
  707. // } else if z.c+n > uint(len(z.b)) {
  708. // z.c = uint(len(z.b))
  709. // panic(io.EOF)
  710. // } else {
  711. // z.c += n
  712. // bs = z.b[z.c-n : z.c]
  713. // }
  714. // return
  715. // if n == 0 {
  716. // return
  717. // }
  718. // if z.c == uint(len(z.b)) {
  719. // panic(io.EOF)
  720. // }
  721. // if z.c+n > uint(len(z.b)) {
  722. // panic(io.ErrUnexpectedEOF)
  723. // }
  724. // // z.a -= n
  725. // z.c += n
  726. // return z.b[z.c-n : z.c]
  727. }
  728. func (z *bytesDecReader) readb(bs []byte) {
  729. copy(bs, z.readx(uint(len(bs))))
  730. }
  731. func (z *bytesDecReader) readn1() (v uint8) {
  732. if z.c == uint(len(z.b)) {
  733. panic(io.EOF)
  734. }
  735. v = z.b[z.c]
  736. z.c++
  737. // z.a--
  738. return
  739. }
  740. // func (z *bytesDecReader) readn1eof() (v uint8, eof bool) {
  741. // if z.a == 0 {
  742. // eof = true
  743. // return
  744. // }
  745. // v = z.b[z.c]
  746. // z.c++
  747. // z.a--
  748. // return
  749. // }
  750. func (z *bytesDecReader) skip(accept *bitset256) (token byte) {
  751. i := z.c
  752. // if i == len(z.b) {
  753. // goto END
  754. // // panic(io.EOF)
  755. // }
  756. // Replace loop with goto construct, so that this can be inlined
  757. // for i := z.c; i < blen; i++ {
  758. // if !accept.isset(z.b[i]) {
  759. // token = z.b[i]
  760. // i++
  761. // z.a -= (i - z.c)
  762. // z.c = i
  763. // return
  764. // }
  765. // }
  766. // i := z.c
  767. LOOP:
  768. if i < uint(len(z.b)) {
  769. token = z.b[i]
  770. i++
  771. if accept.isset(token) {
  772. goto LOOP
  773. }
  774. // z.a -= (i - z.c)
  775. z.c = i
  776. return
  777. }
  778. // END:
  779. panic(io.EOF)
  780. // // z.a = 0
  781. // z.c = blen
  782. // return
  783. }
  784. func (z *bytesDecReader) readTo(accept *bitset256) (out []byte) {
  785. i := z.c
  786. if i == uint(len(z.b)) {
  787. panic(io.EOF)
  788. }
  789. // Replace loop with goto construct, so that this can be inlined
  790. // for i := z.c; i < blen; i++ {
  791. // if !accept.isset(z.b[i]) {
  792. // out = z.b[z.c:i]
  793. // z.a -= (i - z.c)
  794. // z.c = i
  795. // return
  796. // }
  797. // }
  798. // out = z.b[z.c:]
  799. // z.a, z.c = 0, blen
  800. // return
  801. // i := z.c
  802. // LOOP:
  803. // if i < blen {
  804. // if accept.isset(z.b[i]) {
  805. // i++
  806. // goto LOOP
  807. // }
  808. // out = z.b[z.c:i]
  809. // z.a -= (i - z.c)
  810. // z.c = i
  811. // return
  812. // }
  813. // out = z.b[z.c:]
  814. // // z.a, z.c = 0, blen
  815. // z.a = 0
  816. // z.c = blen
  817. // return
  818. // c := i
  819. LOOP:
  820. if i < uint(len(z.b)) {
  821. if accept.isset(z.b[i]) {
  822. i++
  823. goto LOOP
  824. }
  825. }
  826. out = z.b[z.c:i]
  827. // z.a -= (i - z.c)
  828. z.c = i
  829. return // z.b[c:i]
  830. // z.c, i = i, z.c
  831. // return z.b[i:z.c]
  832. }
  833. func (z *bytesDecReader) readUntil(stop byte) (out []byte) {
  834. i := z.c
  835. // if i == len(z.b) {
  836. // panic(io.EOF)
  837. // }
  838. // Replace loop with goto construct, so that this can be inlined
  839. // for i := z.c; i < blen; i++ {
  840. // if z.b[i] == stop {
  841. // i++
  842. // out = z.b[z.c:i]
  843. // z.a -= (i - z.c)
  844. // z.c = i
  845. // return
  846. // }
  847. // }
  848. LOOP:
  849. if i < uint(len(z.b)) {
  850. if z.b[i] == stop {
  851. i++
  852. out = z.b[z.c:i]
  853. // z.a -= (i - z.c)
  854. z.c = i
  855. return
  856. }
  857. i++
  858. goto LOOP
  859. }
  860. // z.a = 0
  861. // z.c = blen
  862. panic(io.EOF)
  863. }
  864. func (z *bytesDecReader) track() {
  865. z.t = z.c
  866. }
  867. func (z *bytesDecReader) stopTrack() (bs []byte) {
  868. return z.b[z.t:z.c]
  869. }
  870. // --------------
  871. type decReaderSwitch struct {
  872. esep bool // has elem separators
  873. mtr, str bool // whether maptype or slicetype are known types
  874. be bool // is binary encoding
  875. js bool // is json handle
  876. jsms bool // is json handle, and MapKeyAsString
  877. // typ entryType
  878. bytes bool // is bytes reader
  879. bufio bool // is this a bufioDecReader?
  880. rb bytesDecReader
  881. ri *ioDecReader
  882. bi *bufioDecReader
  883. }
  884. // numread, track and stopTrack are always inlined, as they just check int fields, etc.
  885. /*
  886. func (z *decReaderSwitch) numread() int {
  887. switch z.typ {
  888. case entryTypeBytes:
  889. return z.rb.numread()
  890. case entryTypeIo:
  891. return z.ri.numread()
  892. default:
  893. return z.bi.numread()
  894. }
  895. }
  896. func (z *decReaderSwitch) track() {
  897. switch z.typ {
  898. case entryTypeBytes:
  899. z.rb.track()
  900. case entryTypeIo:
  901. z.ri.track()
  902. default:
  903. z.bi.track()
  904. }
  905. }
  906. func (z *decReaderSwitch) stopTrack() []byte {
  907. switch z.typ {
  908. case entryTypeBytes:
  909. return z.rb.stopTrack()
  910. case entryTypeIo:
  911. return z.ri.stopTrack()
  912. default:
  913. return z.bi.stopTrack()
  914. }
  915. }
  916. func (z *decReaderSwitch) unreadn1() {
  917. switch z.typ {
  918. case entryTypeBytes:
  919. z.rb.unreadn1()
  920. case entryTypeIo:
  921. z.ri.unreadn1()
  922. default:
  923. z.bi.unreadn1()
  924. }
  925. }
  926. func (z *decReaderSwitch) readx(n int) []byte {
  927. switch z.typ {
  928. case entryTypeBytes:
  929. return z.rb.readx(n)
  930. case entryTypeIo:
  931. return z.ri.readx(n)
  932. default:
  933. return z.bi.readx(n)
  934. }
  935. }
  936. func (z *decReaderSwitch) readb(s []byte) {
  937. switch z.typ {
  938. case entryTypeBytes:
  939. z.rb.readb(s)
  940. case entryTypeIo:
  941. z.ri.readb(s)
  942. default:
  943. z.bi.readb(s)
  944. }
  945. }
  946. func (z *decReaderSwitch) readn1() uint8 {
  947. switch z.typ {
  948. case entryTypeBytes:
  949. return z.rb.readn1()
  950. case entryTypeIo:
  951. return z.ri.readn1()
  952. default:
  953. return z.bi.readn1()
  954. }
  955. }
  956. func (z *decReaderSwitch) skip(accept *bitset256) (token byte) {
  957. switch z.typ {
  958. case entryTypeBytes:
  959. return z.rb.skip(accept)
  960. case entryTypeIo:
  961. return z.ri.skip(accept)
  962. default:
  963. return z.bi.skip(accept)
  964. }
  965. }
  966. func (z *decReaderSwitch) readTo(accept *bitset256) (out []byte) {
  967. switch z.typ {
  968. case entryTypeBytes:
  969. return z.rb.readTo(accept)
  970. case entryTypeIo:
  971. return z.ri.readTo(accept)
  972. default:
  973. return z.bi.readTo(accept)
  974. }
  975. }
  976. func (z *decReaderSwitch) readUntil(stop byte) (out []byte) {
  977. switch z.typ {
  978. case entryTypeBytes:
  979. return z.rb.readUntil(stop)
  980. case entryTypeIo:
  981. return z.ri.readUntil(stop)
  982. default:
  983. return z.bi.readUntil(stop)
  984. }
  985. }
  986. */
  987. // the if/else-if/else block is expensive to inline.
  988. // Each node of this construct costs a lot and dominates the budget.
  989. // Best to only do an if fast-path else block (so fast-path is inlined).
  990. // This is irrespective of inlineExtraCallCost set in $GOROOT/src/cmd/compile/internal/gc/inl.go
  991. //
  992. // In decReaderSwitch methods below, we delegate all IO functions into their own methods.
  993. // This allows for the inlining of the common path when z.bytes=true.
  994. // Go 1.12+ supports inlining methods with up to 1 inlined function (or 2 if no other constructs).
  995. func (z *decReaderSwitch) release() {
  996. if z.bytes {
  997. } else if z.bufio {
  998. z.bi.release()
  999. } else {
  1000. z.ri.release()
  1001. }
  1002. }
  1003. func (z *decReaderSwitch) numread() uint {
  1004. if z.bytes {
  1005. return z.rb.numread()
  1006. } else if z.bufio {
  1007. return z.bi.numread()
  1008. } else {
  1009. return z.ri.numread()
  1010. }
  1011. }
  1012. func (z *decReaderSwitch) track() {
  1013. if z.bytes {
  1014. z.rb.track()
  1015. } else if z.bufio {
  1016. z.bi.track()
  1017. } else {
  1018. z.ri.track()
  1019. }
  1020. }
  1021. func (z *decReaderSwitch) stopTrack() []byte {
  1022. if z.bytes {
  1023. return z.rb.stopTrack()
  1024. } else if z.bufio {
  1025. return z.bi.stopTrack()
  1026. } else {
  1027. return z.ri.stopTrack()
  1028. }
  1029. }
  1030. // func (z *decReaderSwitch) unreadn1() {
  1031. // if z.bytes {
  1032. // z.rb.unreadn1()
  1033. // } else {
  1034. // z.unreadn1IO()
  1035. // }
  1036. // }
  1037. // func (z *decReaderSwitch) unreadn1IO() {
  1038. // if z.bufio {
  1039. // z.bi.unreadn1()
  1040. // } else {
  1041. // z.ri.unreadn1()
  1042. // }
  1043. // }
  1044. func (z *decReaderSwitch) unreadn1() {
  1045. if z.bytes {
  1046. z.rb.unreadn1()
  1047. } else if z.bufio {
  1048. z.bi.unreadn1()
  1049. } else {
  1050. z.ri.unreadn1() // not inlined
  1051. }
  1052. }
  1053. func (z *decReaderSwitch) readx(n uint) []byte {
  1054. if z.bytes {
  1055. return z.rb.readx(n)
  1056. }
  1057. return z.readxIO(n)
  1058. }
  1059. func (z *decReaderSwitch) readxIO(n uint) []byte {
  1060. if z.bufio {
  1061. return z.bi.readx(n)
  1062. }
  1063. return z.ri.readx(n)
  1064. }
  1065. func (z *decReaderSwitch) readb(s []byte) {
  1066. if z.bytes {
  1067. z.rb.readb(s)
  1068. } else {
  1069. z.readbIO(s)
  1070. }
  1071. }
  1072. //go:noinline - fallback for io, ensures z.bytes path is inlined
  1073. func (z *decReaderSwitch) readbIO(s []byte) {
  1074. if z.bufio {
  1075. z.bi.readb(s)
  1076. } else {
  1077. z.ri.readb(s)
  1078. }
  1079. }
  1080. func (z *decReaderSwitch) readn1() uint8 {
  1081. if z.bytes {
  1082. return z.rb.readn1()
  1083. }
  1084. return z.readn1IO()
  1085. }
  1086. func (z *decReaderSwitch) readn1IO() uint8 {
  1087. if z.bufio {
  1088. return z.bi.readn1()
  1089. }
  1090. return z.ri.readn1()
  1091. }
  1092. func (z *decReaderSwitch) skip(accept *bitset256) (token byte) {
  1093. if z.bytes {
  1094. return z.rb.skip(accept)
  1095. }
  1096. return z.skipIO(accept)
  1097. }
  1098. func (z *decReaderSwitch) skipIO(accept *bitset256) (token byte) {
  1099. if z.bufio {
  1100. return z.bi.skip(accept)
  1101. }
  1102. return z.ri.skip(accept)
  1103. }
  1104. func (z *decReaderSwitch) readTo(accept *bitset256) (out []byte) {
  1105. if z.bytes {
  1106. return z.rb.readTo(accept)
  1107. }
  1108. return z.readToIO(accept)
  1109. }
  1110. //go:noinline - fallback for io, ensures z.bytes path is inlined
  1111. func (z *decReaderSwitch) readToIO(accept *bitset256) (out []byte) {
  1112. if z.bufio {
  1113. return z.bi.readTo(accept)
  1114. }
  1115. return z.ri.readTo(accept)
  1116. }
  1117. func (z *decReaderSwitch) readUntil(stop byte) (out []byte) {
  1118. if z.bytes {
  1119. return z.rb.readUntil(stop)
  1120. }
  1121. return z.readUntilIO(stop)
  1122. }
  1123. func (z *decReaderSwitch) readUntilIO(stop byte) (out []byte) {
  1124. if z.bufio {
  1125. return z.bi.readUntil(stop)
  1126. }
  1127. return z.ri.readUntil(stop)
  1128. }
  1129. // // register these here, so that staticcheck stops barfing
  1130. // var _ = (*bytesDecReader).readTo
  1131. // var _ = (*bytesDecReader).readUntil