src/common/sessiond-comm/sessiond-comm.h

   1 /*
   2  * Copyright (C) 2011 - David Goulet <david.goulet@polymtl.ca>
   3  *                      Julien Desfossez <julien.desfossez@polymtl.ca>
   4  *                      Mathieu Desnoyers <mathieu.desnoyers@efficios.com>
   5  *
   6  * This program is free software; you can redistribute it and/or modify
   7  * it under the terms of the GNU General Public License, version 2 only,
   8  * as published by the Free Software Foundation.
   9  *
  10  * This program is distributed in the hope that it will be useful, but WITHOUT
  11  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  12  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
  13  * more details.
  14  *
  15  * You should have received a copy of the GNU General Public License along
  16  * with this program; if not, write to the Free Software Foundation, Inc.,
  17  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  18  */
  19
  20 /*
  21  * This header is meant for liblttng and libust internal use ONLY. These
  22  * declarations should NOT be considered stable API.
  23  */
  24
  25 #ifndef _LTTNG_SESSIOND_COMM_H
  26 #define _LTTNG_SESSIOND_COMM_H
  27
  28 #define _GNU_SOURCE
  29 #include <limits.h>
  30 #include <lttng/lttng.h>
  31 #include <lttng/snapshot-internal.h>
  32 #include <lttng/save-internal.h>
  33 #include <common/compat/socket.h>
  34 #include <common/uri.h>
  35 #include <common/defaults.h>
  36 #include <common/compat/uuid.h>
  37
  38 #include <arpa/inet.h>
  39 #include <netinet/in.h>
  40 #include <sys/un.h>
  41
  42 #include "inet.h"
  43 #include "inet6.h"
  44 #include "unix.h"
  45
  46 /* Queue size of listen(2) */
  47 #define LTTNG_SESSIOND_COMM_MAX_LISTEN 64
  48
  49 /* Maximum number of FDs that can be sent over a Unix socket */
  50 #define LTTCOMM_MAX_SEND_FDS           4
  51
  52 /*
  53  * Get the error code index from 0 since LTTCOMM_OK start at 1000
  54  */
  55 #define LTTCOMM_ERR_INDEX(code) (code - LTTCOMM_CONSUMERD_COMMAND_SOCK_READY)
  56
  57 enum lttcomm_sessiond_command {
  58         /* Tracer command */
  59         LTTNG_ADD_CONTEXT                   = 0,
  60         LTTNG_CALIBRATE                     = 1,
  61         LTTNG_DISABLE_CHANNEL               = 2,
  62         LTTNG_DISABLE_EVENT                 = 3,
  63         LTTNG_DISABLE_ALL_EVENT             = 4,
  64         LTTNG_ENABLE_CHANNEL                = 5,
  65         LTTNG_ENABLE_EVENT                  = 6,
  66         LTTNG_ENABLE_ALL_EVENT              = 7,
  67         /* Session daemon command */
  68         LTTNG_CREATE_SESSION                = 8,
  69         LTTNG_DESTROY_SESSION               = 9,
  70         LTTNG_LIST_CHANNELS                 = 10,
  71         LTTNG_LIST_DOMAINS                  = 11,
  72         LTTNG_LIST_EVENTS                   = 12,
  73         LTTNG_LIST_SESSIONS                 = 13,
  74         LTTNG_LIST_TRACEPOINTS              = 14,
  75         LTTNG_REGISTER_CONSUMER             = 15,
  76         LTTNG_START_TRACE                   = 16,
  77         LTTNG_STOP_TRACE                    = 17,
  78         LTTNG_LIST_TRACEPOINT_FIELDS        = 18,
  79
  80         /* Consumer */
  81         LTTNG_DISABLE_CONSUMER              = 19,
  82         LTTNG_ENABLE_CONSUMER               = 20,
  83         LTTNG_SET_CONSUMER_URI              = 21,
  84         /* 22 */
  85         /* 23 */
  86         LTTNG_DATA_PENDING                  = 24,
  87         LTTNG_SNAPSHOT_ADD_OUTPUT           = 25,
  88         LTTNG_SNAPSHOT_DEL_OUTPUT           = 26,
  89         LTTNG_SNAPSHOT_LIST_OUTPUT          = 27,
  90         LTTNG_SNAPSHOT_RECORD               = 28,
  91         LTTNG_CREATE_SESSION_SNAPSHOT       = 29,
  92         LTTNG_CREATE_SESSION_LIVE           = 30,
  93         LTTNG_SAVE_SESSION                  = 31,
  94 };
  95
  96 enum lttcomm_relayd_command {
  97         RELAYD_ADD_STREAM                   = 1,
  98         RELAYD_CREATE_SESSION               = 2,
  99         RELAYD_START_DATA                   = 3,
 100         RELAYD_UPDATE_SYNC_INFO             = 4,
 101         RELAYD_VERSION                      = 5,
 102         RELAYD_SEND_METADATA                = 6,
 103         RELAYD_CLOSE_STREAM                 = 7,
 104         RELAYD_DATA_PENDING                 = 8,
 105         RELAYD_QUIESCENT_CONTROL            = 9,
 106         RELAYD_BEGIN_DATA_PENDING           = 10,
 107         RELAYD_END_DATA_PENDING             = 11,
 108         RELAYD_ADD_INDEX                    = 12,
 109         RELAYD_SEND_INDEX                   = 13,
 110         RELAYD_CLOSE_INDEX                  = 14,
 111         /* Live-reading commands (2.4+). */
 112         RELAYD_LIST_SESSIONS                = 15,
 113         /* All streams of the channel have been sent to the relayd (2.4+). */
 114         RELAYD_STREAMS_SENT                 = 16,
 115 };
 116
 117 /*
 118  * lttcomm error code.
 119  */
 120 enum lttcomm_return_code {
 121         LTTCOMM_CONSUMERD_SUCCESS            = 0,   /* Everything went fine. */
 122         LTTCOMM_CONSUMERD_COMMAND_SOCK_READY = 1,   /* Command socket ready */
 123         LTTCOMM_CONSUMERD_SUCCESS_RECV_FD,          /* Success on receiving fds */
 124         LTTCOMM_CONSUMERD_ERROR_RECV_FD,            /* Error on receiving fds */
 125         LTTCOMM_CONSUMERD_ERROR_RECV_CMD,           /* Error on receiving command */
 126         LTTCOMM_CONSUMERD_POLL_ERROR,               /* Error in polling thread */
 127         LTTCOMM_CONSUMERD_POLL_NVAL,                /* Poll on closed fd */
 128         LTTCOMM_CONSUMERD_POLL_HUP,                 /* All fds have hungup */
 129         LTTCOMM_CONSUMERD_EXIT_SUCCESS,             /* Consumerd exiting normally */
 130         LTTCOMM_CONSUMERD_EXIT_FAILURE,             /* Consumerd exiting on error */
 131         LTTCOMM_CONSUMERD_OUTFD_ERROR,              /* Error opening the tracefile */
 132         LTTCOMM_CONSUMERD_SPLICE_EBADF,             /* EBADF from splice(2) */
 133         LTTCOMM_CONSUMERD_SPLICE_EINVAL,            /* EINVAL from splice(2) */
 134         LTTCOMM_CONSUMERD_SPLICE_ENOMEM,            /* ENOMEM from splice(2) */
 135         LTTCOMM_CONSUMERD_SPLICE_ESPIPE,            /* ESPIPE from splice(2) */
 136         LTTCOMM_CONSUMERD_ENOMEM,                   /* Consumer is out of memory */
 137         LTTCOMM_CONSUMERD_ERROR_METADATA,           /* Error with metadata. */
 138         LTTCOMM_CONSUMERD_FATAL,                    /* Fatal error. */
 139         LTTCOMM_CONSUMERD_RELAYD_FAIL,              /* Error on remote relayd */
 140         LTTCOMM_CONSUMERD_CHANNEL_FAIL,             /* Channel creation failed. */
 141         LTTCOMM_CONSUMERD_CHAN_NOT_FOUND,           /* Channel not found. */
 142
 143         /* MUST be last element */
 144         LTTCOMM_NR,                                             /* Last element */
 145 };
 146
 147 /* lttng socket protocol. */
 148 enum lttcomm_sock_proto {
 149         LTTCOMM_SOCK_UDP,
 150         LTTCOMM_SOCK_TCP,
 151 };
 152
 153 /*
 154  * Index in the net_families array below. Please keep in sync!
 155  */
 156 enum lttcomm_sock_domain {
 157         LTTCOMM_INET      = 0,
 158         LTTCOMM_INET6     = 1,
 159 };
 160
 161 enum lttcomm_metadata_command {
 162         LTTCOMM_METADATA_REQUEST = 1,
 163 };
 164
 165 /*
 166  * Commands sent from the consumerd to the sessiond to request if new metadata
 167  * is available. This message is used to find the per UID _or_ per PID registry
 168  * for the channel key. For per UID lookup, the triplet
 169  * bits_per_long/uid/session_id is used. On lookup failure, we search for the
 170  * per PID registry indexed by session id ignoring the other values.
 171  */
 172 struct lttcomm_metadata_request_msg {
 173         uint64_t session_id; /* Tracing session id */
 174         uint64_t session_id_per_pid; /* Tracing session id for per-pid */
 175         uint32_t bits_per_long; /* Consumer ABI */
 176         uint32_t uid;
 177         uint64_t key; /* Metadata channel key. */
 178 } LTTNG_PACKED;
 179
 180 struct lttcomm_sockaddr {
 181         enum lttcomm_sock_domain type;
 182         union {
 183                 struct sockaddr_in sin;
 184                 struct sockaddr_in6 sin6;
 185         } addr;
 186 } LTTNG_PACKED;
 187
 188 struct lttcomm_sock {
 189         int32_t fd;
 190         enum lttcomm_sock_proto proto;
 191         struct lttcomm_sockaddr sockaddr;
 192         const struct lttcomm_proto_ops *ops;
 193 } LTTNG_PACKED;
 194
 195 /*
 196  * Relayd sock. Adds the protocol version to use for the communications with
 197  * the relayd.
 198  */
 199 struct lttcomm_relayd_sock {
 200         struct lttcomm_sock sock;
 201         uint32_t major;
 202         uint32_t minor;
 203 } LTTNG_PACKED;
 204
 205 struct lttcomm_net_family {
 206         int family;
 207         int (*create) (struct lttcomm_sock *sock, int type, int proto);
 208 };
 209
 210 struct lttcomm_proto_ops {
 211         int (*bind) (struct lttcomm_sock *sock);
 212         int (*close) (struct lttcomm_sock *sock);
 213         int (*connect) (struct lttcomm_sock *sock);
 214         struct lttcomm_sock *(*accept) (struct lttcomm_sock *sock);
 215         int (*listen) (struct lttcomm_sock *sock, int backlog);
 216         ssize_t (*recvmsg) (struct lttcomm_sock *sock, void *buf, size_t len,
 217                         int flags);
 218         ssize_t (*sendmsg) (struct lttcomm_sock *sock, void *buf, size_t len,
 219                         int flags);
 220 };
 221
 222 /*
 223  * Data structure received from lttng client to session daemon.
 224  */
 225 struct lttcomm_session_msg {
 226         uint32_t cmd_type;      /* enum lttcomm_sessiond_command */
 227         struct lttng_session session;
 228         struct lttng_domain domain;
 229         union {
 230                 struct {
 231                         char channel_name[LTTNG_SYMBOL_NAME_LEN];
 232                         char name[NAME_MAX];
 233                 } LTTNG_PACKED disable;
 234                 /* Event data */
 235                 struct {
 236                         char channel_name[LTTNG_SYMBOL_NAME_LEN];
 237                         struct lttng_event event LTTNG_PACKED;
 238                         /* Length of following filter expression. */
 239                         uint32_t expression_len;
 240                         /* Length of following bytecode for filter. */
 241                         uint32_t bytecode_len;
 242                         /* exclusion data */
 243                         uint32_t exclusion_count;
 244                         /*
 245                          * After this structure, the following variable-length
 246                          * items are transmitted:
 247                          * - char exclusion_names[LTTNG_SYMBOL_NAME_LEN][exclusion_count]
 248                          * - unsigned char filter_expression[expression_len]
 249                          * - unsigned char filter_bytecode[bytecode_len]
 250                          */
 251                 } LTTNG_PACKED enable;
 252                 /* Create channel */
 253                 struct {
 254                         struct lttng_channel chan LTTNG_PACKED;
 255                 } LTTNG_PACKED channel;
 256                 /* Context */
 257                 struct {
 258                         char channel_name[LTTNG_SYMBOL_NAME_LEN];
 259                         struct lttng_event_context ctx LTTNG_PACKED;
 260                 } LTTNG_PACKED context;
 261                 /* Use by register_consumer */
 262                 struct {
 263                         char path[PATH_MAX];
 264                 } LTTNG_PACKED reg;
 265                 /* List */
 266                 struct {
 267                         char channel_name[LTTNG_SYMBOL_NAME_LEN];
 268                 } LTTNG_PACKED list;
 269                 struct lttng_calibrate calibrate;
 270                 /* Used by the set_consumer_url and used by create_session also call */
 271                 struct {
 272                         /* Number of lttng_uri following */
 273                         uint32_t size;
 274                 } LTTNG_PACKED uri;
 275                 struct {
 276                         struct lttng_snapshot_output output LTTNG_PACKED;
 277                 } LTTNG_PACKED snapshot_output;
 278                 struct {
 279                         uint32_t wait;
 280                         struct lttng_snapshot_output output LTTNG_PACKED;
 281                 } LTTNG_PACKED snapshot_record;
 282                 struct {
 283                         uint32_t nb_uri;
 284                         unsigned int timer_interval;    /* usec */
 285                 } LTTNG_PACKED session_live;
 286                 struct {
 287                         struct lttng_save_session_attr attr; /* struct already packed */
 288                 } LTTNG_PACKED save_session;
 289         } u;
 290 } LTTNG_PACKED;
 291
 292 #define LTTNG_FILTER_MAX_LEN    65536
 293
 294 /*
 295  * Filter bytecode data. The reloc table is located at the end of the
 296  * bytecode. It is made of tuples: (uint16_t, var. len. string). It
 297  * starts at reloc_table_offset.
 298  */
 299 #define LTTNG_FILTER_PADDING    32
 300 struct lttng_filter_bytecode {
 301         uint32_t len;   /* len of data */
 302         uint32_t reloc_table_offset;
 303         uint64_t seqnum;
 304         char padding[LTTNG_FILTER_PADDING];
 305         char data[0];
 306 } LTTNG_PACKED;
 307
 308 /*
 309  * Event exclusion data. At the end of the structure, there will actually
 310  * by zero or more names, where the actual number of names is given by
 311  * the 'count' item of the structure.
 312  */
 313 #define LTTNG_EVENT_EXCLUSION_PADDING   32
 314 struct lttng_event_exclusion {
 315         uint32_t count;
 316         char padding[LTTNG_EVENT_EXCLUSION_PADDING];
 317         char names[LTTNG_SYMBOL_NAME_LEN][0];
 318 } LTTNG_PACKED;
 319
 320 /*
 321  * Data structure for the response from sessiond to the lttng client.
 322  */
 323 struct lttcomm_lttng_msg {
 324         uint32_t cmd_type;      /* enum lttcomm_sessiond_command */
 325         uint32_t ret_code;      /* enum lttcomm_return_code */
 326         uint32_t pid;           /* pid_t */
 327         uint32_t data_size;
 328         /* Contains: trace_name + data */
 329         char payload[];
 330 } LTTNG_PACKED;
 331
 332 struct lttcomm_lttng_output_id {
 333         uint32_t id;
 334 } LTTNG_PACKED;
 335
 336 /*
 337  * lttcomm_consumer_msg is the message sent from sessiond to consumerd
 338  * to either add a channel, add a stream, update a stream, or stop
 339  * operation.
 340  */
 341 struct lttcomm_consumer_msg {
 342         uint32_t cmd_type;      /* enum consumerd_command */
 343         union {
 344                 struct {
 345                         uint64_t channel_key;
 346                         uint64_t session_id;
 347                         char pathname[PATH_MAX];
 348                         uint32_t uid;
 349                         uint32_t gid;
 350                         uint64_t relayd_id;
 351                         /* nb_init_streams is the number of streams open initially. */
 352                         uint32_t nb_init_streams;
 353                         char name[LTTNG_SYMBOL_NAME_LEN];
 354                         /* Use splice or mmap to consume this fd */
 355                         enum lttng_event_output output;
 356                         int type; /* Per cpu or metadata. */
 357                         uint64_t tracefile_size; /* bytes */
 358                         uint32_t tracefile_count; /* number of tracefiles */
 359                         /* If the channel's streams have to be monitored or not. */
 360                         uint32_t monitor;
 361                         /* timer to check the streams usage in live mode (usec). */
 362                         unsigned int live_timer_interval;
 363                 } LTTNG_PACKED channel; /* Only used by Kernel. */
 364                 struct {
 365                         uint64_t stream_key;
 366                         uint64_t channel_key;
 367                         int32_t cpu;    /* On which CPU this stream is assigned. */
 368                         /* Tells the consumer if the stream should be or not monitored. */
 369                         uint32_t no_monitor;
 370                 } LTTNG_PACKED stream;  /* Only used by Kernel. */
 371                 struct {
 372                         uint64_t net_index;
 373                         enum lttng_stream_type type;
 374                         /* Open socket to the relayd */
 375                         struct lttcomm_relayd_sock sock;
 376                         /* Tracing session id associated to the relayd. */
 377                         uint64_t session_id;
 378                         /* Relayd session id, only used with control socket. */
 379                         uint64_t relayd_session_id;
 380                 } LTTNG_PACKED relayd_sock;
 381                 struct {
 382                         uint64_t net_seq_idx;
 383                 } LTTNG_PACKED destroy_relayd;
 384                 struct {
 385                         uint64_t session_id;
 386                 } LTTNG_PACKED data_pending;
 387                 struct {
 388                         uint64_t subbuf_size;                   /* bytes */
 389                         uint64_t num_subbuf;                    /* power of 2 */
 390                         int32_t overwrite;                      /* 1: overwrite, 0: discard */
 391                         uint32_t switch_timer_interval;         /* usec */
 392                         uint32_t read_timer_interval;           /* usec */
 393                         unsigned int live_timer_interval;               /* usec */
 394                         int32_t output;                         /* splice, mmap */
 395                         int32_t type;                           /* metadata or per_cpu */
 396                         uint64_t session_id;                    /* Tracing session id */
 397                         char pathname[PATH_MAX];                /* Channel file path. */
 398                         char name[LTTNG_SYMBOL_NAME_LEN];       /* Channel name. */
 399                         uint32_t uid;                           /* User ID of the session */
 400                         uint32_t gid;                           /* Group ID ot the session */
 401                         uint64_t relayd_id;                     /* Relayd id if apply. */
 402                         uint64_t key;                           /* Unique channel key. */
 403                         unsigned char uuid[UUID_LEN];   /* uuid for ust tracer. */
 404                         uint32_t chan_id;                       /* Channel ID on the tracer side. */
 405                         uint64_t tracefile_size;        /* bytes */
 406                         uint32_t tracefile_count;       /* number of tracefiles */
 407                         uint64_t session_id_per_pid;    /* Per-pid session ID. */
 408                         /* Tells the consumer if the stream should be or not monitored. */
 409                         uint32_t monitor;
 410                         /*
 411                          * For UST per UID buffers, this is the application UID of the
 412                          * channel.  This can be different from the user UID requesting the
 413                          * channel creation and used for the rights on the stream file
 414                          * because the application can be in the tracing for instance.
 415                          */
 416                         uint32_t ust_app_uid;
 417                 } LTTNG_PACKED ask_channel;
 418                 struct {
 419                         uint64_t key;
 420                 } LTTNG_PACKED get_channel;
 421                 struct {
 422                         uint64_t key;
 423                 } LTTNG_PACKED destroy_channel;
 424                 struct {
 425                         uint64_t key;   /* Metadata channel key. */
 426                         uint64_t target_offset; /* Offset in the consumer */
 427                         uint64_t len;   /* Length of metadata to be received. */
 428                 } LTTNG_PACKED push_metadata;
 429                 struct {
 430                         uint64_t key;   /* Metadata channel key. */
 431                 } LTTNG_PACKED close_metadata;
 432                 struct {
 433                         uint64_t key;   /* Metadata channel key. */
 434                 } LTTNG_PACKED setup_metadata;
 435                 struct {
 436                         uint64_t key;   /* Channel key. */
 437                 } LTTNG_PACKED flush_channel;
 438                 struct {
 439                         char pathname[PATH_MAX];
 440                         /* Indicate if the snapshot goes on the relayd or locally. */
 441                         uint32_t use_relayd;
 442                         uint32_t metadata;              /* This a metadata snapshot. */
 443                         uint64_t relayd_id;             /* Relayd id if apply. */
 444                         uint64_t key;
 445                         uint64_t max_stream_size;
 446                 } LTTNG_PACKED snapshot_channel;
 447                 struct {
 448                         uint64_t channel_key;
 449                         uint64_t net_seq_idx;
 450                 } LTTNG_PACKED sent_streams;
 451         } u;
 452 } LTTNG_PACKED;
 453
 454 /*
 455  * Status message returned to the sessiond after a received command.
 456  */
 457 struct lttcomm_consumer_status_msg {
 458         enum lttcomm_return_code ret_code;
 459 } LTTNG_PACKED;
 460
 461 struct lttcomm_consumer_status_channel {
 462         enum lttcomm_return_code ret_code;
 463         uint64_t key;
 464         unsigned int stream_count;
 465 } LTTNG_PACKED;
 466
 467 #ifdef HAVE_LIBLTTNG_UST_CTL
 468
 469 #include <lttng/ust-abi.h>
 470
 471 /*
 472  * Data structure for the commands sent from sessiond to UST.
 473  */
 474 struct lttcomm_ust_msg {
 475         uint32_t handle;
 476         uint32_t cmd;
 477         union {
 478                 struct lttng_ust_channel channel;
 479                 struct lttng_ust_stream stream;
 480                 struct lttng_ust_event event;
 481                 struct lttng_ust_context context;
 482                 struct lttng_ust_tracer_version version;
 483         } u;
 484 } LTTNG_PACKED;
 485
 486 /*
 487  * Data structure for the response from UST to the session daemon.
 488  * cmd_type is sent back in the reply for validation.
 489  */
 490 struct lttcomm_ust_reply {
 491         uint32_t handle;
 492         uint32_t cmd;
 493         uint32_t ret_code;      /* enum lttcomm_return_code */
 494         uint32_t ret_val;       /* return value */
 495         union {
 496                 struct {
 497                         uint64_t memory_map_size;
 498                 } LTTNG_PACKED channel;
 499                 struct {
 500                         uint64_t memory_map_size;
 501                 } LTTNG_PACKED stream;
 502                 struct lttng_ust_tracer_version version;
 503         } u;
 504 } LTTNG_PACKED;
 505
 506 #endif /* HAVE_LIBLTTNG_UST_CTL */
 507
 508 extern const char *lttcomm_get_readable_code(enum lttcomm_return_code code);
 509
 510 extern int lttcomm_init_inet_sockaddr(struct lttcomm_sockaddr *sockaddr,
 511                 const char *ip, unsigned int port);
 512 extern int lttcomm_init_inet6_sockaddr(struct lttcomm_sockaddr *sockaddr,
 513                 const char *ip, unsigned int port);
 514
 515 extern struct lttcomm_sock *lttcomm_alloc_sock(enum lttcomm_sock_proto proto);
 516 extern int lttcomm_create_sock(struct lttcomm_sock *sock);
 517 extern struct lttcomm_sock *lttcomm_alloc_sock_from_uri(struct lttng_uri *uri);
 518 extern void lttcomm_destroy_sock(struct lttcomm_sock *sock);
 519 extern struct lttcomm_sock *lttcomm_alloc_copy_sock(struct lttcomm_sock *src);
 520 extern void lttcomm_copy_sock(struct lttcomm_sock *dst,
 521                 struct lttcomm_sock *src);
 522
 523 /* Relayd socket object. */
 524 extern struct lttcomm_relayd_sock *lttcomm_alloc_relayd_sock(
 525                 struct lttng_uri *uri, uint32_t major, uint32_t minor);
 526
 527 extern int lttcomm_setsockopt_rcv_timeout(int sock, unsigned int msec);
 528 extern int lttcomm_setsockopt_snd_timeout(int sock, unsigned int msec);
 529
 530 extern void lttcomm_init(void);
 531 /* Get network timeout, in milliseconds */
 532 extern unsigned long lttcomm_get_network_timeout(void);
 533
 534 #endif  /* _LTTNG_SESSIOND_COMM_H */