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
45static 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
59static void *
60wrk(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
73static int
74getproc(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
89int
90main(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