quickjs-done/qjs.c

804 lines
25 KiB
C
Raw Normal View History

2020-09-06 16:53:08 +00:00
/*
* QuickJS stand alone interpreter
2025-01-06 20:48:15 +00:00
*
2021-03-27 10:17:31 +00:00
* Copyright (c) 2017-2021 Fabrice Bellard
* Copyright (c) 2017-2021 Charlie Gordon
2020-09-06 16:53:08 +00:00
*
* Permission is hereby granted, free of charge, to any person obtaining a copy
* of this software and associated documentation files (the "Software"), to deal
* in the Software without restriction, including without limitation the rights
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
* copies of the Software, and to permit persons to whom the Software is
* furnished to do so, subject to the following conditions:
*
* The above copyright notice and this permission notice shall be included in
* all copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
* THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
* THE SOFTWARE.
*/
#include <stdlib.h>
#include <stdio.h>
#include <stdarg.h>
#include <inttypes.h>
#include <string.h>
#include <assert.h>
2025-01-06 20:48:15 +00:00
#if !defined(_MSC_VER)
2020-09-06 16:53:08 +00:00
#include <unistd.h>
2025-01-06 20:48:15 +00:00
#endif
2020-09-06 16:53:08 +00:00
#include <errno.h>
#include <fcntl.h>
#include <time.h>
#include "cutils.h"
2025-01-06 20:48:15 +00:00
#include "quickjs.h"
2020-09-06 16:53:08 +00:00
#include "quickjs-libc.h"
2025-01-06 20:48:15 +00:00
#ifdef QJS_USE_MIMALLOC
#include <mimalloc.h>
#endif
2020-09-06 16:53:08 +00:00
extern const uint8_t qjsc_repl[];
extern const uint32_t qjsc_repl_size;
2025-01-06 20:48:15 +00:00
extern const uint8_t qjsc_standalone[];
extern const uint32_t qjsc_standalone_size;
// Must match standalone.js
#define TRAILER_SIZE 12
static const char trailer_magic[] = "quickjs2";
static const int trailer_magic_size = sizeof(trailer_magic) - 1;
static const int trailer_size = TRAILER_SIZE;
static int qjs__argc;
static char **qjs__argv;
static BOOL is_standalone(const char *exe)
{
FILE *exe_f = fopen(exe, "rb");
if (!exe_f)
return FALSE;
if (fseek(exe_f, -trailer_size, SEEK_END) < 0)
goto fail;
uint8_t buf[TRAILER_SIZE];
if (fread(buf, 1, trailer_size, exe_f) != trailer_size)
goto fail;
fclose(exe_f);
return !memcmp(buf, trailer_magic, trailer_magic_size);
fail:
fclose(exe_f);
return FALSE;
}
static JSValue load_standalone_module(JSContext *ctx)
{
JSModuleDef *m;
JSValue obj, val;
obj = JS_ReadObject(ctx, qjsc_standalone, qjsc_standalone_size, JS_READ_OBJ_BYTECODE);
if (JS_IsException(obj))
goto exception;
assert(JS_VALUE_GET_TAG(obj) == JS_TAG_MODULE);
if (JS_ResolveModule(ctx, obj) < 0) {
JS_FreeValue(ctx, obj);
goto exception;
}
js_module_set_import_meta(ctx, obj, FALSE, TRUE);
val = JS_EvalFunction(ctx, JS_DupValue(ctx, obj));
val = js_std_await(ctx, val);
if (JS_IsException(val)) {
JS_FreeValue(ctx, obj);
exception:
js_std_dump_error(ctx);
exit(1);
}
JS_FreeValue(ctx, val);
m = JS_VALUE_GET_PTR(obj);
JS_FreeValue(ctx, obj);
return JS_GetModuleNamespace(ctx, m);
}
2020-09-06 16:53:08 +00:00
static int eval_buf(JSContext *ctx, const void *buf, int buf_len,
const char *filename, int eval_flags)
{
JSValue val;
int ret;
if ((eval_flags & JS_EVAL_TYPE_MASK) == JS_EVAL_TYPE_MODULE) {
/* for the modules, we compile then run to be able to set
import.meta */
val = JS_Eval(ctx, buf, buf_len, filename,
eval_flags | JS_EVAL_FLAG_COMPILE_ONLY);
if (!JS_IsException(val)) {
js_module_set_import_meta(ctx, val, TRUE, TRUE);
val = JS_EvalFunction(ctx, val);
}
2025-01-06 20:48:15 +00:00
val = js_std_await(ctx, val);
2020-09-06 16:53:08 +00:00
} else {
val = JS_Eval(ctx, buf, buf_len, filename, eval_flags);
}
if (JS_IsException(val)) {
js_std_dump_error(ctx);
ret = -1;
} else {
ret = 0;
}
JS_FreeValue(ctx, val);
return ret;
}
static int eval_file(JSContext *ctx, const char *filename, int module)
{
uint8_t *buf;
int ret, eval_flags;
size_t buf_len;
2025-01-06 20:48:15 +00:00
2020-09-06 16:53:08 +00:00
buf = js_load_file(ctx, &buf_len, filename);
if (!buf) {
perror(filename);
exit(1);
}
if (module < 0) {
2025-01-06 20:48:15 +00:00
module = (js__has_suffix(filename, ".mjs") ||
2020-09-06 16:53:08 +00:00
JS_DetectModule((const char *)buf, buf_len));
}
if (module)
eval_flags = JS_EVAL_TYPE_MODULE;
else
eval_flags = JS_EVAL_TYPE_GLOBAL;
2024-10-25 18:03:27 +00:00
2025-01-06 20:48:15 +00:00
//POLYFILLS FOR QJS FILES BEGIN
2024-10-25 18:03:27 +00:00
const char *pf = "globalThis.global = globalThis;\n"
"global.console.error = console.log\n"
"global.console.warn = console.log\n"
"globalThis.breakFunction = () => { throw new Error('Function Break'); };\n"
"\n"
"if (typeof os !== 'undefined') {\n"
" globalThis.sleep = os.sleep;\n"
" async function setTimeout2(func, ms) {globalThis.clearTimeout = false; await sleep(ms); if (!clearTimeout) { func(); } }\n"
" globalThis.setTimeout = setTimeout2\n"
"} else {\n"
" console.error('os is not defined.');\n"
"}\n"
"\n"
"if (typeof std !== 'undefined') {\n"
" globalThis.urlGet = std.urlGet;\n"
" globalThis.loadFile = std.loadFile;\n"
" globalThis.printf = console.log;\n"
" globalThis.evalFile = std.loadScript;\n"
2025-01-06 20:48:15 +00:00
//" globalThis.require = std.loadScript;\n"
2024-10-25 18:03:27 +00:00
" globalThis.getURL = std.urlGet;\n"
"} else {\n"
" console.error('std is not defined.');\n"
"}\n";
eval_buf(ctx, pf, strlen(pf), "<input>", JS_EVAL_TYPE_MODULE);
2020-09-06 16:53:08 +00:00
ret = eval_buf(ctx, buf, buf_len, filename, eval_flags);
js_free(ctx, buf);
return ret;
}
2025-01-06 20:48:15 +00:00
static int64_t parse_limit(const char *arg) {
char *p;
unsigned long unit = 1024; /* default to traditional KB */
double d = strtod(arg, &p);
if (p == arg) {
fprintf(stderr, "Invalid limit: %s\n", arg);
return -1;
}
if (*p) {
switch (*p++) {
case 'b': case 'B': unit = 1UL << 0; break;
case 'k': case 'K': unit = 1UL << 10; break; /* IEC kibibytes */
case 'm': case 'M': unit = 1UL << 20; break; /* IEC mebibytes */
case 'g': case 'G': unit = 1UL << 30; break; /* IEC gigibytes */
default:
fprintf(stderr, "Invalid limit: %s, unrecognized suffix, only k,m,g are allowed\n", arg);
return -1;
}
if (*p) {
fprintf(stderr, "Invalid limit: %s, only one suffix allowed\n", arg);
return -1;
}
}
return (int64_t)(d * unit);
}
static JSValue js_gc(JSContext *ctx, JSValue this_val,
int argc, JSValue *argv)
{
JS_RunGC(JS_GetRuntime(ctx));
return JS_UNDEFINED;
}
static JSValue js_navigator_get_userAgent(JSContext *ctx, JSValue this_val)
{
char version[32];
snprintf(version, sizeof(version), "quickjs-ng/%s", JS_GetVersion());
return JS_NewString(ctx, version);
}
static const JSCFunctionListEntry navigator_proto_funcs[] = {
JS_CGETSET_DEF2("userAgent", js_navigator_get_userAgent, NULL, JS_PROP_CONFIGURABLE | JS_PROP_ENUMERABLE),
JS_PROP_STRING_DEF("[Symbol.toStringTag]", "Navigator", JS_PROP_CONFIGURABLE),
};
static const JSCFunctionListEntry global_obj[] = {
JS_CFUNC_DEF("gc", 0, js_gc),
};
2020-09-06 17:10:15 +00:00
/* also used to initialize the worker context */
static JSContext *JS_NewCustomContext(JSRuntime *rt)
{
JSContext *ctx;
ctx = JS_NewContext(rt);
if (!ctx)
return NULL;
/* system modules */
2025-01-06 20:48:15 +00:00
js_init_module_std(ctx, "qjs:std");
js_init_module_os(ctx, "qjs:os");
js_init_module_bjson(ctx, "qjs:bjson");
JSValue global = JS_GetGlobalObject(ctx);
JS_SetPropertyFunctionList(ctx, global, global_obj, countof(global_obj));
JSValue args = JS_NewArray(ctx);
int i;
for(i = 0; i < qjs__argc; i++) {
JS_SetPropertyUint32(ctx, args, i, JS_NewString(ctx, qjs__argv[i]));
}
JS_SetPropertyStr(ctx, global, "execArgv", args);
JS_SetPropertyStr(ctx, global, "argv0", JS_NewString(ctx, qjs__argv[0]));
JSValue navigator_proto = JS_NewObject(ctx);
JS_SetPropertyFunctionList(ctx, navigator_proto, navigator_proto_funcs, countof(navigator_proto_funcs));
JSValue navigator = JS_NewObjectProto(ctx, navigator_proto);
JS_DefinePropertyValueStr(ctx, global, "navigator", navigator, JS_PROP_CONFIGURABLE | JS_PROP_ENUMERABLE);
JS_FreeValue(ctx, global);
JS_FreeValue(ctx, navigator_proto);
2020-09-06 17:10:15 +00:00
return ctx;
}
2020-09-06 16:53:08 +00:00
struct trace_malloc_data {
uint8_t *base;
};
static inline unsigned long long js_trace_malloc_ptr_offset(uint8_t *ptr,
struct trace_malloc_data *dp)
{
return ptr - dp->base;
}
2021-03-27 10:17:31 +00:00
static void
2025-01-06 20:48:15 +00:00
#if defined(_WIN32) && !defined(__clang__)
2021-03-27 10:17:31 +00:00
/* mingw printf is used */
__attribute__((format(gnu_printf, 2, 3)))
#else
__attribute__((format(printf, 2, 3)))
#endif
2025-01-06 20:48:15 +00:00
js_trace_malloc_printf(void *opaque, const char *fmt, ...)
2020-09-06 16:53:08 +00:00
{
va_list ap;
int c;
va_start(ap, fmt);
while ((c = *fmt++) != '\0') {
if (c == '%') {
/* only handle %p and %zd */
if (*fmt == 'p') {
uint8_t *ptr = va_arg(ap, void *);
if (ptr == NULL) {
printf("NULL");
} else {
printf("H%+06lld.%zd",
2025-01-06 20:48:15 +00:00
js_trace_malloc_ptr_offset(ptr, opaque),
js__malloc_usable_size(ptr));
2020-09-06 16:53:08 +00:00
}
fmt++;
continue;
}
if (fmt[0] == 'z' && fmt[1] == 'd') {
size_t sz = va_arg(ap, size_t);
printf("%zd", sz);
fmt += 2;
continue;
}
}
putc(c, stdout);
}
va_end(ap);
}
static void js_trace_malloc_init(struct trace_malloc_data *s)
{
free(s->base = malloc(8));
}
2025-01-06 20:48:15 +00:00
static void *js_trace_calloc(void *opaque, size_t count, size_t size)
2020-09-06 16:53:08 +00:00
{
void *ptr;
2025-01-06 20:48:15 +00:00
ptr = calloc(count, size);
js_trace_malloc_printf(opaque, "C %zd %zd -> %p\n", count, size, ptr);
return ptr;
}
2020-09-06 16:53:08 +00:00
2025-01-06 20:48:15 +00:00
static void *js_trace_malloc(void *opaque, size_t size)
{
void *ptr;
2020-09-06 16:53:08 +00:00
ptr = malloc(size);
2025-01-06 20:48:15 +00:00
js_trace_malloc_printf(opaque, "A %zd -> %p\n", size, ptr);
2020-09-06 16:53:08 +00:00
return ptr;
}
2025-01-06 20:48:15 +00:00
static void js_trace_free(void *opaque, void *ptr)
2020-09-06 16:53:08 +00:00
{
if (!ptr)
return;
2025-01-06 20:48:15 +00:00
js_trace_malloc_printf(opaque, "F %p\n", ptr);
2020-09-06 16:53:08 +00:00
free(ptr);
}
2025-01-06 20:48:15 +00:00
static void *js_trace_realloc(void *opaque, void *ptr, size_t size)
2020-09-06 16:53:08 +00:00
{
2025-01-06 20:48:15 +00:00
js_trace_malloc_printf(opaque, "R %zd %p", size, ptr);
2020-09-06 16:53:08 +00:00
ptr = realloc(ptr, size);
2025-01-06 20:48:15 +00:00
js_trace_malloc_printf(opaque, " -> %p\n", ptr);
2020-09-06 16:53:08 +00:00
return ptr;
}
static const JSMallocFunctions trace_mf = {
2025-01-06 20:48:15 +00:00
js_trace_calloc,
2020-09-06 16:53:08 +00:00
js_trace_malloc,
js_trace_free,
js_trace_realloc,
2025-01-06 20:48:15 +00:00
js__malloc_usable_size
2020-09-06 16:53:08 +00:00
};
2025-01-06 20:48:15 +00:00
#ifdef QJS_USE_MIMALLOC
static void *js_mi_calloc(void *opaque, size_t count, size_t size)
{
return mi_calloc(count, size);
}
static void *js_mi_malloc(void *opaque, size_t size)
{
return mi_malloc(size);
}
static void js_mi_free(void *opaque, void *ptr)
{
if (!ptr)
return;
mi_free(ptr);
}
static void *js_mi_realloc(void *opaque, void *ptr, size_t size)
{
return mi_realloc(ptr, size);
}
static const JSMallocFunctions mi_mf = {
js_mi_calloc,
js_mi_malloc,
js_mi_free,
js_mi_realloc,
mi_malloc_usable_size
};
#endif
2020-09-06 16:53:08 +00:00
#define PROG_NAME "qjs"
void help(void)
{
2025-01-06 20:48:15 +00:00
printf("QuickJS-ng DoneJS Edition version %s\n"
2020-09-06 16:57:11 +00:00
"usage: " PROG_NAME " [options] [file [args]]\n"
2020-09-06 16:53:08 +00:00
"-h --help list options\n"
"-e --eval EXPR evaluate EXPR\n"
"-i --interactive go to interactive mode\n"
"-m --module load as ES6 module (default=autodetect)\n"
" --script load as ES6 script (default=autodetect)\n"
2020-09-06 16:57:11 +00:00
"-I --include file include an additional file\n"
2025-01-06 20:48:15 +00:00
" --std make 'std', 'os' and 'bjson' available to script\n"
2020-09-06 16:53:08 +00:00
"-T --trace trace memory allocation\n"
"-d --dump dump the memory usage stats\n"
2025-01-06 20:48:15 +00:00
"-D --dump-flags flags for dumping debug data (see DUMP_* defines)\n"
"-c --compile FILE compile the given JS file as a standalone executable\n"
"-o --out FILE output file for standalone executables\n"
" --exe select the executable to use as the base, defaults to the current one\n"
" --memory-limit n limit the memory usage to 'n' Kbytes\n"
" --stack-size n limit the stack size to 'n' Kbytes\n"
2020-09-06 16:53:08 +00:00
" --unhandled-rejection dump unhandled promise rejections\n"
2025-01-06 20:48:15 +00:00
"-q --quit just instantiate the interpreter and quit\n", JS_GetVersion());
2020-09-06 16:53:08 +00:00
exit(1);
}
int main(int argc, char **argv)
{
JSRuntime *rt;
JSContext *ctx;
2025-01-06 20:48:15 +00:00
JSValue ret = JS_UNDEFINED;
2020-09-06 16:53:08 +00:00
struct trace_malloc_data trace_data = { NULL };
2025-01-06 20:48:15 +00:00
int optind = 1;
char *compile_file = NULL;
char *exe = NULL;
2020-09-06 16:53:08 +00:00
char *expr = NULL;
2025-01-06 20:48:15 +00:00
char *dump_flags_str = NULL;
char *out = NULL;
int standalone = 0;
2020-09-06 16:53:08 +00:00
int interactive = 0;
int dump_memory = 0;
2025-01-06 20:48:15 +00:00
int dump_flags = 0;
2020-09-06 16:53:08 +00:00
int trace_memory = 0;
int empty_run = 0;
int module = -1;
2025-01-06 20:48:15 +00:00
int load_std = 0;
2020-09-06 16:53:08 +00:00
int dump_unhandled_promise_rejection = 0;
2020-09-06 16:57:11 +00:00
char *include_list[32];
int i, include_count = 0;
2025-01-06 20:48:15 +00:00
int64_t memory_limit = -1;
int64_t stack_size = -1;
/* save for later */
qjs__argc = argc;
qjs__argv = argv;
if (is_standalone(argv[0])) {
standalone = 1;
goto start;
2020-09-06 16:53:08 +00:00
}
2025-01-06 20:48:15 +00:00
dump_flags_str = getenv("QJS_DUMP_FLAGS");
dump_flags = dump_flags_str ? strtol(dump_flags_str, NULL, 16) : 0;
2020-09-06 16:53:08 +00:00
/* cannot use getopt because we want to pass the command line to
the script */
while (optind < argc && *argv[optind] == '-') {
char *arg = argv[optind] + 1;
const char *longopt = "";
2025-01-06 20:48:15 +00:00
char *opt_arg = NULL;
2020-09-06 16:53:08 +00:00
/* a single - is not an option, it also stops argument scanning */
if (!*arg)
break;
optind++;
if (*arg == '-') {
longopt = arg + 1;
2025-01-06 20:48:15 +00:00
opt_arg = strchr(longopt, '=');
if (opt_arg)
*opt_arg++ = '\0';
2020-09-06 16:53:08 +00:00
arg += strlen(arg);
/* -- stops argument scanning */
if (!*longopt)
break;
}
for (; *arg || *longopt; longopt = "") {
char opt = *arg;
2025-01-06 20:48:15 +00:00
if (opt) {
2020-09-06 16:53:08 +00:00
arg++;
2025-01-06 20:48:15 +00:00
if (!opt_arg && *arg)
opt_arg = arg;
}
2020-09-06 16:53:08 +00:00
if (opt == 'h' || opt == '?' || !strcmp(longopt, "help")) {
help();
continue;
}
if (opt == 'e' || !strcmp(longopt, "eval")) {
2025-01-06 20:48:15 +00:00
if (!opt_arg) {
if (optind >= argc) {
fprintf(stderr, "qjs: missing expression for -e\n");
exit(1);
}
opt_arg = argv[optind++];
2020-09-06 16:53:08 +00:00
}
2025-01-06 20:48:15 +00:00
expr = opt_arg;
break;
2020-09-06 16:53:08 +00:00
}
2020-09-06 16:57:11 +00:00
if (opt == 'I' || !strcmp(longopt, "include")) {
if (optind >= argc) {
fprintf(stderr, "expecting filename");
exit(1);
}
if (include_count >= countof(include_list)) {
fprintf(stderr, "too many included files");
exit(1);
}
include_list[include_count++] = argv[optind++];
continue;
}
2020-09-06 16:53:08 +00:00
if (opt == 'i' || !strcmp(longopt, "interactive")) {
interactive++;
continue;
}
if (opt == 'm' || !strcmp(longopt, "module")) {
module = 1;
continue;
}
if (!strcmp(longopt, "script")) {
module = 0;
continue;
}
if (opt == 'd' || !strcmp(longopt, "dump")) {
dump_memory++;
continue;
}
2025-01-06 20:48:15 +00:00
if (opt == 'D' || !strcmp(longopt, "dump-flags")) {
dump_flags = opt_arg ? strtol(opt_arg, NULL, 16) : 0;
break;
}
2020-09-06 16:53:08 +00:00
if (opt == 'T' || !strcmp(longopt, "trace")) {
trace_memory++;
continue;
}
2025-01-06 20:48:15 +00:00
if (!strcmp(longopt, "std")) {
load_std = 1;
2020-09-06 16:53:08 +00:00
continue;
}
2025-01-06 20:48:15 +00:00
if (!strcmp(longopt, "unhandled-rejection")) {
dump_unhandled_promise_rejection = 1;
2020-09-06 16:53:08 +00:00
continue;
}
if (opt == 'q' || !strcmp(longopt, "quit")) {
empty_run++;
continue;
}
if (!strcmp(longopt, "memory-limit")) {
2025-01-06 20:48:15 +00:00
if (!opt_arg) {
if (optind >= argc) {
fprintf(stderr, "expecting memory limit");
exit(1);
}
opt_arg = argv[optind++];
2020-09-06 16:53:08 +00:00
}
2025-01-06 20:48:15 +00:00
memory_limit = parse_limit(opt_arg);
break;
2020-09-06 16:53:08 +00:00
}
2020-09-06 17:07:30 +00:00
if (!strcmp(longopt, "stack-size")) {
2025-01-06 20:48:15 +00:00
if (!opt_arg) {
if (optind >= argc) {
fprintf(stderr, "expecting stack size");
exit(1);
}
opt_arg = argv[optind++];
2020-09-06 17:07:30 +00:00
}
2025-01-06 20:48:15 +00:00
stack_size = parse_limit(opt_arg);
break;
}
if (opt == 'c' || !strcmp(longopt, "compile")) {
if (!opt_arg) {
if (optind >= argc) {
fprintf(stderr, "qjs: missing file for -c\n");
exit(1);
}
opt_arg = argv[optind++];
}
compile_file = opt_arg;
break;
}
if (opt == 'o' || !strcmp(longopt, "out")) {
if (!opt_arg) {
if (optind >= argc) {
fprintf(stderr, "qjs: missing file for -o\n");
exit(1);
}
opt_arg = argv[optind++];
}
out = opt_arg;
break;
}
if (!strcmp(longopt, "exe")) {
if (!opt_arg) {
if (optind >= argc) {
fprintf(stderr, "qjs: missing file for --exe\n");
exit(1);
}
opt_arg = argv[optind++];
}
exe = opt_arg;
break;
2020-09-06 17:07:30 +00:00
}
2020-09-06 16:53:08 +00:00
if (opt) {
fprintf(stderr, "qjs: unknown option '-%c'\n", opt);
} else {
fprintf(stderr, "qjs: unknown option '--%s'\n", longopt);
}
help();
}
}
2025-01-06 20:48:15 +00:00
if (compile_file && !out)
help();
start:
2020-09-06 17:10:15 +00:00
2020-09-06 16:53:08 +00:00
if (trace_memory) {
js_trace_malloc_init(&trace_data);
rt = JS_NewRuntime2(&trace_mf, &trace_data);
} else {
2025-01-06 20:48:15 +00:00
#ifdef QJS_USE_MIMALLOC
rt = JS_NewRuntime2(&mi_mf, NULL);
#else
2020-09-06 16:53:08 +00:00
rt = JS_NewRuntime();
2025-01-06 20:48:15 +00:00
#endif
2020-09-06 16:53:08 +00:00
}
if (!rt) {
fprintf(stderr, "qjs: cannot allocate JS runtime\n");
exit(2);
}
2025-01-06 20:48:15 +00:00
if (memory_limit >= 0)
JS_SetMemoryLimit(rt, (size_t)memory_limit);
if (stack_size >= 0)
JS_SetMaxStackSize(rt, (size_t)stack_size);
if (dump_flags != 0)
JS_SetDumpFlags(rt, dump_flags);
2020-09-06 17:10:15 +00:00
js_std_set_worker_new_context_func(JS_NewCustomContext);
2020-09-06 17:07:30 +00:00
js_std_init_handlers(rt);
2020-09-06 17:10:15 +00:00
ctx = JS_NewCustomContext(rt);
2020-09-06 16:53:08 +00:00
if (!ctx) {
fprintf(stderr, "qjs: cannot allocate JS context\n");
exit(2);
}
/* loader for ES6 modules */
JS_SetModuleLoaderFunc(rt, NULL, js_module_loader, NULL);
if (dump_unhandled_promise_rejection) {
JS_SetHostPromiseRejectionTracker(rt, js_std_promise_rejection_tracker,
NULL);
}
2025-01-06 20:48:15 +00:00
2020-09-06 16:53:08 +00:00
if (!empty_run) {
js_std_add_helpers(ctx, argc - optind, argv + optind);
2025-01-06 20:48:15 +00:00
/* make 'std' and 'os' visible to non module code */
if (load_std) {
const char *str =
"import * as bjson from 'qjs:bjson';\n"
"import * as std from 'qjs:std';\n"
"import * as os from 'qjs:os';\n"
"globalThis.bjson = bjson;\n"
2020-09-06 16:53:08 +00:00
"globalThis.std = std;\n"
"globalThis.os = os;\n";
eval_buf(ctx, str, strlen(str), "<input>", JS_EVAL_TYPE_MODULE);
2025-01-06 20:48:15 +00:00
}
2024-10-24 20:43:37 +00:00
2020-09-06 16:57:11 +00:00
for(i = 0; i < include_count; i++) {
2025-01-06 20:48:15 +00:00
if (eval_file(ctx, include_list[i], 0))
2020-09-06 16:57:11 +00:00
goto fail;
}
2025-01-06 20:48:15 +00:00
if (standalone) {
JSValue ns = load_standalone_module(ctx);
if (JS_IsException(ns))
goto fail;
JSValue func = JS_GetPropertyStr(ctx, ns, "runStandalone");
JS_FreeValue(ctx, ns);
if (JS_IsException(func))
goto fail;
ret = JS_Call(ctx, func, JS_UNDEFINED, 0, NULL);
JS_FreeValue(ctx, func);
} else if (compile_file) {
JSValue ns = load_standalone_module(ctx);
if (JS_IsException(ns))
goto fail;
JSValue func = JS_GetPropertyStr(ctx, ns, "compileStandalone");
JS_FreeValue(ctx, ns);
if (JS_IsException(func))
goto fail;
JSValue args[3];
args[0] = JS_NewString(ctx, compile_file);
args[1] = JS_NewString(ctx, out);
args[2] = JS_NewString(ctx, exe != NULL ? exe : argv[0]);
ret = JS_Call(ctx, func, JS_UNDEFINED, countof(args), args);
JS_FreeValue(ctx, func);
JS_FreeValue(ctx, args[0]);
JS_FreeValue(ctx, args[1]);
JS_FreeValue(ctx, args[2]);
} else if (expr) {
2020-09-06 16:53:08 +00:00
if (eval_buf(ctx, expr, strlen(expr), "<cmdline>", 0))
goto fail;
2025-01-06 20:48:15 +00:00
} else if (optind >= argc) {
2020-09-06 16:53:08 +00:00
/* interactive mode */
interactive = 1;
} else {
const char *filename;
filename = argv[optind];
if (eval_file(ctx, filename, module))
goto fail;
}
if (interactive) {
js_std_eval_binary(ctx, qjsc_repl, qjsc_repl_size, 0);
}
//POLYFILLS FOR QJS REPL BEGIN
2024-10-25 18:03:27 +00:00
const char *pf = "globalThis.global = globalThis;\n"
"global.console.error = console.log\n"
"global.console.warn = console.log\n"
2024-10-25 18:03:27 +00:00
"globalThis.breakFunction = () => { throw new Error('Function Break'); };\n"
"\n"
"if (typeof os !== 'undefined') {\n"
" globalThis.sleep = os.sleep;\n"
2024-10-25 16:34:47 +00:00
" async function setTimeout2(func, ms) {globalThis.clearTimeout = false; await sleep(ms); if (!clearTimeout) { func(); } }\n"
" globalThis.setTimeout = setTimeout2\n"
"} else {\n"
" console.error('os is not defined.');\n"
"}\n"
"\n"
"if (typeof std !== 'undefined') {\n"
" globalThis.urlGet = std.urlGet;\n"
" globalThis.loadFile = std.loadFile;\n"
2024-10-24 21:22:27 +00:00
" globalThis.printf = console.log;\n"
" globalThis.evalFile = std.loadScript;\n"
2025-01-06 20:48:15 +00:00
// " globalThis.require = std.loadScript;\n"
" globalThis.getURL = std.urlGet;\n"
"} else {\n"
" console.error('std is not defined.');\n"
"}\n";
eval_buf(ctx, pf, strlen(pf), "<input>", JS_EVAL_TYPE_MODULE);
2025-01-06 20:48:15 +00:00
if (standalone || compile_file) {
if (JS_IsException(ret)) {
ret = JS_GetException(ctx);
} else {
JS_FreeValue(ctx, ret);
ret = js_std_loop(ctx);
}
} else {
ret = js_std_loop(ctx);
}
if (!JS_IsUndefined(ret)) {
js_std_dump_error1(ctx, ret);
JS_FreeValue(ctx, ret);
goto fail;
}
2020-09-06 16:53:08 +00:00
}
2025-01-06 20:48:15 +00:00
2020-09-06 16:53:08 +00:00
if (dump_memory) {
JSMemoryUsage stats;
JS_ComputeMemoryUsage(rt, &stats);
JS_DumpMemoryUsage(stdout, &stats, rt);
}
js_std_free_handlers(rt);
JS_FreeContext(ctx);
JS_FreeRuntime(rt);
if (empty_run && dump_memory) {
clock_t t[5];
2025-01-06 20:48:15 +00:00
double best[5] = {0};
2020-09-06 16:53:08 +00:00
int i, j;
for (i = 0; i < 100; i++) {
t[0] = clock();
rt = JS_NewRuntime();
t[1] = clock();
ctx = JS_NewContext(rt);
t[2] = clock();
JS_FreeContext(ctx);
t[3] = clock();
JS_FreeRuntime(rt);
t[4] = clock();
for (j = 4; j > 0; j--) {
double ms = 1000.0 * (t[j] - t[j - 1]) / CLOCKS_PER_SEC;
if (i == 0 || best[j] > ms)
best[j] = ms;
}
}
printf("\nInstantiation times (ms): %.3f = %.3f+%.3f+%.3f+%.3f\n",
best[1] + best[2] + best[3] + best[4],
best[1], best[2], best[3], best[4]);
}
return 0;
fail:
js_std_free_handlers(rt);
JS_FreeContext(ctx);
JS_FreeRuntime(rt);
return 1;
}