src/bin/lttng-sessiond/trace-ust.h

   1 /*
   2  * Copyright (C) 2011 - David Goulet <david.goulet@polymtl.ca>
   3  *
   4  * This program is free software; you can redistribute it and/or modify
   5  * it under the terms of the GNU General Public License, version 2 only,
   6  * as published by the Free Software Foundation.
   7  *
   8  * This program is distributed in the hope that it will be useful,
   9  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  10  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  11  * GNU General Public License for more details.
  12  *
  13  * You should have received a copy of the GNU General Public License along
  14  * with this program; if not, write to the Free Software Foundation, Inc.,
  15  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  16  */
  17
  18 #ifndef _LTT_TRACE_UST_H
  19 #define _LTT_TRACE_UST_H
  20
  21 #include <config.h>
  22 #include <limits.h>
  23 #include <urcu/list.h>
  24
  25 #include <lttng/lttng.h>
  26 #include <common/hashtable/hashtable.h>
  27 #include <common/defaults.h>
  28
  29 #include "consumer.h"
  30 #include "ust-ctl.h"
  31
  32 struct ltt_ust_ht_key {
  33         const char *name;
  34         const struct lttng_filter_bytecode *filter;
  35         enum lttng_ust_loglevel_type loglevel;
  36 };
  37
  38 /* Context hash table nodes */
  39 struct ltt_ust_context {
  40         struct lttng_ust_context ctx;
  41         struct lttng_ht_node_ulong node;
  42 };
  43
  44 /* UST event */
  45 struct ltt_ust_event {
  46         unsigned int enabled;
  47         struct lttng_ust_event attr;
  48         struct lttng_ht_node_str node;
  49         struct lttng_ust_filter_bytecode *filter;
  50 };
  51
  52 /* UST channel */
  53 struct ltt_ust_channel {
  54         uint64_t id;    /* unique id per session. */
  55         unsigned int enabled;
  56         char name[LTTNG_UST_SYM_NAME_LEN];
  57         struct lttng_ust_channel_attr attr;
  58         struct lttng_ht *ctx;
  59         struct lttng_ht *events;
  60         struct lttng_ht_node_str node;
  61         uint64_t tracefile_size;
  62         uint64_t tracefile_count;
  63 };
  64
  65 /* UST Metadata */
  66 struct ltt_ust_metadata {
  67         int handle;
  68         struct lttng_ust_object_data *obj;
  69         char pathname[PATH_MAX];              /* Trace file path name */
  70         struct lttng_ust_channel_attr attr;
  71         struct lttng_ust_object_data *stream_obj;
  72 };
  73
  74 /* UST domain global (LTTNG_DOMAIN_UST) */
  75 struct ltt_ust_domain_global {
  76         struct lttng_ht *channels;
  77         struct cds_list_head registry_buffer_uid_list;
  78 };
  79
  80 /* UST session */
  81 struct ltt_ust_session {
  82         uint64_t id;    /* Unique identifier of session */
  83         int start_trace;
  84         struct ltt_ust_domain_global domain_global;
  85         /* UID/GID of the user owning the session */
  86         uid_t uid;
  87         gid_t gid;
  88         /*
  89          * Two consumer_output object are needed where one is for the current
  90          * output object and the second one is the temporary object used to store
  91          * URI being set by the lttng_set_consumer_uri call. Once
  92          * lttng_enable_consumer is called, the two pointers are swapped.
  93          */
  94         struct consumer_output *consumer;
  95         struct consumer_output *tmp_consumer;
  96         /* Sequence number for filters so the tracer knows the ordering. */
  97         uint64_t filter_seq_num;
  98         /* This indicates which type of buffer this session is set for. */
  99         enum lttng_buffer_type buffer_type;
 100         /* If set to 1, the buffer_type can not be changed anymore. */
 101         int buffer_type_changed;
 102         /* For per UID buffer, every buffer reg object is kept of this session */
 103         struct cds_list_head buffer_reg_uid_list;
 104         /* Next channel ID available for a newly registered channel. */
 105         uint64_t next_channel_id;
 106         /* Once this value reaches UINT32_MAX, no more id can be allocated. */
 107         uint64_t used_channel_id;
 108         /* Tell or not if the session has to output the traces. */
 109         unsigned int output_traces;
 110         unsigned int snapshot_mode;
 111         unsigned int has_non_default_channel;
 112 };
 113
 114 /*
 115  * Validate that the id has reached the maximum allowed or not.
 116  *
 117  * Return 0 if NOT else 1.
 118  */
 119 static inline int trace_ust_is_max_id(uint64_t id)
 120 {
 121         return (id == UINT64_MAX) ? 1 : 0;
 122 }
 123
 124 /*
 125  * Return next available channel id and increment the used counter. The
 126  * trace_ust_is_max_id function MUST be called before in order to validate if
 127  * the maximum number of IDs have been reached. If not, it is safe to call this
 128  * function.
 129  *
 130  * Return a unique channel ID. If max is reached, the used_channel_id counter
 131  * is returned.
 132  */
 133 static inline uint64_t trace_ust_get_next_chan_id(struct ltt_ust_session *s)
 134 {
 135         if (trace_ust_is_max_id(s->used_channel_id)) {
 136                 return s->used_channel_id;
 137         }
 138
 139         s->used_channel_id++;
 140         return s->next_channel_id++;
 141 }
 142
 143 #ifdef HAVE_LIBLTTNG_UST_CTL
 144
 145 int trace_ust_ht_match_event(struct cds_lfht_node *node, const void *_key);
 146 int trace_ust_ht_match_event_by_name(struct cds_lfht_node *node,
 147                 const void *_key);
 148
 149 /*
 150  * Lookup functions. NULL is returned if not found.
 151  */
 152 struct ltt_ust_event *trace_ust_find_event(struct lttng_ht *ht,
 153                 char *name, struct lttng_filter_bytecode *filter, int loglevel);
 154 struct ltt_ust_channel *trace_ust_find_channel_by_name(struct lttng_ht *ht,
 155                 char *name);
 156
 157 /*
 158  * Create functions malloc() the data structure.
 159  */
 160 struct ltt_ust_session *trace_ust_create_session(uint64_t session_id);
 161 struct ltt_ust_channel *trace_ust_create_channel(struct lttng_channel *attr);
 162 struct ltt_ust_event *trace_ust_create_event(struct lttng_event *ev,
 163                 struct lttng_filter_bytecode *filter);
 164 struct ltt_ust_metadata *trace_ust_create_metadata(char *path);
 165 struct ltt_ust_context *trace_ust_create_context(
 166                 struct lttng_event_context *ctx);
 167 void trace_ust_delete_channel(struct lttng_ht *ht,
 168                 struct ltt_ust_channel *channel);
 169
 170 /*
 171  * Destroy functions free() the data structure and remove from linked list if
 172  * it's applies.
 173  */
 174 void trace_ust_destroy_session(struct ltt_ust_session *session);
 175 void trace_ust_destroy_metadata(struct ltt_ust_metadata *metadata);
 176 void trace_ust_destroy_channel(struct ltt_ust_channel *channel);
 177 void trace_ust_destroy_event(struct ltt_ust_event *event);
 178
 179 #else /* HAVE_LIBLTTNG_UST_CTL */
 180
 181 static inline int trace_ust_ht_match_event(struct cds_lfht_node *node,
 182                 const void *_key)
 183 {
 184         return 0;
 185 }
 186 static inline int trace_ust_ht_match_event_by_name(struct cds_lfht_node *node,
 187                 const void *_key)
 188 {
 189         return 0;
 190 }
 191 static inline
 192 struct ltt_ust_channel *trace_ust_find_channel_by_name(struct lttng_ht *ht,
 193                 char *name)
 194 {
 195         return NULL;
 196 }
 197
 198 static inline
 199 struct ltt_ust_session *trace_ust_create_session(unsigned int session_id)
 200 {
 201         return NULL;
 202 }
 203 static inline
 204 struct ltt_ust_channel *trace_ust_create_channel(struct lttng_channel *attr)
 205 {
 206         return NULL;
 207 }
 208 static inline
 209 struct ltt_ust_event *trace_ust_create_event(struct lttng_event *ev,
 210                 struct lttng_filter_bytecode *filter)
 211 {
 212         return NULL;
 213 }
 214 static inline
 215 struct ltt_ust_metadata *trace_ust_create_metadata(char *path)
 216 {
 217         return NULL;
 218 }
 219
 220 static inline
 221 void trace_ust_destroy_session(struct ltt_ust_session *session)
 222 {
 223 }
 224
 225 static inline
 226 void trace_ust_destroy_metadata(struct ltt_ust_metadata *metadata)
 227 {
 228 }
 229
 230 static inline
 231 void trace_ust_destroy_channel(struct ltt_ust_channel *channel)
 232 {
 233 }
 234
 235 static inline
 236 void trace_ust_destroy_event(struct ltt_ust_event *event)
 237 {
 238 }
 239 static inline
 240 struct ltt_ust_context *trace_ust_create_context(
 241                 struct lttng_event_context *ctx)
 242 {
 243         return NULL;
 244 }
 245 static inline struct ltt_ust_event *trace_ust_find_event(struct lttng_ht *ht,
 246                 char *name, struct lttng_filter_bytecode *filter, int loglevel)
 247 {
 248         return NULL;
 249 }
 250 static inline
 251 void trace_ust_delete_channel(struct lttng_ht *ht,
 252                 struct ltt_ust_channel *channel)
 253 {
 254         return;
 255 }
 256
 257 #endif /* HAVE_LIBLTTNG_UST_CTL */
 258
 259 #endif /* _LTT_TRACE_UST_H */