1 /* $NetBSD: h_execthr.c,v 1.7 2016/11/24 00:37:29 dholland 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 //#define VERBOSE 46 47 #ifdef VERBOSE 48 #define SAY(...) printf(__VA_ARGS__) 49 #else 50 #define SAY(...) 51 #endif 52 53 static int canreturn = 0; 54 55 /* 56 * Use a fairly large number of threads so that we have 57 * a better chance catching races. XXX: this is rumpuser's 58 * MAXWORKER-1. 59 */ 60 #define NTHR 63 61 62 #define P1_0 3 63 #define P1_1 4 64 #define P2_0 5 65 #define P2_1 6 66 67 static void * 68 wrk(void *arg) 69 { 70 int fd = (uintptr_t)arg; 71 72 rump_sys_read(fd, &fd, sizeof(fd)); 73 if (!canreturn) 74 errx(1, "should not have returned"); 75 if (fd != 37) 76 errx(1, "got invalid magic"); 77 78 return NULL; 79 } 80 81 static int 82 getproc(pid_t mypid, struct kinfo_proc2 *p) 83 { 84 int name[6]; 85 size_t len = sizeof(*p); 86 87 name[0] = CTL_KERN; 88 name[1] = KERN_PROC2; 89 name[2] = KERN_PROC_PID; 90 name[3] = mypid; 91 name[4] = len; 92 name[5] = 1; 93 94 return rump_sys___sysctl(name, __arraycount(name), p, &len, NULL, 0); 95 } 96 97 int 98 main(int argc, char *argv[], char *envp[]) 99 { 100 struct kinfo_proc2 p; 101 char *execarg[3]; 102 int p1[2], p2[2]; 103 pid_t mypid; 104 pthread_t pt; 105 ssize_t n; 106 int i, execd; 107 char nexec[16]; 108 109 if (argc > 1) 110 execd = atoi(argv[1]); 111 else 112 execd = 0; 113 sprintf(nexec, "%d", execd+1); 114 SAY("execd: %d\n", execd); 115 116 if (rumpclient_init() == -1) { 117 if (execd) 118 err(1, "init execd"); 119 else 120 err(1, "init"); 121 } 122 mypid = rump_sys_getpid(); 123 SAY("rumpclient_init finished.\n"); 124 125 if (execd) { 126 canreturn = 1; 127 errno = pthread_create(&pt, NULL, 128 wrk, (void *)(uintptr_t)P2_0); 129 if (errno != 0) 130 err(1, "exec pthread_create"); 131 SAY("startup pthread_create finished.\n"); 132 133 i = 37; 134 rump_sys_write(P2_1, &i, sizeof(i)); 135 pthread_join(pt, NULL); 136 SAY("startup pthread_join finished.\n"); 137 138 n = rump_sys_read(P1_0, &i, sizeof(i)); 139 if (n != -1 || errno != EBADF) 140 errx(1, "post-exec cloexec works"); 141 SAY("startup rump_sys_read finished.\n"); 142 143 getproc(mypid, &p); 144 SAY("startup getproc finished.\n"); 145 if (p.p_nlwps != 2) 146 errx(1, "invalid nlwps: %lld", (long long)p.p_nlwps); 147 148 /* we passed? */ 149 if (execd > 10) { 150 SAY("done.\n"); 151 exit(0); 152 } 153 154 rump_sys_close(P2_0); 155 rump_sys_close(P2_1); 156 } 157 158 SAY("making pipes...\n"); 159 160 if (rump_sys_pipe(p1) == -1) 161 err(1, "pipe1"); 162 if (p1[0] != P1_0 || p1[1] != P1_1) 163 errx(1, "p1 assumptions failed %d %d", p1[0], p1[1]); 164 if (rump_sys_pipe(p2) == -1) 165 err(1, "pipe1"); 166 if (p2[0] != P2_0 || p2[1] != P2_1) 167 errx(1, "p2 assumptions failed"); 168 if (rump_sys_fcntl(p1[0], F_SETFD, FD_CLOEXEC) == -1) 169 err(1, "cloexec"); 170 if (rump_sys_fcntl(p1[1], F_SETFD, FD_CLOEXEC) == -1) 171 err(1, "cloexec"); 172 173 SAY("making threads...\n"); 174 175 for (i = 0; i < NTHR; i++) { 176 errno = pthread_create(&pt, NULL, 177 wrk, (void *)(uintptr_t)p1[0]); 178 if (errno != 0) 179 err(1, "pthread_create 1 %d", i); 180 } 181 182 for (i = 0; i < NTHR; i++) { 183 errno = pthread_create(&pt, NULL, 184 wrk, (void *)(uintptr_t)p2[0]); 185 if (errno != 0) 186 err(1, "pthread_create 2 %d", i); 187 } 188 189 SAY("waiting for threads to start...\n"); 190 191 /* wait for all the threads to be enjoying themselves */ 192 for (;;) { 193 getproc(mypid, &p); 194 SAY("getproc finished.\n"); 195 if (p.p_nlwps == 2*NTHR + 2) 196 break; 197 usleep(10000); 198 } 199 200 SAY("making some more threads start...\n"); 201 202 /* 203 * load up one more (big) set. these won't start executing, though, 204 * but we're interested in if they create blockage 205 */ 206 for (i = 0; i < 3*NTHR; i++) { 207 errno = pthread_create(&pt, NULL, 208 wrk, (void *)(uintptr_t)p1[0]); 209 if (errno != 0) 210 err(1, "pthread_create 3 %d", i); 211 } 212 213 SAY("calling exec...\n"); 214 215 /* then, we exec! */ 216 execarg[0] = argv[0]; 217 execarg[1] = nexec; 218 execarg[2] = NULL; 219 if (rumpclient_exec(argv[0], execarg, envp) == -1) 220 err(1, "exec"); 221 } 222