1#include "cache.h"
2#include "config.h"
3#include "json-writer.h"
4#include "run-command.h"
5#include "version.h"
6#include "trace2/tr2_dst.h"
7#include "trace2/tr2_tbuf.h"
8#include "trace2/tr2_sid.h"
9#include "trace2/tr2_sysenv.h"
10#include "trace2/tr2_tgt.h"
11#include "trace2/tr2_tls.h"
12
13static struct tr2_dst tr2dst_event = { TR2_SYSENV_EVENT, 0, 0, 0 };
14
15/*
16 * The version number of the JSON data generated by the EVENT target
17 * in this source file. Update this if you make a significant change
18 * to the JSON fields or message structure. You probably do not need
19 * to update this if you just add another call to one of the existing
20 * TRACE2 API methods.
21 */
22#define TR2_EVENT_VERSION "1"
23
24/*
25 * Region nesting limit for messages written to the event target.
26 *
27 * The "region_enter" and "region_leave" messages (especially recursive
28 * messages such as those produced while diving the worktree or index)
29 * are primarily intended for the performance target during debugging.
30 *
31 * Some of the outer-most messages, however, may be of interest to the
32 * event target. Use the TR2_SYSENV_EVENT_NESTING setting to increase
33 * region details in the event target.
34 */
35static int tr2env_event_max_nesting_levels = 2;
36
37/*
38 * Use the TR2_SYSENV_EVENT_BRIEF to omit the <time>, <file>, and
39 * <line> fields from most events.
40 */
41static int tr2env_event_be_brief;
42
43static int fn_init(void)
44{
45 int want = tr2_dst_trace_want(&tr2dst_event);
46 int max_nesting;
47 int want_brief;
48 const char *nesting;
49 const char *brief;
50
51 if (!want)
52 return want;
53
54 nesting = tr2_sysenv_get(TR2_SYSENV_EVENT_NESTING);
55 if (nesting && *nesting && ((max_nesting = atoi(nesting)) > 0))
56 tr2env_event_max_nesting_levels = max_nesting;
57
58 brief = tr2_sysenv_get(TR2_SYSENV_EVENT_BRIEF);
59 if (brief && *brief &&
60 ((want_brief = git_parse_maybe_bool(brief)) != -1))
61 tr2env_event_be_brief = want_brief;
62
63 return want;
64}
65
66static void fn_term(void)
67{
68 tr2_dst_trace_disable(&tr2dst_event);
69}
70
71/*
72 * Append common key-value pairs to the currently open JSON object.
73 * "event:"<event_name>"
74 * "sid":"<sid>"
75 * "thread":"<thread_name>"
76 * "time":"<time>"
77 * "file":"<filename>"
78 * "line":<line_number>
79 * "repo":<repo_id>
80 */
81static void event_fmt_prepare(const char *event_name, const char *file,
82 int line, const struct repository *repo,
83 struct json_writer *jw)
84{
85 struct tr2tls_thread_ctx *ctx = tr2tls_get_self();
86 struct tr2_tbuf tb_now;
87
88 jw_object_string(jw, "event", event_name);
89 jw_object_string(jw, "sid", tr2_sid_get());
90 jw_object_string(jw, "thread", ctx->thread_name.buf);
91
92 /*
93 * In brief mode, only emit <time> on these 2 event types.
94 */
95 if (!tr2env_event_be_brief || !strcmp(event_name, "version") ||
96 !strcmp(event_name, "atexit")) {
97 tr2_tbuf_utc_datetime_extended(&tb_now);
98 jw_object_string(jw, "time", tb_now.buf);
99 }
100
101 if (!tr2env_event_be_brief && file && *file) {
102 jw_object_string(jw, "file", file);
103 jw_object_intmax(jw, "line", line);
104 }
105
106 if (repo)
107 jw_object_intmax(jw, "repo", repo->trace2_repo_id);
108}
109
110static void fn_version_fl(const char *file, int line)
111{
112 const char *event_name = "version";
113 struct json_writer jw = JSON_WRITER_INIT;
114
115 jw_object_begin(&jw, 0);
116 event_fmt_prepare(event_name, file, line, NULL, &jw);
117 jw_object_string(&jw, "evt", TR2_EVENT_VERSION);
118 jw_object_string(&jw, "exe", git_version_string);
119 jw_end(&jw);
120
121 tr2_dst_write_line(&tr2dst_event, &jw.json);
122 jw_release(&jw);
123}
124
125static void fn_start_fl(const char *file, int line,
126 uint64_t us_elapsed_absolute, const char **argv)
127{
128 const char *event_name = "start";
129 struct json_writer jw = JSON_WRITER_INIT;
130 double t_abs = (double)us_elapsed_absolute / 1000000.0;
131
132 jw_object_begin(&jw, 0);
133 event_fmt_prepare(event_name, file, line, NULL, &jw);
134 jw_object_double(&jw, "t_abs", 6, t_abs);
135 jw_object_inline_begin_array(&jw, "argv");
136 jw_array_argv(&jw, argv);
137 jw_end(&jw);
138 jw_end(&jw);
139
140 tr2_dst_write_line(&tr2dst_event, &jw.json);
141 jw_release(&jw);
142}
143
144static void fn_exit_fl(const char *file, int line, uint64_t us_elapsed_absolute,
145 int code)
146{
147 const char *event_name = "exit";
148 struct json_writer jw = JSON_WRITER_INIT;
149 double t_abs = (double)us_elapsed_absolute / 1000000.0;
150
151 jw_object_begin(&jw, 0);
152 event_fmt_prepare(event_name, file, line, NULL, &jw);
153 jw_object_double(&jw, "t_abs", 6, t_abs);
154 jw_object_intmax(&jw, "code", code);
155 jw_end(&jw);
156
157 tr2_dst_write_line(&tr2dst_event, &jw.json);
158 jw_release(&jw);
159}
160
161static void fn_signal(uint64_t us_elapsed_absolute, int signo)
162{
163 const char *event_name = "signal";
164 struct json_writer jw = JSON_WRITER_INIT;
165 double t_abs = (double)us_elapsed_absolute / 1000000.0;
166
167 jw_object_begin(&jw, 0);
168 event_fmt_prepare(event_name, __FILE__, __LINE__, NULL, &jw);
169 jw_object_double(&jw, "t_abs", 6, t_abs);
170 jw_object_intmax(&jw, "signo", signo);
171 jw_end(&jw);
172
173 tr2_dst_write_line(&tr2dst_event, &jw.json);
174 jw_release(&jw);
175}
176
177static void fn_atexit(uint64_t us_elapsed_absolute, int code)
178{
179 const char *event_name = "atexit";
180 struct json_writer jw = JSON_WRITER_INIT;
181 double t_abs = (double)us_elapsed_absolute / 1000000.0;
182
183 jw_object_begin(&jw, 0);
184 event_fmt_prepare(event_name, __FILE__, __LINE__, NULL, &jw);
185 jw_object_double(&jw, "t_abs", 6, t_abs);
186 jw_object_intmax(&jw, "code", code);
187 jw_end(&jw);
188
189 tr2_dst_write_line(&tr2dst_event, &jw.json);
190 jw_release(&jw);
191}
192
193static void maybe_add_string_va(struct json_writer *jw, const char *field_name,
194 const char *fmt, va_list ap)
195{
196 if (fmt && *fmt) {
197 va_list copy_ap;
198 struct strbuf buf = STRBUF_INIT;
199
200 va_copy(copy_ap, ap);
201 strbuf_vaddf(&buf, fmt, copy_ap);
202 va_end(copy_ap);
203
204 jw_object_string(jw, field_name, buf.buf);
205 strbuf_release(&buf);
206 return;
207 }
208}
209
210static void fn_error_va_fl(const char *file, int line, const char *fmt,
211 va_list ap)
212{
213 const char *event_name = "error";
214 struct json_writer jw = JSON_WRITER_INIT;
215
216 jw_object_begin(&jw, 0);
217 event_fmt_prepare(event_name, file, line, NULL, &jw);
218 maybe_add_string_va(&jw, "msg", fmt, ap);
219 /*
220 * Also emit the format string as a field in case
221 * post-processors want to aggregate common error
222 * messages by type without argument fields (such
223 * as pathnames or branch names) cluttering it up.
224 */
225 if (fmt && *fmt)
226 jw_object_string(&jw, "fmt", fmt);
227 jw_end(&jw);
228
229 tr2_dst_write_line(&tr2dst_event, &jw.json);
230 jw_release(&jw);
231}
232
233static void fn_command_path_fl(const char *file, int line, const char *pathname)
234{
235 const char *event_name = "cmd_path";
236 struct json_writer jw = JSON_WRITER_INIT;
237
238 jw_object_begin(&jw, 0);
239 event_fmt_prepare(event_name, file, line, NULL, &jw);
240 jw_object_string(&jw, "path", pathname);
241 jw_end(&jw);
242
243 tr2_dst_write_line(&tr2dst_event, &jw.json);
244 jw_release(&jw);
245}
246
247static void fn_command_name_fl(const char *file, int line, const char *name,
248 const char *hierarchy)
249{
250 const char *event_name = "cmd_name";
251 struct json_writer jw = JSON_WRITER_INIT;
252
253 jw_object_begin(&jw, 0);
254 event_fmt_prepare(event_name, file, line, NULL, &jw);
255 jw_object_string(&jw, "name", name);
256 if (hierarchy && *hierarchy)
257 jw_object_string(&jw, "hierarchy", hierarchy);
258 jw_end(&jw);
259
260 tr2_dst_write_line(&tr2dst_event, &jw.json);
261 jw_release(&jw);
262}
263
264static void fn_command_mode_fl(const char *file, int line, const char *mode)
265{
266 const char *event_name = "cmd_mode";
267 struct json_writer jw = JSON_WRITER_INIT;
268
269 jw_object_begin(&jw, 0);
270 event_fmt_prepare(event_name, file, line, NULL, &jw);
271 jw_object_string(&jw, "name", mode);
272 jw_end(&jw);
273
274 tr2_dst_write_line(&tr2dst_event, &jw.json);
275 jw_release(&jw);
276}
277
278static void fn_alias_fl(const char *file, int line, const char *alias,
279 const char **argv)
280{
281 const char *event_name = "alias";
282 struct json_writer jw = JSON_WRITER_INIT;
283
284 jw_object_begin(&jw, 0);
285 event_fmt_prepare(event_name, file, line, NULL, &jw);
286 jw_object_string(&jw, "alias", alias);
287 jw_object_inline_begin_array(&jw, "argv");
288 jw_array_argv(&jw, argv);
289 jw_end(&jw);
290 jw_end(&jw);
291
292 tr2_dst_write_line(&tr2dst_event, &jw.json);
293 jw_release(&jw);
294}
295
296static void fn_child_start_fl(const char *file, int line,
297 uint64_t us_elapsed_absolute,
298 const struct child_process *cmd)
299{
300 const char *event_name = "child_start";
301 struct json_writer jw = JSON_WRITER_INIT;
302
303 jw_object_begin(&jw, 0);
304 event_fmt_prepare(event_name, file, line, NULL, &jw);
305 jw_object_intmax(&jw, "child_id", cmd->trace2_child_id);
306 if (cmd->trace2_hook_name) {
307 jw_object_string(&jw, "child_class", "hook");
308 jw_object_string(&jw, "hook_name", cmd->trace2_hook_name);
309 } else {
310 const char *child_class =
311 cmd->trace2_child_class ? cmd->trace2_child_class : "?";
312 jw_object_string(&jw, "child_class", child_class);
313 }
314 if (cmd->dir)
315 jw_object_string(&jw, "cd", cmd->dir);
316 jw_object_bool(&jw, "use_shell", cmd->use_shell);
317 jw_object_inline_begin_array(&jw, "argv");
318 if (cmd->git_cmd)
319 jw_array_string(&jw, "git");
320 jw_array_argv(&jw, cmd->argv);
321 jw_end(&jw);
322 jw_end(&jw);
323
324 tr2_dst_write_line(&tr2dst_event, &jw.json);
325 jw_release(&jw);
326}
327
328static void fn_child_exit_fl(const char *file, int line,
329 uint64_t us_elapsed_absolute, int cid, int pid,
330 int code, uint64_t us_elapsed_child)
331{
332 const char *event_name = "child_exit";
333 struct json_writer jw = JSON_WRITER_INIT;
334 double t_rel = (double)us_elapsed_child / 1000000.0;
335
336 jw_object_begin(&jw, 0);
337 event_fmt_prepare(event_name, file, line, NULL, &jw);
338 jw_object_intmax(&jw, "child_id", cid);
339 jw_object_intmax(&jw, "pid", pid);
340 jw_object_intmax(&jw, "code", code);
341 jw_object_double(&jw, "t_rel", 6, t_rel);
342 jw_end(&jw);
343
344 tr2_dst_write_line(&tr2dst_event, &jw.json);
345
346 jw_release(&jw);
347}
348
349static void fn_thread_start_fl(const char *file, int line,
350 uint64_t us_elapsed_absolute)
351{
352 const char *event_name = "thread_start";
353 struct json_writer jw = JSON_WRITER_INIT;
354
355 jw_object_begin(&jw, 0);
356 event_fmt_prepare(event_name, file, line, NULL, &jw);
357 jw_end(&jw);
358
359 tr2_dst_write_line(&tr2dst_event, &jw.json);
360 jw_release(&jw);
361}
362
363static void fn_thread_exit_fl(const char *file, int line,
364 uint64_t us_elapsed_absolute,
365 uint64_t us_elapsed_thread)
366{
367 const char *event_name = "thread_exit";
368 struct json_writer jw = JSON_WRITER_INIT;
369 double t_rel = (double)us_elapsed_thread / 1000000.0;
370
371 jw_object_begin(&jw, 0);
372 event_fmt_prepare(event_name, file, line, NULL, &jw);
373 jw_object_double(&jw, "t_rel", 6, t_rel);
374 jw_end(&jw);
375
376 tr2_dst_write_line(&tr2dst_event, &jw.json);
377 jw_release(&jw);
378}
379
380static void fn_exec_fl(const char *file, int line, uint64_t us_elapsed_absolute,
381 int exec_id, const char *exe, const char **argv)
382{
383 const char *event_name = "exec";
384 struct json_writer jw = JSON_WRITER_INIT;
385
386 jw_object_begin(&jw, 0);
387 event_fmt_prepare(event_name, file, line, NULL, &jw);
388 jw_object_intmax(&jw, "exec_id", exec_id);
389 if (exe)
390 jw_object_string(&jw, "exe", exe);
391 jw_object_inline_begin_array(&jw, "argv");
392 jw_array_argv(&jw, argv);
393 jw_end(&jw);
394 jw_end(&jw);
395
396 tr2_dst_write_line(&tr2dst_event, &jw.json);
397 jw_release(&jw);
398}
399
400static void fn_exec_result_fl(const char *file, int line,
401 uint64_t us_elapsed_absolute, int exec_id,
402 int code)
403{
404 const char *event_name = "exec_result";
405 struct json_writer jw = JSON_WRITER_INIT;
406
407 jw_object_begin(&jw, 0);
408 event_fmt_prepare(event_name, file, line, NULL, &jw);
409 jw_object_intmax(&jw, "exec_id", exec_id);
410 jw_object_intmax(&jw, "code", code);
411 jw_end(&jw);
412
413 tr2_dst_write_line(&tr2dst_event, &jw.json);
414 jw_release(&jw);
415}
416
417static void fn_param_fl(const char *file, int line, const char *param,
418 const char *value)
419{
420 const char *event_name = "def_param";
421 struct json_writer jw = JSON_WRITER_INIT;
422
423 jw_object_begin(&jw, 0);
424 event_fmt_prepare(event_name, file, line, NULL, &jw);
425 jw_object_string(&jw, "param", param);
426 jw_object_string(&jw, "value", value);
427 jw_end(&jw);
428
429 tr2_dst_write_line(&tr2dst_event, &jw.json);
430 jw_release(&jw);
431}
432
433static void fn_repo_fl(const char *file, int line,
434 const struct repository *repo)
435{
436 const char *event_name = "def_repo";
437 struct json_writer jw = JSON_WRITER_INIT;
438
439 jw_object_begin(&jw, 0);
440 event_fmt_prepare(event_name, file, line, repo, &jw);
441 jw_object_string(&jw, "worktree", repo->worktree);
442 jw_end(&jw);
443
444 tr2_dst_write_line(&tr2dst_event, &jw.json);
445 jw_release(&jw);
446}
447
448static void fn_region_enter_printf_va_fl(const char *file, int line,
449 uint64_t us_elapsed_absolute,
450 const char *category,
451 const char *label,
452 const struct repository *repo,
453 const char *fmt, va_list ap)
454{
455 const char *event_name = "region_enter";
456 struct tr2tls_thread_ctx *ctx = tr2tls_get_self();
457 if (ctx->nr_open_regions <= tr2env_event_max_nesting_levels) {
458 struct json_writer jw = JSON_WRITER_INIT;
459
460 jw_object_begin(&jw, 0);
461 event_fmt_prepare(event_name, file, line, repo, &jw);
462 jw_object_intmax(&jw, "nesting", ctx->nr_open_regions);
463 if (category)
464 jw_object_string(&jw, "category", category);
465 if (label)
466 jw_object_string(&jw, "label", label);
467 maybe_add_string_va(&jw, "msg", fmt, ap);
468 jw_end(&jw);
469
470 tr2_dst_write_line(&tr2dst_event, &jw.json);
471 jw_release(&jw);
472 }
473}
474
475static void fn_region_leave_printf_va_fl(
476 const char *file, int line, uint64_t us_elapsed_absolute,
477 uint64_t us_elapsed_region, const char *category, const char *label,
478 const struct repository *repo, const char *fmt, va_list ap)
479{
480 const char *event_name = "region_leave";
481 struct tr2tls_thread_ctx *ctx = tr2tls_get_self();
482 if (ctx->nr_open_regions <= tr2env_event_max_nesting_levels) {
483 struct json_writer jw = JSON_WRITER_INIT;
484 double t_rel = (double)us_elapsed_region / 1000000.0;
485
486 jw_object_begin(&jw, 0);
487 event_fmt_prepare(event_name, file, line, repo, &jw);
488 jw_object_double(&jw, "t_rel", 6, t_rel);
489 jw_object_intmax(&jw, "nesting", ctx->nr_open_regions);
490 if (category)
491 jw_object_string(&jw, "category", category);
492 if (label)
493 jw_object_string(&jw, "label", label);
494 maybe_add_string_va(&jw, "msg", fmt, ap);
495 jw_end(&jw);
496
497 tr2_dst_write_line(&tr2dst_event, &jw.json);
498 jw_release(&jw);
499 }
500}
501
502static void fn_data_fl(const char *file, int line, uint64_t us_elapsed_absolute,
503 uint64_t us_elapsed_region, const char *category,
504 const struct repository *repo, const char *key,
505 const char *value)
506{
507 const char *event_name = "data";
508 struct tr2tls_thread_ctx *ctx = tr2tls_get_self();
509 if (ctx->nr_open_regions <= tr2env_event_max_nesting_levels) {
510 struct json_writer jw = JSON_WRITER_INIT;
511 double t_abs = (double)us_elapsed_absolute / 1000000.0;
512 double t_rel = (double)us_elapsed_region / 1000000.0;
513
514 jw_object_begin(&jw, 0);
515 event_fmt_prepare(event_name, file, line, repo, &jw);
516 jw_object_double(&jw, "t_abs", 6, t_abs);
517 jw_object_double(&jw, "t_rel", 6, t_rel);
518 jw_object_intmax(&jw, "nesting", ctx->nr_open_regions);
519 jw_object_string(&jw, "category", category);
520 jw_object_string(&jw, "key", key);
521 jw_object_string(&jw, "value", value);
522 jw_end(&jw);
523
524 tr2_dst_write_line(&tr2dst_event, &jw.json);
525 jw_release(&jw);
526 }
527}
528
529static void fn_data_json_fl(const char *file, int line,
530 uint64_t us_elapsed_absolute,
531 uint64_t us_elapsed_region, const char *category,
532 const struct repository *repo, const char *key,
533 const struct json_writer *value)
534{
535 const char *event_name = "data_json";
536 struct tr2tls_thread_ctx *ctx = tr2tls_get_self();
537 if (ctx->nr_open_regions <= tr2env_event_max_nesting_levels) {
538 struct json_writer jw = JSON_WRITER_INIT;
539 double t_abs = (double)us_elapsed_absolute / 1000000.0;
540 double t_rel = (double)us_elapsed_region / 1000000.0;
541
542 jw_object_begin(&jw, 0);
543 event_fmt_prepare(event_name, file, line, repo, &jw);
544 jw_object_double(&jw, "t_abs", 6, t_abs);
545 jw_object_double(&jw, "t_rel", 6, t_rel);
546 jw_object_intmax(&jw, "nesting", ctx->nr_open_regions);
547 jw_object_string(&jw, "category", category);
548 jw_object_string(&jw, "key", key);
549 jw_object_sub_jw(&jw, "value", value);
550 jw_end(&jw);
551
552 tr2_dst_write_line(&tr2dst_event, &jw.json);
553 jw_release(&jw);
554 }
555}
556
557struct tr2_tgt tr2_tgt_event = {
558 &tr2dst_event,
559
560 fn_init,
561 fn_term,
562
563 fn_version_fl,
564 fn_start_fl,
565 fn_exit_fl,
566 fn_signal,
567 fn_atexit,
568 fn_error_va_fl,
569 fn_command_path_fl,
570 fn_command_name_fl,
571 fn_command_mode_fl,
572 fn_alias_fl,
573 fn_child_start_fl,
574 fn_child_exit_fl,
575 fn_thread_start_fl,
576 fn_thread_exit_fl,
577 fn_exec_fl,
578 fn_exec_result_fl,
579 fn_param_fl,
580 fn_repo_fl,
581 fn_region_enter_printf_va_fl,
582 fn_region_leave_printf_va_fl,
583 fn_data_fl,
584 fn_data_json_fl,
585 NULL, /* printf */
586};