fdc0da6bf2ee921b6438fef2df6676ed88f5ad20
[lttng-tools.git] / 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 }
This page took 0.046065 seconds and 3 git commands to generate.