| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980 |
- /* 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;
- }
|