int main(int argc, char* argv[])
{
if (argc > 1)
{
char *value = argv[1];
// Parameters should start with '/' or '-'.
if (!(*value == '-' || *value == '/'))
{
std::cerr << "Unexpected parameter: " << value << std::endl
<< std::endl;
show_usage(argv[0]);
return 1;
}
// Skip past the first character.
value = value + 1;
if (*value == 'h' || *value == '?')
{
show_usage(argv[0]);
}
else if (strcmp(value, "shell") == 0)
{
while (!shutdown)
{
try
{
pushyd_once(&std::cout);
}
catch (std::exception const& e)
{
std::cerr << "std::exception caught: " << e.what()
<< std::endl;
return 1;
}
catch (...)
{
std::cerr << "Unknown exception caught" << std::endl;
return 1;
}
}
}
else if (strcmp(value, "install") == 0)
{
return service_install();
}
}
else
{
start_service();
}
return 0;
}
int main(int argc, char **argv) {
int i;
int help = 0;
int version = 0;
int check_config = 0;
int install = 0;
int uninstall = 0;
int console = 0;
int log_to_file = 0;
int debug = 0;
int rc;
for (i = 1; i < argc; ++i) {
if (strcmp(argv[i], "--help") == 0) {
help = 1;
} else if (strcmp(argv[i], "--version") == 0) {
version = 1;
} else if (strcmp(argv[i], "--check-config") == 0) {
check_config = 1;
} else if (strcmp(argv[i], "--install") == 0) {
install = 1;
} else if (strcmp(argv[i], "--uninstall") == 0) {
uninstall = 1;
} else if (strcmp(argv[i], "--console") == 0) {
console = 1;
} else if (strcmp(argv[i], "--log-to-file") == 0) {
log_to_file = 1;
} else if (strcmp(argv[i], "--debug") == 0) {
debug = 1;
} else {
fprintf(stderr, "Unknown option '%s'\n\n", argv[i]);
print_usage();
return EXIT_FAILURE;
}
}
if (help) {
print_usage();
return EXIT_SUCCESS;
}
if (version) {
printf("%s\n", VERSION_STRING);
return EXIT_SUCCESS;
}
if (GetModuleFileName(NULL, _config_filename, sizeof(_config_filename)) == 0) {
rc = ERRNO_WINAPI_OFFSET + GetLastError();
fprintf(stderr, "Could not get module file name: %s (%d)\n",
get_errno_name(rc), rc);
return EXIT_FAILURE;
}
i = strlen(_config_filename);
if (i < 4) {
fprintf(stderr, "Module file name '%s' is too short", _config_filename);
return EXIT_FAILURE;
}
strcpy(_config_filename + i - 3, "ini");
if (check_config) {
return config_check(_config_filename) < 0 ? EXIT_FAILURE : EXIT_SUCCESS;
}
if (install && uninstall) {
fprintf(stderr, "Invalid option combination\n");
print_usage();
return EXIT_FAILURE;
}
if (install) {
if (service_install(log_to_file, debug) < 0) {
return EXIT_FAILURE;
}
} else if (uninstall) {
if (service_uninstall() < 0) {
return EXIT_FAILURE;
}
} else {
printf("Starting...\n");
config_init(_config_filename);
log_init();
if (console) {
_run_as_service = 0;
_pause_before_exit = started_by_explorer();
return generic_main(log_to_file, debug);
} else {
return service_run(log_to_file, debug);
}
}
return EXIT_SUCCESS;
}
int main(int argc, char **argv) {
int i;
bool help = false;
bool version = false;
bool check_config = false;
bool install = false;
bool uninstall = false;
bool console = false;
bool log_to_file = false;
bool debug = false;
bool libusb_debug = false;
int rc;
fixes_init();
for (i = 1; i < argc; ++i) {
if (strcmp(argv[i], "--help") == 0) {
help = true;
} else if (strcmp(argv[i], "--version") == 0) {
version = true;
} else if (strcmp(argv[i], "--check-config") == 0) {
check_config = true;
} else if (strcmp(argv[i], "--install") == 0) {
install = true;
} else if (strcmp(argv[i], "--uninstall") == 0) {
uninstall = true;
} else if (strcmp(argv[i], "--console") == 0) {
console = true;
} else if (strcmp(argv[i], "--log-to-file") == 0) {
log_to_file = true;
} else if (strcmp(argv[i], "--debug") == 0) {
debug = true;
} else if (strcmp(argv[i], "--libusb-debug") == 0) {
libusb_debug = true;
} else {
fprintf(stderr, "Unknown option '%s'\n\n", argv[i]);
print_usage();
return EXIT_FAILURE;
}
}
if (help) {
print_usage();
return EXIT_SUCCESS;
}
if (version) {
print_version();
return EXIT_SUCCESS;
}
if (GetModuleFileName(NULL, _config_filename, sizeof(_config_filename)) == 0) {
rc = ERRNO_WINAPI_OFFSET + GetLastError();
fprintf(stderr, "Could not get module file name: %s (%d)\n",
get_errno_name(rc), rc);
return EXIT_FAILURE;
}
i = strlen(_config_filename);
if (i < 4) {
fprintf(stderr, "Module file name '%s' is too short", _config_filename);
return EXIT_FAILURE;
}
_config_filename[i - 3] = '\0';
string_append(_config_filename, "ini", sizeof(_config_filename));
if (check_config) {
rc = config_check(_config_filename);
if (started_by_explorer(false)) {
printf("\nPress any key to exit...\n");
getch();
}
return rc < 0 ? EXIT_FAILURE : EXIT_SUCCESS;
}
if (install && uninstall) {
fprintf(stderr, "Invalid option combination\n");
print_usage();
return EXIT_FAILURE;
}
if (install) {
if (service_install(log_to_file, debug) < 0) {
return EXIT_FAILURE;
}
} else if (uninstall) {
if (service_uninstall() < 0) {
return EXIT_FAILURE;
}
} else {
printf("Starting...\n");
config_init(_config_filename);
log_init();
if (console) {
_run_as_service = false;
_pause_before_exit = started_by_explorer(true);
return generic_main(log_to_file, debug, libusb_debug);
} else {
return service_run(log_to_file, debug, libusb_debug);
}
}
return EXIT_SUCCESS;
}
int WINAPI WinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance,
#ifdef _WIN32_WCE
LPWSTR lpCmdLine,
#else
LPSTR lpCmdLine,
#endif
int nCmdShow) {
LPSTR command_line;
#ifdef _WIN32_WCE
command_line=tstr2str(lpCmdLine);
#else
command_line=lpCmdLine;
#endif
ghInst=hInstance;
parse_cmdline(command_line); /* setup global cmdline structure */
#ifndef _WIN32_WCE
if(set_cwd()) /* set current working directory */
return 1;
#endif
/* setup the windo caption before reading the configuration file
* options.win32_service is not available here and may not be used */
#ifdef _WIN32_WCE
_tcscpy(win32_name, TEXT("stunnel ") TEXT(VERSION)
TEXT(" on Windows CE (not configured)"));
#else
_tcscpy(win32_name, TEXT("stunnel ") TEXT(VERSION)
TEXT(" on Win32 (not configured)"));
#endif
if(initialize_winsock())
return 1;
if(!setjmp(jump_buf)) { /* TRY */
main_initialize(
cmdline.config_file[0] ? cmdline.config_file : NULL, NULL);
#ifdef _WIN32_WCE
_tcscpy(win32_name, TEXT("stunnel ") TEXT(VERSION)
TEXT(" on Windows CE"));
#else
_snprintf(win32_name, STRLEN, "stunnel %s on Win32 (%s)",
VERSION, options.win32_service); /* update the information */
if(!cmdline.service) {
if(cmdline.install)
return service_install(command_line);
if(cmdline.uninstall)
return service_uninstall();
if(cmdline.start)
return service_start();
if(cmdline.stop)
return service_stop();
}
#endif
}
/* CATCH */
#ifndef _WIN32_WCE
if(cmdline.service)
return service_initialize();
else
#endif
return win_main(hInstance, hPrevInstance, command_line, nCmdShow);
}
int main(int argc, char *argv[])
{
int *listensock = NULL;
int listensock_count = 0;
int listensock_index = 0;
struct mqtt3_config config;
char buf[1024];
int i, j;
FILE *pid;
int listener_max;
int rc;
char err[256];
#ifdef WIN32
SYSTEMTIME st;
#else
struct timeval tv;
#endif
#if defined(WIN32) || defined(__CYGWIN__)
if(argc == 2){
if(!strcmp(argv[1], "run")){
service_run();
return 0;
}else if(!strcmp(argv[1], "install")){
service_install();
return 0;
}else if(!strcmp(argv[1], "uninstall")){
service_uninstall();
return 0;
}
}
#endif
#ifdef WIN32
GetSystemTime(&st);
srand(st.wSecond + st.wMilliseconds);
#else
gettimeofday(&tv, NULL);
srand(tv.tv_sec + tv.tv_usec);
#endif
memset(&int_db, 0, sizeof(struct mosquitto_db));
_mosquitto_net_init();
mqtt3_config_init(&config);
rc = mqtt3_config_parse_args(&config, argc, argv);
if(rc != MOSQ_ERR_SUCCESS) return rc;
int_db.config = &config;
if(config.daemon){
#ifndef WIN32
switch(fork()){
case 0:
break;
case -1:
strerror_r(errno, err, 256);
_mosquitto_log_printf(NULL, MOSQ_LOG_ERR, "Error in fork: %s", err);
return 1;
default:
return MOSQ_ERR_SUCCESS;
}
#else
_mosquitto_log_printf(NULL, MOSQ_LOG_WARNING, "Warning: Can't start in daemon mode in Windows.");
#endif
}
if(config.daemon && config.pid_file){
pid = _mosquitto_fopen(config.pid_file, "wt");
if(pid){
fprintf(pid, "%d", getpid());
fclose(pid);
}else{
_mosquitto_log_printf(NULL, MOSQ_LOG_ERR, "Error: Unable to write pid file.");
return 1;
}
}
rc = drop_privileges(&config);
if(rc != MOSQ_ERR_SUCCESS) return rc;
rc = mqtt3_db_open(&config, &int_db);
if(rc != MOSQ_ERR_SUCCESS){
_mosquitto_log_printf(NULL, MOSQ_LOG_ERR, "Error: Couldn't open database.");
return rc;
}
/* Initialise logging only after initialising the database in case we're
* logging to topics */
mqtt3_log_init(config.log_type, config.log_dest);
_mosquitto_log_printf(NULL, MOSQ_LOG_INFO, "mosquitto version %s (build date %s) starting", VERSION, TIMESTAMP);
if(config.config_file){
_mosquitto_log_printf(NULL, MOSQ_LOG_INFO, "Config loaded from %s.", config.config_file);
}else{
_mosquitto_log_printf(NULL, MOSQ_LOG_INFO, "Using default config.");
}
rc = mosquitto_security_module_init(&int_db);
if(rc) return rc;
rc = mosquitto_security_init(&int_db, false);
if(rc) return rc;
#ifdef WITH_SYS_TREE
if(config.sys_interval > 0){
/* Set static $SYS messages */
snprintf(buf, 1024, "mosquitto version %s", VERSION);
mqtt3_db_messages_easy_queue(&int_db, NULL, "$SYS/broker/version", 2, strlen(buf), buf, 1);
snprintf(buf, 1024, "%s", TIMESTAMP);
mqtt3_db_messages_easy_queue(&int_db, NULL, "$SYS/broker/timestamp", 2, strlen(buf), buf, 1);
#ifdef CHANGESET
snprintf(buf, 1024, "%s", CHANGESET);
mqtt3_db_messages_easy_queue(&int_db, NULL, "$SYS/broker/changeset", 2, strlen(buf), buf, 1);
#endif
}
#endif
listener_max = -1;
listensock_index = 0;
for(i=0; i<config.listener_count; i++){
if(mqtt3_socket_listen(&config.listeners[i])){
_mosquitto_free(int_db.contexts);
mqtt3_db_close(&int_db);
if(config.pid_file){
remove(config.pid_file);
}
return 1;
}
listensock_count += config.listeners[i].sock_count;
listensock = _mosquitto_realloc(listensock, sizeof(int)*listensock_count);
if(!listensock){
_mosquitto_free(int_db.contexts);
mqtt3_db_close(&int_db);
if(config.pid_file){
remove(config.pid_file);
}
return 1;
}
for(j=0; j<config.listeners[i].sock_count; j++){
if(config.listeners[i].socks[j] == INVALID_SOCKET){
_mosquitto_free(int_db.contexts);
mqtt3_db_close(&int_db);
if(config.pid_file){
remove(config.pid_file);
}
return 1;
}
listensock[listensock_index] = config.listeners[i].socks[j];
if(listensock[listensock_index] > listener_max){
listener_max = listensock[listensock_index];
}
listensock_index++;
}
}
signal(SIGINT, handle_sigint);
signal(SIGTERM, handle_sigint);
#ifdef SIGHUP
signal(SIGHUP, handle_sighup);
#endif
#ifndef WIN32
signal(SIGUSR1, handle_sigusr1);
signal(SIGUSR2, handle_sigusr2);
signal(SIGPIPE, SIG_IGN);
#endif
#ifdef WITH_BRIDGE
for(i=0; i<config.bridge_count; i++){
if(mqtt3_bridge_new(&int_db, &(config.bridges[i]))){
_mosquitto_log_printf(NULL, MOSQ_LOG_WARNING, "Warning: Unable to connect to bridge %s.",
config.bridges[i].name);
}
}
#endif
run = 1;
rc = mosquitto_main_loop(&int_db, listensock, listensock_count, listener_max);
_mosquitto_log_printf(NULL, MOSQ_LOG_INFO, "mosquitto version %s terminating", VERSION);
mqtt3_log_close();
#ifdef WITH_PERSISTENCE
if(config.persistence){
mqtt3_db_backup(&int_db, true, true);
}
#endif
for(i=0; i<int_db.context_count; i++){
if(int_db.contexts[i]){
mqtt3_context_cleanup(&int_db, int_db.contexts[i], true);
}
}
_mosquitto_free(int_db.contexts);
int_db.contexts = NULL;
mqtt3_db_close(&int_db);
if(listensock){
for(i=0; i<listensock_count; i++){
if(listensock[i] != INVALID_SOCKET){
#ifndef WIN32
close(listensock[i]);
#else
closesocket(listensock[i]);
#endif
}
}
_mosquitto_free(listensock);
}
mosquitto_security_module_cleanup(&int_db);
if(config.pid_file){
remove(config.pid_file);
}
_mosquitto_net_cleanup();
mqtt3_config_cleanup(int_db.config);
return rc;
}
int WINAPI WinMain(HINSTANCE this_instance, HINSTANCE prev_instance,
#ifdef _WIN32_WCE
LPWSTR lpCmdLine,
#else
LPSTR lpCmdLine,
#endif
int nCmdShow) {
TCHAR stunnel_exe_path[MAX_PATH];
LPTSTR c;
#ifndef _WIN32_WCE
LPTSTR errmsg;
#endif
(void)prev_instance; /* squash the unused parameter warning */
(void)lpCmdLine; /* squash the unused parameter warning */
(void)nCmdShow; /* squash the unused parameter warning */
tls_init(); /* initialize thread-local storage */
ghInst=this_instance;
gui_cmdline(); /* setup global cmdline structure */
GetModuleFileName(0, stunnel_exe_path, MAX_PATH);
#ifndef _WIN32_WCE
/* find previous instances of the same executable */
if(!cmdline.service && !cmdline.install && !cmdline.uninstall &&
!cmdline.reload && !cmdline.reopen &&
!cmdline.start && !cmdline.stop) {
EnumWindows(enum_windows, (LPARAM)stunnel_exe_path);
if(cmdline.exit)
return 1; /* in case EnumWindows didn't find a previous instance */
}
#endif
/* set current working directory and engine path */
c=_tcsrchr(stunnel_exe_path, TEXT('\\')); /* last backslash */
if(c) { /* found */
*c=TEXT('\0'); /* truncate the program name */
c=_tcsrchr(stunnel_exe_path, TEXT('\\')); /* previous backslash */
if(c && !_tcscmp(c+1, TEXT("bin")))
*c=TEXT('\0'); /* truncate "bin" */
}
#ifndef _WIN32_WCE
if(!SetCurrentDirectory(stunnel_exe_path)) {
errmsg=str_tprintf(TEXT("Cannot set directory to %s"),
stunnel_exe_path);
message_box(errmsg, MB_ICONERROR);
str_free(errmsg);
return 1;
}
/* try to enter the "config" subdirectory, ignore the result */
SetCurrentDirectory(TEXT("config"));
#endif
_tputenv(str_tprintf(TEXT("OPENSSL_ENGINES=%s\\engines"),
stunnel_exe_path));
if(initialize_winsock())
return 1;
#ifndef _WIN32_WCE
if(cmdline.service) /* "-service" must be processed before "-install" */
return service_initialize();
if(cmdline.install)
return service_install();
if(cmdline.uninstall)
return service_uninstall();
if(cmdline.start)
return service_start();
if(cmdline.stop)
return service_stop();
if(cmdline.reload)
return service_user(SIGNAL_RELOAD_CONFIG);
if(cmdline.reopen)
return service_user(SIGNAL_REOPEN_LOG);
#endif
return gui_loop();
}
int WINAPI WinMain(HINSTANCE this_instance, HINSTANCE prev_instance,
#ifdef _WIN32_WCE
LPWSTR lpCmdLine,
#else
LPSTR lpCmdLine,
#endif
int nCmdShow) {
LPSTR command_line;
char *c, stunnel_exe_path[MAX_PATH];
#ifndef _WIN32_WCE
char *errmsg;
#endif
(void)prev_instance; /* skip warning about unused parameter */
(void)nCmdShow; /* skip warning about unused parameter */
str_init(); /* initialize per-thread string management */
ghInst=this_instance;
#ifdef _WIN32_WCE
command_line=tstr2str(lpCmdLine);
#else
command_line=lpCmdLine;
#endif
parse_cmdline(command_line); /* setup global cmdline structure */
GetModuleFileName(0, stunnel_exe_path, MAX_PATH);
#ifndef _WIN32_WCE
/* find previous instances of the same executable */
if(!cmdline.service && !cmdline.install && !cmdline.uninstall &&
!cmdline.start && !cmdline.stop) {
EnumWindows(enum_windows, (LPARAM)stunnel_exe_path);
if(cmdline.exit)
return 0; /* in case EnumWindows didn't find a previous instance */
}
#endif
/* set current working directory and engine path */
c=strrchr(stunnel_exe_path, '\\'); /* last backslash */
if(c) /* found */
c[1]='\0'; /* truncate program name */
#ifndef _WIN32_WCE
if(!SetCurrentDirectory(stunnel_exe_path)) {
errmsg=str_printf("Cannot set directory to %s", stunnel_exe_path);
message_box(errmsg, MB_ICONERROR);
str_free(errmsg);
return 1;
}
#endif
_putenv_s("OPENSSL_ENGINES", stunnel_exe_path);
if(initialize_winsock())
return 1;
#ifndef _WIN32_WCE
if(cmdline.service) /* "-service" must be processed before "-install" */
return service_initialize();
if(cmdline.install)
return service_install(command_line);
if(cmdline.uninstall)
return service_uninstall();
if(cmdline.start)
return service_start();
if(cmdline.stop)
return service_stop();
#endif
return gui_loop();
}
