歡迎來到Linux教程網
Linux教程網
Linux教程網
Linux教程網
您现在的位置: Linux教程網 >> UnixLinux >  >> Linux編程 >> Linux編程

幾種服務器模型

TCP測試用客戶程序

每次運行客戶程序,在命令行參數指定服務器的ip地址,端口,發起連接的子進程數,和一個待發送的字符串數據,客戶程序將模擬多個客戶根據指定的子進程數創建子進程來並發的連接到服務器,並發送數據,服務器收到數據後都原樣的回發給客戶,是一點典型的回射服務器。

  1. #include "net.h"  
  2.  
  3. char *addr = NULL; 
  4. char *request = NULL; 
  5. unsigned int port; 
  6. int connCount; 
  7. int clientfd; 
  8.  
  9. void client_deal() 
  10.     char *buf = NULL; 
  11.     int len; 
  12.      
  13.     Tcp_connect(addr, port, &clientfd); 
  14.     if (sendAll(clientfd, request, strlen(request)) > 0) 
  15.     { 
  16.         len = recvAll(clientfd, (void**)&buf); 
  17.         if (len > 0) 
  18.         { 
  19.             buf[len] = 0; 
  20.             printf("%s\n", buf); 
  21.         } 
  22.     } 
  23.     freePtr(buf); 
  24.     Close(clientfd); 
  25.     exit(0); 
  26.  
  27. int main(int argc, char **argv) 
  28.     if (argc != 5) 
  29.     { 
  30.         printf("use [ip] [port] [connCount] [request]\n"); 
  31.         exit(-1); 
  32.     } 
  33.      
  34.     addr = argv[1]; 
  35.     port = atoi(argv[2]); 
  36.     connCount = atoi(argv[3]); 
  37.     request = argv[4]; 
  38.      
  39.     for (int i=0; i<connCount; ++i) 
  40.     { 
  41.         if (fork() == 0) 
  42.         { 
  43.             client_deal(); 
  44.         } 
  45.     } 
  46.      
  47.     while (wait(NULL) > 0); 
  48.     if (errno != ECHILD) 
  49.     { 
  50.         perror("wait error"); 
  51.         exit(-1); 
  52.     } 
  53.      
  54.     return 0; 
#include "net.h"

char *addr = NULL;
char *request = NULL;
unsigned int port;
int connCount;
int clientfd;

void client_deal()
{
	char *buf = NULL;
	int len;
	
	Tcp_connect(addr, port, &clientfd);
	if (sendAll(clientfd, request, strlen(request)) > 0)
	{
		len = recvAll(clientfd, (void**)&buf);
		if (len > 0)
		{
			buf[len] = 0;
			printf("%s\n", buf);
		}
	}
	freePtr(buf);
	Close(clientfd);
	exit(0);
}

int main(int argc, char **argv)
{
	if (argc != 5)
	{
		printf("use [ip] [port] [connCount] [request]\n");
		exit(-1);
	}
	
	addr = argv[1];
	port = atoi(argv[2]);
	connCount = atoi(argv[3]);
	request = argv[4];
	
	for (int i=0; i<connCount; ++i)
	{
		if (fork() == 0)
		{
			client_deal();
		}
	}
	
	while (wait(NULL) > 0);
	if (errno != ECHILD)
	{
		perror("wait error");
		exit(-1);
	}
	
	return 0;
}

 

1.迭代服務器

在處理完成某個客戶的請求之後才轉向下一個客戶,比較少見,雖然總的服務時間稍慢,但需要進程控制

  1. #include "net.h"  
  2.  
  3. int listenfd; 
  4.  
  5. void server_deal() 
  6.     char *buf = NULL; 
  7.     ssize_t size; 
  8.     int clifd; 
  9.                      
  10.     Accept(listenfd, NULL, NULL, &clifd); 
  11.     printf("有新連接\n"); 
  12.     if ( (size = recvAll(clifd, (void**)&buf)) > 0) 
  13.         sendAll(clifd, buf, size); 
  14.     freePtr(buf); 
  15.     Close(clifd); 
  16.  
  17. int main() 
  18.     Tcp_listen("INADDR_ANY", 9999, 5, &listenfd); 
  19.      
  20.     while (1) 
  21.     { 
  22.         server_deal(); 
  23.     } 
  24.      
  25.     return 0; 
