Fixed memory leak when websocket hande shake failed. (#2768)

* 修复websocket握手失败不会释放WsContext中set的值而导致内存泄漏的问题

* Update Server.php

* Update CHANGELOG-2.0.md

Co-authored-by: 李铭昕 <715557344@qq.com>

Author: fushiliang321

src/Server.php

@@ -165,37 +165,41 @@ public function onHandShake(SwooleRequest $request, SwooleResponse $response): v
 
             $class = $psr7Response->getAttribute('class');
 
-            if (! empty($class)) {
-                FdCollector::set($fd, $class);
-                $server = $this->getServer();
-                if ($server instanceof \Swoole\Coroutine\Http\Server) {
-                    $response->upgrade();
-                    $this->getSender()->setResponse($fd, $response);
-                    $this->deferOnOpen($request, $class, $response);
-
-                    [, , $callbacks] = ServerManager::get($this->serverName);
-
-                    [$onMessageCallbackClass, $onMessageCallbackMethod] = $callbacks[SwooleEvent::ON_MESSAGE];
-                    $onMessageCallbackInstance = $this->container->get($onMessageCallbackClass);
-
-                    [$onCloseCallbackClass, $onCloseCallbackMethod] = $callbacks[SwooleEvent::ON_CLOSE];
-                    $onCloseCallbackInstance = $this->container->get($onCloseCallbackClass);
-
-                    while (true) {
-                        $frame = $response->recv();
-                        if ($frame === false || $frame instanceof CloseFrame || $frame === '') {
-                            $onCloseCallbackInstance->{$onCloseCallbackMethod}($response, $fd, 0);
-                            break;
-                        }
-                        $onMessageCallbackInstance->{$onMessageCallbackMethod}($response, $frame);
+            if (empty($class)) {
+                throw new WebSocketHandeShakeException('WebSocket hande shake failed, because the class does not exists.');
+            }
+
+            FdCollector::set($fd, $class);
+            $server = $this->getServer();
+            if ($server instanceof \Swoole\Coroutine\Http\Server) {
+                $response->upgrade();
+                $this->getSender()->setResponse($fd, $response);
+                $this->deferOnOpen($request, $class, $response);
+
+                [, , $callbacks] = ServerManager::get($this->serverName);
+
+                [$onMessageCallbackClass, $onMessageCallbackMethod] = $callbacks[SwooleEvent::ON_MESSAGE];
+                $onMessageCallbackInstance = $this->container->get($onMessageCallbackClass);
+
+                [$onCloseCallbackClass, $onCloseCallbackMethod] = $callbacks[SwooleEvent::ON_CLOSE];
+                $onCloseCallbackInstance = $this->container->get($onCloseCallbackClass);
+
+                while (true) {
+                    $frame = $response->recv();
+                    if ($frame === false || $frame instanceof CloseFrame || $frame === '') {
+                        $onCloseCallbackInstance->{$onCloseCallbackMethod}($response, $fd, 0);
+                        break;
                     }
-                } else {
-                    $this->deferOnOpen($request, $class, $server);
+                    $onMessageCallbackInstance->{$onMessageCallbackMethod}($response, $frame);
                 }
+            } else {
+                $this->deferOnOpen($request, $class, $server);
             }
         } catch (Throwable $throwable) {
             // Delegate the exception to exception handler.
             $psr7Response = $this->exceptionHandlerDispatcher->dispatch($throwable, $this->exceptionHandlers);
+            FdCollector::del($request->fd);
+            WsContext::release($request->fd);
         } finally {
             isset($fd) && $this->getSender()->setResponse($fd, null);
             // Send the Response to client.