source4/lib/registry/util.c

   1 /*
   2    Unix SMB/CIFS implementation.
   3    Transparent registry backend handling
   4    Copyright (C) Jelmer Vernooij                        2003-2007.
   5
   6    This program is free software; you can redistribute it and/or modify
   7    it under the terms of the GNU General Public License as published by
   8    the Free Software Foundation; either version 3 of the License, or
   9    (at your option) any later version.
  10
  11    This program is distributed in the hope that it will be useful,
  12    but WITHOUT ANY WARRANTY; without even the implied warranty of
  13    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14    GNU General Public License for more details.
  15
  16    You should have received a copy of the GNU General Public License
  17    along with this program.  If not, see <http://www.gnu.org/licenses/>.
  18 */
  19
  20 #include "includes.h"
  21 #include "lib/registry/registry.h"
  22 #include "librpc/gen_ndr/winreg.h"
  23
  24 /**
  25  * @file
  26  * @brief Registry utility functions
  27  */
  28
  29 static const struct {
  30         uint32_t id;
  31         const char *name;
  32 } reg_value_types[] = {
  33         { REG_SZ, "REG_SZ" },
  34         { REG_DWORD, "REG_DWORD" },
  35         { REG_BINARY, "REG_BINARY" },
  36         { REG_EXPAND_SZ, "REG_EXPAND_SZ" },
  37         { REG_NONE, "REG_NONE" },
  38         { 0, NULL }
  39 };
  40
  41 /** Return string description of registry value type */
  42 _PUBLIC_ const char *str_regtype(int type)
  43 {
  44         unsigned int i;
  45         for (i = 0; reg_value_types[i].name; i++) {
  46                 if (reg_value_types[i].id == type)
  47                         return reg_value_types[i].name;
  48         }
  49
  50         return "Unknown";
  51 }
  52
  53 _PUBLIC_ char *reg_val_data_string(TALLOC_CTX *mem_ctx,
  54                                    struct smb_iconv_convenience *iconv_convenience,
  55                                    uint32_t type,
  56                                    const DATA_BLOB data)
  57 {
  58         char *ret = NULL;
  59
  60         if (data.length == 0)
  61                 return talloc_strdup(mem_ctx, "");
  62
  63         switch (type) {
  64                 case REG_EXPAND_SZ:
  65                 case REG_SZ:
  66                         convert_string_talloc_convenience(mem_ctx,
  67                                                           iconv_convenience,
  68                                                           CH_UTF16, CH_UNIX,
  69                                                           data.data,
  70                                                           data.length,
  71                                                           (void **)&ret,
  72                                                           NULL, false);
  73                         break;
  74                 case REG_BINARY:
  75                         ret = data_blob_hex_string_upper(mem_ctx, &data);
  76                         break;
  77                 case REG_DWORD:
  78                         if (IVAL(data.data, 0) == 0) {
  79                                 ret = talloc_strdup(mem_ctx, "0");
  80                         } else {
  81                                 ret = talloc_asprintf(mem_ctx, "0x%x",
  82                                                       IVAL(data.data, 0));
  83                         }
  84                         break;
  85                 case REG_NONE:
  86                         /* "NULL" is the right return value */
  87                         break;
  88                 case REG_MULTI_SZ:
  89                         /* FIXME: We don't support this yet */
  90                         break;
  91                 default:
  92                         /* FIXME */
  93                         /* Other datatypes aren't supported -> return "NULL" */
  94                         break;
  95         }
  96
  97         return ret;
  98 }
  99
 100 /** Generate a string that describes a registry value */
 101 _PUBLIC_ char *reg_val_description(TALLOC_CTX *mem_ctx,
 102                                    struct smb_iconv_convenience *iconv_convenience,
 103                                    const char *name,
 104                                    uint32_t data_type,
 105                                    const DATA_BLOB data)
 106 {
 107         return talloc_asprintf(mem_ctx, "%s = %s : %s", name?name:"<No Name>",
 108                                str_regtype(data_type),
 109                                reg_val_data_string(mem_ctx, iconv_convenience, data_type, data));
 110 }
 111
 112 _PUBLIC_ bool reg_string_to_val(TALLOC_CTX *mem_ctx,
 113                                 struct smb_iconv_convenience *iconv_convenience,
 114                                 const char *type_str,
 115                                 const char *data_str, uint32_t *type,
 116                                 DATA_BLOB *data)
 117 {
 118         unsigned int i;
 119         *type = -1;
 120
 121         /* Find the correct type */
 122         for (i = 0; reg_value_types[i].name; i++) {
 123                 if (!strcmp(reg_value_types[i].name, type_str)) {
 124                         *type = reg_value_types[i].id;
 125                         break;
 126                 }
 127         }
 128
 129         if (*type == -1)
 130                 return false;
 131
 132         /* Convert data appropriately */
 133
 134         switch (*type) {
 135                 case REG_SZ:
 136                 case REG_EXPAND_SZ:
 137                         convert_string_talloc_convenience(mem_ctx,
 138                                                           iconv_convenience,
 139                                                           CH_UNIX, CH_UTF16,
 140                                                           data_str,
 141                                                           strlen(data_str)+1,
 142                                                           (void **)&data->data,
 143                                                           &data->length, false);
 144                         break;
 145                 case REG_BINARY:
 146                         *data = strhex_to_data_blob(mem_ctx, data_str);
 147                         break;
 148                 case REG_DWORD: {
 149                         uint32_t tmp = strtol(data_str, NULL, 0);
 150                         *data = data_blob_talloc(mem_ctx, NULL, 4);
 151                         SIVAL(data->data, 0, tmp);
 152                         }
 153                         break;
 154                 case REG_NONE:
 155                         ZERO_STRUCTP(data);
 156                         break;
 157                 case REG_MULTI_SZ:
 158                         /* FIXME: We don't support this yet */
 159                         return false;
 160                 default:
 161                         /* FIXME */
 162                         /* Other datatypes aren't supported -> return no success */
 163                         return false;
 164         }
 165         return true;
 166 }
 167
 168 /** Open a key by name (including the predefined key name!) */
 169 WERROR reg_open_key_abs(TALLOC_CTX *mem_ctx, struct registry_context *handle,
 170                         const char *name, struct registry_key **result)
 171 {
 172         struct registry_key *predef;
 173         WERROR error;
 174         size_t predeflength;
 175         char *predefname;
 176
 177         if (strchr(name, '\\') != NULL)
 178                 predeflength = strchr(name, '\\')-name;
 179         else
 180                 predeflength = strlen(name);
 181
 182         predefname = talloc_strndup(mem_ctx, name, predeflength);
 183         error = reg_get_predefined_key_by_name(handle, predefname, &predef);
 184         talloc_free(predefname);
 185
 186         if (!W_ERROR_IS_OK(error)) {
 187                 return error;
 188         }
 189
 190         if (strchr(name, '\\')) {
 191                 return reg_open_key(mem_ctx, predef, strchr(name, '\\')+1,
 192                                     result);
 193         } else {
 194                 *result = predef;
 195                 return WERR_OK;
 196         }
 197 }
 198
 199 static WERROR get_abs_parent(TALLOC_CTX *mem_ctx, struct registry_context *ctx,
 200                              const char *path, struct registry_key **parent,
 201                              const char **name)
 202 {
 203         char *parent_name;
 204         WERROR error;
 205
 206         if (strchr(path, '\\') == NULL) {
 207                 return WERR_FOOBAR;
 208         }
 209
 210         parent_name = talloc_strndup(mem_ctx, path, strrchr(path, '\\')-path);
 211
 212         error = reg_open_key_abs(mem_ctx, ctx, parent_name, parent);
 213         if (!W_ERROR_IS_OK(error)) {
 214                 return error;
 215         }
 216
 217         *name = talloc_strdup(mem_ctx, strrchr(path, '\\')+1);
 218
 219         return WERR_OK;
 220 }
 221
 222 WERROR reg_key_del_abs(struct registry_context *ctx, const char *path)
 223 {
 224         struct registry_key *parent;
 225         const char *n;
 226         TALLOC_CTX *mem_ctx = talloc_init("reg_key_del_abs");
 227         WERROR error;
 228
 229         if (!strchr(path, '\\')) {
 230                 return WERR_FOOBAR;
 231         }
 232
 233         error = get_abs_parent(mem_ctx, ctx, path, &parent, &n);
 234         if (W_ERROR_IS_OK(error)) {
 235                 error = reg_key_del(parent, n);
 236         }
 237
 238         talloc_free(mem_ctx);
 239
 240         return error;
 241 }
 242
 243 WERROR reg_key_add_abs(TALLOC_CTX *mem_ctx, struct registry_context *ctx,
 244                        const char *path, uint32_t access_mask,
 245                        struct security_descriptor *sec_desc,
 246                        struct registry_key **result)
 247 {
 248         struct registry_key *parent;
 249         const char *n;
 250         WERROR error;
 251
 252         if (!strchr(path, '\\')) {
 253                 return WERR_ALREADY_EXISTS;
 254         }
 255
 256         error = get_abs_parent(mem_ctx, ctx, path, &parent, &n);
 257         if (!W_ERROR_IS_OK(error)) {
 258                 DEBUG(2, ("Opening parent of %s failed with %s\n", path,
 259                                   win_errstr(error)));
 260                 return error;
 261         }
 262
 263         error = reg_key_add_name(mem_ctx, parent, n, NULL, sec_desc, result);
 264
 265         return error;
 266 }