mirror of
https://github.com/hrydgard/ppsspp.git
synced 2025-02-20 06:00:58 +00:00
http: Add a simple WebSocket server class.
This commit is contained in:
parent
65d072c7b0
commit
b75b6801c8
@ -893,6 +893,8 @@ add_library(native STATIC
|
||||
ext/native/base/stringutil.h
|
||||
ext/native/base/timeutil.cpp
|
||||
ext/native/base/timeutil.h
|
||||
ext/native/data/base64.cpp
|
||||
ext/native/data/base64.h
|
||||
ext/native/data/compression.cpp
|
||||
ext/native/data/compression.h
|
||||
ext/native/file/chunk_file.cpp
|
||||
@ -985,6 +987,8 @@ add_library(native STATIC
|
||||
ext/native/net/sinks.h
|
||||
ext/native/net/url.cpp
|
||||
ext/native/net/url.h
|
||||
ext/native/net/websocket_server.cpp
|
||||
ext/native/net/websocket_server.h
|
||||
ext/native/profiler/profiler.cpp
|
||||
ext/native/profiler/profiler.h
|
||||
ext/native/thin3d/thin3d.cpp
|
||||
|
@ -13,6 +13,7 @@ LOCAL_SRC_FILES :=\
|
||||
base/timeutil.cpp \
|
||||
base/colorutil.cpp \
|
||||
base/stringutil.cpp \
|
||||
data/base64.cpp \
|
||||
data/compression.cpp \
|
||||
ext/rg_etc1/rg_etc1.cpp \
|
||||
ext/cityhash/city.cpp \
|
||||
@ -64,6 +65,7 @@ LOCAL_SRC_FILES :=\
|
||||
net/resolve.cpp \
|
||||
net/sinks.cpp \
|
||||
net/url.cpp \
|
||||
net/websocket_server.cpp \
|
||||
profiler/profiler.cpp \
|
||||
thread/executor.cpp \
|
||||
thread/threadutil.cpp \
|
||||
|
35
ext/native/data/base64.cpp
Normal file
35
ext/native/data/base64.cpp
Normal file
@ -0,0 +1,35 @@
|
||||
#include "data/base64.h"
|
||||
|
||||
// TODO: This is a simple but not very efficient implementation.
|
||||
std::string Base64Encode(const uint8_t *p, size_t sz) {
|
||||
const char digits[] = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/";
|
||||
|
||||
size_t unpaddedLength = (4 * sz + 3) / 3;
|
||||
std::string result;
|
||||
result.resize((unpaddedLength + 3) & ~3, '=');
|
||||
|
||||
for (size_t i = 0; i < unpaddedLength; ++i) {
|
||||
// This is the index into the original string.
|
||||
size_t pos = (i * 3) / 4;
|
||||
int8_t off = 2 * ((i * 3) % 4);
|
||||
|
||||
int c = p[pos];
|
||||
if (off > 2) {
|
||||
c <<= 8;
|
||||
off -= 8;
|
||||
|
||||
// Grab more bits from the next character.
|
||||
if (pos + 1 < sz) {
|
||||
c |= p[pos + 1];
|
||||
}
|
||||
}
|
||||
|
||||
// Since we take from the big end, off starts at 2 and goes down.
|
||||
int8_t shift = 2 - off;
|
||||
|
||||
// Now take the bits at off and encode the character.
|
||||
result[i] = digits[(c >> shift) & 0x3F];
|
||||
}
|
||||
|
||||
return result;
|
||||
}
|
5
ext/native/data/base64.h
Normal file
5
ext/native/data/base64.h
Normal file
@ -0,0 +1,5 @@
|
||||
#pragma once
|
||||
|
||||
#include <string>
|
||||
|
||||
std::string Base64Encode(const uint8_t *p, size_t sz);
|
@ -202,6 +202,7 @@
|
||||
<ClInclude Include="base\NativeApp.h" />
|
||||
<ClInclude Include="base\stringutil.h" />
|
||||
<ClInclude Include="base\timeutil.h" />
|
||||
<ClInclude Include="data\base64.h" />
|
||||
<ClInclude Include="data\compression.h" />
|
||||
<ClInclude Include="data\listable.h" />
|
||||
<ClInclude Include="ext\cityhash\city.h" />
|
||||
@ -238,6 +239,7 @@
|
||||
<ClInclude Include="gfx_es2\draw_text_qt.h" />
|
||||
<ClInclude Include="gfx_es2\draw_text_win.h" />
|
||||
<ClInclude Include="json\json_reader.h" />
|
||||
<ClInclude Include="net\websocket_server.h" />
|
||||
<ClInclude Include="thin3d\d3d11_loader.h" />
|
||||
<ClInclude Include="thin3d\DataFormat.h" />
|
||||
<ClInclude Include="thin3d\DataFormatGL.h" />
|
||||
@ -317,6 +319,7 @@
|
||||
</ClCompile>
|
||||
<ClCompile Include="base\stringutil.cpp" />
|
||||
<ClCompile Include="base\timeutil.cpp" />
|
||||
<ClCompile Include="data\base64.cpp" />
|
||||
<ClCompile Include="data\compression.cpp" />
|
||||
<ClCompile Include="ext\cityhash\city.cpp">
|
||||
<InlineFunctionExpansion Condition="'$(Configuration)|$(Platform)'=='Debug|Win32'">AnySuitable</InlineFunctionExpansion>
|
||||
@ -700,6 +703,7 @@
|
||||
<ClCompile Include="gfx_es2\draw_text_win.cpp" />
|
||||
<ClCompile Include="json\json_reader.cpp" />
|
||||
<ClCompile Include="math\dataconv.cpp" />
|
||||
<ClCompile Include="net\websocket_server.cpp" />
|
||||
<ClCompile Include="thin3d\d3d11_loader.cpp" />
|
||||
<ClCompile Include="thin3d\DataFormatGL.cpp" />
|
||||
<ClCompile Include="thin3d\GLQueueRunner.cpp" />
|
||||
|
@ -341,6 +341,12 @@
|
||||
<ClInclude Include="json\json_reader.h">
|
||||
<Filter>json</Filter>
|
||||
</ClInclude>
|
||||
<ClInclude Include="net\websocket_server.h">
|
||||
<Filter>net</Filter>
|
||||
</ClInclude>
|
||||
<ClInclude Include="data\base64.h">
|
||||
<Filter>data</Filter>
|
||||
</ClInclude>
|
||||
</ItemGroup>
|
||||
<ItemGroup>
|
||||
<ClCompile Include="gfx\gl_debug_log.cpp">
|
||||
@ -814,6 +820,12 @@
|
||||
<ClCompile Include="json\json_reader.cpp">
|
||||
<Filter>json</Filter>
|
||||
</ClCompile>
|
||||
<ClCompile Include="net\websocket_server.cpp">
|
||||
<Filter>net</Filter>
|
||||
</ClCompile>
|
||||
<ClCompile Include="data\base64.cpp">
|
||||
<Filter>data</Filter>
|
||||
</ClCompile>
|
||||
</ItemGroup>
|
||||
<ItemGroup>
|
||||
<Filter Include="gfx">
|
||||
|
@ -87,8 +87,10 @@ void Request::WriteHttpResponseHeader(int status, int64_t size, const char *mime
|
||||
net::OutputSink *buffer = Out();
|
||||
buffer->Printf("HTTP/1.0 %03d %s\r\n", status, statusStr);
|
||||
buffer->Push("Server: PPSSPPServer v0.1\r\n");
|
||||
buffer->Printf("Content-Type: %s\r\n", mimeType ? mimeType : DEFAULT_MIME_TYPE);
|
||||
buffer->Push("Connection: close\r\n");
|
||||
if (!mimeType || strcmp(mimeType, "websocket") != 0) {
|
||||
buffer->Printf("Content-Type: %s\r\n", mimeType ? mimeType : DEFAULT_MIME_TYPE);
|
||||
buffer->Push("Connection: close\r\n");
|
||||
}
|
||||
if (size >= 0) {
|
||||
buffer->Printf("Content-Length: %llu\r\n", size);
|
||||
}
|
||||
|
449
ext/native/net/websocket_server.cpp
Normal file
449
ext/native/net/websocket_server.cpp
Normal file
@ -0,0 +1,449 @@
|
||||
#include <algorithm>
|
||||
#include <cassert>
|
||||
#include <cctype>
|
||||
#include <cstring>
|
||||
#ifndef _WIN32
|
||||
#include <arpa/inet.h>
|
||||
#include <netinet/in.h>
|
||||
#include <sys/socket.h>
|
||||
#include <sys/types.h>
|
||||
#include <sys/select.h>
|
||||
#include <unistd.h>
|
||||
#else
|
||||
#include <io.h>
|
||||
#include <winsock2.h>
|
||||
#include <ws2tcpip.h>
|
||||
#endif
|
||||
#include "base/stringutil.h"
|
||||
#include "data/base64.h"
|
||||
#include "net/http_server.h"
|
||||
#include "net/sinks.h"
|
||||
#include "net/websocket_server.h"
|
||||
// TODO: Not a great cross dependency.
|
||||
#include "Common/Crypto/sha1.h"
|
||||
|
||||
#ifdef _WIN32
|
||||
// Function Cross-Compatibility
|
||||
#define strcasecmp _stricmp
|
||||
#endif
|
||||
|
||||
static const char *const WEBSOCKET_GUID = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";
|
||||
|
||||
namespace net {
|
||||
|
||||
enum class Opcode {
|
||||
CONTINUE = 0,
|
||||
TEXT = 1,
|
||||
BINARY = 2,
|
||||
|
||||
CLOSE = 8,
|
||||
PING = 9,
|
||||
PONG = 10,
|
||||
|
||||
PAYLOAD_MAX = 2,
|
||||
CONTROL_MIN = 8,
|
||||
CONTROL_MAX = 10,
|
||||
};
|
||||
|
||||
WebSocketServer *WebSocketServer::CreateAsUpgrade(const http::Request &request) {
|
||||
auto requireHeader = [&](const char *name, const char *expected) {
|
||||
std::string val;
|
||||
if (!request.GetHeader(name, &val)) {
|
||||
return false;
|
||||
}
|
||||
return strcasecmp(val.c_str(), expected) == 0;
|
||||
};
|
||||
auto requireHeaderContains = [&](const char *name, const char *expected) {
|
||||
std::string val;
|
||||
if (!request.GetHeader(name, &val)) {
|
||||
return false;
|
||||
}
|
||||
std::transform(val.begin(), val.end(), val.begin(), tolower);
|
||||
return strstr(val.c_str(), expected) != 0;
|
||||
};
|
||||
|
||||
if (!requireHeader("upgrade", "websocket") || !requireHeaderContains("connection", "upgrade")) {
|
||||
request.WriteHttpResponseHeader(400, -1, "text/plain");
|
||||
request.Out()->Push("Must send a websocket request.");
|
||||
return nullptr;
|
||||
}
|
||||
if (!requireHeader("sec-websocket-version", "13")) {
|
||||
request.WriteHttpResponseHeader(400, -1, "text/plain", "Sec-WebSocket-Version: 13\r\n");
|
||||
request.Out()->Push("Unsupported version.");
|
||||
return nullptr;
|
||||
}
|
||||
|
||||
std::string key;
|
||||
if (!request.GetHeader("sec-websocket-key", &key)) {
|
||||
request.WriteHttpResponseHeader(400, -1, "text/plain");
|
||||
request.Out()->Push("Cannot accept without key.");
|
||||
return nullptr;
|
||||
}
|
||||
|
||||
key += WEBSOCKET_GUID;
|
||||
unsigned char accept[20];
|
||||
sha1((unsigned char *)key.c_str(), (int)key.size(), accept);
|
||||
|
||||
std::string acceptKey = Base64Encode(accept, 20);
|
||||
std::string otherHeaders = StringFromFormat("Upgrade: websocket\r\nConnection: Upgrade\r\nSec-WebSocket-Accept: %s\r\n", acceptKey.c_str());
|
||||
|
||||
// Okay, we're good to go then.
|
||||
request.WriteHttpResponseHeader(101, -1, "websocket", otherHeaders.c_str());
|
||||
request.WritePartial();
|
||||
|
||||
return new WebSocketServer(request.fd(), request.In(), request.Out());
|
||||
}
|
||||
|
||||
void WebSocketServer::Send(const std::string &str) {
|
||||
assert(open_);
|
||||
SendHeader(true, (int)Opcode::TEXT, str.size());
|
||||
// TODO: For long strings, this will block. Possibly not ideal.
|
||||
if (!out_->Push(str.c_str(), str.size())) {
|
||||
assert(false);
|
||||
open_ = false;
|
||||
closeReason_ = WebSocketClose::ABNORMAL;
|
||||
}
|
||||
}
|
||||
|
||||
void WebSocketServer::Send(const std::vector<uint8_t> &payload) {
|
||||
assert(open_);
|
||||
SendHeader(true, (int)Opcode::BINARY, payload.size());
|
||||
// TODO: For long strings, this will block. Possibly not ideal.
|
||||
if (!out_->Push((const char *)&payload[0], payload.size())) {
|
||||
assert(false);
|
||||
open_ = false;
|
||||
closeReason_ = WebSocketClose::ABNORMAL;
|
||||
}
|
||||
}
|
||||
|
||||
void WebSocketServer::Ping(const std::vector<uint8_t> &payload) {
|
||||
assert(open_);
|
||||
assert(payload.size() <= 125);
|
||||
SendHeader(true, (int)Opcode::PING, payload.size());
|
||||
if (!out_->Push((const char *)&payload[0], payload.size())) {
|
||||
assert(false);
|
||||
open_ = false;
|
||||
closeReason_ = WebSocketClose::ABNORMAL;
|
||||
}
|
||||
}
|
||||
|
||||
void WebSocketServer::Pong(const std::vector<uint8_t> &payload) {
|
||||
assert(open_);
|
||||
assert(payload.size() <= 125);
|
||||
SendHeader(true, (int)Opcode::PONG, payload.size());
|
||||
if (!out_->Push((const char *)&payload[0], payload.size())) {
|
||||
assert(false);
|
||||
open_ = false;
|
||||
closeReason_ = WebSocketClose::ABNORMAL;
|
||||
}
|
||||
}
|
||||
|
||||
void WebSocketServer::Close(WebSocketClose reason) {
|
||||
closeReason_ = reason;
|
||||
SendHeader(true, (int)Opcode::CLOSE, 2);
|
||||
|
||||
uint16_t r = (uint16_t)reason;
|
||||
uint8_t reasonData[] = {
|
||||
(uint8_t)((r >> 8) & 0xFF),
|
||||
(uint8_t)((r >> 0) & 0xFF),
|
||||
};
|
||||
if (!out_->Push((const char *)reasonData, sizeof(reasonData))) {
|
||||
assert(false);
|
||||
open_ = false;
|
||||
closeReason_ = WebSocketClose::ABNORMAL;
|
||||
}
|
||||
|
||||
sentClose_ = true;
|
||||
}
|
||||
|
||||
bool WebSocketServer::Process(float timeout) {
|
||||
if (!open_) {
|
||||
return false;
|
||||
}
|
||||
|
||||
out_->Flush(false);
|
||||
|
||||
if (out_->Empty() && sentClose_) {
|
||||
// Okay, we've sent the close. Don't wait for anything else (whether we got a close or not.)
|
||||
open_ = false;
|
||||
return false;
|
||||
}
|
||||
|
||||
struct timeval tv;
|
||||
tv.tv_sec = floor(timeout);
|
||||
tv.tv_usec = (timeout - floor(timeout)) * 1000000.0;
|
||||
|
||||
fd_set read;
|
||||
FD_ZERO(&read);
|
||||
// In case we closed due to protocol error, don't even try to read.
|
||||
if (!sentClose_) {
|
||||
FD_SET(fd_, &read);
|
||||
}
|
||||
|
||||
fd_set write;
|
||||
FD_ZERO(&write);
|
||||
if (!out_->Empty()) {
|
||||
FD_SET(fd_, &write);
|
||||
}
|
||||
|
||||
// First argument to select is the highest socket in the set + 1.
|
||||
int rval = select((int)fd_ + 1, &read, &write, nullptr, &tv);
|
||||
if (rval < 0) {
|
||||
// Something went wrong with the select() call.
|
||||
// TODO: Could be EINTR, for now returning true...
|
||||
return true;
|
||||
}
|
||||
|
||||
if (rval == 0) {
|
||||
// Timed out.
|
||||
return true;
|
||||
}
|
||||
|
||||
if (FD_ISSET(fd_, &write)) {
|
||||
out_->Flush(false);
|
||||
}
|
||||
if (FD_ISSET(fd_, &read)) {
|
||||
if (in_->Empty() && !in_->TryFill()) {
|
||||
// Since select said it was readable, we assume this means disconnect.
|
||||
closeReason_ = WebSocketClose::ABNORMAL;
|
||||
open_ = false;
|
||||
return false;
|
||||
}
|
||||
|
||||
while (ReadFrames() && !in_->Empty())
|
||||
continue;
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
bool WebSocketServer::ReadFrames() {
|
||||
if (pendingLeft_ != 0) {
|
||||
return ReadPending();
|
||||
}
|
||||
|
||||
return ReadFrame();
|
||||
}
|
||||
|
||||
bool WebSocketServer::ReadFrame() {
|
||||
assert(pendingLeft_ == 0);
|
||||
|
||||
// TODO: For now blocking on header trickle, shouldn't be common.
|
||||
auto readExact = [&](void *p, size_t sz) {
|
||||
if (!in_->TakeExact((char *)p, sz)) {
|
||||
// TODO: Failing on too slow trickle timeout for now.
|
||||
Close(WebSocketClose::POLICY_VIOLATION);
|
||||
return false;
|
||||
}
|
||||
return true;
|
||||
};
|
||||
|
||||
// Client frames are always between 6 and 14 bytes. We start with 6.
|
||||
uint8_t header[14];
|
||||
if (!readExact(header, 6))
|
||||
return false;
|
||||
|
||||
// Don't allow reserved bits to be set, require masking.
|
||||
if ((header[0] & 0x70) != 0 || (header[1] & 0x80) == 0) {
|
||||
Close(WebSocketClose::PROTOCOL_ERROR);
|
||||
return false;
|
||||
}
|
||||
|
||||
const bool fin = (header[0] & 0x80) != 0;
|
||||
const int opcode = header[0] & 0x0F;
|
||||
uint64_t sz = header[1] & 0x7F;
|
||||
const uint8_t *mask = &header[2];
|
||||
|
||||
if (opcode >= (int)Opcode::CONTROL_MIN && (sz > 125 || !fin)) {
|
||||
// Control frames must be <= 125 bytes.
|
||||
Close(WebSocketClose::PROTOCOL_ERROR);
|
||||
return false;
|
||||
}
|
||||
|
||||
if (opcode > (int)Opcode::CONTROL_MAX || (opcode > (int)Opcode::PAYLOAD_MAX && opcode < (int)Opcode::CONTROL_MIN)) {
|
||||
// Undefined opcode.
|
||||
Close(WebSocketClose::PROTOCOL_ERROR);
|
||||
return false;
|
||||
}
|
||||
|
||||
if (!pendingFin_ && opcode == (int)Opcode::CONTINUE) {
|
||||
// Can't continue what you haven't started.
|
||||
Close(WebSocketClose::PROTOCOL_ERROR);
|
||||
return false;
|
||||
}
|
||||
if (pendingFin_ && opcode != (int)Opcode::CONTINUE && opcode < (int)Opcode::CONTROL_MIN) {
|
||||
// Can't start something else until you finish your thought.
|
||||
Close(WebSocketClose::PROTOCOL_ERROR);
|
||||
return false;
|
||||
}
|
||||
|
||||
if (sz == 126) {
|
||||
// Read the rest of the mask.
|
||||
if (!readExact((char *)&header[6], 2))
|
||||
return false;
|
||||
|
||||
mask = &header[4];
|
||||
sz = (header[2] << 8) | (header[3] << 0);
|
||||
} else if (sz == 127) {
|
||||
// We only have half the size so far - read the rest, and the mask.
|
||||
if (!readExact((char *)&header[6], 8))
|
||||
return false;
|
||||
|
||||
mask = &header[10];
|
||||
// Read from big endian.
|
||||
uint64_t high = (header[2] << 24) | (header[3] << 16) || (header[4] << 8) | (header[5] << 0);
|
||||
uint64_t low = (header[6] << 24) | (header[7] << 16) || (header[8] << 8) | (header[9] << 0);
|
||||
sz = (high << 32) | low;
|
||||
|
||||
if ((sz & 0x8000000000000000ULL) != 0) {
|
||||
Close(WebSocketClose::PROTOCOL_ERROR);
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
if (opcode >= (int)Opcode::CONTROL_MIN) {
|
||||
// It's safe to overwrite this since we can be between fragmented frames, but not inside a frame.
|
||||
memcpy(pendingMask_, mask, sizeof(pendingMask_));
|
||||
return ReadControlFrame(opcode, sz);
|
||||
}
|
||||
|
||||
// The data could be split among many TCP packets, so read it as it comes.
|
||||
if (!pendingFin_)
|
||||
pendingOpcode_ = opcode;
|
||||
pendingFin_ = !fin;
|
||||
pendingLeft_ = sz;
|
||||
memcpy(pendingMask_, mask, sizeof(pendingMask_));
|
||||
|
||||
// Payload data is actually read in ReadPending().
|
||||
return true;
|
||||
}
|
||||
|
||||
bool WebSocketServer::ReadPending() {
|
||||
size_t pos = pendingBuf_.size();
|
||||
pendingBuf_.resize(pendingBuf_.size() + pendingLeft_);
|
||||
|
||||
// Read what we can.
|
||||
size_t readBytes = in_->TakeAtMost((char *)&pendingBuf_[pos], pendingLeft_);
|
||||
for (size_t i = 0; i < readBytes; ++i) {
|
||||
pendingBuf_[pos + i] ^= pendingMask_[i & 3];
|
||||
}
|
||||
pendingLeft_ -= readBytes;
|
||||
|
||||
if (pendingLeft_ != 0) {
|
||||
// Still more to read. Careful: we might need to rotate the mask.
|
||||
// Example: if we read only 3 bytes, next read should start at fourth byte in mask.
|
||||
int offset = readBytes & 3;
|
||||
if (offset) {
|
||||
uint8_t orig[4];
|
||||
memcpy(orig, pendingMask_, sizeof(orig));
|
||||
for (size_t i = 0; i < sizeof(orig); ++i) {
|
||||
pendingMask_[i] = orig[(offset + i) & 3];
|
||||
}
|
||||
}
|
||||
|
||||
// Truncate out the unread bytes for next time.
|
||||
pendingBuf_.resize(pos + readBytes);
|
||||
return true;
|
||||
}
|
||||
|
||||
// We're done, but were we waiting for a FIN packet?
|
||||
if (pendingFin_)
|
||||
return true;
|
||||
|
||||
if (pendingOpcode_ == (int)Opcode::TEXT) {
|
||||
if (text_) {
|
||||
text_(std::string(pendingBuf_.begin(), pendingBuf_.end()));
|
||||
}
|
||||
} else if (pendingOpcode_ == (int)Opcode::BINARY) {
|
||||
if (binary_) {
|
||||
binary_(pendingBuf_);
|
||||
}
|
||||
} else {
|
||||
assert(false);
|
||||
}
|
||||
|
||||
// All done, clear it out.
|
||||
pendingBuf_.clear();
|
||||
pendingOpcode_ = -1;
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
bool WebSocketServer::ReadControlFrame(int opcode, size_t sz) {
|
||||
std::vector<uint8_t> payload;
|
||||
payload.resize(sz);
|
||||
// Just block here to read the payload.
|
||||
if (!in_->TakeExact((char *)&payload[0], sz)) {
|
||||
// TODO: Failing on too slow trickle timeout for now.
|
||||
Close(WebSocketClose::POLICY_VIOLATION);
|
||||
return false;
|
||||
}
|
||||
|
||||
for (size_t i = 0; i < sz; ++i) {
|
||||
payload[i] ^= pendingMask_[i & 3];
|
||||
}
|
||||
|
||||
if (opcode == (int)Opcode::PING) {
|
||||
Pong(payload);
|
||||
// Try to send immediately if possible, but don't block.
|
||||
out_->Flush(false);
|
||||
|
||||
if (ping_) {
|
||||
ping_(payload);
|
||||
}
|
||||
} else if (opcode == (int)Opcode::PONG) {
|
||||
if (pong_) {
|
||||
pong_(payload);
|
||||
}
|
||||
} else if (opcode == (int)Opcode::CLOSE) {
|
||||
if (payload.size() >= 2) {
|
||||
uint16_t reason = (payload[0] << 8) | payload[1];
|
||||
// Send back a close right away.
|
||||
Close(WebSocketClose(reason));
|
||||
} else {
|
||||
Close(WebSocketClose::NO_STATUS);
|
||||
}
|
||||
// Don't read anything more.
|
||||
return false;
|
||||
} else {
|
||||
assert(false);
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
void WebSocketServer::SendHeader(bool fin, int opcode, size_t sz) {
|
||||
assert((opcode & 0x0F) == opcode);
|
||||
uint8_t frameHeader = (fin ? 0x80 : 0x00) | opcode;
|
||||
out_->Push((const char *)&frameHeader, 1);
|
||||
|
||||
// We never mask from the server.
|
||||
if (sz <= 125) {
|
||||
uint8_t frameSize = (int8_t)sz;
|
||||
out_->Push((const char *)&frameSize, 1);
|
||||
} else if (sz <= 0xFFFF) {
|
||||
uint8_t frameSize[] = {
|
||||
126,
|
||||
(uint8_t)((sz >> 8) & 0xFF),
|
||||
(uint8_t)((sz >> 0) & 0xFF),
|
||||
};
|
||||
out_->Push((const char *)frameSize, sizeof(frameSize));
|
||||
} else {
|
||||
uint64_t sz64 = sz;
|
||||
assert((sz64 & 0x8000000000000000ULL) == 0);
|
||||
uint8_t frameSize[] = {
|
||||
127,
|
||||
(uint8_t)((sz64 >> 56) & 0xFF),
|
||||
(uint8_t)((sz64 >> 48) & 0xFF),
|
||||
(uint8_t)((sz64 >> 40) & 0xFF),
|
||||
(uint8_t)((sz64 >> 32) & 0xFF),
|
||||
(uint8_t)((sz64 >> 24) & 0xFF),
|
||||
(uint8_t)((sz64 >> 16) & 0xFF),
|
||||
(uint8_t)((sz64 >> 8) & 0xFF),
|
||||
(uint8_t)((sz64 >> 0) & 0xFF),
|
||||
};
|
||||
out_->Push((const char *)frameSize, sizeof(frameSize));
|
||||
}
|
||||
}
|
||||
|
||||
};
|
95
ext/native/net/websocket_server.h
Normal file
95
ext/native/net/websocket_server.h
Normal file
@ -0,0 +1,95 @@
|
||||
#pragma once
|
||||
|
||||
#include <functional>
|
||||
#include <string>
|
||||
#include "net/http_server.h"
|
||||
#include "net/sinks.h"
|
||||
|
||||
namespace net {
|
||||
|
||||
enum class WebSocketClose : uint16_t {
|
||||
NORMAL = 1000,
|
||||
GOING_AWAY = 1001,
|
||||
PROTOCOL_ERROR = 1002,
|
||||
UNSUPPORTED_DATA = 1003,
|
||||
INVALID_DATA = 1007,
|
||||
POLICY_VIOLATION = 1008,
|
||||
MESSAGE_TOO_LONG = 1009,
|
||||
MISSING_EXTENSION = 1010,
|
||||
INTERNAL_ERROR = 1011,
|
||||
SERVICE_RESTART = 1012,
|
||||
TRY_AGAIN_LATER = 1013,
|
||||
BAD_GATEWAY = 1014,
|
||||
|
||||
NO_STATUS = 1005,
|
||||
ABNORMAL = 1006,
|
||||
};
|
||||
|
||||
// RFC 6455
|
||||
class WebSocketServer {
|
||||
public:
|
||||
static WebSocketServer *CreateAsUpgrade(const http::Request &request);
|
||||
|
||||
// TODO: Doesn't support fragmented data yet.
|
||||
void Send(const std::string &str);
|
||||
void Send(const std::vector<uint8_t> &payload);
|
||||
void Ping(const std::vector<uint8_t> &payload = {});
|
||||
void Pong(const std::vector<uint8_t> &payload = {});
|
||||
void Close(WebSocketClose reason = WebSocketClose::GOING_AWAY);
|
||||
|
||||
// Note: may interrupt early. Call in a loop.
|
||||
bool Process(float timeout = -1.0f);
|
||||
|
||||
void SetTextHandler(std::function<void(const std::string &)> func) {
|
||||
text_ = func;
|
||||
}
|
||||
void SetBinaryHandler(std::function<void(const std::vector<uint8_t> &)> func) {
|
||||
binary_ = func;
|
||||
}
|
||||
// Doesn't need to send a Pong.
|
||||
void SetPingHandler(std::function<void(const std::vector<uint8_t> &)> func) {
|
||||
ping_ = func;
|
||||
}
|
||||
void SetPongHandler(std::function<void(const std::vector<uint8_t> &)> func) {
|
||||
pong_ = func;
|
||||
}
|
||||
|
||||
bool IsOpen() {
|
||||
return open_;
|
||||
}
|
||||
WebSocketClose CloseReason() {
|
||||
return closeReason_;
|
||||
}
|
||||
|
||||
protected:
|
||||
WebSocketServer(size_t fd, InputSink *in, OutputSink *out) : fd_(fd), in_(in), out_(out) {
|
||||
}
|
||||
|
||||
void SendHeader(bool fin, int opcode, size_t sz);
|
||||
bool ReadFrames();
|
||||
bool ReadFrame();
|
||||
bool ReadPending();
|
||||
bool ReadControlFrame(int opcode, size_t sz);
|
||||
|
||||
bool open_ = true;
|
||||
bool sentClose_ = false;
|
||||
size_t fd_ = 0;
|
||||
InputSink *in_ = nullptr;
|
||||
OutputSink *out_ = nullptr;
|
||||
WebSocketClose closeReason_ = WebSocketClose::NO_STATUS;
|
||||
|
||||
std::vector<uint8_t> pendingBuf_;
|
||||
uint8_t pendingMask_[4]{};
|
||||
// Bytes left to read in the frame (in case of a partial frame read.)
|
||||
uint64_t pendingLeft_ = 0;
|
||||
int pendingOpcode_ = -1;
|
||||
// Waiting for a frame with FIN.
|
||||
bool pendingFin_ = false;
|
||||
|
||||
std::function<void(const std::string &)> text_;
|
||||
std::function<void(const std::vector<uint8_t> &)> binary_;
|
||||
std::function<void(const std::vector<uint8_t> &)> ping_;
|
||||
std::function<void(const std::vector<uint8_t> &)> pong_;
|
||||
};
|
||||
|
||||
};
|
Loading…
x
Reference in New Issue
Block a user