Changeset 61 for trunk/src

Timestamp:

07/17/06 21:56:59 (18 years ago)

Author:

tim

Message:

various minor speedups and memory leak fixes. Added support for NONE and LINK types.

File:

• trunk/src/reglookup.c (modified) (24 diffs)

trunk/src/reglookup.c

@@ -3 +3 @@
  * Gerald Carter''s regfio interface.
  *
- * Copyright (C) 2005 Timothy D. Morgan
+ * Copyright (C) 2005-2006 Timothy D. Morgan
  * Copyright (C) 2002 Richard Sharpe, rsharpe@richardsharpe.com
  *
@@ -28 +28 @@
 #include <strings.h>
 #include <time.h>
+#include <iconv.h>
 #include "../include/regfio.h"
 #include "../include/void_stack.h"
@@ -43 +44 @@
 /* Other globals */
 const char* special_chars = ",\"\\";
+iconv_t conv_desc;
+
 
 void bailOut(int code, char* message)
@@ -54 +57 @@
  * except for non-printable or special characters are quoted in hex
  * with the syntax '\xQQ' where QQ is the hex ascii value of the quoted
- * character.  A null terminator is added, as only ascii, not binary, 
+ * character.  A null terminator is added, since only ascii, not binary,
  * is returned.
  */
