// 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., // 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA // // mac_saver_module.cpp // BOINC_Saver_Module // #include #include #include #include #include #include #include #include #include #include // for MAXPATHLEN #include "gui_rpc_client.h" #include "common_defs.h" #include "util.h" #include "Mac_Saver_Module.h" #include "screensaver.h" //#include // It would be nice to always display the scrolled progress info in case the // graphics application fails to show its window, but displaying the scrolled // progress info takes up too much CPU time for this to be practical. #define ALWAYS_DISPLAY_PROGRESS_TEXT 0 // Flags for testing & debugging #define CREATE_LOG 1 #define TEXTLOGOFREQUENCY 60 /* Number of times per second to update moving logo with text */ #define NOTEXTLOGOFREQUENCY 4 /* Times per second to call animateOneFrame if no moving logo with text */ #define STATUSUPDATEINTERVAL 5 /* seconds between status display updates */ #define GFX_STARTING_MSG_DURATION 45 /* seconds to show ScreenSaverAppStartingMsg */ #define STATUSRESULTCHANGETIME 10 /* seconds to show status display for each task */ #define TASK_RUN_CHECK_PERIOD 5 /* Seconds between safety check that task is actually running */ enum SaverState { SaverState_Idle, SaverState_LaunchingCoreClient, SaverState_CoreClientRunning, SaverState_RelaunchCoreClient, SaverState_ConnectedToCoreClient, SaverState_CantLaunchCoreClient, SaverState_ControlPanelTestMode, SaverState_UnrecoverableError }; static CScreensaver* gspScreensaver = NULL; extern int gGoToBlank; // True if we are to blank the screen extern int gBlankingTime; // Delay in minutes before blanking the screen extern CFStringRef gPathToBundleResources; static SaverState saverState = SaverState_Idle; // int gQuitCounter = 0; static long gSystemVersion = 0; const char * CantLaunchCCMsg = "Unable to launch BOINC application."; const char * LaunchingCCMsg = "Launching BOINC application."; const char * ConnectingCCMsg = "Connecting to BOINC application."; const char * BOINCSuspendedMsg = "BOINC is currently suspended."; const char * BOINCNoAppsExecutingMsg = "BOINC is currently idle."; const char * BOINCNoProjectsDetectedMsg = "BOINC is not attached to any projects. Please attach to projects using the BOINC Manager."; const char * BOINCNoGraphicAppsExecutingMsg = "Project does not support graphics:"; const char * BOINCUnrecoverableErrorMsg = "Sorry, an unrecoverable error occurred"; const char * BOINCTestmodeMsg = "BOINC screensaver test: success."; const char * BOINCV5GFXDaemonMsg = "BOINC can't display graphics from older applications when running as a daemon."; const char * ScreenSaverAppStartingMsg = "Starting screensaver graphics.\nPlease wait ..."; //const char * BOINCExitedSaverMode = "BOINC is no longer in screensaver mode."; // Returns desired Animation Frequency (per second) or 0 for no change int initBOINCSaver(Boolean ispreview) { if (ispreview) return 8; gspScreensaver = new CScreensaver(); return gspScreensaver->Create(); } int getSSMessage(char **theMessage, int* coveredFreq) { return gspScreensaver->getSSMessage(theMessage, coveredFreq); }; void drawPreview(CGContextRef myContext) { gspScreensaver->drawPreview(myContext); }; void closeBOINCSaver() { gspScreensaver->ShutdownSaver(); } CScreensaver::CScreensaver() { OSStatus err; m_dwBlankScreen = 0; m_dwBlankTime = 0; m_bErrorMode = false; m_hrError = 0; m_StatusMessageUpdated = false; // Display first status update after 5 seconds m_iStatusUpdateCounter = ((STATUSUPDATEINTERVAL-5) * TEXTLOGOFREQUENCY); m_iGraphicsStartingMsgCounter = 0; m_iLastResultShown = 0; m_tLastResultChangeCounter = ((STATUSRESULTCHANGETIME-5) * TEXTLOGOFREQUENCY); saverState = SaverState_Idle; m_wasAlreadyRunning = false; m_CoreClientPID = nil; m_MsgBuf[0] = 0; setSSMessageText(0); m_CurrentBannerMessage = 0; m_QuitDataManagementProc = false; m_BrandText = "BOINC"; m_updating_results = false; m_hDataManagementThread = NULL; m_hGraphicsApplication = NULL; m_bResetCoreState = TRUE; rpc = 0; err = Gestalt(gestaltSystemVersion, &gSystemVersion); if (err != noErr) { gSystemVersion = 0; } } int CScreensaver::Create() { ProcessSerialNumber psn; ProcessInfoRec pInfo; OSStatus err; // Ugly workaround for a problem with the System Preferences app // For an unknown reason, when this screensaver is run using the // Test button in the System Prefs Screensaver control panel, the // control panel calls our stopAnimation function as soon as the // science application opens a GLUT window. This problem does not // occur when the screensaver is run normally (from the screensaver // engine.) So we just display a message and don't access the core // client. GetCurrentProcess(&psn); memset(&pInfo, 0, sizeof(pInfo)); pInfo.processInfoLength = sizeof( ProcessInfoRec ); pInfo.processName = NULL; err = GetProcessInformation(&psn, &pInfo); if ( (err == noErr) && (pInfo.processSignature == 'sprf') ) { saverState = SaverState_ControlPanelTestMode; } // If there are multiple displays, initBOINCSaver may get called // multiple times (once for each display), so we need to guard // against launching multiple instances of the core client if (saverState == SaverState_Idle) { // Calculate the estimated blank time by adding the starting // time and and the user-specified time which is in minutes m_dwBlankScreen = gGoToBlank; if (gGoToBlank) m_dwBlankTime = time(0) + (gBlankingTime * 60); else m_dwBlankTime = 0; CFStringGetCString(gPathToBundleResources, m_gfx_Switcher_Path, sizeof(m_gfx_Switcher_Path), kCFStringEncodingMacRoman); strlcat(m_gfx_Switcher_Path, "/gfx_switcher", sizeof(m_gfx_Switcher_Path)); err = initBOINCApp(); CGDisplayHideCursor(kCGNullDirectDisplay); if (saverState == SaverState_LaunchingCoreClient) { SetError(FALSE, 0); m_QuitDataManagementProc = false; if (rpc == NULL) { rpc = new RPC_CLIENT; } } } return TEXTLOGOFREQUENCY; } OSStatus CScreensaver::initBOINCApp() { char boincPath[2048]; pid_t myPid; int status; OSStatus err; static int retryCount = 0; long brandId = 0; saverState = SaverState_CantLaunchCoreClient; brandId = GetBrandID(); switch(brandId) { case 1: m_BrandText = "GridRepublic"; break; default: m_BrandText = "BOINC"; break; } m_CoreClientPID = FindProcessPID("boinc", 0); if (m_CoreClientPID) { m_wasAlreadyRunning = true; saverState = SaverState_LaunchingCoreClient; return noErr; } m_wasAlreadyRunning = false; if (++retryCount > 3) // Limit to 3 relaunches to prevent thrashing return -1; err = GetpathToBOINCManagerApp(boincPath, sizeof(boincPath)); if (err) { // If we couldn't find BOINCManager.app, try default path strcpy(boincPath, "/Applications/"); if (brandId) strcat(boincPath, m_BrandText); else strcat(boincPath, "BOINCManager"); strcat(boincPath, ".app"); } strcat(boincPath, "/Contents/Resources/boinc"); if ( (myPid = fork()) < 0) return -1; else if (myPid == 0) // child { // We don't customize BOINC Data directory name for branding #if 0 // Code for separate data in each user's private directory char buf[256]; strcpy(buf, getenv("HOME")); strcat(buf, "/Library/Application Support/BOINC Data"); status = chdir(buf); #else // All users share the same data status = chdir("/Library/Application Support/BOINC Data"); #endif if (status) { perror("chdir"); fflush(NULL); _exit(status); } status = execl(boincPath, boincPath, "-redirectio", "-saver", (char *) 0); fflush(NULL); _exit(127); // execl error (execl should never return) } else { m_CoreClientPID = myPid; // make this available globally saverState = SaverState_LaunchingCoreClient; } return noErr; } // Returns new desired Animation Frequency (per second) or 0 for no change int CScreensaver::getSSMessage(char **theMessage, int* coveredFreq) { int newFrequency = TEXTLOGOFREQUENCY; *coveredFreq = 0; pid_t myPid; OSStatus err; m_iStatusUpdateCounter++; m_tLastResultChangeCounter++; switch (saverState) { case SaverState_RelaunchCoreClient: err = initBOINCApp(); break; case SaverState_LaunchingCoreClient: if (m_wasAlreadyRunning) setSSMessageText(ConnectingCCMsg); else setSSMessageText(LaunchingCCMsg); myPid = FindProcessPID(NULL, m_CoreClientPID); if (myPid) { saverState = SaverState_CoreClientRunning; rpc->init(NULL); // Initialize communications with Core Client // Set up a separate thread for communicating with Core Client CreateDataManagementThread(); // ToDo: Add a timeout after which we display error message } else // Take care of the possible race condition where the Core Client was in the // process of shutting down just as ScreenSaver started, so initBOINCApp() // found it already running but now it has shut down. if (m_wasAlreadyRunning) // If we launched it, then just wait for it to start saverState = SaverState_RelaunchCoreClient; break; case SaverState_CoreClientRunning: // RPC called in DataManagementProc() setSSMessageText(ConnectingCCMsg); if (! m_bResetCoreState) { saverState = SaverState_ConnectedToCoreClient; } break; case SaverState_ConnectedToCoreClient: switch (m_hrError) { case 0: break; // No status response yet from DataManagementProc case SCRAPPERR_SCREENSAVERBLANKED: default: setSSMessageText(0); // No text message break; case SCRAPPERR_BOINCSUSPENDED: setSSMessageText(BOINCSuspendedMsg); break; case SCRAPPERR_BOINCNOAPPSEXECUTING: setSSMessageText(BOINCNoAppsExecutingMsg); break; case SCRAPPERR_BOINCNOPROJECTSDETECTED: setSSMessageText(BOINCNoProjectsDetectedMsg); break; case SCRAPPERR_BOINCAPPFOUNDGRAPHICSLOADING: case SCRAPPERR_SCREENSAVERRUNNING: #if ! ALWAYS_DISPLAY_PROGRESS_TEXT // NOTE: My tests seem to confirm that the first window returned by NSWindowList // is always the top window under OS 10.5, but not under earlier systems. However, // Apple's documentation is unclear whether we can depend on this. So I have added // some safety by doing two things: // [1] Only use the NSWindowList test when we have started project graphics. // [2] Assume that our window is covered 45 seconds after starting project // graphics even if the NSWindowList test did not indicate that is so. // // The -animateOneFrame method in Mac_SaverModuleView.m does the NSWindowList test // only if we return a non-zero value for coveredFreq. // // Tell the calling routine to set the frame rate to NOTEXTLOGOFREQUENCY if // NSWindowList indicates that science app graphics window has covered our window. if (gSystemVersion >= 0x1050) { *coveredFreq = NOTEXTLOGOFREQUENCY; } if (m_iGraphicsStartingMsgCounter > 0) { // Show ScreenSaverAppStartingMsg for GFX_STARTING_MSG_DURATION seconds or until // NSWindowList indicates that science app graphics window has covered our window setSSMessageText(ScreenSaverAppStartingMsg); m_iGraphicsStartingMsgCounter--; } else { // Don't waste CPU cycles when the science app is drawing over our window setSSMessageText(0); // No text message } break; #endif case SCRAPPERR_BOINCNOGRAPHICSAPPSEXECUTING: case SCRAPPERR_DAEMONALLOWSNOGRAPHICS: if (m_StatusMessageUpdated) { setSSMessageText(m_MsgBuf); updateSSMessageText(m_MsgBuf); m_StatusMessageUpdated = false; } break; #if 0 case SCRAPPERR_QUITSCREENSAVERREQUESTED: // setSSMessageText(BOINCExitedSaverMode); // Wait 1 second to allow ScreenSaver engine to close us down if (++gQuitCounter > (m_MessageText[0] ? TEXTLOGOFREQUENCY : NOTEXTLOGOFREQUENCY)) { closeBOINCSaver(); KillScreenSaver(); // Stop the ScreenSaver Engine } break; #endif } // end switch (m_hrError) break; case SaverState_ControlPanelTestMode: setSSMessageText(BOINCTestmodeMsg); break; case SaverState_UnrecoverableError: setSSMessageText(BOINCUnrecoverableErrorMsg); break; case SaverState_CantLaunchCoreClient: setSSMessageText(CantLaunchCCMsg); break; case SaverState_Idle: break; // Should never get here; fixes compiler warning } // end switch (saverState) if (m_MessageText[0]) { newFrequency = TEXTLOGOFREQUENCY; } else { newFrequency = NOTEXTLOGOFREQUENCY; } *theMessage = m_MessageText; return newFrequency; } void CScreensaver::drawPreview(CGContextRef myContext) { // For possible future use } // If there are multiple displays, closeBOINCSaver may get called // multiple times (once for each display), so we need to guard // against any problems that may cause. void CScreensaver::ShutdownSaver() { DestroyDataManagementThread(); if (rpc) { #if 0 // OS X calls closeBOINCSaver() when energy saver puts display // to sleep, but we want to keep crunching. So don't kill it. // Code in core client now quits on user activity if screen // saver launched it (2/28/07). // Also, under sandbox security, screensaver doesn't have access // to rpc password in gui_rpc_auth.cfg file, so core client won't // accept rpc->quit from screensaver. if (m_CoreClientPID && (!m_wasAlreadyRunning)) { rpc->quit(); // Kill core client if we launched it } #endif delete rpc; rpc = NULL; } setSSMessageText(0); m_CoreClientPID = 0; // gQuitCounter = 0; m_wasAlreadyRunning = false; m_QuitDataManagementProc = false; saverState = SaverState_Idle; } // This function forwards to DataManagementProc, which has access to the // "this" pointer. // void * CScreensaver::DataManagementProcStub(void* param) { return gspScreensaver->DataManagementProc(); } void CScreensaver::HandleRPCError() { // Attempt to restart BOINC Client if needed, reinitialize the RPC client and state rpc->close(); // There is a possible race condition where the Core Client was in the // process of shutting down just as ScreenSaver started, so initBOINCApp() // found it already running but now it has shut down. This code takes // care of that and other situations where the Core Client quits unexpectedy. if (FindProcessPID("boinc", 0) == 0) { saverState = SaverState_RelaunchCoreClient; m_bResetCoreState = true; } rpc->init(NULL); // Otherwise just reinitialize the RPC client and state and keep trying // Error message after timeout? } bool CScreensaver::CreateDataManagementThread() { int retval; if (m_hDataManagementThread == NULL) { retval = pthread_create(&m_hDataManagementThread, NULL, DataManagementProcStub, 0); if (retval) { saverState = SaverState_UnrecoverableError; return false; } } return true; } bool CScreensaver::DestroyDataManagementThread() { m_QuitDataManagementProc = true; // Tell DataManagementProc thread to exit if (m_hDataManagementThread) { // Wait for DataManagementProc thread to exit pthread_join(m_hDataManagementThread, NULL); m_hDataManagementThread = NULL; } return true; } bool CScreensaver::SetError(bool bErrorMode, unsigned int hrError) { m_bErrorMode = bErrorMode; m_hrError = hrError; if (hrError == SCRAPPERR_BOINCAPPFOUNDGRAPHICSLOADING) { // Show ScreenSaverAppStartingMsg for GFX_STARTING_MSG_DURATION seconds m_iGraphicsStartingMsgCounter = GFX_STARTING_MSG_DURATION * TEXTLOGOFREQUENCY; } if ((hrError == SCRAPPERR_BOINCNOGRAPHICSAPPSEXECUTING) || (hrError == SCRAPPERR_DAEMONALLOWSNOGRAPHICS) #if ALWAYS_DISPLAY_PROGRESS_TEXT || (hrError == SCRAPPERR_SCREENSAVERRUNNING) #endif ) { UpdateProgressText(hrError); } return true; } void CScreensaver::UpdateProgressText(unsigned int hrError) { int iResultCount; int iIndex; int iModIndex; unsigned int len; RESULT* theResult; PROJECT* pProject; char statusBuf[2048]; if ( (m_iStatusUpdateCounter >= (STATUSUPDATEINTERVAL * TEXTLOGOFREQUENCY) ) && !m_updating_results ) { if (! m_StatusMessageUpdated) { m_iStatusUpdateCounter = 0; strcpy(m_MsgBuf, hrError == SCRAPPERR_DAEMONALLOWSNOGRAPHICS ? BOINCV5GFXDaemonMsg : BOINCNoGraphicAppsExecutingMsg ); iResultCount = results.results.size(); theResult = NULL; for (iIndex = 0; iIndex < iResultCount; iIndex++) { // cycle through the active results starting from the last one iModIndex = (iIndex + m_iLastResultShown+1) % iResultCount; theResult = results.results.at(iModIndex); // The get_state rpc is time-consuming, so we assume the list of // attached projects does not change while the screensaver is active. pProject = state.lookup_project(theResult->project_url); if (pProject != NULL) { RESULT* pResult = state.lookup_result(pProject, results.results.at(iModIndex)->name); if ( pResult != NULL ) { len = snprintf(statusBuf, sizeof(statusBuf), "\nComputing for %s\nApplication: %s\nTask: %s\n%.2f%% complete\n", pProject->project_name.c_str(), pResult->app->user_friendly_name.c_str(), pResult->wu_name.c_str(), results.results.at(iModIndex)->fraction_done*100 ); strlcat(m_MsgBuf, statusBuf, sizeof(m_MsgBuf)); if (m_tLastResultChangeCounter >= (STATUSRESULTCHANGETIME * TEXTLOGOFREQUENCY)) { m_iLastResultShown = iModIndex; m_tLastResultChangeCounter = 0; } break; } else { HandleRPCError(); return; } } else { // (pProject == NULL): re-synch with client HandleRPCError(); return; } } // end for() loop m_StatusMessageUpdated = true; } // end if (! m_StatusMessageUpdated) } // end if (m_iStatusUpdateCounter > time to update) } void CScreensaver::setSSMessageText(const char * msg) { if (msg == 0) m_MessageText[0] = 0; if (m_CurrentBannerMessage != msg) updateSSMessageText((char *)msg); } void CScreensaver::updateSSMessageText(char *msg) { char *p, *s; m_CurrentBannerMessage = msg; if (msg) { s = msg; m_MessageText[0] = '\0'; do { p = strstr(s, "BOINC"); if (p == NULL) { strcat(m_MessageText, s); } else { strncat(m_MessageText, s, p - s); strcat(m_MessageText, m_BrandText); s = p + 5; // s = p + strlen("BOINC"); } } while (p); } } int CScreensaver::GetBrandID() { char buf[1024]; long iBrandId; OSErr err; iBrandId = 0; // Default value err = GetpathToBOINCManagerApp(buf, sizeof(buf)); if (err) { // If we couldn't find our application bundle, look in BOINC Data Directory // (the installer put a copy there for us) strcpy(buf, "/Library/Application Support/BOINC Data/Branding"); } else strcat(buf, "/Contents/Resources/Branding"); FILE *f = fopen(buf, "r"); if (f) { fscanf(f, "BrandId=%ld\n", &iBrandId); fclose(f); } return iBrandId; } char * CScreensaver::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); } pid_t CScreensaver::FindProcessPID(char* name, pid_t thePID) { FILE *f; char buf[1024]; size_t n = 0; pid_t aPID; if (name != NULL) // Search ny name n = strlen(name); f = popen("ps -a -x -c -o command,pid", "r"); if (f == NULL) return 0; while (PersistentFGets(buf, sizeof(buf), f)) { if (name != NULL) { // Search ny name if (strncmp(buf, name, n) == 0) { aPID = atol(buf+16); pclose(f); return aPID; } } else { // Search by PID aPID = atol(buf+16); if (aPID == thePID) { pclose(f); return aPID; } } } pclose(f); return 0; } OSErr CScreensaver::GetpathToBOINCManagerApp(char* path, int maxLen) { CFStringRef bundleID = CFSTR("edu.berkeley.boinc"); OSType creator = 'BNC!'; FSRef theFSRef; OSStatus status = noErr; status = LSFindApplicationForInfo(creator, bundleID, NULL, &theFSRef, NULL); if (status == noErr) status = FSRefMakePath(&theFSRef, (unsigned char *)path, maxLen); return status; } // Send a Quit AppleEvent to the process which called this module // (i.e., tell the ScreenSaver engine to quit) OSErr CScreensaver::KillScreenSaver() { ProcessSerialNumber thisPSN; pid_t thisPID; OSErr err = noErr; GetCurrentProcess(&thisPSN); err = GetProcessPID(&thisPSN , &thisPID); if (err == noErr) err = kill(thisPID, SIGABRT); // SIGINT return err; } void print_to_log_file(const char *format, ...) { #if CREATE_LOG FILE *f; va_list args; char buf[256]; time_t t; strcpy(buf, getenv("HOME")); strcat(buf, "/Documents/test_log.txt"); f = fopen(buf, "a"); if (!f) return; // freopen(buf, "a", stdout); // freopen(buf, "a", stderr); time(&t); strcpy(buf, asctime(localtime(&t))); strip_cr(buf); fputs(buf, f); fputs(" ", f); va_start(args, format); vfprintf(f, format, args); va_end(args); fputs("\n", f); fflush(f); fclose(f); #endif } #if CREATE_LOG void strip_cr(char *buf) { char *theCR; theCR = strrchr(buf, '\n'); if (theCR) *theCR = '\0'; theCR = strrchr(buf, '\r'); if (theCR) *theCR = '\0'; } #endif // CREATE_LOG const char *BOINC_RCSID_7ce0778d35="$Id$";