Rooba
/
boinc
mirror of https://github.com/BOINC/boinc.git
1
0
Fork 0
Code Issues 1 Packages Projects Releases Wiki Activity
boinc/clientgui/BOINCGUIApp.cpp

891 lines
27 KiB
C++
Raw Blame History

// Berkeley Open Infrastructure for Network Computing
// http://boinc.berkeley.edu
// Copyright (C) 2005 University of California
//
// This is free software; you can redistribute it and/or
// modify it under the terms of the GNU Lesser General Public
// License as published by the Free Software Foundation;
// either version 2.1 of the License, or (at your option) any later version.
//
// This software is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
// See the GNU Lesser General Public License for more details.
//
// To view the GNU Lesser General Public License visit
// http://www.gnu.org/copyleft/lesser.html
// or write to the Free Software Foundation, Inc.,
// 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
#if defined(__GNUG__) && !defined(__APPLE__)
#pragma implementation "BOINCGUIApp.h"
#endif
#ifdef __WXMAC__
#include <Carbon/Carbon.h>
#endif
#include "stdwx.h"
#include "BOINCGUIApp.h"
#include "diagnostics.h"
#include "network.h"
#include "MainFrame.h"
#include "Events.h"
#include "MainDocument.h"
////@begin XPM images
#ifdef __APPLE__
#include "res/boinc_mac.xpm"
#include "res/gridrepublic_mac.xpm"
#else
#include "res/boinc.xpm"
#include "res/gridrepublic.xpm"
#include "res/cpdnbbc16.xpm"
#include "res/cpdnbbc32.xpm"
#include "res/cpdnbbcapwizard.xpm"
#endif
#include "res/boincsm.xpm"
#include "res/attachprojectwizard.xpm"
#include "res/gridrepublicamwizard.xpm"
////@end XPM images
#ifdef __WXMSW__
typedef BOOL (*pfnClientLibraryStartup)();
typedef void (*pfnClientLibraryShutdown)();
typedef int (*pfnBOINCIsNetworkAlive)(long* lpdwFlags);
typedef int (*pfnBOINCIsNetworkAlwaysOnline)();
typedef DWORD (*pfnBOINCGetIdleTickCount)();
#endif
IMPLEMENT_APP(CBOINCGUIApp)
IMPLEMENT_DYNAMIC_CLASS(CBOINCGUIApp, wxApp)
bool CBrandingScheme::OnInit( wxConfigBase *pConfig ) {
wxString strBaseConfigLocation = wxEmptyString;
wxInt32 iBrandId = 0;
wxASSERT(pConfig);
#ifdef __WXMAC__
wxChar buf[1024];
ProcessSerialNumber ourPSN;
FSRef ourFSRef;
OSErr err;
iBrandId = 0; // Default value
// Get the full path to core client inside this application's bundle
err = GetCurrentProcess (&ourPSN);
if (err == noErr) {
err = GetProcessBundleLocation(&ourPSN, &ourFSRef);
}
if (err == noErr) {
err = FSRefMakePath (&ourFSRef, (UInt8*)buf, sizeof(buf));
}
if (err == noErr) {
strcat(buf, "/Contents/Resources/Branding");
FILE *f = fopen(buf, "r");
if (f) {
fscanf(f, "BrandId=%d\n", &iBrandId);
fclose(f);
}
}
#else
strBaseConfigLocation = pConfig->GetPath();
pConfig->SetPath(strBaseConfigLocation + wxT("Branding"));
pConfig->Read(wxT("BrandId"), &iBrandId, 0);
pConfig->SetPath(strBaseConfigLocation);
#endif
// If the BrandId is greater than 0 then we are running in
// branded mode
if (iBrandId) {
m_bIsBranded = true;
} else {
m_bIsBranded = false; // wxWidgets automatically sets bools to
// true by default.
}
// Now determine which resources are needed for each BrandId
switch (iBrandId) {
case 1:
// Running as a GridRepublic client.
m_strApplicationName = wxT("GridRepublic Manager");
m_iconApplicationIcon = wxIcon(gridrepublic_xpm);
m_bitmapApplicationLogo = wxBitmap(gridrepublic_xpm);
m_strCompanyName = wxT("GridRepublic");
m_strCompanyWebsite = wxT("http://www.gridrepublic.com/");
m_strProjectName = wxT("GridRepublic");
m_strAPWizardTitle = wxEmptyString;
m_bDefaultTabSpecified = false;
m_iDefaultTab = 0;
m_bitmapAPWizardLogo = wxBitmap(gridrepublicamwizard_xpm);
m_strAPWizardAccountInfoText = wxEmptyString;
m_strAPWizardCompletionTitle = wxEmptyString;
m_strAPWizardCompletionBrandedMessage = wxEmptyString;
m_strAPWizardCompletionMessage = wxEmptyString;
m_strAMWizardTitle = wxEmptyString;
m_bitmapAMWizardLogo = wxBitmap(gridrepublicamwizard_xpm);
m_strAMWizardAttachMessage = wxEmptyString;
m_strExitMessage = wxEmptyString;
break;
#ifndef __APPLE__
case 2:
// Running as a CPDNBBC client.
m_strApplicationName = wxT("The BOINC Manager for the BBC Climate Change Experiment");
m_iconApplicationIcon = wxIcon(cpdnbbc16_xpm);
m_bitmapApplicationLogo = wxBitmap(cpdnbbc32_xpm);
m_strCompanyName = wxT("ClimatePrediction.net");
m_strCompanyWebsite = wxT("http://bbc.cpdn.org/beta/");
m_strProjectName = wxT("BBC Climate Change Experiment");
m_bDefaultTabSpecified = true;
m_iDefaultTab = ID_LIST_WORKVIEW - ID_LIST_BASE;
m_strAPWizardTitle = _("Start Project");
m_bitmapAPWizardLogo = wxBitmap(cpdnbbcapwizard_xpm);
m_strAPWizardAccountInfoText =
_("Please enter an email address and password.\n"
"\n"
"You will need your email address if you want to change your account\n"
"options or use our message boards.\n"
"\n"
"We will send you occasional emails. You can stop these at aynt time.\n"
"We will not pass your email address on to others.\n");
m_strAPWizardCompletionTitle =
_("Project Started");
m_strAPWizardCompletionBrandedMessage =
_("Congratulations, you have now successfully started your Climate\n"
"Change Experiment.\n");
m_strAPWizardCompletionMessage =
_("Click finish to exit. You will be taken to a web page which tells\n"
"you more about your model.\n");
m_strAMWizardTitle = wxEmptyString;
m_bitmapAMWizardLogo = wxBitmap(cpdnbbcapwizard_xpm);
m_strAMWizardAttachMessage = wxEmptyString;
m_strExitMessage =
_("This will shut down your experiment until it restarts automatically\n"
"following your user preferences. Close window to close the manager\n"
"without stopping the experiment.");
break;
#endif __APPLE__
default:
// Running in native mode without any branding
m_strApplicationName = wxT("BOINC Manager");
m_iconApplicationIcon = wxIcon(boinc_xpm);
m_bitmapApplicationLogo = wxBitmap(boincsm_xpm);
m_strCompanyName = wxT("Space Sciences Laboratory, U.C. Berkeley");
m_strCompanyWebsite = wxT("http://boinc.berkeley.edu/");
m_strProjectName = wxT("BOINC");
m_bDefaultTabSpecified = false;
m_iDefaultTab = 0;
m_strAPWizardTitle = wxEmptyString;
m_bitmapAPWizardLogo = wxBitmap(attachprojectwizard_xpm);
m_strAPWizardCompletionTitle = wxEmptyString;
m_strAPWizardCompletionBrandedMessage = wxEmptyString;
m_strAPWizardCompletionMessage = wxEmptyString;
m_strAMWizardTitle = wxEmptyString;
m_strAPWizardAccountInfoText = wxEmptyString;
m_bitmapAMWizardLogo = wxBitmap(attachprojectwizard_xpm);
m_strAMWizardAttachMessage = wxEmptyString;
m_strExitMessage = wxEmptyString;
break;
}
return true;
}
bool CBOINCGUIApp::OnInit() {
// Setup variables with default values
m_bBOINCStartedByManager = false;
m_bFrameVisible = true;
m_lBOINCCoreProcessId = 0;
#ifdef __WXMSW__
m_hBOINCCoreProcess = NULL;
m_hClientLibraryDll = NULL;
#endif
m_strDefaultWindowStation = wxT("");
m_strDefaultDesktop = wxT("");
m_strDefaultDisplay = wxT("");
// Setup application and company information
SetAppName(wxT("BOINC Manager"));
SetVendorName(wxT("Space Sciences Laboratory, U.C. Berkeley"));
// Initialize the configuration storage module
m_pConfig = new wxConfig(GetAppName());
wxConfigBase::Set(m_pConfig);
wxASSERT(m_pConfig);
m_pConfig->SetPath(wxT("/"));
#ifdef __WXMSW__
TCHAR szPath[MAX_PATH-1];
// change the current directory to the boinc install directory
GetModuleFileName(NULL, szPath, (sizeof(szPath)/sizeof(TCHAR)));
TCHAR *pszProg = strrchr(szPath, '\\');
if (pszProg) {
szPath[pszProg - szPath + 1] = 0;
SetCurrentDirectory(szPath);
}
#endif
// Setup the branding scheme
// NOTE: SetAppName(), SetVendorName(), and the creation of the m_pConfig
// variable need to be done before the branding object can be used on
// any platform other than the Mac.
m_pBranding = new CBrandingScheme;
wxASSERT(m_pBranding);
m_pBranding->OnInit(m_pConfig);
#ifdef __WXMAC__
wxString strDirectory = wxEmptyString;
bool success;
// Set the current directory ahead of the application launch so the core
// client can find its files
#if 0 // Code for separate data in each user's private directory
wxChar buf[1024];
wxExpandPath(buf, "~/Library/Application Support");
strDirectory = wxT(buf);
#else // All users share the same data
// The mac installer sets the "setuid & setgid" bits for the
// BOINC Manager and core client so any user can run them and
// they can operate on shared data.
strDirectory = wxT("/Library/Application Support/");
#endif
success = ::wxSetWorkingDirectory(strDirectory);
if (success) {
// If SetWD failed, don't create a directory in wrong place
strDirectory += m_pBranding->GetProjectName();
strDirectory += wxT(" Data");
if (! wxDirExists(strDirectory))
success = wxMkdir(m_pBranding->GetProjectName() + wxT(" Data"), 0777); // Does nothing if dir exists
success = ::wxSetWorkingDirectory(strDirectory);
// wxChar *wd = wxGetWorkingDirectory(buf, 1000); // For debugging
}
#endif // __WXMAC__
// Initialize the BOINC Diagnostics Framework
int dwDiagnosticsFlags =
BOINC_DIAG_DUMPCALLSTACKENABLED |
BOINC_DIAG_HEAPCHECKENABLED |
BOINC_DIAG_MEMORYLEAKCHECKENABLED |
#if defined(__WXMSW__) || defined(__WXMAC__)
BOINC_DIAG_REDIRECTSTDERR |
BOINC_DIAG_REDIRECTSTDOUT |
#endif
BOINC_DIAG_TRACETOSTDOUT;
diagnostics_init(
dwDiagnosticsFlags,
"stdoutgui",
"stderrgui"
);
// Enable Logging and Trace Masks
m_pLog = new wxLogBOINC();
wxLog::SetActiveTarget(m_pLog);
m_pLog->AddTraceMask(wxT("Function Start/End"));
m_pLog->AddTraceMask(wxT("Function Status"));
// Enable known image types
wxImage::AddHandler(new wxXPMHandler);
// Initialize the internationalization module
m_pLocale = new wxLocale();
wxASSERT(m_pLocale);
wxInt32 iSelectedLanguage = m_pConfig->Read(wxT("Language"), 0L);
// Locale information is stored relative to the executable.
m_pLocale->Init(iSelectedLanguage);
m_pLocale->AddCatalogLookupPathPrefix(wxT("locale"));
m_pLocale->AddCatalog(wxT("BOINC Manager"));
InitSupportedLanguages();
// Note: JAWS for Windows will only speak the context-sensitive
// help if you use this help provider:
wxHelpProvider::Set(new wxHelpControllerHelpProvider());
// Commandline parsing is done in wxApp::OnInit()
if (!wxApp::OnInit()) {
return false;
}
// Initialize the main document
m_pDocument = new CMainDocument();
wxASSERT(m_pDocument);
m_pDocument->OnInit();
// Initialize the main gui window
m_pFrame = new CMainFrame(
m_pBranding->GetApplicationName(),
m_pBranding->GetApplicationIcon()
);
wxASSERT(m_pFrame);
// Initialize the task bar icon
#if defined(__WXMSW__) || defined(__WXMAC__)
m_pTaskBarIcon = new CTaskBarIcon(
m_pBranding->GetApplicationName(),
m_pBranding->GetApplicationIcon()
);
wxASSERT(m_pTaskBarIcon);
#endif
#ifdef __WXMAC__
m_pMacSystemMenu = new CMacSystemMenu(
m_pBranding->GetApplicationName(),
m_pBranding->GetApplicationIcon()
);
wxASSERT(m_pMacSystemMenu);
#endif
// Detect the display info and store for later use.
DetectDisplayInfo();
// Startup the System Idle Detection code
ClientLibraryStartup();
// Detect if we need to start the BOINC Core Client due to configuration
StartupBOINCCore();
#ifdef __WXMAC__
ProcessSerialNumber psn;
ProcessInfoRec pInfo;
OSStatus err;
GetCurrentProcess(&psn);
memset(&pInfo, 0, sizeof(pInfo));
pInfo.processInfoLength = sizeof( ProcessInfoRec );
err = GetProcessInformation(&psn, &pInfo);
if (!err) {
psn = pInfo.processLauncher;
memset(&pInfo, 0, sizeof(pInfo));
pInfo.processInfoLength = sizeof( ProcessInfoRec );
err = GetProcessInformation(&psn, &pInfo);
}
// Don't open main window if we were started automatically at login
if (pInfo.processSignature == 'lgnw') { // Login Window app
m_bFrameVisible = false;
// If the system was just started, we usually get a "Connection
// failed" error if we try to connect too soon, so delay a bit.
sleep(10);
}
#endif
// Show the UI
SetTopWindow(m_pFrame);
if (m_bFrameVisible) {
m_pFrame->Show();
} else {
#ifdef __WXMAC__
GetCurrentProcess(&psn);
ShowHideProcess(&psn, false);
#else
m_pFrame->Show();
m_pFrame->Show(false);
#endif
}
return true;
}
int CBOINCGUIApp::OnExit() {
// Detect if we need to stop the BOINC Core Client due to configuration
ShutdownBOINCCore();
// Shutdown the System Idle Detection code
ClientLibraryShutdown();
#if defined(__WXMSW__) || defined(__WXMAC__)
if (m_pTaskBarIcon) {
delete m_pTaskBarIcon;
}
#endif
#ifdef __WXMAC__
if (m_pMacSystemMenu) {
delete m_pMacSystemMenu;
}
#endif
if (m_pDocument) {
m_pDocument->OnExit();
delete m_pDocument;
}
if (m_pBranding) {
delete m_pBranding;
}
if (m_pLocale) {
delete m_pLocale;
}
return wxApp::OnExit();
}
void CBOINCGUIApp::OnInitCmdLine(wxCmdLineParser &parser) {
wxApp::OnInitCmdLine(parser);
static const wxCmdLineEntryDesc cmdLineDesc[] = {
{ wxCMD_LINE_SWITCH, wxT("s"), wxT("systray"), _("Startup BOINC so only the system tray icon is visible")},
{ wxCMD_LINE_NONE} //DON'T forget this line!!
};
parser.SetDesc(cmdLineDesc);
}
bool CBOINCGUIApp::OnCmdLineParsed(wxCmdLineParser &parser) {
// Give default processing (-?, --help and --verbose) the chance to do something.
wxApp::OnCmdLineParsed(parser);
if (parser.Found(wxT("systray"))) {
m_bFrameVisible = false;
}
return true;
}
void CBOINCGUIApp::DetectDisplayInfo() {
#ifdef __WXMSW__
wxChar szWindowStation[256];
memset(szWindowStation, 0, sizeof(szWindowStation)/sizeof(wxChar));
wxChar szDesktop[256];
memset(szDesktop, 0, sizeof(szDesktop)/sizeof(wxChar));
if (wxWIN95 != wxGetOsVersion(NULL, NULL)) {
// Retrieve the current window station and desktop names
GetUserObjectInformation(
GetProcessWindowStation(),
UOI_NAME,
szWindowStation,
(sizeof(szWindowStation) / sizeof(wxChar)),
NULL
);
GetUserObjectInformation(
GetThreadDesktop(GetCurrentThreadId()),
UOI_NAME,
szDesktop,
(sizeof(szDesktop) / sizeof(wxChar)),
NULL
);
m_strDefaultWindowStation = szWindowStation;
m_strDefaultDesktop = szDesktop;
}
#else
char* p = getenv("DISPLAY");
if (p) m_strDefaultDisplay = p;
#endif
}
void CBOINCGUIApp::InitSupportedLanguages() {
wxInt32 iIndex = 0;
const wxLanguageInfo* liLanguage = NULL;
// Prepare the array
m_astrLanguages.Insert(wxEmptyString, 0, wxLANGUAGE_USER_DEFINED+1);
// These are just special tags so deal with them in a special way
m_astrLanguages[wxLANGUAGE_DEFAULT] = _("(Automatic Detection)");
m_astrLanguages[wxLANGUAGE_UNKNOWN] = _("(Unknown)");
m_astrLanguages[wxLANGUAGE_USER_DEFINED] = _("(User Defined)");
for (iIndex = 0; iIndex <= wxLANGUAGE_USER_DEFINED; iIndex++) {
liLanguage = wxLocale::GetLanguageInfo(iIndex);
if (liLanguage) {
m_astrLanguages[iIndex] = liLanguage->Description;
}
}
}
bool CBOINCGUIApp::IsBOINCCoreRunning() {
wxLogTrace(wxT("Function Start/End"), wxT("CBOINCGUIApp::IsBOINCCoreRunning - Function Begin"));
int retval;
bool running;
RPC_CLIENT rpc;
retval = rpc.init("localhost"); // synchronous is OK since local
running = (retval == 0);
rpc.close();
wxLogTrace(wxT("Function Start/End"), wxT("CBOINCGUIApp::IsBOINCCoreRunning - Function End"));
return running;
}
void CBOINCGUIApp::StartupBOINCCore() {
if (!IsBOINCCoreRunning()) {
#ifndef __WXMAC__
wxString strDirectory = wxEmptyString;
#endif // ! __WXMAC__
wxString strExecute = wxEmptyString;
wxChar szExecutableDirectory[4096];
memset(szExecutableDirectory, 0, sizeof(szExecutableDirectory));
#ifdef __WXMSW__
// On the surface it would seem that GetCurrentDirectory would be a better choice
// for determing which directory we should prepend to the execution string before
// starting BOINC, except that we cannot depend on any shortcuts being configured
// to startup in the correct directory, since the user may have created the
// shortcut themselves. So determine where boinc.exe is based off of our
// current execution location and then execute it.
GetModuleFileName(
NULL,
szExecutableDirectory,
(sizeof(szExecutableDirectory) / sizeof(wxChar))
);
#endif
#ifdef __WXMAC__
{
wxChar buf[1024];
wxChar *argv[3];
ProcessSerialNumber ourPSN;
FSRef ourFSRef;
OSErr err;
// Get the full path to core client inside this application's bundle
err = GetCurrentProcess (&ourPSN);
if (err == noErr) {
err = GetProcessBundleLocation(&ourPSN, &ourFSRef);
}
if (err == noErr) {
err = FSRefMakePath (&ourFSRef, (UInt8*)buf, sizeof(buf));
}
if (err == noErr) {
#if 0 // The Mac version of wxExecute(wxString& ...) crashes if there is a space in the path
strExecute = wxT("\"");
strExecute += wxT(buf);
strExecute += wxT("/Contents/Resources/boinc\" -redirectio");
m_lBOINCCoreProcessId = ::wxExecute(strExecute);
#else // Use wxExecute(wxChar **argv ...) instead of wxExecute(wxString& ...)
strcat(buf, "/Contents/Resources/boinc");
argv[0] = buf;
argv[1] = "-redirectio";
argv[2] = NULL;
m_lBOINCCoreProcessId = ::wxExecute(argv);
#endif
} else {
buf[0] = '\0';
}
}
#else // ! __WXMAC__
#ifndef __WXMSW__
// copy the path to the boinmgr from argv[0]
strncpy(szExecutableDirectory, wxGetApp().argv[0], sizeof(szExecutableDirectory));
#endif
// We are only interested in the path component of the fully qualified path.
wxFileName::SplitPath(szExecutableDirectory, &strDirectory, NULL, NULL);
// Set the current directory ahead of the application launch so the core
// client can find its files
::wxSetWorkingDirectory(strDirectory);
#endif // ! __WXMAC__
#ifdef __WXMSW__
// Append boinc.exe to the end of the strExecute string and get ready to rock
strExecute = wxT("\"") + strDirectory + wxT("\\boinc.exe\" -redirectio");
PROCESS_INFORMATION pi;
STARTUPINFO si;
BOOL bProcessStarted;
memset(&pi, 0, sizeof(pi));
memset(&si, 0, sizeof(si));
si.cb = sizeof(si);
si.dwFlags = STARTF_USESHOWWINDOW;
si.wShowWindow = SW_HIDE;
bProcessStarted = CreateProcess(
NULL,
(LPTSTR)strExecute.c_str(),
NULL,
NULL,
FALSE,
CREATE_NEW_PROCESS_GROUP|CREATE_NO_WINDOW,
NULL,
(LPTSTR)strDirectory.c_str(),
&si,
&pi
);
if (bProcessStarted) {
m_lBOINCCoreProcessId = pi.dwProcessId;
m_hBOINCCoreProcess = pi.hProcess;
}
#else
#ifndef __WXMAC__
// Append boinc.exe to the end of the strExecute string and get ready to rock
strExecute = wxT("./boinc -redirectio");
m_lBOINCCoreProcessId = ::wxExecute(strExecute);
#endif // ! __WXMAC__
#endif // ! __WXMSW__
if (0 != m_lBOINCCoreProcessId) {
m_bBOINCStartedByManager = true;
}
}
}
#if defined(__WXMSW__)
void CBOINCGUIApp::ShutdownBOINCCore() {
wxInt32 iCount = 0;
bool bClientQuit = false;
DWORD dwExitCode;
wxString strMachineName = wxT("localhost");
if (m_bBOINCStartedByManager) {
// The user may have gone off to look at another machine on the network, and
// we don't want to leave any dangling processes if we started them up.
m_pDocument->Connect(strMachineName);
if (GetExitCodeProcess(m_hBOINCCoreProcess, &dwExitCode)) {
if (STILL_ACTIVE == dwExitCode) {
m_pDocument->CoreClientQuit();
for (iCount = 0; iCount <= 10; iCount++) {
if (!bClientQuit && GetExitCodeProcess(m_hBOINCCoreProcess, &dwExitCode)) {
if (STILL_ACTIVE != dwExitCode) {
bClientQuit = true;
continue;
}
}
::Sleep(1);
}
}
}
if (!bClientQuit) {
::wxKill(m_lBOINCCoreProcessId);
}
}
}
#elif defined(__WXMAC__)
static char * PersistentFGets(char *buf, size_t buflen, FILE *f) {
char *p = buf;
size_t len = buflen;
size_t datalen = 0;
*buf = '\0';
while (datalen < (buflen - 1)) {
fgets(p, len, f);
if (feof(f)) break;
if (ferror(f) && (errno != EINTR)) break;
if (strchr(buf, '\n')) break;
datalen = strlen(buf);
p = buf + datalen;
len -= datalen;
}
return (buf[0] ? buf : NULL);
}
bool CBOINCGUIApp::ProcessExists(pid_t thePID)
{
FILE *f;
char buf[256];
pid_t aPID;
f = popen("ps -a -x -c -o pid,state", "r");
if (f == NULL)
return false;
while (PersistentFGets(buf, sizeof(buf), f)) {
aPID = atol(buf);
if (aPID == thePID) {
if (strchr(buf, 'Z')) // A 'zombie', stopped but waiting
break; // for us (its parent) to quit
pclose(f);
return true;
}
}
pclose(f);
return false;
}
// wxProcess::Exists and wxKill are unimplemented in WxMac-2.6.0
void CBOINCGUIApp::ShutdownBOINCCore() {
wxInt32 iCount = 0;
wxString strMachineName = wxT("localhost");
if (m_bBOINCStartedByManager) {
// The user may have gone off to look at another machine on the network, and
// we don't want to leave any dangling processes if we started them up.
m_pDocument->Connect(strMachineName);
if (ProcessExists(m_lBOINCCoreProcessId)) {
m_pDocument->CoreClientQuit();
for (iCount = 0; iCount <= 10; iCount++) {
if (!ProcessExists(m_lBOINCCoreProcessId))
return;
::wxSleep(1);
}
}
// Client did not quit after 10 seconds so kill it
kill(m_lBOINCCoreProcessId, SIGKILL);
}
}
#else
void CBOINCGUIApp::ShutdownBOINCCore() {
wxInt32 iCount = 0;
bool bClientQuit = false;
wxString strMachineName = wxT("localhost");
if (m_bBOINCStartedByManager) {
// The user may have gone off to look at another machine on the network, and
// we don't want to leave any dangling processes if we started them up.
m_pDocument->Connect(strMachineName);
if (wxProcess::Exists(m_lBOINCCoreProcessId)) {
m_pDocument->CoreClientQuit();
for (iCount = 0; iCount <= 10; iCount++) {
if (!bClientQuit && !wxProcess::Exists(m_lBOINCCoreProcessId)) {
bClientQuit = true;
continue;
}
::wxSleep(1);
}
}
if (!bClientQuit) {
::wxKill(m_lBOINCCoreProcessId);
}
}
}
#endif
int CBOINCGUIApp::ClientLibraryStartup() {
#ifdef __WXMSW__
// load dll and start idle detection
m_hClientLibraryDll = LoadLibrary("boinc.dll");
if(m_hClientLibraryDll) {
pfnClientLibraryStartup fn;
fn = (pfnClientLibraryStartup)GetProcAddress(m_hClientLibraryDll, wxT("ClientLibraryStartup"));
if(!fn) {
FreeLibrary(m_hClientLibraryDll);
m_hClientLibraryDll = NULL;
return -1;
} else {
if(!fn()) {
FreeLibrary(m_hClientLibraryDll);
m_hClientLibraryDll = NULL;
return -1;
}
}
}
#endif
return 0;
}
int CBOINCGUIApp::ClientLibraryShutdown() {
#ifdef __WXMSW__
if(m_hClientLibraryDll) {
pfnClientLibraryShutdown fn;
fn = (pfnClientLibraryShutdown)GetProcAddress(m_hClientLibraryDll, wxT("ClientLibraryShutdown"));
if(fn) {
fn();
} else {
return -1;
}
FreeLibrary(m_hClientLibraryDll);
m_hClientLibraryDll = NULL;
}
#endif
return 0;
}
int CBOINCGUIApp::IsNetworkAlive(long* lpdwFlags) {
#ifdef __WXMSW__
if(m_hClientLibraryDll) {
pfnBOINCIsNetworkAlive fn;
fn = (pfnBOINCIsNetworkAlive)GetProcAddress(m_hClientLibraryDll, wxT("BOINCIsNetworkAlive"));
if(fn) {
return fn(lpdwFlags);
} else {
return -1;
}
}
#endif
return TRUE;
}
int CBOINCGUIApp::IsNetworkAlwaysOnline() {
#ifdef __WXMSW__
if(m_hClientLibraryDll) {
pfnBOINCIsNetworkAlwaysOnline fn;
fn = (pfnBOINCIsNetworkAlwaysOnline)GetProcAddress(m_hClientLibraryDll, wxT("BOINCIsNetworkAlwaysOnline"));
if(fn) {
return fn();
} else {
return -1;
}
}
#endif
return TRUE;
}
int CBOINCGUIApp::UpdateSystemIdleDetection() {
#ifdef __WXMSW__
if (m_hClientLibraryDll) {
pfnBOINCGetIdleTickCount fn;
fn = (pfnBOINCGetIdleTickCount)GetProcAddress(m_hClientLibraryDll, wxT("BOINCGetIdleTickCount"));
if(fn) {
fn();
} else {
return -1;
}
}
#endif
return 0;
}
const char *BOINC_RCSID_487cbf3018 = "$Id$";
Reference in New Issue View Git Blame Copy Permalink