r23801: The FSF has moved around a lot. This fixes their Mass Ave address.
[bbaumbach/samba-autobuild/.git] / source3 / registry / reg_objects.c
index 7ee2cd841415aa59c89e40cace6aeed412ef4564..aebf0142eec15a2fa55e7c1689e579d63593916a 100644 (file)
@@ -5,7 +5,7 @@
  *
  *  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
+ *  the Free Software Foundation; either version 3 of the License, or
  *  (at your option) any later version.
  *  
  *  This program is distributed in the hope that it will be useful,
@@ -14,8 +14,7 @@
  *  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.
+ *  along with this program; if not, see <http://www.gnu.org/licenses/>.
  */
 
 /* Implementation of registry frontend view functions. */
 #undef DBGC_CLASS
 #define DBGC_CLASS DBGC_RPC_SRV
 
+/**********************************************************************
 
-/***********************************************************************
- Init the talloc context held by a REGSUBKEY_CTR structure
- This now zero's the structure
- **********************************************************************/
+ 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.
 
-void regsubkey_ctr_init( REGSUBKEY_CTR *ctr )
-{
-       ZERO_STRUCTP( ctr );
-       ctr->ctx = talloc_init("regsubkey_ctr_init for ctr %p", ctr);
-}
+ 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 )
+WERROR regsubkey_ctr_addkey( REGSUBKEY_CTR *ctr, const char *keyname )
 {
-       char **pp;
+       char **newkeys;
 
-       if ( !keyname )
-               return ctr->num_subkeys;
+       if ( !keyname ) {
+               return WERR_OK;
+       }
 
        /* 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->ctx, char *);
-       else {
-               pp = TALLOC_REALLOC_ARRAY( ctr->ctx, ctr->subkeys, char *, ctr->num_subkeys+1);
-               if ( pp )
-                       ctr->subkeys = pp;
+       if ( regsubkey_ctr_key_exists( ctr, keyname ) ) {
+               return WERR_OK;
        }
 
-       /* allocate the string and save it in the array */
-       
-       ctr->subkeys[ctr->num_subkeys] = talloc_strdup( ctr->ctx, keyname );
+       if (!(newkeys = TALLOC_REALLOC_ARRAY(ctr, ctr->subkeys, char *,
+                                            ctr->num_subkeys+1))) {
+               return WERR_NOMEM;
+       }
+
+       ctr->subkeys = newkeys;
+
+       if (!(ctr->subkeys[ctr->num_subkeys] = talloc_strdup(ctr->subkeys,
+                                                            keyname ))) {
+               /*
+                * Don't shrink the new array again, this wastes a pointer
+                */
+               return WERR_NOMEM;
+       }
        ctr->num_subkeys++;
-       
-       return ctr->num_subkeys;
+
+       return WERR_OK;
 }
  
  /***********************************************************************
Add a new key to the array
Delete a key from the array
  **********************************************************************/
 
 int regsubkey_ctr_delkey( REGSUBKEY_CTR *ctr, const char *keyname )
@@ -94,7 +96,7 @@ int regsubkey_ctr_delkey( REGSUBKEY_CTR *ctr, const char *keyname )
 
        /* update if we have any keys left */
        ctr->num_subkeys--;
