1 |
/* |
2 |
instrument-js.c - JavaScript instrumentation routines |
3 |
Copyright (C) 2007, 2008 siliconforks.com |
4 |
|
5 |
This program is free software; you can redistribute it and/or modify |
6 |
it under the terms of the GNU General Public License as published by |
7 |
the Free Software Foundation; either version 2 of the License, or |
8 |
(at your option) any later version. |
9 |
|
10 |
This program is distributed in the hope that it will be useful, |
11 |
but WITHOUT ANY WARRANTY; without even the implied warranty of |
12 |
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
13 |
GNU General Public License for more details. |
14 |
|
15 |
You should have received a copy of the GNU General Public License along |
16 |
with this program; if not, write to the Free Software Foundation, Inc., |
17 |
51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. |
18 |
*/ |
19 |
|
20 |
#include <config.h> |
21 |
|
22 |
#include "instrument-js.h" |
23 |
|
24 |
#include <assert.h> |
25 |
#include <stdlib.h> |
26 |
#include <string.h> |
27 |
|
28 |
#include <jsapi.h> |
29 |
#include <jsarena.h> |
30 |
#include <jsatom.h> |
31 |
#include <jsexn.h> |
32 |
#include <jsfun.h> |
33 |
#include <jsinterp.h> |
34 |
#include <jsiter.h> |
35 |
#include <jsparse.h> |
36 |
#include <jsregexp.h> |
37 |
#include <jsscope.h> |
38 |
#include <jsstr.h> |
39 |
|
40 |
#include "encoding.h" |
41 |
#include "global.h" |
42 |
#include "highlight.h" |
43 |
#include "resource-manager.h" |
44 |
#include "util.h" |
45 |
|
46 |
struct IfDirective { |
47 |
const jschar * condition_start; |
48 |
const jschar * condition_end; |
49 |
uint16_t start_line; |
50 |
uint16_t end_line; |
51 |
struct IfDirective * next; |
52 |
}; |
53 |
|
54 |
static bool * exclusive_directives = NULL; |
55 |
|
56 |
static JSRuntime * runtime = NULL; |
57 |
static JSContext * context = NULL; |
58 |
static JSObject * global = NULL; |
59 |
static JSVersion js_version = JSVERSION_ECMA_3; |
60 |
|
61 |
/* |
62 |
JSParseNode objects store line numbers starting from 1. |
63 |
The lines array stores line numbers starting from 0. |
64 |
*/ |
65 |
static const char * file_id = NULL; |
66 |
static char * lines = NULL; |
67 |
static uint16_t num_lines = 0; |
68 |
|
69 |
void jscoverage_set_js_version(const char * version) { |
70 |
js_version = atoi(version); |
71 |
} |
72 |
|
73 |
void jscoverage_init(void) { |
74 |
runtime = JS_NewRuntime(8L * 1024L * 1024L); |
75 |
if (runtime == NULL) { |
76 |
fatal("cannot create runtime"); |
77 |
} |
78 |
|
79 |
context = JS_NewContext(runtime, 8192); |
80 |
if (context == NULL) { |
81 |
fatal("cannot create context"); |
82 |
} |
83 |
|
84 |
JS_SetVersion(context, js_version); |
85 |
|
86 |
global = JS_NewObject(context, NULL, NULL, NULL); |
87 |
if (global == NULL) { |
88 |
fatal("cannot create global object"); |
89 |
} |
90 |
|
91 |
if (! JS_InitStandardClasses(context, global)) { |
92 |
fatal("cannot initialize standard classes"); |
93 |
} |
94 |
} |
95 |
|
96 |
void jscoverage_cleanup(void) { |
97 |
JS_DestroyContext(context); |
98 |
JS_DestroyRuntime(runtime); |
99 |
} |
100 |
|
101 |
static void print_javascript(const jschar * characters, size_t num_characters, Stream * f) { |
102 |
for (size_t i = 0; i < num_characters; i++) { |
103 |
jschar c = characters[i]; |
104 |
/* |
105 |
XXX does not handle no-break space, other unicode "space separator" |
106 |
*/ |
107 |
switch (c) { |
108 |
case 0x9: |
109 |
case 0xB: |
110 |
case 0xC: |
111 |
Stream_write_char(f, c); |
112 |
break; |
113 |
default: |
114 |
if (32 <= c && c <= 126) { |
115 |
Stream_write_char(f, c); |
116 |
} |
117 |
else { |
118 |
Stream_printf(f, "\\u%04x", c); |
119 |
} |
120 |
break; |
121 |
} |
122 |
} |
123 |
} |
124 |
|
125 |
static void print_string(JSString * s, Stream * f) { |
126 |
size_t length = JSSTRING_LENGTH(s); |
127 |
jschar * characters = JSSTRING_CHARS(s); |
128 |
for (size_t i = 0; i < length; i++) { |
129 |
jschar c = characters[i]; |
130 |
if (32 <= c && c <= 126) { |
131 |
switch (c) { |
132 |
case '"': |
133 |
Stream_write_string(f, "\\\""); |
134 |
break; |
135 |
/* |
136 |
case '\'': |
137 |
Stream_write_string(f, "\\'"); |
138 |
break; |
139 |
*/ |
140 |
case '\\': |
141 |
Stream_write_string(f, "\\\\"); |
142 |
break; |
143 |
default: |
144 |
Stream_write_char(f, c); |
145 |
break; |
146 |
} |
147 |
} |
148 |
else { |
149 |
switch (c) { |
150 |
case 0x8: |
151 |
Stream_write_string(f, "\\b"); |
152 |
break; |
153 |
case 0x9: |
154 |
Stream_write_string(f, "\\t"); |
155 |
break; |
156 |
case 0xa: |
157 |
Stream_write_string(f, "\\n"); |
158 |
break; |
159 |
/* IE doesn't support this */ |
160 |
/* |
161 |
case 0xb: |
162 |
Stream_write_string(f, "\\v"); |
163 |
break; |
164 |
*/ |
165 |
case 0xc: |
166 |
Stream_write_string(f, "\\f"); |
167 |
break; |
168 |
case 0xd: |
169 |
Stream_write_string(f, "\\r"); |
170 |
break; |
171 |
default: |
172 |
Stream_printf(f, "\\u%04x", c); |
173 |
break; |
174 |
} |
175 |
} |
176 |
} |
177 |
} |
178 |
|
179 |
static void print_string_atom(JSAtom * atom, Stream * f) { |
180 |
assert(ATOM_IS_STRING(atom)); |
181 |
JSString * s = ATOM_TO_STRING(atom); |
182 |
print_string(s, f); |
183 |
} |
184 |
|
185 |
static void print_regex(jsval value, Stream * f) { |
186 |
assert(JSVAL_IS_STRING(value)); |
187 |
JSString * s = JSVAL_TO_STRING(value); |
188 |
size_t length = JSSTRING_LENGTH(s); |
189 |
jschar * characters = JSSTRING_CHARS(s); |
190 |
for (size_t i = 0; i < length; i++) { |
191 |
jschar c = characters[i]; |
192 |
if (32 <= c && c <= 126) { |
193 |
Stream_write_char(f, c); |
194 |
} |
195 |
else { |
196 |
Stream_printf(f, "\\u%04x", c); |
197 |
} |
198 |
} |
199 |
} |
200 |
|
201 |
static void print_quoted_string_atom(JSAtom * atom, Stream * f) { |
202 |
assert(ATOM_IS_STRING(atom)); |
203 |
JSString * s = ATOM_TO_STRING(atom); |
204 |
Stream_write_char(f, '"'); |
205 |
print_string(s, f); |
206 |
Stream_write_char(f, '"'); |
207 |
} |
208 |
|
209 |
static const char * get_op(uint8 op) { |
210 |
switch(op) { |
211 |
case JSOP_OR: |
212 |
return "||"; |
213 |
case JSOP_AND: |
214 |
return "&&"; |
215 |
case JSOP_BITOR: |
216 |
return "|"; |
217 |
case JSOP_BITXOR: |
218 |
return "^"; |
219 |
case JSOP_BITAND: |
220 |
return "&"; |
221 |
case JSOP_EQ: |
222 |
return "=="; |
223 |
case JSOP_NE: |
224 |
return "!="; |
225 |
case JSOP_STRICTEQ: |
226 |
return "==="; |
227 |
case JSOP_STRICTNE: |
228 |
return "!=="; |
229 |
case JSOP_LT: |
230 |
return "<"; |
231 |
case JSOP_LE: |
232 |
return "<="; |
233 |
case JSOP_GT: |
234 |
return ">"; |
235 |
case JSOP_GE: |
236 |
return ">="; |
237 |
case JSOP_LSH: |
238 |
return "<<"; |
239 |
case JSOP_RSH: |
240 |
return ">>"; |
241 |
case JSOP_URSH: |
242 |
return ">>>"; |
243 |
case JSOP_ADD: |
244 |
return "+"; |
245 |
case JSOP_SUB: |
246 |
return "-"; |
247 |
case JSOP_MUL: |
248 |
return "*"; |
249 |
case JSOP_DIV: |
250 |
return "/"; |
251 |
case JSOP_MOD: |
252 |
return "%"; |
253 |
default: |
254 |
abort(); |
255 |
} |
256 |
} |
257 |
|
258 |
static void instrument_expression(JSParseNode * node, Stream * f); |
259 |
static void instrument_statement(JSParseNode * node, Stream * f, int indent, bool is_jscoverage_if); |
260 |
|
261 |
enum FunctionType { |
262 |
FUNCTION_NORMAL, |
263 |
FUNCTION_GETTER_OR_SETTER |
264 |
}; |
265 |
|
266 |
static void instrument_function(JSParseNode * node, Stream * f, int indent, enum FunctionType type) { |
267 |
assert(node->pn_type == TOK_FUNCTION); |
268 |
assert(node->pn_arity == PN_FUNC); |
269 |
JSObject * object = node->pn_funpob->object; |
270 |
assert(JS_ObjectIsFunction(context, object)); |
271 |
JSFunction * function = (JSFunction *) JS_GetPrivate(context, object); |
272 |
assert(function); |
273 |
assert(object == &function->object); |
274 |
Stream_printf(f, "%*s", indent, ""); |
275 |
if (type == FUNCTION_NORMAL) { |
276 |
Stream_write_string(f, "function"); |
277 |
} |
278 |
|
279 |
/* function name */ |
280 |
if (function->atom) { |
281 |
Stream_write_char(f, ' '); |
282 |
print_string_atom(function->atom, f); |
283 |
} |
284 |
|
285 |
/* |
286 |
function parameters - see JS_DecompileFunction in jsapi.cpp, which calls |
287 |
js_DecompileFunction in jsopcode.cpp |
288 |
*/ |
289 |
Stream_write_string(f, "("); |
290 |
JSArenaPool pool; |
291 |
JS_INIT_ARENA_POOL(&pool, "instrument_function", 256, 1, &context->scriptStackQuota); |
292 |
jsuword * local_names = NULL; |
293 |
if (JS_GET_LOCAL_NAME_COUNT(function)) { |
294 |
local_names = js_GetLocalNameArray(context, function, &pool); |
295 |
if (local_names == NULL) { |
296 |
fatal("out of memory"); |
297 |
} |
298 |
} |
299 |
for (int i = 0; i < function->nargs; i++) { |
300 |
if (i > 0) { |
301 |
Stream_write_string(f, ", "); |
302 |
} |
303 |
JSAtom * param = JS_LOCAL_NAME_TO_ATOM(local_names[i]); |
304 |
print_string_atom(param, f); |
305 |
} |
306 |
JS_FinishArenaPool(&pool); |
307 |
Stream_write_string(f, ") {\n"); |
308 |
|
309 |
/* function body */ |
310 |
instrument_statement(node->pn_body, f, indent + 2, false); |
311 |
|
312 |
Stream_write_string(f, "}\n"); |
313 |
} |
314 |
|
315 |
static void instrument_function_call(JSParseNode * node, Stream * f) { |
316 |
instrument_expression(node->pn_head, f); |
317 |
Stream_write_char(f, '('); |
318 |
for (struct JSParseNode * p = node->pn_head->pn_next; p != NULL; p = p->pn_next) { |
319 |
if (p != node->pn_head->pn_next) { |
320 |
Stream_write_string(f, ", "); |
321 |
} |
322 |
instrument_expression(p, f); |
323 |
} |
324 |
Stream_write_char(f, ')'); |
325 |
} |
326 |
|
327 |
/* |
328 |
See <Expressions> in jsparse.h. |
329 |
TOK_FUNCTION is handled as a statement and as an expression. |
330 |
TOK_DBLDOT is not handled (XML op). |
331 |
TOK_DEFSHARP and TOK_USESHARP are not handled. |
332 |
TOK_ANYNAME is not handled (XML op). |
333 |
TOK_AT is not handled (XML op). |
334 |
TOK_DBLCOLON is not handled. |
335 |
TOK_XML* are not handled. |
336 |
There seem to be some undocumented expressions: |
337 |
TOK_INSTANCEOF binary |
338 |
TOK_IN binary |
339 |
*/ |
340 |
static void instrument_expression(JSParseNode * node, Stream * f) { |
341 |
switch (node->pn_type) { |
342 |
case TOK_FUNCTION: |
343 |
instrument_function(node, f, 0, FUNCTION_NORMAL); |
344 |
break; |
345 |
case TOK_COMMA: |
346 |
for (struct JSParseNode * p = node->pn_head; p != NULL; p = p->pn_next) { |
347 |
if (p != node->pn_head) { |
348 |
Stream_write_string(f, ", "); |
349 |
} |
350 |
instrument_expression(p, f); |
351 |
} |
352 |
break; |
353 |
case TOK_ASSIGN: |
354 |
instrument_expression(node->pn_left, f); |
355 |
Stream_write_char(f, ' '); |
356 |
switch (node->pn_op) { |
357 |
case JSOP_ADD: |
358 |
case JSOP_SUB: |
359 |
case JSOP_MUL: |
360 |
case JSOP_MOD: |
361 |
case JSOP_LSH: |
362 |
case JSOP_RSH: |
363 |
case JSOP_URSH: |
364 |
case JSOP_BITAND: |
365 |
case JSOP_BITOR: |
366 |
case JSOP_BITXOR: |
367 |
case JSOP_DIV: |
368 |
Stream_printf(f, "%s", get_op(node->pn_op)); |
369 |
break; |
370 |
default: |
371 |
/* do nothing - it must be a simple assignment */ |
372 |
break; |
373 |
} |
374 |
Stream_write_string(f, "= "); |
375 |
instrument_expression(node->pn_right, f); |
376 |
break; |
377 |
case TOK_HOOK: |
378 |
instrument_expression(node->pn_kid1, f); |
379 |
Stream_write_string(f, "? "); |
380 |
instrument_expression(node->pn_kid2, f); |
381 |
Stream_write_string(f, ": "); |
382 |
instrument_expression(node->pn_kid3, f); |
383 |
break; |
384 |
case TOK_OR: |
385 |
case TOK_AND: |
386 |
case TOK_BITOR: |
387 |
case TOK_BITXOR: |
388 |
case TOK_BITAND: |
389 |
case TOK_EQOP: |
390 |
case TOK_RELOP: |
391 |
case TOK_SHOP: |
392 |
case TOK_PLUS: |
393 |
case TOK_MINUS: |
394 |
case TOK_STAR: |
395 |
case TOK_DIVOP: |
396 |
switch (node->pn_arity) { |
397 |
case PN_BINARY: |
398 |
instrument_expression(node->pn_left, f); |
399 |
Stream_printf(f, " %s ", get_op(node->pn_op)); |
400 |
instrument_expression(node->pn_right, f); |
401 |
break; |
402 |
case PN_LIST: |
403 |
for (struct JSParseNode * p = node->pn_head; p != NULL; p = p->pn_next) { |
404 |
if (p != node->pn_head) { |
405 |
Stream_printf(f, " %s ", get_op(node->pn_op)); |
406 |
} |
407 |
instrument_expression(p, f); |
408 |
} |
409 |
break; |
410 |
default: |
411 |
abort(); |
412 |
} |
413 |
break; |
414 |
case TOK_UNARYOP: |
415 |
switch (node->pn_op) { |
416 |
case JSOP_NEG: |
417 |
Stream_write_char(f, '-'); |
418 |
instrument_expression(node->pn_kid, f); |
419 |
break; |
420 |
case JSOP_POS: |
421 |
Stream_write_char(f, '+'); |
422 |
instrument_expression(node->pn_kid, f); |
423 |
break; |
424 |
case JSOP_NOT: |
425 |
Stream_write_char(f, '!'); |
426 |
instrument_expression(node->pn_kid, f); |
427 |
break; |
428 |
case JSOP_BITNOT: |
429 |
Stream_write_char(f, '~'); |
430 |
instrument_expression(node->pn_kid, f); |
431 |
break; |
432 |
case JSOP_TYPEOF: |
433 |
Stream_write_string(f, "typeof "); |
434 |
instrument_expression(node->pn_kid, f); |
435 |
break; |
436 |
case JSOP_VOID: |
437 |
Stream_write_string(f, "void "); |
438 |
instrument_expression(node->pn_kid, f); |
439 |
break; |
440 |
default: |
441 |
abort(); |
442 |
break; |
443 |
} |
444 |
break; |
445 |
case TOK_INC: |
446 |
case TOK_DEC: |
447 |
/* |
448 |
This is not documented, but node->pn_op tells whether it is pre- or post-increment. |
449 |
*/ |
450 |
switch (node->pn_op) { |
451 |
case JSOP_INCNAME: |
452 |
case JSOP_INCPROP: |
453 |
case JSOP_INCELEM: |
454 |
Stream_write_string(f, "++"); |
455 |
instrument_expression(node->pn_kid, f); |
456 |
break; |
457 |
case JSOP_DECNAME: |
458 |
case JSOP_DECPROP: |
459 |
case JSOP_DECELEM: |
460 |
Stream_write_string(f, "--"); |
461 |
instrument_expression(node->pn_kid, f); |
462 |
break; |
463 |
case JSOP_NAMEINC: |
464 |
case JSOP_PROPINC: |
465 |
case JSOP_ELEMINC: |
466 |
instrument_expression(node->pn_kid, f); |
467 |
Stream_write_string(f, "++"); |
468 |
break; |
469 |
case JSOP_NAMEDEC: |
470 |
case JSOP_PROPDEC: |
471 |
case JSOP_ELEMDEC: |
472 |
instrument_expression(node->pn_kid, f); |
473 |
Stream_write_string(f, "--"); |
474 |
break; |
475 |
default: |
476 |
abort(); |
477 |
break; |
478 |
} |
479 |
break; |
480 |
case TOK_NEW: |
481 |
Stream_write_string(f, "new "); |
482 |
instrument_function_call(node, f); |
483 |
break; |
484 |
case TOK_DELETE: |
485 |
Stream_write_string(f, "delete "); |
486 |
instrument_expression(node->pn_kid, f); |
487 |
break; |
488 |
case TOK_DOT: |
489 |
/* |
490 |
This may have originally been x['foo-bar']. Because the string 'foo-bar' |
491 |
contains illegal characters, we have to use the subscript syntax instead of |
492 |
the dot syntax. |
493 |
*/ |
494 |
instrument_expression(node->pn_expr, f); |
495 |
assert(ATOM_IS_STRING(node->pn_atom)); |
496 |
{ |
497 |
JSString * s = ATOM_TO_STRING(node->pn_atom); |
498 |
bool is_keyword = (js_CheckKeyword(JSSTRING_CHARS(s), JSSTRING_LENGTH(s)) != TOK_EOF); |
499 |
if (! is_keyword && js_IsIdentifier(s)) { |
500 |
Stream_write_char(f, '.'); |
501 |
print_string_atom(node->pn_atom, f); |
502 |
} |
503 |
else { |
504 |
Stream_write_char(f, '['); |
505 |
print_quoted_string_atom(node->pn_atom, f); |
506 |
Stream_write_char(f, ']'); |
507 |
} |
508 |
} |
509 |
break; |
510 |
case TOK_LB: |
511 |
instrument_expression(node->pn_left, f); |
512 |
Stream_write_char(f, '['); |
513 |
instrument_expression(node->pn_right, f); |
514 |
Stream_write_char(f, ']'); |
515 |
break; |
516 |
case TOK_LP: |
517 |
instrument_function_call(node, f); |
518 |
break; |
519 |
case TOK_RB: |
520 |
Stream_write_char(f, '['); |
521 |
for (struct JSParseNode * p = node->pn_head; p != NULL; p = p->pn_next) { |
522 |
if (p != node->pn_head) { |
523 |
Stream_write_string(f, ", "); |
524 |
} |
525 |
/* TOK_COMMA is a special case: a hole in the array */ |
526 |
if (p->pn_type != TOK_COMMA) { |
527 |
instrument_expression(p, f); |
528 |
} |
529 |
} |
530 |
if (node->pn_extra == PNX_ENDCOMMA) { |
531 |
Stream_write_char(f, ','); |
532 |
} |
533 |
Stream_write_char(f, ']'); |
534 |
break; |
535 |
case TOK_RC: |
536 |
Stream_write_char(f, '{'); |
537 |
for (struct JSParseNode * p = node->pn_head; p != NULL; p = p->pn_next) { |
538 |
assert(p->pn_type == TOK_COLON); |
539 |
if (p != node->pn_head) { |
540 |
Stream_write_string(f, ", "); |
541 |
} |
542 |
|
543 |
/* check whether this is a getter or setter */ |
544 |
switch (p->pn_op) { |
545 |
case JSOP_GETTER: |
546 |
case JSOP_SETTER: |
547 |
if (p->pn_op == JSOP_GETTER) { |
548 |
Stream_write_string(f, "get "); |
549 |
} |
550 |
else { |
551 |
Stream_write_string(f, "set "); |
552 |
} |
553 |
instrument_expression(p->pn_left, f); |
554 |
if (p->pn_right->pn_type != TOK_FUNCTION) { |
555 |
fatal("parse error: expected function"); |
556 |
} |
557 |
instrument_function(p->pn_right, f, 0, FUNCTION_GETTER_OR_SETTER); |
558 |
break; |
559 |
default: |
560 |
instrument_expression(p->pn_left, f); |
561 |
Stream_write_string(f, ": "); |
562 |
instrument_expression(p->pn_right, f); |
563 |
break; |
564 |
} |
565 |
} |
566 |
Stream_write_char(f, '}'); |
567 |
break; |
568 |
case TOK_RP: |
569 |
Stream_write_char(f, '('); |
570 |
instrument_expression(node->pn_kid, f); |
571 |
Stream_write_char(f, ')'); |
572 |
break; |
573 |
case TOK_NAME: |
574 |
print_string_atom(node->pn_atom, f); |
575 |
break; |
576 |
case TOK_STRING: |
577 |
print_quoted_string_atom(node->pn_atom, f); |
578 |
break; |
579 |
case TOK_REGEXP: |
580 |
assert(node->pn_op == JSOP_REGEXP); |
581 |
{ |
582 |
JSObject * object = node->pn_pob->object; |
583 |
jsval result; |
584 |
js_regexp_toString(context, object, &result); |
585 |
print_regex(result, f); |
586 |
} |
587 |
break; |
588 |
case TOK_NUMBER: |
589 |
/* |
590 |
A 64-bit IEEE 754 floating point number has a 52-bit fraction. |
591 |
2^(-52) = 2.22 x 10^(-16) |
592 |
Thus there are 16 significant digits. |
593 |
To keep the output simple, special-case zero. |
594 |
*/ |
595 |
if (node->pn_dval == 0.0) { |
596 |
Stream_write_string(f, "0"); |
597 |
} |
598 |
else { |
599 |
Stream_printf(f, "%.15g", node->pn_dval); |
600 |
} |
601 |
break; |
602 |
case TOK_PRIMARY: |
603 |
switch (node->pn_op) { |
604 |
case JSOP_TRUE: |
605 |
Stream_write_string(f, "true"); |
606 |
break; |
607 |
case JSOP_FALSE: |
608 |
Stream_write_string(f, "false"); |
609 |
break; |
610 |
case JSOP_NULL: |
611 |
Stream_write_string(f, "null"); |
612 |
break; |
613 |
case JSOP_THIS: |
614 |
Stream_write_string(f, "this"); |
615 |
break; |
616 |
/* jsscan.h mentions `super' ??? */ |
617 |
default: |
618 |
abort(); |
619 |
} |
620 |
break; |
621 |
case TOK_INSTANCEOF: |
622 |
instrument_expression(node->pn_left, f); |
623 |
Stream_write_string(f, " instanceof "); |
624 |
instrument_expression(node->pn_right, f); |
625 |
break; |
626 |
case TOK_IN: |
627 |
instrument_expression(node->pn_left, f); |
628 |
Stream_write_string(f, " in "); |
629 |
instrument_expression(node->pn_right, f); |
630 |
break; |
631 |
default: |
632 |
fatal("unsupported node type in file %s: %d", file_id, node->pn_type); |
633 |
} |
634 |
} |
635 |
|
636 |
static void instrument_var_statement(JSParseNode * node, Stream * f, int indent) { |
637 |
assert(node->pn_arity == PN_LIST); |
638 |
Stream_printf(f, "%*s", indent, ""); |
639 |
Stream_write_string(f, "var "); |
640 |
for (struct JSParseNode * p = node->pn_u.list.head; p != NULL; p = p->pn_next) { |
641 |
assert(p->pn_type == TOK_NAME); |
642 |
assert(p->pn_arity == PN_NAME); |
643 |
if (p != node->pn_head) { |
644 |
Stream_write_string(f, ", "); |
645 |
} |
646 |
print_string_atom(p->pn_atom, f); |
647 |
if (p->pn_expr != NULL) { |
648 |
Stream_write_string(f, " = "); |
649 |
instrument_expression(p->pn_expr, f); |
650 |
} |
651 |
} |
652 |
} |
653 |
|
654 |
static void output_statement(JSParseNode * node, Stream * f, int indent, bool is_jscoverage_if) { |
655 |
switch (node->pn_type) { |
656 |
case TOK_FUNCTION: |
657 |
instrument_function(node, f, indent, FUNCTION_NORMAL); |
658 |
break; |
659 |
case TOK_LC: |
660 |
assert(node->pn_arity == PN_LIST); |
661 |
/* |
662 |
Stream_write_string(f, "{\n"); |
663 |
*/ |
664 |
for (struct JSParseNode * p = node->pn_u.list.head; p != NULL; p = p->pn_next) { |
665 |
instrument_statement(p, f, indent, false); |
666 |
} |
667 |
/* |
668 |
Stream_printf(f, "%*s", indent, ""); |
669 |
Stream_write_string(f, "}\n"); |
670 |
*/ |
671 |
break; |
672 |
case TOK_IF: |
673 |
{ |
674 |
assert(node->pn_arity == PN_TERNARY); |
675 |
|
676 |
uint16_t line = node->pn_pos.begin.lineno; |
677 |
if (! is_jscoverage_if) { |
678 |
if (line > num_lines) { |
679 |
fatal("%s: script contains more than 65,535 lines", file_id); |
680 |
} |
681 |
if (line >= 2 && exclusive_directives[line - 2]) { |
682 |
is_jscoverage_if = true; |
683 |
} |
684 |
} |
685 |
|
686 |
Stream_printf(f, "%*s", indent, ""); |
687 |
Stream_write_string(f, "if ("); |
688 |
instrument_expression(node->pn_kid1, f); |
689 |
Stream_write_string(f, ") {\n"); |
690 |
if (is_jscoverage_if && node->pn_kid3) { |
691 |
uint16_t else_start = node->pn_kid3->pn_pos.begin.lineno; |
692 |
uint16_t else_end = node->pn_kid3->pn_pos.end.lineno + 1; |
693 |
Stream_printf(f, "%*s", indent + 2, ""); |
694 |
Stream_printf(f, "_$jscoverage['%s'].conditionals[%d] = %d;\n", file_id, else_start, else_end); |
695 |
} |
696 |
instrument_statement(node->pn_kid2, f, indent + 2, false); |
697 |
Stream_printf(f, "%*s", indent, ""); |
698 |
Stream_write_string(f, "}\n"); |
699 |
|
700 |
if (node->pn_kid3 || is_jscoverage_if) { |
701 |
Stream_printf(f, "%*s", indent, ""); |
702 |
Stream_write_string(f, "else {\n"); |
703 |
|
704 |
if (is_jscoverage_if) { |
705 |
uint16_t if_start = node->pn_kid2->pn_pos.begin.lineno + 1; |
706 |
uint16_t if_end = node->pn_kid2->pn_pos.end.lineno + 1; |
707 |
Stream_printf(f, "%*s", indent + 2, ""); |
708 |
Stream_printf(f, "_$jscoverage['%s'].conditionals[%d] = %d;\n", file_id, if_start, if_end); |
709 |
} |
710 |
|
711 |
if (node->pn_kid3) { |
712 |
instrument_statement(node->pn_kid3, f, indent + 2, is_jscoverage_if); |
713 |
} |
714 |
|
715 |
Stream_printf(f, "%*s", indent, ""); |
716 |
Stream_write_string(f, "}\n"); |
717 |
} |
718 |
|
719 |
break; |
720 |
} |
721 |
case TOK_SWITCH: |
722 |
assert(node->pn_arity == PN_BINARY); |
723 |
Stream_printf(f, "%*s", indent, ""); |
724 |
Stream_write_string(f, "switch ("); |
725 |
instrument_expression(node->pn_left, f); |
726 |
Stream_write_string(f, ") {\n"); |
727 |
for (struct JSParseNode * p = node->pn_right->pn_head; p != NULL; p = p->pn_next) { |
728 |
Stream_printf(f, "%*s", indent, ""); |
729 |
switch (p->pn_type) { |
730 |
case TOK_CASE: |
731 |
Stream_write_string(f, "case "); |
732 |
instrument_expression(p->pn_left, f); |
733 |
Stream_write_string(f, ":\n"); |
734 |
break; |
735 |
case TOK_DEFAULT: |
736 |
Stream_write_string(f, "default:\n"); |
737 |
break; |
738 |
default: |
739 |
abort(); |
740 |
break; |
741 |
} |
742 |
instrument_statement(p->pn_right, f, indent + 2, false); |
743 |
} |
744 |
Stream_printf(f, "%*s", indent, ""); |
745 |
Stream_write_string(f, "}\n"); |
746 |
break; |
747 |
case TOK_CASE: |
748 |
case TOK_DEFAULT: |
749 |
abort(); |
750 |
break; |
751 |
case TOK_WHILE: |
752 |
assert(node->pn_arity == PN_BINARY); |
753 |
Stream_printf(f, "%*s", indent, ""); |
754 |
Stream_write_string(f, "while ("); |
755 |
instrument_expression(node->pn_left, f); |
756 |
Stream_write_string(f, ") {\n"); |
757 |
instrument_statement(node->pn_right, f, indent + 2, false); |
758 |
Stream_write_string(f, "}\n"); |
759 |
break; |
760 |
case TOK_DO: |
761 |
assert(node->pn_arity == PN_BINARY); |
762 |
Stream_printf(f, "%*s", indent, ""); |
763 |
Stream_write_string(f, "do {\n"); |
764 |
instrument_statement(node->pn_left, f, indent + 2, false); |
765 |
Stream_write_string(f, "}\n"); |
766 |
Stream_printf(f, "%*s", indent, ""); |
767 |
Stream_write_string(f, "while ("); |
768 |
instrument_expression(node->pn_right, f); |
769 |
Stream_write_string(f, ");\n"); |
770 |
break; |
771 |
case TOK_FOR: |
772 |
assert(node->pn_arity == PN_BINARY); |
773 |
Stream_printf(f, "%*s", indent, ""); |
774 |
Stream_write_string(f, "for "); |
775 |
if (node->pn_iflags & JSITER_FOREACH) { |
776 |
Stream_write_string(f, "each "); |
777 |
} |
778 |
Stream_write_char(f, '('); |
779 |
switch (node->pn_left->pn_type) { |
780 |
case TOK_IN: |
781 |
/* for/in */ |
782 |
assert(node->pn_left->pn_arity == PN_BINARY); |
783 |
switch (node->pn_left->pn_left->pn_type) { |
784 |
case TOK_VAR: |
785 |
instrument_var_statement(node->pn_left->pn_left, f, 0); |
786 |
break; |
787 |
case TOK_NAME: |
788 |
instrument_expression(node->pn_left->pn_left, f); |
789 |
break; |
790 |
default: |
791 |
/* this is undocumented: for (x.value in y) */ |
792 |
instrument_expression(node->pn_left->pn_left, f); |
793 |
break; |
794 |
/* |
795 |
default: |
796 |
fprintf(stderr, "unexpected node type: %d\n", node->pn_left->pn_left->pn_type); |
797 |
abort(); |
798 |
break; |
799 |
*/ |
800 |
} |
801 |
Stream_write_string(f, " in "); |
802 |
instrument_expression(node->pn_left->pn_right, f); |
803 |
break; |
804 |
case TOK_RESERVED: |
805 |
/* for (;;) */ |
806 |
assert(node->pn_left->pn_arity == PN_TERNARY); |
807 |
if (node->pn_left->pn_kid1) { |
808 |
if (node->pn_left->pn_kid1->pn_type == TOK_VAR) { |
809 |
instrument_var_statement(node->pn_left->pn_kid1, f, 0); |
810 |
} |
811 |
else { |
812 |
instrument_expression(node->pn_left->pn_kid1, f); |
813 |
} |
814 |
} |
815 |
Stream_write_string(f, ";"); |
816 |
if (node->pn_left->pn_kid2) { |
817 |
Stream_write_char(f, ' '); |
818 |
instrument_expression(node->pn_left->pn_kid2, f); |
819 |
} |
820 |
Stream_write_string(f, ";"); |
821 |
if (node->pn_left->pn_kid3) { |
822 |
Stream_write_char(f, ' '); |
823 |
instrument_expression(node->pn_left->pn_kid3, f); |
824 |
} |
825 |
break; |
826 |
default: |
827 |
abort(); |
828 |
break; |
829 |
} |
830 |
Stream_write_string(f, ") {\n"); |
831 |
instrument_statement(node->pn_right, f, indent + 2, false); |
832 |
Stream_write_string(f, "}\n"); |
833 |
break; |
834 |
case TOK_THROW: |
835 |
assert(node->pn_arity == PN_UNARY); |
836 |
Stream_printf(f, "%*s", indent, ""); |
837 |
Stream_write_string(f, "throw "); |
838 |
instrument_expression(node->pn_u.unary.kid, f); |
839 |
Stream_write_string(f, ";\n"); |
840 |
break; |
841 |
case TOK_TRY: |
842 |
Stream_printf(f, "%*s", indent, ""); |
843 |
Stream_write_string(f, "try {\n"); |
844 |
instrument_statement(node->pn_kid1, f, indent + 2, false); |
845 |
Stream_printf(f, "%*s", indent, ""); |
846 |
Stream_write_string(f, "}\n"); |
847 |
if (node->pn_kid2) { |
848 |
assert(node->pn_kid2->pn_type == TOK_RESERVED); |
849 |
for (JSParseNode * scope = node->pn_kid2->pn_head; scope != NULL; scope = scope->pn_next) { |
850 |
assert(scope->pn_type == TOK_LEXICALSCOPE); |
851 |
JSParseNode * catch = scope->pn_expr; |
852 |
assert(catch->pn_type == TOK_CATCH); |
853 |
Stream_printf(f, "%*s", indent, ""); |
854 |
Stream_write_string(f, "catch ("); |
855 |
assert(catch->pn_kid1->pn_arity == PN_NAME); |
856 |
print_string_atom(catch->pn_kid1->pn_atom, f); |
857 |
if (catch->pn_kid2) { |
858 |
Stream_write_string(f, " if "); |
859 |
instrument_expression(catch->pn_kid2, f); |
860 |
} |
861 |
Stream_write_string(f, ") {\n"); |
862 |
instrument_statement(catch->pn_kid3, f, indent + 2, false); |
863 |
Stream_printf(f, "%*s", indent, ""); |
864 |
Stream_write_string(f, "}\n"); |
865 |
} |
866 |
} |
867 |
if (node->pn_kid3) { |
868 |
Stream_printf(f, "%*s", indent, ""); |
869 |
Stream_write_string(f, "finally {\n"); |
870 |
instrument_statement(node->pn_kid3, f, indent + 2, false); |
871 |
Stream_printf(f, "%*s", indent, ""); |
872 |
Stream_write_string(f, "}\n"); |
873 |
} |
874 |
break; |
875 |
case TOK_CATCH: |
876 |
abort(); |
877 |
break; |
878 |
case TOK_BREAK: |
879 |
case TOK_CONTINUE: |
880 |
assert(node->pn_arity == PN_NAME || node->pn_arity == PN_NULLARY); |
881 |
Stream_printf(f, "%*s", indent, ""); |
882 |
Stream_write_string(f, node->pn_type == TOK_BREAK? "break": "continue"); |
883 |
JSAtom * atom = node->pn_u.name.atom; |
884 |
if (atom != NULL) { |
885 |
Stream_write_char(f, ' '); |
886 |
print_string_atom(node->pn_atom, f); |
887 |
} |
888 |
Stream_write_string(f, ";\n"); |
889 |
break; |
890 |
case TOK_WITH: |
891 |
assert(node->pn_arity == PN_BINARY); |
892 |
Stream_printf(f, "%*s", indent, ""); |
893 |
Stream_write_string(f, "with ("); |
894 |
instrument_expression(node->pn_left, f); |
895 |
Stream_write_string(f, ") {\n"); |
896 |
instrument_statement(node->pn_right, f, indent + 2, false); |
897 |
Stream_printf(f, "%*s", indent, ""); |
898 |
Stream_write_string(f, "}\n"); |
899 |
break; |
900 |
case TOK_VAR: |
901 |
instrument_var_statement(node, f, indent); |
902 |
Stream_write_string(f, ";\n"); |
903 |
break; |
904 |
case TOK_RETURN: |
905 |
assert(node->pn_arity == PN_UNARY); |
906 |
Stream_printf(f, "%*s", indent, ""); |
907 |
Stream_write_string(f, "return"); |
908 |
if (node->pn_kid != NULL) { |
909 |
Stream_write_char(f, ' '); |
910 |
instrument_expression(node->pn_kid, f); |
911 |
} |
912 |
Stream_write_string(f, ";\n"); |
913 |
break; |
914 |
case TOK_SEMI: |
915 |
assert(node->pn_arity == PN_UNARY); |
916 |
Stream_printf(f, "%*s", indent, ""); |
917 |
if (node->pn_kid != NULL) { |
918 |
instrument_expression(node->pn_kid, f); |
919 |
} |
920 |
Stream_write_string(f, ";\n"); |
921 |
break; |
922 |
case TOK_COLON: |
923 |
assert(node->pn_arity == PN_NAME); |
924 |
/* |
925 |
This one is tricky: can't output instrumentation between the label and the |
926 |
statement it's supposed to label ... |
927 |
*/ |
928 |
Stream_printf(f, "%*s", indent < 2? 0: indent - 2, ""); |
929 |
print_string_atom(node->pn_atom, f); |
930 |
Stream_write_string(f, ":\n"); |
931 |
/* |
932 |
... use output_statement instead of instrument_statement. |
933 |
*/ |
934 |
output_statement(node->pn_expr, f, indent, false); |
935 |
break; |
936 |
default: |
937 |
fatal("unsupported node type in file %s: %d", file_id, node->pn_type); |
938 |
} |
939 |
} |
940 |
|
941 |
/* |
942 |
See <Statements> in jsparse.h. |
943 |
TOK_FUNCTION is handled as a statement and as an expression. |
944 |
TOK_EXPORT, TOK_IMPORT are not handled. |
945 |
*/ |
946 |
static void instrument_statement(JSParseNode * node, Stream * f, int indent, bool is_jscoverage_if) { |
947 |
if (node->pn_type != TOK_LC) { |
948 |
uint16_t line = node->pn_pos.begin.lineno; |
949 |
if (line > num_lines) { |
950 |
fatal("%s: script contains more than 65,535 lines", file_id); |
951 |
} |
952 |
|
953 |
/* the root node has line number 0 */ |
954 |
if (line != 0) { |
955 |
Stream_printf(f, "%*s", indent, ""); |
956 |
Stream_printf(f, "_$jscoverage['%s'][%d]++;\n", file_id, line); |
957 |
lines[line - 1] = 1; |
958 |
} |
959 |
} |
960 |
output_statement(node, f, indent, is_jscoverage_if); |
961 |
} |
962 |
|
963 |
static bool characters_start_with(const jschar * characters, size_t line_start, size_t line_end, const char * prefix) { |
964 |
const jschar * characters_end = characters + line_end; |
965 |
const jschar * cp = characters + line_start; |
966 |
const char * bp = prefix; |
967 |
for (;;) { |
968 |
if (*bp == '\0') { |
969 |
return true; |
970 |
} |
971 |
else if (cp == characters_end) { |
972 |
return false; |
973 |
} |
974 |
else if (*cp != *bp) { |
975 |
return false; |
976 |
} |
977 |
bp++; |
978 |
cp++; |
979 |
} |
980 |
} |
981 |
|
982 |
static bool characters_are_white_space(const jschar * characters, size_t line_start, size_t line_end) { |
983 |
/* XXX - other Unicode space */ |
984 |
const jschar * end = characters + line_end; |
985 |
for (const jschar * p = characters + line_start; p < end; p++) { |
986 |
jschar c = *p; |
987 |
if (c == 0x9 || c == 0xB || c == 0xC || c == 0x20 || c == 0xA0) { |
988 |
continue; |
989 |
} |
990 |
else { |
991 |
return false; |
992 |
} |
993 |
} |
994 |
return true; |
995 |
} |
996 |
|
997 |
static void error_reporter(JSContext * context, const char * message, JSErrorReport * report) { |
998 |
fprintf(stderr, "jscoverage: parse error: line %u: %s\n", report->lineno, message); |
999 |
} |
1000 |
|
1001 |
void jscoverage_instrument_js(const char * id, const uint16_t * characters, size_t num_characters, Stream * output) { |
1002 |
file_id = id; |
1003 |
|
1004 |
/* parse the javascript */ |
1005 |
JSParseContext parse_context; |
1006 |
if (! js_InitParseContext(context, &parse_context, NULL, NULL, characters, num_characters, NULL, NULL, 1)) { |
1007 |
fatal("cannot create token stream from file: %s", file_id); |
1008 |
} |
1009 |
JSErrorReporter old_error_reporter = JS_SetErrorReporter(context, error_reporter); |
1010 |
JSParseNode * node = js_ParseScript(context, global, &parse_context); |
1011 |
if (node == NULL) { |
1012 |
js_ReportUncaughtException(context); |
1013 |
fatal("parse error in file: %s", file_id); |
1014 |
} |
1015 |
JS_SetErrorReporter(context, old_error_reporter); |
1016 |
num_lines = node->pn_pos.end.lineno; |
1017 |
lines = xmalloc(num_lines); |
1018 |
for (unsigned int i = 0; i < num_lines; i++) { |
1019 |
lines[i] = 0; |
1020 |
} |
1021 |
|
1022 |
/* search code for conditionals */ |
1023 |
exclusive_directives = xnew(bool, num_lines); |
1024 |
for (unsigned int i = 0; i < num_lines; i++) { |
1025 |
exclusive_directives[i] = false; |
1026 |
} |
1027 |
|
1028 |
bool has_conditionals = false; |
1029 |
struct IfDirective * if_directives = NULL; |
1030 |
size_t line_number = 0; |
1031 |
size_t i = 0; |
1032 |
while (i < num_characters) { |
1033 |
if (line_number == UINT16_MAX) { |
1034 |
fatal("%s: script has more than 65,535 lines", file_id); |
1035 |
} |
1036 |
line_number++; |
1037 |
size_t line_start = i; |
1038 |
jschar c; |
1039 |
bool done = false; |
1040 |
while (! done && i < num_characters) { |
1041 |
c = characters[i]; |
1042 |
switch (c) { |
1043 |
case '\r': |
1044 |
case '\n': |
1045 |
case 0x2028: |
1046 |
case 0x2029: |
1047 |
done = true; |
1048 |
break; |
1049 |
default: |
1050 |
i++; |
1051 |
} |
1052 |
} |
1053 |
size_t line_end = i; |
1054 |
if (i < num_characters) { |
1055 |
i++; |
1056 |
if (c == '\r' && i < num_characters && characters[i] == '\n') { |
1057 |
i++; |
1058 |
} |
1059 |
} |
1060 |
|
1061 |
if (characters_start_with(characters, line_start, line_end, "//#JSCOVERAGE_IF")) { |
1062 |
has_conditionals = true; |
1063 |
|
1064 |
if (characters_are_white_space(characters, line_start + 16, line_end)) { |
1065 |
exclusive_directives[line_number - 1] = true; |
1066 |
} |
1067 |
else { |
1068 |
struct IfDirective * if_directive = xnew(struct IfDirective, 1); |
1069 |
if_directive->condition_start = characters + line_start + 16; |
1070 |
if_directive->condition_end = characters + line_end; |
1071 |
if_directive->start_line = line_number; |
1072 |
if_directive->end_line = 0; |
1073 |
if_directive->next = if_directives; |
1074 |
if_directives = if_directive; |
1075 |
} |
1076 |
} |
1077 |
else if (characters_start_with(characters, line_start, line_end, "//#JSCOVERAGE_ENDIF")) { |
1078 |
for (struct IfDirective * p = if_directives; p != NULL; p = p->next) { |
1079 |
if (p->end_line == 0) { |
1080 |
p->end_line = line_number; |
1081 |
break; |
1082 |
} |
1083 |
} |
1084 |
} |
1085 |
} |
1086 |
|
1087 |
/* |
1088 |
An instrumented JavaScript file has 4 sections: |
1089 |
1. initialization |
1090 |
2. instrumented source code |
1091 |
3. conditionals |
1092 |
4. original source code |
1093 |
*/ |
1094 |
|
1095 |
Stream * instrumented = Stream_new(0); |
1096 |
instrument_statement(node, instrumented, 0, false); |
1097 |
js_FinishParseContext(context, &parse_context); |
1098 |
|
1099 |
/* write line number info to the output */ |
1100 |
Stream_write_string(output, "/* automatically generated by JSCoverage - do not edit */\n"); |
1101 |
Stream_write_string(output, "if (! top._$jscoverage) {\n top._$jscoverage = {};\n}\n"); |
1102 |
Stream_write_string(output, "var _$jscoverage = top._$jscoverage;\n"); |
1103 |
Stream_printf(output, "if (! _$jscoverage['%s']) {\n", file_id); |
1104 |
Stream_printf(output, " _$jscoverage['%s'] = [];\n", file_id); |
1105 |
for (int i = 0; i < num_lines; i++) { |
1106 |
if (lines[i]) { |
1107 |
Stream_printf(output, " _$jscoverage['%s'][%d] = 0;\n", file_id, i + 1); |
1108 |
} |
1109 |
} |
1110 |
Stream_write_string(output, "}\n"); |
1111 |
free(lines); |
1112 |
lines = NULL; |
1113 |
free(exclusive_directives); |
1114 |
exclusive_directives = NULL; |
1115 |
|
1116 |
/* conditionals */ |
1117 |
if (has_conditionals) { |
1118 |
Stream_printf(output, "_$jscoverage['%s'].conditionals = [];\n", file_id); |
1119 |
} |
1120 |
|
1121 |
/* copy the instrumented source code to the output */ |
1122 |
Stream_write(output, instrumented->data, instrumented->length); |
1123 |
|
1124 |
/* conditionals */ |
1125 |
for (struct IfDirective * if_directive = if_directives; if_directive != NULL; if_directive = if_directive->next) { |
1126 |
Stream_write_string(output, "if (!("); |
1127 |
print_javascript(if_directive->condition_start, if_directive->condition_end - if_directive->condition_start, output); |
1128 |
Stream_write_string(output, ")) {\n"); |
1129 |
Stream_printf(output, " _$jscoverage['%s'].conditionals[%d] = %d;\n", file_id, if_directive->start_line, if_directive->end_line); |
1130 |
Stream_write_string(output, "}\n"); |
1131 |
} |
1132 |
|
1133 |
/* free */ |
1134 |
while (if_directives != NULL) { |
1135 |
struct IfDirective * if_directive = if_directives; |
1136 |
if_directives = if_directives->next; |
1137 |
free(if_directive); |
1138 |
} |
1139 |
|
1140 |
/* copy the original source to the output */ |
1141 |
Stream_printf(output, "_$jscoverage['%s'].source = ", file_id); |
1142 |
jscoverage_write_source(id, characters, num_characters, output); |
1143 |
Stream_printf(output, ";\n"); |
1144 |
|
1145 |
Stream_delete(instrumented); |
1146 |
|
1147 |
file_id = NULL; |
1148 |
} |
1149 |
|
1150 |
void jscoverage_write_source(const char * id, const jschar * characters, size_t num_characters, Stream * output) { |
1151 |
Stream_write_string(output, "["); |
1152 |
if (jscoverage_highlight) { |
1153 |
Stream * highlighted_stream = Stream_new(num_characters); |
1154 |
jscoverage_highlight_js(context, id, characters, num_characters, highlighted_stream); |
1155 |
size_t i = 0; |
1156 |
while (i < highlighted_stream->length) { |
1157 |
if (i > 0) { |
1158 |
Stream_write_char(output, ','); |
1159 |
} |
1160 |
|
1161 |
Stream_write_char(output, '"'); |
1162 |
bool done = false; |
1163 |
while (! done) { |
1164 |
char c = highlighted_stream->data[i]; |
1165 |
switch (c) { |
1166 |
case 0x8: |
1167 |
/* backspace */ |
1168 |
Stream_write_string(output, "\\b"); |
1169 |
break; |
1170 |
case 0x9: |
1171 |
/* horizontal tab */ |
1172 |
Stream_write_string(output, "\\t"); |
1173 |
break; |
1174 |
case 0xa: |
1175 |
/* line feed (new line) */ |
1176 |
done = true; |
1177 |
break; |
1178 |
/* IE doesn't support this */ |
1179 |
/* |
1180 |
case 0xb: |
1181 |
Stream_write_string(output, "\\v"); |
1182 |
break; |
1183 |
*/ |
1184 |
case 0xc: |
1185 |
/* form feed */ |
1186 |
Stream_write_string(output, "\\f"); |
1187 |
break; |
1188 |
case 0xd: |
1189 |
/* carriage return */ |
1190 |
done = true; |
1191 |
if (i + 1 < highlighted_stream->length && highlighted_stream->data[i + 1] == '\n') { |
1192 |
i++; |
1193 |
} |
1194 |
break; |
1195 |
case '"': |
1196 |
Stream_write_string(output, "\\\""); |
1197 |
break; |
1198 |
case '\\': |
1199 |
Stream_write_string(output, "\\\\"); |
1200 |
break; |
1201 |
default: |
1202 |
Stream_write_char(output, c); |
1203 |
break; |
1204 |
} |
1205 |
i++; |
1206 |
if (i >= highlighted_stream->length) { |
1207 |
done = true; |
1208 |
} |
1209 |
} |
1210 |
Stream_write_char(output, '"'); |
1211 |
} |
1212 |
Stream_delete(highlighted_stream); |
1213 |
} |
1214 |
else { |
1215 |
size_t i = 0; |
1216 |
while (i < num_characters) { |
1217 |
if (i > 0) { |
1218 |
Stream_write_char(output, ','); |
1219 |
} |
1220 |
|
1221 |
Stream_write_char(output, '"'); |
1222 |
bool done = false; |
1223 |
while (! done) { |
1224 |
jschar c = characters[i]; |
1225 |
switch (c) { |
1226 |
case 0x8: |
1227 |
/* backspace */ |
1228 |
Stream_write_string(output, "\\b"); |
1229 |
break; |
1230 |
case 0x9: |
1231 |
/* horizontal tab */ |
1232 |
Stream_write_string(output, "\\t"); |
1233 |
break; |
1234 |
case 0xa: |
1235 |
/* line feed (new line) */ |
1236 |
done = true; |
1237 |
break; |
1238 |
/* IE doesn't support this */ |
1239 |
/* |
1240 |
case 0xb: |
1241 |
Stream_write_string(output, "\\v"); |
1242 |
break; |
1243 |
*/ |
1244 |
case 0xc: |
1245 |
/* form feed */ |
1246 |
Stream_write_string(output, "\\f"); |
1247 |
break; |
1248 |
case 0xd: |
1249 |
/* carriage return */ |
1250 |
done = true; |
1251 |
if (i + 1 < num_characters && characters[i + 1] == '\n') { |
1252 |
i++; |
1253 |
} |
1254 |
break; |
1255 |
case '"': |
1256 |
Stream_write_string(output, "\\\""); |
1257 |
break; |
1258 |
case '\\': |
1259 |
Stream_write_string(output, "\\\\"); |
1260 |
break; |
1261 |
case '&': |
1262 |
Stream_write_string(output, "&"); |
1263 |
break; |
1264 |
case '<': |
1265 |
Stream_write_string(output, "<"); |
1266 |
break; |
1267 |
case '>': |
1268 |
Stream_write_string(output, ">"); |
1269 |
break; |
1270 |
case 0x2028: |
1271 |
case 0x2029: |
1272 |
done = true; |
1273 |
break; |
1274 |
default: |
1275 |
if (32 <= c && c <= 126) { |
1276 |
Stream_write_char(output, c); |
1277 |
} |
1278 |
else { |
1279 |
Stream_printf(output, "&#%d;", c); |
1280 |
} |
1281 |
break; |
1282 |
} |
1283 |
i++; |
1284 |
if (i >= num_characters) { |
1285 |
done = true; |
1286 |
} |
1287 |
} |
1288 |
Stream_write_char(output, '"'); |
1289 |
} |
1290 |
} |
1291 |
Stream_write_string(output, "]"); |
1292 |
} |
1293 |
|
1294 |
void jscoverage_copy_resources(const char * destination_directory) { |
1295 |
copy_resource("jscoverage.html", destination_directory); |
1296 |
copy_resource("jscoverage.css", destination_directory); |
1297 |
copy_resource("jscoverage.js", destination_directory); |
1298 |
copy_resource("jscoverage-ie.css", destination_directory); |
1299 |
copy_resource("jscoverage-throbber.gif", destination_directory); |
1300 |
copy_resource("jscoverage-highlight.css", destination_directory); |
1301 |
} |
1302 |
|
1303 |
/* |
1304 |
coverage reports |
1305 |
*/ |
1306 |
|
1307 |
struct FileCoverageList { |
1308 |
FileCoverage * file_coverage; |
1309 |
struct FileCoverageList * next; |
1310 |
}; |
1311 |
|
1312 |
struct Coverage { |
1313 |
JSHashTable * coverage_table; |
1314 |
struct FileCoverageList * coverage_list; |
1315 |
}; |
1316 |
|
1317 |
static int compare_strings(const void * p1, const void * p2) { |
1318 |
return strcmp(p1, p2) == 0; |
1319 |
} |
1320 |
|
1321 |
Coverage * Coverage_new(void) { |
1322 |
Coverage * result = xmalloc(sizeof(Coverage)); |
1323 |
result->coverage_table = JS_NewHashTable(1024, JS_HashString, compare_strings, NULL, NULL, NULL); |
1324 |
if (result->coverage_table == NULL) { |
1325 |
fatal("cannot create hash table"); |
1326 |
} |
1327 |
result->coverage_list = NULL; |
1328 |
return result; |
1329 |
} |
1330 |
|
1331 |
void Coverage_delete(Coverage * coverage) { |
1332 |
JS_HashTableDestroy(coverage->coverage_table); |
1333 |
struct FileCoverageList * p = coverage->coverage_list; |
1334 |
while (p != NULL) { |
1335 |
free(p->file_coverage->coverage_lines); |
1336 |
if (p->file_coverage->source_lines != NULL) { |
1337 |
for (uint32 i = 0; i < p->file_coverage->num_source_lines; i++) { |
1338 |
free(p->file_coverage->source_lines[i]); |
1339 |
} |
1340 |
free(p->file_coverage->source_lines); |
1341 |
} |
1342 |
free(p->file_coverage->id); |
1343 |
free(p->file_coverage); |
1344 |
struct FileCoverageList * q = p; |
1345 |
p = p->next; |
1346 |
free(q); |
1347 |
} |
1348 |
free(coverage); |
1349 |
} |
1350 |
|
1351 |
struct EnumeratorArg { |
1352 |
CoverageForeachFunction f; |
1353 |
void * p; |
1354 |
}; |
1355 |
|
1356 |
static intN enumerator(JSHashEntry * entry, intN i, void * arg) { |
1357 |
struct EnumeratorArg * enumerator_arg = arg; |
1358 |
enumerator_arg->f(entry->value, i, enumerator_arg->p); |
1359 |
return 0; |
1360 |
} |
1361 |
|
1362 |
void Coverage_foreach_file(Coverage * coverage, CoverageForeachFunction f, void * p) { |
1363 |
struct EnumeratorArg enumerator_arg; |
1364 |
enumerator_arg.f = f; |
1365 |
enumerator_arg.p = p; |
1366 |
JS_HashTableEnumerateEntries(coverage->coverage_table, enumerator, &enumerator_arg); |
1367 |
} |
1368 |
|
1369 |
int jscoverage_parse_json(Coverage * coverage, const uint8_t * json, size_t length) { |
1370 |
int result = 0; |
1371 |
|
1372 |
jschar * base = js_InflateString(context, (char *) json, &length); |
1373 |
if (base == NULL) { |
1374 |
fatal("out of memory"); |
1375 |
} |
1376 |
|
1377 |
jschar * parenthesized_json = xnew(jschar, addst(length, 2)); |
1378 |
parenthesized_json[0] = '('; |
1379 |
memcpy(parenthesized_json + 1, base, mulst(length, sizeof(jschar))); |
1380 |
parenthesized_json[length + 1] = ')'; |
1381 |
|
1382 |
JS_free(context, base); |
1383 |
|
1384 |
JSParseContext parse_context; |
1385 |
if (! js_InitParseContext(context, &parse_context, NULL, NULL, parenthesized_json, length + 2, NULL, NULL, 1)) { |
1386 |
free(parenthesized_json); |
1387 |
return -1; |
1388 |
} |
1389 |
JSParseNode * root = js_ParseScript(context, global, &parse_context); |
1390 |
free(parenthesized_json); |
1391 |
if (root == NULL) { |
1392 |
result = -1; |
1393 |
goto done; |
1394 |
} |
1395 |
|
1396 |
/* root node must be TOK_LC */ |
1397 |
if (root->pn_type != TOK_LC) { |
1398 |
result = -1; |
1399 |
goto done; |
1400 |
} |
1401 |
JSParseNode * semi = root->pn_u.list.head; |
1402 |
|
1403 |
/* the list must be TOK_SEMI and it must contain only one element */ |
1404 |
if (semi->pn_type != TOK_SEMI || semi->pn_next != NULL) { |
1405 |
result = -1; |
1406 |
goto done; |
1407 |
} |
1408 |
JSParseNode * parenthesized = semi->pn_kid; |
1409 |
|
1410 |
/* this must be a parenthesized expression */ |
1411 |
if (parenthesized->pn_type != TOK_RP) { |
1412 |
result = -1; |
1413 |
goto done; |
1414 |
} |
1415 |
JSParseNode * object = parenthesized->pn_kid; |
1416 |
|
1417 |
/* this must be an object literal */ |
1418 |
if (object->pn_type != TOK_RC) { |
1419 |
result = -1; |
1420 |
goto done; |
1421 |
} |
1422 |
|
1423 |
for (JSParseNode * p = object->pn_head; p != NULL; p = p->pn_next) { |
1424 |
/* every element of this list must be TOK_COLON */ |
1425 |
if (p->pn_type != TOK_COLON) { |
1426 |
result = -1; |
1427 |
goto done; |
1428 |
} |
1429 |
|
1430 |
/* the key must be a string representing the file */ |
1431 |
JSParseNode * key = p->pn_left; |
1432 |
if (key->pn_type != TOK_STRING || ! ATOM_IS_STRING(key->pn_atom)) { |
1433 |
result = -1; |
1434 |
goto done; |
1435 |
} |
1436 |
char * id_bytes = JS_GetStringBytes(ATOM_TO_STRING(key->pn_atom)); |
1437 |
|
1438 |
/* the value must be an object literal OR an array */ |
1439 |
JSParseNode * value = p->pn_right; |
1440 |
if (! (value->pn_type == TOK_RC || value->pn_type == TOK_RB)) { |
1441 |
result = -1; |
1442 |
goto done; |
1443 |
} |
1444 |
|
1445 |
JSParseNode * array = NULL; |
1446 |
JSParseNode * source = NULL; |
1447 |
if (value->pn_type == TOK_RB) { |
1448 |
/* an array */ |
1449 |
array = value; |
1450 |
} |
1451 |
else if (value->pn_type == TOK_RC) { |
1452 |
/* an object literal */ |
1453 |
if (value->pn_count != 2) { |
1454 |
result = -1; |
1455 |
goto done; |
1456 |
} |
1457 |
for (JSParseNode * element = value->pn_head; element != NULL; element = element->pn_next) { |
1458 |
if (element->pn_type != TOK_COLON) { |
1459 |
result = -1; |
1460 |
goto done; |
1461 |
} |
1462 |
JSParseNode * left = element->pn_left; |
1463 |
if (left->pn_type != TOK_STRING || ! ATOM_IS_STRING(left->pn_atom)) { |
1464 |
result = -1; |
1465 |
goto done; |
1466 |
} |
1467 |
const char * s = JS_GetStringBytes(ATOM_TO_STRING(left->pn_atom)); |
1468 |
if (strcmp(s, "coverage") == 0) { |
1469 |
array = element->pn_right; |
1470 |
if (array->pn_type != TOK_RB) { |
1471 |
result = -1; |
1472 |
goto done; |
1473 |
} |
1474 |
} |
1475 |
else if (strcmp(s, "source") == 0) { |
1476 |
source = element->pn_right; |
1477 |
if (source->pn_type != TOK_RB) { |
1478 |
result = -1; |
1479 |
goto done; |
1480 |
} |
1481 |
} |
1482 |
else { |
1483 |
result = -1; |
1484 |
goto done; |
1485 |
} |
1486 |
} |
1487 |
} |
1488 |
else { |
1489 |
result = -1; |
1490 |
goto done; |
1491 |
} |
1492 |
|
1493 |
if (array == NULL) { |
1494 |
result = -1; |
1495 |
goto done; |
1496 |
} |
1497 |
|
1498 |
/* look up the file in the coverage table */ |
1499 |
FileCoverage * file_coverage = JS_HashTableLookup(coverage->coverage_table, id_bytes); |
1500 |
if (file_coverage == NULL) { |
1501 |
/* not there: create a new one */ |
1502 |
char * id = xstrdup(id_bytes); |
1503 |
file_coverage = xmalloc(sizeof(FileCoverage)); |
1504 |
file_coverage->id = id; |
1505 |
file_coverage->num_coverage_lines = array->pn_count; |
1506 |
file_coverage->coverage_lines = xnew(int, array->pn_count); |
1507 |
file_coverage->source_lines = NULL; |
1508 |
|
1509 |
/* set coverage for all lines */ |
1510 |
uint32 i = 0; |
1511 |
for (JSParseNode * element = array->pn_head; element != NULL; element = element->pn_next, i++) { |
1512 |
if (element->pn_type == TOK_NUMBER) { |
1513 |
file_coverage->coverage_lines[i] = (int) element->pn_dval; |
1514 |
} |
1515 |
else if (element->pn_type == TOK_PRIMARY && element->pn_op == JSOP_NULL) { |
1516 |
file_coverage->coverage_lines[i] = -1; |
1517 |
} |
1518 |
else { |
1519 |
result = -1; |
1520 |
goto done; |
1521 |
} |
1522 |
} |
1523 |
assert(i == array->pn_count); |
1524 |
|
1525 |
/* add to the hash table */ |
1526 |
JS_HashTableAdd(coverage->coverage_table, id, file_coverage); |
1527 |
struct FileCoverageList * coverage_list = xmalloc(sizeof(struct FileCoverageList)); |
1528 |
coverage_list->file_coverage = file_coverage; |
1529 |
coverage_list->next = coverage->coverage_list; |
1530 |
coverage->coverage_list = coverage_list; |
1531 |
} |
1532 |
else { |
1533 |
/* sanity check */ |
1534 |
assert(strcmp(file_coverage->id, id_bytes) == 0); |
1535 |
if (file_coverage->num_coverage_lines != array->pn_count) { |
1536 |
result = -2; |
1537 |
goto done; |
1538 |
} |
1539 |
|
1540 |
/* merge the coverage */ |
1541 |
uint32 i = 0; |
1542 |
for (JSParseNode * element = array->pn_head; element != NULL; element = element->pn_next, i++) { |
1543 |
if (element->pn_type == TOK_NUMBER) { |
1544 |
if (file_coverage->coverage_lines[i] == -1) { |
1545 |
result = -2; |
1546 |
goto done; |
1547 |
} |
1548 |
file_coverage->coverage_lines[i] += (int) element->pn_dval; |
1549 |
} |
1550 |
else if (element->pn_type == TOK_PRIMARY && element->pn_op == JSOP_NULL) { |
1551 |
if (file_coverage->coverage_lines[i] != -1) { |
1552 |
result = -2; |
1553 |
goto done; |
1554 |
} |
1555 |
} |
1556 |
else { |
1557 |
result = -1; |
1558 |
goto done; |
1559 |
} |
1560 |
} |
1561 |
assert(i == array->pn_count); |
1562 |
} |
1563 |
|
1564 |
/* if this JSON file has source, use it */ |
1565 |
if (file_coverage->source_lines == NULL && source != NULL) { |
1566 |
file_coverage->num_source_lines = source->pn_count; |
1567 |
file_coverage->source_lines = xnew(char *, source->pn_count); |
1568 |
uint32 i = 0; |
1569 |
for (JSParseNode * element = source->pn_head; element != NULL; element = element->pn_next, i++) { |
1570 |
if (element->pn_type != TOK_STRING) { |
1571 |
result = -1; |
1572 |
goto done; |
1573 |
} |
1574 |
file_coverage->source_lines[i] = xstrdup(JS_GetStringBytes(ATOM_TO_STRING(element->pn_atom))); |
1575 |
} |
1576 |
assert(i == source->pn_count); |
1577 |
} |
1578 |
} |
1579 |
|
1580 |
done: |
1581 |
js_FinishParseContext(context, &parse_context); |
1582 |
return result; |
1583 |
} |