TCP并发服务器程序

AllenYao

我把《UNIX网络编程第一卷》的示例代码改了一点点想用于生产环境（AIX），但是还没有想好怎么样监控进程以达到在有较多客户端请求时能再产生新进程和在客户端请求减少时销毁空闲的进程。这个在书里只是提到了却没有给出示例代码。

1. 
   
2. #include <stdio.h>
   
3. #include <unistd.h>
   
4. #include <stdlib.h>
   
5. #include <errno.h>
   
6. #include <string.h>
   
7. #include <sys/types.h>
   
8. #include <sys/socket.h>
   
9. #include <sys/mman.h>
   
10. #include <pthread.h>
    
11. 
    
12. extern int errno;
    
13. 
    
14. pthread_mutex_t *mptr; /* actual mutex will be in shared memory */
    
15. 
    
16. static int nchildren = 4;
    
17. static pid_t *pids;
    
18. 
    
19. static int g_iPort = 11014;
    
20. 
    
21. /********************************************************************/
    
22. int main( int argc, char ** argv )
    
23. {
    
24.     void worldcup( void );
    
25.    
    
26.     pids = calloc( nchildren, sizeof(pid_t) );
    
27.         if (pids == NULL)
    
28.         {
    
29.                 printf( "分配内存失败!\n" );
    
30.             exit( 0 );
    
31.         }
    
32.                
    
33.         setpgrp();
    
34.         if (fork() == 0)
    
35.         {
    
36.                 worldcup();
    
37.         }
    
38.         return( 0 );
    
39. }
    
40. 
    
41. /********************************************************************/
    
42. void worldcup( void )
    
43. {
    
44.         int i, listenfd;
    
45.         pthread_mutexattr_t mattr;
    
46.        
    
47.         void sig_int( int );
    
48.         pid_t child_make( int, int );
    
49.        
    
50.         if ((listenfd = Listen( g_iPort, 5 )) < 0)
    
51.         {
    
52.                 printf( "\t--监听端口:%d --失败!\n", g_iPort );
    
53.                 return;
    
54.         }
    
55.         printf( "\t--监听端口:%d --成功!\n", g_iPort );
    
56.         HDB_Log( " -- 监听端口:%d 成功!", g_iPort );
    
57.        
    
58.         /* get shared memory for parent and children */
    
59.         mptr = mmap( NULL, sizeof(pthread_mutex_t), PROT_READ|PROT_WRITE,
    
60.                      MAP_ANON|MAP_SHARED, -1, 0 );
    
61.         if (mptr == MAP_FAILED)
    
62.         {
    
63.                 printf( "error: mmap()\n" );
    
64.                 exit( 0 );
    
65.         }
    
66.        
    
67.         /* initialize the mutex and lock it */
    
68.         pthread_mutexattr_init( &mattr );
    
69.         pthread_mutexattr_setpshared( &mattr, PTHREAD_PROCESS_SHARED );
    
70.         pthread_mutex_init( mptr, &mattr );
    
71.         pthread_mutexattr_destroy( &mattr );
    
72.        
    
73.         /*--------------------------------------------------------------*/
    
74.         for (i=0; i<nchildren; i++)
    
75.         {
    
76.                 pids[i] = child_make( i, listenfd );  /* parent returns */
    
77.         }
    
78.        
    
79.         signal( SIGINT, sig_int );
    
80.        
    
81.         for (;;)
    
82.         {
    
83.                 pause();  /* everything done by children */
    
84.         }
    
85. }
    
86. 
    
87. /********************************************************************/
    
88. void sig_int( int signo )
    
89. {
    
90.         int i;
    
91.        
    
92.         /* terminate all children */
    
93.         for (i=0; i<nchildren; i++)
    
94.         {
    
95.                 if (kill( pids[i], SIGTERM ) != 0)
    
96.                 {
    
97.                         printf( "%s", strerror(errno) );
    
98.                 }
    
99.         }
    
100.         while(wait(NULL)>0);  /* wait for all children */
     
101.        
     
102.         if (errno != ECHILD)
     
103.         {
     
104.                 printf( "wait error" );
     
105.         }
     
106.         exit( 0 );
     
107. }
     
108. 
     
109. /********************************************************************/
     
110. pid_t child_make( int i, int listenfd )
     
111. {
     
112.         pid_t pid;
     
113.         void child_main( int, int );
     
114.        
     
115.         if ((pid = fork()) > 0)
     
116.         {
     
117.                 return( pid );  /* parent */
     
118.         }
     
119.         child_main( i, listenfd );  /* never returns */
     
120. }
     
121. 
     
122. /********************************************************************/
     
123. void child_main( int i, int listenfd )
     
124. {
     
125.         int connfd;
     
126.         socklen_t addrlen;
     
127.         struct sockaddr cliaddr;
     
128.         void child_deal( int sockfd );
     
129.        
     
130.         HDB_Log( " -- 子进程 [%d] 准备就绪", i );
     
131.        
     
132.         for (;;)
     
133.         {
     
134.                 pthread_mutex_lock( mptr );
     
135.                 if ((connfd = accept( listenfd, &cliaddr, &addrlen )) < 0)
     
136.                 {
     
137.                         HDB_Log( "accept error : %s", strerror(errno) );
     
138.                         pthread_mutex_unlock( mptr );
     
139.                         continue;
     
140.                 }
     
141.                 pthread_mutex_unlock( mptr );
     
142.                 child_deal( connfd );
     
143.                 close( connfd );
     
144.         }
     
145. }
     
146. 
     
147. /********************************************************************/
     
148. void child_deal( int sockfd )
     
149. {
     
150.     /*从sockfd读取客户端的请求报文，处理，然后返回应答报文给客户端。*/
     
151. }
     

复制代码

[ 本帖最后由 AllenYao 于 2006-5-31 08:54 编辑 ]