Something went wrong. Try again.
javascript for 🐜's, a tiny runtime with big ambitions antjs.org
Something went wrong. Try again.
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463#include <stdlib.h>#include <string.h>
#include "ant.h"#include "errors.h"#include "runtime.h"#include "internal.h"#include "descriptors.h"
#include "modules/symbol.h"#include "modules/buffer.h"#include "modules/textcodec.h"#include "streams/codec.h"#include "streams/transform.h"
ant_value_t g_tes_proto;ant_value_t g_tds_proto;
typedef struct { uint8_t pending[3]; uint8_t pending_len;} tes_state_t;
static ant_value_t tes_get_ts(ant_value_t obj) { return js_get_slot(obj, SLOT_ENTRIES);}
static ant_value_t tds_get_ts(ant_value_t obj) { return js_get_slot(obj, SLOT_ENTRIES);}
bool tes_is_stream(ant_value_t obj) { return is_object_type(obj) && vtype(js_get_slot(obj, SLOT_DATA)) == T_NUM && ts_is_stream(tes_get_ts(obj));}
bool tds_is_stream(ant_value_t obj) { return is_object_type(obj) && vtype(js_get_slot(obj, SLOT_DATA)) == T_NUM && ts_is_stream(tds_get_ts(obj));}
ant_value_t tes_stream_readable(ant_value_t obj) { return ts_stream_readable(tes_get_ts(obj));}
ant_value_t tes_stream_writable(ant_value_t obj) { return ts_stream_writable(tes_get_ts(obj));}
ant_value_t tds_stream_readable(ant_value_t obj) { return ts_stream_readable(tds_get_ts(obj));}
ant_value_t tds_stream_writable(ant_value_t obj) { return ts_stream_writable(tds_get_ts(obj));}
static void tes_state_finalize(ant_t *js, ant_object_t *obj) { if (!obj->extra_slots) return; ant_extra_slot_t *entries = (ant_extra_slot_t *)obj->extra_slots; for (uint8_t i = 0; i < obj->extra_count; i++) { if (entries[i].slot == SLOT_DATA && vtype(entries[i].value) == T_NUM) { free((tes_state_t *)(uintptr_t)(size_t)js_getnum(entries[i].value)); return; }}}
static void tds_state_finalize(ant_t *js, ant_object_t *obj) { if (!obj->extra_slots) return; ant_extra_slot_t *entries = (ant_extra_slot_t *)obj->extra_slots; for (uint8_t i = 0; i < obj->extra_count; i++) { if (entries[i].slot == SLOT_DATA && vtype(entries[i].value) == T_NUM) { free((td_state_t *)(uintptr_t)(size_t)js_getnum(entries[i].value)); return; }}}
static ant_value_t codec_transform_controller(ant_value_t *args, int nargs) { return (nargs > 1) ? args[1] : js_mkundef();}
static ant_value_t codec_flush_controller(ant_value_t *args, int nargs) { return (nargs > 0) ? args[0] : js_mkundef();}
static ant_value_t tes_transform(ant_t *js, ant_value_t *args, int nargs) { ant_value_t wrapper = js_get_slot(js->current_func, SLOT_DATA); ant_value_t state_val = js_get_slot(wrapper, SLOT_DATA); tes_state_t *st = (tes_state_t *)(uintptr_t)(size_t)js_getnum(state_val); ant_value_t ctrl_obj = codec_transform_controller(args, nargs); ant_value_t chunk = (nargs > 0) ? args[0] : js_mkundef(); size_t str_len = 0; const char *str = NULL;
if (vtype(chunk) == T_STR) { str = js_getstr(js, chunk, &str_len); } else { ant_value_t sv = js_tostring_val(js, chunk); if (is_err(sv)) return sv; str = js_getstr(js, sv, &str_len); } if (!str) { str = ""; str_len = 0; }
const uint8_t *s = (const uint8_t *)str; size_t total = st->pending_len + str_len;
if (total == 0) return js_mkundef();
uint8_t *buf = malloc(total); if (!buf) return js_mkerr(js, "out of memory");
size_t off = 0; if (st->pending_len > 0) { memcpy(buf, st->pending, st->pending_len); off = st->pending_len; st->pending_len = 0; } memcpy(buf + off, s, str_len);
size_t out_len = total; if (out_len >= 3) { uint8_t b0 = buf[out_len - 3], b1 = buf[out_len - 2], b2 = buf[out_len - 1]; if (b0 == 0xED && b1 >= 0xA0 && b1 <= 0xAF) { st->pending[0] = b0; st->pending[1] = b1; st->pending[2] = b2; st->pending_len = 3; out_len -= 3; }}
if (out_len == 0) { free(buf); return js_mkundef(); }
uint8_t *out = malloc(out_len * 4 / 3 + 4); if (!out) { free(buf); return js_mkerr(js, "out of memory"); }
size_t i = 0, o = 0; while (i < out_len) { if (i + 5 < out_len && buf[i] == 0xED && buf[i+1] >= 0xA0 && buf[i+1] <= 0xAF && buf[i+3] == 0xED && buf[i+4] >= 0xB0 && buf[i+4] <= 0xBF) { uint32_t hi_cp = ((uint32_t)0x0D << 12) | ((uint32_t)(buf[i+1] & 0x3F) << 6) | (buf[i+2] & 0x3F); uint32_t lo_cp = ((uint32_t)0x0D << 12) | ((uint32_t)(buf[i+4] & 0x3F) << 6) | (buf[i+5] & 0x3F); uint32_t cp = 0x10000 + ((hi_cp - 0xD800) << 10) + (lo_cp - 0xDC00); out[o++] = (uint8_t)(0xF0 | (cp >> 18)); out[o++] = (uint8_t)(0x80 | ((cp >> 12) & 0x3F)); out[o++] = (uint8_t)(0x80 | ((cp >> 6) & 0x3F)); out[o++] = (uint8_t)(0x80 | (cp & 0x3F)); i += 6; } else if (buf[i] == 0xED && i + 2 < out_len && buf[i+1] >= 0xA0 && buf[i+1] <= 0xBF) { out[o++] = 0xEF; out[o++] = 0xBF; out[o++] = 0xBD; i += 3; } else out[o++] = buf[i++]; }
free(buf);
ArrayBufferData *ab = create_array_buffer_data(o); if (!ab) { free(out); return js_mkerr(js, "out of memory"); } memcpy(ab->data, out, o); free(out);
ant_value_t result = create_typed_array(js, TYPED_ARRAY_UINT8, ab, 0, o, "Uint8Array"); return ts_is_controller(ctrl_obj) ? ts_ctrl_enqueue(js, ctrl_obj, result) : js_mkerr_typed(js, JS_ERR_TYPE, "Invalid TransformStreamDefaultController");}
static ant_value_t tes_flush(ant_t *js, ant_value_t *args, int nargs) { ant_value_t wrapper = js_get_slot(js->current_func, SLOT_DATA); ant_value_t state_val = js_get_slot(wrapper, SLOT_DATA); tes_state_t *st = (tes_state_t *)(uintptr_t)(size_t)js_getnum(state_val); ant_value_t ctrl_obj = codec_flush_controller(args, nargs);
if (st->pending_len > 0) { uint8_t fffd[3] = { 0xEF, 0xBF, 0xBD }; ArrayBufferData *ab = create_array_buffer_data(3); if (!ab) return js_mkerr(js, "out of memory"); memcpy(ab->data, fffd, 3); st->pending_len = 0; ant_value_t result = create_typed_array(js, TYPED_ARRAY_UINT8, ab, 0, 3, "Uint8Array"); if (!ts_is_controller(ctrl_obj)) return js_mkerr_typed(js, JS_ERR_TYPE, "Invalid TransformStreamDefaultController"); return ts_ctrl_enqueue(js, ctrl_obj, result); }
return js_mkundef();}
static ant_value_t js_tes_get_encoding(ant_t *js, ant_value_t *args, int nargs) { return js_mkstr(js, "utf-8", 5);}
static ant_value_t js_tes_get_readable(ant_t *js, ant_value_t *args, int nargs) { ant_value_t ts_obj = tes_get_ts(js->this_val); if (!ts_is_stream(ts_obj)) return js_mkerr_typed(js, JS_ERR_TYPE, "Invalid TextEncoderStream"); return ts_stream_readable(ts_obj);}
static ant_value_t js_tes_get_writable(ant_t *js, ant_value_t *args, int nargs) { ant_value_t ts_obj = tes_get_ts(js->this_val); if (!ts_is_stream(ts_obj)) return js_mkerr_typed(js, JS_ERR_TYPE, "Invalid TextEncoderStream"); return ts_stream_writable(ts_obj);}
static ant_value_t js_tes_ctor(ant_t *js, ant_value_t *args, int nargs) { if (vtype(js->new_target) == T_UNDEF) return js_mkerr_typed(js, JS_ERR_TYPE, "TextEncoderStream constructor requires 'new'");
tes_state_t *st = calloc(1, sizeof(tes_state_t)); if (!st) return js_mkerr(js, "out of memory");
ant_value_t obj = js_mkobj(js); ant_value_t proto = js_instance_proto_from_new_target(js, g_tes_proto); if (is_object_type(proto)) js_set_proto_init(obj, proto); js_set_slot(obj, SLOT_DATA, ANT_PTR(st)); js_set_finalizer(obj, tes_state_finalize);
ant_value_t wrapper = js_mkobj(js); js_set_slot(wrapper, SLOT_DATA, ANT_PTR(st));
ant_value_t transformer = js_mkobj(js); ant_value_t transform_fn = js_heavy_mkfun(js, tes_transform, wrapper); ant_value_t flush_fn = js_heavy_mkfun(js, tes_flush, wrapper); js_set(js, transformer, "transform", transform_fn); js_set(js, transformer, "flush", flush_fn);
ant_value_t ctor_args[1] = { transformer }; ant_value_t saved_new_target = js->new_target; ant_value_t saved_this = js->this_val; js->new_target = js_mknum(1);
ant_value_t ts_obj = js_ts_ctor(js, ctor_args, 1); js->new_target = saved_new_target; js->this_val = saved_this;
if (is_err(ts_obj)) { free(st); return ts_obj; } js_set_slot(obj, SLOT_ENTRIES, ts_obj); js_set_slot(wrapper, SLOT_ENTRIES, ts_obj);
return obj;}
static ant_value_t tds_transform(ant_t *js, ant_value_t *args, int nargs) { ant_value_t wrapper = js_get_slot(js->current_func, SLOT_DATA); ant_value_t state_val = js_get_slot(wrapper, SLOT_DATA); td_state_t *st = (td_state_t *)(uintptr_t)(size_t)js_getnum(state_val); ant_value_t ctrl_obj = codec_transform_controller(args, nargs);
ant_value_t chunk = (nargs > 0) ? args[0] : js_mkundef(); const uint8_t *input = NULL; size_t input_len = 0;
if (!is_object_type(chunk) || !buffer_source_get_bytes(js, chunk, &input, &input_len)) return js_mkerr_typed(js, JS_ERR_TYPE, "The provided value is not of type '(ArrayBuffer or ArrayBufferView)'");
ant_value_t result = td_decode(js, st, input, input_len, true); if (is_err(result)) return result;
size_t slen = 0; const char *sval = js_getstr(js, result, &slen); if (sval && slen > 0) { if (!ts_is_controller(ctrl_obj)) return js_mkerr_typed(js, JS_ERR_TYPE, "Invalid TransformStreamDefaultController"); return ts_ctrl_enqueue(js, ctrl_obj, result); }
return js_mkundef();}
static ant_value_t tds_flush(ant_t *js, ant_value_t *args, int nargs) { ant_value_t wrapper = js_get_slot(js->current_func, SLOT_DATA); ant_value_t state_val = js_get_slot(wrapper, SLOT_DATA); td_state_t *st = (td_state_t *)(uintptr_t)(size_t)js_getnum(state_val); ant_value_t ctrl_obj = codec_flush_controller(args, nargs);
ant_value_t result = td_decode(js, st, NULL, 0, false); if (is_err(result)) return result;
size_t slen = 0; const char *sval = js_getstr(js, result, &slen); if (sval && slen > 0) { if (!ts_is_controller(ctrl_obj)) return js_mkerr_typed(js, JS_ERR_TYPE, "Invalid TransformStreamDefaultController"); return ts_ctrl_enqueue(js, ctrl_obj, result); }
return js_mkundef();}
static ant_value_t js_tds_get_encoding(ant_t *js, ant_value_t *args, int nargs) { ant_value_t state_val = js_get_slot(js->this_val, SLOT_DATA); td_state_t *st = (td_state_t *)(uintptr_t)(size_t)js_getnum(state_val); if (!st) return js_mkstr(js, "utf-8", 5); switch (st->encoding) { case TD_ENC_UTF16LE: return js_mkstr(js, "utf-16le", 8); case TD_ENC_UTF16BE: return js_mkstr(js, "utf-16be", 8); case TD_ENC_WINDOWS_1252: return js_mkstr(js, "windows-1252", 12); case TD_ENC_ISO_8859_2: return js_mkstr(js, "iso-8859-2", 10); default: return js_mkstr(js, "utf-8", 5); }}
static ant_value_t js_tds_get_fatal(ant_t *js, ant_value_t *args, int nargs) { ant_value_t state_val = js_get_slot(js->this_val, SLOT_DATA); td_state_t *st = (td_state_t *)(uintptr_t)(size_t)js_getnum(state_val); return (st && st->fatal) ? js_true : js_false;}
static ant_value_t js_tds_get_ignore_bom(ant_t *js, ant_value_t *args, int nargs) { ant_value_t state_val = js_get_slot(js->this_val, SLOT_DATA); td_state_t *st = (td_state_t *)(uintptr_t)(size_t)js_getnum(state_val); return (st && st->ignore_bom) ? js_true : js_false;}
static ant_value_t js_tds_get_readable(ant_t *js, ant_value_t *args, int nargs) { ant_value_t ts_obj = tds_get_ts(js->this_val); if (!ts_is_stream(ts_obj)) return js_mkerr_typed(js, JS_ERR_TYPE, "Invalid TextDecoderStream"); return ts_stream_readable(ts_obj);}
static ant_value_t js_tds_get_writable(ant_t *js, ant_value_t *args, int nargs) { ant_value_t ts_obj = tds_get_ts(js->this_val); if (!ts_is_stream(ts_obj)) return js_mkerr_typed(js, JS_ERR_TYPE, "Invalid TextDecoderStream"); return ts_stream_writable(ts_obj);}
static const char *tds_trim_label(const char *s, size_t len, size_t *out_len) { while (len > 0 && (unsigned char)*s <= 0x20) { s++; len--; } while (len > 0 && (unsigned char)s[len - 1] <= 0x20) { len--; } *out_len = len; return s;}
static int tds_resolve_encoding(const char *s, size_t len) { static const struct { const char *label; uint8_t label_len; td_encoding_t enc; } map[] = { {"unicode-1-1-utf-8", 17, TD_ENC_UTF8}, {"unicode11utf8", 13, TD_ENC_UTF8}, {"unicode20utf8", 13, TD_ENC_UTF8}, {"utf-8", 5, TD_ENC_UTF8}, {"utf8", 4, TD_ENC_UTF8}, {"x-unicode20utf8",17, TD_ENC_UTF8}, {"windows-1252", 12, TD_ENC_WINDOWS_1252}, {"ascii", 5, TD_ENC_WINDOWS_1252}, {"unicodefffe", 11, TD_ENC_UTF16BE}, {"utf-16be", 8, TD_ENC_UTF16BE}, {"csunicode", 9, TD_ENC_UTF16LE}, {"iso-10646-ucs-2",16, TD_ENC_UTF16LE}, {"ucs-2", 5, TD_ENC_UTF16LE}, {"unicode", 7, TD_ENC_UTF16LE}, {"unicodefeff", 11, TD_ENC_UTF16LE}, {"utf-16", 6, TD_ENC_UTF16LE}, {"utf-16le", 8, TD_ENC_UTF16LE}, {"iso-8859-2", 10, TD_ENC_ISO_8859_2}, {NULL, 0, 0} }; for (int i = 0; map[i].label; i++) { if (len == map[i].label_len && strncasecmp(s, map[i].label, len) == 0) return (int)map[i].enc; } return -1;}
static ant_value_t js_tds_ctor(ant_t *js, ant_value_t *args, int nargs) { if (vtype(js->new_target) == T_UNDEF) return js_mkerr_typed(js, JS_ERR_TYPE, "TextDecoderStream constructor requires 'new'");
td_encoding_t enc = TD_ENC_UTF8; if (nargs > 0 && !is_undefined(args[0])) { ant_value_t label = (vtype(args[0]) == T_STR) ? args[0] : coerce_to_str(js, args[0]); if (is_err(label)) return label; size_t llen; const char *raw = js_getstr(js, label, &llen); if (raw) { size_t tlen; const char *trimmed = tds_trim_label(raw, llen, &tlen); int resolved = tds_resolve_encoding(trimmed, tlen); if (resolved < 0) return js_mkerr_typed( js, JS_ERR_RANGE, "Failed to construct 'TextDecoderStream': The encoding label provided ('%.*s') is invalid.", (int)tlen, trimmed ); enc = (td_encoding_t)resolved; } }
bool fatal = false; bool ignore_bom = false; if (nargs > 1 && is_object_type(args[1])) { ant_value_t fv = js_getprop_fallback(js, args[1], "fatal"); if (is_err(fv)) return fv; if (vtype(fv) != T_UNDEF) fatal = js_truthy(js, fv); ant_value_t bv = js_getprop_fallback(js, args[1], "ignoreBOM"); if (is_err(bv)) return bv; if (vtype(bv) != T_UNDEF) ignore_bom = js_truthy(js, bv); }
td_state_t *st = td_state_new(enc, fatal, ignore_bom); if (!st) return js_mkerr(js, "out of memory");
ant_value_t obj = js_mkobj(js); ant_value_t proto = js_instance_proto_from_new_target(js, g_tds_proto); if (is_object_type(proto)) js_set_proto_init(obj, proto); js_set_slot(obj, SLOT_DATA, ANT_PTR(st)); js_set_finalizer(obj, tds_state_finalize);
ant_value_t wrapper = js_mkobj(js); js_set_slot(wrapper, SLOT_DATA, ANT_PTR(st));
ant_value_t transformer = js_mkobj(js); ant_value_t transform_fn = js_heavy_mkfun(js, tds_transform, wrapper); ant_value_t flush_fn = js_heavy_mkfun(js, tds_flush, wrapper); js_set(js, transformer, "transform", transform_fn); js_set(js, transformer, "flush", flush_fn);
ant_value_t ctor_args[1] = { transformer };
ant_value_t saved_new_target = js->new_target; ant_value_t saved_this = js->this_val; js->new_target = js_mknum(1);
ant_value_t ts_obj = js_ts_ctor(js, ctor_args, 1);
js->new_target = saved_new_target; js->this_val = saved_this;
if (is_err(ts_obj)) { free(st); return ts_obj; }
js_set_slot(obj, SLOT_ENTRIES, ts_obj); js_set_slot(wrapper, SLOT_ENTRIES, ts_obj);
return obj;}
void init_codec_stream_module(void) { ant_t *js = rt->js; ant_value_t g = js_glob(js);
g_tes_proto = js_mkobj(js); js_set_getter_desc(js, g_tes_proto, "encoding", 8, js_mkfun(js_tes_get_encoding), JS_DESC_C); js_set_getter_desc(js, g_tes_proto, "readable", 8, js_mkfun(js_tes_get_readable), JS_DESC_C); js_set_getter_desc(js, g_tes_proto, "writable", 8, js_mkfun(js_tes_get_writable), JS_DESC_C); js_set_sym(js, g_tes_proto, get_toStringTag_sym(), js_mkstr(js, "TextEncoderStream", 17));
ant_value_t tes_ctor = js_make_ctor(js, js_tes_ctor, g_tes_proto, "TextEncoderStream", 17); js_set(js, g, "TextEncoderStream", tes_ctor); js_set_descriptor(js, g, "TextEncoderStream", 17, JS_DESC_W | JS_DESC_C);
g_tds_proto = js_mkobj(js); js_set_getter_desc(js, g_tds_proto, "encoding", 8, js_mkfun(js_tds_get_encoding), JS_DESC_C); js_set_getter_desc(js, g_tds_proto, "fatal", 5, js_mkfun(js_tds_get_fatal), JS_DESC_C); js_set_getter_desc(js, g_tds_proto, "ignoreBOM", 9, js_mkfun(js_tds_get_ignore_bom), JS_DESC_C); js_set_getter_desc(js, g_tds_proto, "readable", 8, js_mkfun(js_tds_get_readable), JS_DESC_C); js_set_getter_desc(js, g_tds_proto, "writable", 8, js_mkfun(js_tds_get_writable), JS_DESC_C); js_set_sym(js, g_tds_proto, get_toStringTag_sym(), js_mkstr(js, "TextDecoderStream", 17));
ant_value_t tds_ctor = js_make_ctor(js, js_tds_ctor, g_tds_proto, "TextDecoderStream", 17); js_set(js, g, "TextDecoderStream", tds_ctor); js_set_descriptor(js, g, "TextDecoderStream", 17, JS_DESC_W | JS_DESC_C);}
void gc_mark_codec_streams(ant_t *js, void (*mark)(ant_t *, ant_value_t)) { mark(js, g_tes_proto); mark(js, g_tds_proto);}