svn commit: samba r7805 - in branches/SAMBA_4_0/source/lib/ldb/common: .

[idra at samba.org]

Author: idra
Date: 2005-06-21 11:14:54 +0000 (Tue, 21 Jun 2005)
New Revision: 7805

WebSVN: http://websvn.samba.org/cgi-bin/viewcvs.cgi?view=rev&root=samba&rev=7805

Log:

add support to read binary files into attributes data like ldap tools does


Modified:
   branches/SAMBA_4_0/source/lib/ldb/common/ldb_ldif.c


Changeset:
Modified: branches/SAMBA_4_0/source/lib/ldb/common/ldb_ldif.c
===================================================================
--- branches/SAMBA_4_0/source/lib/ldb/common/ldb_ldif.c	2005-06-21 07:52:00 UTC (rev 7804)
+++ branches/SAMBA_4_0/source/lib/ldb/common/ldb_ldif.c	2005-06-21 11:14:54 UTC (rev 7805)
@@ -105,6 +105,62 @@
 }
 
 /*
+  
+*/
+static int ldb_read_data_file(void *mem_ctx, struct ldb_val *value)
+{
+	struct stat statbuf;
+	char *buf;
+	int count, size, bytes;
+	int ret;
+	int f;
+
+	f = open(value->data, O_RDONLY);
+	if (f == -1) {
+		return -1;
+	}
+
+	if (fstat(f, &statbuf) != 0) {
+		ret = -1;
+		goto done;
+	}
+
+	if (statbuf.st_size == 0) {
+		ret = -1;
+		goto done;
+	}
+
+	value->data = talloc_size(mem_ctx, statbuf.st_size + 1);
+	if (value->data == NULL) {
+		ret = -1;
+		goto done;
+	}
+	value->data[statbuf.st_size] = 0;
+
+	count = 0;
+	size = statbuf.st_size;
+	buf = value->data;
+	while (count < statbuf.st_size) {
+		bytes = read(f, buf, size);
+		if (bytes == -1) {
+			talloc_free(value->data);
+			ret = -1;
+			goto done;
+		}
+		count += bytes;
+		buf += bytes;
+		size -= bytes;
+	}
+
+	value->length = statbuf.st_size;
+	ret = statbuf.st_size;
+
+done:
+	close(f);
+	return ret;
+}
+
+/*
   this base64 decoder was taken from jitterbug (written by tridge).
   we might need to replace it with a new version
 */
@@ -426,10 +482,11 @@
 
 
 /* simple ldif attribute parser */
-static int next_attr(char **s, const char **attr, struct ldb_val *value)
+static int next_attr(void *mem_ctx, char **s, const char **attr, struct ldb_val *value)
 {
 	char *p;
 	int base64_encoded = 0;
+	int binary_file = 0;
 
 	if (strncmp(*s, "-\n", 2) == 0) {
 		value->length = 0;
@@ -450,6 +507,11 @@
 		p++;
 	}
 
+	if (*p == '<') {
+		binary_file = 1;
+		p++;
+	}
+
 	*attr = *s;
 
 	while (*p == ' ' || *p == '\t') {
@@ -478,6 +540,14 @@
 		value->length = len;
 	}
 
+	if (binary_file) {
+		int len = ldb_read_data_file(mem_ctx, value);
+		if (len == -1) {
+			/* an error occured hile trying to retrieve the file */
+			return -1;
+		}
+	}
+
 	return 0;
 }
 
@@ -564,7 +634,7 @@
 	msg->private_data = chunk;
 	s = chunk;
 
-	if (next_attr(&s, &attr, &value) != 0) {
+	if (next_attr(ldif, &s, &attr, &value) != 0) {
 		goto failed;
 	}
 	
@@ -577,7 +647,7 @@
 
 	msg->dn = value.data;
 
-	while (next_attr(&s, &attr, &value) == 0) {
+	while (next_attr(ldif, &s, &attr, &value) == 0) {
 		ldb_ldif_handler_t read_fn;
 		struct ldb_message_element *el;
 		int ret, empty = 0;