Changeset 32

Timestamp:

07/16/05 19:16:17 (19 years ago)

Author:

tim

Message:

Added key name retrieval to test utility

Code formatting changes to regfio.c

Added type<->string mapping functions to regfio library

Location:

trunk

Files:

• include/regfio.h (modified) (2 diffs)
• include/smb_deps.h (modified) (1 diff)
• lib/regfio.c (modified) (27 diffs)
• test/lib-regfio.c (modified) (2 diffs)

trunk/include/regfio.h

@@ -48 +48 @@
 #include "smb_deps.h"
 
-
 /******************************************************************************/
 /* Macros */
  
+/* Registry data types */
+#define REG_NONE                       0
+#define REG_SZ                         1
+#define REG_EXPAND_SZ                  2
+#define REG_BINARY                     3
+#define REG_DWORD                      4
+#define REG_DWORD_LE                   4        /* DWORD, little endian */
+#define REG_DWORD_BE                   5        /* DWORD, big endian */
+#define REG_LINK                       6
+#define REG_MULTI_SZ                   7
+#define REG_RESOURCE_LIST              8
+#define REG_FULL_RESOURCE_DESCRIPTOR   9
+#define REG_RESOURCE_REQUIREMENTS_LIST 10
+/* Not a real type in the registry */
+#define REG_KEY                        255
+
+typedef struct _val_str { 
+  unsigned int val;
+  const char * str;
+} VAL_STR;
+
 #define REGF_BLOCKSIZE          0x1000
 #define REGF_ALLOC_BLOCK        0x1000
@@ -230 +250 @@
 /* Function Declarations */
  
+const char* type_val2str(unsigned int val);
+int type_str2val(const char* str);
+
 REGF_FILE*    regfio_open( const char *filename );
 int           regfio_close( REGF_FILE *r );

trunk/include/smb_deps.h

@@ -366 +366 @@
 /* End of stuff from reg_objects.h */
 
-/* From rpc_reg.h */
-
-/* Registry data types */
-
-#define REG_NONE                       0
-#define REG_SZ                         1
-#define REG_EXPAND_SZ                  2
-#define REG_BINARY                     3
-#define REG_DWORD                      4
-#define REG_DWORD_LE                   4        /* DWORD, little endian */
-#define REG_DWORD_BE                   5        /* DWORD, big endian */
-#define REG_LINK                       6
-#define REG_MULTI_SZ                   7
-#define REG_RESOURCE_LIST              8
-#define REG_FULL_RESOURCE_DESCRIPTOR   9
-#define REG_RESOURCE_REQUIREMENTS_LIST 10
-
-/* End of stuff from rpc_reg.h */
-
 /* From rpc_secdes.h */
 

trunk/lib/regfio.c

@@ -35 +35 @@
  ******************************************************************/
 
+/* Registry types mapping */
+const VAL_STR reg_type_names[] = 
+{
+  { REG_SZ,                        "SZ"           },
+  { REG_EXPAND_SZ,                 "EXPAND_SZ"    },
+  { REG_BINARY,                    "BIN"          },
+  { REG_DWORD,                     "DWORD"        },
+  { REG_DWORD_BE,                  "DWORD_BE"     },
+  { REG_LINK,                      "LINK"         },
+  { REG_MULTI_SZ,                  "MULTI_SZ"     },
+  { REG_RESOURCE_LIST,             "RSRC_LIST"    },
+  { REG_FULL_RESOURCE_DESCRIPTOR,  "RSRC_DESC"    },
+  { REG_RESOURCE_REQUIREMENTS_LIST,"RSRC_REQ_LIST"},
+  { REG_KEY,                       "KEY"          },
+  { 0,                             NULL           },
+};
+
+
+/* Returns NULL on error */
+const char* type_val2str(unsigned int val)
+{
+  int i;
+
+  for(i=0; reg_type_names[i].val && reg_type_names[i].str; i++)
+    if (reg_type_names[i].val == val) 
+      return reg_type_names[i].str;
+
+  return NULL;
+}
+
+
+/* Returns 0 on error */
+int type_str2val(const char* str)
+{
+  int i;
+
+  for(i=0; reg_type_names[i].val && reg_type_names[i].str; i++) 
+    if (strcmp(reg_type_names[i].str, str) == 0) 
+      return reg_type_names[i].val;
+
+  return 0;
+}
+
 
 /*******************************************************************
@@ -95 +138 @@
   bytes_read = returned = 0;
 
-  while ( bytes_read < block_size ) {
-    if ( (returned = read( file->fd, buffer+bytes_read, block_size-bytes_read )) == -1 ) {
+  while ( bytes_read < block_size ) 
+  {
+    if((returned = 
+        read(file->fd, buffer+bytes_read, block_size-bytes_read)) == -1)
+    {
       /*DEBUG(0,("read_block: read() failed (%s)\n", strerror(errno) ));*/
       return false;
     }
