--- libaitrpc/src/blob.c	2010/06/23 16:23:31	1.1.2.2
+++ libaitrpc/src/blob.c	2010/07/08 08:02:02	1.1.2.12
@@ -2,6 +2,66 @@
 
 
 /*
+ * rpc_srv_blobCreate() Create map blob to memory region and return object
+ * @srv = RPC Server instance
+ * @len = BLOB length object
+ * return: NULL error or !=NULL allocated BLOB object
+ */
+inline rpc_blob_t *
+rpc_srv_blobCreate(rpc_srv_t * __restrict srv, int len)
+{
+	rpc_blob_t *blob = NULL;
+	char szFName[MAXPATHLEN];
+	int f;
+	u_int rnd;
+
+	srandomdev();
+	srandom(getpid() ^ time(NULL));
+again:
+	rnd = random() % UINT_MAX;
+
+	memset(szFName, 0, MAXPATHLEN);
+	snprintf(szFName, MAXPATHLEN, BLOB_FILE, srv->srv_blob.dir, rnd);
+	f = open(szFName, O_CREAT | O_EXCL | O_RDWR, 0600);
+	if (f == -1) {
+		if (errno == EEXIST)
+			goto again;
+
+		LOGERR;
+		return NULL;
+	}
+	if (lseek(f, len - 1, SEEK_SET) == -1) {
+		LOGERR;
+		close(f);
+		unlink(szFName);
+		return NULL;
+	} else
+		write(f, "", 1);
+
+	blob = malloc(sizeof(rpc_blob_t));
+	if (!blob) {
+		LOGERR;
+		close(f);
+		unlink(szFName);
+		return NULL;
+	}
+
+	blob->blob_data = mmap(NULL, len, PROT_READ | PROT_WRITE, MAP_SHARED, f, 0);
+	if (blob->blob_data == MAP_FAILED) {
+		LOGERR;
+		free(blob);
+		close(f);
+		unlink(szFName);
+		return NULL;
+	} else
+		close(f);
+
+	blob->blob_len = len;
+	blob->blob_var = rnd;
+	return blob;
+}
+
+/*
  * rpc_srv_blobMap() Map blob to memory region 
  * @srv = RPC Server instance
  * @blob = Map to this BLOB element
@@ -11,7 +71,6 @@ inline int
 rpc_srv_blobMap(rpc_srv_t * __restrict srv, rpc_blob_t * __restrict blob)
 {
 	int f;
-	struct stat sb;
 	char szFName[MAXPATHLEN];
 
 	if (!blob) {
@@ -21,26 +80,22 @@ rpc_srv_blobMap(rpc_srv_t * __restrict srv, rpc_blob_t
 
 	memset(szFName, 0, MAXPATHLEN);
 	snprintf(szFName, MAXPATHLEN, BLOB_FILE, srv->srv_blob.dir, blob->blob_var);
-	f = open(szFName, O_RDONLY);
+	f = open(szFName, O_RDWR);
 	if (f == -1) {
 		LOGERR;
 		return -1;
 	}
-	if (fstat(f, &sb) == -1) {
-		LOGERR;
-		close(f);
-		return -1;
-	}
 
-	blob->blob_data = mmap(NULL, sb.st_size, PROT_READ, MAP_PRIVATE, f, 0);
+	blob->blob_data = mmap(NULL, blob->blob_len, PROT_READ | PROT_WRITE, MAP_SHARED, f, 0);
 	if (blob->blob_data == MAP_FAILED) {
 		LOGERR;
 		close(f);
 		blob->blob_data = NULL;
 		return -1;
 	} else {
-		blob->blob_len = sb.st_size;
 		close(f);
+
+		madvise(blob->blob_data, blob->blob_len, MADV_SEQUENTIAL);
 	}
 
 	return 0;
@@ -59,7 +114,6 @@ rpc_srv_blobUnmap(rpc_blob_t * __restrict blob)
 	else {
 		munmap(blob->blob_data, blob->blob_len);
 		blob->blob_data = NULL;
-		blob->blob_len = 0;
 	}
 }
 
@@ -94,19 +148,221 @@ rpc_srv_blobFree(rpc_srv_t * __restrict srv, rpc_blob_
 
 // ------------------------------------------------------------
 
+/*
+ * rpc_srv_sendBLOB() Send mapped BLOB to client
+ * @cli = Client instance
+ * @blob = Mapped BLOB element
+ * return: -1 error, 0 ok
+ */
 int
