82 lines
2.4 KiB
C++
82 lines
2.4 KiB
C++
|
#include "WebSocketChannel.h"
|
||
|
|
|
||
|
|
namespace hv {
|
||
|
|
|
||
|
|
int WebSocketChannel::send(const char* buf, int len, enum ws_opcode opcode /* = WS_OPCODE_BINARY */, bool fin /* = true */) {
|
||
|
|
int fragment = 0xFFFF; // 65535
|
||
|
|
if (len > fragment) {
|
||
|
|
return send(buf, len, fragment, opcode);
|
||
|
|
}
|
||
|
|
std::lock_guard<std::mutex> locker(mutex_);
|
||
|
|
return sendFrame(buf, len, opcode, fin);
|
||
|
|
}
|
||
|
|
|
||
|
|
/*
|
||
|
|
* websocket fragment
|
||
|
|
* lock ->
|
||
|
|
* sendFrame(p, fragment, opcode, false) ->
|
||
|
|
* sendFrame(p, fragment, WS_OPCODE_CONTINUE, false) ->
|
||
|
|
* ... ->
|
||
|
|
* sendFrame(p, remain, WS_OPCODE_CONTINUE, true)
|
||
|
|
* unlock
|
||
|
|
*
|
||
|
|
*/
|
||
|
|
int WebSocketChannel::send(const char* buf, int len, int fragment, enum ws_opcode opcode /* = WS_OPCODE_BINARY */) {
|
||
|
|
std::lock_guard<std::mutex> locker(mutex_);
|
||
|
|
if (len <= fragment) {
|
||
|
|
return sendFrame(buf, len, opcode, true);
|
||
|
|
}
|
||
|
|
|
||
|
|
// first fragment
|
||
|
|
int nsend = sendFrame(buf, fragment, opcode, false);
|
||
|
|
if (nsend < 0) return nsend;
|
||
|
|
|
||
|
|
const char* p = buf + fragment;
|
||
|
|
int remain = len - fragment;
|
||
|
|
while (remain > fragment) {
|
||
|
|
nsend = sendFrame(p, fragment, WS_OPCODE_CONTINUE, false);
|
||
|
|
if (nsend < 0) return nsend;
|
||
|
|
p += fragment;
|
||
|
|
remain -= fragment;
|
||
|
|
}
|
||
|
|
|
||
|
|
// last fragment
|
||
|
|
nsend = sendFrame(p, remain, WS_OPCODE_CONTINUE, true);
|
||
|
|
if (nsend < 0) return nsend;
|
||
|
|
|
||
|
|
return len;
|
||
|
|
}
|
||
|
|
|
||
|
|
int WebSocketChannel::sendPing() {
|
||
|
|
std::lock_guard<std::mutex> locker(mutex_);
|
||
|
|
if (type == WS_CLIENT) {
|
||
|
|
return write(WS_CLIENT_PING_FRAME, WS_CLIENT_MIN_FRAME_SIZE);
|
||
|
|
}
|
||
|
|
return write(WS_SERVER_PING_FRAME, WS_SERVER_MIN_FRAME_SIZE);
|
||
|
|
}
|
||
|
|
|
||
|
|
int WebSocketChannel::sendPong() {
|
||
|
|
std::lock_guard<std::mutex> locker(mutex_);
|
||
|
|
if (type == WS_CLIENT) {
|
||
|
|
return write(WS_CLIENT_PONG_FRAME, WS_CLIENT_MIN_FRAME_SIZE);
|
||
|
|
}
|
||
|
|
return write(WS_SERVER_PONG_FRAME, WS_SERVER_MIN_FRAME_SIZE);
|
||
|
|
}
|
||
|
|
|
||
|
|
int WebSocketChannel::sendFrame(const char* buf, int len, enum ws_opcode opcode /* = WS_OPCODE_BINARY */, bool fin /* = true */) {
|
||
|
|
bool has_mask = false;
|
||
|
|
char mask[4] = {0};
|
||
|
|
if (type == WS_CLIENT) {
|
||
|
|
*(int*)mask = rand();
|
||
|
|
has_mask = true;
|
||
|
|
}
|
||
|
|
int frame_size = ws_calc_frame_size(len, has_mask);
|
||
|
|
if (sendbuf_.len < (size_t)frame_size) {
|
||
|
|
sendbuf_.resize(ceil2e(frame_size));
|
||
|
|
}
|
||
|
|
ws_build_frame(sendbuf_.base, buf, len, mask, has_mask, opcode, fin);
|
||
|
|
return write(sendbuf_.base, frame_size);
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|