svn commit: samba r3026 - in branches/SAMBA_4_0/source: lib/messaging torture/local

[tridge at samba.org]

Author: tridge
Date: 2004-10-17 22:28:26 +0000 (Sun, 17 Oct 2004)
New Revision: 3026

WebSVN: http://websvn.samba.org/websvn/changeset.php?rep=samba&path=/branches/SAMBA_4_0/source&rev=3026&nolog=1

Log:
- added automatic retry to messages when the servers listen queue is
  full. This means callers can just "send and forget" rather than
  having to check for a temporary failure. The mechanism takes nice
  advantage of the timed events handling is our events code. A message
  will only fail now if we completely run out of some resource (such
  as memory).

- changed the test code not to do retries itself, but only to warn on real failures


Modified:
   branches/SAMBA_4_0/source/lib/messaging/messaging.c
   branches/SAMBA_4_0/source/torture/local/messaging.c


Changeset:
Modified: branches/SAMBA_4_0/source/lib/messaging/messaging.c
===================================================================
--- branches/SAMBA_4_0/source/lib/messaging/messaging.c	2004-10-17 22:04:15 UTC (rev 3025)
+++ branches/SAMBA_4_0/source/lib/messaging/messaging.c	2004-10-17 22:28:26 UTC (rev 3026)
@@ -49,8 +49,6 @@
 
 /* an individual message */
 struct messaging_rec {
-	struct messaging_rec *next, *prev;
-
 	struct messaging_state *msg;
 	struct socket_context *sock;
 	struct fd_event *fde;
@@ -324,6 +322,43 @@
 
 
 /*
+  when the servers listen queue is full we use this to backoff the message
+*/
+static void messaging_backoff_handler(struct event_context *ev, struct timed_event *te, time_t t)
+{
+	struct messaging_rec *rec = te->private;
+	struct messaging_state *msg = rec->msg;
+	NTSTATUS status;
+	struct fd_event fde;
+
+	status = socket_connect(rec->sock, NULL, 0, rec->path, 0, 0);
+	if (NT_STATUS_EQUAL(status, STATUS_MORE_ENTRIES)) {
+		/* backoff again */
+		te->next_event = t+1;
+		return;
+	}
+
+	if (!NT_STATUS_IS_OK(status)) {
+		DEBUG(1,("messaging: Lost message from %u to %u of type %u after backoff - %s\n", 
+			 rec->header.from, rec->header.to, rec->header.msg_type, nt_errstr(status)));
+		talloc_free(rec);
+		return;
+	}
+
+	fde.private 	= rec;
+	fde.fd		= socket_get_fd(rec->sock);
+	fde.flags	= EVENT_FD_WRITE;
+	fde.handler	= messaging_send_handler;
+
+	rec->fde	= event_add_fd(msg->event.ev, &fde);
+
+	talloc_set_destructor(rec, rec_destructor);
+
+	messaging_send_handler(msg->event.ev, rec->fde, 0, EVENT_FD_WRITE);
+}
+
+
+/*
   Send a message to a particular server
 */
 NTSTATUS messaging_send(void *msg_ctx, servid_t server, uint32_t msg_type, DATA_BLOB *data)
@@ -361,6 +396,16 @@
 	rec->path = messaging_path(rec, server);
 
 	status = socket_connect(rec->sock, NULL, 0, rec->path, 0, 0);
+	if (NT_STATUS_EQUAL(status, STATUS_MORE_ENTRIES)) {
+		/* backoff on this message - the servers listen queue is full */
+		struct timed_event te;
+		te.next_event = time(NULL)+1;
+		te.handler = messaging_backoff_handler;
+		te.private = rec;
+		event_add_timed(msg->event.ev, &te);
+		return NT_STATUS_OK;
+	}
+
 	if (!NT_STATUS_IS_OK(status)) {
 		talloc_free(rec);
 		return status;

Modified: branches/SAMBA_4_0/source/torture/local/messaging.c
===================================================================
--- branches/SAMBA_4_0/source/torture/local/messaging.c	2004-10-17 22:04:15 UTC (rev 3025)
+++ branches/SAMBA_4_0/source/torture/local/messaging.c	2004-10-17 22:28:26 UTC (rev 3026)
@@ -28,9 +28,10 @@
 			 uint32_t msg_type, servid_t src, DATA_BLOB *data)
 {
 	NTSTATUS status;
-	do {
-		status = messaging_send(msg_ctx, src, MY_PONG, data);
-	} while (NT_STATUS_EQUAL(status, STATUS_MORE_ENTRIES));
+	status = messaging_send(msg_ctx, src, MY_PONG, data);
+	if (!NT_STATUS_IS_OK(status)) {
+		printf("pong failed - %s\n", nt_errstr(status));
+	}
 }
 
 static void pong_message(void *msg_ctx, void *private, 
@@ -85,15 +86,17 @@
 		status1 = messaging_send(msg_ctx, 1, MY_PING, &data);
 		status2 = messaging_send(msg_ctx, 1, MY_PING, NULL);
 
-		if (NT_STATUS_IS_OK(status1)) {
-			ping_count++;
+		if (!NT_STATUS_IS_OK(status1)) {
+			printf("msg1 failed - %s\n", nt_errstr(status1));
 		}
 
-		if (NT_STATUS_IS_OK(status2)) {
-			ping_count++;
+		if (!NT_STATUS_IS_OK(status2)) {
+			printf("msg2 failed - %s\n", nt_errstr(status2));
 		}
 
-		while (ping_count > pong_count + 10) {
+		ping_count += 2;
+
+		while (ping_count > pong_count + 20) {
 			event_loop_once(ev);
 		}
 	}