hijack.c revision 1.65 1 1.65 pooka /* $NetBSD: hijack.c,v 1.65 2011/02/23 15:29:21 pooka Exp $ */
2 1.1 pooka
3 1.1 pooka /*-
4 1.1 pooka * Copyright (c) 2011 Antti Kantee. All Rights Reserved.
5 1.1 pooka *
6 1.1 pooka * Redistribution and use in source and binary forms, with or without
7 1.1 pooka * modification, are permitted provided that the following conditions
8 1.1 pooka * are met:
9 1.1 pooka * 1. Redistributions of source code must retain the above copyright
10 1.1 pooka * notice, this list of conditions and the following disclaimer.
11 1.1 pooka * 2. Redistributions in binary form must reproduce the above copyright
12 1.1 pooka * notice, this list of conditions and the following disclaimer in the
13 1.1 pooka * documentation and/or other materials provided with the distribution.
14 1.1 pooka *
15 1.1 pooka * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS
16 1.1 pooka * OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
17 1.1 pooka * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
18 1.1 pooka * DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
19 1.1 pooka * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
20 1.1 pooka * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
21 1.1 pooka * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
22 1.1 pooka * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
23 1.1 pooka * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
24 1.1 pooka * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
25 1.1 pooka * SUCH DAMAGE.
26 1.1 pooka */
27 1.1 pooka
28 1.1 pooka #include <sys/cdefs.h>
29 1.65 pooka __RCSID("$NetBSD: hijack.c,v 1.65 2011/02/23 15:29:21 pooka Exp $");
30 1.21 christos
31 1.21 christos #define __ssp_weak_name(fun) _hijack_ ## fun
32 1.1 pooka
33 1.1 pooka #include <sys/param.h>
34 1.1 pooka #include <sys/types.h>
35 1.10 pooka #include <sys/event.h>
36 1.1 pooka #include <sys/ioctl.h>
37 1.62 pooka #include <sys/mman.h>
38 1.48 pooka #include <sys/mount.h>
39 1.48 pooka #include <sys/poll.h>
40 1.1 pooka #include <sys/socket.h>
41 1.45 pooka #include <sys/statvfs.h>
42 1.1 pooka
43 1.1 pooka #include <rump/rumpclient.h>
44 1.1 pooka #include <rump/rump_syscalls.h>
45 1.1 pooka
46 1.1 pooka #include <assert.h>
47 1.1 pooka #include <dlfcn.h>
48 1.1 pooka #include <err.h>
49 1.1 pooka #include <errno.h>
50 1.1 pooka #include <fcntl.h>
51 1.1 pooka #include <poll.h>
52 1.1 pooka #include <pthread.h>
53 1.3 pooka #include <signal.h>
54 1.1 pooka #include <stdarg.h>
55 1.8 pooka #include <stdbool.h>
56 1.1 pooka #include <stdio.h>
57 1.1 pooka #include <stdlib.h>
58 1.28 pooka #include <string.h>
59 1.3 pooka #include <time.h>
60 1.1 pooka #include <unistd.h>
61 1.1 pooka
62 1.64 pooka #include "hijack.h"
63 1.64 pooka
64 1.17 pooka enum dualcall {
65 1.60 pooka DUALCALL_WRITE, DUALCALL_WRITEV, DUALCALL_PWRITE, DUALCALL_PWRITEV,
66 1.17 pooka DUALCALL_IOCTL, DUALCALL_FCNTL,
67 1.17 pooka DUALCALL_SOCKET, DUALCALL_ACCEPT, DUALCALL_BIND, DUALCALL_CONNECT,
68 1.17 pooka DUALCALL_GETPEERNAME, DUALCALL_GETSOCKNAME, DUALCALL_LISTEN,
69 1.17 pooka DUALCALL_RECVFROM, DUALCALL_RECVMSG,
70 1.17 pooka DUALCALL_SENDTO, DUALCALL_SENDMSG,
71 1.17 pooka DUALCALL_GETSOCKOPT, DUALCALL_SETSOCKOPT,
72 1.17 pooka DUALCALL_SHUTDOWN,
73 1.60 pooka DUALCALL_READ, DUALCALL_READV, DUALCALL_PREAD, DUALCALL_PREADV,
74 1.41 pooka DUALCALL_DUP2,
75 1.34 pooka DUALCALL_CLOSE,
76 1.17 pooka DUALCALL_POLLTS,
77 1.34 pooka DUALCALL_KEVENT,
78 1.45 pooka DUALCALL_STAT, DUALCALL_LSTAT, DUALCALL_FSTAT,
79 1.45 pooka DUALCALL_CHMOD, DUALCALL_LCHMOD, DUALCALL_FCHMOD,
80 1.45 pooka DUALCALL_CHOWN, DUALCALL_LCHOWN, DUALCALL_FCHOWN,
81 1.45 pooka DUALCALL_OPEN,
82 1.45 pooka DUALCALL_STATVFS1, DUALCALL_FSTATVFS1,
83 1.45 pooka DUALCALL_CHDIR, DUALCALL_FCHDIR,
84 1.45 pooka DUALCALL_LSEEK,
85 1.45 pooka DUALCALL_GETDENTS,
86 1.45 pooka DUALCALL_UNLINK, DUALCALL_SYMLINK, DUALCALL_READLINK,
87 1.45 pooka DUALCALL_RENAME,
88 1.45 pooka DUALCALL_MKDIR, DUALCALL_RMDIR,
89 1.45 pooka DUALCALL_UTIMES, DUALCALL_LUTIMES, DUALCALL_FUTIMES,
90 1.45 pooka DUALCALL_TRUNCATE, DUALCALL_FTRUNCATE,
91 1.45 pooka DUALCALL_FSYNC, DUALCALL_FSYNC_RANGE,
92 1.48 pooka DUALCALL_MOUNT, DUALCALL_UNMOUNT,
93 1.57 pooka DUALCALL___GETCWD,
94 1.60 pooka DUALCALL_CHFLAGS, DUALCALL_LCHFLAGS, DUALCALL_FCHFLAGS,
95 1.65 pooka DUALCALL_ACCESS,
96 1.17 pooka DUALCALL__NUM
97 1.1 pooka };
98 1.1 pooka
99 1.8 pooka #define RSYS_STRING(a) __STRING(a)
100 1.8 pooka #define RSYS_NAME(a) RSYS_STRING(__CONCAT(RUMP_SYS_RENAME_,a))
101 1.8 pooka
102 1.1 pooka /*
103 1.14 pooka * Would be nice to get this automatically in sync with libc.
104 1.14 pooka * Also, this does not work for compat-using binaries!
105 1.14 pooka */
106 1.14 pooka #if !__NetBSD_Prereq__(5,99,7)
107 1.29 pooka #define REALSELECT select
108 1.29 pooka #define REALPOLLTS pollts
109 1.34 pooka #define REALKEVENT kevent
110 1.45 pooka #define REALSTAT __stat30
111 1.45 pooka #define REALLSTAT __lstat30
112 1.45 pooka #define REALFSTAT __fstat30
113 1.45 pooka #define REALUTIMES utimes
114 1.45 pooka #define REALLUTIMES lutimes
115 1.45 pooka #define REALFUTIMES futimes
116 1.14 pooka #else
117 1.29 pooka #define REALSELECT _sys___select50
118 1.29 pooka #define REALPOLLTS _sys___pollts50
119 1.34 pooka #define REALKEVENT _sys___kevent50
120 1.45 pooka #define REALSTAT __stat50
121 1.45 pooka #define REALLSTAT __lstat50
122 1.45 pooka #define REALFSTAT __fstat50
123 1.45 pooka #define REALUTIMES __utimes50
124 1.45 pooka #define REALLUTIMES __lutimes50
125 1.45 pooka #define REALFUTIMES __futimes50
126 1.17 pooka #endif
127 1.31 pooka #define REALREAD _sys_read
128 1.60 pooka #define REALPREAD _sys_pread
129 1.60 pooka #define REALPWRITE _sys_pwrite
130 1.45 pooka #define REALGETDENTS __getdents30
131 1.48 pooka #define REALMOUNT __mount50
132 1.14 pooka
133 1.29 pooka int REALSELECT(int, fd_set *, fd_set *, fd_set *, struct timeval *);
134 1.29 pooka int REALPOLLTS(struct pollfd *, nfds_t,
135 1.20 pooka const struct timespec *, const sigset_t *);
136 1.34 pooka int REALKEVENT(int, const struct kevent *, size_t, struct kevent *, size_t,
137 1.34 pooka const struct timespec *);
138 1.31 pooka ssize_t REALREAD(int, void *, size_t);
139 1.60 pooka ssize_t REALPREAD(int, void *, size_t, off_t);
140 1.60 pooka ssize_t REALPWRITE(int, const void *, size_t, off_t);
141 1.45 pooka int REALSTAT(const char *, struct stat *);
142 1.45 pooka int REALLSTAT(const char *, struct stat *);
143 1.45 pooka int REALFSTAT(int, struct stat *);
144 1.45 pooka int REALGETDENTS(int, char *, size_t);
145 1.45 pooka int REALUTIMES(const char *, const struct timeval [2]);
146 1.45 pooka int REALLUTIMES(const char *, const struct timeval [2]);
147 1.45 pooka int REALFUTIMES(int, const struct timeval [2]);
148 1.48 pooka int REALMOUNT(const char *, const char *, int, void *, size_t);
149 1.57 pooka int __getcwd(char *, size_t);
150 1.17 pooka
151 1.17 pooka #define S(a) __STRING(a)
152 1.17 pooka struct sysnames {
153 1.17 pooka enum dualcall scm_callnum;
154 1.17 pooka const char *scm_hostname;
155 1.17 pooka const char *scm_rumpname;
156 1.17 pooka } syscnames[] = {
157 1.17 pooka { DUALCALL_SOCKET, "__socket30", RSYS_NAME(SOCKET) },
158 1.17 pooka { DUALCALL_ACCEPT, "accept", RSYS_NAME(ACCEPT) },
159 1.17 pooka { DUALCALL_BIND, "bind", RSYS_NAME(BIND) },
160 1.17 pooka { DUALCALL_CONNECT, "connect", RSYS_NAME(CONNECT) },
161 1.17 pooka { DUALCALL_GETPEERNAME, "getpeername", RSYS_NAME(GETPEERNAME) },
162 1.17 pooka { DUALCALL_GETSOCKNAME, "getsockname", RSYS_NAME(GETSOCKNAME) },
163 1.17 pooka { DUALCALL_LISTEN, "listen", RSYS_NAME(LISTEN) },
164 1.17 pooka { DUALCALL_RECVFROM, "recvfrom", RSYS_NAME(RECVFROM) },
165 1.17 pooka { DUALCALL_RECVMSG, "recvmsg", RSYS_NAME(RECVMSG) },
166 1.17 pooka { DUALCALL_SENDTO, "sendto", RSYS_NAME(SENDTO) },
167 1.17 pooka { DUALCALL_SENDMSG, "sendmsg", RSYS_NAME(SENDMSG) },
168 1.17 pooka { DUALCALL_GETSOCKOPT, "getsockopt", RSYS_NAME(GETSOCKOPT) },
169 1.17 pooka { DUALCALL_SETSOCKOPT, "setsockopt", RSYS_NAME(SETSOCKOPT) },
170 1.17 pooka { DUALCALL_SHUTDOWN, "shutdown", RSYS_NAME(SHUTDOWN) },
171 1.31 pooka { DUALCALL_READ, S(REALREAD), RSYS_NAME(READ) },
172 1.17 pooka { DUALCALL_READV, "readv", RSYS_NAME(READV) },
173 1.60 pooka { DUALCALL_PREAD, S(REALPREAD), RSYS_NAME(PREAD) },
174 1.60 pooka { DUALCALL_PREADV, "preadv", RSYS_NAME(PREADV) },
175 1.17 pooka { DUALCALL_WRITE, "write", RSYS_NAME(WRITE) },
176 1.17 pooka { DUALCALL_WRITEV, "writev", RSYS_NAME(WRITEV) },
177 1.60 pooka { DUALCALL_PWRITE, S(REALPWRITE), RSYS_NAME(PWRITE) },
178 1.60 pooka { DUALCALL_PWRITEV, "pwritev", RSYS_NAME(PWRITEV) },
179 1.17 pooka { DUALCALL_IOCTL, "ioctl", RSYS_NAME(IOCTL) },
180 1.17 pooka { DUALCALL_FCNTL, "fcntl", RSYS_NAME(FCNTL) },
181 1.17 pooka { DUALCALL_DUP2, "dup2", RSYS_NAME(DUP2) },
182 1.17 pooka { DUALCALL_CLOSE, "close", RSYS_NAME(CLOSE) },
183 1.29 pooka { DUALCALL_POLLTS, S(REALPOLLTS), RSYS_NAME(POLLTS) },
184 1.34 pooka { DUALCALL_KEVENT, S(REALKEVENT), RSYS_NAME(KEVENT) },
185 1.45 pooka { DUALCALL_STAT, S(REALSTAT), RSYS_NAME(STAT) },
186 1.45 pooka { DUALCALL_LSTAT, S(REALLSTAT), RSYS_NAME(LSTAT) },
187 1.45 pooka { DUALCALL_FSTAT, S(REALFSTAT), RSYS_NAME(FSTAT) },
188 1.45 pooka { DUALCALL_CHOWN, "chown", RSYS_NAME(CHOWN) },
189 1.45 pooka { DUALCALL_LCHOWN, "lchown", RSYS_NAME(LCHOWN) },
190 1.45 pooka { DUALCALL_FCHOWN, "fchown", RSYS_NAME(FCHOWN) },
191 1.45 pooka { DUALCALL_CHMOD, "chmod", RSYS_NAME(CHMOD) },
192 1.45 pooka { DUALCALL_LCHMOD, "lchmod", RSYS_NAME(LCHMOD) },
193 1.45 pooka { DUALCALL_FCHMOD, "fchmod", RSYS_NAME(FCHMOD) },
194 1.45 pooka { DUALCALL_UTIMES, S(REALUTIMES), RSYS_NAME(UTIMES) },
195 1.45 pooka { DUALCALL_LUTIMES, S(REALLUTIMES), RSYS_NAME(LUTIMES) },
196 1.45 pooka { DUALCALL_FUTIMES, S(REALFUTIMES), RSYS_NAME(FUTIMES) },
197 1.45 pooka { DUALCALL_OPEN, "open", RSYS_NAME(OPEN) },
198 1.45 pooka { DUALCALL_STATVFS1, "statvfs1", RSYS_NAME(STATVFS1) },
199 1.45 pooka { DUALCALL_FSTATVFS1, "fstatvfs1", RSYS_NAME(FSTATVFS1) },
200 1.45 pooka { DUALCALL_CHDIR, "chdir", RSYS_NAME(CHDIR) },
201 1.45 pooka { DUALCALL_FCHDIR, "fchdir", RSYS_NAME(FCHDIR) },
202 1.61 pooka { DUALCALL_LSEEK, "lseek", RSYS_NAME(LSEEK) },
203 1.45 pooka { DUALCALL_GETDENTS, "__getdents30", RSYS_NAME(GETDENTS) },
204 1.45 pooka { DUALCALL_UNLINK, "unlink", RSYS_NAME(UNLINK) },
205 1.45 pooka { DUALCALL_SYMLINK, "symlink", RSYS_NAME(SYMLINK) },
206 1.45 pooka { DUALCALL_READLINK, "readlink", RSYS_NAME(READLINK) },
207 1.45 pooka { DUALCALL_RENAME, "rename", RSYS_NAME(RENAME) },
208 1.45 pooka { DUALCALL_MKDIR, "mkdir", RSYS_NAME(MKDIR) },
209 1.45 pooka { DUALCALL_RMDIR, "rmdir", RSYS_NAME(RMDIR) },
210 1.45 pooka { DUALCALL_TRUNCATE, "truncate", RSYS_NAME(TRUNCATE) },
211 1.45 pooka { DUALCALL_FTRUNCATE, "ftruncate", RSYS_NAME(FTRUNCATE) },
212 1.45 pooka { DUALCALL_FSYNC, "fsync", RSYS_NAME(FSYNC) },
213 1.45 pooka { DUALCALL_FSYNC_RANGE, "fsync_range", RSYS_NAME(FSYNC_RANGE) },
214 1.48 pooka { DUALCALL_MOUNT, S(REALMOUNT), RSYS_NAME(MOUNT) },
215 1.48 pooka { DUALCALL_UNMOUNT, "unmount", RSYS_NAME(UNMOUNT) },
216 1.57 pooka { DUALCALL___GETCWD, "__getcwd", RSYS_NAME(__GETCWD) },
217 1.60 pooka { DUALCALL_CHFLAGS, "chflags", RSYS_NAME(CHFLAGS) },
218 1.60 pooka { DUALCALL_LCHFLAGS, "lchflags", RSYS_NAME(LCHFLAGS) },
219 1.60 pooka { DUALCALL_FCHFLAGS, "fchflags", RSYS_NAME(FCHFLAGS) },
220 1.65 pooka { DUALCALL_ACCESS, "access", RSYS_NAME(ACCESS) },
221 1.17 pooka };
222 1.17 pooka #undef S
223 1.17 pooka
224 1.17 pooka struct bothsys {
225 1.17 pooka void *bs_host;
226 1.17 pooka void *bs_rump;
227 1.17 pooka } syscalls[DUALCALL__NUM];
228 1.17 pooka #define GETSYSCALL(which, name) syscalls[DUALCALL_##name].bs_##which
229 1.17 pooka
230 1.25 pooka pid_t (*host_fork)(void);
231 1.25 pooka int (*host_daemon)(int, int);
232 1.39 pooka int (*host_execve)(const char *, char *const[], char *const[]);
233 1.62 pooka void * (*host_mmap)(void *, size_t, int, int, int, off_t);
234 1.17 pooka
235 1.47 pooka /* ok, we need *two* bits per dup2'd fd to track fd+HIJACKOFF aliases */
236 1.40 pooka static uint32_t dup2mask;
237 1.47 pooka #define ISDUP2D(fd) (((fd) < 16) && (1<<(fd) & dup2mask))
238 1.40 pooka #define SETDUP2(fd) \
239 1.47 pooka do { if ((fd) < 16) dup2mask |= (1<<(fd)); } while (/*CONSTCOND*/0)
240 1.40 pooka #define CLRDUP2(fd) \
241 1.47 pooka do { if ((fd) < 16) dup2mask &= ~(1<<(fd)); } while (/*CONSTCOND*/0)
242 1.47 pooka #define ISDUP2ALIAS(fd) (((fd) < 16) && (1<<((fd)+16) & dup2mask))
243 1.47 pooka #define SETDUP2ALIAS(fd) \
244 1.47 pooka do { if ((fd) < 16) dup2mask |= (1<<((fd)+16)); } while (/*CONSTCOND*/0)
245 1.47 pooka #define CLRDUP2ALIAS(fd) \
246 1.47 pooka do { if ((fd) < 16) dup2mask &= ~(1<<((fd)+16)); } while (/*CONSTCOND*/0)
247 1.17 pooka
248 1.17 pooka //#define DEBUGJACK
249 1.17 pooka #ifdef DEBUGJACK
250 1.17 pooka #define DPRINTF(x) mydprintf x
251 1.17 pooka static void
252 1.17 pooka mydprintf(const char *fmt, ...)
253 1.17 pooka {
254 1.17 pooka va_list ap;
255 1.17 pooka
256 1.17 pooka if (ISDUP2D(STDERR_FILENO))
257 1.17 pooka return;
258 1.17 pooka
259 1.17 pooka va_start(ap, fmt);
260 1.17 pooka vfprintf(stderr, fmt, ap);
261 1.17 pooka va_end(ap);
262 1.17 pooka }
263 1.17 pooka
264 1.17 pooka #else
265 1.17 pooka #define DPRINTF(x)
266 1.14 pooka #endif
267 1.14 pooka
268 1.17 pooka #define FDCALL(type, name, rcname, args, proto, vars) \
269 1.17 pooka type name args \
270 1.17 pooka { \
271 1.17 pooka type (*fun) proto; \
272 1.17 pooka \
273 1.33 pooka DPRINTF(("%s -> %d\n", __STRING(name), fd)); \
274 1.17 pooka if (fd_isrump(fd)) { \
275 1.17 pooka fun = syscalls[rcname].bs_rump; \
276 1.17 pooka fd = fd_host2rump(fd); \
277 1.17 pooka } else { \
278 1.17 pooka fun = syscalls[rcname].bs_host; \
279 1.17 pooka } \
280 1.17 pooka \
281 1.17 pooka return fun vars; \
282 1.17 pooka }
283 1.17 pooka
284 1.45 pooka #define PATHCALL(type, name, rcname, args, proto, vars) \
285 1.45 pooka type name args \
286 1.45 pooka { \
287 1.45 pooka type (*fun) proto; \
288 1.45 pooka \
289 1.45 pooka DPRINTF(("%s -> %s\n", __STRING(name), path)); \
290 1.45 pooka if (path_isrump(path)) { \
291 1.45 pooka fun = syscalls[rcname].bs_rump; \
292 1.45 pooka path = path_host2rump(path); \
293 1.45 pooka } else { \
294 1.45 pooka fun = syscalls[rcname].bs_host; \
295 1.45 pooka } \
296 1.45 pooka \
297 1.45 pooka return fun vars; \
298 1.45 pooka }
299 1.45 pooka
300 1.14 pooka /*
301 1.49 pooka * This tracks if our process is in a subdirectory of /rump.
302 1.49 pooka * It's preserved over exec.
303 1.49 pooka */
304 1.49 pooka static bool pwdinrump = false;
305 1.49 pooka
306 1.49 pooka /*
307 1.49 pooka * These variables are set from the RUMPHIJACK string and control
308 1.49 pooka * which operations can product rump kernel file descriptors.
309 1.49 pooka * This should be easily extendable for future needs.
310 1.49 pooka */
311 1.49 pooka #define RUMPHIJACK_DEFAULT "path=/rump,socket=all:nolocal"
312 1.49 pooka static bool rumpsockets[PF_MAX];
313 1.49 pooka static const char *rumpprefix;
314 1.49 pooka static size_t rumpprefixlen;
315 1.49 pooka
316 1.49 pooka static struct {
317 1.49 pooka int pf;
318 1.49 pooka const char *name;
319 1.49 pooka } socketmap[] = {
320 1.51 pooka { PF_LOCAL, "local" },
321 1.49 pooka { PF_INET, "inet" },
322 1.49 pooka { PF_LINK, "link" },
323 1.55 pooka #ifdef PF_OROUTE
324 1.56 pooka { PF_OROUTE, "oroute" },
325 1.56 pooka #endif
326 1.49 pooka { PF_ROUTE, "route" },
327 1.49 pooka { PF_INET6, "inet6" },
328 1.55 pooka #ifdef PF_MPLS
329 1.55 pooka { PF_MPLS, "mpls" },
330 1.55 pooka #endif
331 1.49 pooka { -1, NULL }
332 1.49 pooka };
333 1.49 pooka
334 1.49 pooka static void
335 1.49 pooka sockparser(char *buf)
336 1.49 pooka {
337 1.49 pooka char *p, *l;
338 1.49 pooka bool value;
339 1.49 pooka int i;
340 1.49 pooka
341 1.49 pooka /* if "all" is present, it must be specified first */
342 1.49 pooka if (strncmp(buf, "all", strlen("all")) == 0) {
343 1.50 pooka for (i = 0; i < (int)__arraycount(rumpsockets); i++) {
344 1.49 pooka rumpsockets[i] = true;
345 1.49 pooka }
346 1.49 pooka buf += strlen("all");
347 1.49 pooka if (*buf == ':')
348 1.49 pooka buf++;
349 1.49 pooka }
350 1.49 pooka
351 1.49 pooka for (p = strtok_r(buf, ":", &l); p; p = strtok_r(NULL, ":", &l)) {
352 1.49 pooka value = true;
353 1.49 pooka if (strncmp(p, "no", strlen("no")) == 0) {
354 1.49 pooka value = false;
355 1.49 pooka p += strlen("no");
356 1.49 pooka }
357 1.45 pooka
358 1.49 pooka for (i = 0; socketmap[i].name; i++) {
359 1.49 pooka if (strcmp(p, socketmap[i].name) == 0) {
360 1.49 pooka rumpsockets[socketmap[i].pf] = value;
361 1.49 pooka break;
362 1.49 pooka }
363 1.49 pooka }
364 1.49 pooka if (socketmap[i].name == NULL) {
365 1.49 pooka warnx("invalid socket specifier %s", p);
366 1.49 pooka }
367 1.49 pooka }
368 1.49 pooka }
369 1.49 pooka
370 1.49 pooka static void
371 1.49 pooka pathparser(char *buf)
372 1.49 pooka {
373 1.49 pooka
374 1.57 pooka /* sanity-check */
375 1.49 pooka if (*buf != '/')
376 1.49 pooka errx(1, "hijack path specifier must begin with ``/''");
377 1.57 pooka rumpprefixlen = strlen(buf);
378 1.57 pooka if (rumpprefixlen < 2)
379 1.57 pooka errx(1, "invalid hijack prefix: %s", buf);
380 1.57 pooka if (buf[rumpprefixlen-1] == '/' && strspn(buf, "/") != rumpprefixlen)
381 1.57 pooka errx(1, "hijack prefix may end in slash only if pure "
382 1.57 pooka "slash, gave %s", buf);
383 1.49 pooka
384 1.49 pooka if ((rumpprefix = strdup(buf)) == NULL)
385 1.49 pooka err(1, "strdup");
386 1.49 pooka rumpprefixlen = strlen(rumpprefix);
387 1.49 pooka }
388 1.49 pooka
389 1.49 pooka static struct {
390 1.49 pooka void (*parsefn)(char *);
391 1.49 pooka const char *name;
392 1.49 pooka } hijackparse[] = {
393 1.49 pooka { sockparser, "socket" },
394 1.49 pooka { pathparser, "path" },
395 1.49 pooka { NULL, NULL },
396 1.49 pooka };
397 1.49 pooka
398 1.49 pooka static void
399 1.49 pooka parsehijack(char *hijack)
400 1.49 pooka {
401 1.49 pooka char *p, *p2, *l;
402 1.49 pooka const char *hijackcopy;
403 1.49 pooka int i;
404 1.49 pooka
405 1.49 pooka if ((hijackcopy = strdup(hijack)) == NULL)
406 1.49 pooka err(1, "strdup");
407 1.49 pooka
408 1.49 pooka /* disable everything explicitly */
409 1.49 pooka for (i = 0; i < PF_MAX; i++)
410 1.49 pooka rumpsockets[i] = false;
411 1.49 pooka
412 1.49 pooka for (p = strtok_r(hijack, ",", &l); p; p = strtok_r(NULL, ",", &l)) {
413 1.49 pooka p2 = strchr(p, '=');
414 1.49 pooka if (!p2)
415 1.49 pooka errx(1, "invalid hijack specifier: %s", hijackcopy);
416 1.49 pooka
417 1.49 pooka for (i = 0; hijackparse[i].parsefn; i++) {
418 1.49 pooka if (strncmp(hijackparse[i].name, p,
419 1.49 pooka (size_t)(p2-p)) == 0) {
420 1.49 pooka hijackparse[i].parsefn(p2+1);
421 1.49 pooka break;
422 1.49 pooka }
423 1.49 pooka }
424 1.49 pooka }
425 1.49 pooka
426 1.49 pooka }
427 1.7 pooka
428 1.1 pooka static void __attribute__((constructor))
429 1.1 pooka rcinit(void)
430 1.1 pooka {
431 1.49 pooka char buf[1024];
432 1.23 pooka extern void *(*rumpclient_dlsym)(void *, const char *);
433 1.19 pooka unsigned i, j;
434 1.1 pooka
435 1.64 pooka rumpclient_dlsym = rumphijack_dlsym;
436 1.17 pooka host_fork = dlsym(RTLD_NEXT, "fork");
437 1.25 pooka host_daemon = dlsym(RTLD_NEXT, "daemon");
438 1.39 pooka host_execve = dlsym(RTLD_NEXT, "execve");
439 1.62 pooka host_mmap = dlsym(RTLD_NEXT, "mmap");
440 1.17 pooka
441 1.17 pooka /*
442 1.17 pooka * In theory cannot print anything during lookups because
443 1.17 pooka * we might not have the call vector set up. so, the errx()
444 1.17 pooka * is a bit of a strech, but it might work.
445 1.17 pooka */
446 1.1 pooka
447 1.17 pooka for (i = 0; i < DUALCALL__NUM; i++) {
448 1.17 pooka /* build runtime O(1) access */
449 1.17 pooka for (j = 0; j < __arraycount(syscnames); j++) {
450 1.17 pooka if (syscnames[j].scm_callnum == i)
451 1.17 pooka break;
452 1.17 pooka }
453 1.17 pooka
454 1.17 pooka if (j == __arraycount(syscnames))
455 1.17 pooka errx(1, "rumphijack error: syscall pos %d missing", i);
456 1.17 pooka
457 1.23 pooka syscalls[i].bs_host = dlsym(RTLD_NEXT,
458 1.23 pooka syscnames[j].scm_hostname);
459 1.17 pooka if (syscalls[i].bs_host == NULL)
460 1.17 pooka errx(1, "hostcall %s not found missing",
461 1.17 pooka syscnames[j].scm_hostname);
462 1.17 pooka
463 1.23 pooka syscalls[i].bs_rump = dlsym(RTLD_NEXT,
464 1.23 pooka syscnames[j].scm_rumpname);
465 1.17 pooka if (syscalls[i].bs_rump == NULL)
466 1.17 pooka errx(1, "rumpcall %s not found missing",
467 1.17 pooka syscnames[j].scm_rumpname);
468 1.1 pooka }
469 1.1 pooka
470 1.22 pooka if (rumpclient_init() == -1)
471 1.1 pooka err(1, "rumpclient init");
472 1.28 pooka
473 1.49 pooka /* check which syscalls we're supposed to hijack */
474 1.49 pooka if (getenv_r("RUMPHIJACK", buf, sizeof(buf)) == -1) {
475 1.49 pooka strcpy(buf, RUMPHIJACK_DEFAULT);
476 1.49 pooka }
477 1.49 pooka parsehijack(buf);
478 1.49 pooka
479 1.28 pooka /* set client persistence level */
480 1.44 pooka if (getenv_r("RUMPHIJACK_RETRYCONNECT", buf, sizeof(buf)) != -1) {
481 1.28 pooka if (strcmp(buf, "die") == 0)
482 1.28 pooka rumpclient_setconnretry(RUMPCLIENT_RETRYCONN_DIE);
483 1.28 pooka else if (strcmp(buf, "inftime") == 0)
484 1.28 pooka rumpclient_setconnretry(RUMPCLIENT_RETRYCONN_INFTIME);
485 1.28 pooka else if (strcmp(buf, "once") == 0)
486 1.28 pooka rumpclient_setconnretry(RUMPCLIENT_RETRYCONN_ONCE);
487 1.28 pooka else {
488 1.28 pooka time_t timeout;
489 1.44 pooka char *ep;
490 1.28 pooka
491 1.44 pooka timeout = (time_t)strtoll(buf, &ep, 10);
492 1.44 pooka if (timeout <= 0 || ep != buf + strlen(buf))
493 1.44 pooka errx(1, "RUMPHIJACK_RETRYCONNECT must be "
494 1.44 pooka "keyword or integer, got: %s", buf);
495 1.28 pooka
496 1.28 pooka rumpclient_setconnretry(timeout);
497 1.28 pooka }
498 1.28 pooka }
499 1.39 pooka
500 1.39 pooka if (getenv_r("RUMPHIJACK__DUP2MASK", buf, sizeof(buf)) == 0) {
501 1.40 pooka dup2mask = strtoul(buf, NULL, 10);
502 1.45 pooka unsetenv("RUMPHIJACK__DUP2MASK");
503 1.45 pooka }
504 1.45 pooka if (getenv_r("RUMPHIJACK__PWDINRUMP", buf, sizeof(buf)) == 0) {
505 1.49 pooka pwdinrump = true;
506 1.45 pooka unsetenv("RUMPHIJACK__PWDINRUMP");
507 1.39 pooka }
508 1.1 pooka }
509 1.1 pooka
510 1.2 pooka /* XXX: need runtime selection. low for now due to FD_SETSIZE */
511 1.2 pooka #define HIJACK_FDOFF 128
512 1.2 pooka static int
513 1.2 pooka fd_rump2host(int fd)
514 1.2 pooka {
515 1.2 pooka
516 1.2 pooka if (fd == -1)
517 1.2 pooka return fd;
518 1.2 pooka
519 1.2 pooka if (!ISDUP2D(fd))
520 1.2 pooka fd += HIJACK_FDOFF;
521 1.2 pooka
522 1.2 pooka return fd;
523 1.2 pooka }
524 1.2 pooka
525 1.2 pooka static int
526 1.2 pooka fd_host2rump(int fd)
527 1.2 pooka {
528 1.2 pooka
529 1.2 pooka if (!ISDUP2D(fd))
530 1.2 pooka fd -= HIJACK_FDOFF;
531 1.2 pooka return fd;
532 1.2 pooka }
533 1.2 pooka
534 1.2 pooka static bool
535 1.2 pooka fd_isrump(int fd)
536 1.2 pooka {
537 1.2 pooka
538 1.2 pooka return ISDUP2D(fd) || fd >= HIJACK_FDOFF;
539 1.2 pooka }
540 1.2 pooka
541 1.40 pooka #define assertfd(_fd_) assert(ISDUP2D(_fd_) || (_fd_) >= HIJACK_FDOFF)
542 1.40 pooka
543 1.49 pooka static bool
544 1.45 pooka path_isrump(const char *path)
545 1.45 pooka {
546 1.45 pooka
547 1.49 pooka if (rumpprefix == NULL)
548 1.49 pooka return false;
549 1.49 pooka
550 1.45 pooka if (*path == '/') {
551 1.49 pooka if (strncmp(path, rumpprefix, rumpprefixlen) == 0)
552 1.49 pooka return true;
553 1.49 pooka return false;
554 1.45 pooka } else {
555 1.45 pooka return pwdinrump;
556 1.45 pooka }
557 1.45 pooka }
558 1.45 pooka
559 1.45 pooka static const char *rootpath = "/";
560 1.45 pooka static const char *
561 1.45 pooka path_host2rump(const char *path)
562 1.45 pooka {
563 1.45 pooka const char *rv;
564 1.45 pooka
565 1.45 pooka if (*path == '/') {
566 1.49 pooka rv = path + rumpprefixlen;
567 1.45 pooka if (*rv == '\0')
568 1.45 pooka rv = rootpath;
569 1.45 pooka } else {
570 1.45 pooka rv = path;
571 1.45 pooka }
572 1.45 pooka
573 1.45 pooka return rv;
574 1.45 pooka }
575 1.45 pooka
576 1.40 pooka static int
577 1.40 pooka dodup(int oldd, int minfd)
578 1.40 pooka {
579 1.40 pooka int (*op_fcntl)(int, int, ...);
580 1.40 pooka int newd;
581 1.40 pooka int isrump;
582 1.40 pooka
583 1.40 pooka DPRINTF(("dup -> %d (minfd %d)\n", oldd, minfd));
584 1.40 pooka if (fd_isrump(oldd)) {
585 1.40 pooka op_fcntl = GETSYSCALL(rump, FCNTL);
586 1.40 pooka oldd = fd_host2rump(oldd);
587 1.63 pooka if (minfd >= HIJACK_FDOFF)
588 1.63 pooka minfd -= HIJACK_FDOFF;
589 1.40 pooka isrump = 1;
590 1.40 pooka } else {
591 1.40 pooka op_fcntl = GETSYSCALL(host, FCNTL);
592 1.40 pooka isrump = 0;
593 1.40 pooka }
594 1.40 pooka
595 1.40 pooka newd = op_fcntl(oldd, F_DUPFD, minfd);
596 1.40 pooka
597 1.40 pooka if (isrump)
598 1.40 pooka newd = fd_rump2host(newd);
599 1.40 pooka DPRINTF(("dup <- %d\n", newd));
600 1.40 pooka
601 1.40 pooka return newd;
602 1.40 pooka }
603 1.2 pooka
604 1.47 pooka /*
605 1.47 pooka * dup a host file descriptor so that it doesn't collide with the dup2mask
606 1.47 pooka */
607 1.47 pooka static int
608 1.47 pooka fd_dupgood(int fd)
609 1.47 pooka {
610 1.47 pooka int (*op_fcntl)(int, int, ...) = GETSYSCALL(host, FCNTL);
611 1.47 pooka int (*op_close)(int) = GETSYSCALL(host, CLOSE);
612 1.47 pooka int ofd, i;
613 1.47 pooka
614 1.47 pooka for (i = 1; ISDUP2D(fd); i++) {
615 1.47 pooka ofd = fd;
616 1.47 pooka fd = op_fcntl(ofd, F_DUPFD, i);
617 1.47 pooka op_close(ofd);
618 1.47 pooka }
619 1.47 pooka
620 1.47 pooka return fd;
621 1.47 pooka }
622 1.47 pooka
623 1.45 pooka int
624 1.45 pooka open(const char *path, int flags, ...)
625 1.45 pooka {
626 1.45 pooka int (*op_open)(const char *, int, ...);
627 1.45 pooka bool isrump;
628 1.45 pooka va_list ap;
629 1.45 pooka int fd;
630 1.45 pooka
631 1.45 pooka if (path_isrump(path)) {
632 1.45 pooka path = path_host2rump(path);
633 1.45 pooka op_open = GETSYSCALL(rump, OPEN);
634 1.45 pooka isrump = true;
635 1.45 pooka } else {
636 1.45 pooka op_open = GETSYSCALL(host, OPEN);
637 1.45 pooka isrump = false;
638 1.45 pooka }
639 1.45 pooka
640 1.45 pooka va_start(ap, flags);
641 1.45 pooka fd = op_open(path, flags, va_arg(ap, mode_t));
642 1.45 pooka va_end(ap);
643 1.45 pooka
644 1.45 pooka if (isrump)
645 1.45 pooka fd = fd_rump2host(fd);
646 1.47 pooka else
647 1.47 pooka fd = fd_dupgood(fd);
648 1.45 pooka return fd;
649 1.45 pooka }
650 1.45 pooka
651 1.45 pooka int
652 1.45 pooka chdir(const char *path)
653 1.45 pooka {
654 1.45 pooka int (*op_chdir)(const char *);
655 1.45 pooka bool isrump;
656 1.45 pooka int rv;
657 1.45 pooka
658 1.45 pooka if (path_isrump(path)) {
659 1.45 pooka op_chdir = GETSYSCALL(rump, CHDIR);
660 1.45 pooka isrump = true;
661 1.45 pooka path = path_host2rump(path);
662 1.45 pooka } else {
663 1.45 pooka op_chdir = GETSYSCALL(host, CHDIR);
664 1.45 pooka isrump = false;
665 1.45 pooka }
666 1.45 pooka
667 1.45 pooka rv = op_chdir(path);
668 1.45 pooka if (rv == 0) {
669 1.45 pooka if (isrump)
670 1.45 pooka pwdinrump = true;
671 1.45 pooka else
672 1.45 pooka pwdinrump = false;
673 1.45 pooka }
674 1.45 pooka
675 1.45 pooka return rv;
676 1.45 pooka }
677 1.45 pooka
678 1.45 pooka int
679 1.45 pooka fchdir(int fd)
680 1.45 pooka {
681 1.45 pooka int (*op_fchdir)(int);
682 1.45 pooka bool isrump;
683 1.45 pooka int rv;
684 1.45 pooka
685 1.45 pooka if (fd_isrump(fd)) {
686 1.45 pooka op_fchdir = GETSYSCALL(rump, FCHDIR);
687 1.45 pooka isrump = true;
688 1.45 pooka fd = fd_host2rump(fd);
689 1.45 pooka } else {
690 1.45 pooka op_fchdir = GETSYSCALL(host, FCHDIR);
691 1.45 pooka isrump = false;
692 1.45 pooka }
693 1.45 pooka
694 1.45 pooka rv = op_fchdir(fd);
695 1.45 pooka if (rv == 0) {
696 1.45 pooka if (isrump)
697 1.45 pooka pwdinrump = true;
698 1.45 pooka else
699 1.45 pooka pwdinrump = false;
700 1.45 pooka }
701 1.45 pooka
702 1.45 pooka return rv;
703 1.45 pooka }
704 1.45 pooka
705 1.52 pooka int
706 1.57 pooka __getcwd(char *bufp, size_t len)
707 1.57 pooka {
708 1.57 pooka int (*op___getcwd)(char *, size_t);
709 1.57 pooka int rv;
710 1.57 pooka
711 1.57 pooka if (pwdinrump) {
712 1.57 pooka size_t prefixgap;
713 1.57 pooka bool iamslash;
714 1.57 pooka
715 1.57 pooka if (rumpprefix[rumpprefixlen-1] == '/')
716 1.57 pooka iamslash = true;
717 1.57 pooka else
718 1.57 pooka iamslash = false;
719 1.57 pooka
720 1.57 pooka if (iamslash)
721 1.57 pooka prefixgap = rumpprefixlen - 1; /* ``//+path'' */
722 1.57 pooka else
723 1.57 pooka prefixgap = rumpprefixlen; /* ``/pfx+/path'' */
724 1.57 pooka if (len <= prefixgap) {
725 1.57 pooka return ERANGE;
726 1.57 pooka }
727 1.57 pooka
728 1.57 pooka op___getcwd = GETSYSCALL(rump, __GETCWD);
729 1.57 pooka rv = op___getcwd(bufp + prefixgap, len - prefixgap);
730 1.57 pooka if (rv == -1)
731 1.57 pooka return rv;
732 1.57 pooka
733 1.57 pooka /* augment the "/" part only for a non-root path */
734 1.57 pooka memcpy(bufp, rumpprefix, rumpprefixlen);
735 1.57 pooka
736 1.57 pooka /* append / only to non-root cwd */
737 1.57 pooka if (rv != 2)
738 1.57 pooka bufp[prefixgap] = '/';
739 1.57 pooka
740 1.57 pooka /* don't append extra slash in the purely-slash case */
741 1.57 pooka if (rv == 2 && !iamslash)
742 1.57 pooka bufp[rumpprefixlen] = '\0';
743 1.57 pooka
744 1.57 pooka return rv;
745 1.57 pooka } else {
746 1.57 pooka op___getcwd = GETSYSCALL(host, __GETCWD);
747 1.57 pooka return op___getcwd(bufp, len);
748 1.57 pooka }
749 1.57 pooka }
750 1.57 pooka
751 1.57 pooka int
752 1.52 pooka rename(const char *from, const char *to)
753 1.52 pooka {
754 1.52 pooka int (*op_rename)(const char *, const char *);
755 1.52 pooka
756 1.52 pooka if (path_isrump(from)) {
757 1.52 pooka if (!path_isrump(to))
758 1.52 pooka return EXDEV;
759 1.52 pooka
760 1.52 pooka from = path_host2rump(from);
761 1.52 pooka to = path_host2rump(to);
762 1.52 pooka op_rename = GETSYSCALL(rump, RENAME);
763 1.52 pooka } else {
764 1.53 pooka if (path_isrump(to))
765 1.53 pooka return EXDEV;
766 1.53 pooka
767 1.52 pooka op_rename = GETSYSCALL(host, RENAME);
768 1.52 pooka }
769 1.52 pooka
770 1.52 pooka return op_rename(from, to);
771 1.52 pooka }
772 1.52 pooka
773 1.1 pooka int __socket30(int, int, int);
774 1.1 pooka int
775 1.1 pooka __socket30(int domain, int type, int protocol)
776 1.1 pooka {
777 1.17 pooka int (*op_socket)(int, int, int);
778 1.1 pooka int fd;
779 1.49 pooka bool isrump;
780 1.7 pooka
781 1.49 pooka isrump = domain < PF_MAX && rumpsockets[domain];
782 1.1 pooka
783 1.49 pooka if (isrump)
784 1.49 pooka op_socket = GETSYSCALL(rump, SOCKET);
785 1.49 pooka else
786 1.17 pooka op_socket = GETSYSCALL(host, SOCKET);
787 1.17 pooka fd = op_socket(domain, type, protocol);
788 1.2 pooka
789 1.49 pooka if (isrump)
790 1.7 pooka fd = fd_rump2host(fd);
791 1.47 pooka else
792 1.47 pooka fd = fd_dupgood(fd);
793 1.7 pooka DPRINTF(("socket <- %d\n", fd));
794 1.2 pooka
795 1.7 pooka return fd;
796 1.1 pooka }
797 1.1 pooka
798 1.1 pooka int
799 1.1 pooka accept(int s, struct sockaddr *addr, socklen_t *addrlen)
800 1.1 pooka {
801 1.17 pooka int (*op_accept)(int, struct sockaddr *, socklen_t *);
802 1.1 pooka int fd;
803 1.7 pooka bool isrump;
804 1.7 pooka
805 1.7 pooka isrump = fd_isrump(s);
806 1.1 pooka
807 1.2 pooka DPRINTF(("accept -> %d", s));
808 1.7 pooka if (isrump) {
809 1.17 pooka op_accept = GETSYSCALL(rump, ACCEPT);
810 1.7 pooka s = fd_host2rump(s);
811 1.7 pooka } else {
812 1.17 pooka op_accept = GETSYSCALL(host, ACCEPT);
813 1.7 pooka }
814 1.17 pooka fd = op_accept(s, addr, addrlen);
815 1.7 pooka if (fd != -1 && isrump)
816 1.7 pooka fd = fd_rump2host(fd);
817 1.47 pooka else
818 1.47 pooka fd = fd_dupgood(fd);
819 1.7 pooka
820 1.7 pooka DPRINTF((" <- %d\n", fd));
821 1.2 pooka
822 1.7 pooka return fd;
823 1.1 pooka }
824 1.1 pooka
825 1.17 pooka /*
826 1.17 pooka * ioctl and fcntl are varargs calls and need special treatment
827 1.17 pooka */
828 1.1 pooka int
829 1.17 pooka ioctl(int fd, unsigned long cmd, ...)
830 1.1 pooka {
831 1.17 pooka int (*op_ioctl)(int, unsigned long cmd, ...);
832 1.17 pooka va_list ap;
833 1.17 pooka int rv;
834 1.1 pooka
835 1.17 pooka DPRINTF(("ioctl -> %d\n", fd));
836 1.17 pooka if (fd_isrump(fd)) {
837 1.17 pooka fd = fd_host2rump(fd);
838 1.17 pooka op_ioctl = GETSYSCALL(rump, IOCTL);
839 1.7 pooka } else {
840 1.17 pooka op_ioctl = GETSYSCALL(host, IOCTL);
841 1.7 pooka }
842 1.1 pooka
843 1.17 pooka va_start(ap, cmd);
844 1.17 pooka rv = op_ioctl(fd, cmd, va_arg(ap, void *));
845 1.17 pooka va_end(ap);
846 1.17 pooka return rv;
847 1.1 pooka }
848 1.1 pooka
849 1.40 pooka #include <syslog.h>
850 1.1 pooka int
851 1.17 pooka fcntl(int fd, int cmd, ...)
852 1.1 pooka {
853 1.17 pooka int (*op_fcntl)(int, int, ...);
854 1.17 pooka va_list ap;
855 1.40 pooka int rv, minfd, i;
856 1.40 pooka
857 1.40 pooka DPRINTF(("fcntl -> %d (cmd %d)\n", fd, cmd));
858 1.40 pooka
859 1.40 pooka switch (cmd) {
860 1.40 pooka case F_DUPFD:
861 1.40 pooka va_start(ap, cmd);
862 1.40 pooka minfd = va_arg(ap, int);
863 1.40 pooka va_end(ap);
864 1.40 pooka return dodup(fd, minfd);
865 1.40 pooka
866 1.40 pooka case F_CLOSEM:
867 1.40 pooka /*
868 1.40 pooka * So, if fd < HIJACKOFF, we want to do a host closem.
869 1.40 pooka */
870 1.40 pooka
871 1.40 pooka if (fd < HIJACK_FDOFF) {
872 1.40 pooka int closemfd = fd;
873 1.1 pooka
874 1.40 pooka if (rumpclient__closenotify(&closemfd,
875 1.39 pooka RUMPCLIENT_CLOSE_FCLOSEM) == -1)
876 1.39 pooka return -1;
877 1.40 pooka op_fcntl = GETSYSCALL(host, FCNTL);
878 1.40 pooka rv = op_fcntl(closemfd, cmd);
879 1.40 pooka if (rv)
880 1.40 pooka return rv;
881 1.40 pooka }
882 1.40 pooka
883 1.40 pooka /*
884 1.40 pooka * Additionally, we want to do a rump closem, but only
885 1.40 pooka * for the file descriptors not within the dup2mask.
886 1.40 pooka */
887 1.40 pooka
888 1.40 pooka /* why don't we offer fls()? */
889 1.47 pooka for (i = 15; i >= 0; i--) {
890 1.47 pooka if (ISDUP2D(i))
891 1.40 pooka break;
892 1.40 pooka }
893 1.40 pooka
894 1.40 pooka if (fd >= HIJACK_FDOFF)
895 1.40 pooka fd -= HIJACK_FDOFF;
896 1.40 pooka else
897 1.40 pooka fd = 0;
898 1.40 pooka fd = MAX(i+1, fd);
899 1.40 pooka
900 1.40 pooka /* hmm, maybe we should close rump fd's not within dup2mask? */
901 1.40 pooka
902 1.40 pooka return rump_sys_fcntl(fd, F_CLOSEM);
903 1.40 pooka
904 1.40 pooka case F_MAXFD:
905 1.40 pooka /*
906 1.40 pooka * For maxfd, if there's a rump kernel fd, return
907 1.40 pooka * it hostified. Otherwise, return host's MAXFD
908 1.40 pooka * return value.
909 1.40 pooka */
910 1.40 pooka if ((rv = rump_sys_fcntl(fd, F_MAXFD)) != -1) {
911 1.40 pooka /*
912 1.40 pooka * This might go a little wrong in case
913 1.40 pooka * of dup2 to [012], but I'm not sure if
914 1.40 pooka * there's a justification for tracking
915 1.40 pooka * that info. Consider e.g.
916 1.40 pooka * dup2(rumpfd, 2) followed by rump_sys_open()
917 1.40 pooka * returning 1. We should return 1+HIJACKOFF,
918 1.40 pooka * not 2+HIJACKOFF. However, if [01] is not
919 1.40 pooka * open, the correct return value is 2.
920 1.40 pooka */
921 1.40 pooka return fd_rump2host(fd);
922 1.40 pooka } else {
923 1.40 pooka op_fcntl = GETSYSCALL(host, FCNTL);
924 1.40 pooka return op_fcntl(fd, F_MAXFD);
925 1.40 pooka }
926 1.40 pooka /*NOTREACHED*/
927 1.40 pooka
928 1.40 pooka default:
929 1.40 pooka if (fd_isrump(fd)) {
930 1.40 pooka fd = fd_host2rump(fd);
931 1.40 pooka op_fcntl = GETSYSCALL(rump, FCNTL);
932 1.40 pooka } else {
933 1.40 pooka op_fcntl = GETSYSCALL(host, FCNTL);
934 1.40 pooka }
935 1.40 pooka
936 1.40 pooka va_start(ap, cmd);
937 1.40 pooka rv = op_fcntl(fd, cmd, va_arg(ap, void *));
938 1.40 pooka va_end(ap);
939 1.40 pooka return rv;
940 1.7 pooka }
941 1.40 pooka /*NOTREACHED*/
942 1.1 pooka }
943 1.1 pooka
944 1.39 pooka int
945 1.39 pooka close(int fd)
946 1.39 pooka {
947 1.39 pooka int (*op_close)(int);
948 1.39 pooka int rv;
949 1.39 pooka
950 1.39 pooka DPRINTF(("close -> %d\n", fd));
951 1.39 pooka if (fd_isrump(fd)) {
952 1.39 pooka int undup2 = 0;
953 1.39 pooka
954 1.47 pooka fd = fd_host2rump(fd);
955 1.47 pooka if (ISDUP2ALIAS(fd)) {
956 1.47 pooka _DIAGASSERT(ISDUP2D(fd));
957 1.47 pooka CLRDUP2ALIAS(fd);
958 1.47 pooka return 0;
959 1.47 pooka }
960 1.47 pooka
961 1.39 pooka if (ISDUP2D(fd))
962 1.39 pooka undup2 = 1;
963 1.39 pooka op_close = GETSYSCALL(rump, CLOSE);
964 1.39 pooka rv = op_close(fd);
965 1.39 pooka if (rv == 0 && undup2)
966 1.40 pooka CLRDUP2(fd);
967 1.39 pooka } else {
968 1.39 pooka if (rumpclient__closenotify(&fd, RUMPCLIENT_CLOSE_CLOSE) == -1)
969 1.39 pooka return -1;
970 1.39 pooka op_close = GETSYSCALL(host, CLOSE);
971 1.39 pooka rv = op_close(fd);
972 1.39 pooka }
973 1.39 pooka
974 1.39 pooka return rv;
975 1.39 pooka }
976 1.39 pooka
977 1.17 pooka /*
978 1.17 pooka * write cannot issue a standard debug printf due to recursion
979 1.17 pooka */
980 1.1 pooka ssize_t
981 1.17 pooka write(int fd, const void *buf, size_t blen)
982 1.1 pooka {
983 1.17 pooka ssize_t (*op_write)(int, const void *, size_t);
984 1.1 pooka
985 1.17 pooka if (fd_isrump(fd)) {
986 1.17 pooka fd = fd_host2rump(fd);
987 1.17 pooka op_write = GETSYSCALL(rump, WRITE);
988 1.16 pooka } else {
989 1.17 pooka op_write = GETSYSCALL(host, WRITE);
990 1.16 pooka }
991 1.1 pooka
992 1.17 pooka return op_write(fd, buf, blen);
993 1.2 pooka }
994 1.2 pooka
995 1.2 pooka /*
996 1.2 pooka * dup2 is special. we allow dup2 of a rump kernel fd to 0-2 since
997 1.2 pooka * many programs do that. dup2 of a rump kernel fd to another value
998 1.2 pooka * not >= fdoff is an error.
999 1.2 pooka *
1000 1.2 pooka * Note: cannot rump2host newd, because it is often hardcoded.
1001 1.2 pooka */
1002 1.2 pooka int
1003 1.2 pooka dup2(int oldd, int newd)
1004 1.2 pooka {
1005 1.17 pooka int (*host_dup2)(int, int);
1006 1.2 pooka int rv;
1007 1.2 pooka
1008 1.2 pooka DPRINTF(("dup2 -> %d (o) -> %d (n)\n", oldd, newd));
1009 1.2 pooka
1010 1.2 pooka if (fd_isrump(oldd)) {
1011 1.2 pooka if (!(newd >= 0 && newd <= 2))
1012 1.2 pooka return EBADF;
1013 1.2 pooka oldd = fd_host2rump(oldd);
1014 1.47 pooka if (oldd == newd) {
1015 1.47 pooka SETDUP2(newd);
1016 1.47 pooka SETDUP2ALIAS(newd);
1017 1.47 pooka return newd;
1018 1.47 pooka }
1019 1.2 pooka rv = rump_sys_dup2(oldd, newd);
1020 1.2 pooka if (rv != -1)
1021 1.40 pooka SETDUP2(newd);
1022 1.2 pooka } else {
1023 1.17 pooka host_dup2 = syscalls[DUALCALL_DUP2].bs_host;
1024 1.39 pooka if (rumpclient__closenotify(&newd, RUMPCLIENT_CLOSE_DUP2) == -1)
1025 1.39 pooka return -1;
1026 1.10 pooka rv = host_dup2(oldd, newd);
1027 1.2 pooka }
1028 1.10 pooka
1029 1.10 pooka return rv;
1030 1.2 pooka }
1031 1.2 pooka
1032 1.34 pooka int
1033 1.34 pooka dup(int oldd)
1034 1.34 pooka {
1035 1.34 pooka
1036 1.40 pooka return dodup(oldd, 0);
1037 1.34 pooka }
1038 1.34 pooka
1039 1.2 pooka pid_t
1040 1.2 pooka fork()
1041 1.2 pooka {
1042 1.2 pooka pid_t rv;
1043 1.2 pooka
1044 1.2 pooka DPRINTF(("fork\n"));
1045 1.2 pooka
1046 1.43 pooka rv = rumpclient__dofork(host_fork);
1047 1.2 pooka
1048 1.2 pooka DPRINTF(("fork returns %d\n", rv));
1049 1.2 pooka return rv;
1050 1.1 pooka }
1051 1.43 pooka /* we do not have the luxury of not requiring a stackframe */
1052 1.43 pooka __strong_alias(__vfork14,fork);
1053 1.1 pooka
1054 1.25 pooka int
1055 1.25 pooka daemon(int nochdir, int noclose)
1056 1.25 pooka {
1057 1.25 pooka struct rumpclient_fork *rf;
1058 1.25 pooka
1059 1.25 pooka if ((rf = rumpclient_prefork()) == NULL)
1060 1.25 pooka return -1;
1061 1.25 pooka
1062 1.25 pooka if (host_daemon(nochdir, noclose) == -1)
1063 1.25 pooka return -1;
1064 1.25 pooka
1065 1.25 pooka if (rumpclient_fork_init(rf) == -1)
1066 1.25 pooka return -1;
1067 1.25 pooka
1068 1.25 pooka return 0;
1069 1.25 pooka }
1070 1.25 pooka
1071 1.39 pooka int
1072 1.42 pooka execve(const char *path, char *const argv[], char *const envp[])
1073 1.39 pooka {
1074 1.39 pooka char buf[128];
1075 1.39 pooka char *dup2str;
1076 1.49 pooka const char *pwdinrumpstr;
1077 1.42 pooka char **newenv;
1078 1.42 pooka size_t nelem;
1079 1.42 pooka int rv, sverrno;
1080 1.45 pooka int bonus = 1, i = 0;
1081 1.45 pooka
1082 1.45 pooka if (dup2mask) {
1083 1.45 pooka snprintf(buf, sizeof(buf), "RUMPHIJACK__DUP2MASK=%u", dup2mask);
1084 1.45 pooka dup2str = malloc(strlen(buf)+1);
1085 1.45 pooka if (dup2str == NULL)
1086 1.45 pooka return ENOMEM;
1087 1.45 pooka strcpy(dup2str, buf);
1088 1.45 pooka bonus++;
1089 1.45 pooka } else {
1090 1.45 pooka dup2str = NULL;
1091 1.45 pooka }
1092 1.39 pooka
1093 1.45 pooka if (pwdinrump) {
1094 1.49 pooka pwdinrumpstr = "RUMPHIJACK__PWDINRUMP=true";
1095 1.45 pooka bonus++;
1096 1.45 pooka } else {
1097 1.45 pooka pwdinrumpstr = NULL;
1098 1.45 pooka }
1099 1.39 pooka
1100 1.42 pooka for (nelem = 0; envp && envp[nelem]; nelem++)
1101 1.42 pooka continue;
1102 1.45 pooka newenv = malloc(sizeof(*newenv) * nelem+bonus);
1103 1.42 pooka if (newenv == NULL) {
1104 1.39 pooka free(dup2str);
1105 1.42 pooka return ENOMEM;
1106 1.39 pooka }
1107 1.42 pooka memcpy(newenv, envp, nelem*sizeof(*newenv));
1108 1.45 pooka if (dup2str) {
1109 1.45 pooka newenv[nelem+i] = dup2str;
1110 1.45 pooka i++;
1111 1.45 pooka }
1112 1.45 pooka if (pwdinrumpstr) {
1113 1.49 pooka newenv[nelem+i] = __UNCONST(pwdinrumpstr);
1114 1.45 pooka i++;
1115 1.45 pooka }
1116 1.45 pooka newenv[nelem+i] = NULL;
1117 1.45 pooka _DIAGASSERT(i < bonus);
1118 1.42 pooka
1119 1.42 pooka rv = rumpclient_exec(path, argv, newenv);
1120 1.42 pooka
1121 1.42 pooka _DIAGASSERT(rv != 0);
1122 1.42 pooka sverrno = errno;
1123 1.42 pooka free(newenv);
1124 1.42 pooka free(dup2str);
1125 1.42 pooka errno = sverrno;
1126 1.39 pooka return rv;
1127 1.39 pooka }
1128 1.39 pooka
1129 1.1 pooka /*
1130 1.17 pooka * select is done by calling poll.
1131 1.1 pooka */
1132 1.1 pooka int
1133 1.29 pooka REALSELECT(int nfds, fd_set *readfds, fd_set *writefds, fd_set *exceptfds,
1134 1.4 pooka struct timeval *timeout)
1135 1.1 pooka {
1136 1.4 pooka struct pollfd *pfds;
1137 1.4 pooka struct timespec ts, *tsp = NULL;
1138 1.19 pooka nfds_t realnfds;
1139 1.19 pooka int i, j;
1140 1.4 pooka int rv, incr;
1141 1.4 pooka
1142 1.7 pooka DPRINTF(("select\n"));
1143 1.7 pooka
1144 1.4 pooka /*
1145 1.4 pooka * Well, first we must scan the fds to figure out how many
1146 1.4 pooka * fds there really are. This is because up to and including
1147 1.17 pooka * nb5 poll() silently refuses nfds > process_maxopen_fds.
1148 1.4 pooka * Seems to be fixed in current, thank the maker.
1149 1.4 pooka * god damn cluster...bomb.
1150 1.4 pooka */
1151 1.4 pooka
1152 1.4 pooka for (i = 0, realnfds = 0; i < nfds; i++) {
1153 1.4 pooka if (readfds && FD_ISSET(i, readfds)) {
1154 1.4 pooka realnfds++;
1155 1.4 pooka continue;
1156 1.4 pooka }
1157 1.4 pooka if (writefds && FD_ISSET(i, writefds)) {
1158 1.4 pooka realnfds++;
1159 1.4 pooka continue;
1160 1.4 pooka }
1161 1.4 pooka if (exceptfds && FD_ISSET(i, exceptfds)) {
1162 1.4 pooka realnfds++;
1163 1.4 pooka continue;
1164 1.1 pooka }
1165 1.1 pooka }
1166 1.1 pooka
1167 1.6 pooka if (realnfds) {
1168 1.38 pooka pfds = calloc(realnfds, sizeof(*pfds));
1169 1.6 pooka if (!pfds)
1170 1.6 pooka return -1;
1171 1.6 pooka } else {
1172 1.6 pooka pfds = NULL;
1173 1.6 pooka }
1174 1.1 pooka
1175 1.4 pooka for (i = 0, j = 0; i < nfds; i++) {
1176 1.4 pooka incr = 0;
1177 1.4 pooka if (readfds && FD_ISSET(i, readfds)) {
1178 1.4 pooka pfds[j].fd = i;
1179 1.4 pooka pfds[j].events |= POLLIN;
1180 1.4 pooka incr=1;
1181 1.4 pooka }
1182 1.4 pooka if (writefds && FD_ISSET(i, writefds)) {
1183 1.4 pooka pfds[j].fd = i;
1184 1.4 pooka pfds[j].events |= POLLOUT;
1185 1.4 pooka incr=1;
1186 1.4 pooka }
1187 1.4 pooka if (exceptfds && FD_ISSET(i, exceptfds)) {
1188 1.4 pooka pfds[j].fd = i;
1189 1.4 pooka pfds[j].events |= POLLHUP|POLLERR;
1190 1.4 pooka incr=1;
1191 1.1 pooka }
1192 1.4 pooka if (incr)
1193 1.4 pooka j++;
1194 1.1 pooka }
1195 1.37 pooka assert(j == (int)realnfds);
1196 1.1 pooka
1197 1.4 pooka if (timeout) {
1198 1.4 pooka TIMEVAL_TO_TIMESPEC(timeout, &ts);
1199 1.4 pooka tsp = &ts;
1200 1.4 pooka }
1201 1.29 pooka rv = REALPOLLTS(pfds, realnfds, tsp, NULL);
1202 1.36 pooka /*
1203 1.36 pooka * "If select() returns with an error the descriptor sets
1204 1.36 pooka * will be unmodified"
1205 1.36 pooka */
1206 1.36 pooka if (rv < 0)
1207 1.4 pooka goto out;
1208 1.4 pooka
1209 1.4 pooka /*
1210 1.36 pooka * zero out results (can't use FD_ZERO for the
1211 1.36 pooka * obvious select-me-not reason). whee.
1212 1.36 pooka *
1213 1.36 pooka * We do this here since some software ignores the return
1214 1.36 pooka * value of select, and hence if the timeout expires, it may
1215 1.36 pooka * assume all input descriptors have activity.
1216 1.4 pooka */
1217 1.4 pooka for (i = 0; i < nfds; i++) {
1218 1.4 pooka if (readfds)
1219 1.4 pooka FD_CLR(i, readfds);
1220 1.4 pooka if (writefds)
1221 1.4 pooka FD_CLR(i, writefds);
1222 1.4 pooka if (exceptfds)
1223 1.4 pooka FD_CLR(i, exceptfds);
1224 1.1 pooka }
1225 1.36 pooka if (rv == 0)
1226 1.36 pooka goto out;
1227 1.1 pooka
1228 1.36 pooka /*
1229 1.36 pooka * We have >0 fds with activity. Harvest the results.
1230 1.36 pooka */
1231 1.19 pooka for (i = 0; i < (int)realnfds; i++) {
1232 1.4 pooka if (readfds) {
1233 1.4 pooka if (pfds[i].revents & POLLIN) {
1234 1.4 pooka FD_SET(pfds[i].fd, readfds);
1235 1.4 pooka }
1236 1.4 pooka }
1237 1.4 pooka if (writefds) {
1238 1.4 pooka if (pfds[i].revents & POLLOUT) {
1239 1.4 pooka FD_SET(pfds[i].fd, writefds);
1240 1.4 pooka }
1241 1.4 pooka }
1242 1.4 pooka if (exceptfds) {
1243 1.4 pooka if (pfds[i].revents & (POLLHUP|POLLERR)) {
1244 1.4 pooka FD_SET(pfds[i].fd, exceptfds);
1245 1.4 pooka }
1246 1.4 pooka }
1247 1.1 pooka }
1248 1.1 pooka
1249 1.4 pooka out:
1250 1.4 pooka free(pfds);
1251 1.1 pooka return rv;
1252 1.1 pooka }
1253 1.1 pooka
1254 1.1 pooka static void
1255 1.1 pooka checkpoll(struct pollfd *fds, nfds_t nfds, int *hostcall, int *rumpcall)
1256 1.1 pooka {
1257 1.1 pooka nfds_t i;
1258 1.1 pooka
1259 1.1 pooka for (i = 0; i < nfds; i++) {
1260 1.12 pooka if (fds[i].fd == -1)
1261 1.12 pooka continue;
1262 1.12 pooka
1263 1.2 pooka if (fd_isrump(fds[i].fd))
1264 1.2 pooka (*rumpcall)++;
1265 1.2 pooka else
1266 1.1 pooka (*hostcall)++;
1267 1.1 pooka }
1268 1.1 pooka }
1269 1.1 pooka
1270 1.1 pooka static void
1271 1.2 pooka adjustpoll(struct pollfd *fds, nfds_t nfds, int (*fdadj)(int))
1272 1.1 pooka {
1273 1.1 pooka nfds_t i;
1274 1.1 pooka
1275 1.1 pooka for (i = 0; i < nfds; i++) {
1276 1.2 pooka fds[i].fd = fdadj(fds[i].fd);
1277 1.1 pooka }
1278 1.1 pooka }
1279 1.1 pooka
1280 1.1 pooka /*
1281 1.1 pooka * poll is easy as long as the call comes in the fds only in one
1282 1.1 pooka * kernel. otherwise its quite tricky...
1283 1.1 pooka */
1284 1.1 pooka struct pollarg {
1285 1.1 pooka struct pollfd *pfds;
1286 1.1 pooka nfds_t nfds;
1287 1.3 pooka const struct timespec *ts;
1288 1.3 pooka const sigset_t *sigmask;
1289 1.1 pooka int pipefd;
1290 1.1 pooka int errnum;
1291 1.1 pooka };
1292 1.1 pooka
1293 1.1 pooka static void *
1294 1.1 pooka hostpoll(void *arg)
1295 1.1 pooka {
1296 1.17 pooka int (*op_pollts)(struct pollfd *, nfds_t, const struct timespec *,
1297 1.17 pooka const sigset_t *);
1298 1.1 pooka struct pollarg *parg = arg;
1299 1.1 pooka intptr_t rv;
1300 1.1 pooka
1301 1.35 pooka op_pollts = GETSYSCALL(host, POLLTS);
1302 1.17 pooka rv = op_pollts(parg->pfds, parg->nfds, parg->ts, parg->sigmask);
1303 1.1 pooka if (rv == -1)
1304 1.1 pooka parg->errnum = errno;
1305 1.1 pooka rump_sys_write(parg->pipefd, &rv, sizeof(rv));
1306 1.1 pooka
1307 1.1 pooka return (void *)(intptr_t)rv;
1308 1.1 pooka }
1309 1.1 pooka
1310 1.1 pooka int
1311 1.29 pooka REALPOLLTS(struct pollfd *fds, nfds_t nfds, const struct timespec *ts,
1312 1.3 pooka const sigset_t *sigmask)
1313 1.1 pooka {
1314 1.3 pooka int (*op_pollts)(struct pollfd *, nfds_t, const struct timespec *,
1315 1.3 pooka const sigset_t *);
1316 1.17 pooka int (*host_close)(int);
1317 1.1 pooka int hostcall = 0, rumpcall = 0;
1318 1.1 pooka pthread_t pt;
1319 1.1 pooka nfds_t i;
1320 1.1 pooka int rv;
1321 1.1 pooka
1322 1.2 pooka DPRINTF(("poll\n"));
1323 1.1 pooka checkpoll(fds, nfds, &hostcall, &rumpcall);
1324 1.1 pooka
1325 1.1 pooka if (hostcall && rumpcall) {
1326 1.1 pooka struct pollfd *pfd_host = NULL, *pfd_rump = NULL;
1327 1.1 pooka int rpipe[2] = {-1,-1}, hpipe[2] = {-1,-1};
1328 1.1 pooka struct pollarg parg;
1329 1.1 pooka uintptr_t lrv;
1330 1.1 pooka int sverrno = 0, trv;
1331 1.1 pooka
1332 1.1 pooka /*
1333 1.1 pooka * ok, this is where it gets tricky. We must support
1334 1.1 pooka * this since it's a very common operation in certain
1335 1.1 pooka * types of software (telnet, netcat, etc). We allocate
1336 1.1 pooka * two vectors and run two poll commands in separate
1337 1.1 pooka * threads. Whichever returns first "wins" and the
1338 1.1 pooka * other kernel's fds won't show activity.
1339 1.1 pooka */
1340 1.1 pooka rv = -1;
1341 1.1 pooka
1342 1.1 pooka /* allocate full vector for O(n) joining after call */
1343 1.1 pooka pfd_host = malloc(sizeof(*pfd_host)*(nfds+1));
1344 1.1 pooka if (!pfd_host)
1345 1.1 pooka goto out;
1346 1.1 pooka pfd_rump = malloc(sizeof(*pfd_rump)*(nfds+1));
1347 1.1 pooka if (!pfd_rump) {
1348 1.1 pooka goto out;
1349 1.1 pooka }
1350 1.1 pooka
1351 1.59 pooka /*
1352 1.59 pooka * then, open two pipes, one for notifications
1353 1.59 pooka * to each kernel.
1354 1.59 pooka */
1355 1.59 pooka if ((rv = rump_sys_pipe(rpipe)) == -1) {
1356 1.59 pooka sverrno = errno;
1357 1.59 pooka }
1358 1.59 pooka if (rv == 0 && (rv = pipe(hpipe)) == -1) {
1359 1.59 pooka sverrno = errno;
1360 1.59 pooka }
1361 1.59 pooka
1362 1.59 pooka /* split vectors (or signal errors) */
1363 1.1 pooka for (i = 0; i < nfds; i++) {
1364 1.59 pooka int fd;
1365 1.59 pooka
1366 1.59 pooka fds[i].revents = 0;
1367 1.3 pooka if (fds[i].fd == -1) {
1368 1.3 pooka pfd_host[i].fd = -1;
1369 1.3 pooka pfd_rump[i].fd = -1;
1370 1.3 pooka } else if (fd_isrump(fds[i].fd)) {
1371 1.2 pooka pfd_host[i].fd = -1;
1372 1.59 pooka fd = fd_host2rump(fds[i].fd);
1373 1.59 pooka if (fd == rpipe[0] || fd == rpipe[1]) {
1374 1.59 pooka fds[i].revents = POLLNVAL;
1375 1.59 pooka if (rv != -1)
1376 1.59 pooka rv++;
1377 1.59 pooka }
1378 1.59 pooka pfd_rump[i].fd = fd;
1379 1.2 pooka pfd_rump[i].events = fds[i].events;
1380 1.2 pooka } else {
1381 1.2 pooka pfd_rump[i].fd = -1;
1382 1.59 pooka fd = fds[i].fd;
1383 1.59 pooka if (fd == hpipe[0] || fd == hpipe[1]) {
1384 1.59 pooka fds[i].revents = POLLNVAL;
1385 1.59 pooka if (rv != -1)
1386 1.59 pooka rv++;
1387 1.59 pooka }
1388 1.59 pooka pfd_host[i].fd = fd;
1389 1.1 pooka pfd_host[i].events = fds[i].events;
1390 1.1 pooka }
1391 1.39 pooka pfd_rump[i].revents = pfd_host[i].revents = 0;
1392 1.1 pooka }
1393 1.59 pooka if (rv) {
1394 1.1 pooka goto out;
1395 1.59 pooka }
1396 1.1 pooka
1397 1.1 pooka pfd_host[nfds].fd = hpipe[0];
1398 1.1 pooka pfd_host[nfds].events = POLLIN;
1399 1.1 pooka pfd_rump[nfds].fd = rpipe[0];
1400 1.1 pooka pfd_rump[nfds].events = POLLIN;
1401 1.1 pooka
1402 1.1 pooka /*
1403 1.1 pooka * then, create a thread to do host part and meanwhile
1404 1.1 pooka * do rump kernel part right here
1405 1.1 pooka */
1406 1.1 pooka
1407 1.1 pooka parg.pfds = pfd_host;
1408 1.1 pooka parg.nfds = nfds+1;
1409 1.3 pooka parg.ts = ts;
1410 1.3 pooka parg.sigmask = sigmask;
1411 1.1 pooka parg.pipefd = rpipe[1];
1412 1.1 pooka pthread_create(&pt, NULL, hostpoll, &parg);
1413 1.1 pooka
1414 1.35 pooka op_pollts = GETSYSCALL(rump, POLLTS);
1415 1.3 pooka lrv = op_pollts(pfd_rump, nfds+1, ts, NULL);
1416 1.1 pooka sverrno = errno;
1417 1.1 pooka write(hpipe[1], &rv, sizeof(rv));
1418 1.1 pooka pthread_join(pt, (void *)&trv);
1419 1.1 pooka
1420 1.1 pooka /* check who "won" and merge results */
1421 1.1 pooka if (lrv != 0 && pfd_host[nfds].revents & POLLIN) {
1422 1.1 pooka rv = trv;
1423 1.1 pooka
1424 1.1 pooka for (i = 0; i < nfds; i++) {
1425 1.1 pooka if (pfd_rump[i].fd != -1)
1426 1.1 pooka fds[i].revents = pfd_rump[i].revents;
1427 1.1 pooka }
1428 1.1 pooka sverrno = parg.errnum;
1429 1.1 pooka } else if (trv != 0 && pfd_rump[nfds].revents & POLLIN) {
1430 1.1 pooka rv = trv;
1431 1.1 pooka
1432 1.1 pooka for (i = 0; i < nfds; i++) {
1433 1.1 pooka if (pfd_host[i].fd != -1)
1434 1.1 pooka fds[i].revents = pfd_host[i].revents;
1435 1.1 pooka }
1436 1.1 pooka } else {
1437 1.1 pooka rv = 0;
1438 1.1 pooka }
1439 1.1 pooka
1440 1.1 pooka out:
1441 1.35 pooka host_close = GETSYSCALL(host, CLOSE);
1442 1.1 pooka if (rpipe[0] != -1)
1443 1.1 pooka rump_sys_close(rpipe[0]);
1444 1.1 pooka if (rpipe[1] != -1)
1445 1.1 pooka rump_sys_close(rpipe[1]);
1446 1.1 pooka if (hpipe[0] != -1)
1447 1.9 pooka host_close(hpipe[0]);
1448 1.1 pooka if (hpipe[1] != -1)
1449 1.9 pooka host_close(hpipe[1]);
1450 1.1 pooka free(pfd_host);
1451 1.1 pooka free(pfd_rump);
1452 1.1 pooka errno = sverrno;
1453 1.1 pooka } else {
1454 1.1 pooka if (hostcall) {
1455 1.35 pooka op_pollts = GETSYSCALL(host, POLLTS);
1456 1.1 pooka } else {
1457 1.35 pooka op_pollts = GETSYSCALL(rump, POLLTS);
1458 1.2 pooka adjustpoll(fds, nfds, fd_host2rump);
1459 1.1 pooka }
1460 1.1 pooka
1461 1.3 pooka rv = op_pollts(fds, nfds, ts, sigmask);
1462 1.1 pooka if (rumpcall)
1463 1.2 pooka adjustpoll(fds, nfds, fd_rump2host);
1464 1.1 pooka }
1465 1.1 pooka
1466 1.1 pooka return rv;
1467 1.1 pooka }
1468 1.1 pooka
1469 1.1 pooka int
1470 1.24 pooka poll(struct pollfd *fds, nfds_t nfds, int timeout)
1471 1.1 pooka {
1472 1.3 pooka struct timespec ts;
1473 1.3 pooka struct timespec *tsp = NULL;
1474 1.3 pooka
1475 1.3 pooka if (timeout != INFTIM) {
1476 1.3 pooka ts.tv_sec = timeout / 1000;
1477 1.11 pooka ts.tv_nsec = (timeout % 1000) * 1000*1000;
1478 1.3 pooka
1479 1.3 pooka tsp = &ts;
1480 1.3 pooka }
1481 1.1 pooka
1482 1.29 pooka return REALPOLLTS(fds, nfds, tsp, NULL);
1483 1.1 pooka }
1484 1.10 pooka
1485 1.10 pooka int
1486 1.34 pooka REALKEVENT(int kq, const struct kevent *changelist, size_t nchanges,
1487 1.34 pooka struct kevent *eventlist, size_t nevents,
1488 1.34 pooka const struct timespec *timeout)
1489 1.10 pooka {
1490 1.34 pooka int (*op_kevent)(int, const struct kevent *, size_t,
1491 1.34 pooka struct kevent *, size_t, const struct timespec *);
1492 1.34 pooka const struct kevent *ev;
1493 1.34 pooka size_t i;
1494 1.10 pooka
1495 1.34 pooka /*
1496 1.34 pooka * Check that we don't attempt to kevent rump kernel fd's.
1497 1.34 pooka * That needs similar treatment to select/poll, but is slightly
1498 1.34 pooka * trickier since we need to manage to different kq descriptors.
1499 1.34 pooka * (TODO, in case you're wondering).
1500 1.34 pooka */
1501 1.34 pooka for (i = 0; i < nchanges; i++) {
1502 1.34 pooka ev = &changelist[i];
1503 1.34 pooka if (ev->filter == EVFILT_READ || ev->filter == EVFILT_WRITE ||
1504 1.34 pooka ev->filter == EVFILT_VNODE) {
1505 1.40 pooka if (fd_isrump((int)ev->ident))
1506 1.34 pooka return ENOTSUP;
1507 1.34 pooka }
1508 1.27 pooka }
1509 1.10 pooka
1510 1.35 pooka op_kevent = GETSYSCALL(host, KEVENT);
1511 1.34 pooka return op_kevent(kq, changelist, nchanges, eventlist, nevents, timeout);
1512 1.10 pooka }
1513 1.17 pooka
1514 1.17 pooka /*
1515 1.62 pooka * mmapping from a rump kernel is not supported, so disallow it.
1516 1.62 pooka */
1517 1.62 pooka void *
1518 1.62 pooka mmap(void *addr, size_t len, int prot, int flags, int fd, off_t offset)
1519 1.62 pooka {
1520 1.62 pooka
1521 1.62 pooka if (flags & MAP_FILE && fd_isrump(fd)) {
1522 1.62 pooka errno = ENOSYS;
1523 1.62 pooka return MAP_FAILED;
1524 1.62 pooka }
1525 1.62 pooka return host_mmap(addr, len, prot, flags, fd, offset);
1526 1.62 pooka }
1527 1.62 pooka
1528 1.62 pooka /*
1529 1.17 pooka * Rest are std type calls.
1530 1.17 pooka */
1531 1.17 pooka
1532 1.17 pooka FDCALL(int, bind, DUALCALL_BIND, \
1533 1.17 pooka (int fd, const struct sockaddr *name, socklen_t namelen), \
1534 1.17 pooka (int, const struct sockaddr *, socklen_t), \
1535 1.17 pooka (fd, name, namelen))
1536 1.17 pooka
1537 1.17 pooka FDCALL(int, connect, DUALCALL_CONNECT, \
1538 1.17 pooka (int fd, const struct sockaddr *name, socklen_t namelen), \
1539 1.17 pooka (int, const struct sockaddr *, socklen_t), \
1540 1.17 pooka (fd, name, namelen))
1541 1.17 pooka
1542 1.17 pooka FDCALL(int, getpeername, DUALCALL_GETPEERNAME, \
1543 1.17 pooka (int fd, struct sockaddr *name, socklen_t *namelen), \
1544 1.17 pooka (int, struct sockaddr *, socklen_t *), \
1545 1.17 pooka (fd, name, namelen))
1546 1.17 pooka
1547 1.17 pooka FDCALL(int, getsockname, DUALCALL_GETSOCKNAME, \
1548 1.17 pooka (int fd, struct sockaddr *name, socklen_t *namelen), \
1549 1.17 pooka (int, struct sockaddr *, socklen_t *), \
1550 1.17 pooka (fd, name, namelen))
1551 1.17 pooka
1552 1.17 pooka FDCALL(int, listen, DUALCALL_LISTEN, \
1553 1.17 pooka (int fd, int backlog), \
1554 1.17 pooka (int, int), \
1555 1.17 pooka (fd, backlog))
1556 1.17 pooka
1557 1.17 pooka FDCALL(ssize_t, recvfrom, DUALCALL_RECVFROM, \
1558 1.17 pooka (int fd, void *buf, size_t len, int flags, \
1559 1.17 pooka struct sockaddr *from, socklen_t *fromlen), \
1560 1.17 pooka (int, void *, size_t, int, struct sockaddr *, socklen_t *), \
1561 1.17 pooka (fd, buf, len, flags, from, fromlen))
1562 1.17 pooka
1563 1.17 pooka FDCALL(ssize_t, sendto, DUALCALL_SENDTO, \
1564 1.17 pooka (int fd, const void *buf, size_t len, int flags, \
1565 1.17 pooka const struct sockaddr *to, socklen_t tolen), \
1566 1.17 pooka (int, const void *, size_t, int, \
1567 1.17 pooka const struct sockaddr *, socklen_t), \
1568 1.17 pooka (fd, buf, len, flags, to, tolen))
1569 1.17 pooka
1570 1.17 pooka FDCALL(ssize_t, recvmsg, DUALCALL_RECVMSG, \
1571 1.17 pooka (int fd, struct msghdr *msg, int flags), \
1572 1.17 pooka (int, struct msghdr *, int), \
1573 1.17 pooka (fd, msg, flags))
1574 1.17 pooka
1575 1.17 pooka FDCALL(ssize_t, sendmsg, DUALCALL_SENDMSG, \
1576 1.17 pooka (int fd, const struct msghdr *msg, int flags), \
1577 1.17 pooka (int, const struct msghdr *, int), \
1578 1.17 pooka (fd, msg, flags))
1579 1.17 pooka
1580 1.17 pooka FDCALL(int, getsockopt, DUALCALL_GETSOCKOPT, \
1581 1.17 pooka (int fd, int level, int optn, void *optval, socklen_t *optlen), \
1582 1.17 pooka (int, int, int, void *, socklen_t *), \
1583 1.17 pooka (fd, level, optn, optval, optlen))
1584 1.17 pooka
1585 1.17 pooka FDCALL(int, setsockopt, DUALCALL_SETSOCKOPT, \
1586 1.17 pooka (int fd, int level, int optn, \
1587 1.17 pooka const void *optval, socklen_t optlen), \
1588 1.17 pooka (int, int, int, const void *, socklen_t), \
1589 1.17 pooka (fd, level, optn, optval, optlen))
1590 1.17 pooka
1591 1.17 pooka FDCALL(int, shutdown, DUALCALL_SHUTDOWN, \
1592 1.17 pooka (int fd, int how), \
1593 1.17 pooka (int, int), \
1594 1.17 pooka (fd, how))
1595 1.17 pooka
1596 1.21 christos #if _FORTIFY_SOURCE > 0
1597 1.21 christos #define STUB(fun) __ssp_weak_name(fun)
1598 1.21 christos ssize_t _sys_readlink(const char * __restrict, char * __restrict, size_t);
1599 1.21 christos ssize_t
1600 1.21 christos STUB(readlink)(const char * __restrict path, char * __restrict buf,
1601 1.21 christos size_t bufsiz)
1602 1.21 christos {
1603 1.21 christos return _sys_readlink(path, buf, bufsiz);
1604 1.21 christos }
1605 1.21 christos
1606 1.21 christos char *_sys_getcwd(char *, size_t);
1607 1.21 christos char *
1608 1.21 christos STUB(getcwd)(char *buf, size_t size)
1609 1.21 christos {
1610 1.21 christos return _sys_getcwd(buf, size);
1611 1.21 christos }
1612 1.21 christos #else
1613 1.21 christos #define STUB(fun) fun
1614 1.21 christos #endif
1615 1.21 christos
1616 1.31 pooka FDCALL(ssize_t, REALREAD, DUALCALL_READ, \
1617 1.17 pooka (int fd, void *buf, size_t buflen), \
1618 1.17 pooka (int, void *, size_t), \
1619 1.17 pooka (fd, buf, buflen))
1620 1.17 pooka
1621 1.18 pooka FDCALL(ssize_t, readv, DUALCALL_READV, \
1622 1.17 pooka (int fd, const struct iovec *iov, int iovcnt), \
1623 1.17 pooka (int, const struct iovec *, int), \
1624 1.17 pooka (fd, iov, iovcnt))
1625 1.17 pooka
1626 1.60 pooka FDCALL(ssize_t, REALPREAD, DUALCALL_PREAD, \
1627 1.60 pooka (int fd, void *buf, size_t nbytes, off_t offset), \
1628 1.60 pooka (int, void *, size_t, off_t), \
1629 1.60 pooka (fd, buf, nbytes, offset))
1630 1.60 pooka
1631 1.60 pooka FDCALL(ssize_t, preadv, DUALCALL_PREADV, \
1632 1.60 pooka (int fd, const struct iovec *iov, int iovcnt, off_t offset), \
1633 1.60 pooka (int, const struct iovec *, int, off_t), \
1634 1.60 pooka (fd, iov, iovcnt, offset))
1635 1.60 pooka
1636 1.17 pooka FDCALL(ssize_t, writev, DUALCALL_WRITEV, \
1637 1.17 pooka (int fd, const struct iovec *iov, int iovcnt), \
1638 1.17 pooka (int, const struct iovec *, int), \
1639 1.17 pooka (fd, iov, iovcnt))
1640 1.45 pooka
1641 1.60 pooka FDCALL(ssize_t, REALPWRITE, DUALCALL_PWRITE, \
1642 1.60 pooka (int fd, const void *buf, size_t nbytes, off_t offset), \
1643 1.60 pooka (int, const void *, size_t, off_t), \
1644 1.60 pooka (fd, buf, nbytes, offset))
1645 1.60 pooka
1646 1.60 pooka FDCALL(ssize_t, pwritev, DUALCALL_PWRITEV, \
1647 1.60 pooka (int fd, const struct iovec *iov, int iovcnt, off_t offset), \
1648 1.60 pooka (int, const struct iovec *, int, off_t), \
1649 1.60 pooka (fd, iov, iovcnt, offset))
1650 1.60 pooka
1651 1.45 pooka FDCALL(int, REALFSTAT, DUALCALL_FSTAT, \
1652 1.45 pooka (int fd, struct stat *sb), \
1653 1.45 pooka (int, struct stat *), \
1654 1.45 pooka (fd, sb))
1655 1.45 pooka
1656 1.45 pooka FDCALL(int, fstatvfs1, DUALCALL_FSTATVFS1, \
1657 1.45 pooka (int fd, struct statvfs *buf, int flags), \
1658 1.45 pooka (int, struct statvfs *, int), \
1659 1.45 pooka (fd, buf, flags))
1660 1.45 pooka
1661 1.61 pooka FDCALL(off_t, lseek, DUALCALL_LSEEK, \
1662 1.45 pooka (int fd, off_t offset, int whence), \
1663 1.45 pooka (int, off_t, int), \
1664 1.45 pooka (fd, offset, whence))
1665 1.61 pooka __strong_alias(_lseek,lseek);
1666 1.45 pooka
1667 1.45 pooka FDCALL(int, REALGETDENTS, DUALCALL_GETDENTS, \
1668 1.45 pooka (int fd, char *buf, size_t nbytes), \
1669 1.45 pooka (int, char *, size_t), \
1670 1.45 pooka (fd, buf, nbytes))
1671 1.45 pooka
1672 1.45 pooka FDCALL(int, fchown, DUALCALL_FCHOWN, \
1673 1.45 pooka (int fd, uid_t owner, gid_t group), \
1674 1.45 pooka (int, uid_t, gid_t), \
1675 1.45 pooka (fd, owner, group))
1676 1.45 pooka
1677 1.45 pooka FDCALL(int, fchmod, DUALCALL_FCHMOD, \
1678 1.45 pooka (int fd, mode_t mode), \
1679 1.45 pooka (int, mode_t), \
1680 1.45 pooka (fd, mode))
1681 1.45 pooka
1682 1.45 pooka FDCALL(int, ftruncate, DUALCALL_FTRUNCATE, \
1683 1.45 pooka (int fd, off_t length), \
1684 1.45 pooka (int, off_t), \
1685 1.45 pooka (fd, length))
1686 1.45 pooka
1687 1.45 pooka FDCALL(int, fsync, DUALCALL_FSYNC, \
1688 1.45 pooka (int fd), \
1689 1.45 pooka (int), \
1690 1.45 pooka (fd))
1691 1.45 pooka
1692 1.45 pooka FDCALL(int, fsync_range, DUALCALL_FSYNC_RANGE, \
1693 1.45 pooka (int fd, int how, off_t start, off_t length), \
1694 1.45 pooka (int, int, off_t, off_t), \
1695 1.45 pooka (fd, how, start, length))
1696 1.45 pooka
1697 1.45 pooka FDCALL(int, futimes, DUALCALL_FUTIMES, \
1698 1.45 pooka (int fd, const struct timeval *tv), \
1699 1.45 pooka (int, const struct timeval *), \
1700 1.45 pooka (fd, tv))
1701 1.45 pooka
1702 1.60 pooka FDCALL(int, fchflags, DUALCALL_FCHFLAGS, \
1703 1.60 pooka (int fd, u_long flags), \
1704 1.60 pooka (int, u_long), \
1705 1.60 pooka (fd, flags))
1706 1.60 pooka
1707 1.45 pooka /*
1708 1.45 pooka * path-based selectors
1709 1.45 pooka */
1710 1.45 pooka
1711 1.45 pooka PATHCALL(int, REALSTAT, DUALCALL_STAT, \
1712 1.45 pooka (const char *path, struct stat *sb), \
1713 1.45 pooka (const char *, struct stat *), \
1714 1.45 pooka (path, sb))
1715 1.45 pooka
1716 1.45 pooka PATHCALL(int, REALLSTAT, DUALCALL_LSTAT, \
1717 1.45 pooka (const char *path, struct stat *sb), \
1718 1.45 pooka (const char *, struct stat *), \
1719 1.45 pooka (path, sb))
1720 1.45 pooka
1721 1.45 pooka PATHCALL(int, chown, DUALCALL_CHOWN, \
1722 1.45 pooka (const char *path, uid_t owner, gid_t group), \
1723 1.45 pooka (const char *, uid_t, gid_t), \
1724 1.45 pooka (path, owner, group))
1725 1.45 pooka
1726 1.45 pooka PATHCALL(int, lchown, DUALCALL_LCHOWN, \
1727 1.45 pooka (const char *path, uid_t owner, gid_t group), \
1728 1.45 pooka (const char *, uid_t, gid_t), \
1729 1.45 pooka (path, owner, group))
1730 1.45 pooka
1731 1.45 pooka PATHCALL(int, chmod, DUALCALL_CHMOD, \
1732 1.45 pooka (const char *path, mode_t mode), \
1733 1.45 pooka (const char *, mode_t), \
1734 1.45 pooka (path, mode))
1735 1.45 pooka
1736 1.45 pooka PATHCALL(int, lchmod, DUALCALL_LCHMOD, \
1737 1.45 pooka (const char *path, mode_t mode), \
1738 1.45 pooka (const char *, mode_t), \
1739 1.45 pooka (path, mode))
1740 1.45 pooka
1741 1.45 pooka PATHCALL(int, statvfs1, DUALCALL_STATVFS1, \
1742 1.45 pooka (const char *path, struct statvfs *buf, int flags), \
1743 1.45 pooka (const char *, struct statvfs *, int), \
1744 1.45 pooka (path, buf, flags))
1745 1.45 pooka
1746 1.45 pooka PATHCALL(int, unlink, DUALCALL_UNLINK, \
1747 1.45 pooka (const char *path), \
1748 1.45 pooka (const char *), \
1749 1.45 pooka (path))
1750 1.45 pooka
1751 1.45 pooka PATHCALL(int, symlink, DUALCALL_SYMLINK, \
1752 1.58 pooka (const char *target, const char *path), \
1753 1.45 pooka (const char *, const char *), \
1754 1.58 pooka (target, path))
1755 1.45 pooka
1756 1.46 pooka PATHCALL(ssize_t, readlink, DUALCALL_READLINK, \
1757 1.45 pooka (const char *path, char *buf, size_t bufsiz), \
1758 1.45 pooka (const char *, char *, size_t), \
1759 1.45 pooka (path, buf, bufsiz))
1760 1.45 pooka
1761 1.45 pooka PATHCALL(int, mkdir, DUALCALL_MKDIR, \
1762 1.45 pooka (const char *path, mode_t mode), \
1763 1.45 pooka (const char *, mode_t), \
1764 1.45 pooka (path, mode))
1765 1.45 pooka
1766 1.45 pooka PATHCALL(int, rmdir, DUALCALL_RMDIR, \
1767 1.45 pooka (const char *path), \
1768 1.45 pooka (const char *), \
1769 1.45 pooka (path))
1770 1.45 pooka
1771 1.45 pooka PATHCALL(int, utimes, DUALCALL_UTIMES, \
1772 1.45 pooka (const char *path, const struct timeval *tv), \
1773 1.45 pooka (const char *, const struct timeval *), \
1774 1.45 pooka (path, tv))
1775 1.45 pooka
1776 1.45 pooka PATHCALL(int, lutimes, DUALCALL_LUTIMES, \
1777 1.45 pooka (const char *path, const struct timeval *tv), \
1778 1.45 pooka (const char *, const struct timeval *), \
1779 1.45 pooka (path, tv))
1780 1.45 pooka
1781 1.60 pooka PATHCALL(int, chflags, DUALCALL_CHFLAGS, \
1782 1.60 pooka (const char *path, u_long flags), \
1783 1.60 pooka (const char *, u_long), \
1784 1.60 pooka (path, flags))
1785 1.60 pooka
1786 1.60 pooka PATHCALL(int, lchflags, DUALCALL_LCHFLAGS, \
1787 1.60 pooka (const char *path, u_long flags), \
1788 1.60 pooka (const char *, u_long), \
1789 1.60 pooka (path, flags))
1790 1.60 pooka
1791 1.45 pooka PATHCALL(int, truncate, DUALCALL_TRUNCATE, \
1792 1.45 pooka (const char *path, off_t length), \
1793 1.45 pooka (const char *, off_t), \
1794 1.45 pooka (path, length))
1795 1.48 pooka
1796 1.65 pooka PATHCALL(int, access, DUALCALL_ACCESS, \
1797 1.65 pooka (const char *path, int mode), \
1798 1.65 pooka (const char *, int), \
1799 1.65 pooka (path, mode))
1800 1.65 pooka
1801 1.48 pooka /*
1802 1.48 pooka * Note: with mount the decisive parameter is the mount
1803 1.48 pooka * destination directory. This is because we don't really know
1804 1.48 pooka * about the "source" directory in a generic call (and besides,
1805 1.48 pooka * it might not even exist, cf. nfs).
1806 1.48 pooka */
1807 1.48 pooka PATHCALL(int, REALMOUNT, DUALCALL_MOUNT, \
1808 1.48 pooka (const char *type, const char *path, int flags, \
1809 1.48 pooka void *data, size_t dlen), \
1810 1.48 pooka (const char *, const char *, int, void *, size_t), \
1811 1.48 pooka (type, path, flags, data, dlen))
1812 1.48 pooka
1813 1.48 pooka PATHCALL(int, unmount, DUALCALL_UNMOUNT, \
1814 1.48 pooka (const char *path, int flags), \
1815 1.48 pooka (const char *, int), \
1816 1.48 pooka (path, flags))
1817