网络编程之Client/Server Design Concurrent Server(三)

1. 代码展示

#include <sys/socket.h>
#include <sys/un.h>
#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <errno.h>
#include <err.h>
#include <sys/types.h>
#include <sys/ipc.h>
#include <sys/msg.h>
#include <sys/wait.h>
#include <sys/time.h>
#include <sys/resource.h>
#include <fcntl.h>

#define KEY 0x1 /* key for first message queue */
#define LISTENQ 10
#define MAXLINE 20
#define MAXN 16384
#define handle_error(msg) \
          do { perror(msg); exit(EXIT_FAILURE); } while (0)

typedef struct {
    long mtype;
    char mtext[MAXLINE];
} Mymsg;

static int read_cnt;
static char *read_ptr;
static char read_buf[MAXLINE];
char flag;



void web_child(int sockfd);
ssize_t readline(int fd, void *vptr, size_t maxlen);
static ssize_t my_read(int fd, char *ptr) ;
ssize_t writen(int fd, const void *vptr, size_t n);
void pr_cpu_time(void);



int main(int argc, char **argv)
{
    int listenfd, connfd,ident,flags;
    pid_t childpid;
    socklen_t clilen;
    in_port_t port;
    Mymsg msg;
    int i=0;
    struct sockaddr_in cliaddr, servaddr;
    if (argc != 3)
        errx(1,"tcp_fork_server <addr> <port>\n");

    port = atoi(argv[2]);
    if((listenfd = socket(AF_INET, SOCK_STREAM, 0)) == -1)
        handle_error("socket");
    bzero(&servaddr, sizeof(servaddr));
    servaddr.sin_family = AF_INET;
    servaddr.sin_port = htons(port);
    if(inet_pton(AF_INET, argv[1], &servaddr.sin_addr) == -1)
       handle_error("inet_pton");


    if((flags = fcntl(listenfd, F_GETFL, 0)) == -1)
	handle_error("fcntl");
    else
        if(fcntl(listenfd, F_SETFL, flags | O_NONBLOCK) == -1)
		handle_error("fcntl");

    if(bind(listenfd, (struct sockaddr *) &servaddr, sizeof(servaddr)) == -1)
        handle_error("bind");
    if(listen(listenfd, LISTENQ) == -1)
        handle_error("listen");



    for ( ; ; ){
		if(flag)
            //每次判断是否client发送消息给server
            if (msgrcv(ident,&msg,MAXLINE,1,IPC_NOWAIT) ==  -1){
                if(errno != ENOMSG)
                    handle_error("msgrcv");
            }
            else{
				while (wait(NULL) > 0);
				    if (errno != ECHILD)
				        errx(1,"wait error");
                pr_cpu_time();
				msg.mtype=2;
                memcpy(msg.mtext,"done",5);
                if (msgsnd(ident,&msg,MAXLINE,0) == -1 )
                    handle_error("msgrcv");
                return 0;
            }

	    clilen = sizeof(cliaddr);
		//debug
	    //i++;

        if((connfd = accept(listenfd, (struct sockaddr *) &cliaddr, &clilen)) == -1 )
            if( errno == EAGAIN)
			    continue;
	    else
		    handle_error("accept");
        if ( (childpid = fork()) <0)
            handle_error("fork");
        if(childpid == 0)
        { /* child process */
            if(close(listenfd) == -1)
                handle_error("close");
		    //debug
		    //printf("server child %d have start\n",i);
            web_child(connfd); /* process the request */
		    //printf("server child %d have done\n",i);
            exit(0);
        }
        else{

            if(close(connfd) == -1)
                handle_error("close"); /* parent closes connected socket */
                
            //保证只执行一次，来获取client的消息队列
            if(!flag){
                if((ident=msgget(KEY,0660)) == -1 )
                     handle_error("msgget");
                flag=1;
            }
        }
    }
}


