src/bin/lttng-sessiond/sessiond-config.c

   1 /*
   2  * Copyright (C) 2017 - 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 #include "sessiond-config.h"
  19 #include <assert.h>
  20 #include "lttng-ust-ctl.h"
  21 #include <common/defaults.h>
  22 #include <limits.h>
  23 #include <errno.h>
  24 #include <ctype.h>
  25 #include <common/error.h>
  26 #include <common/utils.h>
  27 #include <common/compat/getenv.h>
  28
  29 static
  30 struct sessiond_config sessiond_config_build_defaults = {
  31         .quiet =                                false,
  32         .verbose =                              0,
  33         .verbose_consumer =                     0,
  34
  35         .agent_tcp_port =                       { .begin = DEFAULT_AGENT_TCP_PORT_RANGE_BEGIN, .end = DEFAULT_AGENT_TCP_PORT_RANGE_END },
  36         .app_socket_timeout =                   DEFAULT_APP_SOCKET_RW_TIMEOUT,
  37
  38         .no_kernel =                            false,
  39         .background =                           false,
  40         .daemonize =                            false,
  41         .sig_parent =                           false,
  42
  43         .tracing_group_name.value =             DEFAULT_TRACING_GROUP,
  44         .kmod_probes_list.value =               NULL,
  45         .kmod_extra_probes_list.value =         NULL,
  46
  47         .rundir.value =                         NULL,
  48
  49         .apps_unix_sock_path.value =            NULL,
  50         .client_unix_sock_path.value =          NULL,
  51         .wait_shm_path.value =                  NULL,
  52         .health_unix_sock_path.value =          NULL,
  53         .lttng_ust_clock_plugin.value =         NULL,
  54         .pid_file_path.value =                  NULL,
  55         .lock_file_path.value =                 NULL,
  56         .agent_port_file_path.value =           NULL,
  57         .load_session_path.value =              NULL,
  58
  59         .consumerd32_path.value =               NULL,
  60         .consumerd32_bin_path.value =           NULL,
  61         .consumerd32_lib_dir.value =            NULL,
  62         .consumerd32_err_unix_sock_path.value = NULL,
  63         .consumerd32_cmd_unix_sock_path.value = NULL,
  64
  65         .consumerd64_path.value =               NULL,
  66         .consumerd64_bin_path.value =           NULL,
  67         .consumerd64_lib_dir.value =            NULL,
  68         .consumerd64_err_unix_sock_path.value = NULL,
  69         .consumerd64_cmd_unix_sock_path.value = NULL,
  70
  71         .kconsumerd_path.value =                NULL,
  72         .kconsumerd_err_unix_sock_path.value =  NULL,
  73         .kconsumerd_cmd_unix_sock_path.value =  NULL,
  74 };
  75
  76 static
  77 void config_string_fini(struct config_string *str)
  78 {
  79         config_string_set(str, NULL);
  80 }
  81
  82 static
  83 void config_string_set_static(struct config_string *config_str,
  84                 const char *value)
  85 {
  86         config_string_set(config_str, (char *) value);
  87         config_str->should_free = false;
  88 }
  89
  90 /* Only use for dynamically-allocated strings. */
  91 LTTNG_HIDDEN
  92 void config_string_set(struct config_string *config_str, char *value)
  93 {
  94         assert(config_str);
  95         if (config_str->should_free) {
  96                 free(config_str->value);
  97                 config_str->should_free = false;
  98         }
  99
 100         config_str->should_free = !!value;
 101         config_str->value = value;
 102 }
 103
 104 LTTNG_HIDDEN
 105 int sessiond_config_apply_env_config(struct sessiond_config *config)
 106 {
 107         int ret = 0;
 108         const char *env_value;
 109
 110         env_value = getenv(DEFAULT_APP_SOCKET_TIMEOUT_ENV);
 111         if (env_value) {
 112                 char *endptr;
 113                 long int_val;
 114
 115                 errno = 0;
 116                 int_val = strtoul(env_value, &endptr, 0);
 117                 if (errno != 0 || int_val > INT_MAX ||
 118                                 (int_val < 0 && int_val != -1)) {
 119                         ERR("Invalid value \"%s\" used for \"%s\" environment variable",
 120                                         env_value, DEFAULT_APP_SOCKET_TIMEOUT_ENV);
 121                         ret = -1;
 122                         goto end;
 123                 }
 124
 125                 config->app_socket_timeout = int_val;
 126         }
 127
 128         env_value = lttng_secure_getenv("LTTNG_CONSUMERD32_BIN");
 129         if (env_value) {
 130                 config_string_set_static(&config->consumerd32_bin_path,
 131                                 env_value);
 132         }
 133         env_value = lttng_secure_getenv("LTTNG_CONSUMERD64_BIN");
 134         if (env_value) {
 135                 config_string_set_static(&config->consumerd64_bin_path,
 136                                 env_value);
 137         }
 138
 139         env_value = lttng_secure_getenv("LTTNG_CONSUMERD32_LIBDIR");
 140         if (env_value) {
 141                 config_string_set_static(&config->consumerd32_lib_dir,
 142                                 env_value);
 143         }
 144         env_value = lttng_secure_getenv("LTTNG_CONSUMERD64_LIBDIR");
 145         if (env_value) {
 146                 config_string_set_static(&config->consumerd64_lib_dir,
 147                                 env_value);
 148         }
 149
 150         env_value = lttng_secure_getenv("LTTNG_UST_CLOCK_PLUGIN");
 151         if (env_value) {
 152                 config_string_set_static(&config->lttng_ust_clock_plugin,
 153                                 env_value);
 154         }
 155
 156         env_value = lttng_secure_getenv(DEFAULT_LTTNG_KMOD_PROBES);
 157         if (env_value) {
 158                 config_string_set_static(&config->kmod_probes_list,
 159                                 env_value);
 160         }
 161
 162         env_value = lttng_secure_getenv(DEFAULT_LTTNG_EXTRA_KMOD_PROBES);
 163         if (env_value) {
 164                 config_string_set_static(&config->kmod_extra_probes_list,
 165                                 env_value);
 166         }
 167 end:
 168         return ret;
 169 }
 170
 171 static
 172 int config_set_paths_root(struct sessiond_config *config)
 173 {
 174         int ret = 0;
 175
 176         config_string_set(&config->rundir, strdup(DEFAULT_LTTNG_RUNDIR));
 177         if (!config->rundir.value) {
 178                 ERR("Failed to set rundir");
 179                 ret = -1;
 180                 goto end;
 181         }
 182
 183         config_string_set_static(&config->apps_unix_sock_path,
 184                         DEFAULT_GLOBAL_APPS_UNIX_SOCK);
 185         config_string_set_static(&config->client_unix_sock_path,
 186                         DEFAULT_GLOBAL_CLIENT_UNIX_SOCK);
 187         config_string_set_static(&config->wait_shm_path,
 188                         DEFAULT_GLOBAL_APPS_WAIT_SHM_PATH);
 189         config_string_set_static(&config->health_unix_sock_path,
 190                         DEFAULT_GLOBAL_HEALTH_UNIX_SOCK);
 191 end:
 192         return ret;
 193 }
 194
 195 static
 196 int config_set_paths_non_root(struct sessiond_config *config)
 197 {
 198         int ret = 0;
 199         const char *home_path = utils_get_home_dir();
 200         char *str;
 201
 202         if (home_path == NULL) {
 203                 ERR("Can't get HOME directory for sockets creation.");
 204                 ret = -1;
 205                 goto end;
 206         }
 207
 208         /*
 209          * Create rundir from home path. This will create something like
 210          * $HOME/.lttng
 211          */
 212         ret = asprintf(&str, DEFAULT_LTTNG_HOME_RUNDIR, home_path);
 213         if (ret < 0) {
 214                 ERR("Failed to set rundir");
 215                 goto end;
 216         }
 217         config_string_set(&config->rundir, str);
 218         str = NULL;
 219
 220         ret = asprintf(&str, DEFAULT_HOME_APPS_UNIX_SOCK, home_path);
 221         if (ret < 0) {
 222                 ERR("Failed to set default home apps unix socket path");
 223                 goto end;
 224         }
 225         config_string_set(&config->apps_unix_sock_path, str);
 226         str = NULL;
 227
 228         ret = asprintf(&str, DEFAULT_HOME_CLIENT_UNIX_SOCK, home_path);
 229         if (ret < 0) {
 230                 ERR("Failed to set default home client unix socket path");
 231                 goto end;
 232         }
 233         config_string_set(&config->client_unix_sock_path, str);
 234         str = NULL;
 235
 236         ret = asprintf(&str, DEFAULT_HOME_APPS_WAIT_SHM_PATH, getuid());
 237         if (ret < 0) {
 238                 ERR("Failed to set default home apps wait shm path");
 239                 goto end;
 240         }
 241         config_string_set(&config->wait_shm_path, str);
 242         str = NULL;
 243
 244         ret = asprintf(&str, DEFAULT_HOME_HEALTH_UNIX_SOCK, home_path);
 245         if (ret < 0) {
 246                 ERR("Failed to set default home health UNIX socket path");
 247                 goto end;
 248         }
 249         config_string_set(&config->health_unix_sock_path, str);
 250         str = NULL;
 251
 252         ret = 0;
 253 end:
 254         return ret;
 255 }
 256
 257 LTTNG_HIDDEN
 258 int sessiond_config_init(struct sessiond_config *config)
 259 {
 260         int ret;
 261         bool is_root = (getuid() == 0);
 262         char *str;
 263
 264         assert(config);
 265         memcpy(config, &sessiond_config_build_defaults, sizeof(*config));
 266
 267         if (is_root) {
 268                 ret = config_set_paths_root(config);
 269         } else {
 270                 ret = config_set_paths_non_root(config);
 271         }
 272
 273         /* 32 bits consumerd path setup */
 274         ret = asprintf(&str, DEFAULT_USTCONSUMERD32_PATH,
 275                         config->rundir.value);
 276         if (ret < 0) {
 277                 ERR("Failed to set 32-bit consumer path");
 278                 goto end;
 279         }
 280         config_string_set(&config->consumerd32_path, str);
 281         str = NULL;
 282
 283         ret = asprintf(&str, DEFAULT_USTCONSUMERD32_ERR_SOCK_PATH,
 284                         config->rundir.value);
 285         if (ret < 0) {
 286                 ERR("Failed to set 32-bit consumer error socket path");
 287                 goto end;
 288         }
 289         config_string_set(&config->consumerd32_err_unix_sock_path, str);
 290         str = NULL;
 291
 292         ret = asprintf(&str, DEFAULT_USTCONSUMERD32_CMD_SOCK_PATH,
 293                         config->rundir.value);
 294         if (ret < 0) {
 295                 ERR("Failed to set 32-bit consumer command socket path");
 296                 goto end;
 297         }
 298         config_string_set(&config->consumerd32_cmd_unix_sock_path, str);
 299         str = NULL;
 300
 301         /* 64 bits consumerd path setup */
 302         ret = asprintf(&str, DEFAULT_USTCONSUMERD64_PATH,
 303                         config->rundir.value);
 304         if (ret < 0) {
 305                 ERR("Failed to set 64-bit consumer path");
 306                 goto end;
 307         }
 308         config_string_set(&config->consumerd64_path, str);
 309         str = NULL;
 310
 311         ret = asprintf(&str, DEFAULT_USTCONSUMERD64_ERR_SOCK_PATH,
 312                         config->rundir.value);
 313         if (ret < 0) {
 314                 ERR("Failed to set 64-bit consumer error socket path");
 315                 goto end;
 316         }
 317         config_string_set(&config->consumerd64_err_unix_sock_path, str);
 318         str = NULL;
 319
 320         ret = asprintf(&str, DEFAULT_USTCONSUMERD64_CMD_SOCK_PATH,
 321                         config->rundir.value);
 322         if (ret < 0) {
 323                 ERR("Failed to set 64-bit consumer command socket path");
 324                 goto end;
 325         }
 326         config_string_set(&config->consumerd64_cmd_unix_sock_path, str);
 327         str = NULL;
 328
 329         /* kconsumerd consumerd path setup */
 330         ret = asprintf(&str, DEFAULT_KCONSUMERD_PATH,
 331                         config->rundir.value);
 332         if (ret < 0) {
 333                 ERR("Failed to set kernel consumer path");
 334                 goto end;
 335         }
 336         config_string_set(&config->kconsumerd_path, str);
 337         str = NULL;
 338
 339         ret = asprintf(&str, DEFAULT_KCONSUMERD_ERR_SOCK_PATH,
 340                         config->rundir.value);
 341         if (ret < 0) {
 342                 ERR("Failed to set kernel consumer error socket path");
 343                 goto end;
 344         }
 345         config_string_set(&config->kconsumerd_err_unix_sock_path, str);
 346         str = NULL;
 347
 348         ret = asprintf(&str, DEFAULT_KCONSUMERD_CMD_SOCK_PATH,
 349                         config->rundir.value);
 350         if (ret < 0) {
 351                 ERR("Failed to set kernel consumer command socket path");
 352                 goto end;
 353         }
 354         config_string_set(&config->kconsumerd_cmd_unix_sock_path, str);
 355         str = NULL;
 356
 357         ret = asprintf(&str, "%s/%s", config->rundir.value,
 358                         DEFAULT_LTTNG_SESSIOND_PIDFILE);
 359         if (ret < 0) {
 360                 ERR("Failed to set PID file path");
 361                 goto end;
 362         }
 363         config_string_set(&config->pid_file_path, str);
 364         str = NULL;
 365
 366         ret = asprintf(&str, "%s/%s", config->rundir.value,
 367                         DEFAULT_LTTNG_SESSIOND_LOCKFILE);
 368         if (ret < 0) {
 369                 ERR("Failed to set lock file path");
 370                 goto end;
 371         }
 372         config_string_set(&config->lock_file_path, str);
 373         str = NULL;
 374
 375         ret = asprintf(&str, "%s/%s", config->rundir.value,
 376                         DEFAULT_LTTNG_SESSIOND_AGENTPORT_FILE);
 377         if (ret < 0) {
 378                 ERR("Failed to set agent port file path");
 379                 goto end;
 380         }
 381         config_string_set(&config->agent_port_file_path, str);
 382         str = NULL;
 383
 384         /*
 385          * Allow INSTALL_BIN_PATH to be used as a target path for the
 386          * native architecture size consumer if CONFIG_CONSUMER*_PATH
 387          * has not been defined.
 388          */
 389 #if (CAA_BITS_PER_LONG == 32)
 390         config_string_set_static(&config->consumerd32_bin_path,
 391                         INSTALL_BIN_PATH "/" DEFAULT_CONSUMERD_FILE);
 392         config_string_set_static(&config->consumerd32_lib_dir,
 393                         INSTALL_LIB_PATH);
 394 #elif (CAA_BITS_PER_LONG == 64)
 395         config_string_set_static(&config->consumerd64_bin_path,
 396                         INSTALL_BIN_PATH "/" DEFAULT_CONSUMERD_FILE);
 397         config_string_set_static(&config->consumerd64_lib_dir,
 398                         INSTALL_LIB_PATH);
 399 #else
 400 #error "Unknown bitness"
 401 #endif
 402         ret = 0;
 403 end:
 404         return ret;
 405 }
 406
 407 LTTNG_HIDDEN
 408 void sessiond_config_fini(struct sessiond_config *config)
 409 {
 410         config_string_fini(&config->tracing_group_name);
 411         config_string_fini(&config->kmod_probes_list);
 412         config_string_fini(&config->kmod_extra_probes_list);
 413         config_string_fini(&config->apps_unix_sock_path);
 414         config_string_fini(&config->client_unix_sock_path);
 415         config_string_fini(&config->wait_shm_path);
 416         config_string_fini(&config->health_unix_sock_path);
 417         config_string_fini(&config->lttng_ust_clock_plugin);
 418         config_string_fini(&config->pid_file_path);
 419         config_string_fini(&config->lock_file_path);
 420         config_string_fini(&config->load_session_path);
 421         config_string_fini(&config->agent_port_file_path);
 422         config_string_fini(&config->consumerd32_path);
 423         config_string_fini(&config->consumerd32_bin_path);
 424         config_string_fini(&config->consumerd32_lib_dir);
 425         config_string_fini(&config->consumerd32_err_unix_sock_path);
 426         config_string_fini(&config->consumerd32_cmd_unix_sock_path);
 427         config_string_fini(&config->consumerd64_path);
 428         config_string_fini(&config->consumerd64_bin_path);
 429         config_string_fini(&config->consumerd64_lib_dir);
 430         config_string_fini(&config->consumerd64_err_unix_sock_path);
 431         config_string_fini(&config->consumerd64_cmd_unix_sock_path);
 432         config_string_fini(&config->kconsumerd_path);
 433         config_string_fini(&config->kconsumerd_err_unix_sock_path);
 434         config_string_fini(&config->kconsumerd_cmd_unix_sock_path);
 435 }
 436
 437 static
 438 int resolve_path(struct config_string *path)
 439 {
 440         int ret = 0;
 441         char *absolute_path;
 442
 443         if (!path->value || path->value[0] == '/') {
 444                 goto end;
 445         }
 446
 447         absolute_path = utils_expand_path(path->value);
 448         if (!absolute_path) {
 449                 ret = -1;
 450                 goto end;
 451         }
 452
 453         config_string_set(path, absolute_path);
 454 end:
 455         return ret;
 456 }
 457
 458 #define RESOLVE_CHECK(path_config_str)          \
 459         if (resolve_path(path_config_str))      \
 460                 return -1
 461
 462 LTTNG_HIDDEN
 463 int sessiond_config_resolve_paths(struct sessiond_config *config)
 464 {
 465         RESOLVE_CHECK(&config->apps_unix_sock_path);
 466         RESOLVE_CHECK(&config->client_unix_sock_path);
 467         RESOLVE_CHECK(&config->wait_shm_path);
 468         RESOLVE_CHECK(&config->health_unix_sock_path);
 469         RESOLVE_CHECK(&config->lttng_ust_clock_plugin);
 470         RESOLVE_CHECK(&config->pid_file_path);
 471         RESOLVE_CHECK(&config->lock_file_path);
 472         RESOLVE_CHECK(&config->load_session_path);
 473         RESOLVE_CHECK(&config->agent_port_file_path);
 474         RESOLVE_CHECK(&config->consumerd32_path);
 475         RESOLVE_CHECK(&config->consumerd32_bin_path);
 476         RESOLVE_CHECK(&config->consumerd32_lib_dir);
 477         RESOLVE_CHECK(&config->consumerd32_err_unix_sock_path);
 478         RESOLVE_CHECK(&config->consumerd32_cmd_unix_sock_path);
 479         RESOLVE_CHECK(&config->consumerd64_path);
 480         RESOLVE_CHECK(&config->consumerd64_bin_path);
 481         RESOLVE_CHECK(&config->consumerd64_lib_dir);
 482         RESOLVE_CHECK(&config->consumerd64_err_unix_sock_path);
 483         RESOLVE_CHECK(&config->consumerd64_cmd_unix_sock_path);
 484         RESOLVE_CHECK(&config->kconsumerd_path);
 485         RESOLVE_CHECK(&config->kconsumerd_err_unix_sock_path);
 486         RESOLVE_CHECK(&config->kconsumerd_cmd_unix_sock_path);
 487         return 0;
 488 }
 489
 490 LTTNG_HIDDEN
 491 void sessiond_config_log(struct sessiond_config *config)
 492 {
 493         DBG_NO_LOC("[sessiond configuration]");
 494         DBG_NO_LOC("\tverbose:                       %i", config->verbose);
 495         DBG_NO_LOC("\tverbose consumer:              %i", config->verbose_consumer);
 496         DBG_NO_LOC("\tquiet mode:                    %s", config->quiet ? "True" : "False");
 497         if (config->agent_tcp_port.begin == config->agent_tcp_port.end) {
 498                 DBG_NO_LOC("\tagent_tcp_port:                %i", config->agent_tcp_port.begin);
 499         } else {
 500                 DBG_NO_LOC("\tagent_tcp_port:                [%i, %i]",
 501                                 config->agent_tcp_port.begin,
 502                                 config->agent_tcp_port.end);
 503         }
 504         DBG_NO_LOC("\tapplication socket timeout:    %i", config->app_socket_timeout);
 505         DBG_NO_LOC("\tno-kernel:                     %s", config->no_kernel ? "True" : "False");
 506         DBG_NO_LOC("\tbackground:                    %s", config->background ? "True" : "False");
 507         DBG_NO_LOC("\tdaemonize:                     %s", config->daemonize ? "True" : "False");
 508         DBG_NO_LOC("\tsignal parent on start:        %s", config->sig_parent ? "True" : "False");
 509         DBG_NO_LOC("\ttracing group name:            %s", config->tracing_group_name.value ? : "Unknown");
 510         DBG_NO_LOC("\tkmod_probe_list:               %s", config->kmod_probes_list.value ? : "None");
 511         DBG_NO_LOC("\tkmod_extra_probe_list:         %s", config->kmod_extra_probes_list.value ? : "None");
 512         DBG_NO_LOC("\trundir:                        %s", config->rundir.value ? : "Unknown");
 513         DBG_NO_LOC("\tapplication socket path:       %s", config->apps_unix_sock_path.value ? : "Unknown");
 514         DBG_NO_LOC("\tclient socket path:            %s", config->client_unix_sock_path.value ? : "Unknown");
 515         DBG_NO_LOC("\twait shm path:                 %s", config->wait_shm_path.value ? : "Unknown");
 516         DBG_NO_LOC("\thealth socket path:            %s", config->health_unix_sock_path.value ? : "Unknown");
 517         DBG_NO_LOC("\tLTTNG_UST_CLOCK_PLUGIN:        %s", config->lttng_ust_clock_plugin.value ? : "None");
 518         DBG_NO_LOC("\tpid file path:                 %s", config->pid_file_path.value ? : "Unknown");
 519         DBG_NO_LOC("\tlock file path:                %s", config->lock_file_path.value ? : "Unknown");
 520         DBG_NO_LOC("\tsession load path:             %s", config->load_session_path.value ? : "None");
 521         DBG_NO_LOC("\tagent port file path:          %s", config->agent_port_file_path.value ? : "Unknown");
 522         DBG_NO_LOC("\tconsumerd32 path:              %s", config->consumerd32_path.value ? : "Unknown");
 523         DBG_NO_LOC("\tconsumerd32 bin path:          %s", config->consumerd32_bin_path.value ? : "Unknown");
 524         DBG_NO_LOC("\tconsumerd32 lib dir:           %s", config->consumerd32_lib_dir.value ? : "Unknown");
 525         DBG_NO_LOC("\tconsumerd32 err unix sock path:%s", config->consumerd32_err_unix_sock_path.value ? : "Unknown");
 526         DBG_NO_LOC("\tconsumerd32 cmd unix sock path:%s", config->consumerd32_cmd_unix_sock_path.value ? : "Unknown");
 527         DBG_NO_LOC("\tconsumerd64 path:              %s", config->consumerd64_path.value ? : "Unknown");
 528         DBG_NO_LOC("\tconsumerd64 bin path:          %s", config->consumerd64_bin_path.value ? : "Unknown");
 529         DBG_NO_LOC("\tconsumerd64 lib dir:           %s", config->consumerd64_lib_dir.value ? : "Unknown");
 530         DBG_NO_LOC("\tconsumerd64 err unix sock path:%s", config->consumerd64_err_unix_sock_path.value ? : "Unknown");
 531         DBG_NO_LOC("\tconsumerd64 cmd unix sock path:%s", config->consumerd64_cmd_unix_sock_path.value ? : "Unknown");
 532         DBG_NO_LOC("\tkconsumerd path:               %s", config->kconsumerd_path.value ? : "Unknown");
 533         DBG_NO_LOC("\tkconsumerd err unix sock path: %s", config->kconsumerd_err_unix_sock_path.value ? : "Unknown");
 534         DBG_NO_LOC("\tkconsumerd cmd unix sock path: %s", config->kconsumerd_cmd_unix_sock_path.value ? : "Unknown");
 535 }