2 files changed, 10 insertions, 11 deletions

diff --git a/pif.c b/pif.c
index 4cf0f6e..8ade587 100644
--- a/pif.c
+++ b/pif.c
@@ -125,11 +125,17 @@ int pif_listen(const struct ctx *c, uint8_t proto, uint8_t pif,
 	ref.listen.pif = pif;
 	ref.listen.rule = rule;
 
-	ret = epoll_add(c->epollfd, EPOLLIN, ref);
-	if (ret < 0) {
-		close(ref.fd);
-		return ret;
+	if (proto == IPPROTO_TCP && listen(ref.fd, 128) < 0) {
+		ret = -errno;
+		goto fail;
 	}
 
+	ret = epoll_add(c->epollfd, EPOLLIN, ref);
+	if (ret < 0)
+		goto fail;
+
 	return ref.fd;
+fail:
+	close(ref.fd);
+	return ret;
 }
diff --git a/util.c b/util.c
index ed874f1..0aa4d42 100644
--- a/util.c
+++ b/util.c
@@ -169,13 +169,6 @@ static int sock_l4_(const struct ctx *c, enum epoll_type type,
 		}
 	}
 
-	if (type == EPOLL_TYPE_TCP_LISTEN && listen(fd, 128) < 0) {
-		ret = -errno;
-		warn("TCP socket listen: %s", strerror_(-ret));
-		close(fd);
-		return ret;
-	}
-
 	return fd;
 }