ngtcp2_pkt.c 63 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119212021212122212321242125212621272128212921302131213221332134213521362137213821392140214121422143214421452146214721482149215021512152215321542155215621572158215921602161216221632164216521662167216821692170217121722173217421752176217721782179218021812182218321842185218621872188218921902191219221932194219521962197219821992200220122022203220422052206220722082209221022112212221322142215221622172218221922202221222222232224222522262227222822292230223122322233223422352236223722382239224022412242224322442245224622472248224922502251225222532254225522562257225822592260226122622263226422652266226722682269227022712272227322742275227622772278227922802281228222832284228522862287228822892290229122922293229422952296229722982299230023012302230323042305230623072308230923102311231223132314231523162317231823192320232123222323232423252326232723282329233023312332233323342335233623372338233923402341234223432344234523462347234823492350235123522353235423552356235723582359236023612362236323642365236623672368236923702371237223732374237523762377237823792380238123822383238423852386238723882389239023912392239323942395239623972398239924002401240224032404240524062407240824092410241124122413241424152416241724182419242024212422242324242425242624272428242924302431243224332434243524362437243824392440244124422443244424452446244724482449245024512452245324542455245624572458245924602461246224632464246524662467246824692470247124722473247424752476247724782479248024812482248324842485248624872488248924902491249224932494249524962497249824992500250125022503250425052506250725082509251025112512251325142515251625172518251925202521252225232524252525262527252825292530253125322533253425352536253725382539254025412542254325442545254625472548254925502551255225532554255525562557255825592560256125622563256425652566256725682569257025712572257325742575
  1. /*
  2. * ngtcp2
  3. *
  4. * Copyright (c) 2017 ngtcp2 contributors
  5. *
  6. * Permission is hereby granted, free of charge, to any person obtaining
  7. * a copy of this software and associated documentation files (the
  8. * "Software"), to deal in the Software without restriction, including
  9. * without limitation the rights to use, copy, modify, merge, publish,
  10. * distribute, sublicense, and/or sell copies of the Software, and to
  11. * permit persons to whom the Software is furnished to do so, subject to
  12. * the following conditions:
  13. *
  14. * The above copyright notice and this permission notice shall be
  15. * included in all copies or substantial portions of the Software.
  16. *
  17. * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
  18. * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
  19. * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
  20. * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
  21. * LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
  22. * OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
  23. * WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
  24. */
  25. #include "ngtcp2_pkt.h"
  26. #include <assert.h>
  27. #include <string.h>
  28. #include "ngtcp2_conv.h"
  29. #include "ngtcp2_str.h"
  30. #include "ngtcp2_macro.h"
  31. #include "ngtcp2_cid.h"
  32. #include "ngtcp2_mem.h"
  33. #include "ngtcp2_vec.h"
  34. #include "ngtcp2_unreachable.h"
  35. int ngtcp2_pkt_chain_new(ngtcp2_pkt_chain **ppc, const ngtcp2_path *path,
  36. const ngtcp2_pkt_info *pi, const uint8_t *pkt,
  37. size_t pktlen, size_t dgramlen, ngtcp2_tstamp ts,
  38. const ngtcp2_mem *mem) {
  39. *ppc = ngtcp2_mem_malloc(mem, sizeof(ngtcp2_pkt_chain) + pktlen);
  40. if (*ppc == NULL) {
  41. return NGTCP2_ERR_NOMEM;
  42. }
  43. ngtcp2_path_storage_init2(&(*ppc)->path, path);
  44. (*ppc)->pi = *pi;
  45. (*ppc)->next = NULL;
  46. (*ppc)->pkt = (uint8_t *)(*ppc) + sizeof(ngtcp2_pkt_chain);
  47. (*ppc)->pktlen = pktlen;
  48. (*ppc)->dgramlen = dgramlen;
  49. (*ppc)->ts = ts;
  50. memcpy((*ppc)->pkt, pkt, pktlen);
  51. return 0;
  52. }
  53. void ngtcp2_pkt_chain_del(ngtcp2_pkt_chain *pc, const ngtcp2_mem *mem) {
  54. ngtcp2_mem_free(mem, pc);
  55. }
  56. int ngtcp2_pkt_decode_version_cid(ngtcp2_version_cid *dest, const uint8_t *data,
  57. size_t datalen, size_t short_dcidlen) {
  58. size_t len;
  59. uint32_t version;
  60. size_t dcidlen, scidlen;
  61. int supported_version;
  62. assert(datalen);
  63. if (data[0] & NGTCP2_HEADER_FORM_BIT) {
  64. /* 1 byte (Header Form, Fixed Bit, Long Packet Type, Type-Specific bits)
  65. * 4 bytes Version
  66. * 1 byte DCID Length
  67. * 1 byte SCID Length
  68. */
  69. len = 1 + 4 + 1 + 1;
  70. if (datalen < len) {
  71. return NGTCP2_ERR_INVALID_ARGUMENT;
  72. }
  73. dcidlen = data[5];
  74. len += dcidlen;
  75. if (datalen < len) {
  76. return NGTCP2_ERR_INVALID_ARGUMENT;
  77. }
  78. scidlen = data[5 + 1 + dcidlen];
  79. len += scidlen;
  80. if (datalen < len) {
  81. return NGTCP2_ERR_INVALID_ARGUMENT;
  82. }
  83. ngtcp2_get_uint32be(&version, &data[1]);
  84. supported_version = ngtcp2_is_supported_version(version);
  85. if (supported_version &&
  86. (dcidlen > NGTCP2_MAX_CIDLEN || scidlen > NGTCP2_MAX_CIDLEN)) {
  87. return NGTCP2_ERR_INVALID_ARGUMENT;
  88. }
  89. if (version && !supported_version &&
  90. datalen < NGTCP2_MAX_UDP_PAYLOAD_SIZE) {
  91. return NGTCP2_ERR_INVALID_ARGUMENT;
  92. }
  93. dest->version = version;
  94. dest->dcid = &data[6];
  95. dest->dcidlen = dcidlen;
  96. dest->scid = &data[6 + dcidlen + 1];
  97. dest->scidlen = scidlen;
  98. if (!version) {
  99. /* VN */
  100. return 0;
  101. }
  102. if (!supported_version) {
  103. return NGTCP2_ERR_VERSION_NEGOTIATION;
  104. }
  105. return 0;
  106. }
  107. assert(short_dcidlen <= NGTCP2_MAX_CIDLEN);
  108. len = 1 + short_dcidlen;
  109. if (datalen < len) {
  110. return NGTCP2_ERR_INVALID_ARGUMENT;
  111. }
  112. dest->version = 0;
  113. dest->dcid = &data[1];
  114. dest->dcidlen = short_dcidlen;
  115. dest->scid = NULL;
  116. dest->scidlen = 0;
  117. return 0;
  118. }
  119. void ngtcp2_pkt_hd_init(ngtcp2_pkt_hd *hd, uint8_t flags, uint8_t type,
  120. const ngtcp2_cid *dcid, const ngtcp2_cid *scid,
  121. int64_t pkt_num, size_t pkt_numlen, uint32_t version,
  122. size_t len) {
  123. hd->flags = flags;
  124. hd->type = type;
  125. if (dcid) {
  126. hd->dcid = *dcid;
  127. } else {
  128. ngtcp2_cid_zero(&hd->dcid);
  129. }
  130. if (scid) {
  131. hd->scid = *scid;
  132. } else {
  133. ngtcp2_cid_zero(&hd->scid);
  134. }
  135. hd->pkt_num = pkt_num;
  136. hd->token = NULL;
  137. hd->tokenlen = 0;
  138. hd->pkt_numlen = pkt_numlen;
  139. hd->version = version;
  140. hd->len = len;
  141. }
  142. ngtcp2_ssize ngtcp2_pkt_decode_hd_long(ngtcp2_pkt_hd *dest, const uint8_t *pkt,
  143. size_t pktlen) {
  144. uint8_t type;
  145. uint32_t version;
  146. size_t dcil, scil;
  147. const uint8_t *p;
  148. size_t len = 0;
  149. size_t ntokenlen = 0;
  150. const uint8_t *token = NULL;
  151. size_t tokenlen = 0;
  152. size_t nlonglen = 0;
  153. size_t longlen = 0;
  154. uint64_t vi;
  155. uint8_t flags = NGTCP2_PKT_FLAG_LONG_FORM;
  156. if (pktlen < 5) {
  157. return NGTCP2_ERR_INVALID_ARGUMENT;
  158. }
  159. if (!(pkt[0] & NGTCP2_HEADER_FORM_BIT)) {
  160. return NGTCP2_ERR_INVALID_ARGUMENT;
  161. }
  162. ngtcp2_get_uint32be(&version, &pkt[1]);
  163. if (version == 0) {
  164. type = NGTCP2_PKT_VERSION_NEGOTIATION;
  165. /* Version Negotiation is not a long header packet. */
  166. flags = NGTCP2_PKT_FLAG_NONE;
  167. /* This must be Version Negotiation packet which lacks packet
  168. number and payload length fields. */
  169. len = 5 + 2;
  170. } else {
  171. if (!(pkt[0] & NGTCP2_FIXED_BIT_MASK)) {
  172. flags |= NGTCP2_PKT_FLAG_FIXED_BIT_CLEAR;
  173. }
  174. type = ngtcp2_pkt_get_type_long(version, pkt[0]);
  175. switch (type) {
  176. case 0:
  177. return NGTCP2_ERR_INVALID_ARGUMENT;
  178. case NGTCP2_PKT_INITIAL:
  179. len = 1 /* Token Length */ + NGTCP2_MIN_LONG_HEADERLEN -
  180. 1; /* Cut packet number field */
  181. break;
  182. case NGTCP2_PKT_RETRY:
  183. /* Retry packet does not have packet number and length fields */
  184. len = 5 + 2;
  185. break;
  186. case NGTCP2_PKT_HANDSHAKE:
  187. case NGTCP2_PKT_0RTT:
  188. len = NGTCP2_MIN_LONG_HEADERLEN - 1; /* Cut packet number field */
  189. break;
  190. default:
  191. /* Unreachable */
  192. ngtcp2_unreachable();
  193. }
  194. }
  195. if (pktlen < len) {
  196. return NGTCP2_ERR_INVALID_ARGUMENT;
  197. }
  198. p = &pkt[5];
  199. dcil = *p;
  200. if (dcil > NGTCP2_MAX_CIDLEN) {
  201. /* QUIC v1 implementation never expect to receive CID length more
  202. than NGTCP2_MAX_CIDLEN. */
  203. return NGTCP2_ERR_INVALID_ARGUMENT;
  204. }
  205. len += dcil;
  206. if (pktlen < len) {
  207. return NGTCP2_ERR_INVALID_ARGUMENT;
  208. }
  209. p += 1 + dcil;
  210. scil = *p;
  211. if (scil > NGTCP2_MAX_CIDLEN) {
  212. return NGTCP2_ERR_INVALID_ARGUMENT;
  213. }
  214. len += scil;
  215. if (pktlen < len) {
  216. return NGTCP2_ERR_INVALID_ARGUMENT;
  217. }
  218. p += 1 + scil;
  219. if (type == NGTCP2_PKT_INITIAL) {
  220. /* Token Length */
  221. ntokenlen = ngtcp2_get_uvarintlen(p);
  222. len += ntokenlen - 1;
  223. if (pktlen < len) {
  224. return NGTCP2_ERR_INVALID_ARGUMENT;
  225. }
  226. p = ngtcp2_get_uvarint(&vi, p);
  227. if (pktlen - len < vi) {
  228. return NGTCP2_ERR_INVALID_ARGUMENT;
  229. }
  230. tokenlen = (size_t)vi;
  231. len += tokenlen;
  232. if (tokenlen) {
  233. token = p;
  234. }
  235. p += tokenlen;
  236. }
  237. switch (type) {
  238. case NGTCP2_PKT_RETRY:
  239. break;
  240. default:
  241. if (!(flags & NGTCP2_PKT_FLAG_LONG_FORM)) {
  242. assert(type == NGTCP2_PKT_VERSION_NEGOTIATION);
  243. /* Version Negotiation is not a long header packet. */
  244. break;
  245. }
  246. /* Length */
  247. nlonglen = ngtcp2_get_uvarintlen(p);
  248. len += nlonglen - 1;
  249. if (pktlen < len) {
  250. return NGTCP2_ERR_INVALID_ARGUMENT;
  251. }
  252. ngtcp2_get_uvarint(&vi, p);
  253. #if SIZE_MAX > UINT32_MAX
  254. if (vi > SIZE_MAX) {
  255. return NGTCP2_ERR_INVALID_ARGUMENT;
  256. }
  257. #endif /* SIZE_MAX > UINT32_MAX */
  258. longlen = (size_t)vi;
  259. }
  260. dest->flags = flags;
  261. dest->type = type;
  262. dest->version = version;
  263. dest->pkt_num = 0;
  264. dest->pkt_numlen = 0;
  265. p = &pkt[6];
  266. ngtcp2_cid_init(&dest->dcid, p, dcil);
  267. p += dcil + 1;
  268. ngtcp2_cid_init(&dest->scid, p, scil);
  269. p += scil;
  270. dest->token = token;
  271. dest->tokenlen = tokenlen;
  272. p += ntokenlen + tokenlen;
  273. dest->len = longlen;
  274. #ifndef NDEBUG
  275. p += nlonglen;
  276. #endif /* !defined(NDEBUG) */
  277. assert((size_t)(p - pkt) == len);
  278. return (ngtcp2_ssize)len;
  279. }
  280. ngtcp2_ssize ngtcp2_pkt_decode_hd_short(ngtcp2_pkt_hd *dest, const uint8_t *pkt,
  281. size_t pktlen, size_t dcidlen) {
  282. size_t len = 1 + dcidlen;
  283. const uint8_t *p = pkt;
  284. uint8_t flags = NGTCP2_PKT_FLAG_NONE;
  285. assert(dcidlen <= NGTCP2_MAX_CIDLEN);
  286. if (pktlen < len) {
  287. return NGTCP2_ERR_INVALID_ARGUMENT;
  288. }
  289. if (pkt[0] & NGTCP2_HEADER_FORM_BIT) {
  290. return NGTCP2_ERR_INVALID_ARGUMENT;
  291. }
  292. if (!(pkt[0] & NGTCP2_FIXED_BIT_MASK)) {
  293. flags |= NGTCP2_PKT_FLAG_FIXED_BIT_CLEAR;
  294. }
  295. p = &pkt[1];
  296. dest->type = NGTCP2_PKT_1RTT;
  297. ngtcp2_cid_init(&dest->dcid, p, dcidlen);
  298. p += dcidlen;
  299. /* Set 0 to SCID so that we don't accidentally reference it and gets
  300. garbage. */
  301. ngtcp2_cid_zero(&dest->scid);
  302. dest->flags = flags;
  303. dest->version = 0;
  304. dest->len = 0;
  305. dest->pkt_num = 0;
  306. dest->pkt_numlen = 0;
  307. dest->token = NULL;
  308. dest->tokenlen = 0;
  309. assert((size_t)(p - pkt) == len);
  310. return (ngtcp2_ssize)len;
  311. }
  312. ngtcp2_ssize ngtcp2_pkt_encode_hd_long(uint8_t *out, size_t outlen,
  313. const ngtcp2_pkt_hd *hd) {
  314. uint8_t *p;
  315. size_t len = NGTCP2_MIN_LONG_HEADERLEN + hd->dcid.datalen + hd->scid.datalen -
  316. 2; /* NGTCP2_MIN_LONG_HEADERLEN includes 1 byte for
  317. len and 1 byte for packet number. */
  318. if (hd->type != NGTCP2_PKT_RETRY) {
  319. len += NGTCP2_PKT_LENGTHLEN /* Length */ + hd->pkt_numlen;
  320. }
  321. if (hd->type == NGTCP2_PKT_INITIAL) {
  322. len += ngtcp2_put_uvarintlen(hd->tokenlen) + hd->tokenlen;
  323. }
  324. if (outlen < len) {
  325. return NGTCP2_ERR_NOBUF;
  326. }
  327. p = out;
  328. *p = (uint8_t)(NGTCP2_HEADER_FORM_BIT |
  329. (ngtcp2_pkt_versioned_type(hd->version, hd->type) << 4) |
  330. (uint8_t)(hd->pkt_numlen - 1));
  331. if (!(hd->flags & NGTCP2_PKT_FLAG_FIXED_BIT_CLEAR)) {
  332. *p |= NGTCP2_FIXED_BIT_MASK;
  333. }
  334. ++p;
  335. p = ngtcp2_put_uint32be(p, hd->version);
  336. *p++ = (uint8_t)hd->dcid.datalen;
  337. if (hd->dcid.datalen) {
  338. p = ngtcp2_cpymem(p, hd->dcid.data, hd->dcid.datalen);
  339. }
  340. *p++ = (uint8_t)hd->scid.datalen;
  341. if (hd->scid.datalen) {
  342. p = ngtcp2_cpymem(p, hd->scid.data, hd->scid.datalen);
  343. }
  344. if (hd->type == NGTCP2_PKT_INITIAL) {
  345. p = ngtcp2_put_uvarint(p, hd->tokenlen);
  346. if (hd->tokenlen) {
  347. p = ngtcp2_cpymem(p, hd->token, hd->tokenlen);
  348. }
  349. }
  350. if (hd->type != NGTCP2_PKT_RETRY) {
  351. p = ngtcp2_put_uvarint30(p, (uint32_t)hd->len);
  352. p = ngtcp2_put_pkt_num(p, hd->pkt_num, hd->pkt_numlen);
  353. }
  354. assert((size_t)(p - out) == len);
  355. return (ngtcp2_ssize)len;
  356. }
  357. ngtcp2_ssize ngtcp2_pkt_encode_hd_short(uint8_t *out, size_t outlen,
  358. const ngtcp2_pkt_hd *hd) {
  359. uint8_t *p;
  360. size_t len = 1 + hd->dcid.datalen + hd->pkt_numlen;
  361. if (outlen < len) {
  362. return NGTCP2_ERR_NOBUF;
  363. }
  364. p = out;
  365. *p = (uint8_t)(hd->pkt_numlen - 1);
  366. if (!(hd->flags & NGTCP2_PKT_FLAG_FIXED_BIT_CLEAR)) {
  367. *p |= NGTCP2_FIXED_BIT_MASK;
  368. }
  369. if (hd->flags & NGTCP2_PKT_FLAG_KEY_PHASE) {
  370. *p |= NGTCP2_SHORT_KEY_PHASE_BIT;
  371. }
  372. ++p;
  373. if (hd->dcid.datalen) {
  374. p = ngtcp2_cpymem(p, hd->dcid.data, hd->dcid.datalen);
  375. }
  376. p = ngtcp2_put_pkt_num(p, hd->pkt_num, hd->pkt_numlen);
  377. assert((size_t)(p - out) == len);
  378. return (ngtcp2_ssize)len;
  379. }
  380. ngtcp2_ssize ngtcp2_pkt_decode_frame(ngtcp2_frame *dest, const uint8_t *payload,
  381. size_t payloadlen) {
  382. uint8_t type;
  383. if (payloadlen == 0) {
  384. return NGTCP2_ERR_FRAME_ENCODING;
  385. }
  386. type = payload[0];
  387. switch (type) {
  388. case NGTCP2_FRAME_PADDING:
  389. return ngtcp2_pkt_decode_padding_frame(&dest->padding, payload, payloadlen);
  390. case NGTCP2_FRAME_RESET_STREAM:
  391. return ngtcp2_pkt_decode_reset_stream_frame(&dest->reset_stream, payload,
  392. payloadlen);
  393. case NGTCP2_FRAME_CONNECTION_CLOSE:
  394. case NGTCP2_FRAME_CONNECTION_CLOSE_APP:
  395. return ngtcp2_pkt_decode_connection_close_frame(&dest->connection_close,
  396. payload, payloadlen);
  397. case NGTCP2_FRAME_MAX_DATA:
  398. return ngtcp2_pkt_decode_max_data_frame(&dest->max_data, payload,
  399. payloadlen);
  400. case NGTCP2_FRAME_MAX_STREAM_DATA:
  401. return ngtcp2_pkt_decode_max_stream_data_frame(&dest->max_stream_data,
  402. payload, payloadlen);
  403. case NGTCP2_FRAME_MAX_STREAMS_BIDI:
  404. case NGTCP2_FRAME_MAX_STREAMS_UNI:
  405. return ngtcp2_pkt_decode_max_streams_frame(&dest->max_streams, payload,
  406. payloadlen);
  407. case NGTCP2_FRAME_PING:
  408. return ngtcp2_pkt_decode_ping_frame(&dest->ping, payload, payloadlen);
  409. case NGTCP2_FRAME_DATA_BLOCKED:
  410. return ngtcp2_pkt_decode_data_blocked_frame(&dest->data_blocked, payload,
  411. payloadlen);
  412. case NGTCP2_FRAME_STREAM_DATA_BLOCKED:
  413. return ngtcp2_pkt_decode_stream_data_blocked_frame(
  414. &dest->stream_data_blocked, payload, payloadlen);
  415. case NGTCP2_FRAME_STREAMS_BLOCKED_BIDI:
  416. case NGTCP2_FRAME_STREAMS_BLOCKED_UNI:
  417. return ngtcp2_pkt_decode_streams_blocked_frame(&dest->streams_blocked,
  418. payload, payloadlen);
  419. case NGTCP2_FRAME_NEW_CONNECTION_ID:
  420. return ngtcp2_pkt_decode_new_connection_id_frame(&dest->new_connection_id,
  421. payload, payloadlen);
  422. case NGTCP2_FRAME_STOP_SENDING:
  423. return ngtcp2_pkt_decode_stop_sending_frame(&dest->stop_sending, payload,
  424. payloadlen);
  425. case NGTCP2_FRAME_ACK:
  426. case NGTCP2_FRAME_ACK_ECN:
  427. return ngtcp2_pkt_decode_ack_frame(&dest->ack, payload, payloadlen);
  428. case NGTCP2_FRAME_PATH_CHALLENGE:
  429. return ngtcp2_pkt_decode_path_challenge_frame(&dest->path_challenge,
  430. payload, payloadlen);
  431. case NGTCP2_FRAME_PATH_RESPONSE:
  432. return ngtcp2_pkt_decode_path_response_frame(&dest->path_response, payload,
  433. payloadlen);
  434. case NGTCP2_FRAME_CRYPTO:
  435. return ngtcp2_pkt_decode_crypto_frame(&dest->stream, payload, payloadlen);
  436. case NGTCP2_FRAME_NEW_TOKEN:
  437. return ngtcp2_pkt_decode_new_token_frame(&dest->new_token, payload,
  438. payloadlen);
  439. case NGTCP2_FRAME_RETIRE_CONNECTION_ID:
  440. return ngtcp2_pkt_decode_retire_connection_id_frame(
  441. &dest->retire_connection_id, payload, payloadlen);
  442. case NGTCP2_FRAME_HANDSHAKE_DONE:
  443. return ngtcp2_pkt_decode_handshake_done_frame(&dest->handshake_done,
  444. payload, payloadlen);
  445. case NGTCP2_FRAME_DATAGRAM:
  446. case NGTCP2_FRAME_DATAGRAM_LEN:
  447. return ngtcp2_pkt_decode_datagram_frame(&dest->datagram, payload,
  448. payloadlen);
  449. default:
  450. if ((type & ~(NGTCP2_FRAME_STREAM - 1)) == NGTCP2_FRAME_STREAM) {
  451. return ngtcp2_pkt_decode_stream_frame(&dest->stream, payload, payloadlen);
  452. }
  453. /* For frame types > 0xff, use ngtcp2_get_uvarintlen and
  454. ngtcp2_get_uvarint to get a frame type, and then switch over
  455. it. Verify that payloadlen >= ngtcp2_get_uvarintlen(payload)
  456. before calling ngtcp2_get_uvarint(payload). */
  457. return NGTCP2_ERR_FRAME_ENCODING;
  458. }
  459. }
  460. ngtcp2_ssize ngtcp2_pkt_decode_stream_frame(ngtcp2_stream *dest,
  461. const uint8_t *payload,
  462. size_t payloadlen) {
  463. uint8_t type;
  464. size_t len = 1 + 1;
  465. const uint8_t *p;
  466. size_t datalen = 0;
  467. size_t ndatalen = 0;
  468. size_t n;
  469. uint64_t vi;
  470. if (payloadlen < len) {
  471. return NGTCP2_ERR_FRAME_ENCODING;
  472. }
  473. type = payload[0];
  474. p = payload + 1;
  475. n = ngtcp2_get_uvarintlen(p);
  476. len += n - 1;
  477. if (payloadlen < len) {
  478. return NGTCP2_ERR_FRAME_ENCODING;
  479. }
  480. p += n;
  481. if (type & NGTCP2_STREAM_OFF_BIT) {
  482. ++len;
  483. if (payloadlen < len) {
  484. return NGTCP2_ERR_FRAME_ENCODING;
  485. }
  486. n = ngtcp2_get_uvarintlen(p);
  487. len += n - 1;
  488. if (payloadlen < len) {
  489. return NGTCP2_ERR_FRAME_ENCODING;
  490. }
  491. p += n;
  492. }
  493. if (type & NGTCP2_STREAM_LEN_BIT) {
  494. ++len;
  495. if (payloadlen < len) {
  496. return NGTCP2_ERR_FRAME_ENCODING;
  497. }
  498. ndatalen = ngtcp2_get_uvarintlen(p);
  499. len += ndatalen - 1;
  500. if (payloadlen < len) {
  501. return NGTCP2_ERR_FRAME_ENCODING;
  502. }
  503. /* p = */ ngtcp2_get_uvarint(&vi, p);
  504. if (payloadlen - len < vi) {
  505. return NGTCP2_ERR_FRAME_ENCODING;
  506. }
  507. datalen = (size_t)vi;
  508. len += datalen;
  509. } else {
  510. len = payloadlen;
  511. }
  512. p = payload + 1;
  513. dest->type = NGTCP2_FRAME_STREAM;
  514. dest->flags = (uint8_t)(type & ~NGTCP2_FRAME_STREAM);
  515. dest->fin = (type & NGTCP2_STREAM_FIN_BIT) != 0;
  516. p = ngtcp2_get_varint(&dest->stream_id, p);
  517. if (type & NGTCP2_STREAM_OFF_BIT) {
  518. p = ngtcp2_get_uvarint(&dest->offset, p);
  519. } else {
  520. dest->offset = 0;
  521. }
  522. if (type & NGTCP2_STREAM_LEN_BIT) {
  523. p += ndatalen;
  524. } else {
  525. datalen = payloadlen - (size_t)(p - payload);
  526. }
  527. if (datalen) {
  528. dest->data[0].len = datalen;
  529. dest->data[0].base = (uint8_t *)p;
  530. dest->datacnt = 1;
  531. p += datalen;
  532. } else {
  533. dest->datacnt = 0;
  534. }
  535. assert((size_t)(p - payload) == len);
  536. return (ngtcp2_ssize)len;
  537. }
  538. ngtcp2_ssize ngtcp2_pkt_decode_ack_frame(ngtcp2_ack *dest,
  539. const uint8_t *payload,
  540. size_t payloadlen) {
  541. size_t rangecnt, max_rangecnt;
  542. size_t nrangecnt;
  543. size_t len = 1 + 1 + 1 + 1 + 1;
  544. const uint8_t *p;
  545. size_t i, j;
  546. ngtcp2_ack_range *range;
  547. size_t n;
  548. uint8_t type;
  549. uint64_t vi;
  550. if (payloadlen < len) {
  551. return NGTCP2_ERR_FRAME_ENCODING;
  552. }
  553. type = payload[0];
  554. p = payload + 1;
  555. /* Largest Acknowledged */
  556. n = ngtcp2_get_uvarintlen(p);
  557. len += n - 1;
  558. if (payloadlen < len) {
  559. return NGTCP2_ERR_FRAME_ENCODING;
  560. }
  561. p += n;
  562. /* ACK Delay */
  563. n = ngtcp2_get_uvarintlen(p);
  564. len += n - 1;
  565. if (payloadlen < len) {
  566. return NGTCP2_ERR_FRAME_ENCODING;
  567. }
  568. p += n;
  569. /* ACK Range Count */
  570. nrangecnt = ngtcp2_get_uvarintlen(p);
  571. len += nrangecnt - 1;
  572. if (payloadlen < len) {
  573. return NGTCP2_ERR_FRAME_ENCODING;
  574. }
  575. p = ngtcp2_get_uvarint(&vi, p);
  576. if (vi > SIZE_MAX / (1 + 1) || payloadlen - len < vi * (1 + 1)) {
  577. return NGTCP2_ERR_FRAME_ENCODING;
  578. }
  579. rangecnt = (size_t)vi;
  580. len += rangecnt * (1 + 1);
  581. /* First ACK Range */
  582. n = ngtcp2_get_uvarintlen(p);
  583. len += n - 1;
  584. if (payloadlen < len) {
  585. return NGTCP2_ERR_FRAME_ENCODING;
  586. }
  587. p += n;
  588. for (i = 0; i < rangecnt; ++i) {
  589. /* Gap, and Additional ACK Range */
  590. for (j = 0; j < 2; ++j) {
  591. n = ngtcp2_get_uvarintlen(p);
  592. len += n - 1;
  593. if (payloadlen < len) {
  594. return NGTCP2_ERR_FRAME_ENCODING;
  595. }
  596. p += n;
  597. }
  598. }
  599. if (type == NGTCP2_FRAME_ACK_ECN) {
  600. len += 3;
  601. if (payloadlen < len) {
  602. return NGTCP2_ERR_FRAME_ENCODING;
  603. }
  604. for (i = 0; i < 3; ++i) {
  605. n = ngtcp2_get_uvarintlen(p);
  606. len += n - 1;
  607. if (payloadlen < len) {
  608. return NGTCP2_ERR_FRAME_ENCODING;
  609. }
  610. p += n;
  611. }
  612. }
  613. /* TODO We might not decode all ranges. It could be very large. */
  614. max_rangecnt = ngtcp2_min_size(NGTCP2_MAX_ACK_RANGES, rangecnt);
  615. p = payload + 1;
  616. dest->type = type;
  617. p = ngtcp2_get_varint(&dest->largest_ack, p);
  618. p = ngtcp2_get_uvarint(&dest->ack_delay, p);
  619. /* This value will be assigned in the upper layer. */
  620. dest->ack_delay_unscaled = 0;
  621. dest->rangecnt = max_rangecnt;
  622. p += nrangecnt;
  623. p = ngtcp2_get_uvarint(&dest->first_ack_range, p);
  624. for (i = 0; i < max_rangecnt; ++i) {
  625. range = &dest->ranges[i];
  626. p = ngtcp2_get_uvarint(&range->gap, p);
  627. p = ngtcp2_get_uvarint(&range->len, p);
  628. }
  629. for (; i < rangecnt; ++i) {
  630. p += ngtcp2_get_uvarintlen(p);
  631. p += ngtcp2_get_uvarintlen(p);
  632. }
  633. if (type == NGTCP2_FRAME_ACK_ECN) {
  634. p = ngtcp2_get_uvarint(&dest->ecn.ect0, p);
  635. p = ngtcp2_get_uvarint(&dest->ecn.ect1, p);
  636. p = ngtcp2_get_uvarint(&dest->ecn.ce, p);
  637. }
  638. assert((size_t)(p - payload) == len);
  639. return (ngtcp2_ssize)len;
  640. }
  641. ngtcp2_ssize ngtcp2_pkt_decode_padding_frame(ngtcp2_padding *dest,
  642. const uint8_t *payload,
  643. size_t payloadlen) {
  644. const uint8_t *p, *ep;
  645. assert(payloadlen > 0);
  646. p = payload + 1;
  647. ep = payload + payloadlen;
  648. for (; p != ep && *p == NGTCP2_FRAME_PADDING; ++p)
  649. ;
  650. dest->type = NGTCP2_FRAME_PADDING;
  651. dest->len = (size_t)(p - payload);
  652. return (ngtcp2_ssize)dest->len;
  653. }
  654. ngtcp2_ssize ngtcp2_pkt_decode_reset_stream_frame(ngtcp2_reset_stream *dest,
  655. const uint8_t *payload,
  656. size_t payloadlen) {
  657. size_t len = 1 + 1 + 1 + 1;
  658. const uint8_t *p;
  659. size_t n;
  660. if (payloadlen < len) {
  661. return NGTCP2_ERR_FRAME_ENCODING;
  662. }
  663. p = payload + 1;
  664. n = ngtcp2_get_uvarintlen(p);
  665. len += n - 1;
  666. if (payloadlen < len) {
  667. return NGTCP2_ERR_FRAME_ENCODING;
  668. }
  669. p += n;
  670. n = ngtcp2_get_uvarintlen(p);
  671. len += n - 1;
  672. if (payloadlen < len) {
  673. return NGTCP2_ERR_FRAME_ENCODING;
  674. }
  675. p += n;
  676. n = ngtcp2_get_uvarintlen(p);
  677. len += n - 1;
  678. if (payloadlen < len) {
  679. return NGTCP2_ERR_FRAME_ENCODING;
  680. }
  681. p = payload + 1;
  682. dest->type = NGTCP2_FRAME_RESET_STREAM;
  683. p = ngtcp2_get_varint(&dest->stream_id, p);
  684. p = ngtcp2_get_uvarint(&dest->app_error_code, p);
  685. p = ngtcp2_get_uvarint(&dest->final_size, p);
  686. assert((size_t)(p - payload) == len);
  687. return (ngtcp2_ssize)len;
  688. }
  689. ngtcp2_ssize ngtcp2_pkt_decode_connection_close_frame(
  690. ngtcp2_connection_close *dest, const uint8_t *payload, size_t payloadlen) {
  691. size_t len = 1 + 1 + 1;
  692. const uint8_t *p;
  693. size_t reasonlen;
  694. size_t nreasonlen;
  695. size_t n;
  696. uint8_t type;
  697. uint64_t vi;
  698. if (payloadlen < len) {
  699. return NGTCP2_ERR_FRAME_ENCODING;
  700. }
  701. type = payload[0];
  702. p = payload + 1;
  703. n = ngtcp2_get_uvarintlen(p);
  704. len += n - 1;
  705. if (payloadlen < len) {
  706. return NGTCP2_ERR_FRAME_ENCODING;
  707. }
  708. p += n;
  709. if (type == NGTCP2_FRAME_CONNECTION_CLOSE) {
  710. ++len;
  711. n = ngtcp2_get_uvarintlen(p);
  712. len += n - 1;
  713. if (payloadlen < len) {
  714. return NGTCP2_ERR_FRAME_ENCODING;
  715. }
  716. p += n;
  717. }
  718. nreasonlen = ngtcp2_get_uvarintlen(p);
  719. len += nreasonlen - 1;
  720. if (payloadlen < len) {
  721. return NGTCP2_ERR_FRAME_ENCODING;
  722. }
  723. p = ngtcp2_get_uvarint(&vi, p);
  724. if (payloadlen - len < vi) {
  725. return NGTCP2_ERR_FRAME_ENCODING;
  726. }
  727. reasonlen = (size_t)vi;
  728. len += reasonlen;
  729. p = payload + 1;
  730. dest->type = type;
  731. p = ngtcp2_get_uvarint(&dest->error_code, p);
  732. if (type == NGTCP2_FRAME_CONNECTION_CLOSE) {
  733. p = ngtcp2_get_uvarint(&dest->frame_type, p);
  734. } else {
  735. dest->frame_type = 0;
  736. }
  737. dest->reasonlen = reasonlen;
  738. p += nreasonlen;
  739. if (reasonlen == 0) {
  740. dest->reason = NULL;
  741. } else {
  742. dest->reason = (uint8_t *)p;
  743. p += reasonlen;
  744. }
  745. assert((size_t)(p - payload) == len);
  746. return (ngtcp2_ssize)len;
  747. }
  748. ngtcp2_ssize ngtcp2_pkt_decode_max_data_frame(ngtcp2_max_data *dest,
  749. const uint8_t *payload,
  750. size_t payloadlen) {
  751. size_t len = 1 + 1;
  752. const uint8_t *p;
  753. size_t n;
  754. if (payloadlen < len) {
  755. return NGTCP2_ERR_FRAME_ENCODING;
  756. }
  757. p = payload + 1;
  758. n = ngtcp2_get_uvarintlen(p);
  759. len += n - 1;
  760. if (payloadlen < len) {
  761. return NGTCP2_ERR_FRAME_ENCODING;
  762. }
  763. dest->type = NGTCP2_FRAME_MAX_DATA;
  764. p = ngtcp2_get_uvarint(&dest->max_data, p);
  765. assert((size_t)(p - payload) == len);
  766. return (ngtcp2_ssize)len;
  767. }
  768. ngtcp2_ssize ngtcp2_pkt_decode_max_stream_data_frame(
  769. ngtcp2_max_stream_data *dest, const uint8_t *payload, size_t payloadlen) {
  770. size_t len = 1 + 1 + 1;
  771. const uint8_t *p;
  772. size_t n;
  773. if (payloadlen < len) {
  774. return NGTCP2_ERR_FRAME_ENCODING;
  775. }
  776. p = payload + 1;
  777. n = ngtcp2_get_uvarintlen(p);
  778. len += n - 1;
  779. if (payloadlen < len) {
  780. return NGTCP2_ERR_FRAME_ENCODING;
  781. }
  782. p += n;
  783. n = ngtcp2_get_uvarintlen(p);
  784. len += n - 1;
  785. if (payloadlen < len) {
  786. return NGTCP2_ERR_FRAME_ENCODING;
  787. }
  788. p = payload + 1;
  789. dest->type = NGTCP2_FRAME_MAX_STREAM_DATA;
  790. p = ngtcp2_get_varint(&dest->stream_id, p);
  791. p = ngtcp2_get_uvarint(&dest->max_stream_data, p);
  792. assert((size_t)(p - payload) == len);
  793. return (ngtcp2_ssize)len;
  794. }
  795. ngtcp2_ssize ngtcp2_pkt_decode_max_streams_frame(ngtcp2_max_streams *dest,
  796. const uint8_t *payload,
  797. size_t payloadlen) {
  798. size_t len = 1 + 1;
  799. const uint8_t *p;
  800. size_t n;
  801. if (payloadlen < len) {
  802. return NGTCP2_ERR_FRAME_ENCODING;
  803. }
  804. p = payload + 1;
  805. n = ngtcp2_get_uvarintlen(p);
  806. len += n - 1;
  807. if (payloadlen < len) {
  808. return NGTCP2_ERR_FRAME_ENCODING;
  809. }
  810. dest->type = payload[0];
  811. p = ngtcp2_get_uvarint(&dest->max_streams, p);
  812. assert((size_t)(p - payload) == len);
  813. return (ngtcp2_ssize)len;
  814. }
  815. ngtcp2_ssize ngtcp2_pkt_decode_ping_frame(ngtcp2_ping *dest,
  816. const uint8_t *payload,
  817. size_t payloadlen) {
  818. (void)payload;
  819. (void)payloadlen;
  820. dest->type = NGTCP2_FRAME_PING;
  821. return 1;
  822. }
  823. ngtcp2_ssize ngtcp2_pkt_decode_data_blocked_frame(ngtcp2_data_blocked *dest,
  824. const uint8_t *payload,
  825. size_t payloadlen) {
  826. size_t len = 1 + 1;
  827. const uint8_t *p;
  828. size_t n;
  829. if (payloadlen < len) {
  830. return NGTCP2_ERR_FRAME_ENCODING;
  831. }
  832. p = payload + 1;
  833. n = ngtcp2_get_uvarintlen(p);
  834. len += n - 1;
  835. if (payloadlen < len) {
  836. return NGTCP2_ERR_FRAME_ENCODING;
  837. }
  838. dest->type = NGTCP2_FRAME_DATA_BLOCKED;
  839. p = ngtcp2_get_uvarint(&dest->offset, p);
  840. assert((size_t)(p - payload) == len);
  841. return (ngtcp2_ssize)len;
  842. }
  843. ngtcp2_ssize ngtcp2_pkt_decode_stream_data_blocked_frame(
  844. ngtcp2_stream_data_blocked *dest, const uint8_t *payload, size_t payloadlen) {
  845. size_t len = 1 + 1 + 1;
  846. const uint8_t *p;
  847. size_t n;
  848. if (payloadlen < len) {
  849. return NGTCP2_ERR_FRAME_ENCODING;
  850. }
  851. p = payload + 1;
  852. n = ngtcp2_get_uvarintlen(p);
  853. len += n - 1;
  854. if (payloadlen < len) {
  855. return NGTCP2_ERR_FRAME_ENCODING;
  856. }
  857. p += n;
  858. n = ngtcp2_get_uvarintlen(p);
  859. len += n - 1;
  860. if (payloadlen < len) {
  861. return NGTCP2_ERR_FRAME_ENCODING;
  862. }
  863. p = payload + 1;
  864. dest->type = NGTCP2_FRAME_STREAM_DATA_BLOCKED;
  865. p = ngtcp2_get_varint(&dest->stream_id, p);
  866. p = ngtcp2_get_uvarint(&dest->offset, p);
  867. assert((size_t)(p - payload) == len);
  868. return (ngtcp2_ssize)len;
  869. }
  870. ngtcp2_ssize ngtcp2_pkt_decode_streams_blocked_frame(
  871. ngtcp2_streams_blocked *dest, const uint8_t *payload, size_t payloadlen) {
  872. size_t len = 1 + 1;
  873. const uint8_t *p;
  874. size_t n;
  875. if (payloadlen < len) {
  876. return NGTCP2_ERR_FRAME_ENCODING;
  877. }
  878. p = payload + 1;
  879. n = ngtcp2_get_uvarintlen(p);
  880. len += n - 1;
  881. if (payloadlen < len) {
  882. return NGTCP2_ERR_FRAME_ENCODING;
  883. }
  884. dest->type = payload[0];
  885. p = ngtcp2_get_uvarint(&dest->max_streams, p);
  886. assert((size_t)(p - payload) == len);
  887. return (ngtcp2_ssize)len;
  888. }
  889. ngtcp2_ssize ngtcp2_pkt_decode_new_connection_id_frame(
  890. ngtcp2_new_connection_id *dest, const uint8_t *payload, size_t payloadlen) {
  891. size_t len = 1 + 1 + 1 + 1 + 16;
  892. const uint8_t *p;
  893. size_t n;
  894. size_t cil;
  895. if (payloadlen < len) {
  896. return NGTCP2_ERR_FRAME_ENCODING;
  897. }
  898. p = payload + 1;
  899. n = ngtcp2_get_uvarintlen(p);
  900. len += n - 1;
  901. if (payloadlen < len) {
  902. return NGTCP2_ERR_FRAME_ENCODING;
  903. }
  904. p += n;
  905. n = ngtcp2_get_uvarintlen(p);
  906. len += n - 1;
  907. if (payloadlen < len) {
  908. return NGTCP2_ERR_FRAME_ENCODING;
  909. }
  910. p += n;
  911. cil = *p;
  912. if (cil < NGTCP2_MIN_CIDLEN || cil > NGTCP2_MAX_CIDLEN) {
  913. return NGTCP2_ERR_FRAME_ENCODING;
  914. }
  915. len += cil;
  916. if (payloadlen < len) {
  917. return NGTCP2_ERR_FRAME_ENCODING;
  918. }
  919. p = payload + 1;
  920. dest->type = NGTCP2_FRAME_NEW_CONNECTION_ID;
  921. p = ngtcp2_get_uvarint(&dest->seq, p);
  922. p = ngtcp2_get_uvarint(&dest->retire_prior_to, p);
  923. ++p;
  924. ngtcp2_cid_init(&dest->cid, p, cil);
  925. p += cil;
  926. p = ngtcp2_get_bytes(dest->stateless_reset_token, p,
  927. NGTCP2_STATELESS_RESET_TOKENLEN);
  928. assert((size_t)(p - payload) == len);
  929. return (ngtcp2_ssize)len;
  930. }
  931. ngtcp2_ssize ngtcp2_pkt_decode_stop_sending_frame(ngtcp2_stop_sending *dest,
  932. const uint8_t *payload,
  933. size_t payloadlen) {
  934. size_t len = 1 + 1 + 1;
  935. const uint8_t *p;
  936. size_t n;
  937. if (payloadlen < len) {
  938. return NGTCP2_ERR_FRAME_ENCODING;
  939. }
  940. p = payload + 1;
  941. n = ngtcp2_get_uvarintlen(p);
  942. len += n - 1;
  943. if (payloadlen < len) {
  944. return NGTCP2_ERR_FRAME_ENCODING;
  945. }
  946. p += n;
  947. n = ngtcp2_get_uvarintlen(p);
  948. len += n - 1;
  949. if (payloadlen < len) {
  950. return NGTCP2_ERR_FRAME_ENCODING;
  951. }
  952. p = payload + 1;
  953. dest->type = NGTCP2_FRAME_STOP_SENDING;
  954. p = ngtcp2_get_varint(&dest->stream_id, p);
  955. p = ngtcp2_get_uvarint(&dest->app_error_code, p);
  956. assert((size_t)(p - payload) == len);
  957. return (ngtcp2_ssize)len;
  958. }
  959. ngtcp2_ssize ngtcp2_pkt_decode_path_challenge_frame(ngtcp2_path_challenge *dest,
  960. const uint8_t *payload,
  961. size_t payloadlen) {
  962. size_t len = 1 + 8;
  963. const uint8_t *p;
  964. if (payloadlen < len) {
  965. return NGTCP2_ERR_FRAME_ENCODING;
  966. }
  967. p = payload + 1;
  968. dest->type = NGTCP2_FRAME_PATH_CHALLENGE;
  969. ngtcp2_cpymem(dest->data, p, sizeof(dest->data));
  970. p += sizeof(dest->data);
  971. assert((size_t)(p - payload) == len);
  972. return (ngtcp2_ssize)len;
  973. }
  974. ngtcp2_ssize ngtcp2_pkt_decode_path_response_frame(ngtcp2_path_response *dest,
  975. const uint8_t *payload,
  976. size_t payloadlen) {
  977. size_t len = 1 + 8;
  978. const uint8_t *p;
  979. if (payloadlen < len) {
  980. return NGTCP2_ERR_FRAME_ENCODING;
  981. }
  982. p = payload + 1;
  983. dest->type = NGTCP2_FRAME_PATH_RESPONSE;
  984. ngtcp2_cpymem(dest->data, p, sizeof(dest->data));
  985. p += sizeof(dest->data);
  986. assert((size_t)(p - payload) == len);
  987. return (ngtcp2_ssize)len;
  988. }
  989. ngtcp2_ssize ngtcp2_pkt_decode_crypto_frame(ngtcp2_stream *dest,
  990. const uint8_t *payload,
  991. size_t payloadlen) {
  992. size_t len = 1 + 1 + 1;
  993. const uint8_t *p;
  994. size_t datalen;
  995. size_t ndatalen;
  996. size_t n;
  997. uint64_t vi;
  998. if (payloadlen < len) {
  999. return NGTCP2_ERR_FRAME_ENCODING;
  1000. }
  1001. p = payload + 1;
  1002. n = ngtcp2_get_uvarintlen(p);
  1003. len += n - 1;
  1004. if (payloadlen < len) {
  1005. return NGTCP2_ERR_FRAME_ENCODING;
  1006. }
  1007. p += n;
  1008. ndatalen = ngtcp2_get_uvarintlen(p);
  1009. len += ndatalen - 1;
  1010. if (payloadlen < len) {
  1011. return NGTCP2_ERR_FRAME_ENCODING;
  1012. }
  1013. p = ngtcp2_get_uvarint(&vi, p);
  1014. if (payloadlen - len < vi) {
  1015. return NGTCP2_ERR_FRAME_ENCODING;
  1016. }
  1017. datalen = (size_t)vi;
  1018. len += datalen;
  1019. p = payload + 1;
  1020. dest->type = NGTCP2_FRAME_CRYPTO;
  1021. dest->flags = 0;
  1022. dest->fin = 0;
  1023. dest->stream_id = 0;
  1024. p = ngtcp2_get_uvarint(&dest->offset, p);
  1025. dest->data[0].len = datalen;
  1026. p += ndatalen;
  1027. if (dest->data[0].len) {
  1028. dest->data[0].base = (uint8_t *)p;
  1029. p += dest->data[0].len;
  1030. dest->datacnt = 1;
  1031. } else {
  1032. dest->data[0].base = NULL;
  1033. dest->datacnt = 0;
  1034. }
  1035. assert((size_t)(p - payload) == len);
  1036. return (ngtcp2_ssize)len;
  1037. }
  1038. ngtcp2_ssize ngtcp2_pkt_decode_new_token_frame(ngtcp2_new_token *dest,
  1039. const uint8_t *payload,
  1040. size_t payloadlen) {
  1041. size_t len = 1 + 1;
  1042. const uint8_t *p;
  1043. size_t n;
  1044. size_t datalen;
  1045. uint64_t vi;
  1046. if (payloadlen < len) {
  1047. return NGTCP2_ERR_FRAME_ENCODING;
  1048. }
  1049. p = payload + 1;
  1050. n = ngtcp2_get_uvarintlen(p);
  1051. len += n - 1;
  1052. if (payloadlen < len) {
  1053. return NGTCP2_ERR_FRAME_ENCODING;
  1054. }
  1055. p = ngtcp2_get_uvarint(&vi, p);
  1056. if (payloadlen - len < vi) {
  1057. return NGTCP2_ERR_FRAME_ENCODING;
  1058. }
  1059. datalen = (size_t)vi;
  1060. len += datalen;
  1061. dest->type = NGTCP2_FRAME_NEW_TOKEN;
  1062. dest->tokenlen = datalen;
  1063. dest->token = (uint8_t *)p;
  1064. p += dest->tokenlen;
  1065. assert((size_t)(p - payload) == len);
  1066. return (ngtcp2_ssize)len;
  1067. }
  1068. ngtcp2_ssize
  1069. ngtcp2_pkt_decode_retire_connection_id_frame(ngtcp2_retire_connection_id *dest,
  1070. const uint8_t *payload,
  1071. size_t payloadlen) {
  1072. size_t len = 1 + 1;
  1073. const uint8_t *p;
  1074. size_t n;
  1075. if (payloadlen < len) {
  1076. return NGTCP2_ERR_FRAME_ENCODING;
  1077. }
  1078. p = payload + 1;
  1079. n = ngtcp2_get_uvarintlen(p);
  1080. len += n - 1;
  1081. if (payloadlen < len) {
  1082. return NGTCP2_ERR_FRAME_ENCODING;
  1083. }
  1084. dest->type = NGTCP2_FRAME_RETIRE_CONNECTION_ID;
  1085. p = ngtcp2_get_uvarint(&dest->seq, p);
  1086. assert((size_t)(p - payload) == len);
  1087. return (ngtcp2_ssize)len;
  1088. }
  1089. ngtcp2_ssize ngtcp2_pkt_decode_handshake_done_frame(ngtcp2_handshake_done *dest,
  1090. const uint8_t *payload,
  1091. size_t payloadlen) {
  1092. (void)payload;
  1093. (void)payloadlen;
  1094. dest->type = NGTCP2_FRAME_HANDSHAKE_DONE;
  1095. return 1;
  1096. }
  1097. ngtcp2_ssize ngtcp2_pkt_decode_datagram_frame(ngtcp2_datagram *dest,
  1098. const uint8_t *payload,
  1099. size_t payloadlen) {
  1100. size_t len = 1;
  1101. const uint8_t *p;
  1102. uint8_t type;
  1103. size_t datalen;
  1104. size_t n;
  1105. uint64_t vi;
  1106. if (payloadlen < len) {
  1107. return NGTCP2_ERR_FRAME_ENCODING;
  1108. }
  1109. type = payload[0];
  1110. p = payload + 1;
  1111. switch (type) {
  1112. case NGTCP2_FRAME_DATAGRAM:
  1113. datalen = payloadlen - 1;
  1114. len = payloadlen;
  1115. break;
  1116. case NGTCP2_FRAME_DATAGRAM_LEN:
  1117. ++len;
  1118. if (payloadlen < len) {
  1119. return NGTCP2_ERR_FRAME_ENCODING;
  1120. }
  1121. n = ngtcp2_get_uvarintlen(p);
  1122. len += n - 1;
  1123. if (payloadlen < len) {
  1124. return NGTCP2_ERR_FRAME_ENCODING;
  1125. }
  1126. p = ngtcp2_get_uvarint(&vi, p);
  1127. if (payloadlen - len < vi) {
  1128. return NGTCP2_ERR_FRAME_ENCODING;
  1129. }
  1130. datalen = (size_t)vi;
  1131. len += datalen;
  1132. break;
  1133. default:
  1134. ngtcp2_unreachable();
  1135. }
  1136. dest->type = type;
  1137. if (datalen == 0) {
  1138. dest->datacnt = 0;
  1139. dest->data = NULL;
  1140. } else {
  1141. dest->datacnt = 1;
  1142. dest->data = dest->rdata;
  1143. dest->rdata[0].len = datalen;
  1144. dest->rdata[0].base = (uint8_t *)p;
  1145. p += datalen;
  1146. }
  1147. assert((size_t)(p - payload) == len);
  1148. return (ngtcp2_ssize)len;
  1149. }
  1150. ngtcp2_ssize ngtcp2_pkt_encode_frame(uint8_t *out, size_t outlen,
  1151. ngtcp2_frame *fr) {
  1152. switch (fr->type) {
  1153. case NGTCP2_FRAME_STREAM:
  1154. return ngtcp2_pkt_encode_stream_frame(out, outlen, &fr->stream);
  1155. case NGTCP2_FRAME_ACK:
  1156. case NGTCP2_FRAME_ACK_ECN:
  1157. return ngtcp2_pkt_encode_ack_frame(out, outlen, &fr->ack);
  1158. case NGTCP2_FRAME_PADDING:
  1159. return ngtcp2_pkt_encode_padding_frame(out, outlen, &fr->padding);
  1160. case NGTCP2_FRAME_RESET_STREAM:
  1161. return ngtcp2_pkt_encode_reset_stream_frame(out, outlen, &fr->reset_stream);
  1162. case NGTCP2_FRAME_CONNECTION_CLOSE:
  1163. case NGTCP2_FRAME_CONNECTION_CLOSE_APP:
  1164. return ngtcp2_pkt_encode_connection_close_frame(out, outlen,
  1165. &fr->connection_close);
  1166. case NGTCP2_FRAME_MAX_DATA:
  1167. return ngtcp2_pkt_encode_max_data_frame(out, outlen, &fr->max_data);
  1168. case NGTCP2_FRAME_MAX_STREAM_DATA:
  1169. return ngtcp2_pkt_encode_max_stream_data_frame(out, outlen,
  1170. &fr->max_stream_data);
  1171. case NGTCP2_FRAME_MAX_STREAMS_BIDI:
  1172. case NGTCP2_FRAME_MAX_STREAMS_UNI:
  1173. return ngtcp2_pkt_encode_max_streams_frame(out, outlen, &fr->max_streams);
  1174. case NGTCP2_FRAME_PING:
  1175. return ngtcp2_pkt_encode_ping_frame(out, outlen, &fr->ping);
  1176. case NGTCP2_FRAME_DATA_BLOCKED:
  1177. return ngtcp2_pkt_encode_data_blocked_frame(out, outlen, &fr->data_blocked);
  1178. case NGTCP2_FRAME_STREAM_DATA_BLOCKED:
  1179. return ngtcp2_pkt_encode_stream_data_blocked_frame(
  1180. out, outlen, &fr->stream_data_blocked);
  1181. case NGTCP2_FRAME_STREAMS_BLOCKED_BIDI:
  1182. case NGTCP2_FRAME_STREAMS_BLOCKED_UNI:
  1183. return ngtcp2_pkt_encode_streams_blocked_frame(out, outlen,
  1184. &fr->streams_blocked);
  1185. case NGTCP2_FRAME_NEW_CONNECTION_ID:
  1186. return ngtcp2_pkt_encode_new_connection_id_frame(out, outlen,
  1187. &fr->new_connection_id);
  1188. case NGTCP2_FRAME_STOP_SENDING:
  1189. return ngtcp2_pkt_encode_stop_sending_frame(out, outlen, &fr->stop_sending);
  1190. case NGTCP2_FRAME_PATH_CHALLENGE:
  1191. return ngtcp2_pkt_encode_path_challenge_frame(out, outlen,
  1192. &fr->path_challenge);
  1193. case NGTCP2_FRAME_PATH_RESPONSE:
  1194. return ngtcp2_pkt_encode_path_response_frame(out, outlen,
  1195. &fr->path_response);
  1196. case NGTCP2_FRAME_CRYPTO:
  1197. return ngtcp2_pkt_encode_crypto_frame(out, outlen, &fr->stream);
  1198. case NGTCP2_FRAME_NEW_TOKEN:
  1199. return ngtcp2_pkt_encode_new_token_frame(out, outlen, &fr->new_token);
  1200. case NGTCP2_FRAME_RETIRE_CONNECTION_ID:
  1201. return ngtcp2_pkt_encode_retire_connection_id_frame(
  1202. out, outlen, &fr->retire_connection_id);
  1203. case NGTCP2_FRAME_HANDSHAKE_DONE:
  1204. return ngtcp2_pkt_encode_handshake_done_frame(out, outlen,
  1205. &fr->handshake_done);
  1206. case NGTCP2_FRAME_DATAGRAM:
  1207. case NGTCP2_FRAME_DATAGRAM_LEN:
  1208. return ngtcp2_pkt_encode_datagram_frame(out, outlen, &fr->datagram);
  1209. default:
  1210. return NGTCP2_ERR_INVALID_ARGUMENT;
  1211. }
  1212. }
  1213. ngtcp2_ssize ngtcp2_pkt_encode_stream_frame(uint8_t *out, size_t outlen,
  1214. ngtcp2_stream *fr) {
  1215. size_t len = 1;
  1216. uint8_t flags = NGTCP2_STREAM_LEN_BIT;
  1217. uint8_t *p;
  1218. size_t i;
  1219. size_t datalen = 0;
  1220. if (fr->fin) {
  1221. flags |= NGTCP2_STREAM_FIN_BIT;
  1222. }
  1223. if (fr->offset) {
  1224. flags |= NGTCP2_STREAM_OFF_BIT;
  1225. len += ngtcp2_put_uvarintlen(fr->offset);
  1226. }
  1227. len += ngtcp2_put_uvarintlen((uint64_t)fr->stream_id);
  1228. for (i = 0; i < fr->datacnt; ++i) {
  1229. datalen += fr->data[i].len;
  1230. }
  1231. len += ngtcp2_put_uvarintlen(datalen);
  1232. len += datalen;
  1233. if (outlen < len) {
  1234. return NGTCP2_ERR_NOBUF;
  1235. }
  1236. p = out;
  1237. *p++ = flags | NGTCP2_FRAME_STREAM;
  1238. fr->flags = flags;
  1239. p = ngtcp2_put_uvarint(p, (uint64_t)fr->stream_id);
  1240. if (fr->offset) {
  1241. p = ngtcp2_put_uvarint(p, fr->offset);
  1242. }
  1243. p = ngtcp2_put_uvarint(p, datalen);
  1244. for (i = 0; i < fr->datacnt; ++i) {
  1245. assert(fr->data[i].len);
  1246. assert(fr->data[i].base);
  1247. p = ngtcp2_cpymem(p, fr->data[i].base, fr->data[i].len);
  1248. }
  1249. assert((size_t)(p - out) == len);
  1250. return (ngtcp2_ssize)len;
  1251. }
  1252. ngtcp2_ssize ngtcp2_pkt_encode_ack_frame(uint8_t *out, size_t outlen,
  1253. const ngtcp2_ack *fr) {
  1254. size_t len = 1 + ngtcp2_put_uvarintlen((uint64_t)fr->largest_ack) +
  1255. ngtcp2_put_uvarintlen(fr->ack_delay) +
  1256. ngtcp2_put_uvarintlen(fr->rangecnt) +
  1257. ngtcp2_put_uvarintlen(fr->first_ack_range);
  1258. uint8_t *p;
  1259. size_t i;
  1260. const ngtcp2_ack_range *range;
  1261. for (i = 0; i < fr->rangecnt; ++i) {
  1262. range = &fr->ranges[i];
  1263. len += ngtcp2_put_uvarintlen(range->gap);
  1264. len += ngtcp2_put_uvarintlen(range->len);
  1265. }
  1266. if (fr->type == NGTCP2_FRAME_ACK_ECN) {
  1267. len += ngtcp2_put_uvarintlen(fr->ecn.ect0) +
  1268. ngtcp2_put_uvarintlen(fr->ecn.ect1) +
  1269. ngtcp2_put_uvarintlen(fr->ecn.ce);
  1270. }
  1271. if (outlen < len) {
  1272. return NGTCP2_ERR_NOBUF;
  1273. }
  1274. p = out;
  1275. *p++ = (uint8_t)fr->type;
  1276. p = ngtcp2_put_uvarint(p, (uint64_t)fr->largest_ack);
  1277. p = ngtcp2_put_uvarint(p, fr->ack_delay);
  1278. p = ngtcp2_put_uvarint(p, fr->rangecnt);
  1279. p = ngtcp2_put_uvarint(p, fr->first_ack_range);
  1280. for (i = 0; i < fr->rangecnt; ++i) {
  1281. range = &fr->ranges[i];
  1282. p = ngtcp2_put_uvarint(p, range->gap);
  1283. p = ngtcp2_put_uvarint(p, range->len);
  1284. }
  1285. if (fr->type == NGTCP2_FRAME_ACK_ECN) {
  1286. p = ngtcp2_put_uvarint(p, fr->ecn.ect0);
  1287. p = ngtcp2_put_uvarint(p, fr->ecn.ect1);
  1288. p = ngtcp2_put_uvarint(p, fr->ecn.ce);
  1289. }
  1290. assert((size_t)(p - out) == len);
  1291. return (ngtcp2_ssize)len;
  1292. }
  1293. ngtcp2_ssize ngtcp2_pkt_encode_padding_frame(uint8_t *out, size_t outlen,
  1294. const ngtcp2_padding *fr) {
  1295. if (outlen < fr->len) {
  1296. return NGTCP2_ERR_NOBUF;
  1297. }
  1298. memset(out, 0, fr->len);
  1299. return (ngtcp2_ssize)fr->len;
  1300. }
  1301. ngtcp2_ssize
  1302. ngtcp2_pkt_encode_reset_stream_frame(uint8_t *out, size_t outlen,
  1303. const ngtcp2_reset_stream *fr) {
  1304. size_t len = 1 + ngtcp2_put_uvarintlen((uint64_t)fr->stream_id) +
  1305. ngtcp2_put_uvarintlen(fr->app_error_code) +
  1306. ngtcp2_put_uvarintlen(fr->final_size);
  1307. uint8_t *p;
  1308. if (outlen < len) {
  1309. return NGTCP2_ERR_NOBUF;
  1310. }
  1311. p = out;
  1312. *p++ = NGTCP2_FRAME_RESET_STREAM;
  1313. p = ngtcp2_put_uvarint(p, (uint64_t)fr->stream_id);
  1314. p = ngtcp2_put_uvarint(p, fr->app_error_code);
  1315. p = ngtcp2_put_uvarint(p, fr->final_size);
  1316. assert((size_t)(p - out) == len);
  1317. return (ngtcp2_ssize)len;
  1318. }
  1319. ngtcp2_ssize
  1320. ngtcp2_pkt_encode_connection_close_frame(uint8_t *out, size_t outlen,
  1321. const ngtcp2_connection_close *fr) {
  1322. size_t len = 1 + ngtcp2_put_uvarintlen(fr->error_code) +
  1323. (fr->type == NGTCP2_FRAME_CONNECTION_CLOSE
  1324. ? ngtcp2_put_uvarintlen(fr->frame_type)
  1325. : 0) +
  1326. ngtcp2_put_uvarintlen(fr->reasonlen) + fr->reasonlen;
  1327. uint8_t *p;
  1328. if (outlen < len) {
  1329. return NGTCP2_ERR_NOBUF;
  1330. }
  1331. p = out;
  1332. *p++ = (uint8_t)fr->type;
  1333. p = ngtcp2_put_uvarint(p, fr->error_code);
  1334. if (fr->type == NGTCP2_FRAME_CONNECTION_CLOSE) {
  1335. p = ngtcp2_put_uvarint(p, fr->frame_type);
  1336. }
  1337. p = ngtcp2_put_uvarint(p, fr->reasonlen);
  1338. if (fr->reasonlen) {
  1339. p = ngtcp2_cpymem(p, fr->reason, fr->reasonlen);
  1340. }
  1341. assert((size_t)(p - out) == len);
  1342. return (ngtcp2_ssize)len;
  1343. }
  1344. ngtcp2_ssize ngtcp2_pkt_encode_max_data_frame(uint8_t *out, size_t outlen,
  1345. const ngtcp2_max_data *fr) {
  1346. size_t len = 1 + ngtcp2_put_uvarintlen(fr->max_data);
  1347. uint8_t *p;
  1348. if (outlen < len) {
  1349. return NGTCP2_ERR_NOBUF;
  1350. }
  1351. p = out;
  1352. *p++ = NGTCP2_FRAME_MAX_DATA;
  1353. p = ngtcp2_put_uvarint(p, fr->max_data);
  1354. assert((size_t)(p - out) == len);
  1355. return (ngtcp2_ssize)len;
  1356. }
  1357. ngtcp2_ssize
  1358. ngtcp2_pkt_encode_max_stream_data_frame(uint8_t *out, size_t outlen,
  1359. const ngtcp2_max_stream_data *fr) {
  1360. size_t len = 1 + ngtcp2_put_uvarintlen((uint64_t)fr->stream_id) +
  1361. ngtcp2_put_uvarintlen(fr->max_stream_data);
  1362. uint8_t *p;
  1363. if (outlen < len) {
  1364. return NGTCP2_ERR_NOBUF;
  1365. }
  1366. p = out;
  1367. *p++ = NGTCP2_FRAME_MAX_STREAM_DATA;
  1368. p = ngtcp2_put_uvarint(p, (uint64_t)fr->stream_id);
  1369. p = ngtcp2_put_uvarint(p, fr->max_stream_data);
  1370. assert((size_t)(p - out) == len);
  1371. return (ngtcp2_ssize)len;
  1372. }
  1373. ngtcp2_ssize ngtcp2_pkt_encode_max_streams_frame(uint8_t *out, size_t outlen,
  1374. const ngtcp2_max_streams *fr) {
  1375. size_t len = 1 + ngtcp2_put_uvarintlen(fr->max_streams);
  1376. uint8_t *p;
  1377. if (outlen < len) {
  1378. return NGTCP2_ERR_NOBUF;
  1379. }
  1380. p = out;
  1381. *p++ = (uint8_t)fr->type;
  1382. p = ngtcp2_put_uvarint(p, fr->max_streams);
  1383. assert((size_t)(p - out) == len);
  1384. return (ngtcp2_ssize)len;
  1385. }
  1386. ngtcp2_ssize ngtcp2_pkt_encode_ping_frame(uint8_t *out, size_t outlen,
  1387. const ngtcp2_ping *fr) {
  1388. (void)fr;
  1389. if (outlen < 1) {
  1390. return NGTCP2_ERR_NOBUF;
  1391. }
  1392. *out++ = NGTCP2_FRAME_PING;
  1393. return 1;
  1394. }
  1395. ngtcp2_ssize
  1396. ngtcp2_pkt_encode_data_blocked_frame(uint8_t *out, size_t outlen,
  1397. const ngtcp2_data_blocked *fr) {
  1398. size_t len = 1 + ngtcp2_put_uvarintlen(fr->offset);
  1399. uint8_t *p;
  1400. if (outlen < len) {
  1401. return NGTCP2_ERR_NOBUF;
  1402. }
  1403. p = out;
  1404. *p++ = NGTCP2_FRAME_DATA_BLOCKED;
  1405. p = ngtcp2_put_uvarint(p, fr->offset);
  1406. assert((size_t)(p - out) == len);
  1407. return (ngtcp2_ssize)len;
  1408. }
  1409. ngtcp2_ssize ngtcp2_pkt_encode_stream_data_blocked_frame(
  1410. uint8_t *out, size_t outlen, const ngtcp2_stream_data_blocked *fr) {
  1411. size_t len = 1 + ngtcp2_put_uvarintlen((uint64_t)fr->stream_id) +
  1412. ngtcp2_put_uvarintlen(fr->offset);
  1413. uint8_t *p;
  1414. if (outlen < len) {
  1415. return NGTCP2_ERR_NOBUF;
  1416. }
  1417. p = out;
  1418. *p++ = NGTCP2_FRAME_STREAM_DATA_BLOCKED;
  1419. p = ngtcp2_put_uvarint(p, (uint64_t)fr->stream_id);
  1420. p = ngtcp2_put_uvarint(p, fr->offset);
  1421. assert((size_t)(p - out) == len);
  1422. return (ngtcp2_ssize)len;
  1423. }
  1424. ngtcp2_ssize
  1425. ngtcp2_pkt_encode_streams_blocked_frame(uint8_t *out, size_t outlen,
  1426. const ngtcp2_streams_blocked *fr) {
  1427. size_t len = 1 + ngtcp2_put_uvarintlen(fr->max_streams);
  1428. uint8_t *p;
  1429. if (outlen < len) {
  1430. return NGTCP2_ERR_NOBUF;
  1431. }
  1432. p = out;
  1433. *p++ = (uint8_t)fr->type;
  1434. p = ngtcp2_put_uvarint(p, fr->max_streams);
  1435. assert((size_t)(p - out) == len);
  1436. return (ngtcp2_ssize)len;
  1437. }
  1438. ngtcp2_ssize
  1439. ngtcp2_pkt_encode_new_connection_id_frame(uint8_t *out, size_t outlen,
  1440. const ngtcp2_new_connection_id *fr) {
  1441. size_t len = 1 + ngtcp2_put_uvarintlen(fr->seq) +
  1442. ngtcp2_put_uvarintlen(fr->retire_prior_to) + 1 +
  1443. fr->cid.datalen + NGTCP2_STATELESS_RESET_TOKENLEN;
  1444. uint8_t *p;
  1445. if (outlen < len) {
  1446. return NGTCP2_ERR_NOBUF;
  1447. }
  1448. p = out;
  1449. *p++ = NGTCP2_FRAME_NEW_CONNECTION_ID;
  1450. p = ngtcp2_put_uvarint(p, fr->seq);
  1451. p = ngtcp2_put_uvarint(p, fr->retire_prior_to);
  1452. *p++ = (uint8_t)fr->cid.datalen;
  1453. p = ngtcp2_cpymem(p, fr->cid.data, fr->cid.datalen);
  1454. p = ngtcp2_cpymem(p, fr->stateless_reset_token,
  1455. NGTCP2_STATELESS_RESET_TOKENLEN);
  1456. assert((size_t)(p - out) == len);
  1457. return (ngtcp2_ssize)len;
  1458. }
  1459. ngtcp2_ssize
  1460. ngtcp2_pkt_encode_stop_sending_frame(uint8_t *out, size_t outlen,
  1461. const ngtcp2_stop_sending *fr) {
  1462. size_t len = 1 + ngtcp2_put_uvarintlen((uint64_t)fr->stream_id) +
  1463. ngtcp2_put_uvarintlen(fr->app_error_code);
  1464. uint8_t *p;
  1465. if (outlen < len) {
  1466. return NGTCP2_ERR_NOBUF;
  1467. }
  1468. p = out;
  1469. *p++ = NGTCP2_FRAME_STOP_SENDING;
  1470. p = ngtcp2_put_uvarint(p, (uint64_t)fr->stream_id);
  1471. p = ngtcp2_put_uvarint(p, fr->app_error_code);
  1472. assert((size_t)(p - out) == len);
  1473. return (ngtcp2_ssize)len;
  1474. }
  1475. ngtcp2_ssize
  1476. ngtcp2_pkt_encode_path_challenge_frame(uint8_t *out, size_t outlen,
  1477. const ngtcp2_path_challenge *fr) {
  1478. size_t len = 1 + 8;
  1479. uint8_t *p;
  1480. if (outlen < len) {
  1481. return NGTCP2_ERR_NOBUF;
  1482. }
  1483. p = out;
  1484. *p++ = NGTCP2_FRAME_PATH_CHALLENGE;
  1485. p = ngtcp2_cpymem(p, fr->data, sizeof(fr->data));
  1486. assert((size_t)(p - out) == len);
  1487. return (ngtcp2_ssize)len;
  1488. }
  1489. ngtcp2_ssize
  1490. ngtcp2_pkt_encode_path_response_frame(uint8_t *out, size_t outlen,
  1491. const ngtcp2_path_response *fr) {
  1492. size_t len = 1 + 8;
  1493. uint8_t *p;
  1494. if (outlen < len) {
  1495. return NGTCP2_ERR_NOBUF;
  1496. }
  1497. p = out;
  1498. *p++ = NGTCP2_FRAME_PATH_RESPONSE;
  1499. p = ngtcp2_cpymem(p, fr->data, sizeof(fr->data));
  1500. assert((size_t)(p - out) == len);
  1501. return (ngtcp2_ssize)len;
  1502. }
  1503. ngtcp2_ssize ngtcp2_pkt_encode_crypto_frame(uint8_t *out, size_t outlen,
  1504. const ngtcp2_stream *fr) {
  1505. size_t len = 1;
  1506. uint8_t *p;
  1507. size_t i;
  1508. size_t datalen = 0;
  1509. len += ngtcp2_put_uvarintlen(fr->offset);
  1510. for (i = 0; i < fr->datacnt; ++i) {
  1511. datalen += fr->data[i].len;
  1512. }
  1513. len += ngtcp2_put_uvarintlen(datalen);
  1514. len += datalen;
  1515. if (outlen < len) {
  1516. return NGTCP2_ERR_NOBUF;
  1517. }
  1518. p = out;
  1519. *p++ = NGTCP2_FRAME_CRYPTO;
  1520. p = ngtcp2_put_uvarint(p, fr->offset);
  1521. p = ngtcp2_put_uvarint(p, datalen);
  1522. for (i = 0; i < fr->datacnt; ++i) {
  1523. assert(fr->data[i].base);
  1524. p = ngtcp2_cpymem(p, fr->data[i].base, fr->data[i].len);
  1525. }
  1526. assert((size_t)(p - out) == len);
  1527. return (ngtcp2_ssize)len;
  1528. }
  1529. ngtcp2_ssize ngtcp2_pkt_encode_new_token_frame(uint8_t *out, size_t outlen,
  1530. const ngtcp2_new_token *fr) {
  1531. size_t len = 1 + ngtcp2_put_uvarintlen(fr->tokenlen) + fr->tokenlen;
  1532. uint8_t *p;
  1533. assert(fr->tokenlen);
  1534. if (outlen < len) {
  1535. return NGTCP2_ERR_NOBUF;
  1536. }
  1537. p = out;
  1538. *p++ = NGTCP2_FRAME_NEW_TOKEN;
  1539. p = ngtcp2_put_uvarint(p, fr->tokenlen);
  1540. p = ngtcp2_cpymem(p, fr->token, fr->tokenlen);
  1541. assert((size_t)(p - out) == len);
  1542. return (ngtcp2_ssize)len;
  1543. }
  1544. ngtcp2_ssize ngtcp2_pkt_encode_retire_connection_id_frame(
  1545. uint8_t *out, size_t outlen, const ngtcp2_retire_connection_id *fr) {
  1546. size_t len = 1 + ngtcp2_put_uvarintlen(fr->seq);
  1547. uint8_t *p;
  1548. if (outlen < len) {
  1549. return NGTCP2_ERR_NOBUF;
  1550. }
  1551. p = out;
  1552. *p++ = NGTCP2_FRAME_RETIRE_CONNECTION_ID;
  1553. p = ngtcp2_put_uvarint(p, fr->seq);
  1554. assert((size_t)(p - out) == len);
  1555. return (ngtcp2_ssize)len;
  1556. }
  1557. ngtcp2_ssize
  1558. ngtcp2_pkt_encode_handshake_done_frame(uint8_t *out, size_t outlen,
  1559. const ngtcp2_handshake_done *fr) {
  1560. (void)fr;
  1561. if (outlen < 1) {
  1562. return NGTCP2_ERR_NOBUF;
  1563. }
  1564. *out++ = NGTCP2_FRAME_HANDSHAKE_DONE;
  1565. return 1;
  1566. }
  1567. ngtcp2_ssize ngtcp2_pkt_encode_datagram_frame(uint8_t *out, size_t outlen,
  1568. const ngtcp2_datagram *fr) {
  1569. uint64_t datalen = ngtcp2_vec_len(fr->data, fr->datacnt);
  1570. uint64_t len =
  1571. 1 +
  1572. (fr->type == NGTCP2_FRAME_DATAGRAM ? 0 : ngtcp2_put_uvarintlen(datalen)) +
  1573. datalen;
  1574. uint8_t *p;
  1575. size_t i;
  1576. assert(fr->type == NGTCP2_FRAME_DATAGRAM ||
  1577. fr->type == NGTCP2_FRAME_DATAGRAM_LEN);
  1578. if (outlen < len) {
  1579. return NGTCP2_ERR_NOBUF;
  1580. }
  1581. p = out;
  1582. *p++ = (uint8_t)fr->type;
  1583. if (fr->type == NGTCP2_FRAME_DATAGRAM_LEN) {
  1584. p = ngtcp2_put_uvarint(p, datalen);
  1585. }
  1586. for (i = 0; i < fr->datacnt; ++i) {
  1587. assert(fr->data[i].len);
  1588. assert(fr->data[i].base);
  1589. p = ngtcp2_cpymem(p, fr->data[i].base, fr->data[i].len);
  1590. }
  1591. assert((size_t)(p - out) == len);
  1592. return (ngtcp2_ssize)len;
  1593. }
  1594. ngtcp2_ssize ngtcp2_pkt_write_version_negotiation(
  1595. uint8_t *dest, size_t destlen, uint8_t unused_random, const uint8_t *dcid,
  1596. size_t dcidlen, const uint8_t *scid, size_t scidlen, const uint32_t *sv,
  1597. size_t nsv) {
  1598. size_t len = 1 + 4 + 1 + dcidlen + 1 + scidlen + nsv * 4;
  1599. uint8_t *p;
  1600. size_t i;
  1601. assert(dcidlen < 256);
  1602. assert(scidlen < 256);
  1603. if (destlen < len) {
  1604. return NGTCP2_ERR_NOBUF;
  1605. }
  1606. p = dest;
  1607. *p++ = 0xc0 | unused_random;
  1608. p = ngtcp2_put_uint32be(p, 0);
  1609. *p++ = (uint8_t)dcidlen;
  1610. if (dcidlen) {
  1611. p = ngtcp2_cpymem(p, dcid, dcidlen);
  1612. }
  1613. *p++ = (uint8_t)scidlen;
  1614. if (scidlen) {
  1615. p = ngtcp2_cpymem(p, scid, scidlen);
  1616. }
  1617. for (i = 0; i < nsv; ++i) {
  1618. p = ngtcp2_put_uint32be(p, sv[i]);
  1619. }
  1620. assert((size_t)(p - dest) == len);
  1621. return (ngtcp2_ssize)len;
  1622. }
  1623. size_t ngtcp2_pkt_decode_version_negotiation(uint32_t *dest,
  1624. const uint8_t *payload,
  1625. size_t payloadlen) {
  1626. const uint8_t *end = payload + payloadlen;
  1627. assert((payloadlen % sizeof(uint32_t)) == 0);
  1628. for (; payload != end;) {
  1629. payload = ngtcp2_get_uint32be(dest++, payload);
  1630. }
  1631. return payloadlen / sizeof(uint32_t);
  1632. }
  1633. int ngtcp2_pkt_decode_stateless_reset(ngtcp2_pkt_stateless_reset *sr,
  1634. const uint8_t *payload,
  1635. size_t payloadlen) {
  1636. const uint8_t *p = payload;
  1637. if (payloadlen <
  1638. NGTCP2_MIN_STATELESS_RESET_RANDLEN + NGTCP2_STATELESS_RESET_TOKENLEN) {
  1639. return NGTCP2_ERR_INVALID_ARGUMENT;
  1640. }
  1641. sr->rand = p;
  1642. sr->randlen = payloadlen - NGTCP2_STATELESS_RESET_TOKENLEN;
  1643. p += sr->randlen;
  1644. memcpy(sr->stateless_reset_token, p, NGTCP2_STATELESS_RESET_TOKENLEN);
  1645. return 0;
  1646. }
  1647. int ngtcp2_pkt_decode_retry(ngtcp2_pkt_retry *dest, const uint8_t *payload,
  1648. size_t payloadlen) {
  1649. size_t len = /* token */ 1 + NGTCP2_RETRY_TAGLEN;
  1650. if (payloadlen < len) {
  1651. return NGTCP2_ERR_INVALID_ARGUMENT;
  1652. }
  1653. dest->token = (uint8_t *)payload;
  1654. dest->tokenlen = (size_t)(payloadlen - NGTCP2_RETRY_TAGLEN);
  1655. ngtcp2_cpymem(dest->tag, payload + dest->tokenlen, NGTCP2_RETRY_TAGLEN);
  1656. return 0;
  1657. }
  1658. int64_t ngtcp2_pkt_adjust_pkt_num(int64_t max_pkt_num, int64_t pkt_num,
  1659. size_t pkt_numlen) {
  1660. int64_t expected = max_pkt_num + 1;
  1661. int64_t win = (int64_t)1 << (pkt_numlen * 8);
  1662. int64_t hwin = win / 2;
  1663. int64_t mask = win - 1;
  1664. int64_t cand = (expected & ~mask) | pkt_num;
  1665. if (cand <= expected - hwin) {
  1666. assert(cand <= (int64_t)NGTCP2_MAX_VARINT - win);
  1667. return cand + win;
  1668. }
  1669. if (cand > expected + hwin && cand >= win) {
  1670. return cand - win;
  1671. }
  1672. return cand;
  1673. }
  1674. int ngtcp2_pkt_validate_ack(const ngtcp2_ack *fr, int64_t min_pkt_num) {
  1675. int64_t largest_ack = fr->largest_ack;
  1676. size_t i;
  1677. if (largest_ack < (int64_t)fr->first_ack_range) {
  1678. return NGTCP2_ERR_ACK_FRAME;
  1679. }
  1680. largest_ack -= (int64_t)fr->first_ack_range;
  1681. if (largest_ack < min_pkt_num) {
  1682. return NGTCP2_ERR_PROTO;
  1683. }
  1684. for (i = 0; i < fr->rangecnt; ++i) {
  1685. if (largest_ack < (int64_t)fr->ranges[i].gap + 2) {
  1686. return NGTCP2_ERR_ACK_FRAME;
  1687. }
  1688. largest_ack -= (int64_t)fr->ranges[i].gap + 2;
  1689. if (largest_ack < (int64_t)fr->ranges[i].len) {
  1690. return NGTCP2_ERR_ACK_FRAME;
  1691. }
  1692. largest_ack -= (int64_t)fr->ranges[i].len;
  1693. if (largest_ack < min_pkt_num) {
  1694. return NGTCP2_ERR_PROTO;
  1695. }
  1696. }
  1697. return 0;
  1698. }
  1699. ngtcp2_ssize
  1700. ngtcp2_pkt_write_stateless_reset(uint8_t *dest, size_t destlen,
  1701. const uint8_t *stateless_reset_token,
  1702. const uint8_t *rand, size_t randlen) {
  1703. uint8_t *p;
  1704. if (destlen <
  1705. NGTCP2_MIN_STATELESS_RESET_RANDLEN + NGTCP2_STATELESS_RESET_TOKENLEN) {
  1706. return NGTCP2_ERR_NOBUF;
  1707. }
  1708. if (randlen < NGTCP2_MIN_STATELESS_RESET_RANDLEN) {
  1709. return NGTCP2_ERR_INVALID_ARGUMENT;
  1710. }
  1711. p = dest;
  1712. randlen = ngtcp2_min_size(destlen - NGTCP2_STATELESS_RESET_TOKENLEN, randlen);
  1713. p = ngtcp2_cpymem(p, rand, randlen);
  1714. p = ngtcp2_cpymem(p, stateless_reset_token, NGTCP2_STATELESS_RESET_TOKENLEN);
  1715. *dest = (uint8_t)((*dest & 0x7fu) | 0x40u);
  1716. return p - dest;
  1717. }
  1718. ngtcp2_ssize ngtcp2_pkt_write_retry(
  1719. uint8_t *dest, size_t destlen, uint32_t version, const ngtcp2_cid *dcid,
  1720. const ngtcp2_cid *scid, const ngtcp2_cid *odcid, const uint8_t *token,
  1721. size_t tokenlen, ngtcp2_encrypt encrypt, const ngtcp2_crypto_aead *aead,
  1722. const ngtcp2_crypto_aead_ctx *aead_ctx) {
  1723. ngtcp2_pkt_hd hd;
  1724. uint8_t pseudo_retry[1500];
  1725. ngtcp2_ssize pseudo_retrylen;
  1726. uint8_t tag[NGTCP2_RETRY_TAGLEN];
  1727. int rv;
  1728. uint8_t *p;
  1729. size_t offset;
  1730. const uint8_t *nonce;
  1731. size_t noncelen;
  1732. assert(tokenlen > 0);
  1733. assert(!ngtcp2_cid_eq(scid, odcid));
  1734. /* Retry packet is sent at most once per one connection attempt. In
  1735. the first connection attempt, client has to send random DCID
  1736. which is at least NGTCP2_MIN_INITIAL_DCIDLEN bytes long. */
  1737. if (odcid->datalen < NGTCP2_MIN_INITIAL_DCIDLEN) {
  1738. return NGTCP2_ERR_INVALID_ARGUMENT;
  1739. }
  1740. ngtcp2_pkt_hd_init(&hd, NGTCP2_PKT_FLAG_LONG_FORM, NGTCP2_PKT_RETRY, dcid,
  1741. scid, /* pkt_num = */ 0, /* pkt_numlen = */ 1, version,
  1742. /* len = */ 0);
  1743. pseudo_retrylen =
  1744. ngtcp2_pkt_encode_pseudo_retry(pseudo_retry, sizeof(pseudo_retry), &hd,
  1745. /* unused = */ 0, odcid, token, tokenlen);
  1746. if (pseudo_retrylen < 0) {
  1747. return pseudo_retrylen;
  1748. }
  1749. switch (version) {
  1750. case NGTCP2_PROTO_VER_V1:
  1751. default:
  1752. nonce = (const uint8_t *)NGTCP2_RETRY_NONCE_V1;
  1753. noncelen = sizeof(NGTCP2_RETRY_NONCE_V1) - 1;
  1754. break;
  1755. case NGTCP2_PROTO_VER_V2:
  1756. nonce = (const uint8_t *)NGTCP2_RETRY_NONCE_V2;
  1757. noncelen = sizeof(NGTCP2_RETRY_NONCE_V2) - 1;
  1758. break;
  1759. }
  1760. /* OpenSSL does not like NULL plaintext. */
  1761. rv = encrypt(tag, aead, aead_ctx, (const uint8_t *)"", 0, nonce, noncelen,
  1762. pseudo_retry, (size_t)pseudo_retrylen);
  1763. if (rv != 0) {
  1764. return rv;
  1765. }
  1766. offset = 1 + odcid->datalen;
  1767. if (destlen < (size_t)pseudo_retrylen + sizeof(tag) - offset) {
  1768. return NGTCP2_ERR_NOBUF;
  1769. }
  1770. p = ngtcp2_cpymem(dest, pseudo_retry + offset,
  1771. (size_t)pseudo_retrylen - offset);
  1772. p = ngtcp2_cpymem(p, tag, sizeof(tag));
  1773. return p - dest;
  1774. }
  1775. ngtcp2_ssize ngtcp2_pkt_encode_pseudo_retry(
  1776. uint8_t *dest, size_t destlen, const ngtcp2_pkt_hd *hd, uint8_t unused,
  1777. const ngtcp2_cid *odcid, const uint8_t *token, size_t tokenlen) {
  1778. uint8_t *p = dest;
  1779. ngtcp2_ssize nwrite;
  1780. if (destlen < 1 + odcid->datalen) {
  1781. return NGTCP2_ERR_NOBUF;
  1782. }
  1783. *p++ = (uint8_t)odcid->datalen;
  1784. p = ngtcp2_cpymem(p, odcid->data, odcid->datalen);
  1785. destlen -= (size_t)(p - dest);
  1786. nwrite = ngtcp2_pkt_encode_hd_long(p, destlen, hd);
  1787. if (nwrite < 0) {
  1788. return nwrite;
  1789. }
  1790. if (destlen < (size_t)nwrite + tokenlen) {
  1791. return NGTCP2_ERR_NOBUF;
  1792. }
  1793. *p &= 0xf0;
  1794. *p |= unused;
  1795. p += nwrite;
  1796. p = ngtcp2_cpymem(p, token, tokenlen);
  1797. return p - dest;
  1798. }
  1799. int ngtcp2_pkt_verify_retry_tag(uint32_t version, const ngtcp2_pkt_retry *retry,
  1800. const uint8_t *pkt, size_t pktlen,
  1801. ngtcp2_encrypt encrypt,
  1802. const ngtcp2_crypto_aead *aead,
  1803. const ngtcp2_crypto_aead_ctx *aead_ctx) {
  1804. uint8_t pseudo_retry[1500];
  1805. size_t pseudo_retrylen;
  1806. uint8_t *p = pseudo_retry;
  1807. int rv;
  1808. uint8_t tag[NGTCP2_RETRY_TAGLEN];
  1809. const uint8_t *nonce;
  1810. size_t noncelen;
  1811. assert(pktlen >= sizeof(retry->tag));
  1812. if (sizeof(pseudo_retry) <
  1813. 1 + retry->odcid.datalen + pktlen - sizeof(retry->tag)) {
  1814. return NGTCP2_ERR_PROTO;
  1815. }
  1816. *p++ = (uint8_t)retry->odcid.datalen;
  1817. p = ngtcp2_cpymem(p, retry->odcid.data, retry->odcid.datalen);
  1818. p = ngtcp2_cpymem(p, pkt, pktlen - sizeof(retry->tag));
  1819. pseudo_retrylen = (size_t)(p - pseudo_retry);
  1820. switch (version) {
  1821. case NGTCP2_PROTO_VER_V1:
  1822. default:
  1823. nonce = (const uint8_t *)NGTCP2_RETRY_NONCE_V1;
  1824. noncelen = sizeof(NGTCP2_RETRY_NONCE_V1) - 1;
  1825. break;
  1826. case NGTCP2_PROTO_VER_V2:
  1827. nonce = (const uint8_t *)NGTCP2_RETRY_NONCE_V2;
  1828. noncelen = sizeof(NGTCP2_RETRY_NONCE_V2) - 1;
  1829. break;
  1830. }
  1831. /* OpenSSL does not like NULL plaintext. */
  1832. rv = encrypt(tag, aead, aead_ctx, (const uint8_t *)"", 0, nonce, noncelen,
  1833. pseudo_retry, pseudo_retrylen);
  1834. if (rv != 0) {
  1835. return rv;
  1836. }
  1837. if (0 != memcmp(retry->tag, tag, sizeof(retry->tag))) {
  1838. return NGTCP2_ERR_PROTO;
  1839. }
  1840. return 0;
  1841. }
  1842. size_t ngtcp2_pkt_stream_max_datalen(int64_t stream_id, uint64_t offset,
  1843. uint64_t len, size_t left) {
  1844. size_t n = 1 /* type */ + ngtcp2_put_uvarintlen((uint64_t)stream_id) +
  1845. (offset ? ngtcp2_put_uvarintlen(offset) : 0);
  1846. if (left <= n) {
  1847. return (size_t)-1;
  1848. }
  1849. left -= n;
  1850. if (left > 8 + 1073741823 && len > 1073741823) {
  1851. #if SIZE_MAX > UINT32_MAX
  1852. len = ngtcp2_min_uint64(len, 4611686018427387903lu);
  1853. #endif /* SIZE_MAX > UINT32_MAX */
  1854. return (size_t)ngtcp2_min_uint64(len, (uint64_t)(left - 8));
  1855. }
  1856. if (left > 4 + 16383 && len > 16383) {
  1857. len = ngtcp2_min_uint64(len, 1073741823);
  1858. return (size_t)ngtcp2_min_uint64(len, (uint64_t)(left - 4));
  1859. }
  1860. if (left > 2 + 63 && len > 63) {
  1861. len = ngtcp2_min_uint64(len, 16383);
  1862. return (size_t)ngtcp2_min_uint64(len, (uint64_t)(left - 2));
  1863. }
  1864. len = ngtcp2_min_uint64(len, 63);
  1865. return (size_t)ngtcp2_min_uint64(len, (uint64_t)(left - 1));
  1866. }
  1867. size_t ngtcp2_pkt_crypto_max_datalen(uint64_t offset, size_t len, size_t left) {
  1868. size_t n = 1 /* type */ + ngtcp2_put_uvarintlen(offset);
  1869. /* CRYPTO frame must contain nonzero length data. Return -1 if
  1870. there is no space to write crypto data. */
  1871. if (left <= n + 1) {
  1872. return (size_t)-1;
  1873. }
  1874. left -= n;
  1875. if (left > 8 + 1073741823 && len > 1073741823) {
  1876. #if SIZE_MAX > UINT32_MAX
  1877. len = ngtcp2_min_size(len, 4611686018427387903lu);
  1878. #endif /* SIZE_MAX > UINT32_MAX */
  1879. return ngtcp2_min_size(len, left - 8);
  1880. }
  1881. if (left > 4 + 16383 && len > 16383) {
  1882. len = ngtcp2_min_size(len, 1073741823);
  1883. return ngtcp2_min_size(len, left - 4);
  1884. }
  1885. if (left > 2 + 63 && len > 63) {
  1886. len = ngtcp2_min_size(len, 16383);
  1887. return ngtcp2_min_size(len, left - 2);
  1888. }
  1889. len = ngtcp2_min_size(len, 63);
  1890. return ngtcp2_min_size(len, left - 1);
  1891. }
  1892. size_t ngtcp2_pkt_datagram_framelen(size_t len) {
  1893. return 1 /* type */ + ngtcp2_put_uvarintlen(len) + len;
  1894. }
  1895. int ngtcp2_is_supported_version(uint32_t version) {
  1896. switch (version) {
  1897. case NGTCP2_PROTO_VER_V1:
  1898. case NGTCP2_PROTO_VER_V2:
  1899. return 1;
  1900. default:
  1901. return 0;
  1902. }
  1903. }
  1904. int ngtcp2_is_reserved_version(uint32_t version) {
  1905. return (version & NGTCP2_RESERVED_VERSION_MASK) ==
  1906. NGTCP2_RESERVED_VERSION_MASK;
  1907. }
  1908. uint8_t ngtcp2_pkt_get_type_long(uint32_t version, uint8_t c) {
  1909. uint8_t pkt_type = (uint8_t)((c & NGTCP2_LONG_TYPE_MASK) >> 4);
  1910. switch (version) {
  1911. case NGTCP2_PROTO_VER_V2:
  1912. switch (pkt_type) {
  1913. case NGTCP2_PKT_TYPE_INITIAL_V2:
  1914. return NGTCP2_PKT_INITIAL;
  1915. case NGTCP2_PKT_TYPE_0RTT_V2:
  1916. return NGTCP2_PKT_0RTT;
  1917. case NGTCP2_PKT_TYPE_HANDSHAKE_V2:
  1918. return NGTCP2_PKT_HANDSHAKE;
  1919. case NGTCP2_PKT_TYPE_RETRY_V2:
  1920. return NGTCP2_PKT_RETRY;
  1921. default:
  1922. return 0;
  1923. }
  1924. default:
  1925. if (!ngtcp2_is_supported_version(version)) {
  1926. return 0;
  1927. }
  1928. switch (pkt_type) {
  1929. case NGTCP2_PKT_TYPE_INITIAL_V1:
  1930. return NGTCP2_PKT_INITIAL;
  1931. case NGTCP2_PKT_TYPE_0RTT_V1:
  1932. return NGTCP2_PKT_0RTT;
  1933. case NGTCP2_PKT_TYPE_HANDSHAKE_V1:
  1934. return NGTCP2_PKT_HANDSHAKE;
  1935. case NGTCP2_PKT_TYPE_RETRY_V1:
  1936. return NGTCP2_PKT_RETRY;
  1937. default:
  1938. return 0;
  1939. }
  1940. }
  1941. }
  1942. uint8_t ngtcp2_pkt_versioned_type(uint32_t version, uint32_t pkt_type) {
  1943. switch (version) {
  1944. case NGTCP2_PROTO_VER_V2:
  1945. switch (pkt_type) {
  1946. case NGTCP2_PKT_INITIAL:
  1947. return NGTCP2_PKT_TYPE_INITIAL_V2;
  1948. case NGTCP2_PKT_0RTT:
  1949. return NGTCP2_PKT_TYPE_0RTT_V2;
  1950. case NGTCP2_PKT_HANDSHAKE:
  1951. return NGTCP2_PKT_TYPE_HANDSHAKE_V2;
  1952. case NGTCP2_PKT_RETRY:
  1953. return NGTCP2_PKT_TYPE_RETRY_V2;
  1954. default:
  1955. ngtcp2_unreachable();
  1956. }
  1957. default:
  1958. /* Assume that unsupported versions share the numeric long packet
  1959. types with QUIC v1 in order to send a packet to elicit Version
  1960. Negotiation packet. */
  1961. switch (pkt_type) {
  1962. case NGTCP2_PKT_INITIAL:
  1963. return NGTCP2_PKT_TYPE_INITIAL_V1;
  1964. case NGTCP2_PKT_0RTT:
  1965. return NGTCP2_PKT_TYPE_0RTT_V1;
  1966. case NGTCP2_PKT_HANDSHAKE:
  1967. return NGTCP2_PKT_TYPE_HANDSHAKE_V1;
  1968. case NGTCP2_PKT_RETRY:
  1969. return NGTCP2_PKT_TYPE_RETRY_V1;
  1970. default:
  1971. ngtcp2_unreachable();
  1972. }
  1973. }
  1974. }
  1975. int ngtcp2_pkt_verify_reserved_bits(uint8_t c) {
  1976. if (c & NGTCP2_HEADER_FORM_BIT) {
  1977. return (c & NGTCP2_LONG_RESERVED_BIT_MASK) == 0 ? 0 : NGTCP2_ERR_PROTO;
  1978. }
  1979. return (c & NGTCP2_SHORT_RESERVED_BIT_MASK) == 0 ? 0 : NGTCP2_ERR_PROTO;
  1980. }