source4/lib/samba3/registry.c

   1 /*
   2  *  Unix SMB/CIFS implementation.
   3  *  Virtual Windows Registry Layer
   4  *  Copyright (C) Gerald Carter                     2002-2005
   5  *  Copyright (C) Jelmer Vernooij                   2005
   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 2 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 /* Implementation of internal registry database functions. */
  23
  24 #include "includes.h"
  25 #include "lib/samba3/samba3.h"
  26 #include "librpc/gen_ndr/winreg.h"
  27 #include "lib/tdb/include/tdbutil.h"
  28 #include "system/filesys.h"
  29 #include "pstring.h"
  30
  31 #define VALUE_PREFIX    "SAMBA_REGVAL"
  32 #define REGVER_V1       1       /* first db version with write support */
  33
  34 /****************************************************************************
  35  Unpack a list of registry values from the TDB
  36  ***************************************************************************/
  37
  38 static int regdb_unpack_values(TDB_CONTEXT *tdb, TALLOC_CTX *ctx, struct samba3_regkey *key, TDB_DATA data )
  39 {
  40         int             len = 0;
  41         uint32_t        type;
  42         uint32_t        size;
  43         uint8_t         *data_p;
  44         uint32_t        num_values = 0;
  45         int             i;
  46         fstring valuename;
  47
  48         /* loop and unpack the rest of the registry values */
  49
  50         len += tdb_unpack(tdb, data.dptr+len, data.dsize-len, "d", &num_values);
  51
  52         for ( i=0; i<num_values; i++ ) {
  53                 struct samba3_regval val;
  54                 /* unpack the next regval */
  55
  56                 type = REG_NONE;
  57                 size = 0;
  58                 data_p = NULL;
  59                 len += tdb_unpack(tdb, data.dptr+len, data.dsize-len, "fdB",
  60                                   valuename,
  61                                   &val.type,
  62                                   &size,
  63                                   &data_p);
  64                 val.name = talloc_strdup(ctx, valuename);
  65                 val.data = data_blob_talloc(ctx, data_p, size);
  66
  67                 key->values = talloc_realloc(ctx, key->values, struct samba3_regval, key->value_count+1);
  68                 key->values[key->value_count] = val;
  69                 key->value_count++;
  70         }
  71
  72         return len;
  73 }
  74
  75
  76
  77 /***********************************************************************
  78  Open the registry database
  79  ***********************************************************************/
  80
  81 NTSTATUS samba3_read_regdb ( const char *fn, TALLOC_CTX *ctx, struct samba3_regdb *db )
  82 {
  83         uint32_t vers_id;
  84         TDB_CONTEXT *tdb;
  85         TDB_DATA kbuf, vbuf;
  86
  87         /* placeholder tdb; reinit upon startup */
  88
  89         if ( !(tdb = tdb_open(fn, 0, TDB_DEFAULT, O_RDONLY, 0600)) )
  90         {
  91                 DEBUG(0, ("Unable to open registry database %s\n", fn));
  92                 return NT_STATUS_UNSUCCESSFUL;
  93         }
  94
  95         vers_id = tdb_fetch_int32(tdb, "INFO/version");
  96
  97         db->key_count = 0;
  98         db->keys = NULL;
  99
 100         if (vers_id != -1 && vers_id >= REGVER_V1) {
 101                 DEBUG(0, ("Registry version mismatch: %d\n", vers_id));
 102                 return NT_STATUS_UNSUCCESSFUL;
 103         }
 104
 105         for (kbuf = tdb_firstkey(tdb); kbuf.dptr; kbuf = tdb_nextkey(tdb, kbuf))
 106         {
 107                 uint32_t len;
 108                 int i;
 109                 struct samba3_regkey key;
 110                 char *skey;
 111
 112                 if (strncmp(kbuf.dptr, VALUE_PREFIX, strlen(VALUE_PREFIX)) == 0)
 113                         continue;
 114
 115                 vbuf = tdb_fetch(tdb, kbuf);
 116
 117                 key.name = talloc_strdup(ctx, kbuf.dptr);
 118
 119                 len = tdb_unpack(tdb, vbuf.dptr, vbuf.dsize, "d", &key.subkey_count);
 120
 121                 key.value_count = 0;
 122                 key.values = NULL;
 123                 key.subkeys = talloc_array(ctx, char *, key.subkey_count);
 124
 125                 for (i = 0; i < key.subkey_count; i++) {
 126                         fstring tmp;
 127                         len += tdb_unpack( tdb, vbuf.dptr+len, vbuf.dsize-len, "f", tmp );
 128                         key.subkeys[i] = talloc_strdup(ctx, tmp);
 129                 }
 130
 131                 skey = talloc_asprintf(ctx, "%s/%s", VALUE_PREFIX, kbuf.dptr );
 132
 133                 vbuf = tdb_fetch_bystring( tdb, skey );
 134
 135                 if ( vbuf.dptr ) {
 136                         regdb_unpack_values( tdb, ctx, &key, vbuf );
 137                 }
 138
 139                 db->keys = talloc_realloc(ctx, db->keys, struct samba3_regkey, db->key_count+1);
 140                 db->keys[db->key_count] = key;
 141                 db->key_count++;
 142         }
 143
 144         tdb_close(tdb);
 145
 146         return NT_STATUS_OK;
 147 }