1 /* $NetBSD: h_execthr.c,v 1.3 2014/08/13 00:03:00 pooka Exp $ */ 2 3 /* 4 * Copyright (c) 2011 The NetBSD Foundation, Inc. 5 * All rights reserved. 6 * 7 * Redistribution and use in source and binary forms, with or without 8 * modification, are permitted provided that the following conditions 9 * are met: 10 * 1. Redistributions of source code must retain the above copyright 11 * notice, this list of conditions and the following disclaimer. 12 * 2. Redistributions in binary form must reproduce the above copyright 13 * notice, this list of conditions and the following disclaimer in the 14 * documentation and/or other materials provided with the distribution. 15 * 16 * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND 17 * CONTRIBUTORS ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, 18 * INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF 19 * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. 20 * IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS BE LIABLE FOR ANY 21 * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 22 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE 23 * GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS 24 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER 25 * IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR 26 * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN 27 * IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 28 */ 29 30 #include <sys/types.h> 31 #include <sys/sysctl.h> 32 33 #include <err.h> 34 #include <errno.h> 35 #include <fcntl.h> 36 #include <pthread.h> 37 #include <stdio.h> 38 #include <stdlib.h> 39 #include <string.h> 40 #include <unistd.h> 41 42 #include <rump/rumpclient.h> 43 #include <rump/rump_syscalls.h> 44 45 static int canreturn = 0; 46 47 /* 48 * Use a fairly large number of threads so that we have 49 * a better chance catching races. XXX: this is rumpuser's 50 * MAXWORKER-1. 51 */ 52 #define NTHR 63 53 54 #define P1_0 3 55 #define P1_1 4 56 #define P2_0 5 57 #define P2_1 6 58 59 static void * 60 wrk(void *arg) 61 { 62 int fd = (uintptr_t)arg; 63 64 rump_sys_read(fd, &fd, sizeof(fd)); 65 if (!canreturn) 66 errx(1, "should not have returned"); 67 if (fd != 37) 68 errx(1, "got invalid magic"); 69 70 return NULL; 71 } 72 73 static int 74 getproc(pid_t mypid, struct kinfo_proc2 *p) 75 { 76 int name[6]; 77 size_t len = sizeof(*p); 78 79 name[0] = CTL_KERN; 80 name[1] = KERN_PROC2; 81 name[2] = KERN_PROC_PID; 82 name[3] = mypid; 83 name[4] = len; 84 name[5] = 1; 85 86 return rump_sys___sysctl(name, __arraycount(name), p, &len, NULL, 0); 87 } 88 89 int 90 main(int argc, char *argv[], char *envp[]) 91 { 92 struct kinfo_proc2 p; 93 char *execarg[3]; 94 int p1[2], p2[2]; 95 pid_t mypid; 96 pthread_t pt; 97 ssize_t n; 98 int i, execd; 99 char nexec[16]; 100 101 if (argc > 1) 102 execd = atoi(argv[1]); 103 else 104 execd = 0; 105 sprintf(nexec, "%d", execd+1); 106 107 if (rumpclient_init() == -1) { 108 if (execd) 109 err(1, "init execd"); 110 else 111 err(1, "init"); 112 } 113 mypid = rump_sys_getpid(); 114 115 if (execd) { 116 canreturn = 1; 117 if (pthread_create(&pt, NULL, 118 wrk, (void *)(uintptr_t)P2_0) != 0) 119 errx(1, "exec pthread_create"); 120 121 i = 37; 122 rump_sys_write(P2_1, &i, sizeof(i)); 123 pthread_join(pt, NULL); 124 125 n = rump_sys_read(P1_0, &i, sizeof(i)); 126 if (n != -1 || errno != EBADF) 127 errx(1, "post-exec cloexec works"); 128 129 getproc(mypid, &p); 130 if (p.p_nlwps != 2) 131 errx(1, "invalid nlwps: %lld", (long long)p.p_nlwps); 132 133 /* we passed? */ 134 if (execd > 10) 135 exit(0); 136 137 rump_sys_close(P2_0); 138 rump_sys_close(P2_1); 139 } 140 141 if (rump_sys_pipe(p1) == -1) 142 err(1, "pipe1"); 143 if (p1[0] != P1_0 || p1[1] != P1_1) 144 errx(1, "p1 assumptions failed %d %d", p1[0], p1[1]); 145 if (rump_sys_pipe(p2) == -1) 146 err(1, "pipe1"); 147 if (p2[0] != P2_0 || p2[1] != P2_1) 148 errx(1, "p2 assumptions failed"); 149 if (rump_sys_fcntl(p1[0], F_SETFD, FD_CLOEXEC) == -1) 150 err(1, "cloexec"); 151 if (rump_sys_fcntl(p1[1], F_SETFD, FD_CLOEXEC) == -1) 152 err(1, "cloexec"); 153 154 for (i = 0; i < NTHR; i++) 155 if (pthread_create(&pt, NULL, 156 wrk, (void *)(uintptr_t)p1[0]) != 0) 157 errx(1, "pthread_create 1 %d", i); 158 159 for (i = 0; i < NTHR; i++) 160 if (pthread_create(&pt, NULL, 161 wrk, (void *)(uintptr_t)p2[0]) != 0) 162 errx(1, "pthread_create 2 %d", i); 163 164 /* wait for all the threads to be enjoying themselves */ 165 for (;;) { 166 getproc(mypid, &p); 167 if (p.p_nlwps == 2*NTHR + 2) 168 break; 169 usleep(10000); 170 } 171 172 /* 173 * load up one more (big) set. these won't start executing, though, 174 * but we're interested in if they create blockage 175 */ 176 for (i = 0; i < 3*NTHR; i++) 177 if (pthread_create(&pt, NULL, 178 wrk, (void *)(uintptr_t)p1[0]) != 0) 179 errx(1, "pthread_create 1 %d", i); 180 181 /* then, we exec! */ 182 execarg[0] = argv[0]; 183 execarg[1] = nexec; 184 execarg[2] = NULL; 185 if (rumpclient_exec(argv[0], execarg, envp) == -1) 186 err(1, "exec"); 187 } 188