uloop: move epoll code into a separate file

[project/libubox.git] / uloop-epoll.c

diff --git a/uloop-epoll.c b/uloop-epoll.c
new file mode 100644 (file)
index 0000000..bb652fd
--- /dev/null
+++ b/uloop-epoll.c
@@ -0,0 +1,108 @@
+/*
+ * uloop - event loop implementation
+ *
+ * Copyright (C) 2010-2016 Felix Fietkau <nbd@openwrt.org>
+ *
+ * Permission to use, copy, modify, and/or distribute this software for any
+ * purpose with or without fee is hereby granted, provided that the above
+ * copyright notice and this permission notice appear in all copies.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
+ * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
+ * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
+ * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
+ * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
+ * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
+ * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
+ */
+
+/**
+ * FIXME: uClibc < 0.9.30.3 does not define EPOLLRDHUP for Linux >= 2.6.17
+ */
+#ifndef EPOLLRDHUP
+#define EPOLLRDHUP 0x2000
+#endif
+
+int uloop_init(void)
+{
+       if (poll_fd >= 0)
+               return 0;
+
+       poll_fd = epoll_create(32);
+       if (poll_fd < 0)
+               return -1;
+
+       fcntl(poll_fd, F_SETFD, fcntl(poll_fd, F_GETFD) | FD_CLOEXEC);
+       return 0;
+}
+
+static int register_poll(struct uloop_fd *fd, unsigned int flags)
+{
+       struct epoll_event ev;
+       int op = fd->registered ? EPOLL_CTL_MOD : EPOLL_CTL_ADD;
+
+       memset(&ev, 0, sizeof(struct epoll_event));
+
+       if (flags & ULOOP_READ)
+               ev.events |= EPOLLIN | EPOLLRDHUP;
+
+       if (flags & ULOOP_WRITE)
+               ev.events |= EPOLLOUT;
+
+       if (flags & ULOOP_EDGE_TRIGGER)
+               ev.events |= EPOLLET;
+
+       ev.data.fd = fd->fd;
+       ev.data.ptr = fd;
+       fd->flags = flags;
+
+       return epoll_ctl(poll_fd, op, fd->fd, &ev);
+}
+
+static struct epoll_event events[ULOOP_MAX_EVENTS];
+
+static int __uloop_fd_delete(struct uloop_fd *sock)
+{
+       sock->flags = 0;
+       return epoll_ctl(poll_fd, EPOLL_CTL_DEL, sock->fd, 0);
+}
+
+static int uloop_fetch_events(int timeout)
+{
+       int n, nfds;
+
+       nfds = epoll_wait(poll_fd, events, ARRAY_SIZE(events), timeout);
+       for (n = 0; n < nfds; ++n) {
+               struct uloop_fd_event *cur = &cur_fds[n];
+               struct uloop_fd *u = events[n].data.ptr;
+               unsigned int ev = 0;
+
+               cur->fd = u;
+               if (!u)
+                       continue;
+
+               if (events[n].events & (EPOLLERR|EPOLLHUP)) {
+                       u->error = true;
+                       if (!(u->flags & ULOOP_ERROR_CB))
+                               uloop_fd_delete(u);
+               }
+
+               if(!(events[n].events & (EPOLLRDHUP|EPOLLIN|EPOLLOUT|EPOLLERR|EPOLLHUP))) {
+                       cur->fd = NULL;
+                       continue;
+               }
+
+               if(events[n].events & EPOLLRDHUP)
+                       u->eof = true;
+
+               if(events[n].events & EPOLLIN)
+                       ev |= ULOOP_READ;
+
+               if(events[n].events & EPOLLOUT)
+                       ev |= ULOOP_WRITE;
+
+               cur->events = ev;
+       }
+
+       return nfds;
+}