BOOL CGuiMPIJobApp::InitInstance()
{
AfxEnableControlContainer();
// Standard initialization
#ifdef _AFXDLL
Enable3dControls(); // Call this when using MFC in a shared DLL
#else
Enable3dControlsStatic(); // Call this when linking to MFC statically
#endif
CGuiMPIJobDlg dlg;
easy_socket_init();
m_pMainWnd = &dlg;
int nResponse = dlg.DoModal();
if (nResponse == IDOK)
{
}
else if (nResponse == IDCANCEL)
{
}
easy_socket_finalize();
// Since the dialog has been closed, return FALSE so that we exit the
// application, rather than start the application's message pump.
return FALSE;
}
void ErrorExit(char *str, int exitcode = -1)
{
err_printf("****%s", str);
ContextFinalize();
dbs_finalize();
easy_socket_finalize();
err_printf("****EXITING\n");
ExitProcess(exitcode);
}
//
// FUNCTION: ServiceStart
//
// PURPOSE: Actual code of the service
// that does the work.
//
// PARAMETERS:
// dwArgc - number of command line arguments
// lpszArgv - array of command line arguments
//
// RETURN VALUE:
// none
//
VOID ServiceStart (DWORD dwArgc, LPTSTR *lpszArgv)
{
int run_retval = RUN_EXIT;
HANDLE stdin_thread = NULL;
int iter;
// report the status to the service control manager.
if (!ReportStatusToSCMgr(SERVICE_START_PENDING, NO_ERROR, 3000))
return;
// Load the path to the service executable
char szExe[1024];
HMODULE hModule = GetModuleHandle(NULL);
if (!GetModuleFileName(hModule, szExe, 1024))
strcpy(szExe, "mpd.exe");
WriteMPDRegistry("path", szExe);
// Initialize
dbs_init();
ContextInit();
if (!bDebug)
{
#ifndef _DEBUG
// If we are not running in debug mode and this is the release
// build then set the error mode to prevent popup message boxes.
#ifdef USE_SET_ERROR_MODE
SetErrorMode(SEM_FAILCRITICALERRORS | SEM_NOGPFAULTERRORBOX);
#endif
#endif
easy_socket_init();
ParseRegistry(false);
}
// report the status to the service control manager.
if (!ReportStatusToSCMgr(SERVICE_START_PENDING, NO_ERROR, 3000))
{
easy_socket_finalize();
dbs_finalize();
ContextFinalize();
return;
}
g_hProcessStructMutex = CreateMutex(NULL, FALSE, NULL);
g_hForwarderMutex = CreateMutex(NULL, FALSE, NULL);
g_hLaunchMutex = CreateMutex(NULL, FALSE, NULL);
g_hBarrierStructMutex = CreateMutex(NULL, FALSE, NULL);
InitMPDUser();
do
{
// report the status to the service control manager.
if (!ReportStatusToSCMgr(SERVICE_RUNNING, NO_ERROR, 0))
{
easy_socket_finalize();
dbs_finalize();
ContextFinalize();
return;
}
////////////////////////////////////////////////////////
//
// Service is now running, perform work until shutdown
//
AddInfoToMessageLog("MPICH_MPD Daemon service started.");
if (stdin_thread == NULL && bDebug)
{
for (iter=0; iter<CREATE_THREAD_RETRIES; iter++)
{
stdin_thread = CreateThread(
NULL, 0,
(LPTHREAD_START_ROUTINE)StdinThread,
NULL, 0, NULL);
if (stdin_thread != NULL)
break;
Sleep(CREATE_THREAD_SLEEP_TIME);
}
if (stdin_thread == NULL)
{
err_printf("ServiceStart:CreateThread(stdin_thread) failed, error %d\n", GetLastError());
}
}
__try {
run_retval = RUN_RESTART;
run_retval = Run();
if (run_retval == RUN_RESTART)
{
warning_printf("Run returned RUN_RESTART, restarting mpd.");
}
} __except ( EvalException(GetExceptionInformation()) ) {}
RemoveAllContexts();
} while (run_retval == RUN_RESTART);
CloseHandle(g_hProcessStructMutex);
CloseHandle(g_hForwarderMutex);
CloseHandle(g_hLaunchMutex);
CloseHandle(g_hBarrierStructMutex);
if (stdin_thread != NULL)
{
TerminateThread(stdin_thread, 0);
CloseHandle(stdin_thread);
}
easy_socket_finalize();
dbs_finalize();
ContextFinalize();
AddInfoToMessageLog("MPICH_MPD Daemon service stopped.");
SetEvent(g_hBombDiffuseEvent);
if (g_hBombThread != NULL)
{
if (WaitForSingleObject(g_hBombThread, 5000) == WAIT_TIMEOUT)
{
TerminateThread(g_hBombThread, 0);
}
CloseHandle(g_hBombThread);
}
CloseHandle(g_hBombDiffuseEvent);
dbg_printf("ServiceStart: exiting.\n");
}
// parseCommandLine ///////////////////////////////////////////////////////////
//
// check for command line parameters and set various flags
//
void parseCommandLine (int *argc, char** argv[])
{
/*
// A little snipet of code to test the update feature.
// Uncomment this code, compile, run mpd -update, then run mpd -loser to see if this new functionality exists
if (GetOpt(*argc, *argv, "-loser"))
{
printf("you are a winner\n");
ExitProcess(0);
}
//*/
if (GetOpt(*argc, *argv, "-norestart"))
bSetupRestart = false;
if (GetOpt(*argc, *argv, "-interact"))
{
interact = true;
}
if (GetOpt(*argc, *argv, "-remove") || GetOpt(*argc, *argv, "-unregserver") || GetOpt(*argc, *argv, "-uninstall"))
{
RegDeleteKey(HKEY_CURRENT_USER, MPICHKEY);
CmdRemoveService(TRUE);
ExitProcess(0);
}
if (GetOpt(*argc, *argv, "-install") || GetOpt(*argc, *argv, "-regserver"))
{
bool bMPDUserCapable = false;
char account[100]="", password[100]="", phrase[MPD_PASSPHRASE_MAX_LENGTH+1]="", port[10]="";
char version[100]="";
if (CmdRemoveService(FALSE) == FALSE)
{
printf("Unable to remove the previous installation, install failed.\n");
ExitProcess(0);
}
easy_socket_init();
CreateMPDRegistry();
bMPDUserCapable = GetOpt(*argc, *argv, "-mpduser");
if (GetOpt(*argc, *argv, "-phrase", phrase))
WriteMPDRegistry("phrase", phrase);
if (GetOpt(*argc, *argv, "-getphrase"))
{
GetPassword("passphrase for mpd: ", NULL, phrase);
WriteMPDRegistry("phrase", phrase);
}
if (GetOpt(*argc, *argv, "-port", port))
WriteMPDRegistry("port", port);
if (GetOpt(*argc, *argv, "-account", account))
{
if (!GetOpt(*argc, *argv, "-password", password))
GetPassword(NULL, account, password);
WriteMPDRegistry("SingleUser", "yes");
ParseRegistry(true);
CmdInstallService(account, password, bMPDUserCapable);
}
else
{
WriteMPDRegistry("SingleUser", "no");
ParseRegistry(true);
CmdInstallService(NULL, NULL, bMPDUserCapable);
}
GetMPDVersion(version, 100);
WriteMPDRegistry("version", version);
easy_socket_finalize();
ExitProcess(0);
}
if (GetOpt(*argc, *argv, "-update"))
{
char account[100]="", password[100]="", phrase[MPD_PASSPHRASE_MAX_LENGTH+1]="", port[10]="";
int nPort;
char pszHost[MAX_HOST_LENGTH], pszHostFile[MAX_PATH];
char pszFileName[MAX_PATH];
char pszError[256];
if (!GetOpt(*argc, *argv, "-mpd", pszFileName))
{
HMODULE hModule = GetModuleHandle(NULL);
if (!GetModuleFileName(hModule, pszFileName, MAX_PATH))
{
printf("Please specify the location of the new mpd.exe with the -mpd option, (-mpd c:\\some\\path\\mpd.exe)\n");
ExitProcess(0);
}
printf("updating mpd to '%s'\n", pszFileName);
}
if (!GetOpt(*argc, *argv, "-singleuser"))
{
if (GetOpt(*argc, *argv, "-account", account))
{
if (!GetOpt(*argc, *argv, "-password", password))
GetPassword(NULL, account, password);
}
else
{
printf("Enter a user to connect to the remote machines as.\naccount: ");fflush(stdout);
gets(account);
GetPassword(NULL, account, password);
}
}
easy_socket_init();
nPort = MPD_DEFAULT_PORT;
if (!ReadMPDRegistry("phrase", phrase, false))
strcpy(phrase, MPD_DEFAULT_PASSPHRASE);
GetOpt(*argc, *argv, "-phrase", phrase);
if (GetOpt(*argc, *argv, "-getphrase"))
{
GetPassword("passphrase for mpd: ", NULL, phrase);
}
if (GetOpt(*argc, *argv, "-port", port))
nPort = atoi(port);
if (GetOpt(*argc, *argv, "-hostfile", pszHostFile))
{
FILE *fin = fopen(pszHostFile, "r");
if (fin == NULL)
{
char pszStr[1024];
Translate_Error(GetLastError(), pszStr);
printf("Unable to open the host file '%s', %s\n", pszHostFile, pszStr);
easy_socket_finalize();
ExitProcess(0);
}
while (GetNextHost(fin, pszHost))
{
pszError[0] = '\0';
if (!UpdateMPD(pszHost, account, password, nPort, phrase, pszFileName, pszError, 256))
{
printf("Failed to update mpd on %s:\n%s\n", pszHost, pszError);
}
}
fclose(fin);
}
else
{
if (!GetOpt(*argc, *argv, "-host", pszHost))
{
printf("Enter the hostname where the mpd that you wish to update is running.\nhost: ");fflush(stdout);
gets(pszHost);
}
pszError[0] = '\0';
if (!UpdateMPD(pszHost, account, password, nPort, phrase, pszFileName, pszError, 256))
{
printf("Failed to update mpd on %s:\n%s\n", pszHost, pszError);
}
}
easy_socket_finalize();
printf("Finished.\n");
ExitProcess(0);
}
if (GetOpt(*argc, *argv, "-iupdate"))
{
// This option is used internally by the update feature
char pszOldFileName[MAX_PATH], pszNewFileName[MAX_PATH];
char pszPid[10];
if (GetOpt(*argc, *argv, "-pid", pszPid) &&
GetOpt(*argc, *argv, "-old", pszOldFileName) &&
GetOpt(*argc, *argv, "-new", pszNewFileName))
{
UpdateMPD(pszOldFileName, pszNewFileName, atoi(pszPid));
}
ExitProcess(0);
}
char host[100];
if (GetOpt(*argc, *argv, "-console", host))
{
char phrase[MPD_PASSPHRASE_MAX_LENGTH+1];
int port = -1;
GetOpt(*argc, *argv, "-port", &port);
/* DoConsole destroys phrase after using it */
DoConsole(
host, port,
GetOpt(*argc, *argv, "-getphrase"),
GetOpt(*argc, *argv, "-phrase", phrase) ? phrase : NULL);
easy_socket_finalize();
ExitProcess(0);
}
if (GetOpt(*argc, *argv, "-console"))
{
char phrase[MPD_PASSPHRASE_MAX_LENGTH+1];
int port = -1;
GetOpt(*argc, *argv, "-port", &port);
/* DoConsole destroys phrase after using it */
DoConsole(
NULL, port,
GetOpt(*argc, *argv, "-getphrase"),
GetOpt(*argc, *argv, "-phrase", phrase) ? phrase : NULL);
easy_socket_finalize();
ExitProcess(0);
}
if (GetOpt(*argc, *argv, "-start"))
{
CmdStartService();
ExitProcess(0);
}
char pszFileName[MAX_PATH];
if (GetOpt(*argc, *argv, "-startdelete", pszFileName))
{
// This option is used by the update feature to start the new service and delete the old one.
char version[100];
// update the version
GetMPDVersion(version, 100);
WriteMPDRegistry("version", version);
// start the new mpd
CmdStartService();
// Give the temporary mpd time to exit
Sleep(1000);
// Then delete it.
DeleteFile(pszFileName);
ExitProcess(0);
}
if (GetOpt(*argc, *argv, "-stop"))
{
CmdStopService();
ExitProcess(0);
}
if (GetOpt(*argc, *argv, "-restart", host))
{
ConnectAndRestart(argc, argv, host);
ExitProcess(0);
}
if (GetOpt(*argc, *argv, "-restart"))
{
CmdStopService();
Sleep(1000);
CmdStartService();
ExitProcess(0);
}
if (GetOpt(*argc, *argv, "-clean"))
{
CleanMPDRegistry();
ExitProcess(0);
}
if (GetOpt(*argc, *argv, "-d"))
{
char account[100]="", password[100]="", phrase[MPD_PASSPHRASE_MAX_LENGTH+1]="", pszPort[20]="";
char str_temp[10];
easy_socket_init();
CreateMPDRegistry();
if (GetOpt(*argc, *argv, "-phrase", phrase))
WriteMPDRegistry("phrase", phrase);
if (GetOpt(*argc, *argv, "-getphrase"))
{
GetPassword("passphrase for mpd: ", NULL, phrase);
WriteMPDRegistry("phrase", phrase);
}
if (GetOpt(*argc, *argv, "-port", pszPort))
{
int g_nSavedPort = g_nPort;
g_nPort = atoi(pszPort);
if (g_nPort > 0)
{
sprintf(pszPort, "%d", g_nPort);
WriteMPDRegistry("port", pszPort);
//printf("using port %d\n", g_nPort);
}
else
g_nPort = g_nSavedPort;
}
g_bSingleUser = true;
g_bStartAlone = GetOpt(*argc, *argv, "-startalone");
if (ReadMPDRegistry("SingleUser", str_temp, false))
{
if (stricmp(str_temp, "no") == 0)
{
WriteMPDRegistry("RevertToMultiUser", "yes");
}
}
WriteMPDRegistry("SingleUser", "yes");
ParseRegistry(true);
CmdDebugService(*argc, *argv);
easy_socket_finalize();
ExitProcess(0);
}
if (GetOpt(*argc, *argv, "-v") || GetOpt(*argc, *argv, "-version"))
{
/*#define USE_BAD_NONREDIRECTABLE_VERSION*/
#ifdef USE_BAD_NONREDIRECTABLE_VERSION
/* stdin, stdout, stderr don't get redirected for some reason when I launch this application
with pipes for redirecting output. If the HANDLE version of stderr is used like the code
after the else, then everything works fine. */
char str[100];
GetMPDVersion(str, 100);
fprintf(stderr, "\nMPD - mpich daemon for Microsoft Windows, version %s\n%s\n", str, COPYRIGHT);
fflush(stderr);
ExitProcess(0);
#else
HANDLE hErr = GetStdHandle(STD_ERROR_HANDLE);
char str[100], out[200];
DWORD num_written;
GetMPDVersion(str, 100);
sprintf(out, "\nMPD - mpich daemon for Microsoft Windows, version %s\n%s\n", str, COPYRIGHT);
WriteFile(hErr, out, strlen(out), &num_written, NULL);
ExitProcess(0);
#endif
}
if (GetOpt(*argc, *argv, "-h") || GetOpt(*argc, *argv, "-?") || GetOpt(*argc, *argv, "-help"))
{
char str[100];
GetMPDVersion(str, 100);
fprintf(stderr, "\nMPD - mpich daemon for Windows NT, version %s\n%s\n", str, COPYRIGHT);
fprintf(stderr, "Usage:\n mpd [ -v -h -install -remove -console ]\n\nCommand line options:\n");
fprintf(stderr, " -install \t:install the service\n -install -interact :allows the mpd to interact with the desktop\n");
fprintf(stderr, " -install -mpduser\t:install the service with single user commands enabled.\n");
fprintf(stderr, " -remove\t:remove the service\n");
fprintf(stderr, " -v\t\t:display version\n");
fprintf(stderr, " -h\t\t:this help screen\n");
fprintf(stderr, " -console\t:start a console session with the mpd on the current host\n");
fprintf(stderr, " -console host [-port x] :start a console session with the mpd on 'host:port'\n");
fprintf(stderr, " -d\t\t:run the mpd from the console\n");
ExitProcess(0);
}
}
