#include <babeltrace/ctf-ir/packet.h>
#include <babeltrace/plugin/plugin-dev.h>
#include <babeltrace/component/component.h>
+#include <babeltrace/component/private-component.h>
#include <babeltrace/component/component-sink.h>
+#include <babeltrace/component/private-component-sink.h>
+#include <babeltrace/component/private-port.h>
+#include <babeltrace/component/private-connection.h>
#include <babeltrace/component/notification/notification.h>
#include <babeltrace/component/notification/iterator.h>
#include <babeltrace/component/notification/event.h>
#include <babeltrace/component/notification/packet.h>
+#include <plugins-common.h>
#include <stdio.h>
#include <stdbool.h>
#include <glib.h>
#include "writer.h"
+#include <assert.h>
static
void destroy_writer_component_data(struct writer_component *writer_component)
{
+ bt_put(writer_component->input_iterator);
g_hash_table_destroy(writer_component->stream_map);
g_hash_table_destroy(writer_component->stream_class_map);
g_hash_table_destroy(writer_component->trace_map);
}
static
-void destroy_writer_component(struct bt_component *component)
+void destroy_writer_component(struct bt_private_component *component)
{
struct writer_component *writer_component = (struct writer_component *)
- bt_component_get_private_data(component);
+ bt_private_component_get_user_data(component);
destroy_writer_component_data(writer_component);
g_free(writer_component);
}
static
-enum bt_component_status run(struct bt_component *component)
+enum bt_component_status writer_component_accept_port_connection(
+ struct bt_private_component *component,
+ struct bt_private_port *self_port)
+{
+ enum bt_component_status ret = BT_COMPONENT_STATUS_OK;
+ struct bt_private_connection *connection;
+ struct writer_component *writer;
+
+ writer = bt_private_component_get_user_data(component);
+ assert(writer);
+ assert(!writer->input_iterator);
+ connection = bt_private_port_get_private_connection(self_port);
+ assert(connection);
+ writer->input_iterator =
+ bt_private_connection_create_notification_iterator(connection);
+
+ if (!writer->input_iterator) {
+ ret = BT_COMPONENT_STATUS_ERROR;
+ }
+ bt_put(connection);
+ return ret;
+}
+
+static
+enum bt_component_status run(struct bt_private_component *component)
{
enum bt_component_status ret;
+ enum bt_notification_iterator_status it_status;
struct bt_notification *notification = NULL;
struct bt_notification_iterator *it;
struct writer_component *writer_component =
- bt_component_get_private_data(component);
+ bt_private_component_get_user_data(component);
- ret = bt_component_sink_get_input_iterator(component, 0, &it);
- if (ret != BT_COMPONENT_STATUS_OK) {
- goto end;
- }
+ it = writer_component->input_iterator;
+ assert(it);
notification = bt_notification_iterator_get_notification(it);
if (!notification) {
goto end;
}
- ret = bt_notification_iterator_next(it);
- if (ret != BT_COMPONENT_STATUS_OK) {
+ it_status = bt_notification_iterator_next(it);
+ if (it_status != BT_COMPONENT_STATUS_OK) {
+ ret = BT_COMPONENT_STATUS_ERROR;
goto end;
}
static
enum bt_component_status writer_component_init(
- struct bt_component *component, struct bt_value *params,
- void *init_method_data)
+ struct bt_private_component *component, struct bt_value *params,
+ UNUSED_VAR void *init_method_data)
{
enum bt_component_status ret;
enum bt_value_status value_ret;
struct writer_component *writer_component = create_writer_component();
struct bt_value *value = NULL;
const char *path;
- (void) init_method_data;
if (!writer_component) {
ret = BT_COMPONENT_STATUS_NOMEM;
goto error;
}
- ret = bt_component_set_private_data(component, writer_component);
+ ret = bt_private_component_set_user_data(component, writer_component);
if (ret != BT_COMPONENT_STATUS_OK) {
goto error;
}
BT_PLUGIN_LICENSE("MIT");
BT_PLUGIN_SINK_COMPONENT_CLASS(writer, run);
BT_PLUGIN_SINK_COMPONENT_CLASS_INIT_METHOD(writer, writer_component_init);
+BT_PLUGIN_SINK_COMPONENT_CLASS_ACCEPT_PORT_CONNECTION_METHOD(writer,
+ writer_component_accept_port_connection);
BT_PLUGIN_SINK_COMPONENT_CLASS_DESTROY_METHOD(writer, destroy_writer_component);
BT_PLUGIN_SINK_COMPONENT_CLASS_DESCRIPTION(writer, "Formats CTF-IR to CTF.");