source4/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->conn->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->conn->private;
  46         struct dcesrv_handle *h;
  47         WERROR error;
  48
  49         h = dcesrv_handle_new(dce_call->conn, 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         h = dcesrv_handle_fetch(dce_call->conn, r->in.handle, HTYPE_REGKEY);
  85         DCESRV_CHECK_HANDLE(h);
  86
  87         dcesrv_handle_destroy(dce_call->conn, h);
  88
  89         return WERR_OK;
  90 }
  91
  92
  93 /*
  94   winreg_CreateKey
  95 */
  96 static WERROR winreg_CreateKey(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
  97                        struct winreg_CreateKey *r)
  98 {
  99         struct dcesrv_handle *h, *newh;
 100         WERROR error;
 101
 102         h = dcesrv_handle_fetch(dce_call->conn, r->in.handle, HTYPE_REGKEY);
 103         DCESRV_CHECK_HANDLE(h);
 104
 105         newh = dcesrv_handle_new(dce_call->conn, HTYPE_REGKEY);
 106
 107         error = reg_key_add_name(newh, (struct registry_key *)h->data, r->in.key.name,
 108                                                          r->in.access_mask,
 109                                                          r->in.sec_desc?r->in.sec_desc->sd:NULL,
 110                                                          (struct registry_key **)&newh->data);
 111
 112         if(W_ERROR_IS_OK(error)) {
 113                 r->out.handle = &newh->wire_handle;
 114         } else {
 115                 dcesrv_handle_destroy(dce_call->conn, newh);
 116         }
 117
 118         return error;
 119 }
 120
 121
 122 /*
 123   winreg_DeleteKey
 124 */
 125 static WERROR winreg_DeleteKey(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
 126                        struct winreg_DeleteKey *r)
 127 {
 128         struct dcesrv_handle *h;
 129         WERROR result;
 130
 131         h = dcesrv_handle_fetch(dce_call->conn, r->in.handle, HTYPE_REGKEY);
 132         DCESRV_CHECK_HANDLE(h);
 133
 134         if (W_ERROR_IS_OK(result)) {
 135                 return reg_key_del((struct registry_key *)h->data, r->in.key.name);
 136         }
 137
 138         return result;
 139 }
 140
 141
 142 /*
 143   winreg_DeleteValue
 144 */
 145 static WERROR winreg_DeleteValue(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
 146                        struct winreg_DeleteValue *r)
 147 {
 148         struct dcesrv_handle *h;
 149         struct registry_key *key;
 150
 151         h = dcesrv_handle_fetch(dce_call->conn, r->in.handle, HTYPE_REGKEY);
 152         DCESRV_CHECK_HANDLE(h);
 153
 154         key = h->data;
 155
 156         return reg_del_value(key, r->in.value.name);
 157 }
 158
 159
 160 /*
 161   winreg_EnumKey
 162 */
 163 static WERROR winreg_EnumKey(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
 164                        struct winreg_EnumKey *r)
 165 {
 166         struct dcesrv_handle *h;
 167         struct registry_key *key;
 168
 169         h = dcesrv_handle_fetch(dce_call->conn, r->in.handle, HTYPE_REGKEY);
 170         DCESRV_CHECK_HANDLE(h);
 171
 172         r->out.result = reg_key_get_subkey_by_index(mem_ctx, (struct registry_key *)h->data, r->in.enum_index, &key);
 173
 174         if (W_ERROR_IS_OK(r->out.result)) {
 175                 r->out.key_name_len = strlen(key->name);
 176                 r->out.out_name = talloc_zero_p(mem_ctx, struct winreg_EnumKeyNameResponse);
 177                 r->out.out_name->name = key->name;
 178                 r->out.class = talloc_zero_p(mem_ctx, struct winreg_String);
 179                 r->out.last_changed_time = talloc_zero_p(mem_ctx, struct winreg_Time);
 180         }
 181
 182         return r->out.result;
 183 }
 184
 185
 186 /*
 187   winreg_EnumValue
 188 */
 189 static WERROR winreg_EnumValue(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
 190                                struct winreg_EnumValue *r)
 191 {
 192         struct dcesrv_handle *h;
 193         struct registry_key *key;
 194         struct registry_value *value;
 195         WERROR result;
 196
 197         h = dcesrv_handle_fetch(dce_call->conn, r->in.handle, HTYPE_REGKEY);
 198         DCESRV_CHECK_HANDLE(h);
 199
 200         key = h->data;
 201
 202         result = reg_key_get_value_by_index(mem_ctx, key, r->in.enum_index, &value);
 203         if (!W_ERROR_IS_OK(result)) {
 204                 return result;
 205         }
 206
 207         r->out.type = talloc_p(mem_ctx, uint32_t);
 208         *r->out.type = value->data_type;
 209         r->out.name_out.name = value->name;
 210         r->out.value = value->data_blk;
 211         r->out.size = talloc_p(mem_ctx, uint32_t);
 212         r->out.length = r->out.size;
 213         *r->out.size = value->data_len;
 214
 215         return WERR_OK;
 216 }
 217
 218
 219 /*
 220   winreg_FlushKey
 221 */
 222 static WERROR winreg_FlushKey(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
 223                        struct winreg_FlushKey *r)
 224 {
 225         struct dcesrv_handle *h;
 226
 227         h = dcesrv_handle_fetch(dce_call->conn, r->in.handle, HTYPE_REGKEY);
 228         DCESRV_CHECK_HANDLE(h);
 229
 230         return reg_key_flush(h->data);
 231 }
 232
 233
 234 /*
 235   winreg_GetKeySecurity
 236 */
 237 static WERROR winreg_GetKeySecurity(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
 238                        struct winreg_GetKeySecurity *r)
 239 {
 240         struct dcesrv_handle *h;
 241
 242         h = dcesrv_handle_fetch(dce_call->conn, r->in.handle, HTYPE_REGKEY);
 243         DCESRV_CHECK_HANDLE(h);
 244
 245         return WERR_NOT_SUPPORTED;
 246 }
 247
 248
 249 /*
 250   winreg_LoadKey
 251 */
 252 static WERROR winreg_LoadKey(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
 253                        struct winreg_LoadKey *r)
 254 {
 255         return WERR_NOT_SUPPORTED;
 256 }
 257
 258
 259 /*
 260   winreg_NotifyChangeKeyValue
 261 */
 262 static WERROR winreg_NotifyChangeKeyValue(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
 263                        struct winreg_NotifyChangeKeyValue *r)
 264 {
 265         return WERR_NOT_SUPPORTED;
 266 }
 267
 268
 269 /*
 270   winreg_OpenKey
 271 */
 272 static WERROR winreg_OpenKey(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
 273                        struct winreg_OpenKey *r)
 274 {
 275         struct dcesrv_handle *h, *newh;
 276         WERROR result;
 277
 278         h = dcesrv_handle_fetch(dce_call->conn, r->in.handle, HTYPE_REGKEY);
 279         DCESRV_CHECK_HANDLE(h);
 280
 281         newh = dcesrv_handle_new(dce_call->conn, HTYPE_REGKEY);
 282
 283         result = reg_open_key(newh, (struct registry_key *)h->data,
 284                                 r->in.keyname.name, (struct registry_key **)&newh->data);
 285
 286         if (W_ERROR_IS_OK(result)) {
 287                 r->out.handle = &newh->wire_handle;
 288         } else {
 289                 dcesrv_handle_destroy(dce_call->conn, newh);
 290         }
 291
 292         return result;
 293 }
 294
 295
 296 /*
 297   winreg_QueryInfoKey
 298 */
 299 static WERROR winreg_QueryInfoKey(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
 300                        struct winreg_QueryInfoKey *r)
 301 {
 302         struct dcesrv_handle *h;
 303         struct registry_key *k;
 304         WERROR ret;
 305
 306         h = dcesrv_handle_fetch(dce_call->conn, r->in.handle, HTYPE_REGKEY);
 307         DCESRV_CHECK_HANDLE(h);
 308         k = h->data;
 309
 310         ret = reg_key_num_subkeys(k, &r->out.num_subkeys);
 311         if (!W_ERROR_IS_OK(ret)) {
 312                 return ret;
 313         }
 314
 315         ret = reg_key_num_values(k, &r->out.num_values);
 316         if (!W_ERROR_IS_OK(ret)) {
 317                 return ret;
 318         }
 319
 320         ret = reg_key_subkeysizes(k, &r->out.max_subkeysize, &r->out.max_subkeylen);
 321         if (!W_ERROR_IS_OK(ret)) {
 322                 return ret;
 323         }
 324
 325         ret = reg_key_valuesizes(k, &r->out.max_valnamelen, &r->out.max_valbufsize);
 326         if (!W_ERROR_IS_OK(ret)) {
 327                 return ret;
 328         }
 329
 330         r->out.secdescsize = 0; /* FIXME */
 331         ZERO_STRUCT(r->out.last_changed_time); /* FIXME */      if (!W_ERROR_IS_OK(ret)) {
 332                 return ret;
 333         }
 334
 335
 336         return WERR_OK;
 337 }
 338
 339
 340 /*
 341   winreg_QueryValue
 342 */
 343 static WERROR winreg_QueryValue(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
 344                        struct winreg_QueryValue *r)
 345 {
 346         struct dcesrv_handle *h;
 347         struct registry_key *key;
 348         struct registry_value *val;
 349         WERROR result;
 350
 351         h = dcesrv_handle_fetch(dce_call->conn, r->in.handle, HTYPE_REGKEY);
 352         DCESRV_CHECK_HANDLE(h);
 353
 354         key = h->data;
 355
 356         result = reg_key_get_value_by_name(mem_ctx, key, r->in.value_name.name, &val);
 357
 358         if (!W_ERROR_IS_OK(result)) {
 359                 return result;
 360         }
 361
 362         r->out.type = &val->data_type;
 363         r->out.size = r->in.size;
 364         r->out.length = &val->data_len;
 365         r->out.data = val->data_blk;
 366
 367         return WERR_OK;
 368 }
 369
 370
 371 /*
 372   winreg_ReplaceKey
 373 */
 374 static WERROR winreg_ReplaceKey(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
 375                        struct winreg_ReplaceKey *r)
 376 {
 377         return WERR_NOT_SUPPORTED;
 378 }
 379
 380
 381 /*
 382   winreg_RestoreKey
 383 */
 384 static WERROR winreg_RestoreKey(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
 385                        struct winreg_RestoreKey *r)
 386 {
 387         return WERR_NOT_SUPPORTED;
 388 }
 389
 390
 391 /*
 392   winreg_SaveKey
 393 */
 394 static WERROR winreg_SaveKey(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
 395                        struct winreg_SaveKey *r)
 396 {
 397         return WERR_NOT_SUPPORTED;
 398 }
 399
 400
 401 /*
 402   winreg_SetKeySecurity
 403 */
 404 static WERROR winreg_SetKeySecurity(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
 405                        struct winreg_SetKeySecurity *r)
 406 {
 407         return WERR_NOT_SUPPORTED;
 408 }
 409
 410
 411 /*
 412   winreg_SetValue
 413 */
 414 static WERROR winreg_SetValue(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
 415                        struct winreg_SetValue *r)
 416 {
 417         return WERR_NOT_SUPPORTED;
 418 }
 419
 420
 421 /*
 422   winreg_UnLoadKey
 423 */
 424 static WERROR winreg_UnLoadKey(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
 425                        struct winreg_UnLoadKey *r)
 426 {
 427         return WERR_NOT_SUPPORTED;
 428 }
 429
 430
 431 /*
 432   winreg_InitiateSystemShutdown
 433 */
 434 static WERROR winreg_InitiateSystemShutdown(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
 435                        struct winreg_InitiateSystemShutdown *r)
 436 {
 437         return WERR_NOT_SUPPORTED;
 438 }
 439
 440
 441 /*
 442   winreg_AbortSystemShutdown
 443 */
 444 static WERROR winreg_AbortSystemShutdown(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
 445                        struct winreg_AbortSystemShutdown *r)
 446 {
 447         return WERR_NOT_SUPPORTED;
 448 }
 449
 450
 451 /*
 452   winreg_GetVersion
 453 */
 454 static WERROR winreg_GetVersion(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
 455                        struct winreg_GetVersion *r)
 456 {
 457         r->out.version = 5;
 458         return WERR_OK;
 459 }
 460
 461
 462 /*
 463   winreg_QueryMultipleValues
 464 */
 465 static WERROR winreg_QueryMultipleValues(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
 466                        struct winreg_QueryMultipleValues *r)
 467 {
 468         return WERR_NOT_SUPPORTED;
 469 }
 470
 471
 472 /*
 473   winreg_InitiateSystemShutdownEx
 474 */
 475 static WERROR winreg_InitiateSystemShutdownEx(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
 476                        struct winreg_InitiateSystemShutdownEx *r)
 477 {
 478         return WERR_NOT_SUPPORTED;
 479 }
 480
 481
 482 /*
 483   winreg_SaveKeyEx
 484 */
 485 static WERROR winreg_SaveKeyEx(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
 486                        struct winreg_SaveKeyEx *r)
 487 {
 488         return WERR_NOT_SUPPORTED;
 489 }
 490
 491
 492 /*
 493   winreg_QueryMultipleValues2
 494 */
 495 static WERROR winreg_QueryMultipleValues2(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
 496                        struct winreg_QueryMultipleValues2 *r)
 497 {
 498         return WERR_NOT_SUPPORTED;
 499 }
 500
 501
 502 /* include the generated boilerplate */
 503 #include "librpc/gen_ndr/ndr_winreg_s.c"