libcli/util Rename common map_nt_error_from_unix to avoid duplicate symbol
[Samba.git] / libgpo / gpext / gpext.c
blob42ab6223312cddd5d79f82fb88663b3f379afb53
1 /*
2 * Unix SMB/CIFS implementation.
3 * Group Policy Support
4 * Copyright (C) Guenther Deschner 2007-2008
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.
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.
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/>.
20 #include "includes.h"
21 #include "../libgpo/gpo.h"
22 #include "../libgpo/gpext/gpext.h"
23 #include "librpc/gen_ndr/ndr_misc.h"
24 #include "lib/util/dlinklist.h"
25 #include "../libcli/registry/util_reg.h"
26 #if _SAMBA_BUILD_ == 3
27 #include "libgpo/gpo_proto.h"
28 #include "registry.h"
29 #include "registry/reg_api.h"
30 #endif
32 static struct gp_extension *extensions = NULL;
34 /****************************************************************
35 ****************************************************************/
37 struct gp_extension *get_gp_extension_list(void)
39 return extensions;
42 /****************************************************************
43 ****************************************************************/
45 /* see http://support.microsoft.com/kb/216358/en-us/ for more info */
47 struct gp_extension_reg_table gpext_reg_vals[] = {
48 { "DllName", REG_EXPAND_SZ },
49 { "ProcessGroupPolicy", REG_SZ },
50 { "NoMachinePolicy", REG_DWORD },
51 { "NoUserPolicy", REG_DWORD },
52 { "NoSlowLink", REG_DWORD },
53 { "NoBackgroundPolicy", REG_DWORD },
54 { "NoGPOListChanges", REG_DWORD },
55 { "PerUserLocalSettings", REG_DWORD },
56 { "RequiresSuccessfulRegistry", REG_DWORD },
57 { "EnableAsynchronousProcessing", REG_DWORD },
58 { "ExtensionDebugLevel", REG_DWORD },
59 /* new */
60 { "GenerateGroupPolicy", REG_SZ }, /* not supported on w2k */
61 { "NotifyLinkTransition", REG_DWORD },
62 { "ProcessGroupPolicyEx", REG_SZ }, /* not supported on w2k */
63 { "ExtensionEventSource", REG_MULTI_SZ }, /* not supported on w2k */
64 { "GenerateGroupPolicy", REG_SZ },
65 { "MaxNoGPOListChangesInterval", REG_DWORD },
66 { NULL, REG_NONE }
69 /****************************************************************
70 ****************************************************************/
72 static struct gp_extension *get_extension_by_name(struct gp_extension *be,
73 const char *name)
75 struct gp_extension *b;
77 for (b = be; b; b = b->next) {
78 if (strequal(b->name, name)) {
79 return b;
83 return NULL;
86 /****************************************************************
87 ****************************************************************/
89 static struct gp_extension_methods *get_methods_by_name(struct gp_extension *be,
90 const char *name)
92 struct gp_extension *b;
94 for (b = be; b; b = b->next) {
95 if (strequal(b->name, name)) {
96 return b->methods;
100 return NULL;
103 /****************************************************************
104 ****************************************************************/
106 NTSTATUS unregister_gp_extension(const char *name)
108 struct gp_extension *ext;
110 ext = get_extension_by_name(extensions, name);
111 if (!ext) {
112 return NT_STATUS_OK;
115 DLIST_REMOVE(extensions, ext);
116 talloc_free(ext);
118 DEBUG(2,("Successfully removed GP extension '%s'\n", name));
120 return NT_STATUS_OK;
123 /****************************************************************
124 ****************************************************************/
126 NTSTATUS register_gp_extension(TALLOC_CTX *gpext_ctx,
127 int version,
128 const char *name,
129 const char *guid,
130 struct gp_extension_methods *methods)
132 struct gp_extension_methods *test;
133 struct gp_extension *entry;
134 NTSTATUS status;
136 if (!gpext_ctx) {
137 return NT_STATUS_INTERNAL_DB_ERROR;
140 if ((version != SMB_GPEXT_INTERFACE_VERSION)) {
141 DEBUG(0,("Failed to register gp extension.\n"
142 "The module was compiled against "
143 "SMB_GPEXT_INTERFACE_VERSION %d,\n"
144 "current SMB_GPEXT_INTERFACE_VERSION is %d.\n"
145 "Please recompile against the current "
146 "version of samba!\n",
147 version, SMB_GPEXT_INTERFACE_VERSION));
148 return NT_STATUS_OBJECT_TYPE_MISMATCH;
151 if (!guid || !name || !name[0] || !methods) {
152 DEBUG(0,("Called with NULL pointer or empty name!\n"));
153 return NT_STATUS_INVALID_PARAMETER;
156 test = get_methods_by_name(extensions, name);
157 if (test) {
158 DEBUG(0,("GP extension module %s already registered!\n",
159 name));
160 return NT_STATUS_OBJECT_NAME_COLLISION;
163 entry = talloc_zero(gpext_ctx, struct gp_extension);
164 NT_STATUS_HAVE_NO_MEMORY(entry);
166 entry->name = talloc_strdup(gpext_ctx, name);
167 NT_STATUS_HAVE_NO_MEMORY(entry->name);
169 entry->guid = talloc_zero(gpext_ctx, struct GUID);
170 NT_STATUS_HAVE_NO_MEMORY(entry->guid);
171 status = GUID_from_string(guid, entry->guid);
172 NT_STATUS_NOT_OK_RETURN(status);
174 entry->methods = methods;
175 DLIST_ADD(extensions, entry);
177 DEBUG(2,("Successfully added GP extension '%s' %s\n",
178 name, GUID_string2(gpext_ctx, entry->guid)));
180 return NT_STATUS_OK;
183 /****************************************************************
184 ****************************************************************/
186 static NTSTATUS gp_extension_init_module(TALLOC_CTX *mem_ctx,
187 const char *name,
188 struct gp_extension **gpext)
190 NTSTATUS status;
191 struct gp_extension *ext = NULL;
193 ext = talloc_zero(mem_ctx, struct gp_extension);
194 NT_STATUS_HAVE_NO_MEMORY(gpext);
196 ext->methods = get_methods_by_name(extensions, name);
197 if (!ext->methods) {
199 status = smb_probe_module(SAMBA_SUBSYSTEM_GPEXT,
200 name);
201 if (!NT_STATUS_IS_OK(status)) {
202 return status;
205 ext->methods = get_methods_by_name(extensions, name);
206 if (!ext->methods) {
207 return NT_STATUS_DLL_INIT_FAILED;
211 *gpext = ext;
213 return NT_STATUS_OK;
216 /****************************************************************
217 ****************************************************************/
219 static bool add_gp_extension_reg_entry_to_array(TALLOC_CTX *mem_ctx,
220 struct gp_extension_reg_entry *entry,
221 struct gp_extension_reg_entry **entries,
222 size_t *num)
224 *entries = talloc_realloc(mem_ctx, *entries,
225 struct gp_extension_reg_entry,
226 (*num)+1);
227 if (*entries == NULL) {
228 *num = 0;
229 return false;
232 (*entries)[*num].value = entry->value;
233 (*entries)[*num].data = entry->data;
235 *num += 1;
236 return true;
239 /****************************************************************
240 ****************************************************************/
242 static bool add_gp_extension_reg_info_entry_to_array(TALLOC_CTX *mem_ctx,
243 struct gp_extension_reg_info_entry *entry,
244 struct gp_extension_reg_info_entry **entries,
245 size_t *num)
247 *entries = talloc_realloc(mem_ctx, *entries,
248 struct gp_extension_reg_info_entry,
249 (*num)+1);
250 if (*entries == NULL) {
251 *num = 0;
252 return false;
255 (*entries)[*num].guid = entry->guid;
256 (*entries)[*num].num_entries = entry->num_entries;
257 (*entries)[*num].entries = entry->entries;
259 *num += 1;
260 return true;
263 /****************************************************************
264 ****************************************************************/
266 static NTSTATUS gp_ext_info_add_reg(TALLOC_CTX *mem_ctx,
267 struct gp_extension_reg_info_entry *entry,
268 const char *value,
269 enum winreg_Type type,
270 const char *data_s)
272 struct gp_extension_reg_entry *reg_entry = NULL;
273 struct registry_value *data = NULL;
275 reg_entry = talloc_zero(mem_ctx, struct gp_extension_reg_entry);
276 NT_STATUS_HAVE_NO_MEMORY(reg_entry);
278 data = talloc_zero(mem_ctx, struct registry_value);
279 NT_STATUS_HAVE_NO_MEMORY(data);
281 data->type = type;
283 switch (type) {
284 case REG_SZ:
285 case REG_EXPAND_SZ:
286 if (!push_reg_sz(mem_ctx, &data->data, data_s)) {
287 return NT_STATUS_NO_MEMORY;
289 break;
290 case REG_DWORD: {
291 uint32_t v = atoi(data_s);
292 data->data = data_blob_talloc(mem_ctx, NULL, 4);
293 SIVAL(data->data.data, 0, v);
294 break;
296 default:
297 return NT_STATUS_NOT_SUPPORTED;
300 reg_entry->value = value;
301 reg_entry->data = data;
303 if (!add_gp_extension_reg_entry_to_array(mem_ctx, reg_entry,
304 &entry->entries,
305 &entry->num_entries)) {
306 return NT_STATUS_NO_MEMORY;
309 return NT_STATUS_OK;
312 /****************************************************************
313 ****************************************************************/
315 static NTSTATUS gp_ext_info_add_reg_table(TALLOC_CTX *mem_ctx,
316 const char *module,
317 struct gp_extension_reg_info_entry *entry,
318 struct gp_extension_reg_table *table)
320 NTSTATUS status;
321 const char *module_name = NULL;
322 int i;
324 module_name = talloc_asprintf(mem_ctx, "%s.%s", module, shlib_ext());
325 NT_STATUS_HAVE_NO_MEMORY(module_name);
327 status = gp_ext_info_add_reg(mem_ctx, entry,
328 "DllName", REG_EXPAND_SZ, module_name);
329 NT_STATUS_NOT_OK_RETURN(status);
331 for (i=0; table[i].val; i++) {
332 status = gp_ext_info_add_reg(mem_ctx, entry,
333 table[i].val,
334 table[i].type,
335 table[i].data);
336 NT_STATUS_NOT_OK_RETURN(status);
339 return status;
342 /****************************************************************
343 ****************************************************************/
345 NTSTATUS gp_ext_info_add_entry(TALLOC_CTX *mem_ctx,
346 const char *module,
347 const char *ext_guid,
348 struct gp_extension_reg_table *table,
349 struct gp_extension_reg_info *info)
351 NTSTATUS status;
352 struct gp_extension_reg_info_entry *entry = NULL;
354 entry = talloc_zero(mem_ctx, struct gp_extension_reg_info_entry);
355 NT_STATUS_HAVE_NO_MEMORY(entry);
357 status = GUID_from_string(ext_guid, &entry->guid);
358 NT_STATUS_NOT_OK_RETURN(status);
360 status = gp_ext_info_add_reg_table(mem_ctx, module, entry, table);
361 NT_STATUS_NOT_OK_RETURN(status);
363 if (!add_gp_extension_reg_info_entry_to_array(mem_ctx, entry,
364 &info->entries,
365 &info->num_entries)) {
366 return NT_STATUS_NO_MEMORY;
369 return NT_STATUS_OK;
372 /****************************************************************
373 ****************************************************************/
375 static bool gp_extension_reg_info_verify_entry(struct gp_extension_reg_entry *entry)
377 int i;
379 for (i=0; gpext_reg_vals[i].val; i++) {
381 if ((strequal(entry->value, gpext_reg_vals[i].val)) &&
382 (entry->data->type == gpext_reg_vals[i].type)) {
383 return true;
387 return false;
390 /****************************************************************
391 ****************************************************************/
393 static bool gp_extension_reg_info_verify(struct gp_extension_reg_info_entry *entry)
395 int i;
397 for (i=0; i < entry->num_entries; i++) {
398 if (!gp_extension_reg_info_verify_entry(&entry->entries[i])) {
399 return false;
403 return true;
406 /****************************************************************
407 ****************************************************************/
409 static WERROR gp_extension_store_reg_vals(TALLOC_CTX *mem_ctx,
410 struct registry_key *key,
411 struct gp_extension_reg_info_entry *entry)
413 WERROR werr = WERR_OK;
414 size_t i;
416 for (i=0; i < entry->num_entries; i++) {
418 werr = reg_setvalue(key,
419 entry->entries[i].value,
420 entry->entries[i].data);
421 W_ERROR_NOT_OK_RETURN(werr);
424 return werr;
427 /****************************************************************
428 ****************************************************************/
430 static WERROR gp_extension_store_reg_entry(TALLOC_CTX *mem_ctx,
431 struct gp_registry_context *reg_ctx,
432 struct gp_extension_reg_info_entry *entry)
434 WERROR werr;
435 struct registry_key *key = NULL;
436 const char *subkeyname = NULL;
438 if (!gp_extension_reg_info_verify(entry)) {
439 return WERR_INVALID_PARAM;
442 subkeyname = GUID_string2(mem_ctx, &entry->guid);
443 W_ERROR_HAVE_NO_MEMORY(subkeyname);
445 strupper_m(discard_const_p(char, subkeyname));
447 werr = gp_store_reg_subkey(mem_ctx,
448 subkeyname,
449 reg_ctx->curr_key,
450 &key);
451 W_ERROR_NOT_OK_RETURN(werr);
453 werr = gp_extension_store_reg_vals(mem_ctx,
454 key,
455 entry);
456 W_ERROR_NOT_OK_RETURN(werr);
458 return werr;
461 /****************************************************************
462 ****************************************************************/
464 static WERROR gp_extension_store_reg(TALLOC_CTX *mem_ctx,
465 struct gp_registry_context *reg_ctx,
466 struct gp_extension_reg_info *info)
468 WERROR werr = WERR_OK;
469 int i;
471 if (!info) {
472 return WERR_OK;
475 for (i=0; i < info->num_entries; i++) {
476 werr = gp_extension_store_reg_entry(mem_ctx,
477 reg_ctx,
478 &info->entries[i]);
479 W_ERROR_NOT_OK_RETURN(werr);
482 return werr;
485 /****************************************************************
486 ****************************************************************/
488 static NTSTATUS gp_glob_ext_list(TALLOC_CTX *mem_ctx,
489 const char ***ext_list,
490 size_t *ext_list_len)
492 SMB_STRUCT_DIR *dir = NULL;
493 SMB_STRUCT_DIRENT *dirent = NULL;
495 dir = sys_opendir(modules_path(talloc_tos(),
496 SAMBA_SUBSYSTEM_GPEXT));
497 if (!dir) {
498 return map_nt_error_from_unix_common(errno);
501 while ((dirent = sys_readdir(dir))) {
503 fstring name; /* forgive me... */
504 char *p;
506 if ((strequal(dirent->d_name, ".")) ||
507 (strequal(dirent->d_name, ".."))) {
508 continue;
511 p = strrchr(dirent->d_name, '.');
512 if (!p) {
513 sys_closedir(dir);
514 return NT_STATUS_NO_MEMORY;
517 if (!strcsequal(p+1, shlib_ext())) {
518 DEBUG(10,("gp_glob_ext_list: not a *.so file: %s\n",
519 dirent->d_name));
520 continue;
523 fstrcpy(name, dirent->d_name);
524 name[PTR_DIFF(p, dirent->d_name)] = 0;
526 if (!add_string_to_array(mem_ctx, name, ext_list,
527 (int *)ext_list_len)) {
528 sys_closedir(dir);
529 return NT_STATUS_NO_MEMORY;
533 sys_closedir(dir);
535 return NT_STATUS_OK;
538 /****************************************************************
539 ****************************************************************/
541 NTSTATUS shutdown_gp_extensions(void)
543 struct gp_extension *ext = NULL;
545 for (ext = extensions; ext; ext = ext->next) {
546 if (ext->methods && ext->methods->shutdown) {
547 ext->methods->shutdown();
551 return NT_STATUS_OK;
554 /****************************************************************
555 ****************************************************************/
557 NTSTATUS init_gp_extensions(TALLOC_CTX *mem_ctx)
559 NTSTATUS status;
560 WERROR werr;
561 int i = 0;
562 const char **ext_array = NULL;
563 size_t ext_array_len = 0;
564 struct gp_extension *gpext = NULL;
565 struct gp_registry_context *reg_ctx = NULL;
567 if (get_gp_extension_list()) {
568 return NT_STATUS_OK;
571 status = gp_glob_ext_list(mem_ctx, &ext_array, &ext_array_len);
572 NT_STATUS_NOT_OK_RETURN(status);
574 for (i=0; i<ext_array_len; i++) {
576 struct gp_extension_reg_info *info = NULL;
578 status = gp_extension_init_module(mem_ctx, ext_array[i],
579 &gpext);
580 if (!NT_STATUS_IS_OK(status)) {
581 goto out;
584 if (gpext->methods->get_reg_config) {
586 status = gpext->methods->initialize(mem_ctx);
587 if (!NT_STATUS_IS_OK(status)) {
588 gpext->methods->shutdown();
589 goto out;
592 status = gpext->methods->get_reg_config(mem_ctx,
593 &info);
594 if (!NT_STATUS_IS_OK(status)) {
595 gpext->methods->shutdown();
596 goto out;
599 if (!reg_ctx) {
600 struct security_token *token;
602 token = registry_create_system_token(mem_ctx);
603 NT_STATUS_HAVE_NO_MEMORY(token);
605 werr = gp_init_reg_ctx(mem_ctx,
606 KEY_WINLOGON_GPEXT_PATH,
607 REG_KEY_WRITE,
608 token,
609 &reg_ctx);
610 if (!W_ERROR_IS_OK(werr)) {
611 status = werror_to_ntstatus(werr);
612 gpext->methods->shutdown();
613 goto out;
617 werr = gp_extension_store_reg(mem_ctx, reg_ctx, info);
618 if (!W_ERROR_IS_OK(werr)) {
619 DEBUG(1,("gp_extension_store_reg failed: %s\n",
620 win_errstr(werr)));
621 TALLOC_FREE(info);
622 gpext->methods->shutdown();
623 status = werror_to_ntstatus(werr);
624 goto out;
626 TALLOC_FREE(info);
631 out:
632 TALLOC_FREE(reg_ctx);
634 return status;
637 /****************************************************************
638 ****************************************************************/
640 NTSTATUS free_gp_extensions(void)
642 struct gp_extension *ext, *ext_next = NULL;
644 for (ext = extensions; ext; ext = ext_next) {
645 ext_next = ext->next;
646 DLIST_REMOVE(extensions, ext);
647 TALLOC_FREE(ext);
650 extensions = NULL;
652 return NT_STATUS_OK;
655 /****************************************************************
656 ****************************************************************/
658 void debug_gpext_header(int lvl,
659 const char *name,
660 uint32_t flags,
661 struct GROUP_POLICY_OBJECT *gpo,
662 const char *extension_guid,
663 const char *snapin_guid)
665 char *flags_str = NULL;
667 DEBUG(lvl,("%s\n", name));
668 DEBUGADD(lvl,("\tgpo: %s (%s)\n", gpo->name,
669 gpo->display_name));
670 DEBUGADD(lvl,("\tcse extension: %s (%s)\n", extension_guid,
671 cse_gpo_guid_string_to_name(extension_guid)));
672 DEBUGADD(lvl,("\tgplink: %s\n", gpo->link));
673 DEBUGADD(lvl,("\tsnapin: %s (%s)\n", snapin_guid,
674 cse_snapin_gpo_guid_string_to_name(snapin_guid)));
676 flags_str = gpo_flag_str(NULL, flags);
677 DEBUGADD(lvl,("\tflags: 0x%08x %s\n", flags, flags_str));
678 TALLOC_FREE(flags_str);
681 NTSTATUS process_gpo_list_with_extension(ADS_STRUCT *ads,
682 TALLOC_CTX *mem_ctx,
683 uint32_t flags,
684 const struct security_token *token,
685 struct GROUP_POLICY_OBJECT *gpo_list,
686 const char *extension_guid,
687 const char *snapin_guid)
689 return NT_STATUS_OK;
692 /****************************************************************
693 ****************************************************************/
695 NTSTATUS gpext_process_extension(ADS_STRUCT *ads,
696 TALLOC_CTX *mem_ctx,
697 uint32_t flags,
698 const struct security_token *token,
699 struct registry_key *root_key,
700 struct GROUP_POLICY_OBJECT *gpo,
701 const char *extension_guid,
702 const char *snapin_guid)
704 NTSTATUS status;
705 struct gp_extension *ext = NULL;
706 struct GUID guid;
707 bool cse_found = false;
709 status = init_gp_extensions(mem_ctx);
710 if (!NT_STATUS_IS_OK(status)) {
711 DEBUG(1,("init_gp_extensions failed: %s\n",
712 nt_errstr(status)));
713 return status;
716 status = GUID_from_string(extension_guid, &guid);
717 if (!NT_STATUS_IS_OK(status)) {
718 return status;
721 for (ext = extensions; ext; ext = ext->next) {
723 if (GUID_equal(ext->guid, &guid)) {
724 cse_found = true;
725 break;
729 if (!cse_found) {
730 goto no_ext;
733 status = ext->methods->initialize(mem_ctx);
734 NT_STATUS_NOT_OK_RETURN(status);
736 status = ext->methods->process_group_policy(ads,
737 mem_ctx,
738 flags,
739 root_key,
740 token,
741 gpo,
742 extension_guid,
743 snapin_guid);
744 if (!NT_STATUS_IS_OK(status)) {
745 ext->methods->shutdown();
748 return status;
750 no_ext:
751 if (flags & GPO_INFO_FLAG_VERBOSE) {
752 DEBUG(0,("process_extension: no extension available for:\n"));
753 DEBUGADD(0,("%s (%s) (snapin: %s)\n",
754 extension_guid,
755 cse_gpo_guid_string_to_name(extension_guid),
756 snapin_guid));
759 return NT_STATUS_OK;