source: vendor/current/source4/lib/registry/patchfile_preg.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: 9.7 KB
Line 
1/*
2 Unix SMB/CIFS implementation.
3 Reading Registry.pol PReg registry files
4
5 Copyright (C) Wilco Baan Hofman 2006-2008
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, write to the Free Software
19 Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
20*/
21
22#include "includes.h"
23#include "lib/registry/registry.h"
24#include "system/filesys.h"
25#include "librpc/gen_ndr/winreg.h"
26
27struct preg_data {
28 int fd;
29 TALLOC_CTX *ctx;
30};
31
32static WERROR preg_read_utf16(int fd, char *c)
33{
34 uint16_t v;
35
36 if (read(fd, &v, sizeof(uint16_t)) < sizeof(uint16_t)) {
37 return WERR_GENERAL_FAILURE;
38 }
39 push_codepoint(c, v);
40 return WERR_OK;
41}
42static WERROR preg_write_utf16(int fd, const char *string)
43{
44 uint16_t v;
45 size_t i, size;
46
47 for (i = 0; i < strlen(string); i+=size) {
48 v = next_codepoint(&string[i], &size);
49 if (write(fd, &v, sizeof(uint16_t)) < sizeof(uint16_t)) {
50 return WERR_GENERAL_FAILURE;
51 }
52 }
53 return WERR_OK;
54}
55/* PReg does not support adding keys. */
56static WERROR reg_preg_diff_add_key(void *_data, const char *key_name)
57{
58 return WERR_OK;
59}
60
61static WERROR reg_preg_diff_set_value(void *_data, const char *key_name,
62 const char *value_name,
63 uint32_t value_type, DATA_BLOB value_data)
64{
65 struct preg_data *data = (struct preg_data *)_data;
66 uint32_t buf;
67
68 preg_write_utf16(data->fd, "[");
69 preg_write_utf16(data->fd, key_name);
70 preg_write_utf16(data->fd, ";");
71 preg_write_utf16(data->fd, value_name);
72 preg_write_utf16(data->fd, ";");
73 SIVAL(&buf, 0, value_type);
74 write(data->fd, &buf, sizeof(uint32_t));
75 preg_write_utf16(data->fd, ";");
76 SIVAL(&buf, 0, value_data.length);
77 write(data->fd, &buf, sizeof(uint32_t));
78 preg_write_utf16(data->fd, ";");
79 write(data->fd, value_data.data, value_data.length);
80 preg_write_utf16(data->fd, "]");
81
82 return WERR_OK;
83}
84
85static WERROR reg_preg_diff_del_key(void *_data, const char *key_name)
86{
87 struct preg_data *data = (struct preg_data *)_data;
88 char *parent_name;
89 DATA_BLOB blob;
90 WERROR werr;
91
92 parent_name = talloc_strndup(data->ctx, key_name,
93 strrchr(key_name, '\\')-key_name);
94 W_ERROR_HAVE_NO_MEMORY(parent_name);
95 blob.data = (uint8_t*)talloc_strndup(data->ctx,
96 key_name+(strrchr(key_name, '\\')-key_name)+1,
97 strlen(key_name)-(strrchr(key_name, '\\')-key_name));
98 W_ERROR_HAVE_NO_MEMORY(blob.data);
99 blob.length = strlen((char *)blob.data)+1;
100
101
102 /* FIXME: These values should be accumulated to be written at done(). */
103 werr = reg_preg_diff_set_value(data, parent_name, "**DeleteKeys",
104 REG_SZ, blob);
105
106 talloc_free(parent_name);
107 talloc_free(blob.data);
108
109 return werr;
110}
111
112static WERROR reg_preg_diff_del_value(void *_data, const char *key_name,
113 const char *value_name)
114{
115 struct preg_data *data = (struct preg_data *)_data;
116 char *val;
117 DATA_BLOB blob;
118 WERROR werr;
119
120 val = talloc_asprintf(data->ctx, "**Del.%s", value_name);
121 W_ERROR_HAVE_NO_MEMORY(val);
122 blob.data = (uint8_t *)talloc(data->ctx, uint32_t);
123 W_ERROR_HAVE_NO_MEMORY(blob.data);
124 SIVAL(blob.data, 0, 0);
125 blob.length = sizeof(uint32_t);
126
127 werr = reg_preg_diff_set_value(data, key_name, val, REG_DWORD, blob);
128
129 talloc_free(val);
130 talloc_free(blob.data);
131
132 return werr;
133}
134
135static WERROR reg_preg_diff_del_all_values(void *_data, const char *key_name)
136{
137 struct preg_data *data = (struct preg_data *)_data;
138 DATA_BLOB blob;
139 WERROR werr;
140
141 blob.data = (uint8_t *)talloc(data->ctx, uint32_t);
142 W_ERROR_HAVE_NO_MEMORY(blob.data);
143 SIVAL(blob.data, 0, 0);
144 blob.length = sizeof(uint32_t);
145
146 werr = reg_preg_diff_set_value(data, key_name, "**DelVals.", REG_DWORD,
147 blob);
148
149 talloc_free(blob.data);
150
151 return werr;
152}
153
154static WERROR reg_preg_diff_done(void *_data)
155{
156 struct preg_data *data = (struct preg_data *)_data;
157
158 close(data->fd);
159 talloc_free(data);
160 return WERR_OK;
161}
162
163/**
164 * Save registry diff
165 */
166_PUBLIC_ WERROR reg_preg_diff_save(TALLOC_CTX *ctx, const char *filename,
167 struct reg_diff_callbacks **callbacks,
168 void **callback_data)
169{
170 struct preg_data *data;
171 struct {
172 char hdr[4];
173 uint32_t version;
174 } preg_header;
175
176
177 data = talloc_zero(ctx, struct preg_data);
178 *callback_data = data;
179
180 if (filename) {
181 data->fd = open(filename, O_CREAT|O_WRONLY, 0755);
182 if (data->fd < 0) {
183 DEBUG(0, ("Unable to open %s\n", filename));
184 return WERR_BADFILE;
185 }
186 } else {
187 data->fd = STDOUT_FILENO;
188 }
189
190 memcpy(preg_header.hdr, "PReg", sizeof(preg_header.hdr));
191 SIVAL(&preg_header.version, 0, 1);
192 write(data->fd, (uint8_t *)&preg_header, sizeof(preg_header));
193
194 data->ctx = ctx;
195
196 *callbacks = talloc(ctx, struct reg_diff_callbacks);
197
198 (*callbacks)->add_key = reg_preg_diff_add_key;
199 (*callbacks)->del_key = reg_preg_diff_del_key;
200 (*callbacks)->set_value = reg_preg_diff_set_value;
201 (*callbacks)->del_value = reg_preg_diff_del_value;
202 (*callbacks)->del_all_values = reg_preg_diff_del_all_values;
203 (*callbacks)->done = reg_preg_diff_done;
204
205 return WERR_OK;
206}
207/**
208 * Load diff file
209 */
210_PUBLIC_ WERROR reg_preg_diff_load(int fd,
211 const struct reg_diff_callbacks *callbacks,
212 void *callback_data)
213{
214 struct {
215 char hdr[4];
216 uint32_t version;
217 } preg_header;
218 char *buf;
219 size_t buf_size = 1024;
220 char *buf_ptr;
221 TALLOC_CTX *mem_ctx = talloc_init("reg_preg_diff_load");
222 WERROR ret = WERR_OK;
223 DATA_BLOB data = {NULL, 0};
224 char *key = NULL;
225 char *value_name = NULL;
226
227 buf = talloc_array(mem_ctx, char, buf_size);
228 buf_ptr = buf;
229
230 /* Read first 8 bytes (the header) */
231 if (read(fd, &preg_header, sizeof(preg_header)) != sizeof(preg_header)) {
232 DEBUG(0, ("Could not read PReg file: %s\n",
233 strerror(errno)));
234 ret = WERR_GENERAL_FAILURE;
235 goto cleanup;
236 }
237 preg_header.version = IVAL(&preg_header.version, 0);
238
239 if (strncmp(preg_header.hdr, "PReg", 4) != 0) {
240 DEBUG(0, ("This file is not a valid preg registry file\n"));
241 ret = WERR_GENERAL_FAILURE;
242 goto cleanup;
243 }
244 if (preg_header.version > 1) {
245 DEBUG(0, ("Warning: file format version is higher than expected.\n"));
246 }
247
248 /* Read the entries */
249 while(1) {
250 uint32_t value_type, length;
251
252 if (!W_ERROR_IS_OK(preg_read_utf16(fd, buf_ptr))) {
253 break;
254 }
255 if (*buf_ptr != '[') {
256 DEBUG(0, ("Error in PReg file.\n"));
257 ret = WERR_GENERAL_FAILURE;
258 goto cleanup;
259 }
260
261 /* Get the path */
262 buf_ptr = buf;
263 while (W_ERROR_IS_OK(preg_read_utf16(fd, buf_ptr)) &&
264 *buf_ptr != ';' && buf_ptr-buf < buf_size) {
265 buf_ptr++;
266 }
267 buf[buf_ptr-buf] = '\0';
268 key = talloc_strdup(mem_ctx, buf);
269
270 /* Get the name */
271 buf_ptr = buf;
272 while (W_ERROR_IS_OK(preg_read_utf16(fd, buf_ptr)) &&
273 *buf_ptr != ';' && buf_ptr-buf < buf_size) {
274 buf_ptr++;
275 }
276 buf[buf_ptr-buf] = '\0';
277 value_name = talloc_strdup(mem_ctx, buf);
278
279 /* Get the type */
280 if (read(fd, &value_type, sizeof(uint32_t)) < sizeof(uint32_t)) {
281 DEBUG(0, ("Error while reading PReg\n"));
282 ret = WERR_GENERAL_FAILURE;
283 goto cleanup;
284 }
285 value_type = IVAL(&value_type, 0);
286
287 /* Read past delimiter */
288 buf_ptr = buf;
289 if (!(W_ERROR_IS_OK(preg_read_utf16(fd, buf_ptr)) &&
290 *buf_ptr == ';') && buf_ptr-buf < buf_size) {
291 DEBUG(0, ("Error in PReg file.\n"));
292 ret = WERR_GENERAL_FAILURE;
293 goto cleanup;
294 }
295
296 /* Get data length */
297 if (read(fd, &length, sizeof(uint32_t)) < sizeof(uint32_t)) {
298 DEBUG(0, ("Error while reading PReg\n"));
299 ret = WERR_GENERAL_FAILURE;
300 goto cleanup;
301 }
302 length = IVAL(&length, 0);
303
304 /* Read past delimiter */
305 buf_ptr = buf;
306 if (!(W_ERROR_IS_OK(preg_read_utf16(fd, buf_ptr)) &&
307 *buf_ptr == ';') && buf_ptr-buf < buf_size) {
308 DEBUG(0, ("Error in PReg file.\n"));
309 ret = WERR_GENERAL_FAILURE;
310 goto cleanup;
311 }
312
313 /* Get the data */
314 buf_ptr = buf;
315 if (length < buf_size &&
316 read(fd, buf_ptr, length) != length) {
317 DEBUG(0, ("Error while reading PReg\n"));
318 ret = WERR_GENERAL_FAILURE;
319 goto cleanup;
320 }
321 data = data_blob_talloc(mem_ctx, buf, length);
322
323 /* Check if delimiter is in place (whine if it isn't) */
324 buf_ptr = buf;
325 if (!(W_ERROR_IS_OK(preg_read_utf16(fd, buf_ptr)) &&
326 *buf_ptr == ']') && buf_ptr-buf < buf_size) {
327 DEBUG(0, ("Warning: Missing ']' in PReg file, expected ']', got '%c' 0x%x.\n",
328 *buf_ptr, *buf_ptr));
329 }
330
331 if (strcasecmp(value_name, "**DelVals") == 0) {
332 callbacks->del_all_values(callback_data, key);
333 } else if (strncasecmp(value_name, "**Del.",6) == 0) {
334 char *p = value_name+6;
335
336 callbacks->del_value(callback_data, key, p);
337 } else if (strcasecmp(value_name, "**DeleteValues") == 0) {
338 char *p, *q;
339
340 p = (char *) data.data;
341
342 while ((q = strchr_m(p, ';'))) {
343 *q = '\0';
344 q++;
345
346 callbacks->del_value(callback_data, key, p);
347
348 p = q;
349 }
350 callbacks->del_value(callback_data, key, p);
351 } else if (strcasecmp(value_name, "**DeleteKeys") == 0) {
352 char *p, *q, *full_key;
353
354 p = (char *) data.data;
355
356 while ((q = strchr_m(p, ';'))) {
357 *q = '\0';
358 q++;
359
360 full_key = talloc_asprintf(mem_ctx, "%s\\%s",
361 key, p);
362 callbacks->del_key(callback_data, full_key);
363 talloc_free(full_key);
364
365 p = q;
366 }
367 full_key = talloc_asprintf(mem_ctx, "%s\\%s", key, p);
368 callbacks->del_key(callback_data, full_key);
369 talloc_free(full_key);
370 } else {
371 callbacks->add_key(callback_data, key);
372 callbacks->set_value(callback_data, key, value_name,
373 value_type, data);
374 }
375 }
376cleanup:
377 close(fd);
378 talloc_free(data.data);
379 talloc_free(key);
380 talloc_free(value_name);
381 talloc_free(buf);
382 return ret;
383}
Note: See TracBrowser for help on using the repository browser.