-rpc_srv_sendBLOB(rpc_cli_t * __restrict cli, rpc_blob_t *blob)
+rpc_srv_sendBLOB(rpc_cli_t * __restrict cli, rpc_blob_t * __restrict blob)
 {
-	int ret = 0;
-	uint8_t buf[BLOBSIZ];
+	int ret, len;
+	uint8_t *pos;
 
-	if (!cli || !blob) {
+	if (!cli || !blob || !blob->blob_data) {
 		rpc_SetErr(EINVAL, "Error:: invalid arguments ...\n");
 		return -1;
 	}
 
-	do {
-	} while (ret);
+	for (ret = blob->blob_len, pos = blob->blob_data; ret > 0; ret -= len, pos += len)
+		if ((len = send(cli->cli_sock, pos, ret > BLOBSIZ ? BLOBSIZ : ret, 0)) == -1) {
+			LOGERR;
+			return -1;
+		}
 
-	return blob->blob_len;
+	return ret;
+}
+
+/*
+ * rpc_srv_recvBLOB() Receive BLOB from client
+ * @cli = Client instance
+ * @blob = Mapped BLOB element
+ * return: -1 error, 0 ok, >0 unreceived data from client, may be error?
+ */
+int
+rpc_srv_recvBLOB(rpc_cli_t * __restrict cli, rpc_blob_t * __restrict blob)
+{
+	int ret, len;
+	uint8_t *pos;
+	fd_set fds;
+	struct timeval tv = { DEF_RPC_TIMEOUT, 0 };
+
+	if (!cli || !blob || !blob->blob_data) {
+		rpc_SetErr(EINVAL, "Error:: invalid arguments ...\n");
+		return -1;
+	}
+
+	for (ret = blob->blob_len, pos = blob->blob_data; ret > 0; ret -= len, pos += len) {
+		FD_ZERO(&fds);
+		FD_SET(cli->cli_sock, &fds);
+		len = select(cli->cli_sock + 1, &fds, NULL, NULL, &tv);
+		if (len < 1) {
+			LOGERR;
+			return -1;
+		}
+
+		if ((len = recv(cli->cli_sock, pos, BLOBSIZ, 0)) == -1) {
+			LOGERR;
+			return -1;
+		}
+	}
+
+	return ret;
+}
+
+// ------------------------------------------------------------
+
+/*
+ * rpc_cli_sendBLOB() Send BLOB to server
+ * @cli = Client instance
+ * @var = BLOB variable
+ * @data = BLOB data
+ * return: -1 error, 0 ok, 1 remote error
+ */
+int
+rpc_cli_sendBLOB(rpc_cli_t * __restrict cli, rpc_val_t * __restrict var, void * __restrict data)
+{
+	int ret, len;
+	uint8_t *pos;
+	struct tagBLOBHdr hdr;
+	fd_set fds;
+	struct timeval tv = { DEF_RPC_TIMEOUT, 0 };
+
+	if (!cli || !var || !data) {
+		rpc_SetErr(EINVAL, "Error:: invalid arguments ...\n");
+		return -1;
+	}
+
+	memcpy(&hdr.hdr_session, cli->cli_parent, sizeof(rpc_sess_t));
+	hdr.hdr_cmd = set;
+	hdr.hdr_var = 0;
+	hdr.hdr_ret = 0;
+	hdr.hdr_len = var->val_len;
+	if (send(cli->cli_sock, &hdr, sizeof hdr, 0) == -1) {
+		LOGERR;
+		return -1;
+	}
+
+	for (ret = var->val_len, pos = data; ret > 0; ret -= len, pos += len)
+		if ((len = send(cli->cli_sock, pos, ret > BLOBSIZ ? BLOBSIZ : ret, 0)) == -1) {
+			LOGERR;
+			return -1;
+		}
+
+	FD_ZERO(&fds);
+	FD_SET(cli->cli_sock, &fds);
+	switch (select(cli->cli_sock + 1, &fds, NULL, NULL, &tv)) {
+		case -1:
+			LOGERR;
+			return -1;
+		case 0:
+			rpc_SetErr(ETIMEDOUT, "Error:: Timeout reached! Server not responde ...\n");
+			return -1;
+	}
+	if (recv(cli->cli_sock, &hdr, sizeof hdr, 0) == -1) {
+		LOGERR;
+		return -1;
+	}
+	if (hdr.hdr_cmd != error) {
+		if (hdr.hdr_len != var->val_len) {
+			rpc_SetErr(EBADMSG, "Error:: Bad return length packet ...\n");
+			return -1;
+		}
+
+		var->val.blob = hdr.hdr_var;
+	}
+
+	return hdr.hdr_cmd == error;
+}
+
+/*
+ * rpc_cli_recvBLOB() Receive BLOB from server
+ * @cli = Client instance
+ * @var = BLOB variable
+ * @data = BLOB data, must be free after use!
+ * return: -1 error, 0 ok, 1 remote error
+ */
+int
+rpc_cli_recvBLOB(rpc_cli_t * __restrict cli, rpc_val_t * __restrict var, void ** data)
+{
+	int ret, len;
+	uint8_t *pos;
+	fd_set fds;
+	struct timeval tv = { DEF_RPC_TIMEOUT, 0 };
+	struct tagBLOBHdr hdr;
+
+	if (!cli || !var || !data) {
+		rpc_SetErr(EINVAL, "Error:: invalid arguments ...\n");
+		return -1;
+	}
+
+	*data = malloc(var->val_len);
+	if (!*data) {
+		LOGERR;
+		return -1;
+	} else
+		memset(*data, 0, var->val_len);
+
+	memcpy(&hdr.hdr_session, cli->cli_parent, sizeof(rpc_sess_t));
+	hdr.hdr_cmd = get;
+	hdr.hdr_var = (uint32_t) RPC_GET_BLOB(var);
+	hdr.hdr_ret = 0;
+	hdr.hdr_len = 0;
+	if (send(cli->cli_sock, &hdr, sizeof hdr, 0) == -1) {
+		LOGERR;
+		free(*data);
+		*data = NULL;
+		return -1;
+	}
+
+	for (ret = var->val_len, pos = *data; ret > 0; ret -= len, pos += len) {
+		FD_ZERO(&fds);
+		FD_SET(cli->cli_sock, &fds);
+		len = select(cli->cli_sock + 1, &fds, NULL, NULL, &tv);
+		if (len < 1) {
+			LOGERR;
+			free(*data);
+			*data = NULL;
+			return -1;
+		}
+
+		if ((len = recv(cli->cli_sock, pos, BLOBSIZ, 0)) == -1) {
+			LOGERR;
+			free(*data);
+			*data = NULL;
+			return -1;
+		}
+	}
+
+	FD_ZERO(&fds);
+	FD_SET(cli->cli_sock, &fds);
+	switch (select(cli->cli_sock + 1, &fds, NULL, NULL, &tv)) {
+		case -1:
+			LOGERR;
+			free(*data);
+			*data = NULL;
+			return -1;
+		case 0:
+			rpc_SetErr(ETIMEDOUT, "Error:: Timeout reached! Server not responde ...\n");
+			free(*data);
+			*data = NULL;
+			return -1;
+	}
+	if (recv(cli->cli_sock, &hdr, sizeof hdr, 0) == -1) {
+		LOGERR;
+		free(*data);
+		*data = NULL;
+		return -1;
+	}
+	if (hdr.hdr_cmd != error) {
+		if (hdr.hdr_len != var->val_len) {
+			rpc_SetErr(EBADMSG, "Error:: Bad return length packet ...\n");
+			free(*data);
+			*data = NULL;
+			return -1;
+		}
+	}
+
+	return hdr.hdr_cmd == error;
 }