duktape-1.5.2/examples/cmdline/duk_cmdline.c

Go to the documentation of this file.

/*
 *  Command line execution tool.  Useful for test cases and manual testing.
 *
 *  To enable linenoise 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_*.
 */
 
/* Helper define to enable a feature set; can also use separate defines. */
#if defined(DUK_CMDLINE_FANCY)
#define DUK_CMDLINE_LINENOISE
#define DUK_CMDLINE_LINENOISE_COMPLETION
#define DUK_CMDLINE_RLIMIT
#define DUK_CMDLINE_SIGNAL
#endif
 
#if defined(WIN32) || defined(_WIN32) || defined(__WIN32__) || \
    defined(WIN64) || defined(_WIN64) || defined(__WIN64__)
/* Suppress warnings about plain fopen() etc. */
#define _CRT_SECURE_NO_WARNINGS
#if defined(_MSC_VER) && (_MSC_VER < 1900)
/* Workaround for snprintf() missing in older MSVC versions.
 * Note that _snprintf() may not NUL terminate the string, but
 * this difference does not matter here as a NUL terminator is
 * always explicitly added.
 */
#define snprintf _snprintf
#endif
#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>
#if defined(DUK_CMDLINE_SIGNAL)
#include <signal.h>
#endif
#if defined(DUK_CMDLINE_RLIMIT)
#include <sys/resource.h>
#endif
#if defined(DUK_CMDLINE_LINENOISE)
#include "linenoise.h"
#endif
#if defined(DUK_CMDLINE_FILEIO)
#include <errno.h>
#endif
#if defined(EMSCRIPTEN)
#include <emscripten.h>
#endif
#if defined(DUK_CMDLINE_ALLOC_LOGGING)
#include "duk_alloc_logging.h"
#endif
#if defined(DUK_CMDLINE_ALLOC_TORTURE)
#include "duk_alloc_torture.h"
#endif
#if defined(DUK_CMDLINE_ALLOC_HYBRID)
#include "duk_alloc_hybrid.h"
#endif
#include "duktape.h"
 
#if defined(DUK_CMDLINE_AJSHEAP)
/* Defined in duk_cmdline_ajduk.c or alljoyn.js headers. */
void ajsheap_init(void);
void ajsheap_free(void);
void ajsheap_dump(void);
void ajsheap_register(duk_context *ctx);
void ajsheap_start_exec_timeout(void);
void ajsheap_clear_exec_timeout(void);
void *ajsheap_alloc_wrapped(void *udata, duk_size_t size);
void *ajsheap_realloc_wrapped(void *udata, void *ptr, duk_size_t size);
void ajsheap_free_wrapped(void *udata, void *ptr);
void *AJS_Alloc(void *udata, duk_size_t size);
void *AJS_Realloc(void *udata, void *ptr, duk_size_t size);
void AJS_Free(void *udata, void *ptr);
#endif
 
#if defined(DUK_CMDLINE_DEBUGGER_SUPPORT)
#include "duk_trans_socket.h"
#endif
 
#define  MEM_LIMIT_NORMAL   (128*1024*1024)   /* 128 MB */
#define  MEM_LIMIT_HIGH     (2047*1024*1024)  /* ~2 GB */
#define  LINEBUF_SIZE       65536
 
static int main_argc = 0;
static char **main_argv = NULL;
static int interactive_mode = 0;
#if defined(DUK_CMDLINE_DEBUGGER_SUPPORT)
static int debugger_reattach = 0;
#endif
 
/*
 *  Misc helpers
 */
 
#if defined(DUK_CMDLINE_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  /* DUK_CMDLINE_RLIMIT */
 
