wslay_frame_test.c 18 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560
  1. /*
  2. * Wslay - The WebSocket Library
  3. *
  4. * Copyright (c) 2011, 2012 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 "wslay_frame_test.h"
  26. #include <assert.h>
  27. #include <CUnit/CUnit.h>
  28. #include "wslay_frame.h"
  29. void test_wslay_frame_context_init(void)
  30. {
  31. wslay_frame_context_ptr ctx;
  32. struct wslay_frame_callbacks callbacks;
  33. int user_data;
  34. CU_ASSERT_FALSE(wslay_frame_context_init(&ctx, &callbacks, &user_data));
  35. wslay_frame_context_free(ctx);
  36. }
  37. struct scripted_data_feed {
  38. uint8_t data[8192];
  39. uint8_t* datamark;
  40. uint8_t* datalimit;
  41. size_t feedseq[8192];
  42. size_t seqidx;
  43. };
  44. static void scripted_data_feed_init(struct scripted_data_feed *df,
  45. uint8_t *data, size_t data_length)
  46. {
  47. memset(df, 0, sizeof(struct scripted_data_feed));
  48. memcpy(df->data, data, data_length);
  49. df->datamark = df->data;
  50. df->datalimit = df->data+data_length;
  51. df->feedseq[0] = data_length;
  52. }
  53. static ssize_t scripted_recv_callback(uint8_t* data, size_t len, int flags,
  54. void *user_data)
  55. {
  56. struct scripted_data_feed *df = (struct scripted_data_feed*)user_data;
  57. size_t wlen = df->feedseq[df->seqidx] > len ? len : df->feedseq[df->seqidx];
  58. memcpy(data, df->datamark, wlen);
  59. df->datamark += wlen;
  60. if(wlen <= len) {
  61. ++df->seqidx;
  62. } else {
  63. df->feedseq[df->seqidx] -= wlen;
  64. }
  65. return wlen;
  66. }
  67. static ssize_t scripted_send_callback(const uint8_t* data, size_t len,
  68. int flags, void *user_data)
  69. {
  70. struct scripted_data_feed *df = (struct scripted_data_feed*)user_data;
  71. size_t wlen = df->feedseq[df->seqidx] > len ? len : df->feedseq[df->seqidx];
  72. memcpy(df->datamark, data, wlen);
  73. df->datamark += wlen;
  74. if(wlen <= len) {
  75. ++df->seqidx;
  76. } else {
  77. df->feedseq[df->seqidx] -= wlen;
  78. }
  79. return wlen;
  80. }
  81. void test_wslay_frame_recv(void)
  82. {
  83. wslay_frame_context_ptr ctx;
  84. struct wslay_frame_callbacks callbacks = { NULL,
  85. scripted_recv_callback,
  86. NULL };
  87. struct scripted_data_feed df;
  88. struct wslay_frame_iocb iocb;
  89. /* Masked text frame containing "Hello" */
  90. uint8_t msg[] = { 0x81u, 0x85u, 0x37u, 0xfau, 0x21u, 0x3du, 0x7fu, 0x9fu,
  91. 0x4du, 0x51u, 0x58u };
  92. scripted_data_feed_init(&df, msg, sizeof(msg));
  93. wslay_frame_context_init(&ctx, &callbacks, &df);
  94. CU_ASSERT(5 == wslay_frame_recv(ctx, &iocb));
  95. CU_ASSERT_EQUAL(1, iocb.fin);
  96. CU_ASSERT_EQUAL(0, iocb.rsv);
  97. CU_ASSERT_EQUAL(0x1, iocb.opcode);
  98. CU_ASSERT_EQUAL(5, iocb.payload_length);
  99. CU_ASSERT_EQUAL(1, iocb.mask);
  100. CU_ASSERT_EQUAL(5, iocb.data_length);
  101. CU_ASSERT(memcmp("Hello", iocb.data, iocb.data_length) == 0);
  102. wslay_frame_context_free(ctx);
  103. }
  104. void test_wslay_frame_recv_1byte(void)
  105. {
  106. wslay_frame_context_ptr ctx;
  107. struct wslay_frame_callbacks callbacks = { NULL,
  108. scripted_recv_callback,
  109. NULL };
  110. struct scripted_data_feed df;
  111. struct wslay_frame_iocb iocb;
  112. size_t i;
  113. /* Masked text frame containing "Hello" */
  114. uint8_t msg[] = { 0x81u, 0x85u, 0x37u, 0xfau, 0x21u, 0x3du, 0x7fu, 0x9fu,
  115. 0x4du, 0x51u, 0x58u };
  116. scripted_data_feed_init(&df, msg, sizeof(msg));
  117. for(i = 0; i < sizeof(msg); ++i) {
  118. df.feedseq[i] = 1;
  119. }
  120. wslay_frame_context_init(&ctx, &callbacks, &df);
  121. for(i = 0; i < 4; ++i) {
  122. CU_ASSERT(WSLAY_ERR_WANT_READ == wslay_frame_recv(ctx, &iocb));
  123. }
  124. for(i = 0; i < 5; ++i) {
  125. CU_ASSERT(1 == wslay_frame_recv(ctx, &iocb));
  126. CU_ASSERT_EQUAL(1, iocb.fin);
  127. CU_ASSERT_EQUAL(0, iocb.rsv);
  128. CU_ASSERT_EQUAL(0x1, iocb.opcode);
  129. CU_ASSERT_EQUAL(5, iocb.payload_length);
  130. CU_ASSERT_EQUAL(1, iocb.mask);
  131. CU_ASSERT_EQUAL(1, iocb.data_length);
  132. CU_ASSERT_EQUAL(msg[6+i]^msg[2+i%4], iocb.data[0]);
  133. }
  134. CU_ASSERT(WSLAY_ERR_WANT_READ == wslay_frame_recv(ctx, &iocb));
  135. wslay_frame_context_free(ctx);
  136. }
  137. void test_wslay_frame_recv_fragmented(void)
  138. {
  139. wslay_frame_context_ptr ctx;
  140. struct wslay_frame_callbacks callbacks = { NULL,
  141. scripted_recv_callback,
  142. NULL };
  143. struct scripted_data_feed df;
  144. struct wslay_frame_iocb iocb;
  145. /* Unmasked message */
  146. uint8_t msg[] = { 0x01, 0x03, 0x48, 0x65, 0x6c, /* "Hel" */
  147. 0x80, 0x02, 0x6c, 0x6f }; /* "lo" */
  148. scripted_data_feed_init(&df, msg, sizeof(msg));
  149. df.feedseq[0] = 5;
  150. df.feedseq[1] = 4;
  151. wslay_frame_context_init(&ctx, &callbacks, &df);
  152. CU_ASSERT(3 == wslay_frame_recv(ctx, &iocb));
  153. CU_ASSERT_EQUAL(0, iocb.fin);
  154. CU_ASSERT_EQUAL(0, iocb.rsv);
  155. CU_ASSERT_EQUAL(WSLAY_TEXT_FRAME, iocb.opcode);
  156. CU_ASSERT_EQUAL(3, iocb.payload_length);
  157. CU_ASSERT_EQUAL(0, iocb.mask);
  158. CU_ASSERT_EQUAL(3, iocb.data_length);
  159. CU_ASSERT(memcmp("Hel", iocb.data, iocb.data_length) == 0);
  160. CU_ASSERT(2 == wslay_frame_recv(ctx, &iocb));
  161. CU_ASSERT_EQUAL(1, iocb.fin);
  162. CU_ASSERT_EQUAL(0, iocb.rsv);
  163. CU_ASSERT_EQUAL(WSLAY_CONTINUATION_FRAME, iocb.opcode);
  164. CU_ASSERT_EQUAL(2, iocb.payload_length);
  165. CU_ASSERT_EQUAL(0, iocb.mask);
  166. CU_ASSERT_EQUAL(2, iocb.data_length);
  167. CU_ASSERT(memcmp("lo", iocb.data, iocb.data_length) == 0);
  168. wslay_frame_context_free(ctx);
  169. }
  170. void test_wslay_frame_recv_interleaved_ctrl_frame(void)
  171. {
  172. wslay_frame_context_ptr ctx;
  173. struct wslay_frame_callbacks callbacks = { NULL,
  174. scripted_recv_callback,
  175. NULL };
  176. struct scripted_data_feed df;
  177. struct wslay_frame_iocb iocb;
  178. /* Unmasked message */
  179. uint8_t msg[] = { 0x01, 0x03, 0x48, 0x65, 0x6c, /* "Hel" */
  180. /* ping with "Hello" */
  181. 0x89, 0x05, 0x48, 0x65, 0x6c, 0x6c, 0x6f,
  182. 0x80, 0x02, 0x6c, 0x6f }; /* "lo" */
  183. scripted_data_feed_init(&df, msg, sizeof(msg));
  184. df.feedseq[0] = 5;
  185. df.feedseq[1] = 7,
  186. df.feedseq[2] = 4;
  187. wslay_frame_context_init(&ctx, &callbacks, &df);
  188. CU_ASSERT(3 == wslay_frame_recv(ctx, &iocb));
  189. CU_ASSERT_EQUAL(0, iocb.fin);
  190. CU_ASSERT_EQUAL(0, iocb.rsv);
  191. CU_ASSERT_EQUAL(WSLAY_TEXT_FRAME, iocb.opcode);
  192. CU_ASSERT_EQUAL(3, iocb.payload_length);
  193. CU_ASSERT_EQUAL(0, iocb.mask);
  194. CU_ASSERT_EQUAL(3, iocb.data_length);
  195. CU_ASSERT(memcmp("Hel", iocb.data, iocb.data_length) == 0);
  196. CU_ASSERT(5 == wslay_frame_recv(ctx, &iocb));
  197. CU_ASSERT_EQUAL(1, iocb.fin);
  198. CU_ASSERT_EQUAL(0, iocb.rsv);
  199. CU_ASSERT_EQUAL(WSLAY_PING, iocb.opcode);
  200. CU_ASSERT_EQUAL(5, iocb.payload_length);
  201. CU_ASSERT_EQUAL(0, iocb.mask);
  202. CU_ASSERT_EQUAL(5, iocb.data_length);
  203. CU_ASSERT(memcmp("Hello", iocb.data, iocb.data_length) == 0);
  204. CU_ASSERT(2 == wslay_frame_recv(ctx, &iocb));
  205. CU_ASSERT_EQUAL(1, iocb.fin);
  206. CU_ASSERT_EQUAL(0, iocb.rsv);
  207. CU_ASSERT_EQUAL(WSLAY_CONTINUATION_FRAME, iocb.opcode);
  208. CU_ASSERT_EQUAL(2, iocb.payload_length);
  209. CU_ASSERT_EQUAL(0, iocb.mask);
  210. CU_ASSERT_EQUAL(2, iocb.data_length);
  211. CU_ASSERT(memcmp("lo", iocb.data, iocb.data_length) == 0);
  212. wslay_frame_context_free(ctx);
  213. }
  214. void test_wslay_frame_recv_zero_payloadlen(void)
  215. {
  216. wslay_frame_context_ptr ctx;
  217. struct wslay_frame_callbacks callbacks = { NULL,
  218. scripted_recv_callback,
  219. NULL };
  220. struct scripted_data_feed df;
  221. struct wslay_frame_iocb iocb;
  222. /* Unmasked message */
  223. uint8_t msg[] = { 0x81, 0x00 }; /* "" */
  224. scripted_data_feed_init(&df, msg, sizeof(msg));
  225. df.feedseq[0] = 2;
  226. wslay_frame_context_init(&ctx, &callbacks, &df);
  227. CU_ASSERT(0 == wslay_frame_recv(ctx, &iocb));
  228. CU_ASSERT_EQUAL(1, iocb.fin);
  229. CU_ASSERT_EQUAL(0, iocb.rsv);
  230. CU_ASSERT_EQUAL(WSLAY_TEXT_FRAME, iocb.opcode);
  231. CU_ASSERT_EQUAL(0, iocb.payload_length);
  232. CU_ASSERT_EQUAL(0, iocb.mask);
  233. CU_ASSERT_EQUAL(0, iocb.data_length);
  234. wslay_frame_context_free(ctx);
  235. }
  236. void test_wslay_frame_recv_too_large_payload(void)
  237. {
  238. wslay_frame_context_ptr ctx;
  239. struct wslay_frame_callbacks callbacks = { NULL,
  240. scripted_recv_callback,
  241. NULL };
  242. struct scripted_data_feed df;
  243. struct wslay_frame_iocb iocb;
  244. uint8_t msg[] = { 0x81, 0x7f, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff};
  245. scripted_data_feed_init(&df, msg, sizeof(msg));
  246. df.feedseq[0] = sizeof(msg);
  247. wslay_frame_context_init(&ctx, &callbacks, &df);
  248. CU_ASSERT_EQUAL(WSLAY_ERR_PROTO, wslay_frame_recv(ctx, &iocb));
  249. wslay_frame_context_free(ctx);
  250. }
  251. void test_wslay_frame_recv_ctrl_frame_too_large_payload(void)
  252. {
  253. wslay_frame_context_ptr ctx;
  254. struct wslay_frame_callbacks callbacks = { NULL,
  255. scripted_recv_callback,
  256. NULL };
  257. struct scripted_data_feed df;
  258. struct wslay_frame_iocb iocb;
  259. uint8_t msg[] = { 0x88, 0x7e };
  260. scripted_data_feed_init(&df, msg, sizeof(msg));
  261. df.feedseq[0] = sizeof(msg);
  262. wslay_frame_context_init(&ctx, &callbacks, &df);
  263. CU_ASSERT_EQUAL(WSLAY_ERR_PROTO, wslay_frame_recv(ctx, &iocb));
  264. wslay_frame_context_free(ctx);
  265. }
  266. void test_wslay_frame_recv_minimum_ext_payload16(void)
  267. {
  268. wslay_frame_context_ptr ctx;
  269. struct wslay_frame_callbacks callbacks = { NULL,
  270. scripted_recv_callback,
  271. NULL };
  272. struct scripted_data_feed df;
  273. struct wslay_frame_iocb iocb;
  274. uint8_t msg[] = { 0x81, 0x7e, 0x00, 0x7d };
  275. scripted_data_feed_init(&df, msg, sizeof(msg));
  276. df.feedseq[0] = sizeof(msg);
  277. wslay_frame_context_init(&ctx, &callbacks, &df);
  278. CU_ASSERT_EQUAL(WSLAY_ERR_PROTO, wslay_frame_recv(ctx, &iocb));
  279. wslay_frame_context_free(ctx);
  280. }
  281. void test_wslay_frame_recv_minimum_ext_payload64(void)
  282. {
  283. wslay_frame_context_ptr ctx;
  284. struct wslay_frame_callbacks callbacks = { NULL,
  285. scripted_recv_callback,
  286. NULL };
  287. struct scripted_data_feed df;
  288. struct wslay_frame_iocb iocb;
  289. uint8_t msg[] = { 0x81, 0x7f, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0xff, 0xff};
  290. scripted_data_feed_init(&df, msg, sizeof(msg));
  291. df.feedseq[0] = sizeof(msg);
  292. wslay_frame_context_init(&ctx, &callbacks, &df);
  293. CU_ASSERT_EQUAL(WSLAY_ERR_PROTO, wslay_frame_recv(ctx, &iocb));
  294. wslay_frame_context_free(ctx);
  295. }
  296. struct accumulator {
  297. uint8_t buf[4096];
  298. size_t length;
  299. };
  300. static ssize_t accumulator_send_callback(const uint8_t *buf, size_t len,
  301. int flags, void* user_data)
  302. {
  303. struct accumulator *acc = (struct accumulator*)user_data;
  304. assert(acc->length+len < sizeof(acc->buf));
  305. memcpy(acc->buf+acc->length, buf, len);
  306. acc->length += len;
  307. return len;
  308. }
  309. static int static_genmask_callback(uint8_t *buf, size_t len,
  310. void* user_data)
  311. {
  312. static const uint8_t makskey[] = { 0x37u, 0xfau, 0x21u, 0x3du };
  313. memcpy(buf, makskey, 4);
  314. return 0;
  315. }
  316. void test_wslay_frame_send(void)
  317. {
  318. wslay_frame_context_ptr ctx;
  319. struct wslay_frame_callbacks callbacks = { accumulator_send_callback,
  320. NULL,
  321. static_genmask_callback };
  322. struct accumulator acc;
  323. struct wslay_frame_iocb iocb;
  324. /* Masked text frame containing "Hello" */
  325. uint8_t msg[] = { 0x81u, 0x85u, 0x37u, 0xfau, 0x21u, 0x3du, 0x7fu, 0x9fu,
  326. 0x4du, 0x51u, 0x58u };
  327. wslay_frame_context_init(&ctx, &callbacks, &acc);
  328. memset(&iocb, 0, sizeof(iocb));
  329. acc.length = 0;
  330. iocb.fin = 1;
  331. iocb.opcode = WSLAY_TEXT_FRAME;
  332. iocb.mask = 1;
  333. iocb.payload_length = 5;
  334. iocb.data = (const uint8_t*)"Hello";
  335. iocb.data_length = 5;
  336. CU_ASSERT(5 == wslay_frame_send(ctx, &iocb));
  337. CU_ASSERT_EQUAL(sizeof(msg), acc.length);
  338. CU_ASSERT(memcmp(msg, acc.buf, sizeof(msg)) == 0);
  339. wslay_frame_context_free(ctx);
  340. }
  341. void test_wslay_frame_send_fragmented(void)
  342. {
  343. wslay_frame_context_ptr ctx;
  344. struct wslay_frame_callbacks callbacks = { accumulator_send_callback,
  345. NULL,
  346. static_genmask_callback };
  347. struct accumulator acc;
  348. struct wslay_frame_iocb iocb;
  349. /* Unmasked message */
  350. uint8_t msg1[] = { 0x01, 0x03, 0x48, 0x65, 0x6c }; /* "Hel" */
  351. uint8_t msg2[] = { 0x80, 0x02, 0x6c, 0x6f }; /* "lo" */
  352. wslay_frame_context_init(&ctx, &callbacks, &acc);
  353. memset(&iocb, 0, sizeof(iocb));
  354. acc.length = 0;
  355. iocb.fin = 0;
  356. iocb.opcode = WSLAY_TEXT_FRAME;
  357. iocb.mask = 0;
  358. iocb.payload_length = 3;
  359. iocb.data = (const uint8_t*)"Hel";
  360. iocb.data_length = 3;
  361. CU_ASSERT(3 == wslay_frame_send(ctx, &iocb));
  362. CU_ASSERT_EQUAL(sizeof(msg1), acc.length);
  363. CU_ASSERT(memcmp(msg1, acc.buf, sizeof(msg1)) == 0);
  364. acc.length = 0;
  365. iocb.fin = 1;
  366. iocb.opcode = WSLAY_CONTINUATION_FRAME;
  367. iocb.payload_length = 2;
  368. iocb.data = (const uint8_t*)"lo";
  369. iocb.data_length = 2;
  370. CU_ASSERT(2 == wslay_frame_send(ctx, &iocb));
  371. CU_ASSERT_EQUAL(sizeof(msg2), acc.length);
  372. CU_ASSERT(memcmp(msg2, acc.buf, sizeof(msg2)) == 0);
  373. wslay_frame_context_free(ctx);
  374. }
  375. void test_wslay_frame_send_interleaved_ctrl_frame(void)
  376. {
  377. wslay_frame_context_ptr ctx;
  378. struct wslay_frame_callbacks callbacks = { accumulator_send_callback,
  379. NULL,
  380. static_genmask_callback };
  381. struct accumulator acc;
  382. struct wslay_frame_iocb iocb;
  383. /* Unmasked message */
  384. /* text with "Hel", with fin = 0 */
  385. uint8_t msg1[] = { 0x01, 0x03, 0x48, 0x65, 0x6c };
  386. /* ping with "Hello" */
  387. uint8_t msg2[] = { 0x89, 0x05, 0x48, 0x65, 0x6c, 0x6c, 0x6f };
  388. /* text with "lo", continuation frame for msg1, with fin = 1 */
  389. uint8_t msg3[] = { 0x80, 0x02, 0x6c, 0x6f };
  390. wslay_frame_context_init(&ctx, &callbacks, &acc);
  391. memset(&iocb, 0, sizeof(iocb));
  392. acc.length = 0;
  393. iocb.fin = 0;
  394. iocb.opcode = WSLAY_TEXT_FRAME;
  395. iocb.mask = 0;
  396. iocb.payload_length = 3;
  397. iocb.data = (const uint8_t*)"Hel";
  398. iocb.data_length = 3;
  399. CU_ASSERT(3 == wslay_frame_send(ctx, &iocb));
  400. CU_ASSERT_EQUAL(sizeof(msg1), acc.length);
  401. CU_ASSERT(memcmp(msg1, acc.buf, sizeof(msg1)) == 0);
  402. acc.length = 0;
  403. iocb.fin = 1;
  404. iocb.opcode = WSLAY_PING;
  405. iocb.payload_length = 5;
  406. iocb.data = (const uint8_t*)"Hello";
  407. iocb.data_length = 5;
  408. CU_ASSERT(5 == wslay_frame_send(ctx, &iocb));
  409. CU_ASSERT_EQUAL(sizeof(msg2), acc.length);
  410. CU_ASSERT(memcmp(msg2, acc.buf, sizeof(msg2)) == 0);
  411. acc.length = 0;
  412. iocb.fin = 1;
  413. iocb.opcode = WSLAY_CONTINUATION_FRAME;
  414. iocb.payload_length = 2;
  415. iocb.data = (const uint8_t*)"lo";
  416. iocb.data_length = 2;
  417. CU_ASSERT(2 == wslay_frame_send(ctx, &iocb));
  418. CU_ASSERT_EQUAL(sizeof(msg3), acc.length);
  419. CU_ASSERT(memcmp(msg3, acc.buf, sizeof(msg3)) == 0);
  420. wslay_frame_context_free(ctx);
  421. }
  422. void test_wslay_frame_send_1byte_masked(void)
  423. {
  424. wslay_frame_context_ptr ctx;
  425. struct wslay_frame_callbacks callbacks = { scripted_send_callback,
  426. NULL,
  427. static_genmask_callback };
  428. struct wslay_frame_iocb iocb;
  429. /* Masked text frame containing "Hello" */
  430. uint8_t msg[] = { 0x81u, 0x85u, 0x37u, 0xfau, 0x21u, 0x3du, 0x7fu, 0x9fu,
  431. 0x4du, 0x51u, 0x58u };
  432. uint8_t hello[] = "Hello";
  433. struct scripted_data_feed df;
  434. size_t i;
  435. scripted_data_feed_init(&df, NULL, 0);
  436. for(i = 0; i < sizeof(msg); ++i) {
  437. df.feedseq[i] = 1;
  438. }
  439. wslay_frame_context_init(&ctx, &callbacks, &df);
  440. memset(&iocb, 0, sizeof(iocb));
  441. iocb.fin = 1;
  442. iocb.opcode = WSLAY_TEXT_FRAME;
  443. iocb.mask = 1;
  444. iocb.payload_length = 5;
  445. iocb.data = hello;
  446. iocb.data_length = sizeof(hello)-1;
  447. for(i = 0; i < 5; ++i) {
  448. CU_ASSERT_EQUAL(WSLAY_ERR_WANT_WRITE, wslay_frame_send(ctx, &iocb));
  449. }
  450. CU_ASSERT_EQUAL(5, wslay_frame_send(ctx, &iocb));
  451. wslay_frame_context_free(ctx);
  452. }
  453. void test_wslay_frame_send_zero_payloadlen(void)
  454. {
  455. wslay_frame_context_ptr ctx;
  456. struct wslay_frame_callbacks callbacks = { accumulator_send_callback,
  457. NULL,
  458. static_genmask_callback };
  459. struct accumulator acc;
  460. struct wslay_frame_iocb iocb;
  461. /* Unmasked message */
  462. uint8_t msg[] = { 0x81, 0x00 }; /* "" */
  463. acc.length = 0;
  464. wslay_frame_context_init(&ctx, &callbacks, &acc);
  465. memset(&iocb, 0, sizeof(iocb));
  466. iocb.fin = 1;
  467. iocb.opcode = WSLAY_TEXT_FRAME;
  468. iocb.mask = 0;
  469. iocb.payload_length = 0;
  470. iocb.data_length = 0;
  471. CU_ASSERT(0 == wslay_frame_send(ctx, &iocb));
  472. CU_ASSERT_EQUAL(sizeof(msg), acc.length);
  473. CU_ASSERT(memcmp(msg, acc.buf, sizeof(msg)) == 0);
  474. wslay_frame_context_free(ctx);
  475. }
  476. void test_wslay_frame_send_too_large_payload(void)
  477. {
  478. wslay_frame_context_ptr ctx;
  479. struct wslay_frame_callbacks callbacks;
  480. struct wslay_frame_iocb iocb;
  481. wslay_frame_context_init(&ctx, &callbacks, NULL);
  482. memset(&iocb, 0, sizeof(iocb));
  483. iocb.fin = 1;
  484. iocb.opcode = WSLAY_TEXT_FRAME;
  485. iocb.mask = 0;
  486. iocb.payload_length = UINT64_MAX;
  487. CU_ASSERT_EQUAL(WSLAY_ERR_INVALID_ARGUMENT,
  488. wslay_frame_send(ctx, &iocb));
  489. wslay_frame_context_free(ctx);
  490. }
  491. void test_wslay_frame_send_ctrl_frame_too_large_payload(void)
  492. {
  493. wslay_frame_context_ptr ctx;
  494. struct wslay_frame_callbacks callbacks;
  495. struct wslay_frame_iocb iocb;
  496. wslay_frame_context_init(&ctx, &callbacks, NULL);
  497. memset(&iocb, 0, sizeof(iocb));
  498. iocb.fin = 1;
  499. iocb.opcode = WSLAY_PING;
  500. iocb.mask = 0;
  501. iocb.payload_length = 1024;
  502. CU_ASSERT_EQUAL(WSLAY_ERR_INVALID_ARGUMENT,
  503. wslay_frame_send(ctx, &iocb));
  504. wslay_frame_context_free(ctx);
  505. }