/*- * BSD LICENSE * * Copyright (c) Intel Corporation. * All rights reserved. * * Redistribution and use in source and binary forms, with or without * modification, are permitted provided that the following conditions * are met: * * * Redistributions of source code must retain the above copyright * notice, this list of conditions and the following disclaimer. * * Redistributions in binary form must reproduce the above copyright * notice, this list of conditions and the following disclaimer in * the documentation and/or other materials provided with the * distribution. * * Neither the name of Intel Corporation nor the names of its * contributors may be used to endorse or promote products derived * from this software without specific prior written permission. * * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. */ #include "spdk/stdinc.h" #include "spdk_internal/event.h" #include "spdk/env.h" #include "spdk/log.h" #include "spdk/conf.h" #include "spdk/thread.h" #include "spdk/trace.h" #include "spdk/string.h" #include "spdk/rpc.h" #include "spdk/util.h" #define SPDK_APP_DEFAULT_LOG_LEVEL SPDK_LOG_NOTICE #define SPDK_APP_DEFAULT_LOG_PRINT_LEVEL SPDK_LOG_INFO #define SPDK_APP_DEFAULT_BACKTRACE_LOG_LEVEL SPDK_LOG_ERROR #define SPDK_APP_DPDK_DEFAULT_MEM_SIZE -1 #define SPDK_APP_DPDK_DEFAULT_MASTER_CORE -1 #define SPDK_APP_DPDK_DEFAULT_MEM_CHANNEL -1 #define SPDK_APP_DPDK_DEFAULT_CORE_MASK "0x1" struct spdk_app { struct spdk_conf *config; int shm_id; spdk_app_shutdown_cb shutdown_cb; int rc; }; static struct spdk_app g_spdk_app; static struct spdk_event *g_app_start_event = NULL; static struct spdk_event *g_shutdown_event = NULL; static int g_init_lcore; static bool g_delay_subsystem_init = false; static bool g_shutdown_sig_received = false; static char *g_executable_name; static struct spdk_app_opts g_default_opts; int spdk_app_get_shm_id(void) { return g_spdk_app.shm_id; } /* append one empty option to indicate the end of the array */ static const struct option g_cmdline_options[] = { #define CONFIG_FILE_OPT_IDX 'c' {"config", required_argument, NULL, CONFIG_FILE_OPT_IDX}, #define LIMIT_COREDUMP_OPT_IDX 'd' {"limit-coredump", no_argument, NULL, LIMIT_COREDUMP_OPT_IDX}, #define TPOINT_GROUP_MASK_OPT_IDX 'e' {"tpoint-group-mask", required_argument, NULL, TPOINT_GROUP_MASK_OPT_IDX}, #define SINGLE_FILE_SEGMENTS_OPT_IDX 'g' {"single-file-segments", no_argument, NULL, SINGLE_FILE_SEGMENTS_OPT_IDX}, #define HELP_OPT_IDX 'h' {"help", no_argument, NULL, HELP_OPT_IDX}, #define SHM_ID_OPT_IDX 'i' {"shm-id", required_argument, NULL, SHM_ID_OPT_IDX}, #define CPUMASK_OPT_IDX 'm' {"cpumask", required_argument, NULL, CPUMASK_OPT_IDX}, #define MEM_CHANNELS_OPT_IDX 'n' {"mem-channels", required_argument, NULL, MEM_CHANNELS_OPT_IDX}, #define MASTER_CORE_OPT_IDX 'p' {"master-core", required_argument, NULL, MASTER_CORE_OPT_IDX}, #define RPC_SOCKET_OPT_IDX 'r' {"rpc-socket", required_argument, NULL, RPC_SOCKET_OPT_IDX}, #define MEM_SIZE_OPT_IDX 's' {"mem-size", required_argument, NULL, MEM_SIZE_OPT_IDX}, #define NO_PCI_OPT_IDX 'u' {"no-pci", no_argument, NULL, NO_PCI_OPT_IDX}, #define PCI_BLACKLIST_OPT_IDX 'B' {"pci-blacklist", required_argument, NULL, PCI_BLACKLIST_OPT_IDX}, #define TRACEFLAG_OPT_IDX 'L' {"traceflag", required_argument, NULL, TRACEFLAG_OPT_IDX}, #define HUGE_UNLINK_OPT_IDX 'R' {"huge-unlink", no_argument, NULL, HUGE_UNLINK_OPT_IDX}, #define PCI_WHITELIST_OPT_IDX 'W' {"pci-whitelist", required_argument, NULL, PCI_WHITELIST_OPT_IDX}, #define SILENCE_NOTICELOG_OPT_IDX 257 {"silence-noticelog", no_argument, NULL, SILENCE_NOTICELOG_OPT_IDX}, #define WAIT_FOR_RPC_OPT_IDX 258 {"wait-for-rpc", no_argument, NULL, WAIT_FOR_RPC_OPT_IDX}, }; /* Global section */ #define GLOBAL_CONFIG_TMPL \ "# Configuration file\n" \ "#\n" \ "# Please write all parameters using ASCII.\n" \ "# The parameter must be quoted if it includes whitespace.\n" \ "#\n" \ "# Configuration syntax:\n" \ "# Spaces at head of line are deleted, other spaces are as separator\n" \ "# Lines starting with '#' are comments and not evaluated.\n" \ "# Lines ending with '\\' are concatenated with the next line.\n" \ "# Bracketed keys are section keys grouping the following value keys.\n" \ "# Number of section key is used as a tag number.\n" \ "# Ex. [TargetNode1] = TargetNode section key with tag number 1\n" \ "[Global]\n" \ " Comment \"Global section\"\n" \ "\n" \ " # Users can restrict work items to only run on certain cores by\n" \ " # specifying a ReactorMask. Default is to allow work items to run\n" \ " # on all cores. Core 0 must be set in the mask if one is specified.\n" \ " # Default: 0xFFFF (cores 0-15)\n" \ " ReactorMask \"0x%s\"\n" \ "\n" \ " # Tracepoint group mask for spdk trace buffers\n" \ " # Default: 0x0 (all tracepoint groups disabled)\n" \ " # Set to 0xFFFF to enable all tracepoint groups.\n" \ " TpointGroupMask \"0x%" PRIX64 "\"\n" \ "\n" \ static void spdk_app_config_dump_global_section(FILE *fp) { struct spdk_cpuset *coremask; if (NULL == fp) { return; } coremask = spdk_app_get_core_mask(); fprintf(fp, GLOBAL_CONFIG_TMPL, spdk_cpuset_fmt(coremask), spdk_trace_get_tpoint_group_mask()); } int spdk_app_get_running_config(char **config_str, char *name) { FILE *fp = NULL; int fd = -1; long length = 0, ret = 0; char vbuf[BUFSIZ]; char config_template[64]; snprintf(config_template, sizeof(config_template), "/tmp/%s.XXXXXX", name); /* Create temporary file to hold config */ fd = mkstemp(config_template); if (fd == -1) { SPDK_ERRLOG("mkstemp failed\n"); return -1; } fp = fdopen(fd, "wb+"); if (NULL == fp) { SPDK_ERRLOG("error opening tmpfile fd = %d\n", fd); return -1; } /* Buffered IO */ setvbuf(fp, vbuf, _IOFBF, BUFSIZ); spdk_app_config_dump_global_section(fp); spdk_subsystem_config(fp); length = ftell(fp); *config_str = malloc(length + 1); if (!*config_str) { SPDK_ERRLOG("out-of-memory for config\n"); fclose(fp); return -1; } fseek(fp, 0, SEEK_SET); ret = fread(*config_str, sizeof(char), length, fp); if (ret < length) { SPDK_ERRLOG("short read\n"); } fclose(fp); (*config_str)[length] = '\0'; return 0; } void spdk_app_start_shutdown(void) { if (g_shutdown_event != NULL) { spdk_event_call(g_shutdown_event); g_shutdown_event = NULL; } else { spdk_app_stop(0); } } static void __shutdown_signal(int signo) { if (!g_shutdown_sig_received) { g_shutdown_sig_received = true; spdk_app_start_shutdown(); } } static void __shutdown_event_cb(void *arg1, void *arg2) { g_spdk_app.shutdown_cb(); } static int spdk_app_opts_validate(const char *app_opts) { int i = 0, j; for (i = 0; app_opts[i] != '\0'; i++) { /* ignore getopt control characters */ if (app_opts[i] == ':' || app_opts[i] == '+' || app_opts[i] == '-') { continue; } for (j = 0; SPDK_APP_GETOPT_STRING[j] != '\0'; j++) { if (app_opts[i] == SPDK_APP_GETOPT_STRING[j]) { return app_opts[i]; } } } return 0; } void spdk_app_opts_init(struct spdk_app_opts *opts) { if (!opts) { return; } memset(opts, 0, sizeof(*opts)); opts->enable_coredump = true; opts->shm_id = -1; opts->mem_size = SPDK_APP_DPDK_DEFAULT_MEM_SIZE; opts->master_core = SPDK_APP_DPDK_DEFAULT_MASTER_CORE; opts->mem_channel = SPDK_APP_DPDK_DEFAULT_MEM_CHANNEL; opts->reactor_mask = NULL; opts->max_delay_us = 0; opts->print_level = SPDK_APP_DEFAULT_LOG_PRINT_LEVEL; opts->rpc_addr = SPDK_DEFAULT_RPC_ADDR; opts->delay_subsystem_init = false; } static int spdk_app_setup_signal_handlers(struct spdk_app_opts *opts) { struct sigaction sigact; sigset_t sigmask; int rc; /* Set up custom shutdown handling if the user requested it. */ if (opts->shutdown_cb != NULL) { g_shutdown_event = spdk_event_allocate(spdk_env_get_current_core(), __shutdown_event_cb, NULL, NULL); } sigemptyset(&sigmask); memset(&sigact, 0, sizeof(sigact)); sigemptyset(&sigact.sa_mask); sigact.sa_handler = SIG_IGN; rc = sigaction(SIGPIPE, &sigact, NULL); if (rc < 0) { SPDK_ERRLOG("sigaction(SIGPIPE) failed\n"); return rc; } /* Install the same handler for SIGINT and SIGTERM */ sigact.sa_handler = __shutdown_signal; rc = sigaction(SIGINT, &sigact, NULL); if (rc < 0) { SPDK_ERRLOG("sigaction(SIGINT) failed\n"); return rc; } sigaddset(&sigmask, SIGINT); rc = sigaction(SIGTERM, &sigact, NULL); if (rc < 0) { SPDK_ERRLOG("sigaction(SIGTERM) failed\n"); return rc; } sigaddset(&sigmask, SIGTERM); if (opts->usr1_handler != NULL) { sigact.sa_handler = opts->usr1_handler; rc = sigaction(SIGUSR1, &sigact, NULL); if (rc < 0) { SPDK_ERRLOG("sigaction(SIGUSR1) failed\n"); return rc; } sigaddset(&sigmask, SIGUSR1); } pthread_sigmask(SIG_UNBLOCK, &sigmask, NULL); return 0; } static void spdk_app_start_application(void) { spdk_rpc_set_state(SPDK_RPC_RUNTIME); spdk_event_call(g_app_start_event); } static void spdk_app_start_rpc(void *arg1, void *arg2) { const char *rpc_addr = arg1; spdk_rpc_initialize(rpc_addr); if (!g_delay_subsystem_init) { spdk_app_start_application(); } } static struct spdk_conf * spdk_app_setup_conf(const char *config_file) { struct spdk_conf *config; int rc; config = spdk_conf_allocate(); assert(config != NULL); if (config_file) { rc = spdk_conf_read(config, config_file); if (rc != 0) { SPDK_ERRLOG("Could not read config file %s\n", config_file); goto error; } if (spdk_conf_first_section(config) == NULL) { SPDK_ERRLOG("Invalid config file %s\n", config_file); goto error; } } spdk_conf_set_as_default(config); return config; error: spdk_conf_free(config); return NULL; } static int spdk_app_opts_add_pci_addr(struct spdk_app_opts *opts, struct spdk_pci_addr **list, char *bdf) { struct spdk_pci_addr *tmp = *list; size_t i = opts->num_pci_addr; tmp = realloc(tmp, sizeof(*tmp) * (i + 1)); if (tmp == NULL) { SPDK_ERRLOG("realloc error\n"); return -ENOMEM; } *list = tmp; if (spdk_pci_addr_parse(*list + i, bdf) < 0) { SPDK_ERRLOG("Invalid address %s\n", bdf); return -EINVAL; } opts->num_pci_addr++; return 0; } static int spdk_app_read_config_file_global_params(struct spdk_app_opts *opts) { struct spdk_conf_section *sp; char *bdf; int i, rc = 0; sp = spdk_conf_find_section(NULL, "Global"); if (opts->shm_id == -1) { if (sp != NULL) { opts->shm_id = spdk_conf_section_get_intval(sp, "SharedMemoryID"); } } if (opts->reactor_mask == NULL) { if (sp && spdk_conf_section_get_val(sp, "ReactorMask")) { SPDK_ERRLOG("ReactorMask config option is deprecated. Use -m/--cpumask\n" "command line parameter instead.\n"); opts->reactor_mask = spdk_conf_section_get_val(sp, "ReactorMask"); } else { opts->reactor_mask = SPDK_APP_DPDK_DEFAULT_CORE_MASK; } } if (!opts->no_pci && sp) { opts->no_pci = spdk_conf_section_get_boolval(sp, "NoPci", false); } if (opts->tpoint_group_mask == NULL) { if (sp != NULL) { opts->tpoint_group_mask = spdk_conf_section_get_val(sp, "TpointGroupMask"); } } if (sp == NULL) { return 0; } for (i = 0; ; i++) { bdf = spdk_conf_section_get_nmval(sp, "PciBlacklist", i, 0); if (!bdf) { break; } rc = spdk_app_opts_add_pci_addr(opts, &opts->pci_blacklist, bdf); if (rc != 0) { free(opts->pci_blacklist); return rc; } } for (i = 0; ; i++) { bdf = spdk_conf_section_get_nmval(sp, "PciWhitelist", i, 0); if (!bdf) { break; } if (opts->pci_blacklist != NULL) { SPDK_ERRLOG("PciBlacklist and PciWhitelist cannot be used at the same time\n"); free(opts->pci_blacklist); return -EINVAL; } rc = spdk_app_opts_add_pci_addr(opts, &opts->pci_whitelist, bdf); if (rc != 0) { free(opts->pci_whitelist); return rc; } } return 0; } static int spdk_app_setup_env(struct spdk_app_opts *opts) { struct spdk_env_opts env_opts = {}; int rc; spdk_env_opts_init(&env_opts); env_opts.name = opts->name; env_opts.core_mask = opts->reactor_mask; env_opts.shm_id = opts->shm_id; env_opts.mem_channel = opts->mem_channel; env_opts.master_core = opts->master_core; env_opts.mem_size = opts->mem_size; env_opts.hugepage_single_segments = opts->hugepage_single_segments; env_opts.unlink_hugepage = opts->unlink_hugepage; env_opts.no_pci = opts->no_pci; env_opts.num_pci_addr = opts->num_pci_addr; env_opts.pci_blacklist = opts->pci_blacklist; env_opts.pci_whitelist = opts->pci_whitelist; rc = spdk_env_init(&env_opts); free(env_opts.pci_blacklist); free(env_opts.pci_whitelist); if (rc < 0) { fprintf(stderr, "Unable to initialize SPDK env\n"); } return rc; } static int spdk_app_setup_trace(struct spdk_app_opts *opts) { char shm_name[64]; uint64_t tpoint_group_mask; char *end; if (opts->shm_id >= 0) { snprintf(shm_name, sizeof(shm_name), "/%s_trace.%d", opts->name, opts->shm_id); } else { snprintf(shm_name, sizeof(shm_name), "/%s_trace.pid%d", opts->name, (int)getpid()); } if (spdk_trace_init(shm_name) != 0) { return -1; } if (opts->tpoint_group_mask != NULL) { errno = 0; tpoint_group_mask = strtoull(opts->tpoint_group_mask, &end, 16); if (*end != '\0' || errno) { SPDK_ERRLOG("invalid tpoint mask %s\n", opts->tpoint_group_mask); } else { SPDK_NOTICELOG("Tracepoint Group Mask %s specified.\n", opts->tpoint_group_mask); SPDK_NOTICELOG("Use 'spdk_trace -s %s %s %d' to capture a snapshot of events at runtime.\n", opts->name, opts->shm_id >= 0 ? "-i" : "-p", opts->shm_id >= 0 ? opts->shm_id : getpid()); #if defined(__linux__) SPDK_NOTICELOG("Or copy /dev/shm%s for offline analysis/debug.\n", shm_name); #endif spdk_trace_set_tpoint_group_mask(tpoint_group_mask); } } return 0; } int spdk_app_start(struct spdk_app_opts *opts, spdk_event_fn start_fn, void *arg1, void *arg2) { struct spdk_conf *config = NULL; int rc; struct spdk_event *rpc_start_event; char *tty; if (!opts) { SPDK_ERRLOG("opts should not be NULL\n"); return 1; } if (!start_fn) { SPDK_ERRLOG("start_fn should not be NULL\n"); return 1; } tty = ttyname(STDERR_FILENO); if (opts->print_level > SPDK_LOG_WARN && isatty(STDERR_FILENO) && tty && !strncmp(tty, "/dev/tty", strlen("/dev/tty"))) { printf("Warning: printing stderr to console terminal without -q option specified.\n"); printf("Suggest using --silence-noticelog to disable logging to stderr and\n"); printf("monitor syslog, or redirect stderr to a file.\n"); printf("(Delaying for 10 seconds...)\n"); sleep(10); } spdk_log_set_print_level(opts->print_level); #ifndef SPDK_NO_RLIMIT if (opts->enable_coredump) { struct rlimit core_limits; core_limits.rlim_cur = core_limits.rlim_max = RLIM_INFINITY; setrlimit(RLIMIT_CORE, &core_limits); } #endif config = spdk_app_setup_conf(opts->config_file); if (config == NULL) { goto app_start_setup_conf_err; } if (spdk_app_read_config_file_global_params(opts) < 0) { goto app_start_setup_conf_err; } spdk_log_set_level(SPDK_APP_DEFAULT_LOG_LEVEL); spdk_log_set_backtrace_level(SPDK_APP_DEFAULT_BACKTRACE_LOG_LEVEL); if (spdk_app_setup_env(opts) < 0) { goto app_start_setup_conf_err; } spdk_log_open(); SPDK_NOTICELOG("Total cores available: %d\n", spdk_env_get_core_count()); spdk_thread_lib_init(); /* * If mask not specified on command line or in configuration file, * reactor_mask will be 0x1 which will enable core 0 to run one * reactor. */ if ((rc = spdk_reactors_init(opts->max_delay_us)) != 0) { SPDK_ERRLOG("Invalid reactor mask.\n"); goto app_start_log_close_err; } /* * Note the call to spdk_app_setup_trace() is located here * ahead of spdk_app_setup_signal_handlers(). * That's because there is not an easy/direct clean * way of unwinding alloc'd resources that can occur * in spdk_app_setup_signal_handlers(). */ if (spdk_app_setup_trace(opts) != 0) { goto app_start_log_close_err; } if ((rc = spdk_app_setup_signal_handlers(opts)) != 0) { goto app_start_trace_cleanup_err; } memset(&g_spdk_app, 0, sizeof(g_spdk_app)); g_spdk_app.config = config; g_spdk_app.shm_id = opts->shm_id; g_spdk_app.shutdown_cb = opts->shutdown_cb; g_spdk_app.rc = 0; g_init_lcore = spdk_env_get_current_core(); g_delay_subsystem_init = opts->delay_subsystem_init; g_app_start_event = spdk_event_allocate(g_init_lcore, start_fn, arg1, arg2); rpc_start_event = spdk_event_allocate(g_init_lcore, spdk_app_start_rpc, (void *)opts->rpc_addr, NULL); if (!g_delay_subsystem_init) { spdk_subsystem_init(rpc_start_event); } else { spdk_event_call(rpc_start_event); } /* This blocks until spdk_app_stop is called */ spdk_reactors_start(); return g_spdk_app.rc; app_start_trace_cleanup_err: spdk_trace_cleanup(); app_start_log_close_err: spdk_log_close(); app_start_setup_conf_err: return 1; } void spdk_app_fini(void) { spdk_trace_cleanup(); spdk_reactors_fini(); spdk_conf_free(g_spdk_app.config); spdk_log_close(); spdk_thread_lib_fini(); } static void _spdk_app_stop(void *arg1, void *arg2) { struct spdk_event *app_stop_event; spdk_rpc_finish(); app_stop_event = spdk_event_allocate(spdk_env_get_current_core(), spdk_reactors_stop, NULL, NULL); spdk_subsystem_fini(app_stop_event); } void spdk_app_stop(int rc) { if (rc) { SPDK_WARNLOG("spdk_app_stop'd on non-zero\n"); } g_spdk_app.rc = rc; /* * We want to run spdk_subsystem_fini() from the same lcore where spdk_subsystem_init() * was called. */ spdk_event_call(spdk_event_allocate(g_init_lcore, _spdk_app_stop, NULL, NULL)); } static void usage(void (*app_usage)(void)) { printf("%s [options]\n", g_executable_name); printf("options:\n"); printf(" -c, --config config file (default %s)\n", g_default_opts.config_file); printf(" -d, --limit-coredump do not set max coredump size to RLIM_INFINITY\n"); printf(" -e, --tpoint-group-mask \n"); printf(" tracepoint group mask for spdk trace buffers (default 0x0)\n"); printf(" -g, --single-file-segments\n"); printf(" force creating just one hugetlbfs file\n"); printf(" -h, --help show this usage\n"); printf(" -i, --shm-id shared memory ID (optional)\n"); printf(" -m, --cpumask core mask for DPDK\n"); printf(" -n, --mem-channels channel number of memory channels used for DPDK\n"); printf(" -p, --master-core master (primary) core for DPDK\n"); printf(" -r, --rpc-socket RPC listen address (default %s)\n", SPDK_DEFAULT_RPC_ADDR); printf(" -s, --mem-size memory size in MB for DPDK (default: "); if (g_default_opts.mem_size > 0) { printf("%dMB)\n", g_default_opts.mem_size); } else { printf("all hugepage memory)\n"); } printf(" --silence-noticelog disable notice level logging to stderr\n"); printf(" -u, --no-pci disable PCI access\n"); printf(" --wait-for-rpc wait for RPCs to initialize subsystems\n"); printf(" -B, --pci-blacklist \n"); printf(" pci addr to blacklist (can be used more than once)\n"); printf(" -R, --huge-unlink unlink huge files after initialization\n"); printf(" -W, --pci-whitelist \n"); printf(" pci addr to whitelist (-B and -W cannot be used at the same time)\n"); spdk_tracelog_usage(stdout, "-L"); if (app_usage) { app_usage(); } } spdk_app_parse_args_rvals_t spdk_app_parse_args(int argc, char **argv, struct spdk_app_opts *opts, const char *app_getopt_str, struct option *app_long_opts, void (*app_parse)(int ch, char *arg), void (*app_usage)(void)) { int ch, rc, opt_idx, global_long_opts_len, app_long_opts_len; struct option *cmdline_options; char *cmdline_short_opts = NULL; enum spdk_app_parse_args_rvals retval = SPDK_APP_PARSE_ARGS_FAIL; memcpy(&g_default_opts, opts, sizeof(g_default_opts)); if (opts->config_file && access(opts->config_file, F_OK) != 0) { opts->config_file = NULL; } if (app_long_opts == NULL) { app_long_opts_len = 0; } else { for (app_long_opts_len = 0; app_long_opts[app_long_opts_len].name != NULL; app_long_opts_len++); } global_long_opts_len = SPDK_COUNTOF(g_cmdline_options); cmdline_options = calloc(global_long_opts_len + app_long_opts_len + 1, sizeof(*cmdline_options)); if (!cmdline_options) { fprintf(stderr, "Out of memory\n"); return SPDK_APP_PARSE_ARGS_FAIL; } memcpy(&cmdline_options[0], g_cmdline_options, sizeof(g_cmdline_options)); if (app_long_opts) { memcpy(&cmdline_options[global_long_opts_len], app_long_opts, app_long_opts_len * sizeof(*app_long_opts)); } if (app_getopt_str != NULL) { ch = spdk_app_opts_validate(app_getopt_str); if (ch) { fprintf(stderr, "Duplicated option '%c' between the generic and application specific spdk opts.\n", ch); goto out; } } cmdline_short_opts = spdk_sprintf_alloc("%s%s", app_getopt_str, SPDK_APP_GETOPT_STRING); if (!cmdline_short_opts) { fprintf(stderr, "Out of memory\n"); goto out; } g_executable_name = argv[0]; while ((ch = getopt_long(argc, argv, cmdline_short_opts, cmdline_options, &opt_idx)) != -1) { switch (ch) { case CONFIG_FILE_OPT_IDX: opts->config_file = optarg; break; case LIMIT_COREDUMP_OPT_IDX: opts->enable_coredump = false; break; case TPOINT_GROUP_MASK_OPT_IDX: opts->tpoint_group_mask = optarg; break; case SINGLE_FILE_SEGMENTS_OPT_IDX: opts->hugepage_single_segments = true; break; case HELP_OPT_IDX: usage(app_usage); retval = SPDK_APP_PARSE_ARGS_HELP; goto out; case SHM_ID_OPT_IDX: if (optarg == NULL) { goto out; } opts->shm_id = atoi(optarg); break; case CPUMASK_OPT_IDX: opts->reactor_mask = optarg; break; case MEM_CHANNELS_OPT_IDX: if (optarg == NULL) { goto out; } opts->mem_channel = atoi(optarg); break; case MASTER_CORE_OPT_IDX: if (optarg == NULL) { goto out; } opts->master_core = atoi(optarg); break; case SILENCE_NOTICELOG_OPT_IDX: opts->print_level = SPDK_LOG_WARN; break; case RPC_SOCKET_OPT_IDX: opts->rpc_addr = optarg; break; case MEM_SIZE_OPT_IDX: { uint64_t mem_size_mb; bool mem_size_has_prefix; rc = spdk_parse_capacity(optarg, &mem_size_mb, &mem_size_has_prefix); if (rc != 0) { fprintf(stderr, "invalid memory pool size `-s %s`\n", optarg); usage(app_usage); goto out; } if (mem_size_has_prefix) { /* the mem size is in MB by default, so if a prefix was * specified, we need to manually convert to MB. */ mem_size_mb /= 1024 * 1024; } if (mem_size_mb > INT_MAX) { fprintf(stderr, "invalid memory pool size `-s %s`\n", optarg); usage(app_usage); goto out; } opts->mem_size = (int) mem_size_mb; break; } case NO_PCI_OPT_IDX: opts->no_pci = true; break; case WAIT_FOR_RPC_OPT_IDX: opts->delay_subsystem_init = true; break; case PCI_BLACKLIST_OPT_IDX: if (opts->pci_whitelist) { free(opts->pci_whitelist); opts->pci_whitelist = NULL; fprintf(stderr, "-B and -W cannot be used at the same time\n"); usage(app_usage); goto out; } rc = spdk_app_opts_add_pci_addr(opts, &opts->pci_blacklist, optarg); if (rc != 0) { free(opts->pci_blacklist); opts->pci_blacklist = NULL; goto out; } break; case TRACEFLAG_OPT_IDX: #ifndef DEBUG fprintf(stderr, "%s must be built with CONFIG_DEBUG=y for -L flag\n", argv[0]); usage(app_usage); goto out; #else rc = spdk_log_set_trace_flag(optarg); if (rc < 0) { fprintf(stderr, "unknown flag\n"); usage(app_usage); goto out; } opts->print_level = SPDK_LOG_DEBUG; break; #endif case HUGE_UNLINK_OPT_IDX: opts->unlink_hugepage = true; break; case PCI_WHITELIST_OPT_IDX: if (opts->pci_blacklist) { free(opts->pci_blacklist); opts->pci_blacklist = NULL; fprintf(stderr, "-B and -W cannot be used at the same time\n"); usage(app_usage); goto out; } rc = spdk_app_opts_add_pci_addr(opts, &opts->pci_whitelist, optarg); if (rc != 0) { free(opts->pci_whitelist); opts->pci_whitelist = NULL; goto out; } break; case '?': /* * In the event getopt() above detects an option * in argv that is NOT in the getopt_str, * getopt() will return a '?' indicating failure. */ usage(app_usage); goto out; default: app_parse(ch, optarg); } } /* TBD: Replace warning by failure when RPCs for startup are prepared. */ if (opts->config_file && opts->delay_subsystem_init) { fprintf(stderr, "WARNING: --wait-for-rpc and config file are used at the same time. " "- Please be careful one options might overwrite others.\n"); } retval = SPDK_APP_PARSE_ARGS_SUCCESS; out: if (retval != SPDK_APP_PARSE_ARGS_SUCCESS) { free(opts->pci_blacklist); opts->pci_blacklist = NULL; free(opts->pci_whitelist); opts->pci_whitelist = NULL; } free(cmdline_short_opts); free(cmdline_options); return retval; } void spdk_app_usage(void) { if (g_executable_name == NULL) { fprintf(stderr, "%s not valid before calling spdk_app_parse_args()\n", __func__); return; } usage(NULL); } static void spdk_rpc_start_subsystem_init_cpl(void *arg1, void *arg2) { struct spdk_jsonrpc_request *request = arg1; struct spdk_json_write_ctx *w; spdk_app_start_application(); w = spdk_jsonrpc_begin_result(request); if (w == NULL) { return; } spdk_json_write_bool(w, true); spdk_jsonrpc_end_result(request, w); } static void spdk_rpc_start_subsystem_init(struct spdk_jsonrpc_request *request, const struct spdk_json_val *params) { struct spdk_event *cb_event; if (params != NULL) { spdk_jsonrpc_send_error_response(request, SPDK_JSONRPC_ERROR_INVALID_PARAMS, "start_subsystem_init requires no parameters"); return; } cb_event = spdk_event_allocate(g_init_lcore, spdk_rpc_start_subsystem_init_cpl, request, NULL); spdk_subsystem_init(cb_event); } SPDK_RPC_REGISTER("start_subsystem_init", spdk_rpc_start_subsystem_init, SPDK_RPC_STARTUP)