waitqueue.h 2.2 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374
  1. /**
  2. * RT-Thread RuiChing
  3. *
  4. * COPYRIGHT (C) 2024-2025 Shanghai Real-Thread Electronic Technology Co., Ltd.
  5. * All rights reserved.
  6. *
  7. * The license and distribution terms for this file may be
  8. * found in the file LICENSE in this distribution.
  9. */
  10. #ifndef __WAITQUEUE_H__
  11. #define __WAITQUEUE_H__
  12. #include <kpi.h>
  13. #include <rtconfig.h>
  14. #include <rtdef.h>
  15. #define RT_WQ_FLAG_CLEAN 0x00
  16. #define RT_WQ_FLAG_WAKEUP 0x01
  17. struct rt_wqueue_node;
  18. typedef int (*rt_wqueue_func_t)(struct rt_wqueue_node *wait, void *key);
  19. struct rt_wqueue_node
  20. {
  21. rt_thread_t polling_thread;
  22. rt_list_t list;
  23. rt_wqueue_t *wqueue;
  24. rt_wqueue_func_t wakeup;
  25. rt_uint32_t key;
  26. };
  27. typedef struct rt_wqueue_node rt_wqueue_node_t;
  28. typedef int (*__kpi___wqueue_default_wake)(
  29. struct rt_wqueue_node *wait, void *key);
  30. rt_inline void rt_wqueue_init(rt_wqueue_t *queue)
  31. {
  32. RT_ASSERT(queue != RT_NULL);
  33. queue->flag = RT_WQ_FLAG_CLEAN;
  34. rt_list_init(&(queue->waiting_list));
  35. rt_spin_lock_init(&(queue->spinlock));
  36. }
  37. typedef int (*__kpi_rt_wqueue_wait_interruptible)(
  38. rt_wqueue_t *queue, int condition, int timeout);
  39. typedef void (*__kpi_rt_wqueue_add)(
  40. rt_wqueue_t *queue, struct rt_wqueue_node *node);
  41. typedef void (*__kpi_rt_wqueue_remove)(struct rt_wqueue_node *node);
  42. typedef int (*__kpi_rt_wqueue_wait)(
  43. rt_wqueue_t *queue, int condition, int timeout);
  44. typedef int (*__kpi_rt_wqueue_wait_killable)(
  45. rt_wqueue_t *queue, int condition, int timeout);
  46. typedef void (*__kpi_rt_wqueue_wakeup)(rt_wqueue_t *queue, void *key);
  47. typedef void (*__kpi_rt_wqueue_wakeup_all)(rt_wqueue_t *queue, void *key);
  48. #define DEFINE_WAIT_FUNC(name, function) \
  49. struct rt_wqueue_node name = { rt_current_thread, \
  50. RT_LIST_OBJECT_INIT(((name).list)), function, 0 }
  51. #define DEFINE_WAIT(name) DEFINE_WAIT_FUNC(name, __wqueue_default_wake)
  52. KPI_EXTERN(__wqueue_default_wake);
  53. KPI_EXTERN(rt_wqueue_add);
  54. KPI_EXTERN(rt_wqueue_remove);
  55. KPI_EXTERN(rt_wqueue_wait);
  56. KPI_EXTERN(rt_wqueue_wait_killable);
  57. KPI_EXTERN(rt_wqueue_wait_interruptible);
  58. KPI_EXTERN(rt_wqueue_wakeup);
  59. KPI_EXTERN(rt_wqueue_wakeup_all);
  60. #endif /* __WAITQUEUE_H__ */