From deebc934edb9a0a70e7615b1161d7a60fcb1d78f Mon Sep 17 00:00:00 2001
From: Wilco Baan Hofman <wilco@baanhofman.nl>
Date: Wed, 28 Jul 2010 14:46:38 +0200
Subject: Fix hex():00,00,00 bugs and strtol dword bugs, these values are
 explicitly hex.

Signed-off-by: Jelmer Vernooij <jelmer@samba.org>
---
 source4/lib/registry/util.c | 44 +++++++++++++++++++++++++++++++++++++-------
 1 file changed, 37 insertions(+), 7 deletions(-)

(limited to 'source4/lib')

diff --git a/source4/lib/registry/util.c b/source4/lib/registry/util.c
index 2d95559200..29417f86d3 100644
--- a/source4/lib/registry/util.c
+++ b/source4/lib/registry/util.c
@@ -21,6 +21,7 @@
 #include "includes.h"
 #include "lib/registry/registry.h"
 #include "librpc/gen_ndr/winreg.h"
+#include "lib/util/data_blob.h"
 
 _PUBLIC_ char *reg_val_data_string(TALLOC_CTX *mem_ctx, uint32_t type,
 				   const DATA_BLOB data)
@@ -77,6 +78,38 @@ _PUBLIC_ char *reg_val_description(TALLOC_CTX *mem_ctx,
 			       reg_val_data_string(mem_ctx, data_type, data));
 }
 
+static DATA_BLOB reg_strhex_to_data_blob(TALLOC_CTX *mem_ctx, const char *str)
+{
+	DATA_BLOB ret;
+	const char *HEXCHARS = "0123456789ABCDEF";
+	size_t i, j;
+	char *hi, *lo;
+
+	ret = data_blob_talloc_zero(mem_ctx, (strlen(str)+(strlen(str) % 3))/3);
+	j = 0;
+	for (i = 0; i < strlen(str); i++) {
+		hi = strchr(HEXCHARS, toupper(str[i]));
+		if (hi == NULL)
+			continue;
+
+		i++;
+		lo = strchr(HEXCHARS, toupper(str[i]));
+		if (lo == NULL)
+			break;
+		
+		ret.data[j] = PTR_DIFF(hi, HEXCHARS) << 4;
+		ret.data[j] += PTR_DIFF(lo, HEXCHARS);
+		j++;
+
+		if (j > ret.length) {
+			DEBUG(0, ("Trouble converting hex string to bin\n"));
+			break;
+		}
+	}
+	return ret;
+} 
+
+
 _PUBLIC_ bool reg_string_to_val(TALLOC_CTX *mem_ctx, const char *type_str,
 				const char *data_str, uint32_t *type, DATA_BLOB *data)
 {
@@ -104,7 +137,6 @@ _PUBLIC_ bool reg_string_to_val(TALLOC_CTX *mem_ctx, const char *type_str,
 				DEBUG(0, ("Could not convert hex to int\n"));
 				return false;
 			}
-			DEBUG(10, ("Found string type: %s: %d\n", p, *type));
 			talloc_free(tmp_type_str);
 		} else if (strcmp(type_str, "hex") == 0) {
 			*type = REG_BINARY;
@@ -120,19 +152,20 @@ _PUBLIC_ bool reg_string_to_val(TALLOC_CTX *mem_ctx, const char *type_str,
 
 	switch (*type) {
 		case REG_SZ:
-		case REG_EXPAND_SZ:
 			return convert_string_talloc(mem_ctx,
 								 CH_UNIX, CH_UTF16, data_str,
 								 strlen(data_str)+1,
 								 (void **)&data->data,
 								 &data->length, false);
 			break;
+		case REG_MULTI_SZ:
+		case REG_EXPAND_SZ:
 		case REG_BINARY:
-			*data = strhex_to_data_blob(mem_ctx, data_str);
+			*data = reg_strhex_to_data_blob(mem_ctx, data_str);
 			break;
 		case REG_DWORD:
 		case REG_DWORD_BIG_ENDIAN: {
-			uint32_t tmp = strtol(data_str, NULL, 0);
+			uint32_t tmp = strtol(data_str, NULL, 16);
 			*data = data_blob_talloc(mem_ctx, NULL, sizeof(uint32_t));
 			if (data->data == NULL) return false;
 			SIVAL(data->data, 0, tmp);
@@ -148,9 +181,6 @@ _PUBLIC_ bool reg_string_to_val(TALLOC_CTX *mem_ctx, const char *type_str,
 		case REG_NONE:
 			ZERO_STRUCTP(data);
 			break;
-		case REG_MULTI_SZ:
-			/* FIXME: We don't support this yet */
-			return false;
 		default:
 			/* FIXME */
 			/* Other datatypes aren't supported -> return no success */
-- 
cgit