#if defined(DUK_CMDLINE_SIGNAL)
static void my_sighandler(int x) {
    fprintf(stderr, "Got signal %d\n", x);
    fflush(stderr);
}
static void set_sigint_handler(void) {
    (void) signal(SIGINT, my_sighandler);
    (void) signal(SIGPIPE, SIG_IGN);  /* avoid SIGPIPE killing process */
}
#endif  /* DUK_CMDLINE_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) {
    const char *src_data;
    duk_size_t src_len;
    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.
     */
 
    /* Use duk_compile_lstring_filename() variant which avoids interning
     * the source code.  This only really matters for low memory environments.
     */
 
    /* [ ... bytecode_filename src_data src_len filename ] */
 
    src_data = (const char *) duk_require_pointer(ctx, -3);
    src_len = (duk_size_t) duk_require_uint(ctx, -2);
 
    if (src_data != NULL && src_len >= 2 && src_data[0] == (char) 0xff) {
        /* Bytecode. */
        duk_push_lstring(ctx, src_data, src_len);
        duk_to_buffer(ctx, -1, NULL);
        duk_load_function(ctx);
    } else {
        /* Source code. */
        comp_flags = 0;
        duk_compile_lstring_filename(ctx, comp_flags, src_data, src_len);
    }
 
    /* [ ... bytecode_filename src_data src_len function ] */
 
    /* Optional bytecode dump. */
    if (duk_is_string(ctx, -4)) {
        FILE *f;
        void *bc_ptr;
        duk_size_t bc_len;
        size_t wrote;
        char fnbuf[256];
        const char *filename;
 
        duk_dup_top(ctx);
        duk_dump_function(ctx);
        bc_ptr = duk_require_buffer(ctx, -1, &bc_len);
        filename = duk_require_string(ctx, -5);
#if defined(EMSCRIPTEN)
        if (filename[0] == '/') {
            snprintf(fnbuf, sizeof(fnbuf), "%s", filename);
        } else {
            snprintf(fnbuf, sizeof(fnbuf), "/working/%s", filename);
        }
#else
        snprintf(fnbuf, sizeof(fnbuf), "%s", filename);
#endif
        fnbuf[sizeof(fnbuf) - 1] = (char) 0;
 
        f = fopen(fnbuf, "wb");
        if (!f) {
            duk_error(ctx, DUK_ERR_ERROR, "failed to open bytecode output file");
        }
        wrote = fwrite(bc_ptr, 1, (size_t) bc_len, f);  /* XXX: handle partial writes */
        (void) fclose(f);
        if (wrote != bc_len) {
            duk_error(ctx, DUK_ERR_ERROR, "failed to write all bytecode");
        }
 
        return 0;  /* duk_safe_call() cleans up */
    }
 
#if 0
    /* Manual test for bytecode dump/load cycle: dump and load before
     * execution.  Enable manually, then run "make qecmatest" for a
     * reasonably good coverage of different functions and programs.
     */
    duk_dump_function(ctx);
    duk_load_function(ctx);
#endif
 
#if defined(DUK_CMDLINE_AJSHEAP)
    ajsheap_start_exec_timeout();
#endif
 
    duk_push_global_object(ctx);  /* 'this' binding */
    duk_call_method(ctx, 0);
 
#if defined(DUK_CMDLINE_AJSHEAP)
    ajsheap_clear_exec_timeout();
#endif
 
    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.
         */
    }
 
    return 0;  /* duk_safe_call() cleans up */
}
 
/*
 *  Minimal Linenoise completion support
 */
 
#if defined(DUK_CMDLINE_LINENOISE_COMPLETION)
static duk_context *completion_ctx;
 
static int completion_idpart(unsigned char c) {
    /* Very simplified "is identifier part" check. */
    if ((c >= (unsigned char) 'a' && c <= (unsigned char) 'z') ||
        (c >= (unsigned char) 'A' && c <= (unsigned char) 'Z') ||
        (c >= (unsigned char) '0' && c <= (unsigned char) '9') ||
        c == (unsigned char) '$' || c == (unsigned char) '_') {
        return 1;
    }
    return 0;
}
 
static int completion_digit(unsigned char c) {
    return (c >= (unsigned char) '0' && c <= (unsigned char) '9');
}
 
