rumpfiber.c revision 1.4 1 1.4 pooka /* $NetBSD: rumpfiber.c,v 1.4 2014/08/24 14:37:31 pooka 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.4 pooka __RCSID("$NetBSD: rumpfiber.c,v 1.4 2014/08/24 14:37:31 pooka 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.1 justin
123 1.1 justin clock_gettime(CLOCK_MONOTONIC, &ts);
124 1.1 justin return (ts.tv_sec * 1000LL) + (ts.tv_nsec / 1000000LL);
125 1.1 justin }
126 1.1 justin
127 1.1 justin void
128 1.1 justin schedule(void)
129 1.1 justin {
130 1.1 justin struct thread *prev, *next, *thread, *tmp;
131 1.1 justin int64_t tm, wakeup;
132 1.1 justin struct timespec sl;
133 1.1 justin
134 1.1 justin prev = get_current();
135 1.1 justin
136 1.1 justin do {
137 1.1 justin tm = now();
138 1.1 justin wakeup = tm + 1000; /* wake up in 1s max */
139 1.1 justin next = NULL;
140 1.1 justin TAILQ_FOREACH_SAFE(thread, &thread_list, thread_list, tmp) {
141 1.1 justin if (!is_runnable(thread) && thread->wakeup_time >= 0) {
142 1.1 justin if (thread->wakeup_time <= tm) {
143 1.1 justin thread->flags |= THREAD_TIMEDOUT;
144 1.1 justin wake(thread);
145 1.1 justin } else if (thread->wakeup_time < wakeup)
146 1.1 justin wakeup = thread->wakeup_time;
147 1.1 justin }
148 1.1 justin if (is_runnable(thread)) {
149 1.1 justin next = thread;
150 1.1 justin /* Put this thread on the end of the list */
151 1.1 justin TAILQ_REMOVE(&thread_list, thread, thread_list);
152 1.1 justin TAILQ_INSERT_TAIL(&thread_list, thread, thread_list);
153 1.1 justin break;
154 1.1 justin }
155 1.1 justin }
156 1.1 justin if (next)
157 1.1 justin break;
158 1.1 justin sl.tv_sec = (wakeup - tm) / 1000;
159 1.1 justin sl.tv_nsec = ((wakeup - tm) - 1000 * sl.tv_sec) * 1000000;
160 1.1 justin #ifdef HAVE_CLOCK_NANOSLEEP
161 1.1 justin clock_nanosleep(CLOCK_MONOTONIC, 0, &sl, NULL);
162 1.1 justin #else
163 1.1 justin nanosleep(&sl, NULL);
164 1.1 justin #endif
165 1.1 justin } while (1);
166 1.1 justin
167 1.1 justin if (prev != next)
168 1.1 justin switch_threads(prev, next);
169 1.1 justin
170 1.1 justin TAILQ_FOREACH_SAFE(thread, &exited_threads, thread_list, tmp) {
171 1.1 justin if (thread != prev) {
172 1.1 justin TAILQ_REMOVE(&exited_threads, thread, thread_list);
173 1.1 justin if ((thread->flags & THREAD_EXTSTACK) == 0)
174 1.1 justin munmap(thread->ctx.uc_stack.ss_sp, STACKSIZE);
175 1.1 justin free(thread->name);
176 1.1 justin free(thread);
177 1.1 justin }
178 1.1 justin }
179 1.1 justin }
180 1.1 justin
181 1.1 justin static void
182 1.1 justin create_ctx(ucontext_t *ctx, void *stack, size_t stack_size)
183 1.1 justin {
184 1.1 justin
185 1.1 justin getcontext(ctx);
186 1.1 justin ctx->uc_stack.ss_sp = stack;
187 1.1 justin ctx->uc_stack.ss_size = stack_size;
188 1.1 justin ctx->uc_stack.ss_flags = 0;
189 1.1 justin ctx->uc_link = NULL; /* TODO may link to main thread */
190 1.1 justin }
191 1.1 justin
192 1.1 justin /* may have to do bounce function to call, if args to makecontext are ints */
193 1.1 justin /* TODO see notes in rumpuser_thread_create, have flags here */
194 1.1 justin struct thread *
195 1.1 justin create_thread(const char *name, void *cookie, void (*f)(void *), void *data,
196 1.1 justin void *stack, size_t stack_size)
197 1.1 justin {
198 1.1 justin struct thread *thread = calloc(1, sizeof(struct thread));
199 1.1 justin
200 1.1 justin if (!stack) {
201 1.1 justin assert(stack_size == 0);
202 1.1 justin stack = mmap(NULL, STACKSIZE, PROT_READ | PROT_WRITE,
203 1.1 justin MAP_SHARED | MAP_ANON, -1, 0);
204 1.1 justin if (stack == MAP_FAILED) {
205 1.1 justin return NULL;
206 1.1 justin }
207 1.1 justin stack_size = STACKSIZE;
208 1.1 justin } else {
209 1.1 justin thread->flags = THREAD_EXTSTACK;
210 1.1 justin }
211 1.1 justin create_ctx(&thread->ctx, stack, stack_size);
212 1.1 justin makecontext(&thread->ctx, (void (*)(void))f, 1, data);
213 1.1 justin
214 1.1 justin thread->name = strdup(name);
215 1.1 justin thread->cookie = cookie;
216 1.1 justin
217 1.1 justin /* Not runnable, not exited, not sleeping */
218 1.1 justin thread->wakeup_time = -1;
219 1.1 justin thread->lwp = NULL;
220 1.1 justin set_runnable(thread);
221 1.1 justin TAILQ_INSERT_TAIL(&thread_list, thread, thread_list);
222 1.1 justin
223 1.1 justin return thread;
224 1.1 justin }
225 1.1 justin
226 1.1 justin static void
227 1.1 justin switch_threads(struct thread *prev, struct thread *next)
228 1.1 justin {
229 1.1 justin int ret;
230 1.1 justin
231 1.1 justin current_thread = next;
232 1.1 justin if (scheduler_hook)
233 1.1 justin scheduler_hook(prev->cookie, next->cookie);
234 1.1 justin ret = swapcontext(&prev->ctx, &next->ctx);
235 1.1 justin if (ret < 0) {
236 1.1 justin printk("swapcontext failed\n");
237 1.1 justin abort();
238 1.1 justin }
239 1.1 justin }
240 1.1 justin
241 1.1 justin struct join_waiter {
242 1.1 justin struct thread *jw_thread;
243 1.1 justin struct thread *jw_wanted;
244 1.1 justin TAILQ_ENTRY(join_waiter) jw_entries;
245 1.1 justin };
246 1.1 justin static TAILQ_HEAD(, join_waiter) joinwq = TAILQ_HEAD_INITIALIZER(joinwq);
247 1.1 justin
248 1.1 justin void
249 1.1 justin exit_thread(void)
250 1.1 justin {
251 1.1 justin struct thread *thread = get_current();
252 1.1 justin struct join_waiter *jw_iter;
253 1.1 justin
254 1.1 justin /* if joinable, gate until we are allowed to exit */
255 1.1 justin while (thread->flags & THREAD_MUSTJOIN) {
256 1.1 justin thread->flags |= THREAD_JOINED;
257 1.1 justin
258 1.1 justin /* see if the joiner is already there */
259 1.1 justin TAILQ_FOREACH(jw_iter, &joinwq, jw_entries) {
260 1.1 justin if (jw_iter->jw_wanted == thread) {
261 1.1 justin wake(jw_iter->jw_thread);
262 1.1 justin break;
263 1.1 justin }
264 1.1 justin }
265 1.1 justin block(thread);
266 1.1 justin schedule();
267 1.1 justin }
268 1.1 justin
269 1.1 justin /* Remove from the thread list */
270 1.1 justin TAILQ_REMOVE(&thread_list, thread, thread_list);
271 1.1 justin clear_runnable(thread);
272 1.1 justin /* Put onto exited list */
273 1.1 justin TAILQ_INSERT_HEAD(&exited_threads, thread, thread_list);
274 1.1 justin
275 1.1 justin /* Schedule will free the resources */
276 1.1 justin while (1) {
277 1.1 justin schedule();
278 1.1 justin printk("schedule() returned! Trying again\n");
279 1.1 justin }
280 1.1 justin }
281 1.1 justin
282 1.1 justin static void
283 1.1 justin join_thread(struct thread *joinable)
284 1.1 justin {
285 1.1 justin struct join_waiter jw;
286 1.1 justin struct thread *thread = get_current();
287 1.1 justin
288 1.1 justin assert(joinable->flags & THREAD_MUSTJOIN);
289 1.1 justin
290 1.1 justin /* wait for exiting thread to hit thread_exit() */
291 1.1 justin while (! (joinable->flags & THREAD_JOINED)) {
292 1.1 justin
293 1.1 justin jw.jw_thread = thread;
294 1.1 justin jw.jw_wanted = joinable;
295 1.1 justin TAILQ_INSERT_TAIL(&joinwq, &jw, jw_entries);
296 1.1 justin block(thread);
297 1.1 justin schedule();
298 1.1 justin TAILQ_REMOVE(&joinwq, &jw, jw_entries);
299 1.1 justin }
300 1.1 justin
301 1.1 justin /* signal exiting thread that we have seen it and it may now exit */
302 1.1 justin assert(joinable->flags & THREAD_JOINED);
303 1.1 justin joinable->flags &= ~THREAD_MUSTJOIN;
304 1.1 justin
305 1.1 justin wake(joinable);
306 1.1 justin }
307 1.1 justin
308 1.1 justin static void msleep(uint64_t millisecs)
309 1.1 justin {
310 1.1 justin struct thread *thread = get_current();
311 1.1 justin
312 1.1 justin thread->wakeup_time = now() + millisecs;
313 1.1 justin clear_runnable(thread);
314 1.1 justin schedule();
315 1.1 justin }
316 1.1 justin
317 1.1 justin static void abssleep(uint64_t millisecs)
318 1.1 justin {
319 1.1 justin struct thread *thread = get_current();
320 1.1 justin
321 1.1 justin thread->wakeup_time = millisecs;
322 1.1 justin clear_runnable(thread);
323 1.1 justin schedule();
324 1.1 justin }
325 1.1 justin
326 1.1 justin /* like abssleep, except against realtime clock instead of monotonic clock */
327 1.1 justin int abssleep_real(uint64_t millisecs)
328 1.1 justin {
329 1.1 justin struct thread *thread = get_current();
330 1.1 justin struct timespec ts;
331 1.1 justin uint64_t real_now;
332 1.1 justin int rv;
333 1.1 justin
334 1.1 justin clock_gettime(CLOCK_REALTIME, &ts);
335 1.1 justin real_now = 1000*ts.tv_sec + ts.tv_nsec/(1000*1000);
336 1.1 justin thread->wakeup_time = now() + (millisecs - real_now);
337 1.1 justin
338 1.1 justin clear_runnable(thread);
339 1.1 justin schedule();
340 1.1 justin
341 1.1 justin rv = !!(thread->flags & THREAD_TIMEDOUT);
342 1.1 justin thread->flags &= ~THREAD_TIMEDOUT;
343 1.1 justin return rv;
344 1.1 justin }
345 1.1 justin
346 1.1 justin void wake(struct thread *thread)
347 1.1 justin {
348 1.1 justin
349 1.1 justin thread->wakeup_time = -1;
350 1.1 justin set_runnable(thread);
351 1.1 justin }
352 1.1 justin
353 1.1 justin void block(struct thread *thread)
354 1.1 justin {
355 1.1 justin
356 1.1 justin thread->wakeup_time = -1;
357 1.1 justin clear_runnable(thread);
358 1.1 justin }
359 1.1 justin
360 1.1 justin int is_runnable(struct thread *thread)
361 1.1 justin {
362 1.1 justin
363 1.1 justin return thread->flags & RUNNABLE_FLAG;
364 1.1 justin }
365 1.1 justin
366 1.1 justin void set_runnable(struct thread *thread)
367 1.1 justin {
368 1.1 justin
369 1.1 justin thread->flags |= RUNNABLE_FLAG;
370 1.1 justin }
371 1.1 justin
372 1.1 justin void clear_runnable(struct thread *thread)
373 1.1 justin {
374 1.1 justin
375 1.1 justin thread->flags &= ~RUNNABLE_FLAG;
376 1.1 justin }
377 1.1 justin
378 1.1 justin static void
379 1.1 justin init_sched(void)
380 1.1 justin {
381 1.1 justin struct thread *thread = calloc(1, sizeof(struct thread));
382 1.1 justin
383 1.1 justin getcontext(&thread->ctx);
384 1.1 justin thread->name = strdup("init");
385 1.1 justin thread->flags = 0;
386 1.1 justin thread->wakeup_time = -1;
387 1.1 justin thread->lwp = NULL;
388 1.1 justin set_runnable(thread);
389 1.1 justin TAILQ_INSERT_TAIL(&thread_list, thread, thread_list);
390 1.1 justin current_thread = thread;
391 1.1 justin }
392 1.1 justin
393 1.1 justin void
394 1.1 justin set_sched_hook(void (*f)(void *, void *))
395 1.1 justin {
396 1.1 justin
397 1.1 justin scheduler_hook = f;
398 1.1 justin }
399 1.1 justin
400 1.1 justin struct thread *
401 1.1 justin init_mainthread(void *cookie)
402 1.1 justin {
403 1.1 justin
404 1.1 justin current_thread->cookie = cookie;
405 1.1 justin return current_thread;
406 1.1 justin }
407 1.1 justin
408 1.1 justin /* rump functions below */
409 1.1 justin
410 1.1 justin struct rumpuser_hyperup rumpuser__hyp;
411 1.1 justin
412 1.1 justin int
413 1.1 justin rumpuser_init(int version, const struct rumpuser_hyperup *hyp)
414 1.1 justin {
415 1.3 justin int rv;
416 1.1 justin
417 1.1 justin if (version != RUMPUSER_VERSION) {
418 1.1 justin printk("rumpuser version mismatch\n");
419 1.3 justin abort();
420 1.1 justin }
421 1.1 justin
422 1.3 justin rv = rumpuser__random_init();
423 1.3 justin if (rv != 0) {
424 1.3 justin ET(rv);
425 1.1 justin }
426 1.1 justin
427 1.1 justin rumpuser__hyp = *hyp;
428 1.1 justin
429 1.1 justin init_sched();
430 1.1 justin
431 1.1 justin return 0;
432 1.1 justin }
433 1.1 justin
434 1.1 justin int
435 1.1 justin rumpuser_clock_gettime(int enum_rumpclock, int64_t *sec, long *nsec)
436 1.1 justin {
437 1.1 justin enum rumpclock rclk = enum_rumpclock;
438 1.1 justin struct timespec ts;
439 1.1 justin clockid_t clk;
440 1.1 justin int rv;
441 1.1 justin
442 1.1 justin switch (rclk) {
443 1.1 justin case RUMPUSER_CLOCK_RELWALL:
444 1.1 justin clk = CLOCK_REALTIME;
445 1.1 justin break;
446 1.1 justin case RUMPUSER_CLOCK_ABSMONO:
447 1.1 justin clk = CLOCK_MONOTONIC;
448 1.1 justin break;
449 1.1 justin default:
450 1.1 justin abort();
451 1.1 justin }
452 1.1 justin
453 1.1 justin if (clock_gettime(clk, &ts) == -1) {
454 1.1 justin rv = errno;
455 1.1 justin } else {
456 1.1 justin *sec = ts.tv_sec;
457 1.1 justin *nsec = ts.tv_nsec;
458 1.1 justin rv = 0;
459 1.1 justin }
460 1.1 justin
461 1.1 justin ET(rv);
462 1.1 justin }
463 1.1 justin
464 1.1 justin int
465 1.1 justin rumpuser_clock_sleep(int enum_rumpclock, int64_t sec, long nsec)
466 1.1 justin {
467 1.1 justin enum rumpclock rclk = enum_rumpclock;
468 1.1 justin uint32_t msec;
469 1.1 justin int nlocks;
470 1.1 justin
471 1.1 justin rumpkern_unsched(&nlocks, NULL);
472 1.1 justin switch (rclk) {
473 1.1 justin case RUMPUSER_CLOCK_RELWALL:
474 1.1 justin msec = sec * 1000 + nsec / (1000*1000UL);
475 1.1 justin msleep(msec);
476 1.1 justin break;
477 1.1 justin case RUMPUSER_CLOCK_ABSMONO:
478 1.1 justin msec = sec * 1000 + nsec / (1000*1000UL);
479 1.1 justin abssleep(msec);
480 1.1 justin break;
481 1.1 justin }
482 1.1 justin rumpkern_sched(nlocks, NULL);
483 1.1 justin
484 1.1 justin return 0;
485 1.1 justin }
486 1.1 justin
487 1.1 justin int
488 1.1 justin rumpuser_getparam(const char *name, void *buf, size_t blen)
489 1.1 justin {
490 1.1 justin int rv;
491 1.1 justin const char *ncpu = "1";
492 1.1 justin
493 1.1 justin if (strcmp(name, RUMPUSER_PARAM_NCPU) == 0) {
494 1.1 justin strncpy(buf, ncpu, blen);
495 1.1 justin rv = 0;
496 1.1 justin } else if (strcmp(name, RUMPUSER_PARAM_HOSTNAME) == 0) {
497 1.1 justin char tmp[MAXHOSTNAMELEN];
498 1.1 justin
499 1.1 justin if (gethostname(tmp, sizeof(tmp)) == -1) {
500 1.1 justin snprintf(buf, blen, "rump-%05d", (int)getpid());
501 1.1 justin } else {
502 1.1 justin snprintf(buf, blen, "rump-%05d.%s",
503 1.1 justin (int)getpid(), tmp);
504 1.1 justin }
505 1.1 justin rv = 0;
506 1.1 justin } else if (*name == '_') {
507 1.1 justin rv = EINVAL;
508 1.1 justin } else {
509 1.1 justin if (getenv_r(name, buf, blen) == -1)
510 1.1 justin rv = errno;
511 1.1 justin else
512 1.1 justin rv = 0;
513 1.1 justin }
514 1.1 justin
515 1.1 justin ET(rv);
516 1.1 justin }
517 1.1 justin
518 1.1 justin void
519 1.1 justin rumpuser_putchar(int c)
520 1.1 justin {
521 1.1 justin
522 1.1 justin putchar(c);
523 1.1 justin }
524 1.1 justin
525 1.1 justin __dead void
526 1.1 justin rumpuser_exit(int rv)
527 1.1 justin {
528 1.1 justin
529 1.1 justin if (rv == RUMPUSER_PANIC)
530 1.1 justin abort();
531 1.1 justin else
532 1.1 justin exit(rv);
533 1.1 justin }
534 1.1 justin
535 1.1 justin void
536 1.1 justin rumpuser_seterrno(int error)
537 1.1 justin {
538 1.1 justin
539 1.1 justin errno = error;
540 1.1 justin }
541 1.1 justin
542 1.1 justin /*
543 1.1 justin * This is meant for safe debugging prints from the kernel.
544 1.1 justin */
545 1.1 justin void
546 1.1 justin rumpuser_dprintf(const char *format, ...)
547 1.1 justin {
548 1.1 justin va_list ap;
549 1.1 justin
550 1.1 justin va_start(ap, format);
551 1.1 justin vfprintf(stderr, format, ap);
552 1.1 justin va_end(ap);
553 1.1 justin }
554 1.1 justin
555 1.1 justin int
556 1.1 justin rumpuser_kill(int64_t pid, int rumpsig)
557 1.1 justin {
558 1.1 justin int sig;
559 1.1 justin
560 1.1 justin sig = rumpuser__sig_rump2host(rumpsig);
561 1.1 justin if (sig > 0)
562 1.1 justin raise(sig);
563 1.1 justin return 0;
564 1.1 justin }
565 1.1 justin
566 1.1 justin /* thread functions */
567 1.1 justin
568 1.1 justin TAILQ_HEAD(waithead, waiter);
569 1.1 justin struct waiter {
570 1.1 justin struct thread *who;
571 1.1 justin TAILQ_ENTRY(waiter) entries;
572 1.1 justin int onlist;
573 1.1 justin };
574 1.1 justin
575 1.1 justin static int
576 1.1 justin wait(struct waithead *wh, uint64_t msec)
577 1.1 justin {
578 1.1 justin struct waiter w;
579 1.1 justin
580 1.1 justin w.who = get_current();
581 1.1 justin TAILQ_INSERT_TAIL(wh, &w, entries);
582 1.1 justin w.onlist = 1;
583 1.1 justin block(w.who);
584 1.1 justin if (msec)
585 1.1 justin w.who->wakeup_time = now() + msec;
586 1.1 justin schedule();
587 1.1 justin
588 1.1 justin /* woken up by timeout? */
589 1.1 justin if (w.onlist)
590 1.1 justin TAILQ_REMOVE(wh, &w, entries);
591 1.1 justin
592 1.1 justin return w.onlist ? ETIMEDOUT : 0;
593 1.1 justin }
594 1.1 justin
595 1.1 justin static void
596 1.1 justin wakeup_one(struct waithead *wh)
597 1.1 justin {
598 1.1 justin struct waiter *w;
599 1.1 justin
600 1.1 justin if ((w = TAILQ_FIRST(wh)) != NULL) {
601 1.1 justin TAILQ_REMOVE(wh, w, entries);
602 1.1 justin w->onlist = 0;
603 1.1 justin wake(w->who);
604 1.1 justin }
605 1.1 justin }
606 1.1 justin
607 1.1 justin static void
608 1.1 justin wakeup_all(struct waithead *wh)
609 1.1 justin {
610 1.1 justin struct waiter *w;
611 1.1 justin
612 1.1 justin while ((w = TAILQ_FIRST(wh)) != NULL) {
613 1.1 justin TAILQ_REMOVE(wh, w, entries);
614 1.1 justin w->onlist = 0;
615 1.1 justin wake(w->who);
616 1.1 justin }
617 1.1 justin }
618 1.1 justin
619 1.1 justin int
620 1.1 justin rumpuser_thread_create(void *(*f)(void *), void *arg, const char *thrname,
621 1.1 justin int joinable, int pri, int cpuidx, void **tptr)
622 1.1 justin {
623 1.1 justin struct thread *thr;
624 1.1 justin
625 1.1 justin thr = create_thread(thrname, NULL, (void (*)(void *))f, arg, NULL, 0);
626 1.1 justin /*
627 1.1 justin * XXX: should be supplied as a flag to create_thread() so as to
628 1.1 justin * _ensure_ it's set before the thread runs (and could exit).
629 1.1 justin * now we're trusting unclear semantics of create_thread()
630 1.1 justin */
631 1.1 justin if (thr && joinable)
632 1.1 justin thr->flags |= THREAD_MUSTJOIN;
633 1.1 justin
634 1.1 justin if (!thr)
635 1.1 justin return EINVAL;
636 1.1 justin
637 1.1 justin *tptr = thr;
638 1.1 justin return 0;
639 1.1 justin }
640 1.1 justin
641 1.1 justin void
642 1.1 justin rumpuser_thread_exit(void)
643 1.1 justin {
644 1.1 justin
645 1.1 justin exit_thread();
646 1.1 justin }
647 1.1 justin
648 1.1 justin int
649 1.1 justin rumpuser_thread_join(void *p)
650 1.1 justin {
651 1.1 justin
652 1.1 justin join_thread(p);
653 1.1 justin return 0;
654 1.1 justin }
655 1.1 justin
656 1.1 justin struct rumpuser_mtx {
657 1.1 justin struct waithead waiters;
658 1.1 justin int v;
659 1.1 justin int flags;
660 1.1 justin struct lwp *o;
661 1.1 justin };
662 1.1 justin
663 1.1 justin void
664 1.1 justin rumpuser_mutex_init(struct rumpuser_mtx **mtxp, int flags)
665 1.1 justin {
666 1.1 justin struct rumpuser_mtx *mtx;
667 1.1 justin
668 1.1 justin mtx = malloc(sizeof(*mtx));
669 1.1 justin memset(mtx, 0, sizeof(*mtx));
670 1.1 justin mtx->flags = flags;
671 1.1 justin TAILQ_INIT(&mtx->waiters);
672 1.1 justin *mtxp = mtx;
673 1.1 justin }
674 1.1 justin
675 1.1 justin void
676 1.1 justin rumpuser_mutex_enter(struct rumpuser_mtx *mtx)
677 1.1 justin {
678 1.1 justin int nlocks;
679 1.1 justin
680 1.1 justin if (rumpuser_mutex_tryenter(mtx) != 0) {
681 1.1 justin rumpkern_unsched(&nlocks, NULL);
682 1.1 justin while (rumpuser_mutex_tryenter(mtx) != 0)
683 1.1 justin wait(&mtx->waiters, 0);
684 1.1 justin rumpkern_sched(nlocks, NULL);
685 1.1 justin }
686 1.1 justin }
687 1.1 justin
688 1.1 justin void
689 1.1 justin rumpuser_mutex_enter_nowrap(struct rumpuser_mtx *mtx)
690 1.1 justin {
691 1.1 justin int rv;
692 1.1 justin
693 1.1 justin rv = rumpuser_mutex_tryenter(mtx);
694 1.1 justin /* one VCPU supported, no preemption => must succeed */
695 1.1 justin if (rv != 0) {
696 1.1 justin printk("no voi ei\n");
697 1.1 justin }
698 1.1 justin }
699 1.1 justin
700 1.1 justin int
701 1.1 justin rumpuser_mutex_tryenter(struct rumpuser_mtx *mtx)
702 1.1 justin {
703 1.1 justin struct lwp *l = get_current()->lwp;
704 1.1 justin
705 1.1 justin if (mtx->v && mtx->o != l)
706 1.1 justin return EBUSY;
707 1.1 justin
708 1.1 justin mtx->v++;
709 1.1 justin mtx->o = l;
710 1.1 justin
711 1.1 justin return 0;
712 1.1 justin }
713 1.1 justin
714 1.1 justin void
715 1.1 justin rumpuser_mutex_exit(struct rumpuser_mtx *mtx)
716 1.1 justin {
717 1.1 justin
718 1.1 justin assert(mtx->v > 0);
719 1.1 justin if (--mtx->v == 0) {
720 1.1 justin mtx->o = NULL;
721 1.1 justin wakeup_one(&mtx->waiters);
722 1.1 justin }
723 1.1 justin }
724 1.1 justin
725 1.1 justin void
726 1.1 justin rumpuser_mutex_destroy(struct rumpuser_mtx *mtx)
727 1.1 justin {
728 1.1 justin
729 1.1 justin assert(TAILQ_EMPTY(&mtx->waiters) && mtx->o == NULL);
730 1.1 justin free(mtx);
731 1.1 justin }
732 1.1 justin
733 1.1 justin void
734 1.1 justin rumpuser_mutex_owner(struct rumpuser_mtx *mtx, struct lwp **lp)
735 1.1 justin {
736 1.1 justin
737 1.1 justin *lp = mtx->o;
738 1.1 justin }
739 1.1 justin
740 1.1 justin struct rumpuser_rw {
741 1.1 justin struct waithead rwait;
742 1.1 justin struct waithead wwait;
743 1.1 justin int v;
744 1.1 justin struct lwp *o;
745 1.1 justin };
746 1.1 justin
747 1.1 justin void
748 1.1 justin rumpuser_rw_init(struct rumpuser_rw **rwp)
749 1.1 justin {
750 1.1 justin struct rumpuser_rw *rw;
751 1.1 justin
752 1.1 justin rw = malloc(sizeof(*rw));
753 1.1 justin memset(rw, 0, sizeof(*rw));
754 1.1 justin TAILQ_INIT(&rw->rwait);
755 1.1 justin TAILQ_INIT(&rw->wwait);
756 1.1 justin
757 1.1 justin *rwp = rw;
758 1.1 justin }
759 1.1 justin
760 1.1 justin void
761 1.1 justin rumpuser_rw_enter(int enum_rumprwlock, struct rumpuser_rw *rw)
762 1.1 justin {
763 1.1 justin enum rumprwlock lk = enum_rumprwlock;
764 1.1 justin struct waithead *w = NULL;
765 1.1 justin int nlocks;
766 1.1 justin
767 1.1 justin switch (lk) {
768 1.1 justin case RUMPUSER_RW_WRITER:
769 1.1 justin w = &rw->wwait;
770 1.1 justin break;
771 1.1 justin case RUMPUSER_RW_READER:
772 1.1 justin w = &rw->rwait;
773 1.1 justin break;
774 1.1 justin }
775 1.1 justin
776 1.1 justin if (rumpuser_rw_tryenter(enum_rumprwlock, rw) != 0) {
777 1.1 justin rumpkern_unsched(&nlocks, NULL);
778 1.1 justin while (rumpuser_rw_tryenter(enum_rumprwlock, rw) != 0)
779 1.1 justin wait(w, 0);
780 1.1 justin rumpkern_sched(nlocks, NULL);
781 1.1 justin }
782 1.1 justin }
783 1.1 justin
784 1.1 justin int
785 1.1 justin rumpuser_rw_tryenter(int enum_rumprwlock, struct rumpuser_rw *rw)
786 1.1 justin {
787 1.1 justin enum rumprwlock lk = enum_rumprwlock;
788 1.1 justin int rv;
789 1.1 justin
790 1.1 justin switch (lk) {
791 1.1 justin case RUMPUSER_RW_WRITER:
792 1.1 justin if (rw->o == NULL) {
793 1.1 justin rw->o = rumpuser_curlwp();
794 1.1 justin rv = 0;
795 1.1 justin } else {
796 1.1 justin rv = EBUSY;
797 1.1 justin }
798 1.1 justin break;
799 1.1 justin case RUMPUSER_RW_READER:
800 1.1 justin if (rw->o == NULL && TAILQ_EMPTY(&rw->wwait)) {
801 1.1 justin rw->v++;
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 default:
808 1.1 justin rv = EINVAL;
809 1.1 justin }
810 1.1 justin
811 1.1 justin return rv;
812 1.1 justin }
813 1.1 justin
814 1.1 justin void
815 1.1 justin rumpuser_rw_exit(struct rumpuser_rw *rw)
816 1.1 justin {
817 1.1 justin
818 1.1 justin if (rw->o) {
819 1.1 justin rw->o = NULL;
820 1.1 justin } else {
821 1.1 justin rw->v--;
822 1.1 justin }
823 1.1 justin
824 1.1 justin /* standard procedure, don't let readers starve out writers */
825 1.1 justin if (!TAILQ_EMPTY(&rw->wwait)) {
826 1.1 justin if (rw->o == NULL)
827 1.1 justin wakeup_one(&rw->wwait);
828 1.1 justin } else if (!TAILQ_EMPTY(&rw->rwait) && rw->o == NULL) {
829 1.1 justin wakeup_all(&rw->rwait);
830 1.1 justin }
831 1.1 justin }
832 1.1 justin
833 1.1 justin void
834 1.1 justin rumpuser_rw_destroy(struct rumpuser_rw *rw)
835 1.1 justin {
836 1.1 justin
837 1.1 justin free(rw);
838 1.1 justin }
839 1.1 justin
840 1.1 justin void
841 1.1 justin rumpuser_rw_held(int enum_rumprwlock, struct rumpuser_rw *rw, int *rvp)
842 1.1 justin {
843 1.1 justin enum rumprwlock lk = enum_rumprwlock;
844 1.1 justin
845 1.1 justin switch (lk) {
846 1.1 justin case RUMPUSER_RW_WRITER:
847 1.1 justin *rvp = rw->o == rumpuser_curlwp();
848 1.1 justin break;
849 1.1 justin case RUMPUSER_RW_READER:
850 1.1 justin *rvp = rw->v > 0;
851 1.1 justin break;
852 1.1 justin }
853 1.1 justin }
854 1.1 justin
855 1.1 justin void
856 1.1 justin rumpuser_rw_downgrade(struct rumpuser_rw *rw)
857 1.1 justin {
858 1.1 justin
859 1.1 justin assert(rw->o == rumpuser_curlwp());
860 1.1 justin rw->v = -1;
861 1.1 justin }
862 1.1 justin
863 1.1 justin int
864 1.1 justin rumpuser_rw_tryupgrade(struct rumpuser_rw *rw)
865 1.1 justin {
866 1.1 justin
867 1.1 justin if (rw->v == -1) {
868 1.1 justin rw->v = 1;
869 1.1 justin rw->o = rumpuser_curlwp();
870 1.1 justin return 0;
871 1.1 justin }
872 1.1 justin
873 1.1 justin return EBUSY;
874 1.1 justin }
875 1.1 justin
876 1.1 justin struct rumpuser_cv {
877 1.1 justin struct waithead waiters;
878 1.1 justin int nwaiters;
879 1.1 justin };
880 1.1 justin
881 1.1 justin void
882 1.1 justin rumpuser_cv_init(struct rumpuser_cv **cvp)
883 1.1 justin {
884 1.1 justin struct rumpuser_cv *cv;
885 1.1 justin
886 1.1 justin cv = malloc(sizeof(*cv));
887 1.1 justin memset(cv, 0, sizeof(*cv));
888 1.1 justin TAILQ_INIT(&cv->waiters);
889 1.1 justin *cvp = cv;
890 1.1 justin }
891 1.1 justin
892 1.1 justin void
893 1.1 justin rumpuser_cv_destroy(struct rumpuser_cv *cv)
894 1.1 justin {
895 1.1 justin
896 1.1 justin assert(cv->nwaiters == 0);
897 1.1 justin free(cv);
898 1.1 justin }
899 1.1 justin
900 1.1 justin static void
901 1.1 justin cv_unsched(struct rumpuser_mtx *mtx, int *nlocks)
902 1.1 justin {
903 1.1 justin
904 1.1 justin rumpkern_unsched(nlocks, mtx);
905 1.1 justin rumpuser_mutex_exit(mtx);
906 1.1 justin }
907 1.1 justin
908 1.1 justin static void
909 1.1 justin cv_resched(struct rumpuser_mtx *mtx, int nlocks)
910 1.1 justin {
911 1.1 justin
912 1.1 justin /* see rumpuser(3) */
913 1.1 justin if ((mtx->flags & (RUMPUSER_MTX_KMUTEX | RUMPUSER_MTX_SPIN)) ==
914 1.1 justin (RUMPUSER_MTX_KMUTEX | RUMPUSER_MTX_SPIN)) {
915 1.1 justin rumpkern_sched(nlocks, mtx);
916 1.1 justin rumpuser_mutex_enter_nowrap(mtx);
917 1.1 justin } else {
918 1.1 justin rumpuser_mutex_enter_nowrap(mtx);
919 1.1 justin rumpkern_sched(nlocks, mtx);
920 1.1 justin }
921 1.1 justin }
922 1.1 justin
923 1.1 justin void
924 1.1 justin rumpuser_cv_wait(struct rumpuser_cv *cv, struct rumpuser_mtx *mtx)
925 1.1 justin {
926 1.1 justin int nlocks;
927 1.1 justin
928 1.1 justin cv->nwaiters++;
929 1.1 justin cv_unsched(mtx, &nlocks);
930 1.1 justin wait(&cv->waiters, 0);
931 1.1 justin cv_resched(mtx, nlocks);
932 1.1 justin cv->nwaiters--;
933 1.1 justin }
934 1.1 justin
935 1.1 justin void
936 1.1 justin rumpuser_cv_wait_nowrap(struct rumpuser_cv *cv, struct rumpuser_mtx *mtx)
937 1.1 justin {
938 1.1 justin
939 1.1 justin cv->nwaiters++;
940 1.1 justin rumpuser_mutex_exit(mtx);
941 1.1 justin wait(&cv->waiters, 0);
942 1.1 justin rumpuser_mutex_enter_nowrap(mtx);
943 1.1 justin cv->nwaiters--;
944 1.1 justin }
945 1.1 justin
946 1.1 justin int
947 1.1 justin rumpuser_cv_timedwait(struct rumpuser_cv *cv, struct rumpuser_mtx *mtx,
948 1.1 justin int64_t sec, int64_t nsec)
949 1.1 justin {
950 1.1 justin int nlocks;
951 1.1 justin int rv;
952 1.1 justin
953 1.1 justin cv->nwaiters++;
954 1.1 justin cv_unsched(mtx, &nlocks);
955 1.1 justin rv = wait(&cv->waiters, sec * 1000 + nsec / (1000*1000));
956 1.1 justin cv_resched(mtx, nlocks);
957 1.1 justin cv->nwaiters--;
958 1.1 justin
959 1.1 justin return rv;
960 1.1 justin }
961 1.1 justin
962 1.1 justin void
963 1.1 justin rumpuser_cv_signal(struct rumpuser_cv *cv)
964 1.1 justin {
965 1.1 justin
966 1.1 justin wakeup_one(&cv->waiters);
967 1.1 justin }
968 1.1 justin
969 1.1 justin void
970 1.1 justin rumpuser_cv_broadcast(struct rumpuser_cv *cv)
971 1.1 justin {
972 1.1 justin
973 1.1 justin wakeup_all(&cv->waiters);
974 1.1 justin }
975 1.1 justin
976 1.1 justin void
977 1.1 justin rumpuser_cv_has_waiters(struct rumpuser_cv *cv, int *rvp)
978 1.1 justin {
979 1.1 justin
980 1.1 justin *rvp = cv->nwaiters != 0;
981 1.1 justin }
982 1.1 justin
983 1.1 justin /*
984 1.1 justin * curlwp
985 1.1 justin */
986 1.1 justin
987 1.1 justin void
988 1.1 justin rumpuser_curlwpop(int enum_rumplwpop, struct lwp *l)
989 1.1 justin {
990 1.1 justin struct thread *thread;
991 1.1 justin enum rumplwpop op = enum_rumplwpop;
992 1.1 justin
993 1.1 justin switch (op) {
994 1.1 justin case RUMPUSER_LWP_CREATE:
995 1.1 justin case RUMPUSER_LWP_DESTROY:
996 1.1 justin break;
997 1.1 justin case RUMPUSER_LWP_SET:
998 1.1 justin thread = get_current();
999 1.1 justin thread->lwp = l;
1000 1.1 justin break;
1001 1.1 justin case RUMPUSER_LWP_CLEAR:
1002 1.1 justin thread = get_current();
1003 1.1 justin assert(thread->lwp == l);
1004 1.1 justin thread->lwp = NULL;
1005 1.1 justin break;
1006 1.1 justin }
1007 1.1 justin }
1008 1.1 justin
1009 1.1 justin struct lwp *
1010 1.1 justin rumpuser_curlwp(void)
1011 1.1 justin {
1012 1.1 justin
1013 1.1 justin return get_current()->lwp;
1014 1.1 justin }
1015