Commit | Line | Data |
---|---|---|
642ec66d MD |
1 | /* |
2 | * BabelTrace - Common Trace Format (CTF) | |
3 | * | |
4 | * CTF Text Format registration. | |
5 | * | |
6 | * Copyright 2010, 2011 - Mathieu Desnoyers <mathieu.desnoyers@efficios.com> | |
7 | * | |
8 | * Permission is hereby granted, free of charge, to any person obtaining a copy | |
9 | * of this software and associated documentation files (the "Software"), to deal | |
10 | * in the Software without restriction, including without limitation the rights | |
11 | * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell | |
12 | * copies of the Software, and to permit persons to whom the Software is | |
13 | * furnished to do so, subject to the following conditions: | |
14 | * | |
15 | * The above copyright notice and this permission notice shall be included in | |
16 | * all copies or substantial portions of the Software. | |
17 | */ | |
18 | ||
19 | #include <babeltrace/format.h> | |
20 | #include <babeltrace/ctf-text/types.h> | |
31262354 | 21 | #include <babeltrace/ctf/metadata.h> |
642ec66d MD |
22 | #include <babeltrace/babeltrace.h> |
23 | #include <inttypes.h> | |
24 | #include <uuid/uuid.h> | |
25 | #include <sys/mman.h> | |
26 | #include <errno.h> | |
27 | #include <sys/types.h> | |
28 | #include <sys/stat.h> | |
29 | #include <fcntl.h> | |
30 | #include <dirent.h> | |
31 | #include <glib.h> | |
32 | #include <unistd.h> | |
33 | #include <stdlib.h> | |
34 | ||
35 | struct trace_descriptor *ctf_text_open_trace(const char *path, int flags); | |
36 | void ctf_text_close_trace(struct trace_descriptor *descriptor); | |
37 | ||
642ec66d MD |
38 | static |
39 | rw_dispatch write_dispatch_table[] = { | |
40 | [ CTF_TYPE_INTEGER ] = ctf_text_integer_write, | |
41 | [ CTF_TYPE_FLOAT ] = ctf_text_float_write, | |
42 | [ CTF_TYPE_ENUM ] = ctf_text_enum_write, | |
43 | [ CTF_TYPE_STRING ] = ctf_text_string_write, | |
44 | [ CTF_TYPE_STRUCT ] = ctf_text_struct_write, | |
45 | [ CTF_TYPE_VARIANT ] = ctf_text_variant_write, | |
46 | [ CTF_TYPE_ARRAY ] = ctf_text_array_write, | |
47 | [ CTF_TYPE_SEQUENCE ] = ctf_text_sequence_write, | |
48 | }; | |
49 | ||
50 | static | |
51 | struct format ctf_text_format = { | |
52 | .open_trace = ctf_text_open_trace, | |
53 | .close_trace = ctf_text_close_trace, | |
54 | }; | |
55 | ||
31262354 MD |
56 | static |
57 | int ctf_text_write_event(struct stream_pos *ppos, | |
764af3f4 | 58 | struct ctf_stream *stream) |
31262354 MD |
59 | { |
60 | struct ctf_text_stream_pos *pos = | |
61 | container_of(ppos, struct ctf_text_stream_pos, parent); | |
764af3f4 | 62 | struct ctf_stream_class *stream_class = stream->stream_class; |
31262354 MD |
63 | struct ctf_event *event_class; |
64 | uint64_t id = 0; | |
65 | int len_index; | |
66 | int ret; | |
aa6bffae | 67 | int field_nr = 0; |
31262354 MD |
68 | |
69 | /* print event header */ | |
70 | if (stream_class->event_header) { | |
71 | /* lookup event id */ | |
72 | len_index = struct_declaration_lookup_field_index(stream_class->event_header_decl, | |
73 | g_quark_from_static_string("id")); | |
74 | if (len_index >= 0) { | |
75 | struct definition_integer *defint; | |
76 | struct definition *field; | |
77 | ||
78 | field = struct_definition_get_field_from_index(stream_class->event_header, len_index); | |
79 | assert(field->declaration->id == CTF_TYPE_INTEGER); | |
80 | defint = container_of(field, struct definition_integer, p); | |
81 | assert(defint->declaration->signedness == FALSE); | |
82 | id = defint->value._unsigned; /* set id */ | |
83 | } | |
84 | } | |
85 | ||
86 | if (id >= stream_class->events_by_id->len) { | |
87 | fprintf(stdout, "[error] Event id %" PRIu64 " is outside range.\n", id); | |
88 | return -EINVAL; | |
89 | } | |
90 | event_class = g_ptr_array_index(stream_class->events_by_id, id); | |
91 | if (!event_class) { | |
92 | fprintf(stdout, "[error] Event id %" PRIu64 " is unknown.\n", id); | |
93 | return -EINVAL; | |
94 | } | |
95 | ||
aa6bffae MD |
96 | if (pos->print_names) |
97 | fprintf(pos->fp, "timestamp = "); | |
98 | else | |
99 | fprintf(pos->fp, "["); | |
764af3f4 | 100 | fprintf(pos->fp, "%12" PRIu64, stream->timestamp); |
aa6bffae MD |
101 | if (!pos->print_names) |
102 | fprintf(pos->fp, "]"); | |
103 | ||
8178fe48 | 104 | if (pos->print_names) |
aa6bffae | 105 | fprintf(pos->fp, ", "); |
8178fe48 MD |
106 | else |
107 | fprintf(pos->fp, " "); | |
31262354 MD |
108 | if (pos->print_names) |
109 | fprintf(pos->fp, "name = "); | |
8178fe48 MD |
110 | fprintf(pos->fp, "%s", g_quark_to_string(event_class->name)); |
111 | if (pos->print_names) | |
112 | field_nr++; | |
113 | else | |
114 | fprintf(pos->fp, ": "); | |
31262354 | 115 | |
764af3f4 MD |
116 | /* Only show the event header in verbose mode */ |
117 | if (babeltrace_verbose && stream_class->event_header) { | |
aa6bffae MD |
118 | if (field_nr++ != 0) |
119 | fprintf(pos->fp, ", "); | |
31262354 | 120 | if (pos->print_names) |
aa6bffae | 121 | fprintf(pos->fp, "stream.event.header = "); |
31262354 MD |
122 | ret = generic_rw(ppos, &stream_class->event_header->p); |
123 | if (ret) | |
124 | goto error; | |
125 | } | |
126 | ||
127 | /* print stream-declared event context */ | |
128 | if (stream_class->event_context) { | |
aa6bffae MD |
129 | if (field_nr++ != 0) |
130 | fprintf(pos->fp, ", "); | |
31262354 | 131 | if (pos->print_names) |
aa6bffae | 132 | fprintf(pos->fp, "stream.event.context = "); |
31262354 MD |
133 | ret = generic_rw(ppos, &stream_class->event_context->p); |
134 | if (ret) | |
135 | goto error; | |
136 | } | |
137 | ||
138 | /* print event-declared event context */ | |
139 | if (event_class->context) { | |
aa6bffae MD |
140 | if (field_nr++ != 0) |
141 | fprintf(pos->fp, ", "); | |
31262354 | 142 | if (pos->print_names) |
aa6bffae | 143 | fprintf(pos->fp, "event.context = "); |
31262354 MD |
144 | ret = generic_rw(ppos, &event_class->context->p); |
145 | if (ret) | |
146 | goto error; | |
147 | } | |
148 | ||
149 | /* Read and print event payload */ | |
150 | if (event_class->fields) { | |
aa6bffae MD |
151 | if (field_nr++ != 0) |
152 | fprintf(pos->fp, ", "); | |
31262354 | 153 | if (pos->print_names) |
aa6bffae | 154 | fprintf(pos->fp, "event.fields = "); |
31262354 MD |
155 | ret = generic_rw(ppos, &event_class->fields->p); |
156 | if (ret) | |
157 | goto error; | |
158 | } | |
159 | fprintf(pos->fp, "\n"); | |
160 | ||
161 | return 0; | |
162 | ||
163 | error: | |
164 | fprintf(stdout, "[error] Unexpected end of stream. Either the trace data stream is corrupted or metadata description does not match data layout.\n"); | |
165 | return ret; | |
166 | } | |
167 | ||
168 | ||
642ec66d MD |
169 | struct trace_descriptor *ctf_text_open_trace(const char *path, int flags) |
170 | { | |
171 | struct ctf_text_stream_pos *pos; | |
172 | FILE *fp; | |
173 | ||
174 | pos = g_new0(struct ctf_text_stream_pos, 1); | |
175 | ||
176 | switch (flags & O_ACCMODE) { | |
989c73bc | 177 | case O_RDWR: |
b61922b5 | 178 | if (!path) |
6cf7957b MD |
179 | fp = stdout; |
180 | else | |
181 | fp = fopen(path, "w"); | |
642ec66d MD |
182 | if (!fp) |
183 | goto error; | |
184 | pos->fp = fp; | |
185 | pos->parent.rw_table = write_dispatch_table; | |
31262354 | 186 | pos->parent.event_cb = ctf_text_write_event; |
d63ca2cd | 187 | pos->print_names = opt_field_names; |
642ec66d MD |
188 | break; |
189 | case O_RDONLY: | |
190 | default: | |
191 | fprintf(stdout, "[error] Incorrect open flags.\n"); | |
192 | goto error; | |
193 | } | |
194 | ||
195 | return &pos->trace_descriptor; | |
196 | error: | |
197 | g_free(pos); | |
198 | return NULL; | |
199 | } | |
200 | ||
201 | void ctf_text_close_trace(struct trace_descriptor *td) | |
202 | { | |
203 | struct ctf_text_stream_pos *pos = | |
204 | container_of(td, struct ctf_text_stream_pos, trace_descriptor); | |
205 | fclose(pos->fp); | |
206 | g_free(pos); | |
207 | } | |
208 | ||
209 | void __attribute__((constructor)) ctf_text_init(void) | |
210 | { | |
211 | int ret; | |
212 | ||
213 | ctf_text_format.name = g_quark_from_static_string("text"); | |
214 | ret = bt_register_format(&ctf_text_format); | |
215 | assert(!ret); | |
216 | } | |
217 | ||
218 | /* TODO: finalize */ |