-       if ( 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;
@@ -108,6 +110,10 @@ BOOL regsubkey_ctr_key_exists( REGSUBKEY_CTR *ctr, const char *keyname )
 {
        int     i;
        
+       if (!ctr->subkeys) {
+               return False;
+       }
+
        for ( i=0; i<ctr->num_subkeys; i++ ) {
                if ( strequal( ctr->subkeys[i],keyname ) )
                        return True;
@@ -137,34 +143,10 @@ char* regsubkey_ctr_specific_key( REGSUBKEY_CTR *ctr, uint32 key_index )
        return ctr->subkeys[key_index];
 }
 
-/***********************************************************************
- free memory held by a REGSUBKEY_CTR structure
- **********************************************************************/
-
-void regsubkey_ctr_destroy( REGSUBKEY_CTR *ctr )
-{
-       if ( ctr ) {
-               talloc_destroy( ctr->ctx );     
-               ZERO_STRUCTP( ctr );
-       }
-}
-
-
 /*
  * Utility functions for REGVAL_CTR
  */
 
-/***********************************************************************
- Init the talloc context held by a REGSUBKEY_CTR structure
- This now zero's the structure
- **********************************************************************/
-
-void regval_ctr_init( REGVAL_CTR *ctr )
-{
-       ZERO_STRUCTP( ctr );
-       ctr->ctx = talloc_init("regval_ctr_init for ctr %p", ctr);
-}
-
 /***********************************************************************
  How many keys does the container hold ?
  **********************************************************************/
@@ -194,13 +176,20 @@ REGISTRY_VALUE* dup_registry_value( REGISTRY_VALUE *val )
        /* copy all the non-pointer initial data */
        
        memcpy( copy, val, sizeof(REGISTRY_VALUE) );
-       if ( val->data_p ) 
+       
+       copy->size = 0;
+       copy->data_p = NULL;
+       
+       if ( val->data_p && val->size ) 
        {
-               if ( !(copy->data_p = memdup( val->data_p, val->size )) ) {
+               if ( !(copy->data_p = (uint8 *)memdup( val->data_p,
+                                                      val->size )) ) {
                        DEBUG(0,("dup_registry_value: memdup() failed for [%d] bytes!\n",
                                val->size));
                        SAFE_FREE( copy );
+                       return NULL;
                }
+               copy->size = val->size;
        }
        
        return copy;    
@@ -232,7 +221,7 @@ uint8* regval_data_p( REGISTRY_VALUE *val )
 /**********************************************************************
  *********************************************************************/
 
-int regval_size( REGISTRY_VALUE *val )
+uint32 regval_size( REGISTRY_VALUE *val )
 {
        return val->size;
 }
@@ -255,7 +244,7 @@ uint32 regval_type( REGISTRY_VALUE *val )
 
 /***********************************************************************
  Retreive a pointer to a specific value.  Caller shoud dup the structure
- since this memory may go away with a regval_ctr_destroy()
+ since this memory will go away when the ctr is free()'d
  **********************************************************************/
 
 REGISTRY_VALUE* regval_ctr_specific_value( REGVAL_CTR *ctr, uint32 idx )
@@ -266,17 +255,6 @@ REGISTRY_VALUE* regval_ctr_specific_value( REGVAL_CTR *ctr, uint32 idx )
        return ctr->values[idx];
 }
 
-/***********************************************************************
- Retrive the TALLOC_CTX associated with a REGISTRY_VALUE 
- **********************************************************************/
-
-TALLOC_CTX* regval_ctr_getctx( REGVAL_CTR *val )
-{
-       if ( !val )
-               return NULL;
-
-       return val->ctx; }
-
 /***********************************************************************
  Check for the existance of a value
  **********************************************************************/
@@ -299,8 +277,6 @@ BOOL regval_ctr_key_exists( REGVAL_CTR *ctr, const char *value )
 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;
 
@@ -310,23 +286,39 @@ int regval_ctr_addvalue( REGVAL_CTR *ctr, const char *name, uint16 type,
 
        /* allocate a slot in the array of pointers */
                
-       if (  ctr->num_values == 0 )
-               ctr->values = TALLOC_P( ctr->ctx, REGISTRY_VALUE *);
-       else {
-               ppreg = TALLOC_REALLOC_ARRAY( ctr->ctx, ctr->values, REGISTRY_VALUE *, ctr->num_values+1 );
-               if ( ppreg )
-                       ctr->values = ppreg;
+       if (  ctr->num_values == 0 ) {
+               ctr->values = TALLOC_P( ctr, REGISTRY_VALUE *);
+       } else {
+               ctr->values = TALLOC_REALLOC_ARRAY( ctr, ctr->values, REGISTRY_VALUE *, ctr->num_values+1 );
+       }
+
+       if (!ctr->values) {
+               ctr->num_values = 0;
+               return 0;
        }
 
        /* allocate a new value and store the pointer in the arrya */
                
-       ctr->values[ctr->num_values] = TALLOC_P( ctr->ctx, REGISTRY_VALUE);
+       ctr->values[ctr->num_values] = TALLOC_P( ctr, REGISTRY_VALUE);
+       if (!ctr->values[ctr->num_values]) {
+               ctr->num_values = 0;
+               return 0;
+       }
 
        /* 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->ctx, data_p, size );
+       if (size) {
+               ctr->values[ctr->num_values]->data_p = (uint8 *)TALLOC_MEMDUP(
+                       ctr, data_p, size );
+               if (!ctr->values[ctr->num_values]->data_p) {
+                       ctr->num_values = 0;
+                       return 0;
+               }
+       } else {
+               ctr->values[ctr->num_values]->data_p = NULL;
+       }
        ctr->values[ctr->num_values]->size = size;
        ctr->num_values++;
 
@@ -339,29 +331,42 @@ int regval_ctr_addvalue( REGVAL_CTR *ctr, const char *name, uint16 type,
 
 int regval_ctr_copyvalue( REGVAL_CTR *ctr, REGISTRY_VALUE *val )
 {
-       REGISTRY_VALUE **ppreg;
-       
-       if ( val )
-       {
+       if ( val ) {
                /* allocate a slot in the array of pointers */
                
-               if (  ctr->num_values == 0 )
-                       ctr->values = TALLOC_P( ctr->ctx, REGISTRY_VALUE *);
-               else {
-                       ppreg = TALLOC_REALLOC_ARRAY( ctr->ctx, ctr->values, REGISTRY_VALUE *, ctr->num_values+1 );
-                       if ( ppreg )
-                               ctr->values = ppreg;
+               if (  ctr->num_values == 0 ) {
+                       ctr->values = TALLOC_P( ctr, REGISTRY_VALUE *);
+               } else {
+                       ctr->values = TALLOC_REALLOC_ARRAY( ctr, ctr->values, REGISTRY_VALUE *, ctr->num_values+1 );
+               }
+
+               if (!ctr->values) {
+                       ctr->num_values = 0;
+                       return 0;
                }
 
                /* allocate a new value and store the pointer in the arrya */
                
-               ctr->values[ctr->num_values] = TALLOC_P( ctr->ctx, REGISTRY_VALUE);
+               ctr->values[ctr->num_values] = TALLOC_P( ctr, REGISTRY_VALUE);
+               if (!ctr->values[ctr->num_values]) {
+                       ctr->num_values = 0;
+                       return 0;
+               }
 
                /* 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->ctx, val->data_p, val->size );
+               if (val->size) {
+                       ctr->values[ctr->num_values]->data_p = (uint8 *)TALLOC_MEMDUP(
+                               ctr, val->data_p, val->size );
+                       if (!ctr->values[ctr->num_values]->data_p) {
+                               ctr->num_values = 0;
+                               return 0;
+                       }
+               } else {
+                       ctr->values[ctr->num_values]->data_p = NULL;
+               }
                ctr->values[ctr->num_values]->size = val->size;
                ctr->num_values++;
        }
@@ -378,10 +383,6 @@ int regval_ctr_delvalue( REGVAL_CTR *ctr, const char *name )
 {
        int     i;
        
-       /* search for the value */
-       if (!(ctr->num_values))
-               return 0;
-       
        for ( i=0; i<ctr->num_values; i++ ) {
                if ( strequal( ctr->values[i]->valuename, name ) )
                        break;
@@ -392,10 +393,10 @@ int regval_ctr_delvalue( REGVAL_CTR *ctr, const char *name )
        if ( i == ctr->num_values )
                return ctr->num_values;
        
-       /* just shift everything down one */
+       /* If 'i' was not the last element, just shift everything down one */
        ctr->num_values--;
-       if ( ctr->num_values )
-               memmove( ctr->values[i], ctr->values[i+1], sizeof(REGISTRY_VALUE)*(ctr->num_values-i) );
+       if ( i < ctr->num_values )
+               memmove( &ctr->values[i], &ctr->values[i+1], sizeof(REGISTRY_VALUE*)*(ctr->num_values-i) );
        
        return ctr->num_values;
 }
@@ -420,15 +421,27 @@ REGISTRY_VALUE* regval_ctr_getvalue( REGVAL_CTR *ctr, const char *name )
 }
 
 /***********************************************************************
- free memory held by a REGVAL_CTR structure
+ return the data_p as a uint32
  **********************************************************************/
 
-void regval_ctr_destroy( REGVAL_CTR *ctr )
+uint32 regval_dword( REGISTRY_VALUE *val )
 {
-       if ( ctr ) {
-               talloc_destroy( ctr->ctx );
-               ZERO_STRUCTP( ctr );
-       }
+       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;
+}