mirror of https://github.com/BOINC/boinc.git
141 lines
4.2 KiB
C++
141 lines
4.2 KiB
C++
// 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 <http://www.gnu.org/licenses/>.
|
|
|
|
// switcher.C
|
|
//
|
|
// When run as
|
|
// switcher Full-Path Executable-Name X1 ... Xn
|
|
// runs program at Full-Path with args X1. ... Xn
|
|
// note that the executable name nust be specified twice:
|
|
// once as part of the Full_Path and again as just the name
|
|
|
|
#include <unistd.h>
|
|
#include <cstdio>
|
|
#include <cstring>
|
|
#include <cerrno>
|
|
#include <pwd.h> // getpwuid
|
|
#include <grp.h>
|
|
|
|
#include "app_ipc.h"
|
|
#include "filesys.h"
|
|
#include "str_replace.h"
|
|
|
|
using std::strcpy;
|
|
|
|
int main(int /*argc*/, char** argv) {
|
|
passwd *pw;
|
|
group *grp;
|
|
char user_name[256], group_name[256];
|
|
APP_INIT_DATA aid;
|
|
FILE *f;
|
|
int retval = -1;
|
|
char libpath[8192];
|
|
char newlibs[256];
|
|
char *projectDirName;
|
|
|
|
strcpy(user_name, "boinc_project");
|
|
strcpy(group_name, "boinc_project");
|
|
|
|
#if 0 // For debugging only
|
|
char current_dir[MAXPATHLEN];
|
|
|
|
getcwd( current_dir, sizeof(current_dir));
|
|
fprintf(stderr, "current directory = %s\n", current_dir);
|
|
|
|
for (int i=0; i<argc; i++) {
|
|
fprintf(stderr, "switcher arg %d: %s\n", i, argv[i]);
|
|
}
|
|
fflush(stderr);
|
|
#endif
|
|
|
|
#if 0 // For debugging only
|
|
// Allow debugging without running as user or group boinc_project
|
|
pw = getpwuid(getuid());
|
|
if (pw) strcpy(user_name, pw->pw_name);
|
|
grp = getgrgid(getgid());
|
|
if (grp) {
|
|
strcpy(group_name, grp->gr_gid);
|
|
}
|
|
|
|
#endif
|
|
|
|
// We are running setuid root, so setgid() sets real group ID,
|
|
// effective group ID and saved set_group-ID for this process
|
|
grp = getgrnam(group_name);
|
|
if (grp) {
|
|
(void) setgid(grp->gr_gid);
|
|
}
|
|
|
|
// We are running setuid root, so setuid() sets real user ID,
|
|
// effective user ID and saved set_user-ID for this process
|
|
pw = getpwnam(user_name);
|
|
if (pw) {
|
|
(void) setuid(pw->pw_uid);
|
|
}
|
|
|
|
// For unknown reasons, the LD_LIBRARY_PATH and DYLD_LIBRARY_PATH
|
|
// environment variables are not passed in to switcher, though all
|
|
// other environment variables do get propagated. So we recreate
|
|
// LD_LIBRARY_PATH and DYLD_LIBRARY_PATH here.
|
|
f = fopen(INIT_DATA_FILE, "r");
|
|
if (f) {
|
|
retval = parse_init_data_file(f, aid);
|
|
fclose(f);
|
|
}
|
|
|
|
if (!retval) {
|
|
// Get project name without leading path
|
|
projectDirName = strrchr(aid.project_dir, '/');
|
|
if (projectDirName) {
|
|
++projectDirName;
|
|
} else {
|
|
projectDirName = aid.project_dir;
|
|
}
|
|
sprintf(newlibs, "../../%s:.:../..", projectDirName);
|
|
#ifdef __APPLE__
|
|
strcat(newlibs, ":/usr/local/cuda/lib/");
|
|
#endif
|
|
char* p = getenv("LD_LIBRARY_PATH");
|
|
if (p) {
|
|
snprintf(libpath, sizeof(libpath), "%s:%s", newlibs, p);
|
|
} else {
|
|
safe_strcpy(libpath, newlibs);
|
|
}
|
|
setenv("LD_LIBRARY_PATH", libpath, 1);
|
|
|
|
// On the Mac, do the same for DYLD_LIBRARY_PATH
|
|
//
|
|
#ifdef __APPLE__
|
|
p = getenv("DYLD_LIBRARY_PATH");
|
|
if (p) {
|
|
sprintf(libpath, "%s:%s", newlibs, p);
|
|
} else {
|
|
safe_strcpy(libpath, newlibs);
|
|
}
|
|
setenv("DYLD_LIBRARY_PATH", libpath, 1);
|
|
#endif
|
|
}
|
|
|
|
retval = execv(argv[1], argv+2);
|
|
if (retval == -1) {
|
|
// If we got here execv failed
|
|
fprintf(stderr, "Process creation (%s) failed: %s (errno = %d)\n", argv[1], strerror(errno), retval);
|
|
}
|
|
|
|
return retval;
|
|
}
|