s4:registry - add support for REG_QWORD values

Basically the same as REG_DWORD but these are eight byte long.

1 files changed, 65 insertions, 0 deletions

diff --git a/source4/lib/registry/ldb.c b/source4/lib/registry/ldb.c
index 9e77f1f6d5..5c52cb238f 100644
--- a/source4/lib/registry/ldb.c
+++ b/source4/lib/registry/ldb.c
@@ -110,6 +110,33 @@ static void reg_ldb_unpack_value(TALLOC_CTX *mem_ctx,
 		}
 		break;
 
+	case REG_QWORD:
+		if (val != NULL) {
+			if (val->data[0] != '\0') {
+				/* The data is a plain QWORD */
+				uint64_t tmp = strtoull((char *)val->data, NULL, 0);
+				data->data = talloc_size(mem_ctx, sizeof(uint64_t) + 1);
+				if (data->data != NULL) {
+					SBVAL(data->data, 0, tmp);
+				}
+				data->length = sizeof(uint64_t);
+			} else {
+				/* Provide a possibility to store also UTF8
+				 * REG_QWORD values. This is done by adding a
+				 * '\0' in front of the data */
+				data->data = talloc_size(mem_ctx, val->length - 1);
+				if (data->data != NULL) {
+					memcpy(data->data, val->data + 1,
+					       val->length - 1);
+				}
+				data->length = val->length - 1;
+			}
+		} else {
+			data->data = NULL;
+			data->length = 0;
+		}
+		break;
+
 	case REG_BINARY:
 	default:
 		if (val != NULL) {
@@ -231,6 +258,44 @@ static struct ldb_message *reg_ldb_pack_value(struct ldb_context *ctx,
 		}
 		break;
 
+	case REG_QWORD:
+		if ((data.length > 0) && (data.data != NULL)) {
+			if (data.length == sizeof(uint64_t)) {
+				char *conv_str;
+
+				conv_str = talloc_asprintf(msg, "0x%llx", BVAL(data.data, 0));
+				if (conv_str == NULL) {
+					talloc_free(msg);
+					return NULL;
+				}
+				ret = ldb_msg_add_string(msg, "data", conv_str);
+			} else {
+				/* Provide a possibility to store also UTF8
+				 * REG_QWORD values. This is done by adding a
+				 * '\0' in front of the data */
+				struct ldb_val *val;
+
+				val = talloc_zero(msg, struct ldb_val);
+				if (val == NULL) {
+					talloc_free(msg);
+					return NULL;
+				}
+
+				val->data = talloc_size(msg, data.length + 1);
+				if (val->data == NULL) {
+					talloc_free(msg);
+					return NULL;
+				}
+				val->data[0] = '\0';
+				memcpy(val->data + 1, data.data, data.length);
+				val->length = data.length + 1;
+				ret = ldb_msg_add_value(msg, "data", val, NULL);
+			}
+		} else {
+			ret = ldb_msg_add_empty(msg, "data", LDB_FLAG_MOD_DELETE, NULL);
+		}
+		break;
+
 	case REG_BINARY:
 	default:
 		if ((data.length > 0) && (data.data != NULL)