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         /* TODO: mi support */
  77         if (lttng_opt_mi) {
  78                 ret = -LTTNG_ERR_MI_NOT_IMPLEMENTED;
  79                 ERR("mi option not supported");
  80                 goto end;
  81         }
  82
  83         while ((opt = poptGetNextOpt(pc)) != -1) {
  84                 switch (opt) {
  85                 case OPT_HELP:
  86                         usage(stdout);
  87                         goto end;
  88                 case OPT_ALL:
  89                         opt_save_all = 1;
  90                         break;
  91                 case OPT_FORCE:
  92                         opt_force = 1;
  93                         break;
  94                 default:
  95                         usage(stderr);
  96                         ret = CMD_UNDEFINED;
  97                         goto end;
  98                 }
  99         }
 100
 101         if (!opt_save_all) {
 102                 session_name = poptGetArg(pc);
 103                 if (session_name) {
 104                         DBG2("Session name: %s", session_name);
 105                 }
 106         }
 107
 108         attr = lttng_save_session_attr_create();
 109         if (!attr) {
 110                 ret = CMD_FATAL;
 111                 goto end_destroy;
 112         }
 113
 114         if (lttng_save_session_attr_set_session_name(attr, session_name)) {
 115                 ret = CMD_ERROR;
 116                 goto end_destroy;
 117         }
 118
 119         if (lttng_save_session_attr_set_overwrite(attr, opt_force)) {
 120                 ret = CMD_ERROR;
 121                 goto end_destroy;
 122         }
 123
 124         if (lttng_save_session_attr_set_output_url(attr, opt_output_path)) {
 125                 ret = CMD_ERROR;
 126                 goto end_destroy;
 127         }
 128
 129         ret = lttng_save_session(attr);
 130         if (ret < 0) {
 131                 ERR("%s", lttng_strerror(ret));
 132         } else {
 133                 /* Inform the user of what just happened on success. */
 134                 if (session_name && opt_output_path) {
 135                         MSG("Session %s saved successfully in %s.", session_name,
 136                                         opt_output_path);
 137                 } else if (session_name && !opt_output_path) {
 138                         MSG("Session %s saved successfully.", session_name);
 139                 } else if (!session_name && opt_output_path) {
 140                         MSG("All sessions have been saved successfully in %s.",
 141                                         opt_output_path);
 142                 } else {
 143                         MSG("All sessions have been saved successfully.");
 144                 }
 145         }
 146 end_destroy:
 147         lttng_save_session_attr_destroy(attr);
 148 end:
 149         poptFreeContext(pc);
 150         return ret;
 151 }