1 /* This case tests chunked resposne. */
2 #include "tx_api.h"
3 #include "nx_api.h"
4 #include "fx_api.h"
5 #include "nx_web_http_client.h"
6 #include "nx_web_http_server.h"
7
8 extern void test_control_return(UINT);
9
10 #if !defined(NX_DISABLE_IPV4)
11
12 #include "test_device_cert.c"
13 #include "test_ca_cert.c"
14 #define ca_cert_der test_ca_cert_der
15 #define ca_cert_der_len test_ca_cert_der_len
16
17 #define DEMO_STACK_SIZE 4096
18
19 /* Set up FileX and file memory resources. */
20 static CHAR ram_disk_memory[4096];
21 static FX_MEDIA ram_disk;
22 static UCHAR media_memory[4096];
23
24 static UCHAR server_stack[16000];
25
26 /* Define device drivers. */
27 extern void _fx_ram_driver(FX_MEDIA *media_ptr);
28 extern void _nx_ram_network_driver_1024(NX_IP_DRIVER *driver_req_ptr);
29
30 /* Set up the HTTP client global variables. */
31
32 #define CLIENT_PACKET_SIZE (NX_WEB_HTTP_CLIENT_MIN_PACKET_SIZE * 2)
33
34 static TX_THREAD client_thread;
35 static NX_PACKET_POOL client_pool;
36 static NX_WEB_HTTP_CLIENT my_client;
37 static NX_IP client_ip;
38 static UINT error_counter;
39
40 /* Set up the HTTP server global variables */
41
42 #define SERVER_PACKET_SIZE (NX_WEB_HTTP_SERVER_MIN_PACKET_SIZE * 2)
43
44 static NX_WEB_HTTP_SERVER my_server;
45 static NX_PACKET_POOL server_pool;
46 static TX_THREAD server_thread;
47 static NX_IP server_ip;
48 static NXD_ADDRESS server_ip_address;
49 static UINT http_server_start = 0;
50 static UINT http_client_stop = 0;
51
52 static void thread_client_entry(ULONG thread_input);
53 static void thread_server_entry(ULONG thread_input);
54
55 #define HTTP_SERVER_ADDRESS IP_ADDRESS(192,168,0,105)
56 #define HTTP_CLIENT_ADDRESS IP_ADDRESS(192,168,0,123)
57
58 #ifdef NX_WEB_HTTPS_ENABLE
59 static UINT https_server_start = 0;
60 static UINT https_client_stop = 0;
61 static UINT loop = 2;
62 extern const NX_SECURE_TLS_CRYPTO nx_crypto_tls_ciphers;
63 static CHAR crypto_metadata_server[20000 * NX_WEB_HTTP_SERVER_SESSION_MAX];
64 static CHAR crypto_metadata_client[20000 * NX_WEB_HTTP_SERVER_SESSION_MAX];
65 static UCHAR tls_packet_buffer[18500];
66 static NX_SECURE_X509_CERT certificate;
67 static NX_SECURE_X509_CERT trusted_certificate;
68 static NX_SECURE_X509_CERT remote_certificate, remote_issuer;
69 static UCHAR remote_cert_buffer[2000];
70 static UCHAR remote_issuer_buffer[2000];
71 #else
72 static UINT loop = 1;
73 #endif /* NX_WEB_HTTPS_ENABLE */
74
75 static UINT server_request_callback(NX_WEB_HTTP_SERVER *server_ptr, UINT request_type, CHAR *resource, NX_PACKET *packet_ptr);
76
77 static char pkt[] = {
78 0x74, 0x65, 0x73, 0x74, 0x74, 0x65, 0x73, 0x74, /* testtest */
79 0x74, 0x65, 0x73, 0x74, 0x74, 0x65, 0x73, 0x74, /* testtest */
80 0x74, 0x65, 0x73, 0x74, 0x74, 0x65, 0x73, 0x74, /* testtest */
81 0x74, 0x65, 0x73, 0x74, 0x74, 0x65, 0x73, 0x74, /* testtest */
82 0x74, 0x65, 0x73, 0x74, 0x74, 0x65, 0x73, 0x74, /* testtest */
83 0x74, 0x65, 0x73, 0x74, 0x74, 0x65, 0x73, 0x74, /* testtest */
84 0x74, 0x65, 0x73, 0x74, 0x74, 0x65, 0x73, 0x74, /* testtest */
85 0x74, 0x65, 0x73, 0x74, 0x74, 0x65, 0x73, 0x74, /* testtest */
86 0x74, 0x65, 0x73, 0x74, 0x74, 0x65, 0x73, 0x74, /* testtest */
87 0x74, 0x65, 0x73, 0x74, 0x74, 0x65, 0x73, 0x74, /* testtest */
88 0x74, 0x65, 0x73, 0x74, 0x74, 0x65, 0x73, 0x74, /* testtest */
89 0x74, 0x65, 0x73, 0x74, 0x74, 0x65, 0x73, 0x74, /* testtest */
90 0x74, 0x65, 0x73, 0x74, 0x74, 0x65, 0x73, 0x74, /* testtest */
91 0x74, 0x65, 0x73, 0x74, 0x74, 0x65, 0x73, 0x74, /* testtest */
92 0x74, 0x65, 0x73, 0x74, 0x74, 0x65, 0x73, 0x74, /* testtest */
93 0x74, 0x65, 0x73, 0x74, 0x74, 0x65, 0x73, 0x74, /* testtest */
94 0x74, 0x65, 0x73, 0x74, 0x74, 0x65, 0x73, 0x74, /* testtest */
95 0x74, 0x65, 0x73, 0x74, 0x74, 0x65, 0x73, 0x74, /* testtest */
96 0x74, 0x65, 0x73, 0x74, 0x74, 0x65, 0x73, 0x74, /* testtest */
97 0x74, 0x65, 0x73, 0x74, 0x74, 0x65, 0x73, 0x74, /* testtest */
98 0x74, 0x65, 0x73, 0x74, 0x74, 0x65, 0x73, 0x74, /* testtest */
99 0x74, 0x65, 0x73, 0x74, 0x74, 0x65, 0x73, 0x74, /* testtest */
100 0x74, 0x65, 0x73, 0x74, 0x74, 0x65, 0x73, 0x74, /* testtest */
101 0x74, 0x65, 0x73, 0x74, 0x74, 0x65, 0x73, 0x74, /* testtest */
102 0x74, 0x65, 0x73, 0x74, 0x74, 0x65, 0x73, 0x74, /* testtest */
103 0x74, 0x65, 0x73, 0x74, 0x74, 0x65, 0x73, 0x74, /* testtest */
104 0x74, 0x65, 0x73, 0x74, 0x74, 0x65, 0x73, 0x74, /* testtest */
105 0x74, 0x65, 0x73, 0x74, 0x74, 0x65, 0x73, 0x74, /* testtest */
106 0x74, 0x65, 0x73, 0x74, 0x74, 0x65, 0x73, 0x74, /* testtest */
107 0x74, 0x65, 0x73, 0x74, 0x74, 0x65, 0x73, 0x74, /* testtest */
108 0x74, 0x65, 0x73, 0x74, 0x74, 0x65, 0x73, 0x74, /* testtest */
109 0x74, 0x65, 0x73, 0x74, 0x74, 0x65, 0x73, 0x74, /* testtest */
110 0x74, 0x65, 0x73, 0x74, 0x74, 0x65, 0x73, 0x74, /* testtest */
111 0x74, 0x65, 0x73, 0x74, 0x74, 0x65, 0x73, 0x74, /* testtest */
112 0x74, 0x65, 0x73, 0x74, 0x74, 0x65, 0x73, 0x74, /* testtest */
113 0x74, 0x65, 0x73, 0x74, 0x74, 0x65, 0x73, 0x74, /* testtest */
114 };
115
116 static UINT test_chunk[] = {20, 120, 19, 34};
117 static UINT test_num = sizeof(test_chunk)/sizeof(UINT);
118
119 #ifdef CTEST
test_application_define(void * first_unused_memory)120 VOID test_application_define(void *first_unused_memory)
121 #else
122 void netx_web_chunked_response_test_application_define(void *first_unused_memory)
123 #endif
124 {
125 CHAR *pointer;
126 UINT status;
127
128
129 error_counter = 0;
130
131 /* Setup the working pointer. */
132 pointer = (CHAR *) first_unused_memory;
133
134 /* Create a helper thread for the server. */
135 tx_thread_create(&server_thread, "HTTP Server thread", thread_server_entry, 0,
136 pointer, DEMO_STACK_SIZE,
137 NX_WEB_HTTP_SERVER_PRIORITY, NX_WEB_HTTP_SERVER_PRIORITY, TX_NO_TIME_SLICE, TX_AUTO_START);
138
139 pointer = pointer + DEMO_STACK_SIZE;
140
141 /* Initialize the NetX system. */
142 nx_system_initialize();
143
144 /* Create the server packet pool. */
145 status = nx_packet_pool_create(&server_pool, "HTTP Server Packet Pool", SERVER_PACKET_SIZE,
146 pointer, SERVER_PACKET_SIZE*16);
147 pointer = pointer + SERVER_PACKET_SIZE * 16;
148 if (status)
149 error_counter++;
150
151 /* Create an IP instance. */
152 status = nx_ip_create(&server_ip, "HTTP Server IP", HTTP_SERVER_ADDRESS,
153 0xFFFFFF00UL, &server_pool, _nx_ram_network_driver_1024,
154 pointer, 4096, 1);
155 pointer = pointer + 4096;
156 if (status)
157 error_counter++;
158
159 /* Enable ARP and supply ARP cache memory for the server IP instance. */
160 status = nx_arp_enable(&server_ip, (void *) pointer, 1024);
161 pointer = pointer + 1024;
162 if (status)
163 error_counter++;
164
165 /* Enable TCP traffic. */
166 status = nx_tcp_enable(&server_ip);
167 if (status)
168 error_counter++;
169
170 /* Create the HTTP Client thread. */
171 status = tx_thread_create(&client_thread, "HTTP Client", thread_client_entry, 0,
172 pointer, DEMO_STACK_SIZE,
173 NX_WEB_HTTP_SERVER_PRIORITY + 2, NX_WEB_HTTP_SERVER_PRIORITY + 2, TX_NO_TIME_SLICE, TX_AUTO_START);
174 pointer = pointer + DEMO_STACK_SIZE;
175 if (status)
176 error_counter++;
177
178 /* Create the Client packet pool. */
179 status = nx_packet_pool_create(&client_pool, "HTTP Client Packet Pool", CLIENT_PACKET_SIZE,
180 pointer, CLIENT_PACKET_SIZE*16);
181 pointer = pointer + CLIENT_PACKET_SIZE * 16;
182 if (status)
183 error_counter++;
184
185 /* Create an IP instance. */
186 status = nx_ip_create(&client_ip, "HTTP Client IP", HTTP_CLIENT_ADDRESS,
187 0xFFFFFF00UL, &client_pool, _nx_ram_network_driver_1024,
188 pointer, 2048, 1);
189 pointer = pointer + 2048;
190 if (status)
191 error_counter++;
192
193 status = nx_arp_enable(&client_ip, (void *) pointer, 1024);
194 pointer = pointer + 2048;
195 if (status)
196 error_counter++;
197
198 /* Enable TCP traffic. */
199 status = nx_tcp_enable(&client_ip);
200 if (status)
201 error_counter++;
202 }
203
204 #ifdef NX_WEB_HTTPS_ENABLE
205 /* Define the TLS setup callback function. */
tls_setup_callback(NX_WEB_HTTP_CLIENT * client_ptr,NX_SECURE_TLS_SESSION * tls_session)206 static UINT tls_setup_callback(NX_WEB_HTTP_CLIENT *client_ptr, NX_SECURE_TLS_SESSION *tls_session)
207 {
208 UINT status;
209
210
211 /* Initialize and create TLS session. */
212 status = nx_secure_tls_session_create(tls_session, &nx_crypto_tls_ciphers, crypto_metadata_client, sizeof(crypto_metadata_client));
213
214 /* Check status. */
215 if (status)
216 {
217 return(status);
218 }
219
220 /* Allocate space for packet reassembly. */
221 status = nx_secure_tls_session_packet_buffer_set(&(client_ptr -> nx_web_http_client_tls_session), tls_packet_buffer, sizeof(tls_packet_buffer));
222
223 /* Check status. */
224 if (status)
225 {
226 return(status);
227 }
228
229 /* Add a CA Certificate to our trusted store for verifying incoming server certificates. */
230 nx_secure_x509_certificate_initialize(&trusted_certificate, ca_cert_der, ca_cert_der_len, NX_NULL, 0, NULL, 0, NX_SECURE_X509_KEY_TYPE_NONE);
231 nx_secure_tls_trusted_certificate_add(&(client_ptr -> nx_web_http_client_tls_session), &trusted_certificate);
232
233 /* Need to allocate space for the certificate coming in from the remote host. */
234 nx_secure_tls_remote_certificate_allocate(&(client_ptr -> nx_web_http_client_tls_session), &remote_certificate, remote_cert_buffer, sizeof(remote_cert_buffer));
235 nx_secure_tls_remote_certificate_allocate(&(client_ptr -> nx_web_http_client_tls_session), &remote_issuer, remote_issuer_buffer, sizeof(remote_issuer_buffer));
236
237 return(NX_SUCCESS);
238 }
239 #endif /* NX_WEB_HTTPS_ENABLE */
240
thread_client_entry(ULONG thread_input)241 void thread_client_entry(ULONG thread_input)
242 {
243 UINT i;
244 UINT status;
245 NX_PACKET *recv_packet;
246 UINT content_length = 0;
247 UINT total_length = 0;
248
249 for (i = 0; i < test_num; i++)
250 {
251 total_length += test_chunk[i];
252 }
253
254 /* Give IP task and driver a chance to initialize the system. */
255 tx_thread_sleep(NX_IP_PERIODIC_RATE);
256
257 /* Set server IP address. */
258 server_ip_address.nxd_ip_address.v4 = HTTP_SERVER_ADDRESS;
259 server_ip_address.nxd_ip_version = NX_IP_VERSION_V4;
260
261 /* First loop test HTTP, second loop test HTTPS. */
262 for (i = 0; i < loop ; i++)
263 {
264 if (i == 0)
265 {
266
267 /* Wait HTTP server started. */
268 while(!http_server_start)
269 {
270 tx_thread_sleep(NX_IP_PERIODIC_RATE);
271 }
272 }
273 #ifdef NX_WEB_HTTPS_ENABLE
274 else
275 {
276
277 /* Wait HTTPS server started. */
278 while(!https_server_start)
279 {
280 tx_thread_sleep(NX_IP_PERIODIC_RATE);
281 }
282 }
283 #endif /* NX_WEB_HTTPS_ENABLE */
284
285
286 /* Create an HTTP client instance. */
287 status = nx_web_http_client_create(&my_client, "HTTP Client", &client_ip, &client_pool, 1536);
288
289 /* Check status. */
290 if (status)
291 error_counter++;
292
293
294 /* Check status. */
295 if (status)
296 error_counter++;
297
298 /* Connect to server. */
299 if (i == 0)
300 {
301 status = nx_web_http_client_connect(&my_client, &server_ip_address, NX_WEB_HTTP_SERVER_PORT, NX_WAIT_FOREVER);
302 }
303 #ifdef NX_WEB_HTTPS_ENABLE
304 else
305 {
306 status = nx_web_http_client_secure_connect(&my_client, &server_ip_address, NX_WEB_HTTPS_SERVER_PORT,
307 tls_setup_callback, NX_WAIT_FOREVER);
308 }
309 #endif /* NX_WEB_HTTPS_ENABLE */
310
311 /* Check status. */
312 if (status)
313 error_counter++;
314
315 /* Initialize the request. */
316 status = nx_web_http_client_request_initialize(&my_client, NX_WEB_HTTP_METHOD_GET, "/index.htm", "www.abc.com",
317 0, NX_TRUE, "name", "password", NX_WAIT_FOREVER);
318
319 /* Check status. */
320 if (status)
321 error_counter++;
322
323 /* Send the request. */
324 status = nx_web_http_client_request_send(&my_client, NX_WAIT_FOREVER);
325
326 /* Check status. */
327 if (status)
328 error_counter++;
329
330 content_length = 0;
331
332 /* Get response from server. */
333 while (1)
334 {
335 status = nx_web_http_client_response_body_get(&my_client, &recv_packet, 1 * NX_IP_PERIODIC_RATE);
336
337 if (status)
338 {
339 if (status == NX_WEB_HTTP_GET_DONE)
340 {
341 nx_packet_release(recv_packet);
342 }
343 break;
344 }
345 else
346 {
347 content_length += recv_packet -> nx_packet_length;
348 nx_packet_release(recv_packet);
349 }
350 }
351
352 /* Check status. */
353 if ((status != NX_WEB_HTTP_GET_DONE) || (content_length != total_length))
354 error_counter++;
355
356 status = nx_web_http_client_delete(&my_client);
357 if (status)
358 error_counter++;
359
360 /* Set the flag. */
361 if (i == 0)
362 {
363 http_client_stop = 1;
364 }
365 #ifdef NX_WEB_HTTPS_ENABLE
366 else
367 {
368 https_client_stop = 1;
369 }
370 #endif /* NX_WEB_HTTPS_ENABLE */
371 }
372 }
373
374
375 /* Define the helper HTTP server thread. */
thread_server_entry(ULONG thread_input)376 void thread_server_entry(ULONG thread_input)
377 {
378 UINT i;
379 UINT status;
380 FX_FILE my_file;
381 UINT server_port = NX_WEB_HTTP_SERVER_PORT;
382
383
384 /* Print out test information banner. */
385 printf("NetX Test: Web Chunked Response Test.................................");
386
387 /* Check for earlier error. */
388 if(error_counter)
389 {
390 printf("ERROR!\n");
391 test_control_return(1);
392 }
393
394 fx_media_format(&ram_disk,
395 _fx_ram_driver, // Driver entry
396 ram_disk_memory, // RAM disk memory pointer
397 media_memory, // Media buffer pointer
398 sizeof(media_memory), // Media buffer size
399 "MY_RAM_DISK", // Volume Name
400 1, // Number of FATs
401 32, // Directory Entries
402 0, // Hidden sectors
403 256, // Total sectors
404 512, // Sector size
405 8, // Sectors per cluster
406 1, // Heads
407 1); // Sectors per track
408
409 /* Open the RAM disk. */
410 status = fx_media_open(&ram_disk, "RAM DISK", _fx_ram_driver, ram_disk_memory, media_memory, sizeof(media_memory)) ;
411 status += fx_file_create(&ram_disk, "index.htm");
412 status += fx_file_open(&ram_disk, &my_file, "index.htm", FX_OPEN_FOR_WRITE);
413 status += fx_file_write(&my_file, "https server", 12);
414 status += fx_file_close(&my_file);
415 if(status)
416 error_counter++;
417
418 /* Give NetX a chance to initialize the system. */
419 tx_thread_sleep(NX_IP_PERIODIC_RATE);
420
421 /* First loop test HTTP, second loop test HTTPS. */
422 for (i = 0; i < loop; i++)
423 {
424
425 if (i == 1)
426 {
427 server_port = NX_WEB_HTTPS_SERVER_PORT;
428 }
429
430 /* Create the HTTP Server. */
431 status = nx_web_http_server_create(&my_server, "My HTTP Server", &server_ip, server_port, &ram_disk,
432 &server_stack, sizeof(server_stack), &server_pool,
433 NX_NULL, server_request_callback);
434 if (status)
435 error_counter++;
436
437 #ifdef NX_WEB_HTTPS_ENABLE
438 /* Set TLS for HTTPS. */
439 if (i == 1)
440 {
441 /* Initialize device certificate (used for all sessions in HTTPS server). */
442 memset(&certificate, 0, sizeof(certificate));
443 nx_secure_x509_certificate_initialize(&certificate, test_device_cert_der, test_device_cert_der_len, NX_NULL, 0, test_device_cert_key_der, test_device_cert_key_der_len, NX_SECURE_X509_KEY_TYPE_RSA_PKCS1_DER);
444
445 /* Setup TLS session data for the TCP server. */
446 status = nx_web_http_server_secure_configure(&my_server, &nx_crypto_tls_ciphers,
447 crypto_metadata_server, sizeof(crypto_metadata_server), tls_packet_buffer, sizeof(tls_packet_buffer),
448 &certificate, NX_NULL, 0, NX_NULL, 0, NX_NULL, 0);
449 if (status)
450 error_counter++;
451 }
452 #endif /* NX_WEB_HTTPS_ENABLE */
453
454 /* OK to start the HTTP Server. */
455 status = nx_web_http_server_start(&my_server);
456 if (status)
457 error_counter++;
458
459 /* Set the flag. */
460 if (i == 0)
461 {
462 http_server_start = 1;
463
464 /* Wait HTTP test finished. */
465 while(!http_client_stop)
466 {
467 tx_thread_sleep(NX_IP_PERIODIC_RATE);
468 }
469 }
470 #ifdef NX_WEB_HTTPS_ENABLE
471 else
472 {
473 https_server_start = 1;
474
475 /* Wait HTTPS test finished. */
476 while(!https_client_stop)
477 {
478 tx_thread_sleep(NX_IP_PERIODIC_RATE);
479 }
480 }
481 #endif /* NX_WEB_HTTPS_ENABLE */
482
483 status = nx_web_http_server_delete(&my_server);
484 if (status)
485 error_counter++;
486 }
487
488 /* Check packet pool. */
489 if (server_pool.nx_packet_pool_available != server_pool.nx_packet_pool_total)
490 {
491 error_counter++;
492 }
493
494 if (client_pool.nx_packet_pool_available != client_pool.nx_packet_pool_total)
495 {
496 error_counter++;
497 }
498
499 if(error_counter)
500 {
501 printf("ERROR!\n");
502 test_control_return(1);
503 }
504 else
505 {
506 printf("SUCCESS!\n");
507 test_control_return(0);
508 }
509 }
510
511 /* Define the server request callback function. */
server_request_callback(NX_WEB_HTTP_SERVER * server_ptr,UINT request_type,CHAR * resource,NX_PACKET * packet_ptr)512 static UINT server_request_callback(NX_WEB_HTTP_SERVER *server_ptr, UINT request_type, CHAR *resource, NX_PACKET *packet_ptr)
513 {
514 NX_PACKET *response_pkt;
515 UINT status;
516 UINT i = 0;
517
518 /* Process multipart data. */
519 if(request_type == NX_WEB_HTTP_SERVER_GET_REQUEST)
520 {
521
522 /* Generate HTTP header. */
523 status = nx_web_http_server_callback_generate_response_header(server_ptr,
524 &response_pkt, NX_WEB_HTTP_STATUS_OK, 0, "text/html",
525 "Transfer-Encoding: chunked\r\n");
526 if (status)
527 {
528 return(status);
529 }
530
531 status = nx_web_http_server_callback_packet_send(server_ptr, response_pkt);
532 if (status)
533 {
534 nx_packet_release(response_pkt);
535 return(status);
536 }
537
538 /* Send the chunk data. */
539 for (i = 0; i < test_num; i++)
540 {
541
542 status = nx_web_http_server_response_packet_allocate(server_ptr, &response_pkt, NX_WAIT_FOREVER);
543 if (status)
544 {
545 return(status);
546 }
547
548 status = nx_web_http_server_response_chunked_set(server_ptr, test_chunk[i], response_pkt);
549 if (status)
550 {
551 return(status);
552 }
553
554 nx_packet_data_append(response_pkt, pkt, test_chunk[i], server_ptr -> nx_web_http_server_packet_pool_ptr, NX_WAIT_FOREVER);
555
556 status = nx_web_http_server_callback_packet_send(server_ptr, response_pkt);
557 if (status)
558 {
559 nx_packet_release(response_pkt);
560 return(status);
561 }
562 }
563 }
564 else
565 {
566 /* Indicate we have not processed the response to client yet. */
567 return(NX_SUCCESS);
568 }
569
570 /* Indicate the response to client is transmitted. */
571 return(NX_WEB_HTTP_CALLBACK_COMPLETED);
572 }
573
574 #else
575
576 #ifdef CTEST
test_application_define(void * first_unused_memory)577 VOID test_application_define(void *first_unused_memory)
578 #else
579 void netx_web_chunked_response_test_application_define(void *first_unused_memory)
580 #endif
581 {
582
583 /* Print out test information banner. */
584 printf("NetX Test: Web Chunked Response Test.................................N/A\n");
585
586 test_control_return(3);
587 }
588 #endif
589
