FreeRDP
Loading...
Searching...
No Matches
TestAddinArgv.c
1#include <winpr/crt.h>
2#include <winpr/crypto.h>
3
4#include <freerdp/settings.h>
5
6static BOOL test_alloc(void)
7{
8 BOOL rc = FALSE;
9 int rng = 0;
10 const char* param[] = { "foo:", "bar", "bla", "rdp", NULL };
11 ADDIN_ARGV* arg1 = NULL;
12 ADDIN_ARGV* arg2 = NULL;
13 ADDIN_ARGV* arg3 = NULL;
14 ADDIN_ARGV* arg4 = NULL;
15
16 /* Test empty allocation */
17 arg1 = freerdp_addin_argv_new(0, NULL);
18 if (!arg1 || (arg1->argc != 0) || (arg1->argv))
19 goto fail;
20
21 /* Test allocation without initializing arguments of random size > 0 */
22 winpr_RAND(&rng, sizeof(rng));
23 rng = abs(rng % 8192) + 1;
24
25 arg2 = freerdp_addin_argv_new(rng, NULL);
26 if (!arg2 || (arg2->argc != rng) || (!arg2->argv))
27 goto fail;
28 for (int x = 0; x < arg2->argc; x++)
29 {
30 if (arg2->argv[x])
31 goto fail;
32 }
33
34 /* Test allocation with initializing arguments of size > 0 */
35 arg3 = freerdp_addin_argv_new(ARRAYSIZE(param) - 1, param);
36 if (!arg3 || (arg3->argc != ARRAYSIZE(param) - 1) || (!arg3->argv))
37 goto fail;
38
39 for (int x = 0; x < arg3->argc; x++)
40 {
41 if (strcmp(arg3->argv[x], param[x]) != 0)
42 goto fail;
43 }
44
45 /* Input lists with NULL elements are not allowed */
46 arg4 = freerdp_addin_argv_new(ARRAYSIZE(param), param);
47 if (arg4)
48 goto fail;
49 rc = TRUE;
50fail:
51 freerdp_addin_argv_free(arg1);
52 freerdp_addin_argv_free(arg2);
53 freerdp_addin_argv_free(arg3);
54 freerdp_addin_argv_free(arg4);
55 printf("%s: %d\n", __func__, rc);
56 return rc;
57}
58
59static BOOL test_clone(void)
60{
61 BOOL rc = FALSE;
62 const char* param[] = { "foo:", "bar", "bla", "rdp" };
63 ADDIN_ARGV* arg = NULL;
64 ADDIN_ARGV* clone = NULL;
65 ADDIN_ARGV* clone2 = NULL;
66
67 arg = freerdp_addin_argv_new(ARRAYSIZE(param), param);
68 if (!arg || (arg->argc != ARRAYSIZE(param)))
69 goto fail;
70 clone = freerdp_addin_argv_clone(arg);
71 if (!clone || (clone->argc != arg->argc))
72 goto fail;
73
74 for (int x = 0; x < arg->argc; x++)
75 {
76 if (strcmp(param[x], arg->argv[x]) != 0)
77 goto fail;
78 if (strcmp(param[x], clone->argv[x]) != 0)
79 goto fail;
80 }
81
82 clone2 = freerdp_addin_argv_clone(NULL);
83 if (clone2)
84 goto fail;
85 rc = TRUE;
86fail:
87 freerdp_addin_argv_free(arg);
88 freerdp_addin_argv_free(clone);
89 freerdp_addin_argv_free(clone2);
90 printf("%s: %d\n", __func__, rc);
91 return rc;
92}
93
94static BOOL test_add_remove(void)
95{
96 const char* args[] = { "foo", "bar", "bla", "gaga" };
97 BOOL rc = FALSE;
98 ADDIN_ARGV* arg = freerdp_addin_argv_new(0, NULL);
99
100 if (!arg || (arg->argc != 0) || arg->argv)
101 goto fail;
102 for (size_t y = 0; y < ARRAYSIZE(args); y++)
103 {
104 const char* param = args[y];
105 if (!freerdp_addin_argv_add_argument(arg, param))
106 goto fail;
107 if (arg->argc != (int)y + 1)
108 goto fail;
109 if (!arg->argv)
110 goto fail;
111 if (strcmp(arg->argv[y], param) != 0)
112 goto fail;
113 }
114
115 /* Try to remove non existing element, must not return TRUE */
116 if (freerdp_addin_argv_del_argument(arg, "foobar"))
117 goto fail;
118
119 /* Invalid parameters, must return FALSE */
120 if (freerdp_addin_argv_del_argument(NULL, "foobar"))
121 goto fail;
122
123 /* Invalid parameters, must return FALSE */
124 if (freerdp_addin_argv_del_argument(arg, NULL))
125 goto fail;
126
127 /* Remove elements one by one to test argument index move */
128 for (size_t y = 0; y < ARRAYSIZE(args); y++)
129 {
130 const char* param = args[y];
131 if (!freerdp_addin_argv_del_argument(arg, param))
132 goto fail;
133 for (size_t x = y + 1; x < ARRAYSIZE(args); x++)
134 {
135 if (strcmp(arg->argv[x - y - 1], args[x]) != 0)
136 goto fail;
137 }
138 }
139 rc = TRUE;
140fail:
141 freerdp_addin_argv_free(arg);
142 printf("%s: %d\n", __func__, rc);
143 return rc;
144}
145
146static BOOL test_set_argument(void)
147{
148 int ret = 0;
149 const char* newarg = "foobar";
150 const char* args[] = { "foo", "bar", "bla", "gaga" };
151 BOOL rc = FALSE;
152 ADDIN_ARGV* arg = NULL;
153
154 arg = freerdp_addin_argv_new(ARRAYSIZE(args), args);
155 if (!arg || (arg->argc != ARRAYSIZE(args)) || !arg->argv)
156 goto fail;
157
158 /* Check invalid parameters */
159 ret = freerdp_addin_set_argument(NULL, "foo");
160 if (ret >= 0)
161 goto fail;
162 ret = freerdp_addin_set_argument(arg, NULL);
163 if (ret >= 0)
164 goto fail;
165
166 /* Try existing parameter */
167 ret = freerdp_addin_set_argument(arg, "foo");
168 if ((ret != 1) || (arg->argc != ARRAYSIZE(args)))
169 goto fail;
170
171 /* Try new parameter */
172 ret = freerdp_addin_set_argument(arg, newarg);
173 if ((ret != 0) || (arg->argc != ARRAYSIZE(args) + 1) ||
174 (strcmp(newarg, arg->argv[ARRAYSIZE(args)]) != 0))
175 goto fail;
176
177 rc = TRUE;
178fail:
179 freerdp_addin_argv_free(arg);
180 printf("%s: %d\n", __func__, rc);
181 return rc;
182}
183
184static BOOL test_replace_argument(void)
185{
186 int ret = 0;
187 const char* newarg = "foobar";
188 const char* args[] = { "foo", "bar", "bla", "gaga" };
189 BOOL rc = FALSE;
190 ADDIN_ARGV* arg = NULL;
191
192 arg = freerdp_addin_argv_new(ARRAYSIZE(args), args);
193 if (!arg || (arg->argc != ARRAYSIZE(args)) || !arg->argv)
194 goto fail;
195
196 /* Check invalid parameters */
197 ret = freerdp_addin_replace_argument(NULL, "foo", newarg);
198 if (ret >= 0)
199 goto fail;
200 ret = freerdp_addin_replace_argument(arg, NULL, newarg);
201 if (ret >= 0)
202 goto fail;
203 ret = freerdp_addin_replace_argument(arg, "foo", NULL);
204 if (ret >= 0)
205 goto fail;
206
207 /* Try existing parameter */
208 ret = freerdp_addin_replace_argument(arg, "foo", newarg);
209 if ((ret != 1) || (arg->argc != ARRAYSIZE(args)) || (strcmp(arg->argv[0], newarg) != 0))
210 goto fail;
211
212 /* Try new parameter */
213 ret = freerdp_addin_replace_argument(arg, "lalala", newarg);
214 if ((ret != 0) || (arg->argc != ARRAYSIZE(args) + 1) ||
215 (strcmp(newarg, arg->argv[ARRAYSIZE(args)]) != 0))
216 goto fail;
217
218 rc = TRUE;
219fail:
220 freerdp_addin_argv_free(arg);
221 printf("%s: %d\n", __func__, rc);
222 return rc;
223}
224
225static BOOL test_set_argument_value(void)
226{
227 int ret = 0;
228 const char* newarg1 = "foobar";
229 const char* newarg2 = "lalala";
230 const char* fullnewarg1 = "foo:foobar";
231 const char* fullnewarg2 = "foo:lalala";
232 const char* fullnewvalue = "lalala:foobar";
233 const char* args[] = { "foo", "foo:", "bar", "bla", "gaga" };
234 BOOL rc = FALSE;
235 ADDIN_ARGV* arg = NULL;
236
237 arg = freerdp_addin_argv_new(ARRAYSIZE(args), args);
238 if (!arg || (arg->argc != ARRAYSIZE(args)) || !arg->argv)
239 goto fail;
240
241 /* Check invalid parameters */
242 ret = freerdp_addin_set_argument_value(NULL, "foo", newarg1);
243 if (ret >= 0)
244 goto fail;
245 ret = freerdp_addin_set_argument_value(arg, NULL, newarg1);
246 if (ret >= 0)
247 goto fail;
248 ret = freerdp_addin_set_argument_value(arg, "foo", NULL);
249 if (ret >= 0)
250 goto fail;
251
252 /* Try existing parameter */
253 ret = freerdp_addin_set_argument_value(arg, "foo", newarg1);
254 if ((ret != 1) || (arg->argc != ARRAYSIZE(args)) || (strcmp(arg->argv[1], fullnewarg1) != 0))
255 goto fail;
256 ret = freerdp_addin_set_argument_value(arg, "foo", newarg2);
257 if ((ret != 1) || (arg->argc != ARRAYSIZE(args)) || (strcmp(arg->argv[1], fullnewarg2) != 0))
258 goto fail;
259
260 /* Try new parameter */
261 ret = freerdp_addin_set_argument_value(arg, newarg2, newarg1);
262 if ((ret != 0) || (arg->argc != ARRAYSIZE(args) + 1) ||
263 (strcmp(fullnewvalue, arg->argv[ARRAYSIZE(args)]) != 0))
264 goto fail;
265
266 rc = TRUE;
267fail:
268 freerdp_addin_argv_free(arg);
269 printf("%s: %d\n", __func__, rc);
270 return rc;
271}
272
273static BOOL test_replace_argument_value(void)
274{
275 int ret = 0;
276 const char* newarg1 = "foobar";
277 const char* newarg2 = "lalala";
278 const char* fullnewarg1 = "foo:foobar";
279 const char* fullnewarg2 = "foo:lalala";
280 const char* fullnewvalue = "lalala:foobar";
281 const char* args[] = { "foo", "foo:", "bar", "bla", "gaga" };
282 BOOL rc = FALSE;
283 ADDIN_ARGV* arg = NULL;
284
285 arg = freerdp_addin_argv_new(ARRAYSIZE(args), args);
286 if (!arg || (arg->argc != ARRAYSIZE(args)) || !arg->argv)
287 goto fail;
288
289 /* Check invalid parameters */
290 ret = freerdp_addin_replace_argument_value(NULL, "bar", "foo", newarg1);
291 if (ret >= 0)
292 goto fail;
293 ret = freerdp_addin_replace_argument_value(arg, NULL, "foo", newarg1);
294 if (ret >= 0)
295 goto fail;
296 ret = freerdp_addin_replace_argument_value(arg, "foo", NULL, newarg1);
297 if (ret >= 0)
298 goto fail;
299 ret = freerdp_addin_replace_argument_value(arg, "bar", "foo", NULL);
300 if (ret >= 0)
301 goto fail;
302
303 /* Try existing parameter */
304 ret = freerdp_addin_replace_argument_value(arg, "bla", "foo", newarg1);
305 if ((ret != 1) || (arg->argc != ARRAYSIZE(args)) || (strcmp(arg->argv[3], fullnewarg1) != 0))
306 goto fail;
307 ret = freerdp_addin_replace_argument_value(arg, "foo", "foo", newarg2);
308 if ((ret != 1) || (arg->argc != ARRAYSIZE(args)) || (strcmp(arg->argv[0], fullnewarg2) != 0))
309 goto fail;
310
311 /* Try new parameter */
312 ret = freerdp_addin_replace_argument_value(arg, "hahaha", newarg2, newarg1);
313 if ((ret != 0) || (arg->argc != ARRAYSIZE(args) + 1) ||
314 (strcmp(fullnewvalue, arg->argv[ARRAYSIZE(args)]) != 0))
315 goto fail;
316
317 rc = TRUE;
318fail:
319 freerdp_addin_argv_free(arg);
320 printf("%s: %d\n", __func__, rc);
321 return rc;
322}
323
324int TestAddinArgv(int argc, char* argv[])
325{
326
327 WINPR_UNUSED(argc);
328 WINPR_UNUSED(argv);
329
330 if (!test_alloc())
331 return -1;
332 if (!test_clone())
333 return -1;
334 if (!test_add_remove())
335 return -1;
336 if (!test_set_argument())
337 return -1;
338 if (!test_replace_argument())
339 return -1;
340 if (!test_set_argument_value())
341 return -1;
342 if (!test_replace_argument_value())
343 return -1;
344 return 0;
345}
ADDIN_ARGV
Definition settings_types.h:367