Blame view

RIOT/core/thread_flags.c 3.63 KB
a752c7ab   elopes   add first test an...
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
  /*
   * Copyright (C) 2016 Kaspar Schleiser <kaspar@schleiser.de>
   *
   * This file is subject to the terms and conditions of the GNU Lesser
   * General Public License v2.1. See the file LICENSE in the top level
   * directory for more details.
   */
  
  /**
   * @ingroup     core_thread
   * @{
   *
   * @file
   * @brief       thread flags implementation
   *
   * @author      Kaspar Schleiser <kaspar@schleiser.de>
   *
   * @}
   */
  
  
  #include "thread_flags.h"
  #include "irq.h"
  #include "thread.h"
  
  #define ENABLE_DEBUG (0)
  #include "debug.h"
  
  static thread_flags_t _thread_flags_clear_atomic(thread_t *thread, thread_flags_t mask)
  {
      unsigned state = irq_disable();
      mask &= thread->flags;
      thread->flags &= ~mask;
      irq_restore(state);
      return mask;
  }
  
  static void _thread_flags_wait(thread_flags_t mask, thread_t *thread, unsigned threadstate, unsigned irqstate)
  {
      DEBUG("_thread_flags_wait: me->flags=0x%08x me->mask=0x%08x. going blocked.\n",
              (unsigned)thread->flags, (unsigned)mask);
  
      thread->wait_data = (void *)(unsigned)mask;
      sched_set_status(thread, threadstate);
      irq_restore(irqstate);
      thread_yield_higher();
  }
  
  thread_flags_t thread_flags_clear(thread_flags_t mask)
  {
      thread_t *me = (thread_t*) sched_active_thread;
      mask = _thread_flags_clear_atomic(me, mask);
      DEBUG("thread_flags_clear(): pid %"PRIkernel_pid" clearing 0x%08x\n", thread_getpid(), mask);
      return mask;
  }
  
  static void _thread_flags_wait_any(thread_flags_t mask)
  {
      thread_t *me = (thread_t*) sched_active_thread;
      unsigned state = irq_disable();
      if (!(me->flags & mask)) {
          _thread_flags_wait(mask, me, STATUS_FLAG_BLOCKED_ANY, state);
      }
      else {
          irq_restore(state);
      }
  }
  
  thread_flags_t thread_flags_wait_any(thread_flags_t mask)
  {
      thread_t *me = (thread_t*) sched_active_thread;
      _thread_flags_wait_any(mask);
      return _thread_flags_clear_atomic(me, mask);
  }
  
  thread_flags_t thread_flags_wait_one(thread_flags_t mask)
  {
      _thread_flags_wait_any(mask);
      thread_t *me = (thread_t*) sched_active_thread;
      thread_flags_t tmp = me->flags & mask;
      /* clear all but least significant bit */
      tmp &= (~tmp + 1);
      return _thread_flags_clear_atomic(me, tmp);
  }
  
  thread_flags_t thread_flags_wait_all(thread_flags_t mask)
  {
      unsigned state = irq_disable();
      thread_t *me = (thread_t*) sched_active_thread;
      if (!((me->flags & mask) == mask)) {
          DEBUG("thread_flags_wait_all(): pid %"PRIkernel_pid" waiting for %08x\n", thread_getpid(), (unsigned)mask);
          _thread_flags_wait(mask, me, STATUS_FLAG_BLOCKED_ALL, state);
      }
      else {
          irq_restore(state);
      }
  
      return _thread_flags_clear_atomic(me, mask);
  }
  
  inline int __attribute__((always_inline)) thread_flags_wake(thread_t *thread)
  {
      unsigned wakeup = 0;
      thread_flags_t mask = (uint16_t)(unsigned)thread->wait_data;
      switch(thread->status) {
          case STATUS_FLAG_BLOCKED_ANY:
              wakeup = (thread->flags & mask);
              break;
          case STATUS_FLAG_BLOCKED_ALL:
              wakeup = ((thread->flags & mask) == mask);
              break;
      }
  
      if (wakeup) {
          DEBUG("_thread_flags_wake(): waking up pid %"PRIkernel_pid"\n", thread->pid);
          sched_set_status(thread, STATUS_PENDING);
          sched_context_switch_request = 1;
      }
  
      return wakeup;
  }
  
  void thread_flags_set(thread_t *thread, thread_flags_t mask)
  {
      DEBUG("thread_flags_set(): setting 0x%08x for pid %"PRIkernel_pid"\n", mask, thread->pid);
      unsigned state = irq_disable();
      thread->flags |= mask;
      if (thread_flags_wake(thread)) {
          irq_restore(state);
          thread_yield_higher();
      }
      else {
          irq_restore(state);
      }
  }