#include "net.h"

int listenfd;

void server_deal()
{
	char *buf = NULL;
	ssize_t size;
	int clifd;
					
	Accept(listenfd, NULL, NULL, &clifd);
	printf("有新連接\n");
	if ( (size = recvAll(clifd, (void**)&buf)) > 0)
	    sendAll(clifd, buf, size);
	freePtr(buf);
	Close(clifd);
}

int main()
{
	Tcp_listen("INADDR_ANY", 9999, 5, &listenfd);
	
	while (1)
	{
		server_deal();
	}
	
	return 0;
}

 

2.TCP多進程並發服務器

每個客戶fork出一個子進程並發的去處理請求,總服務器時間稍短,fork子進程比較耗費CPU時間

  1. #include "net.h"  
  2.  
  3. int listenfd; 
  4. int clifd; 
  5.  
  6. void server_deal() 
  7.     char *buf = NULL; 
  8.     ssize_t size; 
  9.                      
  10.     if ( (size = recvAll(clifd, (void**)&buf)) > 0) 
  11.         sendAll(clifd, buf, size); 
  12.     freePtr(buf); 
  13.     Close(clifd); 
  14.     exit(0); 
  15.  
  16. int main() 
  17.     Tcp_listen("INADDR_ANY", 9999, 5, &listenfd); 
  18.      
  19.     while (1) 
  20.     { 
  21.         Accept(listenfd, NULL, NULL, &clifd); 
  22.         printf("有新連接\n"); 
  23.         if (fork() == 0) 
  24.         { 
  25.             Close(listenfd); 
  26.             server_deal(); 
  27.         } 
  28.         Close(clifd); 
  29.     } 
  30.      
  31.     return 0; 
#include "net.h"

int listenfd;
int clifd;

void server_deal()
{
	char *buf = NULL;
	ssize_t size;
					
	if ( (size = recvAll(clifd, (void**)&buf)) > 0)
	    sendAll(clifd, buf, size);
	freePtr(buf);
	Close(clifd);
	exit(0);
}

int main()
{
	Tcp_listen("INADDR_ANY", 9999, 5, &listenfd);
	
	while (1)
	{
		Accept(listenfd, NULL, NULL, &clifd);
		printf("有新連接\n");
		if (fork() == 0)
		{
			Close(listenfd);
			server_deal();
		}
		Close(clifd);
	}
	
	return 0;
}

 


3.TCP預先派生子進程服務器

與之前的每一個客戶請求臨時fork一個進程處理不同,在啟動的時候就fork出一些子進程,優點是節省了臨時fork的開銷,缺點是父進程在啟動階段要先知道預先派生的子進程數,如果連接較多而無可用子進程,那麼客戶請求超過了連接排隊數就可能會被忽略

  1. #include "net.h"  
  2.  
  3. const int PROCESS_COUNT = 5; 
  4. int listenfd; 
  5.  
  6. void server_deal() 
  7.     int clifd; 
  8.     char *buf = NULL; 
  9.     ssize_t size; 
  10.              
  11.     Accept(listenfd, NULL, NULL, &clifd); 
  12.     printf("子進程%ld有新連接\n", (long)getpid()); 
  13.     if ( (size = recvAll(clifd, (void**)&buf)) > 0) 
  14.         sendAll(clifd, buf, size); 
  15.     freePtr(buf); 
  16.     Close(clifd); 
  17.  
  18. int main() 
  19.     Tcp_listen("INADDR_ANY", 9999, 5, &listenfd); 
  20.      
  21.     for (int i=0; i<PROCESS_COUNT; ++i) 
  22.     { 
  23.         if (fork() == 0) 
  24.         { 
  25.             while (1) 
  26.             { 
  27.                 server_deal(); 
  28.             } 
  29.         } 
  30.     } 
  31.      
  32.     while (1); 
  33.     return 0; 
#include "net.h"

const int PROCESS_COUNT = 5;
int listenfd;

void server_deal()
{
	int clifd;
	char *buf = NULL;
	ssize_t size;
			
	Accept(listenfd, NULL, NULL, &clifd);
	printf("子進程%ld有新連接\n", (long)getpid());
	if ( (size = recvAll(clifd, (void**)&buf)) > 0)
	    sendAll(clifd, buf, size);
	freePtr(buf);
	Close(clifd);
}

