LINUX网络编程(TCP)(2 )
LINUX网络编程(TCP)(2 )
庸俗武士 发表于4年前
LINUX网络编程(TCP)(2 )
  • 发表于 4年前
  • 阅读 168
  • 收藏 4
  • 点赞 0
  • 评论 0

标题:腾讯云 新注册用户域名抢购1元起>>>   

在 LINUX网络编程(TCP)(1) 中的 程序。。。。


如果启用2个client连接 Server。 先连上server的哪一个Client只要不执行 命令3.

那么第2个client貌似不管怎么发 命令,都得不到 服务器的响应。这是为什么呢?

简单来说:由于通信是使用TCP协议的,而TCP协议是面向连接的, 只要有连接没有断开,那么 服务器就没有办法响应了。只有等连接断开之后,才能处理其他client的命令。


于是,可以对服务器端的程序 做 如下的 修改:

#1. fork()函数的使用。

服务器端程序 变化为:

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <sys/socket.h>
#include <sys/types.h>
#include <netdb.h>

#define PORT_NUM	(2227)		//定义端口号。
#define BUFFSIZE	(1024)

#define	CMD_ONE			("1")
#define CMD_TWO			("2")
#define CMD_THREE		("3")

#define CMD_CODE_ERR	(0)
#define CMD_CODE_ONE	(1)
#define CMD_CODE_TWO	(2)
#define CMD_CODE_THREE	(3)

#define LISTEN_QUE		(10)

#define LOG_MSG(format, ...)	do{fprintf(stderr, "[MessageInfo]: "format" \n", ##__VA_ARGS__);}while(0)

typedef struct socklnk_st{
	int32_t sd;
	struct socklnk_st *next;
}socklnk_t;


socklnk_t*
create_soclnk(int32_t data)
{
	socklnk_t *soclnk = NULL;
	
	soclnk = (socklnk_t *)malloc(sizeof(socklnk_t));
	if(soclnk != NULL){
		soclnk->next = NULL;
		soclnk->sd = data;
	}
	
	return soclnk;
}

void
free_soclnk(socklnk_t *dst)
{
	socklnk_t *tmp = dst;
	socklnk_t *next;

	while(tmp != NULL){
		next = tmp->next;
		close(tmp->sd);
		free(tmp);
		tmp = next;
	}
}

//定义每个Code返回值。
static int8_t *server_msg[] = {
	[CMD_CODE_ONE] = "Hello",
	[CMD_CODE_TWO] = "Good",
	[CMD_CODE_THREE] = "Bye",
	[CMD_CODE_ERR]	= "Error Command",
};

socklnk_t *
start_serice(uint16_t port)
{
	int8_t serv[NI_MAXSERV];
	int32_t sd;
	int32_t opt = 1;
	struct addrinfo *ai, *res = NULL;
	struct addrinfo hints;
	socklnk_t *head = NULL;
	socklnk_t *cur  = NULL;	
	
	if(snprintf(serv, NI_MAXSERV, "%u", port) < 0){
		goto func_end;
	}
	
	(void)memset(&hints, 0x00, sizeof hints);
	hints.ai_family = AF_UNSPEC;
	hints.ai_socktype = SOCK_STREAM;
	hints.ai_flags = AI_PASSIVE;
	
	if(getaddrinfo(NULL, serv, &hints, &res) != 0){
		goto func_end;
	}
	
	for(ai = res; ai != NULL; ai = ai->ai_next){
		sd = socket(ai->ai_family, ai->ai_socktype, ai->ai_protocol);
		if(sd < 0){
			continue;
		}
		
		if(setsockopt(sd, SOL_SOCKET, SO_REUSEADDR, &opt, sizeof opt) < 0){
			close(sd);
			free_soclnk(head);
			goto func_end;
		}
		
		if(bind(sd, ai->ai_addr, ai->ai_addrlen) < 0){
			close(sd);
			continue;
		}
		
		if(listen(sd, LISTEN_QUE) < 0){
			close(sd);
			continue;
		}
		
		if(head == NULL){
			head = create_soclnk(sd);
			if(head == NULL){
				close(sd);
				free_soclnk(head);
				goto func_end;			
			}
			cur = head;
		}else{
			cur->next = create_soclnk(sd);
			if(cur->next == NULL){
				close(sd);
				free_soclnk(head);
				goto func_end;			
			}
			cur = cur->next;
		}
	}
	
func_end:
	if(res != NULL){
		freeaddrinfo(res);
	}

	return head;
}

