Major code cleanup

[lttng-tools.git] / ltt-sessiond / event.c

diff --git a/ltt-sessiond/event.c b/ltt-sessiond/event.c
new file mode 100644 (file)
index 0000000..3e3ecd3
--- /dev/null
+++ b/ltt-sessiond/event.c
@@ -0,0 +1,155 @@
+/*
+ * Copyright (C) 2011 - David Goulet <david.goulet@polymtl.ca>
+ *
+ * This program is free software; you can redistribute it and/or modify it
+ * under the terms of the GNU General Public License as published by the Free
+ * Software Foundation; only version 2 of the License.
+ *
+ * This program is distributed in the hope that it will be useful, but WITHOUT
+ * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
+ * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
+ * more details.
+ *
+ * You should have received a copy of the GNU General Public License along with
+ * this program; if not, write to the Free Software Foundation, Inc., 59 Temple
+ * Place - Suite 330, Boston, MA  02111-1307, USA.
+ */
+
+#include <urcu/list.h>
+
+#include <lttng/lttng.h>
+#include <lttng-sessiond-comm.h>
+#include <lttngerr.h>
+
+#include "channel.h"
+#include "event.h"
+#include "kernel-ctl.h"
+
+/*
+ * Disable kernel event for a channel from the kernel session.
+ */
+int event_kernel_disable(struct ltt_kernel_session *ksession,
+               struct ltt_kernel_channel *kchan, char *event_name)
+{
+       int ret;
+       struct ltt_kernel_event *kevent;
+
+       kevent = trace_kernel_get_event_by_name(event_name, kchan);
+       if (kevent == NULL) {
+               ret = LTTCOMM_NO_EVENT;
+               goto error;
+       }
+
+       ret = kernel_disable_event(kevent);
+       if (ret < 0) {
+               ret = LTTCOMM_KERN_DISABLE_FAIL;
+               goto error;
+       }
+
+       DBG("Kernel event %s disable for channel %s.",
+                       kevent->event->name, kchan->channel->name);
+
+       ret = LTTCOMM_OK;
+
+error:
+       return ret;
+}
+
+/*
+ * Disable all kernel event for a channel from the kernel session.
+ */
+int event_kernel_disable_all(struct ltt_kernel_session *ksession,
+               struct ltt_kernel_channel *kchan)
+{
+       int ret;
+       struct ltt_kernel_event *kevent;
+
+       /* For each event in the kernel session */
+       cds_list_for_each_entry(kevent, &kchan->events_list.head, list) {
+               ret = kernel_disable_event(kevent);
+               if (ret < 0) {
+                       /* We continue disabling the rest */
+                       continue;
+               }
+       }
+
+       ret = LTTCOMM_OK;
+
+       return ret;
+}
+
+/*
+ * Enable kernel event for a channel from the kernel session.
+ */
+int event_kernel_enable(struct ltt_kernel_session *ksession,
+               struct ltt_kernel_channel *kchan, struct lttng_event *event)
+{
+       int ret;
+       struct ltt_kernel_event *kevent;
+
+       kevent = trace_kernel_get_event_by_name(event->name, kchan);
+       if (kevent == NULL) {
+               ret = kernel_create_event(event, kchan);
+               if (ret < 0) {
+                       ret = LTTCOMM_KERN_ENABLE_FAIL;
+                       goto error;
+               }
+       } else if (kevent->enabled == 0) {
+               ret = kernel_enable_event(kevent);
+               if (ret < 0) {
+                       ret = LTTCOMM_KERN_ENABLE_FAIL;
+                       goto error;
+               }
+       }
+
+       ret = LTTCOMM_OK;
+
+error:
+       return ret;
+}
+
+/*
+ * Enable all kernel event of a channel of the kernel session.
+ */
+int event_kernel_enable_all(struct ltt_kernel_session *ksession,
+               struct ltt_kernel_channel *kchan, int kernel_tracer_fd)
+{
+       int size, i, ret;
+       struct ltt_kernel_event *kevent;
+       struct lttng_event *event_list;
+
+       /* For each event in the kernel session */
+       cds_list_for_each_entry(kevent, &kchan->events_list.head, list) {
+               ret = kernel_enable_event(kevent);
+               if (ret < 0) {
+                       /* Enable failed but still continue */
+                       continue;
+               }
+       }
+
+       size = kernel_list_events(kernel_tracer_fd, &event_list);
+       if (size < 0) {
+               ret = LTTCOMM_KERN_LIST_FAIL;
+               goto error;
+       }
+
+       for (i = 0; i < size; i++) {
+               kevent = trace_kernel_get_event_by_name(event_list[i].name, kchan);
+               if (kevent == NULL) {
+                       /* Default event type for enable all */
+                       event_list[i].type = LTTNG_EVENT_TRACEPOINT;
+                       /* Enable each single tracepoint event */
+                       ret = kernel_create_event(&event_list[i], kchan);
+                       if (ret < 0) {
+                               /* Ignore error here and continue */
+                       }
+               }
+       }
+
+       free(event_list);
+
+       ret = LTTCOMM_OK;
+
+error:
+       return ret;
+}