int main()
{
	Tcp_listen("INADDR_ANY", 9999, 5, &listenfd);
	
	for (int i=0; i<PROCESS_COUNT; ++i)
	{
		if (fork() == 0)
		{
			while (1)
			{
				server_deal();
			}
		}
	}
	
	while (1);
	return 0;
}



 

4.TCP預先派生子進程服務器,accept使用文件上鎖保護

因為某些內核實現中不允許多個進程引用對同一個監聽套接字調用accept,所以對accept加鎖成為原子操作為對上一種模型的改進

  1. #include "net.h"  
  2.  
  3. const int PROCESS_COUNT = 5; 
  4. int listenfd; 
  5. int lock_fd; 
  6. struct flock lock_it, unlock_it; 
  7.  
  8. void my_lock_init(const char *pathname) 
  9.     char lock_file[1024]; 
  10.      
  11.     strncpy(lock_file, pathname, sizeof(lock_file)); 
  12.     lock_fd = Mkstemp(lock_file); 
  13.     Unlink(lock_file); 
  14.      
  15.     lock_it.l_type = F_WRLCK; 
  16.     lock_it.l_whence = SEEK_SET; 
  17.     lock_it.l_start = 0; 
  18.     lock_it.l_len = 0; 
  19.      
  20.     unlock_it.l_type = F_UNLCK; 
  21.     unlock_it.l_whence = SEEK_SET; 
  22.     unlock_it.l_start = 0; 
  23.     unlock_it.l_len = 0; 
  24.  
  25. void my_lock_wait() 
  26.     while (fcntl(lock_fd, F_SETLKW, &lock_it) < 0) 
  27.     { 
  28.         if (errno == EINTR) 
  29.             continue; 
  30.         else 
  31.             printErrExit("my_lock_wait error"); 
  32.     } 
  33.  
  34. void my_lock_release() 
  35.     while (fcntl(lock_fd, F_SETLKW, &unlock_it) < 0) 
  36.     { 
  37.         if (errno == EINTR) 
  38.             continue; 
  39.         else 
  40.             printErrExit("my_lock_release error"); 
  41.     } 
  42.  
  43. void server_deal() 
  44.     int clifd; 
  45.     char *buf = NULL; 
  46.     ssize_t size; 
  47.              
  48.     my_lock_wait(); 
  49.     Accept(listenfd, NULL, NULL, &clifd); 
  50.     printf("子進程%ld有新連接\n", (long)getpid()); 
  51.     my_lock_release(); 
  52.     if ( (size = recvAll(clifd, (void**)&buf)) > 0) 
  53.         sendAll(clifd, buf, size); 
  54.     freePtr(buf); 
  55.     Close(clifd); 
  56.  
  57. int main() 
  58.     Tcp_listen("INADDR_ANY", 9999, 5, &listenfd); 
  59.     my_lock_init("/tmp/lock.XXXXXX"); 
  60.      
  61.     for (int i=0; i<PROCESS_COUNT; ++i) 
  62.     { 
  63.         if (fork() == 0) 
  64.         { 
  65.             while (1) 
  66.             { 
  67.                 server_deal(); 
  68.             } 
  69.         } 
  70.     } 
  71.      
  72.     while (1); 
  73.     return 0; 
#include "net.h"

const int PROCESS_COUNT = 5;
int listenfd;
int lock_fd;
struct flock lock_it, unlock_it;

void my_lock_init(const char *pathname)
{
	char lock_file[1024];
	
	strncpy(lock_file, pathname, sizeof(lock_file));
	lock_fd = Mkstemp(lock_file);
	Unlink(lock_file);
	
	lock_it.l_type = F_WRLCK;
	lock_it.l_whence = SEEK_SET;
	lock_it.l_start = 0;
	lock_it.l_len = 0;
	
	unlock_it.l_type = F_UNLCK;
	unlock_it.l_whence = SEEK_SET;
	unlock_it.l_start = 0;
	unlock_it.l_len = 0;
}

void my_lock_wait()
{
	while (fcntl(lock_fd, F_SETLKW, &lock_it) < 0)
	{
		if (errno == EINTR)
			continue;
		else
			printErrExit("my_lock_wait error");
	}
}

