123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041 |
- /*
- * Copyright (c) 2006-2023, RT-Thread Development Team
- *
- * SPDX-License-Identifier: Apache-2.0
- *
- * Change Logs:
- * Date Author Notes
- * 2023-07-29 zmq810150896 first version
- * 2024-03-26 TroyMitchelle Add comments for all functions, members within structure members and fix incorrect naming of triggered
- * 2023-12-14 Shell When poll goes to sleep before the waitqueue has added a
- * record and finished enumerating all the fd's, it may be
- * incorrectly woken up. This is basically because the poll
- * mechanism wakeup algorithm does not correctly distinguish
- * the current wait state.
- */
- #include <rtthread.h>
- #include <fcntl.h>
- #include <stdint.h>
- #include <unistd.h>
- #include <dfs_file.h>
- #include "sys/epoll.h"
- #include "poll.h"
- #include <lwp_signal.h>
- #define EPOLL_MUTEX_NAME "EVENTEPOLL"
- #define EFD_SHARED_EPOLL_TYPE (EPOLL_CTL_ADD | EPOLL_CTL_DEL | EPOLL_CTL_MOD)
- #define EPOLLINOUT_BITS (EPOLLIN | EPOLLOUT | EPOLLRDNORM | EPOLLWRNORM)
- #define EPOLLEXCLUSIVE_BITS (EPOLLINOUT_BITS | EPOLLERR | EPOLLHUP | \
- EPOLLET | EPOLLEXCLUSIVE)
- struct rt_eventpoll;
- enum rt_epoll_status {
- RT_EPOLL_STAT_INIT,
- RT_EPOLL_STAT_TRIG,
- RT_EPOLL_STAT_WAITING,
- };
- /* Monitor queue */
- struct rt_fd_list
- {
- rt_uint32_t revents; /**< Monitored events */
- struct epoll_event epev; /**< Epoll event structure */
- rt_pollreq_t req; /**< Poll request structure */
- struct rt_eventpoll *ep; /**< Pointer to the associated event poll */
- struct rt_wqueue_node wqn; /**< Wait queue node */
- int exclusive; /**< Indicates if the event is exclusive */
- rt_bool_t is_rdl_node; /**< Indicates if the node is in the ready list */
- int fd; /**< File descriptor */
- struct rt_fd_list *next; /**< Pointer to the next file descriptor list */
- rt_slist_t rdl_node; /**< Ready list node */
- };
- struct rt_eventpoll
- {
- rt_wqueue_t epoll_read; /**< Epoll read queue */
- rt_thread_t polling_thread; /**< Polling thread */
- struct rt_mutex lock; /**< Mutex lock */
- struct rt_fd_list *fdlist; /**< Monitor list */
- int eventpoll_num; /**< Number of ready lists */
- rt_pollreq_t req; /**< Poll request structure */
- struct rt_spinlock spinlock; /**< Spin lock */
- rt_slist_t rdl_head; /**< Ready list head */
- enum rt_epoll_status status; /* the waited thread whether triggered */
- };
- static int epoll_close(struct dfs_file *file);
- static int epoll_poll(struct dfs_file *file, struct rt_pollreq *req);
- static int epoll_get_event(struct rt_fd_list *fl, rt_pollreq_t *req);
- static int epoll_do_ctl(int epfd, int op, int fd, struct epoll_event *event);
- static const struct dfs_file_ops epoll_fops =
- {
- .close = epoll_close,
- .poll = epoll_poll,
- };
- /**
- * @brief Closes the file descriptor list associated with epoll.
- *
- * This function closes the file descriptor list associated with epoll and frees the allocated memory.
- *
- * @param fdlist Pointer to the file descriptor list.
- *
- * @return Returns 0 on success.
- */
- static int epoll_close_fdlist(struct rt_fd_list *fdlist)
- {
- struct rt_fd_list *fre_node, *list;
- if (fdlist != RT_NULL)
- {
- list = fdlist;
- while (list->next != RT_NULL)
- {
- fre_node = list->next;
- rt_wqueue_remove(&fre_node->wqn);
- list->next = fre_node->next;
- rt_free(fre_node);
- }
- rt_free(fdlist);
- }
- return 0;
- }
- /**
- * @brief Closes the epoll file descriptor.
- *
- * This function closes the epoll file descriptor and cleans up associated resources.
- *
- * @param file Pointer to the file structure.
- *
- * @return Returns 0 on success.
- */
- static int epoll_close(struct dfs_file *file)
- {
- struct rt_eventpoll *ep;
- if (file->vnode->ref_count != 1)
- return 0;
- if (file->vnode)
- {
- if (file->vnode->data)
- {
- ep = file->vnode->data;
- if (ep)
- {
- rt_mutex_take(&ep->lock, RT_WAITING_FOREVER);
- if (ep->fdlist)
- {
- epoll_close_fdlist(ep->fdlist);
- }
- rt_mutex_release(&ep->lock);
- rt_mutex_detach(&ep->lock);
- rt_free(ep);
- }
- }
- }
- return 0;
- }
- /**
- * @brief Polls the epoll file descriptor for events.
- *
- * This function polls the epoll file descriptor for events and updates the poll request accordingly.
- *
- * @param file Pointer to the file structure.
- * @param req Pointer to the poll request structure.
- *
- * @return Returns the events occurred on success.
- */
- static int epoll_poll(struct dfs_file *file, struct rt_pollreq *req)
- {
- struct rt_eventpoll *ep;
- int events = 0;
- rt_base_t level;
- if (file->vnode->data)
- {
- ep = file->vnode->data;
- ep->req._key = req->_key;
- rt_mutex_take(&ep->lock, RT_WAITING_FOREVER);
- rt_poll_add(&ep->epoll_read, req);
- level = rt_spin_lock_irqsave(&ep->spinlock);
- if (!rt_slist_isempty(&ep->rdl_head))
- events |= POLLIN | EPOLLRDNORM | POLLOUT;
- rt_spin_unlock_irqrestore(&ep->spinlock, level);
- rt_mutex_release(&ep->lock);
- }
- return events;
- }
- /**
- * @brief Callback function for the wait queue.
- *
- * This function is called when the file descriptor is ready for polling.
- *
- * @param wait Pointer to the wait queue node.
- * @param key Key associated with the wait queue node.
- *
- * @return Returns 0 on success.
- */
- static int epoll_wqueue_callback(struct rt_wqueue_node *wait, void *key)
- {
- struct rt_fd_list *fdlist;
- struct rt_eventpoll *ep;
- rt_base_t level;
- int is_waiting = 0;
- if (key && !((rt_ubase_t)key & wait->key))
- return -1;
- fdlist = rt_container_of(wait, struct rt_fd_list, wqn);
- ep = fdlist->ep;
- if (ep)
- {
- level = rt_spin_lock_irqsave(&ep->spinlock);
- if (fdlist->is_rdl_node == RT_FALSE)
- {
- rt_slist_append(&ep->rdl_head, &fdlist->rdl_node);
- fdlist->exclusive = 0;
- fdlist->is_rdl_node = RT_TRUE;
- ep->eventpoll_num++;
- is_waiting = (ep->status == RT_EPOLL_STAT_WAITING);
- ep->status = RT_EPOLL_STAT_TRIG;
- rt_wqueue_wakeup(&ep->epoll_read, (void *)POLLIN);
- }
- rt_spin_unlock_irqrestore(&ep->spinlock, level);
- }
- if (is_waiting)
- {
- return __wqueue_default_wake(wait, key);
- }
- return -1;
- }
- /**
- * @brief Adds a callback function to the wait queue associated with epoll.
- *
- * This function adds a callback function to the wait queue associated with epoll.
- *
- * @param wq Pointer to the wait queue.
- * @param req Pointer to the poll request structure.
- */
- static void epoll_wqueue_add_callback(rt_wqueue_t *wq, rt_pollreq_t *req)
- {
- struct rt_fd_list *fdlist;
- struct rt_eventpoll *ep;
- fdlist = rt_container_of(req, struct rt_fd_list, req);
- ep = fdlist->ep;
- fdlist->wqn.key = req->_key;
- rt_list_init(&(fdlist->wqn.list));
- fdlist->wqn.polling_thread = ep->polling_thread;
- fdlist->wqn.wakeup = epoll_wqueue_callback;
- rt_wqueue_add(wq, &fdlist->wqn);
- }
- /**
- * @brief Installs a file descriptor list into the epoll control structure.
- *
- * This function installs a file descriptor list into the epoll control structure.
- *
- * @param fdlist Pointer to the file descriptor list.
- * @param ep Pointer to the epoll control structure.
- */
- static void epoll_ctl_install(struct rt_fd_list *fdlist, struct rt_eventpoll *ep)
- {
- rt_uint32_t mask = 0;
- rt_base_t level;
- fdlist->req._key = fdlist->revents;
- mask = epoll_get_event(fdlist, &fdlist->req);
- if (mask & fdlist->revents)
- {
- if (ep)
- {
- rt_mutex_take(&ep->lock, RT_WAITING_FOREVER);
- level = rt_spin_lock_irqsave(&ep->spinlock);
- rt_slist_append(&ep->rdl_head, &fdlist->rdl_node);
- fdlist->exclusive = 0;
- fdlist->is_rdl_node = RT_TRUE;
- ep->status = RT_EPOLL_STAT_TRIG;
- ep->eventpoll_num ++;
- rt_spin_unlock_irqrestore(&ep->spinlock, level);
- rt_mutex_release(&ep->lock);
- }
- }
- }
- /**
- * @brief Initializes the epoll control structure.
- *
- * This function initializes the epoll control structure.
- *
- * @param ep Pointer to the epoll control structure.
- */
- static void epoll_member_init(struct rt_eventpoll *ep)
- {
- ep->status = RT_EPOLL_STAT_INIT;
- ep->eventpoll_num = 0;
- ep->polling_thread = rt_thread_self();
- ep->fdlist = RT_NULL;
- ep->req._key = 0;
- rt_slist_init(&(ep->rdl_head));
- rt_wqueue_init(&ep->epoll_read);
- rt_mutex_init(&ep->lock, EPOLL_MUTEX_NAME, RT_IPC_FLAG_FIFO);
- rt_spin_lock_init(&ep->spinlock);
- }
- /**
- * @brief Initializes the epoll file descriptor.
- *
- * This function initializes the epoll file descriptor.
- *
- * @param fd File descriptor.
- *
- * @return Returns 0 on success.
- */
- static int epoll_epf_init(int fd)
- {
- struct dfs_file *df;
- struct rt_eventpoll *ep;
- rt_err_t ret = 0;
- df = fd_get(fd);
- if (df)
- {
- ep = (struct rt_eventpoll *)rt_malloc(sizeof(struct rt_eventpoll));
- if (ep)
- {
- epoll_member_init(ep);
- #ifdef RT_USING_DFS_V2
- df->fops = &epoll_fops;
- #endif
- df->vnode = (struct dfs_vnode *)rt_malloc(sizeof(struct dfs_vnode));
- if (df->vnode)
- {
- ep->fdlist = (struct rt_fd_list *)rt_malloc(sizeof(struct rt_fd_list));
- if (ep->fdlist)
- {
- ep->fdlist->next = RT_NULL;
- ep->fdlist->fd = fd;
- ep->fdlist->ep = ep;
- ep->fdlist->exclusive = 0;
- ep->fdlist->is_rdl_node = RT_FALSE;
- dfs_vnode_init(df->vnode, FT_REGULAR, &epoll_fops);
- df->vnode->data = ep;
- rt_slist_init(&ep->fdlist->rdl_node);
- }
- else
- {
- ret = -ENOMEM;
- rt_free(df->vnode);
- rt_free(ep);
- }
- }
- else
- {
- ret = -ENOMEM;
- rt_free(ep);
- }
- }
- else
- {
- ret = -ENOMEM;
- }
- }
- return ret;
- }
- /**
- * @brief Creates an epoll file descriptor.
- *
- * This function creates an epoll file descriptor.
- *
- * @param size Size of the epoll instance.
- *
- * @return Returns the file descriptor on success, or -1 on failure.
- */
- static int epoll_do_create(int size)
- {
- rt_err_t ret = -1;
- int status;
- int fd;
- if (size < 0)
- {
- rt_set_errno(EINVAL);
- }
- else
- {
- fd = fd_new();
- if (fd >= 0)
- {
- ret = fd;
- status = epoll_epf_init(fd);
- if (status < 0)
- {
- fd_release(fd);
- rt_set_errno(-status);
- }
- }
- else
- {
- rt_set_errno(-fd);
- }
- }
- return ret;
- }
- /**
- * @brief Adds a file descriptor to the epoll instance.
- *
- * This function adds a file descriptor to the epoll instance.
- *
- * @param df Pointer to the file structure.
- * @param fd File descriptor to add.
- * @param event Pointer to the epoll event structure.
- *
- * @return Returns 0 on success, or an error code on failure.
- */
- static int epoll_ctl_add(struct dfs_file *df, int fd, struct epoll_event *event)
- {
- struct rt_fd_list *fdlist;
- struct rt_eventpoll *ep;
- rt_err_t ret = -EINVAL;
- if (df->vnode->data)
- {
- ep = df->vnode->data;
- fdlist = ep->fdlist;
- ret = 0;
- rt_mutex_take(&ep->lock, RT_WAITING_FOREVER);
- while (fdlist->next != RT_NULL)
- {
- if (fdlist->next->fd == fd)
- {
- rt_mutex_release(&ep->lock);
- return 0;
- }
- fdlist = fdlist->next;
- }
- rt_mutex_release(&ep->lock);
- fdlist = (struct rt_fd_list *)rt_malloc(sizeof(struct rt_fd_list));
- if (fdlist)
- {
- fdlist->fd = fd;
- memcpy(&fdlist->epev.data, &event->data, sizeof(event->data));
- fdlist->epev.events = 0;
- fdlist->ep = ep;
- fdlist->exclusive = 0;
- fdlist->is_rdl_node = RT_FALSE;
- fdlist->req._proc = epoll_wqueue_add_callback;
- fdlist->revents = event->events;
- rt_mutex_take(&ep->lock, RT_WAITING_FOREVER);
- fdlist->next = ep->fdlist->next;
- ep->fdlist->next = fdlist;
- rt_mutex_release(&ep->lock);
- rt_slist_init(&fdlist->rdl_node);
- epoll_ctl_install(fdlist, ep);
- }
- else
- {
- ret = -ENOMEM;
- }
- }
- return ret;
- }
- /**
- * @brief Removes a file descriptor from the epoll instance.
- *
- * This function removes a file descriptor from the epoll instance.
- *
- * @param df Pointer to the file structure.
- * @param fd File descriptor to remove.
- *
- * @return Returns 0 on success, or an error code on failure.
- */
- static int epoll_ctl_del(struct dfs_file *df, int fd)
- {
- struct rt_fd_list *fdlist, *fre_fd, *rdlist;
- struct rt_eventpoll *ep = RT_NULL;
- rt_slist_t *node = RT_NULL;
- rt_err_t ret = -EINVAL;
- rt_base_t level;
- if (df->vnode->data)
- {
- ep = df->vnode->data;
- if (ep)
- {
- rt_mutex_take(&ep->lock, RT_WAITING_FOREVER);
- level = rt_spin_lock_irqsave(&ep->spinlock);
- rt_slist_for_each(node, &ep->rdl_head)
- {
- rdlist = rt_slist_entry(node, struct rt_fd_list, rdl_node);
- if (rdlist->fd == fd)
- rt_slist_remove(&ep->rdl_head, node);
- }
- rt_spin_unlock_irqrestore(&ep->spinlock, level);
- fdlist = ep->fdlist;
- while (fdlist->next != RT_NULL)
- {
- if (fdlist->next->fd == fd)
- {
- fre_fd = fdlist->next;
- fdlist->next = fdlist->next->next;
- if (fre_fd->wqn.wqueue)
- rt_wqueue_remove(&fre_fd->wqn);
- rt_free(fre_fd);
- break;
- }
- else
- {
- fdlist = fdlist->next;
- }
- }
- rt_mutex_release(&ep->lock);
- }
- ret = 0;
- }
- return ret;
- }
- /**
- * @brief Modifies the events associated with a file descriptor in the epoll instance.
- *
- * This function modifies the events associated with a file descriptor in the epoll instance.
- *
- * @param df Pointer to the file structure.
- * @param fd File descriptor to modify.
- * @param event Pointer to the epoll event structure.
- *
- * @return Returns 0 on success, or an error code on failure.
- */
- static int epoll_ctl_mod(struct dfs_file *df, int fd, struct epoll_event *event)
- {
- struct rt_fd_list *fdlist;
- struct rt_eventpoll *ep = RT_NULL;
- rt_err_t ret = -EINVAL;
- if (df->vnode->data)
- {
- ep = df->vnode->data;
- fdlist = ep->fdlist;
- while (fdlist->next != RT_NULL)
- {
- if (fdlist->next->fd == fd)
- {
- rt_mutex_take(&ep->lock, RT_WAITING_FOREVER);
- memcpy(&fdlist->next->epev.data, &event->data, sizeof(event->data));
- fdlist->next->revents = event->events;
- if (fdlist->next->wqn.wqueue)
- rt_wqueue_remove(&fdlist->next->wqn);
- rt_mutex_release(&ep->lock);
- epoll_ctl_install(fdlist->next, ep);
- break;
- }
- fdlist = fdlist->next;
- }
- ret = 0;
- }
- return ret;
- }
- /**
- * @brief Controls an epoll instance.
- *
- * This function controls an epoll instance, performing operations such as adding,
- * modifying, or removing file descriptors associated with the epoll instance.
- *
- * @param epfd File descriptor of the epoll instance.
- * @param op Operation to perform (EPOLL_CTL_ADD, EPOLL_CTL_DEL, or EPOLL_CTL_MOD).
- * @param fd File descriptor to add, modify, or remove.
- * @param event Pointer to the epoll event structure.
- *
- * @return Returns 0 on success, or -1 on failure with errno set appropriately.
- */
- static int epoll_do_ctl(int epfd, int op, int fd, struct epoll_event *event)
- {
- struct dfs_file *epdf;
- struct rt_eventpoll *ep;
- rt_err_t ret = 0;
- if (op & ~EFD_SHARED_EPOLL_TYPE)
- {
- rt_set_errno(EINVAL);
- return -1;
- }
- if ((epfd == fd) || (epfd < 0))
- {
- rt_set_errno(EINVAL);
- return -1;
- }
- if (!(op & EPOLL_CTL_DEL))
- {
- if (!(event->events & EPOLLEXCLUSIVE_BITS))
- {
- rt_set_errno(EINVAL);
- return -1;
- }
- event->events |= EPOLLERR | EPOLLHUP;
- }
- if (!fd_get(fd))
- {
- rt_set_errno(EBADF);
- return -1;
- }
- epdf = fd_get(epfd);
- if (epdf->vnode->data)
- {
- ep = epdf->vnode->data;
- switch (op)
- {
- case EPOLL_CTL_ADD:
- ret = epoll_ctl_add(epdf, fd, event);
- break;
- case EPOLL_CTL_DEL:
- ret = epoll_ctl_del(epdf, fd);
- break;
- case EPOLL_CTL_MOD:
- ret = epoll_ctl_mod(epdf, fd, event);
- break;
- default:
- rt_set_errno(EINVAL);
- break;
- }
- if (ret < 0)
- {
- rt_set_errno(-ret);
- ret = -1;
- }
- else
- {
- ep->polling_thread = rt_thread_self();
- }
- }
- return ret;
- }
- /**
- * @brief Waits for events on an epoll instance with a specified timeout.
- *
- * This function waits for events on the specified epoll instance within the given timeout.
- *
- * @param ep Pointer to the epoll instance.
- * @param msec Timeout in milliseconds.
- *
- * @return Returns 0 if no events occurred within the timeout, or 1 if events were triggered.
- */
- static int epoll_wait_timeout(struct rt_eventpoll *ep, int msec)
- {
- rt_int32_t timeout;
- struct rt_thread *thread;
- rt_base_t level;
- int ret = 0;
- thread = ep->polling_thread;
- timeout = rt_tick_from_millisecond(msec);
- level = rt_spin_lock_irqsave(&ep->spinlock);
- if (timeout != 0 && ep->status != RT_EPOLL_STAT_TRIG)
- {
- if (rt_thread_suspend_with_flag(thread, RT_KILLABLE) == RT_EOK)
- {
- if (timeout > 0)
- {
- rt_timer_control(&(thread->thread_timer),
- RT_TIMER_CTRL_SET_TIME,
- &timeout);
- rt_timer_start(&(thread->thread_timer));
- }
- ep->status = RT_EPOLL_STAT_WAITING;
- rt_spin_unlock_irqrestore(&ep->spinlock, level);
- rt_schedule();
- level = rt_spin_lock_irqsave(&ep->spinlock);
- if (ep->status == RT_EPOLL_STAT_WAITING)
- ep->status = RT_EPOLL_STAT_INIT;
- }
- }
- ret = !(ep->status == RT_EPOLL_STAT_TRIG);
- rt_spin_unlock_irqrestore(&ep->spinlock, level);
- return ret;
- }
- /**
- * @brief Gets events associated with a file descriptor in the epoll instance.
- *
- * This function gets events associated with a file descriptor in the epoll instance.
- *
- * @param fl Pointer to the file descriptor list structure.
- * @param req Pointer to the poll request structure.
- *
- * @return Returns the bitmask of events associated with the file descriptor.
- */
- static int epoll_get_event(struct rt_fd_list *fl, rt_pollreq_t *req)
- {
- struct dfs_file *df;
- int mask = 0;
- int fd = 0;
- fd = fl->fd;
- if (fd >= 0)
- {
- df = fd_get(fd);
- if (df)
- {
- if (df->vnode->fops->poll)
- {
- req->_key = fl->revents | POLLERR | POLLHUP;
- mask = df->vnode->fops->poll(df, req);
- if (mask < 0)
- return mask;
- }
- mask &= fl->revents | EPOLLOUT | POLLERR;
- }
- }
- return mask;
- }
- /**
- * @brief Performs epoll operation to get triggered events.
- *
- * This function performs epoll operation to get triggered events.
- *
- * @param ep Pointer to the epoll instance.
- * @param events Pointer to the array to store triggered events.
- * @param maxevents Maximum number of events to store in the array.
- * @param timeout Timeout value in milliseconds.
- *
- * @return Returns the number of triggered events.
- */
- static int epoll_do(struct rt_eventpoll *ep, struct epoll_event *events, int maxevents, int timeout)
- {
- struct rt_fd_list *rdlist;
- rt_slist_t *node = RT_NULL;
- int event_num = 0;
- int istimeout = 0;
- int isn_add = 0;
- int isfree = 0;
- int mask = 0;
- rt_base_t level;
- while (1)
- {
- rt_mutex_take(&ep->lock, RT_WAITING_FOREVER);
- level = rt_spin_lock_irqsave(&ep->spinlock);
- if (ep->eventpoll_num > 0)
- {
- rt_slist_for_each(node,&ep->rdl_head)
- {
- rdlist = rt_slist_entry(node, struct rt_fd_list, rdl_node);
- ep->eventpoll_num --;
- rt_slist_remove(&ep->rdl_head, &rdlist->rdl_node);
- rdlist->is_rdl_node = RT_FALSE;
- rt_spin_unlock_irqrestore(&ep->spinlock, level);
- isfree = 0;
- isn_add = 0;
- if (event_num < maxevents)
- {
- if (rdlist->wqn.wqueue)
- {
- rt_wqueue_remove(&rdlist->wqn);
- }
- mask = epoll_get_event(rdlist, &rdlist->req);
- if (mask & rdlist->revents)
- {
- rdlist->epev.events = mask & rdlist->revents;
- }
- else
- {
- isfree = 1;
- isn_add = 1;
- }
- if (rdlist->revents & EPOLLONESHOT)
- {
- rdlist->revents = 0;
- isfree = 1;
- if (rdlist->wqn.wqueue)
- rt_wqueue_remove(&rdlist->wqn);
- }
- else
- {
- if (rdlist->revents & EPOLLET)
- {
- isfree = 1;
- }
- else
- {
- level = rt_spin_lock_irqsave(&ep->spinlock);
- if (rdlist->exclusive != 1)
- {
- rdlist->exclusive = 1;
- }
- rt_spin_unlock_irqrestore(&ep->spinlock, level);
- }
- }
- if (!isn_add)
- {
- memcpy(&events[event_num], &rdlist->epev, sizeof(rdlist->epev));
- event_num ++;
- }
- if (!isfree)
- {
- if (rdlist->is_rdl_node == RT_FALSE)
- {
- level = rt_spin_lock_irqsave(&ep->spinlock);
- ep->eventpoll_num ++;
- rt_slist_append(&ep->rdl_head, &rdlist->rdl_node);
- rdlist->is_rdl_node = RT_TRUE;
- rt_spin_unlock_irqrestore(&ep->spinlock, level);
- }
- else
- {
- level = rt_spin_lock_irqsave(&ep->spinlock);
- if (!rdlist->wqn.wqueue)
- {
- epoll_get_event(rdlist, &rdlist->req);
- }
- rt_spin_unlock_irqrestore(&ep->spinlock, level);
- }
- }
- }
- else
- {
- level = rt_spin_lock_irqsave(&ep->spinlock);
- break;
- }
- level = rt_spin_lock_irqsave(&ep->spinlock);
- }
- }
- rt_spin_unlock_irqrestore(&ep->spinlock, level);
- rt_mutex_release(&ep->lock);
- if (event_num || istimeout)
- {
- level = rt_spin_lock_irqsave(&ep->spinlock);
- ep->status = RT_EPOLL_STAT_INIT;
- rt_spin_unlock_irqrestore(&ep->spinlock, level);
- if ((timeout >= 0) || (event_num > 0))
- break;
- }
- if (epoll_wait_timeout(ep, timeout))
- {
- istimeout = 1;
- }
- }
- return event_num;
- }
- /**
- * @brief Waits for events on an epoll instance with specified parameters.
- *
- * This function waits for events on the specified epoll instance within the given timeout, optionally blocking signals based on the provided signal set.
- *
- * @param epfd File descriptor referring to the epoll instance.
- * @param events Pointer to the array to store triggered events.
- * @param maxevents Maximum number of events to store in the array.
- * @param timeout Timeout value in milliseconds.
- * @param ss Pointer to the signal set indicating signals to block during the wait operation. Pass NULL if no signals need to be blocked.
- *
- * @return Returns the number of triggered events on success, or -1 on failure.
- */
- static int epoll_do_wait(int epfd, struct epoll_event *events, int maxevents, int timeout, const sigset_t *ss)
- {
- struct rt_eventpoll *ep;
- struct dfs_file *df;
- lwp_sigset_t old_sig, new_sig;
- rt_err_t ret = 0;
- if (ss)
- {
- memcpy(&new_sig, ss, sizeof(lwp_sigset_t));
- lwp_thread_signal_mask(rt_thread_self(), LWP_SIG_MASK_CMD_BLOCK, &new_sig, &old_sig);
- }
- if ((maxevents > 0) && (epfd >= 0))
- {
- df = fd_get(epfd);
- if (df && df->vnode)
- {
- ep = (struct rt_eventpoll *)df->vnode->data;
- if (ep)
- {
- ret = epoll_do(ep, events, maxevents, timeout);
- }
- }
- }
- if (ss)
- {
- lwp_thread_signal_mask(rt_thread_self(), LWP_SIG_MASK_CMD_SET_MASK, &old_sig, RT_NULL);
- }
- if (ret < 0)
- {
- rt_set_errno(-ret);
- ret = -1;
- }
- return ret;
- }
- /**
- * @brief Creates an epoll instance.
- *
- * This function creates an epoll instance with the specified size.
- *
- * @param size Size of the epoll instance.
- *
- * @return Returns the file descriptor referring to the created epoll instance on success, or -1 on failure.
- */
- int epoll_create(int size)
- {
- return epoll_do_create(size);
- }
- /**
- * @brief Modifies an epoll instance.
- *
- * This function modifies the epoll instance referred to by 'epfd' according to the specified operation 'op', associated with the file descriptor 'fd', and the event structure 'event'.
- *
- * @param epfd File descriptor referring to the epoll instance.
- * @param op Operation to perform (EPOLL_CTL_ADD, EPOLL_CTL_DEL, or EPOLL_CTL_MOD).
- * @param fd File descriptor to associate with the epoll instance or remove from it.
- * @param event Pointer to the event structure containing the events to modify.
- *
- * @return Returns 0 on success, or -1 on failure.
- */
- int epoll_ctl(int epfd, int op, int fd, struct epoll_event *event)
- {
- return epoll_do_ctl(epfd, op, fd, event);
- }
- /**
- * @brief Waits for events on an epoll instance.
- *
- * This function waits for events on the epoll instance referred to by 'epfd' within the given timeout.
- *
- * @param epfd File descriptor referring to the epoll instance.
- * @param events Pointer to the array to store triggered events.
- * @param maxevents Maximum number of events to store in the array.
- * @param timeout Timeout value in milliseconds.
- *
- * @return Returns the number of triggered events on success, or -1 on failure.
- */
- int epoll_wait(int epfd, struct epoll_event *events, int maxevents, int timeout)
- {
- return epoll_do_wait(epfd, events, maxevents, timeout, RT_NULL);
- }
- /**
- * @brief Waits for events on an epoll instance, blocking signals.
- *
- * This function waits for events on the epoll instance referred to by 'epfd' within the given timeout, blocking signals based on the provided signal set 'ss'.
- *
- * @param epfd File descriptor referring to the epoll instance.
- * @param events Pointer to the array to store triggered events.
- * @param maxevents Maximum number of events to store in the array.
- * @param timeout Timeout value in milliseconds.
- * @param ss Pointer to the signal set indicating signals to block during the wait operation.
- *
- * @return Returns the number of triggered events on success, or -1 on failure.
- */
- int epoll_pwait(int epfd, struct epoll_event *events, int maxevents, int timeout, const sigset_t *ss)
- {
- return epoll_do_wait(epfd, events, maxevents, timeout, ss);
- }
- /**
- * @brief Waits for events on an epoll instance, blocking signals.
- *
- * This function waits for events on the epoll instance referred to by 'epfd' within the given timeout, blocking signals based on the provided signal set 'ss'.
- *
- * @param epfd File descriptor referring to the epoll instance.
- * @param events Pointer to the array to store triggered events.
- * @param maxevents Maximum number of events to store in the array.
- * @param timeout Timeout value in milliseconds.
- * @param ss Pointer to the signal set indicating signals to block during the wait operation.
- *
- * @return Returns the number of triggered events on success, or -1 on failure.
- */
- int epoll_pwait2(int epfd, struct epoll_event *events, int maxevents, int timeout, const sigset_t *ss)
- {
- return epoll_do_wait(epfd, events, maxevents, timeout, ss);
- }
|