s3-gpo: Fix CID #1405972 Resource leak
[samba.git] / source3 / libgpo / gpext / registry.c
1 /*
2  *  Unix SMB/CIFS implementation.
3  *  Group Policy Support
4  *  Copyright (C) Guenther Deschner 2007-2008,2010
5  *
6  *  This program is free software; you can redistribute it and/or modify
7  *  it under the terms of the GNU General Public License as published by
8  *  the Free Software Foundation; either version 3 of the License, or
9  *  (at your option) any later version.
10  *
11  *  This program is distributed in the hope that it will be useful,
12  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
13  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14  *  GNU General Public License for more details.
15  *
16  *  You should have received a copy of the GNU General Public License
17  *  along with this program; if not, see <http://www.gnu.org/licenses/>.
18  */
19
20 #include "includes.h"
21 #include "../libgpo/gpo_ini.h"
22 #include "../libgpo/gpo.h"
23 #include "libgpo/gpo_proto.h"
24 #include "registry.h"
25 #include "../librpc/gen_ndr/ndr_preg.h"
26 #include "libgpo/gpext/gpext.h"
27
28 #define GP_EXT_NAME "registry"
29
30 /* more info can be found at:
31  * http://msdn2.microsoft.com/en-us/library/aa374407.aspx */
32
33 #define GP_REGPOL_FILE  "Registry.pol"
34
35 #define GP_REGPOL_FILE_SIGNATURE 0x67655250 /* 'PReg' */
36 #define GP_REGPOL_FILE_VERSION 1
37
38 static TALLOC_CTX *ctx = NULL;
39
40 NTSTATUS gpext_registry_init(TALLOC_CTX *mem_ctx);
41
42 /****************************************************************
43 ****************************************************************/
44
45 static bool reg_parse_value(TALLOC_CTX *mem_ctx,
46                             const char **value,
47                             enum gp_reg_action *action)
48 {
49         if (!*value) {
50                 *action = GP_REG_ACTION_ADD_KEY;
51                 return true;
52         }
53
54         if (strncmp(*value, "**", 2) != 0) {
55                 *action = GP_REG_ACTION_ADD_VALUE;
56                 return true;
57         }
58
59         if (strnequal(*value, "**DelVals.", 10)) {
60                 *action = GP_REG_ACTION_DEL_ALL_VALUES;
61                 return true;
62         }
63
64         if (strnequal(*value, "**Del.", 6)) {
65                 *value = talloc_strdup(mem_ctx, *value + 6);
66                 *action = GP_REG_ACTION_DEL_VALUE;
67                 return true;
68         }
69
70         if (strnequal(*value, "**SecureKey", 11)) {
71                 if (strnequal(*value, "**SecureKey=1", 13)) {
72                         *action = GP_REG_ACTION_SEC_KEY_SET;
73                         return true;
74                 }
75
76  /*************** not tested from here on ***************/
77                 if (strnequal(*value, "**SecureKey=0", 13)) {
78                         smb_panic("not supported: **SecureKey=0");
79                         *action = GP_REG_ACTION_SEC_KEY_RESET;
80                         return true;
81                 }
82                 DEBUG(0,("unknown: SecureKey: %s\n", *value));
83                 smb_panic("not supported SecureKey method");
84                 return false;
85         }
86
87         if (strnequal(*value, "**DeleteValues", strlen("**DeleteValues"))) {
88                 smb_panic("not supported: **DeleteValues");
89                 *action = GP_REG_ACTION_DEL_VALUES;
90                 return false;
91         }
92
93         if (strnequal(*value, "**DeleteKeys", strlen("**DeleteKeys"))) {
94                 smb_panic("not supported: **DeleteKeys");
95                 *action = GP_REG_ACTION_DEL_KEYS;
96                 return false;
97         }
98
99         DEBUG(0,("unknown value: %s\n", *value));
100         smb_panic(*value);
101         return false;
102 }
103
104 /****************************************************************
105 ****************************************************************/
106
107 static bool gp_reg_entry_from_file_entry(TALLOC_CTX *mem_ctx,
108                                          struct preg_entry *r,
109                                          struct gp_registry_entry **reg_entry)
110 {
111         struct registry_value *data = NULL;
112         struct gp_registry_entry *entry = NULL;
113         enum gp_reg_action action = GP_REG_ACTION_NONE;
114
115         ZERO_STRUCTP(*reg_entry);
116
117         data = talloc_zero(mem_ctx, struct registry_value);
118         if (!data)
119                 return false;
120
121         data->type = r->type;
122         data->data = data_blob_talloc(data, r->data, r->size);
123
124         entry = talloc_zero(mem_ctx, struct gp_registry_entry);
125         if (!entry)
126                 return false;
127
128         if (!reg_parse_value(mem_ctx, &r->valuename, &action))
129                 return false;
130
131         entry->key = talloc_strdup(entry, r->keyname);
132         entry->value = talloc_strdup(entry, r->valuename);
133         entry->data = data;
134         entry->action = action;
135
136         *reg_entry = entry;
137
138         return true;
139 }
140
141 /****************************************************************
142 ****************************************************************/
143
144 static NTSTATUS reg_parse_registry(TALLOC_CTX *mem_ctx,
145                                    uint32_t flags,
146                                    const char *filename,
147                                    struct gp_registry_entry **entries_p,
148                                    size_t *num_entries_p)
149 {
150         DATA_BLOB blob;
151         NTSTATUS status;
152         enum ndr_err_code ndr_err;
153         const char *real_filename = NULL;
154         struct preg_file r;
155         struct gp_registry_entry *entries = NULL;
156         size_t num_entries = 0;
157         int i;
158
159         status = gp_find_file(mem_ctx,
160                               flags,
161                               filename,
162                               GP_REGPOL_FILE,
163                               &real_filename);
164         if (!NT_STATUS_IS_OK(status)) {
165                 return status;
166         }
167
168         blob.data = (uint8_t *)file_load(real_filename, &blob.length, 0, NULL);
169         if (!blob.data) {
170                 return NT_STATUS_CANNOT_LOAD_REGISTRY_FILE;
171         }
172
173         ndr_err = ndr_pull_struct_blob(&blob, mem_ctx, &r,
174                         (ndr_pull_flags_fn_t)ndr_pull_preg_file);
175         if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
176                 status = ndr_map_error2ntstatus(ndr_err);
177                 goto out;
178         }
179
180         if (flags & GPO_INFO_FLAG_VERBOSE) {
181                 NDR_PRINT_DEBUG(preg_file, &r);
182         }
183
184         if (!strequal(r.header.signature, "PReg")) {
185                 status = NT_STATUS_INVALID_PARAMETER;
186                 goto out;
187         }
188
189         if (r.header.version != GP_REGPOL_FILE_VERSION) {
190                 status = NT_STATUS_INVALID_PARAMETER;
191                 goto out;
192         }
193
194         for (i=0; i < r.num_entries; i++) {
195
196                 struct gp_registry_entry *r_entry = NULL;
197
198                 if (!gp_reg_entry_from_file_entry(mem_ctx,
199                                                   &r.entries[i],
200                                                   &r_entry)) {
201                         status = NT_STATUS_NO_MEMORY;
202                         goto out;
203                 }
204
205                 if (!add_gp_registry_entry_to_array(mem_ctx,
206                                                     r_entry,
207                                                     &entries,
208                                                     &num_entries)) {
209                         status = NT_STATUS_NO_MEMORY;
210                         goto out;
211                 }
212         }
213
214         *entries_p = entries;
215         *num_entries_p = num_entries;
216
217         status = NT_STATUS_OK;
218
219  out:
220         data_blob_free(&blob);
221         return status;
222 }
223
224 /****************************************************************
225 ****************************************************************/
226
227 static WERROR reg_apply_registry(TALLOC_CTX *mem_ctx,
228                                  const struct security_token *token,
229                                  struct registry_key *root_key,
230                                  uint32_t flags,
231                                  struct gp_registry_entry *entries,
232                                  size_t num_entries)
233 {
234         struct gp_registry_context *reg_ctx = NULL;
235         WERROR werr;
236         size_t i;
237
238         if (num_entries == 0) {
239                 return WERR_OK;
240         }
241
242 #if 0
243         if (flags & GPO_LIST_FLAG_MACHINE) {
244                 werr = gp_init_reg_ctx(mem_ctx, KEY_HKLM, REG_KEY_WRITE,
245                                        get_system_token(),
246                                        &reg_ctx);
247         } else {
248                 werr = gp_init_reg_ctx(mem_ctx, KEY_HKCU, REG_KEY_WRITE,
249                                        token,
250                                        &reg_ctx);
251         }
252         W_ERROR_NOT_OK_RETURN(werr);
253 #endif
254         for (i=0; i<num_entries; i++) {
255
256                 /* FIXME: maybe we should check here if we attempt to go beyond
257                  * the 4 allowed reg keys */
258
259                 werr = reg_apply_registry_entry(mem_ctx, root_key,
260                                                 reg_ctx,
261                                                 &(entries)[i],
262                                                 token, flags);
263                 if (!W_ERROR_IS_OK(werr)) {
264                         DEBUG(0,("failed to apply registry: %s\n",
265                                 win_errstr(werr)));
266                         goto done;
267                 }
268         }
269
270 done:
271         gp_free_reg_ctx(reg_ctx);
272         return werr;
273 }
274
275
276 /****************************************************************
277 ****************************************************************/
278
279 static NTSTATUS registry_process_group_policy(TALLOC_CTX *mem_ctx,
280                                               uint32_t flags,
281                                               struct registry_key *root_key,
282                                               const struct security_token *token,
283                                               const struct GROUP_POLICY_OBJECT *deleted_gpo_list,
284                                               const struct GROUP_POLICY_OBJECT *changed_gpo_list)
285 {
286         NTSTATUS status;
287         WERROR werr;
288         struct gp_registry_entry *entries = NULL;
289         size_t num_entries = 0;
290         char *unix_path = NULL;
291         const struct GROUP_POLICY_OBJECT *gpo;
292         char *gpo_cache_path = cache_path(GPO_CACHE_DIR);
293         if (gpo_cache_path == NULL) {
294                 return NT_STATUS_NO_MEMORY;
295         }
296
297         /* implementation of the policy callback function, see
298          * http://msdn.microsoft.com/en-us/library/aa373494%28v=vs.85%29.aspx
299          * for details - gd */
300
301         /* for now do not process the list of deleted group policies
302
303         for (gpo = deleted_gpo_list; gpo; gpo = gpo->next) {
304         }
305
306         */
307
308         for (gpo = changed_gpo_list; gpo; gpo = gpo->next) {
309
310                 gpext_debug_header(0, "registry_process_group_policy", flags,
311                                    gpo, GP_EXT_GUID_REGISTRY, NULL);
312
313                 status = gpo_get_unix_path(mem_ctx, gpo_cache_path,
314                                            gpo, &unix_path);
315                 if (!NT_STATUS_IS_OK(status)) {
316                         goto err_cache_path_free;
317                 }
318
319                 status = reg_parse_registry(mem_ctx,
320                                             flags,
321                                             unix_path,
322                                             &entries,
323                                             &num_entries);
324                 if (!NT_STATUS_IS_OK(status)) {
325                         DEBUG(0,("failed to parse registry: %s\n",
326                                 nt_errstr(status)));
327                         goto err_cache_path_free;
328                 }
329
330                 dump_reg_entries(flags, "READ", entries, num_entries);
331
332                 werr = reg_apply_registry(mem_ctx, token, root_key, flags,
333                                           entries, num_entries);
334                 if (!W_ERROR_IS_OK(werr)) {
335                         DEBUG(0,("failed to apply registry: %s\n",
336                                 win_errstr(werr)));
337                         status = werror_to_ntstatus(werr);
338                         goto err_cache_path_free;
339                 }
340         }
341         status = NT_STATUS_OK;
342
343 err_cache_path_free:
344         talloc_free(gpo_cache_path);
345         talloc_free(entries);
346         return status;
347 }
348
349 /****************************************************************
350 ****************************************************************/
351
352 static NTSTATUS registry_get_reg_config(TALLOC_CTX *mem_ctx,
353                                         struct gp_extension_reg_info **reg_info)
354 {
355         NTSTATUS status;
356         struct gp_extension_reg_info *info = NULL;
357         struct gp_extension_reg_table table[] = {
358                 { "ProcessGroupPolicy", REG_SZ, "registry_process_group_policy" },
359                 { NULL, REG_NONE, NULL }
360         };
361
362         info = talloc_zero(mem_ctx, struct gp_extension_reg_info);
363         NT_STATUS_HAVE_NO_MEMORY(info);
364
365         status = gpext_info_add_entry(mem_ctx, GP_EXT_NAME,
366                                       GP_EXT_GUID_REGISTRY,
367                                       table, info);
368         NT_STATUS_NOT_OK_RETURN(status);
369
370         *reg_info = info;
371
372         return NT_STATUS_OK;
373 }
374
375 /****************************************************************
376 ****************************************************************/
377
378 static NTSTATUS registry_initialize(TALLOC_CTX *mem_ctx)
379 {
380         return NT_STATUS_OK;
381 }
382
383 /****************************************************************
384 ****************************************************************/
385
386 static NTSTATUS registry_shutdown(void)
387 {
388         NTSTATUS status;
389
390         status = gpext_unregister_gp_extension(GP_EXT_NAME);
391         if (NT_STATUS_IS_OK(status)) {
392                 return status;
393         }
394
395         TALLOC_FREE(ctx);
396
397         return NT_STATUS_OK;
398 }
399
400 /****************************************************************
401 ****************************************************************/
402
403 static struct gp_extension_methods registry_methods = {
404         .initialize             = registry_initialize,
405         .process_group_policy   = registry_process_group_policy,
406         .get_reg_config         = registry_get_reg_config,
407         .shutdown               = registry_shutdown
408 };
409
410 /****************************************************************
411 ****************************************************************/
412
413 NTSTATUS gpext_registry_init(TALLOC_CTX *mem_ctx)
414 {
415         NTSTATUS status;
416
417         ctx = talloc_init("gpext_registry_init");
418         NT_STATUS_HAVE_NO_MEMORY(ctx);
419
420         status = gpext_register_gp_extension(ctx, SMB_GPEXT_INTERFACE_VERSION,
421                                              GP_EXT_NAME, GP_EXT_GUID_REGISTRY,
422                                              &registry_methods);
423         if (!NT_STATUS_IS_OK(status)) {
424                 TALLOC_FREE(ctx);
425         }
426
427         return status;
428 }