void pr_cpu_time(void)
{
    double user, sys;
    struct rusage myusage, childusage;
    if (getrusage(RUSAGE_SELF, &myusage) < 0)
        handle_error("getrusage error");
    if (getrusage(RUSAGE_CHILDREN, &childusage) < 0)
        handle_error("getrusage error");
    user = (double) myusage.ru_utime.tv_sec +myusage.ru_utime.tv_usec / 1000000.0;
    user += (double) childusage.ru_utime.tv_sec +childusage.ru_utime.tv_usec / 1000000.0;
    sys = (double) myusage.ru_stime.tv_sec + myusage.ru_stime.tv_usec / 1000000.0;
    sys += (double) childusage.ru_stime.tv_sec + childusage.ru_stime.tv_usec / 1000000.0;
    printf("\nuser time = %g, sys time = %g\n", user, sys);
}




void web_child(int sockfd)
{
    int ntowrite;
    ssize_t nread;
    char line[MAXLINE], result[MAXN];
    for ( ; ; ) {

        if((nread=readline(sockfd, line, MAXLINE)) == -1)
            handle_error("readline");
        else if(nread == 0)
            return ;
        ntowrite = atol(line);
        if ((ntowrite <= 0) || (ntowrite > MAXN))
            errx(1,"client request for %d bytes,max size is %d\n", ntowrite,MAXN);
        if(writen(sockfd, result, ntowrite) == -1)
            handle_error("writen");
    }
}

ssize_t writen(int fd, const void *vptr, size_t n)
{
    size_t nleft;
    ssize_t nwritten;
    const char *ptr;
    ptr = vptr;
    nleft = n;
    while (nleft > 0)
    {
        if ( (nwritten = write(fd, ptr, nleft)) <= 0)
        {
            if (nwritten < 0 && errno == EINTR)
                nwritten = 0; /* and call write() again */
            else
                return (-1); /* error */
        }
        nleft -= nwritten;
        ptr += nwritten;
    }
    return (n);
}


static ssize_t my_read(int fd, char *ptr)
{
    if (read_cnt <= 0) {
        again:
            if ( (read_cnt = read(fd, read_buf, sizeof(read_buf))) < 0) {
                if (errno == EINTR)
                    goto again;
                return (-1);
            } else if (read_cnt == 0)
                return (0);
            read_ptr = read_buf;
    }
    read_cnt--;
    *ptr = *read_ptr++;
    return (1);
}

ssize_t readline(int fd, void *vptr, size_t maxlen)
{
    ssize_t n, rc;
    char c, *ptr;
    ptr = vptr;
    for (n = 1; n < maxlen; n++) {
        if ( (rc = my_read(fd, &c)) == 1) {
            *ptr++ = c;
            if (c == '\n')
                break; /* newline is stored, like fgets() */
        } else if (rc == 0) {
            *ptr = 0;
            return (n - 1); /* EOF, n - 1 bytes were read */
        } else
            return (-1); /* error, errno set by read() */
    }
    *ptr = 0; /* null terminate like fgets() */
    return (n);
}

2. 几个强调的地方

除了配合上一篇的Testbench以外，还有几个地方需要强调：

• if((ident=msgget(KEY,0660)) == -1 )放在第一次fork之后，因为笔者考虑一般都是先运行server，后运行testbench，为了满足这个逻辑，必须保证server IPC的获取一定位于Testbench运行起来之后。
• if(flag)的判断必须放在每次accept之前，且必须保证listenfd为非阻塞模式，否则都无法实现预期的功能。

3. 结果

//testbench端
[root@localhost ~]# ./30_testbench 127.0.0.1 9877 5 100 1000   
[root@localhost ~]# ./30_testbench 127.0.0.1 9877 5 100 2000
[root@localhost ~]# ./30_testbench 127.0.0.1 9877 10 100 2000 
[root@localhost ~]# 



//server端
[root@localhost ~]# ./30_fork_ser 127.0.0.1 9877

user time = 0.15411, sys time = 2.2219
[root@localhost ~]# ./30_fork_ser 127.0.0.1 9877

user time = 0.158257, sys time = 2.37561
[root@localhost ~]# ./30_fork_ser 127.0.0.1 9877

user time = 0.190075, sys time = 1.76053
[root@localhost ~]#