s3-netlogon: use dsgetdcname() instead of get_dc_name()
[Samba/vl.git] / libgpo / gpext / gpext.c
blob1f86b5f5d71d4cfdccc09899293ac629d46521d4
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 #include "libgpo/gpo_proto.h"
27 #include "registry.h"
28 #include "registry/reg_api.h"
30 static struct gp_extension *extensions = NULL;
32 /****************************************************************
33 ****************************************************************/
35 struct gp_extension *get_gp_extension_list(void)
37 return extensions;
40 /****************************************************************
41 ****************************************************************/
43 /* see http://support.microsoft.com/kb/216358/en-us/ for more info */
45 struct gp_extension_reg_table gpext_reg_vals[] = {
46 { "DllName", REG_EXPAND_SZ },
47 { "ProcessGroupPolicy", REG_SZ },
48 { "NoMachinePolicy", REG_DWORD },
49 { "NoUserPolicy", REG_DWORD },
50 { "NoSlowLink", REG_DWORD },
51 { "NoBackgroundPolicy", REG_DWORD },
52 { "NoGPOListChanges", REG_DWORD },
53 { "PerUserLocalSettings", REG_DWORD },
54 { "RequiresSuccessfulRegistry", REG_DWORD },
55 { "EnableAsynchronousProcessing", REG_DWORD },
56 { "ExtensionDebugLevel", REG_DWORD },
57 /* new */
58 { "GenerateGroupPolicy", REG_SZ }, /* not supported on w2k */
59 { "NotifyLinkTransition", REG_DWORD },
60 { "ProcessGroupPolicyEx", REG_SZ }, /* not supported on w2k */
61 { "ExtensionEventSource", REG_MULTI_SZ }, /* not supported on w2k */
62 { "GenerateGroupPolicy", REG_SZ },
63 { "MaxNoGPOListChangesInterval", REG_DWORD },
64 { NULL, REG_NONE }
67 /****************************************************************
68 ****************************************************************/
70 static struct gp_extension *get_extension_by_name(struct gp_extension *be,
71 const char *name)
73 struct gp_extension *b;
75 for (b = be; b; b = b->next) {
76 if (strequal(b->name, name)) {
77 return b;
81 return NULL;
84 /****************************************************************
85 ****************************************************************/
87 static struct gp_extension_methods *get_methods_by_name(struct gp_extension *be,
88 const char *name)
90 struct gp_extension *b;
92 for (b = be; b; b = b->next) {
93 if (strequal(b->name, name)) {
94 return b->methods;
98 return NULL;
101 /****************************************************************
102 ****************************************************************/
104 NTSTATUS unregister_gp_extension(const char *name)
106 struct gp_extension *ext;
108 ext = get_extension_by_name(extensions, name);
109 if (!ext) {
110 return NT_STATUS_OK;
113 DLIST_REMOVE(extensions, ext);
114 talloc_free(ext);
116 DEBUG(2,("Successfully removed GP extension '%s'\n", name));
118 return NT_STATUS_OK;
121 /****************************************************************
122 ****************************************************************/
124 NTSTATUS register_gp_extension(TALLOC_CTX *gpext_ctx,
125 int version,
126 const char *name,
127 const char *guid,
128 struct gp_extension_methods *methods)
130 struct gp_extension_methods *test;
131 struct gp_extension *entry;
132 NTSTATUS status;
134 if (!gpext_ctx) {
135 return NT_STATUS_INTERNAL_DB_ERROR;
138 if ((version != SMB_GPEXT_INTERFACE_VERSION)) {
139 DEBUG(0,("Failed to register gp extension.\n"
140 "The module was compiled against "
141 "SMB_GPEXT_INTERFACE_VERSION %d,\n"
142 "current SMB_GPEXT_INTERFACE_VERSION is %d.\n"
143 "Please recompile against the current "
144 "version of samba!\n",
145 version, SMB_GPEXT_INTERFACE_VERSION));
146 return NT_STATUS_OBJECT_TYPE_MISMATCH;
149 if (!guid || !name || !name[0] || !methods) {
150 DEBUG(0,("Called with NULL pointer or empty name!\n"));
151 return NT_STATUS_INVALID_PARAMETER;
154 test = get_methods_by_name(extensions, name);
155 if (test) {
156 DEBUG(0,("GP extension module %s already registered!\n",
157 name));
158 return NT_STATUS_OBJECT_NAME_COLLISION;
161 entry = talloc_zero(gpext_ctx, struct gp_extension);
162 NT_STATUS_HAVE_NO_MEMORY(entry);
164 entry->name = talloc_strdup(gpext_ctx, name);
165 NT_STATUS_HAVE_NO_MEMORY(entry->name);
167 entry->guid = talloc_zero(gpext_ctx, struct GUID);
168 NT_STATUS_HAVE_NO_MEMORY(entry->guid);
169 status = GUID_from_string(guid, entry->guid);
170 NT_STATUS_NOT_OK_RETURN(status);
172 entry->methods = methods;
173 DLIST_ADD(extensions, entry);
175 DEBUG(2,("Successfully added GP extension '%s' %s\n",
176 name, GUID_string2(gpext_ctx, entry->guid)));
178 return NT_STATUS_OK;
181 /****************************************************************
182 ****************************************************************/
184 static NTSTATUS gp_extension_init_module(TALLOC_CTX *mem_ctx,
185 const char *name,
186 struct gp_extension **gpext)
188 NTSTATUS status;
189 struct gp_extension *ext = NULL;
191 ext = talloc_zero(mem_ctx, struct gp_extension);
192 NT_STATUS_HAVE_NO_MEMORY(gpext);
194 ext->methods = get_methods_by_name(extensions, name);
195 if (!ext->methods) {
197 status = smb_probe_module(SAMBA_SUBSYSTEM_GPEXT,
198 name);
199 if (!NT_STATUS_IS_OK(status)) {
200 return status;
203 ext->methods = get_methods_by_name(extensions, name);
204 if (!ext->methods) {
205 return NT_STATUS_DLL_INIT_FAILED;
209 *gpext = ext;
211 return NT_STATUS_OK;
214 /****************************************************************
215 ****************************************************************/
217 static bool add_gp_extension_reg_entry_to_array(TALLOC_CTX *mem_ctx,
218 struct gp_extension_reg_entry *entry,
219 struct gp_extension_reg_entry **entries,
220 size_t *num)
222 *entries = talloc_realloc(mem_ctx, *entries,
223 struct gp_extension_reg_entry,
224 (*num)+1);
225 if (*entries == NULL) {
226 *num = 0;
227 return false;
230 (*entries)[*num].value = entry->value;
231 (*entries)[*num].data = entry->data;
233 *num += 1;
234 return true;
237 /****************************************************************
238 ****************************************************************/
240 static bool add_gp_extension_reg_info_entry_to_array(TALLOC_CTX *mem_ctx,
241 struct gp_extension_reg_info_entry *entry,
242 struct gp_extension_reg_info_entry **entries,
243 size_t *num)
245 *entries = talloc_realloc(mem_ctx, *entries,
246 struct gp_extension_reg_info_entry,
247 (*num)+1);
248 if (*entries == NULL) {
249 *num = 0;
250 return false;
253 (*entries)[*num].guid = entry->guid;
254 (*entries)[*num].num_entries = entry->num_entries;
255 (*entries)[*num].entries = entry->entries;
257 *num += 1;
258 return true;
261 /****************************************************************
262 ****************************************************************/
264 static NTSTATUS gp_ext_info_add_reg(TALLOC_CTX *mem_ctx,
265 struct gp_extension_reg_info_entry *entry,
266 const char *value,
267 enum winreg_Type type,
268 const char *data_s)
270 struct gp_extension_reg_entry *reg_entry = NULL;
271 struct registry_value *data = NULL;
273 reg_entry = talloc_zero(mem_ctx, struct gp_extension_reg_entry);
274 NT_STATUS_HAVE_NO_MEMORY(reg_entry);
276 data = talloc_zero(mem_ctx, struct registry_value);
277 NT_STATUS_HAVE_NO_MEMORY(data);
279 data->type = type;
281 switch (type) {
282 case REG_SZ:
283 case REG_EXPAND_SZ:
284 if (!push_reg_sz(mem_ctx, &data->data, data_s)) {
285 return NT_STATUS_NO_MEMORY;
287 break;
288 case REG_DWORD: {
289 uint32_t v = atoi(data_s);
290 data->data = data_blob_talloc(mem_ctx, NULL, 4);
291 SIVAL(data->data.data, 0, v);
292 break;
294 default:
295 return NT_STATUS_NOT_SUPPORTED;
298 reg_entry->value = value;
299 reg_entry->data = data;
301 if (!add_gp_extension_reg_entry_to_array(mem_ctx, reg_entry,
302 &entry->entries,
303 &entry->num_entries)) {
304 return NT_STATUS_NO_MEMORY;
307 return NT_STATUS_OK;
310 /****************************************************************
311 ****************************************************************/
313 static NTSTATUS gp_ext_info_add_reg_table(TALLOC_CTX *mem_ctx,
314 const char *module,
315 struct gp_extension_reg_info_entry *entry,
316 struct gp_extension_reg_table *table)
318 NTSTATUS status;
319 const char *module_name = NULL;
320 int i;
322 module_name = talloc_asprintf(mem_ctx, "%s.%s", module, shlib_ext());
323 NT_STATUS_HAVE_NO_MEMORY(module_name);
325 status = gp_ext_info_add_reg(mem_ctx, entry,
326 "DllName", REG_EXPAND_SZ, module_name);
327 NT_STATUS_NOT_OK_RETURN(status);
329 for (i=0; table[i].val; i++) {
330 status = gp_ext_info_add_reg(mem_ctx, entry,
331 table[i].val,
332 table[i].type,
333 table[i].data);
334 NT_STATUS_NOT_OK_RETURN(status);
337 return status;
340 /****************************************************************
341 ****************************************************************/
343 NTSTATUS gp_ext_info_add_entry(TALLOC_CTX *mem_ctx,
344 const char *module,
345 const char *ext_guid,
346 struct gp_extension_reg_table *table,
347 struct gp_extension_reg_info *info)
349 NTSTATUS status;
350 struct gp_extension_reg_info_entry *entry = NULL;
352 entry = talloc_zero(mem_ctx, struct gp_extension_reg_info_entry);
353 NT_STATUS_HAVE_NO_MEMORY(entry);
355 status = GUID_from_string(ext_guid, &entry->guid);
356 NT_STATUS_NOT_OK_RETURN(status);
358 status = gp_ext_info_add_reg_table(mem_ctx, module, entry, table);
359 NT_STATUS_NOT_OK_RETURN(status);
361 if (!add_gp_extension_reg_info_entry_to_array(mem_ctx, entry,
362 &info->entries,
363 &info->num_entries)) {
364 return NT_STATUS_NO_MEMORY;
367 return NT_STATUS_OK;
370 /****************************************************************
371 ****************************************************************/
373 static bool gp_extension_reg_info_verify_entry(struct gp_extension_reg_entry *entry)
375 int i;
377 for (i=0; gpext_reg_vals[i].val; i++) {
379 if ((strequal(entry->value, gpext_reg_vals[i].val)) &&
380 (entry->data->type == gpext_reg_vals[i].type)) {
381 return true;
385 return false;
388 /****************************************************************
389 ****************************************************************/
391 static bool gp_extension_reg_info_verify(struct gp_extension_reg_info_entry *entry)
393 int i;
395 for (i=0; i < entry->num_entries; i++) {
396 if (!gp_extension_reg_info_verify_entry(&entry->entries[i])) {
397 return false;
401 return true;
404 /****************************************************************
405 ****************************************************************/
407 static WERROR gp_extension_store_reg_vals(TALLOC_CTX *mem_ctx,
408 struct registry_key *key,
409 struct gp_extension_reg_info_entry *entry)
411 WERROR werr = WERR_OK;
412 size_t i;
414 for (i=0; i < entry->num_entries; i++) {
416 werr = reg_setvalue(key,
417 entry->entries[i].value,
418 entry->entries[i].data);
419 W_ERROR_NOT_OK_RETURN(werr);
422 return werr;
425 /****************************************************************
426 ****************************************************************/
428 static WERROR gp_extension_store_reg_entry(TALLOC_CTX *mem_ctx,
429 struct gp_registry_context *reg_ctx,
430 struct gp_extension_reg_info_entry *entry)
432 WERROR werr;
433 struct registry_key *key = NULL;
434 const char *subkeyname = NULL;
436 if (!gp_extension_reg_info_verify(entry)) {
437 return WERR_INVALID_PARAM;
440 subkeyname = GUID_string2(mem_ctx, &entry->guid);
441 W_ERROR_HAVE_NO_MEMORY(subkeyname);
443 strupper_m(discard_const_p(char, subkeyname));
445 werr = gp_store_reg_subkey(mem_ctx,
446 subkeyname,
447 reg_ctx->curr_key,
448 &key);
449 W_ERROR_NOT_OK_RETURN(werr);
451 werr = gp_extension_store_reg_vals(mem_ctx,
452 key,
453 entry);
454 W_ERROR_NOT_OK_RETURN(werr);
456 return werr;
459 /****************************************************************
460 ****************************************************************/
462 static WERROR gp_extension_store_reg(TALLOC_CTX *mem_ctx,
463 struct gp_registry_context *reg_ctx,
464 struct gp_extension_reg_info *info)
466 WERROR werr = WERR_OK;
467 int i;
469 if (!info) {
470 return WERR_OK;
473 for (i=0; i < info->num_entries; i++) {
474 werr = gp_extension_store_reg_entry(mem_ctx,
475 reg_ctx,
476 &info->entries[i]);
477 W_ERROR_NOT_OK_RETURN(werr);
480 return werr;
483 /****************************************************************
484 ****************************************************************/
486 static NTSTATUS gp_glob_ext_list(TALLOC_CTX *mem_ctx,
487 const char ***ext_list,
488 size_t *ext_list_len)
490 SMB_STRUCT_DIR *dir = NULL;
491 SMB_STRUCT_DIRENT *dirent = NULL;
493 dir = sys_opendir(modules_path(talloc_tos(),
494 SAMBA_SUBSYSTEM_GPEXT));
495 if (!dir) {
496 return map_nt_error_from_unix_common(errno);
499 while ((dirent = sys_readdir(dir))) {
501 fstring name; /* forgive me... */
502 char *p;
504 if ((strequal(dirent->d_name, ".")) ||
505 (strequal(dirent->d_name, ".."))) {
506 continue;
509 p = strrchr(dirent->d_name, '.');
510 if (!p) {
511 sys_closedir(dir);
512 return NT_STATUS_NO_MEMORY;
515 if (!strcsequal(p+1, shlib_ext())) {
516 DEBUG(10,("gp_glob_ext_list: not a *.so file: %s\n",
517 dirent->d_name));
518 continue;
521 fstrcpy(name, dirent->d_name);
522 name[PTR_DIFF(p, dirent->d_name)] = 0;
524 if (!add_string_to_array(mem_ctx, name, ext_list,
525 (int *)ext_list_len)) {
526 sys_closedir(dir);
527 return NT_STATUS_NO_MEMORY;
531 sys_closedir(dir);
533 return NT_STATUS_OK;
536 /****************************************************************
537 ****************************************************************/
539 NTSTATUS shutdown_gp_extensions(void)
541 struct gp_extension *ext = NULL;
543 for (ext = extensions; ext; ext = ext->next) {
544 if (ext->methods && ext->methods->shutdown) {
545 ext->methods->shutdown();
549 return NT_STATUS_OK;
552 /****************************************************************
553 ****************************************************************/
555 NTSTATUS init_gp_extensions(TALLOC_CTX *mem_ctx)
557 NTSTATUS status;
558 WERROR werr;
559 int i = 0;
560 const char **ext_array = NULL;
561 size_t ext_array_len = 0;
562 struct gp_extension *gpext = NULL;
563 struct gp_registry_context *reg_ctx = NULL;
565 if (get_gp_extension_list()) {
566 return NT_STATUS_OK;
569 status = gp_glob_ext_list(mem_ctx, &ext_array, &ext_array_len);
570 NT_STATUS_NOT_OK_RETURN(status);
572 for (i=0; i<ext_array_len; i++) {
574 struct gp_extension_reg_info *info = NULL;
576 status = gp_extension_init_module(mem_ctx, ext_array[i],
577 &gpext);
578 if (!NT_STATUS_IS_OK(status)) {
579 goto out;
582 if (gpext->methods->get_reg_config) {
584 status = gpext->methods->initialize(mem_ctx);
585 if (!NT_STATUS_IS_OK(status)) {
586 gpext->methods->shutdown();
587 goto out;
590 status = gpext->methods->get_reg_config(mem_ctx,
591 &info);
592 if (!NT_STATUS_IS_OK(status)) {
593 gpext->methods->shutdown();
594 goto out;
597 if (!reg_ctx) {
598 struct security_token *token;
600 token = registry_create_system_token(mem_ctx);
601 NT_STATUS_HAVE_NO_MEMORY(token);
603 werr = gp_init_reg_ctx(mem_ctx,
604 KEY_WINLOGON_GPEXT_PATH,
605 REG_KEY_WRITE,
606 token,
607 &reg_ctx);
608 if (!W_ERROR_IS_OK(werr)) {
609 status = werror_to_ntstatus(werr);
610 gpext->methods->shutdown();
611 goto out;
615 werr = gp_extension_store_reg(mem_ctx, reg_ctx, info);
616 if (!W_ERROR_IS_OK(werr)) {
617 DEBUG(1,("gp_extension_store_reg failed: %s\n",
618 win_errstr(werr)));
619 TALLOC_FREE(info);
620 gpext->methods->shutdown();
621 status = werror_to_ntstatus(werr);
622 goto out;
624 TALLOC_FREE(info);
629 out:
630 TALLOC_FREE(reg_ctx);
632 return status;
635 /****************************************************************
636 ****************************************************************/
638 NTSTATUS free_gp_extensions(void)
640 struct gp_extension *ext, *ext_next = NULL;
642 for (ext = extensions; ext; ext = ext_next) {
643 ext_next = ext->next;
644 DLIST_REMOVE(extensions, ext);
645 TALLOC_FREE(ext);
648 extensions = NULL;
650 return NT_STATUS_OK;
653 /****************************************************************
654 ****************************************************************/
656 void debug_gpext_header(int lvl,
657 const char *name,
658 uint32_t flags,
659 struct GROUP_POLICY_OBJECT *gpo,
660 const char *extension_guid,
661 const char *snapin_guid)
663 char *flags_str = NULL;
665 DEBUG(lvl,("%s\n", name));
666 DEBUGADD(lvl,("\tgpo: %s (%s)\n", gpo->name,
667 gpo->display_name));
668 DEBUGADD(lvl,("\tcse extension: %s (%s)\n", extension_guid,
669 cse_gpo_guid_string_to_name(extension_guid)));
670 DEBUGADD(lvl,("\tgplink: %s\n", gpo->link));
671 DEBUGADD(lvl,("\tsnapin: %s (%s)\n", snapin_guid,
672 cse_snapin_gpo_guid_string_to_name(snapin_guid)));
674 flags_str = gpo_flag_str(NULL, flags);
675 DEBUGADD(lvl,("\tflags: 0x%08x %s\n", flags, flags_str));
676 TALLOC_FREE(flags_str);
679 NTSTATUS process_gpo_list_with_extension(ADS_STRUCT *ads,
680 TALLOC_CTX *mem_ctx,
681 uint32_t flags,
682 const struct security_token *token,
683 struct GROUP_POLICY_OBJECT *gpo_list,
684 const char *extension_guid,
685 const char *snapin_guid)
687 return NT_STATUS_OK;
690 /****************************************************************
691 ****************************************************************/
693 NTSTATUS gpext_process_extension(ADS_STRUCT *ads,
694 TALLOC_CTX *mem_ctx,
695 uint32_t flags,
696 const struct security_token *token,
697 struct registry_key *root_key,
698 struct GROUP_POLICY_OBJECT *gpo,
699 const char *extension_guid,
700 const char *snapin_guid)
702 NTSTATUS status;
703 struct gp_extension *ext = NULL;
704 struct GUID guid;
705 bool cse_found = false;
707 status = init_gp_extensions(mem_ctx);
708 if (!NT_STATUS_IS_OK(status)) {
709 DEBUG(1,("init_gp_extensions failed: %s\n",
710 nt_errstr(status)));
711 return status;
714 status = GUID_from_string(extension_guid, &guid);
715 if (!NT_STATUS_IS_OK(status)) {
716 return status;
719 for (ext = extensions; ext; ext = ext->next) {
721 if (GUID_equal(ext->guid, &guid)) {
722 cse_found = true;
723 break;
727 if (!cse_found) {
728 goto no_ext;
731 status = ext->methods->initialize(mem_ctx);
732 NT_STATUS_NOT_OK_RETURN(status);
734 status = ext->methods->process_group_policy(ads,
735 mem_ctx,
736 flags,
737 root_key,
738 token,
739 gpo,
740 extension_guid,
741 snapin_guid);
742 if (!NT_STATUS_IS_OK(status)) {
743 ext->methods->shutdown();
746 return status;
748 no_ext:
749 if (flags & GPO_INFO_FLAG_VERBOSE) {
750 DEBUG(0,("process_extension: no extension available for:\n"));
751 DEBUGADD(0,("%s (%s) (snapin: %s)\n",
752 extension_guid,
753 cse_gpo_guid_string_to_name(extension_guid),
754 snapin_guid));
757 return NT_STATUS_OK;