src/bin/lttng/commands/load.c

   1 /*
   2  * Copyright (C) 2014 - 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 <common/config/config.h>
  27 #include "../command.h"
  28
  29 static char *opt_input_path;
  30 static int opt_force;
  31 static int opt_load_all;
  32
  33 static const char *session_name;
  34
  35 enum {
  36         OPT_HELP = 1,
  37         OPT_ALL,
  38         OPT_FORCE,
  39 };
  40
  41 static struct poptOption load_opts[] = {
  42         /* longName, shortName, argInfo, argPtr, value, descrip, argDesc */
  43         {"help",        'h',  POPT_ARG_NONE, 0, OPT_HELP, 0, 0},
  44         {"all",         'a',  POPT_ARG_NONE, 0, OPT_ALL, 0, 0},
  45         {"input-path",  'i',  POPT_ARG_STRING, &opt_input_path, 0, 0, 0},
  46         {"force",       'f',  POPT_ARG_NONE, 0, OPT_FORCE, 0, 0},
  47         {0, 0, 0, 0, 0, 0, 0}
  48 };
  49
  50 /*
  51  * usage
  52  */
  53 static void usage(FILE *ofp)
  54 {
  55         fprintf(ofp, "usage: lttng load [OPTIONS] [SESSION]\n");
  56         fprintf(ofp, "\n");
  57         fprintf(ofp, "Options:\n");
  58         fprintf(ofp, "  -h, --help               Show this help\n");
  59         fprintf(ofp, "  -a, --all                Load all sessions (default)\n");
  60         fprintf(ofp, "  -i, --input-path PATH    Input path of the session file(s).\n");
  61         fprintf(ofp, "                           If a directory, load all files in it\n");
  62         fprintf(ofp, "                           else try to load the given file.\n");
  63         fprintf(ofp, "  -f, --force              Override existing session(s).\n");
  64         fprintf(ofp, "                           This will destroy existing session(s)\n");
  65         fprintf(ofp, "                           before creating new one(s).\n");
  66 }
  67
  68 /*
  69  * The 'load <options>' first level command
  70  */
  71 int cmd_load(int argc, const char **argv)
  72 {
  73         int ret = CMD_SUCCESS;
  74         int opt;
  75         poptContext pc;
  76
  77         pc = poptGetContext(NULL, argc, argv, load_opts, 0);
  78         poptReadDefaultConfig(pc, 0);
  79
  80         /* TODO: mi support */
  81         if (lttng_opt_mi) {
  82                 ret = -LTTNG_ERR_MI_NOT_IMPLEMENTED;
  83                 ERR("mi option not supported");
  84                 goto end;
  85         }
  86
  87         while ((opt = poptGetNextOpt(pc)) != -1) {
  88                 switch (opt) {
  89                 case OPT_HELP:
  90                         usage(stdout);
  91                         goto end;
  92                 case OPT_ALL:
  93                         opt_load_all = 1;
  94                         break;
  95                 case OPT_FORCE:
  96                         opt_force = 1;
  97                         break;
  98                 default:
  99                         usage(stderr);
 100                         ret = CMD_UNDEFINED;
 101                         goto end;
 102                 }
 103         }
 104
 105         if (!opt_load_all) {
 106                 session_name = poptGetArg(pc);
 107                 if (session_name) {
 108                         DBG2("Loading session name: %s", session_name);
 109                 }
 110         }
 111
 112         ret = config_load_session(opt_input_path, session_name, opt_force, 0);
 113         if (ret) {
 114                 ERR("%s", lttng_strerror(ret));
 115                 ret = -ret;
 116         } else {
 117                 if (opt_load_all) {
 118                         MSG("All sessions have been loaded successfully");
 119                 } else if (session_name) {
 120                         ret = config_init((char *)session_name);
 121                         if (ret < 0) {
 122                                 ret = CMD_WARNING;
 123                         }
 124                         MSG("Session %s has been loaded successfully", session_name);
 125                 } else {
 126                         MSG("Session has been loaded successfully");
 127                 }
 128         }
 129 end:
 130         poptFreeContext(pc);
 131         return ret;
 132 }