src/bin/lttng-relayd/utils.c

   1 /*
   2  * Copyright (C) 2013 - Julien Desfossez <jdesfossez@efficios.com>
   3  *                      David Goulet <dgoulet@efficios.com>
   4  *
   5  * This program is free software; you can redistribute it and/or modify it
   6  * under the terms of the GNU General Public License, version 2 only, as
   7  * published by the Free Software Foundation.
   8  *
   9  * This program is distributed in the hope that it will be useful, but WITHOUT
  10  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  11  * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
  12  * more details.
  13  *
  14  * You should have received a copy of the GNU General Public License along with
  15  * this program; if not, write to the Free Software Foundation, Inc., 51
  16  * Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  17  */
  18
  19 #define _LGPL_SOURCE
  20 #include <assert.h>
  21 #include <stdio.h>
  22 #include <stdlib.h>
  23 #include <string.h>
  24
  25 #include <common/common.h>
  26 #include <common/defaults.h>
  27 #include <common/utils.h>
  28
  29 #include "lttng-relayd.h"
  30 #include "utils.h"
  31
  32 static char *create_output_path_auto(const char *path_name)
  33 {
  34         int ret;
  35         char *traces_path = NULL;
  36         char *alloc_path = NULL;
  37         char *default_path;
  38
  39         default_path = utils_get_home_dir();
  40         if (default_path == NULL) {
  41                 ERR("Home path not found.\n \
  42                                 Please specify an output path using -o, --output PATH");
  43                 goto exit;
  44         }
  45         alloc_path = strdup(default_path);
  46         if (alloc_path == NULL) {
  47                 PERROR("Path allocation");
  48                 goto exit;
  49         }
  50         ret = asprintf(&traces_path, "%s/" DEFAULT_TRACE_DIR_NAME
  51                         "/%s", alloc_path, path_name);
  52         if (ret < 0) {
  53                 PERROR("asprintf trace dir name");
  54                 goto exit;
  55         }
  56 exit:
  57         free(alloc_path);
  58         return traces_path;
  59 }
  60
  61 static char *create_output_path_noauto(const char *path_name)
  62 {
  63         int ret;
  64         char *traces_path = NULL;
  65         char *full_path;
  66
  67         full_path = utils_expand_path(opt_output_path);
  68         if (!full_path) {
  69                 goto exit;
  70         }
  71
  72         ret = asprintf(&traces_path, "%s/%s", full_path, path_name);
  73         if (ret < 0) {
  74                 PERROR("asprintf trace dir name");
  75                 goto exit;
  76         }
  77 exit:
  78         free(full_path);
  79         return traces_path;
  80 }
  81
  82 /*
  83  * Create the output trace directory path name string.
  84  *
  85  * Return the allocated string containing the path name or else NULL.
  86  */
  87 char *create_output_path(const char *path_name)
  88 {
  89         assert(path_name);
  90
  91         if (opt_output_path == NULL) {
  92                 return create_output_path_auto(path_name);
  93         } else {
  94                 return create_output_path_noauto(path_name);
  95         }
  96 }