libgpo: add gp_inifile_enum_section()
[ambi/samba-autobuild/.git] / libgpo / gpo_ini.c
1 /*
2  *  Unix SMB/CIFS implementation.
3  *  Group Policy Support
4  *  Copyright (C) Guenther Deschner 2007
5  *  Copyright (C) Wilco Baan Hofman 2009
6  *
7  *  This program is free software; you can redistribute it and/or modify
8  *  it under the terms of the GNU General Public License as published by
9  *  the Free Software Foundation; either version 3 of the License, or
10  *  (at your option) any later version.
11  *
12  *  This program is distributed in the hope that it will be useful,
13  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
14  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15  *  GNU General Public License for more details.
16  *
17  *  You should have received a copy of the GNU General Public License
18  *  along with this program; if not, see <http://www.gnu.org/licenses/>.
19  */
20
21 #include "includes.h"
22 #include "gpo.h"
23 #include "gpo_ini.h"
24 #include "system/filesys.h"
25
26
27 static bool change_section(const char *section, void *ctx_ptr)
28 {
29         struct gp_inifile_context *ctx = (struct gp_inifile_context *) ctx_ptr;
30
31         if (ctx->current_section) {
32                 talloc_free(ctx->current_section);
33         }
34         ctx->current_section = talloc_strdup(ctx, section);
35         if (!ctx->current_section) {
36                 return false;
37         }
38         return true;
39 }
40
41 /****************************************************************
42 ****************************************************************/
43
44 static bool store_keyval_pair(const char *key, const char *value, void *ctx_ptr)
45 {
46         struct gp_inifile_context *ctx = (struct gp_inifile_context *) ctx_ptr;
47
48         ctx->data = talloc_realloc(ctx, ctx->data, struct keyval_pair *, ctx->keyval_count+1);
49         if (!ctx->data) {
50                 return false;
51         }
52
53         ctx->data[ctx->keyval_count] = talloc_zero(ctx, struct keyval_pair);
54         if (!ctx->data[ctx->keyval_count]) {
55                 return false;
56         }
57
58         ctx->data[ctx->keyval_count]->key = talloc_asprintf(ctx, "%s:%s", ctx->current_section, key);
59         ctx->data[ctx->keyval_count]->val = talloc_strdup(ctx, value);
60
61         if (!ctx->data[ctx->keyval_count]->key ||
62             !ctx->data[ctx->keyval_count]->val) {
63                 return false;
64         }
65
66         ctx->keyval_count++;
67         return true;
68 }
69
70 /****************************************************************
71 ****************************************************************/
72
73 static NTSTATUS convert_file_from_ucs2(TALLOC_CTX *mem_ctx,
74                                        const char *filename_in,
75                                        char **filename_out)
76 {
77         int tmp_fd = -1;
78         uint8_t *data_in = NULL;
79         uint8_t *data_out = NULL;
80         char *tmp_name = NULL;
81         NTSTATUS status;
82         size_t n = 0;
83         size_t converted_size;
84         mode_t mask;
85
86         if (!filename_out) {
87                 return NT_STATUS_INVALID_PARAMETER;
88         }
89
90         data_in = (uint8_t *)file_load(filename_in, &n, 0, NULL);
91         if (!data_in) {
92                 status = NT_STATUS_NO_SUCH_FILE;
93                 goto out;
94         }
95
96         tmp_name = talloc_asprintf(mem_ctx, "%s/convert_file_from_ucs2.XXXXXX",
97                 tmpdir());
98         if (!tmp_name) {
99                 status = NT_STATUS_NO_MEMORY;
100                 goto out;
101         }
102
103         mask = umask(S_IRWXO | S_IRWXG);
104         tmp_fd = mkstemp(tmp_name);
105         umask(mask);
106         if (tmp_fd == -1) {
107                 status = NT_STATUS_ACCESS_DENIED;
108                 goto out;
109         }
110
111         if (!convert_string_talloc(mem_ctx, CH_UTF16LE, CH_UNIX, data_in, n,
112                                    (void *)&data_out, &converted_size))
113         {
114                 status = NT_STATUS_INVALID_BUFFER_SIZE;
115                 goto out;
116         }
117
118         /* skip utf8 BOM */
119         DEBUG(11,("convert_file_from_ucs2: "
120                "data_out[0]: 0x%x, data_out[1]: 0x%x, data_out[2]: 0x%x\n",
121                 data_out[0], data_out[1], data_out[2]));
122
123         if ((data_out[0] == 0xef) && (data_out[1] == 0xbb) &&
124             (data_out[2] == 0xbf)) {
125                 DEBUG(11,("convert_file_from_ucs2: "
126                          "%s skipping utf8 BOM\n", tmp_name));
127                 data_out += 3;
128                 converted_size -= 3;
129         }
130
131         if (write(tmp_fd, data_out, converted_size) != converted_size) {
132                 status = map_nt_error_from_unix_common(errno);
133                 goto out;
134         }
135
136         *filename_out = tmp_name;
137
138         status = NT_STATUS_OK;
139
140  out:
141         if (tmp_fd != -1) {
142                 close(tmp_fd);
143         }
144
145         talloc_free(data_in);
146
147         return status;
148 }
149
150 /****************************************************************
151 ****************************************************************/
152
153 NTSTATUS gp_inifile_getstring(struct gp_inifile_context *ctx, const char *key, char **ret)
154 {
155         int i;
156
157         for (i = 0; i < ctx->keyval_count; i++) {
158                 if (strcmp(ctx->data[i]->key, key) == 0) {
159                         if (ret) {
160                                 *ret = ctx->data[i]->val;
161                         }
162                         return NT_STATUS_OK;
163                 }
164         }
165         return NT_STATUS_NOT_FOUND;
166 }
167
168 /****************************************************************
169 ****************************************************************/
170
171 NTSTATUS gp_inifile_getint(struct gp_inifile_context *ctx, const char *key, int *ret)
172 {
173         char *value;
174         NTSTATUS result;
175
176         result = gp_inifile_getstring(ctx,key, &value);
177         if (!NT_STATUS_IS_OK(result)) {
178                 return result;
179         }
180
181         if (ret) {
182                 *ret = (int)strtol(value, NULL, 10);
183         }
184         return NT_STATUS_OK;
185 }
186
187 /****************************************************************
188 ****************************************************************/
189
190 NTSTATUS gp_inifile_getbool(struct gp_inifile_context *ctx, const char *key, bool *ret)
191 {
192         char *value;
193         NTSTATUS result;
194
195         result = gp_inifile_getstring(ctx,key, &value);
196         if (!NT_STATUS_IS_OK(result)) {
197                 return result;
198         }
199
200         if (strequal(value, "Yes") ||
201             strequal(value, "True")) {
202                 if (ret) {
203                         *ret = true;
204                 }
205                 return NT_STATUS_OK;
206         } else if (strequal(value, "No") ||
207                    strequal(value, "False")) {
208                 if (ret) {
209                         *ret = false;
210                 }
211                 return NT_STATUS_OK;
212         }
213
214         return NT_STATUS_NOT_FOUND;
215 }
216
217 /****************************************************************
218 ****************************************************************/
219
220 NTSTATUS gp_inifile_enum_section(struct gp_inifile_context *ctx,
221                                  const char *section,
222                                  size_t *num_ini_keys,
223                                  const char ***ini_keys,
224                                  const char ***ini_values)
225 {
226         NTSTATUS status;
227         int i;
228         size_t num_keys = 0, num_vals = 0;
229         const char **keys = NULL;
230         const char **values = NULL;
231
232         if (section == NULL || num_ini_keys == NULL ||
233             ini_keys == NULL || ini_values == NULL) {
234                 return NT_STATUS_INVALID_PARAMETER;
235         }
236
237         for (i = 0; i < ctx->keyval_count; i++) {
238
239                 bool ok;
240
241                 /*
242                  * section: KEYNAME
243                  * KEYNAME:value matches
244                  * KEYNAME_OEM:value not
245                  */
246
247                 if (strlen(section)+1 > strlen(ctx->data[i]->key)) {
248                         continue;
249                 }
250
251                 if (!strnequal(section, ctx->data[i]->key, strlen(section))) {
252                         continue;
253                 }
254
255                 if (ctx->data[i]->key[strlen(section)] != ':') {
256                         continue;
257                 }
258
259                 ok = add_string_to_array(ctx, ctx->data[i]->key, &keys, &num_keys);
260                 if (!ok) {
261                         status = NT_STATUS_NO_MEMORY;
262                         goto failed;
263                 }
264
265                 ok = add_string_to_array(ctx, ctx->data[i]->val, &values, &num_vals);
266                 if (!ok) {
267                         status = NT_STATUS_NO_MEMORY;
268                         goto failed;
269                 }
270
271                 if (num_keys != num_vals) {
272                         status = NT_STATUS_INTERNAL_DB_CORRUPTION;
273                         goto failed;
274                 }
275         }
276
277         *num_ini_keys = num_keys;
278         *ini_keys = keys;
279         *ini_values = values;
280
281         return NT_STATUS_OK;
282
283  failed:
284         talloc_free(keys);
285         talloc_free(values);
286
287         return status;
288 }
289
290
291 /****************************************************************
292 ****************************************************************/
293
294 NTSTATUS gp_inifile_init_context(TALLOC_CTX *mem_ctx,
295                                  uint32_t flags,
296                                  const char *unix_path,
297                                  const char *suffix,
298                                  struct gp_inifile_context **ctx_ret)
299 {
300         struct gp_inifile_context *ctx = NULL;
301         NTSTATUS status;
302         int rv;
303         char *tmp_filename = NULL;
304         const char *ini_filename = NULL;
305
306         if (!unix_path || !ctx_ret) {
307                 return NT_STATUS_INVALID_PARAMETER;
308         }
309
310         ctx = talloc_zero(mem_ctx, struct gp_inifile_context);
311         NT_STATUS_HAVE_NO_MEMORY(ctx);
312
313         status = gp_find_file(mem_ctx, flags, unix_path, suffix,
314                               &ini_filename);
315
316         if (!NT_STATUS_IS_OK(status)) {
317                 goto failed;
318         }
319
320         status = convert_file_from_ucs2(mem_ctx, ini_filename,
321                                         &tmp_filename);
322         if (!NT_STATUS_IS_OK(status)) {
323                 goto failed;
324         }
325
326         rv = pm_process(tmp_filename, change_section, store_keyval_pair, ctx);
327         if (!rv) {
328                 return NT_STATUS_NO_SUCH_FILE;
329         }
330
331
332         ctx->generated_filename = tmp_filename;
333         ctx->mem_ctx = mem_ctx;
334
335         *ctx_ret = ctx;
336
337         return NT_STATUS_OK;
338
339  failed:
340
341         DEBUG(1,("gp_inifile_init_context failed: %s\n",
342                 nt_errstr(status)));
343
344         talloc_free(ctx);
345
346         return status;
347 }
348
349 /****************************************************************
350 ****************************************************************/
351
352 NTSTATUS gp_inifile_init_context_direct(TALLOC_CTX *mem_ctx,
353                                         const char *unix_path,
354                                         struct gp_inifile_context **pgp_ctx)
355 {
356         struct gp_inifile_context *gp_ctx = NULL;
357         NTSTATUS status;
358         int rv;
359         char *tmp_filename = NULL;
360
361         if (unix_path == NULL || pgp_ctx == NULL) {
362                 return NT_STATUS_INVALID_PARAMETER;
363         }
364
365         gp_ctx = talloc_zero(mem_ctx, struct gp_inifile_context);
366         if (gp_ctx == NULL) {
367                 return NT_STATUS_NO_MEMORY;
368         }
369
370         status = convert_file_from_ucs2(mem_ctx, unix_path,
371                                         &tmp_filename);
372         if (!NT_STATUS_IS_OK(status)) {
373                 goto failed;
374         }
375
376         rv = pm_process(tmp_filename,
377                         change_section,
378                         store_keyval_pair,
379                         gp_ctx);
380         if (rv != 0) {
381                 return NT_STATUS_NO_SUCH_FILE;
382         }
383
384         gp_ctx->generated_filename = tmp_filename;
385         gp_ctx->mem_ctx = mem_ctx;
386
387         *pgp_ctx = gp_ctx;
388
389         return NT_STATUS_OK;
390
391  failed:
392
393         DEBUG(1,("gp_inifile_init_context_direct failed: %s\n",
394                 nt_errstr(status)));
395
396         talloc_free(gp_ctx);
397
398         return status;
399 }
400
401
402 /****************************************************************
403  parse the local gpt.ini file
404 ****************************************************************/
405
406 #define GPT_INI_SECTION_GENERAL "General"
407 #define GPT_INI_PARAMETER_VERSION "Version"
408 #define GPT_INI_PARAMETER_DISPLAYNAME "displayName"
409
410 NTSTATUS parse_gpt_ini(TALLOC_CTX *mem_ctx,
411                        const char *filename,
412                        uint32_t *version,
413                        char **display_name)
414 {
415         NTSTATUS result;
416         int rv;
417         int v = 0;
418         char *name = NULL;
419         struct gp_inifile_context *ctx;
420
421         if (!filename) {
422                 return NT_STATUS_INVALID_PARAMETER;
423         }
424
425         ctx = talloc_zero(mem_ctx, struct gp_inifile_context);
426         NT_STATUS_HAVE_NO_MEMORY(ctx);
427
428         rv = pm_process(filename, change_section, store_keyval_pair, ctx);
429         if (!rv) {
430                 return NT_STATUS_NO_SUCH_FILE;
431         }
432
433
434         result = gp_inifile_getstring(ctx, GPT_INI_SECTION_GENERAL
435                         ":"GPT_INI_PARAMETER_DISPLAYNAME, &name);
436         if (!NT_STATUS_IS_OK(result)) {
437                 /* the default domain policy and the default domain controller
438                  * policy never have a displayname in their gpt.ini file */
439                 DEBUG(10,("parse_gpt_ini: no name in %s\n", filename));
440         }
441
442         if (name && display_name) {
443                 *display_name = talloc_strdup(ctx, name);
444                 if (*display_name == NULL) {
445                         return NT_STATUS_NO_MEMORY;
446                 }
447         }
448
449         result = gp_inifile_getint(ctx, GPT_INI_SECTION_GENERAL
450                         ":"GPT_INI_PARAMETER_VERSION, &v);
451         if (!NT_STATUS_IS_OK(result)) {
452                 DEBUG(10,("parse_gpt_ini: no version\n"));
453                 return NT_STATUS_INTERNAL_DB_CORRUPTION;
454         }
455
456         if (version) {
457                 *version = v;
458         }
459
460         talloc_free(ctx);
461
462         return NT_STATUS_OK;
463 }