From bdae7d71473289a9e50c8d03188bd90cec8a5d57 Mon Sep 17 00:00:00 2001 From: Partha Pratim Mukherjee Date: Sun, 5 Jul 2015 15:31:15 -0400 Subject: [PATCH] Fix: destroy session removes the default config file MIME-Version: 1.0 Content-Type: text/plain; charset=utf8 Content-Transfer-Encoding: 8bit Destroy session command by default removes the default config file without checking the current session. As a result when we call any other command which expects a default session by calling get_session_name() function, it fails. This patch will fix this by checking that the default config file gets removed only when destroy session is called with the current session. Fixes: #887 Signed-off-by: Partha Pratim Mukherjee Signed-off-by: Jérémie Galarneau --- src/bin/lttng/commands/destroy.c | 10 ++++- src/bin/lttng/conf.c | 66 +++++++++++++++++++++++--------- src/bin/lttng/conf.h | 1 + src/bin/lttng/utils.c | 34 ++++++++++++---- src/bin/lttng/utils.h | 1 + 5 files changed, 84 insertions(+), 28 deletions(-) diff --git a/src/bin/lttng/commands/destroy.c b/src/bin/lttng/commands/destroy.c index 3fb5fb232..03f10dac9 100644 --- a/src/bin/lttng/commands/destroy.c +++ b/src/bin/lttng/commands/destroy.c @@ -27,6 +27,7 @@ #include "../command.h" #include +#include static char *opt_session_name; static int opt_destroy_all; @@ -70,6 +71,7 @@ static void usage(FILE *ofp) static int destroy_session(const char *session_name) { int ret; + char *default_session_name = NULL; ret = lttng_destroy_session(session_name); if (ret < 0) { @@ -85,9 +87,15 @@ static int destroy_session(const char *session_name) } MSG("Session %s destroyed", session_name); - config_destroy_default(); + + default_session_name = get_session_name_quiet(); + if (default_session_name && + !strncmp(session_name, session_name, NAME_MAX)) { + config_destroy_default(); + } ret = CMD_SUCCESS; error: + free(default_session_name); return ret; } diff --git a/src/bin/lttng/conf.c b/src/bin/lttng/conf.c index d3f031368..1e2e5950d 100644 --- a/src/bin/lttng/conf.c +++ b/src/bin/lttng/conf.c @@ -23,6 +23,7 @@ #include #include #include +#include #include #include @@ -176,14 +177,10 @@ int config_exists(const char *path) return S_ISREG(info.st_mode) || S_ISDIR(info.st_mode); } -/* - * Returns the session name from the config file. - * The caller is responsible for freeing the returned string. - * On error, NULL is returned. - */ -char *config_read_session_name(char *path) +static +int _config_read_session_name(char *path, char **name) { - int ret; + int ret = 0; FILE *fp; char var[NAME_MAX], *session_name; #if (NAME_MAX == 255) @@ -192,15 +189,14 @@ char *config_read_session_name(char *path) session_name = zmalloc(NAME_MAX); if (session_name == NULL) { + ret = -ENOMEM; ERR("Out of memory"); goto error; } fp = open_config(path, "r"); if (fp == NULL) { - ERR("Can't find valid lttng config %s/.lttngrc", path); - MSG("Did you create a session? (lttng create )"); - free(session_name); + ret = -ENOENT; goto error; } @@ -221,22 +217,54 @@ char *config_read_session_name(char *path) } error_close: - free(session_name); - ret = fclose(fp); - if (ret < 0) { + if (fclose(fp) < 0) { PERROR("close config read session name"); } - error: - return NULL; - + free(session_name); + return ret; found: - ret = fclose(fp); - if (ret < 0) { + *name = session_name; + if (fclose(fp) < 0) { PERROR("close config read session name found"); } - return session_name; + return ret; +} + +/* + * Returns the session name from the config file. + * + * The caller is responsible for freeing the returned string. + * On error, NULL is returned. + */ +char *config_read_session_name(char *path) +{ + int ret; + char *name = NULL; + + ret = _config_read_session_name(path, &name); + if (ret == -ENOENT) { + const char *home_dir = utils_get_home_dir(); + + ERR("Can't find valid lttng config %s/.lttngrc", home_dir); + MSG("Did you create a session? (lttng create )"); + } + + return name; +} + +/* + * Returns the session name from the config file. (no warnings/errors emitted) + * + * The caller is responsible for freeing the returned string. + * On error, NULL is returned. + */ +char *config_read_session_name_quiet(char *path) +{ + char *name = NULL; + (void) _config_read_session_name(path, &name); + return name; } /* diff --git a/src/bin/lttng/conf.h b/src/bin/lttng/conf.h index 3bed59c2a..a26a6b8e3 100644 --- a/src/bin/lttng/conf.h +++ b/src/bin/lttng/conf.h @@ -28,6 +28,7 @@ int config_add_session_name(char *path, char *name); /* Must free() the return pointer */ char *config_read_session_name(char *path); +char *config_read_session_name_quiet(char *path); char *config_get_file_path(char *path); #endif /* _LTTNG_CONFIG_H */ diff --git a/src/bin/lttng/utils.c b/src/bin/lttng/utils.c index 257dfc89a..be020befc 100644 --- a/src/bin/lttng/utils.c +++ b/src/bin/lttng/utils.c @@ -37,13 +37,8 @@ static const char *str_kernel = "Kernel"; static const char *str_ust = "UST"; static const char *str_jul = "JUL"; -/* - * get_session_name - * - * Return allocated string with the session name found in the config - * directory. - */ -char *get_session_name(void) +static +char *_get_session_name(int quiet) { char *path, *session_name = NULL; @@ -54,7 +49,8 @@ char *get_session_name(void) } /* Get session name from config */ - session_name = config_read_session_name(path); + session_name = quiet ? config_read_session_name_quiet(path) : + config_read_session_name(path); if (session_name == NULL) { goto error; } @@ -67,6 +63,28 @@ error: return NULL; } +/* + * get_session_name + * + * Return allocated string with the session name found in the config + * directory. + */ +char *get_session_name(void) +{ + return _get_session_name(0); +} + +/* + * get_session_name_quiet (no warnings/errors emitted) + * + * Return allocated string with the session name found in the config + * directory. + */ +char *get_session_name_quiet(void) +{ + return _get_session_name(1); +} + /* * list_commands * diff --git a/src/bin/lttng/utils.h b/src/bin/lttng/utils.h index d3941279a..ea92bb9bc 100644 --- a/src/bin/lttng/utils.h +++ b/src/bin/lttng/utils.h @@ -28,6 +28,7 @@ extern char *opt_relayd_path; struct cmd_struct; char *get_session_name(void); +char *get_session_name_quiet(void); void list_commands(struct cmd_struct *commands, FILE *ofp); void list_cmd_options(FILE *ofp, struct poptOption *options); -- 2.34.1