Commit | Line | Data |
---|---|---|
7a278c8e | 1 | /* |
2e339de1 | 2 | * Copyright 2016 Jérémie Galarneau <jeremie.galarneau@efficios.com> |
f504043c | 3 | * Copyright 2016 Mathieu Desnoyers <mathieu.desnoyers@efficios.com> |
7a278c8e JG |
4 | * |
5 | * Author: Jérémie Galarneau <jeremie.galarneau@efficios.com> | |
6 | * | |
7 | * Permission is hereby granted, free of charge, to any person obtaining a copy | |
8 | * of this software and associated documentation files (the "Software"), to deal | |
9 | * in the Software without restriction, including without limitation the rights | |
10 | * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell | |
11 | * copies of the Software, and to permit persons to whom the Software is | |
12 | * furnished to do so, subject to the following conditions: | |
13 | * | |
14 | * The above copyright notice and this permission notice shall be included in | |
15 | * all copies or substantial portions of the Software. | |
16 | * | |
17 | * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR | |
18 | * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, | |
19 | * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE | |
20 | * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER | |
21 | * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, | |
22 | * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE | |
23 | * SOFTWARE. | |
24 | */ | |
25 | ||
71c5da58 | 26 | #include <babeltrace2/babeltrace.h> |
57952005 MJ |
27 | #include "compat/compiler.h" |
28 | #include "common/common.h" | |
bfd20a42 | 29 | #include <stdio.h> |
39cfa40f | 30 | #include <stdbool.h> |
bac67f0f | 31 | #include <glib.h> |
8d81b92f | 32 | #include <string.h> |
57952005 | 33 | #include "common/assert.h" |
6405967d | 34 | |
3228cc1d PP |
35 | #include "pretty.h" |
36 | ||
b7c70610 MJ |
37 | GQuark stream_packet_context_quarks[STREAM_PACKET_CONTEXT_QUARKS_LEN]; |
38 | ||
6e1bc0df MD |
39 | static |
40 | const char *plugin_options[] = { | |
ad96d936 | 41 | "color", |
a0ad501b | 42 | "path", |
6e1bc0df MD |
43 | "no-delta", |
44 | "clock-cycles", | |
45 | "clock-seconds", | |
46 | "clock-date", | |
47 | "clock-gmt", | |
a263021c | 48 | "verbose", |
6e1bc0df MD |
49 | "name-default", /* show/hide */ |
50 | "name-payload", | |
51 | "name-context", | |
52 | "name-scope", | |
53 | "name-header", | |
54 | "field-default", /* show/hide */ | |
55 | "field-trace", | |
56 | "field-trace:hostname", | |
57 | "field-trace:domain", | |
58 | "field-trace:procname", | |
59 | "field-trace:vpid", | |
60 | "field-loglevel", | |
61 | "field-emf", | |
60535549 | 62 | "field-callsite", |
6e1bc0df MD |
63 | }; |
64 | ||
1043fdea PP |
65 | static |
66 | const char * const in_port_name = "in"; | |
67 | ||
bfd20a42 | 68 | static |
3228cc1d | 69 | void destroy_pretty_data(struct pretty_component *pretty) |
bac67f0f | 70 | { |
b09a5592 | 71 | bt_self_component_port_input_message_iterator_put_ref(pretty->iterator); |
5280f742 PP |
72 | |
73 | if (pretty->string) { | |
74 | (void) g_string_free(pretty->string, TRUE); | |
75 | } | |
76 | ||
77 | if (pretty->tmp_string) { | |
78 | (void) g_string_free(pretty->tmp_string, TRUE); | |
79 | } | |
80 | ||
3228cc1d | 81 | if (pretty->out != stdout) { |
77986bad JD |
82 | int ret; |
83 | ||
3228cc1d | 84 | ret = fclose(pretty->out); |
77986bad JD |
85 | if (ret) { |
86 | perror("close output file"); | |
87 | } | |
88 | } | |
3228cc1d | 89 | g_free(pretty->options.output_path); |
3228cc1d | 90 | g_free(pretty); |
bac67f0f JG |
91 | } |
92 | ||
b25bd455 | 93 | static |
3228cc1d | 94 | struct pretty_component *create_pretty(void) |
bac67f0f | 95 | { |
3228cc1d | 96 | struct pretty_component *pretty; |
541b0a11 | 97 | |
3228cc1d PP |
98 | pretty = g_new0(struct pretty_component, 1); |
99 | if (!pretty) { | |
541b0a11 JG |
100 | goto end; |
101 | } | |
3228cc1d PP |
102 | pretty->string = g_string_new(""); |
103 | if (!pretty->string) { | |
6a18b281 MD |
104 | goto error; |
105 | } | |
5280f742 PP |
106 | pretty->tmp_string = g_string_new(""); |
107 | if (!pretty->tmp_string) { | |
108 | goto error; | |
109 | } | |
541b0a11 | 110 | end: |
3228cc1d | 111 | return pretty; |
6a18b281 MD |
112 | |
113 | error: | |
3228cc1d | 114 | g_free(pretty); |
6a18b281 | 115 | return NULL; |
bac67f0f JG |
116 | } |
117 | ||
3228cc1d | 118 | BT_HIDDEN |
8eee8ea2 | 119 | void pretty_finalize(bt_self_component_sink *comp) |
b25bd455 | 120 | { |
834e9996 PP |
121 | destroy_pretty_data( |
122 | bt_self_component_get_data( | |
bb61965b | 123 | bt_self_component_sink_as_self_component(comp))); |
b25bd455 JG |
124 | } |
125 | ||
bac67f0f | 126 | static |
fb25b9e3 | 127 | bt_component_class_message_iterator_next_method_status handle_message( |
834e9996 | 128 | struct pretty_component *pretty, |
b09a5592 | 129 | const bt_message *message) |
4c1456f0 | 130 | { |
fb25b9e3 PP |
131 | bt_component_class_message_iterator_next_method_status ret = |
132 | BT_COMPONENT_CLASS_MESSAGE_ITERATOR_NEXT_METHOD_STATUS_OK; | |
541b0a11 | 133 | |
8b45963b | 134 | BT_ASSERT(pretty); |
541b0a11 | 135 | |
b09a5592 | 136 | switch (bt_message_get_type(message)) { |
b09a5592 PP |
137 | case BT_MESSAGE_TYPE_EVENT: |
138 | if (pretty_print_event(pretty, message)) { | |
fb25b9e3 | 139 | ret = BT_COMPONENT_CLASS_MESSAGE_ITERATOR_NEXT_METHOD_STATUS_ERROR; |
834e9996 | 140 | } |
7cdc2bab | 141 | break; |
22dfa677 PP |
142 | case BT_MESSAGE_TYPE_DISCARDED_EVENTS: |
143 | case BT_MESSAGE_TYPE_DISCARDED_PACKETS: | |
144 | if (pretty_print_discarded_items(pretty, message)) { | |
fb25b9e3 | 145 | ret = BT_COMPONENT_CLASS_MESSAGE_ITERATOR_NEXT_METHOD_STATUS_ERROR; |
22dfa677 PP |
146 | } |
147 | break; | |
7cdc2bab | 148 | default: |
6ff151ad | 149 | break; |
78586d8a | 150 | } |
b5e978f4 | 151 | |
541b0a11 | 152 | return ret; |
4c1456f0 | 153 | } |
bac67f0f | 154 | |
3228cc1d | 155 | BT_HIDDEN |
fb25b9e3 PP |
156 | bt_component_class_sink_graph_is_configured_method_status |
157 | pretty_graph_is_configured(bt_self_component_sink *comp) | |
75e1829b | 158 | { |
fb25b9e3 PP |
159 | bt_component_class_sink_graph_is_configured_method_status status = |
160 | BT_COMPONENT_CLASS_SINK_GRAPH_IS_CONFIGURED_METHOD_STATUS_OK; | |
3228cc1d | 161 | struct pretty_component *pretty; |
75e1829b | 162 | |
834e9996 | 163 | pretty = bt_self_component_get_data( |
bb61965b | 164 | bt_self_component_sink_as_self_component(comp)); |
8b45963b | 165 | BT_ASSERT(pretty); |
834e9996 | 166 | BT_ASSERT(!pretty->iterator); |
b09a5592 | 167 | pretty->iterator = bt_self_component_port_input_message_iterator_create( |
1043fdea PP |
168 | bt_self_component_sink_borrow_input_port_by_name(comp, |
169 | in_port_name)); | |
834e9996 | 170 | if (!pretty->iterator) { |
fb25b9e3 | 171 | status = BT_COMPONENT_CLASS_MESSAGE_ITERATOR_NEXT_METHOD_STATUS_MEMORY_ERROR; |
75e1829b | 172 | } |
72b913fb | 173 | |
634f394c | 174 | return status; |
75e1829b JG |
175 | } |
176 | ||
3228cc1d | 177 | BT_HIDDEN |
fb25b9e3 | 178 | bt_component_class_sink_consume_method_status pretty_consume( |
8eee8ea2 | 179 | bt_self_component_sink *comp) |
fec2a9f2 | 180 | { |
fb25b9e3 | 181 | bt_component_class_sink_consume_method_status ret; |
b09a5592 PP |
182 | bt_message_array_const msgs; |
183 | bt_self_component_port_input_message_iterator *it; | |
834e9996 | 184 | struct pretty_component *pretty = bt_self_component_get_data( |
bb61965b | 185 | bt_self_component_sink_as_self_component(comp)); |
fb25b9e3 | 186 | bt_message_iterator_next_status next_status; |
3fd7b79d PP |
187 | uint64_t count = 0; |
188 | uint64_t i = 0; | |
fec2a9f2 | 189 | |
834e9996 | 190 | it = pretty->iterator; |
fb25b9e3 | 191 | next_status = bt_self_component_port_input_message_iterator_next(it, |
b09a5592 | 192 | &msgs, &count); |
0d8b4d8e | 193 | |
fb25b9e3 PP |
194 | switch (next_status) { |
195 | case BT_MESSAGE_ITERATOR_NEXT_STATUS_OK: | |
834e9996 | 196 | break; |
fb25b9e3 PP |
197 | case BT_MESSAGE_ITERATOR_NEXT_STATUS_MEMORY_ERROR: |
198 | case BT_MESSAGE_ITERATOR_NEXT_STATUS_AGAIN: | |
199 | ret = (int) next_status; | |
834e9996 | 200 | goto end; |
fb25b9e3 PP |
201 | case BT_MESSAGE_ITERATOR_NEXT_STATUS_END: |
202 | ret = (int) next_status; | |
b09a5592 | 203 | BT_SELF_COMPONENT_PORT_INPUT_MESSAGE_ITERATOR_PUT_REF_AND_RESET( |
8c6884d9 | 204 | pretty->iterator); |
dfa6653c | 205 | goto end; |
dfa6653c | 206 | default: |
fb25b9e3 | 207 | ret = BT_COMPONENT_CLASS_SINK_CONSUME_METHOD_STATUS_ERROR; |
dfa6653c | 208 | goto end; |
fec2a9f2 JG |
209 | } |
210 | ||
fb25b9e3 | 211 | BT_ASSERT(next_status == BT_MESSAGE_ITERATOR_NEXT_STATUS_OK); |
3fd7b79d PP |
212 | |
213 | for (i = 0; i < count; i++) { | |
b09a5592 | 214 | ret = handle_message(pretty, msgs[i]); |
3fd7b79d PP |
215 | if (ret) { |
216 | goto end; | |
217 | } | |
218 | ||
b09a5592 | 219 | bt_message_put_ref(msgs[i]); |
3fd7b79d | 220 | } |
dfa6653c | 221 | |
fec2a9f2 | 222 | end: |
3fd7b79d | 223 | for (; i < count; i++) { |
b09a5592 | 224 | bt_message_put_ref(msgs[i]); |
3fd7b79d PP |
225 | } |
226 | ||
fec2a9f2 JG |
227 | return ret; |
228 | } | |
229 | ||
6e1bc0df | 230 | static |
8eee8ea2 | 231 | int add_params_to_map(bt_value *plugin_opt_map) |
6e1bc0df | 232 | { |
834e9996 | 233 | int ret = 0; |
6e1bc0df MD |
234 | unsigned int i; |
235 | ||
236 | for (i = 0; i < BT_ARRAY_SIZE(plugin_options); i++) { | |
237 | const char *key = plugin_options[i]; | |
fb25b9e3 PP |
238 | |
239 | if (bt_value_map_insert_entry(plugin_opt_map, key, | |
240 | bt_value_null) < 0) { | |
834e9996 | 241 | ret = -1; |
6e1bc0df MD |
242 | goto end; |
243 | } | |
244 | } | |
fb25b9e3 | 245 | |
6e1bc0df MD |
246 | end: |
247 | return ret; | |
248 | } | |
249 | ||
250 | static | |
8eee8ea2 | 251 | bt_bool check_param_exists(const char *key, const bt_value *object, |
ce141536 | 252 | void *data) |
6e1bc0df | 253 | { |
3228cc1d | 254 | struct pretty_component *pretty = data; |
6e1bc0df | 255 | |
ce141536 PP |
256 | if (!bt_value_map_has_entry(pretty->plugin_opt_map, |
257 | key)) { | |
3228cc1d PP |
258 | fprintf(pretty->err, |
259 | "[warning] Parameter \"%s\" unknown to \"text.pretty\" sink component\n", key); | |
6e1bc0df | 260 | } |
c55a9f58 | 261 | return BT_TRUE; |
6e1bc0df MD |
262 | } |
263 | ||
264 | static | |
8eee8ea2 | 265 | void apply_one_string(const char *key, const bt_value *params, char **option) |
6e1bc0df | 266 | { |
8eee8ea2 | 267 | const bt_value *value = NULL; |
6e1bc0df MD |
268 | const char *str; |
269 | ||
ce141536 | 270 | value = bt_value_map_borrow_entry_value_const(params, key); |
6e1bc0df MD |
271 | if (!value) { |
272 | goto end; | |
273 | } | |
274 | if (bt_value_is_null(value)) { | |
275 | goto end; | |
276 | } | |
b5cdc106 | 277 | str = bt_value_string_get(value); |
6e1bc0df | 278 | *option = g_strdup(str); |
b5cdc106 | 279 | |
6e1bc0df | 280 | end: |
834e9996 | 281 | return; |
6e1bc0df MD |
282 | } |
283 | ||
284 | static | |
8eee8ea2 | 285 | void apply_one_bool(const char *key, const bt_value *params, bool *option, |
6e1bc0df MD |
286 | bool *found) |
287 | { | |
8eee8ea2 | 288 | const bt_value *value = NULL; |
c55a9f58 | 289 | bt_bool bool_val; |
6e1bc0df | 290 | |
ce141536 | 291 | value = bt_value_map_borrow_entry_value_const(params, key); |
6e1bc0df MD |
292 | if (!value) { |
293 | goto end; | |
294 | } | |
b5cdc106 | 295 | bool_val = bt_value_bool_get(value); |
c55a9f58 | 296 | *option = (bool) bool_val; |
6e1bc0df MD |
297 | if (found) { |
298 | *found = true; | |
299 | } | |
f78d5dd1 | 300 | |
6e1bc0df | 301 | end: |
834e9996 | 302 | return; |
6e1bc0df MD |
303 | } |
304 | ||
ad96d936 | 305 | static |
3228cc1d | 306 | void warn_wrong_color_param(struct pretty_component *pretty) |
ad96d936 | 307 | { |
3228cc1d | 308 | fprintf(pretty->err, |
ad96d936 PP |
309 | "[warning] Accepted values for the \"color\" parameter are:\n \"always\", \"auto\", \"never\"\n"); |
310 | } | |
311 | ||
77986bad | 312 | static |
834e9996 | 313 | int open_output_file(struct pretty_component *pretty) |
77986bad | 314 | { |
834e9996 | 315 | int ret = 0; |
77986bad | 316 | |
3228cc1d | 317 | if (!pretty->options.output_path) { |
77986bad JD |
318 | goto end; |
319 | } | |
320 | ||
3228cc1d PP |
321 | pretty->out = fopen(pretty->options.output_path, "w"); |
322 | if (!pretty->out) { | |
77986bad JD |
323 | goto error; |
324 | } | |
325 | ||
326 | goto end; | |
327 | ||
328 | error: | |
834e9996 PP |
329 | ret = -1; |
330 | ||
77986bad JD |
331 | end: |
332 | return ret; | |
333 | } | |
334 | ||
6e1bc0df | 335 | static |
8eee8ea2 | 336 | int apply_params(struct pretty_component *pretty, const bt_value *params) |
6e1bc0df | 337 | { |
834e9996 | 338 | int ret = 0; |
6e1bc0df MD |
339 | bool value, found; |
340 | char *str = NULL; | |
341 | ||
ce141536 | 342 | pretty->plugin_opt_map = bt_value_map_create(); |
3228cc1d | 343 | if (!pretty->plugin_opt_map) { |
834e9996 | 344 | ret = -1; |
6e1bc0df MD |
345 | goto end; |
346 | } | |
3228cc1d | 347 | ret = add_params_to_map(pretty->plugin_opt_map); |
834e9996 | 348 | if (ret) { |
6e1bc0df MD |
349 | goto end; |
350 | } | |
351 | /* Report unknown parameters. */ | |
fb25b9e3 PP |
352 | if (bt_value_map_foreach_entry_const(params, |
353 | check_param_exists, pretty) < 0) { | |
834e9996 | 354 | ret = -1; |
6e1bc0df MD |
355 | goto end; |
356 | } | |
fb25b9e3 | 357 | |
6e1bc0df | 358 | /* Known parameters. */ |
3228cc1d | 359 | pretty->options.color = PRETTY_COLOR_OPT_AUTO; |
44514773 | 360 | if (bt_value_map_has_entry(params, "color")) { |
8eee8ea2 | 361 | const bt_value *color_value; |
ad96d936 PP |
362 | const char *color; |
363 | ||
ce141536 PP |
364 | color_value = bt_value_map_borrow_entry_value_const(params, |
365 | "color"); | |
ad96d936 PP |
366 | if (!color_value) { |
367 | goto end; | |
368 | } | |
369 | ||
b5cdc106 PP |
370 | color = bt_value_string_get(color_value); |
371 | ||
372 | if (strcmp(color, "never") == 0) { | |
373 | pretty->options.color = PRETTY_COLOR_OPT_NEVER; | |
374 | } else if (strcmp(color, "auto") == 0) { | |
375 | pretty->options.color = PRETTY_COLOR_OPT_AUTO; | |
376 | } else if (strcmp(color, "always") == 0) { | |
377 | pretty->options.color = PRETTY_COLOR_OPT_ALWAYS; | |
ad96d936 | 378 | } else { |
b5cdc106 | 379 | warn_wrong_color_param(pretty); |
ad96d936 | 380 | } |
ad96d936 PP |
381 | } |
382 | ||
834e9996 | 383 | apply_one_string("path", params, &pretty->options.output_path); |
3228cc1d | 384 | ret = open_output_file(pretty); |
834e9996 | 385 | if (ret) { |
77986bad JD |
386 | goto end; |
387 | } | |
6e1bc0df | 388 | |
6e1bc0df | 389 | value = false; /* Default. */ |
834e9996 | 390 | apply_one_bool("no-delta", params, &value, NULL); |
3228cc1d | 391 | pretty->options.print_delta_field = !value; /* Reverse logic. */ |
6e1bc0df MD |
392 | |
393 | value = false; /* Default. */ | |
834e9996 | 394 | apply_one_bool("clock-cycles", params, &value, NULL); |
3228cc1d | 395 | pretty->options.print_timestamp_cycles = value; |
6e1bc0df MD |
396 | |
397 | value = false; /* Default. */ | |
834e9996 | 398 | apply_one_bool("clock-seconds", params, &value, NULL); |
3228cc1d | 399 | pretty->options.clock_seconds = value; |
6e1bc0df MD |
400 | |
401 | value = false; /* Default. */ | |
834e9996 | 402 | apply_one_bool("clock-date", params, &value, NULL); |
3228cc1d | 403 | pretty->options.clock_date = value; |
6e1bc0df MD |
404 | |
405 | value = false; /* Default. */ | |
834e9996 | 406 | apply_one_bool("clock-gmt", params, &value, NULL); |
3228cc1d | 407 | pretty->options.clock_gmt = value; |
6e1bc0df | 408 | |
a263021c | 409 | value = false; /* Default. */ |
834e9996 | 410 | apply_one_bool("verbose", params, &value, NULL); |
3228cc1d | 411 | pretty->options.verbose = value; |
a263021c | 412 | |
6e1bc0df | 413 | /* Names. */ |
834e9996 | 414 | apply_one_string("name-default", params, &str); |
6e1bc0df | 415 | if (!str) { |
3228cc1d | 416 | pretty->options.name_default = PRETTY_DEFAULT_UNSET; |
6e1bc0df | 417 | } else if (!strcmp(str, "show")) { |
3228cc1d | 418 | pretty->options.name_default = PRETTY_DEFAULT_SHOW; |
6e1bc0df | 419 | } else if (!strcmp(str, "hide")) { |
3228cc1d | 420 | pretty->options.name_default = PRETTY_DEFAULT_HIDE; |
6e1bc0df | 421 | } else { |
834e9996 | 422 | ret = -1; |
6e1bc0df MD |
423 | goto end; |
424 | } | |
425 | g_free(str); | |
426 | str = NULL; | |
427 | ||
3228cc1d PP |
428 | switch (pretty->options.name_default) { |
429 | case PRETTY_DEFAULT_UNSET: | |
430 | pretty->options.print_payload_field_names = true; | |
431 | pretty->options.print_context_field_names = true; | |
432 | pretty->options.print_header_field_names = false; | |
433 | pretty->options.print_scope_field_names = false; | |
6e1bc0df | 434 | break; |
3228cc1d PP |
435 | case PRETTY_DEFAULT_SHOW: |
436 | pretty->options.print_payload_field_names = true; | |
437 | pretty->options.print_context_field_names = true; | |
438 | pretty->options.print_header_field_names = true; | |
439 | pretty->options.print_scope_field_names = true; | |
6e1bc0df | 440 | break; |
3228cc1d PP |
441 | case PRETTY_DEFAULT_HIDE: |
442 | pretty->options.print_payload_field_names = false; | |
443 | pretty->options.print_context_field_names = false; | |
444 | pretty->options.print_header_field_names = false; | |
445 | pretty->options.print_scope_field_names = false; | |
6e1bc0df MD |
446 | break; |
447 | default: | |
834e9996 | 448 | ret = -1; |
6e1bc0df MD |
449 | goto end; |
450 | } | |
451 | ||
452 | value = false; | |
453 | found = false; | |
834e9996 | 454 | apply_one_bool("name-payload", params, &value, &found); |
6e1bc0df | 455 | if (found) { |
3228cc1d | 456 | pretty->options.print_payload_field_names = value; |
6e1bc0df MD |
457 | } |
458 | ||
459 | value = false; | |
460 | found = false; | |
834e9996 | 461 | apply_one_bool("name-context", params, &value, &found); |
6e1bc0df | 462 | if (found) { |
3228cc1d | 463 | pretty->options.print_context_field_names = value; |
6e1bc0df MD |
464 | } |
465 | ||
466 | value = false; | |
467 | found = false; | |
834e9996 | 468 | apply_one_bool("name-header", params, &value, &found); |
6e1bc0df | 469 | if (found) { |
3228cc1d | 470 | pretty->options.print_header_field_names = value; |
6e1bc0df MD |
471 | } |
472 | ||
473 | value = false; | |
474 | found = false; | |
834e9996 | 475 | apply_one_bool("name-scope", params, &value, &found); |
6e1bc0df | 476 | if (found) { |
3228cc1d | 477 | pretty->options.print_scope_field_names = value; |
6e1bc0df MD |
478 | } |
479 | ||
480 | /* Fields. */ | |
834e9996 | 481 | apply_one_string("field-default", params, &str); |
6e1bc0df | 482 | if (!str) { |
3228cc1d | 483 | pretty->options.field_default = PRETTY_DEFAULT_UNSET; |
6e1bc0df | 484 | } else if (!strcmp(str, "show")) { |
3228cc1d | 485 | pretty->options.field_default = PRETTY_DEFAULT_SHOW; |
6e1bc0df | 486 | } else if (!strcmp(str, "hide")) { |
3228cc1d | 487 | pretty->options.field_default = PRETTY_DEFAULT_HIDE; |
6e1bc0df | 488 | } else { |
834e9996 | 489 | ret = -1; |
6e1bc0df MD |
490 | goto end; |
491 | } | |
492 | g_free(str); | |
493 | str = NULL; | |
494 | ||
3228cc1d PP |
495 | switch (pretty->options.field_default) { |
496 | case PRETTY_DEFAULT_UNSET: | |
497 | pretty->options.print_trace_field = false; | |
498 | pretty->options.print_trace_hostname_field = true; | |
499 | pretty->options.print_trace_domain_field = false; | |
500 | pretty->options.print_trace_procname_field = true; | |
501 | pretty->options.print_trace_vpid_field = true; | |
502 | pretty->options.print_loglevel_field = false; | |
503 | pretty->options.print_emf_field = false; | |
504 | pretty->options.print_callsite_field = false; | |
6e1bc0df | 505 | break; |
3228cc1d PP |
506 | case PRETTY_DEFAULT_SHOW: |
507 | pretty->options.print_trace_field = true; | |
508 | pretty->options.print_trace_hostname_field = true; | |
509 | pretty->options.print_trace_domain_field = true; | |
510 | pretty->options.print_trace_procname_field = true; | |
511 | pretty->options.print_trace_vpid_field = true; | |
512 | pretty->options.print_loglevel_field = true; | |
513 | pretty->options.print_emf_field = true; | |
514 | pretty->options.print_callsite_field = true; | |
6e1bc0df | 515 | break; |
3228cc1d PP |
516 | case PRETTY_DEFAULT_HIDE: |
517 | pretty->options.print_trace_field = false; | |
518 | pretty->options.print_trace_hostname_field = false; | |
519 | pretty->options.print_trace_domain_field = false; | |
520 | pretty->options.print_trace_procname_field = false; | |
521 | pretty->options.print_trace_vpid_field = false; | |
522 | pretty->options.print_loglevel_field = false; | |
523 | pretty->options.print_emf_field = false; | |
524 | pretty->options.print_callsite_field = false; | |
6e1bc0df MD |
525 | break; |
526 | default: | |
834e9996 | 527 | ret = -1; |
6e1bc0df MD |
528 | goto end; |
529 | } | |
530 | ||
531 | value = false; | |
532 | found = false; | |
834e9996 | 533 | apply_one_bool("field-trace", params, &value, &found); |
6e1bc0df | 534 | if (found) { |
3228cc1d | 535 | pretty->options.print_trace_field = value; |
6e1bc0df MD |
536 | } |
537 | ||
538 | value = false; | |
539 | found = false; | |
834e9996 | 540 | apply_one_bool("field-trace:hostname", params, &value, &found); |
6e1bc0df | 541 | if (found) { |
3228cc1d | 542 | pretty->options.print_trace_hostname_field = value; |
6e1bc0df MD |
543 | } |
544 | ||
545 | value = false; | |
546 | found = false; | |
834e9996 | 547 | apply_one_bool("field-trace:domain", params, &value, &found); |
6e1bc0df | 548 | if (found) { |
3228cc1d | 549 | pretty->options.print_trace_domain_field = value; |
6e1bc0df MD |
550 | } |
551 | ||
552 | value = false; | |
553 | found = false; | |
834e9996 | 554 | apply_one_bool("field-trace:procname", params, &value, &found); |
6e1bc0df | 555 | if (found) { |
3228cc1d | 556 | pretty->options.print_trace_procname_field = value; |
6e1bc0df MD |
557 | } |
558 | ||
559 | value = false; | |
560 | found = false; | |
834e9996 | 561 | apply_one_bool("field-trace:vpid", params, &value, &found); |
6e1bc0df | 562 | if (found) { |
3228cc1d | 563 | pretty->options.print_trace_vpid_field = value; |
6e1bc0df MD |
564 | } |
565 | ||
566 | value = false; | |
567 | found = false; | |
834e9996 | 568 | apply_one_bool("field-loglevel", params, &value, &found); |
6e1bc0df | 569 | if (found) { |
3228cc1d | 570 | pretty->options.print_loglevel_field = value; |
6e1bc0df MD |
571 | } |
572 | ||
573 | value = false; | |
574 | found = false; | |
834e9996 | 575 | apply_one_bool("field-emf", params, &value, &found); |
6e1bc0df | 576 | if (found) { |
3228cc1d | 577 | pretty->options.print_emf_field = value; |
6e1bc0df MD |
578 | } |
579 | ||
580 | value = false; | |
581 | found = false; | |
834e9996 | 582 | apply_one_bool("field-callsite", params, &value, &found); |
6e1bc0df | 583 | if (found) { |
3228cc1d | 584 | pretty->options.print_callsite_field = value; |
6e1bc0df MD |
585 | } |
586 | ||
6e1bc0df | 587 | end: |
8c6884d9 | 588 | bt_value_put_ref(pretty->plugin_opt_map); |
3228cc1d | 589 | pretty->plugin_opt_map = NULL; |
6e1bc0df MD |
590 | g_free(str); |
591 | return ret; | |
592 | } | |
593 | ||
ad96d936 | 594 | static |
3228cc1d | 595 | void set_use_colors(struct pretty_component *pretty) |
ad96d936 | 596 | { |
3228cc1d PP |
597 | switch (pretty->options.color) { |
598 | case PRETTY_COLOR_OPT_ALWAYS: | |
599 | pretty->use_colors = true; | |
ad96d936 | 600 | break; |
3228cc1d PP |
601 | case PRETTY_COLOR_OPT_AUTO: |
602 | pretty->use_colors = pretty->out == stdout && | |
ad96d936 PP |
603 | bt_common_colors_supported(); |
604 | break; | |
3228cc1d PP |
605 | case PRETTY_COLOR_OPT_NEVER: |
606 | pretty->use_colors = false; | |
ad96d936 PP |
607 | break; |
608 | } | |
609 | } | |
610 | ||
2b4c4a7c JD |
611 | static |
612 | void init_stream_packet_context_quarks(void) | |
613 | { | |
614 | stream_packet_context_quarks[Q_TIMESTAMP_BEGIN] = | |
615 | g_quark_from_string("timestamp_begin"); | |
616 | stream_packet_context_quarks[Q_TIMESTAMP_BEGIN] = | |
617 | g_quark_from_string("timestamp_begin"); | |
618 | stream_packet_context_quarks[Q_TIMESTAMP_END] = | |
619 | g_quark_from_string("timestamp_end"); | |
620 | stream_packet_context_quarks[Q_EVENTS_DISCARDED] = | |
621 | g_quark_from_string("events_discarded"); | |
622 | stream_packet_context_quarks[Q_CONTENT_SIZE] = | |
623 | g_quark_from_string("content_size"); | |
624 | stream_packet_context_quarks[Q_PACKET_SIZE] = | |
625 | g_quark_from_string("packet_size"); | |
626 | stream_packet_context_quarks[Q_PACKET_SEQ_NUM] = | |
627 | g_quark_from_string("packet_seq_num"); | |
628 | } | |
629 | ||
3228cc1d | 630 | BT_HIDDEN |
fb25b9e3 PP |
631 | bt_component_class_init_method_status pretty_init( |
632 | bt_self_component_sink *comp, const bt_value *params, | |
f807570c | 633 | __attribute__((unused)) void *init_method_data) |
bac67f0f | 634 | { |
fb25b9e3 PP |
635 | bt_component_class_init_method_status ret = |
636 | BT_COMPONENT_CLASS_INIT_METHOD_STATUS_OK; | |
3228cc1d | 637 | struct pretty_component *pretty = create_pretty(); |
bac67f0f | 638 | |
3228cc1d | 639 | if (!pretty) { |
fb25b9e3 | 640 | ret = BT_COMPONENT_CLASS_INIT_METHOD_STATUS_MEMORY_ERROR; |
bac67f0f JG |
641 | goto end; |
642 | } | |
643 | ||
fb25b9e3 PP |
644 | if (bt_self_component_sink_add_input_port(comp, |
645 | in_port_name, NULL, NULL) < 0) { | |
646 | ret = BT_COMPONENT_CLASS_INIT_METHOD_STATUS_MEMORY_ERROR; | |
647 | goto error; | |
b9d103be PP |
648 | } |
649 | ||
3228cc1d PP |
650 | pretty->out = stdout; |
651 | pretty->err = stderr; | |
6e1bc0df | 652 | |
3228cc1d PP |
653 | pretty->delta_cycles = -1ULL; |
654 | pretty->last_cycles_timestamp = -1ULL; | |
3af83b5a | 655 | |
3228cc1d PP |
656 | pretty->delta_real_timestamp = -1ULL; |
657 | pretty->last_real_timestamp = -1ULL; | |
3af83b5a | 658 | |
834e9996 | 659 | if (apply_params(pretty, params)) { |
fb25b9e3 | 660 | ret = BT_COMPONENT_CLASS_INIT_METHOD_STATUS_ERROR; |
6e1bc0df MD |
661 | goto error; |
662 | } | |
663 | ||
3228cc1d | 664 | set_use_colors(pretty); |
834e9996 | 665 | bt_self_component_set_data( |
bb61965b | 666 | bt_self_component_sink_as_self_component(comp), pretty); |
2b4c4a7c JD |
667 | init_stream_packet_context_quarks(); |
668 | ||
bac67f0f JG |
669 | end: |
670 | return ret; | |
834e9996 | 671 | |
bac67f0f | 672 | error: |
3228cc1d | 673 | destroy_pretty_data(pretty); |
bac67f0f JG |
674 | return ret; |
675 | } |