static bool process(cchar *operation)
{
bool rc;
rc = 1;
if (smatch(operation, "install")) {
rc = installService(app->serviceArgs);
} else if (smatch(operation, "uninstall")) {
rc = removeService(1);
} else if (smatch(operation, "enable")) {
rc = enableService(1);
} else if (smatch(operation, "disable")) {
rc = enableService(0);
} else if (smatch(operation, "start")) {
rc = startService();
} else if (smatch(operation, "stop")) {
rc = removeService(0);
} else if (smatch(operation, "reload")) {
rc = process("restart");
} else if (smatch(operation, "restart")) {
process("stop");
return process("start");
} else if (smatch(operation, "run")) {
/*
This thread will block if being started by SCM. While blocked, the serviceMain
will be called which becomes the effective main program.
*/
startDispatcher(serviceMain);
}
return rc;
}
MorseEventDelayer::MorseEventDelayer()
: currentTime(0),
delayInMilliseconds(0),
timeToDie(false) {
startDispatcher();
}
int APIENTRY WinMain(HINSTANCE inst, HINSTANCE junk, char *args, int junk2)
{
char **argv, *argp, *serviceArgs, *homeDir;
int argc, err, nextArg, installFlag;
mpr = mprCreate(0, NULL, NULL);
appInst = inst;
serviceArgs = 0;
err = 0;
exiting = 0;
installFlag = 0;
homeDir = 0;
heartBeatPeriod = HEART_BEAT_PERIOD;
initService();
mprSetAppName(mpr, BLD_PRODUCT "Angel", BLD_NAME "Angel", BLD_VERSION);
appName = mprGetAppName(mpr);
appTitle = mprGetAppTitle(mpr);
mprSetLogHandler(mpr, logHandler, NULL);
/*
* Create the window
*/
if (initWindow() < 0) {
mprError(mpr, "Can't initialize application Window");
return FALSE;
}
/*
* Parse command line arguments
*/
if (mprMakeArgv(mpr, "", args, &argc, &argv) < 0) {
return FALSE;
}
for (nextArg = 1; nextArg < argc; nextArg++) {
argp = argv[nextArg];
if (*argp != '-' || strcmp(argp, "--") == 0) {
break;
}
if (strcmp(argp, "--args") == 0) {
/*
* Args to pass to service when it starts
*/
if (nextArg >= argc) {
err++;
} else {
serviceArgs = argv[++nextArg];
}
} else if (strcmp(argp, "--console") == 0) {
/*
* Allow the service to interact with the console
*/
createConsole++;
} else if (strcmp(argp, "--daemon") == 0) {
/* Ignored on windows */
} else if (strcmp(argp, "--heartBeat") == 0) {
/*
* Set the heart beat period
*/
if (nextArg >= argc) {
err++;
} else {
heartBeatPeriod = atoi(argv[++nextArg]) * 1000;
}
} else if (strcmp(argp, "--home") == 0) {
/*
* Change to this directory before starting the service
*/
if (nextArg >= argc) {
err++;
} else {
homeDir = argv[++nextArg];
}
} else if (strcmp(argp, "--program") == 0) {
if (nextArg >= argc) {
err++;
} else {
serviceProgram = argv[++nextArg];
}
} else if (strcmp(argp, "--install") == 0) {
installFlag++;
} else if (strcmp(argp, "--start") == 0) {
/*
* Start the angel
*/
if (startService() < 0) {
return FALSE;
}
mprSleep(mpr, 2000); /* Time for service to really start */
} else if (strcmp(argp, "--stop") == 0) {
/*
* Stop the angel
*/
if (removeService(0) < 0) {
return FALSE;
}
} else if (strcmp(argp, "--uninstall") == 0) {
/*
* Remove the angel
*/
if (removeService(1) < 0) {
return FALSE;
}
} else if (strcmp(argp, "--verbose") == 0 || strcmp(argp, "-v") == 0) {
verbose++;
} else {
err++;
}
if (err) {
mprUserError(mpr,
"Bad command line: %s\n"
" Usage: %s [options] [program args]\n"
" Switches:\n"
" --args # Args to pass to service\n"
" --console # Display the service console\n"
" --heartBeat interval # Heart beat interval period (secs)\n"
" --home path # Home directory for service\n"
" --install # Install the service\n"
" --program # Service program to start\n"
" --start # Start the service\n"
" --stop # Stop the service\n"
" --uninstall # Uninstall the service",
args, appName);
return -1;
}
}
if (installFlag) {
/*
* Install the angel
*/
if (installService(homeDir, serviceArgs) < 0) {
return FALSE;
}
}
if (argc <= 1) {
/*
* This will block if we are a service and are being started by the
* service control manager. While blocked, the svcMain will be called
* which becomes the effective main program.
*/
startDispatcher(serviceMain);
}
return 0;
}