1 files changed, 495 insertions, 0 deletions
diff --git a/fluent-bit/lib/wasm-micro-runtime-WAMR-1.2.2/assembly-script/wamr_app_lib/request.ts b/fluent-bit/lib/wasm-micro-runtime-WAMR-1.2.2/assembly-script/wamr_app_lib/request.ts
new file mode 100644
index 000000000..16a229277
--- /dev/null
+++ b/fluent-bit/lib/wasm-micro-runtime-WAMR-1.2.2/assembly-script/wamr_app_lib/request.ts
@@ -0,0 +1,495 @@
+/*
+ * Copyright (C) 2019 Intel Corporation.  All rights reserved.
+ * SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+ */
+
+import * as console from './console'
+import * as timer from './timer'
+
+@external("env", "wasm_response_send")
+declare function wasm_response_send(buffer: ArrayBuffer, size: i32): bool;
+
+@external("env", "wasm_register_resource")
+declare function wasm_register_resource(url: ArrayBuffer): void;
+
+@external("env", "wasm_post_request")
+declare function wasm_post_request(buffer: ArrayBuffer, size: i32): void;
+
+@external("env", "wasm_sub_event")
+declare function wasm_sub_event(url: ArrayBuffer): void;
+
+var COAP_GET = 1;
+var COAP_POST = 2;
+var COAP_PUT = 3;
+var COAP_DELETE = 4;
+var COAP_EVENT = COAP_DELETE + 2;
+
+/* CoAP response codes */
+export enum CoAP_Status {
+    NO_ERROR = 0,
+
+    CREATED_2_01 = 65, /* CREATED */
+    DELETED_2_02 = 66, /* DELETED */
+    VALID_2_03 = 67, /* NOT_MODIFIED */
+    CHANGED_2_04 = 68, /* CHANGED */
+    CONTENT_2_05 = 69, /* OK */
+    CONTINUE_2_31 = 95, /* CONTINUE */
+
+    BAD_REQUEST_4_00 = 128, /* BAD_REQUEST */
+    UNAUTHORIZED_4_01 = 129, /* UNAUTHORIZED */
+    BAD_OPTION_4_02 = 130, /* BAD_OPTION */
+    FORBIDDEN_4_03 = 131, /* FORBIDDEN */
+    NOT_FOUND_4_04 = 132, /* NOT_FOUND */
+    METHOD_NOT_ALLOWED_4_05 = 133, /* METHOD_NOT_ALLOWED */
+    NOT_ACCEPTABLE_4_06 = 134, /* NOT_ACCEPTABLE */
+    PRECONDITION_FAILED_4_12 = 140, /* BAD_REQUEST */
+    REQUEST_ENTITY_TOO_LARGE_4_13 = 141, /* REQUEST_ENTITY_TOO_LARGE */
+    UNSUPPORTED_MEDIA_TYPE_4_15 = 143, /* UNSUPPORTED_MEDIA_TYPE */
+
+    INTERNAL_SERVER_ERROR_5_00 = 160, /* INTERNAL_SERVER_ERROR */
+    NOT_IMPLEMENTED_5_01 = 161, /* NOT_IMPLEMENTED */
+    BAD_GATEWAY_5_02 = 162, /* BAD_GATEWAY */
+    SERVICE_UNAVAILABLE_5_03 = 163, /* SERVICE_UNAVAILABLE */
+    GATEWAY_TIMEOUT_5_04 = 164, /* GATEWAY_TIMEOUT */
+    PROXYING_NOT_SUPPORTED_5_05 = 165, /* PROXYING_NOT_SUPPORTED */
+
+    /* Erbium errors */
+    MEMORY_ALLOCATION_ERROR = 192, PACKET_SERIALIZATION_ERROR,
+
+    /* Erbium hooks */
+    MANUAL_RESPONSE, PING_RESPONSE
+};
+
+var g_mid: i32 = 0;
+class wamr_request {
+    mid: i32 = 0;
+    url: string = "";
+    action: i32 = 0;
+    fmt: i32 = 0;
+    payload: ArrayBuffer;
+    payload_len: i32 = 0;
+
+    sender: i32 = 0;
+
+    constructor(mid: i32, url: string, action: i32, fmt: i32,
+        payload: ArrayBuffer, payload_len: number) {
+        this.mid = mid;
+        this.url = url;
+        this.action = action;
+        this.fmt = fmt;
+        this.payload = payload;
+        this.payload_len = i32(payload_len);
+    }
+}
+
+class wamr_response {
+    mid: i32 = 0;
+    status: i32 = 0;
+    fmt: i32 = 0;
+    payload: ArrayBuffer | null;
+    payload_len: i32 = 0;
+
+    receiver: i32 = 0;
+
+    constructor(mid: i32, status: i32, fmt: i32,
+        payload: ArrayBuffer | null, payload_len: i32) {
+        this.mid = mid;
+        this.status = status;
+        this.fmt = fmt;
+        this.payload = payload;
+        this.payload_len = payload_len;
+    }
+
+    set_status(status: number): void {
+        this.status = i32(status);
+    }
+
+    set_payload(payload: ArrayBuffer, payload_len: number): void {
+        this.payload = payload;
+        this.payload_len = i32(payload_len);
+    }
+}
+
+class wamr_resource {
+    url: string;
+    type: number;
+    cb: request_handler_f;
+
+    constructor(url: string, type: number, cb: request_handler_f) {
+        this.url = url;
+        this.type = type;
+        this.cb = cb;
+    }
+}
+
+function is_expire(trans: wamr_transaction, index: i32, array: Array<wamr_transaction>): bool {
+    var now = timer.now();
+
+    var elapsed_ms = (now < trans.time) ?
+        (now + (0xFFFFFFFF - trans.time) + 1) : (now - trans.time);
+
+    return elapsed_ms >= TRANSACTION_TIMEOUT_MS;
+}
+
+function not_expire(trans: wamr_transaction, index: i32, array: Array<wamr_transaction>): bool {
+    var now = timer.now();
+
+    var elapsed_ms = (now < trans.time) ?
+        (now + (0xFFFFFFFF - trans.time) + 1) : (now - trans.time);
+
+    return elapsed_ms < TRANSACTION_TIMEOUT_MS;
+}
+
+function transaction_timeout_handler(): void {
+    var now = timer.now();
+
+    var expired = transaction_list.filter(is_expire);
+    transaction_list = transaction_list.filter(not_expire);
+
+    expired.forEach(item => {
+        item.cb(null);
+        transaction_remove(item);
+    })
+
+    if (transaction_list.length > 0) {
+        var elpased_ms: number, ms_to_expiry: number;
+        now = timer.now();
+        if (now < transaction_list[0].time) {
+            elpased_ms = now + (0xFFFFFFFF - transaction_list[0].time) + 1;
+        } else {
+            elpased_ms = now - transaction_list[0].time;
+        }
+        ms_to_expiry = TRANSACTION_TIMEOUT_MS - elpased_ms;
+        timer.timer_restart(g_trans_timer, ms_to_expiry);
+    } else {
+        timer.timer_cancel(g_trans_timer);
+    }
+}
+
+function transaction_find(mid: number): wamr_transaction | null {
+    for (let i = 0; i < transaction_list.length; i++) {
+        if (transaction_list[i].mid == mid)
+            return transaction_list[i];
+    }
+    return null;
+}
+
+function transaction_add(trans: wamr_transaction): void {
+    transaction_list.push(trans);
+
+    if (transaction_list.length == 1) {
+        g_trans_timer = timer.setTimeout(
+            transaction_timeout_handler,
+            TRANSACTION_TIMEOUT_MS
+        );
+    }
+}
+
+function transaction_remove(trans: wamr_transaction): void {
+    var index = transaction_list.indexOf(trans);
+    transaction_list.splice(index, 1);
+}
+
+var transaction_list = new Array<wamr_transaction>();
+class wamr_transaction {
+    mid: number;
+    time: number;
+    cb: (resp: wamr_response | null) => void;
+
+    constructor(mid: number, time: number, cb: (resp: wamr_response) => void) {
+        this.mid = mid;
+        this.time = time;
+        this.cb = cb;
+    }
+}
+
+var REQUEST_PACKET_FIX_PART_LEN = 18;
+var RESPONSE_PACKET_FIX_PART_LEN = 16;
+var TRANSACTION_TIMEOUT_MS = 5000;
+var g_trans_timer: timer.user_timer;
+
+var Reg_Event = 0;
+var Reg_Request = 1;
+
+function pack_request(req: wamr_request): DataView {
+    var url_len = req.url.length + 1;
+    var len = REQUEST_PACKET_FIX_PART_LEN + url_len + req.payload_len
+    var buf = new ArrayBuffer(len);
+
+    var dataview = new DataView(buf, 0, len);
+
+    dataview.setUint8(0, 1);
+    dataview.setUint8(1, u8(req.action));
+    dataview.setUint16(2, u16(req.fmt));
+    dataview.setUint32(4, req.mid);
+    dataview.setUint32(8, req.sender);
+    dataview.setUint16(12, u16(url_len))
+    dataview.setUint32(14, req.payload_len);
+
+    var i = 0;
+    for (i = 0; i < url_len - 1; i++) {
+        dataview.setUint8(i + 18, u8(req.url.codePointAt(i)));
+    }
+    dataview.setUint8(i + 18, 0);
+
+    var payload_view = new DataView(req.payload);
+    for (i = 0; i < req.payload_len; i++) {
+        dataview.setUint8(i + 18 + url_len, u8(payload_view.getUint8(i)));
+    }
+
+    return dataview;
+}
+
+function unpack_request(packet: ArrayBuffer, size: i32): wamr_request {
+    var dataview = new DataView(packet, 0, size);
+
+    if (dataview.getUint8(0) != 1)
+        throw new Error("packet version mismatch");
+
+    if (size < REQUEST_PACKET_FIX_PART_LEN)
+        throw new Error("packet size error");
+
+    var url_len = dataview.getUint16(12);
+    var payload_len = dataview.getUint32(14);
+
+    if (size != (REQUEST_PACKET_FIX_PART_LEN + url_len + payload_len))
+        throw new Error("packet size error");
+
+    var action = dataview.getUint8(1);
+    var fmt = dataview.getUint16(2);
+    var mid = dataview.getUint32(4);
+    var sender = dataview.getUint32(8);
+
+    var url = packet.slice(REQUEST_PACKET_FIX_PART_LEN, REQUEST_PACKET_FIX_PART_LEN + url_len - 1);
+    var payload = packet.slice(REQUEST_PACKET_FIX_PART_LEN + url_len, REQUEST_PACKET_FIX_PART_LEN + url_len + payload_len);
+
+    var req = new wamr_request(mid, String.UTF8.decode(url), action, fmt, payload, payload_len);
+    req.sender = sender;
+
+    return req;
+}
+
+function pack_response(resp: wamr_response): DataView {
+    var len = RESPONSE_PACKET_FIX_PART_LEN + resp.payload_len
+    var buf = new ArrayBuffer(len);
+
+    var dataview = new DataView(buf, 0, len);
+
+    dataview.setUint8(0, 1);
+    dataview.setUint8(1, u8(resp.status));
+    dataview.setUint16(2, u16(resp.fmt));
+    dataview.setUint32(4, resp.mid);
+    dataview.setUint32(8, resp.receiver);
+    dataview.setUint32(12, resp.payload_len)
+
+    if (resp.payload != null) {
+        var payload_view = new DataView(resp.payload!);
+        for (let i = 0; i < resp.payload_len; i++) {
+            dataview.setUint8(i + 16, payload_view.getUint8(i));
+        }
+    }
+
+    return dataview;
+}
+
+function unpack_response(packet: ArrayBuffer, size: i32): wamr_response {
+    var dataview = new DataView(packet, 0, size);
+
+    if (dataview.getUint8(0) != 1)
+        throw new Error("packet version mismatch");
+
+    if (size < RESPONSE_PACKET_FIX_PART_LEN)
+        throw new Error("packet size error");
+
+    var payload_len = dataview.getUint32(12);
+    if (size != RESPONSE_PACKET_FIX_PART_LEN + payload_len)
+        throw new Error("packet size error");
+
+    var status = dataview.getUint8(1);
+    var fmt = dataview.getUint16(2);
+    var mid = dataview.getUint32(4);
+    var receiver = dataview.getUint32(8);
+
+    var payload = packet.slice(RESPONSE_PACKET_FIX_PART_LEN);
+
+    var resp = new wamr_response(mid, status, fmt, payload, payload_len);
+    resp.receiver = receiver;
+
+    return resp;
+}
+
+function do_request(req: wamr_request, cb: (resp: wamr_response) => void): void {
+    var trans = new wamr_transaction(req.mid, timer.now(), cb);
+    var msg = pack_request(req);
+
+    transaction_add(trans);
+
+    wasm_post_request(msg.buffer, msg.byteLength);
+}
+
+function do_response(resp: wamr_response): void {
+    var msg = pack_response(resp);
+
+    wasm_response_send(msg.buffer, msg.byteLength);
+}
+
+var resource_list = new Array<wamr_resource>();
+type request_handler_f = (req: wamr_request) => void;
+
+function registe_url_handler(url: string, cb: request_handler_f, type: number): void {
+    for (let i = 0; i < resource_list.length; i++) {
+        if (resource_list[i].type == type && resource_list[i].url == url) {
+            resource_list[i].cb = cb;
+            return;
+        }
+    }
+
+    var res = new wamr_resource(url, type, cb);
+    resource_list.push(res);
+
+    if (type == Reg_Request)
+        wasm_register_resource(String.UTF8.encode(url));
+    else
+        wasm_sub_event(String.UTF8.encode(url));
+}
+
+function is_event_type(req: wamr_request): bool {
+    return req.action == COAP_EVENT;
+}
+
+function check_url_start(url: string, leading_str: string): bool {
+    return url.split('/')[0] == leading_str.split('/')[0];
+}
+
+/* User APIs below */
+export function post(url: string, payload: ArrayBuffer, payload_len: number, tag: string,
+                     cb: (resp: wamr_response) => void): void {
+    var req = new wamr_request(g_mid++, url, COAP_POST, 0, payload, payload_len);
+
+    do_request(req, cb);
+}
+
+export function get(url: string, tag: string,
+                    cb: (resp: wamr_response) => void): void {
+    var req = new wamr_request(g_mid++, url, COAP_GET, 0, new ArrayBuffer(0), 0);
+
+    do_request(req, cb);
+}
+
+export function put(url: string, payload: ArrayBuffer, payload_len: number, tag: string,
+                    cb: (resp: wamr_response) => void): void {
+    var req = new wamr_request(g_mid++, url, COAP_PUT, 0, payload, payload_len);
+
+    do_request(req, cb);
+}
+
+export function del(url: string, tag: string,
+                    cb: (resp: wamr_response) => void): void {
+    var req = new wamr_request(g_mid++, url, COAP_DELETE, 0, new ArrayBuffer(0), 0);
+
+    do_request(req, cb);
+}
+
+export function make_response_for_request(req: wamr_request): wamr_response {
+    var resp = new wamr_response(req.mid, CoAP_Status.CONTENT_2_05, 0, null, 0);
+    resp.receiver = req.sender;
+
+    return resp;
+}
+
+export function api_response_send(resp: wamr_response): void {
+    do_response(resp);
+}
+
+export function register_resource_handler(url: string,
+    request_handle: request_handler_f): void {
+    registe_url_handler(url, request_handle, Reg_Request);
+}
+
+export function publish_event(url: string, fmt: number,
+    payload: ArrayBuffer, payload_len: number): void {
+    var req = new wamr_request(g_mid++, url, COAP_EVENT, i32(fmt), payload, payload_len);
+
+    var msg = pack_request(req);
+
+    wasm_post_request(msg.buffer, msg.byteLength);
+}
+
+export function subscribe_event(url: string, cb: request_handler_f): void {
+    registe_url_handler(url, cb, Reg_Event);
+}
+
+
+/* These two APIs are required by wamr runtime,
+    use a wrapper to export them in the entry file
+
+    e.g:
+
+    import * as request from '.wamr_app_lib/request'
+
+    // Your code here ...
+
+    export function _on_request(buffer_offset: i32, size: i32): void {
+        on_request(buffer_offset, size);
+    }
+
+    export function _on_response(buffer_offset: i32, size: i32): void {
+        on_response(buffer_offset, size);
+    }
+*/
+export function on_request(buffer_offset: i32, size: i32): void {
+    var buffer = new ArrayBuffer(size);
+    var dataview = new DataView(buffer);
+
+    for (let i = 0; i < size; i++) {
+        dataview.setUint8(i, load<i8>(buffer_offset + i, 0, 1));
+    }
+
+    var req = unpack_request(buffer, size);
+
+    var is_event = is_event_type(req);
+
+    for (let i = 0; i < resource_list.length; i++) {
+        if ((is_event && resource_list[i].type == Reg_Event)
+            || (!is_event && resource_list[i].type == Reg_Request)) {
+            if (check_url_start(req.url, resource_list[i].url)) {
+                resource_list[i].cb(req);
+                return;
+            }
+        }
+    }
+
+    console.log("on_request: exit. no service handler.");
+}
+
+export function on_response(buffer_offset: i32, size: i32): void {
+    var buffer = new ArrayBuffer(size);
+    var dataview = new DataView(buffer);
+
+    for (let i = 0; i < size; i++) {
+        dataview.setUint8(i, load<i8>(buffer_offset + i, 0, 1));
+    }
+
+    var resp = unpack_response(buffer, size);
+    var trans = transaction_find(resp.mid);
+
+    if (trans != null) {
+        if (transaction_list.indexOf(trans) == 0) {
+            if (transaction_list.length >= 2) {
+                var elpased_ms: number, ms_to_expiry: number;
+                var now = timer.now();
+                if (now < transaction_list[1].time) {
+                    elpased_ms = now + (0xFFFFFFFF - transaction_list[1].time) + 1;
+                } else {
+                    elpased_ms = now - transaction_list[1].time;
+                }
+                ms_to_expiry = TRANSACTION_TIMEOUT_MS - elpased_ms;
+                timer.timer_restart(g_trans_timer, ms_to_expiry);
+            } else {
+                timer.timer_cancel(g_trans_timer);
+            }
+        }
+
+        trans.cb(resp);
+    }
+}