source: vendor/current/source4/lib/registry/util.c

Last change on this file was 988, checked in by Silvan Scherrer, 9 years ago

Samba Server: update vendor to version 4.4.3
File size: 7.6 KB
Line 
1/*
2 Unix SMB/CIFS implementation.
3 Transparent registry backend handling
4 Copyright (C) Jelmer Vernooij 2003-2007.
5 Copyright (C) Wilco Baan Hofman 2010.
6
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.
11
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.
16
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/>.
19*/
20
21#include "includes.h"
22#include "lib/registry/registry.h"
23#include "librpc/gen_ndr/winreg.h"
24#include "lib/util/data_blob.h"
25
26_PUBLIC_ char *reg_val_data_string(TALLOC_CTX *mem_ctx, uint32_t type,
27 const DATA_BLOB data)
28{
29 size_t converted_size = 0;
30 char *ret = NULL;
31
32 if (data.length == 0)
33 return talloc_strdup(mem_ctx, "");
34
35 switch (type) {
36 case REG_EXPAND_SZ:
37 case REG_SZ:
38 convert_string_talloc(mem_ctx,
39 CH_UTF16, CH_UNIX,
40 data.data, data.length,
41 (void **)&ret, &converted_size);
42 break;
43 case REG_DWORD:
44 case REG_DWORD_BIG_ENDIAN:
45 SMB_ASSERT(data.length == sizeof(uint32_t));
46 ret = talloc_asprintf(mem_ctx, "0x%8.8x",
47 IVAL(data.data, 0));
48 break;
49 case REG_QWORD:
50 SMB_ASSERT(data.length == sizeof(uint64_t));
51 ret = talloc_asprintf(mem_ctx, "0x%16.16llx",
52 (long long)BVAL(data.data, 0));
53 break;
54 case REG_BINARY:
55 ret = data_blob_hex_string_upper(mem_ctx, &data);
56 break;
57 case REG_NONE:
58 /* "NULL" is the right return value */
59 break;
60 case REG_MULTI_SZ:
61 /* FIXME: We don't support this yet */
62 break;
63 default:
64 /* FIXME */
65 /* Other datatypes aren't supported -> return "NULL" */
66 break;
67 }
68
69 return ret;
70}
71
72/** Generate a string that describes a registry value */
73_PUBLIC_ char *reg_val_description(TALLOC_CTX *mem_ctx,
74 const char *name,
75 uint32_t data_type,
76 const DATA_BLOB data)
77{
78 return talloc_asprintf(mem_ctx, "%s = %s : %s", name?name:"<No Name>",
79 str_regtype(data_type),
80 reg_val_data_string(mem_ctx, data_type, data));
81}
82
83/*
84 * This implements reading hex bytes that include comma's.
85 * It was previously handled by strhex_to_data_blob, but that did not cover
86 * the format used by windows.
87 */
88static DATA_BLOB reg_strhex_to_data_blob(TALLOC_CTX *mem_ctx, const char *str)
89{
90 DATA_BLOB ret;
91 const char *HEXCHARS = "0123456789ABCDEF";
92 size_t i, j;
93 char *hi, *lo;
94
95 ret = data_blob_talloc_zero(mem_ctx, (strlen(str)+(strlen(str) % 3))/3);
96 j = 0;
97 for (i = 0; i < strlen(str); i++) {
98 hi = strchr(HEXCHARS, toupper(str[i]));
99 if (hi == NULL)
100 continue;
101
102 i++;
103 lo = strchr(HEXCHARS, toupper(str[i]));
104 if (lo == NULL)
105 break;
106
107 ret.data[j] = PTR_DIFF(hi, HEXCHARS) << 4;
108 ret.data[j] += PTR_DIFF(lo, HEXCHARS);
109 j++;
110
111 if (j > ret.length) {
112 DEBUG(0, ("Trouble converting hex string to bin\n"));
113 break;
114 }
115 }
116 return ret;
117}
118
119
120_PUBLIC_ bool reg_string_to_val(TALLOC_CTX *mem_ctx, const char *type_str,
121 const char *data_str, uint32_t *type, DATA_BLOB *data)
122{
123 char *tmp_type_str, *p, *q;
124 int result;
125
126 *type = regtype_by_string(type_str);
127
128 if (*type == -1) {
129 /* Normal windows format is hex, hex(type int as string),
130 dword or just a string. */
131 if (strncmp(type_str, "hex(", 4) == 0) {
132 /* there is a hex string with the value type between
133 the braces */
134 tmp_type_str = talloc_strdup(mem_ctx, type_str);
135 q = p = tmp_type_str + strlen("hex(");
136
137 /* Go to the closing brace or end of the string */
138 while (*q != ')' && *q != '\0') q++;
139 *q = '\0';
140
141 /* Convert hex string to int, store it in type */
142 result = sscanf(p, "%x", type);
143 if (!result) {
144 DEBUG(0, ("Could not convert hex to int\n"));
145 return false;
146 }
147 talloc_free(tmp_type_str);
148 } else if (strcmp(type_str, "hex") == 0) {
149 *type = REG_BINARY;
150 } else if (strcmp(type_str, "dword") == 0) {
151 *type = REG_DWORD;
152 }
153 }
154
155 if (*type == -1)
156 return false;
157
158 /* Convert data appropriately */
159
160 switch (*type) {
161 case REG_SZ:
162 return convert_string_talloc(mem_ctx,
163 CH_UNIX, CH_UTF16,
164 data_str, strlen(data_str)+1,
165 (void **)&data->data,
166 &data->length);
167 break;
168 case REG_MULTI_SZ:
169 case REG_EXPAND_SZ:
170 case REG_BINARY:
171 *data = reg_strhex_to_data_blob(mem_ctx, data_str);
172 break;
173 case REG_DWORD:
174 case REG_DWORD_BIG_ENDIAN: {
175 uint32_t tmp = strtol(data_str, NULL, 16);
176 *data = data_blob_talloc(mem_ctx, NULL, sizeof(uint32_t));
177 if (data->data == NULL) return false;
178 SIVAL(data->data, 0, tmp);
179 }
180 break;
181 case REG_QWORD: {
182 uint64_t tmp = strtoll(data_str, NULL, 16);
183 *data = data_blob_talloc(mem_ctx, NULL, sizeof(uint64_t));
184 if (data->data == NULL) return false;
185 SBVAL(data->data, 0, tmp);
186 }
187 break;
188 case REG_NONE:
189 ZERO_STRUCTP(data);
190 break;
191 default:
192 /* FIXME */
193 /* Other datatypes aren't supported -> return no success */
194 return false;
195 }
196 return true;
197}
198
199/** Open a key by name (including the predefined key name!) */
200WERROR reg_open_key_abs(TALLOC_CTX *mem_ctx, struct registry_context *handle,
201 const char *name, struct registry_key **result)
202{
203 struct registry_key *predef;
204 WERROR error;
205 size_t predeflength;
206 char *predefname;
207
208 if (strchr(name, '\\') != NULL)
209 predeflength = strchr(name, '\\')-name;
210 else
211 predeflength = strlen(name);
212
213 predefname = talloc_strndup(mem_ctx, name, predeflength);
214 W_ERROR_HAVE_NO_MEMORY(predefname);
215 error = reg_get_predefined_key_by_name(handle, predefname, &predef);
216 talloc_free(predefname);
217
218 if (!W_ERROR_IS_OK(error)) {
219 return error;
220 }
221
222 if (strchr(name, '\\')) {
223 return reg_open_key(mem_ctx, predef, strchr(name, '\\')+1,
224 result);
225 } else {
226 *result = predef;
227 return WERR_OK;
228 }
229}
230
231static WERROR get_abs_parent(TALLOC_CTX *mem_ctx, struct registry_context *ctx,
232 const char *path, struct registry_key **parent,
233 const char **name)
234{
235 char *parent_name;
236 WERROR error;
237
238 if (strchr(path, '\\') == NULL) {
239 return WERR_FOOBAR;
240 }
241
242 parent_name = talloc_strndup(mem_ctx, path, strrchr(path, '\\')-path);
243 W_ERROR_HAVE_NO_MEMORY(parent_name);
244 error = reg_open_key_abs(mem_ctx, ctx, parent_name, parent);
245 talloc_free(parent_name);
246 if (!W_ERROR_IS_OK(error)) {
247 return error;
248 }
249
250 *name = talloc_strdup(mem_ctx, strrchr(path, '\\')+1);
251 W_ERROR_HAVE_NO_MEMORY(*name);
252
253 return WERR_OK;
254}
255
256WERROR reg_key_del_abs(struct registry_context *ctx, const char *path)
257{
258 struct registry_key *parent;
259 const char *n;
260 TALLOC_CTX *mem_ctx = talloc_init("reg_key_del_abs");
261 WERROR error;
262
263 if (!strchr(path, '\\')) {
264 return WERR_FOOBAR;
265 }
266
267 error = get_abs_parent(mem_ctx, ctx, path, &parent, &n);
268 if (W_ERROR_IS_OK(error)) {
269 error = reg_key_del(mem_ctx, parent, n);
270 }
271
272 talloc_free(mem_ctx);
273
274 return error;
275}
276
277WERROR reg_key_add_abs(TALLOC_CTX *mem_ctx, struct registry_context *ctx,
278 const char *path, uint32_t access_mask,
279 struct security_descriptor *sec_desc,
280 struct registry_key **result)
281{
282 struct registry_key *parent;
283 const char *n;
284 WERROR error;
285
286 *result = NULL;
287
288 if (!strchr(path, '\\')) {
289 return WERR_ALREADY_EXISTS;
290 }
291
292 error = get_abs_parent(mem_ctx, ctx, path, &parent, &n);
293 if (!W_ERROR_IS_OK(error)) {
294 DEBUG(2, ("Opening parent of %s failed with %s\n", path,
295 win_errstr(error)));
296 return error;
297 }
298
299 error = reg_key_add_name(mem_ctx, parent, n, NULL, sec_desc, result);
300
301 return error;
302}
Note: See TracBrowser for help on using the repository browser.