b0fd119d14
* Quick hack to host HTTP and WS on the same port #373 * Quick hack to host HTTP and WS on the same port #373 - simplify code * ran clang-format Co-authored-by: En Shih <seanstone5923@gmail.com> Co-authored-by: The Artful Bodger <TheArtfulBodger@users.noreply.github.com>
58 lines
2.0 KiB
C++
58 lines
2.0 KiB
C++
/*
|
|
* IXWebSocketHandshake.h
|
|
* Author: Benjamin Sergeant
|
|
* Copyright (c) 2019 Machine Zone, Inc. All rights reserved.
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include "IXCancellationRequest.h"
|
|
#include "IXHttp.h"
|
|
#include "IXSocket.h"
|
|
#include "IXWebSocketHttpHeaders.h"
|
|
#include "IXWebSocketInitResult.h"
|
|
#include "IXWebSocketPerMessageDeflate.h"
|
|
#include "IXWebSocketPerMessageDeflateOptions.h"
|
|
#include <atomic>
|
|
#include <chrono>
|
|
#include <memory>
|
|
#include <string>
|
|
|
|
namespace ix
|
|
{
|
|
class WebSocketHandshake
|
|
{
|
|
public:
|
|
WebSocketHandshake(std::atomic<bool>& requestInitCancellation,
|
|
std::unique_ptr<Socket>& _socket,
|
|
WebSocketPerMessageDeflatePtr& perMessageDeflate,
|
|
WebSocketPerMessageDeflateOptions& perMessageDeflateOptions,
|
|
std::atomic<bool>& enablePerMessageDeflate);
|
|
|
|
WebSocketInitResult clientHandshake(const std::string& url,
|
|
const WebSocketHttpHeaders& extraHeaders,
|
|
const std::string& host,
|
|
const std::string& path,
|
|
int port,
|
|
int timeoutSecs);
|
|
|
|
WebSocketInitResult serverHandshake(int timeoutSecs,
|
|
bool enablePerMessageDeflate,
|
|
HttpRequestPtr request = nullptr);
|
|
|
|
private:
|
|
std::string genRandomString(const int len);
|
|
|
|
// Parse HTTP headers
|
|
WebSocketInitResult sendErrorResponse(int code, const std::string& reason);
|
|
|
|
bool insensitiveStringCompare(const std::string& a, const std::string& b);
|
|
|
|
std::atomic<bool>& _requestInitCancellation;
|
|
std::unique_ptr<Socket>& _socket;
|
|
WebSocketPerMessageDeflatePtr& _perMessageDeflate;
|
|
WebSocketPerMessageDeflateOptions& _perMessageDeflateOptions;
|
|
std::atomic<bool>& _enablePerMessageDeflate;
|
|
};
|
|
} // namespace ix
|