void my_lock_release()
{
	while (fcntl(lock_fd, F_SETLKW, &unlock_it) < 0)
	{
		if (errno == EINTR)
			continue;
		else
			printErrExit("my_lock_release error");
	}
}

void server_deal()
{
	int clifd;
	char *buf = NULL;
	ssize_t size;
			
	my_lock_wait();
	Accept(listenfd, NULL, NULL, &clifd);
	printf("子進程%ld有新連接\n", (long)getpid());
	my_lock_release();
	if ( (size = recvAll(clifd, (void**)&buf)) > 0)
	    sendAll(clifd, buf, size);
	freePtr(buf);
	Close(clifd);
}

int main()
{
	Tcp_listen("INADDR_ANY", 9999, 5, &listenfd);
	my_lock_init("/tmp/lock.XXXXXX");
	
	for (int i=0; i<PROCESS_COUNT; ++i)
	{
		if (fork() == 0)
		{
			while (1)
			{
				server_deal();
			}
		}
	}
	
	while (1);
	return 0;
}



 

5.TCP預先派生子進程服務器,accept使用線程上鎖保護

與上一模型類似,采用多進程間共享線程鎖進行的方式對預先派生進程服務器的改進

  1. #include "net.h"  
  2.  
  3. const int PROCESS_COUNT = 5; 
  4. int listenfd; 
  5. pthread_mutex_t *mptr; 
  6.  
  7. void my_lock_init() 
  8.     int fd; 
  9.      
  10.     pthread_mutexattr_t mattr; 
  11.     fd = Open("/dev/zero", O_RDWR, 0); 
  12.     mptr = (pthread_mutex_t*)Mmap(0, sizeof(pthread_mutex_t),   
  13.             PROT_READ | PROT_WRITE, MAP_SHARED, fd, 0); 
  14.     Close(fd); 
  15.     pthread_mutexattr_init(&mattr); 
  16.     pthread_mutexattr_setpshared(&mattr, PTHREAD_PROCESS_SHARED); 
  17.     pthread_mutex_init(mptr, &mattr); 
  18.  
  19. void my_lock_wait() 
  20.     pthread_mutex_lock(mptr); 
  21.  
  22. void my_lock_release() 
  23.     pthread_mutex_unlock(mptr); 
  24.  
  25. void server_deal() 
  26.     int clifd; 
  27.     char *buf = NULL; 
  28.     ssize_t size; 
  29.              
  30.     my_lock_wait(); 
  31.     Accept(listenfd, NULL, NULL, &clifd); 
  32.     printf("子進程%ld有新連接\n", (long)getpid()); 
  33.     my_lock_release(); 
  34.     if ( (size = recvAll(clifd, (void**)&buf)) > 0) 
  35.         sendAll(clifd, buf, size); 
  36.     freePtr(buf); 
  37.     Close(clifd); 
  38.  
  39. int main() 
  40.     Tcp_listen("INADDR_ANY", 9999, 5, &listenfd); 
  41.     my_lock_init(); 
  42.      
  43.     for (int i=0; i<PROCESS_COUNT; ++i) 
  44.     { 
  45.         if (fork() == 0) 
  46.         { 
  47.             while (1) 
  48.             { 
  49.                 server_deal(); 
  50.             } 
  51.         } 
  52.     } 
  53.      
  54.     while (1); 
  55.     return 0; 
#include "net.h"

const int PROCESS_COUNT = 5;
int listenfd;
pthread_mutex_t *mptr;

void my_lock_init()
{
	int fd;
	
	pthread_mutexattr_t mattr;
	fd = Open("/dev/zero", O_RDWR, 0);
	mptr = (pthread_mutex_t*)Mmap(0, sizeof(pthread_mutex_t), 
			PROT_READ | PROT_WRITE, MAP_SHARED, fd, 0);
	Close(fd);
	pthread_mutexattr_init(&mattr);
	pthread_mutexattr_setpshared(&mattr, PTHREAD_PROCESS_SHARED);
	pthread_mutex_init(mptr, &mattr);
}

void my_lock_wait()
{
	pthread_mutex_lock(mptr);
}

