[GNUnet-SVN] [libmicrohttpd] branch master updated: Prevent repetitive analysing of headers to detect Keep-Alive/Close of connection.

[gnunet]

This is an automated email from the git hooks/post-receive script.

karlson2k pushed a commit to branch master
in repository libmicrohttpd.

The following commit(s) were added to refs/heads/master by this push:
     new b99c2902 Prevent repetitive analysing of headers to detect 
Keep-Alive/Close of connection.
b99c2902 is described below

commit b99c290230f3c8fd8adacca0e2d482b42b1d3c50
Author: Evgeny Grin (Karlson2k) <address@hidden>
AuthorDate: Fri May 5 21:05:49 2017 +0300

    Prevent repetitive analysing of headers to detect Keep-Alive/Close of 
connection.
---
 ChangeLog                   |  5 +++++
 src/microhttpd/connection.c | 47 ++++++++++++++++++++-------------------------
 src/microhttpd/internal.h   | 28 +++++++++++++++++++++++++++
 3 files changed, 54 insertions(+), 26 deletions(-)

diff --git a/ChangeLog b/ChangeLog
index 655a703b..7de19bd1 100644
--- a/ChangeLog
+++ b/ChangeLog
@@ -1,3 +1,8 @@
+Tue May  5 20:57:00 MSK 2017
+       Fixed null dereference when connection has "Upgrade" request and 
+       connection is not upgraded.  -JB/EG
+       Better handle Keep-Alive/Close. -EG
+
 Tue May  2 18:37:53 CEST 2017
        Update manual. -CG
        Add MHD_CONNECTION_INFO_REQUEST_HEADER_SIZE.
diff --git a/src/microhttpd/connection.c b/src/microhttpd/connection.c
index d4f25242..e4873073 100644
--- a/src/microhttpd/connection.c
+++ b/src/microhttpd/connection.c
@@ -863,6 +863,8 @@ keepalive_possible (struct MHD_Connection *connection)
 {
   const char *end;
 
+  if (MHD_CONN_MUST_CLOSE == connection->keepalive)
+    return MHD_NO;
   if (NULL == connection->version)
     return MHD_NO;
   if ( (NULL != connection->response) &&
@@ -1103,6 +1105,9 @@ build_header_response (struct MHD_Connection *connection)
                                        "close")) )
         client_requested_close = NULL;
 
+      if (0 != (connection->response->flags & MHD_RF_HTTP_VERSION_1_0_ONLY))
+        connection->keepalive = MHD_CONN_MUST_CLOSE;
+
       /* now analyze chunked encoding situation */
       connection->have_chunked_upload = false;
 
@@ -1148,7 +1153,8 @@ build_header_response (struct MHD_Connection *connection)
 
       /* check for other reasons to add 'close' header */
       if ( ( (NULL != client_requested_close) ||
-             (connection->read_closed) ) &&
+             (connection->read_closed) ||
+             (MHD_CONN_MUST_CLOSE == connection->keepalive)) &&
            (NULL == response_has_close) &&
            (0 == (connection->response->flags & MHD_RF_HTTP_VERSION_1_0_ONLY) 
) )
         must_add_close = MHD_YES;
@@ -1196,7 +1202,7 @@ build_header_response (struct MHD_Connection *connection)
       if ( (NULL == response_has_keepalive) &&
            (NULL == response_has_close) &&
            (MHD_NO == must_add_close) &&
-           (0 == (connection->response->flags & MHD_RF_HTTP_VERSION_1_0_ONLY) 
) &&
+           (MHD_CONN_MUST_CLOSE != connection->keepalive) &&
            (MHD_YES == keepalive_possible (connection)) )
         must_add_keep_alive = MHD_YES;
       break;
@@ -1207,6 +1213,14 @@ build_header_response (struct MHD_Connection *connection)
       EXTRA_CHECK (0);
     }
 
+  if (MHD_CONN_MUST_CLOSE != connection->keepalive)
+    {
+      if ( (must_add_close) || (NULL != response_has_close) )
+        connection->keepalive = MHD_CONN_MUST_CLOSE;
+      else if ( (must_add_keep_alive) || (NULL != response_has_keepalive) )
+        connection->keepalive = MHD_CONN_USE_KEEPALIVE;
+    }
+
   if (must_add_close)
     size += MHD_STATICSTR_LEN_ ("Connection: close\r\n");
   if (must_add_keep_alive)
