src/lib/lttng-ctl/save.c

   1 /*
   2  * Copyright (C) 2013 - Jérémie Galarneau <jeremie.galarneau@efficios.com>
   3  *
   4  * This library is free software; you can redistribute it and/or modify it
   5  * under the terms of the GNU Lesser General Public License, version 2.1 only,
   6  * as published by the Free Software Foundation.
   7  *
   8  * This library 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 Lesser General Public License
  11  * for more details.
  12  *
  13  * You should have received a copy of the GNU Lesser General Public License
  14  * along with this library; if not, write to the Free Software Foundation,
  15  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  16  */
  17
  18 #define _LGPL_SOURCE
  19 #include <assert.h>
  20 #include <string.h>
  21
  22 #include <lttng/lttng-error.h>
  23 #include <lttng/save.h>
  24 #include <lttng/save-internal.h>
  25 #include <common/sessiond-comm/sessiond-comm.h>
  26
  27 #include "lttng-ctl-helper.h"
  28
  29 struct lttng_save_session_attr *lttng_save_session_attr_create(void)
  30 {
  31         return zmalloc(sizeof(struct lttng_save_session_attr));
  32 }
  33
  34 void lttng_save_session_attr_destroy(struct lttng_save_session_attr *output)
  35 {
  36         if (output) {
  37                 free(output);
  38         }
  39 }
  40
  41 const char *lttng_save_session_attr_get_session_name(
  42         struct lttng_save_session_attr *attr)
  43 {
  44         const char *ret = NULL;
  45
  46         if (attr && attr->session_name[0]) {
  47                 ret = attr->session_name;
  48         }
  49
  50         return ret;
  51 }
  52
  53 const char *lttng_save_session_attr_get_output_url(
  54         struct lttng_save_session_attr *attr)
  55 {
  56         const char *ret = NULL;
  57
  58         if (attr && attr->configuration_url[0]) {
  59                 ret = attr->configuration_url;
  60         }
  61
  62         return ret;
  63 }
  64
  65 int lttng_save_session_attr_get_overwrite(
  66         struct lttng_save_session_attr *attr)
  67 {
  68         return attr ? attr->overwrite : -LTTNG_ERR_INVALID;
  69 }
  70
  71 int lttng_save_session_attr_set_session_name(
  72         struct lttng_save_session_attr *attr, const char *session_name)
  73 {
  74         int ret = 0;
  75
  76         if (!attr) {
  77                 ret = -LTTNG_ERR_INVALID;
  78                 goto error;
  79         }
  80
  81         if (session_name) {
  82                 size_t len;
  83
  84                 len = strlen(session_name);
  85                 if (len >= LTTNG_NAME_MAX) {
  86                         ret = -LTTNG_ERR_INVALID;
  87                         goto error;
  88                 }
  89
  90                 strncpy(attr->session_name, session_name, len);
  91         } else {
  92                 attr->session_name[0] = '\0';
  93         }
  94 error:
  95         return ret;
  96 }
  97
  98 int lttng_save_session_attr_set_output_url(
  99         struct lttng_save_session_attr *attr, const char *url)
 100 {
 101         int ret = 0;
 102         size_t len;
 103         ssize_t size;
 104         struct lttng_uri *uris = NULL;
 105
 106         if (!attr) {
 107                 ret = -LTTNG_ERR_INVALID;
 108                 goto error;
 109         }
 110
 111         if (!url) {
 112                 attr->configuration_url[0] = '\0';
 113                 ret = 0;
 114                 goto end;
 115         }
 116
 117         len = strlen(url);
 118         if (len >= PATH_MAX) {
 119                 ret = -LTTNG_ERR_INVALID;
 120                 goto error;
 121         }
 122
 123         size = uri_parse_str_urls(url, NULL, &uris);
 124         if (size <= 0 || uris[0].dtype != LTTNG_DST_PATH) {
 125                 ret = -LTTNG_ERR_INVALID;
 126                 goto error;
 127         }
 128
 129         /* Copy string plus the NULL terminated byte. */
 130         lttng_ctl_copy_string(attr->configuration_url, uris[0].dst.path,
 131                         sizeof(attr->configuration_url));
 132
 133 end:
 134 error:
 135         free(uris);
 136         return ret;
 137 }
 138
 139 int lttng_save_session_attr_set_overwrite(
 140         struct lttng_save_session_attr *attr, int overwrite)
 141 {
 142         int ret = 0;
 143
 144         if (!attr) {
 145                 ret = -LTTNG_ERR_INVALID;
 146                 goto end;
 147         }
 148
 149         attr->overwrite = !!overwrite;
 150 end:
 151         return ret;
 152 }
 153
 154 /*
 155  * The lttng-ctl API does not expose all the information needed to save the
 156  * session configurations. Thus, we must send a save command to the session
 157  * daemon which will, in turn, save its current session configuration.
 158  */
 159 int lttng_save_session(struct lttng_save_session_attr *attr)
 160 {
 161         struct lttcomm_session_msg lsm;
 162         int ret;
 163
 164         if (!attr) {
 165                 ret = -LTTNG_ERR_INVALID;
 166                 goto end;
 167         }
 168
 169         memset(&lsm, 0, sizeof(lsm));
 170         lsm.cmd_type = LTTNG_SAVE_SESSION;
 171
 172         memcpy(&lsm.u.save_session.attr, attr,
 173                 sizeof(struct lttng_save_session_attr));
 174         ret = lttng_ctl_ask_sessiond(&lsm, NULL);
 175 end:
 176         return ret;
 177 }