source: mainline/kernel/generic/src/proc/thread.c

Last change on this file was 597fa24, checked in by Jiří Zárevúcky <zarevucky.jiri@…>, 3 months ago

Enable static initialization of kernel synchronization primitives

  • Property mode set to 100644
File size: 26.5 KB
Line 
1/*
2 * Copyright (c) 2025 Jiri Svoboda
3 * Copyright (c) 2010 Jakub Jermar
4 * All rights reserved.
5 *
6 * Redistribution and use in source and binary forms, with or without
7 * modification, are permitted provided that the following conditions
8 * are met:
9 *
10 * - Redistributions of source code must retain the above copyright
11 * notice, this list of conditions and the following disclaimer.
12 * - 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 * - The name of the author may not be used to endorse or promote products
16 * derived from this software without specific prior written permission.
17 *
18 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
19 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
20 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
21 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
22 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
23 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
24 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
25 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
26 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
27 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
28 */
29
30/** @addtogroup kernel_generic_proc
31 * @{
32 */
33
34/**
35 * @file
36 * @brief Thread management functions.
37 */
38
39#include <assert.h>
40#include <proc/scheduler.h>
41#include <proc/thread.h>
42#include <proc/task.h>
43#include <mm/frame.h>
44#include <mm/page.h>
45#include <arch/asm.h>
46#include <arch/cycle.h>
47#include <arch.h>
48#include <synch/spinlock.h>
49#include <synch/waitq.h>
50#include <synch/syswaitq.h>
51#include <cpu.h>
52#include <str.h>
53#include <context.h>
54#include <adt/list.h>
55#include <adt/odict.h>
56#include <time/clock.h>
57#include <time/timeout.h>
58#include <time/delay.h>
59#include <config.h>
60#include <arch/interrupt.h>
61#include <smp/ipi.h>
62#include <atomic.h>
63#include <memw.h>
64#include <stdio.h>
65#include <stdlib.h>
66#include <main/uinit.h>
67#include <syscall/copy.h>
68#include <errno.h>
69#include <debug.h>
70#include <halt.h>
71
72/** Thread states */
73const char *thread_states[] = {
74 "Invalid",
75 "Running",
76 "Sleeping",
77 "Ready",
78 "Entering",
79 "Exiting",
80 "Lingering"
81};
82
83/** Lock protecting the @c threads ordered dictionary .
84 *
85 * For locking rules, see declaration thereof.
86 */
87IRQ_SPINLOCK_INITIALIZE(threads_lock);
88
89/** Ordered dictionary of all threads by their address (i.e. pointer to
90 * the thread_t structure).
91 *
92 * When a thread is found in the @c threads ordered dictionary, it is
93 * guaranteed to exist as long as the @c threads_lock is held.
94 *
95 * Members are of type thread_t.
96 *
97 * This structure contains weak references. Any reference from it must not leave
98 * threads_lock critical section unless strengthened via thread_try_ref().
99 */
100odict_t threads;
101
102IRQ_SPINLOCK_STATIC_INITIALIZE(tidlock);
103static thread_id_t last_tid = 0;
104
105static slab_cache_t *thread_cache;
106
107static void *threads_getkey(odlink_t *);
108static int threads_cmp(void *, void *);
109
110/** Initialization and allocation for thread_t structure
111 *
112 */
113static errno_t thr_constructor(void *obj, unsigned int kmflags)
114{
115 thread_t *thread = (thread_t *) obj;
116
117 link_initialize(&thread->rq_link);
118 link_initialize(&thread->wq_link);
119 link_initialize(&thread->th_link);
120
121 /* call the architecture-specific part of the constructor */
122 thr_constructor_arch(thread);
123
124 /*
125 * Allocate the kernel stack from the low-memory to prevent an infinite
126 * nesting of TLB-misses when accessing the stack from the part of the
127 * TLB-miss handler written in C.
128 *
129 * Note that low-memory is safe to be used for the stack as it will be
130 * covered by the kernel identity mapping, which guarantees not to
131 * nest TLB-misses infinitely (either via some hardware mechanism or
132 * by the construction of the assembly-language part of the TLB-miss
133 * handler).
134 *
135 * This restriction can be lifted once each architecture provides
136 * a similar guarantee, for example, by locking the kernel stack
137 * in the TLB whenever it is allocated from the high-memory and the
138 * thread is being scheduled to run.
139 */
140 kmflags |= FRAME_LOWMEM;
141 kmflags &= ~FRAME_HIGHMEM;
142
143 /*
144 * NOTE: All kernel stacks must be aligned to STACK_SIZE,
145 * see CURRENT.
146 */
147
148 uintptr_t stack_phys =
149 frame_alloc(STACK_FRAMES, kmflags, STACK_SIZE - 1);
150 if (!stack_phys)
151 return ENOMEM;
152
153 thread->kstack = (uint8_t *) PA2KA(stack_phys);
154
155#ifdef CONFIG_UDEBUG
156 mutex_initialize(&thread->udebug.lock, MUTEX_PASSIVE);
157#endif
158
159 return EOK;
160}
161
162/** Destruction of thread_t object */
163static size_t thr_destructor(void *obj)
164{
165 thread_t *thread = (thread_t *) obj;
166
167 /* call the architecture-specific part of the destructor */
168 thr_destructor_arch(thread);
169
170 frame_free(KA2PA(thread->kstack), STACK_FRAMES);
171
172 return STACK_FRAMES; /* number of frames freed */
173}
174
175/** Initialize threads
176 *
177 * Initialize kernel threads support.
178 *
179 */
180void thread_init(void)
181{
182 THREAD = NULL;
183
184 atomic_store(&nrdy, 0);
185 thread_cache = slab_cache_create("thread_t", sizeof(thread_t), _Alignof(thread_t),
186 thr_constructor, thr_destructor, 0);
187
188 odict_initialize(&threads, threads_getkey, threads_cmp);
189}
190
191/** Wire thread to the given CPU
192 *
193 * @param cpu CPU to wire the thread to.
194 *
195 */
196void thread_wire(thread_t *thread, cpu_t *cpu)
197{
198 ipl_t ipl = interrupts_disable();
199 atomic_set_unordered(&thread->cpu, cpu);
200 thread->nomigrate++;
201 interrupts_restore(ipl);
202}
203
204/** Start a thread that wasn't started yet since it was created.
205 *
206 * @param thread A reference to the newly created thread.
207 */
208void thread_start(thread_t *thread)
209{
210 assert(atomic_get_unordered(&thread->state) == Entering);
211 thread_requeue_sleeping(thread_ref(thread));
212}
213
214/** Create new thread
215 *
216 * Create a new thread.
217 *
218 * @param func Thread's implementing function.
219 * @param arg Thread's implementing function argument.
220 * @param task Task to which the thread belongs. The caller must
221 * guarantee that the task won't cease to exist during the
222 * call. The task's lock may not be held.
223 * @param flags Thread flags.
224 * @param name Symbolic name (a copy is made).
225 *
226 * @return New thread's structure on success, NULL on failure.
227 *
228 */
229thread_t *thread_create(void (*func)(void *), void *arg, task_t *task,
230 thread_flags_t flags, const char *name)
231{
232 thread_t *thread = (thread_t *) slab_alloc(thread_cache, FRAME_ATOMIC);
233 if (!thread)
234 return NULL;
235
236 refcount_init(&thread->refcount);
237
238 if (thread_create_arch(thread, flags) != EOK) {
239 slab_free(thread_cache, thread);
240 return NULL;
241 }
242
243 /* Not needed, but good for debugging */
244 memsetb(thread->kstack, STACK_SIZE, 0);
245
246 irq_spinlock_lock(&tidlock, true);
247 thread->tid = ++last_tid;
248 irq_spinlock_unlock(&tidlock, true);
249
250 context_create(&thread->saved_context, thread_main_func,
251 thread->kstack, STACK_SIZE);
252
253 current_initialize((current_t *) thread->kstack);
254
255 str_cpy(thread->name, THREAD_NAME_BUFLEN, name);
256
257 thread->thread_code = func;
258 thread->thread_arg = arg;
259 thread->ucycles = ATOMIC_TIME_INITIALIZER();
260 thread->kcycles = ATOMIC_TIME_INITIALIZER();
261 thread->uncounted =
262 ((flags & THREAD_FLAG_UNCOUNTED) == THREAD_FLAG_UNCOUNTED);
263 atomic_init(&thread->priority, 0);
264 atomic_init(&thread->cpu, NULL);
265 thread->stolen = false;
266 thread->uspace =
267 ((flags & THREAD_FLAG_USPACE) == THREAD_FLAG_USPACE);
268
269 thread->nomigrate = 0;
270 atomic_init(&thread->state, Entering);
271
272 atomic_init(&thread->sleep_queue, NULL);
273
274 thread->in_copy_from_uspace = false;
275 thread->in_copy_to_uspace = false;
276
277 thread->interrupted = false;
278 atomic_init(&thread->sleep_state, SLEEP_INITIAL);
279
280 waitq_initialize(&thread->join_wq);
281
282 thread->task = task;
283
284 thread->fpu_context_exists = false;
285
286 odlink_initialize(&thread->lthreads);
287
288#ifdef CONFIG_UDEBUG
289 /* Initialize debugging stuff */
290 atomic_init(&thread->btrace, false);
291 udebug_thread_initialize(&thread->udebug);
292#endif
293
294 if ((flags & THREAD_FLAG_NOATTACH) != THREAD_FLAG_NOATTACH)
295 thread_attach(thread, task);
296
297 return thread;
298}
299
300/** Destroy thread memory structure
301 *
302 * Detach thread from all queues, cpus etc. and destroy it.
303 *
304 * @param obj Thread to be destroyed.
305 *
306 */
307static void thread_destroy(void *obj)
308{
309 thread_t *thread = (thread_t *) obj;
310
311 assert_link_not_used(&thread->rq_link);
312 assert_link_not_used(&thread->wq_link);
313
314 assert(thread->task);
315
316 ipl_t ipl = interrupts_disable();
317
318 /* Remove thread from global list. */
319 irq_spinlock_lock(&threads_lock, false);
320 odict_remove(&thread->lthreads);
321 irq_spinlock_unlock(&threads_lock, false);
322
323 /* Remove thread from task's list and accumulate accounting. */
324 irq_spinlock_lock(&thread->task->lock, false);
325
326 list_remove(&thread->th_link);
327
328 /*
329 * No other CPU has access to this thread anymore, so we don't need
330 * thread->lock for accessing thread's fields after this point.
331 */
332
333 if (!thread->uncounted) {
334 thread->task->ucycles += atomic_time_read(&thread->ucycles);
335 thread->task->kcycles += atomic_time_read(&thread->kcycles);
336 }
337
338 irq_spinlock_unlock(&thread->task->lock, false);
339
340 assert((atomic_get_unordered(&thread->state) == Entering) ||
341 (atomic_get_unordered(&thread->state) == Exiting) ||
342 (atomic_get_unordered(&thread->state) == Lingering));
343
344 /* Clear cpu->fpu_owner if set to this thread. */
345#ifdef CONFIG_FPU_LAZY
346 cpu_t *cpu = atomic_get_unordered(&thread->cpu);
347 if (cpu) {
348 /*
349 * We need to lock for this because the old CPU can concurrently try
350 * to dump this thread's FPU state, in which case we need to wait for
351 * it to finish. An atomic compare-and-swap wouldn't be enough.
352 */
353 irq_spinlock_lock(&cpu->fpu_lock, false);
354
355 if (atomic_get_unordered(&cpu->fpu_owner) == thread)
356 atomic_set_unordered(&cpu->fpu_owner, NULL);
357
358 irq_spinlock_unlock(&cpu->fpu_lock, false);
359 }
360#endif
361
362 interrupts_restore(ipl);
363
364 /*
365 * Drop the reference to the containing task.
366 */
367 task_release(thread->task);
368 thread->task = NULL;
369
370 slab_free(thread_cache, thread);
371}
372
373void thread_put(thread_t *thread)
374{
375 if (refcount_down(&thread->refcount)) {
376 thread_destroy(thread);
377 }
378}
379
380/** Make the thread visible to the system.
381 *
382 * Attach the thread structure to the current task and make it visible in the
383 * threads_tree.
384 *
385 * @param t Thread to be attached to the task.
386 * @param task Task to which the thread is to be attached.
387 *
388 */
389void thread_attach(thread_t *thread, task_t *task)
390{
391 ipl_t ipl = interrupts_disable();
392
393 /*
394 * Attach to the specified task.
395 */
396 irq_spinlock_lock(&task->lock, false);
397
398 /* Hold a reference to the task. */
399 task_hold(task);
400
401 /* Must not count kbox thread into lifecount */
402 if (thread->uspace)
403 atomic_inc(&task->lifecount);
404
405 list_append(&thread->th_link, &task->threads);
406
407 irq_spinlock_unlock(&task->lock, false);
408
409 /*
410 * Register this thread in the system-wide dictionary.
411 */
412 irq_spinlock_lock(&threads_lock, false);
413 odict_insert(&thread->lthreads, &threads, NULL);
414 irq_spinlock_unlock(&threads_lock, false);
415
416 interrupts_restore(ipl);
417}
418
419/** Terminate thread.
420 *
421 * End current thread execution and switch it to the exiting state.
422 * All pending timeouts are executed.
423 *
424 */
425void thread_exit(void)
426{
427 if (THREAD->uspace) {
428#ifdef CONFIG_UDEBUG
429 /* Generate udebug THREAD_E event */
430 udebug_thread_e_event();
431
432 /*
433 * This thread will not execute any code or system calls from
434 * now on.
435 */
436 udebug_stoppable_begin();
437#endif
438 if (atomic_predec(&TASK->lifecount) == 0) {
439 /*
440 * We are the last userspace thread in the task that
441 * still has not exited. With the exception of the
442 * moment the task was created, new userspace threads
443 * can only be created by threads of the same task.
444 * We are safe to perform cleanup.
445 *
446 */
447 ipc_cleanup();
448 sys_waitq_task_cleanup();
449 LOG("Cleanup of task %" PRIu64 " completed.", TASK->taskid);
450 }
451 }
452
453 scheduler_enter(Exiting);
454 unreachable();
455}
456
457/** Interrupts an existing thread so that it may exit as soon as possible.
458 *
459 * Threads that are blocked waiting for a synchronization primitive
460 * are woken up with a return code of EINTR if the
461 * blocking call was interruptable. See waitq_sleep_timeout().
462 *
463 * Interrupted threads automatically exit when returning back to user space.
464 *
465 * @param thread A valid thread object.
466 */
467void thread_interrupt(thread_t *thread)
468{
469 assert(thread != NULL);
470 thread->interrupted = true;
471 thread_wakeup(thread);
472}
473
474/** Prepare for putting the thread to sleep.
475 *
476 * @returns whether the thread is currently terminating. If THREAD_OK
477 * is returned, the thread is guaranteed to be woken up instantly if the thread
478 * is terminated at any time between this function's return and
479 * thread_wait_finish(). If THREAD_TERMINATING is returned, the thread can still
480 * go to sleep, but doing so will delay termination.
481 */
482thread_termination_state_t thread_wait_start(void)
483{
484 assert(THREAD != NULL);
485
486 /*
487 * This is an exchange rather than a store so that we can use the acquire
488 * semantics, which is needed to ensure that code after this operation sees
489 * memory ops made before thread_wakeup() in other thread, if that wakeup
490 * was reset by this operation.
491 *
492 * In particular, we need this to ensure we can't miss the thread being
493 * terminated concurrently with a synchronization primitive preparing to
494 * sleep.
495 */
496 (void) atomic_exchange_explicit(&THREAD->sleep_state, SLEEP_INITIAL,
497 memory_order_acquire);
498
499 return THREAD->interrupted ? THREAD_TERMINATING : THREAD_OK;
500}
501
502static void thread_wait_timeout_callback(void *arg)
503{
504 thread_wakeup(arg);
505}
506
507/**
508 * Suspends this thread's execution until thread_wakeup() is called on it,
509 * or deadline is reached.
510 *
511 * The way this would normally be used is that the current thread call
512 * thread_wait_start(), and if interruption has not been signaled, stores
513 * a reference to itself in a synchronized structure (such as waitq).
514 * After that, it releases any spinlocks it might hold and calls this function.
515 *
516 * The thread doing the wakeup will acquire the thread's reference from said
517 * synchronized structure and calls thread_wakeup() on it.
518 *
519 * Notably, there can be more than one thread performing wakeup.
520 * The number of performed calls to thread_wakeup(), or their relative
521 * ordering with thread_wait_finish(), does not matter. However, calls to
522 * thread_wakeup() are expected to be synchronized with thread_wait_start()
523 * with which they are associated, otherwise wakeups may be missed.
524 * However, the operation of thread_wakeup() is defined at any time,
525 * synchronization notwithstanding (in the sense of C un/defined behavior),
526 * and is in fact used to interrupt waiting threads by external events.
527 * The waiting thread must operate correctly in face of spurious wakeups,
528 * and clean up its reference in the synchronization structure if necessary.
529 *
530 * Returns THREAD_WAIT_TIMEOUT if timeout fired, which is a necessary condition
531 * for it to have been waken up by the timeout, but the caller must assume
532 * that proper wakeups, timeouts and interrupts may occur concurrently, so
533 * the fact timeout has been registered does not necessarily mean the thread
534 * has not been woken up or interrupted.
535 */
536thread_wait_result_t thread_wait_finish(deadline_t deadline)
537{
538 assert(THREAD != NULL);
539
540 timeout_t timeout;
541
542 /* Extra check to avoid going to scheduler if we don't need to. */
543 if (atomic_load_explicit(&THREAD->sleep_state, memory_order_acquire) !=
544 SLEEP_INITIAL)
545 return THREAD_WAIT_SUCCESS;
546
547 if (deadline != DEADLINE_NEVER) {
548 timeout_initialize(&timeout);
549 timeout_register_deadline(&timeout, deadline,
550 thread_wait_timeout_callback, THREAD);
551 }
552
553 scheduler_enter(Sleeping);
554
555 if (deadline != DEADLINE_NEVER && !timeout_unregister(&timeout)) {
556 return THREAD_WAIT_TIMEOUT;
557 } else {
558 return THREAD_WAIT_SUCCESS;
559 }
560}
561
562void thread_wakeup(thread_t *thread)
563{
564 assert(thread != NULL);
565
566 int state = atomic_exchange_explicit(&thread->sleep_state, SLEEP_WOKE,
567 memory_order_acq_rel);
568
569 if (state == SLEEP_ASLEEP) {
570 /*
571 * Only one thread gets to do this.
572 * The reference consumed here is the reference implicitly passed to
573 * the waking thread by the sleeper in thread_wait_finish().
574 */
575 thread_requeue_sleeping(thread);
576 }
577}
578
579/** Prevent the current thread from being migrated to another processor. */
580void thread_migration_disable(void)
581{
582 ipl_t ipl = interrupts_disable();
583
584 assert(THREAD);
585 THREAD->nomigrate++;
586
587 interrupts_restore(ipl);
588}
589
590/** Allow the current thread to be migrated to another processor. */
591void thread_migration_enable(void)
592{
593 ipl_t ipl = interrupts_disable();
594
595 assert(THREAD);
596 assert(THREAD->nomigrate > 0);
597
598 if (THREAD->nomigrate > 0)
599 THREAD->nomigrate--;
600
601 interrupts_restore(ipl);
602}
603
604/** Thread sleep
605 *
606 * Suspend execution of the current thread.
607 *
608 * @param sec Number of seconds to sleep.
609 *
610 */
611void thread_sleep(uint32_t sec)
612{
613 /*
614 * Sleep in 1000 second steps to support
615 * full argument range
616 */
617 while (sec > 0) {
618 uint32_t period = (sec > 1000) ? 1000 : sec;
619
620 thread_usleep(period * 1000000);
621 sec -= period;
622 }
623}
624
625errno_t thread_join(thread_t *thread)
626{
627 return thread_join_timeout(thread, SYNCH_NO_TIMEOUT, SYNCH_FLAGS_NONE);
628}
629
630/** Wait for another thread to exit.
631 * After successful wait, the thread reference is destroyed.
632 *
633 * @param thread Thread to join on exit.
634 * @param usec Timeout in microseconds.
635 * @param flags Mode of operation.
636 *
637 * @return An error code from errno.h or an error code from synch.h.
638 *
639 */
640errno_t thread_join_timeout(thread_t *thread, uint32_t usec, unsigned int flags)
641{
642 assert(thread != NULL);
643
644 if (thread == THREAD)
645 return EINVAL;
646
647 errno_t rc = _waitq_sleep_timeout(&thread->join_wq, usec, flags);
648
649 if (rc == EOK)
650 thread_put(thread);
651
652 return rc;
653}
654
655void thread_detach(thread_t *thread)
656{
657 thread_put(thread);
658}
659
660/** Thread usleep
661 *
662 * Suspend execution of the current thread.
663 *
664 * @param usec Number of microseconds to sleep.
665 *
666 */
667void thread_usleep(uint32_t usec)
668{
669 WAITQ_INITIALIZE(wq);
670 (void) waitq_sleep_timeout(&wq, usec);
671}
672
673/** Allow other threads to run. */
674void thread_yield(void)
675{
676 assert(THREAD != NULL);
677 scheduler_enter(Running);
678}
679
680static void thread_print(thread_t *thread, bool additional)
681{
682 uint64_t ucycles, kcycles;
683 char usuffix, ksuffix;
684 order_suffix(atomic_time_read(&thread->ucycles), &ucycles, &usuffix);
685 order_suffix(atomic_time_read(&thread->kcycles), &kcycles, &ksuffix);
686
687 state_t state = atomic_get_unordered(&thread->state);
688
689 char *name;
690 if (str_cmp(thread->name, "uinit") == 0)
691 name = thread->task->name;
692 else
693 name = thread->name;
694
695 if (additional)
696 printf("%-8" PRIu64 " %p %p %9" PRIu64 "%c %9" PRIu64 "%c ",
697 thread->tid, thread->thread_code, thread->kstack,
698 ucycles, usuffix, kcycles, ksuffix);
699 else
700 printf("%-8" PRIu64 " %-14s %p %-8s %p %-5" PRIu32 "\n",
701 thread->tid, name, thread, thread_states[state],
702 thread->task, thread->task->container);
703
704 if (additional) {
705 cpu_t *cpu = atomic_get_unordered(&thread->cpu);
706 if (cpu)
707 printf("%-5u", cpu->id);
708 else
709 printf("none ");
710
711 if (state == Sleeping) {
712 printf(" %p", thread->sleep_queue);
713 }
714
715 printf("\n");
716 }
717}
718
719/** Print list of threads debug info
720 *
721 * @param additional Print additional information.
722 *
723 */
724void thread_print_list(bool additional)
725{
726 thread_t *thread;
727
728 /* Accessing system-wide threads list through thread_first()/thread_next(). */
729 irq_spinlock_lock(&threads_lock, true);
730
731 if (sizeof(void *) <= 4) {
732 if (additional)
733 printf("[id ] [code ] [stack ] [ucycles ] [kcycles ]"
734 " [cpu] [waitqueue]\n");
735 else
736 printf("[id ] [name ] [address ] [state ] [task ]"
737 " [ctn]\n");
738 } else {
739 if (additional) {
740 printf("[id ] [code ] [stack ] [ucycles ] [kcycles ]"
741 " [cpu] [waitqueue ]\n");
742 } else
743 printf("[id ] [name ] [address ] [state ]"
744 " [task ] [ctn]\n");
745 }
746
747 thread = thread_first();
748 while (thread != NULL) {
749 thread_print(thread, additional);
750 thread = thread_next(thread);
751 }
752
753 irq_spinlock_unlock(&threads_lock, true);
754}
755
756static bool thread_exists(thread_t *thread)
757{
758 odlink_t *odlink = odict_find_eq(&threads, thread, NULL);
759 return odlink != NULL;
760}
761
762/** Check whether the thread exists, and if so, return a reference to it.
763 */
764thread_t *thread_try_get(thread_t *thread)
765{
766 irq_spinlock_lock(&threads_lock, true);
767
768 if (thread_exists(thread)) {
769 /* Try to strengthen the reference. */
770 thread = thread_try_ref(thread);
771 } else {
772 thread = NULL;
773 }
774
775 irq_spinlock_unlock(&threads_lock, true);
776
777 return thread;
778}
779
780/** Update accounting of current thread.
781 *
782 * Note that thread_lock on THREAD must be already held and
783 * interrupts must be already disabled.
784 *
785 * @param user True to update user accounting, false for kernel.
786 *
787 */
788void thread_update_accounting(bool user)
789{
790 assert(interrupts_disabled());
791
792 uint64_t time = get_cycle();
793
794 if (user)
795 atomic_time_increment(&THREAD->ucycles, time - THREAD->last_cycle);
796 else
797 atomic_time_increment(&THREAD->kcycles, time - THREAD->last_cycle);
798
799 THREAD->last_cycle = time;
800}
801
802/** Find thread structure corresponding to thread ID.
803 *
804 * The threads_lock must be already held by the caller of this function and
805 * interrupts must be disabled.
806 *
807 * The returned reference is weak.
808 * If the caller needs to keep it, thread_try_ref() must be used to upgrade
809 * to a strong reference _before_ threads_lock is released.
810 *
811 * @param id Thread ID.
812 *
813 * @return Thread structure address or NULL if there is no such thread ID.
814 *
815 */
816thread_t *thread_find_by_id(thread_id_t thread_id)
817{
818 thread_t *thread;
819
820 assert(interrupts_disabled());
821 assert(irq_spinlock_locked(&threads_lock));
822
823 thread = thread_first();
824 while (thread != NULL) {
825 if (thread->tid == thread_id)
826 return thread;
827
828 thread = thread_next(thread);
829 }
830
831 return NULL;
832}
833
834/** Get count of threads.
835 *
836 * @return Number of threads in the system
837 */
838size_t thread_count(void)
839{
840 assert(interrupts_disabled());
841 assert(irq_spinlock_locked(&threads_lock));
842
843 return odict_count(&threads);
844}
845
846/** Get first thread.
847 *
848 * @return Pointer to first thread or @c NULL if there are none.
849 */
850thread_t *thread_first(void)
851{
852 odlink_t *odlink;
853
854 assert(interrupts_disabled());
855 assert(irq_spinlock_locked(&threads_lock));
856
857 odlink = odict_first(&threads);
858 if (odlink == NULL)
859 return NULL;
860
861 return odict_get_instance(odlink, thread_t, lthreads);
862}
863
864/** Get next thread.
865 *
866 * @param cur Current thread
867 * @return Pointer to next thread or @c NULL if there are no more threads.
868 */
869thread_t *thread_next(thread_t *cur)
870{
871 odlink_t *odlink;
872
873 assert(interrupts_disabled());
874 assert(irq_spinlock_locked(&threads_lock));
875
876 odlink = odict_next(&cur->lthreads, &threads);
877 if (odlink == NULL)
878 return NULL;
879
880 return odict_get_instance(odlink, thread_t, lthreads);
881}
882
883#ifdef CONFIG_UDEBUG
884
885void thread_stack_trace(thread_id_t thread_id)
886{
887 irq_spinlock_lock(&threads_lock, true);
888 thread_t *thread = thread_try_ref(thread_find_by_id(thread_id));
889 irq_spinlock_unlock(&threads_lock, true);
890
891 if (thread == NULL) {
892 printf("No such thread.\n");
893 return;
894 }
895
896 /*
897 * Schedule a stack trace to be printed
898 * just before the thread is scheduled next.
899 *
900 * If the thread is sleeping then try to interrupt
901 * the sleep. Any request for printing an uspace stack
902 * trace from within the kernel should be always
903 * considered a last resort debugging means, therefore
904 * forcing the thread's sleep to be interrupted
905 * is probably justifiable.
906 */
907
908 printf("Scheduling thread stack trace.\n");
909 atomic_set_unordered(&thread->btrace, true);
910
911 thread_wakeup(thread);
912 thread_put(thread);
913}
914
915#endif /* CONFIG_UDEBUG */
916
917/** Get key function for the @c threads ordered dictionary.
918 *
919 * @param odlink Link
920 * @return Pointer to thread structure cast as 'void *'
921 */
922static void *threads_getkey(odlink_t *odlink)
923{
924 thread_t *thread = odict_get_instance(odlink, thread_t, lthreads);
925 return (void *) thread;
926}
927
928/** Key comparison function for the @c threads ordered dictionary.
929 *
930 * @param a Pointer to thread A
931 * @param b Pointer to thread B
932 * @return -1, 0, 1 iff pointer to A is less than, equal to, greater than B
933 */
934static int threads_cmp(void *a, void *b)
935{
936 if (a > b)
937 return -1;
938 else if (a == b)
939 return 0;
940 else
941 return +1;
942}
943
944/** Process syscall to create new thread.
945 * The started thread will have initial pc and sp set to the exact values passed
946 * to the syscall. The kernel will not touch any stack data below the stack
947 * pointer, but some architectures may require some space to be available
948 * for use above it. See userspace() in kernel, and <libarch/thread.h> in libc.
949 *
950 */
951sys_errno_t sys_thread_create(sysarg_t pc, sysarg_t sp,
952 uspace_ptr_char uspace_name, size_t name_len)
953{
954 if (name_len > THREAD_NAME_BUFLEN - 1)
955 name_len = THREAD_NAME_BUFLEN - 1;
956
957 char namebuf[THREAD_NAME_BUFLEN];
958 errno_t rc = copy_from_uspace(namebuf, uspace_name, name_len);
959 if (rc != EOK)
960 return (sys_errno_t) rc;
961
962 namebuf[name_len] = 0;
963
964 /*
965 * In case of failure, kernel_uarg will be deallocated in this function.
966 * In case of success, kernel_uarg will be freed in uinit().
967 */
968 uinit_arg_t *kernel_uarg = malloc(sizeof(uinit_arg_t));
969 if (!kernel_uarg)
970 return (sys_errno_t) ENOMEM;
971
972 kernel_uarg->pc = pc;
973 kernel_uarg->sp = sp;
974
975 // TODO: fix some unnecessary inconsistencies between architectures
976
977 thread_t *thread = thread_create(uinit, kernel_uarg, TASK,
978 THREAD_FLAG_USPACE | THREAD_FLAG_NOATTACH, namebuf);
979 if (!thread) {
980 free(kernel_uarg);
981 return (sys_errno_t) ENOMEM;
982 }
983
984#ifdef CONFIG_UDEBUG
985 /*
986 * Generate udebug THREAD_B event and attach the thread.
987 * This must be done atomically (with the debug locks held),
988 * otherwise we would either miss some thread or receive
989 * THREAD_B events for threads that already existed
990 * and could be detected with THREAD_READ before.
991 */
992 udebug_thread_b_event_attach(thread, TASK);
993#else
994 thread_attach(thread, TASK);
995#endif
996 thread_start(thread);
997 thread_put(thread);
998
999 return (sys_errno_t) EOK;
1000}
1001
1002/** Process syscall to terminate thread.
1003 *
1004 */
1005sys_errno_t sys_thread_exit(int uspace_status)
1006{
1007 thread_exit();
1008}
1009
1010/** Syscall for getting TID.
1011 *
1012 * @param uspace_thread_id Userspace address of 8-byte buffer where to store
1013 * current thread ID.
1014 *
1015 * @return 0 on success or an error code from @ref errno.h.
1016 *
1017 */
1018sys_errno_t sys_thread_get_id(uspace_ptr_thread_id_t uspace_thread_id)
1019{
1020 /*
1021 * No need to acquire lock on THREAD because tid
1022 * remains constant for the lifespan of the thread.
1023 *
1024 */
1025 return (sys_errno_t) copy_to_uspace(uspace_thread_id, &THREAD->tid,
1026 sizeof(THREAD->tid));
1027}
1028
1029/** Syscall wrapper for sleeping. */
1030sys_errno_t sys_thread_usleep(uint32_t usec)
1031{
1032 thread_usleep(usec);
1033 return 0;
1034}
1035
1036sys_errno_t sys_thread_udelay(uint32_t usec)
1037{
1038 delay(usec);
1039 return 0;
1040}
1041
1042/** @}
1043 */
Note: See TracBrowser for help on using the repository browser.