From d5fa76ceb931c7294a2d3fd7b81297ef4303b9f5 Mon Sep 17 00:00:00 2001
From: Steven Dake <sdake@broked.org>
Date: Tue, 31 Aug 2004 08:21:44 +0000
Subject: [PATCH] Major rewrite of saRecvQueue.  Fixed is: activate poll
 messages are not sent over and over The message header is read into the
 return buffer if the message doesn't match, a new message is allocated This
 allows the message buffer for saRecvQueue to be small, but allows receiving
 very large messages.

(Logical change 1.62)


git-svn-id: http://svn.fedorahosted.org/svn/corosync/trunk@219 fd59a12c-fef9-0310-b244-a6a79926bd2f
---
 lib/util.c | 72 +++++++++++++++++++++++++++++++++++-------------------
 1 file changed, 47 insertions(+), 25 deletions(-)

diff --git a/lib/util.c b/lib/util.c
index f24c5a4c..17c525aa 100644
--- a/lib/util.c
+++ b/lib/util.c
@@ -1,6 +1,5 @@
-
 /*
- * Copyright (c) 2002-2003 MontaVista Software, Inc.
+ * Copyright (c) 2002-2004 MontaVista Software, Inc.
  *
  * All rights reserved.
  *
@@ -156,12 +155,13 @@ retry_recv:
 	if (result == -1 || result != len) {
 		error = SA_ERR_SYSTEM;
 	}
+	assert (result == len);
 	return (error);
 }
 
-struct message_overlay {
+struct res_overlay {
 	struct res_header header;
-	char payload[4096];
+	char payload[0];
 };
 
 SaErrorT
@@ -171,46 +171,68 @@ saRecvQueue (
 	struct queue *queue,
 	int findMessageId)
 {
-	struct message_overlay message;
+	struct res_overlay *overlay;
 	void *inq_msg;
 	int match;
 	SaErrorT error;
 
 	do {
-		error = saRecvRetry (s, &message.header, sizeof (struct res_header),
+		overlay = (struct res_overlay *)msg;
+		error = saRecvRetry (s, overlay, sizeof (struct res_header),
 			MSG_WAITALL | MSG_NOSIGNAL);
 		if (error != SA_OK) {
 			goto error_exit;
 		}
-		if (message.header.size > sizeof (struct res_header)) {
-			error = saRecvRetry (s, &message.payload,
-				message.header.size - sizeof (struct res_header),
-				MSG_WAITALL | MSG_NOSIGNAL);
-			if (error != SA_OK) {
-				goto error_exit;
-			}
-		}
-		match = (message.header.id == findMessageId);
+		assert (overlay->header.size != 0);
 
+		match = (overlay->header.id == findMessageId);
+
+		/*
+		 * Item doesn't match, queue it
+		 */
 		if (match == 0 && queue) {
-			inq_msg = (void *)malloc (message.header.size);
+			inq_msg = (void *)malloc (overlay->header.size);
 			if (inq_msg == 0) {
 				error = SA_ERR_NO_MEMORY;
 				goto error_exit;
 			}
-			memcpy (inq_msg, &message, message.header.size);
-			error = saQueueItemAdd (queue, &inq_msg);
-			if (error != SA_OK) {
-				free (inq_msg);
-				goto error_exit;
+			memcpy (inq_msg, overlay, sizeof (struct res_header));
+			overlay = (struct res_overlay *)inq_msg;
+			if (overlay->header.size > sizeof (struct res_header)) {
+				error = saRecvRetry (s, &overlay->payload,
+					overlay->header.size - sizeof (struct res_header),
+					MSG_WAITALL | MSG_NOSIGNAL);
+				if (error != SA_OK) {
+					goto error_exit;
+				}
 			}
 
-			error = saActivatePoll (s);
-			if (error != SA_OK) {
-				goto error_exit;
+			if (overlay->header.id != MESSAGE_RES_LIB_ACTIVATEPOLL) {
+				error = saQueueItemAdd (queue, &inq_msg);
+				if (error != SA_OK) {
+					free (inq_msg);
+					goto error_exit;
+				}
+
+				error = saActivatePoll (s);
+				if (error != SA_OK) {
+					goto error_exit;
+				}
 			}
 		} else {
-			memcpy (msg, &message, message.header.size);
+		/*
+		 *  its a match, so deliver it
+		 */
+			overlay = (struct res_overlay *)msg;
+			if (overlay->header.size > sizeof (struct res_header)) {
+				error = saRecvRetry (s, &overlay->payload,
+					overlay->header.size - sizeof (struct res_header),
+					MSG_WAITALL | MSG_NOSIGNAL);
+				if (error != SA_OK) {
+					goto error_exit;
+				}
+			}
+			break;
 		}
 	} while (match == 0);