1/* Copyright (c) 2013, Vsevolod Stakhov
2 * All rights reserved.
3 *
4 * Redistribution and use in source and binary forms, with or without
5 * modification, are permitted provided that the following conditions are met:
6 *       * Redistributions of source code must retain the above copyright
7 *         notice, this list of conditions and the following disclaimer.
8 *       * Redistributions in binary form must reproduce the above copyright
9 *         notice, this list of conditions and the following disclaimer in the
10 *         documentation and/or other materials provided with the distribution.
11 *
12 * THIS SOFTWARE IS PROVIDED ''AS IS'' AND ANY
13 * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
14 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
15 * DISCLAIMED. IN NO EVENT SHALL AUTHOR BE LIABLE FOR ANY
16 * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
17 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
18 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
19 * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
20 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
21 * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
22 */
23
24#include <stdio.h>
25#include <errno.h>
26#include <assert.h>
27#include "ucl.h"
28
29static void
30ud_dtor (void *ptr)
31{
32	assert (ptr == NULL);
33}
34
35static const char *
36ud_emit (void *ptr)
37{
38	return "test userdata emit";
39}
40
41int
42main (int argc, char **argv)
43{
44	ucl_object_t *obj, *cur, *ar, *ar1, *ref, *test_obj, *comments;
45	ucl_object_iter_t it;
46	const ucl_object_t *found, *it_obj, *test;
47	struct ucl_emitter_functions *fn;
48	FILE *out;
49	unsigned char *emitted;
50	const char *fname_out = NULL;
51	struct ucl_parser *parser;
52	int ret = 0;
53
54	switch (argc) {
55	case 2:
56		fname_out = argv[1];
57		break;
58	}
59
60
61	if (fname_out != NULL) {
62		out = fopen (fname_out, "w");
63		if (out == NULL) {
64			exit (-errno);
65		}
66	}
67	else {
68		out = stdout;
69	}
70
71	obj = ucl_object_typed_new (UCL_OBJECT);
72
73	/* Keys replacing */
74	cur = ucl_object_fromstring_common ("value1", 0, UCL_STRING_TRIM);
75	ucl_object_insert_key (obj, cur, "key0", 0, false);
76	cur = ucl_object_fromdouble (0.1);
77	assert (ucl_object_replace_key (obj, cur, "key0", 0, false));
78
79	/* Create some strings */
80	cur = ucl_object_fromstring_common ("  test string    ", 0, UCL_STRING_TRIM);
81	ucl_object_insert_key (obj, cur, "key1", 0, false);
82	cur = ucl_object_fromstring_common ("  test \nstring\n\r\n\b\t\f\\\"    ", 0,
83			UCL_STRING_TRIM | UCL_STRING_ESCAPE);
84	ucl_object_insert_key (obj, cur, "key2", 0, false);
85	cur = ucl_object_fromstring_common ("  test string    \n", 0, 0);
86	ucl_object_insert_key (obj, cur, "key3", 0, false);
87	/* Array of numbers */
88	ar = ucl_object_typed_new (UCL_ARRAY);
89	cur = ucl_object_fromint (10);
90	ucl_array_append (ar, cur);
91	assert (ucl_array_index_of (ar, cur) == 0);
92	cur = ucl_object_fromdouble (10.1);
93	ucl_array_append (ar, cur);
94	assert (ucl_array_index_of (ar, cur) == 1);
95	cur = ucl_object_fromdouble (9.999);
96	ucl_array_prepend (ar, cur);
97	assert (ucl_array_index_of (ar, cur) == 0);
98
99	ar1 = ucl_object_copy (ar);
100	cur = ucl_object_fromstring ("abc");
101	ucl_array_prepend (ar1, cur);
102	cur = ucl_object_fromstring ("cde");
103	ucl_array_prepend (ar1, cur);
104	cur = ucl_object_fromstring ("������"); /* UTF8 */
105	ucl_array_prepend (ar1, cur);
106	cur = ucl_object_fromstring ("������"); /* UTF8 */
107	ucl_array_prepend (ar1, cur);
108/*
109 * This is ususally broken or fragile as utf collate is far from perfect
110	cur = ucl_object_fromstring ("������");
111	ucl_array_prepend (ar1, cur);
112	cur = ucl_object_fromstring ("������"); // hello to @bapt
113	ucl_array_prepend (ar1, cur);
114*/
115	cur = ucl_object_fromstring ("����"); /* everybody likes emoji in the code */
116	ucl_array_prepend (ar1, cur);
117
118	ucl_object_array_sort (ar1, ucl_object_compare_qsort);
119
120	/* Removing from an array */
121	cur = ucl_object_fromdouble (1.0);
122	ucl_array_append (ar, cur);
123	cur = ucl_array_delete (ar, cur);
124	assert (ucl_object_todouble (cur) == 1.0);
125	ucl_object_unref (cur);
126	cur = ucl_object_fromdouble (2.0);
127	ucl_array_append (ar, cur);
128	cur = ucl_array_pop_last (ar);
129	assert (ucl_object_todouble (cur) == 2.0);
130	ucl_object_unref (cur);
131	cur = ucl_object_fromdouble (3.0);
132	ucl_array_prepend (ar, cur);
133	cur = ucl_array_pop_first (ar);
134	assert (ucl_object_todouble (cur) == 3.0);
135	ucl_object_unref (cur);
136
137	ucl_object_insert_key (obj, ar, "key4", 0, false);
138	cur = ucl_object_frombool (true);
139	/* Ref object to test refcounts */
140	ref = ucl_object_ref (cur);
141	ucl_object_insert_key (obj, cur, "key4", 0, false);
142	/* Empty strings */
143	cur = ucl_object_fromstring_common ("      ", 0, UCL_STRING_TRIM);
144	ucl_object_insert_key (obj, cur, "key5", 0, false);
145	cur = ucl_object_fromstring_common ("", 0, UCL_STRING_ESCAPE);
146	ucl_object_insert_key (obj, cur, "key6", 0, false);
147	cur = ucl_object_fromstring_common ("   \n", 0, UCL_STRING_ESCAPE);
148	ucl_object_insert_key (obj, cur, "key7", 0, false);
149	/* Numbers and booleans */
150	cur = ucl_object_fromstring_common ("1mb", 0, UCL_STRING_ESCAPE | UCL_STRING_PARSE);
151	ucl_object_insert_key (obj, cur, "key8", 0, false);
152	cur = ucl_object_fromstring_common ("3.14", 0, UCL_STRING_PARSE);
153	ucl_object_insert_key (obj, cur, "key9", 0, false);
154	cur = ucl_object_fromstring_common ("true", 0, UCL_STRING_PARSE);
155	ucl_object_insert_key (obj, cur, "key10", 0, false);
156	cur = ucl_object_fromstring_common ("  off  ", 0, UCL_STRING_PARSE | UCL_STRING_TRIM);
157	ucl_object_insert_key (obj, cur, "key11", 0, false);
158	cur = ucl_object_fromstring_common ("gslin@gslin.org", 0, UCL_STRING_PARSE_INT);
159	ucl_object_insert_key (obj, cur, "key12", 0, false);
160	cur = ucl_object_fromstring_common ("#test", 0, UCL_STRING_PARSE_INT);
161	ucl_object_insert_key (obj, cur, "key13", 0, false);
162	cur = ucl_object_frombool (true);
163	ucl_object_insert_key (obj, cur, "k=3", 0, false);
164	ucl_object_insert_key (obj, ar1, "key14", 0, false);
165	cur = ucl_object_new_userdata (ud_dtor, ud_emit, NULL);
166	ucl_object_insert_key (obj, cur, "key15", 0, false);
167
168	/* More tests for keys */
169	cur = ucl_object_fromlstring ("test", 3);
170	ucl_object_insert_key (obj, cur, "key16", 0, false);
171	test = ucl_object_lookup_any (obj, "key100", "key200", "key300", "key16", NULL);
172	assert (test == cur);
173	test = ucl_object_lookup_len (obj, "key160", 5);
174	assert (test == cur);
175	cur = ucl_object_pop_key (obj, "key16");
176	assert (test == cur);
177	test = ucl_object_pop_key (obj, "key16");
178	assert (test == NULL);
179	test = ucl_object_lookup_len (obj, "key160", 5);
180	assert (test == NULL);
181	/* Objects merging tests */
182	test_obj = ucl_object_new_full (UCL_OBJECT, 2);
183	ucl_object_insert_key (test_obj, cur, "key16", 0, true);
184	ucl_object_merge (obj, test_obj, true);
185	ucl_object_unref (test_obj);
186	/* Array merging test */
187	test_obj = ucl_object_new_full (UCL_ARRAY, 3);
188	ucl_array_append (test_obj, ucl_object_fromstring ("test"));
189	ucl_array_merge (test_obj, ar1, false);
190	ucl_object_insert_key (obj, test_obj, "key17", 0, true);
191	/* Object deletion */
192	cur = ucl_object_fromstring ("test");
193	ucl_object_insert_key (obj, cur, "key18", 0, true);
194	assert (ucl_object_delete_key (obj, "key18"));
195	assert (!ucl_object_delete_key (obj, "key18"));
196	cur = ucl_object_fromlstring ("test", 4);
197	ucl_object_insert_key (obj, cur, "key18\0\0", 7, true);
198	assert (ucl_object_lookup_len (obj, "key18\0\0", 7) == cur);
199	assert (ucl_object_lookup (obj, "key18") == NULL);
200	assert (ucl_object_lookup_len (obj, "key18\0\1", 7) == NULL);
201	assert (ucl_object_delete_keyl (obj, "key18\0\0", 7));
202
203	/* Comments */
204
205	comments = ucl_object_typed_new (UCL_OBJECT);
206	found = ucl_object_lookup (obj, "key17");
207	test = ucl_object_lookup (obj, "key16");
208	ucl_comments_add (comments, found, "# test comment");
209	assert (ucl_comments_find (comments, found) != NULL);
210	assert (ucl_comments_find (comments, test) == NULL);
211	ucl_comments_move (comments, found, test);
212	assert (ucl_comments_find (comments, found) == NULL);
213	assert (ucl_comments_find (comments, test) != NULL);
214
215	/* Array replace */
216	ar1 = ucl_object_typed_new (UCL_ARRAY);
217	cur = ucl_object_fromstring ("test");
218	cur = ucl_elt_append (cur, ucl_object_fromstring ("test1"));
219	ucl_array_append (ar1, cur);
220	test = ucl_array_replace_index (ar1, ucl_object_fromstring ("test2"), 0);
221	assert (test == cur);
222
223	/* Try to find using path */
224	/* Should exist */
225	found = ucl_object_lookup_path (obj, "key4.1");
226	assert (found != NULL && ucl_object_toint (found) == 10);
227	/* . should be ignored */
228	found = ucl_object_lookup_path (obj, ".key4.1");
229	assert (found != NULL && ucl_object_toint (found) == 10);
230	/* moar dots... */
231	found = ucl_object_lookup_path (obj, ".key4........1...");
232	assert (found != NULL && ucl_object_toint (found) == 10);
233	/* No such index */
234	found = ucl_object_lookup_path (obj, ".key4.3");
235	assert (found == NULL);
236	/* No such key */
237	found = ucl_object_lookup_path (obj, "key9..key1");
238	assert (found == NULL);
239
240	/* Test iteration */
241	it = ucl_object_iterate_new (obj);
242	it_obj = ucl_object_iterate_safe (it, true);
243	/* key0 = 0.1 */
244	assert (ucl_object_type (it_obj) == UCL_FLOAT);
245	it_obj = ucl_object_iterate_safe (it, true);
246	/* key1 = "" */
247	assert (ucl_object_type (it_obj) == UCL_STRING);
248	it_obj = ucl_object_iterate_safe (it, true);
249	/* key2 = "" */
250	assert (ucl_object_type (it_obj) == UCL_STRING);
251	it_obj = ucl_object_iterate_safe (it, true);
252	/* key3 = "" */
253	assert (ucl_object_type (it_obj) == UCL_STRING);
254	it_obj = ucl_object_iterate_safe (it, true);
255	/* key4 = ([float, int, float], boolean) */
256	ucl_object_iterate_reset (it, it_obj);
257	it_obj = ucl_object_iterate_safe (it, true);
258	assert (ucl_object_type (it_obj) == UCL_FLOAT);
259	it_obj = ucl_object_iterate_safe (it, true);
260	assert (ucl_object_type (it_obj) == UCL_INT);
261	it_obj = ucl_object_iterate_safe (it, true);
262	assert (ucl_object_type (it_obj) == UCL_FLOAT);
263	it_obj = ucl_object_iterate_safe (it, true);
264	assert (ucl_object_type (it_obj) == UCL_BOOLEAN);
265	ucl_object_iterate_free (it);
266
267	fn = ucl_object_emit_memory_funcs (&emitted);
268	assert (ucl_object_emit_full (obj, UCL_EMIT_CONFIG, fn, comments));
269	fprintf (out, "%s\n", emitted);
270	ucl_object_emit_funcs_free (fn);
271	ucl_object_unref (obj);
272	ucl_object_unref (comments);
273
274	parser = ucl_parser_new (UCL_PARSER_NO_IMPLICIT_ARRAYS);
275
276	if (ucl_parser_add_chunk_full (parser, emitted, strlen (emitted),
277			3, UCL_DUPLICATE_ERROR, UCL_PARSE_UCL)) {
278		/* Should fail due to duplicate */
279		assert (0);
280	}
281	else {
282		assert (ucl_parser_get_error (parser) != NULL);
283		ucl_parser_clear_error (parser);
284		ucl_parser_free (parser);
285		parser = ucl_parser_new (0);
286		ucl_parser_add_chunk_full (parser, emitted, strlen (emitted),
287					3, UCL_DUPLICATE_MERGE, UCL_PARSE_UCL);
288	}
289
290	assert (ucl_parser_get_column (parser) == 0);
291	assert (ucl_parser_get_linenum (parser) != 0);
292	ucl_parser_clear_error (parser);
293	assert (ucl_parser_get_error_code (parser) == 0);
294	obj = ucl_parser_get_object (parser);
295	ucl_parser_free (parser);
296	ucl_object_free (obj);
297
298	if (emitted != NULL) {
299		free (emitted);
300	}
301	fclose (out);
302
303	/* Ref should still be accessible */
304	ref->value.iv = 100500;
305	ucl_object_unref (ref);
306
307	return ret;
308}
309