1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
|
// 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
// Satisfy an error / warning from rpmlint: ensure that
// we drop any supplementary groups associated with root
setgroups(0, NULL);
// 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;
}
|
