summary refs log tree commit diff
diff options
context:
space:
mode:
-rw-r--r--bounce.h1
-rw-r--r--client.c7
-rw-r--r--ring.c5
3 files changed, 12 insertions, 1 deletions
diff --git a/bounce.h b/bounce.h
index d5f4170..f8ab0c0 100644
--- a/bounce.h
+++ b/bounce.h
@@ -144,6 +144,7 @@ extern bool verbose;
 void ringAlloc(size_t len);
 void ringProduce(const char *line);
 size_t ringConsumer(const char *name);
+void ringSet(size_t consumer, size_t pos);
 size_t ringPos(size_t consumer);
 size_t ringDiff(size_t consumer);
 const char *ringPeek(struct timeval *time, size_t consumer);
diff --git a/client.c b/client.c
index a335b99..66d07d0 100644
--- a/client.c
+++ b/client.c
@@ -47,6 +47,7 @@ struct Client {
 	struct tls *tls;
 	enum Need need;
 	size_t consumer;
+	size_t setPos;
 	enum Cap caps;
 	char buf[MessageCap];
 	size_t len;
@@ -118,6 +119,7 @@ static void maybeSync(struct Client *client) {
 	if (client->need == NeedPass) passRequired(client);
 	if (!client->need) {
 		stateSync(client);
+		if (client->setPos) ringSet(client->consumer, client->setPos);
 		if (!(client->caps & CapPassive) && !active++) {
 			serverFormat("AWAY\r\n");
 		}
@@ -193,9 +195,12 @@ static void handleCap(struct Client *client, struct Message *msg) {
 
 	} else if (!strcmp(msg->params[0], "REQ") && msg->params[1]) {
 		if (client->need) client->need |= NeedCapEnd;
-		enum Cap caps = capParse(msg->params[1], NULL);
+		enum Cap caps = capParse(msg->params[1], values);
 		if (caps == (avail & caps)) {
 			client->caps |= caps;
+			if (caps & CapConsumer && values[CapConsumerBit]) {
+				client->setPos = strtoull(values[CapConsumerBit], NULL, 10);
+			}
 			clientFormat(client, ":%s CAP * ACK :%s\r\n", ORIGIN, msg->params[1]);
 		} else {
 			clientFormat(client, ":%s CAP * NAK :%s\r\n", ORIGIN, msg->params[1]);
diff --git a/ring.c b/ring.c
index 90e6f2a..254c285 100644
--- a/ring.c
+++ b/ring.c
@@ -81,6 +81,11 @@ size_t ringConsumer(const char *name) {
 	return consumers.len++;
 }
 
+void ringSet(size_t consumer, size_t pos) {
+	assert(consumer < consumers.len);
+	if (pos <= producer) consumers.ptr[consumer].pos = pos;
+}
+
 size_t ringPos(size_t consumer) {
 	assert(consumer < consumers.len);
 	return consumers.ptr[consumer].pos;