]> CyberLeo.Net >> Repos - FreeBSD/FreeBSD.git/blob - contrib/libucl/src/ucl_util.c
Bring down 0.4.5 vendor files and other catchups with the distribution tarball.
[FreeBSD/FreeBSD.git] / contrib / libucl / src / ucl_util.c
1 /* Copyright (c) 2013, Vsevolod Stakhov
2  * Copyright (c) 2015 Allan Jude <allanjude@freebsd.org>
3  * All rights reserved.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions are met:
7  *       * Redistributions of source code must retain the above copyright
8  *         notice, this list of conditions and the following disclaimer.
9  *       * Redistributions in binary form must reproduce the above copyright
10  *         notice, this list of conditions and the following disclaimer in the
11  *         documentation and/or other materials provided with the distribution.
12  *
13  * THIS SOFTWARE IS PROVIDED ''AS IS'' AND ANY
14  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
15  * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
16  * DISCLAIMED. IN NO EVENT SHALL AUTHOR BE LIABLE FOR ANY
17  * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
18  * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
19  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
20  * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
21  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
22  * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
23  */
24
25 #include "ucl.h"
26 #include "ucl_internal.h"
27 #include "ucl_chartable.h"
28 #include "kvec.h"
29 #include <stdarg.h>
30
31 #ifndef _WIN32
32 #include <glob.h>
33 #endif
34
35 #ifdef HAVE_LIBGEN_H
36 #include <libgen.h> /* For dirname */
37 #endif
38
39 typedef kvec_t(ucl_object_t *) ucl_array_t;
40
41 #define UCL_ARRAY_GET(ar, obj) ucl_array_t *ar = \
42         (ucl_array_t *)((obj) != NULL ? (obj)->value.av : NULL)
43
44 #ifdef HAVE_OPENSSL
45 #include <openssl/err.h>
46 #include <openssl/sha.h>
47 #include <openssl/rsa.h>
48 #include <openssl/ssl.h>
49 #include <openssl/evp.h>
50 #endif
51
52 #ifdef CURL_FOUND
53 #include <curl/curl.h>
54 #endif
55 #ifdef HAVE_FETCH_H
56 #include <fetch.h>
57 #endif
58
59 #ifdef _WIN32
60 #include <windows.h>
61
62 #ifndef PROT_READ
63 #define PROT_READ       1
64 #endif
65 #ifndef PROT_WRITE
66 #define PROT_WRITE      2
67 #endif
68 #ifndef PROT_READWRITE
69 #define PROT_READWRITE  3
70 #endif
71 #ifndef MAP_SHARED
72 #define MAP_SHARED      1
73 #endif
74 #ifndef MAP_PRIVATE
75 #define MAP_PRIVATE     2
76 #endif
77 #ifndef MAP_FAILED
78 #define MAP_FAILED      ((void *) -1)
79 #endif
80
81 #ifdef _WIN32
82 #include <limits.h>
83 #define NBBY CHAR_BIT
84 #endif
85
86 static void *ucl_mmap(char *addr, size_t length, int prot, int access, int fd, off_t offset)
87 {
88         void *map = NULL;
89         HANDLE handle = INVALID_HANDLE_VALUE;
90
91         switch (prot) {
92         default:
93         case PROT_READ:
94                 {
95                         handle = CreateFileMapping((HANDLE) _get_osfhandle(fd), 0, PAGE_READONLY, 0, length, 0);
96                         if (!handle) break;
97                         map = (void *) MapViewOfFile(handle, FILE_MAP_READ, 0, 0, length);
98                         CloseHandle(handle);
99                         break;
100                 }
101         case PROT_WRITE:
102                 {
103                         handle = CreateFileMapping((HANDLE) _get_osfhandle(fd), 0, PAGE_READWRITE, 0, length, 0);
104                         if (!handle) break;
105                         map = (void *) MapViewOfFile(handle, FILE_MAP_WRITE, 0, 0, length);
106                         CloseHandle(handle);
107                         break;
108                 }
109         case PROT_READWRITE:
110                 {
111                         handle = CreateFileMapping((HANDLE) _get_osfhandle(fd), 0, PAGE_READWRITE, 0, length, 0);
112                         if (!handle) break;
113                         map = (void *) MapViewOfFile(handle, FILE_MAP_ALL_ACCESS, 0, 0, length);
114                         CloseHandle(handle);
115                         break;
116                 }
117         }
118         if (map == (void *) NULL) {
119                 return (void *) MAP_FAILED;
120         }
121         return (void *) ((char *) map + offset);
122 }
123
124 static int ucl_munmap(void *map,size_t length)
125 {
126         if (!UnmapViewOfFile(map)) {
127                 return(-1);
128         }
129         return(0);
130 }
131
132 static char* ucl_realpath(const char *path, char *resolved_path) {
133     char *p;
134     char tmp[MAX_PATH + 1];
135     strncpy(tmp, path, sizeof(tmp)-1);
136     p = tmp;
137     while(*p) {
138         if (*p == '/') *p = '\\';
139         p++;
140     }
141     return _fullpath(resolved_path, tmp, MAX_PATH);
142 }
143 #else
144 #define ucl_mmap mmap
145 #define ucl_munmap munmap
146 #define ucl_realpath realpath
147 #endif
148
149 typedef void (*ucl_object_dtor) (ucl_object_t *obj);
150 static void ucl_object_free_internal (ucl_object_t *obj, bool allow_rec,
151                 ucl_object_dtor dtor);
152 static void ucl_object_dtor_unref (ucl_object_t *obj);
153
154 static void
155 ucl_object_dtor_free (ucl_object_t *obj)
156 {
157         if (obj->trash_stack[UCL_TRASH_KEY] != NULL) {
158                 UCL_FREE (obj->hh.keylen, obj->trash_stack[UCL_TRASH_KEY]);
159         }
160         if (obj->trash_stack[UCL_TRASH_VALUE] != NULL) {
161                 UCL_FREE (obj->len, obj->trash_stack[UCL_TRASH_VALUE]);
162         }
163         /* Do not free ephemeral objects */
164         if ((obj->flags & UCL_OBJECT_EPHEMERAL) == 0) {
165                 if (obj->type != UCL_USERDATA) {
166                         UCL_FREE (sizeof (ucl_object_t), obj);
167                 }
168                 else {
169                         struct ucl_object_userdata *ud = (struct ucl_object_userdata *)obj;
170                         if (ud->dtor) {
171                                 ud->dtor (obj->value.ud);
172                         }
173                         UCL_FREE (sizeof (*ud), obj);
174                 }
175         }
176 }
177
178 /*
179  * This is a helper function that performs exactly the same as
180  * `ucl_object_unref` but it doesn't iterate over elements allowing
181  * to use it for individual elements of arrays and multiple values
182  */
183 static void
184 ucl_object_dtor_unref_single (ucl_object_t *obj)
185 {
186         if (obj != NULL) {
187 #ifdef HAVE_ATOMIC_BUILTINS
188                 unsigned int rc = __sync_sub_and_fetch (&obj->ref, 1);
189                 if (rc == 0) {
190 #else
191                 if (--obj->ref == 0) {
192 #endif
193                         ucl_object_free_internal (obj, false, ucl_object_dtor_unref);
194                 }
195         }
196 }
197
198 static void
199 ucl_object_dtor_unref (ucl_object_t *obj)
200 {
201         if (obj->ref == 0) {
202                 ucl_object_dtor_free (obj);
203         }
204         else {
205                 /* This may cause dtor unref being called one more time */
206                 ucl_object_dtor_unref_single (obj);
207         }
208 }
209
210 static void
211 ucl_object_free_internal (ucl_object_t *obj, bool allow_rec, ucl_object_dtor dtor)
212 {
213         ucl_object_t *tmp, *sub;
214
215         while (obj != NULL) {
216                 if (obj->type == UCL_ARRAY) {
217                         UCL_ARRAY_GET (vec, obj);
218                         unsigned int i;
219
220                         if (vec != NULL) {
221                                 for (i = 0; i < vec->n; i ++) {
222                                         sub = kv_A (*vec, i);
223                                         if (sub != NULL) {
224                                                 tmp = sub;
225                                                 while (sub) {
226                                                         tmp = sub->next;
227                                                         dtor (sub);
228                                                         sub = tmp;
229                                                 }
230                                         }
231                                 }
232                                 kv_destroy (*vec);
233                                 UCL_FREE (sizeof (*vec), vec);
234                         }
235                         obj->value.av = NULL;
236                 }
237                 else if (obj->type == UCL_OBJECT) {
238                         if (obj->value.ov != NULL) {
239                                 ucl_hash_destroy (obj->value.ov, (ucl_hash_free_func *)dtor);
240                         }
241                         obj->value.ov = NULL;
242                 }
243                 tmp = obj->next;
244                 dtor (obj);
245                 obj = tmp;
246
247                 if (!allow_rec) {
248                         break;
249                 }
250         }
251 }
252
253 void
254 ucl_object_free (ucl_object_t *obj)
255 {
256         ucl_object_free_internal (obj, true, ucl_object_dtor_free);
257 }
258
259 size_t
260 ucl_unescape_json_string (char *str, size_t len)
261 {
262         char *t = str, *h = str;
263         int i, uval;
264
265         if (len <= 1) {
266                 return len;
267         }
268         /* t is target (tortoise), h is source (hare) */
269
270         while (len) {
271                 if (*h == '\\') {
272                         h ++;
273
274                         if (len == 1) {
275                                 /*
276                                  * If \ is last, then do not try to go further
277                                  * Issue: #74
278                                  */
279                                 len --;
280                                 *t++ = '\\';
281                                 continue;
282                         }
283
284                         switch (*h) {
285                         case 'n':
286                                 *t++ = '\n';
287                                 break;
288                         case 'r':
289                                 *t++ = '\r';
290                                 break;
291                         case 'b':
292                                 *t++ = '\b';
293                                 break;
294                         case 't':
295                                 *t++ = '\t';
296                                 break;
297                         case 'f':
298                                 *t++ = '\f';
299                                 break;
300                         case '\\':
301                                 *t++ = '\\';
302                                 break;
303                         case '"':
304                                 *t++ = '"';
305                                 break;
306                         case 'u':
307                                 /* Unicode escape */
308                                 uval = 0;
309                                 if (len > 3) {
310                                         for (i = 0; i < 4; i++) {
311                                                 uval <<= 4;
312                                                 if (isdigit (h[i])) {
313                                                         uval += h[i] - '0';
314                                                 }
315                                                 else if (h[i] >= 'a' && h[i] <= 'f') {
316                                                         uval += h[i] - 'a' + 10;
317                                                 }
318                                                 else if (h[i] >= 'A' && h[i] <= 'F') {
319                                                         uval += h[i] - 'A' + 10;
320                                                 }
321                                                 else {
322                                                         break;
323                                                 }
324                                         }
325                                         h += 3;
326                                         len -= 3;
327                                         /* Encode */
328                                         if(uval < 0x80) {
329                                                 t[0] = (char)uval;
330                                                 t ++;
331                                         }
332                                         else if(uval < 0x800) {
333                                                 t[0] = 0xC0 + ((uval & 0x7C0) >> 6);
334                                                 t[1] = 0x80 + ((uval & 0x03F));
335                                                 t += 2;
336                                         }
337                                         else if(uval < 0x10000) {
338                                                 t[0] = 0xE0 + ((uval & 0xF000) >> 12);
339                                                 t[1] = 0x80 + ((uval & 0x0FC0) >> 6);
340                                                 t[2] = 0x80 + ((uval & 0x003F));
341                                                 t += 3;
342                                         }
343                                         else if(uval <= 0x10FFFF) {
344                                                 t[0] = 0xF0 + ((uval & 0x1C0000) >> 18);
345                                                 t[1] = 0x80 + ((uval & 0x03F000) >> 12);
346                                                 t[2] = 0x80 + ((uval & 0x000FC0) >> 6);
347                                                 t[3] = 0x80 + ((uval & 0x00003F));
348                                                 t += 4;
349                                         }
350                                         else {
351                                                 *t++ = '?';
352                                         }
353                                 }
354                                 else {
355                                         *t++ = 'u';
356                                 }
357                                 break;
358                         default:
359                                 *t++ = *h;
360                                 break;
361                         }
362                         h ++;
363                         len --;
364                 }
365                 else {
366                         *t++ = *h++;
367                 }
368
369                 if (len > 0) {
370                         len --;
371                 }
372         }
373         *t = '\0';
374
375         return (t - str);
376 }
377
378 char *
379 ucl_copy_key_trash (const ucl_object_t *obj)
380 {
381         ucl_object_t *deconst;
382
383         if (obj == NULL) {
384                 return NULL;
385         }
386         if (obj->trash_stack[UCL_TRASH_KEY] == NULL && obj->key != NULL) {
387                 deconst = __DECONST (ucl_object_t *, obj);
388                 deconst->trash_stack[UCL_TRASH_KEY] = malloc (obj->keylen + 1);
389                 if (deconst->trash_stack[UCL_TRASH_KEY] != NULL) {
390                         memcpy (deconst->trash_stack[UCL_TRASH_KEY], obj->key, obj->keylen);
391                         deconst->trash_stack[UCL_TRASH_KEY][obj->keylen] = '\0';
392                 }
393                 deconst->key = obj->trash_stack[UCL_TRASH_KEY];
394                 deconst->flags |= UCL_OBJECT_ALLOCATED_KEY;
395         }
396
397         return obj->trash_stack[UCL_TRASH_KEY];
398 }
399
400 char *
401 ucl_copy_value_trash (const ucl_object_t *obj)
402 {
403         ucl_object_t *deconst;
404
405         if (obj == NULL) {
406                 return NULL;
407         }
408         if (obj->trash_stack[UCL_TRASH_VALUE] == NULL) {
409                 deconst = __DECONST (ucl_object_t *, obj);
410                 if (obj->type == UCL_STRING) {
411
412                         /* Special case for strings */
413                         if (obj->flags & UCL_OBJECT_BINARY) {
414                                 deconst->trash_stack[UCL_TRASH_VALUE] = malloc (obj->len);
415                                 if (deconst->trash_stack[UCL_TRASH_VALUE] != NULL) {
416                                         memcpy (deconst->trash_stack[UCL_TRASH_VALUE],
417                                                         obj->value.sv,
418                                                         obj->len);
419                                         deconst->value.sv = obj->trash_stack[UCL_TRASH_VALUE];
420                                 }
421                         }
422                         else {
423                                 deconst->trash_stack[UCL_TRASH_VALUE] = malloc (obj->len + 1);
424                                 if (deconst->trash_stack[UCL_TRASH_VALUE] != NULL) {
425                                         memcpy (deconst->trash_stack[UCL_TRASH_VALUE],
426                                                         obj->value.sv,
427                                                         obj->len);
428                                         deconst->trash_stack[UCL_TRASH_VALUE][obj->len] = '\0';
429                                         deconst->value.sv = obj->trash_stack[UCL_TRASH_VALUE];
430                                 }
431                         }
432                 }
433                 else {
434                         /* Just emit value in json notation */
435                         deconst->trash_stack[UCL_TRASH_VALUE] = ucl_object_emit_single_json (obj);
436                         deconst->len = strlen (obj->trash_stack[UCL_TRASH_VALUE]);
437                 }
438                 deconst->flags |= UCL_OBJECT_ALLOCATED_VALUE;
439         }
440         
441         return obj->trash_stack[UCL_TRASH_VALUE];
442 }
443
444 ucl_object_t*
445 ucl_parser_get_object (struct ucl_parser *parser)
446 {
447         if (parser->state != UCL_STATE_ERROR && parser->top_obj != NULL) {
448                 return ucl_object_ref (parser->top_obj);
449         }
450
451         return NULL;
452 }
453
454 void
455 ucl_parser_free (struct ucl_parser *parser)
456 {
457         struct ucl_stack *stack, *stmp;
458         struct ucl_macro *macro, *mtmp;
459         struct ucl_chunk *chunk, *ctmp;
460         struct ucl_pubkey *key, *ktmp;
461         struct ucl_variable *var, *vtmp;
462         ucl_object_t *tr, *trtmp;
463
464         if (parser == NULL) {
465                 return;
466         }
467
468         if (parser->top_obj != NULL) {
469                 ucl_object_unref (parser->top_obj);
470         }
471
472         if (parser->includepaths != NULL) {
473                 ucl_object_unref (parser->includepaths);
474         }
475
476         LL_FOREACH_SAFE (parser->stack, stack, stmp) {
477                 free (stack);
478         }
479         HASH_ITER (hh, parser->macroes, macro, mtmp) {
480                 free (macro->name);
481                 HASH_DEL (parser->macroes, macro);
482                 UCL_FREE (sizeof (struct ucl_macro), macro);
483         }
484         LL_FOREACH_SAFE (parser->chunks, chunk, ctmp) {
485                 UCL_FREE (sizeof (struct ucl_chunk), chunk);
486         }
487         LL_FOREACH_SAFE (parser->keys, key, ktmp) {
488                 UCL_FREE (sizeof (struct ucl_pubkey), key);
489         }
490         LL_FOREACH_SAFE (parser->variables, var, vtmp) {
491                 free (var->value);
492                 free (var->var);
493                 UCL_FREE (sizeof (struct ucl_variable), var);
494         }
495         LL_FOREACH_SAFE (parser->trash_objs, tr, trtmp) {
496                 ucl_object_free_internal (tr, false, ucl_object_dtor_free);
497         }
498
499         if (parser->err != NULL) {
500                 utstring_free (parser->err);
501         }
502
503         if (parser->cur_file) {
504                 free (parser->cur_file);
505         }
506
507         UCL_FREE (sizeof (struct ucl_parser), parser);
508 }
509
510 const char *
511 ucl_parser_get_error(struct ucl_parser *parser)
512 {
513         if (parser == NULL) {
514                 return NULL;
515         }
516
517         if (parser->err == NULL) {
518                 return NULL;
519         }
520
521         return utstring_body (parser->err);
522 }
523
524 int
525 ucl_parser_get_error_code(struct ucl_parser *parser)
526 {
527         if (parser == NULL) {
528                 return 0;
529         }
530
531         return parser->err_code;
532 }
533
534 unsigned
535 ucl_parser_get_column(struct ucl_parser *parser)
536 {
537         if (parser == NULL || parser->chunks == NULL) {
538                 return 0;
539         }
540
541         return parser->chunks->column;
542 }
543
544 unsigned
545 ucl_parser_get_linenum(struct ucl_parser *parser)
546 {
547         if (parser == NULL || parser->chunks == NULL) {
548                 return 0;
549         }
550
551         return parser->chunks->line;
552 }
553
554 void
555 ucl_parser_clear_error(struct ucl_parser *parser)
556 {
557         if (parser != NULL && parser->err != NULL) {
558                 utstring_free(parser->err);
559                 parser->err = NULL;
560                 parser->err_code = 0;
561         }
562 }
563
564 bool
565 ucl_pubkey_add (struct ucl_parser *parser, const unsigned char *key, size_t len)
566 {
567 #ifndef HAVE_OPENSSL
568         ucl_create_err (&parser->err, "cannot check signatures without openssl");
569         return false;
570 #else
571 # if (OPENSSL_VERSION_NUMBER < 0x10000000L)
572         ucl_create_err (&parser->err, "cannot check signatures, openssl version is unsupported");
573         return EXIT_FAILURE;
574 # else
575         struct ucl_pubkey *nkey;
576         BIO *mem;
577
578         mem = BIO_new_mem_buf ((void *)key, len);
579         nkey = UCL_ALLOC (sizeof (struct ucl_pubkey));
580         if (nkey == NULL) {
581                 ucl_create_err (&parser->err, "cannot allocate memory for key");
582                 return false;
583         }
584         nkey->key = PEM_read_bio_PUBKEY (mem, &nkey->key, NULL, NULL);
585         BIO_free (mem);
586         if (nkey->key == NULL) {
587                 UCL_FREE (sizeof (struct ucl_pubkey), nkey);
588                 ucl_create_err (&parser->err, "%s",
589                                 ERR_error_string (ERR_get_error (), NULL));
590                 return false;
591         }
592         LL_PREPEND (parser->keys, nkey);
593 # endif
594 #endif
595         return true;
596 }
597
598 #ifdef CURL_FOUND
599 struct ucl_curl_cbdata {
600         unsigned char *buf;
601         size_t buflen;
602 };
603
604 static size_t
605 ucl_curl_write_callback (void* contents, size_t size, size_t nmemb, void* ud)
606 {
607         struct ucl_curl_cbdata *cbdata = ud;
608         size_t realsize = size * nmemb;
609
610         cbdata->buf = realloc (cbdata->buf, cbdata->buflen + realsize + 1);
611         if (cbdata->buf == NULL) {
612                 return 0;
613         }
614
615         memcpy (&(cbdata->buf[cbdata->buflen]), contents, realsize);
616         cbdata->buflen += realsize;
617         cbdata->buf[cbdata->buflen] = 0;
618
619         return realsize;
620 }
621 #endif
622
623 /**
624  * Fetch a url and save results to the memory buffer
625  * @param url url to fetch
626  * @param len length of url
627  * @param buf target buffer
628  * @param buflen target length
629  * @return
630  */
631 static bool
632 ucl_fetch_url (const unsigned char *url, unsigned char **buf, size_t *buflen,
633                 UT_string **err, bool must_exist)
634 {
635
636 #ifdef HAVE_FETCH_H
637         struct url *fetch_url;
638         struct url_stat us;
639         FILE *in;
640
641         fetch_url = fetchParseURL (url);
642         if (fetch_url == NULL) {
643                 ucl_create_err (err, "invalid URL %s: %s",
644                                 url, strerror (errno));
645                 return false;
646         }
647         if ((in = fetchXGet (fetch_url, &us, "")) == NULL) {
648                 if (!must_exist) {
649                         ucl_create_err (err, "cannot fetch URL %s: %s",
650                                 url, strerror (errno));
651                 }
652                 fetchFreeURL (fetch_url);
653                 return false;
654         }
655
656         *buflen = us.size;
657         *buf = malloc (*buflen);
658         if (*buf == NULL) {
659                 ucl_create_err (err, "cannot allocate buffer for URL %s: %s",
660                                 url, strerror (errno));
661                 fclose (in);
662                 fetchFreeURL (fetch_url);
663                 return false;
664         }
665
666         if (fread (*buf, *buflen, 1, in) != 1) {
667                 ucl_create_err (err, "cannot read URL %s: %s",
668                                 url, strerror (errno));
669                 fclose (in);
670                 fetchFreeURL (fetch_url);
671                 return false;
672         }
673
674         fetchFreeURL (fetch_url);
675         return true;
676 #elif defined(CURL_FOUND)
677         CURL *curl;
678         int r;
679         struct ucl_curl_cbdata cbdata;
680
681         curl = curl_easy_init ();
682         if (curl == NULL) {
683                 ucl_create_err (err, "CURL interface is broken");
684                 return false;
685         }
686         if ((r = curl_easy_setopt (curl, CURLOPT_URL, url)) != CURLE_OK) {
687                 ucl_create_err (err, "invalid URL %s: %s",
688                                 url, curl_easy_strerror (r));
689                 curl_easy_cleanup (curl);
690                 return false;
691         }
692         curl_easy_setopt (curl, CURLOPT_WRITEFUNCTION, ucl_curl_write_callback);
693         cbdata.buf = *buf;
694         cbdata.buflen = *buflen;
695         curl_easy_setopt (curl, CURLOPT_WRITEDATA, &cbdata);
696
697         if ((r = curl_easy_perform (curl)) != CURLE_OK) {
698                 if (!must_exist) {
699                         ucl_create_err (err, "error fetching URL %s: %s",
700                                 url, curl_easy_strerror (r));
701                 }
702                 curl_easy_cleanup (curl);
703                 if (cbdata.buf) {
704                         free (cbdata.buf);
705                 }
706                 return false;
707         }
708         *buf = cbdata.buf;
709         *buflen = cbdata.buflen;
710
711         return true;
712 #else
713         ucl_create_err (err, "URL support is disabled");
714         return false;
715 #endif
716 }
717
718 /**
719  * Fetch a file and save results to the memory buffer
720  * @param filename filename to fetch
721  * @param len length of filename
722  * @param buf target buffer
723  * @param buflen target length
724  * @return
725  */
726 static bool
727 ucl_fetch_file (const unsigned char *filename, unsigned char **buf, size_t *buflen,
728                 UT_string **err, bool must_exist)
729 {
730         int fd;
731         struct stat st;
732
733         if (stat (filename, &st) == -1 || !S_ISREG (st.st_mode)) {
734                 if (must_exist) {
735                         ucl_create_err (err, "cannot stat file %s: %s",
736                                         filename, strerror (errno));
737                 }
738                 return false;
739         }
740         if (st.st_size == 0) {
741                 /* Do not map empty files */
742                 *buf = "";
743                 *buflen = 0;
744         }
745         else {
746                 if ((fd = open (filename, O_RDONLY)) == -1) {
747                         ucl_create_err (err, "cannot open file %s: %s",
748                                         filename, strerror (errno));
749                         return false;
750                 }
751                 if ((*buf = ucl_mmap (NULL, st.st_size, PROT_READ, MAP_SHARED, fd, 0)) == MAP_FAILED) {
752                         close (fd);
753                         ucl_create_err (err, "cannot mmap file %s: %s",
754                                         filename, strerror (errno));
755                         return false;
756                 }
757                 *buflen = st.st_size;
758                 close (fd);
759         }
760
761         return true;
762 }
763
764
765 #if (defined(HAVE_OPENSSL) && OPENSSL_VERSION_NUMBER >= 0x10000000L)
766 static inline bool
767 ucl_sig_check (const unsigned char *data, size_t datalen,
768                 const unsigned char *sig, size_t siglen, struct ucl_parser *parser)
769 {
770         struct ucl_pubkey *key;
771         char dig[EVP_MAX_MD_SIZE];
772         unsigned int diglen;
773         EVP_PKEY_CTX *key_ctx;
774         EVP_MD_CTX *sign_ctx = NULL;
775
776         sign_ctx = EVP_MD_CTX_create ();
777
778         LL_FOREACH (parser->keys, key) {
779                 key_ctx = EVP_PKEY_CTX_new (key->key, NULL);
780                 if (key_ctx != NULL) {
781                         if (EVP_PKEY_verify_init (key_ctx) <= 0) {
782                                 EVP_PKEY_CTX_free (key_ctx);
783                                 continue;
784                         }
785                         if (EVP_PKEY_CTX_set_rsa_padding (key_ctx, RSA_PKCS1_PADDING) <= 0) {
786                                 EVP_PKEY_CTX_free (key_ctx);
787                                 continue;
788                         }
789                         if (EVP_PKEY_CTX_set_signature_md (key_ctx, EVP_sha256 ()) <= 0) {
790                                 EVP_PKEY_CTX_free (key_ctx);
791                                 continue;
792                         }
793                         EVP_DigestInit (sign_ctx, EVP_sha256 ());
794                         EVP_DigestUpdate (sign_ctx, data, datalen);
795                         EVP_DigestFinal (sign_ctx, dig, &diglen);
796
797                         if (EVP_PKEY_verify (key_ctx, sig, siglen, dig, diglen) == 1) {
798                                 EVP_MD_CTX_destroy (sign_ctx);
799                                 EVP_PKEY_CTX_free (key_ctx);
800                                 return true;
801                         }
802
803                         EVP_PKEY_CTX_free (key_ctx);
804                 }
805         }
806
807         EVP_MD_CTX_destroy (sign_ctx);
808
809         return false;
810 }
811 #endif
812
813 struct ucl_include_params {
814         bool check_signature;
815         bool must_exist;
816         bool use_glob;
817         bool use_prefix;
818         bool soft_fail;
819         bool allow_glob;
820         unsigned priority;
821         enum ucl_duplicate_strategy strat;
822         enum ucl_parse_type parse_type;
823         const char *prefix;
824         const char *target;
825 };
826
827 /**
828  * Include an url to configuration
829  * @param data
830  * @param len
831  * @param parser
832  * @param err
833  * @return
834  */
835 static bool
836 ucl_include_url (const unsigned char *data, size_t len,
837                 struct ucl_parser *parser,
838                 struct ucl_include_params *params)
839 {
840
841         bool res;
842         unsigned char *buf = NULL;
843         size_t buflen = 0;
844         struct ucl_chunk *chunk;
845         char urlbuf[PATH_MAX];
846         int prev_state;
847
848         snprintf (urlbuf, sizeof (urlbuf), "%.*s", (int)len, data);
849
850         if (!ucl_fetch_url (urlbuf, &buf, &buflen, &parser->err, params->must_exist)) {
851                 return (!params->must_exist || false);
852         }
853
854         if (params->check_signature) {
855 #if (defined(HAVE_OPENSSL) && OPENSSL_VERSION_NUMBER >= 0x10000000L)
856                 unsigned char *sigbuf = NULL;
857                 size_t siglen = 0;
858                 /* We need to check signature first */
859                 snprintf (urlbuf, sizeof (urlbuf), "%.*s.sig", (int)len, data);
860                 if (!ucl_fetch_url (urlbuf, &sigbuf, &siglen, &parser->err, true)) {
861                         return false;
862                 }
863                 if (!ucl_sig_check (buf, buflen, sigbuf, siglen, parser)) {
864                         ucl_create_err (&parser->err, "cannot verify url %s: %s",
865                                                         urlbuf,
866                                                         ERR_error_string (ERR_get_error (), NULL));
867                         if (siglen > 0) {
868                                 ucl_munmap (sigbuf, siglen);
869                         }
870                         return false;
871                 }
872                 if (siglen > 0) {
873                         ucl_munmap (sigbuf, siglen);
874                 }
875 #endif
876         }
877
878         prev_state = parser->state;
879         parser->state = UCL_STATE_INIT;
880
881         res = ucl_parser_add_chunk_full (parser, buf, buflen, params->priority,
882                         params->strat, params->parse_type);
883         if (res == true) {
884                 /* Remove chunk from the stack */
885                 chunk = parser->chunks;
886                 if (chunk != NULL) {
887                         parser->chunks = chunk->next;
888                         UCL_FREE (sizeof (struct ucl_chunk), chunk);
889                 }
890         }
891
892         parser->state = prev_state;
893         free (buf);
894
895         return res;
896 }
897
898 /**
899  * Include a single file to the parser
900  * @param data
901  * @param len
902  * @param parser
903  * @param check_signature
904  * @param must_exist
905  * @param allow_glob
906  * @param priority
907  * @return
908  */
909 static bool
910 ucl_include_file_single (const unsigned char *data, size_t len,
911                 struct ucl_parser *parser, struct ucl_include_params *params)
912 {
913         bool res;
914         struct ucl_chunk *chunk;
915         unsigned char *buf = NULL;
916         char *old_curfile, *ext;
917         size_t buflen = 0;
918         char filebuf[PATH_MAX], realbuf[PATH_MAX];
919         int prev_state;
920         struct ucl_variable *cur_var, *tmp_var, *old_curdir = NULL,
921                         *old_filename = NULL;
922         ucl_object_t *nest_obj = NULL, *old_obj = NULL, *new_obj = NULL;
923         ucl_hash_t *container = NULL;
924         struct ucl_stack *st = NULL;
925
926         snprintf (filebuf, sizeof (filebuf), "%.*s", (int)len, data);
927         if (ucl_realpath (filebuf, realbuf) == NULL) {
928                 if (params->soft_fail) {
929                         return false;
930                 }
931                 if (!params->must_exist) {
932                         return true;
933                 }
934                 ucl_create_err (&parser->err, "cannot open file %s: %s",
935                                                                         filebuf,
936                                                                         strerror (errno));
937                 return false;
938         }
939
940         if (parser->cur_file && strcmp (realbuf, parser->cur_file) == 0) {
941                 /* We are likely including the file itself */
942                 if (params->soft_fail) {
943                         return false;
944                 }
945
946                 ucl_create_err (&parser->err, "trying to include the file %s from itself",
947                                 realbuf);
948                 return false;
949         }
950
951         if (!ucl_fetch_file (realbuf, &buf, &buflen, &parser->err, params->must_exist)) {
952                 if (params->soft_fail) {
953                         return false;
954                 }
955                 return (!params->must_exist || false);
956         }
957
958         if (params->check_signature) {
959 #if (defined(HAVE_OPENSSL) && OPENSSL_VERSION_NUMBER >= 0x10000000L)
960                 unsigned char *sigbuf = NULL;
961                 size_t siglen = 0;
962                 /* We need to check signature first */
963                 snprintf (filebuf, sizeof (filebuf), "%s.sig", realbuf);
964                 if (!ucl_fetch_file (filebuf, &sigbuf, &siglen, &parser->err, true)) {
965                         return false;
966                 }
967                 if (!ucl_sig_check (buf, buflen, sigbuf, siglen, parser)) {
968                         ucl_create_err (&parser->err, "cannot verify file %s: %s",
969                                                         filebuf,
970                                                         ERR_error_string (ERR_get_error (), NULL));
971                         if (siglen > 0) {
972                                 ucl_munmap (sigbuf, siglen);
973                         }
974                         return false;
975                 }
976                 if (siglen > 0) {
977                         ucl_munmap (sigbuf, siglen);
978                 }
979 #endif
980         }
981
982         old_curfile = parser->cur_file;
983         parser->cur_file = strdup (realbuf);
984
985         /* Store old file vars */
986         DL_FOREACH_SAFE (parser->variables, cur_var, tmp_var) {
987                 if (strcmp (cur_var->var, "CURDIR") == 0) {
988                         old_curdir = cur_var;
989                         DL_DELETE (parser->variables, cur_var);
990                 }
991                 else if (strcmp (cur_var->var, "FILENAME") == 0) {
992                         old_filename = cur_var;
993                         DL_DELETE (parser->variables, cur_var);
994                 }
995         }
996
997         ucl_parser_set_filevars (parser, realbuf, false);
998
999         prev_state = parser->state;
1000         parser->state = UCL_STATE_INIT;
1001
1002         if (params->use_prefix && params->prefix == NULL) {
1003                 /* Auto generate a key name based on the included filename */
1004                 params->prefix = basename (realbuf);
1005                 ext = strrchr (params->prefix, '.');
1006                 if (ext != NULL && (strcmp (ext, ".conf") == 0 || strcmp (ext, ".ucl") == 0)) {
1007                         /* Strip off .conf or .ucl */
1008                         *ext = '\0';
1009                 }
1010         }
1011         if (params->prefix != NULL) {
1012                 /* This is a prefixed include */
1013                 container = parser->stack->obj->value.ov;
1014
1015                 old_obj = __DECONST (ucl_object_t *, ucl_hash_search (container,
1016                                 params->prefix, strlen (params->prefix)));
1017
1018                 if (strcasecmp (params->target, "array") == 0 && old_obj == NULL) {
1019                         /* Create an array with key: prefix */
1020                         old_obj = ucl_object_new_full (UCL_ARRAY, params->priority);
1021                         old_obj->key = params->prefix;
1022                         old_obj->keylen = strlen (params->prefix);
1023                         ucl_copy_key_trash(old_obj);
1024                         old_obj->prev = old_obj;
1025                         old_obj->next = NULL;
1026
1027                         container = ucl_hash_insert_object (container, old_obj,
1028                                         parser->flags & UCL_PARSER_KEY_LOWERCASE);
1029                         parser->stack->obj->len ++;
1030
1031                         nest_obj = ucl_object_new_full (UCL_OBJECT, params->priority);
1032                         nest_obj->prev = nest_obj;
1033                         nest_obj->next = NULL;
1034
1035                         ucl_array_append (old_obj, nest_obj);
1036                 }
1037                 else if (old_obj == NULL) {
1038                         /* Create an object with key: prefix */
1039                         nest_obj = ucl_object_new_full (UCL_OBJECT, params->priority);
1040                         nest_obj->key = params->prefix;
1041                         nest_obj->keylen = strlen (params->prefix);
1042                         ucl_copy_key_trash(nest_obj);
1043                         nest_obj->prev = nest_obj;
1044                         nest_obj->next = NULL;
1045
1046                         container = ucl_hash_insert_object (container, nest_obj,
1047                                         parser->flags & UCL_PARSER_KEY_LOWERCASE);
1048                         parser->stack->obj->len ++;
1049                 }
1050                 else if (strcasecmp (params->target, "array") == 0 ||
1051                                 ucl_object_type(old_obj) == UCL_ARRAY) {
1052                         if (ucl_object_type(old_obj) == UCL_ARRAY) {
1053                                 /* Append to the existing array */
1054                                 nest_obj = ucl_object_new_full (UCL_OBJECT, params->priority);
1055                                 nest_obj->prev = nest_obj;
1056                                 nest_obj->next = NULL;
1057
1058                                 ucl_array_append (old_obj, nest_obj);
1059                         }
1060                         else {
1061                                 /* Convert the object to an array */
1062                                 new_obj = ucl_object_typed_new (UCL_ARRAY);
1063                                 new_obj->key = old_obj->key;
1064                                 new_obj->keylen = old_obj->keylen;
1065                                 new_obj->flags |= UCL_OBJECT_MULTIVALUE;
1066                                 new_obj->prev = new_obj;
1067                                 new_obj->next = NULL;
1068
1069                                 nest_obj = ucl_object_new_full (UCL_OBJECT, params->priority);
1070                                 nest_obj->prev = nest_obj;
1071                                 nest_obj->next = NULL;
1072
1073                                 ucl_array_append (new_obj, old_obj);
1074                                 ucl_array_append (new_obj, nest_obj);
1075                                 ucl_hash_replace (container, old_obj, new_obj);
1076                         }
1077                 }
1078                 else {
1079                         if (ucl_object_type (old_obj) == UCL_OBJECT) {
1080                                 /* Append to existing Object*/
1081                                 nest_obj = old_obj;
1082                         }
1083                         else {
1084                                 /* The key is not an object */
1085                                 ucl_create_err (&parser->err,
1086                                                 "Conflicting type for key: %s",
1087                                                 params->prefix);
1088                                 return false;
1089                         }
1090                 }
1091
1092                  /* Put all of the content of the include inside that object */
1093                 parser->stack->obj->value.ov = container;
1094
1095                 if (nest_obj != NULL) {
1096                         st = UCL_ALLOC (sizeof (struct ucl_stack));
1097                         if (st == NULL) {
1098                                 ucl_create_err (&parser->err, "cannot allocate memory for an object");
1099                                 ucl_object_unref (nest_obj);
1100                                 return NULL;
1101                         }
1102                         st->obj = nest_obj;
1103                         st->level = parser->stack->level;
1104                         LL_PREPEND (parser->stack, st);
1105                         parser->cur_obj = nest_obj;
1106                 }
1107         }
1108
1109         res = ucl_parser_add_chunk_full (parser, buf, buflen, params->priority,
1110                         params->strat, params->parse_type);
1111         if (!res && !params->must_exist) {
1112                 /* Free error */
1113                 utstring_free (parser->err);
1114                 parser->err = NULL;
1115                 parser->state = UCL_STATE_AFTER_VALUE;
1116         }
1117
1118         /* Stop nesting the include, take 1 level off the stack */
1119         if (params->prefix != NULL && nest_obj != NULL) {
1120                 parser->stack = st->next;
1121                 UCL_FREE (sizeof (struct ucl_stack), st);
1122         }
1123
1124         /* Remove chunk from the stack */
1125         chunk = parser->chunks;
1126         if (chunk != NULL) {
1127                 parser->chunks = chunk->next;
1128                 UCL_FREE (sizeof (struct ucl_chunk), chunk);
1129                 parser->recursion --;
1130         }
1131
1132         /* Restore old file vars */
1133         if (parser->cur_file) {
1134                 free (parser->cur_file);
1135         }
1136
1137         parser->cur_file = old_curfile;
1138         DL_FOREACH_SAFE (parser->variables, cur_var, tmp_var) {
1139                 if (strcmp (cur_var->var, "CURDIR") == 0 && old_curdir) {
1140                         DL_DELETE (parser->variables, cur_var);
1141                         free (cur_var->var);
1142                         free (cur_var->value);
1143                         UCL_FREE (sizeof (struct ucl_variable), cur_var);
1144                 }
1145                 else if (strcmp (cur_var->var, "FILENAME") == 0 && old_filename) {
1146                         DL_DELETE (parser->variables, cur_var);
1147                         free (cur_var->var);
1148                         free (cur_var->value);
1149                         UCL_FREE (sizeof (struct ucl_variable), cur_var);
1150                 }
1151         }
1152         if (old_filename) {
1153                 DL_APPEND (parser->variables, old_filename);
1154         }
1155         if (old_curdir) {
1156                 DL_APPEND (parser->variables, old_curdir);
1157         }
1158
1159         parser->state = prev_state;
1160
1161         if (buflen > 0) {
1162                 ucl_munmap (buf, buflen);
1163         }
1164
1165         return res;
1166 }
1167
1168 /**
1169  * Include a file to configuration
1170  * @param data
1171  * @param len
1172  * @param parser
1173  * @param err
1174  * @return
1175  */
1176 static bool
1177 ucl_include_file (const unsigned char *data, size_t len,
1178                 struct ucl_parser *parser, struct ucl_include_params *params)
1179 {
1180         const unsigned char *p = data, *end = data + len;
1181         bool need_glob = false;
1182         int cnt = 0;
1183         char glob_pattern[PATH_MAX];
1184         size_t i;
1185
1186 #ifndef _WIN32
1187         if (!params->allow_glob) {
1188                 return ucl_include_file_single (data, len, parser, params);
1189         }
1190         else {
1191                 /* Check for special symbols in a filename */
1192                 while (p != end) {
1193                         if (*p == '*' || *p == '?') {
1194                                 need_glob = true;
1195                                 break;
1196                         }
1197                         p ++;
1198                 }
1199                 if (need_glob) {
1200                         glob_t globbuf;
1201                         memset (&globbuf, 0, sizeof (globbuf));
1202                         ucl_strlcpy (glob_pattern, (const char *)data,
1203                                 (len + 1 < sizeof (glob_pattern) ? len + 1 : sizeof (glob_pattern)));
1204                         if (glob (glob_pattern, 0, NULL, &globbuf) != 0) {
1205                                 return (!params->must_exist || false);
1206                         }
1207                         for (i = 0; i < globbuf.gl_pathc; i ++) {
1208                                 if (!ucl_include_file_single ((unsigned char *)globbuf.gl_pathv[i],
1209                                                 strlen (globbuf.gl_pathv[i]), parser, params)) {
1210                                         if (params->soft_fail) {
1211                                                 continue;
1212                                         }
1213                                         globfree (&globbuf);
1214                                         return false;
1215                                 }
1216                                 cnt ++;
1217                         }
1218                         globfree (&globbuf);
1219
1220                         if (cnt == 0 && params->must_exist) {
1221                                 ucl_create_err (&parser->err, "cannot match any files for pattern %s",
1222                                         glob_pattern);
1223                                 return false;
1224                         }
1225                 }
1226                 else {
1227                         return ucl_include_file_single (data, len, parser, params);
1228                 }
1229         }
1230 #else
1231         /* Win32 compilers do not support globbing. Therefore, for Win32,
1232            treat allow_glob/need_glob as a NOOP and just return */
1233         return ucl_include_file_single (data, len, parser, params);
1234 #endif
1235         
1236         return true;
1237 }
1238
1239 /**
1240  * Common function to handle .*include* macros
1241  * @param data
1242  * @param len
1243  * @param args
1244  * @param parser
1245  * @param default_try
1246  * @param default_sign
1247  * @return
1248  */
1249 static bool
1250 ucl_include_common (const unsigned char *data, size_t len,
1251                 const ucl_object_t *args, struct ucl_parser *parser,
1252                 bool default_try,
1253                 bool default_sign)
1254 {
1255         bool allow_url, search;
1256         const char *duplicate;
1257         const ucl_object_t *param;
1258         ucl_object_iter_t it = NULL, ip = NULL;
1259         char ipath[PATH_MAX];
1260         struct ucl_include_params params;
1261
1262         /* Default values */
1263         params.soft_fail = default_try;
1264         params.allow_glob = false;
1265         params.check_signature = default_sign;
1266         params.use_prefix = false;
1267         params.target = "object";
1268         params.prefix = NULL;
1269         params.priority = 0;
1270         params.parse_type = UCL_PARSE_UCL;
1271         params.strat = UCL_DUPLICATE_APPEND;
1272         params.must_exist = !default_try;
1273
1274         search = false;
1275
1276         /* Process arguments */
1277         if (args != NULL && args->type == UCL_OBJECT) {
1278                 while ((param = ucl_iterate_object (args, &it, true)) != NULL) {
1279                         if (param->type == UCL_BOOLEAN) {
1280                                 if (strncmp (param->key, "try", param->keylen) == 0) {
1281                                         params.must_exist = !ucl_object_toboolean (param);
1282                                 }
1283                                 else if (strncmp (param->key, "sign", param->keylen) == 0) {
1284                                         params.check_signature = ucl_object_toboolean (param);
1285                                 }
1286                                 else if (strncmp (param->key, "glob", param->keylen) == 0) {
1287                                         params.allow_glob = ucl_object_toboolean (param);
1288                                 }
1289                                 else if (strncmp (param->key, "url", param->keylen) == 0) {
1290                                         allow_url = ucl_object_toboolean (param);
1291                                 }
1292                                 else if (strncmp (param->key, "prefix", param->keylen) == 0) {
1293                                         params.use_prefix = ucl_object_toboolean (param);
1294                                 }
1295                         }
1296                         else if (param->type == UCL_STRING) {
1297                                 if (strncmp (param->key, "key", param->keylen) == 0) {
1298                                         params.prefix = ucl_object_tostring (param);
1299                                 }
1300                                 else if (strncmp (param->key, "target", param->keylen) == 0) {
1301                                         params.target = ucl_object_tostring (param);
1302                                 }
1303                                 else if (strncmp (param->key, "duplicate", param->keylen) == 0) {
1304                                         duplicate = ucl_object_tostring (param);
1305
1306                                         if (strcmp (duplicate, "append") == 0) {
1307                                                 params.strat = UCL_DUPLICATE_APPEND;
1308                                         }
1309                                         else if (strcmp (duplicate, "merge") == 0) {
1310                                                 params.strat = UCL_DUPLICATE_MERGE;
1311                                         }
1312                                         else if (strcmp (duplicate, "rewrite") == 0) {
1313                                                 params.strat = UCL_DUPLICATE_REWRITE;
1314                                         }
1315                                         else if (strcmp (duplicate, "error") == 0) {
1316                                                 params.strat = UCL_DUPLICATE_ERROR;
1317                                         }
1318                                 }
1319                         }
1320                         else if (param->type == UCL_ARRAY) {
1321                                 if (strncmp (param->key, "path", param->keylen) == 0) {
1322                                         ucl_set_include_path (parser, __DECONST(ucl_object_t *, param));
1323                                 }
1324                         }
1325                         else if (param->type == UCL_INT) {
1326                                 if (strncmp (param->key, "priority", param->keylen) == 0) {
1327                                         params.priority = ucl_object_toint (param);
1328                                 }
1329                         }
1330                 }
1331         }
1332
1333         if (parser->includepaths == NULL) {
1334                 if (allow_url && ucl_strnstr (data, "://", len) != NULL) {
1335                         /* Globbing is not used for URL's */
1336                         return ucl_include_url (data, len, parser, &params);
1337                 }
1338                 else if (data != NULL) {
1339                         /* Try to load a file */
1340                         return ucl_include_file (data, len, parser, &params);
1341                 }
1342         }
1343         else {
1344                 if (allow_url && ucl_strnstr (data, "://", len) != NULL) {
1345                         /* Globbing is not used for URL's */
1346                         return ucl_include_url (data, len, parser, &params);
1347                 }
1348
1349                 ip = ucl_object_iterate_new (parser->includepaths);
1350                 while ((param = ucl_object_iterate_safe (ip, true)) != NULL) {
1351                         if (ucl_object_type(param) == UCL_STRING) {
1352                                 snprintf (ipath, sizeof (ipath), "%s/%.*s", ucl_object_tostring(param),
1353                                                 (int)len, data);
1354                                 if ((search = ucl_include_file (ipath, strlen (ipath),
1355                                                 parser, &params))) {
1356                                         if (!params.allow_glob) {
1357                                                 break;
1358                                         }
1359                                 }
1360                         }
1361                 }
1362                 ucl_object_iterate_free (ip);
1363                 if (search == true) {
1364                         return true;
1365                 }
1366                 else {
1367                         ucl_create_err (&parser->err,
1368                                         "cannot find file: %.*s in search path",
1369                                         (int)len, data);
1370                         return false;
1371                 }
1372         }
1373
1374         return false;
1375 }
1376
1377 /**
1378  * Handle include macro
1379  * @param data include data
1380  * @param len length of data
1381  * @param args UCL object representing arguments to the macro
1382  * @param ud user data
1383  * @return
1384  */
1385 bool
1386 ucl_include_handler (const unsigned char *data, size_t len,
1387                 const ucl_object_t *args, void* ud)
1388 {
1389         struct ucl_parser *parser = ud;
1390
1391         return ucl_include_common (data, len, args, parser, false, false);
1392 }
1393
1394 /**
1395  * Handle includes macro
1396  * @param data include data
1397  * @param len length of data
1398  * @param args UCL object representing arguments to the macro
1399  * @param ud user data
1400  * @return
1401  */
1402 bool
1403 ucl_includes_handler (const unsigned char *data, size_t len,
1404                 const ucl_object_t *args, void* ud)
1405 {
1406         struct ucl_parser *parser = ud;
1407
1408         return ucl_include_common (data, len, args, parser, false, true);
1409 }
1410
1411 /**
1412  * Handle tryinclude macro
1413  * @param data include data
1414  * @param len length of data
1415  * @param args UCL object representing arguments to the macro
1416  * @param ud user data
1417  * @return
1418  */
1419 bool
1420 ucl_try_include_handler (const unsigned char *data, size_t len,
1421                 const ucl_object_t *args, void* ud)
1422 {
1423         struct ucl_parser *parser = ud;
1424
1425         return ucl_include_common (data, len, args, parser, true, false);
1426 }
1427
1428 /**
1429  * Handle priority macro
1430  * @param data include data
1431  * @param len length of data
1432  * @param args UCL object representing arguments to the macro
1433  * @param ud user data
1434  * @return
1435  */
1436 bool
1437 ucl_priority_handler (const unsigned char *data, size_t len,
1438                 const ucl_object_t *args, void* ud)
1439 {
1440         struct ucl_parser *parser = ud;
1441         unsigned priority = 255;
1442         const ucl_object_t *param;
1443         bool found = false;
1444         char *value = NULL, *leftover = NULL;
1445         ucl_object_iter_t it = NULL;
1446
1447         if (parser == NULL) {
1448                 return false;
1449         }
1450
1451         /* Process arguments */
1452         if (args != NULL && args->type == UCL_OBJECT) {
1453                 while ((param = ucl_iterate_object (args, &it, true)) != NULL) {
1454                         if (param->type == UCL_INT) {
1455                                 if (strncmp (param->key, "priority", param->keylen) == 0) {
1456                                         priority = ucl_object_toint (param);
1457                                         found = true;
1458                                 }
1459                         }
1460                 }
1461         }
1462
1463         if (len > 0) {
1464                 value = malloc(len + 1);
1465                 ucl_strlcpy(value, (const char *)data, len + 1);
1466                 priority = strtol(value, &leftover, 10);
1467                 if (*leftover != '\0') {
1468                         ucl_create_err (&parser->err, "Invalid priority value in macro: %s",
1469                                 value);
1470                         free(value);
1471                         return false;
1472                 }
1473                 free(value);
1474                 found = true;
1475         }
1476
1477         if (found == true) {
1478                 parser->chunks->priority = priority;
1479                 return true;
1480         }
1481
1482         ucl_create_err (&parser->err, "Unable to parse priority macro");
1483         return false;
1484 }
1485
1486 /**
1487  * Handle load macro
1488  * @param data include data
1489  * @param len length of data
1490  * @param args UCL object representing arguments to the macro
1491  * @param ud user data
1492  * @return
1493  */
1494 bool
1495 ucl_load_handler (const unsigned char *data, size_t len,
1496                 const ucl_object_t *args, void* ud)
1497 {
1498         struct ucl_parser *parser = ud;
1499         const ucl_object_t *param;
1500         ucl_object_t *obj, *old_obj;
1501         ucl_object_iter_t it = NULL;
1502         bool try_load, multiline, test;
1503         const char *target, *prefix;
1504         char *load_file, *tmp;
1505         unsigned char *buf;
1506         size_t buflen;
1507         unsigned priority;
1508         int64_t iv;
1509         ucl_hash_t *container = NULL;
1510         enum ucl_string_flags flags;
1511
1512         /* Default values */
1513         try_load = false;
1514         multiline = false;
1515         test = false;
1516         target = "string";
1517         prefix = NULL;
1518         load_file = NULL;
1519         buf = NULL;
1520         buflen = 0;
1521         priority = 0;
1522         obj = NULL;
1523         old_obj = NULL;
1524         flags = 0;
1525
1526         if (parser == NULL) {
1527                 return false;
1528         }
1529
1530         /* Process arguments */
1531         if (args != NULL && args->type == UCL_OBJECT) {
1532                 while ((param = ucl_iterate_object (args, &it, true)) != NULL) {
1533                         if (param->type == UCL_BOOLEAN) {
1534                                 if (strncmp (param->key, "try", param->keylen) == 0) {
1535                                         try_load = ucl_object_toboolean (param);
1536                                 }
1537                                 else if (strncmp (param->key, "multiline", param->keylen) == 0) {
1538                                         multiline = ucl_object_toboolean (param);
1539                                 }
1540                                 else if (strncmp (param->key, "escape", param->keylen) == 0) {
1541                                         test = ucl_object_toboolean (param);
1542                                         if (test) {
1543                                                 flags |= UCL_STRING_ESCAPE;
1544                                         }
1545                                 }
1546                                 else if (strncmp (param->key, "trim", param->keylen) == 0) {
1547                                         test = ucl_object_toboolean (param);
1548                                         if (test) {
1549                                                 flags |= UCL_STRING_TRIM;
1550                                         }
1551                                 }
1552                         }
1553                         else if (param->type == UCL_STRING) {
1554                                 if (strncmp (param->key, "key", param->keylen) == 0) {
1555                                         prefix = ucl_object_tostring (param);
1556                                 }
1557                                 else if (strncmp (param->key, "target", param->keylen) == 0) {
1558                                         target = ucl_object_tostring (param);
1559                                 }
1560                         }
1561                         else if (param->type == UCL_INT) {
1562                                 if (strncmp (param->key, "priority", param->keylen) == 0) {
1563                                         priority = ucl_object_toint (param);
1564                                 }
1565                         }
1566                 }
1567         }
1568
1569         if (prefix == NULL || strlen(prefix) == 0) {
1570                 ucl_create_err (&parser->err, "No Key specified in load macro");
1571                 return false;
1572         }
1573
1574         if (len > 0) {
1575                 asprintf (&load_file, "%.*s", (int)len, data);
1576                 if (!ucl_fetch_file (load_file, &buf, &buflen, &parser->err, !try_load)) {
1577                         return (try_load || false);
1578                 }
1579
1580                 container = parser->stack->obj->value.ov;
1581                 old_obj = __DECONST (ucl_object_t *, ucl_hash_search (container, prefix, strlen (prefix)));
1582                 if (old_obj != NULL) {
1583                         ucl_create_err (&parser->err, "Key %s already exists", prefix);
1584                         return false;
1585                 }
1586
1587                 if (strcasecmp (target, "string") == 0) {
1588                         obj = ucl_object_fromstring_common (buf, buflen, flags);
1589                         ucl_copy_value_trash (obj);
1590                         if (multiline) {
1591                                 obj->flags |= UCL_OBJECT_MULTILINE;
1592                         }
1593                 }
1594                 else if (strcasecmp (target, "int") == 0) {
1595                         asprintf(&tmp, "%.*s", (int)buflen, buf);
1596                         iv = strtoll(tmp, NULL, 10);
1597                         obj = ucl_object_fromint(iv);
1598                 }
1599
1600                 if (buflen > 0) {
1601                         ucl_munmap (buf, buflen);
1602                 }
1603
1604                 if (obj != NULL) {
1605                         obj->key = prefix;
1606                         obj->keylen = strlen (prefix);
1607                         ucl_copy_key_trash(obj);
1608                         obj->prev = obj;
1609                         obj->next = NULL;
1610                         ucl_object_set_priority (obj, priority);
1611                         container = ucl_hash_insert_object (container, obj,
1612                                         parser->flags & UCL_PARSER_KEY_LOWERCASE);
1613                         parser->stack->obj->value.ov = container;
1614                 }
1615                 return true;
1616         }
1617
1618         ucl_create_err (&parser->err, "Unable to parse load macro");
1619         return false;
1620 }
1621
1622 bool
1623 ucl_inherit_handler (const unsigned char *data, size_t len,
1624                 const ucl_object_t *args, const ucl_object_t *ctx, void* ud)
1625 {
1626         const ucl_object_t *parent, *cur;
1627         ucl_object_t *target, *copy;
1628         ucl_object_iter_t it = NULL;
1629         bool replace = false;
1630         struct ucl_parser *parser = ud;
1631
1632         parent = ucl_object_find_keyl (ctx, data, len);
1633
1634         /* Some sanity checks */
1635         if (parent == NULL || ucl_object_type (parent) != UCL_OBJECT) {
1636                 ucl_create_err (&parser->err, "Unable to find inherited object %*.s",
1637                                 (int)len, data);
1638                 return false;
1639         }
1640
1641         if (parser->stack == NULL || parser->stack->obj == NULL ||
1642                         ucl_object_type (parser->stack->obj) != UCL_OBJECT) {
1643                 ucl_create_err (&parser->err, "Invalid inherit context");
1644                 return false;
1645         }
1646
1647         target = parser->stack->obj;
1648
1649         if (args && (cur = ucl_object_find_key (args, "replace")) != NULL) {
1650                 replace = ucl_object_toboolean (cur);
1651         }
1652
1653         while ((cur = ucl_iterate_object (parent, &it, true))) {
1654                 /* We do not replace existing keys */
1655                 if (!replace && ucl_object_find_keyl (target, cur->key, cur->keylen)) {
1656                         continue;
1657                 }
1658
1659                 copy = ucl_object_copy (cur);
1660
1661                 if (!replace) {
1662                         copy->flags |= UCL_OBJECT_INHERITED;
1663                 }
1664
1665                 ucl_object_insert_key (target, copy, copy->key,
1666                                 copy->keylen, false);
1667         }
1668
1669         return true;
1670 }
1671
1672 bool
1673 ucl_parser_set_filevars (struct ucl_parser *parser, const char *filename, bool need_expand)
1674 {
1675         char realbuf[PATH_MAX], *curdir;
1676
1677         if (filename != NULL) {
1678                 if (need_expand) {
1679                         if (ucl_realpath (filename, realbuf) == NULL) {
1680                                 return false;
1681                         }
1682                 }
1683                 else {
1684                         ucl_strlcpy (realbuf, filename, sizeof (realbuf));
1685                 }
1686
1687                 /* Define variables */
1688                 ucl_parser_register_variable (parser, "FILENAME", realbuf);
1689                 curdir = dirname (realbuf);
1690                 ucl_parser_register_variable (parser, "CURDIR", curdir);
1691         }
1692         else {
1693                 /* Set everything from the current dir */
1694                 curdir = getcwd (realbuf, sizeof (realbuf));
1695                 ucl_parser_register_variable (parser, "FILENAME", "undef");
1696                 ucl_parser_register_variable (parser, "CURDIR", curdir);
1697         }
1698
1699         return true;
1700 }
1701
1702 bool
1703 ucl_parser_add_file_priority (struct ucl_parser *parser, const char *filename,
1704                 unsigned priority)
1705 {
1706         unsigned char *buf;
1707         size_t len;
1708         bool ret;
1709         char realbuf[PATH_MAX];
1710
1711         if (ucl_realpath (filename, realbuf) == NULL) {
1712                 ucl_create_err (&parser->err, "cannot open file %s: %s",
1713                                 filename,
1714                                 strerror (errno));
1715                 return false;
1716         }
1717
1718         if (!ucl_fetch_file (realbuf, &buf, &len, &parser->err, true)) {
1719                 return false;
1720         }
1721
1722         if (parser->cur_file) {
1723                 free (parser->cur_file);
1724         }
1725         parser->cur_file = strdup (realbuf);
1726         ucl_parser_set_filevars (parser, realbuf, false);
1727         ret = ucl_parser_add_chunk_priority (parser, buf, len, priority);
1728
1729         if (len > 0) {
1730                 ucl_munmap (buf, len);
1731         }
1732
1733         return ret;
1734 }
1735
1736 bool
1737 ucl_parser_add_file (struct ucl_parser *parser, const char *filename)
1738 {
1739         if (parser == NULL) {
1740                 return false;
1741         }
1742
1743         return ucl_parser_add_file_priority(parser, filename,
1744                         parser->default_priority);
1745 }
1746
1747 bool
1748 ucl_parser_add_fd_priority (struct ucl_parser *parser, int fd,
1749                 unsigned priority)
1750 {
1751         unsigned char *buf;
1752         size_t len;
1753         bool ret;
1754         struct stat st;
1755
1756         if (fstat (fd, &st) == -1) {
1757                 ucl_create_err (&parser->err, "cannot stat fd %d: %s",
1758                         fd, strerror (errno));
1759                 return false;
1760         }
1761         if ((buf = ucl_mmap (NULL, st.st_size, PROT_READ, MAP_SHARED, fd, 0)) == MAP_FAILED) {
1762                 ucl_create_err (&parser->err, "cannot mmap fd %d: %s",
1763                         fd, strerror (errno));
1764                 return false;
1765         }
1766
1767         if (parser->cur_file) {
1768                 free (parser->cur_file);
1769         }
1770         parser->cur_file = NULL;
1771         len = st.st_size;
1772         ret = ucl_parser_add_chunk_priority (parser, buf, len, priority);
1773
1774         if (len > 0) {
1775                 ucl_munmap (buf, len);
1776         }
1777
1778         return ret;
1779 }
1780
1781 bool
1782 ucl_parser_add_fd (struct ucl_parser *parser, int fd)
1783 {
1784         if (parser == NULL) {
1785                 return false;
1786         }
1787
1788         return ucl_parser_add_fd_priority(parser, fd, parser->default_priority);
1789 }
1790
1791 size_t
1792 ucl_strlcpy (char *dst, const char *src, size_t siz)
1793 {
1794         char *d = dst;
1795         const char *s = src;
1796         size_t n = siz;
1797
1798         /* Copy as many bytes as will fit */
1799         if (n != 0) {
1800                 while (--n != 0) {
1801                         if ((*d++ = *s++) == '\0') {
1802                                 break;
1803                         }
1804                 }
1805         }
1806
1807         if (n == 0 && siz != 0) {
1808                 *d = '\0';
1809         }
1810
1811         return (s - src - 1);    /* count does not include NUL */
1812 }
1813
1814 size_t
1815 ucl_strlcpy_unsafe (char *dst, const char *src, size_t siz)
1816 {
1817         memcpy (dst, src, siz - 1);
1818         dst[siz - 1] = '\0';
1819
1820         return siz - 1;
1821 }
1822
1823 size_t
1824 ucl_strlcpy_tolower (char *dst, const char *src, size_t siz)
1825 {
1826         char *d = dst;
1827         const char *s = src;
1828         size_t n = siz;
1829
1830         /* Copy as many bytes as will fit */
1831         if (n != 0) {
1832                 while (--n != 0) {
1833                         if ((*d++ = tolower (*s++)) == '\0') {
1834                                 break;
1835                         }
1836                 }
1837         }
1838
1839         if (n == 0 && siz != 0) {
1840                 *d = '\0';
1841         }
1842
1843         return (s - src);    /* count does not include NUL */
1844 }
1845
1846 /*
1847  * Find the first occurrence of find in s
1848  */
1849 char *
1850 ucl_strnstr (const char *s, const char *find, int len)
1851 {
1852         char c, sc;
1853         int mlen;
1854
1855         if ((c = *find++) != 0) {
1856                 mlen = strlen (find);
1857                 do {
1858                         do {
1859                                 if ((sc = *s++) == 0 || len-- == 0)
1860                                         return (NULL);
1861                         } while (sc != c);
1862                 } while (strncmp (s, find, mlen) != 0);
1863                 s--;
1864         }
1865         return ((char *)s);
1866 }
1867
1868 /*
1869  * Find the first occurrence of find in s, ignore case.
1870  */
1871 char *
1872 ucl_strncasestr (const char *s, const char *find, int len)
1873 {
1874         char c, sc;
1875         int mlen;
1876
1877         if ((c = *find++) != 0) {
1878                 c = tolower (c);
1879                 mlen = strlen (find);
1880                 do {
1881                         do {
1882                                 if ((sc = *s++) == 0 || len-- == 0)
1883                                         return (NULL);
1884                         } while (tolower (sc) != c);
1885                 } while (strncasecmp (s, find, mlen) != 0);
1886                 s--;
1887         }
1888         return ((char *)s);
1889 }
1890
1891 ucl_object_t *
1892 ucl_object_fromstring_common (const char *str, size_t len, enum ucl_string_flags flags)
1893 {
1894         ucl_object_t *obj;
1895         const char *start, *end, *p, *pos;
1896         char *dst, *d;
1897         size_t escaped_len;
1898
1899         if (str == NULL) {
1900                 return NULL;
1901         }
1902
1903         obj = ucl_object_new ();
1904         if (obj) {
1905                 if (len == 0) {
1906                         len = strlen (str);
1907                 }
1908                 if (flags & UCL_STRING_TRIM) {
1909                         /* Skip leading spaces */
1910                         for (start = str; (size_t)(start - str) < len; start ++) {
1911                                 if (!ucl_test_character (*start, UCL_CHARACTER_WHITESPACE_UNSAFE)) {
1912                                         break;
1913                                 }
1914                         }
1915                         /* Skip trailing spaces */
1916                         for (end = str + len - 1; end > start; end --) {
1917                                 if (!ucl_test_character (*end, UCL_CHARACTER_WHITESPACE_UNSAFE)) {
1918                                         break;
1919                                 }
1920                         }
1921                         end ++;
1922                 }
1923                 else {
1924                         start = str;
1925                         end = str + len;
1926                 }
1927
1928                 obj->type = UCL_STRING;
1929                 if (flags & UCL_STRING_ESCAPE) {
1930                         for (p = start, escaped_len = 0; p < end; p ++, escaped_len ++) {
1931                                 if (ucl_test_character (*p, UCL_CHARACTER_JSON_UNSAFE)) {
1932                                         escaped_len ++;
1933                                 }
1934                         }
1935                         dst = malloc (escaped_len + 1);
1936                         if (dst != NULL) {
1937                                 for (p = start, d = dst; p < end; p ++, d ++) {
1938                                         if (ucl_test_character (*p, UCL_CHARACTER_JSON_UNSAFE)) {
1939                                                 switch (*p) {
1940                                                 case '\n':
1941                                                         *d++ = '\\';
1942                                                         *d = 'n';
1943                                                         break;
1944                                                 case '\r':
1945                                                         *d++ = '\\';
1946                                                         *d = 'r';
1947                                                         break;
1948                                                 case '\b':
1949                                                         *d++ = '\\';
1950                                                         *d = 'b';
1951                                                         break;
1952                                                 case '\t':
1953                                                         *d++ = '\\';
1954                                                         *d = 't';
1955                                                         break;
1956                                                 case '\f':
1957                                                         *d++ = '\\';
1958                                                         *d = 'f';
1959                                                         break;
1960                                                 case '\\':
1961                                                         *d++ = '\\';
1962                                                         *d = '\\';
1963                                                         break;
1964                                                 case '"':
1965                                                         *d++ = '\\';
1966                                                         *d = '"';
1967                                                         break;
1968                                                 }
1969                                         }
1970                                         else {
1971                                                 *d = *p;
1972                                         }
1973                                 }
1974                                 *d = '\0';
1975                                 obj->value.sv = dst;
1976                                 obj->trash_stack[UCL_TRASH_VALUE] = dst;
1977                                 obj->len = escaped_len;
1978                         }
1979                 }
1980                 else {
1981                         dst = malloc (end - start + 1);
1982                         if (dst != NULL) {
1983                                 ucl_strlcpy_unsafe (dst, start, end - start + 1);
1984                                 obj->value.sv = dst;
1985                                 obj->trash_stack[UCL_TRASH_VALUE] = dst;
1986                                 obj->len = end - start;
1987                         }
1988                 }
1989                 if ((flags & UCL_STRING_PARSE) && dst != NULL) {
1990                         /* Parse what we have */
1991                         if (flags & UCL_STRING_PARSE_BOOLEAN) {
1992                                 if (!ucl_maybe_parse_boolean (obj, dst, obj->len) && (flags & UCL_STRING_PARSE_NUMBER)) {
1993                                         ucl_maybe_parse_number (obj, dst, dst + obj->len, &pos,
1994                                                         flags & UCL_STRING_PARSE_DOUBLE,
1995                                                         flags & UCL_STRING_PARSE_BYTES,
1996                                                         flags & UCL_STRING_PARSE_TIME);
1997                                 }
1998                         }
1999                         else {
2000                                 ucl_maybe_parse_number (obj, dst, dst + obj->len, &pos,
2001                                                 flags & UCL_STRING_PARSE_DOUBLE,
2002                                                 flags & UCL_STRING_PARSE_BYTES,
2003                                                 flags & UCL_STRING_PARSE_TIME);
2004                         }
2005                 }
2006         }
2007
2008         return obj;
2009 }
2010
2011 static bool
2012 ucl_object_insert_key_common (ucl_object_t *top, ucl_object_t *elt,
2013                 const char *key, size_t keylen, bool copy_key, bool merge, bool replace)
2014 {
2015         ucl_object_t *found, *tmp;
2016         const ucl_object_t *cur;
2017         ucl_object_iter_t it = NULL;
2018         const char *p;
2019         int ret = true;
2020
2021         if (elt == NULL || key == NULL) {
2022                 return false;
2023         }
2024
2025         if (top == NULL) {
2026                 return false;
2027         }
2028
2029         if (top->type != UCL_OBJECT) {
2030                 /* It is possible to convert NULL type to an object */
2031                 if (top->type == UCL_NULL) {
2032                         top->type = UCL_OBJECT;
2033                 }
2034                 else {
2035                         /* Refuse converting of other object types */
2036                         return false;
2037                 }
2038         }
2039
2040         if (top->value.ov == NULL) {
2041                 top->value.ov = ucl_hash_create (false);
2042         }
2043
2044         if (keylen == 0) {
2045                 keylen = strlen (key);
2046         }
2047
2048         for (p = key; p < key + keylen; p ++) {
2049                 if (ucl_test_character (*p, UCL_CHARACTER_UCL_UNSAFE)) {
2050                         elt->flags |= UCL_OBJECT_NEED_KEY_ESCAPE;
2051                         break;
2052                 }
2053         }
2054
2055         /* workaround for some use cases */
2056         if (elt->trash_stack[UCL_TRASH_KEY] != NULL &&
2057                         key != (const char *)elt->trash_stack[UCL_TRASH_KEY]) {
2058                 /* Remove copied key */
2059                 free (elt->trash_stack[UCL_TRASH_KEY]);
2060                 elt->trash_stack[UCL_TRASH_KEY] = NULL;
2061                 elt->flags &= ~UCL_OBJECT_ALLOCATED_KEY;
2062         }
2063
2064         elt->key = key;
2065         elt->keylen = keylen;
2066
2067         if (copy_key) {
2068                 ucl_copy_key_trash (elt);
2069         }
2070
2071         found = __DECONST (ucl_object_t *, ucl_hash_search_obj (top->value.ov, elt));
2072
2073         if (found == NULL) {
2074                 top->value.ov = ucl_hash_insert_object (top->value.ov, elt, false);
2075                 top->len ++;
2076                 if (replace) {
2077                         ret = false;
2078                 }
2079         }
2080         else {
2081                 if (replace) {
2082                         ucl_hash_replace (top->value.ov, found, elt);
2083                         ucl_object_unref (found);
2084                 }
2085                 else if (merge) {
2086                         if (found->type != UCL_OBJECT && elt->type == UCL_OBJECT) {
2087                                 /* Insert old elt to new one */
2088                                 ucl_object_insert_key_common (elt, found, found->key,
2089                                                 found->keylen, copy_key, false, false);
2090                                 ucl_hash_delete (top->value.ov, found);
2091                                 top->value.ov = ucl_hash_insert_object (top->value.ov, elt, false);
2092                         }
2093                         else if (found->type == UCL_OBJECT && elt->type != UCL_OBJECT) {
2094                                 /* Insert new to old */
2095                                 ucl_object_insert_key_common (found, elt, elt->key,
2096                                                 elt->keylen, copy_key, false, false);
2097                         }
2098                         else if (found->type == UCL_OBJECT && elt->type == UCL_OBJECT) {
2099                                 /* Mix two hashes */
2100                                 while ((cur = ucl_iterate_object (elt, &it, true)) != NULL) {
2101                                         tmp = ucl_object_ref (cur);
2102                                         ucl_object_insert_key_common (found, tmp, cur->key,
2103                                                         cur->keylen, copy_key, false, false);
2104                                 }
2105                                 ucl_object_unref (elt);
2106                         }
2107                         else {
2108                                 /* Just make a list of scalars */
2109                                 DL_APPEND (found, elt);
2110                         }
2111                 }
2112                 else {
2113                         DL_APPEND (found, elt);
2114                 }
2115         }
2116
2117         return ret;
2118 }
2119
2120 bool
2121 ucl_object_delete_keyl (ucl_object_t *top, const char *key, size_t keylen)
2122 {
2123         ucl_object_t *found;
2124
2125         if (top == NULL || key == NULL) {
2126                 return false;
2127         }
2128
2129         found = __DECONST (ucl_object_t *, ucl_object_find_keyl (top, key, keylen));
2130
2131         if (found == NULL) {
2132                 return false;
2133         }
2134
2135         ucl_hash_delete (top->value.ov, found);
2136         ucl_object_unref (found);
2137         top->len --;
2138
2139         return true;
2140 }
2141
2142 bool
2143 ucl_object_delete_key (ucl_object_t *top, const char *key)
2144 {
2145         return ucl_object_delete_keyl (top, key, strlen (key));
2146 }
2147
2148 ucl_object_t*
2149 ucl_object_pop_keyl (ucl_object_t *top, const char *key, size_t keylen)
2150 {
2151         const ucl_object_t *found;
2152
2153         if (top == NULL || key == NULL) {
2154                 return false;
2155         }
2156         found = ucl_object_find_keyl (top, key, keylen);
2157
2158         if (found == NULL) {
2159                 return NULL;
2160         }
2161         ucl_hash_delete (top->value.ov, found);
2162         top->len --;
2163
2164         return __DECONST (ucl_object_t *, found);
2165 }
2166
2167 ucl_object_t*
2168 ucl_object_pop_key (ucl_object_t *top, const char *key)
2169 {
2170         return ucl_object_pop_keyl (top, key, strlen (key));
2171 }
2172
2173 bool
2174 ucl_object_insert_key (ucl_object_t *top, ucl_object_t *elt,
2175                 const char *key, size_t keylen, bool copy_key)
2176 {
2177         return ucl_object_insert_key_common (top, elt, key, keylen, copy_key, false, false);
2178 }
2179
2180 bool
2181 ucl_object_insert_key_merged (ucl_object_t *top, ucl_object_t *elt,
2182                 const char *key, size_t keylen, bool copy_key)
2183 {
2184         return ucl_object_insert_key_common (top, elt, key, keylen, copy_key, true, false);
2185 }
2186
2187 bool
2188 ucl_object_replace_key (ucl_object_t *top, ucl_object_t *elt,
2189                 const char *key, size_t keylen, bool copy_key)
2190 {
2191         return ucl_object_insert_key_common (top, elt, key, keylen, copy_key, false, true);
2192 }
2193
2194 bool
2195 ucl_object_merge (ucl_object_t *top, ucl_object_t *elt, bool copy)
2196 {
2197         ucl_object_t *cur = NULL, *cp = NULL, *found = NULL;
2198         ucl_object_iter_t iter = NULL;
2199
2200         if (top == NULL || top->type != UCL_OBJECT || elt == NULL || elt->type != UCL_OBJECT) {
2201                 return false;
2202         }
2203
2204         /* Mix two hashes */
2205         while ((cur = (ucl_object_t*)ucl_hash_iterate (elt->value.ov, &iter))) {
2206                 if (copy) {
2207                         cp = ucl_object_copy (cur);
2208                 }
2209                 else {
2210                         cp = ucl_object_ref (cur);
2211                 }
2212                 found = __DECONST(ucl_object_t *, ucl_hash_search (top->value.ov, cp->key, cp->keylen));
2213                 if (found == NULL) {
2214                         /* The key does not exist */
2215                         top->value.ov = ucl_hash_insert_object (top->value.ov, cp, false);
2216                         top->len ++;
2217                 }
2218                 else {
2219                         /* The key already exists, replace it */
2220                         ucl_hash_replace (top->value.ov, found, cp);
2221                         ucl_object_unref (found);
2222                 }
2223         }
2224
2225         return true;
2226 }
2227
2228 const ucl_object_t *
2229 ucl_object_find_keyl (const ucl_object_t *obj, const char *key, size_t klen)
2230 {
2231         const ucl_object_t *ret;
2232         ucl_object_t srch;
2233
2234         if (obj == NULL || obj->type != UCL_OBJECT || key == NULL) {
2235                 return NULL;
2236         }
2237
2238         srch.key = key;
2239         srch.keylen = klen;
2240         ret = ucl_hash_search_obj (obj->value.ov, &srch);
2241
2242         return ret;
2243 }
2244
2245 const ucl_object_t *
2246 ucl_object_find_key (const ucl_object_t *obj, const char *key)
2247 {
2248         if (key == NULL) {
2249                 return NULL;
2250         }
2251
2252         return ucl_object_find_keyl (obj, key, strlen (key));
2253 }
2254
2255 const ucl_object_t*
2256 ucl_object_find_any_key (const ucl_object_t *obj,
2257                 const char *key, ...)
2258 {
2259         va_list ap;
2260         const ucl_object_t *ret = NULL;
2261         const char *nk = NULL;
2262
2263         if (obj == NULL || key == NULL) {
2264                 return NULL;
2265         }
2266
2267         ret = ucl_object_find_keyl (obj, key, strlen (key));
2268
2269         if (ret == NULL) {
2270                 va_start (ap, key);
2271
2272                 while (ret == NULL) {
2273                         nk = va_arg (ap, const char *);
2274
2275                         if (nk == NULL) {
2276                                 break;
2277                         }
2278                         else {
2279                                 ret = ucl_object_find_keyl (obj, nk, strlen (nk));
2280                         }
2281                 }
2282
2283                 va_end (ap);
2284         }
2285
2286         return ret;
2287 }
2288
2289 const ucl_object_t*
2290 ucl_iterate_object (const ucl_object_t *obj, ucl_object_iter_t *iter, bool expand_values)
2291 {
2292         const ucl_object_t *elt = NULL;
2293
2294         if (obj == NULL || iter == NULL) {
2295                 return NULL;
2296         }
2297
2298         if (expand_values) {
2299                 switch (obj->type) {
2300                 case UCL_OBJECT:
2301                         return (const ucl_object_t*)ucl_hash_iterate (obj->value.ov, iter);
2302                         break;
2303                 case UCL_ARRAY: {
2304                         unsigned int idx;
2305                         UCL_ARRAY_GET (vec, obj);
2306                         idx = (unsigned int)(uintptr_t)(*iter);
2307
2308                         if (vec != NULL) {
2309                                 while (idx < kv_size (*vec)) {
2310                                         if ((elt = kv_A (*vec, idx)) != NULL) {
2311                                                 idx ++;
2312                                                 break;
2313                                         }
2314                                         idx ++;
2315                                 }
2316                                 *iter = (void *)(uintptr_t)idx;
2317                         }
2318
2319                         return elt;
2320                         break;
2321                 }
2322                 default:
2323                         /* Go to linear iteration */
2324                         break;
2325                 }
2326         }
2327         /* Treat everything as a linear list */
2328         elt = *iter;
2329         if (elt == NULL) {
2330                 elt = obj;
2331         }
2332         else if (elt == obj) {
2333                 return NULL;
2334         }
2335         *iter = __DECONST (void *, elt->next ? elt->next : obj);
2336         return elt;
2337
2338         /* Not reached */
2339         return NULL;
2340 }
2341
2342 const char safe_iter_magic[4] = {'u', 'i', 't', 'e'};
2343 struct ucl_object_safe_iter {
2344         char magic[4]; /* safety check */
2345         const ucl_object_t *impl_it; /* implicit object iteration */
2346         ucl_object_iter_t expl_it; /* explicit iteration */
2347 };
2348
2349 #define UCL_SAFE_ITER(ptr) (struct ucl_object_safe_iter *)(ptr)
2350 #define UCL_SAFE_ITER_CHECK(it) do { \
2351         assert (it != NULL); \
2352         assert (memcmp (it->magic, safe_iter_magic, sizeof (it->magic)) == 0); \
2353  } while (0)
2354
2355 ucl_object_iter_t
2356 ucl_object_iterate_new (const ucl_object_t *obj)
2357 {
2358         struct ucl_object_safe_iter *it;
2359
2360         it = UCL_ALLOC (sizeof (*it));
2361         if (it != NULL) {
2362                 memcpy (it->magic, safe_iter_magic, sizeof (it->magic));
2363                 it->expl_it = NULL;
2364                 it->impl_it = obj;
2365         }
2366
2367         return (ucl_object_iter_t)it;
2368 }
2369
2370
2371 ucl_object_iter_t
2372 ucl_object_iterate_reset (ucl_object_iter_t it, const ucl_object_t *obj)
2373 {
2374         struct ucl_object_safe_iter *rit = UCL_SAFE_ITER (it);
2375
2376         UCL_SAFE_ITER_CHECK (rit);
2377
2378         rit->impl_it = obj;
2379         rit->expl_it = NULL;
2380
2381         return it;
2382 }
2383
2384 const ucl_object_t*
2385 ucl_object_iterate_safe (ucl_object_iter_t it, bool expand_values)
2386 {
2387         struct ucl_object_safe_iter *rit = UCL_SAFE_ITER (it);
2388         const ucl_object_t *ret = NULL;
2389
2390         UCL_SAFE_ITER_CHECK (rit);
2391
2392         if (rit->impl_it == NULL) {
2393                 return NULL;
2394         }
2395
2396         if (rit->impl_it->type == UCL_OBJECT || rit->impl_it->type == UCL_ARRAY) {
2397                 ret = ucl_iterate_object (rit->impl_it, &rit->expl_it, true);
2398
2399                 if (ret == NULL) {
2400                         /* Need to switch to another implicit object in chain */
2401                         rit->impl_it = rit->impl_it->next;
2402                         rit->expl_it = NULL;
2403                         return ucl_object_iterate_safe (it, expand_values);
2404                 }
2405         }
2406         else {
2407                 /* Just iterate over the implicit array */
2408                 ret = rit->impl_it;
2409                 rit->impl_it = rit->impl_it->next;
2410                 if (expand_values) {
2411                         /* We flatten objects if need to expand values */
2412                         if (ret->type == UCL_OBJECT || ret->type == UCL_ARRAY) {
2413                                 return ucl_object_iterate_safe (it, expand_values);
2414                         }
2415                 }
2416         }
2417
2418         return ret;
2419 }
2420
2421 void
2422 ucl_object_iterate_free (ucl_object_iter_t it)
2423 {
2424         struct ucl_object_safe_iter *rit = UCL_SAFE_ITER (it);
2425
2426         UCL_SAFE_ITER_CHECK (rit);
2427
2428         UCL_FREE (sizeof (*rit), it);
2429 }
2430
2431 const ucl_object_t *
2432 ucl_lookup_path (const ucl_object_t *top, const char *path_in) {
2433         return ucl_lookup_path_char (top, path_in, '.');
2434 }
2435
2436
2437 const ucl_object_t *
2438 ucl_lookup_path_char (const ucl_object_t *top, const char *path_in, const char sep) {
2439         const ucl_object_t *o = NULL, *found;
2440         const char *p, *c;
2441         char *err_str;
2442         unsigned index;
2443
2444         if (path_in == NULL || top == NULL) {
2445                 return NULL;
2446         }
2447
2448         found = NULL;
2449         p = path_in;
2450
2451         /* Skip leading dots */
2452         while (*p == sep) {
2453                 p ++;
2454         }
2455
2456         c = p;
2457         while (*p != '\0') {
2458                 p ++;
2459                 if (*p == sep || *p == '\0') {
2460                         if (p > c) {
2461                                 switch (top->type) {
2462                                 case UCL_ARRAY:
2463                                         /* Key should be an int */
2464                                         index = strtoul (c, &err_str, 10);
2465                                         if (err_str != NULL && (*err_str != sep && *err_str != '\0')) {
2466                                                 return NULL;
2467                                         }
2468                                         o = ucl_array_find_index (top, index);
2469                                         break;
2470                                 default:
2471                                         o = ucl_object_find_keyl (top, c, p - c);
2472                                         break;
2473                                 }
2474                                 if (o == NULL) {
2475                                         return NULL;
2476                                 }
2477                                 top = o;
2478                         }
2479                         if (*p != '\0') {
2480                                 c = p + 1;
2481                         }
2482                 }
2483         }
2484         found = o;
2485
2486         return found;
2487 }
2488
2489
2490 ucl_object_t *
2491 ucl_object_new (void)
2492 {
2493         return ucl_object_typed_new (UCL_NULL);
2494 }
2495
2496 ucl_object_t *
2497 ucl_object_typed_new (ucl_type_t type)
2498 {
2499         return ucl_object_new_full (type, 0);
2500 }
2501
2502 ucl_object_t *
2503 ucl_object_new_full (ucl_type_t type, unsigned priority)
2504 {
2505         ucl_object_t *new;
2506
2507         if (type != UCL_USERDATA) {
2508                 new = UCL_ALLOC (sizeof (ucl_object_t));
2509                 if (new != NULL) {
2510                         memset (new, 0, sizeof (ucl_object_t));
2511                         new->ref = 1;
2512                         new->type = (type <= UCL_NULL ? type : UCL_NULL);
2513                         new->next = NULL;
2514                         new->prev = new;
2515                         ucl_object_set_priority (new, priority);
2516
2517                         if (type == UCL_ARRAY) {
2518                                 new->value.av = UCL_ALLOC (sizeof (ucl_array_t));
2519                                 if (new->value.av) {
2520                                         memset (new->value.av, 0, sizeof (ucl_array_t));
2521                                         UCL_ARRAY_GET (vec, new);
2522
2523                                         /* Preallocate some space for arrays */
2524                                         kv_resize (ucl_object_t *, *vec, 8);
2525                                 }
2526                         }
2527                 }
2528         }
2529         else {
2530                 new = ucl_object_new_userdata (NULL, NULL);
2531                 ucl_object_set_priority (new, priority);
2532         }
2533
2534         return new;
2535 }
2536
2537 ucl_object_t*
2538 ucl_object_new_userdata (ucl_userdata_dtor dtor, ucl_userdata_emitter emitter)
2539 {
2540         struct ucl_object_userdata *new;
2541         size_t nsize = sizeof (*new);
2542
2543         new = UCL_ALLOC (nsize);
2544         if (new != NULL) {
2545                 memset (new, 0, nsize);
2546                 new->obj.ref = 1;
2547                 new->obj.type = UCL_USERDATA;
2548                 new->obj.next = NULL;
2549                 new->obj.prev = (ucl_object_t *)new;
2550                 new->dtor = dtor;
2551                 new->emitter = emitter;
2552         }
2553
2554         return (ucl_object_t *)new;
2555 }
2556
2557 ucl_type_t
2558 ucl_object_type (const ucl_object_t *obj)
2559 {
2560         if (obj == NULL) {
2561                 return UCL_NULL;
2562         }
2563
2564         return obj->type;
2565 }
2566
2567 ucl_object_t*
2568 ucl_object_fromstring (const char *str)
2569 {
2570         return ucl_object_fromstring_common (str, 0, UCL_STRING_ESCAPE);
2571 }
2572
2573 ucl_object_t *
2574 ucl_object_fromlstring (const char *str, size_t len)
2575 {
2576         return ucl_object_fromstring_common (str, len, UCL_STRING_ESCAPE);
2577 }
2578
2579 ucl_object_t *
2580 ucl_object_fromint (int64_t iv)
2581 {
2582         ucl_object_t *obj;
2583
2584         obj = ucl_object_new ();
2585         if (obj != NULL) {
2586                 obj->type = UCL_INT;
2587                 obj->value.iv = iv;
2588         }
2589
2590         return obj;
2591 }
2592
2593 ucl_object_t *
2594 ucl_object_fromdouble (double dv)
2595 {
2596         ucl_object_t *obj;
2597
2598         obj = ucl_object_new ();
2599         if (obj != NULL) {
2600                 obj->type = UCL_FLOAT;
2601                 obj->value.dv = dv;
2602         }
2603
2604         return obj;
2605 }
2606
2607 ucl_object_t*
2608 ucl_object_frombool (bool bv)
2609 {
2610         ucl_object_t *obj;
2611
2612         obj = ucl_object_new ();
2613         if (obj != NULL) {
2614                 obj->type = UCL_BOOLEAN;
2615                 obj->value.iv = bv;
2616         }
2617
2618         return obj;
2619 }
2620
2621 bool
2622 ucl_array_append (ucl_object_t *top, ucl_object_t *elt)
2623 {
2624         UCL_ARRAY_GET (vec, top);
2625
2626         if (elt == NULL || top == NULL) {
2627                 return false;
2628         }
2629
2630         if (vec == NULL) {
2631                 vec = UCL_ALLOC (sizeof (*vec));
2632
2633                 if (vec == NULL) {
2634                         return false;
2635                 }
2636
2637                 kv_init (*vec);
2638                 top->value.av = (void *)vec;
2639         }
2640
2641         kv_push (ucl_object_t *, *vec, elt);
2642
2643         top->len ++;
2644
2645         return true;
2646 }
2647
2648 bool
2649 ucl_array_prepend (ucl_object_t *top, ucl_object_t *elt)
2650 {
2651         UCL_ARRAY_GET (vec, top);
2652
2653         if (elt == NULL || top == NULL) {
2654                 return false;
2655         }
2656
2657         if (vec == NULL) {
2658                 vec = UCL_ALLOC (sizeof (*vec));
2659                 kv_init (*vec);
2660                 top->value.av = (void *)vec;
2661                 kv_push (ucl_object_t *, *vec, elt);
2662         }
2663         else {
2664                 /* Slow O(n) algorithm */
2665                 kv_prepend (ucl_object_t *, *vec, elt);
2666         }
2667
2668         top->len ++;
2669
2670         return true;
2671 }
2672
2673 bool
2674 ucl_array_merge (ucl_object_t *top, ucl_object_t *elt, bool copy)
2675 {
2676         unsigned i;
2677         ucl_object_t *cp = NULL;
2678         ucl_object_t **obj;
2679
2680         if (elt == NULL || top == NULL || top->type != UCL_ARRAY || elt->type != UCL_ARRAY) {
2681                 return false;
2682         }
2683
2684         if (copy) {
2685                 cp = ucl_object_copy (elt);
2686         }
2687         else {
2688                 cp = ucl_object_ref (elt);
2689         }
2690
2691         UCL_ARRAY_GET (v1, top);
2692         UCL_ARRAY_GET (v2, cp);
2693
2694         kv_concat (ucl_object_t *, *v1, *v2);
2695
2696         for (i = v2->n; i < v1->n; i ++) {
2697                 obj = &kv_A (*v1, i);
2698                 if (*obj == NULL) {
2699                         continue;
2700                 }
2701                 top->len ++;
2702         }
2703
2704         return true;
2705 }
2706
2707 ucl_object_t *
2708 ucl_array_delete (ucl_object_t *top, ucl_object_t *elt)
2709 {
2710         UCL_ARRAY_GET (vec, top);
2711         ucl_object_t *ret = NULL;
2712         unsigned i;
2713
2714         if (vec == NULL) {
2715                 return NULL;
2716         }
2717
2718         for (i = 0; i < vec->n; i ++) {
2719                 if (kv_A (*vec, i) == elt) {
2720                         kv_del (ucl_object_t *, *vec, i);
2721                         ret = elt;
2722                         top->len --;
2723                         break;
2724                 }
2725         }
2726
2727         return ret;
2728 }
2729
2730 const ucl_object_t *
2731 ucl_array_head (const ucl_object_t *top)
2732 {
2733         UCL_ARRAY_GET (vec, top);
2734
2735         if (vec == NULL || top == NULL || top->type != UCL_ARRAY ||
2736                         top->value.av == NULL) {
2737                 return NULL;
2738         }
2739
2740         return (vec->n > 0 ? vec->a[0] : NULL);
2741 }
2742
2743 const ucl_object_t *
2744 ucl_array_tail (const ucl_object_t *top)
2745 {
2746         UCL_ARRAY_GET (vec, top);
2747
2748         if (top == NULL || top->type != UCL_ARRAY || top->value.av == NULL) {
2749                 return NULL;
2750         }
2751
2752         return (vec->n > 0 ? vec->a[vec->n - 1] : NULL);
2753 }
2754
2755 ucl_object_t *
2756 ucl_array_pop_last (ucl_object_t *top)
2757 {
2758         UCL_ARRAY_GET (vec, top);
2759         ucl_object_t **obj, *ret = NULL;
2760
2761         if (vec != NULL && vec->n > 0) {
2762                 obj = &kv_A (*vec, vec->n - 1);
2763                 ret = *obj;
2764                 kv_del (ucl_object_t *, *vec, vec->n - 1);
2765                 top->len --;
2766         }
2767
2768         return ret;
2769 }
2770
2771 ucl_object_t *
2772 ucl_array_pop_first (ucl_object_t *top)
2773 {
2774         UCL_ARRAY_GET (vec, top);
2775         ucl_object_t **obj, *ret = NULL;
2776
2777         if (vec != NULL && vec->n > 0) {
2778                 obj = &kv_A (*vec, 0);
2779                 ret = *obj;
2780                 kv_del (ucl_object_t *, *vec, 0);
2781                 top->len --;
2782         }
2783
2784         return ret;
2785 }
2786
2787 const ucl_object_t *
2788 ucl_array_find_index (const ucl_object_t *top, unsigned int index)
2789 {
2790         UCL_ARRAY_GET (vec, top);
2791
2792         if (vec != NULL && vec->n > 0 && index < vec->n) {
2793                 return kv_A (*vec, index);
2794         }
2795
2796         return NULL;
2797 }
2798
2799 unsigned int
2800 ucl_array_index_of (ucl_object_t *top, ucl_object_t *elt)
2801 {
2802         UCL_ARRAY_GET (vec, top);
2803         unsigned i;
2804
2805         if (vec == NULL) {
2806                 return (unsigned int)(-1);
2807         }
2808
2809         for (i = 0; i < vec->n; i ++) {
2810                 if (kv_A (*vec, i) == elt) {
2811                         return i;
2812                 }
2813         }
2814
2815         return (unsigned int)(-1);
2816 }
2817
2818 ucl_object_t *
2819 ucl_array_replace_index (ucl_object_t *top, ucl_object_t *elt,
2820         unsigned int index)
2821 {
2822         UCL_ARRAY_GET (vec, top);
2823         ucl_object_t *ret = NULL;
2824
2825         if (vec != NULL && vec->n > 0 && index < vec->n) {
2826                 ret = kv_A (*vec, index);
2827                 kv_A (*vec, index) = elt;
2828         }
2829
2830         return ret;
2831 }
2832
2833 ucl_object_t *
2834 ucl_elt_append (ucl_object_t *head, ucl_object_t *elt)
2835 {
2836
2837         if (head == NULL) {
2838                 elt->next = NULL;
2839                 elt->prev = elt;
2840                 head = elt;
2841         }
2842         else {
2843                 elt->prev = head->prev;
2844                 head->prev->next = elt;
2845                 head->prev = elt;
2846                 elt->next = NULL;
2847         }
2848
2849         return head;
2850 }
2851
2852 bool
2853 ucl_object_todouble_safe (const ucl_object_t *obj, double *target)
2854 {
2855         if (obj == NULL || target == NULL) {
2856                 return false;
2857         }
2858         switch (obj->type) {
2859         case UCL_INT:
2860                 *target = obj->value.iv; /* Probaly could cause overflow */
2861                 break;
2862         case UCL_FLOAT:
2863         case UCL_TIME:
2864                 *target = obj->value.dv;
2865                 break;
2866         default:
2867                 return false;
2868         }
2869
2870         return true;
2871 }
2872
2873 double
2874 ucl_object_todouble (const ucl_object_t *obj)
2875 {
2876         double result = 0.;
2877
2878         ucl_object_todouble_safe (obj, &result);
2879         return result;
2880 }
2881
2882 bool
2883 ucl_object_toint_safe (const ucl_object_t *obj, int64_t *target)
2884 {
2885         if (obj == NULL || target == NULL) {
2886                 return false;
2887         }
2888         switch (obj->type) {
2889         case UCL_INT:
2890                 *target = obj->value.iv;
2891                 break;
2892         case UCL_FLOAT:
2893         case UCL_TIME:
2894                 *target = obj->value.dv; /* Loosing of decimal points */
2895                 break;
2896         default:
2897                 return false;
2898         }
2899
2900         return true;
2901 }
2902
2903 int64_t
2904 ucl_object_toint (const ucl_object_t *obj)
2905 {
2906         int64_t result = 0;
2907
2908         ucl_object_toint_safe (obj, &result);
2909         return result;
2910 }
2911
2912 bool
2913 ucl_object_toboolean_safe (const ucl_object_t *obj, bool *target)
2914 {
2915         if (obj == NULL || target == NULL) {
2916                 return false;
2917         }
2918         switch (obj->type) {
2919         case UCL_BOOLEAN:
2920                 *target = (obj->value.iv == true);
2921                 break;
2922         default:
2923                 return false;
2924         }
2925
2926         return true;
2927 }
2928
2929 bool
2930 ucl_object_toboolean (const ucl_object_t *obj)
2931 {
2932         bool result = false;
2933
2934         ucl_object_toboolean_safe (obj, &result);
2935         return result;
2936 }
2937
2938 bool
2939 ucl_object_tostring_safe (const ucl_object_t *obj, const char **target)
2940 {
2941         if (obj == NULL || target == NULL) {
2942                 return false;
2943         }
2944
2945         switch (obj->type) {
2946         case UCL_STRING:
2947                 if (!(obj->flags & UCL_OBJECT_BINARY)) {
2948                         *target = ucl_copy_value_trash (obj);
2949                 }
2950                 break;
2951         default:
2952                 return false;
2953         }
2954
2955         return true;
2956 }
2957
2958 const char *
2959 ucl_object_tostring (const ucl_object_t *obj)
2960 {
2961         const char *result = NULL;
2962
2963         ucl_object_tostring_safe (obj, &result);
2964         return result;
2965 }
2966
2967 const char *
2968 ucl_object_tostring_forced (const ucl_object_t *obj)
2969 {
2970         /* TODO: For binary strings we might encode string here */
2971         if (!(obj->flags & UCL_OBJECT_BINARY)) {
2972                 return ucl_copy_value_trash (obj);
2973         }
2974
2975         return NULL;
2976 }
2977
2978 bool
2979 ucl_object_tolstring_safe (const ucl_object_t *obj, const char **target, size_t *tlen)
2980 {
2981         if (obj == NULL || target == NULL) {
2982                 return false;
2983         }
2984         switch (obj->type) {
2985         case UCL_STRING:
2986                 *target = obj->value.sv;
2987                 if (tlen != NULL) {
2988                         *tlen = obj->len;
2989                 }
2990                 break;
2991         default:
2992                 return false;
2993         }
2994
2995         return true;
2996 }
2997
2998 const char *
2999 ucl_object_tolstring (const ucl_object_t *obj, size_t *tlen)
3000 {
3001         const char *result = NULL;
3002
3003         ucl_object_tolstring_safe (obj, &result, tlen);
3004         return result;
3005 }
3006
3007 const char *
3008 ucl_object_key (const ucl_object_t *obj)
3009 {
3010         return ucl_copy_key_trash (obj);
3011 }
3012
3013 const char *
3014 ucl_object_keyl (const ucl_object_t *obj, size_t *len)
3015 {
3016         if (len == NULL || obj == NULL) {
3017                 return NULL;
3018         }
3019         *len = obj->keylen;
3020         return obj->key;
3021 }
3022
3023 ucl_object_t *
3024 ucl_object_ref (const ucl_object_t *obj)
3025 {
3026         ucl_object_t *res = NULL;
3027
3028         if (obj != NULL) {
3029                 if (obj->flags & UCL_OBJECT_EPHEMERAL) {
3030                         /*
3031                          * Use deep copy for ephemeral objects, note that its refcount
3032                          * is NOT increased, since ephemeral objects does not need refcount
3033                          * at all
3034                          */
3035                         res = ucl_object_copy (obj);
3036                 }
3037                 else {
3038                         res = __DECONST (ucl_object_t *, obj);
3039 #ifdef HAVE_ATOMIC_BUILTINS
3040                         (void)__sync_add_and_fetch (&res->ref, 1);
3041 #else
3042                         res->ref ++;
3043 #endif
3044                 }
3045         }
3046         return res;
3047 }
3048
3049 static ucl_object_t *
3050 ucl_object_copy_internal (const ucl_object_t *other, bool allow_array)
3051 {
3052
3053         ucl_object_t *new;
3054         ucl_object_iter_t it = NULL;
3055         const ucl_object_t *cur;
3056
3057         new = malloc (sizeof (*new));
3058
3059         if (new != NULL) {
3060                 memcpy (new, other, sizeof (*new));
3061                 if (other->flags & UCL_OBJECT_EPHEMERAL) {
3062                         /* Copied object is always non ephemeral */
3063                         new->flags &= ~UCL_OBJECT_EPHEMERAL;
3064                 }
3065                 new->ref = 1;
3066                 /* Unlink from others */
3067                 new->next = NULL;
3068                 new->prev = new;
3069
3070                 /* deep copy of values stored */
3071                 if (other->trash_stack[UCL_TRASH_KEY] != NULL) {
3072                         new->trash_stack[UCL_TRASH_KEY] =
3073                                         strdup (other->trash_stack[UCL_TRASH_KEY]);
3074                         if (other->key == (const char *)other->trash_stack[UCL_TRASH_KEY]) {
3075                                 new->key = new->trash_stack[UCL_TRASH_KEY];
3076                         }
3077                 }
3078                 if (other->trash_stack[UCL_TRASH_VALUE] != NULL) {
3079                         new->trash_stack[UCL_TRASH_VALUE] =
3080                                         strdup (other->trash_stack[UCL_TRASH_VALUE]);
3081                         if (new->type == UCL_STRING) {
3082                                 new->value.sv = new->trash_stack[UCL_TRASH_VALUE];
3083                         }
3084                 }
3085
3086                 if (other->type == UCL_ARRAY || other->type == UCL_OBJECT) {
3087                         /* reset old value */
3088                         memset (&new->value, 0, sizeof (new->value));
3089
3090                         while ((cur = ucl_iterate_object (other, &it, true)) != NULL) {
3091                                 if (other->type == UCL_ARRAY) {
3092                                         ucl_array_append (new, ucl_object_copy_internal (cur, false));
3093                                 }
3094                                 else {
3095                                         ucl_object_t *cp = ucl_object_copy_internal (cur, true);
3096                                         if (cp != NULL) {
3097                                                 ucl_object_insert_key (new, cp, cp->key, cp->keylen,
3098                                                                 false);
3099                                         }
3100                                 }
3101                         }
3102                 }
3103                 else if (allow_array && other->next != NULL) {
3104                         LL_FOREACH (other->next, cur) {
3105                                 ucl_object_t *cp = ucl_object_copy_internal (cur, false);
3106                                 if (cp != NULL) {
3107                                         DL_APPEND (new, cp);
3108                                 }
3109                         }
3110                 }
3111         }
3112
3113         return new;
3114 }
3115
3116 ucl_object_t *
3117 ucl_object_copy (const ucl_object_t *other)
3118 {
3119         return ucl_object_copy_internal (other, true);
3120 }
3121
3122 void
3123 ucl_object_unref (ucl_object_t *obj)
3124 {
3125         if (obj != NULL) {
3126 #ifdef HAVE_ATOMIC_BUILTINS
3127                 unsigned int rc = __sync_sub_and_fetch (&obj->ref, 1);
3128                 if (rc == 0) {
3129 #else
3130                 if (--obj->ref == 0) {
3131 #endif
3132                         ucl_object_free_internal (obj, true, ucl_object_dtor_unref);
3133                 }
3134         }
3135 }
3136
3137 int
3138 ucl_object_compare (const ucl_object_t *o1, const ucl_object_t *o2)
3139 {
3140         const ucl_object_t *it1, *it2;
3141         ucl_object_iter_t iter = NULL;
3142         int ret = 0;
3143
3144         if (o1->type != o2->type) {
3145                 return (o1->type) - (o2->type);
3146         }
3147
3148         switch (o1->type) {
3149         case UCL_STRING:
3150                 if (o1->len == o2->len && o1->len > 0) {
3151                         ret = strcmp (ucl_object_tostring(o1), ucl_object_tostring(o2));
3152                 }
3153                 else {
3154                         ret = o1->len - o2->len;
3155                 }
3156                 break;
3157         case UCL_FLOAT:
3158         case UCL_INT:
3159         case UCL_TIME:
3160                 ret = ucl_object_todouble (o1) - ucl_object_todouble (o2);
3161                 break;
3162         case UCL_BOOLEAN:
3163                 ret = ucl_object_toboolean (o1) - ucl_object_toboolean (o2);
3164                 break;
3165         case UCL_ARRAY:
3166                 if (o1->len == o2->len && o1->len > 0) {
3167                         UCL_ARRAY_GET (vec1, o1);
3168                         UCL_ARRAY_GET (vec2, o2);
3169                         unsigned i;
3170
3171                         /* Compare all elements in both arrays */
3172                         for (i = 0; i < vec1->n; i ++) {
3173                                 it1 = kv_A (*vec1, i);
3174                                 it2 = kv_A (*vec2, i);
3175
3176                                 if (it1 == NULL && it2 != NULL) {
3177                                         return -1;
3178                                 }
3179                                 else if (it2 == NULL && it1 != NULL) {
3180                                         return 1;
3181                                 }
3182                                 else if (it1 != NULL && it2 != NULL) {
3183                                         ret = ucl_object_compare (it1, it2);
3184                                         if (ret != 0) {
3185                                                 break;
3186                                         }
3187                                 }
3188                         }
3189                 }
3190                 else {
3191                         ret = o1->len - o2->len;
3192                 }
3193                 break;
3194         case UCL_OBJECT:
3195                 if (o1->len == o2->len && o1->len > 0) {
3196                         while ((it1 = ucl_iterate_object (o1, &iter, true)) != NULL) {
3197                                 it2 = ucl_object_find_key (o2, ucl_object_key (it1));
3198                                 if (it2 == NULL) {
3199                                         ret = 1;
3200                                         break;
3201                                 }
3202                                 ret = ucl_object_compare (it1, it2);
3203                                 if (ret != 0) {
3204                                         break;
3205                                 }
3206                         }
3207                 }
3208                 else {
3209                         ret = o1->len - o2->len;
3210                 }
3211                 break;
3212         default:
3213                 ret = 0;
3214                 break;
3215         }
3216
3217         return ret;
3218 }
3219
3220 void
3221 ucl_object_array_sort (ucl_object_t *ar,
3222                 int (*cmp)(const ucl_object_t **o1, const ucl_object_t **o2))
3223 {
3224         UCL_ARRAY_GET (vec, ar);
3225
3226         if (cmp == NULL || ar == NULL || ar->type != UCL_ARRAY) {
3227                 return;
3228         }
3229
3230         qsort (vec->a, vec->n, sizeof (ucl_object_t *),
3231                         (int (*)(const void *, const void *))cmp);
3232 }
3233
3234 #define PRIOBITS 4
3235
3236 unsigned int
3237 ucl_object_get_priority (const ucl_object_t *obj)
3238 {
3239         if (obj == NULL) {
3240                 return 0;
3241         }
3242
3243         return (obj->flags >> ((sizeof (obj->flags) * NBBY) - PRIOBITS));
3244 }
3245
3246 void
3247 ucl_object_set_priority (ucl_object_t *obj,
3248                 unsigned int priority)
3249 {
3250         if (obj != NULL) {
3251                 priority &= (0x1 << PRIOBITS) - 1;
3252                 priority <<= ((sizeof (obj->flags) * NBBY) - PRIOBITS);
3253                 priority |= obj->flags & ((1 << ((sizeof (obj->flags) * NBBY) -
3254                                 PRIOBITS)) - 1);
3255                 obj->flags = priority;
3256         }
3257 }