src/bin/lttng/commands/save.c

   1 /*
   2  * Copyright (C) 2013 - Jérémie Galarneau <jeremie.galarneau@efficios.com>
   3  *
   4  * This program is free software; you can redistribute it and/or modify
   5  * it under the terms of the GNU General Public License, version 2 only,
   6  * as published by the Free Software Foundation.
   7  *
   8  * This program is distributed in the hope that it will be useful,
   9  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  10  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  11  * GNU General Public License for more details.
  12  *
  13  * You should have received a copy of the GNU General Public License along
  14  * with this program; if not, write to the Free Software Foundation, Inc.,
  15  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  16  */
  17
  18 #define _GNU_SOURCE
  19 #include <inttypes.h>
  20 #include <popt.h>
  21 #include <stdio.h>
  22 #include <stdlib.h>
  23 #include <string.h>
  24 #include <assert.h>
  25
  26 #include "../command.h"
  27 #include <lttng/save.h>
  28
  29 static char *opt_output_path;
  30 static int opt_force;
  31 static int opt_save_all;
  32
  33 enum {
  34         OPT_HELP = 1,
  35         OPT_ALL,
  36         OPT_FORCE,
  37 };
  38
  39 static struct poptOption save_opts[] = {
  40         /* longName, shortName, argInfo, argPtr, value, descrip, argDesc */
  41         {"help",        'h', POPT_ARG_NONE, 0, OPT_HELP, 0, 0},
  42         {"all",         'a', POPT_ARG_NONE, 0, OPT_ALL, 0, 0},
  43         {"output-path", 'o', POPT_ARG_STRING, &opt_output_path, 0, 0, 0},
  44         {"force",       'f', POPT_ARG_NONE, 0, OPT_FORCE, 0, 0},
  45         {0, 0, 0, 0, 0, 0, 0}
  46 };
  47
  48 /*
  49  * usage
  50  */
  51 static void usage(FILE *ofp)
  52 {
  53         fprintf(ofp, "usage: lttng save [OPTIONS] [SESSION]\n");
  54         fprintf(ofp, "\n");
  55         fprintf(ofp, "Options:\n");
  56         fprintf(ofp, "  -h, --help           Show this help\n");
  57         fprintf(ofp, "  -a, --all            Save all sessions (default)\n");
  58         fprintf(ofp, "  -o, --output-path    Output path of the session configuration(s)\n");
  59         fprintf(ofp, "  -f, --force          Overwrite existing session configuration(s)\n");
  60 }
  61
  62 /*
  63  * The 'save <options>' first level command
  64  */
  65 int cmd_save(int argc, const char **argv)
  66 {
  67         int ret = CMD_SUCCESS;
  68         int opt;
  69         const char *session_name = NULL;
  70         poptContext pc;
  71         struct lttng_save_session_attr *attr;
  72
  73         pc = poptGetContext(NULL, argc, argv, save_opts, 0);
  74         poptReadDefaultConfig(pc, 0);
  75
  76         while ((opt = poptGetNextOpt(pc)) != -1) {
  77                 switch (opt) {
  78                 case OPT_HELP:
  79                         usage(stdout);
  80                         goto end;
  81                 case OPT_ALL:
  82                         opt_save_all = 1;
  83                         break;
  84                 case OPT_FORCE:
  85                         opt_force = 1;
  86                         break;
  87                 default:
  88                         usage(stderr);
  89                         ret = CMD_UNDEFINED;
  90                         goto end;
  91                 }
  92         }
  93
  94         if (!opt_save_all) {
  95                 session_name = poptGetArg(pc);
  96                 if (session_name) {
  97                         DBG2("Session name: %s", session_name);
  98                 }
  99         }
 100
 101         attr = lttng_save_session_attr_create();
 102         if (!attr) {
 103                 ret = CMD_FATAL;
 104                 goto end_destroy;
 105         }
 106
 107         if (lttng_save_session_attr_set_session_name(attr, session_name)) {
 108                 ret = CMD_ERROR;
 109                 goto end_destroy;
 110         }
 111
 112         if (lttng_save_session_attr_set_overwrite(attr, opt_force)) {
 113                 ret = CMD_ERROR;
 114                 goto end_destroy;
 115         }
 116
 117         if (lttng_save_session_attr_set_output_url(attr, opt_output_path)) {
 118                 ret = CMD_ERROR;
 119                 goto end_destroy;
 120         }
 121
 122         ret = lttng_save_session(attr);
 123         if (ret < 0) {
 124                 ERR("%s", lttng_strerror(ret));
 125         } else {
 126                 /* Inform the user of what just happened on success. */
 127                 if (session_name && opt_output_path) {
 128                         MSG("Session %s saved successfully in %s.", session_name,
 129                                         opt_output_path);
 130                 } else if (session_name && !opt_output_path) {
 131                         MSG("Session %s saved successfully.", session_name);
 132                 } else if (!session_name && opt_output_path) {
 133                         MSG("All sessions have been saved successfully in %s.",
 134                                         opt_output_path);
 135                 } else {
 136                         MSG("All sessions have been saved successfully.");
 137                 }
 138         }
 139 end_destroy:
 140         lttng_save_session_attr_destroy(attr);
 141 end:
 142         poptFreeContext(pc);
 143         return ret;
 144 }