Implement ctf-metadata output plugin
[babeltrace.git] / formats / ctf-text / ctf-text.c
index cbf074e07e76db4f03bfd028803ffbdf94745081..b44ed8440f69b87f2dd1a70114869e03b7bf855d 100644 (file)
  *
  * The above copyright notice and this permission notice shall be included in
  * all copies or substantial portions of the Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+ * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+ * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+ * SOFTWARE.
  */
 
 #include <babeltrace/format.h>
@@ -79,11 +87,18 @@ enum bt_loglevel {
 };
 
 static
-struct trace_descriptor *ctf_text_open_trace(const char *path, int flags,
-               void (*packet_seek)(struct stream_pos *pos, size_t index,
+struct bt_trace_descriptor *ctf_text_open_trace(const char *path, int flags,
+               void (*packet_seek)(struct bt_stream_pos *pos, size_t index,
+                       int whence), FILE *metadata_fp);
+static
+int ctf_text_close_trace(struct bt_trace_descriptor *descriptor);
+
+static
+struct bt_trace_descriptor *ctf_metadata_open_trace(const char *path, int flags,
+               void (*packet_seek)(struct bt_stream_pos *pos, size_t index,
                        int whence), FILE *metadata_fp);
 static
-void ctf_text_close_trace(struct trace_descriptor *descriptor);
+int ctf_metadata_close_trace(struct bt_trace_descriptor *descriptor);
 
 static
 rw_dispatch write_dispatch_table[] = {
@@ -98,11 +113,17 @@ rw_dispatch write_dispatch_table[] = {
 };
 
 static
-struct format ctf_text_format = {
+struct bt_format ctf_text_format = {
        .open_trace = ctf_text_open_trace,
        .close_trace = ctf_text_close_trace,
 };
 
+static
+struct bt_format ctf_metadata_format = {
+       .open_trace = ctf_metadata_open_trace,
+       .close_trace = ctf_metadata_close_trace,
+};
+
 static GQuark Q_STREAM_PACKET_CONTEXT_TIMESTAMP_BEGIN,
        Q_STREAM_PACKET_CONTEXT_TIMESTAMP_END,
        Q_STREAM_PACKET_CONTEXT_EVENTS_DISCARDED,
@@ -127,7 +148,7 @@ struct ctf_callsite_dups *ctf_trace_callsite_lookup(struct ctf_trace *trace,
                        (gpointer) (unsigned long) callsite_name);
 }
 
-int print_field(struct definition *definition)
+int print_field(struct bt_definition *definition)
 {
        /* Print all fields in verbose mode */
        if (babeltrace_verbose)
@@ -224,7 +245,7 @@ const char *print_loglevel(int value)
 }
 
 static
-int ctf_text_write_event(struct stream_pos *ppos, struct ctf_stream_definition *stream)
+int ctf_text_write_event(struct bt_stream_pos *ppos, struct ctf_stream_definition *stream)
                         
 {
        struct ctf_text_stream_pos *pos =
@@ -257,23 +278,7 @@ int ctf_text_write_event(struct stream_pos *ppos, struct ctf_stream_definition *
        /* Print events discarded */
        if (stream->events_discarded) {
                fflush(pos->fp);
-               fprintf(stderr, "[warning] Tracer discarded %" PRIu64 " events between [",
-                       stream->events_discarded);
-               if (opt_clock_cycles) {
-                       ctf_print_timestamp(stderr, stream,
-                                       stream->prev_cycles_timestamp);
-                       fprintf(stderr, "] and [");
-                       ctf_print_timestamp(stderr, stream,
-                                       stream->prev_cycles_timestamp_end);
-               } else {
-                       ctf_print_timestamp(stderr, stream,
-                                       stream->prev_real_timestamp);
-                       fprintf(stderr, "] and [");
-                       ctf_print_timestamp(stderr, stream,
-                                       stream->prev_real_timestamp_end);
-               }
-               fprintf(stderr, "]. You should consider recording a new trace with larger buffers or with fewer events enabled.\n");
-               fflush(stderr);
+               ctf_print_discarded(stderr, stream, 0);
                stream->events_discarded = 0;
        }
 
@@ -296,7 +301,7 @@ int ctf_text_write_event(struct stream_pos *ppos, struct ctf_stream_definition *
                else
                        fprintf(pos->fp, " ");
        }
-       if ((opt_delta_field || opt_all_fields) && stream->has_timestamp) {
+       if (opt_delta_field && stream->has_timestamp) {
                uint64_t delta, delta_sec, delta_nsec;
 
                set_field_names_print(pos, ITEM_HEADER);
@@ -547,8 +552,8 @@ error:
 }
 
 static
-struct trace_descriptor *ctf_text_open_trace(const char *path, int flags,
-               void (*packet_seek)(struct stream_pos *pos, size_t index,
+struct bt_trace_descriptor *ctf_text_open_trace(const char *path, int flags,
+               void (*packet_seek)(struct bt_stream_pos *pos, size_t index,
                        int whence), FILE *metadata_fp)
 {
        struct ctf_text_stream_pos *pos;
@@ -584,12 +589,92 @@ error:
 }
 
 static
-void ctf_text_close_trace(struct trace_descriptor *td)
+int ctf_text_close_trace(struct bt_trace_descriptor *td)
+{
+       int ret;
+       struct ctf_text_stream_pos *pos =
+               container_of(td, struct ctf_text_stream_pos, trace_descriptor);
+       if (pos->fp != stdout) {
+               ret = fclose(pos->fp);
+               if (ret) {
+                       perror("Error on fclose");
+                       return -1;
+               }
+       }
+       g_free(pos);
+       return 0;
+}
+
+static
+int ctf_metadata_trace_pre_handler(struct bt_stream_pos *ppos,
+                       struct bt_trace_descriptor *td)
+{
+       struct ctf_text_stream_pos *pos =
+               container_of(ppos, struct ctf_text_stream_pos, parent);
+       struct ctf_trace *trace;
+
+       trace = container_of(td, struct ctf_trace, parent);
+       if (!trace->metadata_string)
+               return -EINVAL;
+       if (trace->metadata_packetized) {
+               fprintf(pos->fp, "/* CTF %u.%u */\n",
+                       BT_CTF_MAJOR, BT_CTF_MINOR);
+       }
+       fprintf(pos->fp, "%s", trace->metadata_string);
+       return 0;
+}
+
+static
+struct bt_trace_descriptor *ctf_metadata_open_trace(const char *path, int flags,
+               void (*packet_seek)(struct bt_stream_pos *pos, size_t index,
+                       int whence), FILE *metadata_fp)
+{
+       struct ctf_text_stream_pos *pos;
+       FILE *fp;
+
+       pos = g_new0(struct ctf_text_stream_pos, 1);
+
+       pos->last_real_timestamp = -1ULL;
+       pos->last_cycles_timestamp = -1ULL;
+       switch (flags & O_ACCMODE) {
+       case O_RDWR:
+               if (!path)
+                       fp = stdout;
+               else
+                       fp = fopen(path, "w");
+               if (!fp)
+                       goto error;
+               pos->fp = fp;
+               pos->parent.pre_trace_cb = ctf_metadata_trace_pre_handler;
+               pos->print_names = 0;
+               break;
+       case O_RDONLY:
+       default:
+               fprintf(stderr, "[error] Incorrect open flags.\n");
+               goto error;
+       }
+
+       return &pos->trace_descriptor;
+error:
+       g_free(pos);
+       return NULL;
+}
+
+static
+int ctf_metadata_close_trace(struct bt_trace_descriptor *td)
 {
+       int ret;
        struct ctf_text_stream_pos *pos =
                container_of(td, struct ctf_text_stream_pos, trace_descriptor);
-       fclose(pos->fp);
+       if (pos->fp != stdout) {
+               ret = fclose(pos->fp);
+               if (ret) {
+                       perror("Error on fclose");
+                       return -1;
+               }
+       }
        g_free(pos);
+       return 0;
 }
 
 static
@@ -600,10 +685,14 @@ void __attribute__((constructor)) ctf_text_init(void)
        ctf_text_format.name = g_quark_from_static_string("text");
        ret = bt_register_format(&ctf_text_format);
        assert(!ret);
+       ctf_metadata_format.name = g_quark_from_static_string("ctf-metadata");
+       ret = bt_register_format(&ctf_metadata_format);
+       assert(!ret);
 }
 
 static
 void __attribute__((destructor)) ctf_text_exit(void)
 {
+       bt_unregister_format(&ctf_metadata_format);
        bt_unregister_format(&ctf_text_format);
 }
This page took 0.02585 seconds and 4 git commands to generate.