void my_lock_release()
{
	pthread_mutex_unlock(mptr);
}

void server_deal()
{
	int clifd;
	char *buf = NULL;
	ssize_t size;
			
	my_lock_wait();
	Accept(listenfd, NULL, NULL, &clifd);
	printf("子進程%ld有新連接\n", (long)getpid());
	my_lock_release();
	if ( (size = recvAll(clifd, (void**)&buf)) > 0)
	    sendAll(clifd, buf, size);
	freePtr(buf);
	Close(clifd);
}

int main()
{
	Tcp_listen("INADDR_ANY", 9999, 5, &listenfd);
	my_lock_init();
	
	for (int i=0; i<PROCESS_COUNT; ++i)
	{
		if (fork() == 0)
		{
			while (1)
			{
				server_deal();
			}
		}
	}
	
	while (1);
	return 0;
}

 

 

6.TCP預先派生子進程服務器,主進程傳遞描述符

主進程中accept後將已連接的套接字通過進程間通信的方式傳遞給預先派生的空閒進程,預先派生的進程處理完成後向主進程發送消息,主進程負責維護所有預先派生進程的狀態以及可用數目

  1. #include "net.h"  
  2.  
  3. #define THREAD_COUNT 5  
  4.  
  5. typedef struct 
  6.     pid_t pid; 
  7.     int   pipefd; 
  8.     int   status; 
  9.     long  count; 
  10. } Child; 
  11.  
  12. int listenfd; 
  13. int navail; 
  14. Child carr[THREAD_COUNT]; 
  15. int tmp_conn_count; 
  16.  
  17. void sig_int(int sig) 
  18.     int i; 
  19.     int sum = 0; 
  20.      
  21.     sum += tmp_conn_count; 
  22.     printf("tmp_conn_count:%d\n", tmp_conn_count); 
  23.     for (i=0; i<THREAD_COUNT; i++) 
  24.     { 
  25.         sum += carr[i].count; 
  26.         printf("carr[%d]'s conn is %ld\n", i, carr[i].count); 
  27.     } 
  28.     printf("sum is %d\n", sum); 
  29.     exit(-1); 
  30.  
  31. void server_deal(int i) 
  32.     int ret; 
  33.     int clifd; 
  34.     char *buf = NULL; 
  35.     char c = 'w'; 
  36.     int size; 
  37.     struct strrecvfd recv_stru; 
  38.      
  39.     while (1) 
  40.     { 
  41.         recvfd(STDERR_FILENO, &clifd); 
  42.         if ( (size = recvAll(clifd, (void**)&buf)) > 0) 
  43.             sendAll(clifd, buf, size); 
  44.         Close(clifd); 
  45.         freePtr(buf); 
  46.         buf = NULL; 
  47.         write(STDERR_FILENO, &c, 1);     
  48.     } 
  49.  
  50. void child_make(int i) 
  51.     int sockfd[2]; 
  52.     pid_t pid; 
  53.      
  54.     Socketpair(AF_LOCAL, SOCK_STREAM, 0, sockfd); 
  55.     //Socketpair(AF_UNIX, SOCK_STREAM, 0, sockfd);  
  56.     if ( (pid = fork()) > 0) 
  57.     { 
  58.         Close(sockfd[1]); 
  59.         carr[i].pipefd = sockfd[0]; 
  60.         carr[i].status = 0; 
  61.         carr[i].count = 0; 
  62.         carr[i].pid = pid; 
  63.     } 
  64.     else 
  65.     { 
  66.         if (dup2(sockfd[1], STDERR_FILENO) < 0) 
  67.             printErrExit("dup2 error"); 
  68.         Close(sockfd[1]); 
  69.         Close(sockfd[0]); 
  70.         Close(listenfd); 
  71.         carr[i].pipefd = sockfd[1]; 
  72.         server_deal(i); 
  73.     } 
  74.  
  75. void temp_child(int clifd) 
  76.     char *buf = NULL; 
  77.     int size; 
  78.      
  79.     if (fork() > 0) 
  80.     { 
  81.         Close(clifd); 
  82.         ++tmp_conn_count; 
  83.     } 
  84.     else 
  85.     { 
  86.         if ( (size = recvAll(clifd, (void**)&buf)) > 0) 
  87.             sendAll(clifd, buf, size); 
  88.         Close(clifd); 
  89.         freePtr(buf); 
  90.         exit(0); 
  91.     } 
  92.  
  93. int main() 
  94.     int maxfd; 
  95.     fd_set rset, master; 
  96.     int nsel; 
  97.     int clifd; 
  98.     int i; 
  99.      
  100.     printf("pid:%d\n", getpid()); 
  101.     Tcp_listen("INADDR_ANY", 9999, 5, &listenfd); 
  102.     FD_ZERO(&rset); 
  103.     FD_SET(listenfd, &master); 
  104.     maxfd = listenfd; 
  105.     tmp_conn_count = 0; 
  106.      
  107.     for (i=0; i<THREAD_COUNT; i++) 
  108.     { 
  109.         child_make(i); 
  110.         FD_SET(carr[i].pipefd, &master); 
  111.         if (maxfd < carr[i].pipefd) 
  112.             maxfd = carr[i].pipefd; 
  113.     } 
  114.      
  115.     navail = THREAD_COUNT; 
  116.     Signal(SIGINT, sig_int); 
  117.     while (1) 
  118.     { 
  119.         printf("navail: %d\n", navail); 
  120.         rset = master; 
  121.         nsel = Select(maxfd+1, &rset, NULL, NULL, NULL); 
  122.         if (FD_ISSET(listenfd, &rset)) 
  123.         { 
  124.             Accept(listenfd, NULL, NULL, &clifd); 
  125.              
  126.             if (navail > 0) 
  127.             { 
  128.                 for (i=0; i<THREAD_COUNT; i++) 
  129.                     if (carr[i].status == 0) 
  130.                         break; 
  131.                  
  132.                 //向子進程傳遞連接上來的套接字描述符  
  133.                 sendfd(carr[i].pipefd, clifd); 
  134.                 carr[i].status = 1; 
  135.                 --navail; 
  136.             } 
  137.             else 
  138.             { 
  139.                 temp_child(clifd);   
  140.             } 
  141.              
  142.             if (--nsel == 0) 
  143.                 continue;         
  144.         } 
  145.          
  146.         for(int i=0; i<THREAD_COUNT; i++) 
  147.         { 
  148.             if (FD_ISSET(carr[i].pipefd, &rset)) 
  149.             { 
  150.                 char c; 
  151.                 read(carr[i].pipefd, &c, sizeof(c)); 
  152.                 carr[i].count++; 
  153.                 carr[i].status = 0; 
  154.                 ++navail; 
  155.                                  
  156.                 if (--nsel == 0) 
  157.                     break;   
  158.             } 
  159.         } 
  160.     } 
  161.          
  162.     return 0; 
