lttng-sessiond/channel.c

   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 it
   5  * under the terms of the GNU General Public License as published by the Free
   6  * Software Foundation; only version 2 of the License.
   7  *
   8  * This program is distributed in the hope that it will be useful, but WITHOUT
   9  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  10  * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
  11  * more details.
  12  *
  13  * You should have received a copy of the GNU General Public License along with
  14  * this program; if not, write to the Free Software Foundation, Inc., 59 Temple
  15  * Place - Suite 330, Boston, MA  02111-1307, USA.
  16  */
  17
  18 #include <string.h>
  19 #include <unistd.h>
  20
  21 #include <lttng/lttng.h>
  22 #include <lttng-sessiond-comm.h>
  23 #include <lttngerr.h>
  24
  25 #include "channel.h"
  26 #include "hashtable.h"
  27 #include "kernel-ctl.h"
  28 #include "ust-ctl.h"
  29 #include "utils.h"
  30
  31 /*
  32  * Return allocated channel attributes.
  33  */
  34 struct lttng_channel *channel_new_default_attr(int dom)
  35 {
  36         struct lttng_channel *chan;
  37
  38         chan = zmalloc(sizeof(struct lttng_channel));
  39         if (chan == NULL) {
  40                 perror("zmalloc channel init");
  41                 goto error_alloc;
  42         }
  43
  44         if (snprintf(chan->name, sizeof(chan->name), "%s",
  45                                 DEFAULT_CHANNEL_NAME) < 0) {
  46                 perror("snprintf default channel name");
  47                 goto error;
  48         }
  49
  50         chan->attr.overwrite = DEFAULT_CHANNEL_OVERWRITE;
  51         chan->attr.switch_timer_interval = DEFAULT_CHANNEL_SWITCH_TIMER;
  52         chan->attr.read_timer_interval = DEFAULT_CHANNEL_READ_TIMER;
  53
  54         switch (dom) {
  55         case LTTNG_DOMAIN_KERNEL:
  56                 chan->attr.subbuf_size = DEFAULT_KERNEL_CHANNEL_SUBBUF_SIZE;
  57                 chan->attr.num_subbuf = DEFAULT_KERNEL_CHANNEL_SUBBUF_NUM;
  58                 chan->attr.output = DEFAULT_KERNEL_CHANNEL_OUTPUT;
  59                 break;
  60         case LTTNG_DOMAIN_UST:
  61         case LTTNG_DOMAIN_UST_PID:
  62                 chan->attr.subbuf_size = DEFAULT_UST_CHANNEL_SUBBUF_SIZE;
  63                 chan->attr.num_subbuf = DEFAULT_UST_CHANNEL_SUBBUF_NUM;
  64                 chan->attr.output = DEFAULT_UST_CHANNEL_OUTPUT;
  65                 break;
  66         default:
  67                 goto error;     /* Not implemented */
  68         }
  69
  70         return chan;
  71
  72 error:
  73         free(chan);
  74 error_alloc:
  75         return NULL;
  76 }
  77
  78 /*
  79  * Copy two ltt ust channel. Dst and src must be already allocated.
  80  */
  81 int channel_ust_copy(struct ltt_ust_channel *dst,
  82                 struct ltt_ust_channel *src)
  83 {
  84         //struct ltt_ust_event *uevent, *new_uevent;
  85
  86         memcpy(dst, src, sizeof(struct ltt_ust_channel));
  87         dst->events = hashtable_new_str(0);
  88
  89         /*
  90         cds_list_for_each_entry(uevent, &src->events.head, list) {
  91                 new_uevent = zmalloc(sizeof(struct ltt_ust_event));
  92                 if (new_uevent == NULL) {
  93                         perror("zmalloc ltt_ust_event");
  94                         goto error;
  95                 }
  96
  97                 memcpy(new_uevent, uevent, sizeof(struct ltt_ust_event));
  98                 cds_list_add(&new_uevent->list, &dst->events.head);
  99                 dst->events.count++;
 100         }
 101         */
 102
 103         return 0;
 104
 105 //error:
 106 //      return -1;
 107 }
 108
 109 /*
 110  * Disable kernel channel of the kernel session.
 111  */
 112 int channel_kernel_disable(struct ltt_kernel_session *ksession,
 113                 char *channel_name)
 114 {
 115         int ret;
 116         struct ltt_kernel_channel *kchan;
 117
 118         kchan = trace_kernel_get_channel_by_name(channel_name, ksession);
 119         if (kchan == NULL) {
 120                 ret = LTTCOMM_KERN_CHAN_NOT_FOUND;
 121                 goto error;
 122         } else if (kchan->enabled == 1) {
 123                 ret = kernel_disable_channel(kchan);
 124                 if (ret < 0) {
 125                         if (ret != EEXIST) {
 126                                 ret = LTTCOMM_KERN_CHAN_DISABLE_FAIL;
 127                         }
 128                         goto error;
 129                 }
 130         }
 131
 132         ret = LTTCOMM_OK;
 133
 134 error:
 135         return ret;
 136 }
 137
 138 /*
 139  * Enable kernel channel of the kernel session.
 140  */
 141 int channel_kernel_enable(struct ltt_kernel_session *ksession,
 142                 struct ltt_kernel_channel *kchan)
 143 {
 144         int ret;
 145
 146         if (kchan->enabled == 0) {
 147                 ret = kernel_enable_channel(kchan);
 148                 if (ret < 0) {
 149                         ret = LTTCOMM_KERN_CHAN_ENABLE_FAIL;
 150                         goto error;
 151                 }
 152         }
 153
 154         ret = LTTCOMM_OK;
 155
 156 error:
 157         return ret;
 158 }
 159
 160 /*
 161  * Create kernel channel of the kernel session and notify kernel thread.
 162  */
 163 int channel_kernel_create(struct ltt_kernel_session *ksession,
 164                 struct lttng_channel *attr, int kernel_pipe)
 165 {
 166         int ret;
 167         struct lttng_channel *defattr = NULL;
 168
 169         /* Creating channel attributes if needed */
 170         if (attr == NULL) {
 171                 defattr = channel_new_default_attr(LTTNG_DOMAIN_KERNEL);
 172                 if (defattr == NULL) {
 173                         ret = LTTCOMM_FATAL;
 174                         goto error;
 175                 }
 176                 attr = defattr;
 177         }
 178
 179         /* Channel not found, creating it */
 180         ret = kernel_create_channel(ksession, attr, ksession->trace_path);
 181         if (ret < 0) {
 182                 ret = LTTCOMM_KERN_CHAN_FAIL;
 183                 goto error;
 184         }
 185
 186         /* Notify kernel thread that there is a new channel */
 187         ret = notify_thread_pipe(kernel_pipe);
 188         if (ret < 0) {
 189                 ret = LTTCOMM_FATAL;
 190                 goto error;
 191         }
 192
 193         ret = LTTCOMM_OK;
 194 error:
 195         free(defattr);
 196         return ret;
 197 }
 198
 199 /*
 200  * Create UST channel and enable it on the tracer.
 201  */
 202 int channel_ust_create(struct ltt_ust_session *usess,
 203                 struct lttng_channel *attr)
 204 {
 205         int ret;
 206         struct ltt_ust_channel *uchan;
 207         //struct lttng_ust_channel_attr uattr;
 208         //struct object_data *obj;
 209
 210         uchan = trace_ust_find_channel_by_name(usess->domain_global.channels,
 211                         attr->name);
 212         if (uchan == NULL) {
 213                 uchan = trace_ust_create_channel(attr, usess->pathname);
 214                 if (uchan == NULL) {
 215                         ret = LTTCOMM_UST_CHAN_FAIL;
 216                         goto error;
 217                 }
 218                 rcu_read_lock();
 219                 hashtable_add_unique(usess->domain_global.channels, &uchan->node);
 220                 rcu_read_unlock();
 221         } else {
 222                 ret = LTTCOMM_UST_CHAN_EXIST;
 223                 goto error;
 224         }
 225
 226         /* TODO: NOTIFY ust application to update */
 227         /*
 228         ret = ustctl_create_channel(sock, usession->handle, &uattr, &obj);
 229         if (ret < 0) {
 230                 ret = LTTCOMM_UST_CHAN_FAIL;
 231                 goto error;
 232         }
 233         */
 234
 235         /*
 236         uchan->attr.overwrite = uattr.overwrite;
 237         uchan->attr.subbuf_size = uattr.subbuf_size;
 238         uchan->attr.num_subbuf = uattr.num_subbuf;
 239         uchan->attr.switch_timer_interval = uattr.switch_timer_interval;
 240         uchan->attr.read_timer_interval = uattr.read_timer_interval;
 241         uchan->attr.output = uattr.output;
 242         uchan->handle = obj->handle;
 243         uchan->attr.shm_fd = obj->shm_fd;
 244         uchan->attr.wait_fd = obj->wait_fd;
 245         uchan->attr.memory_map_size = obj->memory_map_size;
 246         uchan->obj = obj;
 247         */
 248
 249         /* Add channel to session */
 250         //rcu_read_lock();
 251         //cds_list_add(&uchan->list, &usession->channels.head);
 252         //usession->channels.count++;
 253         //rcu_read_unlock();
 254
 255         //DBG2("Channel %s UST create successfully for sock:%d", uchan->name, sock);
 256
 257         ret = LTTCOMM_OK;
 258
 259 error:
 260         return ret;
 261 }
 262
 263 /*
 264  * Enable UST channel on the tracer.
 265  */
 266 int channel_ust_enable(struct ltt_ust_session *usession,
 267                 struct ltt_ust_channel *uchan, int sock)
 268 {
 269         int ret = LTTCOMM_OK;
 270 #ifdef DISABLE
 271         struct object_data obj;
 272
 273         obj.shm_fd = uchan->attr.shm_fd;
 274         obj.wait_fd = uchan->attr.wait_fd;
 275         obj.memory_map_size = uchan->attr.memory_map_size;
 276         ret = ustctl_enable(sock, &obj);
 277         if (ret < 0) {
 278                 ret = LTTCOMM_UST_CHAN_FAIL;
 279                 goto end;
 280         }
 281         ret = LTTCOMM_OK;
 282 end:
 283 #endif
 284         return ret;
 285 }
 286
 287 /*
 288  * Disable UST channel on the tracer.
 289  */
 290 int channel_ust_disable(struct ltt_ust_session *usession,
 291                 struct ltt_ust_channel *uchan, int sock)
 292 {
 293         int ret = LTTCOMM_OK;
 294 #ifdef DISABLE
 295         struct object_data obj;
 296
 297         obj.shm_fd = uchan->attr.shm_fd;
 298         obj.wait_fd = uchan->attr.wait_fd;
 299         obj.memory_map_size = uchan->attr.memory_map_size;
 300         ret = ustctl_disable(sock, &obj);
 301         if (ret < 0) {
 302                 ret = LTTCOMM_UST_CHAN_FAIL;
 303                 goto end;
 304         }
 305         ret = LTTCOMM_OK;
 306 end:
 307 #endif
 308         return ret;
 309 }