liba 0.1.15

An algorithm library based on C/C++
Documentation
#include "a.h"
#include "a/crc.h"

struct crc32
{
    a_u32 table[0x100];
    a_u32 (*eval)(a_u32 const table[0x100], void const *pdata, a_size nbyte, a_u32 value);
};

static JSClassID liba_crc32_class_id;

static void liba_crc32_finalizer(JSRuntime *rt, JSValue val)
{
    js_free_rt(rt, JS_GetOpaque(val, liba_crc32_class_id));
}

static JSClassDef liba_crc32_class = {"crc32", .finalizer = liba_crc32_finalizer};

static JSValue liba_crc32_ctor(JSContext *ctx, JSValueConst new_target, int argc, JSValueConst *argv)
{
    JSValue proto, clazz = JS_UNDEFINED;
    struct crc32 *const self = (struct crc32 *)js_mallocz(ctx, sizeof(struct crc32));
    if (!self) { return JS_EXCEPTION; }
    a_u32 poly = 0;
    if (JS_ToUint32(ctx, &poly, argv[0])) { goto fail; }
    int reversed = 0;
    if (argc > 1)
    {
        reversed = JS_ToBool(ctx, argv[1]);
        if (reversed < 0) { goto fail; }
    }
    if (reversed)
    {
        a_crc32l_init(self->table, poly);
        self->eval = a_crc32l;
    }
    else
    {
        a_crc32m_init(self->table, poly);
        self->eval = a_crc32m;
    }
    proto = JS_GetPropertyStr(ctx, new_target, "prototype");
    if (JS_IsException(proto)) { goto fail; }
    clazz = JS_NewObjectProtoClass(ctx, proto, liba_crc32_class_id);
    JS_FreeValue(ctx, proto);
    if (JS_IsException(clazz)) { goto fail; }
    JS_SetOpaque(clazz, self);
    return clazz;
fail:
    js_free(ctx, self);
    JS_FreeValue(ctx, clazz);
    return JS_UNDEFINED;
}

static JSValue liba_crc32_gen(JSContext *ctx, JSValueConst this_val, int argc, JSValueConst *argv)
{
    struct crc32 *const self = (struct crc32 *)JS_GetOpaque2(ctx, this_val, liba_crc32_class_id);
    if (!self) { return JS_EXCEPTION; }
    a_u32 poly = 0;
    if (JS_ToUint32(ctx, &poly, argv[0])) { return JS_EXCEPTION; }
    int reversed = 0;
    if (argc > 1)
    {
        reversed = JS_ToBool(ctx, argv[1]);
        if (reversed < 0) { return JS_EXCEPTION; }
    }
    if (reversed)
    {
        a_crc32l_init(self->table, poly);
        self->eval = a_crc32l;
    }
    else
    {
        a_crc32m_init(self->table, poly);
        self->eval = a_crc32m;
    }
    return JS_UNDEFINED;
}

static JSValue liba_crc32_eval(JSContext *ctx, JSValueConst this_val, int argc, JSValueConst *argv)
{
    struct crc32 *const self = (struct crc32 *)JS_GetOpaque2(ctx, this_val, liba_crc32_class_id);
    if (!self) { return JS_EXCEPTION; }
    a_u32 value = 0;
    if (argc > 1)
    {
        if (JS_ToUint32(ctx, &value, argv[1])) { return JS_EXCEPTION; }
    }
    size_t n = 0;
    if (JS_IsArray(ctx, argv[0]))
    {
        a_byte *p = JS_GetArrayBuffer(ctx, &n, argv[0]);
        if (p) { value = self->eval(self->table, p, n, value); }
    }
    else
    {
        char const *const p = JS_ToCStringLen(ctx, &n, argv[0]);
        value = self->eval(self->table, p, n, value);
        JS_FreeCString(ctx, p);
    }
    return JS_NewUint32(ctx, value);
}

static JSValue liba_crc32_pack(JSContext *ctx, JSValueConst this_val, int argc, JSValueConst *argv)
{
    JSValue val = JS_UNDEFINED;
    struct crc32 *const self = (struct crc32 *)JS_GetOpaque2(ctx, this_val, liba_crc32_class_id);
    if (!self) { return JS_EXCEPTION; }
    a_u32 value = 0;
    if (argc > 1)
    {
        if (JS_ToUint32(ctx, &value, argv[1])) { return JS_EXCEPTION; }
    }
    size_t n = 0;
    char const *const s = JS_ToCStringLen(ctx, &n, argv[0]);
    value = self->eval(self->table, s, n, value);
    a_byte *p = (a_byte *)js_malloc(ctx, n + 4);
    if (p) { a_copy(p, s, n); }
    else { goto fail; }
    self->eval == a_crc32m
        ? a_u32_setb(p + n, value)
        : a_u32_setl(p + n, value);
    val = js_array_u8_new(ctx, p, (uint32_t)n + 4);
fail:
    JS_FreeCString(ctx, s);
    js_free(ctx, p);
    return val;
}

enum
{
    self_table,
};

static JSValue liba_crc32_get(JSContext *ctx, JSValueConst this_val, int magic)
{
    struct crc32 *const self = (struct crc32 *)JS_GetOpaque2(ctx, this_val, liba_crc32_class_id);
    if (!self) { return JS_EXCEPTION; }
    if (magic == self_table)
    {
        return js_array_u32_new(ctx, self->table, 0x100);
    }
    return JS_UNDEFINED;
}

static JSCFunctionListEntry const liba_crc32_proto[] = {
    JS_PROP_STRING_DEF("[Symbol.toStringTag]", "a.crc32", 0),
    JS_CGETSET_MAGIC_DEF("table", liba_crc32_get, NULL, self_table),
    JS_CFUNC_DEF("gen", 2, liba_crc32_gen),
    JS_CFUNC_DEF("eval", 2, liba_crc32_eval),
    JS_CFUNC_DEF("pack", 2, liba_crc32_pack),
};

int js_liba_crc32_init(JSContext *ctx, JSModuleDef *m)
{
    JS_NewClassID(&liba_crc32_class_id);
    JS_NewClass(JS_GetRuntime(ctx), liba_crc32_class_id, &liba_crc32_class);

    JSValue const proto = JS_NewObject(ctx);
    JS_SetPropertyFunctionList(ctx, proto, liba_crc32_proto, A_LEN(liba_crc32_proto));

    JSValue const clazz = JS_NewCFunction2(ctx, liba_crc32_ctor, "crc32", 2, JS_CFUNC_constructor, 0);
    JS_SetConstructor(ctx, clazz, proto);
    JS_SetClassProto(ctx, liba_crc32_class_id, proto);

    return JS_SetModuleExport(ctx, m, "crc32", clazz);
}