rumpfiber.c revision 1.7 1 1.7 justin /* $NetBSD: rumpfiber.c,v 1.7 2014/11/08 23:20:23 justin Exp $ */
2 1.4 pooka
3 1.1 justin /*
4 1.1 justin * Copyright (c) 2007-2013 Antti Kantee. All Rights Reserved.
5 1.1 justin * Copyright (c) 2014 Justin Cormack. All Rights Reserved.
6 1.1 justin *
7 1.1 justin * Redistribution and use in source and binary forms, with or without
8 1.1 justin * modification, are permitted provided that the following conditions
9 1.1 justin * are met:
10 1.1 justin * 1. Redistributions of source code must retain the above copyright
11 1.1 justin * notice, this list of conditions and the following disclaimer.
12 1.1 justin * 2. Redistributions in binary form must reproduce the above copyright
13 1.1 justin * notice, this list of conditions and the following disclaimer in the
14 1.1 justin * documentation and/or other materials provided with the distribution.
15 1.1 justin *
16 1.1 justin * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS
17 1.1 justin * OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
18 1.1 justin * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
19 1.1 justin * DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
20 1.1 justin * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
21 1.1 justin * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
22 1.1 justin * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
23 1.1 justin * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
24 1.1 justin * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
25 1.1 justin * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
26 1.1 justin * SUCH DAMAGE.
27 1.1 justin */
28 1.1 justin
29 1.1 justin /* Based partly on code from Xen Minios with the following license */
30 1.1 justin
31 1.1 justin /*
32 1.1 justin ****************************************************************************
33 1.1 justin * (C) 2005 - Grzegorz Milos - Intel Research Cambridge
34 1.1 justin ****************************************************************************
35 1.1 justin *
36 1.1 justin * File: sched.c
37 1.1 justin * Author: Grzegorz Milos
38 1.1 justin * Changes: Robert Kaiser
39 1.1 justin *
40 1.1 justin * Date: Aug 2005
41 1.1 justin *
42 1.1 justin * Environment: Xen Minimal OS
43 1.1 justin * Description: simple scheduler for Mini-Os
44 1.1 justin *
45 1.1 justin * The scheduler is non-preemptive (cooperative), and schedules according
46 1.1 justin * to Round Robin algorithm.
47 1.1 justin *
48 1.1 justin ****************************************************************************
49 1.1 justin * Permission is hereby granted, free of charge, to any person obtaining a copy
50 1.1 justin * of this software and associated documentation files (the "Software"), to
51 1.1 justin * deal in the Software without restriction, including without limitation the
52 1.1 justin * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
53 1.1 justin * sell copies of the Software, and to permit persons to whom the Software is
54 1.1 justin * furnished to do so, subject to the following conditions:
55 1.1 justin *
56 1.1 justin * The above copyright notice and this permission notice shall be included in
57 1.1 justin * all copies or substantial portions of the Software.
58 1.1 justin *
59 1.1 justin * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
60 1.1 justin * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
61 1.1 justin * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
62 1.1 justin * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
63 1.1 justin * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
64 1.1 justin * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
65 1.1 justin * DEALINGS IN THE SOFTWARE.
66 1.1 justin */
67 1.1 justin
68 1.1 justin #include "rumpuser_port.h"
69 1.1 justin
70 1.1 justin #if !defined(lint)
71 1.7 justin __RCSID("$NetBSD: rumpfiber.c,v 1.7 2014/11/08 23:20:23 justin Exp $");
72 1.1 justin #endif /* !lint */
73 1.1 justin
74 1.1 justin #include <sys/ioctl.h>
75 1.1 justin #include <sys/mman.h>
76 1.1 justin #include <sys/time.h>
77 1.1 justin
78 1.1 justin #include <assert.h>
79 1.1 justin #include <errno.h>
80 1.1 justin #include <fcntl.h>
81 1.1 justin #include <signal.h>
82 1.1 justin #include <stdarg.h>
83 1.1 justin #include <stdint.h>
84 1.1 justin #include <stdio.h>
85 1.1 justin #include <stdlib.h>
86 1.1 justin #include <string.h>
87 1.1 justin #include <time.h>
88 1.1 justin #include <unistd.h>
89 1.1 justin
90 1.1 justin #include <rump/rumpuser.h>
91 1.1 justin
92 1.1 justin #include "rumpuser_int.h"
93 1.1 justin #include "rumpfiber.h"
94 1.1 justin
95 1.1 justin static void init_sched(void);
96 1.1 justin static void join_thread(struct thread *);
97 1.1 justin static void switch_threads(struct thread *prev, struct thread *next);
98 1.1 justin static struct thread *get_current(void);
99 1.1 justin static int64_t now(void);
100 1.1 justin static void msleep(uint64_t millisecs);
101 1.1 justin static void abssleep(uint64_t millisecs);
102 1.1 justin
103 1.1 justin TAILQ_HEAD(thread_list, thread);
104 1.1 justin
105 1.1 justin static struct thread_list exited_threads = TAILQ_HEAD_INITIALIZER(exited_threads);
106 1.1 justin static struct thread_list thread_list = TAILQ_HEAD_INITIALIZER(thread_list);
107 1.1 justin static struct thread *current_thread = NULL;
108 1.1 justin
109 1.1 justin static void (*scheduler_hook)(void *, void *);
110 1.1 justin
111 1.1 justin static struct thread *
112 1.1 justin get_current(void)
113 1.1 justin {
114 1.1 justin
115 1.1 justin return current_thread;
116 1.1 justin }
117 1.1 justin
118 1.1 justin static int64_t
119 1.1 justin now(void)
120 1.1 justin {
121 1.1 justin struct timespec ts;
122 1.5 pooka int rv;
123 1.1 justin
124 1.5 pooka rv = clock_gettime(CLOCK_MONOTONIC, &ts);
125 1.5 pooka assert(rv == 0);
126 1.1 justin return (ts.tv_sec * 1000LL) + (ts.tv_nsec / 1000000LL);
127 1.1 justin }
128 1.1 justin
129 1.1 justin void
130 1.1 justin schedule(void)
131 1.1 justin {
132 1.1 justin struct thread *prev, *next, *thread, *tmp;
133 1.1 justin int64_t tm, wakeup;
134 1.1 justin struct timespec sl;
135 1.1 justin
136 1.1 justin prev = get_current();
137 1.1 justin
138 1.1 justin do {
139 1.1 justin tm = now();
140 1.1 justin wakeup = tm + 1000; /* wake up in 1s max */
141 1.1 justin next = NULL;
142 1.1 justin TAILQ_FOREACH_SAFE(thread, &thread_list, thread_list, tmp) {
143 1.1 justin if (!is_runnable(thread) && thread->wakeup_time >= 0) {
144 1.1 justin if (thread->wakeup_time <= tm) {
145 1.1 justin thread->flags |= THREAD_TIMEDOUT;
146 1.1 justin wake(thread);
147 1.1 justin } else if (thread->wakeup_time < wakeup)
148 1.1 justin wakeup = thread->wakeup_time;
149 1.1 justin }
150 1.1 justin if (is_runnable(thread)) {
151 1.1 justin next = thread;
152 1.1 justin /* Put this thread on the end of the list */
153 1.1 justin TAILQ_REMOVE(&thread_list, thread, thread_list);
154 1.1 justin TAILQ_INSERT_TAIL(&thread_list, thread, thread_list);
155 1.1 justin break;
156 1.1 justin }
157 1.1 justin }
158 1.1 justin if (next)
159 1.1 justin break;
160 1.1 justin sl.tv_sec = (wakeup - tm) / 1000;
161 1.1 justin sl.tv_nsec = ((wakeup - tm) - 1000 * sl.tv_sec) * 1000000;
162 1.1 justin #ifdef HAVE_CLOCK_NANOSLEEP
163 1.1 justin clock_nanosleep(CLOCK_MONOTONIC, 0, &sl, NULL);
164 1.1 justin #else
165 1.1 justin nanosleep(&sl, NULL);
166 1.1 justin #endif
167 1.1 justin } while (1);
168 1.1 justin
169 1.1 justin if (prev != next)
170 1.1 justin switch_threads(prev, next);
171 1.1 justin
172 1.1 justin TAILQ_FOREACH_SAFE(thread, &exited_threads, thread_list, tmp) {
173 1.1 justin if (thread != prev) {
174 1.1 justin TAILQ_REMOVE(&exited_threads, thread, thread_list);
175 1.1 justin if ((thread->flags & THREAD_EXTSTACK) == 0)
176 1.1 justin munmap(thread->ctx.uc_stack.ss_sp, STACKSIZE);
177 1.1 justin free(thread->name);
178 1.1 justin free(thread);
179 1.1 justin }
180 1.1 justin }
181 1.1 justin }
182 1.1 justin
183 1.1 justin static void
184 1.7 justin create_ctx(ucontext_t *ctx, void *stack, size_t stack_size,
185 1.7 justin void (*f)(void *), void *data)
186 1.1 justin {
187 1.1 justin
188 1.1 justin getcontext(ctx);
189 1.1 justin ctx->uc_stack.ss_sp = stack;
190 1.1 justin ctx->uc_stack.ss_size = stack_size;
191 1.1 justin ctx->uc_stack.ss_flags = 0;
192 1.1 justin ctx->uc_link = NULL; /* TODO may link to main thread */
193 1.7 justin /* may have to do bounce function to call, if args to makecontext are ints */
194 1.7 justin makecontext(ctx, (void (*)(void))f, 1, data);
195 1.1 justin }
196 1.1 justin
197 1.1 justin /* TODO see notes in rumpuser_thread_create, have flags here */
198 1.1 justin struct thread *
199 1.1 justin create_thread(const char *name, void *cookie, void (*f)(void *), void *data,
200 1.1 justin void *stack, size_t stack_size)
201 1.1 justin {
202 1.1 justin struct thread *thread = calloc(1, sizeof(struct thread));
203 1.1 justin
204 1.6 justin if (!thread) {
205 1.6 justin return NULL;
206 1.6 justin }
207 1.6 justin
208 1.1 justin if (!stack) {
209 1.1 justin assert(stack_size == 0);
210 1.1 justin stack = mmap(NULL, STACKSIZE, PROT_READ | PROT_WRITE,
211 1.1 justin MAP_SHARED | MAP_ANON, -1, 0);
212 1.1 justin if (stack == MAP_FAILED) {
213 1.6 justin free(thread);
214 1.1 justin return NULL;
215 1.1 justin }
216 1.1 justin stack_size = STACKSIZE;
217 1.1 justin } else {
218 1.1 justin thread->flags = THREAD_EXTSTACK;
219 1.1 justin }
220 1.7 justin create_ctx(&thread->ctx, stack, stack_size, f, data);
221 1.1 justin
222 1.1 justin thread->name = strdup(name);
223 1.1 justin thread->cookie = cookie;
224 1.1 justin
225 1.1 justin /* Not runnable, not exited, not sleeping */
226 1.1 justin thread->wakeup_time = -1;
227 1.1 justin thread->lwp = NULL;
228 1.1 justin set_runnable(thread);
229 1.1 justin TAILQ_INSERT_TAIL(&thread_list, thread, thread_list);
230 1.1 justin
231 1.1 justin return thread;
232 1.1 justin }
233 1.1 justin
234 1.1 justin static void
235 1.1 justin switch_threads(struct thread *prev, struct thread *next)
236 1.1 justin {
237 1.1 justin int ret;
238 1.1 justin
239 1.1 justin current_thread = next;
240 1.1 justin if (scheduler_hook)
241 1.1 justin scheduler_hook(prev->cookie, next->cookie);
242 1.1 justin ret = swapcontext(&prev->ctx, &next->ctx);
243 1.1 justin if (ret < 0) {
244 1.1 justin printk("swapcontext failed\n");
245 1.1 justin abort();
246 1.1 justin }
247 1.1 justin }
248 1.1 justin
249 1.1 justin struct join_waiter {
250 1.1 justin struct thread *jw_thread;
251 1.1 justin struct thread *jw_wanted;
252 1.1 justin TAILQ_ENTRY(join_waiter) jw_entries;
253 1.1 justin };
254 1.1 justin static TAILQ_HEAD(, join_waiter) joinwq = TAILQ_HEAD_INITIALIZER(joinwq);
255 1.1 justin
256 1.1 justin void
257 1.1 justin exit_thread(void)
258 1.1 justin {
259 1.1 justin struct thread *thread = get_current();
260 1.1 justin struct join_waiter *jw_iter;
261 1.1 justin
262 1.1 justin /* if joinable, gate until we are allowed to exit */
263 1.1 justin while (thread->flags & THREAD_MUSTJOIN) {
264 1.1 justin thread->flags |= THREAD_JOINED;
265 1.1 justin
266 1.1 justin /* see if the joiner is already there */
267 1.1 justin TAILQ_FOREACH(jw_iter, &joinwq, jw_entries) {
268 1.1 justin if (jw_iter->jw_wanted == thread) {
269 1.1 justin wake(jw_iter->jw_thread);
270 1.1 justin break;
271 1.1 justin }
272 1.1 justin }
273 1.1 justin block(thread);
274 1.1 justin schedule();
275 1.1 justin }
276 1.1 justin
277 1.1 justin /* Remove from the thread list */
278 1.1 justin TAILQ_REMOVE(&thread_list, thread, thread_list);
279 1.1 justin clear_runnable(thread);
280 1.1 justin /* Put onto exited list */
281 1.1 justin TAILQ_INSERT_HEAD(&exited_threads, thread, thread_list);
282 1.1 justin
283 1.1 justin /* Schedule will free the resources */
284 1.1 justin while (1) {
285 1.1 justin schedule();
286 1.1 justin printk("schedule() returned! Trying again\n");
287 1.1 justin }
288 1.1 justin }
289 1.1 justin
290 1.1 justin static void
291 1.1 justin join_thread(struct thread *joinable)
292 1.1 justin {
293 1.1 justin struct join_waiter jw;
294 1.1 justin struct thread *thread = get_current();
295 1.1 justin
296 1.1 justin assert(joinable->flags & THREAD_MUSTJOIN);
297 1.1 justin
298 1.1 justin /* wait for exiting thread to hit thread_exit() */
299 1.1 justin while (! (joinable->flags & THREAD_JOINED)) {
300 1.1 justin
301 1.1 justin jw.jw_thread = thread;
302 1.1 justin jw.jw_wanted = joinable;
303 1.1 justin TAILQ_INSERT_TAIL(&joinwq, &jw, jw_entries);
304 1.1 justin block(thread);
305 1.1 justin schedule();
306 1.1 justin TAILQ_REMOVE(&joinwq, &jw, jw_entries);
307 1.1 justin }
308 1.1 justin
309 1.1 justin /* signal exiting thread that we have seen it and it may now exit */
310 1.1 justin assert(joinable->flags & THREAD_JOINED);
311 1.1 justin joinable->flags &= ~THREAD_MUSTJOIN;
312 1.1 justin
313 1.1 justin wake(joinable);
314 1.1 justin }
315 1.1 justin
316 1.1 justin static void msleep(uint64_t millisecs)
317 1.1 justin {
318 1.1 justin struct thread *thread = get_current();
319 1.1 justin
320 1.1 justin thread->wakeup_time = now() + millisecs;
321 1.1 justin clear_runnable(thread);
322 1.1 justin schedule();
323 1.1 justin }
324 1.1 justin
325 1.1 justin static void abssleep(uint64_t millisecs)
326 1.1 justin {
327 1.1 justin struct thread *thread = get_current();
328 1.1 justin
329 1.1 justin thread->wakeup_time = millisecs;
330 1.1 justin clear_runnable(thread);
331 1.1 justin schedule();
332 1.1 justin }
333 1.1 justin
334 1.1 justin /* like abssleep, except against realtime clock instead of monotonic clock */
335 1.1 justin int abssleep_real(uint64_t millisecs)
336 1.1 justin {
337 1.1 justin struct thread *thread = get_current();
338 1.1 justin struct timespec ts;
339 1.1 justin uint64_t real_now;
340 1.1 justin int rv;
341 1.1 justin
342 1.1 justin clock_gettime(CLOCK_REALTIME, &ts);
343 1.1 justin real_now = 1000*ts.tv_sec + ts.tv_nsec/(1000*1000);
344 1.1 justin thread->wakeup_time = now() + (millisecs - real_now);
345 1.1 justin
346 1.1 justin clear_runnable(thread);
347 1.1 justin schedule();
348 1.1 justin
349 1.1 justin rv = !!(thread->flags & THREAD_TIMEDOUT);
350 1.1 justin thread->flags &= ~THREAD_TIMEDOUT;
351 1.1 justin return rv;
352 1.1 justin }
353 1.1 justin
354 1.1 justin void wake(struct thread *thread)
355 1.1 justin {
356 1.1 justin
357 1.1 justin thread->wakeup_time = -1;
358 1.1 justin set_runnable(thread);
359 1.1 justin }
360 1.1 justin
361 1.1 justin void block(struct thread *thread)
362 1.1 justin {
363 1.1 justin
364 1.1 justin thread->wakeup_time = -1;
365 1.1 justin clear_runnable(thread);
366 1.1 justin }
367 1.1 justin
368 1.1 justin int is_runnable(struct thread *thread)
369 1.1 justin {
370 1.1 justin
371 1.1 justin return thread->flags & RUNNABLE_FLAG;
372 1.1 justin }
373 1.1 justin
374 1.1 justin void set_runnable(struct thread *thread)
375 1.1 justin {
376 1.1 justin
377 1.1 justin thread->flags |= RUNNABLE_FLAG;
378 1.1 justin }
379 1.1 justin
380 1.1 justin void clear_runnable(struct thread *thread)
381 1.1 justin {
382 1.1 justin
383 1.1 justin thread->flags &= ~RUNNABLE_FLAG;
384 1.1 justin }
385 1.1 justin
386 1.1 justin static void
387 1.1 justin init_sched(void)
388 1.1 justin {
389 1.1 justin struct thread *thread = calloc(1, sizeof(struct thread));
390 1.1 justin
391 1.1 justin getcontext(&thread->ctx);
392 1.1 justin thread->name = strdup("init");
393 1.1 justin thread->flags = 0;
394 1.1 justin thread->wakeup_time = -1;
395 1.1 justin thread->lwp = NULL;
396 1.1 justin set_runnable(thread);
397 1.1 justin TAILQ_INSERT_TAIL(&thread_list, thread, thread_list);
398 1.1 justin current_thread = thread;
399 1.1 justin }
400 1.1 justin
401 1.1 justin void
402 1.1 justin set_sched_hook(void (*f)(void *, void *))
403 1.1 justin {
404 1.1 justin
405 1.1 justin scheduler_hook = f;
406 1.1 justin }
407 1.1 justin
408 1.1 justin struct thread *
409 1.1 justin init_mainthread(void *cookie)
410 1.1 justin {
411 1.1 justin
412 1.1 justin current_thread->cookie = cookie;
413 1.1 justin return current_thread;
414 1.1 justin }
415 1.1 justin
416 1.1 justin /* rump functions below */
417 1.1 justin
418 1.1 justin struct rumpuser_hyperup rumpuser__hyp;
419 1.1 justin
420 1.1 justin int
421 1.1 justin rumpuser_init(int version, const struct rumpuser_hyperup *hyp)
422 1.1 justin {
423 1.3 justin int rv;
424 1.1 justin
425 1.1 justin if (version != RUMPUSER_VERSION) {
426 1.1 justin printk("rumpuser version mismatch\n");
427 1.3 justin abort();
428 1.1 justin }
429 1.1 justin
430 1.3 justin rv = rumpuser__random_init();
431 1.3 justin if (rv != 0) {
432 1.3 justin ET(rv);
433 1.1 justin }
434 1.1 justin
435 1.1 justin rumpuser__hyp = *hyp;
436 1.1 justin
437 1.1 justin init_sched();
438 1.1 justin
439 1.1 justin return 0;
440 1.1 justin }
441 1.1 justin
442 1.1 justin int
443 1.1 justin rumpuser_clock_gettime(int enum_rumpclock, int64_t *sec, long *nsec)
444 1.1 justin {
445 1.1 justin enum rumpclock rclk = enum_rumpclock;
446 1.1 justin struct timespec ts;
447 1.1 justin clockid_t clk;
448 1.1 justin int rv;
449 1.1 justin
450 1.1 justin switch (rclk) {
451 1.1 justin case RUMPUSER_CLOCK_RELWALL:
452 1.1 justin clk = CLOCK_REALTIME;
453 1.1 justin break;
454 1.1 justin case RUMPUSER_CLOCK_ABSMONO:
455 1.1 justin clk = CLOCK_MONOTONIC;
456 1.1 justin break;
457 1.1 justin default:
458 1.1 justin abort();
459 1.1 justin }
460 1.1 justin
461 1.1 justin if (clock_gettime(clk, &ts) == -1) {
462 1.1 justin rv = errno;
463 1.1 justin } else {
464 1.1 justin *sec = ts.tv_sec;
465 1.1 justin *nsec = ts.tv_nsec;
466 1.1 justin rv = 0;
467 1.1 justin }
468 1.1 justin
469 1.1 justin ET(rv);
470 1.1 justin }
471 1.1 justin
472 1.1 justin int
473 1.1 justin rumpuser_clock_sleep(int enum_rumpclock, int64_t sec, long nsec)
474 1.1 justin {
475 1.1 justin enum rumpclock rclk = enum_rumpclock;
476 1.1 justin uint32_t msec;
477 1.1 justin int nlocks;
478 1.1 justin
479 1.1 justin rumpkern_unsched(&nlocks, NULL);
480 1.1 justin switch (rclk) {
481 1.1 justin case RUMPUSER_CLOCK_RELWALL:
482 1.1 justin msec = sec * 1000 + nsec / (1000*1000UL);
483 1.1 justin msleep(msec);
484 1.1 justin break;
485 1.1 justin case RUMPUSER_CLOCK_ABSMONO:
486 1.1 justin msec = sec * 1000 + nsec / (1000*1000UL);
487 1.1 justin abssleep(msec);
488 1.1 justin break;
489 1.1 justin }
490 1.1 justin rumpkern_sched(nlocks, NULL);
491 1.1 justin
492 1.1 justin return 0;
493 1.1 justin }
494 1.1 justin
495 1.1 justin int
496 1.1 justin rumpuser_getparam(const char *name, void *buf, size_t blen)
497 1.1 justin {
498 1.1 justin int rv;
499 1.1 justin const char *ncpu = "1";
500 1.1 justin
501 1.1 justin if (strcmp(name, RUMPUSER_PARAM_NCPU) == 0) {
502 1.1 justin strncpy(buf, ncpu, blen);
503 1.1 justin rv = 0;
504 1.1 justin } else if (strcmp(name, RUMPUSER_PARAM_HOSTNAME) == 0) {
505 1.1 justin char tmp[MAXHOSTNAMELEN];
506 1.1 justin
507 1.1 justin if (gethostname(tmp, sizeof(tmp)) == -1) {
508 1.1 justin snprintf(buf, blen, "rump-%05d", (int)getpid());
509 1.1 justin } else {
510 1.1 justin snprintf(buf, blen, "rump-%05d.%s",
511 1.1 justin (int)getpid(), tmp);
512 1.1 justin }
513 1.1 justin rv = 0;
514 1.1 justin } else if (*name == '_') {
515 1.1 justin rv = EINVAL;
516 1.1 justin } else {
517 1.1 justin if (getenv_r(name, buf, blen) == -1)
518 1.1 justin rv = errno;
519 1.1 justin else
520 1.1 justin rv = 0;
521 1.1 justin }
522 1.1 justin
523 1.1 justin ET(rv);
524 1.1 justin }
525 1.1 justin
526 1.1 justin void
527 1.1 justin rumpuser_putchar(int c)
528 1.1 justin {
529 1.1 justin
530 1.1 justin putchar(c);
531 1.1 justin }
532 1.1 justin
533 1.1 justin __dead void
534 1.1 justin rumpuser_exit(int rv)
535 1.1 justin {
536 1.1 justin
537 1.1 justin if (rv == RUMPUSER_PANIC)
538 1.1 justin abort();
539 1.1 justin else
540 1.1 justin exit(rv);
541 1.1 justin }
542 1.1 justin
543 1.1 justin void
544 1.1 justin rumpuser_seterrno(int error)
545 1.1 justin {
546 1.1 justin
547 1.1 justin errno = error;
548 1.1 justin }
549 1.1 justin
550 1.1 justin /*
551 1.1 justin * This is meant for safe debugging prints from the kernel.
552 1.1 justin */
553 1.1 justin void
554 1.1 justin rumpuser_dprintf(const char *format, ...)
555 1.1 justin {
556 1.1 justin va_list ap;
557 1.1 justin
558 1.1 justin va_start(ap, format);
559 1.1 justin vfprintf(stderr, format, ap);
560 1.1 justin va_end(ap);
561 1.1 justin }
562 1.1 justin
563 1.1 justin int
564 1.1 justin rumpuser_kill(int64_t pid, int rumpsig)
565 1.1 justin {
566 1.1 justin int sig;
567 1.1 justin
568 1.1 justin sig = rumpuser__sig_rump2host(rumpsig);
569 1.1 justin if (sig > 0)
570 1.1 justin raise(sig);
571 1.1 justin return 0;
572 1.1 justin }
573 1.1 justin
574 1.1 justin /* thread functions */
575 1.1 justin
576 1.1 justin TAILQ_HEAD(waithead, waiter);
577 1.1 justin struct waiter {
578 1.1 justin struct thread *who;
579 1.1 justin TAILQ_ENTRY(waiter) entries;
580 1.1 justin int onlist;
581 1.1 justin };
582 1.1 justin
583 1.1 justin static int
584 1.1 justin wait(struct waithead *wh, uint64_t msec)
585 1.1 justin {
586 1.1 justin struct waiter w;
587 1.1 justin
588 1.1 justin w.who = get_current();
589 1.1 justin TAILQ_INSERT_TAIL(wh, &w, entries);
590 1.1 justin w.onlist = 1;
591 1.1 justin block(w.who);
592 1.1 justin if (msec)
593 1.1 justin w.who->wakeup_time = now() + msec;
594 1.1 justin schedule();
595 1.1 justin
596 1.1 justin /* woken up by timeout? */
597 1.1 justin if (w.onlist)
598 1.1 justin TAILQ_REMOVE(wh, &w, entries);
599 1.1 justin
600 1.1 justin return w.onlist ? ETIMEDOUT : 0;
601 1.1 justin }
602 1.1 justin
603 1.1 justin static void
604 1.1 justin wakeup_one(struct waithead *wh)
605 1.1 justin {
606 1.1 justin struct waiter *w;
607 1.1 justin
608 1.1 justin if ((w = TAILQ_FIRST(wh)) != NULL) {
609 1.1 justin TAILQ_REMOVE(wh, w, entries);
610 1.1 justin w->onlist = 0;
611 1.1 justin wake(w->who);
612 1.1 justin }
613 1.1 justin }
614 1.1 justin
615 1.1 justin static void
616 1.1 justin wakeup_all(struct waithead *wh)
617 1.1 justin {
618 1.1 justin struct waiter *w;
619 1.1 justin
620 1.1 justin while ((w = TAILQ_FIRST(wh)) != NULL) {
621 1.1 justin TAILQ_REMOVE(wh, w, entries);
622 1.1 justin w->onlist = 0;
623 1.1 justin wake(w->who);
624 1.1 justin }
625 1.1 justin }
626 1.1 justin
627 1.1 justin int
628 1.1 justin rumpuser_thread_create(void *(*f)(void *), void *arg, const char *thrname,
629 1.1 justin int joinable, int pri, int cpuidx, void **tptr)
630 1.1 justin {
631 1.1 justin struct thread *thr;
632 1.1 justin
633 1.1 justin thr = create_thread(thrname, NULL, (void (*)(void *))f, arg, NULL, 0);
634 1.1 justin /*
635 1.1 justin * XXX: should be supplied as a flag to create_thread() so as to
636 1.1 justin * _ensure_ it's set before the thread runs (and could exit).
637 1.1 justin * now we're trusting unclear semantics of create_thread()
638 1.1 justin */
639 1.1 justin if (thr && joinable)
640 1.1 justin thr->flags |= THREAD_MUSTJOIN;
641 1.1 justin
642 1.1 justin if (!thr)
643 1.1 justin return EINVAL;
644 1.1 justin
645 1.1 justin *tptr = thr;
646 1.1 justin return 0;
647 1.1 justin }
648 1.1 justin
649 1.1 justin void
650 1.1 justin rumpuser_thread_exit(void)
651 1.1 justin {
652 1.1 justin
653 1.1 justin exit_thread();
654 1.1 justin }
655 1.1 justin
656 1.1 justin int
657 1.1 justin rumpuser_thread_join(void *p)
658 1.1 justin {
659 1.1 justin
660 1.1 justin join_thread(p);
661 1.1 justin return 0;
662 1.1 justin }
663 1.1 justin
664 1.1 justin struct rumpuser_mtx {
665 1.1 justin struct waithead waiters;
666 1.1 justin int v;
667 1.1 justin int flags;
668 1.1 justin struct lwp *o;
669 1.1 justin };
670 1.1 justin
671 1.1 justin void
672 1.1 justin rumpuser_mutex_init(struct rumpuser_mtx **mtxp, int flags)
673 1.1 justin {
674 1.1 justin struct rumpuser_mtx *mtx;
675 1.1 justin
676 1.1 justin mtx = malloc(sizeof(*mtx));
677 1.1 justin memset(mtx, 0, sizeof(*mtx));
678 1.1 justin mtx->flags = flags;
679 1.1 justin TAILQ_INIT(&mtx->waiters);
680 1.1 justin *mtxp = mtx;
681 1.1 justin }
682 1.1 justin
683 1.1 justin void
684 1.1 justin rumpuser_mutex_enter(struct rumpuser_mtx *mtx)
685 1.1 justin {
686 1.1 justin int nlocks;
687 1.1 justin
688 1.1 justin if (rumpuser_mutex_tryenter(mtx) != 0) {
689 1.1 justin rumpkern_unsched(&nlocks, NULL);
690 1.1 justin while (rumpuser_mutex_tryenter(mtx) != 0)
691 1.1 justin wait(&mtx->waiters, 0);
692 1.1 justin rumpkern_sched(nlocks, NULL);
693 1.1 justin }
694 1.1 justin }
695 1.1 justin
696 1.1 justin void
697 1.1 justin rumpuser_mutex_enter_nowrap(struct rumpuser_mtx *mtx)
698 1.1 justin {
699 1.1 justin int rv;
700 1.1 justin
701 1.1 justin rv = rumpuser_mutex_tryenter(mtx);
702 1.1 justin /* one VCPU supported, no preemption => must succeed */
703 1.1 justin if (rv != 0) {
704 1.1 justin printk("no voi ei\n");
705 1.1 justin }
706 1.1 justin }
707 1.1 justin
708 1.1 justin int
709 1.1 justin rumpuser_mutex_tryenter(struct rumpuser_mtx *mtx)
710 1.1 justin {
711 1.1 justin struct lwp *l = get_current()->lwp;
712 1.1 justin
713 1.1 justin if (mtx->v && mtx->o != l)
714 1.1 justin return EBUSY;
715 1.1 justin
716 1.1 justin mtx->v++;
717 1.1 justin mtx->o = l;
718 1.1 justin
719 1.1 justin return 0;
720 1.1 justin }
721 1.1 justin
722 1.1 justin void
723 1.1 justin rumpuser_mutex_exit(struct rumpuser_mtx *mtx)
724 1.1 justin {
725 1.1 justin
726 1.1 justin assert(mtx->v > 0);
727 1.1 justin if (--mtx->v == 0) {
728 1.1 justin mtx->o = NULL;
729 1.1 justin wakeup_one(&mtx->waiters);
730 1.1 justin }
731 1.1 justin }
732 1.1 justin
733 1.1 justin void
734 1.1 justin rumpuser_mutex_destroy(struct rumpuser_mtx *mtx)
735 1.1 justin {
736 1.1 justin
737 1.1 justin assert(TAILQ_EMPTY(&mtx->waiters) && mtx->o == NULL);
738 1.1 justin free(mtx);
739 1.1 justin }
740 1.1 justin
741 1.1 justin void
742 1.1 justin rumpuser_mutex_owner(struct rumpuser_mtx *mtx, struct lwp **lp)
743 1.1 justin {
744 1.1 justin
745 1.1 justin *lp = mtx->o;
746 1.1 justin }
747 1.1 justin
748 1.1 justin struct rumpuser_rw {
749 1.1 justin struct waithead rwait;
750 1.1 justin struct waithead wwait;
751 1.1 justin int v;
752 1.1 justin struct lwp *o;
753 1.1 justin };
754 1.1 justin
755 1.1 justin void
756 1.1 justin rumpuser_rw_init(struct rumpuser_rw **rwp)
757 1.1 justin {
758 1.1 justin struct rumpuser_rw *rw;
759 1.1 justin
760 1.1 justin rw = malloc(sizeof(*rw));
761 1.1 justin memset(rw, 0, sizeof(*rw));
762 1.1 justin TAILQ_INIT(&rw->rwait);
763 1.1 justin TAILQ_INIT(&rw->wwait);
764 1.1 justin
765 1.1 justin *rwp = rw;
766 1.1 justin }
767 1.1 justin
768 1.1 justin void
769 1.1 justin rumpuser_rw_enter(int enum_rumprwlock, struct rumpuser_rw *rw)
770 1.1 justin {
771 1.1 justin enum rumprwlock lk = enum_rumprwlock;
772 1.1 justin struct waithead *w = NULL;
773 1.1 justin int nlocks;
774 1.1 justin
775 1.1 justin switch (lk) {
776 1.1 justin case RUMPUSER_RW_WRITER:
777 1.1 justin w = &rw->wwait;
778 1.1 justin break;
779 1.1 justin case RUMPUSER_RW_READER:
780 1.1 justin w = &rw->rwait;
781 1.1 justin break;
782 1.1 justin }
783 1.1 justin
784 1.1 justin if (rumpuser_rw_tryenter(enum_rumprwlock, rw) != 0) {
785 1.1 justin rumpkern_unsched(&nlocks, NULL);
786 1.1 justin while (rumpuser_rw_tryenter(enum_rumprwlock, rw) != 0)
787 1.1 justin wait(w, 0);
788 1.1 justin rumpkern_sched(nlocks, NULL);
789 1.1 justin }
790 1.1 justin }
791 1.1 justin
792 1.1 justin int
793 1.1 justin rumpuser_rw_tryenter(int enum_rumprwlock, struct rumpuser_rw *rw)
794 1.1 justin {
795 1.1 justin enum rumprwlock lk = enum_rumprwlock;
796 1.1 justin int rv;
797 1.1 justin
798 1.1 justin switch (lk) {
799 1.1 justin case RUMPUSER_RW_WRITER:
800 1.1 justin if (rw->o == NULL) {
801 1.1 justin rw->o = rumpuser_curlwp();
802 1.1 justin rv = 0;
803 1.1 justin } else {
804 1.1 justin rv = EBUSY;
805 1.1 justin }
806 1.1 justin break;
807 1.1 justin case RUMPUSER_RW_READER:
808 1.1 justin if (rw->o == NULL && TAILQ_EMPTY(&rw->wwait)) {
809 1.1 justin rw->v++;
810 1.1 justin rv = 0;
811 1.1 justin } else {
812 1.1 justin rv = EBUSY;
813 1.1 justin }
814 1.1 justin break;
815 1.1 justin default:
816 1.1 justin rv = EINVAL;
817 1.1 justin }
818 1.1 justin
819 1.1 justin return rv;
820 1.1 justin }
821 1.1 justin
822 1.1 justin void
823 1.1 justin rumpuser_rw_exit(struct rumpuser_rw *rw)
824 1.1 justin {
825 1.1 justin
826 1.1 justin if (rw->o) {
827 1.1 justin rw->o = NULL;
828 1.1 justin } else {
829 1.1 justin rw->v--;
830 1.1 justin }
831 1.1 justin
832 1.1 justin /* standard procedure, don't let readers starve out writers */
833 1.1 justin if (!TAILQ_EMPTY(&rw->wwait)) {
834 1.1 justin if (rw->o == NULL)
835 1.1 justin wakeup_one(&rw->wwait);
836 1.1 justin } else if (!TAILQ_EMPTY(&rw->rwait) && rw->o == NULL) {
837 1.1 justin wakeup_all(&rw->rwait);
838 1.1 justin }
839 1.1 justin }
840 1.1 justin
841 1.1 justin void
842 1.1 justin rumpuser_rw_destroy(struct rumpuser_rw *rw)
843 1.1 justin {
844 1.1 justin
845 1.1 justin free(rw);
846 1.1 justin }
847 1.1 justin
848 1.1 justin void
849 1.1 justin rumpuser_rw_held(int enum_rumprwlock, struct rumpuser_rw *rw, int *rvp)
850 1.1 justin {
851 1.1 justin enum rumprwlock lk = enum_rumprwlock;
852 1.1 justin
853 1.1 justin switch (lk) {
854 1.1 justin case RUMPUSER_RW_WRITER:
855 1.1 justin *rvp = rw->o == rumpuser_curlwp();
856 1.1 justin break;
857 1.1 justin case RUMPUSER_RW_READER:
858 1.1 justin *rvp = rw->v > 0;
859 1.1 justin break;
860 1.1 justin }
861 1.1 justin }
862 1.1 justin
863 1.1 justin void
864 1.1 justin rumpuser_rw_downgrade(struct rumpuser_rw *rw)
865 1.1 justin {
866 1.1 justin
867 1.1 justin assert(rw->o == rumpuser_curlwp());
868 1.1 justin rw->v = -1;
869 1.1 justin }
870 1.1 justin
871 1.1 justin int
872 1.1 justin rumpuser_rw_tryupgrade(struct rumpuser_rw *rw)
873 1.1 justin {
874 1.1 justin
875 1.1 justin if (rw->v == -1) {
876 1.1 justin rw->v = 1;
877 1.1 justin rw->o = rumpuser_curlwp();
878 1.1 justin return 0;
879 1.1 justin }
880 1.1 justin
881 1.1 justin return EBUSY;
882 1.1 justin }
883 1.1 justin
884 1.1 justin struct rumpuser_cv {
885 1.1 justin struct waithead waiters;
886 1.1 justin int nwaiters;
887 1.1 justin };
888 1.1 justin
889 1.1 justin void
890 1.1 justin rumpuser_cv_init(struct rumpuser_cv **cvp)
891 1.1 justin {
892 1.1 justin struct rumpuser_cv *cv;
893 1.1 justin
894 1.1 justin cv = malloc(sizeof(*cv));
895 1.1 justin memset(cv, 0, sizeof(*cv));
896 1.1 justin TAILQ_INIT(&cv->waiters);
897 1.1 justin *cvp = cv;
898 1.1 justin }
899 1.1 justin
900 1.1 justin void
901 1.1 justin rumpuser_cv_destroy(struct rumpuser_cv *cv)
902 1.1 justin {
903 1.1 justin
904 1.1 justin assert(cv->nwaiters == 0);
905 1.1 justin free(cv);
906 1.1 justin }
907 1.1 justin
908 1.1 justin static void
909 1.1 justin cv_unsched(struct rumpuser_mtx *mtx, int *nlocks)
910 1.1 justin {
911 1.1 justin
912 1.1 justin rumpkern_unsched(nlocks, mtx);
913 1.1 justin rumpuser_mutex_exit(mtx);
914 1.1 justin }
915 1.1 justin
916 1.1 justin static void
917 1.1 justin cv_resched(struct rumpuser_mtx *mtx, int nlocks)
918 1.1 justin {
919 1.1 justin
920 1.1 justin /* see rumpuser(3) */
921 1.1 justin if ((mtx->flags & (RUMPUSER_MTX_KMUTEX | RUMPUSER_MTX_SPIN)) ==
922 1.1 justin (RUMPUSER_MTX_KMUTEX | RUMPUSER_MTX_SPIN)) {
923 1.1 justin rumpkern_sched(nlocks, mtx);
924 1.1 justin rumpuser_mutex_enter_nowrap(mtx);
925 1.1 justin } else {
926 1.1 justin rumpuser_mutex_enter_nowrap(mtx);
927 1.1 justin rumpkern_sched(nlocks, mtx);
928 1.1 justin }
929 1.1 justin }
930 1.1 justin
931 1.1 justin void
932 1.1 justin rumpuser_cv_wait(struct rumpuser_cv *cv, struct rumpuser_mtx *mtx)
933 1.1 justin {
934 1.1 justin int nlocks;
935 1.1 justin
936 1.1 justin cv->nwaiters++;
937 1.1 justin cv_unsched(mtx, &nlocks);
938 1.1 justin wait(&cv->waiters, 0);
939 1.1 justin cv_resched(mtx, nlocks);
940 1.1 justin cv->nwaiters--;
941 1.1 justin }
942 1.1 justin
943 1.1 justin void
944 1.1 justin rumpuser_cv_wait_nowrap(struct rumpuser_cv *cv, struct rumpuser_mtx *mtx)
945 1.1 justin {
946 1.1 justin
947 1.1 justin cv->nwaiters++;
948 1.1 justin rumpuser_mutex_exit(mtx);
949 1.1 justin wait(&cv->waiters, 0);
950 1.1 justin rumpuser_mutex_enter_nowrap(mtx);
951 1.1 justin cv->nwaiters--;
952 1.1 justin }
953 1.1 justin
954 1.1 justin int
955 1.1 justin rumpuser_cv_timedwait(struct rumpuser_cv *cv, struct rumpuser_mtx *mtx,
956 1.1 justin int64_t sec, int64_t nsec)
957 1.1 justin {
958 1.1 justin int nlocks;
959 1.1 justin int rv;
960 1.1 justin
961 1.1 justin cv->nwaiters++;
962 1.1 justin cv_unsched(mtx, &nlocks);
963 1.1 justin rv = wait(&cv->waiters, sec * 1000 + nsec / (1000*1000));
964 1.1 justin cv_resched(mtx, nlocks);
965 1.1 justin cv->nwaiters--;
966 1.1 justin
967 1.1 justin return rv;
968 1.1 justin }
969 1.1 justin
970 1.1 justin void
971 1.1 justin rumpuser_cv_signal(struct rumpuser_cv *cv)
972 1.1 justin {
973 1.1 justin
974 1.1 justin wakeup_one(&cv->waiters);
975 1.1 justin }
976 1.1 justin
977 1.1 justin void
978 1.1 justin rumpuser_cv_broadcast(struct rumpuser_cv *cv)
979 1.1 justin {
980 1.1 justin
981 1.1 justin wakeup_all(&cv->waiters);
982 1.1 justin }
983 1.1 justin
984 1.1 justin void
985 1.1 justin rumpuser_cv_has_waiters(struct rumpuser_cv *cv, int *rvp)
986 1.1 justin {
987 1.1 justin
988 1.1 justin *rvp = cv->nwaiters != 0;
989 1.1 justin }
990 1.1 justin
991 1.1 justin /*
992 1.1 justin * curlwp
993 1.1 justin */
994 1.1 justin
995 1.1 justin void
996 1.1 justin rumpuser_curlwpop(int enum_rumplwpop, struct lwp *l)
997 1.1 justin {
998 1.1 justin struct thread *thread;
999 1.1 justin enum rumplwpop op = enum_rumplwpop;
1000 1.1 justin
1001 1.1 justin switch (op) {
1002 1.1 justin case RUMPUSER_LWP_CREATE:
1003 1.1 justin case RUMPUSER_LWP_DESTROY:
1004 1.1 justin break;
1005 1.1 justin case RUMPUSER_LWP_SET:
1006 1.1 justin thread = get_current();
1007 1.1 justin thread->lwp = l;
1008 1.1 justin break;
1009 1.1 justin case RUMPUSER_LWP_CLEAR:
1010 1.1 justin thread = get_current();
1011 1.1 justin assert(thread->lwp == l);
1012 1.1 justin thread->lwp = NULL;
1013 1.1 justin break;
1014 1.1 justin }
1015 1.1 justin }
1016 1.1 justin
1017 1.1 justin struct lwp *
1018 1.1 justin rumpuser_curlwp(void)
1019 1.1 justin {
1020 1.1 justin
1021 1.1 justin return get_current()->lwp;
1022 1.1 justin }
1023