int
main(int ac, char **av)
{
	int8_t *buf;
	int32_t connfd;
	int32_t maxfd = -1;
	int32_t cmd;
	int32_t sendlen;
	fd_set fds;
	pid_t pid;        //+新追加
	
	socklnk_t *sdlnk = NULL;
	socklnk_t *this;
	
	
	buf = (int8_t*)malloc(BUFFSIZE);
	if(buf == NULL){
		LOG_MSG("Service Start Failed");
		goto func_end;		
	}
	
	sdlnk = start_serice(PORT_NUM);
	if(sdlnk == NULL){
		LOG_MSG("Service Start Failed");
		goto func_end;
	}
	
	//main Loop
	for(;;){
		FD_ZERO(&fds);
		for(this = sdlnk; this != NULL; this = this->next){
			FD_SET(this->sd, &fds);
			if(this->sd > maxfd){
				maxfd = this->sd;
			}
		}
		
		if(select(maxfd + 1, &fds, NULL, NULL, NULL) > 0){
			for(this = sdlnk; this != NULL; this = this->next){
				if(FD_ISSET(this->sd, &fds) != 0){
					if((connfd = accept(this->sd, NULL, NULL)) < 0){
						continue;
					}
					break;
				}
			}
		}else{
			goto func_end;
		}
	
		if(this == NULL){
			goto func_end;
		}
		
		
		//+新追加。。
		pid = fork();
		if(pid < 0){
			close(connfd);
			goto func_end;
		}else if(pid > 0){
			close(connfd);
			waitpid(-1, NULL, WNOHANG);
		}else{
			for(;;){
				(void)memset(buf, 0x00, BUFFSIZE);
			
				if(read(connfd, buf, BUFFSIZE) > 0){
					if(strncmp(buf, CMD_ONE, strlen(CMD_ONE)) == 0){
						cmd = CMD_CODE_ONE;
					}else if(strncmp(buf, CMD_TWO, strlen(CMD_TWO)) == 0){
						cmd = CMD_CODE_TWO;
					}else if(strncmp(buf, CMD_THREE, strlen(CMD_THREE)) == 0){
						cmd = CMD_CODE_THREE;
					}else{
						cmd = CMD_CODE_ERR;
					}
				
					sendlen = strlen(server_msg[cmd]);
				
					if(write(connfd, server_msg[cmd], sendlen) != sendlen){
						close(connfd);
						goto func_end;
					}
				
					if(cmd == CMD_CODE_THREE){
						break;
					}
				}else{
					break;
				}
			}
		
			close(connfd);
			goto func_end;
		
		}	
	}
	
	
func_end:
	if(sdlnk != NULL){
		free_soclnk(sdlnk);
	}
	
	if(buf != NULL){
		free(buf);
	}
	
	return 0;
}

哦,谢天谢地,服务 可以响应多个  客户端啦~

但是 ,又出现了一个新的问题。。 竟然 产生了 Defunct进程。。。

这是怎么回事?。。

理由是 子进程 结束后 会想 父进程 发送一个SIGCHLD的信号。默认对这个信号的处理是 忽略。

而在上面的代码中,waitpid用了WNOHANG参数。也就是不等待 子进程结束。因此子进程 结束了 ,就没有对他做处理了。

下面的 代码 则是改进了 ,已经不会产生 Defunct了。

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <sys/socket.h>
#include <sys/types.h>
#include <netdb.h>
#include <signal.h>

#define PORT_NUM	(2227)		//定义端口号。
#define BUFFSIZE	(1024)

#define	CMD_ONE			("1")
#define CMD_TWO			("2")
#define CMD_THREE		("3")

#define CMD_CODE_ERR	(0)
#define CMD_CODE_ONE	(1)
#define CMD_CODE_TWO	(2)
#define CMD_CODE_THREE	(3)

#define LISTEN_QUE		(10)

#define LOG_MSG(format, ...)	do{fprintf(stderr, "[MessageInfo]: "format" \n", ##__VA_ARGS__);}while(0)

typedef struct socklnk_st{
	int32_t sd;
	struct socklnk_st *next;
}socklnk_t;


socklnk_t*
create_soclnk(int32_t data)
{
	socklnk_t *soclnk = NULL;
	
	soclnk = (socklnk_t *)malloc(sizeof(socklnk_t));
	if(soclnk != NULL){
		soclnk->next = NULL;
		soclnk->sd = data;
	}
	
	return soclnk;
}

void
free_soclnk(socklnk_t *dst)
{
	socklnk_t *tmp = dst;
	socklnk_t *next;

	while(tmp != NULL){
		next = tmp->next;
		close(tmp->sd);
		free(tmp);
		tmp = next;
	}
}

//定义每个Code返回值。
static int8_t *server_msg[] = {
	[CMD_CODE_ONE] = "Hello",
	[CMD_CODE_TWO] = "Good",
	[CMD_CODE_THREE] = "Bye",
	[CMD_CODE_ERR]	= "Error Command",
};

