1 /*
2 * Unix SMB/CIFS implementation.
3 * Registry helper routines
4 * Copyright (C) Volker Lendecke 2006
5 * Copyright (C) Guenther Deschner 2009
6 * Copyright (C) Jelmer Vernooij 2003-2007
7 *
8 * This program is free software; you can redistribute it and/or modify it
9 * under the terms of the GNU General Public License as published by the Free
10 * Software Foundation; either version 3 of the License, or (at your option)
11 * any later version.
12 *
13 * This program is distributed in the hope that it will be useful, but WITHOUT
14 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
15 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
16 * more details.
17 *
18 * You should have received a copy of the GNU General Public License along with
19 * this program; if not, see <http://www.gnu.org/licenses/>.
20 */
21
22 #include "includes.h"
23 #include "../librpc/gen_ndr/ndr_misc.h"
24 #include "../libcli/registry/util_reg.h"
25
26 /**
27 * @file
28 * @brief Registry utility functions
29 */
30
31 static const struct {
32 uint32_t id;
33 const char *name;
34 } reg_value_types[] = {
35 { REG_NONE, "REG_NONE" },
36 { REG_SZ, "REG_SZ" },
37 { REG_EXPAND_SZ, "REG_EXPAND_SZ" },
38 { REG_BINARY, "REG_BINARY" },
39 { REG_DWORD, "REG_DWORD" },
40 { REG_DWORD_BIG_ENDIAN, "REG_DWORD_BIG_ENDIAN" },
41 { REG_LINK, "REG_LINK" },
42 { REG_MULTI_SZ, "REG_MULTI_SZ" },
43 { REG_RESOURCE_LIST, "REG_RESOURCE_LIST" },
44 { REG_FULL_RESOURCE_DESCRIPTOR, "REG_FULL_RESOURCE_DESCRIPTOR" },
45 { REG_RESOURCE_REQUIREMENTS_LIST, "REG_RESOURCE_REQUIREMENTS_LIST" },
46 { REG_QWORD, "REG_QWORD" },
47
48 { 0, NULL }
49 };
50
51 /** Return string description of registry value type */
str_regtype(int type)52 _PUBLIC_ const char *str_regtype(int type)
53 {
54 unsigned int i;
55 for (i = 0; reg_value_types[i].name; i++) {
56 if (reg_value_types[i].id == type)
57 return reg_value_types[i].name;
58 }
59
60 return "Unknown";
61 }
62
63 /** Return registry value type for string description */
regtype_by_string(const char * str)64 _PUBLIC_ int regtype_by_string(const char *str)
65 {
66 unsigned int i;
67 for (i = 0; reg_value_types[i].name; i++) {
68 if (strequal(reg_value_types[i].name, str))
69 return reg_value_types[i].id;
70 }
71
72 return -1;
73 }
74
75 /*******************************************************************
76 push a string in unix charset into a REG_SZ UCS2 null terminated blob
77 ********************************************************************/
78
push_reg_sz(TALLOC_CTX * mem_ctx,DATA_BLOB * blob,const char * s)79 bool push_reg_sz(TALLOC_CTX *mem_ctx, DATA_BLOB *blob, const char *s)
80 {
81 union winreg_Data data;
82 enum ndr_err_code ndr_err;
83 data.string = s;
84 ndr_err = ndr_push_union_blob(blob, mem_ctx, &data, REG_SZ,
85 (ndr_push_flags_fn_t)ndr_push_winreg_Data);
86 return NDR_ERR_CODE_IS_SUCCESS(ndr_err);
87 }
88
89 /*******************************************************************
90 push a string_array in unix charset into a REG_MULTI_SZ UCS2 double-null
91 terminated blob
92 ********************************************************************/
93
push_reg_multi_sz(TALLOC_CTX * mem_ctx,DATA_BLOB * blob,const char ** a)94 bool push_reg_multi_sz(TALLOC_CTX *mem_ctx, DATA_BLOB *blob, const char **a)
95 {
96 union winreg_Data data;
97 enum ndr_err_code ndr_err;
98 data.string_array = a;
99 ndr_err = ndr_push_union_blob(blob, mem_ctx, &data, REG_MULTI_SZ,
100 (ndr_push_flags_fn_t)ndr_push_winreg_Data);
101 return NDR_ERR_CODE_IS_SUCCESS(ndr_err);
102 }
103
104 /*******************************************************************
105 pull a string in unix charset out of a REG_SZ UCS2 null terminated blob
106 ********************************************************************/
107
pull_reg_sz(TALLOC_CTX * mem_ctx,const DATA_BLOB * blob,const char ** s)108 bool pull_reg_sz(TALLOC_CTX *mem_ctx, const DATA_BLOB *blob, const char **s)
109 {
110 union winreg_Data data;
111 enum ndr_err_code ndr_err;
112 ndr_err = ndr_pull_union_blob(blob, mem_ctx, &data, REG_SZ,
113 (ndr_pull_flags_fn_t)ndr_pull_winreg_Data);
114 if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
115 return false;
116 }
117 *s = data.string;
118 return true;
119 }
120
121 /*******************************************************************
122 pull a string_array in unix charset out of a REG_MULTI_SZ UCS2 double-null
123 terminated blob
124 ********************************************************************/
125
pull_reg_multi_sz(TALLOC_CTX * mem_ctx,const DATA_BLOB * blob,const char *** a)126 bool pull_reg_multi_sz(TALLOC_CTX *mem_ctx,
127 const DATA_BLOB *blob, const char ***a)
128 {
129 union winreg_Data data;
130 enum ndr_err_code ndr_err;
131 ndr_err = ndr_pull_union_blob(blob, mem_ctx, &data, REG_MULTI_SZ,
132 (ndr_pull_flags_fn_t)ndr_pull_winreg_Data);
133 if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
134 return false;
135 }
136 *a = data.string_array;
137 return true;
138 }
139