evtloop_device.c 2.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157
  1. /*
  2. * Copyright (C) 2019-2020 Alibaba Group Holding Limited
  3. */
  4. #include <stdlib.h>
  5. #include <string.h>
  6. #include <aos/kernel.h>
  7. #include <aos/list.h>
  8. #include "evtloop_main.h"
  9. #ifndef DEBUG_MALLOC
  10. #define DEBUG_MALLOC malloc
  11. #endif
  12. #ifndef DEBUG_FREE
  13. #define DEBUG_FREE free
  14. #endif
  15. typedef struct {
  16. aos_mutex_t mutex;
  17. void *sem;
  18. int counter;
  19. dlist_t bufs;
  20. int cache_count;
  21. dlist_t buf_cache;
  22. } event_dev_t;
  23. typedef struct {
  24. dlist_t node;
  25. int len;
  26. char buf[];
  27. } dev_event_t;
  28. static event_dev_t dev_pri;
  29. int eventloop_local_event_init(void)
  30. {
  31. memset(&dev_pri, 0, sizeof(dev_pri));
  32. event_dev_t *pdev = &dev_pri;
  33. aos_mutex_new(&pdev->mutex);
  34. dlist_init(&pdev->bufs);
  35. dlist_init(&pdev->buf_cache);
  36. return 0;
  37. }
  38. int eventloop_local_event_deinit(void)
  39. {
  40. int i = 0;
  41. event_dev_t *pdev = &dev_pri;
  42. dlist_t *node = NULL;
  43. /* clear bufs */
  44. for (i = 0; i < pdev->counter; i++) {
  45. node = (dlist_t *)pdev->bufs.next;
  46. dlist_del(node);
  47. DEBUG_FREE(node);
  48. }
  49. /* clear buf cache */
  50. for (i = 0; i < pdev->cache_count; i++) {
  51. node = (dlist_t *)pdev->buf_cache.next;
  52. dlist_del(node);
  53. DEBUG_FREE(node);
  54. }
  55. aos_mutex_free(&pdev->mutex);
  56. return 0;
  57. }
  58. int event_write(const void *buf, size_t len)
  59. {
  60. event_dev_t *pdev = &dev_pri;
  61. aos_mutex_lock(&pdev->mutex, AOS_WAIT_FOREVER);
  62. dev_event_t *evt;
  63. evt = (dev_event_t *)pdev->buf_cache.next;
  64. if (pdev->cache_count > 0 && evt->len == len) {
  65. dlist_del(&evt->node);
  66. pdev->cache_count --;
  67. } else {
  68. evt = DEBUG_MALLOC(sizeof(*evt) + len);
  69. }
  70. if (evt == NULL) {
  71. len = -1;
  72. goto out;
  73. }
  74. pdev->counter ++;
  75. evt->len = len;
  76. memcpy(evt->buf, buf, len);
  77. dlist_add_tail(&evt->node, &pdev->bufs);
  78. if (pdev->sem != NULL) {
  79. aos_sem_signal((aos_sem_t *)pdev->sem);
  80. }
  81. out:
  82. aos_mutex_unlock(&pdev->mutex);
  83. return len;
  84. }
  85. int event_read(void *buf, size_t len)
  86. {
  87. event_dev_t *pdev = &dev_pri;
  88. int cnt = pdev->counter;
  89. if (!cnt) {
  90. return 0;
  91. }
  92. aos_mutex_lock(&pdev->mutex, AOS_WAIT_FOREVER);
  93. dev_event_t *evt = (dev_event_t *)pdev->bufs.next;
  94. dlist_del(&evt->node);
  95. cnt = (len > evt->len) ? evt->len : len;
  96. memcpy(buf, evt->buf, cnt);
  97. if (pdev->cache_count < 4) {
  98. dlist_add(&evt->node, &pdev->buf_cache);
  99. pdev->cache_count ++;
  100. } else {
  101. DEBUG_FREE(evt);
  102. }
  103. pdev->counter --;
  104. aos_mutex_unlock(&pdev->mutex);
  105. return cnt;
  106. }
  107. int event_poll(void *reader, bool setup, void *sem)
  108. {
  109. event_dev_t *pdev = &dev_pri;
  110. if (!setup) {
  111. pdev->sem = NULL;
  112. return 0;
  113. }
  114. pdev->sem = sem;
  115. if (pdev->counter) {
  116. ((eventloop_sock_t *)reader)->event = 1; /* set event flag*/
  117. aos_sem_signal((aos_sem_t *)sem);
  118. }
  119. return 0;
  120. }