Add a save API to lttng-ctl

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

diff --git a/src/lib/lttng-ctl/save.c b/src/lib/lttng-ctl/save.c
new file mode 100644 (file)
index 0000000..f215468
--- /dev/null
+++ b/src/lib/lttng-ctl/save.c
@@ -0,0 +1,163 @@
+/*
+ * Copyright (C) 2013 - Jérémie Galarneau <jeremie.galarneau@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/save.h>
+#include <lttng/save-internal.h>
+#include <common/sessiond-comm/sessiond-comm.h>
+
+#include "lttng-ctl-helper.h"
+
+struct lttng_save_session_attr *lttng_save_session_attr_create(void)
+{
+       return zmalloc(sizeof(struct lttng_save_session_attr));
+}
+
+void lttng_save_session_attr_destroy(struct lttng_save_session_attr *output)
+{
+       if (output) {
+               free(output);
+       }
+}
+
+const char *lttng_save_session_attr_get_session_name(
+       struct lttng_save_session_attr *attr)
+{
+       const char *ret = NULL;
+
+       if (attr && attr->session_name[0]) {
+               ret = attr->session_name;
+       }
+
+       return ret;
+}
+
+const char *lttng_save_session_attr_get_output_url(
+       struct lttng_save_session_attr *attr)
+{
+       const char *ret = NULL;
+
+       if (attr && attr->configuration_url[0]) {
+               ret = attr->configuration_url;
+       }
+
+       return ret;
+}
+
+int lttng_save_session_attr_get_overwrite(
+       struct lttng_save_session_attr *attr)
+{
+       return attr ? attr->overwrite : -LTTNG_ERR_INVALID;
+}
+
+int lttng_save_session_attr_set_session_name(
+       struct lttng_save_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_save_session_attr_set_output_url(
+       struct lttng_save_session_attr *attr, const char *url)
+{
+       int ret = 0;
+
+       if (!attr) {
+               ret = -LTTNG_ERR_INVALID;
+               goto error;
+       }
+
+       if (url) {
+               size_t len;
+
+               len = strlen(url);
+               if (len >= PATH_MAX) {
+                       ret = -LTTNG_ERR_INVALID;
+                       goto error;
+               }
+
+               strncpy(attr->configuration_url, url, len);
+       } else {
+               attr->configuration_url[0] = '\0';
+       }
+error:
+       return ret;
+}
+
+int lttng_save_session_attr_set_overwrite(
+       struct lttng_save_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 save the
+ * session configurations. Thus, we must send a save command to the session
+ * daemon which will, in turn, save its current session configuration.
+ */
+int lttng_save_session(struct lttng_save_session_attr *attr)
+{
+       struct lttcomm_session_msg lsm;
+       int ret;
+
+       if (!attr) {
+               ret = -LTTNG_ERR_INVALID;
+               goto end;
+       }
+
+       memset(&lsm, 0, sizeof(lsm));
+       lsm.cmd_type = LTTNG_SAVE_SESSION;
+
+       memcpy(&lsm.u.save_session.attr, attr,
+               sizeof(struct lttng_save_session_attr));
+       ret = lttng_ctl_ask_sessiond(&lsm, NULL);
+end:
+       return ret;
+}