X-Git-Url: https://git.ozlabs.org/?a=blobdiff_plain;f=ccan%2Foserver%2Foserver.c;fp=ccan%2Foserver%2Foserver.c;h=b8959035e1cadc8056f879892f541a74ea7c55d3;hb=a88d58b2667c6fbd20c7fea563eed0d92654ea39;hp=076ba70b4c15930eeb7cfde0ffb6e2eed1a38a57;hpb=e78c302eef5bf006cd0994c03a98b1e1c078b67b;p=ccan-lca-2011.git

diff --git a/ccan/oserver/oserver.c b/ccan/oserver/oserver.c
index 076ba70..b895903 100644
--- a/ccan/oserver/oserver.c
+++ b/ccan/oserver/oserver.c
@@ -17,13 +17,17 @@
 #include <signal.h>
 
 enum state {
+	SENDING_GREETING,
 	RECEIVING_USER_QUESTION,
+	SENDING_ANSWER_PREFIX,
 	SENDING_ANSWER,
 	FINISHED
 };
 
 static uint16_t state_flag_map[] = {
+	[SENDING_GREETING]		= TEVENT_FD_WRITE,
 	[RECEIVING_USER_QUESTION]	= TEVENT_FD_READ,
+	[SENDING_ANSWER_PREFIX]		= TEVENT_FD_WRITE,
 	[SENDING_ANSWER]		= TEVENT_FD_WRITE,
 	[FINISHED]			= 0
 };
@@ -105,6 +109,10 @@ static void service_client(struct tevent_context *ev,
 	ssize_t len;
 
 	switch (c->state) {
+	case SENDING_GREETING:
+		if (!send_string(c, "Welcome.  Please ask your question.\n"))
+			goto fail;
+		break;
 	case RECEIVING_USER_QUESTION:
 		len = read_string(c->fd, &c->question);
 		if (len <= 0)
@@ -114,9 +122,13 @@ static void service_client(struct tevent_context *ev,
 
 			for (i = 0; c->question[i]; i++)
 				c->question[i] = toupper(c->question[i]);
-			set_state(c, SENDING_ANSWER);
+			set_state(c, SENDING_ANSWER_PREFIX);
 		}
 		break;
+	case SENDING_ANSWER_PREFIX:
+		if (!send_string(c, "Our answer is:\n"))
+			goto fail;
+		break;
 	case SENDING_ANSWER:
 		if (!send_string(c, c->question))
 			goto fail;
@@ -159,7 +171,7 @@ static void add_client(struct tevent_context *ev,
 	if (client->fd < 0)
 		err(1, "Accepting client connection");
 
-	client->state = RECEIVING_USER_QUESTION;
+	client->state = SENDING_GREETING;
 	client->bytes_sent = 0;
 	client->question = talloc_strdup(client, "");
 	client->oserver = oserver;