static duk_ret_t linenoise_completion_lookup(duk_context *ctx) {
    duk_size_t len;
    const char *orig;
    const unsigned char *p;
    const unsigned char *p_curr;
    const unsigned char *p_end;
    const char *key;
    const char *prefix;
    linenoiseCompletions *lc;
    duk_idx_t idx_obj;
 
    orig = duk_require_string(ctx, -3);
    p_curr = (const unsigned char *) duk_require_lstring(ctx, -2, &len);
    p_end = p_curr + len;
    lc = duk_require_pointer(ctx, -1);
 
    duk_push_global_object(ctx);
    idx_obj = duk_require_top_index(ctx);
 
    while (p_curr <= p_end) {
        /* p_curr == p_end allowed on purpose, to handle 'Math.' for example. */
        p = p_curr;
        while (p < p_end && p[0] != (unsigned char) '.') {
            p++;
        }
        /* 'p' points to a NUL (p == p_end) or a period. */
        prefix = duk_push_lstring(ctx, (const char *) p_curr, (duk_size_t) (p - p_curr));
 
#if 0
        fprintf(stderr, "Completion check: '%s'\n", prefix);
        fflush(stderr);
#endif
 
        if (p == p_end) {
            /* 'idx_obj' points to the object matching the last
             * full component, use [p_curr,p[ as a filter for
             * that object.
             */
 
            duk_enum(ctx, idx_obj, DUK_ENUM_INCLUDE_NONENUMERABLE);
            while (duk_next(ctx, -1, 0 /*get_value*/)) {
                key = duk_get_string(ctx, -1);
#if 0
                fprintf(stderr, "Key: %s\n", key ? key : "");
                fflush(stderr);
#endif
                if (!key) {
                    /* Should never happen, just in case. */
                    goto next;
                }
 
                /* Ignore array index keys: usually not desirable, and would
                 * also require ['0'] quoting.
                 */
                if (completion_digit(key[0])) {
                    goto next;
                }
 
                /* XXX: There's no key quoting now, it would require replacing the
                 * last component with a ['foo\nbar'] style lookup when appropriate.
                 */
 
                if (strlen(prefix) == 0) {
                    /* Partial ends in a period, e.g. 'Math.' -> complete all Math properties. */
                    duk_push_string(ctx, orig);  /* original, e.g. 'Math.' */
                    duk_push_string(ctx, key);
                    duk_concat(ctx, 2);
                    linenoiseAddCompletion(lc, duk_require_string(ctx, -1));
                    duk_pop(ctx);
                } else if (prefix && strcmp(key, prefix) == 0) {
                    /* Full completion, add a period, e.g. input 'Math' -> 'Math.'. */
                    duk_push_string(ctx, orig);  /* original, including partial last component */
                    duk_push_string(ctx, ".");
                    duk_concat(ctx, 2);
                    linenoiseAddCompletion(lc, duk_require_string(ctx, -1));
                    duk_pop(ctx);
                } else if (prefix && strncmp(key, prefix, strlen(prefix)) == 0) {
                    /* Last component is partial, complete. */
                    duk_push_string(ctx, orig);  /* original, including partial last component */
                    duk_push_string(ctx, key + strlen(prefix));  /* completion to last component */
                    duk_concat(ctx, 2);
                    linenoiseAddCompletion(lc, duk_require_string(ctx, -1));
                    duk_pop(ctx);
                }
 
             next:
                duk_pop(ctx);
            }
            return 0;
        } else {
            if (duk_get_prop(ctx, idx_obj)) {
                duk_to_object(ctx, -1);  /* for properties of plain strings etc */
                duk_replace(ctx, idx_obj);
                p_curr = p + 1;
            } else {
                /* Not found. */
                return 0;
            }
        }
    }
 
    return 0;
}
 
static void linenoise_completion(const char *buf, linenoiseCompletions *lc) {
    duk_context *ctx;
    const unsigned char *p_start;
    const unsigned char *p_end;
    const unsigned char *p;
    duk_int_t rc;
 
    if (!buf) {
        return;
    }
    ctx = completion_ctx;
    if (!ctx) {
        return;
    }
 
    p_start = (const unsigned char *) buf;
    p_end = (const unsigned char *) (buf + strlen(buf));
    p = p_end;
 
    /* Scan backwards for a maximal string which looks like a property
     * chain (e.g. foo.bar.quux).
     */
 
    while (--p >= p_start) {
        if (p[0] == (unsigned char) '.') {
            if (p <= p_start) {
                break;
            }
            if (!completion_idpart(p[-1])) {
                /* Catches e.g. 'foo..bar' -> we want 'bar' only. */
                break;
            }
        } else if (!completion_idpart(p[0])) {
            break;
        }
    }
    /* 'p' will either be p_start - 1 (ran out of buffer) or point to
     * the first offending character.
     */
    p++;
    if (p < p_start || p >= p_end) {
        return;  /* should never happen, but just in case */
    }
 
    /* 'p' now points to a string of the form 'foo.bar.quux'.  Look up
     * all the components except the last; treat the last component as
     * a partial name which is used as a filter for the previous full
     * component.  All lookups are from the global object now.
     */
 
#if 0
    fprintf(stderr, "Completion starting point: '%s'\n", p);
    fflush(stderr);
#endif
 
    duk_push_string(ctx, (const char *) buf);
    duk_push_lstring(ctx, (const char *) p, (duk_size_t) (p_end - p));
    duk_push_pointer(ctx, (void *) lc);
 
    rc = duk_safe_call(ctx, linenoise_completion_lookup, 3 /*nargs*/, 1 /*nrets*/);
    if (rc != DUK_EXEC_SUCCESS) {
        fprintf(stderr, "Completion handling failure: %s\n", duk_safe_to_string(ctx, -1));
    }
    duk_pop(ctx);
}
#endif  /* DUK_CMDLINE_LINENOISE_COMPLETION */
 
/*
 *  Execute from file handle etc
 */
 
