1 files changed, 2 insertions, 6 deletions
diff --git a/eventloop.c b/eventloop.c
index f79114d..9a26b98 100644
--- a/eventloop.c
+++ b/eventloop.c
@@ -19,10 +19,7 @@ int eventLoopAddSocket(struct eventLoop *el, struct tsocket *sock, int flag)
 {
 struct epoll_event ev;
 ev.events = flag;
- ev.data.fd = sock->fd;
-
- // TODO: How the fd ranges? Will it be larger than the MAX_EVENTS?
- el->socks[sock->fd] = sock;
+ ev.data.ptr = sock;
 
 return epoll_ctl(el->epollfd, EPOLL_CTL_ADD, sock->fd, &ev);
 }
@@ -34,6 +31,5 @@ int eventLoopWait(struct eventLoop *el, int timeout)
 
 struct tsocket *eventLoopGetSocket(struct eventLoop *el, int index)
 {
- int fd = el->events[index].data.fd;
- return el->socks[fd];
+ return (struct tsocket *)el->events[index].data.ptr;
 }