Merge pull request #145 from neanton/feature-sticky-sessions

Cookie-based sticky sessions support for handshake and upgrade transport

Author: Taluu

src/Engine/AbstractSocketIO.php

@@ -34,6 +34,9 @@ abstract class AbstractSocketIO implements EngineInterface
     /** @var string[] Parse url result */
     protected $url;
 
+    /** @var array cookies received during handshake */
+    protected $cookies = [];
+
     /** @var string[] Session information */
     protected $session;
 

src/Engine/SocketIO/Version0X.php

@@ -87,6 +87,7 @@ public function close()
         fclose($this->stream);
         $this->stream = null;
         $this->session = null;
+        $this->cookies = [];
     }
 
     /** {@inheritDoc} */
@@ -176,6 +177,14 @@ protected function handshake()
             throw new UnsupportedTransportException('websocket');
         }
 
+        $cookies = [];
+        foreach ($http_response_header as $header) {
+            if (preg_match('/^Set-Cookie:\s*([^;]*)/i', $header, $matches)) {
+                $cookies[] = $matches[1];
+            }
+        }
+        $this->cookies = $cookies;
+
         $this->session = new Session($decoded['sid'], $decoded['pingInterval'], $decoded['pingTimeout'], $decoded['upgrades']);
     }
 
@@ -211,7 +220,13 @@ private function upgradeTransport()
                  . "Connection: Upgrade\r\n"
                  . "Sec-WebSocket-Key: {$key}\r\n"
                  . "Sec-WebSocket-Version: 13\r\n"
-                 . "Origin: {$origin}\r\n\r\n";
+                 . "Origin: {$origin}\r\n";
+
+        if (!empty($this->cookies)) {
+            $request .= "Cookie: " . implode('; ', $this->cookies) . "\r\n";
+        }
+
+        $request .= "\r\n";
 
         fwrite($this->stream, $request);
         $result = fread($this->stream, 12);

src/Engine/SocketIO/Version1X.php

@@ -77,6 +77,7 @@ public function close()
         fclose($this->stream);
         $this->stream = null;
         $this->session = null;
+        $this->cookies = [];
     }
 
     /** {@inheritDoc} */
@@ -172,6 +173,14 @@ protected function handshake()
             throw new UnsupportedTransportException('websocket');
         }
 
+        $cookies = [];
+        foreach ($http_response_header as $header) {
+            if (preg_match('/^Set-Cookie:\s*([^;]*)/i', $header, $matches)) {
+                $cookies[] = $matches[1];
+            }
+        }
+        $this->cookies = $cookies;
+
         $this->session = new Session($decoded['sid'], $decoded['pingInterval'], $decoded['pingTimeout'], $decoded['upgrades']);
     }
 
@@ -204,7 +213,13 @@ private function upgradeTransport()
                  . "Connection: Upgrade\r\n"
                  . "Sec-WebSocket-Key: {$key}\r\n"
                  . "Sec-WebSocket-Version: 13\r\n"
-                 . "Origin: {$origin}\r\n\r\n";
+                 . "Origin: {$origin}\r\n";
+
+        if (!empty($this->cookies)) {
+            $request .= "Cookie: " . implode('; ', $this->cookies) . "\r\n";
+        }
+
+        $request .= "\r\n";
 
         fwrite($this->stream, $request);
         $result = fread($this->stream, 12);