From 4414ab120c07accb749ade48e0bc4289f94e2bea Mon Sep 17 00:00:00 2001
From: ed <s@ocv.me>
Date: Sat, 6 Jul 2019 23:59:45 +0000
Subject: [PATCH] support connection: close

---
 README.md            |  1 +
 copyparty/httpcli.py | 15 ++++++++++-----
 2 files changed, 11 insertions(+), 5 deletions(-)

diff --git a/README.md b/README.md
index 97e3be8e..5edfe53b 100644
--- a/README.md
+++ b/README.md
@@ -86,5 +86,6 @@ roughly sorted by priority
 * support pillow-simd
 * cache sha512 chunks on client
 * symlink existing files on upload
+* comment field
 * figure out the deal with pixel3a not being connectable as hotspot
   * pixel3a having unpredictable 3sec latency in general :||||
diff --git a/copyparty/httpcli.py b/copyparty/httpcli.py
index e47193ec..95672a8c 100644
--- a/copyparty/httpcli.py
+++ b/copyparty/httpcli.py
@@ -41,6 +41,7 @@ class HttpCli(object):
         self.log_func(self.log_src, msg)
 
     def run(self):
+        """returns true if connection can be reused"""
         try:
             headerlines = read_header(self.sr)
             if not headerlines:
@@ -60,6 +61,9 @@ class HttpCli(object):
             k, v = header_line.split(":", 1)
             self.headers[k.lower()] = v.strip()
 
+        v = self.headers.get("connection", "").lower()
+        self.keepalive = not v.startswith("close")
+
         self.uname = "*"
         if "cookie" in self.headers:
             cookies = self.headers["cookie"].split(";")
@@ -102,9 +106,9 @@ class HttpCli(object):
 
         try:
             if self.mode in ["GET", "HEAD"]:
-                return self.handle_get()
+                return self.handle_get() and self.keepalive
             elif self.mode == "POST":
-                return self.handle_post()
+                return self.handle_post() and self.keepalive
             else:
                 raise Pebkac(400, 'invalid HTTP mode "{0}"'.format(self.mode))
 
@@ -120,9 +124,9 @@ class HttpCli(object):
         # TODO something to reply with user-supplied values safely
         response = [
             "HTTP/1.1 {} {}".format(status, HTTPCODE[status]),
-            "Connection: Keep-Alive",
             "Content-Type: " + mime,
             "Content-Length: " + str(len(body)),
+            "Connection: " + ("Keep-Alive" if self.keepalive else "Close"),
         ]
         for k, v in self.out_headers.items():
             response.append("{}: {}".format(k, v))
@@ -266,7 +270,7 @@ class HttpCli(object):
         response = json.dumps(response)
 
         self.log(response)
-        self.reply(response.encode("utf-8"), headers=["Content-Type: application/json"])
+        self.reply(response.encode("utf-8"), mime="application/json")
         return True
 
     def handle_post_binary(self):
@@ -553,12 +557,13 @@ class HttpCli(object):
 
         headers = [
             "HTTP/1.1 {} {}".format(status, HTTPCODE[status]),
-            "Connection: Keep-Alive",
             "Content-Type: " + mime,
             "Content-Length: " + str(upper - lower),
             "Accept-Ranges: bytes",
             "Last-Modified: " + file_lastmod,
+            "Connection: " + ("Keep-Alive" if self.keepalive else "Close"),
         ]
+
         headers.extend(extra_headers)
         headers = "\r\n".join(headers).encode("utf-8") + b"\r\n\r\n"
         self.s.sendall(headers)