-    if ( (returned == 0) && (bytes_read < block_size) ) {
+    if ((returned == 0) && (bytes_read < block_size)) 
+    {
       /*DEBUG(0,("read_block: not a vald registry file ?\n" ));*/
       return false;
     }   
-                
+
     bytes_read += returned;
   }
@@ -701 +748 @@
   /* get the data if necessary */
 
-  if ( vk->data_size != 0 ) {
+  if ( vk->data_size != 0 ) 
+  {
     bool charmode = false;
 
@@ -708 +756 @@
 
     /* the data is stored in the offset if the size <= 4 */
-
-    if ( !(vk->data_size & VK_DATA_IN_OFFSET) ) {
+    if ( !(vk->data_size & VK_DATA_IN_OFFSET) ) 
+    {
       REGF_HBIN *hblock = hbin;
       uint32 data_rec_size;
 
-      if ( hbin->ps.io ) {
+      if ( hbin->ps.io ) 
+      {
         if ( !(vk->data = (uint8*)zcalloc(sizeof(uint8), vk->data_size) ) )
           return false;
@@ -719 +768 @@
 
       /* this data can be in another hbin */
-      if ( !hbin_contains_offset( hbin, vk->data_off ) ) {
+      if ( !hbin_contains_offset( hbin, vk->data_off ) ) 
+      {
         if ( !(hblock = lookup_hbin_block( file, vk->data_off )) )
           return false;
       }
-      if ( !(prs_set_offset( &hblock->ps, (vk->data_off+HBIN_HDR_SIZE-hblock->first_hbin_off)-sizeof(uint32) )) )
-        return false;
-
-      if ( !hblock->ps.io ) {
+      if (!(prs_set_offset(&hblock->ps, 
+                           (vk->data_off
+                            + HBIN_HDR_SIZE
+                            - hblock->first_hbin_off)
+                           - sizeof(uint32))))
+      { return false; }
+
+      if ( !hblock->ps.io ) 
+      {
         data_rec_size = ( (vk->data_size+sizeof(uint32)) & 0xfffffff8 ) + 8;
         data_rec_size = ( data_rec_size - 1 ) ^ 0xFFFFFFFF;
@@ -732 +787 @@
       if ( !prs_uint32( "data_rec_size", &hblock->ps, depth, &data_rec_size ))
         return false;
-      if ( !prs_uint8s( charmode, "data", &hblock->ps, depth, vk->data, vk->data_size) )
+      if(!prs_uint8s(charmode, "data", &hblock->ps, depth, 
+                     vk->data, vk->data_size))
         return false;
 
@@ -738 +794 @@
         hblock->dirty = true;
     }
-    else {
-      if ( !(vk->data = zcalloc(sizeof(uint8), 4) ) )
+    else 
+    {
+      if(!(vk->data = zcalloc(sizeof(uint8), 4)))
         return false;
       SIVAL( vk->data, 0, vk->data_off );
@@ -749 +806 @@
 
   data_size = ((start_off - end_off ) & 0xfffffff8 );
-  if ( data_size !=  vk->rec_size )
-    /*DEBUG(10,("prs_vk_rec: data_size check failed (0x%x < 0x%x)\n", data_size, vk->rec_size));*/
-
-    if ( !hbin->ps.io )
-      hbin->dirty = true;
+  /*if ( data_size !=  vk->rec_size )
+    DEBUG(10,("prs_vk_rec: data_size check failed (0x%x < 0x%x)\n", data_size, vk->rec_size));*/
+
+  if ( !hbin->ps.io )
+    hbin->dirty = true;
 
   return true;
@@ -763 +820 @@
  in the prs_struct *ps.
 *******************************************************************/
-static bool hbin_prs_vk_records( const char *desc, REGF_HBIN *hbin, 
-                                 int depth, REGF_NK_REC *nk, REGF_FILE *file )
+static bool hbin_prs_vk_records(const char *desc, REGF_HBIN *hbin, 
+                                int depth, REGF_NK_REC *nk, REGF_FILE *file)
 {
   int i;
@@ -772 +829 @@
         
   /* check if we have anything to do first */
-        
-  if ( nk->num_values == 0 )
+  if(nk->num_values == 0)
     return true;
                 
-  if ( hbin->ps.io ) {
-    if ( !(nk->values = (REGF_VK_REC*)zcalloc(sizeof(REGF_VK_REC), nk->num_values ) ) )
+  if(hbin->ps.io)
+  {
+    if (!(nk->values = (REGF_VK_REC*)zcalloc(sizeof(REGF_VK_REC), 
+                                              nk->num_values )))
       return false;
   }
         
   /* convert the offset to something relative to this HBIN block */
-        
-  if ( !prs_set_offset( &hbin->ps, nk->values_off+HBIN_HDR_SIZE-hbin->first_hbin_off-sizeof(uint32)) )
-    return false;
-
-  if ( !hbin->ps.io ) { 
+  if (!prs_set_offset(&hbin->ps, 
+                      nk->values_off
+                      + HBIN_HDR_SIZE
+                      - hbin->first_hbin_off
+                      - sizeof(uint32)))
+  { return false; }
+
+  if ( !hbin->ps.io ) 
+  { 
     record_size = ( ( nk->num_values * sizeof(uint32) ) & 0xfffffff8 ) + 8;
     record_size = (record_size - 1) ^ 0xFFFFFFFF;
@@ -794 +856 @@
     return false;
                 
-  for ( i=0; i<nk->num_values; i++ ) {
+  for ( i=0; i<nk->num_values; i++ ) 
+  {
     if ( !prs_uint32( "vk_off", &hbin->ps, depth, &nk->values[i].rec_off ) )
       return false;
   }
 
-  for ( i=0; i<nk->num_values; i++ ) {
+  for ( i=0; i<nk->num_values; i++ ) 
+  {
     REGF_HBIN *sub_hbin = hbin;
     uint32 new_offset;
         
-    if ( !hbin_contains_offset( hbin, nk->values[i].rec_off ) ) {
+    if ( !hbin_contains_offset( hbin, nk->values[i].rec_off ) ) 
+    {
       sub_hbin = lookup_hbin_block( file, nk->values[i].rec_off );
-      if ( !sub_hbin ) {
+      if ( !sub_hbin ) 
+      {
         /*DEBUG(0,("hbin_prs_vk_records: Failed to find HBIN block containing offset [0x%x]\n", 
           nk->values[i].hbin_off));*/
@@ -812 +878 @@
     }
                 
-    new_offset = nk->values[i].rec_off + HBIN_HDR_SIZE - sub_hbin->first_hbin_off;
-    if ( !prs_set_offset( &sub_hbin->ps, new_offset ) )
-      return false;
-    if ( !hbin_prs_vk_rec( "vk_rec", sub_hbin, depth, &nk->values[i], file ) )
+    new_offset = nk->values[i].rec_off 
+      + HBIN_HDR_SIZE 
+      - sub_hbin->first_hbin_off;
+
+    if (!prs_set_offset(&sub_hbin->ps, new_offset))
+      return false;
+    if (!hbin_prs_vk_rec("vk_rec", sub_hbin, depth, &nk->values[i], file))
       return false;
   }
@@ -822 +891 @@
     hbin->dirty = true;
 
-
   return true;
 }
@@ -829 +897 @@
 /*******************************************************************
  *******************************************************************/
-
 static REGF_SK_REC* find_sk_record_by_offset( REGF_FILE *file, uint32 offset )
 {
@@ -842 +909 @@
 }
 
-/*******************************************************************
- *******************************************************************/
-
+
+/*******************************************************************
+ *******************************************************************/
 static REGF_SK_REC* find_sk_record_by_sec_desc( REGF_FILE *file, SEC_DESC *sd )
 {
@@ -859 +926 @@
 }
 
-/*******************************************************************
- *******************************************************************/
-
+
+/*******************************************************************
+ *******************************************************************/
 static bool hbin_prs_key( REGF_FILE *file, REGF_HBIN *hbin, REGF_NK_REC *nk )
 {
@@ -876 +943 @@
   /* fill in values */
         
-  if ( nk->num_values && (nk->values_off!=REGF_OFFSET_NONE) ) {
+  if ( nk->num_values && (nk->values_off!=REGF_OFFSET_NONE) ) 
+  {
     sub_hbin = hbin;
-    if ( !hbin_contains_offset( hbin, nk->values_off ) ) {
+    if ( !hbin_contains_offset( hbin, nk->values_off ) ) 
+    {
       sub_hbin = lookup_hbin_block( file, nk->values_off );
-      if ( !sub_hbin ) {
+      if ( !sub_hbin ) 
+      {
         /*DEBUG(0,("hbin_prs_key: Failed to find HBIN block containing value_list_offset [0x%x]\n", 
           nk->values_off));*/
@@ -887 +957 @@
     }
                 
-    if ( !hbin_prs_vk_records( "vk_rec", sub_hbin, depth, nk, file ))
+    if(!hbin_prs_vk_records("vk_rec", sub_hbin, depth, nk, file))
       return false;
   }
                 
   /* now get subkeys */
-        
-  if ( nk->num_subkeys && (nk->subkeys_off!=REGF_OFFSET_NONE) ) {
+  if ( nk->num_subkeys && (nk->subkeys_off!=REGF_OFFSET_NONE) ) 
+  {
     sub_hbin = hbin;
-    if ( !hbin_contains_offset( hbin, nk->subkeys_off ) ) {
+    if ( !hbin_contains_offset( hbin, nk->subkeys_off ) ) 
+    {
       sub_hbin = lookup_hbin_block( file, nk->subkeys_off );
-      if ( !sub_hbin ) {
+      if ( !sub_hbin ) 
+      {
         /*DEBUG(0,("hbin_prs_key: Failed to find HBIN block containing subkey_offset [0x%x]\n", 
           nk->subkeys_off));*/
@@ -904 +976 @@
     }
                 
-    if ( !hbin_prs_lf_records( "lf_rec", sub_hbin, depth, nk ))
+    if (!hbin_prs_lf_records("lf_rec", sub_hbin, depth, nk))
       return false;
   }
@@ -910 +982 @@
   /* get the to the security descriptor.  First look if we have already parsed it */
         
-  if ( (nk->sk_off!=REGF_OFFSET_NONE) && !( nk->sec_desc = find_sk_record_by_offset( file, nk->sk_off )) ) {
-
+  if ((nk->sk_off!=REGF_OFFSET_NONE) 
+      && !(nk->sec_desc = find_sk_record_by_offset( file, nk->sk_off )))
+  {
     sub_hbin = hbin;
-    if ( !hbin_contains_offset( hbin, nk->sk_off ) ) {
+    if (!hbin_contains_offset(hbin, nk->sk_off))
+    {
       sub_hbin = lookup_hbin_block( file, nk->sk_off );
       if ( !sub_hbin ) {
@@ -937 +1011 @@
 }
 
-/*******************************************************************
- *******************************************************************/
-
+
+/*******************************************************************
+ *******************************************************************/
 static bool next_record( REGF_HBIN *hbin, const char *hdr, bool *eob )
 {
@@ -954 +1028 @@
   /* assume that the current offset is at the reacord header 
      and we need to backup to read the record size */
-
   curr_off -= sizeof(uint32);
 
   block_size = ps->buffer_size;
   record_size = 0;
-  while ( !found ) {
-
+  while ( !found ) 
+  {
     curr_off = curr_off+record_size;
     if ( curr_off >= block_size ) 
@@ -985 +1058 @@
 
   /* mark prs_struct as done ( at end ) if no more SK records */
-  /* mark end-of-block as true */
-        
-  if ( !found ) 
+  /* mark end-of-block as true */       
+  if ( !found )
   {
     prs_set_offset( &hbin->ps, hbin->ps.buffer_size );
@@ -993 +1065 @@
     return false;
   }
-                
-  if ( !prs_set_offset( ps, curr_off ) )
+
+  if (!prs_set_offset(ps, curr_off))
     return false;
 
@@ -1003 +1075 @@
 /*******************************************************************
  *******************************************************************/
-static bool next_nk_record( REGF_FILE *file, REGF_HBIN *hbin, REGF_NK_REC *nk, bool *eob )
-{
-  if ( next_record( hbin, "nk", eob ) && hbin_prs_key( file, hbin, nk ) )
+static bool next_nk_record(REGF_FILE *file, REGF_HBIN *hbin, 
+                           REGF_NK_REC *nk, bool *eob)
+{
+  if (next_record(hbin, "nk", eob) 
+      && hbin_prs_key(file, hbin, nk))
     return true;
         
@@ -1161 +1235 @@
 
   /* see if there is anything left to report */
-        
-  if ( !nk || (nk->subkeys_off==REGF_OFFSET_NONE) || (nk->subkey_index >= nk->num_subkeys) )
+  if (!nk || (nk->subkeys_off==REGF_OFFSET_NONE) 
+      || (nk->subkey_index >= nk->num_subkeys))
     return NULL;
 
   /* find the HBIN block which should contain the nk record */
-        
   if(!(hbin 
        = lookup_hbin_block(file, nk->subkeys.hashes[nk->subkey_index].nk_off )))
@@ -1176 +1249 @@
         
   nk_offset = nk->subkeys.hashes[nk->subkey_index].nk_off;
-  if ( !prs_set_offset( &hbin->ps, (HBIN_HDR_SIZE + nk_offset - hbin->first_hbin_off) ) )
+  if(!prs_set_offset(&hbin->ps, 
+                     (HBIN_HDR_SIZE + nk_offset - hbin->first_hbin_off)))
     return NULL;
                 
   nk->subkey_index++;
-  if (!(subkey = (REGF_NK_REC*)zalloc(sizeof(REGF_NK_REC))))
-    return NULL;
-
-  if ( !hbin_prs_key( file, hbin, subkey ) )
-    return NULL;
-        
+  if(!(subkey = (REGF_NK_REC*)zalloc(sizeof(REGF_NK_REC))))
+    return NULL;
+
+  if(!hbin_prs_key(file, hbin, subkey))
+    return NULL;
+
   return subkey;
 }

trunk/test/lib-regfio.c

@@ -74 +74 @@
   }
 
+  /* Cut trailing slash */  
+  if(buf[buf_len-buf_left-2] == '/')
+    buf[buf_len-buf_left-2] = '\0';
+
   return buf;
 }
 
 
-void printKeyTree(REGF_FILE* f,  void_stack* nk_stack)
+void printKeyList(REGF_NK_REC* nk, char* prefix)
+{
+  unsigned int i;
+  const char* str_type;
+  
+  for(i=0; i < nk->num_values; i++)
+  {
+    str_type = type_val2str(nk->values[i].type);
+    printf("%s/%s:%s=\n", prefix, nk->values[i].valuename, str_type);
+  }
+}
+
+
+void printKeyTree(REGF_FILE* f, void_stack* nk_stack)
 {
   REGF_NK_REC* cur;
@@ -84 +101 @@
   char* path;
 
-  while((cur = (REGF_NK_REC*)void_stack_cur(nk_stack)) != NULL)
+  if((cur = (REGF_NK_REC*)void_stack_cur(nk_stack)) != NULL)
   {
-    if((sub = regfio_fetch_subkey(f, cur)) != NULL)
+    printf("%s:KEY\n", cur->keyname);
+    while((cur = (REGF_NK_REC*)void_stack_cur(nk_stack)) != NULL)
     {
-      path = getStackPath(nk_stack);
-      if(path != NULL)
+      if((sub = regfio_fetch_subkey(f, cur)) != NULL)
       {
-        printf("%s%s:KEY\n", path, sub->keyname);
-        free(path);
+        void_stack_push(nk_stack, sub);
+        path = getStackPath(nk_stack);
+        if(path != NULL)
+        {
+          printKeyList(cur, path);
+          printf("%s:KEY\n", path);
+          free(path);
+        }
       }
-      void_stack_push(nk_stack, sub);
-    }
-    else
-    {
-      cur = void_stack_pop(nk_stack);
-      /* XXX: This is just a shallow free.  Need to write deep free
-       * routines to replace the Samba code for this. 
-       */ 
-      if(cur != NULL)
-        free(cur);
+      else
+      {
+        cur = void_stack_pop(nk_stack);
+        /* XXX: This is just a shallow free.  Need to write deep free
+         * routines to replace the Samba code for this. 
+         */ 
+        if(cur != NULL)
+          free(cur);
+      }
     }
   }