// This file is part of BOINC.
// http://boinc.berkeley.edu
// Copyright (C) 2008 University of California
//
// BOINC 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 3 of the License, or (at your option) any later version.
//
// BOINC 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.
//
// You should have received a copy of the GNU Lesser General Public License
// along with BOINC. If not, see .
/* CustomInstall.cpp */
/* Customizable installer to allow use of features described at
http://boinc.berkeley.edu/client_startup.php
Directions for creating a customized installer for the Macintosh:
[1] Create a directory, with a name such as SETI@home_Mac_Installer
[2] Place this CustomInstall application inside that directory. You
may rename this application if you wish.
[3] Create a new directory named "boinc_startup_files" inside the
directory created in step 1.
[4] Place the custom .xml files in the "boinc_startup_files" directory.
[5] Zip the directory created in step 1 (with all its contents).
*/
#define CREATE_LOG 1 /* for debugging */
#include
#include
#include
#include "error_numbers.h"
#include "filesys.h"
#include "str_util.h"
#ifdef __cplusplus
extern "C" {
#endif
static void Initialize(void); /* function prototypes */
static OSStatus download_thread(void* param);
static OSErr download_boinc(FILE * out_file);
//static curl_progress_callback show_prog;
static int show_prog(void *clientp, double dltotal, double dlnow,
double ultotal, double ulnow);
static OSErr runProgressDlog();
static pascal Boolean ProgDlgFilterProc(DialogPtr dp, EventRecord *event, short *item);
static void show_message(StringPtr s1);
static OSErr QuitAppleEventHandler(const AppleEvent *appleEvt, AppleEvent* reply, UInt32 refcon);
static void print_to_log_file(const char *format, ...);
static void strip_cr(char *buf);
#ifdef __cplusplus
}
#endif
/* globals */
Boolean gQuitFlag = false;
short gProgressValue = 0, gOldProgressValue = 0;
char gCurlError[CURL_ERROR_SIZE];
char * gDL_URL = "http://boinc.berkeley.edu/dl/boinc_5.2.1_macOSX.zip";
char * gDownLoadFileName = "boinc_installer.zip";
char * gCustomDirectoryName = "boinc_startup_files";
MPQueueID gTerminationQueue; /* This queue will report the completion of our threads. */
MPTaskID gDownload_thread_id; /* IDs of the threads we create. */
CURLcode gResult;
DialogPtr gProgressDlog;
int main(int argc, char *argv[])
{
FILE * boinc_installer_zip_file;
char path[1024], buf[256], *p;
DIRREF dirp;
OSStatus err;
int retval;
Initialize();
#ifdef __APPLE__
if (strlen(argv[0]) >= sizeof(path)) {
show_message((StringPtr)"\pPath to application is too long.");
return 0;
}
safe_strcpy(path, argv[0]); // Path to this application.
p = strstr(path, "/Contents/MacOS/");
*p = '\0';
p = strrchr(path, '/');
*(p+1) = '\0';
chdir(path); // Directory containing this application
#if 0 // For testing under debugger
chdir("/Users/charliefenton/Desktop/FTPTEST/");
getcwd(path, 256);
#endif
#else // Not __APPLE__
getcwd(path, 256); // Directory containing this application
#endif
// we use multiple threads to be able to run our progress dialogs during
// copy and search operations
if (!MPLibraryIsLoaded()) {
printf("MultiProcessing Library not available.\n");
ExitToShell();
}
err = MPCreateQueue(&gTerminationQueue); /* Create the queue which will report the completion of the task. */
if (err != noErr) {
printf("Cannot create the termination queue. err = %d\n", (short)err);
ExitToShell();
}
boinc_installer_zip_file = boinc_fopen(gDownLoadFileName, "w");
if (boinc_installer_zip_file == NULL)
{
show_message((StringPtr)"\pFailed to create zip file.");
return -1;
}
err = MPCreateTask(download_thread, /* This is the task function. */
boinc_installer_zip_file, /* This is the parameter to the task function. */
(500*1024), /* Stack size for thread. */
gTerminationQueue, /* We'll use this to sense task completion. */
0, /* We won't use the first part of the termination message. */
0, /* We won't use the second part of the termination message. */
0, /* Use the normal task options. (Currently this *must* be zero!) */
&gDownload_thread_id); /* Here's where the ID of the new task will go. */
if (err != noErr) {
(void) MPDeleteQueue(gTerminationQueue);
printf("Cannot create the copy thread. err = %d\n", (short)err);
ExitToShell();
}
err = runProgressDlog();
fclose(boinc_installer_zip_file);
if (gQuitFlag)
{
show_message((StringPtr)"\pCancelled by user.");
return 0;
}
if (gResult)
{
buf[0] = sprintf(buf+1, "Download error %d:\n%s", gResult, curl_easy_strerror(gResult));
show_message((StringPtr)buf);
return 0;
}
sprintf(buf, "unzip -o %s", gDownLoadFileName);
retval = system(buf);
if (retval)
{
show_message((StringPtr)"\pError expanding downloaded zip file.");
return 0;
}
// Copy the custom XML files
#ifdef __APPLE__
// On the Macintosh, BOINC puts its data at a fixed, predetermined path
// so we can just copy the custom files there. On other platforms, this
// application should copy the custom files to a temporary, intermediate
// location which the standard installer can then find; the standard
// installer should then copy the files to the correct directory and
// possibly delete the temporary ones.
retval = system("mkdir -p /Library/Application\\ Support/BOINC\\ Data");
if (!retval)
{
sprintf(buf, "chmod 0644 .%s%s%s*.xml", PATH_SEPARATOR, gCustomDirectoryName, PATH_SEPARATOR);
retval = system (buf);
}
if (!retval)
{
sprintf(buf, "cp -f .%s%s%s*.xml /Library/Application\\ Support/BOINC\\ Data/",
PATH_SEPARATOR, gCustomDirectoryName, PATH_SEPARATOR);
retval = system(buf);
}
#endif
if (retval)
{
show_message((StringPtr)"\pCouldn't copy custom BOINC startup files.");
retval = 0; // Should we continue anyway?
}
// Search this directory for the expanded BOINC installer directory; it
// should be the only directory other than the custom files directory.
sprintf(buf, ".%s", PATH_SEPARATOR);
dirp = dir_open(path);
if (dirp == NULL)
{
retval = ERR_OPENDIR;
} else {
do {
retval = dir_scan(buf+2, dirp, sizeof(buf)-2);
if (!is_dir(buf))
continue;
if (strcmp(buf+2, gCustomDirectoryName))
break;
} while (retval == BOINC_SUCCESS);
}
if (retval)
{
show_message((StringPtr)"\pCouldn't find downloaded additional BOINC installer software.");
return 0;
}
// Run the installer
retval = chdir(buf);
if (!retval)
retval = system("open ./BOINC.pkg");
if (retval)
show_message((StringPtr)"\pError running additional BOINC installer software.");
return 0;
}
static void Initialize() /* Initialize some managers */
{
OSErr err;
InitCursor();
err = AEInstallEventHandler( kCoreEventClass, kAEQuitApplication, NewAEEventHandlerUPP((AEEventHandlerProcPtr)QuitAppleEventHandler), 0, false );
if (err != noErr)
ExitToShell();
}
////////////////////////////////////////////////////////////////////////
// //
// CAUTION - download_thread is a second MPThread, so all Mac //
// API calls must be thread-safe! //
// //
////////////////////////////////////////////////////////////////////////
static OSStatus download_thread(void* param)
{
return download_boinc((FILE *)param);
}
static OSErr download_boinc(FILE * out_file)
{
CURL *myHandle;
CURLcode curlErr;
curlErr = curl_global_init(0);
myHandle = curl_easy_init();
if (myHandle == NULL)
{
show_message((StringPtr)"\pDownload initialization error.");
return -1;
}
curlErr = curl_easy_setopt(myHandle, CURLOPT_VERBOSE, 1);
curlErr = curl_easy_setopt(myHandle, CURLOPT_NOPROGRESS, 0);
curlErr = curl_easy_setopt(myHandle, CURLOPT_WRITEDATA, out_file);
curlErr = curl_easy_setopt(myHandle, CURLOPT_PROGRESSFUNCTION, show_prog);
curlErr = curl_easy_setopt(myHandle, CURLOPT_PROGRESSDATA, &gProgressValue);
curlErr = curl_easy_setopt(myHandle, CURLOPT_ERRORBUFFER, gCurlError);
curlErr = curl_easy_setopt(myHandle, CURLOPT_URL, gDL_URL);
sleep(2); // Give progress dialog a chance to be drawn
curlErr = curl_easy_perform(myHandle);
curl_easy_cleanup(myHandle);
curl_global_cleanup();
return curlErr;
}
static int show_prog(void *clientp, double dltotal, double dlnow,
double ultotal, double ulnow)
{
gProgressValue = (short)(dlnow * 100.0 / dltotal);
printf("Progress = %d%%\n", gProgressValue);
return gQuitFlag;
}
static OSErr runProgressDlog()
{
AlertStdAlertParamRec alertParams;
DialogItemIndex itemHit = 0;
ModalFilterUPP myFilterProcUPP = NewModalFilterUPP(ProgDlgFilterProc);
OSStatus err;
gOldProgressValue = -1;
alertParams.movable = false;
alertParams.helpButton = false;
alertParams.filterProc = myFilterProcUPP;
alertParams.defaultText = "\pCancel";
alertParams.cancelText = NULL;
alertParams.otherText = NULL;
alertParams.defaultButton = kAlertStdAlertOKButton;
alertParams.cancelButton = 0;
alertParams.position = kWindowDefaultPosition;
ParamText("\pDownloading additional BOINC installer software:\n 0% complete", "\p", "\p", "\p");
err = StandardAlert(kAlertNoteAlert, "\p^0", NULL, &alertParams, &itemHit);
if (itemHit == kAlertStdAlertOKButton)
gQuitFlag = true;
return noErr;
}
static pascal Boolean ProgDlgFilterProc(DialogPtr dp, EventRecord *event, short *item)
{
char buf[256];
OSStatus err;
if (gTerminationQueue)
{
err = MPWaitOnQueue(gTerminationQueue, 0, 0, (void **)&gResult, kDurationImmediate);
if (err == noErr)
{
*item = kAlertStdAlertCancelButton;
return true;
}
}
if (gProgressValue == gOldProgressValue)
return false;
gOldProgressValue = gProgressValue;
buf[0] = sprintf(buf+1, "Downloading additional BOINC installer software:\n %3d%% complete", gProgressValue);
ParamText("\pDownloading additional BOINC installer software: 0% complete", "\p", "\p", "\p");
ParamText((StringPtr)buf, "\p", "\p", "\p");
DrawDialog(dp);
return false;
}
/********************************************************************
ShowMessage
********************************************************************/
static void show_message(StringPtr s1)
{
DialogItemIndex itemHit;
OSErr err;
err = StandardAlert (kAlertStopAlert, s1, NULL, NULL, &itemHit);
}
static OSErr QuitAppleEventHandler( const AppleEvent *appleEvt, AppleEvent* reply, UInt32 refcon )
{
gQuitFlag = true;
return noErr;
}
// For debugging
static void print_to_log_file(const char *format, ...) {
#if CREATE_LOG
FILE *f;
va_list args;
char buf[256];
time_t t;
safe_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);
safe_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
static 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