2 Unix SMB/CIFS implementation.
3 Reading Registry.pol PReg registry files
5 Copyright (C) Wilco Baan Hofman 2006-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, write to the Free Software
19 Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
23 #include "lib/registry/registry.h"
24 #include "system/filesys.h"
25 #include "librpc/gen_ndr/winreg.h"
32 static WERROR
preg_read_utf16(int fd
, char *c
)
36 if (read(fd
, &v
, 2) < 2) {
37 return WERR_GENERAL_FAILURE
;
42 static WERROR
preg_write_utf16(int fd
, const char *string
)
48 for (i
= 0; i
< strlen(string
); i
+=size
) {
49 v
= next_codepoint(&string
[i
], &size
);
50 if (write(fd
, &v
, 2) < 2) {
51 return WERR_GENERAL_FAILURE
;
56 /* PReg does not support adding keys. */
57 static WERROR
reg_preg_diff_add_key(void *_data
, const char *key_name
)
62 static WERROR
reg_preg_diff_set_value(void *_data
, const char *key_name
,
63 const char *value_name
,
64 uint32_t value_type
, DATA_BLOB value_data
)
66 struct preg_data
*data
= (struct preg_data
*)_data
;
69 preg_write_utf16(data
->fd
, "[");
70 preg_write_utf16(data
->fd
, key_name
);
71 preg_write_utf16(data
->fd
, ";");
72 preg_write_utf16(data
->fd
, value_name
);
73 preg_write_utf16(data
->fd
, ";");
74 SIVAL(&buf
, 0, value_type
);
75 write(data
->fd
, &buf
, sizeof(uint32_t));
76 preg_write_utf16(data
->fd
, ";");
77 SIVAL(&buf
, 0, value_data
.length
);
78 write(data
->fd
, &buf
, sizeof(uint32_t));
79 preg_write_utf16(data
->fd
, ";");
80 write(data
->fd
, value_data
.data
, value_data
.length
);
81 preg_write_utf16(data
->fd
, "]");
86 static WERROR
reg_preg_diff_del_key(void *_data
, const char *key_name
)
88 struct preg_data
*data
= (struct preg_data
*)_data
;
92 parent_name
= talloc_strndup(data
->ctx
, key_name
, strrchr(key_name
, '\\')-key_name
);
93 blob
.data
= (uint8_t *)talloc_strndup(data
->ctx
, key_name
+(strrchr(key_name
, '\\')-key_name
)+1,
94 strlen(key_name
)-(strrchr(key_name
, '\\')-key_name
));
95 blob
.length
= strlen((char *)blob
.data
)+1;
98 /* FIXME: These values should be accumulated to be written at done(). */
99 return reg_preg_diff_set_value(data
, parent_name
, "**DeleteKeys", REG_SZ
, blob
);
102 static WERROR
reg_preg_diff_del_value(void *_data
, const char *key_name
,
103 const char *value_name
)
105 struct preg_data
*data
= (struct preg_data
*)_data
;
109 val
= talloc_asprintf(data
->ctx
, "**Del.%s", value_name
);
111 blob
.data
= (uint8_t *)talloc(data
->ctx
, uint32_t);
112 SIVAL(blob
.data
, 0, 0);
114 return reg_preg_diff_set_value(data
, key_name
, val
, REG_DWORD
, blob
);
117 static WERROR
reg_preg_diff_del_all_values(void *_data
, const char *key_name
)
119 struct preg_data
*data
= (struct preg_data
*)_data
;
122 blob
.data
= (uint8_t *)talloc(data
->ctx
, uint32_t);
123 SIVAL(blob
.data
, 0, 0);
126 return reg_preg_diff_set_value(data
, key_name
, "**DelVals.", REG_DWORD
, blob
);
129 static WERROR
reg_preg_diff_done(void *_data
)
131 struct preg_data
*data
= (struct preg_data
*)_data
;
141 _PUBLIC_ WERROR
reg_preg_diff_save(TALLOC_CTX
*ctx
, const char *filename
,
142 struct smb_iconv_convenience
*ic
,
143 struct reg_diff_callbacks
**callbacks
,
144 void **callback_data
)
146 struct preg_data
*data
;
153 data
= talloc_zero(ctx
, struct preg_data
);
154 *callback_data
= data
;
157 data
->fd
= open(filename
, O_CREAT
|O_WRONLY
, 0755);
159 DEBUG(0, ("Unable to open %s\n", filename
));
163 data
->fd
= STDOUT_FILENO
;
166 strncpy(preg_header
.hdr
, "PReg", 4);
167 SIVAL(&preg_header
, 4, 1);
168 write(data
->fd
, (uint8_t *)&preg_header
,8);
172 *callbacks
= talloc(ctx
, struct reg_diff_callbacks
);
174 (*callbacks
)->add_key
= reg_preg_diff_add_key
;
175 (*callbacks
)->del_key
= reg_preg_diff_del_key
;
176 (*callbacks
)->set_value
= reg_preg_diff_set_value
;
177 (*callbacks
)->del_value
= reg_preg_diff_del_value
;
178 (*callbacks
)->del_all_values
= reg_preg_diff_del_all_values
;
179 (*callbacks
)->done
= reg_preg_diff_done
;
186 _PUBLIC_ WERROR
reg_preg_diff_load(int fd
,
187 struct smb_iconv_convenience
*iconv_convenience
,
188 const struct reg_diff_callbacks
*callbacks
,
196 size_t buf_size
= 1024;
198 TALLOC_CTX
*mem_ctx
= talloc_init("reg_preg_diff_load");
199 WERROR ret
= WERR_OK
;
200 DATA_BLOB data
= {NULL
, 0};
202 char *value_name
= NULL
;
204 buf
= talloc_array(mem_ctx
, char, buf_size
);
207 /* Read first 8 bytes (the header) */
208 if (read(fd
, &preg_header
, 8) != 8) {
209 DEBUG(0, ("Could not read PReg file: %s\n",
211 ret
= WERR_GENERAL_FAILURE
;
214 preg_header
.version
= IVAL(&preg_header
.version
, 0);
216 if (strncmp(preg_header
.hdr
, "PReg", 4) != 0) {
217 DEBUG(0, ("This file is not a valid preg registry file\n"));
218 ret
= WERR_GENERAL_FAILURE
;
221 if (preg_header
.version
> 1) {
222 DEBUG(0, ("Warning: file format version is higher than expected.\n"));
225 /* Read the entries */
227 uint32_t value_type
, length
;
229 if (!W_ERROR_IS_OK(preg_read_utf16(fd
, buf_ptr
))) {
232 if (*buf_ptr
!= '[') {
233 DEBUG(0, ("Error in PReg file.\n"));
234 ret
= WERR_GENERAL_FAILURE
;
240 while (W_ERROR_IS_OK(preg_read_utf16(fd
, buf_ptr
)) &&
241 *buf_ptr
!= ';' && buf_ptr
-buf
< buf_size
) {
244 buf
[buf_ptr
-buf
] = '\0';
245 key
= talloc_strdup(mem_ctx
, buf
);
249 while (W_ERROR_IS_OK(preg_read_utf16(fd
, buf_ptr
)) &&
250 *buf_ptr
!= ';' && buf_ptr
-buf
< buf_size
) {
253 buf
[buf_ptr
-buf
] = '\0';
254 value_name
= talloc_strdup(mem_ctx
, buf
);
257 if (read(fd
, &value_type
, 4) < 4) {
258 DEBUG(0, ("Error while reading PReg\n"));
259 ret
= WERR_GENERAL_FAILURE
;
262 value_type
= IVAL(&value_type
, 0);
264 /* Read past delimiter */
266 if (!(W_ERROR_IS_OK(preg_read_utf16(fd
, buf_ptr
)) &&
267 *buf_ptr
== ';') && buf_ptr
-buf
< buf_size
) {
268 DEBUG(0, ("Error in PReg file.\n"));
269 ret
= WERR_GENERAL_FAILURE
;
272 /* Get data length */
273 if (read(fd
, &length
, 4) < 4) {
274 DEBUG(0, ("Error while reading PReg\n"));
275 ret
= WERR_GENERAL_FAILURE
;
278 /* Read past delimiter */
280 if (!(W_ERROR_IS_OK(preg_read_utf16(fd
, buf_ptr
)) &&
281 *buf_ptr
== ';') && buf_ptr
-buf
< buf_size
) {
282 DEBUG(0, ("Error in PReg file.\n"));
283 ret
= WERR_GENERAL_FAILURE
;
288 if (length
< buf_size
&&
289 read(fd
, buf_ptr
, length
) != length
) {
290 DEBUG(0, ("Error while reading PReg\n"));
291 ret
= WERR_GENERAL_FAILURE
;
294 data
= data_blob_talloc(mem_ctx
, buf
, length
);
296 /* Check if delimiter is in place (whine if it isn't) */
298 if (!(W_ERROR_IS_OK(preg_read_utf16(fd
, buf_ptr
)) &&
299 *buf_ptr
== ']') && buf_ptr
-buf
< buf_size
) {
300 DEBUG(0, ("Warning: Missing ']' in PReg file, expected ']', got '%c' 0x%x.\n",
301 *buf_ptr
, *buf_ptr
));
304 if (strcasecmp(value_name
, "**DelVals") == 0) {
305 callbacks
->del_all_values(callback_data
, key
);
306 } else if (strncasecmp(value_name
, "**Del.",6) == 0) {
307 char *p
= value_name
+6;
309 callbacks
->del_value(callback_data
, key
, p
);
310 } else if (strcasecmp(value_name
, "**DeleteValues") == 0) {
313 p
= (char *) data
.data
;
315 while ((q
= strchr_m(p
, ';'))) {
319 callbacks
->del_value(callback_data
, key
, p
);
323 callbacks
->del_value(callback_data
, key
, p
);
324 } else if (strcasecmp(value_name
, "**DeleteKeys") == 0) {
325 char *p
, *q
, *full_key
;
327 p
= (char *) data
.data
;
329 while ((q
= strchr_m(p
, ';'))) {
333 full_key
= talloc_asprintf(mem_ctx
, "%s\\%s",
335 callbacks
->del_key(callback_data
, full_key
);
336 talloc_free(full_key
);
340 full_key
= talloc_asprintf(mem_ctx
, "%s\\%s", key
, p
);
341 callbacks
->del_key(callback_data
, full_key
);
342 talloc_free(full_key
);
344 callbacks
->add_key(callback_data
, key
);
345 callbacks
->set_value(callback_data
, key
, value_name
,
351 talloc_free(data
.data
);
353 talloc_free(value_name
);