initial commit

third_party/libhv/http/wsdef.c

@@ -0,0 +1,46 @@
+#include "wsdef.h"
+
+#include <string.h>
+
+#include "sha1.h"
+#include "base64.h"
+
+#include "websocket_parser.h"
+
+// base64_encode( SHA1(key + magic) )
+void ws_encode_key(const char* key, char accept[]) {
+    char magic[] = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";
+    unsigned char digest[20] = {0};
+    HV_SHA1_CTX ctx;
+    HV_SHA1Init(&ctx);
+    HV_SHA1Update(&ctx, (unsigned char*)key, (uint32_t)strlen(key));
+    HV_SHA1Update(&ctx, (unsigned char*)magic, (uint32_t)strlen(magic));
+    HV_SHA1Final(digest, &ctx);
+    hv_base64_encode(digest, 20, accept);
+}
+
+// fix-header[2] + var-length[2/8] + mask[4] + data[data_len]
+int ws_calc_frame_size(int data_len, bool has_mask) {
+    int size = data_len + 2;
+    if (data_len >=126) {
+        if (data_len > 0xFFFF) {
+            size += 8;
+        } else {
+            size += 2;
+        }
+    }
+    if (has_mask) size += 4;
+    return size;
+}
+
+int ws_build_frame(
+    char* out,
+    const char* data, int data_len,
+    const char mask[4], bool has_mask,
+    enum ws_opcode opcode,
+    bool fin) {
+    int flags = opcode;
+    if (fin) flags |= WS_FIN;
+    if (has_mask) flags |=  WS_HAS_MASK;
+    return (int)websocket_build_frame(out, (websocket_flags)flags, mask, data, data_len);
+}