@@ -1353,6 +1367,8 @@ transmit_error_response (struct MHD_Connection 
*connection,
                       response);
   EXTRA_CHECK (NULL != connection->response);
   MHD_destroy_response (response);
+  /* Do not reuse this connection. */
+  connection->keepalive = MHD_CONN_MUST_CLOSE;
   if (MHD_NO == build_header_response (connection))
     {
       /* oops - close! */
@@ -2799,10 +2815,8 @@ int
 MHD_connection_handle_idle (struct MHD_Connection *connection)
 {
   struct MHD_Daemon *daemon = connection->daemon;
-  const char *end;
   char *line;
   size_t line_len;
-  int client_close;
   int ret;
 
   connection->in_idle = true;
@@ -3184,12 +3198,6 @@ MHD_connection_handle_idle (struct MHD_Connection 
*connection)
           else
             socket_start_normal_buffering (connection);
 
-          end =
-            MHD_get_response_header (connection->response,
-                                    MHD_HTTP_HEADER_CONNECTION);
-          client_close = ( (NULL != end) &&
-                           (MHD_str_equal_caseless_(end,
-                                                    "close")));
           MHD_destroy_response (connection->response);
           connection->response = NULL;
           if ( (NULL != daemon->notify_completed) &&
@@ -3201,22 +3209,8 @@ MHD_connection_handle_idle (struct MHD_Connection 
*connection)
                                      &connection->client_context,
                                      MHD_REQUEST_TERMINATED_COMPLETED_OK);
           }
-          end =
-            MHD_lookup_connection_value (connection,
-                                        MHD_HEADER_KIND,
-                                         MHD_HTTP_HEADER_CONNECTION);
-          if ( (connection->read_closed) ||
-               (client_close) ||
-               ( (NULL != end) &&
-                (MHD_str_equal_caseless_ (end,
-                                           "close")) ) )
-            {
-              connection->read_closed = true;
-              connection->read_buffer_offset = 0;
-            }
-          if ( ( (connection->read_closed) &&
-                 (0 == connection->read_buffer_offset) ) ||
-               (MHD_NO == keepalive_possible (connection) ) )
+          if ( (MHD_CONN_USE_KEEPALIVE != connection->keepalive) ||
+               (connection->read_closed) )
             {
               /* have to close for some reason */
               MHD_connection_close_ (connection,
@@ -3237,6 +3231,7 @@ MHD_connection_handle_idle (struct MHD_Connection 
*connection)
               connection->last = NULL;
               connection->colon = NULL;
               connection->header_size = 0;
+              connection->keepalive = MHD_CONN_KEEPALIVE_UNKOWN;
               /* Reset the read buffer to the starting size,
                  preserving the bytes we have already read. */
               connection->read_buffer
diff --git a/src/microhttpd/internal.h b/src/microhttpd/internal.h
index 3595f105..cb6753c9 100644
--- a/src/microhttpd/internal.h
+++ b/src/microhttpd/internal.h
@@ -570,6 +570,27 @@ typedef ssize_t
 
 
 /**
+ * Ability to use same connection for next request
+ */
+enum MHD_ConnKeepAlive
+{
+  /**
+   * Connection must be closed after sending response.
+   */
+  MHD_CONN_MUST_CLOSE = -1,
+
+  /**
+   * KeelAlive state is not yet determined
+   */
+  MHD_CONN_KEEPALIVE_UNKOWN = 0,
+
+  /**
+   * Connection can be used for serving next request
+   */
+  MHD_CONN_USE_KEEPALIVE = 1
+};
+
+/**
  * State kept for each HTTP request.
  */
 struct MHD_Connection
@@ -676,6 +697,13 @@ struct MHD_Connection
   char *version;
 
   /**
+   * Close connection after sending response?
+   * Functions may change value from "Unknown" or "KeepAlive" to "Must close",
+   * but no functions reset value "Must Close" to any other value.
+   */
+  enum MHD_ConnKeepAlive keepalive;
+
+  /**
    * Buffer for reading requests.  Allocated in pool.  Actually one
    * byte larger than @e read_buffer_size (if non-NULL) to allow for
    * 0-termination.

-- 
To stop receiving notification emails like this one, please contact
address@hidden