Add the load session API call

[lttng-tools.git] / src / lib / lttng-ctl / load.c

diff --git a/src/lib/lttng-ctl/load.c b/src/lib/lttng-ctl/load.c
new file mode 100644 (file)
index 0000000..3811981
--- /dev/null
+++ b/src/lib/lttng-ctl/load.c
@@ -0,0 +1,173 @@
+/*
+ * Copyright (C) 2014 - David Goulet <dgoulet@efficios.com>
+ *
+ * This library is free software; you can redistribute it and/or modify it
+ * under the terms of the GNU Lesser General Public License, version 2.1 only,
+ * as published by the Free Software Foundation.
+ *
+ * This library 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 Lesser General Public License
+ * for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public License
+ * along with this library; if not, write to the Free Software Foundation,
+ * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
+ */
+
+#define _GNU_SOURCE
+#include <assert.h>
+#include <string.h>
+
+#include <lttng/lttng-error.h>
+#include <lttng/load.h>
+#include <lttng/load-internal.h>
+#include <common/sessiond-comm/sessiond-comm.h>
+#include <common/config/config.h>
+
+#include "lttng-ctl-helper.h"
+
+struct lttng_load_session_attr *lttng_load_session_attr_create(void)
+{
+       return zmalloc(sizeof(struct lttng_load_session_attr));
+}
+
+void lttng_load_session_attr_destroy(struct lttng_load_session_attr *attr)
+{
+       if (attr) {
+               free(attr);
+       }
+}
+
+const char *lttng_load_session_attr_get_session_name(
+       struct lttng_load_session_attr *attr)
+{
+       const char *ret = NULL;
+
+       if (attr && attr->session_name[0]) {
+               ret = attr->session_name;
+       }
+
+       return ret;
+}
+
+const char *lttng_load_session_attr_get_input_url(
+       struct lttng_load_session_attr *attr)
+{
+       const char *ret = NULL;
+
+       if (attr && attr->input_url[0]) {
+               ret = attr->input_url;
+       }
+
+       return ret;
+}
+
+int lttng_load_session_attr_get_overwrite(
+       struct lttng_load_session_attr *attr)
+{
+       return attr ? attr->overwrite : -LTTNG_ERR_INVALID;
+}
+
+int lttng_load_session_attr_set_session_name(
+       struct lttng_load_session_attr *attr, const char *session_name)
+{
+       int ret = 0;
+
+       if (!attr) {
+               ret = -LTTNG_ERR_INVALID;
+               goto error;
+       }
+
+       if (session_name) {
+               size_t len;
+
+               len = strlen(session_name);
+               if (len >= NAME_MAX) {
+                       ret = -LTTNG_ERR_INVALID;
+                       goto error;
+               }
+
+               strncpy(attr->session_name, session_name, len);
+       } else {
+               attr->session_name[0] = '\0';
+       }
+error:
+       return ret;
+}
+
+int lttng_load_session_attr_set_input_url(
+       struct lttng_load_session_attr *attr, const char *url)
+{
+       int ret = 0;
+       size_t len, size;
+       struct lttng_uri *uris = NULL;
+
+       if (!attr) {
+               ret = -LTTNG_ERR_INVALID;
+               goto error;
+       }
+
+       if (!url) {
+               attr->input_url[0] = '\0';
+               ret = 0;
+               goto end;
+       }
+
+       len = strlen(url);
+       if (len >= PATH_MAX) {
+               ret = -LTTNG_ERR_INVALID;
+               goto error;
+       }
+
+       size = uri_parse_str_urls(url, NULL, &uris);
+       if (size <= 0 || uris[0].dtype != LTTNG_DST_PATH) {
+               ret = -LTTNG_ERR_INVALID;
+               goto error;
+       }
+
+       /* Copy string plus the NULL terminated byte. */
+       lttng_ctl_copy_string(attr->input_url, uris[0].dst.path,
+                       sizeof(attr->input_url));
+
+end:
+error:
+       free(uris);
+       return ret;
+}
+
+int lttng_load_session_attr_set_overwrite(
+       struct lttng_load_session_attr *attr, int overwrite)
+{
+       int ret = 0;
+
+       if (!attr) {
+               ret = -LTTNG_ERR_INVALID;
+               goto end;
+       }
+
+       attr->overwrite = !!overwrite;
+end:
+       return ret;
+}
+
+/*
+ * The lttng-ctl API does not expose all the information needed to load the
+ * session configurations. Thus, we must send a load command to the session
+ * daemon which will, in turn, load its current session configuration.
+ */
+int lttng_load_session(struct lttng_load_session_attr *attr)
+{
+       int ret;
+
+       if (!attr) {
+               ret = -LTTNG_ERR_INVALID;
+               goto end;
+       }
+
+       ret = config_load_session(attr->input_url, attr->session_name,
+                       attr->overwrite);
+
+end:
+       return ret;
+}