xref: /freebsd/contrib/libucl/tests/test_basic.c (revision dda5b39711dab90ae1c5624bdd6ff7453177df31)
1 /* Copyright (c) 2013, Vsevolod Stakhov
2  * All rights reserved.
3  *
4  * Redistribution and use in source and binary forms, with or without
5  * modification, are permitted provided that the following conditions are met:
6  *       * Redistributions of source code must retain the above copyright
7  *         notice, this list of conditions and the following disclaimer.
8  *       * Redistributions in binary form must reproduce the above copyright
9  *         notice, this list of conditions and the following disclaimer in the
10  *         documentation and/or other materials provided with the distribution.
11  *
12  * THIS SOFTWARE IS PROVIDED ''AS IS'' AND ANY
13  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
14  * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
15  * DISCLAIMED. IN NO EVENT SHALL AUTHOR BE LIABLE FOR ANY
16  * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
17  * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
18  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
19  * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
20  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
21  * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
22  */
23 
24 #include <stdio.h>
25 #include <errno.h>
26 #include <unistd.h>
27 #include "ucl.h"
28 
29 int
30 main (int argc, char **argv)
31 {
32 	char inbuf[8192], *test_in = NULL;
33 	struct ucl_parser *parser = NULL, *parser2 = NULL;
34 	ucl_object_t *obj;
35 	FILE *in, *out;
36 	unsigned char *emitted = NULL;
37 	const char *fname_in = NULL, *fname_out = NULL;
38 	int ret = 0, inlen, opt, json = 0;
39 
40 	while ((opt = getopt(argc, argv, "j")) != -1) {
41 		switch (opt) {
42 		case 'j':
43 			json = 1;
44 			break;
45 		default: /* '?' */
46 			fprintf (stderr, "Usage: %s [-j] [in] [out]\n",
47 					argv[0]);
48 			exit (EXIT_FAILURE);
49 		}
50 	}
51 
52 	argc -= optind;
53 	argv += optind;
54 
55 	switch (argc) {
56 	case 1:
57 		fname_in = argv[0];
58 		break;
59 	case 2:
60 		fname_in = argv[0];
61 		fname_out = argv[1];
62 		break;
63 	}
64 
65 	if (fname_in != NULL) {
66 		in = fopen (fname_in, "r");
67 		if (in == NULL) {
68 			exit (-errno);
69 		}
70 	}
71 	else {
72 		in = stdin;
73 	}
74 	parser = ucl_parser_new (UCL_PARSER_KEY_LOWERCASE);
75 	ucl_parser_register_variable (parser, "ABI", "unknown");
76 
77 	if (fname_in != NULL) {
78 		ucl_parser_set_filevars (parser, fname_in, true);
79 	}
80 
81 	while (!feof (in)) {
82 		memset (inbuf, 0, sizeof (inbuf));
83 		(void)fread (inbuf, sizeof (inbuf) - 1, 1, in);
84 		inlen = strlen (inbuf);
85 		test_in = malloc (inlen);
86 		memcpy (test_in, inbuf, inlen);
87 		ucl_parser_add_chunk (parser, test_in, inlen);
88 	}
89 	fclose (in);
90 
91 	if (fname_out != NULL) {
92 		out = fopen (fname_out, "w");
93 		if (out == NULL) {
94 			exit (-errno);
95 		}
96 	}
97 	else {
98 		out = stdout;
99 	}
100 	if (ucl_parser_get_error(parser) != NULL) {
101 		fprintf (out, "Error occurred: %s\n", ucl_parser_get_error(parser));
102 		ret = 1;
103 		goto end;
104 	}
105 	obj = ucl_parser_get_object (parser);
106 	if (json) {
107 		emitted = ucl_object_emit (obj, UCL_EMIT_JSON);
108 	}
109 	else {
110 		emitted = ucl_object_emit (obj, UCL_EMIT_CONFIG);
111 	}
112 	ucl_parser_free (parser);
113 	ucl_object_unref (obj);
114 	parser2 = ucl_parser_new (UCL_PARSER_KEY_LOWERCASE);
115 	ucl_parser_add_chunk (parser2, emitted, strlen (emitted));
116 
117 	if (ucl_parser_get_error(parser2) != NULL) {
118 		fprintf (out, "Error occurred: %s\n", ucl_parser_get_error(parser2));
119 		fprintf (out, "%s\n", emitted);
120 		ret = 1;
121 		goto end;
122 	}
123 	if (emitted != NULL) {
124 		free (emitted);
125 	}
126 	obj = ucl_parser_get_object (parser2);
127 	if (json) {
128 		emitted = ucl_object_emit (obj, UCL_EMIT_JSON);
129 	}
130 	else {
131 		emitted = ucl_object_emit (obj, UCL_EMIT_CONFIG);
132 	}
133 
134 	fprintf (out, "%s\n", emitted);
135 	ucl_object_unref (obj);
136 
137 end:
138 	if (emitted != NULL) {
139 		free (emitted);
140 	}
141 	if (parser2 != NULL) {
142 		ucl_parser_free (parser2);
143 	}
144 	if (test_in != NULL) {
145 		free (test_in);
146 	}
147 
148 	fclose (out);
149 
150 	return ret;
151 }
152