s4-registry: fixed byte order assumptions
[ira/wip.git] / source4 / lib / registry / patchfile_preg.c
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
27 struct preg_data {
28         int fd;
29         TALLOC_CTX *ctx;
30 };
31
32 static WERROR preg_read_utf16(int fd, char *c)
33 {
34         uint16_t v;
35
36         if (read(fd, &v, 2) < 2) {
37                 return WERR_GENERAL_FAILURE;
38         }
39         push_codepoint(c, v);
40         return WERR_OK;
41 }
42 static WERROR preg_write_utf16(int fd, const char *string)
43 {
44         codepoint_t v;
45         uint16_t i;
46         size_t size;
47
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;
52                 }
53         }
54         return WERR_OK;
55 }
56 /* PReg does not support adding keys. */
57 static WERROR reg_preg_diff_add_key(void *_data, const char *key_name)
58 {
59         return WERR_OK;
60 }
61
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)
65 {
66         struct preg_data *data = (struct preg_data *)_data;
67         uint32_t buf;
68         
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, "]");
82         
83         return WERR_OK;
84 }
85
86 static WERROR reg_preg_diff_del_key(void *_data, const char *key_name)
87 {
88         struct preg_data *data = (struct preg_data *)_data;
89         char *parent_name;
90         DATA_BLOB blob;
91
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;
96         
97
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);
100 }
101
102 static WERROR reg_preg_diff_del_value(void *_data, const char *key_name,
103                                       const char *value_name)
104 {
105         struct preg_data *data = (struct preg_data *)_data;
106         char *val;
107         DATA_BLOB blob;
108
109         val = talloc_asprintf(data->ctx, "**Del.%s", value_name);
110
111         blob.data = (uint8_t *)talloc(data->ctx, uint32_t);
112         SIVAL(blob.data, 0, 0);
113         blob.length = 4;
114         return reg_preg_diff_set_value(data, key_name, val, REG_DWORD, blob);
115 }
116
117 static WERROR reg_preg_diff_del_all_values(void *_data, const char *key_name)
118 {
119         struct preg_data *data = (struct preg_data *)_data;
120         DATA_BLOB blob;
121
122         blob.data = (uint8_t *)talloc(data->ctx, uint32_t);
123         SIVAL(blob.data, 0, 0);
124         blob.length = 4;
125
126         return reg_preg_diff_set_value(data, key_name, "**DelVals.", REG_DWORD, blob);
127 }
128
129 static WERROR reg_preg_diff_done(void *_data)
130 {
131         struct preg_data *data = (struct preg_data *)_data;
132         
133         close(data->fd);
134         talloc_free(data);
135         return WERR_OK;
136 }
137
138 /**
139  * Save registry diff
140  */
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)
145 {
146         struct preg_data *data;
147         struct {
148                 char hdr[4];
149                 uint32_t version;
150         } preg_header;
151
152
153         data = talloc_zero(ctx, struct preg_data);
154         *callback_data = data;
155
156         if (filename) {
157                 data->fd = open(filename, O_CREAT|O_WRONLY, 0755);
158                 if (data->fd < 0) {
159                         DEBUG(0, ("Unable to open %s\n", filename));
160                         return WERR_BADFILE;
161                 }
162         } else {
163                 data->fd = STDOUT_FILENO;
164         }
165
166         strncpy(preg_header.hdr, "PReg", 4);
167         SIVAL(&preg_header, 4, 1);
168         write(data->fd, (uint8_t *)&preg_header,8);
169
170         data->ctx = ctx;
171
172         *callbacks = talloc(ctx, struct reg_diff_callbacks);
173
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;
180
181         return WERR_OK;
182 }
183 /**
184  * Load diff file
185  */
186 _PUBLIC_ WERROR reg_preg_diff_load(int fd,
187                                    struct smb_iconv_convenience *iconv_convenience, 
188                                    const struct reg_diff_callbacks *callbacks,
189                                    void *callback_data)
190 {
191         struct {
192                 char hdr[4];
193                 uint32_t version;
194         } preg_header;
195         char *buf;
196         size_t buf_size = 1024;
197         char *buf_ptr;
198         TALLOC_CTX *mem_ctx = talloc_init("reg_preg_diff_load");
199         WERROR ret = WERR_OK;
200         DATA_BLOB data = {NULL, 0};
201         char *key = NULL;
202         char *value_name = NULL;
203
204         buf = talloc_array(mem_ctx, char, buf_size);
205         buf_ptr = buf;
206
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",
210                                 strerror(errno)));
211                 ret = WERR_GENERAL_FAILURE;
212                 goto cleanup;
213         }
214         preg_header.version = IVAL(&preg_header.version, 0);
215
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;
219                 goto cleanup;
220         }
221         if (preg_header.version > 1) {
222                 DEBUG(0, ("Warning: file format version is higher than expected.\n"));
223         }
224
225         /* Read the entries */
226         while(1) {
227                 uint32_t value_type, length;
228
229                 if (!W_ERROR_IS_OK(preg_read_utf16(fd, buf_ptr))) {
230                         break;
231                 }
232                 if (*buf_ptr != '[') {
233                         DEBUG(0, ("Error in PReg file.\n"));
234                         ret = WERR_GENERAL_FAILURE;
235                         goto cleanup;
236                 }
237
238                 /* Get the path */
239                 buf_ptr = buf;
240                 while (W_ERROR_IS_OK(preg_read_utf16(fd, buf_ptr)) &&
241                        *buf_ptr != ';' && buf_ptr-buf < buf_size) {
242                         buf_ptr++;
243                 }
244                 buf[buf_ptr-buf] = '\0';
245                 key = talloc_strdup(mem_ctx, buf);
246
247                 /* Get the name */
248                 buf_ptr = buf;
249                 while (W_ERROR_IS_OK(preg_read_utf16(fd, buf_ptr)) &&
250                        *buf_ptr != ';' && buf_ptr-buf < buf_size) {
251                         buf_ptr++;
252                 }
253                 buf[buf_ptr-buf] = '\0';
254                 value_name = talloc_strdup(mem_ctx, buf);
255
256                 /* Get the type */
257                 if (read(fd, &value_type, 4) < 4) {
258                         DEBUG(0, ("Error while reading PReg\n"));
259                         ret = WERR_GENERAL_FAILURE;
260                         goto cleanup;
261                 }
262                 value_type = IVAL(&value_type, 0);
263
264                 /* Read past delimiter */
265                 buf_ptr = buf;
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;
270                         goto cleanup;
271                 }
272                 /* Get data length */
273                 if (read(fd, &length, 4) < 4) {
274                         DEBUG(0, ("Error while reading PReg\n"));
275                         ret = WERR_GENERAL_FAILURE;
276                         goto cleanup;
277                 }
278                 /* Read past delimiter */
279                 buf_ptr = buf;
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;
284                         goto cleanup;
285                 }
286                 /* Get the data */
287                 buf_ptr = buf;
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;
292                         goto cleanup;
293                 }
294                 data = data_blob_talloc(mem_ctx, buf, length);
295
296                 /* Check if delimiter is in place (whine if it isn't) */
297                 buf_ptr = buf;
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));
302                 }
303
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;
308
309                         callbacks->del_value(callback_data, key, p);
310                 } else  if (strcasecmp(value_name, "**DeleteValues") == 0) {
311                         char *p, *q;
312
313                         p = (char *) data.data;
314
315                         while ((q = strchr_m(p, ';'))) {
316                                 *q = '\0';
317                                 q++;
318
319                                 callbacks->del_value(callback_data, key, p);
320
321                                 p = q;
322                         }
323                         callbacks->del_value(callback_data, key, p);
324                 } else if (strcasecmp(value_name, "**DeleteKeys") == 0) {
325                         char *p, *q, *full_key;
326
327                         p = (char *) data.data;
328
329                         while ((q = strchr_m(p, ';'))) {
330                                 *q = '\0';
331                                 q++;
332
333                                 full_key = talloc_asprintf(mem_ctx, "%s\\%s",
334                                                            key, p);
335                                 callbacks->del_key(callback_data, full_key);
336                                 talloc_free(full_key);
337
338                                 p = q;
339                         }
340                         full_key = talloc_asprintf(mem_ctx, "%s\\%s", key, p);
341                         callbacks->del_key(callback_data, full_key);
342                         talloc_free(full_key);
343                 } else {
344                         callbacks->add_key(callback_data, key);
345                         callbacks->set_value(callback_data, key, value_name,
346                                              value_type, data);
347                 }
348         }
349 cleanup:
350         close(fd);
351         talloc_free(data.data);
352         talloc_free(key);
353         talloc_free(value_name);
354         talloc_free(buf);
355         return ret;
356 }