static int handle_fh(duk_context *ctx, FILE *f, const char *filename, const char *bytecode_filename) {
    char *buf = NULL;
    size_t bufsz;
    size_t bufoff;
    size_t got;
    int rc;
    int retval = -1;
 
    buf = (char *) malloc(1024);
    if (!buf) {
        goto error;
    }
    bufsz = 1024;
    bufoff = 0;
 
    /* Read until EOF, avoid fseek/stat because it won't work with stdin. */
    for (;;) {
        size_t avail;
 
        avail = bufsz - bufoff;
        if (avail < 1024) {
            size_t newsz;
            char *buf_new;
#if 0
            fprintf(stderr, "resizing read buffer: %ld -> %ld\n", (long) bufsz, (long) (bufsz * 2));
#endif
            newsz = bufsz + (bufsz >> 2) + 1024;  /* +25% and some extra */
            buf_new = (char *) realloc(buf, newsz);
            if (!buf_new) {
                goto error;
            }
            buf = buf_new;
            bufsz = newsz;
        }
 
        avail = bufsz - bufoff;
#if 0
        fprintf(stderr, "reading input: buf=%p bufsz=%ld bufoff=%ld avail=%ld\n",
                (void *) buf, (long) bufsz, (long) bufoff, (long) avail);
#endif
 
        got = fread((void *) (buf + bufoff), (size_t) 1, avail, f);
#if 0
        fprintf(stderr, "got=%ld\n", (long) got);
#endif
        if (got == 0) {
            break;
        }
        bufoff += got;
 
        /* Emscripten specific: stdin EOF doesn't work as expected.
         * Instead, when 'emduk' is executed using Node.js, a file
         * piped to stdin repeats (!).  Detect that repeat and cut off
         * the stdin read.  Ensure the loop repeats enough times to
         * avoid detecting spurious loops.
         *
         * This only seems to work for inputs up to 256 bytes long.
         */
#if defined(EMSCRIPTEN)
        if (bufoff >= 16384) {
            size_t i, j, nloops;
            int looped = 0;
 
            for (i = 16; i < bufoff / 8; i++) {
                int ok;
 
                nloops = bufoff / i;
                ok = 1;
                for (j = 1; j < nloops; j++) {
                    if (memcmp((void *) buf, (void *) (buf + i * j), i) != 0) {
                        ok = 0;
                        break;
                    }
                }
                if (ok) {
                    fprintf(stderr, "emscripten workaround: detect looping at index %ld, verified with %ld loops\n", (long) i, (long) (nloops - 1));
                    bufoff = i;
                    looped = 1;
                    break;
                }
            }
 
            if (looped) {
                break;
            }
        }
#endif
    }
 
    duk_push_string(ctx, bytecode_filename);
    duk_push_pointer(ctx, (void *) buf);
    duk_push_uint(ctx, (duk_uint_t) bufoff);
    duk_push_string(ctx, filename);
 
    interactive_mode = 0;  /* global */
 
    rc = duk_safe_call(ctx, wrapped_compile_execute, 4 /*nargs*/, 1 /*nret*/);
 
#if defined(DUK_CMDLINE_AJSHEAP)
    ajsheap_clear_exec_timeout();
#endif
 
    free(buf);
    buf = NULL;
 
    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);
        buf = NULL;
    }
    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, const char *bytecode_filename) {
    FILE *f = NULL;
    int retval;
    char fnbuf[256];
 
    /* Example of sending an application specific debugger notification. */
    duk_push_string(ctx, "DebuggerHandleFile");
    duk_push_string(ctx, filename);
    duk_debugger_notify(ctx, 2);
 
#if defined(EMSCRIPTEN)
    if (filename[0] == '/') {
        snprintf(fnbuf, sizeof(fnbuf), "%s", filename);
    } else {
        snprintf(fnbuf, sizeof(fnbuf), "/working/%s", filename);
    }
#else
    snprintf(fnbuf, sizeof(fnbuf), "%s", filename);
#endif
    fnbuf[sizeof(fnbuf) - 1] = (char) 0;
 
    f = fopen(fnbuf, "rb");
    if (!f) {
        fprintf(stderr, "failed to open source file: %s\n", filename);
        fflush(stderr);
        goto error;
    }
 
    retval = handle_fh(ctx, f, filename, bytecode_filename);
 
    fclose(f);
    return retval;
 
 error:
    return -1;
}
 
static int handle_eval(duk_context *ctx, char *code) {
    int rc;
    int retval = -1;
 
    duk_push_pointer(ctx, (void *) code);
    duk_push_uint(ctx, (duk_uint_t) strlen(code));
    duk_push_string(ctx, "eval");
 
    interactive_mode = 0;  /* global */
 
    rc = duk_safe_call(ctx, wrapped_compile_execute, 3 /*nargs*/, 1 /*nret*/);
 
#if defined(DUK_CMDLINE_AJSHEAP)
    ajsheap_clear_exec_timeout();
#endif
 
    if (rc != DUK_EXEC_SUCCESS) {
        print_pop_error(ctx, stderr);
    } else {
        duk_pop(ctx);
        retval = 0;
    }
 
    return retval;
}
 
#if defined(DUK_CMDLINE_LINENOISE)
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(" [linenoise]"));
    duk_pop(ctx);
 
    linenoiseSetMultiLine(1);
    linenoiseHistorySetMaxLen(64);
