2 * Copyright (c) 2011 David Goulet <david.goulet@polymtl.ca>
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.
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.
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.
24 #include <sys/types.h>
28 #include <common/common.h>
29 #include <common/utils.h>
34 * Returns the path with '/CONFIG_FILENAME' added to it;
35 * path will be NULL if an error occurs.
37 char *config_get_file_path(const char *path
)
42 ret
= asprintf(&file_path
, "%s/%s", path
, CONFIG_FILENAME
);
44 ERR("Fail allocating config file path");
52 * Returns an open FILE pointer to the config file;
53 * on error, NULL is returned.
55 static FILE *open_config(const char *path
, const char *mode
)
60 file_path
= config_get_file_path(path
);
61 if (file_path
== NULL
) {
65 fp
= fopen(file_path
, mode
);
76 * Creates the empty config file at the path.
77 * On success, returns 0;
78 * on error, returns -1.
80 static int create_config_file(const char *path
)
85 fp
= open_config(path
, "w+");
87 ERR("Unable to create config file");
99 * Append data to the config file in file_path
100 * On success, returns 0;
101 * on error, returns -1.
103 static int write_config(const char *file_path
, size_t size
, char *data
)
109 fp
= open_config(file_path
, "a");
115 /* Write session name into config file */
116 len
= fwrite(data
, size
, 1, fp
);
121 PERROR("close write_config");
128 * Destroys directory config and file config.
130 void config_destroy(const char *path
)
135 config_path
= config_get_file_path(path
);
136 if (config_path
== NULL
) {
140 if (!config_exists(config_path
)) {
144 DBG("Removing %s\n", config_path
);
145 ret
= remove(config_path
);
147 PERROR("remove config file");
154 * Destroys the default config
156 void config_destroy_default(void)
158 const char *path
= utils_get_home_dir();
162 config_destroy(path
);
166 * Returns 1 if config exists, 0 otherwise
168 int config_exists(const char *path
)
173 ret
= stat(path
, &info
);
177 return S_ISREG(info
.st_mode
) || S_ISDIR(info
.st_mode
);
181 int _config_read_session_name(const char *path
, char **name
)
185 char var
[NAME_MAX
], *session_name
;
187 #if (NAME_MAX == 255)
188 #define NAME_MAX_SCANF_IS_A_BROKEN_API "254"
191 session_name
= zmalloc(NAME_MAX
);
192 if (session_name
== NULL
) {
194 ERR("Out of memory");
198 fp
= open_config(path
, "r");
205 if ((ret
= fscanf(fp
, "%" NAME_MAX_SCANF_IS_A_BROKEN_API
206 "[^'=']=%" NAME_MAX_SCANF_IS_A_BROKEN_API
"s\n",
207 var
, session_name
)) != 2) {
209 ERR("Missing session=NAME in config file.");
215 if (strcmp(var
, "session") == 0) {
221 if (fclose(fp
) < 0) {
222 PERROR("close config read session name");
228 *name
= session_name
;
229 if (fclose(fp
) < 0) {
230 PERROR("close config read session name found");
236 * Returns the session name from the config file.
238 * The caller is responsible for freeing the returned string.
239 * On error, NULL is returned.
241 char *config_read_session_name(const char *path
)
246 ret
= _config_read_session_name(path
, &name
);
247 if (ret
== -ENOENT
) {
248 const char *home_dir
= utils_get_home_dir();
250 ERR("Can't find valid lttng config %s/.lttngrc", home_dir
);
251 MSG("Did you create a session? (lttng create <my_session>)");
258 * Returns the session name from the config file. (no warnings/errors emitted)
260 * The caller is responsible for freeing the returned string.
261 * On error, NULL is returned.
263 char *config_read_session_name_quiet(const char *path
)
267 (void) _config_read_session_name(path
, &name
);
272 * Write session name option to the config file.
273 * On success, returns 0;
274 * on error, returns -1.
276 int config_add_session_name(const char *path
, const char *name
)
279 char *attr
= "session=";
280 /* Max name len accepted plus attribute's len and the NULL byte. */
281 char session_name
[NAME_MAX
+ strlen(attr
) + 1];
284 * With GNU C < 2.1, snprintf returns -1 if the target buffer is too small;
285 * With GNU C >= 2.1, snprintf returns the required size (excluding closing null)
287 ret
= snprintf(session_name
, sizeof(session_name
), "%s%s\n", attr
, name
);
292 ret
= write_config(path
, ret
, session_name
);
298 * Init configuration directory and file.
299 * On success, returns 0;
300 * on error, returns -1.
302 int config_init(const char *session_name
)
307 path
= utils_get_home_dir();
313 /* Create default config file */
314 ret
= create_config_file(path
);
319 ret
= config_add_session_name(path
, session_name
);
324 DBG("Init config session in %s", path
);