ref: 7e27931468a7c0f41b2c8a64c9cb6b069f47a5ac
dir: /pp.c/
/* Pretty-print input source by emitting parse tree back as syntax. * with no flags: pretty-printed source * with -m: minified source with line breaks * with -mm: minified source without line breaks * with -s: s-expression syntax tree */ #include <stdio.h> #include <assert.h> #include "jsi.h" #include "utf.h" static const char *astname[] = { #include "astnames.h" NULL }; static const char *opname[] = { #include "opnames.h" NULL }; static int format = 0; static int minify = 0; static void pc(int c) { putchar(c); } static void ps(const char *s) { fputs(s, stdout); } static void in(int d) { if (minify < 1) while (d-- > 0) putchar('\t'); } static void nl(void) { if (minify < 2) putchar('\n'); } static void sp(void) { if (minify < 1) putchar(' '); } static void comma(void) { putchar(','); sp(); } static void pstr(const char *s) { static const char *HEX = "0123456789ABCDEF"; Rune c; pc(minify ? '\'' : '"'); while (*s) { s += chartorune(&c, s); switch (c) { case '\'': ps("\\'"); break; case '"': ps("\\\""); break; case '\\': ps("\\\\"); break; case '\b': ps("\\b"); break; case '\f': ps("\\f"); break; case '\n': ps("\\n"); break; case '\r': ps("\\r"); break; case '\t': ps("\\t"); break; default: if (c < ' ' || c > 127) { ps("\\u"); pc(HEX[(c>>12)&15]); pc(HEX[(c>>8)&15]); pc(HEX[(c>>4)&15]); pc(HEX[c&15]); } else { pc(c); break; } } } pc(minify ? '\'' : '"'); } static void pregexp(const char *prog, int flags) { pc('/'); while (*prog) { if (*prog == '/') pc('\\'); pc(*prog); ++prog; } pc('/'); if (flags & JS_REGEXP_G) pc('g'); if (flags & JS_REGEXP_I) pc('i'); if (flags & JS_REGEXP_M) pc('m'); } /* Bytecode */ static void jsC_dumpfunction(js_State *J, js_Function *F) { js_Instruction *p = F->code; js_Instruction *end = F->code + F->codelen; char *s; double n; int i; printf("%s(%d)\n", F->name, F->numparams); if (F->strict) printf("\tstrict\n"); if (F->lightweight) printf("\tlightweight\n"); if (F->arguments) printf("\targuments\n"); printf("\tsource %s:%d\n", F->filename, F->line); for (i = 0; i < F->funlen; ++i) printf("\tfunction %d %s\n", i, F->funtab[i]->name); for (i = 0; i < F->varlen; ++i) printf("\tlocal %d %s\n", i + 1, F->vartab[i]); printf("{\n"); while (p < end) { int ln = *p++; int c = *p++; printf("%5d(%3d): ", (int)(p - F->code) - 2, ln); ps(opname[c]); switch (c) { case OP_INTEGER: printf(" %ld", (long)((*p++) - 32768)); break; case OP_NUMBER: memcpy(&n, p, sizeof(n)); p += sizeof(n) / sizeof(*p); printf(" %.9g", n); break; case OP_STRING: memcpy(&s, p, sizeof(s)); p += sizeof(s) / sizeof(*p); pc(' '); pstr(s); break; case OP_NEWREGEXP: pc(' '); memcpy(&s, p, sizeof(s)); p += sizeof(s) / sizeof(*p); pregexp(s, *p++); break; case OP_GETVAR: case OP_HASVAR: case OP_SETVAR: case OP_DELVAR: case OP_GETPROP_S: case OP_SETPROP_S: case OP_DELPROP_S: case OP_CATCH: memcpy(&s, p, sizeof(s)); p += sizeof(s) / sizeof(*p); pc(' '); ps(s); break; case OP_GETLOCAL: case OP_SETLOCAL: case OP_DELLOCAL: printf(" %s", F->vartab[*p++ - 1]); break; case OP_CLOSURE: case OP_CALL: case OP_NEW: case OP_JUMP: case OP_JTRUE: case OP_JFALSE: case OP_JCASE: case OP_TRY: printf(" %ld", (long)*p++); break; } putchar('\n'); } printf("}\n"); for (i = 0; i < F->funlen; ++i) { if (F->funtab[i] != F) { printf("function %d ", i); jsC_dumpfunction(J, F->funtab[i]); } } } /* Pretty-printed Javascript syntax */ static int prec(enum js_AstType type) { switch (type) { case AST_IDENTIFIER: case EXP_IDENTIFIER: case EXP_NUMBER: case EXP_STRING: case EXP_REGEXP: case EXP_ELISION: case EXP_NULL: case EXP_TRUE: case EXP_FALSE: case EXP_THIS: case EXP_ARRAY: case EXP_OBJECT: return 170; case EXP_FUN: case EXP_INDEX: case EXP_MEMBER: case EXP_CALL: case EXP_NEW: return 160; case EXP_POSTINC: case EXP_POSTDEC: return 150; case EXP_DELETE: case EXP_VOID: case EXP_TYPEOF: case EXP_PREINC: case EXP_PREDEC: case EXP_POS: case EXP_NEG: case EXP_BITNOT: case EXP_LOGNOT: return 140; case EXP_MOD: case EXP_DIV: case EXP_MUL: return 130; case EXP_SUB: case EXP_ADD: return 120; case EXP_USHR: case EXP_SHR: case EXP_SHL: return 110; case EXP_IN: case EXP_INSTANCEOF: case EXP_GE: case EXP_LE: case EXP_GT: case EXP_LT: return 100; case EXP_STRICTNE: case EXP_STRICTEQ: case EXP_NE: case EXP_EQ: return 90; case EXP_BITAND: return 80; case EXP_BITXOR: return 70; case EXP_BITOR: return 60; case EXP_LOGAND: return 50; case EXP_LOGOR: return 40; case EXP_COND: return 30; case EXP_ASS: case EXP_ASS_MUL: case EXP_ASS_DIV: case EXP_ASS_MOD: case EXP_ASS_ADD: case EXP_ASS_SUB: case EXP_ASS_SHL: case EXP_ASS_SHR: case EXP_ASS_USHR: case EXP_ASS_BITAND: case EXP_ASS_BITXOR: case EXP_ASS_BITOR: return 20; #define COMMA 15 case EXP_COMMA: return 10; default: return 0; } } static void pstmlist(int d, js_Ast *list); static void pexpi(int d, int i, js_Ast *exp); static void pstm(int d, js_Ast *stm); static void slist(int d, js_Ast *list); static void sblock(int d, js_Ast *list); static void pargs(int d, js_Ast *list) { while (list) { assert(list->type == AST_LIST); pexpi(d, COMMA, list->a); list = list->b; if (list) comma(); } } static void parray(int d, js_Ast *list) { pc('['); while (list) { assert(list->type == AST_LIST); pexpi(d, COMMA, list->a); list = list->b; if (list) comma(); } pc(']'); } static void pobject(int d, js_Ast *list) { pc('{'); if (list) { nl(); in(d+1); } while (list) { js_Ast *kv = list->a; assert(list->type == AST_LIST); switch (kv->type) { default: break; case EXP_PROP_VAL: pexpi(d+1, COMMA, kv->a); pc(':'); sp(); pexpi(d+1, COMMA, kv->b); break; case EXP_PROP_GET: ps("get "); pexpi(d+1, COMMA, kv->a); ps("()"); sp(); pc('{'); nl(); pstmlist(d+1, kv->c); in(d+1); pc('}'); break; case EXP_PROP_SET: ps("set "); pexpi(d+1, COMMA, kv->a); pc('('); pargs(d+1, kv->b); pc(')'); sp(); pc('{'); nl(); pstmlist(d+1, kv->c); in(d+1); pc('}'); break; } list = list->b; if (list) { pc(','); nl(); in(d+1); } else { nl(); in(d); } } pc('}'); } static void pbin(int d, int p, js_Ast *exp, const char *op) { pexpi(d, p, exp->a); sp(); ps(op); sp(); pexpi(d, p, exp->b); } static void puna(int d, int p, js_Ast *exp, const char *pre, const char *suf) { ps(pre); pexpi(d, p, exp->a); ps(suf); } static void pexpi(int d, int p, js_Ast *exp) { int tp, paren; if (!exp) return; tp = prec(exp->type); paren = 0; if (tp < p) { pc('('); paren = 1; } p = tp; switch (exp->type) { case AST_IDENTIFIER: ps(exp->string); break; case EXP_IDENTIFIER: ps(exp->string); break; case EXP_NUMBER: printf("%.9g", exp->number); break; case EXP_STRING: pstr(exp->string); break; case EXP_REGEXP: pregexp(exp->string, exp->number); break; case EXP_ELISION: ps("elision"); break; case EXP_NULL: ps("null"); break; case EXP_TRUE: ps("true"); break; case EXP_FALSE: ps("false"); break; case EXP_THIS: ps("this"); break; case EXP_OBJECT: pobject(d, exp->a); break; case EXP_ARRAY: parray(d, exp->a); break; case EXP_DELETE: puna(d, p, exp, "delete ", ""); break; case EXP_VOID: puna(d, p, exp, "void ", ""); break; case EXP_TYPEOF: puna(d, p, exp, "typeof ", ""); break; case EXP_PREINC: puna(d, p, exp, "++", ""); break; case EXP_PREDEC: puna(d, p, exp, "--", ""); break; case EXP_POSTINC: puna(d, p, exp, "", "++"); break; case EXP_POSTDEC: puna(d, p, exp, "", "--"); break; case EXP_POS: puna(d, p, exp, "+", ""); break; case EXP_NEG: puna(d, p, exp, "-", ""); break; case EXP_BITNOT: puna(d, p, exp, "~", ""); break; case EXP_LOGNOT: puna(d, p, exp, "!", ""); break; case EXP_LOGOR: pbin(d, p, exp, "||"); break; case EXP_LOGAND: pbin(d, p, exp, "&&"); break; case EXP_BITOR: pbin(d, p, exp, "|"); break; case EXP_BITXOR: pbin(d, p, exp, "^"); break; case EXP_BITAND: pbin(d, p, exp, "&"); break; case EXP_EQ: pbin(d, p, exp, "=="); break; case EXP_NE: pbin(d, p, exp, "!="); break; case EXP_STRICTEQ: pbin(d, p, exp, "==="); break; case EXP_STRICTNE: pbin(d, p, exp, "!=="); break; case EXP_LT: pbin(d, p, exp, "<"); break; case EXP_GT: pbin(d, p, exp, ">"); break; case EXP_LE: pbin(d, p, exp, "<="); break; case EXP_GE: pbin(d, p, exp, ">="); break; case EXP_IN: pbin(d, p, exp, "in"); break; case EXP_SHL: pbin(d, p, exp, "<<"); break; case EXP_SHR: pbin(d, p, exp, ">>"); break; case EXP_USHR: pbin(d, p, exp, ">>>"); break; case EXP_ADD: pbin(d, p, exp, "+"); break; case EXP_SUB: pbin(d, p, exp, "-"); break; case EXP_MUL: pbin(d, p, exp, "*"); break; case EXP_DIV: pbin(d, p, exp, "/"); break; case EXP_MOD: pbin(d, p, exp, "%"); break; case EXP_ASS: pbin(d, p, exp, "="); break; case EXP_ASS_MUL: pbin(d, p, exp, "*="); break; case EXP_ASS_DIV: pbin(d, p, exp, "/="); break; case EXP_ASS_MOD: pbin(d, p, exp, "%="); break; case EXP_ASS_ADD: pbin(d, p, exp, "+="); break; case EXP_ASS_SUB: pbin(d, p, exp, "-="); break; case EXP_ASS_SHL: pbin(d, p, exp, "<<="); break; case EXP_ASS_SHR: pbin(d, p, exp, ">>="); break; case EXP_ASS_USHR: pbin(d, p, exp, ">>>="); break; case EXP_ASS_BITAND: pbin(d, p, exp, "&="); break; case EXP_ASS_BITXOR: pbin(d, p, exp, "^="); break; case EXP_ASS_BITOR: pbin(d, p, exp, "|="); break; case EXP_INSTANCEOF: pexpi(d, p, exp->a); ps(" instanceof "); pexpi(d, p, exp->b); break; case EXP_COMMA: pexpi(d, p, exp->a); pc(','); sp(); pexpi(d, p, exp->b); break; case EXP_COND: pexpi(d, p, exp->a); sp(); pc('?'); sp(); pexpi(d, p, exp->b); sp(); pc(':'); sp(); pexpi(d, p, exp->c); break; case EXP_INDEX: pexpi(d, p, exp->a); pc('['); pexpi(d, 0, exp->b); pc(']'); break; case EXP_MEMBER: pexpi(d, p, exp->a); pc('.'); pexpi(d, 0, exp->b); break; case EXP_CALL: pexpi(d, p, exp->a); pc('('); pargs(d, exp->b); pc(')'); break; case EXP_NEW: ps("new "); pexpi(d, p, exp->a); pc('('); pargs(d, exp->b); pc(')'); break; case EXP_FUN: if (p == 0) pc('('); ps("function "); pexpi(d, 0, exp->a); pc('('); pargs(d, exp->b); pc(')'); sp(); pc('{'); nl(); pstmlist(d, exp->c); in(d); pc('}'); if (p == 0) pc(')'); break; default: ps("<UNKNOWN>"); break; } if (paren) pc(')'); } static void pexp(int d, js_Ast *exp) { pexpi(d, 0, exp); } static void pvar(int d, js_Ast *var) { assert(var->type == EXP_VAR); pexp(d, var->a); if (var->b) { sp(); pc('='); sp(); pexp(d, var->b); } } static void pvarlist(int d, js_Ast *list) { while (list) { assert(list->type == AST_LIST); pvar(d, list->a); list = list->b; if (list) comma(); } } static void pblock(int d, js_Ast *block) { assert(block->type == STM_BLOCK); pc('{'); nl(); pstmlist(d, block->a); in(d); pc('}'); } static void pstmh(int d, js_Ast *stm) { if (stm->type == STM_BLOCK) { sp(); pblock(d, stm); } else { nl(); pstm(d+1, stm); } } static void pcaselist(int d, js_Ast *list) { while (list) { js_Ast *stm = list->a; if (stm->type == STM_CASE) { in(d); ps("case "); pexp(d, stm->a); pc(':'); nl(); pstmlist(d, stm->b); } if (stm->type == STM_DEFAULT) { in(d); ps("default:"); nl(); pstmlist(d, stm->a); } list = list->b; } } static void pstm(int d, js_Ast *stm) { if (stm->type == STM_BLOCK) { pblock(d, stm); return; } in(d); switch (stm->type) { case AST_FUNDEC: ps("function "); pexp(d, stm->a); pc('('); pargs(d, stm->b); pc(')'); sp(); pc('{'); nl(); pstmlist(d, stm->c); in(d); pc('}'); break; case STM_EMPTY: pc(';'); break; case STM_VAR: ps("var "); pvarlist(d, stm->a); pc(';'); break; case STM_IF: ps("if"); sp(); pc('('); pexp(d, stm->a); pc(')'); pstmh(d, stm->b); if (stm->c) { nl(); in(d); ps("else"); pstmh(d, stm->c); } break; case STM_DO: ps("do"); pstmh(d, stm->a); nl(); in(d); ps("while"); sp(); pc('('); pexp(d, stm->b); pc(')'); pc(';'); break; case STM_WHILE: ps("while"); sp(); pc('('); pexp(d, stm->a); pc(')'); pstmh(d, stm->b); break; case STM_FOR: ps("for"); sp(); pc('('); pexp(d, stm->a); pc(';'); sp(); pexp(d, stm->b); pc(';'); sp(); pexp(d, stm->c); pc(')'); pstmh(d, stm->d); break; case STM_FOR_VAR: ps("for"); sp(); ps("(var "); pvarlist(d, stm->a); pc(';'); sp(); pexp(d, stm->b); pc(';'); sp(); pexp(d, stm->c); pc(')'); pstmh(d, stm->d); break; case STM_FOR_IN: ps("for"); sp(); pc('('); pexp(d, stm->a); ps(" in "); pexp(d, stm->b); pc(')'); pstmh(d, stm->c); break; case STM_FOR_IN_VAR: ps("for"); sp(); ps("(var "); pvarlist(d, stm->a); ps(" in "); pexp(d, stm->b); pc(')'); pstmh(d, stm->c); break; case STM_CONTINUE: ps("continue"); if (stm->a) { pc(' '); pexp(d, stm->a); } pc(';'); break; case STM_BREAK: ps("break"); if (stm->a) { pc(' '); pexp(d, stm->a); } pc(';'); break; case STM_RETURN: ps("return"); if (stm->a) { pc(' '); pexp(d, stm->a); } pc(';'); break; case STM_WITH: ps("with"); sp(); pc('('); pexp(d, stm->a); pc(')'); pstmh(d, stm->b); break; case STM_SWITCH: ps("switch"); sp(); pc('('); pexp(d, stm->a); pc(')'); sp(); pc('{'); nl(); pcaselist(d, stm->b); in(d); pc('}'); break; case STM_THROW: ps("throw "); pexp(d, stm->a); pc(';'); break; case STM_TRY: ps("try"); if (minify && stm->a->type != STM_BLOCK) pc(' '); pstmh(d, stm->a); if (stm->b && stm->c) { nl(); in(d); ps("catch"); sp(); pc('('); pexp(d, stm->b); pc(')'); pstmh(d, stm->c); } if (stm->d) { nl(); in(d); ps("finally"); pstmh(d, stm->d); } break; case STM_LABEL: pexp(d, stm->a); pc(':'); sp(); pstm(d, stm->b); break; case STM_DEBUGGER: ps("debugger"); pc(';'); break; default: pexp(d, stm); pc(';'); } } static void pstmlist(int d, js_Ast *list) { while (list) { assert(list->type == AST_LIST); pstm(d+1, list->a); nl(); list = list->b; } } static void jsP_dumpsyntax(js_State *J, js_Ast *prog) { if (prog) { if (prog->type == AST_LIST) pstmlist(-1, prog); else { pstm(0, prog); nl(); } } if (minify > 1) putchar('\n'); } /* S-expression list representation */ static void snode(int d, js_Ast *node) { void (*afun)(int,js_Ast*) = snode; void (*bfun)(int,js_Ast*) = snode; void (*cfun)(int,js_Ast*) = snode; void (*dfun)(int,js_Ast*) = snode; if (!node) { return; } if (node->type == AST_LIST) { slist(d, node); return; } pc('('); ps(astname[node->type]); switch (node->type) { default: break; case AST_IDENTIFIER: pc(' '); ps(node->string); break; case EXP_IDENTIFIER: pc(' '); ps(node->string); break; case EXP_STRING: pc(' '); pstr(node->string); break; case EXP_REGEXP: pc(' '); pregexp(node->string, node->number); break; case EXP_NUMBER: printf(" %.9g", node->number); break; case STM_BLOCK: afun = sblock; break; case AST_FUNDEC: case EXP_FUN: cfun = sblock; break; case EXP_PROP_GET: cfun = sblock; break; case EXP_PROP_SET: cfun = sblock; break; case STM_SWITCH: bfun = sblock; break; case STM_CASE: bfun = sblock; break; case STM_DEFAULT: afun = sblock; break; } if (node->a) { pc(' '); afun(d, node->a); } if (node->b) { pc(' '); bfun(d, node->b); } if (node->c) { pc(' '); cfun(d, node->c); } if (node->d) { pc(' '); dfun(d, node->d); } pc(')'); } static void slist(int d, js_Ast *list) { pc('['); while (list) { assert(list->type == AST_LIST); snode(d, list->a); list = list->b; if (list) pc(' '); } pc(']'); } static void sblock(int d, js_Ast *list) { ps("[\n"); in(d+1); while (list) { assert(list->type == AST_LIST); snode(d+1, list->a); list = list->b; if (list) { nl(); in(d+1); } } nl(); in(d); pc(']'); } static void jsP_dumplist(js_State *J, js_Ast *prog) { if (prog) { if (prog->type == AST_LIST) sblock(0, prog); else snode(0, prog); nl(); } } static void js_ppstring(js_State *J, const char *filename, const char *source) { js_Ast *P; js_Function *F; if (js_try(J)) { jsP_freeparse(J); js_throw(J); } P = jsP_parse(J, filename, source); F = jsC_compilescript(J, P, J->default_strict); switch (format) { case 0: jsP_dumpsyntax(J, P); break; case 1: jsP_dumplist(J, P); break; case 2: jsC_dumpfunction(J, F); break; } jsP_freeparse(J); js_endtry(J); } static void js_ppfile(js_State *J, const char *filename) { FILE * volatile f = NULL; char * volatile s = NULL; int n, t; if (js_try(J)) { js_free(J, s); if (f) fclose(f); js_throw(J); } f = fopen(filename, "rb"); if (!f) { js_error(J, "cannot open file: '%s'", filename); } if (fseek(f, 0, SEEK_END) < 0) { js_error(J, "cannot seek in file: '%s'", filename); } n = ftell(f); if (n < 0) { js_error(J, "cannot tell in file: '%s'", filename); } if (fseek(f, 0, SEEK_SET) < 0) { js_error(J, "cannot seek in file: '%s'", filename); } s = js_malloc(J, n + 1); /* add space for string terminator */ if (!s) { js_error(J, "cannot allocate storage for file contents: '%s'", filename); } t = fread(s, 1, (size_t)n, f); if (t != n) { js_error(J, "cannot read data from file: '%s'", filename); } s[n] = 0; /* zero-terminate string containing file data */ js_ppstring(J, filename, s); js_endtry(J); js_free(J, s); fclose(f); } static void js_tryppfile(js_State *J, const char *file) { if (js_try(J)) { js_report(J, js_trystring(J, -1, "Error")); js_pop(J, 1); return; } js_ppfile(J, file); js_endtry(J); } int main(int argc, char **argv) { js_State *J; int i; if (argc < 2) { fprintf(stderr, "usage: mujs-pp [-m | -mm | -s | -c] input.js\n"); fprintf(stderr, " -m\tminify output\n"); fprintf(stderr, " -mm\tminify output more\n"); fprintf(stderr, " -s\tprint syntax tree\n"); fprintf(stderr, " -c\tprint bytecode\n"); } J = js_newstate(NULL, NULL, 0); for (i = 1; i < argc; ++i) { if (!strcmp(argv[i], "-m")) format = 0, minify = 1; else if (!strcmp(argv[i], "-mm")) format = 0, minify = 2; else if (!strcmp(argv[i], "-s")) format = 1, minify = 0; else if (!strcmp(argv[i], "-c")) format = 2, minify = 0; else js_tryppfile(J, argv[i]); } js_gc(J, 0); js_freestate(J); return 0; }