|
|
|
/*
|
|
|
|
* Command line execution tool. Useful for test cases and manual testing.
|
|
|
|
*
|
|
|
|
* To enable readline and other fancy stuff, compile with -DDUK_CMDLINE_FANCY.
|
|
|
|
* It is not the default to maximize portability. You can also compile in
|
|
|
|
* support for example allocators, grep for DUK_CMDLINE_*.
|
|
|
|
*/
|
|
|
|
|
|
|
|
#ifndef DUK_CMDLINE_FANCY
|
|
|
|
#define NO_READLINE
|
|
|
|
#define NO_RLIMIT
|
|
|
|
#define NO_SIGNAL
|
|
|
|
#endif
|
|
|
|
|
|
|
|
#define GREET_CODE(variant) \
|
|
|
|
"print('((o) Duktape" variant " ' + " \
|
|
|
|
"Math.floor(Duktape.version / 10000) + '.' + " \
|
|
|
|
"Math.floor(Duktape.version / 100) % 100 + '.' + " \
|
|
|
|
"Duktape.version % 100" \
|
|
|
|
", '(" DUK_GIT_DESCRIBE ")');"
|
|
|
|
|
|
|
|
#include <stdio.h>
|
|
|
|
#include <stdlib.h>
|
|
|
|
#include <string.h>
|
|
|
|
#ifndef NO_SIGNAL
|
|
|
|
#include <signal.h>
|
|
|
|
#endif
|
|
|
|
#ifndef NO_RLIMIT
|
|
|
|
#include <sys/resource.h>
|
|
|
|
#endif
|
|
|
|
#ifndef NO_READLINE
|
|
|
|
#include <readline/readline.h>
|
|
|
|
#include <readline/history.h>
|
|
|
|
#endif
|
|
|
|
#ifdef DUK_CMDLINE_ALLOC_LOGGING
|
|
|
|
#include "duk_alloc_logging.h"
|
|
|
|
#endif
|
|
|
|
#ifdef DUK_CMDLINE_ALLOC_TORTURE
|
|
|
|
#include "duk_alloc_torture.h"
|
|
|
|
#endif
|
|
|
|
#ifdef DUK_CMDLINE_ALLOC_HYBRID
|
|
|
|
#include "duk_alloc_hybrid.h"
|
|
|
|
#endif
|
|
|
|
#include "duktape.h"
|
|
|
|
|
|
|
|
#define MEM_LIMIT_NORMAL (128*1024*1024) /* 128 MB */
|
|
|
|
#define MEM_LIMIT_HIGH (2047*1024*1024) /* ~2 GB */
|
|
|
|
#define LINEBUF_SIZE 65536
|
|
|
|
|
|
|
|
static int interactive_mode = 0;
|
|
|
|
|
|
|
|
#ifndef NO_RLIMIT
|
|
|
|
static void set_resource_limits(rlim_t mem_limit_value) {
|
|
|
|
int rc;
|
|
|
|
struct rlimit lim;
|
|
|
|
|
|
|
|
rc = getrlimit(RLIMIT_AS, &lim);
|
|
|
|
if (rc != 0) {
|
|
|
|
fprintf(stderr, "Warning: cannot read RLIMIT_AS\n");
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (lim.rlim_max < mem_limit_value) {
|
|
|
|
fprintf(stderr, "Warning: rlim_max < mem_limit_value (%d < %d)\n", (int) lim.rlim_max, (int) mem_limit_value);
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
lim.rlim_cur = mem_limit_value;
|
|
|
|
lim.rlim_max = mem_limit_value;
|
|
|
|
|
|
|
|
rc = setrlimit(RLIMIT_AS, &lim);
|
|
|
|
if (rc != 0) {
|
|
|
|
fprintf(stderr, "Warning: setrlimit failed\n");
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
#if 0
|
|
|
|
fprintf(stderr, "Set RLIMIT_AS to %d\n", (int) mem_limit_value);
|
|
|
|
#endif
|
|
|
|
}
|
|
|
|
#endif /* NO_RLIMIT */
|
|
|
|
|
|
|
|
#ifndef NO_SIGNAL
|
|
|
|
static void my_sighandler(int x) {
|
|
|
|
fprintf(stderr, "Got signal %d\n", x);
|
|
|
|
|
|
|
|
}
|
|
|
|
static void set_sigint_handler(void) {
|
|
|
|
(void) signal(SIGINT, my_sighandler);
|
|
|
|
}
|
|
|
|
#endif /* NO_SIGNAL */
|
|
|
|
|
|
|
|
static int get_stack_raw(duk_context *ctx) {
|
|
|
|
if (!duk_is_object(ctx, -1)) {
|
|
|
|
return 1;
|
|
|
|
}
|
|
|
|
if (!duk_has_prop_string(ctx, -1, "stack")) {
|
|
|
|
return 1;
|
|
|
|
}
|
|
|
|
if (!duk_is_error(ctx, -1)) {
|
|
|
|
/* Not an Error instance, don't read "stack". */
|
|
|
|
return 1;
|
|
|
|
}
|
|
|
|
|
|
|
|
duk_get_prop_string(ctx, -1, "stack"); /* caller coerces */
|
|
|
|
duk_remove(ctx, -2);
|
|
|
|
return 1;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Print error to stderr and pop error. */
|
|
|
|
static void print_pop_error(duk_context *ctx, FILE *f) {
|
|
|
|
/* Print error objects with a stack trace specially.
|
|
|
|
* Note that getting the stack trace may throw an error
|
|
|
|
* so this also needs to be safe call wrapped.
|
|
|
|
*/
|
|
|
|
(void) duk_safe_call(ctx, get_stack_raw, 1 /*nargs*/, 1 /*nrets*/);
|
|
|
|
fprintf(f, "%s\n", duk_safe_to_string(ctx, -1));
|
|
|
|
fflush(f);
|
|
|
|
duk_pop(ctx);
|
|
|
|
}
|
|
|
|
|
|
|
|
static int wrapped_compile_execute(duk_context *ctx) {
|
|
|
|
int comp_flags;
|
|
|
|
|
|
|
|
/* XXX: Here it'd be nice to get some stats for the compilation result
|
|
|
|
* when a suitable command line is given (e.g. code size, constant
|
|
|
|
* count, function count. These are available internally but not through
|
|
|
|
* the public API.
|
|
|
|
*/
|
|
|
|
|
|
|
|
comp_flags = 0;
|
|
|
|
duk_compile(ctx, comp_flags);
|
|
|
|
|
|
|
|
duk_push_global_object(ctx); /* 'this' binding */
|
|
|
|
duk_call_method(ctx, 0);
|
|
|
|
|
|
|
|
if (interactive_mode) {
|
|
|
|
/*
|
|
|
|
* In interactive mode, write to stdout so output won't
|
|
|
|
* interleave as easily.
|
|
|
|
*
|
|
|
|
* NOTE: the ToString() coercion may fail in some cases;
|
|
|
|
* for instance, if you evaluate:
|
|
|
|
*
|
|
|
|
* ( {valueOf: function() {return {}},
|
|
|
|
* toString: function() {return {}}});
|
|
|
|
*
|
|
|
|
* The error is:
|
|
|
|
*
|
|
|
|
* TypeError: failed to coerce with [[DefaultValue]]
|
|
|
|
* duk_api.c:1420
|
|
|
|
*
|
|
|
|
* These are handled now by the caller which also has stack
|
|
|
|
* trace printing support. User code can print out errors
|
|
|
|
* safely using duk_safe_to_string().
|
|
|
|
*/
|
|
|
|
|
|
|
|
fprintf(stdout, "= %s\n", duk_to_string(ctx, -1));
|
|
|
|
fflush(stdout);
|
|
|
|
} else {
|
|
|
|
/* In non-interactive mode, success results are not written at all.
|
|
|
|
* It is important that the result value is not string coerced,
|
|
|
|
* as the string coercion may cause an error in some cases.
|
|
|
|
*/
|
|
|
|
}
|
|
|
|
|
|
|
|
duk_pop(ctx);
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
static int handle_fh(duk_context *ctx, FILE *f, const char *filename) {
|
|
|
|
char *buf = NULL;
|
|
|
|
int len;
|
|
|
|
int got;
|
|
|
|
int rc;
|
|
|
|
int retval = -1;
|
|
|
|
|
|
|
|
if (fseek(f, 0, SEEK_END) < 0) {
|
|
|
|
goto error;
|
|
|
|
}
|
|
|
|
len = (int) ftell(f);
|
|
|
|
if (fseek(f, 0, SEEK_SET) < 0) {
|
|
|
|
goto error;
|
|
|
|
}
|
|
|
|
buf = (char *) malloc(len);
|
|
|
|
if (!buf) {
|
|
|
|
goto error;
|
|
|
|
}
|
|
|
|
|
|
|
|
got = fread((void *) buf, (size_t) 1, (size_t) len, f);
|
|
|
|
|
|
|
|
duk_push_lstring(ctx, buf, got);
|
|
|
|
duk_push_string(ctx, filename);
|
|
|
|
|
|
|
|
free(buf);
|
|
|
|
buf = NULL;
|
|
|
|
|
|
|
|
interactive_mode = 0; /* global */
|
|
|
|
|
|
|
|
rc = duk_safe_call(ctx, wrapped_compile_execute, 2 /*nargs*/, 1 /*nret*/);
|
|
|
|
if (rc != DUK_EXEC_SUCCESS) {
|
|
|
|
print_pop_error(ctx, stderr);
|
|
|
|
goto error;
|
|
|
|
} else {
|
|
|
|
duk_pop(ctx);
|
|
|
|
retval = 0;
|
|
|
|
}
|
|
|
|
/* fall thru */
|
|
|
|
|
|
|
|
cleanup:
|
|
|
|
if (buf) {
|
|
|
|
free(buf);
|
|
|
|
}
|
|
|
|
return retval;
|
|
|
|
|
|
|
|
error:
|
|
|
|
fprintf(stderr, "error in executing file %s\n", filename);
|
|
|
|
fflush(stderr);
|
|
|
|
goto cleanup;
|
|
|
|
}
|
|
|
|
|
|
|
|
static int handle_file(duk_context *ctx, const char *filename) {
|
|
|
|
FILE *f = NULL;
|
|
|
|
int retval;
|
|
|
|
|
|
|
|
f = fopen(filename, "rb");
|
|
|
|
if (!f) {
|
|
|
|
fprintf(stderr, "failed to open source file: %s\n", filename);
|
|
|
|
fflush(stderr);
|
|
|
|
goto error;
|
|
|
|
}
|
|
|
|
|
|
|
|
retval = handle_fh(ctx, f, filename);
|
|
|
|
|
|
|
|
fclose(f);
|
|
|
|
return retval;
|
|
|
|
|
|
|
|
error:
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
|
|
|
static int handle_eval(duk_context *ctx, const char *code) {
|
|
|
|
int rc;
|
|
|
|
int retval = -1;
|
|
|
|
|
|
|
|
duk_push_string(ctx, code);
|
|
|
|
duk_push_string(ctx, "eval");
|
|
|
|
|
|
|
|
interactive_mode = 0; /* global */
|
|
|
|
|
|
|
|
rc = duk_safe_call(ctx, wrapped_compile_execute, 2 /*nargs*/, 1 /*nret*/);
|
|
|
|
if (rc != DUK_EXEC_SUCCESS) {
|
|
|
|
print_pop_error(ctx, stderr);
|
|
|
|
} else {
|
|
|
|
duk_pop(ctx);
|
|
|
|
retval = 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
return retval;
|
|
|
|
}
|
|
|
|
|
|
|
|
#ifdef NO_READLINE
|
|
|
|
static int handle_interactive(duk_context *ctx) {
|
|
|
|
const char *prompt = "duk> ";
|
|
|
|
char *buffer = NULL;
|
|
|
|
int retval = 0;
|
|
|
|
int rc;
|
|
|
|
int got_eof = 0;
|
|
|
|
|
|
|
|
duk_eval_string(ctx, GREET_CODE(" [no readline]"));
|
|
|
|
duk_pop(ctx);
|
|
|
|
|
|
|
|
buffer = (char *) malloc(LINEBUF_SIZE);
|
|
|
|
if (!buffer) {
|
|
|
|
fprintf(stderr, "failed to allocated a line buffer\n");
|
|
|
|
fflush(stderr);
|
|
|
|
retval = -1;
|
|
|
|
goto done;
|
|
|
|
}
|
|
|
|
|
|
|
|
while (!got_eof) {
|
|
|
|
size_t idx = 0;
|
|
|
|
|
|
|
|
fwrite(prompt, 1, strlen(prompt), stdout);
|
|
|
|
fflush(stdout);
|
|
|
|
|
|
|
|
for (;;) {
|
|
|
|
int c = fgetc(stdin);
|
|
|
|
if (c == EOF) {
|
|
|
|
got_eof = 1;
|
|
|
|
break;
|
|
|
|
} else if (c == '\n') {
|
|
|
|
break;
|
|
|
|
} else if (idx >= LINEBUF_SIZE) {
|
|
|
|
fprintf(stderr, "line too long\n");
|
|
|
|
fflush(stderr);
|
|
|
|
retval = -1;
|
|
|
|
goto done;
|
|
|
|
} else {
|
|
|
|
buffer[idx++] = (char) c;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
duk_push_lstring(ctx, buffer, idx);
|
|
|
|
duk_push_string(ctx, "input");
|
|
|
|
|
|
|
|
interactive_mode = 1; /* global */
|
|
|
|
|
|
|
|
rc = duk_safe_call(ctx, wrapped_compile_execute, 2 /*nargs*/, 1 /*nret*/);
|
|
|
|
if (rc != DUK_EXEC_SUCCESS) {
|
|
|
|
/* in interactive mode, write to stdout */
|
|
|
|
print_pop_error(ctx, stdout);
|
|
|
|
retval = -1; /* an error 'taints' the execution */
|
|
|
|
} else {
|
|
|
|
duk_pop(ctx);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
done:
|
|
|
|
if (buffer) {
|
|
|
|
free(buffer);
|
|
|
|
buffer = NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
return retval;
|
|
|
|
}
|
|
|
|
#else /* NO_READLINE */
|
|
|
|
static int handle_interactive(duk_context *ctx) {
|
|
|
|
const char *prompt = "duk> ";
|
|
|
|
char *buffer = NULL;
|
|
|
|
int retval = 0;
|
|
|
|
int rc;
|
|
|
|
|
|
|
|
duk_eval_string(ctx, GREET_CODE(""));
|
|
|
|
duk_pop(ctx);
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Note: using readline leads to valgrind-reported leaks inside
|
|
|
|
* readline itself. Execute code from an input file (and not
|
|
|
|
* through stdin) for clean valgrind runs.
|
|
|
|
*/
|
|
|
|
|
|
|
|
rl_initialize();
|
|
|
|
|
|
|
|
for (;;) {
|
|
|
|
if (buffer) {
|
|
|
|
free(buffer);
|
|
|
|
buffer = NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
buffer = readline(prompt);
|
|
|
|
if (!buffer) {
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (buffer && buffer[0] != (char) 0) {
|
|
|
|
add_history(buffer);
|
|
|
|
}
|
|
|
|
|
|
|
|
duk_push_lstring(ctx, buffer, strlen(buffer));
|
|
|
|
duk_push_string(ctx, "input");
|
|
|
|
|
|
|
|
if (buffer) {
|
|
|
|
free(buffer);
|
|
|
|
buffer = NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
interactive_mode = 1; /* global */
|
|
|
|
|
|
|
|
rc = duk_safe_call(ctx, wrapped_compile_execute, 2 /*nargs*/, 1 /*nret*/);
|
|
|
|
if (rc != DUK_EXEC_SUCCESS) {
|
|
|
|
/* in interactive mode, write to stdout */
|
|
|
|
print_pop_error(ctx, stdout);
|
|
|
|
retval = -1; /* an error 'taints' the execution */
|
|
|
|
} else {
|
|
|
|
duk_pop(ctx);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
if (buffer) {
|
|
|
|
free(buffer);
|
|
|
|
buffer = NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
return retval;
|
|
|
|
}
|
|
|
|
#endif /* NO_READLINE */
|
|
|
|
|
|
|
|
#ifdef DUK_CMDLINE_AJSHEAP
|
|
|
|
/*
|
|
|
|
* Heap initialization when using AllJoyn.js pool allocator (without any
|
|
|
|
* other AllJoyn.js integration). This serves as an example of how to
|
|
|
|
* integrate Duktape with a pool allocator and is useful for low memory
|
|
|
|
* testing.
|
|
|
|
*
|
|
|
|
* The pool sizes are not optimized here. The sizes are chosen so that
|
|
|
|
* you can look at the high water mark (hwm) and use counts (use) and see
|
|
|
|
* how much allocations are needed for each pool size. To optimize pool
|
|
|
|
* sizes more accurately, you can use --alloc-logging and inspect the memory
|
|
|
|
* allocation log which provides exact byte counts etc.
|
|
|
|
*
|
|
|
|
* https://git.allseenalliance.org/cgit/core/alljoyn-js.git
|
|
|
|
* https://git.allseenalliance.org/cgit/core/alljoyn-js.git/tree/ajs.c
|
|
|
|
*/
|
|
|
|
|
|
|
|
#include "ajs.h"
|
|
|
|
#include "ajs_heap.h"
|
|
|
|
|
|
|
|
static const AJS_HeapConfig ajsheap_config[] = {
|
|
|
|
{ 8, 10, AJS_POOL_BORROW, 0 },
|
|
|
|
{ 12, 10, AJS_POOL_BORROW, 0 },
|
|
|
|
{ 16, 200, AJS_POOL_BORROW, 0 },
|
|
|
|
{ 20, 400, AJS_POOL_BORROW, 0 },
|
|
|
|
{ 24, 400, AJS_POOL_BORROW, 0 },
|
|
|
|
{ 28, 200, AJS_POOL_BORROW, 0 },
|
|
|
|
{ 32, 200, AJS_POOL_BORROW, 0 },
|
|
|
|
{ 40, 200, AJS_POOL_BORROW, 0 },
|
|
|
|
{ 48, 50, AJS_POOL_BORROW, 0 },
|
|
|
|
{ 52, 50, AJS_POOL_BORROW, 0 },
|
|
|
|
{ 56, 50, AJS_POOL_BORROW, 0 },
|
|
|
|
{ 60, 50, AJS_POOL_BORROW, 0 },
|
|
|
|
{ 64, 50, 0, 0 },
|
|
|
|
{ 128, 80, 0, 0 },
|
|
|
|
{ 256, 16, 0, 0 },
|
|
|
|
{ 512, 16, 0, 0 },
|
|
|
|
{ 1024, 6, 0, 0 },
|
|
|
|
{ 2048, 5, 0, 0 },
|
|
|
|
{ 4096, 3, 0, 0 },
|
|
|
|
{ 8192, 1, 0, 0 }
|
|
|
|
};
|
|
|
|
|
|
|
|
uint8_t *ajsheap_ram = NULL;
|
|
|
|
|
|
|
|
/* Example pointer compression functions.
|
|
|
|
* 'base' is chosen so that no non-NULL pointer results in a zero result
|
|
|
|
* which is reserved for NULL pointers.
|
|
|
|
*/
|
|
|
|
duk_uint16_t ajsheap_enc16(void *p) {
|
|
|
|
duk_uint32_t ret;
|
|
|
|
char *base = (char *) ajsheap_ram - 4;
|
|
|
|
|
|
|
|
if (p == NULL) {
|
|
|
|
ret = 0;
|
|
|
|
} else {
|
|
|
|
ret = (duk_uint32_t) (((char *) p - base) >> 2);
|
|
|
|
}
|
|
|
|
#if 0
|
|
|
|
printf("ajsheap_enc16: %p -> %u\n", p, (unsigned int) ret);
|
|
|
|
#endif
|
|
|
|
if (ret > 0xffffUL) {
|
|
|
|
fprintf(stderr, "Failed to compress pointer\n");
|
|
|
|
fflush(stderr);
|
|
|
|
abort();
|
|
|
|
}
|
|
|
|
return (duk_uint16_t) ret;
|
|
|
|
}
|
|
|
|
void *ajsheap_dec16(duk_uint16_t x) {
|
|
|
|
void *ret;
|
|
|
|
char *base = (char *) ajsheap_ram - 4;
|
|
|
|
|
|
|
|
if (x == 0) {
|
|
|
|
ret = NULL;
|
|
|
|
} else {
|
|
|
|
ret = (void *) (base + (((duk_uint32_t) x) << 2));
|
|
|
|
}
|
|
|
|
#if 0
|
|
|
|
printf("ajsheap_dec16: %u -> %p\n", (unsigned int) x, ret);
|
|
|
|
#endif
|
|
|
|
return ret;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Simplified example of an external strings strategy where incoming strings
|
|
|
|
* are writted sequentially into a fixed flash memory area which is memory
|
|
|
|
* mapped. The example first scans if the string is already in the flash
|
|
|
|
* (which may happen if the same string is interned multiple times), then
|
|
|
|
* adds it to flash if there is space.
|
|
|
|
*
|
|
|
|
* This example is too slow to be used in a real world application: there
|
|
|
|
* should be e.g. a hash table to quickly check for strings that are already
|
|
|
|
* present in the string data (similarly to how string interning works in
|
|
|
|
* Duktape itself).
|
|
|
|
*/
|
|
|
|
static uint8_t ajsheap_strdata[65536];
|
|
|
|
static size_t ajsheap_strdata_used = 0;
|
|
|
|
|
|
|
|
const void *ajsheap_ext_str_check(const void *ptr, duk_size_t len) {
|
|
|
|
uint8_t *p, *p_end;
|
|
|
|
uint8_t initial;
|
|
|
|
uint8_t *ret;
|
|
|
|
size_t left;
|
|
|
|
|
|
|
|
if (len <= 3) {
|
|
|
|
/* It's not worth it to make very small strings external, as
|
|
|
|
* they would take the same space anyway. Also avoids zero
|
|
|
|
* length degenerate case.
|
|
|
|
*/
|
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Check if we already have the string. Be careful to compare for
|
|
|
|
* NUL terminator too, it is NOT present in 'ptr'. This algorithm
|
|
|
|
* is too simplistic and way too slow for actual use.
|
|
|
|
*/
|
|
|
|
|
|
|
|
initial = ((const uint8_t *) ptr)[0];
|
|
|
|
for (p = ajsheap_strdata, p_end = p + ajsheap_strdata_used; p != p_end; p++) {
|
|
|
|
if (*p != initial) {
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
left = (size_t) (p_end - p);
|
|
|
|
if (left >= len + 1 &&
|
|
|
|
memcmp(p, ptr, len) == 0 &&
|
|
|
|
p[len] == 0) {
|
|
|
|
ret = p;
|
|
|
|
#if 0
|
|
|
|
printf("ajsheap_ext_str_check: ptr=%p, len=%ld -> existing %p (used=%ld)\n", (void *) ptr, (long) len, (void *) ret, (long) ajsheap_strdata_used);
|
|
|
|
#endif
|
|
|
|
return ret;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Not present yet, check if we have space. Again, be careful to
|
|
|
|
* ensure there is space for a NUL following the input data.
|
|
|
|
*/
|
|
|
|
|
|
|
|
if (ajsheap_strdata_used + len + 1 > sizeof(ajsheap_strdata)) {
|
|
|
|
#if 0
|
|
|
|
printf("ajsheap_ext_str_check: ptr=%p, len=%ld -> no space (used=%ld)\n", (void *) ptr, (long) len, (long) ajsheap_strdata_used);
|
|
|
|
#endif
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* There is space, add the string to our collection, being careful
|
|
|
|
* to append the NUL.
|
|
|
|
*/
|
|
|
|
|
|
|
|
ret = ajsheap_strdata + ajsheap_strdata_used;
|
|
|
|
memcpy(ret, ptr, len);
|
|
|
|
ret[len] = (uint8_t) 0;
|
|
|
|
ajsheap_strdata_used += len + 1;
|
|
|
|
|
|
|
|
#if 0
|
|
|
|
printf("ajsheap_ext_str_check: ptr=%p, len=%ld -> %p (used=%ld)\n", (void *) ptr, (long) len, (void *) ret, (long) ajsheap_strdata_used);
|
|
|
|
#endif
|
|
|
|
return (const void *) ret;
|
|
|
|
}
|
|
|
|
|
|
|
|
static void ajsheap_init(void) {
|
|
|
|
size_t heap_sz[1];
|
|
|
|
uint8_t *heap_array[1];
|
|
|
|
uint8_t num_pools, i;
|
|
|
|
AJ_Status ret;
|
|
|
|
|
|
|
|
num_pools = (uint8_t) (sizeof(ajsheap_config) / sizeof(AJS_HeapConfig));
|
|
|
|
heap_sz[0] = AJS_HeapRequired(ajsheap_config, /* heapConfig */
|
|
|
|
num_pools, /* numPools */
|
|
|
|
0); /* heapNum */
|
|
|
|
ajsheap_ram = (uint8_t *) malloc(heap_sz[0]);
|
|
|
|
if (!ajsheap_ram) {
|
|
|
|
fprintf(stderr, "Failed to allocate AJS heap\n");
|
|
|
|
fflush(stderr);
|
|
|
|
exit(1);
|
|
|
|
}
|
|
|
|
heap_array[0] = ajsheap_ram;
|
|
|
|
|
|
|
|
fprintf(stderr, "Allocated AJS heap of %ld bytes, pools:", (long) heap_sz[0]);
|
|
|
|
for (i = 0; i < num_pools; i++) {
|
|
|
|
fprintf(stderr, " (sz:%ld,num:%ld,brw:%ld,idx:%ld)",
|
|
|
|
(long) ajsheap_config[i].size, (long) ajsheap_config[i].entries,
|
|
|
|
(long) ajsheap_config[i].borrow, (long) ajsheap_config[i].heapIndex);
|
|
|
|
}
|
|
|
|
fprintf(stderr, "\n");
|
|
|
|
fflush(stderr);
|
|
|
|
|
|
|
|
ret = AJS_HeapInit(heap_array, /* heap */
|
|
|
|
heap_sz, /* heapSz */
|
|
|
|
ajsheap_config, /* heapConfig */
|
|
|
|
num_pools, /* numPools */
|
|
|
|
1); /* numHeaps */
|
|
|
|
fprintf(stderr, "AJS_HeapInit() -> %ld\n", (long) ret);
|
|
|
|
fflush(stderr);
|
|
|
|
}
|
|
|
|
|
|
|
|
/* AjsHeap.dump(), allows Ecmascript code to dump heap status at suitable
|
|
|
|
* points.
|
|
|
|
*/
|
|
|
|
static duk_ret_t ajsheap_dump(duk_context *ctx) {
|
|
|
|
AJS_HeapDump();
|
|
|
|
fflush(stdout);
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
static void ajsheap_register(duk_context *ctx) {
|
|
|
|
duk_push_object(ctx);
|
|
|
|
duk_push_c_function(ctx, ajsheap_dump, 0);
|
|
|
|
duk_put_prop_string(ctx, -2, "dump");
|
|
|
|
duk_put_global_string(ctx, "AjsHeap");
|
|
|
|
}
|
|
|
|
#endif /* DUK_CMDLINE_AJSHEAP */
|
|
|
|
|
|
|
|
#define ALLOC_DEFAULT 0
|
|
|
|
#define ALLOC_LOGGING 1
|
|
|
|
#define ALLOC_TORTURE 2
|
|
|
|
#define ALLOC_HYBRID 3
|
|
|
|
#define ALLOC_AJSHEAP 4
|
|
|
|
|
|
|
|
int main(int argc, char *argv[]) {
|
|
|
|
duk_context *ctx = NULL;
|
|
|
|
int retval = 0;
|
|
|
|
int have_files = 0;
|
|
|
|
int have_eval = 0;
|
|
|
|
int interactive = 0;
|
|
|
|
int memlimit_high = 1;
|
|
|
|
int alloc_provider = ALLOC_DEFAULT;
|
|
|
|
int i;
|
|
|
|
|
|
|
|
#ifdef DUK_CMDLINE_AJSHEAP
|
|
|
|
alloc_provider = ALLOC_AJSHEAP;
|
|
|
|
#endif
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Signal handling setup
|
|
|
|
*/
|
|
|
|
|
|
|
|
#ifndef NO_SIGNAL
|
|
|
|
set_sigint_handler();
|
|
|
|
|
|
|
|
/* This is useful at the global level; libraries should avoid SIGPIPE though */
|
|
|
|
/*signal(SIGPIPE, SIG_IGN);*/
|
|
|
|
#endif
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Parse options
|
|
|
|
*/
|
|
|
|
|
|
|
|
for (i = 1; i < argc; i++) {
|
|
|
|
char *arg = argv[i];
|
|
|
|
if (!arg) {
|
|
|
|
goto usage;
|
|
|
|
}
|
|
|
|
if (strcmp(arg, "--restrict-memory") == 0) {
|
|
|
|
memlimit_high = 0;
|
|
|
|
} else if (strcmp(arg, "-i") == 0) {
|
|
|
|
interactive = 1;
|
|
|
|
} else if (strcmp(arg, "-e") == 0) {
|
|
|
|
have_eval = 1;
|
|
|
|
if (i == argc - 1) {
|
|
|
|
goto usage;
|
|
|
|
}
|
|
|
|
i++; /* skip code */
|
|
|
|
} else if (strcmp(arg, "--alloc-default") == 0) {
|
|
|
|
alloc_provider = ALLOC_DEFAULT;
|
|
|
|
} else if (strcmp(arg, "--alloc-logging") == 0) {
|
|
|
|
alloc_provider = ALLOC_LOGGING;
|
|
|
|
} else if (strcmp(arg, "--alloc-torture") == 0) {
|
|
|
|
alloc_provider = ALLOC_TORTURE;
|
|
|
|
} else if (strcmp(arg, "--alloc-hybrid") == 0) {
|
|
|
|
alloc_provider = ALLOC_HYBRID;
|
|
|
|
} else if (strcmp(arg, "--alloc-ajsheap") == 0) {
|
|
|
|
alloc_provider = ALLOC_AJSHEAP;
|
|
|
|
} else if (strlen(arg) >= 1 && arg[0] == '-') {
|
|
|
|
goto usage;
|
|
|
|
} else {
|
|
|
|
have_files = 1;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
if (!have_files && !have_eval) {
|
|
|
|
interactive = 1;
|
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Memory limit
|
|
|
|
*/
|
|
|
|
|
|
|
|
#ifndef NO_RLIMIT
|
|
|
|
set_resource_limits(memlimit_high ? MEM_LIMIT_HIGH : MEM_LIMIT_NORMAL);
|
|
|
|
#else
|
|
|
|
if (memlimit_high == 0) {
|
|
|
|
fprintf(stderr, "Warning: option --restrict-memory ignored, no rlimit support\n");
|
|
|
|
fflush(stderr);
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Create context
|
|
|
|
*/
|
|
|
|
|
|
|
|
ctx = NULL;
|
|
|
|
if (!ctx && alloc_provider == ALLOC_LOGGING) {
|
|
|
|
#ifdef DUK_CMDLINE_ALLOC_LOGGING
|
|
|
|
ctx = duk_create_heap(duk_alloc_logging,
|
|
|
|
duk_realloc_logging,
|
|
|
|
duk_free_logging,
|
|
|
|
NULL,
|
|
|
|
NULL);
|
|
|
|
#else
|
|
|
|
fprintf(stderr, "Warning: option --alloc-logging ignored, no logging allocator support\n");
|
|
|
|
fflush(stderr);
|
|
|
|
#endif
|
|
|
|
}
|
|
|
|
if (!ctx && alloc_provider == ALLOC_TORTURE) {
|
|
|
|
#ifdef DUK_CMDLINE_ALLOC_TORTURE
|
|
|
|
ctx = duk_create_heap(duk_alloc_torture,
|
|
|
|
duk_realloc_torture,
|
|
|
|
duk_free_torture,
|
|
|
|
NULL,
|
|
|
|
NULL);
|
|
|
|
#else
|
|
|
|
fprintf(stderr, "Warning: option --alloc-torture ignored, no torture allocator support\n");
|
|
|
|
fflush(stderr);
|
|
|
|
#endif
|
|
|
|
}
|
|
|
|
if (!ctx && alloc_provider == ALLOC_HYBRID) {
|
|
|
|
#ifdef DUK_CMDLINE_ALLOC_HYBRID
|
|
|
|
void *udata = duk_alloc_hybrid_init();
|
|
|
|
if (!udata) {
|
|
|
|
fprintf(stderr, "Failed to init hybrid allocator\n");
|
|
|
|
fflush(stderr);
|
|
|
|
} else {
|
|
|
|
ctx = duk_create_heap(duk_alloc_hybrid,
|
|
|
|
duk_realloc_hybrid,
|
|
|
|
duk_free_hybrid,
|
|
|
|
udata,
|
|
|
|
NULL);
|
|
|
|
}
|
|
|
|
#else
|
|
|
|
fprintf(stderr, "Warning: option --alloc-hybrid ignored, no hybrid allocator support\n");
|
|
|
|
fflush(stderr);
|
|
|
|
#endif
|
|
|
|
}
|
|
|
|
if (!ctx && alloc_provider == ALLOC_AJSHEAP) {
|
|
|
|
#ifdef DUK_CMDLINE_AJSHEAP
|
|
|
|
ajsheap_init();
|
|
|
|
|
|
|
|
ctx = duk_create_heap(AJS_Alloc,
|
|
|
|
AJS_Realloc,
|
|
|
|
AJS_Free,
|
|
|
|
(void *) &ctx, /* alloc_udata */
|
|
|
|
NULL); /* fatal_handler */
|
|
|
|
#else
|
|
|
|
fprintf(stderr, "Warning: option --alloc-ajsheap ignored, no ajsheap allocator support\n");
|
|
|
|
fflush(stderr);
|
|
|
|
#endif
|
|
|
|
}
|
|
|
|
if (!ctx && alloc_provider == ALLOC_DEFAULT) {
|
|
|
|
ctx = duk_create_heap_default();
|
|
|
|
}
|
|
|
|
|
|
|
|
if (!ctx) {
|
|
|
|
fprintf(stderr, "Failed to create Duktape heap\n");
|
|
|
|
fflush(stderr);
|
|
|
|
exit(-1);
|
|
|
|
}
|
|
|
|
|
|
|
|
#ifdef DUK_CMDLINE_AJSHEAP
|
|
|
|
if (alloc_provider == ALLOC_AJSHEAP) {
|
|
|
|
fprintf(stdout, "Pool dump after heap creation\n");
|
|
|
|
fflush(stdout);
|
|
|
|
AJS_HeapDump();
|
|
|
|
fflush(stdout);
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
|
|
|
#ifdef DUK_CMDLINE_AJSHEAP
|
|
|
|
if (alloc_provider == ALLOC_AJSHEAP) {
|
|
|
|
ajsheap_register(ctx);
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Execute any argument file(s)
|
|
|
|
*/
|
|
|
|
|
|
|
|
for (i = 1; i < argc; i++) {
|
|
|
|
char *arg = argv[i];
|
|
|
|
if (!arg) {
|
|
|
|
continue;
|
|
|
|
} else if (strlen(arg) == 2 && strcmp(arg, "-e") == 0) {
|
|
|
|
/* Here we know the eval arg exists but check anyway */
|
|
|
|
if (i == argc - 1) {
|
|
|
|
retval = 1;
|
|
|
|
goto cleanup;
|
|
|
|
}
|
|
|
|
if (handle_eval(ctx, argv[i + 1]) != 0) {
|
|
|
|
retval = 1;
|
|
|
|
goto cleanup;
|
|
|
|
}
|
|
|
|
i++; /* skip code */
|
|
|
|
continue;
|
|
|
|
} else if (strlen(arg) >= 1 && arg[0] == '-') {
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (handle_file(ctx, arg) != 0) {
|
|
|
|
retval = 1;
|
|
|
|
goto cleanup;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Enter interactive mode if options indicate it
|
|
|
|
*/
|
|
|
|
|
|
|
|
if (interactive) {
|
|
|
|
if (handle_interactive(ctx) != 0) {
|
|
|
|
retval = 1;
|
|
|
|
goto cleanup;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Cleanup and exit
|
|
|
|
*/
|
|
|
|
|
|
|
|
cleanup:
|
|
|
|
if (interactive) {
|
|
|
|
fprintf(stderr, "Cleaning up...\n");
|
|
|
|
fflush(stderr);
|
|
|
|
}
|
|
|
|
|
|
|
|
#ifdef DUK_CMDLINE_AJSHEAP
|
|
|
|
if (alloc_provider == ALLOC_AJSHEAP) {
|
|
|
|
fprintf(stdout, "Pool dump before duk_destroy_heap(), before forced gc\n");
|
|
|
|
fflush(stdout);
|
|
|
|
AJS_HeapDump();
|
|
|
|
fflush(stdout);
|
|
|
|
|
|
|
|
duk_gc(ctx, 0);
|
|
|
|
|
|
|
|
fprintf(stdout, "Pool dump before duk_destroy_heap(), after forced gc\n");
|
|
|
|
fflush(stdout);
|
|
|
|
AJS_HeapDump();
|
|
|
|
fflush(stdout);
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
|
|
|
if (ctx) {
|
|
|
|
duk_destroy_heap(ctx);
|
|
|
|
}
|
|
|
|
|
|
|
|
#ifdef DUK_CMDLINE_AJSHEAP
|
|
|
|
if (alloc_provider == ALLOC_AJSHEAP) {
|
|
|
|
fprintf(stdout, "Pool dump after duk_destroy_heap() (should have zero allocs)\n");
|
|
|
|
fflush(stdout);
|
|
|
|
AJS_HeapDump();
|
|
|
|
fflush(stdout);
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
|
|
|
return retval;
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Usage
|
|
|
|
*/
|
|
|
|
|
|
|
|
usage:
|
|
|
|
fprintf(stderr, "Usage: duk [options] [<filenames>]\n"
|
|
|
|
"\n"
|
|
|
|
" -i enter interactive mode after executing argument file(s) / eval code\n"
|
|
|
|
" -e CODE evaluate code\n"
|
|
|
|
" --restrict-memory use lower memory limit (used by test runner)\n"
|
|
|
|
" --alloc-default use Duktape default allocator\n"
|
|
|
|
#ifdef DUK_CMDLINE_ALLOC_LOGGING
|
|
|
|
" --alloc-logging use logging allocator (writes to /tmp)\n"
|
|
|
|
#endif
|
|
|
|
#ifdef DUK_CMDLINE_ALLOC_TORTURE
|
|
|
|
" --alloc-torture use torture allocator\n"
|
|
|
|
#endif
|
|
|
|
#ifdef DUK_CMDLINE_ALLOC_HYBRID
|
|
|
|
" --alloc-hybrid use hybrid allocator\n"
|
|
|
|
#endif
|
|
|
|
#ifdef DUK_CMDLINE_AJSHEAP
|
|
|
|
" --alloc-ajsheap use ajsheap allocator (enabled by default with 'ajduk')\n"
|
|
|
|
#endif
|
|
|
|
"\n"
|
|
|
|
"If <filename> is omitted, interactive mode is started automatically.\n");
|
|
|
|
fflush(stderr);
|
|
|
|
exit(1);
|
|
|
|
}
|