init

.gitignore

@@ -0,0 +1,3 @@
+server
+client
+

Makefile

@@ -0,0 +1,5 @@
+
+all:
+	gcc -O2 -o server server.c
+	gcc -O2 -o client client.c
+

README.md

@@ -1,4 +1,34 @@
 c1000k
 ======
 
-A tool to test if you system supports 1 million connections
+This is the TCP server-client suit to help you to test if your OS supports c1000k(1 million connections).
+
+## Usage
+
+### Download and compile
+
+	wget --no-check-certificate https://github.com/ideawu/c1000k/archive/master.zip
+	unzip master.zip
+	cd c1000k-master
+	make
+
+### start the server, listen on port 7000~7099
+
+	./server 7000
+
+### run the client
+
+	./client 127.0.0.1 7000
+
+The server will ouput message like this and quit:
+
+	connections: 153
+	error: Too many open files
+
+It says the server can only accept 153 connections, it reaches the max-open-files limitation so it quit.
+
+The client will output message like this:
+
+	connections: 154
+	error: Connection refused
+

client.c

@@ -0,0 +1,73 @@
+/*****************************************************
+ * The TCP socket client to help you to test if your
+ * OS supports c1000k(1 million connections).
+ * @author: ideawu
+ * @link: http://www.ideawu.com/
+ *****************************************************/
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <unistd.h>
+#include <errno.h>
+#include <arpa/inet.h>
+#include <netinet/tcp.h>
+
+#define MAX_PORTS 100
+
+int main(int argc, char **argv){
+	if(argc <=  2){
+		printf("Usage: %s ip port\n", argv[0]);
+		exit(0);
+	}
+
+	struct sockaddr_in addr;
+	const char *ip = argv[1];
+	int base_port = atoi(argv[2]);
+	int opt = 1;
+	int bufsize;
+	socklen_t optlen;
+	int connections = 0;
+
+	bzero(&addr, sizeof(addr));
+	addr.sin_family = AF_INET;
+	inet_pton(AF_INET, ip, &addr.sin_addr);
+
+	char tmp_data[10];
+	int index = 0;
+	while(1){
+		if(++index >= MAX_PORTS){
+			index = 0;
+		}
+		int port = base_port + index;
+		//printf("connect to %s:%d\n", ip, port);
+
+		addr.sin_port = htons((short)port);
+
+		int sock;
+		if((sock = socket(AF_INET, SOCK_STREAM, 0)) == -1){
+			goto sock_err;
+		}
+		if(connect(sock, (struct sockaddr *)&addr, sizeof(addr)) == -1){
+			goto sock_err;
+		}
+
+		connections ++;
+
+		if(connections % 1000 == 999){
+			//printf("press Enter to continue: ");
+			//getchar();
+			printf("connections: %d, fd: %d\n", connections, sock);
+		}
+		usleep(1 * 1000);
+
+		bufsize = 5000;
+		setsockopt(sock, SOL_SOCKET, SO_SNDBUF, &bufsize, sizeof(bufsize));
+		setsockopt(sock, SOL_SOCKET, SO_RCVBUF, &bufsize, sizeof(bufsize));
+	}
+
+	return 0;
+sock_err:
+	printf("connections: %d\n", connections);
+	printf("error: %s\n", strerror(errno));
+	return 0;
+}

server.c

@@ -0,0 +1,114 @@
+/*****************************************************
+ * The TCP socket server to help you to test if your
+ * OS supports c1000k(1 million connections).
+ * @author: ideawu
+ * @link: http://www.ideawu.com/
+ *****************************************************/
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <unistd.h>
+#include <errno.h>
+#include <arpa/inet.h>
+#include <netinet/tcp.h>
+#include <sys/select.h>
+
+#define MAX_PORTS 100
+
+int main(int argc, char **argv){
+	if(argc <  2){
+		printf("Usage: %s port\n", argv[0]);
+		exit(0);
+	}
+
+	struct sockaddr_in addr;
+	const char *ip = "0.0.0.0";
+	int opt = 1;
+	int bufsize;
+	socklen_t optlen;
+	int connections = 0;
+	int base_port = 7000;
+	if(argc > 2){
+		base_port = atoi(argv[1]);
+	}
+
+	int server_socks[MAX_PORTS];
+
+	for(int i=0; i<MAX_PORTS; i++){
+		int port = base_port + i;
+		bzero(&addr, sizeof(addr));
+		addr.sin_family = AF_INET;
+		addr.sin_port = htons((short)port);
+		inet_pton(AF_INET, ip, &addr.sin_addr);
+
+		int serv_sock;
+		if((serv_sock = socket(AF_INET, SOCK_STREAM, 0)) == -1){
+			goto sock_err;
+		}
+		if(setsockopt(serv_sock, SOL_SOCKET, SO_REUSEADDR, &opt, sizeof(opt)) == -1){
+			goto sock_err;
+		}
+		if(bind(serv_sock, (struct sockaddr *)&addr, sizeof(addr)) == -1){
+			goto sock_err;
+		}
+		if(listen(serv_sock, 1024) == -1){
+			goto sock_err;
+		}
+
+		server_socks[i] = serv_sock;
+		printf("server listen on port: %d\n", port);
+	}
+
+	while(1){
+		fd_set readset;
+		FD_ZERO(&readset);
+		int maxfd = 0;
+		for(int i=0; i<MAX_PORTS; i++){
+			FD_SET(server_socks[i], &readset);
+			if(server_socks[i] > maxfd){
+				maxfd = server_socks[i];
+			}
+		}
+		int ret = select(maxfd + 1, &readset, NULL, NULL, NULL);
+		if(ret < 0){
+			if(errno == EINTR){
+				continue;
+			}else{
+				printf("select error! %s\n", strerror(errno));
+				exit(0);
+			}
+		}
+		if(ret == 0){
+			continue;
+		}
+
+		for(int i=0; i<MAX_PORTS; i++){
+			if(!FD_ISSET(server_socks[i], &readset)){
+				continue;
+			}
+			socklen_t addrlen = sizeof(addr);
+			int sock = accept(server_socks[i], (struct sockaddr *)&addr, &addrlen);
+			if(sock == -1){
+				goto sock_err;
+			}
+			connections ++;
+			if(connections % 1000 == 999){
+				//printf("press Enter to continue: ");
+				//getchar();
+				printf("connections: %d, fd: %d\n", connections, sock);
+			}
+
+			bufsize = 5000;
+			setsockopt(sock, SOL_SOCKET, SO_SNDBUF, &bufsize, sizeof(bufsize));
+			setsockopt(sock, SOL_SOCKET, SO_RCVBUF, &bufsize, sizeof(bufsize));
+		}
+	}
+
+	return 0;
+sock_err:
+	printf("connections: %d\n", connections);
+	printf("error: %s\n", strerror(errno));
+	return 0;
+}
+
+