#if defined(DUK_CMDLINE_LINENOISE_COMPLETION)
    linenoiseSetCompletionCallback(linenoise_completion);
#endif
 
    for (;;) {
        if (buffer) {
            linenoiseFree(buffer);
            buffer = NULL;
        }
 
#if defined(DUK_CMDLINE_LINENOISE_COMPLETION)
        completion_ctx = ctx;
#endif
        buffer = linenoise(prompt);
#if defined(DUK_CMDLINE_LINENOISE_COMPLETION)
        completion_ctx = NULL;
#endif
 
        if (!buffer) {
            break;
        }
 
        if (buffer && buffer[0] != (char) 0) {
            linenoiseHistoryAdd(buffer);
        }
 
        duk_push_pointer(ctx, (void *) buffer);
        duk_push_uint(ctx, (duk_uint_t) strlen(buffer));
        duk_push_string(ctx, "input");
 
        interactive_mode = 1;  /* global */
 
        rc = duk_safe_call(ctx, wrapped_compile_execute, 3 /*nargs*/, 1 /*nret*/);
 
#if defined(DUK_CMDLINE_AJSHEAP)
        ajsheap_clear_exec_timeout();
#endif
 
        if (buffer) {
            linenoiseFree(buffer);
            buffer = NULL;
        }
 
        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) {
        linenoiseFree(buffer);
        buffer = NULL;
    }
 
    return retval;
}
#else  /* DUK_CMDLINE_LINENOISE */
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(""));
    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_pointer(ctx, (void *) buffer);
        duk_push_uint(ctx, (duk_uint_t) idx);
        duk_push_string(ctx, "input");
 
        interactive_mode = 1;  /* global */
 
        rc = duk_safe_call(ctx, wrapped_compile_execute, 3 /*nargs*/, 1 /*nret*/);
 
#if defined(DUK_CMDLINE_AJSHEAP)
        ajsheap_clear_exec_timeout();
#endif
 
        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;
}
#endif  /* DUK_CMDLINE_LINENOISE */
 
/*
 *  Simple file read/write bindings
 */
 
#if defined(DUK_CMDLINE_FILEIO)
static duk_ret_t fileio_read_file(duk_context *ctx) {
    const char *fn;
    char *buf;
    size_t len;
    size_t off;
    int rc;
    FILE *f;
 
    fn = duk_require_string(ctx, 0);
    f = fopen(fn, "rb");
    if (!f) {
        duk_error(ctx, DUK_ERR_TYPE_ERROR, "cannot open file %s for reading, errno %ld: %s",
                  fn, (long) errno, strerror(errno));
    }
 
    rc = fseek(f, 0, SEEK_END);
    if (rc < 0) {
        (void) fclose(f);
        duk_error(ctx, DUK_ERR_TYPE_ERROR, "fseek() failed for %s, errno %ld: %s",
                  fn, (long) errno, strerror(errno));
    }
    len = (size_t) ftell(f);
    rc = fseek(f, 0, SEEK_SET);
    if (rc < 0) {
        (void) fclose(f);
        duk_error(ctx, DUK_ERR_TYPE_ERROR, "fseek() failed for %s, errno %ld: %s",
                  fn, (long) errno, strerror(errno));
    }
 
    buf = (char *) duk_push_fixed_buffer(ctx, (duk_size_t) len);
    for (off = 0; off < len;) {
        size_t got;
        got = fread((void *) (buf + off), 1, len - off, f);
        if (ferror(f)) {
            (void) fclose(f);
            duk_error(ctx, DUK_ERR_TYPE_ERROR, "error while reading %s", fn);
        }
        if (got == 0) {
            if (feof(f)) {
                break;
            } else {
                (void) fclose(f);
                duk_error(ctx, DUK_ERR_TYPE_ERROR, "error while reading %s", fn);
            }
        }
        off += got;
    }
 
    if (f) {
        (void) fclose(f);
    }
 
    return 1;
}
 
static duk_ret_t fileio_write_file(duk_context *ctx) {
    const char *fn;
    const char *buf;
    size_t len;
    size_t off;
    FILE *f;
 
    fn = duk_require_string(ctx, 0);
    f = fopen(fn, "wb");
    if (!f) {
        duk_error(ctx, DUK_ERR_TYPE_ERROR, "cannot open file %s for writing, errno %ld: %s",
                  fn, (long) errno, strerror(errno));
    }
 
    len = 0;
    buf = (char *) duk_to_buffer(ctx, 1, &len);
    for (off = 0; off < len;) {
        size_t got;
        got = fwrite((const void *) (buf + off), 1, len - off, f);
        if (ferror(f)) {
            (void) fclose(f);
            duk_error(ctx, DUK_ERR_TYPE_ERROR, "error while writing %s", fn);
        }
        if (got == 0) {
            (void) fclose(f);
            duk_error(ctx, DUK_ERR_TYPE_ERROR, "error while writing %s", fn);
        }
        off += got;
    }
 
    if (f) {
        (void) fclose(f);
    }
 
    return 0;
}
#endif  /* DUK_CMDLINE_FILEIO */
 