@@ -60 +63 @@
                           unsigned int len, const char* special)
 {
-  unsigned int i;
-  unsigned int num_written=0;
-  unsigned int out_len = sizeof(char)*len+1;
-  char* ret_val = malloc(out_len);
+  unsigned int i, added_len;
+  unsigned int num_written = 0;
+
+  unsigned int buf_len = sizeof(char)*(len+1);
+  char* ret_val = malloc(buf_len);
+  char* tmp_buf;
 
   if(ret_val == NULL)
@@ -70 +75 @@
   for(i=0; i<len; i++)
   {
+    if(buf_len <= (num_written+5))
+    {
+      /* Expand the buffer by the memory consumption rate seen so far 
+       * times the amount of input left to process.  The expansion is bounded 
+       * below by a minimum safety increase, and above by the maximum possible 
+       * output string length.
+       */
+      added_len = (len-i)*num_written/(i+1);
+      if((buf_len+added_len) > (len*4+1))
+        buf_len = len*4+1;
+      else
+      {
+        if (added_len < 5)
+          buf_len += 5;
+        else
+          buf_len += added_len;
+      }
+
+      tmp_buf = realloc(ret_val, buf_len);
+      if(tmp_buf == NULL)
+      {
+        free(ret_val);
+        return NULL;
+      }
+      ret_val = tmp_buf;
+    }
+    
     if(str[i] < 32 || str[i] > 126 || strchr(special, str[i]) != NULL)
     {
-      out_len += 3;
-      /* XXX: may not be the most efficient way of getting enough memory. */
-      ret_val = realloc(ret_val, out_len);
-      if(ret_val == NULL)
-        break;
-      num_written += snprintf(ret_val+num_written, (out_len)-num_written,
+      num_written += snprintf(ret_val + num_written, buf_len - num_written,
                               "\\x%.2X", str[i]);
     }
@@ -107 +134 @@
 
 /*
- * Convert from UniCode to Ascii ... Does not take into account other lang
- * Restrict by ascii_max if > 0
+ * Convert from Unicode to ASCII.
  */
-static int uni_to_ascii(unsigned char *uni, unsigned char *ascii, 
-                        int ascii_max, int uni_max)
-{
-  int i = 0; 
-
-  while (i < ascii_max && (uni[i*2] || uni[i*2+1]))
-  {
-    if (uni_max > 0 && (i*2) >= uni_max) break;
-    ascii[i] = uni[i*2];
-    i++;
-  }
-  ascii[i] = '\0';
-
-  return i;
+static int uni_to_ascii(unsigned char* uni, char* ascii, 
+                        unsigned int uni_max, unsigned int ascii_max)
+{
+  char* inbuf = (char*)uni;
+  char* outbuf = ascii;
+  int ret;
+
+  /* Set up conversion descriptor. */
+  conv_desc = iconv_open("US-ASCII", "UTF-16LE");
+
+  ret = iconv(conv_desc, &inbuf, &uni_max, &outbuf, &ascii_max);
+  if(ret == -1)
+  {
+    iconv_close(conv_desc);
+    return -1;
+  }
+
+  iconv_close(conv_desc);  
+  return strlen(ascii);
 }
 
@@ -130 +161 @@
  * Convert a data value to a string for display
  */
-static unsigned char* data_to_ascii(unsigned char *datap, int len, int type)
-{
-  unsigned char *asciip;
+static char* data_to_ascii(unsigned char *datap, int len, int type)
+{
+  char* asciip;
   unsigned int i;
   unsigned short num_nulls;
-  unsigned char* ascii;
+  char* ascii;
   unsigned char* cur_str;
-  unsigned char* cur_ascii;
+  char* cur_ascii;
   char* cur_quoted;
   unsigned int cur_str_len;
@@ -145 +176 @@
   switch (type) 
   {
+    /* REG_LINK is a symbolic link, stored as a unicode string. */
+  case REG_LINK:
   case REG_SZ:
     ascii_max = sizeof(char)*len;
@@ -151 +184 @@
       return NULL;
     
-    /* XXX: This has to be fixed. It has to be UNICODE */
-    uni_to_ascii(datap, ascii, len, ascii_max);
-    cur_quoted = quote_string((char*)ascii, special_chars);
+    if(uni_to_ascii(datap, ascii, len, ascii_max) < 0)
+    {
+      free(ascii);
+      return NULL;
+    }
+    cur_quoted = quote_string(ascii, special_chars);
     free(ascii);
-    return (unsigned char*)cur_quoted;
+    return cur_quoted;
     break;
 
+    /* XXX: This could be Unicode or ASCII.  Are we processing it 
+     *      correctly in both cases? 
+     */
   case REG_EXPAND_SZ:
     ascii_max = sizeof(char)*len;
@@ -164 +203 @@
       return NULL;
 
-    uni_to_ascii(datap, ascii, len, ascii_max);
-    cur_quoted = quote_string((char*)ascii, special_chars);
+    if(uni_to_ascii(datap, ascii, len, ascii_max) < 0)
+    {
+      free(ascii);
+      return NULL;
+    }
+    cur_quoted = quote_string(ascii, special_chars);
     free(ascii);
-    return (unsigned char*)cur_quoted;
+    return cur_quoted;
     break;
 
@@ -176 +219 @@
       return NULL;
 
-    snprintf((char*)ascii, ascii_max, "0x%.2X%.2X%.2X%.2X", 
+    snprintf(ascii, ascii_max, "0x%.2X%.2X%.2X%.2X", 
              datap[0], datap[1], datap[2], datap[3]);
     return ascii;
@@ -187 +230 @@
       return NULL;
 
-    snprintf((char*)ascii, ascii_max, "0x%.2X%.2X%.2X%.2X", 
+    snprintf(ascii, ascii_max, "0x%.2X%.2X%.2X%.2X", 
              datap[3], datap[2], datap[1], datap[0]);
     return ascii;
@@ -226 +269 @@
       if(num_nulls == 2)
       {
-        uni_to_ascii(cur_str, cur_ascii, cur_str_max, 0);
-        cur_quoted = quote_string((char*)cur_ascii, ",|\"\\");
-        alen = snprintf((char*)asciip, str_rem, "%s", cur_quoted);
+        if(uni_to_ascii(cur_str, cur_ascii, cur_str_max, cur_str_max) < 0)
+        {
+          free(ascii);
+          free(cur_ascii);
+          free(cur_str);
+          return NULL;
+        }
+
+        cur_quoted = quote_string(cur_ascii, ",|\"\\");
+        alen = snprintf(asciip, str_rem, "%s", cur_quoted);
         asciip += alen;
         str_rem -= alen;
@@ -237 +287 @@
         else
         {
-          alen = snprintf((char*)asciip, str_rem, "%c", '|');
-          asciip += alen;
-          str_rem -= alen;
+          if(str_rem > 0)
+          {
+            asciip[0] = '|';
+            asciip[1] = '\0';
+            asciip++;
+            str_rem--;
+          }
           memset(cur_str, 0, cur_str_max);
           cur_str_len = 0;
@@ -255 +309 @@
 
   /* XXX: Dont know what to do with these yet, just print as binary... */
+  case REG_NONE:
   case REG_RESOURCE_LIST:
   case REG_FULL_RESOURCE_DESCRIPTOR:
@@ -260 +315 @@
 
   case REG_BINARY:
-    return (unsigned char*)quote_buffer(datap, len, special_chars);
+    return quote_buffer(datap, len, special_chars);
     break;
 
@@ -369 +424 @@
   uint32 size;
   uint8 tmp_buf[4];
-  unsigned char* quoted_value;
+  char* quoted_value;
   char* quoted_prefix;
   char* quoted_name;
@@ -397 +452 @@
       size = 16384;
     }
+
     quoted_value = data_to_ascii(vk->data, vk->data_size, vk->type);
   }
@@ -443 +499 @@
   time_t tmp_time[1];
   struct tm* tmp_time_s = NULL;
-
+  char* quoted_path = NULL;
+
+  /* XXX: it would be faster to always pass a quoted string to this
+   *      function, instead of re-quoting it every call. */
+  /* XXX: check return of quote_string */
+  quoted_path = quote_string(full_path, special_chars);
   *tmp_time = nt_time_to_unix(&k->mtime);
   tmp_time_s = gmtime(tmp_time);
@@ -463 +524 @@
       dacl = empty_str;
 
-    printf("%s,KEY,,%s,%s,%s,%s,%s\n", full_path, mtime, 
+    printf("%s,KEY,,%s,%s,%s,%s,%s\n", quoted_path, mtime, 
            owner, group, sacl, dacl);
 
@@ -476 +537 @@
   }
   else
-    printf("%s,KEY,,%s\n", full_path, mtime);
+    printf("%s,KEY,,%s\n", quoted_path, mtime);
+
+  free(quoted_path);
 }
 
@@ -673 +736 @@
 static void usage(void)
 {
-  fprintf(stderr, "Usage: readreg [-v] [-s]"
+  fprintf(stderr, "Usage: reglookup [-v] [-s]"
           " [-p <PATH_FILTER>] [-t <TYPE_FILTER>]"
           " <REGISTRY_FILE>\n");
   /* XXX: replace version string with Subversion property? */
-  fprintf(stderr, "Version: 0.2.2\n");
+  fprintf(stderr, "Version: 0.3.0\n");
   fprintf(stderr, "Options:\n");
   fprintf(stderr, "\t-v\t sets verbose mode.\n");
@@ -728 +791 @@
         bailOut(1, "ERROR: '-t' option requires parameter.\n");
       }
-      if((type_filter = regfio_type_str2val(argv[argi])) == 0)
+      if((type_filter = regfio_type_str2val(argv[argi])) < 0)
       {
         fprintf(stderr, "ERROR: Invalid type specified: %s.\n", argv[argi]);
         bailOut(1, "");
       }
-
       type_filter_enabled = true;
     }