2 * Unix SMB/CIFS implementation.
3 * Virtual Windows Registry Layer
4 * Copyright (C) Volker Lendecke 2006
5 * Copyright (C) Michael Adam 2007-2008
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.
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.
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/>.
21 /* Attempt to wrap the existing API in a more winreg.idl-like way */
24 * Here is a list of winreg.idl functions and corresponding implementations
27 * 0x00 winreg_OpenHKCR
28 * 0x01 winreg_OpenHKCU
29 * 0x02 winreg_OpenHKLM
30 * 0x03 winreg_OpenHKPD
32 * 0x05 winreg_CloseKey
33 * 0x06 winreg_CreateKey reg_createkey
34 * 0x07 winreg_DeleteKey reg_deletekey
35 * 0x08 winreg_DeleteValue reg_deletevalue
36 * 0x09 winreg_EnumKey reg_enumkey
37 * 0x0a winreg_EnumValue reg_enumvalue
38 * 0x0b winreg_FlushKey
39 * 0x0c winreg_GetKeySecurity reg_getkeysecurity
41 * 0x0e winreg_NotifyChangeKeyValue
42 * 0x0f winreg_OpenKey reg_openkey
43 * 0x10 winreg_QueryInfoKey reg_queryinfokey
44 * 0x11 winreg_QueryValue reg_queryvalue
45 * 0x12 winreg_ReplaceKey
46 * 0x13 winreg_RestoreKey reg_restorekey
47 * 0x14 winreg_SaveKey reg_savekey
48 * 0x15 winreg_SetKeySecurity reg_setkeysecurity
49 * 0x16 winreg_SetValue reg_setvalue
50 * 0x17 winreg_UnLoadKey
51 * 0x18 winreg_InitiateSystemShutdown
52 * 0x19 winreg_AbortSystemShutdown
53 * 0x1a winreg_GetVersion reg_getversion
54 * 0x1b winreg_OpenHKCC
55 * 0x1c winreg_OpenHKDD
56 * 0x1d winreg_QueryMultipleValues
57 * 0x1e winreg_InitiateSystemShutdownEx
58 * 0x1f winreg_SaveKeyEx
59 * 0x20 winreg_OpenHKPT
60 * 0x21 winreg_OpenHKPN
61 * 0x22 winreg_QueryMultipleValues2
69 #define DBGC_CLASS DBGC_REGISTRY
72 /**********************************************************************
74 **********************************************************************/
76 static WERROR
fill_value_cache(struct registry_key
*key
)
78 if (key
->values
!= NULL
) {
79 if (!reg_values_need_update(key
->key
, key
->values
)) {
84 if (!(key
->values
= TALLOC_ZERO_P(key
, REGVAL_CTR
))) {
87 if (fetch_reg_values(key
->key
, key
->values
) == -1) {
88 TALLOC_FREE(key
->values
);
95 static WERROR
fill_subkey_cache(struct registry_key
*key
)
97 if (key
->subkeys
!= NULL
) {
98 if (!reg_subkeys_need_update(key
->key
, key
->subkeys
)) {
103 if (!(key
->subkeys
= TALLOC_ZERO_P(key
, REGSUBKEY_CTR
))) {
107 if (fetch_reg_keys(key
->key
, key
->subkeys
) == -1) {
108 TALLOC_FREE(key
->subkeys
);
109 return WERR_NO_MORE_ITEMS
;
115 static int regkey_destructor(REGISTRY_KEY
*key
)
117 return regdb_close();
120 static WERROR
regkey_open_onelevel(TALLOC_CTX
*mem_ctx
,
121 struct registry_key
*parent
,
123 const struct nt_user_token
*token
,
124 uint32 access_desired
,
125 struct registry_key
**pregkey
)
127 WERROR result
= WERR_OK
;
128 struct registry_key
*regkey
;
130 REGSUBKEY_CTR
*subkeys
= NULL
;
132 DEBUG(7,("regkey_open_onelevel: name = [%s]\n", name
));
134 SMB_ASSERT(strchr(name
, '\\') == NULL
);
136 if (!(regkey
= TALLOC_ZERO_P(mem_ctx
, struct registry_key
)) ||
137 !(regkey
->token
= dup_nt_token(regkey
, token
)) ||
138 !(regkey
->key
= TALLOC_ZERO_P(regkey
, REGISTRY_KEY
))) {
143 if ( !(W_ERROR_IS_OK(result
= regdb_open())) ) {
148 talloc_set_destructor(key
, regkey_destructor
);
152 key
->type
= REG_KEY_GENERIC
;
154 if (name
[0] == '\0') {
156 * Open a copy of the parent key
159 result
= WERR_BADFILE
;
162 key
->name
= talloc_strdup(key
, parent
->key
->name
);
168 key
->name
= talloc_asprintf(key
, "%s%s%s",
169 parent
? parent
->key
->name
: "",
174 if (key
->name
== NULL
) {
179 /* Tag this as a Performance Counter Key */
181 if( StrnCaseCmp(key
->name
, KEY_HKPD
, strlen(KEY_HKPD
)) == 0 )
182 key
->type
= REG_KEY_HKPD
;
184 /* Look up the table of registry I/O operations */
186 if ( !(key
->hook
= reghook_cache_find( key
->name
)) ) {
187 DEBUG(0,("reg_open_onelevel: Failed to assign a "
188 "REGISTRY_HOOK to [%s]\n", key
->name
));
189 result
= WERR_BADFILE
;
193 /* check if the path really exists; failed is indicated by -1 */
194 /* if the subkey count failed, bail out */
196 if ( !(subkeys
= TALLOC_ZERO_P( key
, REGSUBKEY_CTR
)) ) {
201 if ( fetch_reg_keys( key
, subkeys
) == -1 ) {
202 result
= WERR_BADFILE
;
206 TALLOC_FREE( subkeys
);
208 if ( !regkey_access_check( key
, access_desired
, &key
->access_granted
,
210 result
= WERR_ACCESS_DENIED
;
218 if ( !W_ERROR_IS_OK(result
) ) {
225 WERROR
reg_openhive(TALLOC_CTX
*mem_ctx
, const char *hive
,
226 uint32 desired_access
,
227 const struct nt_user_token
*token
,
228 struct registry_key
**pkey
)
230 SMB_ASSERT(hive
!= NULL
);
231 SMB_ASSERT(hive
[0] != '\0');
232 SMB_ASSERT(strchr(hive
, '\\') == NULL
);
234 return regkey_open_onelevel(mem_ctx
, NULL
, hive
, token
, desired_access
,
239 /**********************************************************************
241 **********************************************************************/
243 WERROR
reg_openkey(TALLOC_CTX
*mem_ctx
, struct registry_key
*parent
,
244 const char *name
, uint32 desired_access
,
245 struct registry_key
**pkey
)
247 struct registry_key
*direct_parent
= parent
;
249 char *p
, *path
, *to_free
;
252 if (!(path
= SMB_STRDUP(name
))) {
259 if ((len
> 0) && (path
[len
-1] == '\\')) {
263 while ((p
= strchr(path
, '\\')) != NULL
) {
264 char *name_component
;
265 struct registry_key
*tmp
;
267 if (!(name_component
= SMB_STRNDUP(path
, (p
- path
)))) {
272 err
= regkey_open_onelevel(mem_ctx
, direct_parent
,
273 name_component
, parent
->token
,
274 SEC_RIGHTS_ENUM_SUBKEYS
, &tmp
);
275 SAFE_FREE(name_component
);
277 if (!W_ERROR_IS_OK(err
)) {
280 if (direct_parent
!= parent
) {
281 TALLOC_FREE(direct_parent
);
288 err
= regkey_open_onelevel(mem_ctx
, direct_parent
, path
, parent
->token
,
289 desired_access
, pkey
);
291 if (direct_parent
!= parent
) {
292 TALLOC_FREE(direct_parent
);
298 WERROR
reg_enumkey(TALLOC_CTX
*mem_ctx
, struct registry_key
*key
,
299 uint32 idx
, char **name
, NTTIME
*last_write_time
)
303 if (!(key
->key
->access_granted
& SEC_RIGHTS_ENUM_SUBKEYS
)) {
304 return WERR_ACCESS_DENIED
;
307 if (!W_ERROR_IS_OK(err
= fill_subkey_cache(key
))) {
311 if (idx
>= key
->subkeys
->num_subkeys
) {
312 return WERR_NO_MORE_ITEMS
;
315 if (!(*name
= talloc_strdup(mem_ctx
, key
->subkeys
->subkeys
[idx
]))) {
319 if (last_write_time
) {
320 *last_write_time
= 0;
326 WERROR
reg_enumvalue(TALLOC_CTX
*mem_ctx
, struct registry_key
*key
,
327 uint32 idx
, char **pname
, struct registry_value
**pval
)
329 struct registry_value
*val
;
332 if (!(key
->key
->access_granted
& SEC_RIGHTS_QUERY_VALUE
)) {
333 return WERR_ACCESS_DENIED
;
336 if (!(W_ERROR_IS_OK(err
= fill_value_cache(key
)))) {
340 if (idx
>= key
->values
->num_values
) {
341 return WERR_NO_MORE_ITEMS
;
344 err
= registry_pull_value(mem_ctx
, &val
,
345 key
->values
->values
[idx
]->type
,
346 key
->values
->values
[idx
]->data_p
,
347 key
->values
->values
[idx
]->size
,
348 key
->values
->values
[idx
]->size
);
349 if (!W_ERROR_IS_OK(err
)) {
354 && !(*pname
= talloc_strdup(
355 mem_ctx
, key
->values
->values
[idx
]->valuename
))) {
364 WERROR
reg_queryvalue(TALLOC_CTX
*mem_ctx
, struct registry_key
*key
,
365 const char *name
, struct registry_value
**pval
)
370 if (!(key
->key
->access_granted
& SEC_RIGHTS_QUERY_VALUE
)) {
371 return WERR_ACCESS_DENIED
;
374 if (!(W_ERROR_IS_OK(err
= fill_value_cache(key
)))) {
378 for (i
=0; i
<key
->values
->num_values
; i
++) {
379 if (strequal(key
->values
->values
[i
]->valuename
, name
)) {
380 return reg_enumvalue(mem_ctx
, key
, i
, NULL
, pval
);
387 WERROR
reg_queryinfokey(struct registry_key
*key
, uint32_t *num_subkeys
,
388 uint32_t *max_subkeylen
, uint32_t *max_subkeysize
,
389 uint32_t *num_values
, uint32_t *max_valnamelen
,
390 uint32_t *max_valbufsize
, uint32_t *secdescsize
,
391 NTTIME
*last_changed_time
)
397 struct security_descriptor
*secdesc
;
399 if (!(key
->key
->access_granted
& SEC_RIGHTS_QUERY_VALUE
)) {
400 return WERR_ACCESS_DENIED
;
403 if (!W_ERROR_IS_OK(fill_subkey_cache(key
)) ||
404 !W_ERROR_IS_OK(fill_value_cache(key
))) {
409 for (i
=0; i
<key
->subkeys
->num_subkeys
; i
++) {
410 max_len
= MAX(max_len
, strlen(key
->subkeys
->subkeys
[i
]));
413 *num_subkeys
= key
->subkeys
->num_subkeys
;
414 *max_subkeylen
= max_len
;
415 *max_subkeysize
= 0; /* Class length? */
419 for (i
=0; i
<key
->values
->num_values
; i
++) {
420 max_len
= MAX(max_len
,
421 strlen(key
->values
->values
[i
]->valuename
));
422 max_size
= MAX(max_size
, key
->values
->values
[i
]->size
);
425 *num_values
= key
->values
->num_values
;
426 *max_valnamelen
= max_len
;
427 *max_valbufsize
= max_size
;
429 if (!(mem_ctx
= talloc_new(key
))) {
433 err
= regkey_get_secdesc(mem_ctx
, key
->key
, &secdesc
);
434 if (!W_ERROR_IS_OK(err
)) {
435 TALLOC_FREE(mem_ctx
);
439 *secdescsize
= ndr_size_security_descriptor(secdesc
, 0);
440 TALLOC_FREE(mem_ctx
);
442 *last_changed_time
= 0;
447 WERROR
reg_createkey(TALLOC_CTX
*ctx
, struct registry_key
*parent
,
448 const char *subkeypath
, uint32 desired_access
,
449 struct registry_key
**pkey
,
450 enum winreg_CreateAction
*paction
)
452 struct registry_key
*key
= parent
;
453 struct registry_key
*create_parent
;
458 if (!(mem_ctx
= talloc_new(ctx
))) return WERR_NOMEM
;
460 if (!(path
= talloc_strdup(mem_ctx
, subkeypath
))) {
465 while ((end
= strchr(path
, '\\')) != NULL
) {
466 struct registry_key
*tmp
;
467 enum winreg_CreateAction action
;
471 err
= reg_createkey(mem_ctx
, key
, path
,
472 SEC_RIGHTS_ENUM_SUBKEYS
, &tmp
, &action
);
473 if (!W_ERROR_IS_OK(err
)) {
486 * At this point, "path" contains the one-element subkey of "key". We
487 * can try to open it.
490 err
= reg_openkey(ctx
, key
, path
, desired_access
, pkey
);
491 if (W_ERROR_IS_OK(err
)) {
492 if (paction
!= NULL
) {
493 *paction
= REG_OPENED_EXISTING_KEY
;
498 if (!W_ERROR_EQUAL(err
, WERR_BADFILE
)) {
500 * Something but "notfound" has happened, so bail out
506 * We have to make a copy of the current key, as we opened it only
507 * with ENUM_SUBKEY access.
510 err
= reg_openkey(mem_ctx
, key
, "", SEC_RIGHTS_CREATE_SUBKEY
,
512 if (!W_ERROR_IS_OK(err
)) {
517 * Actually create the subkey
520 err
= fill_subkey_cache(create_parent
);
521 if (!W_ERROR_IS_OK(err
)) goto done
;
523 err
= regsubkey_ctr_addkey(create_parent
->subkeys
, path
);
524 if (!W_ERROR_IS_OK(err
)) goto done
;
526 if (!store_reg_keys(create_parent
->key
, create_parent
->subkeys
)) {
527 TALLOC_FREE(create_parent
->subkeys
);
528 err
= WERR_REG_IO_FAILURE
;
533 * Now open the newly created key
536 err
= reg_openkey(ctx
, create_parent
, path
, desired_access
, pkey
);
537 if (W_ERROR_IS_OK(err
) && (paction
!= NULL
)) {
538 *paction
= REG_CREATED_NEW_KEY
;
542 TALLOC_FREE(mem_ctx
);
546 WERROR
reg_deletekey(struct registry_key
*parent
, const char *path
)
552 struct registry_key
*tmp_key
, *key
;
554 if (!(mem_ctx
= talloc_init("reg_createkey"))) return WERR_NOMEM
;
556 if (!(name
= talloc_strdup(mem_ctx
, path
))) {
561 /* check if the key has subkeys */
562 err
= reg_openkey(mem_ctx
, parent
, name
, REG_KEY_READ
, &key
);
563 if (!W_ERROR_IS_OK(err
)) {
566 if (!W_ERROR_IS_OK(err
= fill_subkey_cache(key
))) {
569 if (key
->subkeys
->num_subkeys
> 0) {
570 err
= WERR_ACCESS_DENIED
;
574 /* no subkeys - proceed with delete */
575 if ((end
= strrchr(name
, '\\')) != NULL
) {
578 err
= reg_openkey(mem_ctx
, parent
, name
,
579 SEC_RIGHTS_CREATE_SUBKEY
, &tmp_key
);
580 if (!W_ERROR_IS_OK(err
)) {
588 if (name
[0] == '\0') {
589 err
= WERR_INVALID_PARAM
;
593 if (!W_ERROR_IS_OK(err
= fill_subkey_cache(parent
))) {
597 num_subkeys
= parent
->subkeys
->num_subkeys
;
599 if (regsubkey_ctr_delkey(parent
->subkeys
, name
) == num_subkeys
) {
604 if (!store_reg_keys(parent
->key
, parent
->subkeys
)) {
605 TALLOC_FREE(parent
->subkeys
);
606 err
= WERR_REG_IO_FAILURE
;
610 regkey_set_secdesc(key
->key
, NULL
);
615 TALLOC_FREE(mem_ctx
);
619 WERROR
reg_setvalue(struct registry_key
*key
, const char *name
,
620 const struct registry_value
*val
)
623 DATA_BLOB value_data
;
626 if (!(key
->key
->access_granted
& SEC_RIGHTS_SET_VALUE
)) {
627 return WERR_ACCESS_DENIED
;
630 if (!W_ERROR_IS_OK(err
= fill_value_cache(key
))) {
634 err
= registry_push_value(key
, val
, &value_data
);
635 if (!W_ERROR_IS_OK(err
)) {
639 res
= regval_ctr_addvalue(key
->values
, name
, val
->type
,
640 (char *)value_data
.data
, value_data
.length
);
641 TALLOC_FREE(value_data
.data
);
644 TALLOC_FREE(key
->values
);
648 if (!store_reg_values(key
->key
, key
->values
)) {
649 TALLOC_FREE(key
->values
);
650 return WERR_REG_IO_FAILURE
;
656 static WERROR
reg_value_exists(struct registry_key
*key
, const char *name
)
660 for (i
=0; i
<key
->values
->num_values
; i
++) {
661 if (strequal(key
->values
->values
[i
]->valuename
, name
)) {
669 WERROR
reg_deletevalue(struct registry_key
*key
, const char *name
)
673 if (!(key
->key
->access_granted
& SEC_RIGHTS_SET_VALUE
)) {
674 return WERR_ACCESS_DENIED
;
677 if (!W_ERROR_IS_OK(err
= fill_value_cache(key
))) {
681 err
= reg_value_exists(key
, name
);
682 if (!W_ERROR_IS_OK(err
)) {
686 regval_ctr_delvalue(key
->values
, name
);
688 if (!store_reg_values(key
->key
, key
->values
)) {
689 TALLOC_FREE(key
->values
);
690 return WERR_REG_IO_FAILURE
;
696 WERROR
reg_getkeysecurity(TALLOC_CTX
*mem_ctx
, struct registry_key
*key
,
697 struct security_descriptor
**psecdesc
)
699 return regkey_get_secdesc(mem_ctx
, key
->key
, psecdesc
);
702 WERROR
reg_setkeysecurity(struct registry_key
*key
,
703 struct security_descriptor
*psecdesc
)
705 return regkey_set_secdesc(key
->key
, psecdesc
);
708 WERROR
reg_getversion(uint32_t *version
)
710 if (version
== NULL
) {
711 return WERR_INVALID_PARAM
;
714 *version
= 0x00000005; /* Windows 2000 registry API version */
718 /*******************************************************************
719 Note: topkeypat is the *full* path that this *key will be
720 loaded into (including the name of the key)
721 ********************************************************************/
723 static WERROR
reg_load_tree(REGF_FILE
*regfile
, const char *topkeypath
,
727 REGISTRY_KEY registry_key
;
729 REGSUBKEY_CTR
*subkeys
;
732 WERROR result
= WERR_OK
;
734 /* initialize the REGISTRY_KEY structure */
736 registry_key
.hook
= reghook_cache_find(topkeypath
);
737 if (!registry_key
.hook
) {
738 DEBUG(0, ("reg_load_tree: Failed to assigned a REGISTRY_HOOK "
739 "to [%s]\n", topkeypath
));
743 registry_key
.name
= talloc_strdup(regfile
->mem_ctx
, topkeypath
);
744 if (!registry_key
.name
) {
745 DEBUG(0, ("reg_load_tree: Talloc failed for reg_key.name!\n"));
749 /* now start parsing the values and subkeys */
751 subkeys
= TALLOC_ZERO_P(regfile
->mem_ctx
, REGSUBKEY_CTR
);
752 if (subkeys
== NULL
) {
756 values
= TALLOC_ZERO_P(subkeys
, REGVAL_CTR
);
757 if (values
== NULL
) {
761 /* copy values into the REGVAL_CTR */
763 for (i
=0; i
<key
->num_values
; i
++) {
764 regval_ctr_addvalue(values
, key
->values
[i
].valuename
,
766 (char*)key
->values
[i
].data
,
767 (key
->values
[i
].data_size
& ~VK_DATA_IN_OFFSET
));
770 /* copy subkeys into the REGSUBKEY_CTR */
772 key
->subkey_index
= 0;
773 while ((subkey
= regfio_fetch_subkey( regfile
, key
))) {
774 regsubkey_ctr_addkey(subkeys
, subkey
->keyname
);
777 /* write this key and values out */
779 if (!store_reg_values(®istry_key
, values
)
780 || !store_reg_keys(®istry_key
, subkeys
))
782 DEBUG(0,("reg_load_tree: Failed to load %s!\n", topkeypath
));
783 result
= WERR_REG_IO_FAILURE
;
786 TALLOC_FREE(subkeys
);
788 if (!W_ERROR_IS_OK(result
)) {
792 /* now continue to load each subkey registry tree */
794 key
->subkey_index
= 0;
795 while ((subkey
= regfio_fetch_subkey(regfile
, key
))) {
796 path
= talloc_asprintf(regfile
->mem_ctx
,
803 result
= reg_load_tree(regfile
, path
, subkey
);
804 if (!W_ERROR_IS_OK(result
)) {
812 /*******************************************************************
813 ********************************************************************/
815 static WERROR
restore_registry_key(REGISTRY_KEY
*krecord
, const char *fname
)
818 REGF_NK_REC
*rootkey
;
821 /* open the registry file....fail if the file already exists */
823 regfile
= regfio_open(fname
, (O_RDONLY
), 0);
824 if (regfile
== NULL
) {
825 DEBUG(0, ("restore_registry_key: failed to open \"%s\" (%s)\n",
826 fname
, strerror(errno
)));
827 return ntstatus_to_werror(map_nt_error_from_unix(errno
));
830 /* get the rootkey from the regf file and then load the tree
831 via recursive calls */
833 if (!(rootkey
= regfio_rootkey(regfile
))) {
834 regfio_close(regfile
);
835 return WERR_REG_FILE_INVALID
;
838 result
= reg_load_tree(regfile
, krecord
->name
, rootkey
);
842 regfio_close(regfile
);
847 WERROR
reg_restorekey(struct registry_key
*key
, const char *fname
)
849 return restore_registry_key(key
->key
, fname
);
852 /********************************************************************
853 ********************************************************************/
855 static WERROR
reg_write_tree(REGF_FILE
*regfile
, const char *keypath
,
860 REGSUBKEY_CTR
*subkeys
;
862 char *key_tmp
= NULL
;
863 char *keyname
, *parentpath
;
864 char *subkeypath
= NULL
;
866 REGISTRY_KEY registry_key
;
867 WERROR result
= WERR_OK
;
868 SEC_DESC
*sec_desc
= NULL
;
871 return WERR_GENERAL_FAILURE
;
875 return WERR_OBJECT_PATH_INVALID
;
878 /* split up the registry key path */
880 key_tmp
= talloc_strdup(regfile
->mem_ctx
, keypath
);
884 if (!reg_split_key(key_tmp
, &parentpath
, &keyname
)) {
885 return WERR_OBJECT_PATH_INVALID
;
889 keyname
= parentpath
;
892 /* we need a REGISTRY_KEY object here to enumerate subkeys and values */
894 ZERO_STRUCT(registry_key
);
896 registry_key
.name
= talloc_strdup(regfile
->mem_ctx
, keypath
);
897 if (registry_key
.name
== NULL
) {
901 registry_key
.hook
= reghook_cache_find(registry_key
.name
);
902 if (registry_key
.hook
== NULL
) {
906 /* lookup the values and subkeys */
908 subkeys
= TALLOC_ZERO_P(regfile
->mem_ctx
, REGSUBKEY_CTR
);
909 if (subkeys
== NULL
) {
913 values
= TALLOC_ZERO_P(subkeys
, REGVAL_CTR
);
914 if (values
== NULL
) {
918 fetch_reg_keys(®istry_key
, subkeys
);
919 fetch_reg_values(®istry_key
, values
);
921 result
= regkey_get_secdesc(regfile
->mem_ctx
, ®istry_key
, &sec_desc
);
922 if (!W_ERROR_IS_OK(result
)) {
926 /* write out this key */
928 key
= regfio_write_key(regfile
, keyname
, values
, subkeys
, sec_desc
,
931 result
= WERR_CAN_NOT_COMPLETE
;
935 /* write each one of the subkeys out */
937 num_subkeys
= regsubkey_ctr_numkeys(subkeys
);
938 for (i
=0; i
<num_subkeys
; i
++) {
939 subkeyname
= regsubkey_ctr_specific_key(subkeys
, i
);
940 subkeypath
= talloc_asprintf(regfile
->mem_ctx
, "%s\\%s",
941 keypath
, subkeyname
);
942 if (subkeypath
== NULL
) {
946 result
= reg_write_tree(regfile
, subkeypath
, key
);
947 if (!W_ERROR_IS_OK(result
))
951 DEBUG(6, ("reg_write_tree: wrote key [%s]\n", keypath
));
954 TALLOC_FREE(subkeys
);
955 TALLOC_FREE(registry_key
.name
);
960 static WERROR
backup_registry_key(REGISTRY_KEY
*krecord
, const char *fname
)
965 /* open the registry file....fail if the file already exists */
967 regfile
= regfio_open(fname
, (O_RDWR
|O_CREAT
|O_EXCL
),
969 if (regfile
== NULL
) {
970 DEBUG(0,("backup_registry_key: failed to open \"%s\" (%s)\n",
971 fname
, strerror(errno
) ));
972 return ntstatus_to_werror(map_nt_error_from_unix(errno
));
975 /* write the registry tree to the file */
977 result
= reg_write_tree(regfile
, krecord
->name
, NULL
);
981 regfio_close(regfile
);
986 WERROR
reg_savekey(struct registry_key
*key
, const char *fname
)
988 return backup_registry_key(key
->key
, fname
);
991 /**********************************************************************
992 * Higher level utility functions
993 **********************************************************************/
995 WERROR
reg_deleteallvalues(struct registry_key
*key
)
1000 if (!(key
->key
->access_granted
& SEC_RIGHTS_SET_VALUE
)) {
1001 return WERR_ACCESS_DENIED
;
1004 if (!W_ERROR_IS_OK(err
= fill_value_cache(key
))) {
1008 for (i
=0; i
<key
->values
->num_values
; i
++) {
1009 regval_ctr_delvalue(key
->values
, key
->values
->values
[i
]->valuename
);
1012 if (!store_reg_values(key
->key
, key
->values
)) {
1013 TALLOC_FREE(key
->values
);
1014 return WERR_REG_IO_FAILURE
;
1021 * Utility function to open a complete registry path including the hive prefix.
1024 WERROR
reg_open_path(TALLOC_CTX
*mem_ctx
, const char *orig_path
,
1025 uint32 desired_access
, const struct nt_user_token
*token
,
1026 struct registry_key
**pkey
)
1028 struct registry_key
*hive
, *key
;
1032 if (!(path
= SMB_STRDUP(orig_path
))) {
1036 p
= strchr(path
, '\\');
1038 if ((p
== NULL
) || (p
[1] == '\0')) {
1040 * No key behind the hive, just return the hive
1043 err
= reg_openhive(mem_ctx
, path
, desired_access
, token
,
1045 if (!W_ERROR_IS_OK(err
)) {
1056 err
= reg_openhive(mem_ctx
, path
, SEC_RIGHTS_ENUM_SUBKEYS
, token
,
1058 if (!W_ERROR_IS_OK(err
)) {
1063 err
= reg_openkey(mem_ctx
, hive
, p
+1, desired_access
, &key
);
1068 if (!W_ERROR_IS_OK(err
)) {
1077 * Utility function to delete a registry key with all its subkeys.
1078 * Note that reg_deletekey returns ACCESS_DENIED when called on a
1079 * key that has subkeys.
1081 static WERROR
reg_deletekey_recursive_internal(TALLOC_CTX
*ctx
,
1082 struct registry_key
*parent
,
1086 TALLOC_CTX
*mem_ctx
= NULL
;
1087 WERROR werr
= WERR_OK
;
1088 struct registry_key
*key
;
1089 char *subkey_name
= NULL
;
1091 mem_ctx
= talloc_new(ctx
);
1092 if (mem_ctx
== NULL
) {
1097 /* recurse through subkeys first */
1098 werr
= reg_openkey(mem_ctx
, parent
, path
, REG_KEY_ALL
, &key
);
1099 if (!W_ERROR_IS_OK(werr
)) {
1103 while (W_ERROR_IS_OK(werr
= reg_enumkey(mem_ctx
, key
, 0,
1104 &subkey_name
, NULL
)))
1106 werr
= reg_deletekey_recursive_internal(mem_ctx
, key
,
1109 if (!W_ERROR_IS_OK(werr
)) {
1113 if (!W_ERROR_EQUAL(WERR_NO_MORE_ITEMS
, werr
)) {
1114 DEBUG(1, ("reg_deletekey_recursive_internal: "
1115 "Error enumerating subkeys: %s\n",
1123 /* now delete the actual key */
1124 werr
= reg_deletekey(parent
, path
);
1128 TALLOC_FREE(mem_ctx
);
1132 WERROR
reg_deletekey_recursive(TALLOC_CTX
*ctx
,
1133 struct registry_key
*parent
,
1136 return reg_deletekey_recursive_internal(ctx
, parent
, path
, true);
1139 WERROR
reg_deletesubkeys_recursive(TALLOC_CTX
*ctx
,
1140 struct registry_key
*parent
,
1143 return reg_deletekey_recursive_internal(ctx
, parent
, path
, false);
1147 /* these two functions are unused. */
1150 * Utility function to create a registry key without opening the hive
1151 * before. Assumes the hive already exists.
1154 WERROR
reg_create_path(TALLOC_CTX
*mem_ctx
, const char *orig_path
,
1155 uint32 desired_access
,
1156 const struct nt_user_token
*token
,
1157 enum winreg_CreateAction
*paction
,
1158 struct registry_key
**pkey
)
1160 struct registry_key
*hive
;
1164 if (!(path
= SMB_STRDUP(orig_path
))) {
1168 p
= strchr(path
, '\\');
1170 if ((p
== NULL
) || (p
[1] == '\0')) {
1172 * No key behind the hive, just return the hive
1175 err
= reg_openhive(mem_ctx
, path
, desired_access
, token
,
1177 if (!W_ERROR_IS_OK(err
)) {
1183 *paction
= REG_OPENED_EXISTING_KEY
;
1189 err
= reg_openhive(mem_ctx
, path
,
1190 (strchr(p
+1, '\\') != NULL
) ?
1191 SEC_RIGHTS_ENUM_SUBKEYS
: SEC_RIGHTS_CREATE_SUBKEY
,
1193 if (!W_ERROR_IS_OK(err
)) {
1198 err
= reg_createkey(mem_ctx
, hive
, p
+1, desired_access
, pkey
, paction
);
1205 * Utility function to create a registry key without opening the hive
1206 * before. Will not delete a hive.
1209 WERROR
reg_delete_path(const struct nt_user_token
*token
,
1210 const char *orig_path
)
1212 struct registry_key
*hive
;
1216 if (!(path
= SMB_STRDUP(orig_path
))) {
1220 p
= strchr(path
, '\\');
1222 if ((p
== NULL
) || (p
[1] == '\0')) {
1224 return WERR_INVALID_PARAM
;
1229 err
= reg_openhive(NULL
, path
,
1230 (strchr(p
+1, '\\') != NULL
) ?
1231 SEC_RIGHTS_ENUM_SUBKEYS
: SEC_RIGHTS_CREATE_SUBKEY
,
1233 if (!W_ERROR_IS_OK(err
)) {
1238 err
= reg_deletekey(hive
, p
+1);