网络编程（day5）

 select的TCP服务器端

#include <myhead.h>

#define ERR_MSG(msg) do{\
	fprintf(stderr,"__%d__:",__LINE__);\
	perror(msg);\
}while(0)

#define PORT 8888
#define IP "192.168.114.41"

int main(int argc, const char *argv[])
{
	//创建一个流式套接字
	int sfd=socket(AF_INET,SOCK_STREAM,0);
	if(-1 == sfd){
		ERR_MSG("socket");
		return -1;
	}
	printf("socket success sfd:%d\n",sfd);

	//允许端口快速重用
	int reuse=1;
	if(setsockopt(sfd,SOL_SOCKET,SO_REUSEADDR,&reuse,sizeof(reuse))){
		ERR_MSG("setsockopt");
		return -1;
	}

	//定义一个地址信息结构体，供bind函数使用
	struct sockaddr_in sin;
	sin.sin_family 		= AF_INET;
	sin.sin_port 		= htons(PORT);
	sin.sin_addr.s_addr = inet_addr(IP);

	//将套接字与IP、端口绑定
	if(bind(sfd,(struct sockaddr*)&sin,sizeof(sin)) == -1){
		ERR_MSG("bind");
		return -1;
	}
	printf("bind success\n");

	//将套接字状态设置为被动监听状态
	if(listen(sfd,128) == -1){
		ERR_MSG("listen");
		return -1;
	}

	int newfd=-1;

	fd_set readfds,tempfds;
	
	FD_ZERO(&readfds);
	FD_ZERO(&tempfds);
	
	FD_SET(0,&readfds);
	FD_SET(sfd,&readfds);

	int maxfd=sfd;
	struct sockaddr_in saveCin[1024];

	int s_res=0;
	ssize_t res=0;

	char buf[128]="";

	struct sockaddr_in cin;
	socklen_t addrlen = sizeof(cin);

	while(1){
		tempfds = readfds;

		s_res = select(maxfd+1,&tempfds,NULL,NULL,NULL);
		if(-1 == s_res){
			ERR_MSG("select");
			return -1;
		}
		else if(0 == s_res){
			printf("time out...\n");
			return -1;
		}

		for(int i=0;i<=maxfd;i++){
			if(!FD_ISSET(i,&tempfds)){
				continue;
			}
			if(0 == i){
				printf("触发键盘输入事件>>>\n");
				bzero(buf,sizeof(buf));
				fgets(buf,sizeof(buf),stdin);
				buf[strlen(buf)-1]=0;
				printf(":%s\n",buf);
			}
			else if(sfd == i){
				printf("触发客户端连接事件>>>\n");
				if((newfd=accept(sfd,(struct sockaddr*)&cin,&addrlen)) == -1){
					ERR_MSG("accept");
					return -1;
				}
				printf("[%s:%d] newfd=%d 连接成功\n",inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),newfd);

				saveCin[newfd]=cin;

				FD_SET(newfd,&readfds);
				maxfd = maxfd>newfd ? maxfd : newfd;
			}
			else{
				printf("触发客户端交互事件>>>\n");
				bzero(buf,sizeof(buf));

				res = recv(i,buf,sizeof(buf),0);
				if(-1 == res){
					ERR_MSG("recv");
					return -1;
				}
				else if(0 == res){
					printf("[%s:%d] newfd=%d 下线了\n",inet_ntoa(saveCin[i].sin_addr),ntohs(saveCin[i].sin_port),i);

					close(i);

					FD_CLR(i,&readfds);

					while(FD_ISSET(maxfd,&readfds) == 0 && maxfd-->=0);
					
					continue;
				}
				printf("[%s:%d] newfd=%d : %s\n",inet_ntoa(saveCin[i].sin_addr),ntohs(saveCin[i].sin_port),i,buf);

				strcat(buf,"*-*");
				if(send(i,buf,sizeof(buf),0) == -1){
					ERR_MSG("send");
					return -1;
				}
				printf("发送成功\n");
			}
		}

	}
	close(sfd);
	return 0;
}

 select的TCP客户端

#include <myhead.h>

#define ERR_MSG(msg) do{\
	fprintf(stderr,"__%d__:",__LINE__);\
	perror(msg);\
}while(0)

#define PORT 8888
#define IP "192.168.114.41"

int main(int argc, const char *argv[])
{
	//创建一个流式套接字
	int cfd=socket(AF_INET,SOCK_STREAM,0);
	if(-1 == cfd){
		ERR_MSG("socket");
		return -1;
	}
	printf("socket success cfd:%d\n",cfd);

	//允许端口快速重用
	int reuse=1;
	if(setsockopt(cfd,SOL_SOCKET,SO_REUSEADDR,&reuse,sizeof(reuse))){
		ERR_MSG("setsockopt");
		return -1;
	}

	//定义一个地址信息结构体，供bind函数使用
	struct sockaddr_in sin;
	sin.sin_family 		= AF_INET;
	sin.sin_port 		= htons(PORT);
	sin.sin_addr.s_addr = inet_addr(IP);

	//连接服务器
	if(connect(cfd,(struct sockaddr*)&sin,sizeof(sin)) == -1){
		ERR_MSG("connect");
		return -1;
	}
	printf("connect success\n");

	fd_set readfds,tempfds;
	
	FD_ZERO(&readfds);
	FD_ZERO(&tempfds);
	
	FD_SET(0,&readfds);
	FD_SET(cfd,&readfds);

	int s_res=0;
	ssize_t res=0;

	char buf[128]="";


	while(1){
		tempfds=readfds;
		s_res = select(cfd+1,&tempfds,NULL,NULL,NULL);
		if(-1 == s_res){
			ERR_MSG("select");
			return -1;
		}
		else if(0 == s_res){
			printf("time out...\n");
			return -1;
		}

		for(int i=0;i<=cfd;i++){
			if(!FD_ISSET(i,&tempfds)){
				continue;
			}
			if(0 == i){
				printf("触发键盘输入事件>>>\n");
				bzero(buf,sizeof(buf));
				fgets(buf,sizeof(buf),stdin);
				buf[strlen(buf)-1]=0;
				if(send(cfd,buf,sizeof(buf),0) == -1){
					ERR_MSG("send");
					return -1;
				}
				printf("send success\n");
			}
			else{
				printf("触发服务端交互事件>>>\n");
				bzero(buf,sizeof(buf));

				res = recv(cfd,buf,sizeof(buf),0);
				if(-1 == res){
					ERR_MSG("recv");
					return -1;
				}
				else if(0 == res){
					printf("cfd=%d 服务器下线了\n",cfd);
					return -1;
				}
				printf("cfd=%d buf=%s\n",cfd,buf);

			}
		}

	}
	close(cfd);
	return 0;
}