From 4fc24014987028ff2a6ad8b58adf1b17809be419 Mon Sep 17 00:00:00 2001
From: Curtis McEnroe <june@causal.agency>
Date: Thu, 24 Oct 2019 01:35:59 -0400
Subject: Only set NeedCapEnd if unregistered

---
 client.c | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

(limited to 'client.c')

diff --git a/client.c b/client.c
index 9e0d2bc..6bbfdb5 100644
--- a/client.c
+++ b/client.c
@@ -139,11 +139,11 @@ static void handleCap(struct Client *client, struct Message msg) {
 		}
 
 	} else if (!strcmp(msg.params[0], "LS")) {
-		client->need |= NeedCapEnd;
+		if (client->need) client->need |= NeedCapEnd;
 		clientFormat(client, ":%s CAP * LS :server-time\r\n", Origin);
 
 	} else if (!strcmp(msg.params[0], "REQ") && msg.params[1]) {
-		client->need |= NeedCapEnd;
+		if (client->need) client->need |= NeedCapEnd;
 		if (!strcmp(msg.params[1], "server-time")) {
 			client->serverTime = true;
 			clientFormat(client, ":%s CAP * ACK :server-time\r\n", Origin);
-- 
cgit 1.4.1