1 /***************************************************************************
2 * _ _ ____ _
3 * Project ___| | | | _ \| |
4 * / __| | | | |_) | |
5 * | (__| |_| | _ <| |___
6 * \___|\___/|_| \_\_____|
7 *
8 * Copyright (C) Daniel Stenberg, <daniel@haxx.se>, et al.
9 *
10 * This software is licensed as described in the file COPYING, which
11 * you should have received as part of this distribution. The terms
12 * are also available at https://curl.se/docs/copyright.html.
13 *
14 * You may opt to use, copy, modify, merge, publish, distribute and/or sell
15 * copies of the Software, and permit persons to whom the Software is
16 * furnished to do so, under the terms of the COPYING file.
17 *
18 * This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY
19 * KIND, either express or implied.
20 *
21 * SPDX-License-Identifier: curl
22 *
23 ***************************************************************************/
24 #include "curlcheck.h"
25
26 #include "urldata.h"
27 #include "dynbuf.h"
28 #include "dynhds.h"
29 #include "curl_trc.h"
30
unit_setup(void)31 static CURLcode unit_setup(void)
32 {
33 return CURLE_OK;
34 }
35
unit_stop(void)36 static void unit_stop(void)
37 {
38 }
39
40
41 UNITTEST_START
42
43 struct dynhds hds;
44 struct dynbuf dbuf;
45 CURLcode result;
46 size_t i;
47
48 /* add 1 more header than allowed */
49 Curl_dynhds_init(&hds, 2, 128);
50 fail_if(Curl_dynhds_count(&hds), "should be empty");
51 fail_if(Curl_dynhds_add(&hds, "test1", 5, "123", 3), "add failed");
52 fail_if(Curl_dynhds_add(&hds, "test2", 5, "456", 3), "add failed");
53 /* remove and add without exceeding limits */
54 for(i = 0; i < 100; ++i) {
55 if(Curl_dynhds_remove(&hds, "test2", 5) != 1) {
56 fail_if(TRUE, "should");
57 break;
58 }
59 if(Curl_dynhds_add(&hds, "test2", 5, "456", 3)) {
60 fail_if(TRUE, "add failed");
61 break;
62 }
63 }
64 fail_unless(Curl_dynhds_count(&hds) == 2, "should hold 2");
65 /* set, replacing previous entry without exceeding limits */
66 for(i = 0; i < 100; ++i) {
67 if(Curl_dynhds_set(&hds, "test2", 5, "456", 3)) {
68 fail_if(TRUE, "add failed");
69 break;
70 }
71 }
72 fail_unless(Curl_dynhds_count(&hds) == 2, "should hold 2");
73 /* exceed limit on # of entries */
74 result = Curl_dynhds_add(&hds, "test3", 5, "789", 3);
75 fail_unless(result, "add should have failed");
76
77 fail_unless(Curl_dynhds_count_name(&hds, "test", 4) == 0, "false positive");
78 fail_unless(Curl_dynhds_count_name(&hds, "test1", 4) == 0, "false positive");
79 fail_if(Curl_dynhds_get(&hds, "test1", 4), "false positive");
80 fail_unless(Curl_dynhds_get(&hds, "test1", 5), "false negative");
81 fail_unless(Curl_dynhds_count_name(&hds, "test1", 5) == 1, "should");
82 fail_unless(Curl_dynhds_ccount_name(&hds, "test2") == 1, "should");
83 fail_unless(Curl_dynhds_cget(&hds, "test2"), "should");
84 fail_unless(Curl_dynhds_ccount_name(&hds, "TEST2") == 1, "should");
85 fail_unless(Curl_dynhds_ccontains(&hds, "TesT2"), "should");
86 fail_unless(Curl_dynhds_contains(&hds, "TeSt2", 5), "should");
87 Curl_dynhds_free(&hds);
88
89 /* add header exceeding max overall length */
90 Curl_dynhds_init(&hds, 128, 10);
91 fail_if(Curl_dynhds_add(&hds, "test1", 5, "123", 3), "add failed");
92 fail_unless(Curl_dynhds_add(&hds, "test2", 5, "456", 3), "should fail");
93 fail_if(Curl_dynhds_add(&hds, "t", 1, "1", 1), "add failed");
94 Curl_dynhds_reset(&hds);
95 Curl_dynhds_free(&hds);
96
97 Curl_dynhds_init(&hds, 128, 4*1024);
98 fail_if(Curl_dynhds_add(&hds, "test1", 5, "123", 3), "add failed");
99 fail_if(Curl_dynhds_add(&hds, "test1", 5, "123", 3), "add failed");
100 fail_if(Curl_dynhds_cadd(&hds, "blablabla", "thingies"), "add failed");
101 fail_if(Curl_dynhds_h1_cadd_line(&hds, "blablabla: thingies"), "add failed");
102 fail_unless(Curl_dynhds_ccount_name(&hds, "blablabla") == 2, "should");
103 fail_unless(Curl_dynhds_cremove(&hds, "blablabla") == 2, "should");
104 fail_if(Curl_dynhds_ccontains(&hds, "blablabla"), "should not");
105
106 result = Curl_dynhds_h1_cadd_line(&hds, "blablabla thingies");
107 fail_unless(result, "add should have failed");
108 if(!result) {
109 fail_unless(Curl_dynhds_ccount_name(&hds, "bLABlaBlA") == 0, "should");
110 fail_if(Curl_dynhds_cadd(&hds, "Bla-Bla", "thingies"), "add failed");
111
112 Curl_dyn_init(&dbuf, 32*1024);
113 fail_if(Curl_dynhds_h1_dprint(&hds, &dbuf), "h1 print failed");
114 if(Curl_dyn_ptr(&dbuf)) {
115 fail_if(strcmp(Curl_dyn_ptr(&dbuf),
116 "test1: 123\r\ntest1: 123\r\nBla-Bla: thingies\r\n"),
117 "h1 format differs");
118 }
119 Curl_dyn_free(&dbuf);
120 }
121
122 Curl_dynhds_free(&hds);
123 Curl_dynhds_init(&hds, 128, 4*1024);
124 /* continuation without previous header fails */
125 result = Curl_dynhds_h1_cadd_line(&hds, " indented value");
126 fail_unless(result, "add should have failed");
127
128 /* continuation with previous header must succeed */
129 fail_if(Curl_dynhds_h1_cadd_line(&hds, "ti1: val1"), "add");
130 fail_if(Curl_dynhds_h1_cadd_line(&hds, " val2"), "add indent");
131 fail_if(Curl_dynhds_h1_cadd_line(&hds, "ti2: val1"), "add");
132 fail_if(Curl_dynhds_h1_cadd_line(&hds, "\tval2"), "add indent");
133 fail_if(Curl_dynhds_h1_cadd_line(&hds, "ti3: val1"), "add");
134 fail_if(Curl_dynhds_h1_cadd_line(&hds, " val2"), "add indent");
135
136 Curl_dyn_init(&dbuf, 32*1024);
137 fail_if(Curl_dynhds_h1_dprint(&hds, &dbuf), "h1 print failed");
138 if(Curl_dyn_ptr(&dbuf)) {
139 fprintf(stderr, "indent concat: %s\n", Curl_dyn_ptr(&dbuf));
140 fail_if(strcmp(Curl_dyn_ptr(&dbuf),
141 "ti1: val1 val2\r\nti2: val1 val2\r\nti3: val1 val2\r\n"),
142 "wrong format");
143 }
144 Curl_dyn_free(&dbuf);
145
146 Curl_dynhds_free(&hds);
147
148 UNITTEST_STOP
149