int main(int argc, char *argv[])
{
/* Don't bother checking for privilege if they only want usage() */
if (argc < 2)
goto usage;
vbetool_init();
if (!strcmp(argv[1], "vbestate")) {
/* VBE save/restore tends to break when done underneath X */
int err = check_console();
if (err) {
return err;
}
if (!strcmp(argv[2], "save")) {
save_state();
} else if (!strcmp(argv[2], "restore")) {
restore_state();
} else {
goto usage;
}
} else if (!strcmp(argv[1], "dpms")) {
if (!strcmp(argv[2], "on")) {
return do_blank(DPMS_STATE_ON);
} else if (!strcmp(argv[2], "suspend")) {
return do_blank(DPMS_STATE_SUSPEND);
} else if (!strcmp(argv[2], "standby")) {
return do_blank(DPMS_STATE_STANDBY);
} else if (!strcmp(argv[2], "off")) {
return do_blank(DPMS_STATE_OFF);
} else if (!strcmp(argv[2], "reduced")) {
return do_blank(DPMS_STATE_LOW);
} else {
goto usage;
}
} else if (!strcmp(argv[1], "vbemode")) {
if (!strcmp(argv[2], "set")) {
return do_set_mode(atoi(argv[3]),0);
} else if (!strcmp(argv[2], "get")) {
return do_get_mode();
} else {
goto usage;
}
} else if (!strcmp(argv[1], "vgamode")) {
if (!strcmp(argv[2], "set")) {
return do_set_mode(atoi(argv[3]),1);
} else {
return do_set_mode(atoi(argv[2]),1);
}
} else if (!strcmp(argv[1], "post")) {
/* Again, we don't really want to do this while X is in
control */
int err = check_console();
if (err) {
return err;
}
return do_post();
} else if (!strcmp(argv[1], "vgastate")) {
if (!strcmp(argv[2], "on")) {
return enable_vga();
} else {
return disable_vga();
}
} else if (!strcmp(argv[1], "vbefp")) {
if (!strcmp(argv[2], "id") || !strcmp(argv[2], "panelid")) {
return do_get_panel_id(0);
} else if (!strcmp(argv[2], "panelsize")) {
return do_get_panel_id(1);
} else if (!strcmp(argv[2], "getbrightness")) {
return do_get_panel_brightness();
} else if (!strcmp(argv[2], "setbrightness")) {
return do_set_panel_brightness(atoi(argv[3]));
} else if (!strcmp(argv[2], "invert")) {
return do_invert_panel();
} else {
return 1;
}
} else {
usage:
fprintf(stderr,
"%s: Usage %s [[vbestate save|restore]|[vbemode set|get]|[vgamode]|[dpms on|off|standby|suspend|reduced]|[post]|[vgastate on|off]|[vbefp panelid|panelsize|getbrightness|setbrightness|invert]]\n",
argv[0], argv[0]);
return 1;
}
return 0;
}
int _tmain(int argc, TCHAR **argv) {
check_console();
#ifdef UNICODE
/*
Ensure we write in UTF-16 mode, so that non-ASCII characters don't get
mangled. If we were compiled in ANSI mode it won't work.
*/
_setmode(_fileno(stdout), _O_U16TEXT);
_setmode(_fileno(stderr), _O_U16TEXT);
#endif
/* Remember if we are admin */
check_admin();
/* Set up function pointers. */
if (get_imports()) exit(111);
/* Remember our path for later. */
GetModuleFileName(0, unquoted_imagepath, _countof(unquoted_imagepath));
GetModuleFileName(0, imagepath, _countof(imagepath));
PathQuoteSpaces(imagepath);
/* Elevate */
if (argc > 1) {
/*
Valid commands are:
start, stop, pause, continue, install, edit, get, set, reset, unset, remove
*/
if (str_equiv(argv[1], _T("start"))) exit(control_service(NSSM_SERVICE_CONTROL_START, argc - 2, argv + 2));
if (str_equiv(argv[1], _T("stop"))) exit(control_service(SERVICE_CONTROL_STOP, argc - 2, argv + 2));
if (str_equiv(argv[1], _T("restart"))) {
int ret = control_service(SERVICE_CONTROL_STOP, argc - 2, argv + 2);
if (ret) exit(ret);
exit(control_service(NSSM_SERVICE_CONTROL_START, argc - 2, argv + 2));
}
if (str_equiv(argv[1], _T("pause"))) exit(control_service(SERVICE_CONTROL_PAUSE, argc - 2, argv + 2));
if (str_equiv(argv[1], _T("continue"))) exit(control_service(SERVICE_CONTROL_CONTINUE, argc - 2, argv + 2));
if (str_equiv(argv[1], _T("status"))) exit(control_service(SERVICE_CONTROL_INTERROGATE, argc - 2, argv + 2));
if (str_equiv(argv[1], _T("rotate"))) exit(control_service(NSSM_SERVICE_CONTROL_ROTATE, argc - 2, argv + 2));
if (str_equiv(argv[1], _T("install"))) {
if (! is_admin) exit(elevate(argc, argv, NSSM_MESSAGE_NOT_ADMINISTRATOR_CANNOT_INSTALL));
create_messages();
exit(pre_install_service(argc - 2, argv + 2));
}
if (str_equiv(argv[1], _T("edit")) || str_equiv(argv[1], _T("get")) || str_equiv(argv[1], _T("set")) || str_equiv(argv[1], _T("reset")) || str_equiv(argv[1], _T("unset"))) {
int ret = pre_edit_service(argc - 1, argv + 1);
if (ret == 3 && ! is_admin && argc == 3) exit(elevate(argc, argv, NSSM_MESSAGE_NOT_ADMINISTRATOR_CANNOT_EDIT));
/* There might be a password here. */
for (int i = 0; i < argc; i++) SecureZeroMemory(argv[i], _tcslen(argv[i]) * sizeof(TCHAR));
exit(ret);
}
if (str_equiv(argv[1], _T("list"))) exit(list_nssm_services());
if (str_equiv(argv[1], _T("remove"))) {
if (! is_admin) exit(elevate(argc, argv, NSSM_MESSAGE_NOT_ADMINISTRATOR_CANNOT_REMOVE));
exit(pre_remove_service(argc - 2, argv + 2));
}
}
/* Thread local storage for error message buffer */
tls_index = TlsAlloc();
/* Register messages */
if (is_admin) create_messages();
/*
Optimisation for Windows 2000:
When we're run from the command line the StartServiceCtrlDispatcher() call
will time out after a few seconds on Windows 2000. On newer versions the
call returns instantly. Check for stdin first and only try to call the
function if there's no input stream found. Although it's possible that
we're running with input redirected it's much more likely that we're
actually running as a service.
This will save time when running with no arguments from a command prompt.
*/
if (! GetStdHandle(STD_INPUT_HANDLE)) {
/* Start service magic */
SERVICE_TABLE_ENTRY table[] = { { NSSM, service_main }, { 0, 0 } };
if (! StartServiceCtrlDispatcher(table)) {
unsigned long error = GetLastError();
/* User probably ran nssm with no argument */
if (error == ERROR_FAILED_SERVICE_CONTROLLER_CONNECT) exit(usage(1));
log_event(EVENTLOG_ERROR_TYPE, NSSM_EVENT_DISPATCHER_FAILED, error_string(error), 0);
free_imports();
exit(100);
}
}
else exit(usage(1));
/* And nothing more to do */
exit(0);
}
