lib/component/notification/packet.c

   1 /*
   2  * Babeltrace Plug-in Packet-related Notifications
   3  *
   4  * Copyright 2016 Jérémie Galarneau <jeremie.galarneau@efficios.com>
   5  *
   6  * Author: Jérémie Galarneau <jeremie.galarneau@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  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  19  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  20  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  21  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  22  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  23  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
  24  * SOFTWARE.
  25  */
  26
  27 #include <babeltrace/compiler.h>
  28 #include <babeltrace/component/notification/packet-internal.h>
  29
  30 static
  31 void bt_notification_packet_begin_destroy(struct bt_object *obj)
  32 {
  33         struct bt_notification_packet_begin *notification =
  34                         (struct bt_notification_packet_begin *) obj;
  35
  36         BT_PUT(notification->packet);
  37         g_free(notification);
  38 }
  39
  40 static
  41 void bt_notification_packet_end_destroy(struct bt_object *obj)
  42 {
  43         struct bt_notification_packet_end *notification =
  44                         (struct bt_notification_packet_end *) obj;
  45
  46         BT_PUT(notification->packet);
  47         g_free(notification);
  48 }
  49
  50 struct bt_notification *bt_notification_packet_begin_create(
  51                 struct bt_ctf_packet *packet)
  52 {
  53         struct bt_notification_packet_begin *notification;
  54
  55         if (!packet) {
  56                 goto error;
  57         }
  58
  59         notification = g_new0(struct bt_notification_packet_begin, 1);
  60         bt_notification_init(&notification->parent,
  61                         BT_NOTIFICATION_TYPE_PACKET_BEGIN,
  62                         bt_notification_packet_begin_destroy);
  63         notification->packet = bt_get(packet);
  64         return &notification->parent;
  65 error:
  66         return NULL;
  67 }
  68
  69 struct bt_ctf_packet *bt_notification_packet_begin_get_packet(
  70                 struct bt_notification *notification)
  71 {
  72         struct bt_ctf_packet *ret = NULL;
  73         struct bt_notification_packet_begin *packet_begin;
  74
  75         if (notification->type != BT_NOTIFICATION_TYPE_PACKET_BEGIN) {
  76                 goto end;
  77         }
  78
  79         packet_begin = container_of(notification,
  80                         struct bt_notification_packet_begin, parent);
  81         ret = bt_get(packet_begin->packet);
  82 end:
  83         return ret;
  84 }
  85
  86 struct bt_notification *bt_notification_packet_end_create(
  87                 struct bt_ctf_packet *packet)
  88 {
  89         struct bt_notification_packet_end *notification;
  90
  91         if (!packet) {
  92                 goto error;
  93         }
  94
  95         notification = g_new0(struct bt_notification_packet_end, 1);
  96         bt_notification_init(&notification->parent,
  97                         BT_NOTIFICATION_TYPE_PACKET_END,
  98                         bt_notification_packet_end_destroy);
  99         notification->packet = bt_get(packet);
 100         return &notification->parent;
 101 error:
 102         return NULL;
 103 }
 104
 105 struct bt_ctf_packet *bt_notification_packet_end_get_packet(
 106                 struct bt_notification *notification)
 107 {
 108         struct bt_ctf_packet *ret = NULL;
 109         struct bt_notification_packet_end *packet_end;
 110
 111         if (notification->type != BT_NOTIFICATION_TYPE_PACKET_END) {
 112                 goto end;
 113         }
 114
 115         packet_end = container_of(notification,
 116                         struct bt_notification_packet_end, parent);
 117         ret = bt_get(packet_end->packet);
 118 end:
 119         return ret;
 120 }