/*
 *  Duktape heap lifecycle
 */
 
#if defined(DUK_CMDLINE_DEBUGGER_SUPPORT)
static duk_idx_t debugger_request(duk_context *ctx, void *udata, duk_idx_t nvalues) {
    const char *cmd;
    int i;
 
    (void) udata;
 
    if (nvalues < 1) {
        duk_push_string(ctx, "missing AppRequest argument(s)");
        return -1;
    }
 
    cmd = duk_get_string(ctx, -nvalues + 0);
 
    if (cmd && strcmp(cmd, "CommandLine") == 0) {
        if (!duk_check_stack(ctx, main_argc)) {
            /* Callback should avoid errors for now, so use
             * duk_check_stack() rather than duk_require_stack().
             */
            duk_push_string(ctx, "failed to extend stack");
            return -1;
        }
        for (i = 0; i < main_argc; i++) {
            duk_push_string(ctx, main_argv[i]);
        }
        return main_argc;
    }
    duk_push_sprintf(ctx, "command not supported");
    return -1;
}
 
static void debugger_detached(void *udata) {
    duk_context *ctx = (duk_context *) udata;
    (void) ctx;
    fprintf(stderr, "Debugger detached, udata: %p\n", (void *) udata);
    fflush(stderr);
 
    /* Ensure socket is closed even when detach is initiated by Duktape
     * rather than debug client.
     */
        duk_trans_socket_finish();
 
    if (debugger_reattach) {
        /* For automatic reattach testing. */
        duk_trans_socket_init();
        duk_trans_socket_waitconn();
        fprintf(stderr, "Debugger reconnected, call duk_debugger_attach()\n");
        fflush(stderr);
#if 0
        /* This is not necessary but should be harmless. */
        duk_debugger_detach(ctx);
#endif
        duk_debugger_attach_custom(ctx,
                                   duk_trans_socket_read_cb,
                                   duk_trans_socket_write_cb,
                                   duk_trans_socket_peek_cb,
                                   duk_trans_socket_read_flush_cb,
                                   duk_trans_socket_write_flush_cb,
                                   debugger_request,
                                   debugger_detached,
                                   (void *) ctx);
    }
}
#endif
 
#define  ALLOC_DEFAULT  0
#define  ALLOC_LOGGING  1
#define  ALLOC_TORTURE  2
#define  ALLOC_HYBRID   3
#define  ALLOC_AJSHEAP  4
 
static duk_context *create_duktape_heap(int alloc_provider, int debugger, int ajsheap_log) {
    duk_context *ctx;
 
    (void) ajsheap_log;  /* suppress warning */
 
    ctx = NULL;
    if (!ctx && alloc_provider == ALLOC_LOGGING) {
#if defined(DUK_CMDLINE_ALLOC_LOGGING)
        ctx = duk_create_heap(duk_alloc_logging,
                              duk_realloc_logging,
                              duk_free_logging,
                              (void *) 0xdeadbeef,
                              NULL);
#else
        fprintf(stderr, "Warning: option --alloc-logging ignored, no logging allocator support\n");
        fflush(stderr);
#endif
    }
    if (!ctx && alloc_provider == ALLOC_TORTURE) {
#if defined(DUK_CMDLINE_ALLOC_TORTURE)
        ctx = duk_create_heap(duk_alloc_torture,
                              duk_realloc_torture,
                              duk_free_torture,
                              (void *) 0xdeadbeef,
                              NULL);
#else
        fprintf(stderr, "Warning: option --alloc-torture ignored, no torture allocator support\n");
        fflush(stderr);
#endif
    }
    if (!ctx && alloc_provider == ALLOC_HYBRID) {
#if defined(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) {
#if defined(DUK_CMDLINE_AJSHEAP)
        ajsheap_init();
 
        ctx = duk_create_heap(
            ajsheap_log ? ajsheap_alloc_wrapped : AJS_Alloc,
            ajsheap_log ? ajsheap_realloc_wrapped : AJS_Realloc,
            ajsheap_log ? ajsheap_free_wrapped : AJS_Free,
            (void *) 0xdeadbeef,  /* heap_udata: ignored by AjsHeap, use as marker */
            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);
    }
 
#if defined(DUK_CMDLINE_AJSHEAP)
    if (alloc_provider == ALLOC_AJSHEAP) {
        fprintf(stdout, "Pool dump after heap creation\n");
        ajsheap_dump();
    }
#endif
 
#if defined(DUK_CMDLINE_AJSHEAP)
    if (alloc_provider == ALLOC_AJSHEAP) {
        ajsheap_register(ctx);
    }
#endif
 
#if defined(DUK_CMDLINE_FILEIO)
    duk_push_c_function(ctx, fileio_read_file, 1 /*nargs*/);
    duk_put_global_string(ctx, "readFile");
    duk_push_c_function(ctx, fileio_write_file, 2 /*nargs*/);
    duk_put_global_string(ctx, "writeFile");
#endif
 
    if (debugger) {
#if defined(DUK_CMDLINE_DEBUGGER_SUPPORT)
        fprintf(stderr, "Debugger enabled, create socket and wait for connection\n");
        fflush(stderr);
        duk_trans_socket_init();
        duk_trans_socket_waitconn();
        fprintf(stderr, "Debugger connected, call duk_debugger_attach() and then execute requested file(s)/eval\n");
        fflush(stderr);
        duk_debugger_attach_custom(ctx,
                                   duk_trans_socket_read_cb,
                                   duk_trans_socket_write_cb,
                                   duk_trans_socket_peek_cb,
                                   duk_trans_socket_read_flush_cb,
                                   duk_trans_socket_write_flush_cb,
                                   debugger_request,
                                   debugger_detached,
                                   (void *) ctx);
#else
        fprintf(stderr, "Warning: option --debugger ignored, no debugger support\n");
        fflush(stderr);
#endif
    }
 
#if 0
    /* Manual test for duk_debugger_cooperate() */
    {
        for (i = 0; i < 60; i++) {
            printf("cooperate: %d\n", i);
            usleep(1000000);
            duk_debugger_cooperate(ctx);
        }
    }
#endif
 
    return ctx;
}
 
