From 24116c426f22f8ce64b3dd80f74ce61125e409ee Mon Sep 17 00:00:00 2001
From: He Sun <bochang.sh@antfin.com>
Date: Fri, 11 Oct 2019 14:07:05 +0800
Subject: [PATCH] Fix address-in-use error in server and server_epoll tests

---
 test/server/main.c       | 7 ++++++-
 test/server_epoll/main.c | 4 ++++
 2 files changed, 10 insertions(+), 1 deletion(-)

diff --git a/test/server/main.c b/test/server/main.c
index c34eaa26..6bfa6ce1 100644
--- a/test/server/main.c
+++ b/test/server/main.c
@@ -24,6 +24,10 @@ int main(int argc, const char *argv[]) {
 	servaddr.sin_addr.s_addr = htonl(INADDR_ANY);
 	servaddr.sin_port = htons(6666);
 
+	int reuse = 1;
+	if (setsockopt(listenfd, SOL_SOCKET, SO_REUSEADDR, &reuse, sizeof(reuse)) < 0)
+		perror("setsockopt port to reuse failed");
+
 	ret = bind(listenfd, (struct sockaddr *) &servaddr, sizeof(servaddr));
 	if (ret < 0) {
 		printf("bind socket error: %s(errno: %d)\n", strerror(errno), errno);
@@ -54,7 +58,8 @@ int main(int argc, const char *argv[]) {
 	int n = recv(connect_fd, buff, BUF_SIZE, 0);
 	buff[n] = '\0';
 	printf("recv msg from client: %s\n", buff);
+	
 	close(connect_fd);
-
 	close(listenfd);
+	return 0;
 }
diff --git a/test/server_epoll/main.c b/test/server_epoll/main.c
index 30fcd92a..59b14753 100644
--- a/test/server_epoll/main.c
+++ b/test/server_epoll/main.c
@@ -28,6 +28,10 @@ create_and_bind() {
 	servaddr.sin_addr.s_addr = htonl(INADDR_ANY);
 	servaddr.sin_port = htons(6667);
 
+	int reuse = 1;
+	if (setsockopt(listenfd, SOL_SOCKET, SO_REUSEADDR, &reuse, sizeof(reuse)) < 0)
+		perror("setsockopt port to reuse failed");
+
 	int ret = bind(listenfd, (struct sockaddr *) &servaddr, sizeof(servaddr));
 	if (ret < 0) {
 		printf("bind socket error: %s(errno: %d)\n", strerror(errno), errno);