// Copyright 2024 The Chromium Authors // Use of this source code is governed by a BSD-style license that can be // found in the LICENSE file. #include "net/test/embedded_test_server/websocket_echo_handler.h" #include #include "base/memory/scoped_refptr.h" #include "net/test/embedded_test_server/websocket_connection.h" #include "net/test/embedded_test_server/websocket_handler.h" namespace net::test_server { WebSocketEchoHandler::WebSocketEchoHandler( scoped_refptr connection) : WebSocketHandler(std::move(connection)) {} void WebSocketEchoHandler::OnHandshake(const HttpRequest& request) { CHECK(connection()); connection()->SetResponseHeader("X-Custom-Header", "WebSocketEcho"); } void WebSocketEchoHandler::OnTextMessage(std::string_view message) { CHECK(connection()); connection()->SendTextMessage(message); } void WebSocketEchoHandler::OnBinaryMessage(base::span message) { CHECK(connection()); connection()->SendBinaryMessage(message); } } // namespace net::test_server