Parallel socket server C

shijiang1130

1. /*
   
2. * Parallel socket server.
   
3. * Forks off a new process for every new connection.
   
4. * Adapted from Stevens - Unix Network Programming.
   
5. *
   
6. * Author: Joe Armstrong
   
7. * Date: 2003-10-02
   
8. * Usage:
   
9. *   server <port>
   
10. */
    
11. 
    
12. #include        <stdio.h>
    
13. #include        <sys/socket.h>
    
14. #include        <arpa/inet.h>
    
15. #include        <varargs.h>
    
16. 
    
17. /* this buffer is used to store all the socket data */
    
18. 
    
19. char buf[65536];
    
20. 
    
21. /*VARARGS1*/
    
22. err_quit(va_alist)
    
23. va_dcl
    
24. {
    
25.   va_list         args;
    
26.   char            *fmt;
    
27.   
    
28.   va_start(args);
    
29.   fmt = va_arg(args, char *);
    
30.   vfprintf(stderr, fmt, args);
    
31.   fputc('\n', stderr);
    
32.   va_end(args);
    
33.   
    
34.   exit(1);
    
35. }
    
36. 
    
37. /*
    
38. * Read "n" bytes from a descriptor.
    
39. * Use in place of read() when fd is a stream socket.
    
40. */
    
41. 
    
42. int readn(int fd, char *ptr, int nbytes)
    
43. {
    
44.   int        nleft, nread;
    
45.   
    
46.   nleft = nbytes;
    
47.   while (nleft > 0) {
    
48.     nread = read(fd, ptr, nleft);
    
49.     if (nread < 0)
    
50.       return(nread);                /* error, return < 0 */
    
51.     else if (nread == 0)
    
52.       break;                        /* EOF */
    
53.     nleft -= nread;
    
54.     ptr   += nread;
    
55.   }
    
56.   return(nbytes - nleft);                /* return >= 0 */
    
57. }
    
58. 
    
59. /*
    
60. * Write "n" bytes to a descriptor.
    
61. * Use in place of write() when fd is a stream socket.
    
62. */
    
63. 
    
64. int writen(int fd, char *ptr, int nbytes)
    
65. {
    
66.   int        nleft, nwritten;
    
67.   
    
68.   nleft = nbytes;
    
69.   while (nleft > 0) {
    
70.     nwritten = write(fd, ptr, nleft);
    
71.     if (nwritten <= 0)
    
72.       return(nwritten);                /* error */
    
73.     nleft -= nwritten;
    
74.     ptr   += nwritten;
    
75.   }
    
76.   return(nbytes - nleft);
    
77. }
    
78. 
    
79. main(int argc, char* argv[])
    
80. {
    
81.   int port,sockfd, newsockfd, clilen, childpid, err;
    
82.   struct sockaddr_in cli_addr, serv_addr;
    
83. 
    
84.   if (argc != 2)
    
85.     err_quit("usage: server <port>\n");
    
86. 
    
87.   if ( (sockfd = socket(AF_INET, SOCK_STREAM, 0)) < 0)
    
88.     err_quit("server: can't open stream socket");
    
89. 
    
90.   port = atoi(argv[1]);
    
91.   printf("opening port %d\n", port);
    
92. 
    
93.   bzero((char *) &serv_addr, sizeof(serv_addr));
    
94.   serv_addr.sin_family      = AF_INET;
    
95.   serv_addr.sin_addr.s_addr = htonl(INADDR_ANY);
    
96.   serv_addr.sin_port        = htons(port);
    
97.   
    
98.   if ((err = bind(sockfd, (struct sockaddr *) &serv_addr,
    
99.                   sizeof(serv_addr))) < 0)
    
100.     err_quit("server: can't bind local address %d", err);
     
101.   listen(sockfd, 5);
     
102.   for ( ; ; ) {
     
103.     clilen = sizeof(cli_addr);
     
104.     newsockfd = accept(sockfd, (struct sockaddr *) &cli_addr, &clilen);
     
105.     if (newsockfd < 0)
     
106.       err_quit("server: accept error");
     
107.     if ( (childpid = fork()) < 0)
     
108.       err_quit("server: fork error");
     
109.     else if (childpid == 0) {
     
110.       close(sockfd);               
     
111.       handle(1, newsockfd, 0);
     
112.       loop(newsockfd);       
     
113.       handle(3, newsockfd, 0);
     
114.       exit(0);
     
115.     }
     
116.     close(newsockfd);
     
117.   }
     
118. }
     
119. 
     
120. /* The socket protocol is 2 byte length then the data */
     
121. 
     
122. loop(int fd)
     
123. {
     
124.   char *p;
     
125.   int again, n, i;
     
126. 
     
127.   while(1){
     
128.     p = buf;
     
129.     i = readn(fd, buf, 2);
     
130.     if(i==0)
     
131.       /* socket closed */
     
132.       break;
     
133.     else if (i == 2){
     
134.       n = (*buf)*256 + *(buf+1);
     
135.       i = readn(fd, buf, n);
     
136.       if (i != n)
     
137.         break;
     
138.       handle(2, fd, n);
     
139.     } else {
     
140.       /* protocol error */
     
141.       break;
     
142.     }
     
143.   }
     
144. }
     
145. 
     
146. /*
     
147. * handle *must* call gen_reply
     
148. */
     
149. 
     
150. handle(int phase, int fd, int n)
     
151. {
     
152.   int i;
     
153.   switch (phase)
     
154.     {
     
155.     case 1:
     
156.       printf("handle %d starting\n", getpid());
     
157.       break;
     
158.     case 2:
     
159.       printf("handle %d received %d bytes:", getpid(), n);
     
160.       for(i=0;i<n;i++)
     
161.         putchar(buf[i]);
     
162.       printf("\r\n");
     
163.       /* just for fun crash if buf[0] = 42 */
     
164.       if(buf[0] == 42)
     
165.         exit(1);
     
166.       strcpy(buf, "ack");
     
167.       gen_reply(fd, buf, 3);
     
168.       printf("handle %d sending ack\n", getpid());
     
169.       break;
     
170.     case 3:
     
171.       printf("handle %d stopping\n", getpid());
     
172.       break;
     
173.     }
     
174. }
     
175. 
     
176. gen_reply(int fd, char *p, int n)
     
177. {
     
178.   char out[2];
     
179.   out[0] = n >> 8;
     
180.   out[1] = n & 0xff;
     
181.   writen(fd, out, 2);
     
182.   writen(fd, p, n);
     
183. }
     
184. 
     
185.   

复制代码

patagonia2

好评如潮，er现在是公认的高并发项目的好工具。