并发服务器的压力测试

tcp并发服务器的压力测试

我修改了最大进程个数，所以做压力测试的时候，我只用时间来进行比较，即写一个shell脚本，每0.01执行一次连接操作，这个时候服务器是接受不过来的。但是我0.02执行一次连接操作，就可以处理。

1、shell脚本

#! /bin/bash
#!/bin/bash

for((i=1;i<=1100;i=i+1))
do
            ./a & //后台执行，即不依次执行
            echo "$i"   
            sleep 0.03
    done
//统计"O" 和 "F" 出现的个数，因为做压力测试的时候连接成功我输出
//OK，错误的时候我用的F
cat test.log | grep "F"|wc -l
cat test.log | grep "O"|wc -l

客户端代码

这个是最基础的客户端连接服务器的代码，具体的怎么创建套接字和连接的我就不说了。里面的客户端地址大家需要自行了解，因为这个需要本机地址转换成网络地址。

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <errno.h>
#include <sys/types.h>          /* See NOTES */
#include <sys/socket.h>
 #include <arpa/inet.h>
#include <sys/socket.h>
       #include <netinet/in.h>
       #include <arpa/inet.h>
#define MSG "hello"
int main(){
        int                             cli_fd;
        struct sockaddr_in              cli_addr;
        socklen_t                       addrlen;
        int                             rv;
        char                            buf[1024];
        cli_fd = socket(AF_INET, SOCK_STREAM, 0);

        memset(&cli_addr,0,sizeof(cli_addr));
        cli_addr.sin_family = AF_INET;
        cli_addr.sin_port = htons(8899);
        inet_aton("59.110.42.24", &cli_addr.sin_addr);
        if(cli_fd < 0){
                printf("creat socket is faild[%s]\n",strerror(errno));
                printf("F");
                return -1;

        }
        //printf("create socket is successful:[%d]\n",cli_fd);

        if(connect(cli_fd, (struct sockaddr *)&cli_addr,sizeof(cli_addr))<0){
                printf("connect is faild[%s]\n",strerror(errno));
                printf("F");
                return 0;
        }
        //printf("connect is successful\n");
                /*
                rv = read(cli_fd, buf, sizeof(buf));
                if(rv <0){
                        printf("read is faild[%s]\n",strerror(errno));
                        continue;
                }
                printf("data is :[%s]",buf);
                */


                if(write(cli_fd, &MSG, sizeof(MSG))<0){
                        printf("write is faild[%s]\n",strerror(errno));
                        close(cli_fd);
        //              printf("失败");
                        exit(-1);
                }
        //      printf("write is successful,\n");
                memset(buf,0,sizeof(buf));
                rv = read(cli_fd, buf, sizeof(buf));
                if(rv<0){
                        printf("read is failed:[%s]\n",strerror(errno));
                        close(cli_fd);
        //      printf("失败");
                        exit(-1);
                }
        //      printf("read is successful,data is [%s]\n",buf);
                printf("OK");
                close(cli_fd);
                exit(0);


}          

3、并发服务器代码

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <sys/types.h>
       #include <sys/wait.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <errno.h>
#include <arpa/inet.h>
 #include <sys/types.h>
 #include <sys/socket.h>
void handle_sigchld2(int signo)
{
        int mypid;
        while (( mypid = waitpid(-1, NULL, WNOHANG)) > 0){
                 printf("child %d terminated\n", mypid);
        }
}

int main(){

        int                     listen_fd,client_fd;
        struct sockaddr_in      serv_addr;
        struct sockaddr_in      cli_addr;
        socklen_t               socklen;
        char                    buf[1024];
        int                     rv;
        listen_fd = socket(AF_INET, SOCK_STREAM,0);
        if(listen_fd < 0){
                printf("creat socket_fd is failed:[%s]\n",strerror(errno));
                return -1;
        }

        printf("create socket successful\n");
        //将socket_in结构体清空为0
        memset(&serv_addr, 0, sizeof(serv_addr));
        serv_addr.sin_family = AF_INET;
        serv_addr.sin_port = htons(8899);
        serv_addr.sin_addr.s_addr = htonl( INADDR_ANY);
		if( bind(listen_fd, (struct sockaddr *)&serv_addr,sizeof(serv_addr))<0){
                printf("bind is failed:[%s]\n",strerror(errno));
                return -1;
        }
        printf("socket:[%d],bind on port:[%d]\n",listen_fd,8899);

        listen(listen_fd,14);
        signal(SIGCHLD, handle_sigchld2);
        while(1){
                client_fd = accept(listen_fd, (struct sockaddr *) &cli_addr, &socklen);
                if(client_fd < 0)       {

                        printf("Accept new client failure: %s\n", strerror(errno));
                        continue;
                }
                printf("Accept new client[%s:%d] successfully\n" ,inet_ntoa(cli_addr.sin_addr), ntohs(cli_addr.sin_port));
                pid_t           pid;
                int             status = -1;
                pid = fork();
                if( pid < 0 )
                {
                 printf("fork() create child process failure: %s\n", strerror(errno));
                 close(client_fd);
                 continue;
                }
                else if( pid > 0 )
                {
                        /* Parent process close client fd and goes to  accept new socket client again */
                        waitpid(pid, &status, WNOHANG);
                        close(client_fd);
                        continue;

                }
                else{
                        char                 buf[1024];
						close(listen_fd);
                        printf("Child process start to commuicate with socket client...\n");
                        memset(buf, 0, sizeof(buf));
                        rv=read(client_fd, buf, sizeof(buf));
                        if( rv < 0 )
                        {

                                printf("Read data from client sockfd[%d] failure: %s\n", client_fd, strerror(errno));
                                close(client_fd);
                                exit(0);
                        }
                        printf("Read %d bytes data from Server: %s\n", rv, buf);
                        rv = write(client_fd,buf,sizeof(buf));
                        if(rv<0){
                                printf("write is failed:[%s]\n",strerror(errno));
                                close(client_fd);
                                exit(0);
                        }
                        close(client_fd);
                        exit(0);

                }


        }
        close(listen_fd);
}
                                  

这个代码大家可以直接用，值得注意的是当大家在写服务器的端的时候一定注意处理僵尸进程的问题，我是用到了信号跟waitpid来处理，我也推荐大家使用这种方式。
另外在我用0.01来执行的时候，会产生比较多的僵尸进程，但是我用ps -aux命令查看，只发现了几个僵尸进程，或许是因为tcp三次握手还没有建立起来就拉闸了吧。。。。
大家可以看到1100个客户端只有多少个正常的。