/*  *  Unix SMB/CIFS implementation. *  Virtual Windows Registry Layer *  Copyright (C) Gerald Carter                     2002-2005 * *  This program is free software; you can redistribute it and/or modify *  it under the terms of the GNU General Public License as published by *  the Free Software Foundation; either version 2 of the License, or *  (at your option) any later version. *   *  This program is distributed in the hope that it will be useful, *  but WITHOUT ANY WARRANTY; without even the implied warranty of *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the *  GNU General Public License for more details. *   *  You should have received a copy of the GNU General Public License *  along with this program; if not, write to the Free Software *  Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA. *//* Implementation of internal registry database functions. */#include "includes.h"#undef DBGC_CLASS#define DBGC_CLASS DBGC_RPC_SRVstatic TDB_CONTEXT *tdb_reg;static int tdb_refcount;#define VALUE_PREFIX	"SAMBA_REGVAL"/* List the deepest path into the registry.  All part components will be created.*//* If you want to have a part of the path controlled by the tdb and part by   a virtual registry db (e.g. printing), then you have to list the deepest path.   For example,"HKLM/SOFTWARE/Microsoft/Windows NT/CurrentVersion/Print"    allows the reg_db backend to handle everything up to    "HKLM/SOFTWARE/Microsoft/Windows NT/CurrentVersion" and then we'll hook    the reg_printing backend onto the last component of the path (see    KEY_PRINTING_2K in include/rpc_reg.h)   --jerry */static const char *builtin_registry_paths[] = {	KEY_PRINTING_2K,	KEY_PRINTING_PORTS,	KEY_PRINTING,	KEY_SHARES,	KEY_EVENTLOG,	"HKLM\\SOFTWARE\\Microsoft\\Windows NT\\CurrentVersion\\Perflib",	"HKLM\\SOFTWARE\\Microsoft\\Windows NT\\CurrentVersion\\Perflib\\009",	"HKLM\\SYSTEM\\CurrentControlSet\\Control\\Print\\Monitors",	"HKLM\\SYSTEM\\CurrentControlSet\\Control\\ProductOptions",	"HKLM\\SYSTEM\\CurrentControlSet\\Control\\Terminal Server\\DefaultUserConfiguration",	"HKLM\\SYSTEM\\CurrentControlSet\\Services\\TcpIp\\Parameters",	"HKLM\\SYSTEM\\CurrentControlSet\\Services\\Netlogon\\Parameters",	"HKU",	"HKCR",	"HKPD",	"HKPT",	 NULL };struct builtin_regkey_value {	const char *path;	const char *valuename;	uint32 type;	union {		const char *string;		uint32 dw_value;	} data;};static struct builtin_regkey_value builtin_registry_values[] = {	{ KEY_PRINTING_PORTS,		SAMBA_PRINTER_PORT_NAME, REG_SZ, { "" } },	{ KEY_PRINTING_2K,		"DefaultSpoolDirectory", REG_SZ, { "C:\\Windows\\System32\\Spool\\Printers" } },	{ KEY_EVENTLOG,		"DisplayName", REG_SZ, { "Event Log" } }, 	{ KEY_EVENTLOG,		"ErrorControl", REG_DWORD, { (char*)0x00000001 } },	{ NULL, NULL, 0, { NULL } }};#define REGVER_V1	1	/* first db version with write support */	/*********************************************************************** Open the registry data in the tdb ***********************************************************************/ static BOOL init_registry_data( void ){	pstring path, base, remaining;	fstring keyname, subkeyname;	REGSUBKEY_CTR *subkeys;	REGVAL_CTR *values;	int i;	const char *p, *p2;	UNISTR2 data;		/* loop over all of the predefined paths and add each component */		for ( i=0; builtin_registry_paths[i] != NULL; i++ ) {		DEBUG(6,("init_registry_data: Adding [%s]\n", builtin_registry_paths[i]));		pstrcpy( path, builtin_registry_paths[i] );		pstrcpy( base, "" );		p = path;				while ( next_token(&p, keyname, "\\", sizeof(keyname)) ) {					/* build up the registry path from the components */						if ( *base )				pstrcat( base, "\\" );			pstrcat( base, keyname );						/* get the immediate subkeyname (if we have one ) */						*subkeyname = '\0';			if ( *p ) {				pstrcpy( remaining, p );				p2 = remaining;								if ( !next_token(&p2, subkeyname, "\\", sizeof(subkeyname)) )					fstrcpy( subkeyname, p2 );			}			DEBUG(10,("init_registry_data: Storing key [%s] with subkey [%s]\n",				base, *subkeyname ? subkeyname : "NULL"));						/* we don't really care if the lookup succeeds or not since			   we are about to update the record.  We just want any 			   subkeys already present */						if ( !(subkeys = TALLOC_ZERO_P( NULL, REGSUBKEY_CTR )) ) {				DEBUG(0,("talloc() failure!\n"));				return False;			}			regdb_fetch_keys( base, subkeys );			if ( *subkeyname ) 				regsubkey_ctr_addkey( subkeys, subkeyname );			if ( !regdb_store_keys( base, subkeys ))				return False;						TALLOC_FREE( subkeys );		}	}	/* loop over all of the predefined values and add each component */		for ( i=0; builtin_registry_values[i].path != NULL; i++ ) {		if ( !(values = TALLOC_ZERO_P( NULL, REGVAL_CTR )) ) {			DEBUG(0,("talloc() failure!\n"));			return False;		}		regdb_fetch_values( builtin_registry_values[i].path, values );		/* preserve existing values across restarts.  Only add new ones */		if ( !regval_ctr_key_exists( values, builtin_registry_values[i].valuename ) ) 		{			switch( builtin_registry_values[i].type ) {			case REG_DWORD:				regval_ctr_addvalue( values, 				                     builtin_registry_values[i].valuename,						     REG_DWORD,						     (char*)&builtin_registry_values[i].data.dw_value,						     sizeof(uint32) );				break;							case REG_SZ:				init_unistr2( &data, builtin_registry_values[i].data.string, UNI_STR_TERMINATE);				regval_ctr_addvalue( values, 				                     builtin_registry_values[i].valuename,						     REG_SZ,						     (char*)data.buffer,						     data.uni_str_len*sizeof(uint16) );				break;						default:				DEBUG(0,("init_registry_data: invalid value type in builtin_registry_values [%d]\n",					builtin_registry_values[i].type));			}			regdb_store_values( builtin_registry_values[i].path, values );		}				TALLOC_FREE( values );	}		return True;}/*********************************************************************** Open the registry database ***********************************************************************/ BOOL regdb_init( void ){	const char *vstring = "INFO/version";	uint32 vers_id;	if ( tdb_reg )		return True;	if ( !(tdb_reg = tdb_open_log(lock_path("registry.tdb"), 0, TDB_DEFAULT, O_RDWR, 0600)) )	{		tdb_reg = tdb_open_log(lock_path("registry.tdb"), 0, TDB_DEFAULT, O_RDWR|O_CREAT, 0600);		if ( !tdb_reg ) {			DEBUG(0,("regdb_init: Failed to open registry %s (%s)\n",				lock_path("registry.tdb"), strerror(errno) ));			return False;		}				DEBUG(10,("regdb_init: Successfully created registry tdb\n"));	}	tdb_refcount = 1;			vers_id = tdb_fetch_int32(tdb_reg, vstring);	if ( vers_id != REGVER_V1 ) {		/* any upgrade code here if needed */	}	/* always setup the necessary keys and values */	if ( !init_registry_data() ) {		DEBUG(0,("init_registry: Failed to initiailize data in registry!\n"));		return False;	}	return True;}/*********************************************************************** Open the registry.  Must already have been initialized by regdb_init() ***********************************************************************/WERROR regdb_open( void ){	WERROR result = WERR_OK;	if ( tdb_reg ) {		DEBUG(10,("regdb_open: incrementing refcount (%d)\n", tdb_refcount));		tdb_refcount++;		return WERR_OK;	}		become_root();	tdb_reg = tdb_open_log(lock_path("registry.tdb"), 0, TDB_DEFAULT, O_RDWR, 0600);	if ( !tdb_reg ) {		result = ntstatus_to_werror( map_nt_error_from_unix( errno ) );		DEBUG(0,("regdb_open: Failed to open %s! (%s)\n", 			lock_path("registry.tdb"), strerror(errno) ));	}	unbecome_root();	tdb_refcount = 1;	DEBUG(10,("regdb_open: refcount reset (%d)\n", tdb_refcount));	return result;}/*********************************************************************** ***********************************************************************/int regdb_close( void ){	int ret;	tdb_refcount--;	DEBUG(10,("regdb_close: decrementing refcount (%d)\n", tdb_refcount));	if ( tdb_refcount > 0 )		return 0;	SMB_ASSERT( tdb_refcount >= 0 );	ret = tdb_close( tdb_reg );	tdb_reg = NULL;	return ret;}/*********************************************************************** Add subkey strings to the registry tdb under a defined key fmt is the same format as tdb_pack except this function only supports fstrings ***********************************************************************/ static BOOL regdb_store_keys_internal( const char *key, REGSUBKEY_CTR *ctr ){	TDB_DATA kbuf, dbuf;	char *buffer, *tmpbuf;	int i = 0;	uint32 len, buflen;	BOOL ret = True;	uint32 num_subkeys = regsubkey_ctr_numkeys( ctr );	pstring keyname;		if ( !key )		return False;	pstrcpy( keyname, key );	normalize_reg_path( keyname );	/* allocate some initial memory */			buffer = SMB_MALLOC(sizeof(pstring));	buflen = sizeof(pstring);