boinc/tools/updater.C

123 lines
3.4 KiB
C

// Berkeley Open Infrastructure for Network Computing
// http://boinc.berkeley.edu
// Copyright (C) 2006 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
// Program to install files as part of auto-update.
// Run in a version directory.
// Arguments:
//
// --install_dir X copy files to X (required)
// --run_manager when done, run Manager
// --run_core when done, run core client
// --run_as_service when done, run core client as service
#include <stdio.h>
#ifdef _WIN32
#else
#include <unistd.h>
#include <errno.h>
#endif
#include "filesys.h"
#include "util.h"
#ifdef _WIN32
#define CORE_NAME "boinc.exe"
#define MANAGER_NAME "boincmgr.exe"
#else
#define CORE_NAME "boinc_client"
#define MANAGER_NAME "boinc_mgr"
#endif
char* install_dir;
int prepare_prev_dir() {
char prev_dir[256];
int retval;
sprintf(prev_dir, "%s/prev_version", install_dir);
if (is_dir(prev_dir)) {
retval = clean_out_dir(prev_dir);
if (retval) return retval;
} else {
retval = boinc_mkdir(prev_dir);
if (retval) return retval;
}
return 0;
}
int move_to_prev(char* file) {
char oldname[1024], newname[1024];
sprintf(oldname, "%s/%s", install_dir, file);
sprintf(newname, "%s/prev_version/%s", install_dir, file);
return boinc_rename(oldname, newname);
}
int copy_to_main(char* file) {
char newname[1024];
sprintf(newname, "%s/%s", install_dir, file);
return boinc_copy(file, newname);
}
int main(int argc, char** argv) {
int i, retval, argc2;
bool run_as_service = false;
bool run_core = false;
bool run_manager = false;
char* argv2[10];
char path[1024];
install_dir = 0;
for (i=1; i<argc; i++) {
printf("updater: argv[%d] is %s\n", i, argv[i]);
if (!strcmp(argv[i], "--run_as_service")) {
run_as_service = true;
} else if (!strcmp(argv[i], "--run_manager")) {
run_manager = true;
} else if (!strcmp(argv[i], "--run_core")) {
run_core = true;
} else if (!strcmp(argv[i], "--install_dir")) {
install_dir = argv[++i];
}
}
if (!install_dir) {
fprintf(stderr, "updater: install dir not specified\n");
exit(1);
}
wait_client_mutex(install_dir, 30);
retval = prepare_prev_dir();
if (retval) exit(retval);
move_to_prev(CORE_NAME);
//move_to_prev(MANAGER_NAME);
copy_to_main(CORE_NAME);
//copy_to_main(MANAGER_NAME);
if (run_core) {
argv2[0] = CORE_NAME;
argv2[1] = 0;
argc2 = 1;
run_program(install_dir, CORE_NAME, argc2, argv2);
}
if (run_manager) {
argv2[0] = MANAGER_NAME;
argv2[1] = 0;
argc2 = 1;
run_program(install_dir, MANAGER_NAME, argc2, argv2);
}
}