static void destroy_duktape_heap(duk_context *ctx, int alloc_provider) {
    (void) alloc_provider;
 
#if defined(DUK_CMDLINE_AJSHEAP)
    if (alloc_provider == ALLOC_AJSHEAP) {
        fprintf(stdout, "Pool dump before duk_destroy_heap(), before forced gc\n");
        ajsheap_dump();
 
        duk_gc(ctx, 0);
 
        fprintf(stdout, "Pool dump before duk_destroy_heap(), after forced gc\n");
        ajsheap_dump();
    }
#endif
 
    if (ctx) {
        duk_destroy_heap(ctx);
    }
 
#if defined(DUK_CMDLINE_AJSHEAP)
    if (alloc_provider == ALLOC_AJSHEAP) {
        fprintf(stdout, "Pool dump after duk_destroy_heap() (should have zero allocs)\n");
        ajsheap_dump();
    }
    ajsheap_free();
#endif
}
 
/*
 *  Main
 */
 
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 ajsheap_log = 0;
    int debugger = 0;
    int recreate_heap = 0;
    int no_heap_destroy = 0;
    int verbose = 0;
    int run_stdin = 0;
    const char *compile_filename = NULL;
    int i;
 
    main_argc = argc;
    main_argv = (char **) argv;
 
#if defined(EMSCRIPTEN)
    /* Try to use NODEFS to provide access to local files.  Mount the
     * CWD as /working, and then prepend "/working/" to relative native
     * paths in file calls to get something that works reasonably for
     * relative paths.  Emscripten doesn't support replacing virtual
     * "/" with host "/" (the default MEMFS at "/" can't be unmounted)
     * but we can mount "/tmp" as host "/tmp" to allow testcase runs.
     *
     * https://kripken.github.io/emscripten-site/docs/api_reference/Filesystem-API.html#filesystem-api-nodefs
     * https://github.com/kripken/emscripten/blob/master/tests/fs/test_nodefs_rw.c
     */
    EM_ASM(
        /* At the moment it's not possible to replace the default MEMFS mounted at '/':
         * https://github.com/kripken/emscripten/issues/2040
         * https://github.com/kripken/emscripten/blob/incoming/src/library_fs.js#L1341-L1358
         */
        /*
        try {
            FS.unmount("/");
        } catch (e) {
            console.log("Failed to unmount default '/' MEMFS mount: " + e);
        }
        */
        try {
            FS.mkdir("/working");
            FS.mount(NODEFS, { root: "." }, "/working");
        } catch (e) {
            console.log("Failed to mount NODEFS /working: " + e);
        }
        /* A virtual '/tmp' exists by default:
         * https://gist.github.com/evanw/e6be28094f34451bd5bd#file-temp-js-L3806-L3809
         */
        /*
        try {
            FS.mkdir("/tmp");
        } catch (e) {
            console.log("Failed to create virtual /tmp: " + e);
        }
        */
        try {
            FS.mount(NODEFS, { root: "/tmp" }, "/tmp");
        } catch (e) {
            console.log("Failed to mount NODEFS /tmp: " + e);
        }
    );
#endif  /* EMSCRIPTEN */
 