#include "net.h"

#define THREAD_COUNT 5

typedef struct
{
	pid_t pid;
	int   pipefd;
	int   status;
	long  count;
} Child;

int listenfd;
int navail;
Child carr[THREAD_COUNT];
int tmp_conn_count;

void sig_int(int sig)
{
	int i;
	int sum = 0;
	
	sum += tmp_conn_count;
	printf("tmp_conn_count:%d\n", tmp_conn_count);
	for (i=0; i<THREAD_COUNT; i++)
	{
		sum += carr[i].count;
		printf("carr[%d]'s conn is %ld\n", i, carr[i].count);
	}
	printf("sum is %d\n", sum);
	exit(-1);
}

void server_deal(int i)
{
	int ret;
	int clifd;
	char *buf = NULL;
	char c = 'w';
	int size;
	struct strrecvfd recv_stru;
	
	while (1)
	{
		recvfd(STDERR_FILENO, &clifd);
		if ( (size = recvAll(clifd, (void**)&buf)) > 0)
	    	sendAll(clifd, buf, size);
	    Close(clifd);
		freePtr(buf);
		buf = NULL;
		write(STDERR_FILENO, &c, 1);	
	}
}

void child_make(int i)
{
	int sockfd[2];
	pid_t pid;
	
	Socketpair(AF_LOCAL, SOCK_STREAM, 0, sockfd);
	//Socketpair(AF_UNIX, SOCK_STREAM, 0, sockfd);
	if ( (pid = fork()) > 0)
	{
		Close(sockfd[1]);
		carr[i].pipefd = sockfd[0];
		carr[i].status = 0;
		carr[i].count = 0;
		carr[i].pid = pid;
	}
	else
	{
		if (dup2(sockfd[1], STDERR_FILENO) < 0)
			printErrExit("dup2 error");
		Close(sockfd[1]);
		Close(sockfd[0]);
		Close(listenfd);
		carr[i].pipefd = sockfd[1];
		server_deal(i);
	}
}

