upgrade: fixed connection states with same value

2 files changed, 6 insertions, 1 deletions

diff --git a/src/microhttpd/connection.c b/src/microhttpd/connection.c
index 3a121b9a..da762614 100644
--- a/src/microhttpd/connection.c
+++ b/src/microhttpd/connection.c
@@ -2354,6 +2354,8 @@ MHD_connection_handle_read (struct MHD_Connection *connection)
         case MHD_CONNECTION_UPGRADE:
           EXTRA_CHECK (0);
           break;
+        case MHD_CONNECTION_UPGRADE_CLOSED:
+          break;
         default:
           /* shrink read buffer to how much is actually used */
           MHD_pool_reallocate (connection->pool,
@@ -2538,6 +2540,8 @@ MHD_connection_handle_write (struct MHD_Connection *connection)
         case MHD_CONNECTION_UPGRADE:
           EXTRA_CHECK (0);
           break;
+        case MHD_CONNECTION_UPGRADE_CLOSED:
+          break;
         default:
           EXTRA_CHECK (0);
           CONNECTION_CLOSE_ERROR (connection,
@@ -3057,6 +3061,7 @@ MHD_connection_handle_idle (struct MHD_Connection *connection)
           cleanup_connection (connection);
           return MHD_NO;
         case MHD_CONNECTION_UPGRADE:
+        case MHD_CONNECTION_UPGRADE_CLOSED:
           return MHD_YES; /* keep open */
         default:
           EXTRA_CHECK (0);
diff --git a/src/microhttpd/internal.h b/src/microhttpd/internal.h
index 89516d86..4a3fe58d 100644
--- a/src/microhttpd/internal.h
+++ b/src/microhttpd/internal.h
@@ -508,7 +508,7 @@ enum MHD_CONNECTION_STATE
    * by the application.  We now need to do our own
    * internal cleanup.
    */
-  MHD_CONNECTION_UPGRADE_CLOSED = MHD_TLS_CONNECTION_INIT + 1
+  MHD_CONNECTION_UPGRADE_CLOSED = MHD_CONNECTION_UPGRADE + 1
 
 };