#if defined(DUK_CMDLINE_AJSHEAP)
    alloc_provider = ALLOC_AJSHEAP;
#endif
    (void) ajsheap_log;
 
    /*
     *  Signal handling setup
     */
 
#if defined(DUK_CMDLINE_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, "-c") == 0) {
            if (i == argc - 1) {
                goto usage;
            }
            i++;
            compile_filename = argv[i];
        } 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 (strcmp(arg, "--ajsheap-log") == 0) {
            ajsheap_log = 1;
        } else if (strcmp(arg, "--debugger") == 0) {
            debugger = 1;
#if defined(DUK_CMDLINE_DEBUGGER_SUPPORT)
        } else if (strcmp(arg, "--reattach") == 0) {
            debugger_reattach = 1;
#endif
        } else if (strcmp(arg, "--recreate-heap") == 0) {
            recreate_heap = 1;
        } else if (strcmp(arg, "--no-heap-destroy") == 0) {
            no_heap_destroy = 1;
        } else if (strcmp(arg, "--verbose") == 0) {
            verbose = 1;
        } else if (strcmp(arg, "--run-stdin") == 0) {
            run_stdin = 1;
        } else if (strlen(arg) >= 1 && arg[0] == '-') {
            goto usage;
        } else {
            have_files = 1;
        }
    }
    if (!have_files && !have_eval && !run_stdin) {
        interactive = 1;
    }
 
    /*
     *  Memory limit
     */
 
#if defined(DUK_CMDLINE_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 heap
     */
 
    ctx = create_duktape_heap(alloc_provider, debugger, ajsheap_log);
 
    /*
     *  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) == 2 && strcmp(arg, "-c") == 0) {
            i++;  /* skip filename */
            continue;
        } else if (strlen(arg) >= 1 && arg[0] == '-') {
            continue;
        }
 
        if (verbose) {
            fprintf(stderr, "*** Executing file: %s\n", arg);
            fflush(stderr);
        }
 
        if (handle_file(ctx, arg, compile_filename) != 0) {
            retval = 1;
            goto cleanup;
        }
 
        if (recreate_heap) {
            if (verbose) {
                fprintf(stderr, "*** Recreating heap...\n");
                fflush(stderr);
            }
 
            destroy_duktape_heap(ctx, alloc_provider);
            ctx = create_duktape_heap(alloc_provider, debugger, ajsheap_log);
        }
    }
 
    if (run_stdin) {
        /* Running stdin like a full file (reading all lines before
         * compiling) is useful with emduk:
         * cat test.js | ./emduk --run-stdin
         */
        if (handle_fh(ctx, stdin, "stdin", compile_filename) != 0) {
            retval = 1;
            goto cleanup;
        }
 
        if (recreate_heap) {
            if (verbose) {
                fprintf(stderr, "*** Recreating heap...\n");
                fflush(stderr);
            }
 
            destroy_duktape_heap(ctx, alloc_provider);
            ctx = create_duktape_heap(alloc_provider, debugger, ajsheap_log);
        }
    }
 
    /*
     *  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);
    }
 
    if (ctx && no_heap_destroy) {
        duk_gc(ctx, 0);
    }
    if (ctx && !no_heap_destroy) {
        destroy_duktape_heap(ctx, alloc_provider);
    }
    ctx = NULL;
 
    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"
            "   -c FILE            compile into bytecode (use with only one file argument)\n"
            "   --run-stdin        treat stdin like a file, i.e. compile full input (not line by line)\n"
            "   --verbose          verbose messages to stderr\n"
                    "   --restrict-memory  use lower memory limit (used by test runner)\n"
                    "   --alloc-default    use Duktape default allocator\n"
#if defined(DUK_CMDLINE_ALLOC_LOGGING)
                    "   --alloc-logging    use logging allocator (writes to /tmp)\n"
#endif
#if defined(DUK_CMDLINE_ALLOC_TORTURE)
                    "   --alloc-torture    use torture allocator\n"
#endif
#if defined(DUK_CMDLINE_ALLOC_HYBRID)
                    "   --alloc-hybrid     use hybrid allocator\n"
#endif
#if defined(DUK_CMDLINE_AJSHEAP)
                    "   --alloc-ajsheap    use ajsheap allocator (enabled by default with 'ajduk')\n"
                    "   --ajsheap-log      write alloc log to /tmp/ajduk-alloc-log.txt\n"
#endif
#if defined(DUK_CMDLINE_DEBUGGER_SUPPORT)
            "   --debugger         start example debugger\n"
            "   --reattach         automatically reattach debugger on detach\n"
#endif
            "   --recreate-heap    recreate heap after every file\n"
            "   --no-heap-destroy  force GC, but don't destroy heap at end (leak testing)\n"
                    "\n"
                    "If <filename> is omitted, interactive mode is started automatically.\n");
    fflush(stderr);
    exit(1);
}