r9388: we should fault bad handles given to winreg_GetVersion()
[samba.git] / source / rpc_server / winreg / rpc_winreg.c
1 /* 
2    Unix SMB/CIFS implementation.
3
4    endpoint server for the winreg pipe
5
6    Copyright (C) Jelmer Vernooij 2004
7    
8    This program is free software; you can redistribute it and/or modify
9    it under the terms of the GNU General Public License as published by
10    the Free Software Foundation; either version 2 of the License, or
11    (at your option) any later version.
12    
13    This program is distributed in the hope that it will be useful,
14    but WITHOUT ANY WARRANTY; without even the implied warranty of
15    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
16    GNU General Public License for more details.
17    
18    You should have received a copy of the GNU General Public License
19    along with this program; if not, write to the Free Software
20    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
21 */
22
23 #include "includes.h"
24 #include "rpc_server/dcerpc_server.h"
25 #include "registry.h"
26 #include "librpc/gen_ndr/ndr_winreg.h"
27 #include "rpc_server/common/common.h"
28
29 enum handle_types { HTYPE_REGVAL, HTYPE_REGKEY };
30
31 static NTSTATUS dcerpc_winreg_bind(struct dcesrv_call_state *dce_call, const struct dcesrv_interface *iface)
32 {
33         struct registry_context *ctx;
34         reg_open_local(&ctx);
35
36         dce_call->context->private = ctx;
37
38         return NT_STATUS_OK;
39 }
40
41 #define DCESRV_INTERFACE_WINREG_BIND dcerpc_winreg_bind
42
43 static WERROR winreg_openhive (struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx, uint32_t hkey, struct policy_handle **outh)
44 {
45         struct registry_context *ctx = dce_call->context->private;
46         struct dcesrv_handle *h; 
47         WERROR error;
48
49         h = dcesrv_handle_new(dce_call->context, HTYPE_REGKEY); 
50
51         error = reg_get_predefined_key(ctx, hkey, (struct registry_key **)&h->data);
52         if (!W_ERROR_IS_OK(error)) {
53                 return error;
54         }
55         
56         *outh = &h->wire_handle; 
57
58         return error; 
59 }
60
61 #define func_winreg_OpenHive(k,n) static WERROR winreg_Open ## k (struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx, struct winreg_Open ## k *r) \
62 { \
63         return winreg_openhive (dce_call, mem_ctx, n, &r->out.handle);\
64 }
65
66 func_winreg_OpenHive(HKCR,HKEY_CLASSES_ROOT)
67 func_winreg_OpenHive(HKCU,HKEY_CURRENT_USER)
68 func_winreg_OpenHive(HKLM,HKEY_LOCAL_MACHINE)
69 func_winreg_OpenHive(HKPD,HKEY_PERFORMANCE_DATA)
70 func_winreg_OpenHive(HKU,HKEY_USERS)
71 func_winreg_OpenHive(HKCC,HKEY_CURRENT_CONFIG)
72 func_winreg_OpenHive(HKDD,HKEY_DYN_DATA)
73 func_winreg_OpenHive(HKPT,HKEY_PERFORMANCE_TEXT)
74 func_winreg_OpenHive(HKPN,HKEY_PERFORMANCE_NLSTEXT)
75
76 /* 
77   winreg_CloseKey 
78 */
79 static WERROR winreg_CloseKey(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
80                               struct winreg_CloseKey *r)
81 {
82         struct dcesrv_handle *h; 
83
84         DCESRV_PULL_HANDLE_FAULT(h, r->in.handle, HTYPE_REGKEY);
85
86         talloc_free(h);
87
88         return WERR_OK;
89 }
90
91
92 /* 
93   winreg_CreateKey 
94 */
95 static WERROR winreg_CreateKey(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
96                                struct winreg_CreateKey *r)
97 {
98         struct dcesrv_handle *h, *newh;
99         WERROR error;
100
101         DCESRV_PULL_HANDLE_FAULT(h, r->in.handle, HTYPE_REGKEY);
102         
103         newh = dcesrv_handle_new(dce_call->context, HTYPE_REGKEY);
104
105         error = reg_key_add_name(newh, (struct registry_key *)h->data, r->in.key.name, 
106                                  r->in.access_mask, 
107                                  r->in.sec_desc?r->in.sec_desc->sd:NULL, 
108                                  (struct registry_key **)&newh->data);
109         if (W_ERROR_IS_OK(error)) {
110                 r->out.handle = &newh->wire_handle;
111         } else {
112                 talloc_free(newh);
113         }
114
115         return error;
116 }
117
118
119 /* 
120   winreg_DeleteKey 
121 */
122 static WERROR winreg_DeleteKey(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
123                        struct winreg_DeleteKey *r)
124 {
125         struct dcesrv_handle *h;
126         WERROR result;
127
128         DCESRV_PULL_HANDLE_FAULT(h, r->in.handle, HTYPE_REGKEY);
129
130         if (W_ERROR_IS_OK(result)) {
131                 return reg_key_del((struct registry_key *)h->data, r->in.key.name);
132         }
133
134         return result;
135 }
136
137
138 /* 
139   winreg_DeleteValue 
140 */
141 static WERROR winreg_DeleteValue(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
142                        struct winreg_DeleteValue *r)
143 {
144         struct dcesrv_handle *h;
145         struct registry_key *key;
146
147         DCESRV_PULL_HANDLE_FAULT(h, r->in.handle, HTYPE_REGKEY);
148
149         key = h->data;
150         
151         return reg_del_value(key, r->in.value.name);
152 }
153
154
155 /* 
156   winreg_EnumKey 
157 */
158 static WERROR winreg_EnumKey(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
159                        struct winreg_EnumKey *r)
160 {
161         struct dcesrv_handle *h;
162         struct registry_key *key;
163
164         DCESRV_PULL_HANDLE_FAULT(h, r->in.handle, HTYPE_REGKEY);
165
166         r->out.result = reg_key_get_subkey_by_index(mem_ctx, (struct registry_key *)h->data, r->in.enum_index, &key);
167
168         if (W_ERROR_IS_OK(r->out.result)) {
169                 if (2*strlen_m_term(key->name) > r->in.name->size) {
170                         return WERR_MORE_DATA;
171                 }
172                 r->out.name->length = 2*strlen_m_term(key->name);
173                 r->out.name->name = key->name;
174                 r->out.class = talloc_zero(mem_ctx, struct winreg_StringBuf);
175                 if (r->in.last_changed_time) {
176                         r->out.last_changed_time = &key->last_mod;
177                 }
178         }
179         
180         return r->out.result;
181 }
182
183
184 /* 
185   winreg_EnumValue 
186 */
187 static WERROR winreg_EnumValue(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
188                                struct winreg_EnumValue *r)
189 {
190         struct dcesrv_handle *h;
191         struct registry_key *key;
192         struct registry_value *value;
193         WERROR result;
194
195         DCESRV_PULL_HANDLE_FAULT(h, r->in.handle, HTYPE_REGKEY);
196
197         key = h->data;
198
199         result = reg_key_get_value_by_index(mem_ctx, key, r->in.enum_index, &value);
200         if (!W_ERROR_IS_OK(result)) {
201                 return result;
202         }
203
204         /* the client can optionally pass a NULL for type, meaning they don't
205            want that back */
206         if (r->in.type != NULL) {
207                 r->out.type = talloc(mem_ctx, uint32_t);
208                 *r->out.type = value->data_type;
209         }
210
211         /* check the client has enough room for the value */
212         if (r->in.value != NULL &&
213             r->in.size != NULL && 
214             value->data_len > *r->in.size) {
215                 return WERR_MORE_DATA;
216         }
217         
218         /* and enough room for the name */
219         if (r->in.name->size < 2*strlen_m_term(value->name)) {
220                 return WERR_MORE_DATA;          
221         }
222
223         r->out.name->name = value->name;
224         r->out.name->length = 2*strlen_m_term(value->name);
225         r->out.name->size = 2*strlen_m_term(value->name);
226
227         if (r->in.value) {
228                 r->out.value = value->data_blk;
229         }
230
231         if (r->in.size) {
232                 r->out.size = talloc(mem_ctx, uint32_t);
233                 *r->out.size = value->data_len;
234                 r->out.length = r->out.size;
235         }
236         
237         return WERR_OK;
238 }
239
240
241 /* 
242   winreg_FlushKey 
243 */
244 static WERROR winreg_FlushKey(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
245                        struct winreg_FlushKey *r)
246 {
247         struct dcesrv_handle *h;
248
249         DCESRV_PULL_HANDLE_FAULT(h, r->in.handle, HTYPE_REGKEY);
250
251         return reg_key_flush(h->data);
252 }
253
254
255 /* 
256   winreg_GetKeySecurity 
257 */
258 static WERROR winreg_GetKeySecurity(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
259                        struct winreg_GetKeySecurity *r)
260 {
261         struct dcesrv_handle *h;
262
263         DCESRV_PULL_HANDLE_FAULT(h, r->in.handle, HTYPE_REGKEY);
264
265         return WERR_NOT_SUPPORTED;
266 }
267
268
269 /* 
270   winreg_LoadKey 
271 */
272 static WERROR winreg_LoadKey(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
273                        struct winreg_LoadKey *r)
274 {
275         return WERR_NOT_SUPPORTED;
276 }
277
278
279 /* 
280   winreg_NotifyChangeKeyValue 
281 */
282 static WERROR winreg_NotifyChangeKeyValue(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
283                        struct winreg_NotifyChangeKeyValue *r)
284 {
285         return WERR_NOT_SUPPORTED;
286 }
287
288
289 /* 
290   winreg_OpenKey 
291 */
292 static WERROR winreg_OpenKey(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
293                        struct winreg_OpenKey *r)
294 {
295         struct dcesrv_handle *h, *newh;
296         WERROR result;
297
298         DCESRV_PULL_HANDLE_FAULT(h, r->in.handle, HTYPE_REGKEY);
299
300         if (r->in.keyname.name && strcmp(r->in.keyname.name, "") == 0) {
301                 newh = talloc_reference(dce_call->context, h);
302                 result = WERR_OK;
303         } else {
304                 newh = dcesrv_handle_new(dce_call->context, HTYPE_REGKEY);
305                 result = reg_open_key(newh, (struct registry_key *)h->data, 
306                                       r->in.keyname.name, (struct registry_key **)&newh->data);
307         }
308         
309         if (W_ERROR_IS_OK(result)) {
310                 r->out.handle = &newh->wire_handle; 
311         } else {
312                 talloc_free(newh);
313         }
314         
315         return result;
316 }
317
318
319 /* 
320   winreg_QueryInfoKey 
321 */
322 static WERROR winreg_QueryInfoKey(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
323                        struct winreg_QueryInfoKey *r)
324 {
325         struct dcesrv_handle *h;
326         struct registry_key *k;
327         WERROR ret;
328
329         DCESRV_PULL_HANDLE_FAULT(h, r->in.handle, HTYPE_REGKEY);
330
331         k = h->data;
332
333         ret = reg_key_num_subkeys(k, &r->out.num_subkeys);
334         if (!W_ERROR_IS_OK(ret)) { 
335                 return ret;
336         }
337
338         ret = reg_key_num_values(k, &r->out.num_values);
339         if (!W_ERROR_IS_OK(ret)) { 
340                 return ret;
341         }
342
343         ret = reg_key_subkeysizes(k, &r->out.max_subkeysize, &r->out.max_subkeylen);
344         if (!W_ERROR_IS_OK(ret)) { 
345                 return ret;
346         }
347
348         ret = reg_key_valuesizes(k, &r->out.max_valnamelen, &r->out.max_valbufsize);
349         if (!W_ERROR_IS_OK(ret)) { 
350                 return ret;
351         }
352
353         r->out.secdescsize = 0; /* FIXME */
354         ZERO_STRUCT(r->out.last_changed_time); /* FIXME */
355         if (!W_ERROR_IS_OK(ret)) { 
356                 return ret;
357         }
358
359
360         return WERR_OK;
361 }
362
363
364 /* 
365   winreg_QueryValue 
366 */
367 static WERROR winreg_QueryValue(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
368                        struct winreg_QueryValue *r)
369 {
370         struct dcesrv_handle *h;
371         struct registry_key *key;
372         struct registry_value *val;
373         WERROR result;
374
375         DCESRV_PULL_HANDLE_FAULT(h, r->in.handle, HTYPE_REGKEY);
376
377         key = h->data;
378         
379         result = reg_key_get_value_by_name(mem_ctx, key, r->in.value_name.name, &val);
380
381         if (!W_ERROR_IS_OK(result)) { 
382                 return result;
383         }
384
385         /* Just asking for the size of the buffer */
386         r->out.type = &val->data_type;
387         r->out.length = &val->data_len;
388         if (!r->in.data) {
389                 r->out.size = talloc(mem_ctx, uint32_t);
390                 *r->out.size = val->data_len;
391         } else {
392                 r->out.size = r->in.size;
393                 r->out.data = val->data_blk;
394         }
395
396         return WERR_OK;
397 }
398
399
400 /* 
401   winreg_ReplaceKey 
402 */
403 static WERROR winreg_ReplaceKey(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
404                        struct winreg_ReplaceKey *r)
405 {
406         return WERR_NOT_SUPPORTED;
407 }
408
409
410 /* 
411   winreg_RestoreKey 
412 */
413 static WERROR winreg_RestoreKey(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
414                        struct winreg_RestoreKey *r)
415 {
416         return WERR_NOT_SUPPORTED;
417 }
418
419
420 /* 
421   winreg_SaveKey 
422 */
423 static WERROR winreg_SaveKey(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
424                        struct winreg_SaveKey *r)
425 {
426         return WERR_NOT_SUPPORTED;
427 }
428
429
430 /* 
431   winreg_SetKeySecurity 
432 */
433 static WERROR winreg_SetKeySecurity(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
434                        struct winreg_SetKeySecurity *r)
435 {
436         return WERR_NOT_SUPPORTED;
437 }
438
439
440 /* 
441   winreg_SetValue 
442 */
443 static WERROR winreg_SetValue(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
444                        struct winreg_SetValue *r)
445 {
446         struct dcesrv_handle *h;
447         struct registry_key *key;
448         WERROR result;
449
450         DCESRV_PULL_HANDLE_FAULT(h, r->in.handle, HTYPE_REGKEY);
451
452         key = h->data;
453         
454         result = reg_val_set(key, r->in.name.name, r->in.type, r->in.data, r->in.size);
455
456         if (!W_ERROR_IS_OK(result)) { 
457                 return result;
458         }
459
460         return WERR_OK;
461 }
462
463
464 /* 
465   winreg_UnLoadKey 
466 */
467 static WERROR winreg_UnLoadKey(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
468                        struct winreg_UnLoadKey *r)
469 {
470         return WERR_NOT_SUPPORTED;
471 }
472
473
474 /* 
475   winreg_InitiateSystemShutdown 
476 */
477 static WERROR winreg_InitiateSystemShutdown(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
478                        struct winreg_InitiateSystemShutdown *r)
479 {
480         return WERR_NOT_SUPPORTED;
481 }
482
483
484 /* 
485   winreg_AbortSystemShutdown 
486 */
487 static WERROR winreg_AbortSystemShutdown(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
488                        struct winreg_AbortSystemShutdown *r)
489 {
490         return WERR_NOT_SUPPORTED;
491 }
492
493
494 /* 
495   winreg_GetVersion 
496 */
497 static WERROR winreg_GetVersion(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
498                                 struct winreg_GetVersion *r)
499 {
500         struct dcesrv_handle *h;
501
502         DCESRV_PULL_HANDLE_FAULT(h, r->in.handle, HTYPE_REGKEY);
503
504         r->out.version = 5;
505         return WERR_OK;
506 }
507
508
509 /* 
510   winreg_QueryMultipleValues 
511 */
512 static WERROR winreg_QueryMultipleValues(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
513                        struct winreg_QueryMultipleValues *r)
514 {
515         return WERR_NOT_SUPPORTED;
516 }
517
518
519 /* 
520   winreg_InitiateSystemShutdownEx 
521 */
522 static WERROR winreg_InitiateSystemShutdownEx(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
523                        struct winreg_InitiateSystemShutdownEx *r)
524 {
525         return WERR_NOT_SUPPORTED;
526 }
527
528
529 /* 
530   winreg_SaveKeyEx 
531 */
532 static WERROR winreg_SaveKeyEx(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
533                        struct winreg_SaveKeyEx *r)
534 {
535         return WERR_NOT_SUPPORTED;
536 }
537
538
539 /* 
540   winreg_QueryMultipleValues2 
541 */
542 static WERROR winreg_QueryMultipleValues2(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
543                        struct winreg_QueryMultipleValues2 *r)
544 {
545         return WERR_NOT_SUPPORTED;
546 }
547
548
549 /* include the generated boilerplate */
550 #include "librpc/gen_ndr/ndr_winreg_s.c"