liblttng-ust/ltt-ring-buffer-metadata-client.h

   1 /*
   2  * ltt-ring-buffer-client.h
   3  *
   4  * LTTng lib ring buffer client template.
   5  *
   6  * Copyright (C) 2010-2012 Mathieu Desnoyers <mathieu.desnoyers@efficios.com>
   7  *
   8  * This library is free software; you can redistribute it and/or
   9  * modify it under the terms of the GNU Lesser General Public
  10  * License as published by the Free Software Foundation; only
  11  * version 2.1 of the License.
  12  *
  13  * This library is distributed in the hope that it will be useful,
  14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  16  * Lesser General Public License for more details.
  17  *
  18  * You should have received a copy of the GNU Lesser General Public
  19  * License along with this library; if not, write to the Free Software
  20  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  21  */
  22
  23 #include <stdint.h>
  24 #include <lttng/ust-events.h>
  25 #include "lttng/bitfield.h"
  26 #include "ltt-tracer.h"
  27 #include "../libringbuffer/frontend_types.h"
  28
  29 struct metadata_packet_header {
  30         uint32_t magic;                 /* 0x75D11D57 */
  31         uint8_t  uuid[LTTNG_UST_UUID_LEN]; /* Unique Universal Identifier */
  32         uint32_t checksum;              /* 0 if unused */
  33         uint32_t content_size;          /* in bits */
  34         uint32_t packet_size;           /* in bits */
  35         uint8_t  compression_scheme;    /* 0 if unused */
  36         uint8_t  encryption_scheme;     /* 0 if unused */
  37         uint8_t  checksum_scheme;       /* 0 if unused */
  38         uint8_t  major;                 /* CTF spec major version number */
  39         uint8_t  minor;                 /* CTF spec minor version number */
  40         uint8_t  header_end[0];
  41 };
  42
  43 struct metadata_record_header {
  44         uint8_t header_end[0];          /* End of header */
  45 };
  46
  47 static const struct lttng_ust_lib_ring_buffer_config client_config;
  48
  49 static inline uint64_t lib_ring_buffer_clock_read(struct channel *chan)
  50 {
  51         return 0;
  52 }
  53
  54 static inline
  55 unsigned char record_header_size(const struct lttng_ust_lib_ring_buffer_config *config,
  56                                  struct channel *chan, size_t offset,
  57                                  size_t *pre_header_padding,
  58                                  struct lttng_ust_lib_ring_buffer_ctx *ctx)
  59 {
  60         return 0;
  61 }
  62
  63 #include "../libringbuffer/api.h"
  64
  65 static uint64_t client_ring_buffer_clock_read(struct channel *chan)
  66 {
  67         return 0;
  68 }
  69
  70 static
  71 size_t client_record_header_size(const struct lttng_ust_lib_ring_buffer_config *config,
  72                                  struct channel *chan, size_t offset,
  73                                  size_t *pre_header_padding,
  74                                  struct lttng_ust_lib_ring_buffer_ctx *ctx)
  75 {
  76         return 0;
  77 }
  78
  79 /**
  80  * client_packet_header_size - called on buffer-switch to a new sub-buffer
  81  *
  82  * Return header size without padding after the structure. Don't use packed
  83  * structure because gcc generates inefficient code on some architectures
  84  * (powerpc, mips..)
  85  */
  86 static size_t client_packet_header_size(void)
  87 {
  88         return offsetof(struct metadata_packet_header, header_end);
  89 }
  90
  91 static void client_buffer_begin(struct lttng_ust_lib_ring_buffer *buf, uint64_t tsc,
  92                                 unsigned int subbuf_idx,
  93                                 struct lttng_ust_shm_handle *handle)
  94 {
  95         struct channel *chan = shmp(handle, buf->backend.chan);
  96         struct metadata_packet_header *header =
  97                 (struct metadata_packet_header *)
  98                         lib_ring_buffer_offset_address(&buf->backend,
  99                                 subbuf_idx * chan->backend.subbuf_size,
 100                                 handle);
 101         struct ltt_channel *ltt_chan = channel_get_private(chan);
 102
 103         header->magic = TSDL_MAGIC_NUMBER;
 104         memcpy(header->uuid, ltt_chan->uuid, sizeof(ltt_chan->uuid));
 105         header->checksum = 0;           /* 0 if unused */
 106         header->content_size = 0xFFFFFFFF; /* in bits, for debugging */
 107         header->packet_size = 0xFFFFFFFF;  /* in bits, for debugging */
 108         header->compression_scheme = 0; /* 0 if unused */
 109         header->encryption_scheme = 0;  /* 0 if unused */
 110         header->checksum_scheme = 0;    /* 0 if unused */
 111         header->major = CTF_SPEC_MAJOR;
 112         header->minor = CTF_SPEC_MINOR;
 113
 114 }
 115
 116 /*
 117  * offset is assumed to never be 0 here : never deliver a completely empty
 118  * subbuffer. data_size is between 1 and subbuf_size.
 119  */
 120 static void client_buffer_end(struct lttng_ust_lib_ring_buffer *buf, uint64_t tsc,
 121                               unsigned int subbuf_idx, unsigned long data_size,
 122                               struct lttng_ust_shm_handle *handle)
 123 {
 124         struct channel *chan = shmp(handle, buf->backend.chan);
 125         struct metadata_packet_header *header =
 126                 (struct metadata_packet_header *)
 127                         lib_ring_buffer_offset_address(&buf->backend,
 128                                 subbuf_idx * chan->backend.subbuf_size,
 129                                 handle);
 130         unsigned long records_lost = 0;
 131
 132         header->content_size = data_size * CHAR_BIT;            /* in bits */
 133         header->packet_size = PAGE_ALIGN(data_size) * CHAR_BIT; /* in bits */
 134         /*
 135          * We do not care about the records lost count, because the metadata
 136          * channel waits and retry.
 137          */
 138         (void) lib_ring_buffer_get_records_lost_full(&client_config, buf);
 139         records_lost += lib_ring_buffer_get_records_lost_wrap(&client_config, buf);
 140         records_lost += lib_ring_buffer_get_records_lost_big(&client_config, buf);
 141         WARN_ON_ONCE(records_lost != 0);
 142 }
 143
 144 static int client_buffer_create(struct lttng_ust_lib_ring_buffer *buf, void *priv,
 145                                 int cpu, const char *name,
 146                                 struct lttng_ust_shm_handle *handle)
 147 {
 148         return 0;
 149 }
 150
 151 static void client_buffer_finalize(struct lttng_ust_lib_ring_buffer *buf,
 152                                    void *priv, int cpu,
 153                                    struct lttng_ust_shm_handle *handle)
 154 {
 155 }
 156
 157 static const struct lttng_ust_lib_ring_buffer_config client_config = {
 158         .cb.ring_buffer_clock_read = client_ring_buffer_clock_read,
 159         .cb.record_header_size = client_record_header_size,
 160         .cb.subbuffer_header_size = client_packet_header_size,
 161         .cb.buffer_begin = client_buffer_begin,
 162         .cb.buffer_end = client_buffer_end,
 163         .cb.buffer_create = client_buffer_create,
 164         .cb.buffer_finalize = client_buffer_finalize,
 165
 166         .tsc_bits = 0,
 167         .alloc = RING_BUFFER_ALLOC_GLOBAL,
 168         .sync = RING_BUFFER_SYNC_GLOBAL,
 169         .mode = RING_BUFFER_MODE_TEMPLATE,
 170         .backend = RING_BUFFER_PAGE,
 171         .output = RING_BUFFER_MMAP,
 172         .oops = RING_BUFFER_OOPS_CONSISTENCY,
 173         .ipi = RING_BUFFER_NO_IPI_BARRIER,
 174         .wakeup = RING_BUFFER_WAKEUP_BY_WRITER,
 175         .client_type = LTTNG_CLIENT_TYPE,
 176 };
 177
 178 const struct lttng_ust_lib_ring_buffer_client_cb *LTTNG_CLIENT_CALLBACKS = &client_config.cb;
 179
 180 static
 181 struct ltt_channel *_channel_create(const char *name,
 182                                 void *buf_addr,
 183                                 size_t subbuf_size, size_t num_subbuf,
 184                                 unsigned int switch_timer_interval,
 185                                 unsigned int read_timer_interval,
 186                                 int **shm_fd, char **shm_path,
 187                                 int **wait_fd, char **wait_pipe_path,
 188                                 uint64_t **memory_map_size,
 189                                 struct ltt_channel *chan_priv_init)
 190 {
 191         void *priv;
 192         struct ltt_channel *ltt_chan = NULL;
 193         struct lttng_ust_shm_handle *handle;
 194
 195         handle = channel_create(&client_config, name,
 196                         &priv, __alignof__(*ltt_chan), sizeof(*ltt_chan),
 197                         chan_priv_init,
 198                         buf_addr, subbuf_size, num_subbuf,
 199                         switch_timer_interval, read_timer_interval,
 200                         shm_fd, shm_path, wait_fd, wait_pipe_path,
 201                         memory_map_size);
 202         if (!handle)
 203                 return NULL;
 204         ltt_chan = priv;
 205         ltt_chan->handle = handle;
 206         ltt_chan->chan = shmp(ltt_chan->handle, ltt_chan->handle->chan);
 207         return ltt_chan;
 208 }
 209
 210 static
 211 int ltt_channel_open_pipe(struct channel *chan,
 212                           struct lttng_ust_shm_handle *handle)
 213 {
 214         struct shm_ref *ref;
 215         int cpu = 0;
 216
 217         if (client_config.alloc == RING_BUFFER_ALLOC_GLOBAL) {
 218                 ref = &chan->backend.buf[0].shmp._ref;
 219                 return shm_open_wakeup_pipe(handle, ref);
 220         } else {
 221                 if (cpu >= num_possible_cpus()) {
 222                         goto error;
 223                 }
 224
 225                 for_each_possible_cpu(cpu) {
 226                         ref = &chan->backend.buf[cpu].shmp._ref;
 227                         if (shm_open_wakeup_pipe(handle, ref) < 0) {
 228                                 goto error;
 229                         }
 230                 }
 231         }
 232
 233         return 0;
 234 error:
 235         return -1;
 236 }
 237
 238 static
 239 void ltt_channel_destroy(struct ltt_channel *ltt_chan)
 240 {
 241         channel_destroy(ltt_chan->chan, ltt_chan->handle, 0);
 242 }
 243
 244 static
 245 struct lttng_ust_lib_ring_buffer *ltt_buffer_read_open(struct channel *chan,
 246                                                        struct lttng_ust_shm_handle *handle,
 247                                                        int **shm_fd, char **shm_path,
 248                                                        int **wait_fd, char **wait_pipe_path,
 249                                                        uint64_t **memory_map_size)
 250 {
 251         struct lttng_ust_lib_ring_buffer *buf;
 252
 253         buf = channel_get_ring_buffer(&client_config, chan,
 254                                       0, handle, shm_fd, shm_path,
 255                                       wait_fd, wait_pipe_path,
 256                                       memory_map_size);
 257         if (!lib_ring_buffer_open_read(buf, handle, 0))
 258                 return buf;
 259         return NULL;
 260 }
 261
 262 static
 263 void ltt_buffer_read_close(struct lttng_ust_lib_ring_buffer *buf,
 264                            struct lttng_ust_shm_handle *handle)
 265 {
 266         lib_ring_buffer_release_read(buf, handle, 0);
 267 }
 268
 269 static
 270 int ltt_event_reserve(struct lttng_ust_lib_ring_buffer_ctx *ctx, uint32_t event_id)
 271 {
 272         return lib_ring_buffer_reserve(&client_config, ctx);
 273 }
 274
 275 static
 276 void ltt_event_commit(struct lttng_ust_lib_ring_buffer_ctx *ctx)
 277 {
 278         lib_ring_buffer_commit(&client_config, ctx);
 279 }
 280
 281 static
 282 void ltt_event_write(struct lttng_ust_lib_ring_buffer_ctx *ctx, const void *src,
 283                      size_t len)
 284 {
 285         lib_ring_buffer_write(&client_config, ctx, src, len);
 286 }
 287
 288 static
 289 size_t ltt_packet_avail_size(struct channel *chan, struct lttng_ust_shm_handle *handle)
 290
 291 {
 292         unsigned long o_begin;
 293         struct lttng_ust_lib_ring_buffer *buf;
 294
 295         buf = shmp(handle, chan->backend.buf[0].shmp);  /* Only for global buffer ! */
 296         o_begin = v_read(&client_config, &buf->offset);
 297         if (subbuf_offset(o_begin, chan) != 0) {
 298                 return chan->backend.subbuf_size - subbuf_offset(o_begin, chan);
 299         } else {
 300                 return chan->backend.subbuf_size - subbuf_offset(o_begin, chan)
 301                         - sizeof(struct metadata_packet_header);
 302         }
 303 }
 304
 305 #if 0
 306 static
 307 wait_queue_head_t *ltt_get_reader_wait_queue(struct channel *chan)
 308 {
 309         return &chan->read_wait;
 310 }
 311
 312 static
 313 wait_queue_head_t *ltt_get_hp_wait_queue(struct channel *chan)
 314 {
 315         return &chan->hp_wait;
 316 }
 317 #endif //0
 318
 319 static
 320 int ltt_is_finalized(struct channel *chan)
 321 {
 322         return lib_ring_buffer_channel_is_finalized(chan);
 323 }
 324
 325 static
 326 int ltt_is_disabled(struct channel *chan)
 327 {
 328         return lib_ring_buffer_channel_is_disabled(chan);
 329 }
 330
 331 static
 332 int ltt_flush_buffer(struct channel *chan, struct lttng_ust_shm_handle *handle)
 333 {
 334         struct lttng_ust_lib_ring_buffer *buf;
 335         int *shm_fd, *wait_fd;
 336         char *shm_path, *wait_pipe_path;
 337         uint64_t *memory_map_size;
 338
 339         buf = channel_get_ring_buffer(&client_config, chan,
 340                                       0, handle, &shm_fd, &shm_path,
 341                                       &wait_fd, &wait_pipe_path,
 342                                       &memory_map_size);
 343         lib_ring_buffer_switch(&client_config, buf,
 344                         SWITCH_ACTIVE, handle);
 345         return 0;
 346 }
 347
 348 static struct ltt_transport ltt_relay_transport = {
 349         .name = "relay-" RING_BUFFER_MODE_TEMPLATE_STRING "-mmap",
 350         .ops = {
 351                 .channel_create = _channel_create,
 352                 .channel_open_pipe = ltt_channel_open_pipe,
 353                 .channel_destroy = ltt_channel_destroy,
 354                 .buffer_read_open = ltt_buffer_read_open,
 355                 .buffer_read_close = ltt_buffer_read_close,
 356                 .event_reserve = ltt_event_reserve,
 357                 .event_commit = ltt_event_commit,
 358                 .event_write = ltt_event_write,
 359                 .packet_avail_size = ltt_packet_avail_size,
 360                 //.get_reader_wait_queue = ltt_get_reader_wait_queue,
 361                 //.get_hp_wait_queue = ltt_get_hp_wait_queue,
 362                 .is_finalized = ltt_is_finalized,
 363                 .is_disabled = ltt_is_disabled,
 364                 .flush_buffer = ltt_flush_buffer,
 365         },
 366 };
 367
 368 void RING_BUFFER_MODE_TEMPLATE_INIT(void)
 369 {
 370         DBG("LTT : ltt ring buffer client init\n");
 371         ltt_transport_register(&ltt_relay_transport);
 372 }
 373
 374 void RING_BUFFER_MODE_TEMPLATE_EXIT(void)
 375 {
 376         DBG("LTT : ltt ring buffer client exit\n");
 377         ltt_transport_unregister(&ltt_relay_transport);
 378 }