?? reg_objects.c
字號:
/* * 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 registry frontend view functions. */#include "includes.h"#undef DBGC_CLASS#define DBGC_CLASS DBGC_RPC_SRV/********************************************************************** Note that the REGSUB_CTR and REGVAL_CTR objects *must* be talloc()'d since the methods use the object pointer as the talloc context for internal private data. There is no longer a regXXX_ctr_intit() and regXXX_ctr_destroy() pair of functions. Simply TALLOC_ZERO_P() and TALLOC_FREE() the object. **********************************************************************//*********************************************************************** Add a new key to the array **********************************************************************/int regsubkey_ctr_addkey( REGSUBKEY_CTR *ctr, const char *keyname ){ char **pp; if ( !keyname ) return ctr->num_subkeys; /* make sure the keyname is not already there */ if ( regsubkey_ctr_key_exists( ctr, keyname ) ) return ctr->num_subkeys; /* allocate a space for the char* in the array */ if ( ctr->subkeys == 0 ) ctr->subkeys = TALLOC_P( ctr, char *); else { pp = TALLOC_REALLOC_ARRAY( ctr, ctr->subkeys, char *, ctr->num_subkeys+1); if ( pp ) ctr->subkeys = pp; } /* allocate the string and save it in the array */ ctr->subkeys[ctr->num_subkeys] = talloc_strdup( ctr, keyname ); ctr->num_subkeys++; return ctr->num_subkeys;} /*********************************************************************** Add a new key to the array **********************************************************************/int regsubkey_ctr_delkey( REGSUBKEY_CTR *ctr, const char *keyname ){ int i; if ( !keyname ) return ctr->num_subkeys; /* make sure the keyname is actually already there */ for ( i=0; i<ctr->num_subkeys; i++ ) { if ( strequal( ctr->subkeys[i], keyname ) ) break; } if ( i == ctr->num_subkeys ) return ctr->num_subkeys; /* update if we have any keys left */ ctr->num_subkeys--; if ( i < ctr->num_subkeys ) memmove( &ctr->subkeys[i], &ctr->subkeys[i+1], sizeof(char*) * (ctr->num_subkeys-i) ); return ctr->num_subkeys;}/*********************************************************************** Check for the existance of a key **********************************************************************/BOOL regsubkey_ctr_key_exists( REGSUBKEY_CTR *ctr, const char *keyname ){ int i; for ( i=0; i<ctr->num_subkeys; i++ ) { if ( strequal( ctr->subkeys[i],keyname ) ) return True; } return False;}/*********************************************************************** How many keys does the container hold ? **********************************************************************/int regsubkey_ctr_numkeys( REGSUBKEY_CTR *ctr ){ return ctr->num_subkeys;}/*********************************************************************** Retreive a specific key string **********************************************************************/char* regsubkey_ctr_specific_key( REGSUBKEY_CTR *ctr, uint32 key_index ){ if ( ! (key_index < ctr->num_subkeys) ) return NULL; return ctr->subkeys[key_index];}/* * Utility functions for REGVAL_CTR *//*********************************************************************** How many keys does the container hold ? **********************************************************************/int regval_ctr_numvals( REGVAL_CTR *ctr ){ return ctr->num_values;}/*********************************************************************** allocate memory for and duplicate a REGISTRY_VALUE. This is malloc'd memory so the caller should free it when done **********************************************************************/REGISTRY_VALUE* dup_registry_value( REGISTRY_VALUE *val ){ REGISTRY_VALUE *copy = NULL; if ( !val ) return NULL; if ( !(copy = SMB_MALLOC_P( REGISTRY_VALUE)) ) { DEBUG(0,("dup_registry_value: malloc() failed!\n")); return NULL; } /* copy all the non-pointer initial data */ memcpy( copy, val, sizeof(REGISTRY_VALUE) ); copy->size = 0; copy->data_p = NULL; if ( val->data_p && val->size ) { if ( !(copy->data_p = memdup( val->data_p, val->size )) ) { DEBUG(0,("dup_registry_value: memdup() failed for [%d] bytes!\n", val->size)); SAFE_FREE( copy ); } copy->size = val->size; } return copy; }/********************************************************************** free the memory allocated to a REGISTRY_VALUE *********************************************************************/ void free_registry_value( REGISTRY_VALUE *val ){ if ( !val ) return; SAFE_FREE( val->data_p ); SAFE_FREE( val ); return;}/********************************************************************** *********************************************************************/uint8* regval_data_p( REGISTRY_VALUE *val ){ return val->data_p;}/********************************************************************** *********************************************************************/uint32 regval_size( REGISTRY_VALUE *val ){ return val->size;}/********************************************************************** *********************************************************************/char* regval_name( REGISTRY_VALUE *val ){ return val->valuename;}/********************************************************************** *********************************************************************/uint32 regval_type( REGISTRY_VALUE *val ){ return val->type;}/*********************************************************************** Retreive a pointer to a specific value. Caller shoud dup the structure since this memory will go away when the ctr is free()'d **********************************************************************/REGISTRY_VALUE* regval_ctr_specific_value( REGVAL_CTR *ctr, uint32 idx ){ if ( !(idx < ctr->num_values) ) return NULL; return ctr->values[idx];}/*********************************************************************** Check for the existance of a value **********************************************************************/BOOL regval_ctr_key_exists( REGVAL_CTR *ctr, const char *value ){ int i; for ( i=0; i<ctr->num_values; i++ ) { if ( strequal( ctr->values[i]->valuename, value) ) return True; } return False;}/*********************************************************************** Add a new registry value to the array **********************************************************************/int regval_ctr_addvalue( REGVAL_CTR *ctr, const char *name, uint16 type, const char *data_p, size_t size ){ REGISTRY_VALUE **ppreg; if ( !name ) return ctr->num_values; /* Delete the current value (if it exists) and add the new one */ regval_ctr_delvalue( ctr, name ); /* allocate a slot in the array of pointers */ if ( ctr->num_values == 0 ) ctr->values = TALLOC_P( ctr, REGISTRY_VALUE *); else { ppreg = TALLOC_REALLOC_ARRAY( ctr, ctr->values, REGISTRY_VALUE *, ctr->num_values+1 ); if ( ppreg ) ctr->values = ppreg; } /* allocate a new value and store the pointer in the arrya */ ctr->values[ctr->num_values] = TALLOC_P( ctr, REGISTRY_VALUE); /* init the value */ fstrcpy( ctr->values[ctr->num_values]->valuename, name ); ctr->values[ctr->num_values]->type = type; ctr->values[ctr->num_values]->data_p = TALLOC_MEMDUP( ctr, data_p, size ); ctr->values[ctr->num_values]->size = size; ctr->num_values++; return ctr->num_values;}/*********************************************************************** Add a new registry value to the array **********************************************************************/int regval_ctr_copyvalue( REGVAL_CTR *ctr, REGISTRY_VALUE *val ){ REGISTRY_VALUE **ppreg; if ( val ) { /* allocate a slot in the array of pointers */ if ( ctr->num_values == 0 ) ctr->values = TALLOC_P( ctr, REGISTRY_VALUE *); else { ppreg = TALLOC_REALLOC_ARRAY( ctr, ctr->values, REGISTRY_VALUE *, ctr->num_values+1 ); if ( ppreg ) ctr->values = ppreg; } /* allocate a new value and store the pointer in the arrya */ ctr->values[ctr->num_values] = TALLOC_P( ctr, REGISTRY_VALUE); /* init the value */ fstrcpy( ctr->values[ctr->num_values]->valuename, val->valuename ); ctr->values[ctr->num_values]->type = val->type; ctr->values[ctr->num_values]->data_p = TALLOC_MEMDUP( ctr, val->data_p, val->size ); ctr->values[ctr->num_values]->size = val->size; ctr->num_values++; } return ctr->num_values;}/*********************************************************************** Delete a single value from the registry container. No need to free memory since it is talloc'd. **********************************************************************/int regval_ctr_delvalue( REGVAL_CTR *ctr, const char *name ){ int i; for ( i=0; i<ctr->num_values; i++ ) { if ( strequal( ctr->values[i]->valuename, name ) ) break; } /* just return if we don't find it */ if ( i == ctr->num_values ) return ctr->num_values; /* If 'i' was not the last element, just shift everything down one */ ctr->num_values--; if ( i < ctr->num_values ) memmove( &ctr->values[i], &ctr->values[i+1], sizeof(REGISTRY_VALUE*)*(ctr->num_values-i) ); return ctr->num_values;}/*********************************************************************** Retrieve single value from the registry container. No need to free memory since it is talloc'd. **********************************************************************/REGISTRY_VALUE* regval_ctr_getvalue( REGVAL_CTR *ctr, const char *name ){ int i; /* search for the value */ for ( i=0; i<ctr->num_values; i++ ) { if ( strequal( ctr->values[i]->valuename, name ) ) return ctr->values[i]; } return NULL;}/*********************************************************************** return the data_p as a uint32 **********************************************************************/uint32 regval_dword( REGISTRY_VALUE *val ){ uint32 data; data = IVAL( regval_data_p(val), 0 ); return data;}/*********************************************************************** return the data_p as a character string **********************************************************************/char* regval_sz( REGISTRY_VALUE *val ){ static pstring data; rpcstr_pull( data, regval_data_p(val), sizeof(data), regval_size(val), 0 ); return data;}
?? 快捷鍵說明
復制代碼
Ctrl + C
搜索代碼
Ctrl + F
全屏模式
F11
切換主題
Ctrl + Shift + D
顯示快捷鍵
?
增大字號
Ctrl + =
減小字號
Ctrl + -