| // Copyright 2012 The Chromium Authors |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #ifndef NET_SERVER_WEB_SOCKET_H_ |
| #define NET_SERVER_WEB_SOCKET_H_ |
| |
| #include <memory> |
| #include <string> |
| #include <string_view> |
| |
| #include "base/memory/raw_ptr.h" |
| #include "net/traffic_annotation/network_traffic_annotation.h" |
| #include "net/websockets/websocket_frame.h" |
| |
| namespace net { |
| |
| class HttpConnection; |
| class HttpServer; |
| class HttpServerRequestInfo; |
| class WebSocketEncoder; |
| |
| class WebSocket final { |
| public: |
| enum ParseResult { |
| // Final frame of a text message or compressed frame. |
| FRAME_OK_FINAL, |
| // Other frame of a text message. |
| FRAME_OK_MIDDLE, |
| FRAME_PING, |
| FRAME_PONG, |
| FRAME_INCOMPLETE, |
| FRAME_CLOSE, |
| FRAME_ERROR |
| }; |
| |
| WebSocket(HttpServer* server, HttpConnection* connection); |
| |
| void Accept(const HttpServerRequestInfo& request, |
| const NetworkTrafficAnnotationTag traffic_annotation); |
| ParseResult Read(std::string* message); |
| void Send(std::string_view message, |
| WebSocketFrameHeader::OpCodeEnum op_code, |
| const NetworkTrafficAnnotationTag traffic_annotation); |
| |
| WebSocket(const WebSocket&) = delete; |
| WebSocket& operator=(const WebSocket&) = delete; |
| |
| ~WebSocket(); |
| |
| private: |
| void Fail(); |
| void SendErrorResponse(const std::string& message, |
| const NetworkTrafficAnnotationTag traffic_annotation); |
| |
| // This dangling raw_ptr occurred in: |
| // browser_tests: PortForwardingDisconnectTest.DisconnectOnRelease |
| // https://ci.chromium.org/ui/p/chromium/builders/try/win-rel/170974/test-results?q=ExactID%3Aninja%3A%2F%2Fchrome%2Ftest%3Abrowser_tests%2FPortForwardingDisconnectTest.DisconnectOnRelease+VHash%3Abdbee181b3e0309b |
| const raw_ptr<HttpServer, |
| AcrossTasksDanglingUntriaged | FlakyDanglingUntriaged> |
| server_; |
| const raw_ptr<HttpConnection> connection_; |
| std::unique_ptr<WebSocketEncoder> encoder_; |
| bool closed_ = false; |
| std::unique_ptr<NetworkTrafficAnnotationTag> traffic_annotation_ = nullptr; |
| }; |
| |
| } // namespace net |
| |
| #endif // NET_SERVER_WEB_SOCKET_H_ |