mirror of
https://github.com/klzgrad/naiveproxy.git
synced 2024-12-01 01:36:09 +03:00
56 lines
1.3 KiB
C++
56 lines
1.3 KiB
C++
// Copyright (c) 2012 The Chromium Authors. All rights reserved.
|
|
// 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 "base/macros.h"
|
|
#include "base/strings/string_piece.h"
|
|
#include "net/traffic_annotation/network_traffic_annotation.h"
|
|
|
|
namespace net {
|
|
|
|
class HttpConnection;
|
|
class HttpServer;
|
|
class HttpServerRequestInfo;
|
|
class WebSocketEncoder;
|
|
|
|
class WebSocket final {
|
|
public:
|
|
enum ParseResult {
|
|
FRAME_OK,
|
|
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(const std::string& message,
|
|
const NetworkTrafficAnnotationTag traffic_annotation);
|
|
~WebSocket();
|
|
|
|
private:
|
|
void Fail();
|
|
void SendErrorResponse(const std::string& message,
|
|
const NetworkTrafficAnnotationTag traffic_annotation);
|
|
|
|
HttpServer* const server_;
|
|
HttpConnection* const connection_;
|
|
std::unique_ptr<WebSocketEncoder> encoder_;
|
|
bool closed_;
|
|
|
|
DISALLOW_COPY_AND_ASSIGN(WebSocket);
|
|
};
|
|
|
|
} // namespace net
|
|
|
|
#endif // NET_SERVER_WEB_SOCKET_H_
|