initial commit

third_party/libhv/http/WebSocketParser.cpp

@@ -0,0 +1,75 @@
+#include "WebSocketParser.h"
+
+#include "websocket_parser.h"
+#include "hdef.h"
+
+#define MAX_PAYLOAD_LENGTH  (1 << 24)   // 16M
+
+static int on_frame_header(websocket_parser* parser) {
+    WebSocketParser* wp = (WebSocketParser*)parser->data;
+    int opcode = parser->flags & WS_OP_MASK;
+    // printf("on_frame_header opcode=%d\n", opcode);
+    if (opcode != WS_OP_CONTINUE) {
+        wp->opcode = opcode;
+    }
+    int length = parser->length;
+    int reserve_length = MIN(length + 1, MAX_PAYLOAD_LENGTH);
+    if (reserve_length > wp->message.capacity()) {
+        wp->message.reserve(reserve_length);
+    }
+    if (wp->state == WS_FRAME_BEGIN ||
+        wp->state == WS_FRAME_FIN) {
+        wp->message.clear();
+    }
+    wp->state = WS_FRAME_HEADER;
+    return 0;
+}
+
+static int on_frame_body(websocket_parser* parser, const char * at, size_t length) {
+    // printf("on_frame_body length=%d\n", (int)length);
+    WebSocketParser* wp = (WebSocketParser*)parser->data;
+    wp->state = WS_FRAME_BODY;
+    if (wp->parser->flags & WS_HAS_MASK) {
+        websocket_parser_decode((char*)at, at, length, wp->parser);
+    }
+    wp->message.append(at, length);
+    return 0;
+}
+
+static int on_frame_end(websocket_parser* parser) {
+    // printf("on_frame_end\n");
+    WebSocketParser* wp = (WebSocketParser*)parser->data;
+    wp->state = WS_FRAME_END;
+    if (wp->parser->flags & WS_FIN) {
+        wp->state = WS_FRAME_FIN;
+        if (wp->onMessage) {
+            wp->onMessage(wp->opcode, wp->message);
+        }
+    }
+    return 0;
+}
+
+static websocket_parser_settings cbs = {
+    on_frame_header,
+    on_frame_body,
+    on_frame_end
+};
+
+WebSocketParser::WebSocketParser() {
+    parser = (websocket_parser*)malloc(sizeof(websocket_parser));
+    websocket_parser_init(parser);
+    parser->data = this;
+    state = WS_FRAME_BEGIN;
+    opcode = WS_OP_CLOSE;
+}
+
+WebSocketParser::~WebSocketParser() {
+    if (parser) {
+        free(parser);
+        parser = NULL;
+    }
+}
+
+int WebSocketParser::FeedRecvData(const char* data, size_t len) {
+    return websocket_parser_execute(parser, &cbs, data, len);
+}