void temp_child(int clifd)
{
	char *buf = NULL;
	int size;
	
	if (fork() > 0)
	{
		Close(clifd);
		++tmp_conn_count;
	}
	else
	{
		if ( (size = recvAll(clifd, (void**)&buf)) > 0)
			sendAll(clifd, buf, size);
		Close(clifd);
		freePtr(buf);
		exit(0);
	}
}

int main()
{
	int maxfd;
	fd_set rset, master;
	int nsel;
	int clifd;
	int i;
	
	printf("pid:%d\n", getpid());
	Tcp_listen("INADDR_ANY", 9999, 5, &listenfd);
	FD_ZERO(&rset);
	FD_SET(listenfd, &master);
	maxfd = listenfd;
	tmp_conn_count = 0;
	
	for (i=0; i<THREAD_COUNT; i++)
	{
		child_make(i);
		FD_SET(carr[i].pipefd, &master);
		if (maxfd < carr[i].pipefd)
			maxfd = carr[i].pipefd;
	}
	
	navail = THREAD_COUNT;
	Signal(SIGINT, sig_int);
	while (1)
	{
		printf("navail: %d\n", navail);
		rset = master;
		nsel = Select(maxfd+1, &rset, NULL, NULL, NULL);
		if (FD_ISSET(listenfd, &rset))
		{
			Accept(listenfd, NULL, NULL, &clifd);
			
			if (navail > 0)
			{
				for (i=0; i<THREAD_COUNT; i++)
					if (carr[i].status == 0)
						break;
				
				//向子進程傳遞連接上來的套接字描述符
				sendfd(carr[i].pipefd, clifd);
				carr[i].status = 1;
				--navail;
			}
			else
			{
				temp_child(clifd);	
			}
			
			if (--nsel == 0)
				continue;		
		}
		
		for(int i=0; i<THREAD_COUNT; i++)
		{
			if (FD_ISSET(carr[i].pipefd, &rset))
			{
				char c;
				read(carr[i].pipefd, &c, sizeof(c));
				carr[i].count++;
				carr[i].status = 0;
				++navail;
								
				if (--nsel == 0)
					break;	
			}
		}
	}
		
	return 0;
}

 

客戶程序創建30個子進程連接時,向服務器進程發送SIGINT信號查看各個進程服務數目的分布

     

7.TCP多線程並發服務器

對於每一個客戶請求創建一個線程來處理,與多進程並發服務器相比,創建線程比創建進程的開銷更低

  1. #include "net.h"  
  2.  
  3. int listenfd; 
  4.  
  5. void* server_deal(void *arg) 
  6.     int clifd = *((int*)arg); 
  7.     printf("clifd: %d\n", clifd); 
  8.     char *buf = NULL; 
  9.     ssize_t size; 
  10.                      
  11.     if ( (size = recvAll(clifd, (void**)&buf)) > 0) 
  12.         sendAll(clifd, buf, size); 
  13.     freePtr(buf); 
  14.     freePtr(arg); 
  15.     Close(clifd); 
  16.  
  17. int main() 
  18.     Tcp_listen("INADDR_ANY", 9999, 5, &listenfd); 
  19.      
  20.     while (1) 
  21.     { 
  22.         int clifd; 
  23.         pthread_t tid; 
  24.         int *arg = NULL; 
  25.          
  26.         Accept(listenfd, NULL, NULL, &clifd); 
  27.         printf("有新連接\n"); 
  28.         arg = (int*)Malloc(sizeof(int)); 
  29.         *arg = clifd; 
  30.         Pthread_create(&tid, NULL, server_deal, arg); 
  31.     } 
  32.      
  33.     return 0; 

更多詳情見請繼續閱讀下一頁的精彩內容: http://www.linuxidc.com/Linux/2014-10/107871p2.htm

Copyright © Linux教程網 All Rights Reserved