initial commit

third_party/libhv/http/WebSocketChannel.cpp

@@ -0,0 +1,81 @@
+#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);
+}
+
+}