nghttp2_submit.c 24 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864
  1. /*
  2. * nghttp2 - HTTP/2 C Library
  3. *
  4. * Copyright (c) 2012, 2013 Tatsuhiro Tsujikawa
  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 "nghttp2_submit.h"
  26. #include <string.h>
  27. #include <assert.h>
  28. #include "nghttp2_session.h"
  29. #include "nghttp2_frame.h"
  30. #include "nghttp2_helper.h"
  31. #include "nghttp2_priority_spec.h"
  32. /* This function takes ownership of |nva_copy|. Regardless of the
  33. return value, the caller must not free |nva_copy| after this
  34. function returns. */
  35. static int32_t submit_headers_shared(nghttp2_session *session, uint8_t flags,
  36. int32_t stream_id, nghttp2_nv *nva_copy,
  37. size_t nvlen,
  38. const nghttp2_data_provider_wrap *dpw,
  39. void *stream_user_data) {
  40. int rv;
  41. uint8_t flags_copy;
  42. nghttp2_outbound_item *item = NULL;
  43. nghttp2_frame *frame = NULL;
  44. nghttp2_headers_category hcat;
  45. nghttp2_mem *mem;
  46. mem = &session->mem;
  47. item = nghttp2_mem_malloc(mem, sizeof(nghttp2_outbound_item));
  48. if (item == NULL) {
  49. rv = NGHTTP2_ERR_NOMEM;
  50. goto fail;
  51. }
  52. nghttp2_outbound_item_init(item);
  53. if (dpw != NULL && dpw->data_prd.read_callback != NULL) {
  54. item->aux_data.headers.dpw = *dpw;
  55. }
  56. item->aux_data.headers.stream_user_data = stream_user_data;
  57. flags_copy =
  58. (uint8_t)((flags & (NGHTTP2_FLAG_END_STREAM | NGHTTP2_FLAG_PRIORITY)) |
  59. NGHTTP2_FLAG_END_HEADERS);
  60. if (stream_id == -1) {
  61. if (session->next_stream_id > INT32_MAX) {
  62. rv = NGHTTP2_ERR_STREAM_ID_NOT_AVAILABLE;
  63. goto fail;
  64. }
  65. stream_id = (int32_t)session->next_stream_id;
  66. session->next_stream_id += 2;
  67. hcat = NGHTTP2_HCAT_REQUEST;
  68. } else {
  69. /* More specific categorization will be done later. */
  70. hcat = NGHTTP2_HCAT_HEADERS;
  71. }
  72. frame = &item->frame;
  73. nghttp2_frame_headers_init(&frame->headers, flags_copy, stream_id, hcat, NULL,
  74. nva_copy, nvlen);
  75. rv = nghttp2_session_add_item(session, item);
  76. if (rv != 0) {
  77. nghttp2_frame_headers_free(&frame->headers, mem);
  78. goto fail2;
  79. }
  80. if (hcat == NGHTTP2_HCAT_REQUEST) {
  81. return stream_id;
  82. }
  83. return 0;
  84. fail:
  85. /* nghttp2_frame_headers_init() takes ownership of nva_copy. */
  86. nghttp2_nv_array_del(nva_copy, mem);
  87. fail2:
  88. nghttp2_mem_free(mem, item);
  89. return rv;
  90. }
  91. static int32_t submit_headers_shared_nva(nghttp2_session *session,
  92. uint8_t flags, int32_t stream_id,
  93. const nghttp2_nv *nva, size_t nvlen,
  94. const nghttp2_data_provider_wrap *dpw,
  95. void *stream_user_data) {
  96. int rv;
  97. nghttp2_nv *nva_copy;
  98. nghttp2_mem *mem;
  99. mem = &session->mem;
  100. rv = nghttp2_nv_array_copy(&nva_copy, nva, nvlen, mem);
  101. if (rv < 0) {
  102. return rv;
  103. }
  104. return submit_headers_shared(session, flags, stream_id, nva_copy, nvlen, dpw,
  105. stream_user_data);
  106. }
  107. int nghttp2_submit_trailer(nghttp2_session *session, int32_t stream_id,
  108. const nghttp2_nv *nva, size_t nvlen) {
  109. if (stream_id <= 0) {
  110. return NGHTTP2_ERR_INVALID_ARGUMENT;
  111. }
  112. return (int)submit_headers_shared_nva(session, NGHTTP2_FLAG_END_STREAM,
  113. stream_id, nva, nvlen, NULL, NULL);
  114. }
  115. int32_t nghttp2_submit_headers(nghttp2_session *session, uint8_t flags,
  116. int32_t stream_id,
  117. const nghttp2_priority_spec *pri_spec,
  118. const nghttp2_nv *nva, size_t nvlen,
  119. void *stream_user_data) {
  120. (void)pri_spec;
  121. if (stream_id == -1) {
  122. if (session->server) {
  123. return NGHTTP2_ERR_PROTO;
  124. }
  125. } else if (stream_id <= 0) {
  126. return NGHTTP2_ERR_INVALID_ARGUMENT;
  127. }
  128. flags &= NGHTTP2_FLAG_END_STREAM;
  129. return submit_headers_shared_nva(session, flags, stream_id, nva, nvlen, NULL,
  130. stream_user_data);
  131. }
  132. int nghttp2_submit_ping(nghttp2_session *session, uint8_t flags,
  133. const uint8_t *opaque_data) {
  134. flags &= NGHTTP2_FLAG_ACK;
  135. return nghttp2_session_add_ping(session, flags, opaque_data);
  136. }
  137. int nghttp2_submit_priority(nghttp2_session *session, uint8_t flags,
  138. int32_t stream_id,
  139. const nghttp2_priority_spec *pri_spec) {
  140. (void)session;
  141. (void)flags;
  142. (void)stream_id;
  143. (void)pri_spec;
  144. return 0;
  145. }
  146. int nghttp2_submit_rst_stream(nghttp2_session *session, uint8_t flags,
  147. int32_t stream_id, uint32_t error_code) {
  148. (void)flags;
  149. if (stream_id == 0) {
  150. return NGHTTP2_ERR_INVALID_ARGUMENT;
  151. }
  152. return nghttp2_session_add_rst_stream_continue(
  153. session, stream_id, error_code, /* continue_without_stream = */ 0);
  154. }
  155. int nghttp2_submit_goaway(nghttp2_session *session, uint8_t flags,
  156. int32_t last_stream_id, uint32_t error_code,
  157. const uint8_t *opaque_data, size_t opaque_data_len) {
  158. (void)flags;
  159. if (session->goaway_flags & NGHTTP2_GOAWAY_TERM_ON_SEND) {
  160. return 0;
  161. }
  162. return nghttp2_session_add_goaway(session, last_stream_id, error_code,
  163. opaque_data, opaque_data_len,
  164. NGHTTP2_GOAWAY_AUX_NONE);
  165. }
  166. int nghttp2_submit_shutdown_notice(nghttp2_session *session) {
  167. if (!session->server) {
  168. return NGHTTP2_ERR_INVALID_STATE;
  169. }
  170. if (session->goaway_flags) {
  171. return 0;
  172. }
  173. return nghttp2_session_add_goaway(session, (1u << 31) - 1, NGHTTP2_NO_ERROR,
  174. NULL, 0,
  175. NGHTTP2_GOAWAY_AUX_SHUTDOWN_NOTICE);
  176. }
  177. int nghttp2_submit_settings(nghttp2_session *session, uint8_t flags,
  178. const nghttp2_settings_entry *iv, size_t niv) {
  179. (void)flags;
  180. return nghttp2_session_add_settings(session, NGHTTP2_FLAG_NONE, iv, niv);
  181. }
  182. int32_t nghttp2_submit_push_promise(nghttp2_session *session, uint8_t flags,
  183. int32_t stream_id, const nghttp2_nv *nva,
  184. size_t nvlen,
  185. void *promised_stream_user_data) {
  186. nghttp2_outbound_item *item;
  187. nghttp2_frame *frame;
  188. nghttp2_nv *nva_copy;
  189. uint8_t flags_copy;
  190. int32_t promised_stream_id;
  191. int rv;
  192. nghttp2_mem *mem;
  193. (void)flags;
  194. mem = &session->mem;
  195. if (stream_id <= 0 || nghttp2_session_is_my_stream_id(session, stream_id)) {
  196. return NGHTTP2_ERR_INVALID_ARGUMENT;
  197. }
  198. if (!session->server) {
  199. return NGHTTP2_ERR_PROTO;
  200. }
  201. /* All 32bit signed stream IDs are spent. */
  202. if (session->next_stream_id > INT32_MAX) {
  203. return NGHTTP2_ERR_STREAM_ID_NOT_AVAILABLE;
  204. }
  205. item = nghttp2_mem_malloc(mem, sizeof(nghttp2_outbound_item));
  206. if (item == NULL) {
  207. return NGHTTP2_ERR_NOMEM;
  208. }
  209. nghttp2_outbound_item_init(item);
  210. item->aux_data.headers.stream_user_data = promised_stream_user_data;
  211. frame = &item->frame;
  212. rv = nghttp2_nv_array_copy(&nva_copy, nva, nvlen, mem);
  213. if (rv < 0) {
  214. nghttp2_mem_free(mem, item);
  215. return rv;
  216. }
  217. flags_copy = NGHTTP2_FLAG_END_HEADERS;
  218. promised_stream_id = (int32_t)session->next_stream_id;
  219. session->next_stream_id += 2;
  220. nghttp2_frame_push_promise_init(&frame->push_promise, flags_copy, stream_id,
  221. promised_stream_id, nva_copy, nvlen);
  222. rv = nghttp2_session_add_item(session, item);
  223. if (rv != 0) {
  224. nghttp2_frame_push_promise_free(&frame->push_promise, mem);
  225. nghttp2_mem_free(mem, item);
  226. return rv;
  227. }
  228. return promised_stream_id;
  229. }
  230. int nghttp2_submit_window_update(nghttp2_session *session, uint8_t flags,
  231. int32_t stream_id,
  232. int32_t window_size_increment) {
  233. int rv;
  234. nghttp2_stream *stream = 0;
  235. (void)flags;
  236. if (window_size_increment == 0) {
  237. return 0;
  238. }
  239. if (stream_id == 0) {
  240. rv = nghttp2_adjust_local_window_size(
  241. &session->local_window_size, &session->recv_window_size,
  242. &session->recv_reduction, &window_size_increment);
  243. if (rv != 0) {
  244. return rv;
  245. }
  246. } else {
  247. stream = nghttp2_session_get_stream(session, stream_id);
  248. if (!stream) {
  249. return 0;
  250. }
  251. rv = nghttp2_adjust_local_window_size(
  252. &stream->local_window_size, &stream->recv_window_size,
  253. &stream->recv_reduction, &window_size_increment);
  254. if (rv != 0) {
  255. return rv;
  256. }
  257. }
  258. if (window_size_increment > 0) {
  259. if (stream_id == 0) {
  260. session->consumed_size =
  261. nghttp2_max_int32(0, session->consumed_size - window_size_increment);
  262. } else {
  263. stream->consumed_size =
  264. nghttp2_max_int32(0, stream->consumed_size - window_size_increment);
  265. }
  266. return nghttp2_session_add_window_update(session, 0, stream_id,
  267. window_size_increment);
  268. }
  269. return 0;
  270. }
  271. int nghttp2_session_set_local_window_size(nghttp2_session *session,
  272. uint8_t flags, int32_t stream_id,
  273. int32_t window_size) {
  274. int32_t window_size_increment;
  275. nghttp2_stream *stream;
  276. int rv;
  277. (void)flags;
  278. if (window_size < 0) {
  279. return NGHTTP2_ERR_INVALID_ARGUMENT;
  280. }
  281. if (stream_id == 0) {
  282. window_size_increment = window_size - session->local_window_size;
  283. if (window_size_increment == 0) {
  284. return 0;
  285. }
  286. if (window_size_increment < 0) {
  287. return nghttp2_adjust_local_window_size(
  288. &session->local_window_size, &session->recv_window_size,
  289. &session->recv_reduction, &window_size_increment);
  290. }
  291. rv = nghttp2_increase_local_window_size(
  292. &session->local_window_size, &session->recv_window_size,
  293. &session->recv_reduction, &window_size_increment);
  294. if (rv != 0) {
  295. return rv;
  296. }
  297. if (window_size_increment > 0) {
  298. return nghttp2_session_add_window_update(session, 0, stream_id,
  299. window_size_increment);
  300. }
  301. return nghttp2_session_update_recv_connection_window_size(session, 0);
  302. } else {
  303. stream = nghttp2_session_get_stream(session, stream_id);
  304. if (stream == NULL) {
  305. return 0;
  306. }
  307. window_size_increment = window_size - stream->local_window_size;
  308. if (window_size_increment == 0) {
  309. return 0;
  310. }
  311. if (window_size_increment < 0) {
  312. return nghttp2_adjust_local_window_size(
  313. &stream->local_window_size, &stream->recv_window_size,
  314. &stream->recv_reduction, &window_size_increment);
  315. }
  316. rv = nghttp2_increase_local_window_size(
  317. &stream->local_window_size, &stream->recv_window_size,
  318. &stream->recv_reduction, &window_size_increment);
  319. if (rv != 0) {
  320. return rv;
  321. }
  322. if (window_size_increment > 0) {
  323. return nghttp2_session_add_window_update(session, 0, stream_id,
  324. window_size_increment);
  325. }
  326. return nghttp2_session_update_recv_stream_window_size(session, stream, 0,
  327. 1);
  328. }
  329. }
  330. int nghttp2_submit_altsvc(nghttp2_session *session, uint8_t flags,
  331. int32_t stream_id, const uint8_t *origin,
  332. size_t origin_len, const uint8_t *field_value,
  333. size_t field_value_len) {
  334. nghttp2_mem *mem;
  335. uint8_t *buf, *p;
  336. uint8_t *origin_copy;
  337. uint8_t *field_value_copy;
  338. nghttp2_outbound_item *item;
  339. nghttp2_frame *frame;
  340. nghttp2_ext_altsvc *altsvc;
  341. int rv;
  342. (void)flags;
  343. mem = &session->mem;
  344. if (!session->server) {
  345. return NGHTTP2_ERR_INVALID_STATE;
  346. }
  347. if (2 + origin_len + field_value_len > NGHTTP2_MAX_PAYLOADLEN) {
  348. return NGHTTP2_ERR_INVALID_ARGUMENT;
  349. }
  350. if (stream_id == 0) {
  351. if (origin_len == 0) {
  352. return NGHTTP2_ERR_INVALID_ARGUMENT;
  353. }
  354. } else if (origin_len != 0) {
  355. return NGHTTP2_ERR_INVALID_ARGUMENT;
  356. }
  357. buf = nghttp2_mem_malloc(mem, origin_len + field_value_len + 2);
  358. if (buf == NULL) {
  359. return NGHTTP2_ERR_NOMEM;
  360. }
  361. p = buf;
  362. origin_copy = p;
  363. if (origin_len) {
  364. p = nghttp2_cpymem(p, origin, origin_len);
  365. }
  366. *p++ = '\0';
  367. field_value_copy = p;
  368. if (field_value_len) {
  369. p = nghttp2_cpymem(p, field_value, field_value_len);
  370. }
  371. *p++ = '\0';
  372. item = nghttp2_mem_malloc(mem, sizeof(nghttp2_outbound_item));
  373. if (item == NULL) {
  374. rv = NGHTTP2_ERR_NOMEM;
  375. goto fail_item_malloc;
  376. }
  377. nghttp2_outbound_item_init(item);
  378. item->aux_data.ext.builtin = 1;
  379. altsvc = &item->ext_frame_payload.altsvc;
  380. frame = &item->frame;
  381. frame->ext.payload = altsvc;
  382. nghttp2_frame_altsvc_init(&frame->ext, stream_id, origin_copy, origin_len,
  383. field_value_copy, field_value_len);
  384. rv = nghttp2_session_add_item(session, item);
  385. if (rv != 0) {
  386. nghttp2_frame_altsvc_free(&frame->ext, mem);
  387. nghttp2_mem_free(mem, item);
  388. return rv;
  389. }
  390. return 0;
  391. fail_item_malloc:
  392. nghttp2_mem_free(mem, buf);
  393. return rv;
  394. }
  395. int nghttp2_submit_origin(nghttp2_session *session, uint8_t flags,
  396. const nghttp2_origin_entry *ov, size_t nov) {
  397. nghttp2_mem *mem;
  398. uint8_t *p;
  399. nghttp2_outbound_item *item;
  400. nghttp2_frame *frame;
  401. nghttp2_ext_origin *origin;
  402. nghttp2_origin_entry *ov_copy;
  403. size_t len = 0;
  404. size_t i;
  405. int rv;
  406. (void)flags;
  407. mem = &session->mem;
  408. if (!session->server) {
  409. return NGHTTP2_ERR_INVALID_STATE;
  410. }
  411. if (nov) {
  412. for (i = 0; i < nov; ++i) {
  413. len += ov[i].origin_len;
  414. }
  415. if (2 * nov + len > NGHTTP2_MAX_PAYLOADLEN) {
  416. return NGHTTP2_ERR_INVALID_ARGUMENT;
  417. }
  418. /* The last nov is added for terminal NULL character. */
  419. ov_copy =
  420. nghttp2_mem_malloc(mem, nov * sizeof(nghttp2_origin_entry) + len + nov);
  421. if (ov_copy == NULL) {
  422. return NGHTTP2_ERR_NOMEM;
  423. }
  424. p = (uint8_t *)ov_copy + nov * sizeof(nghttp2_origin_entry);
  425. for (i = 0; i < nov; ++i) {
  426. ov_copy[i].origin = p;
  427. ov_copy[i].origin_len = ov[i].origin_len;
  428. p = nghttp2_cpymem(p, ov[i].origin, ov[i].origin_len);
  429. *p++ = '\0';
  430. }
  431. assert((size_t)(p - (uint8_t *)ov_copy) ==
  432. nov * sizeof(nghttp2_origin_entry) + len + nov);
  433. } else {
  434. ov_copy = NULL;
  435. }
  436. item = nghttp2_mem_malloc(mem, sizeof(nghttp2_outbound_item));
  437. if (item == NULL) {
  438. rv = NGHTTP2_ERR_NOMEM;
  439. goto fail_item_malloc;
  440. }
  441. nghttp2_outbound_item_init(item);
  442. item->aux_data.ext.builtin = 1;
  443. origin = &item->ext_frame_payload.origin;
  444. frame = &item->frame;
  445. frame->ext.payload = origin;
  446. nghttp2_frame_origin_init(&frame->ext, ov_copy, nov);
  447. rv = nghttp2_session_add_item(session, item);
  448. if (rv != 0) {
  449. nghttp2_frame_origin_free(&frame->ext, mem);
  450. nghttp2_mem_free(mem, item);
  451. return rv;
  452. }
  453. return 0;
  454. fail_item_malloc:
  455. nghttp2_mem_free(mem, ov_copy);
  456. return rv;
  457. }
  458. int nghttp2_submit_priority_update(nghttp2_session *session, uint8_t flags,
  459. int32_t stream_id,
  460. const uint8_t *field_value,
  461. size_t field_value_len) {
  462. nghttp2_mem *mem;
  463. uint8_t *buf, *p;
  464. nghttp2_outbound_item *item;
  465. nghttp2_frame *frame;
  466. nghttp2_ext_priority_update *priority_update;
  467. int rv;
  468. (void)flags;
  469. mem = &session->mem;
  470. if (session->server) {
  471. return NGHTTP2_ERR_INVALID_STATE;
  472. }
  473. if (session->remote_settings.no_rfc7540_priorities == 0) {
  474. return 0;
  475. }
  476. if (stream_id == 0 || 4 + field_value_len > NGHTTP2_MAX_PAYLOADLEN) {
  477. return NGHTTP2_ERR_INVALID_ARGUMENT;
  478. }
  479. if (field_value_len) {
  480. buf = nghttp2_mem_malloc(mem, field_value_len + 1);
  481. if (buf == NULL) {
  482. return NGHTTP2_ERR_NOMEM;
  483. }
  484. p = nghttp2_cpymem(buf, field_value, field_value_len);
  485. *p = '\0';
  486. } else {
  487. buf = NULL;
  488. }
  489. item = nghttp2_mem_malloc(mem, sizeof(nghttp2_outbound_item));
  490. if (item == NULL) {
  491. rv = NGHTTP2_ERR_NOMEM;
  492. goto fail_item_malloc;
  493. }
  494. nghttp2_outbound_item_init(item);
  495. item->aux_data.ext.builtin = 1;
  496. priority_update = &item->ext_frame_payload.priority_update;
  497. frame = &item->frame;
  498. frame->ext.payload = priority_update;
  499. nghttp2_frame_priority_update_init(&frame->ext, stream_id, buf,
  500. field_value_len);
  501. rv = nghttp2_session_add_item(session, item);
  502. if (rv != 0) {
  503. nghttp2_frame_priority_update_free(&frame->ext, mem);
  504. nghttp2_mem_free(mem, item);
  505. return rv;
  506. }
  507. return 0;
  508. fail_item_malloc:
  509. nghttp2_mem_free(mem, buf);
  510. return rv;
  511. }
  512. static uint8_t set_request_flags(const nghttp2_data_provider_wrap *dpw) {
  513. uint8_t flags = NGHTTP2_FLAG_NONE;
  514. if (dpw == NULL || dpw->data_prd.read_callback == NULL) {
  515. flags |= NGHTTP2_FLAG_END_STREAM;
  516. }
  517. return flags;
  518. }
  519. static int32_t submit_request_shared(nghttp2_session *session,
  520. const nghttp2_nv *nva, size_t nvlen,
  521. const nghttp2_data_provider_wrap *dpw,
  522. void *stream_user_data) {
  523. uint8_t flags;
  524. if (session->server) {
  525. return NGHTTP2_ERR_PROTO;
  526. }
  527. flags = set_request_flags(dpw);
  528. return submit_headers_shared_nva(session, flags, -1, nva, nvlen, dpw,
  529. stream_user_data);
  530. }
  531. int32_t nghttp2_submit_request(nghttp2_session *session,
  532. const nghttp2_priority_spec *pri_spec,
  533. const nghttp2_nv *nva, size_t nvlen,
  534. const nghttp2_data_provider *data_prd,
  535. void *stream_user_data) {
  536. nghttp2_data_provider_wrap dpw;
  537. (void)pri_spec;
  538. return submit_request_shared(session, nva, nvlen,
  539. nghttp2_data_provider_wrap_v1(&dpw, data_prd),
  540. stream_user_data);
  541. }
  542. int32_t nghttp2_submit_request2(nghttp2_session *session,
  543. const nghttp2_priority_spec *pri_spec,
  544. const nghttp2_nv *nva, size_t nvlen,
  545. const nghttp2_data_provider2 *data_prd,
  546. void *stream_user_data) {
  547. nghttp2_data_provider_wrap dpw;
  548. (void)pri_spec;
  549. return submit_request_shared(session, nva, nvlen,
  550. nghttp2_data_provider_wrap_v2(&dpw, data_prd),
  551. stream_user_data);
  552. }
  553. static uint8_t set_response_flags(const nghttp2_data_provider_wrap *dpw) {
  554. uint8_t flags = NGHTTP2_FLAG_NONE;
  555. if (dpw == NULL || dpw->data_prd.read_callback == NULL) {
  556. flags |= NGHTTP2_FLAG_END_STREAM;
  557. }
  558. return flags;
  559. }
  560. static int submit_response_shared(nghttp2_session *session, int32_t stream_id,
  561. const nghttp2_nv *nva, size_t nvlen,
  562. const nghttp2_data_provider_wrap *dpw) {
  563. uint8_t flags;
  564. if (stream_id <= 0) {
  565. return NGHTTP2_ERR_INVALID_ARGUMENT;
  566. }
  567. if (!session->server) {
  568. return NGHTTP2_ERR_PROTO;
  569. }
  570. flags = set_response_flags(dpw);
  571. return submit_headers_shared_nva(session, flags, stream_id, nva, nvlen, dpw,
  572. NULL);
  573. }
  574. int nghttp2_submit_response(nghttp2_session *session, int32_t stream_id,
  575. const nghttp2_nv *nva, size_t nvlen,
  576. const nghttp2_data_provider *data_prd) {
  577. nghttp2_data_provider_wrap dpw;
  578. return submit_response_shared(session, stream_id, nva, nvlen,
  579. nghttp2_data_provider_wrap_v1(&dpw, data_prd));
  580. }
  581. int nghttp2_submit_response2(nghttp2_session *session, int32_t stream_id,
  582. const nghttp2_nv *nva, size_t nvlen,
  583. const nghttp2_data_provider2 *data_prd) {
  584. nghttp2_data_provider_wrap dpw;
  585. return submit_response_shared(session, stream_id, nva, nvlen,
  586. nghttp2_data_provider_wrap_v2(&dpw, data_prd));
  587. }
  588. int nghttp2_submit_data_shared(nghttp2_session *session, uint8_t flags,
  589. int32_t stream_id,
  590. const nghttp2_data_provider_wrap *dpw) {
  591. int rv;
  592. nghttp2_outbound_item *item;
  593. nghttp2_frame *frame;
  594. nghttp2_data_aux_data *aux_data;
  595. uint8_t nflags = flags & NGHTTP2_FLAG_END_STREAM;
  596. nghttp2_mem *mem;
  597. mem = &session->mem;
  598. if (stream_id == 0) {
  599. return NGHTTP2_ERR_INVALID_ARGUMENT;
  600. }
  601. item = nghttp2_mem_malloc(mem, sizeof(nghttp2_outbound_item));
  602. if (item == NULL) {
  603. return NGHTTP2_ERR_NOMEM;
  604. }
  605. nghttp2_outbound_item_init(item);
  606. frame = &item->frame;
  607. aux_data = &item->aux_data.data;
  608. aux_data->dpw = *dpw;
  609. aux_data->eof = 0;
  610. aux_data->flags = nflags;
  611. /* flags are sent on transmission */
  612. nghttp2_frame_data_init(&frame->data, NGHTTP2_FLAG_NONE, stream_id);
  613. rv = nghttp2_session_add_item(session, item);
  614. if (rv != 0) {
  615. nghttp2_frame_data_free(&frame->data);
  616. nghttp2_mem_free(mem, item);
  617. return rv;
  618. }
  619. return 0;
  620. }
  621. int nghttp2_submit_data(nghttp2_session *session, uint8_t flags,
  622. int32_t stream_id,
  623. const nghttp2_data_provider *data_prd) {
  624. nghttp2_data_provider_wrap dpw;
  625. assert(data_prd);
  626. return nghttp2_submit_data_shared(
  627. session, flags, stream_id, nghttp2_data_provider_wrap_v1(&dpw, data_prd));
  628. }
  629. int nghttp2_submit_data2(nghttp2_session *session, uint8_t flags,
  630. int32_t stream_id,
  631. const nghttp2_data_provider2 *data_prd) {
  632. nghttp2_data_provider_wrap dpw;
  633. assert(data_prd);
  634. return nghttp2_submit_data_shared(
  635. session, flags, stream_id, nghttp2_data_provider_wrap_v2(&dpw, data_prd));
  636. }
  637. ssize_t nghttp2_pack_settings_payload(uint8_t *buf, size_t buflen,
  638. const nghttp2_settings_entry *iv,
  639. size_t niv) {
  640. return (ssize_t)nghttp2_pack_settings_payload2(buf, buflen, iv, niv);
  641. }
  642. nghttp2_ssize nghttp2_pack_settings_payload2(uint8_t *buf, size_t buflen,
  643. const nghttp2_settings_entry *iv,
  644. size_t niv) {
  645. if (!nghttp2_iv_check(iv, niv)) {
  646. return NGHTTP2_ERR_INVALID_ARGUMENT;
  647. }
  648. if (buflen < (niv * NGHTTP2_FRAME_SETTINGS_ENTRY_LENGTH)) {
  649. return NGHTTP2_ERR_INSUFF_BUFSIZE;
  650. }
  651. return (nghttp2_ssize)nghttp2_frame_pack_settings_payload(buf, iv, niv);
  652. }
  653. int nghttp2_submit_extension(nghttp2_session *session, uint8_t type,
  654. uint8_t flags, int32_t stream_id, void *payload) {
  655. int rv;
  656. nghttp2_outbound_item *item;
  657. nghttp2_frame *frame;
  658. nghttp2_mem *mem;
  659. mem = &session->mem;
  660. if (type <= NGHTTP2_CONTINUATION) {
  661. return NGHTTP2_ERR_INVALID_ARGUMENT;
  662. }
  663. if (!session->callbacks.pack_extension_callback2 &&
  664. !session->callbacks.pack_extension_callback) {
  665. return NGHTTP2_ERR_INVALID_STATE;
  666. }
  667. item = nghttp2_mem_malloc(mem, sizeof(nghttp2_outbound_item));
  668. if (item == NULL) {
  669. return NGHTTP2_ERR_NOMEM;
  670. }
  671. nghttp2_outbound_item_init(item);
  672. frame = &item->frame;
  673. nghttp2_frame_extension_init(&frame->ext, type, flags, stream_id, payload);
  674. rv = nghttp2_session_add_item(session, item);
  675. if (rv != 0) {
  676. nghttp2_frame_extension_free(&frame->ext);
  677. nghttp2_mem_free(mem, item);
  678. return rv;
  679. }
  680. return 0;
  681. }