From d6dbb6c979bc785e501cf109a7d155635b1647ea Mon Sep 17 00:00:00 2001
From: tiptorrent development team <tiptorrent@soleta.eu>
Date: Fri, 10 Sep 2021 17:06:35 +0200
Subject: add a close wait state

Wait for clients to close the connection, then:

- create redirection.
- activate pending clients.

Do no refresh timeout while in close wait state.
---
 src/core.c | 31 +++++++++++++++++++++++--------
 1 file changed, 23 insertions(+), 8 deletions(-)

(limited to 'src/core.c')

diff --git a/src/core.c b/src/core.c
index 6c7c542..b018207 100644
--- a/src/core.c
+++ b/src/core.c
@@ -107,6 +107,8 @@ static int tip_client_recv(struct tip_client *cli, int events)
 	return ret;
 }
 
+static int tip_client_redirect_create(const struct tip_client *cli);
+
 static void tip_client_read_cb(struct ev_loop *loop, struct ev_io *io, int events)
 {
 	struct tip_client *cli;
@@ -115,10 +117,11 @@ static void tip_client_read_cb(struct ev_loop *loop, struct ev_io *io, int event
 	cli = container_of(io, struct tip_client, io);
 
 	ret = tip_client_recv(cli, events);
-	if (ret <= 0)
+	if (ret < 0)
 		goto close;
 
-	ev_timer_again(loop, &cli->timer);
+	if (cli->state != TIP_CLIENT_CLOSE_WAIT)
+		ev_timer_again(loop, &cli->timer);
 
 	cli->buf_len += ret;
 	if (cli->buf_len >= sizeof(cli->buf)) {
@@ -167,12 +170,24 @@ static void tip_client_read_cb(struct ev_loop *loop, struct ev_io *io, int event
 		ev_io_set(&cli->io, tip_client_socket(cli), EV_READ | EV_WRITE);
 		ev_io_start(loop, &cli->io);
 		break;
+	case TIP_CLIENT_CLOSE_WAIT:
+		if (ret == 0) {
+			syslog(LOG_INFO, "client %s:%hu has closed the connection with us\n",
+			       inet_ntoa(cli->addr.sin_addr), ntohs(cli->addr.sin_port));
+			goto shutdown;
+		}
+		syslog(LOG_ERR, "unexpected data from client %s:%hu while waiting to close\n",
+		       inet_ntoa(cli->addr.sin_addr), ntohs(cli->addr.sin_port));
+		goto close;
 	default:
 		syslog(LOG_ERR, "unknown read state, critical internal error for %s:%hu\n",
 		       inet_ntoa(cli->addr.sin_addr), ntohs(cli->addr.sin_port));
 		goto close;
 	}
 	return;
+shutdown:
+	if (cli->size > FILE_SIZE_THRESHOLD)
+		tip_client_redirect_create(cli);
 close:
 	ev_timer_stop(loop, &cli->timer);
 	tip_client_release(loop, cli);
@@ -259,9 +274,12 @@ static void tip_client_write_cb(struct ev_loop *loop, struct ev_io *io, int even
 		break;
 	case TIP_CLIENT_PROCESSING_REQUEST_3:
 		ret = tip_client_state_process_payload_bulk(cli);
-		if (ret > 0)
-			goto shutdown;
-		else if (ret < 0) {
+		if (ret > 0) {
+			/* entering TIP_CLIENT_CLOSE_WAIT state. */
+			ev_io_stop(loop, &cli->io);
+			ev_io_set(&cli->io, tip_client_socket(cli), EV_READ);
+			ev_io_start(loop, &cli->io);
+		} else if (ret < 0) {
 			syslog(LOG_ERR, "Failed to process HTTP request from %s:%hu\n",
 			       inet_ntoa(cli->addr.sin_addr),
 			       ntohs(cli->addr.sin_port));
@@ -274,9 +292,6 @@ static void tip_client_write_cb(struct ev_loop *loop, struct ev_io *io, int even
 		goto close;
 	}
 	return;
-shutdown:
-	if (cli->size > FILE_SIZE_THRESHOLD)
-		tip_client_redirect_create(cli);
 close:
 	ev_timer_stop(loop, &cli->timer);
 	tip_client_release(loop, cli);
-- 
cgit v1.2.3-18-g5258