1: /*************************************************************************
2: * (C) 2008 AITNET ltd - Sofia/Bulgaria - <misho@aitbg.com>
3: * by Michael Pounov <misho@openbsd-bg.org>
4: *
5: * $Author: misho $
6: * $Id: parse.c,v 1.18 2019/11/21 14:30:02 misho Exp $
7: *
8: **************************************************************************
9: The ELWIX and AITNET software is distributed under the following
10: terms:
11:
12: All of the documentation and software included in the ELWIX and AITNET
13: Releases is copyrighted by ELWIX - Sofia/Bulgaria <info@elwix.org>
14:
15: Copyright 2004 - 2018
16: by Michael Pounov <misho@elwix.org>. All rights reserved.
17:
18: Redistribution and use in source and binary forms, with or without
19: modification, are permitted provided that the following conditions
20: are met:
21: 1. Redistributions of source code must retain the above copyright
22: notice, this list of conditions and the following disclaimer.
23: 2. Redistributions in binary form must reproduce the above copyright
24: notice, this list of conditions and the following disclaimer in the
25: documentation and/or other materials provided with the distribution.
26: 3. All advertising materials mentioning features or use of this software
27: must display the following acknowledgement:
28: This product includes software developed by Michael Pounov <misho@elwix.org>
29: ELWIX - Embedded LightWeight unIX and its contributors.
30: 4. Neither the name of AITNET nor the names of its contributors
31: may be used to endorse or promote products derived from this software
32: without specific prior written permission.
33:
34: THIS SOFTWARE IS PROVIDED BY AITNET AND CONTRIBUTORS ``AS IS'' AND
35: ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
36: IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
37: ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
38: FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
39: DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
40: OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
41: HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
42: LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
43: OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
44: SUCH DAMAGE.
45: */
46: #include "global.h"
47:
48:
49: /*
50: * cfgReadConfig() - Read file and add new item at config root
51: *
52: * @f = File resource
53: * @cfg = Config root
54: * return: -1 error or 0 ok
55: */
56: int
57: cfgReadConfig(FILE *f, cfg_root_t * __restrict cfg)
58: {
59: char line[BUFSIZ], origin[BUFSIZ];
60: struct tagCfg *av = NULL;
61: int flg = 0;
62: char *psAttr, *psVal, szSection[STRSIZ] = { 0 };
63: FILE *ff;
64:
65: if (!f || !cfg) {
66: cfg_SetErr(EINVAL, "Invalid parameter(s)");
67: return -1;
68: }
69:
70: while (!feof(f)) {
71: memset(line, 0, sizeof line);
72: fgets(line, sizeof line - 1, f);
73: #ifdef SUPPORT_USER_EOF
74: /* check for user end-of-file */
75: if (line[0] == '.' && line[1] == '\n')
76: break;
77: #endif
78: if (!(psAttr = strpbrk(line, "\r\n"))) {
79: /* skip line, too long */
80: continue;
81: } else {
82: *psAttr = 0;
83: strlcpy(origin, line, sizeof origin);
84: str_Trim(line);
85: }
86:
87: if (flg) {
88: /* continues line */
89: if (!av)
90: continue;
91: else
92: psAttr = line + strlen(line) - 1;
93: if (*psAttr == '\\')
94: *psAttr = 0;
95: else
96: flg = 0;
97: /* concat line to value */
98: AIT_SET_STRCAT(&av->cfg_val, line);
99: if (!flg && AIT_ADDR(&av->cfg_val))
100: str_Unquot((char*) AIT_GET_STR(&av->cfg_val));
101:
102: /* read include file */
103: if (!flg && AIT_ADDR(&av->cfg_val) &&
104: *AIT_GET_STR(&av->cfg_attr) == '%' &&
105: !strcmp(AIT_GET_STR(&av->cfg_attr), "%include")) {
106: ff = fopen(AIT_GET_STR(&av->cfg_val), "r");
107: if (ff) {
108: cfgReadConfig(ff, cfg);
109: fclose(ff);
110: } else
111: EDEBUG(7, "Error:: Can't open %s file",
112: AIT_GET_STR(&av->cfg_val));
113: }
114: continue;
115: }
116:
117: /* *NEW PAIR* alloc new pair element */
118: av = e_malloc(sizeof(struct tagCfg));
119: if (!av) {
120: cfg_SetErr(elwix_GetErrno(), "%s", elwix_GetError());
121: return -1;
122: } else {
123: memset(av, 0, sizeof(struct tagCfg));
124: CFG_RC_LOCK(cfg);
125: TAILQ_INSERT_TAIL(cfg, av, cfg_next);
126: CFG_RC_UNLOCK(cfg);
127: }
128:
129: /* check for comment or empty line */
130: if (!*line || *line == '#' || *line == ';') {
131: AIT_SET_STR(&av->cfg_val, line);
132: continue;
133: }
134:
135: /* check for continues line */
136: psAttr = line + (*line ? strlen(line) : 1) - 1;
137: if (*psAttr == '\\') {
138: *psAttr = 0;
139: flg = 1;
140: }
141:
142: /* section */
143: if (*line == '[') {
144: psAttr = line + strlen(line) - 1;
145: if (*psAttr == ']') {
146: *psAttr = 0;
147: flg = 0;
148: strlcpy(szSection, line + 1, sizeof szSection);
149: AIT_SET_STR(&av->cfg_sec, line);
150: } else
151: EDEBUG(7, "Ignore section '%s' ... not found ']'", line);
152: continue;
153: }
154: /* parse pair */
155: if (!(psAttr = strchr(line, '='))) {
156: AIT_SET_STR(&av->cfg_val, origin);
157: EDEBUG(7, "Ignore a/v '%s' ... not found '='", line);
158: continue;
159: } else {
160: *psAttr = 0;
161: psVal = psAttr + 1;
162: psAttr = line;
163: }
164:
165: /* if exists, added section name to element */
166: if (*szSection) {
167: AIT_SET_STR(&av->cfg_sec, szSection);
168: AIT_KEY(&av->cfg_sec) = crcFletcher16(AIT_GET_LIKE(&av->cfg_sec, u_short*),
169: E_ALIGN(AIT_LEN(&av->cfg_sec) - 1, 2) / 2);
170: }
171:
172: str_RTrim(psAttr);
173: str_LTrim(psVal);
174: if (!flg)
175: str_Unquot(psVal);
176: AIT_SET_STR(&av->cfg_val, psVal);
177: AIT_SET_STR(&av->cfg_attr, psAttr);
178: AIT_KEY(&av->cfg_attr) = crcFletcher16(AIT_GET_LIKE(&av->cfg_attr, u_short*),
179: E_ALIGN(AIT_LEN(&av->cfg_attr) - 1, 2) / 2);
180:
181: CFG_RC_LOCK(cfg);
182: RB_INSERT(tagRC, cfg, av);
183: CFG_RC_UNLOCK(cfg);
184:
185: /* read include file */
186: if (!flg && *AIT_GET_STR(&av->cfg_attr) == '%' &&
187: !strcmp(AIT_GET_STR(&av->cfg_attr), "%include")) {
188: ff = fopen(AIT_GET_STR(&av->cfg_val), "r");
189: if (ff) {
190: cfgReadConfig(ff, cfg);
191: fclose(ff);
192: } else
193: EDEBUG(7, "Error:: Can't open %s file",
194: AIT_GET_STR(&av->cfg_val));
195: }
196: }
197:
198: return 0;
199: }
200:
201: /*
202: * cfgWriteConfig() - Write config from memory
203: *
204: * @f = File handle
205: * @cfg = Config root
206: * @whitespace = Additional whitespace characters to file
207: * return: -1 error or 0 ok
208: */
209: int
210: cfgWriteConfig(FILE *f, cfg_root_t * __restrict cfg, int whitespace)
211: {
212: struct tagCfg *av;
213: char line[BUFSIZ] = { 0 }, szSection[STRSIZ] = { [0 ... STRSIZ - 1] = 0 };
214:
215: if (!f || !cfg) {
216: cfg_SetErr(EINVAL, "Invalid parameter(s)");
217: return -1;
218: }
219:
220: CFG_RC_LOCK(cfg);
221: RB_FOREACH(av, tagRC, cfg) {
222: /* empty lines or comment */
223: if (AIT_ISEMPTY(&av->cfg_attr)) {
224: if (AIT_ISEMPTY(&av->cfg_val))
225: continue;
226: strlcpy(line, AIT_GET_STR(&av->cfg_val), sizeof line);
227: goto skip_sec;
228: }
229:
230: /* section [] */
231: if (!AIT_ISEMPTY(&av->cfg_sec) && AIT_ADDR(&av->cfg_sec) &&
232: strcmp(AIT_GET_STRZ(&av->cfg_sec), szSection)) {
233: strlcpy(szSection, AIT_GET_STR(&av->cfg_sec), sizeof szSection);
234: if (!cfg_Write(f, "\n[%s]\n", AIT_GET_STR(&av->cfg_sec))) {
235: LOGERR;
236: CFG_RC_UNLOCK(cfg);
237: return -1;
238: }
239: } else if (AIT_ISEMPTY(&av->cfg_sec) && *szSection) {
240: memset(szSection, 0, sizeof szSection);
241: if (!cfg_Write(f, "\n[]\n")) {
242: LOGERR;
243: CFG_RC_UNLOCK(cfg);
244: return -1;
245: }
246: }
247:
248: /* build line */
249: memset(line, 0, sizeof line);
250: if (!AIT_ISEMPTY(&av->cfg_attr) && AIT_TYPE(&av->cfg_attr) == string) {
251: strlcpy(line, AIT_GET_STRZ(&av->cfg_attr), sizeof line);
252: if (whitespace)
253: strlcat(line, " = ", sizeof line);
254: else
255: strlcat(line, "=", sizeof line);
256: }
257: if (!AIT_ISEMPTY(&av->cfg_val) && AIT_TYPE(&av->cfg_val) == string)
258: strlcat(line, AIT_GET_STRZ(&av->cfg_val), sizeof line);
259: skip_sec:
260: /* write */
261: if (!cfg_Write(f, "%s\n", line)) {
262: LOGERR;
263: CFG_RC_UNLOCK(cfg);
264: return -1;
265: }
266: }
267: CFG_RC_UNLOCK(cfg);
268:
269: return 0;
270: }
271:
272: /*
273: * cfgConcatConfig() - Concat two configs into one
274: *
275: * @cfg = Config root
276: * @add_cfg = Concated config will be destroy after merge
277: * return: -1 error or 0 ok
278: */
279: int
280: cfgConcatConfig(cfg_root_t * __restrict cfg, cfg_root_t * __restrict add_cfg)
281: {
282: struct tagCfg *item;
283:
284: if (!cfg || !add_cfg)
285: return -1;
286:
287: CFG_RC_LOCK(add_cfg);
288: CFG_RC_LOCK(cfg);
289:
290: /* concat lists & red-black trees */
291: TAILQ_FOREACH(item, add_cfg, cfg_next) {
292: TAILQ_INSERT_TAIL(cfg, item, cfg_next);
293: RB_INSERT(tagRC, cfg, item);
294: }
295:
296: CFG_RC_UNLOCK(cfg);
297:
298: TAILQ_INIT(add_cfg);
299: RB_INIT(add_cfg);
300: CFG_RC_UNLOCK(add_cfg);
301: pthread_mutex_destroy(&add_cfg->rc_mtx);
302: return 0;
303: }
304:
305: /*
306: * cfgMergeConfig() - Marge two list in one cfg and destroy add_cfg
307: *
308: * @cfg = Config root of main list
309: * @add_cfg = Merged config will be destroy after merge
310: * return: -1 error or 0 ok
311: */
312: int
313: cfgMergeConfig(cfg_root_t * __restrict cfg, cfg_root_t * __restrict add_cfg)
314: {
315: struct tagCfg *item, *merge, *add_next, *next;
316: int flg;
317:
318: if (!cfg || !add_cfg)
319: return -1;
320:
321: CFG_RC_LOCK(add_cfg);
322: CFG_RC_LOCK(cfg);
323:
324: /* merge lists */
325: TAILQ_FOREACH_SAFE(item, add_cfg, cfg_next, add_next) {
326: flg = 0;
327: TAILQ_FOREACH_SAFE(merge, cfg, cfg_next, next) {
328: if (AIT_ISEMPTY(&merge->cfg_sec) && AIT_ISEMPTY(&item->cfg_sec)) {
329: flg = 1;
330: break;
331: }
332: if (!AIT_ISEMPTY(&merge->cfg_sec) && !AIT_ISEMPTY(&item->cfg_sec) &&
333: AIT_ADDR(&merge->cfg_sec) && AIT_ADDR(&item->cfg_sec) &&
334: !strcmp(AIT_GET_STR(&merge->cfg_sec), AIT_GET_STR(&item->cfg_sec))) {
335: flg = 1;
336: break;
337: }
338: }
339:
340: if (!flg)
341: TAILQ_INSERT_TAIL(cfg, item, cfg_next);
342: else
343: TAILQ_INSERT_AFTER(cfg, merge, item, cfg_next);
344: RB_INSERT(tagRC, cfg, item);
345: }
346:
347: CFG_RC_UNLOCK(cfg);
348:
349: TAILQ_INIT(add_cfg);
350: RB_INIT(add_cfg);
351: CFG_RC_UNLOCK(add_cfg);
352: pthread_mutex_destroy(&add_cfg->rc_mtx);
353: return 0;
354: }
355:
356: /*
357: * cfgReadLines() - Read custom lines and add new item at config root
358: *
359: * @f = File resource
360: * @delim = Custom delimiter, if =NULL default is '='
361: * @end = Custom user end of file, if =NULL default is EOF
362: * @cfg = Config root
363: * return: -1 error or 0 ok
364: */
365: int
366: cfgReadLines(FILE *f, const char *delim, const char *end, cfg_root_t * __restrict cfg)
367: {
368: char line[BUFSIZ];
369: struct tagCfg *d, *av = NULL;
370: char *p, *psSec, *psAttr, *psVal;
371:
372: if (!cfg)
373: return -1;
374: if (!delim)
375: delim = ATR_LINES_DELIM;
376:
377: while (!feof(f)) {
378: psSec = psAttr = psVal = NULL;
379: memset(line, 0, sizeof line);
380: fgets(line, sizeof line - 1, f);
381: /* check for user end-of-file */
382: if (strspn(line, end))
383: break;
384:
385: if (!(psAttr = strpbrk(line, "\r\n"))) {
386: /* skip line, too long */
387: continue;
388: } else {
389: *psAttr = 0;
390: str_Trim(line);
391: if (!*line)
392: continue;
393: }
394:
395: if (!av_MakeExt(line, delim, &p, &psVal))
396: continue;
397: else {
398: str_RTrim(p);
399: str_LTrim(psVal);
400: }
401: if (!av_MakeExt(p, SEC_LINES_DELIM, &psSec, &psAttr))
402: psAttr = p;
403:
404: /* *NEW PAIR* alloc new pair element */
405: av = e_malloc(sizeof(struct tagCfg));
406: if (!av) {
407: LOGERR;
408: return -1;
409: } else
410: memset(av, 0, sizeof(struct tagCfg));
411:
412: if (psSec) {
413: AIT_SET_STR(&av->cfg_sec, psSec);
414: AIT_KEY(&av->cfg_sec) = crcFletcher16(AIT_GET_LIKE(&av->cfg_sec, u_short*),
415: E_ALIGN(AIT_LEN(&av->cfg_sec) - 1, 2) / 2);
416: }
417: if (psVal)
418: AIT_SET_STR(&av->cfg_val, psVal);
419: AIT_SET_STR(&av->cfg_attr, psAttr);
420: AIT_KEY(&av->cfg_attr) = crcFletcher16(AIT_GET_LIKE(&av->cfg_attr, u_short*),
421: E_ALIGN(AIT_LEN(&av->cfg_attr) - 1, 2) / 2);
422:
423: CFG_RC_LOCK(cfg);
424: /* find & delete duplicates */
425: if ((d = RB_FIND(tagRC, cfg, av))) {
426: RB_REMOVE(tagRC, cfg, d);
427: TAILQ_REMOVE(cfg, d, cfg_next);
428:
429: AIT_FREE_VAL(&d->cfg_val);
430: AIT_FREE_VAL(&d->cfg_attr);
431: AIT_FREE_VAL(&d->cfg_sec);
432: e_free(d);
433: }
434:
435: TAILQ_INSERT_TAIL(cfg, av, cfg_next);
436: RB_INSERT(tagRC, cfg, av);
437: CFG_RC_UNLOCK(cfg);
438: }
439:
440: return 0;
441: }
442:
443: /*
444: * cfgWriteLines() - Write custom lines and export data to variable
445: *
446: * @f = File resource
447: * @delim = Custom delimiter, if =NULL default is '='
448: * @eol = End of line string, if =NULL default is "\n"
449: * @section = Export only section, if =NULL default is all
450: * @cfg = Config root
451: * return: =NULL error or !=NULL exported data, must be free after use with ait_freeVar()
452: */
453: ait_val_t *
454: cfgWriteLines(FILE *f, const char *delim, const char *eol, const char *section, cfg_root_t * __restrict cfg)
455: {
456: ait_val_t *v = NULL;
457: struct tagCfg *av;
458:
459: if (!cfg)
460: return NULL;
461: if (!delim)
462: delim = ATR_LINES_DELIM;
463: if (!eol)
464: eol = EOL_LINES_DELIM;
465: if (!(v = ait_allocVar())) {
466: cfg_SetErr(elwix_GetErrno(), "%s", elwix_GetError());
467: return NULL;
468: } else
469: AIT_INIT_VAL2(v, string);
470:
471: TAILQ_FOREACH(av, cfg, cfg_next) {
472: if (section) {
473: if (!AIT_ISEMPTY(&av->cfg_sec) && *section)
474: continue;
475: if (strcmp(section, AIT_GET_STRZ(&av->cfg_sec)))
476: continue;
477: }
478:
479: if (!AIT_ISEMPTY(&av->cfg_sec)) {
480: AIT_SET_STRCAT(v, AIT_GET_STR(&av->cfg_sec));
481: AIT_SET_STRCAT(v, SEC_LINES_DELIM);
482: }
483: if (!AIT_ISEMPTY(&av->cfg_attr)) {
484: AIT_SET_STRCAT(v, AIT_GET_STR(&av->cfg_attr));
485: AIT_SET_STRCAT(v, delim);
486: }
487: if (!AIT_ISEMPTY(&av->cfg_val))
488: AIT_SET_STRCAT(v, AIT_GET_STR(&av->cfg_val));
489: AIT_SET_STRCAT(v, eol);
490: }
491:
492: if (f)
493: fputs(AIT_GET_STR(v), f);
494: return v;
495: }
FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>