X-Git-Url: https://tinc-vpn.org/git/browse?a=blobdiff_plain;f=fd%2Ffd_epoll.c;fp=fd%2Ffd_epoll.c;h=43bdc06eae33bab345b4dae728215f20b42ca886;hb=02b9e09173fbc1c481fb7f9aeff7d268f1af99e9;hp=0000000000000000000000000000000000000000;hpb=e10d80f4c2fbfd2f542e385a4da6a5515b5aad74;p=tinc

diff --git a/fd/fd_epoll.c b/fd/fd_epoll.c
new file mode 100644
index 00000000..43bdc06e
--- /dev/null
+++ b/fd/fd_epoll.c
@@ -0,0 +1,166 @@
+/*
+    fd_epoll.c -- I/O and event multiplexing using epoll
+
+    Copyright (C) 2003-2004 Guus Sliepen <guus@tinc-vpn.org>,
+
+    This program is free software; you can redistribute it and/or modify
+    it under the terms of the GNU General Public License as published by
+    the Free Software Foundation; either version 2 of the License, or
+    (at your option) any later version.
+
+    This program is distributed in the hope that it will be useful,
+    but WITHOUT ANY WARRANTY; without even the implied warranty of
+    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+    GNU General Public License for more details.
+
+    You should have received a copy of the GNU General Public License
+    along with this program; if not, write to the Free Software
+    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
+
+    $Id: fd.c 1375 2004-03-22 12:30:39Z guus $
+*/
+
+#include "system.h"
+
+#include "support/avl.h"
+#include "support/xalloc.h"
+#include "fd/event.h"
+#include "fd/fd.h"
+
+static int epollfd;
+static avl_tree_t *fds;
+
+volatile bool fd_running = false;
+
+int fd_compare(struct fd *a, struct fd *b) {
+	return (a->fd - b->fd) ?: (a->mode - b->mode);
+};
+
+bool fd_init(void) {
+	int i;
+
+	epollfd = epoll_create(32);
+
+	if(epollfd == -1) {
+		logger(LOG_ERR, "fd: could not open an epoll file descriptor: %s", strerror(errno));
+		return false;
+	}
+
+	fds = avl_tree_new((avl_compare_t)fd_compare, NULL);
+
+	event_init();
+}
+
+bool fd_exit(void) {
+	event_exit();
+
+	avl_tree_del(fds);
+
+	close(epollfd);
+}
+
+bool fd_add(struct fd *fd) {
+	if(!avl_add(fds, fd))
+		return false;
+	
+	fd->event.events = 0;
+
+	if(fd->read)
+		fd->event.events |= EPOLLIN;
+
+	if(fd->write)
+		fd->event.events |= EPOLLOUT;
+
+	if(fd->error)
+		fd->event.events |= EPOLLPRI | EPOLLERR | EPOLLHUP;
+
+	fd->event.data.ptr = fd;
+
+	if(epoll_ctl(epollfd, EPOLL_CTL_ADD, fd->fd, &fd->event) == -1) {
+		logger(LOG_ERR, "fd: failed to add file descriptor: %s", strerror(errno));
+		return false;
+	}
+
+	return true;
+};
+
+bool fd_del(struct fd *fd) {
+	if(epoll_ctl(epollfd, EPOLL_CTL_DEL, fd->fd, &fd->event) == -1) {
+		logger(LOG_ERR, "fd: failed to delete file descriptor: %s", strerror(errno));
+		return false;
+	}
+
+	return avl_del(fds, fd);
+};
+
+bool fd_mod(struct fd *fd) {
+	fd->event.events = 0;
+
+	if(fd->read)
+		fd->event.events |= EPOLLIN;
+
+	if(fd->write)
+		fd->event.events |= EPOLLOUT;
+
+	if(fd->error)
+		fd->event.events |= EPOLLPRI | EPOLLERR | EPOLLHUP;
+
+	if(epoll_ctl(epollfd, EPOLL_CTL_MOD, fd->fd, &fd->event) == -1) {
+		logger(LOG_ERR, "fd: failed to modify file descriptor: %s", strerror(errno));
+		return false;
+	}
+
+	return true;
+}	
+
+bool fd_run(void) {
+	struct timeval tv;
+	int result;
+	struct epoll_event *events[10];
+
+	fd_running = true;
+
+	logger(LOG_INFO, "fd: running");
+		
+	while(fd_running) {
+		tv = event_timeout();
+
+		result = epoll_wait(epollfd, events, sizeof events / sizeof *events, tv.tv_sec >= 0 ? tv.tv_sec * 1000 + tv.tv_usec / 1000: -1);
+
+		if(result < 0) {
+			if(errno != EINTR && errno != EAGAIN) {
+				logger(LOG_ERR, _("fd: error while waiting for input: %s"), strerror(errno));
+				return false;
+			}
+
+			continue;
+		}
+
+		if(result) {
+			struct fd *fd;
+
+			while(result--) {
+				fd = events[result].data.ptr;
+
+				if(events[result].events & EPOLLIN)
+					fd->read(fd);
+
+				if(events[result].events & EPOLLOUT)
+					fd->write(fd);
+
+				if(events[result].events & (EPOLLPRI | EPOLLERR | EPOLLHUP))
+					fd->error(fd);
+			}
+		} else {
+			event_handle();
+		}
+	}
+
+	logger(LOG_INFO, "fd: stopping");
+
+	return true;
+}
+
+void fd_stop(void) {
+	fd_running = false;
+}