1*4882a593Smuzhiyun /**
2*4882a593Smuzhiyun * Copyright (c) 2016 rxi
3*4882a593Smuzhiyun *
4*4882a593Smuzhiyun * Permission is hereby granted, free of charge, to any person obtaining a copy
5*4882a593Smuzhiyun * of this software and associated documentation files (the "Software"), to deal
6*4882a593Smuzhiyun * in the Software without restriction, including without limitation the rights
7*4882a593Smuzhiyun * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
8*4882a593Smuzhiyun * copies of the Software, and to permit persons to whom the Software is
9*4882a593Smuzhiyun * furnished to do so, subject to the following conditions:
10*4882a593Smuzhiyun *
11*4882a593Smuzhiyun * The above copyright notice and this permission notice shall be included in
12*4882a593Smuzhiyun * all copies or substantial portions of the Software.
13*4882a593Smuzhiyun *
14*4882a593Smuzhiyun * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
15*4882a593Smuzhiyun * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
16*4882a593Smuzhiyun * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
17*4882a593Smuzhiyun * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
18*4882a593Smuzhiyun * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
19*4882a593Smuzhiyun * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
20*4882a593Smuzhiyun * SOFTWARE.
21*4882a593Smuzhiyun */
22*4882a593Smuzhiyun
23*4882a593Smuzhiyun #include <stdio.h>
24*4882a593Smuzhiyun #include <stdlib.h>
25*4882a593Smuzhiyun #include <string.h>
26*4882a593Smuzhiyun #include <ctype.h>
27*4882a593Smuzhiyun
28*4882a593Smuzhiyun #include "rkaiq_ini.h"
29*4882a593Smuzhiyun
30*4882a593Smuzhiyun struct ini_t {
31*4882a593Smuzhiyun char *data;
32*4882a593Smuzhiyun char *end;
33*4882a593Smuzhiyun };
34*4882a593Smuzhiyun
35*4882a593Smuzhiyun
36*4882a593Smuzhiyun /* Case insensitive string compare */
strcmpci(const char * a,const char * b)37*4882a593Smuzhiyun static int strcmpci(const char *a, const char *b) {
38*4882a593Smuzhiyun for (;;) {
39*4882a593Smuzhiyun int d = tolower(*a) - tolower(*b);
40*4882a593Smuzhiyun if (d != 0 || !*a) {
41*4882a593Smuzhiyun return d;
42*4882a593Smuzhiyun }
43*4882a593Smuzhiyun a++, b++;
44*4882a593Smuzhiyun }
45*4882a593Smuzhiyun }
46*4882a593Smuzhiyun
47*4882a593Smuzhiyun /* Returns the next string in the split data */
next(ini_t * ini,char * p)48*4882a593Smuzhiyun static char* next(ini_t *ini, char *p) {
49*4882a593Smuzhiyun p += strlen(p);
50*4882a593Smuzhiyun while (p < ini->end && *p == '\0') {
51*4882a593Smuzhiyun p++;
52*4882a593Smuzhiyun }
53*4882a593Smuzhiyun return p;
54*4882a593Smuzhiyun }
55*4882a593Smuzhiyun
trim_back(ini_t * ini,char * p)56*4882a593Smuzhiyun static void trim_back(ini_t *ini, char *p) {
57*4882a593Smuzhiyun while (p >= ini->data && (*p == ' ' || *p == '\t' || *p == '\r')) {
58*4882a593Smuzhiyun *p-- = '\0';
59*4882a593Smuzhiyun }
60*4882a593Smuzhiyun }
61*4882a593Smuzhiyun
discard_line(ini_t * ini,char * p)62*4882a593Smuzhiyun static char* discard_line(ini_t *ini, char *p) {
63*4882a593Smuzhiyun while (p < ini->end && *p != '\n') {
64*4882a593Smuzhiyun *p++ = '\0';
65*4882a593Smuzhiyun }
66*4882a593Smuzhiyun return p;
67*4882a593Smuzhiyun }
68*4882a593Smuzhiyun
69*4882a593Smuzhiyun
unescape_quoted_value(ini_t * ini,char * p)70*4882a593Smuzhiyun static char *unescape_quoted_value(ini_t *ini, char *p) {
71*4882a593Smuzhiyun /* Use `q` as write-head and `p` as read-head, `p` is always ahead of `q`
72*4882a593Smuzhiyun * as escape sequences are always larger than their resultant data */
73*4882a593Smuzhiyun char *q = p;
74*4882a593Smuzhiyun p++;
75*4882a593Smuzhiyun while (p < ini->end && *p != '"' && *p != '\r' && *p != '\n') {
76*4882a593Smuzhiyun if (*p == '\\') {
77*4882a593Smuzhiyun /* Handle escaped char */
78*4882a593Smuzhiyun p++;
79*4882a593Smuzhiyun switch (*p) {
80*4882a593Smuzhiyun default : *q = *p; break;
81*4882a593Smuzhiyun case 'r' : *q = '\r'; break;
82*4882a593Smuzhiyun case 'n' : *q = '\n'; break;
83*4882a593Smuzhiyun case 't' : *q = '\t'; break;
84*4882a593Smuzhiyun case '\r' :
85*4882a593Smuzhiyun case '\n' :
86*4882a593Smuzhiyun case '\0' : goto end;
87*4882a593Smuzhiyun }
88*4882a593Smuzhiyun
89*4882a593Smuzhiyun } else {
90*4882a593Smuzhiyun /* Handle normal char */
91*4882a593Smuzhiyun *q = *p;
92*4882a593Smuzhiyun }
93*4882a593Smuzhiyun q++, p++;
94*4882a593Smuzhiyun }
95*4882a593Smuzhiyun end:
96*4882a593Smuzhiyun return q;
97*4882a593Smuzhiyun }
98*4882a593Smuzhiyun
99*4882a593Smuzhiyun
100*4882a593Smuzhiyun /* Splits data in place into strings containing section-headers, keys and
101*4882a593Smuzhiyun * values using one or more '\0' as a delimiter. Unescapes quoted values */
split_data(ini_t * ini)102*4882a593Smuzhiyun static void split_data(ini_t *ini) {
103*4882a593Smuzhiyun char *value_start, *line_start;
104*4882a593Smuzhiyun char *p = ini->data;
105*4882a593Smuzhiyun
106*4882a593Smuzhiyun while (p < ini->end) {
107*4882a593Smuzhiyun switch (*p) {
108*4882a593Smuzhiyun case '\r':
109*4882a593Smuzhiyun case '\n':
110*4882a593Smuzhiyun case '\t':
111*4882a593Smuzhiyun case ' ':
112*4882a593Smuzhiyun *p = '\0';
113*4882a593Smuzhiyun p++;
114*4882a593Smuzhiyun break;
115*4882a593Smuzhiyun case '\0':
116*4882a593Smuzhiyun p++;
117*4882a593Smuzhiyun break;
118*4882a593Smuzhiyun
119*4882a593Smuzhiyun case '[':
120*4882a593Smuzhiyun p += strcspn(p, "]\n");
121*4882a593Smuzhiyun *p = '\0';
122*4882a593Smuzhiyun break;
123*4882a593Smuzhiyun
124*4882a593Smuzhiyun case ';':
125*4882a593Smuzhiyun p = discard_line(ini, p);
126*4882a593Smuzhiyun break;
127*4882a593Smuzhiyun
128*4882a593Smuzhiyun default:
129*4882a593Smuzhiyun line_start = p;
130*4882a593Smuzhiyun p += strcspn(p, "=\n");
131*4882a593Smuzhiyun
132*4882a593Smuzhiyun /* Is line missing a '='? */
133*4882a593Smuzhiyun if (*p != '=') {
134*4882a593Smuzhiyun p = discard_line(ini, line_start);
135*4882a593Smuzhiyun break;
136*4882a593Smuzhiyun }
137*4882a593Smuzhiyun trim_back(ini, p - 1);
138*4882a593Smuzhiyun
139*4882a593Smuzhiyun /* Replace '=' and whitespace after it with '\0' */
140*4882a593Smuzhiyun do {
141*4882a593Smuzhiyun *p++ = '\0';
142*4882a593Smuzhiyun } while (*p == ' ' || *p == '\r' || *p == '\t');
143*4882a593Smuzhiyun
144*4882a593Smuzhiyun /* Is a value after '=' missing? */
145*4882a593Smuzhiyun if (*p == '\n' || *p == '\0') {
146*4882a593Smuzhiyun p = discard_line(ini, line_start);
147*4882a593Smuzhiyun break;
148*4882a593Smuzhiyun }
149*4882a593Smuzhiyun
150*4882a593Smuzhiyun if (*p == '"') {
151*4882a593Smuzhiyun /* Handle quoted string value */
152*4882a593Smuzhiyun value_start = p;
153*4882a593Smuzhiyun p = unescape_quoted_value(ini, p);
154*4882a593Smuzhiyun
155*4882a593Smuzhiyun /* Was the string empty? */
156*4882a593Smuzhiyun if (p == value_start) {
157*4882a593Smuzhiyun p = discard_line(ini, line_start);
158*4882a593Smuzhiyun break;
159*4882a593Smuzhiyun }
160*4882a593Smuzhiyun
161*4882a593Smuzhiyun /* Discard the rest of the line after the string value */
162*4882a593Smuzhiyun p = discard_line(ini, p);
163*4882a593Smuzhiyun
164*4882a593Smuzhiyun } else {
165*4882a593Smuzhiyun /* Handle normal value */
166*4882a593Smuzhiyun p += strcspn(p, "\n");
167*4882a593Smuzhiyun trim_back(ini, p - 1);
168*4882a593Smuzhiyun }
169*4882a593Smuzhiyun break;
170*4882a593Smuzhiyun }
171*4882a593Smuzhiyun }
172*4882a593Smuzhiyun }
173*4882a593Smuzhiyun
174*4882a593Smuzhiyun
175*4882a593Smuzhiyun
rkaiq_ini_load(const char * filename)176*4882a593Smuzhiyun ini_t* rkaiq_ini_load(const char *filename) {
177*4882a593Smuzhiyun ini_t *ini = NULL;
178*4882a593Smuzhiyun FILE *fp = NULL;
179*4882a593Smuzhiyun int n, sz;
180*4882a593Smuzhiyun
181*4882a593Smuzhiyun /* Init ini struct */
182*4882a593Smuzhiyun ini = (ini_t*)malloc(sizeof(*ini));
183*4882a593Smuzhiyun if (!ini) {
184*4882a593Smuzhiyun goto fail;
185*4882a593Smuzhiyun }
186*4882a593Smuzhiyun memset(ini, 0, sizeof(*ini));
187*4882a593Smuzhiyun
188*4882a593Smuzhiyun /* Open file */
189*4882a593Smuzhiyun fp = fopen(filename, "rb");
190*4882a593Smuzhiyun if (!fp) {
191*4882a593Smuzhiyun goto fail;
192*4882a593Smuzhiyun }
193*4882a593Smuzhiyun
194*4882a593Smuzhiyun /* Get file size */
195*4882a593Smuzhiyun fseek(fp, 0, SEEK_END);
196*4882a593Smuzhiyun sz = ftell(fp);
197*4882a593Smuzhiyun rewind(fp);
198*4882a593Smuzhiyun
199*4882a593Smuzhiyun /* Load file content into memory, null terminate, init end var */
200*4882a593Smuzhiyun ini->data = (char*)malloc(sz + 1);
201*4882a593Smuzhiyun ini->data[sz] = '\0';
202*4882a593Smuzhiyun ini->end = ini->data + sz;
203*4882a593Smuzhiyun n = fread(ini->data, 1, sz, fp);
204*4882a593Smuzhiyun if (n != sz) {
205*4882a593Smuzhiyun goto fail;
206*4882a593Smuzhiyun }
207*4882a593Smuzhiyun
208*4882a593Smuzhiyun /* Prepare data */
209*4882a593Smuzhiyun split_data(ini);
210*4882a593Smuzhiyun
211*4882a593Smuzhiyun /* Clean up and return */
212*4882a593Smuzhiyun fclose(fp);
213*4882a593Smuzhiyun return ini;
214*4882a593Smuzhiyun
215*4882a593Smuzhiyun fail:
216*4882a593Smuzhiyun if (fp) fclose(fp);
217*4882a593Smuzhiyun if (ini) rkaiq_ini_free(ini);
218*4882a593Smuzhiyun return NULL;
219*4882a593Smuzhiyun }
220*4882a593Smuzhiyun
221*4882a593Smuzhiyun
rkaiq_ini_free(ini_t * ini)222*4882a593Smuzhiyun void rkaiq_ini_free(ini_t *ini) {
223*4882a593Smuzhiyun free(ini->data);
224*4882a593Smuzhiyun free(ini);
225*4882a593Smuzhiyun }
226*4882a593Smuzhiyun
227*4882a593Smuzhiyun
rkaiq_ini_get(ini_t * ini,const char * section,const char * key)228*4882a593Smuzhiyun const char* rkaiq_ini_get(ini_t *ini, const char *section, const char *key) {
229*4882a593Smuzhiyun char *current_section = "";
230*4882a593Smuzhiyun char *val;
231*4882a593Smuzhiyun char *p = ini->data;
232*4882a593Smuzhiyun
233*4882a593Smuzhiyun if (*p == '\0') {
234*4882a593Smuzhiyun p = next(ini, p);
235*4882a593Smuzhiyun }
236*4882a593Smuzhiyun
237*4882a593Smuzhiyun while (p < ini->end) {
238*4882a593Smuzhiyun if (*p == '[') {
239*4882a593Smuzhiyun /* Handle section */
240*4882a593Smuzhiyun current_section = p + 1;
241*4882a593Smuzhiyun
242*4882a593Smuzhiyun } else {
243*4882a593Smuzhiyun /* Handle key */
244*4882a593Smuzhiyun val = next(ini, p);
245*4882a593Smuzhiyun if (!section || !strcmpci(section, current_section)) {
246*4882a593Smuzhiyun if (!strcmpci(p, key)) {
247*4882a593Smuzhiyun return val;
248*4882a593Smuzhiyun }
249*4882a593Smuzhiyun }
250*4882a593Smuzhiyun p = val;
251*4882a593Smuzhiyun }
252*4882a593Smuzhiyun
253*4882a593Smuzhiyun p = next(ini, p);
254*4882a593Smuzhiyun }
255*4882a593Smuzhiyun
256*4882a593Smuzhiyun return NULL;
257*4882a593Smuzhiyun }
258*4882a593Smuzhiyun
259*4882a593Smuzhiyun
rkaiq_ini_sget(ini_t * ini,const char * section,const char * key,const char * scanfmt,void * dst)260*4882a593Smuzhiyun int rkaiq_ini_sget(
261*4882a593Smuzhiyun ini_t *ini, const char *section, const char *key,
262*4882a593Smuzhiyun const char *scanfmt, void *dst
263*4882a593Smuzhiyun ) {
264*4882a593Smuzhiyun const char *val = rkaiq_ini_get(ini, section, key);
265*4882a593Smuzhiyun if (!val) {
266*4882a593Smuzhiyun return 0;
267*4882a593Smuzhiyun }
268*4882a593Smuzhiyun if (scanfmt) {
269*4882a593Smuzhiyun sscanf(val, scanfmt, dst);
270*4882a593Smuzhiyun } else {
271*4882a593Smuzhiyun *((const char**) dst) = val;
272*4882a593Smuzhiyun }
273*4882a593Smuzhiyun return 1;
274*4882a593Smuzhiyun }
275