socklnk_t *
start_serice(uint16_t port)
{
	int8_t serv[NI_MAXSERV];
	int32_t sd;
	int32_t opt = 1;
	struct addrinfo *ai, *res = NULL;
	struct addrinfo hints;
	socklnk_t *head = NULL;
	socklnk_t *cur  = NULL;	
	
	if(snprintf(serv, NI_MAXSERV, "%u", port) < 0){
		goto func_end;
	}
	
	(void)memset(&hints, 0x00, sizeof hints);
	hints.ai_family = AF_UNSPEC;
	hints.ai_socktype = SOCK_STREAM;
	hints.ai_flags = AI_PASSIVE;
	
	if(getaddrinfo(NULL, serv, &hints, &res) != 0){
		goto func_end;
	}
	
	for(ai = res; ai != NULL; ai = ai->ai_next){
		sd = socket(ai->ai_family, ai->ai_socktype, ai->ai_protocol);
		if(sd < 0){
			continue;
		}
		
		if(setsockopt(sd, SOL_SOCKET, SO_REUSEADDR, &opt, sizeof opt) < 0){
			close(sd);
			free_soclnk(head);
			goto func_end;
		}
		
		if(bind(sd, ai->ai_addr, ai->ai_addrlen) < 0){
			close(sd);
			continue;
		}
		
		if(listen(sd, LISTEN_QUE) < 0){
			close(sd);
			continue;
		}
		
		if(head == NULL){
			head = create_soclnk(sd);
			if(head == NULL){
				close(sd);
				free_soclnk(head);
				goto func_end;			
			}
			cur = head;
		}else{
			cur->next = create_soclnk(sd);
			if(cur->next == NULL){
				close(sd);
				free_soclnk(head);
				goto func_end;			
			}
			cur = cur->next;
		}
	}
	
func_end:
	if(res != NULL){
		freeaddrinfo(res);
	}

	return head;
}


void
sig_child(int32_t signo)
{
	for(;;){
		if(waitpid(-1, NULL, WNOHANG) > 0){
			break;
		}
	}
}



int
main(int ac, char **av)
{
	int8_t *buf;
	int32_t connfd;
	int32_t maxfd = -1;
	int32_t cmd;
	int32_t sendlen;
	fd_set fds;
	pid_t pid;
	
	socklnk_t *sdlnk = NULL;
	socklnk_t *this;
	
	
	buf = (int8_t*)malloc(BUFFSIZE);
	if(buf == NULL){
		LOG_MSG("Service Start Failed");
		goto func_end;		
	}
	
	sdlnk = start_serice(PORT_NUM);
	if(sdlnk == NULL){
		LOG_MSG("Service Start Failed");
		goto func_end;
	}
	
	signal(SIGCHLD, sig_child);
	
	//main Loop
	for(;;){
		FD_ZERO(&fds);
		for(this = sdlnk; this != NULL; this = this->next){
			FD_SET(this->sd, &fds);
			if(this->sd > maxfd){
				maxfd = this->sd;
			}
		}
		
		if(select(maxfd + 1, &fds, NULL, NULL, NULL) > 0){
			for(this = sdlnk; this != NULL; this = this->next){
				if(FD_ISSET(this->sd, &fds) != 0){
					if((connfd = accept(this->sd, NULL, NULL)) < 0){
						continue;
					}
					break;
				}
			}
		}else{
			goto func_end;
		}
	
		if(this == NULL){
			goto func_end;
		}
		
		pid = fork();
		if(pid < 0){
			close(connfd);
			goto func_end;
		}else if(pid > 0){
			close(connfd);
			
		}else{
			for(;;){
				(void)memset(buf, 0x00, BUFFSIZE);
			
				if(read(connfd, buf, BUFFSIZE) > 0){
					if(strncmp(buf, CMD_ONE, strlen(CMD_ONE)) == 0){
						cmd = CMD_CODE_ONE;
					}else if(strncmp(buf, CMD_TWO, strlen(CMD_TWO)) == 0){
						cmd = CMD_CODE_TWO;
					}else if(strncmp(buf, CMD_THREE, strlen(CMD_THREE)) == 0){
						cmd = CMD_CODE_THREE;
					}else{
						cmd = CMD_CODE_ERR;
					}
				
					sendlen = strlen(server_msg[cmd]);
				
					if(write(connfd, server_msg[cmd], sendlen) != sendlen){
						close(connfd);
						goto func_end;
					}
				
					if(cmd == CMD_CODE_THREE){
						break;
					}
				}else{
					break;
				}
			}
		
			close(connfd);
			goto func_end;
		
		}	
	}
	
	
func_end:
	if(sdlnk != NULL){
		free_soclnk(sdlnk);
	}
	
	if(buf != NULL){
		free(buf);
	}
	
	return 0;
}


共有 人打赏支持
粉丝 2
博文 15
码字总数 5935
×
庸俗武士
如果觉得我的文章对您有用,请随意打赏。您的支持将鼓励我继续创作!
* 金额(元)
¥1 ¥5 ¥10 ¥20 其他金额
打赏人
留言
* 支付类型
微信扫码支付
打赏金额:
已支付成功
打赏金额: