TCP多线程并发服务器,select的TCP服务端,select的TCP客户端

1.TCP多线程并发服务器

#include <stdio.h>
#include <sys/socket.h>
#include <sys/types.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <string.h>
#include <unistd.h>
#include <pthread.h>

#define PORT 6666
#define IP "192.168.124.124"

#define ERR_MSG(err) do{\
	fprintf(stderr,"%d",__LINE__);\
	perror(err);\
}while(0)

struct msg
{
	int newsfd;
	struct sockaddr_in cin;
};



void* recv_send_pthread(void* arg);

int main(int argc, const char *argv[])
{
	//创建报式套接字
	int sfd=socket(AF_INET,SOCK_STREAM,0);
	if(sfd<0)
	{
		ERR_MSG("socket");
		return -1;
	}
	printf("报式套接字创建成功\n");

	//允许端口快速被复用
	int reuse=1;
	if(setsockopt(sfd,SOL_SOCKET,SO_REUSEADDR,&reuse,sizeof(reuse))<0)
	{
		ERR_MSG("setsockopt");
		return -1;
	}
	printf("允许端口快速被复用\n");


	//补参数
	struct sockaddr_in sin;
	sin.sin_family   =AF_INET;
	sin.sin_port   =htons(PORT);
	sin.sin_addr.s_addr=inet_addr(IP);

	//bind
	if(bind(sfd,(struct sockaddr*)&sin,sizeof(sin))<0)
	{
		ERR_MSG("bind");
		return -1;
	}
	printf("绑定成功\n");

	//将套接字文件设置为监听
	if(listen(sfd,128)<0)
	{
		ERR_MSG("listen");
		return -1;
	}
	printf("监听成功\n");

	struct sockaddr_in cin;    //获取客户端信息
	socklen_t size=sizeof(cin);  //客户端信息,真实结构体大小

	int newsfd;
	pthread_t tid;
	struct msg clinfo;
	while(1){
		//accept 阻塞函数
		//当客户端请求连接时,创建新的套接字文件
		newsfd= accept(sfd,(struct sockaddr*)&cin,&size);
		if(newsfd<0)
		{
			ERR_MSG("accept");
			return -1;
		}
		printf("[%s : %d] newsfd=%d 连接成功\n",inet_ntoa(cin.sin_addr),htons(cin.sin_port),newsfd);
		
		clinfo.newsfd=newsfd;
		clinfo.cin=cin;
		
		if(pthread_create(&tid,NULL,recv_send_pthread,(void*)&clinfo)!=0)
		{
			fprintf(stderr,"pthread_create failed __%d__",__LINE__);
			break;
		}
		
		pthread_detach(tid); //分离线程	
		
	}

	//关闭文件描述符
	if(close(sfd)<0)
	{
		ERR_MSG("close");
		return -1;
	}
	return 0;
}

void* recv_send_pthread(void* arg)
{

	int newsfd=((struct msg*)arg)->newsfd;
	struct sockaddr_in cin = ((struct msg*)arg)->cin;

	char buf[128]="";
	ssize_t res;
	while(1)
	{
		//接收数据
		bzero(buf,sizeof(buf));
		res=recv(newsfd,buf,sizeof(buf),0);
		if(res<0)
		{
			ERR_MSG("recv");
			break;
		}
		else if(0==res)
		{
			printf("[%s : %d]客户端关闭\n",inet_ntoa(cin.sin_addr),htons(cin.sin_port));
			break;
		}
		printf("客户端[%s : %d] 发送的数据:%s\n",inet_ntoa(cin.sin_addr),\
				htons(cin.sin_port),buf);

		//发送数据
		strcat(buf ,"*_**_*");
		if(send(newsfd,buf,sizeof(buf),0)<0)
		{
			ERR_MSG("send");
			break;
		}
		printf("[%s : %d]服务器发送数据成功\n",IP,PORT);	
	}

	close(newsfd);
	pthread_exit(NULL);
}



2.select的TCP服务端

#include <stdio.h>
#include <sys/socket.h>
#include <sys/types.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <string.h>
#include <unistd.h>
#include <sys/select.h>
#include <sys/time.h>

#define PORT 6666
#define IP "192.168.124.124"

#define ERR_MSG(err) do{\
	fprintf(stderr,"%d",__LINE__);\
	perror(err);\
}while(0)



int deal_key_send(fd_set readfds);
int deal_CliConnect(int sfd,struct sockaddr_in clinfo[],fd_set *preadfds,int *pmaxfd);

int deal_CliRecvSend(int newfd,struct sockaddr_in clinfo[],fd_set *preadfds,int *pmaxfd);

int main(int argc, const char *argv[])
{
	int sfd=socket(AF_INET,SOCK_STREAM,0);
	if(sfd<0)
	{
		ERR_MSG("socket");
		return -1;
	}
	printf("流式套接字创建成功\n");

	int reuse=1;
	if(setsockopt(sfd,SOL_SOCKET,SO_REUSEADDR,&reuse,sizeof(reuse))<0)
	{
		ERR_MSG("setsockopt");
		return -1;
	}
	printf("允许端口被快速复用\n");

	struct sockaddr_in sin;
	sin.sin_family=AF_INET;
	sin.sin_port=htons(PORT);
	sin.sin_addr.s_addr=inet_addr(IP);

	if(bind(sfd,(struct sockaddr*)&sin,sizeof(sin))<0)
	{
		ERR_MSG("bind");
		return -1;
	}
	printf("绑定成功\n");

	if(listen(sfd,128)<0)
	{
		ERR_MSG("listen");
		return -1;
	}
	printf("监听成功\n");

	//添加到select集合
	fd_set readfds,tempfds;
	FD_ZERO(&readfds);  //将集合清空
	FD_ZERO(&tempfds);

	FD_SET(0,&readfds);	
	FD_SET(sfd,&readfds);

	struct sockaddr_in clinfo[1024];//定义数组存储每个客户端的消息
	
	int maxfd=sfd;
	int num;
	while(1)
	{
		tempfds=readfds;//readfds不变,每次重新赋值给tempfds
						//
		num=select(maxfd+1,&tempfds,NULL,NULL,NULL);
		if(num<0)
		{
			ERR_MSG("select");
			return -1;
		}
		else if(num==0)
		{
			printf("timeout....\n");
			break;
		}

		//判断select中哪个资源准备就绪
		//准备就绪的资源可解除select阻塞,执行对应的功能
		for(int i=0;i<=maxfd;i++)
		{
			if(FD_ISSET(i,&tempfds)==0)
			{
				continue;//不在集合中,就不是外面监听的文件描述符
						//直接跳出,不做后续的判断
			}
			
			if(i==0)
			{
				printf("执行键盘输入事件,向客户端发送消息\n");
				deal_key_send(readfds);
			}
			else if(i==sfd)
			{
				printf("执行连接客户端事件\n");
				deal_CliConnect(sfd,clinfo,&readfds,&maxfd);	
			}
			else 
			{
				printf("执行与客户端的交互事件\n");
				deal_CliRecvSend(i,clinfo,&readfds,&maxfd);
			}
		}
	}
	
	if(close(sfd)<0)
	{
		ERR_MSG("close");
		return -1;
	}
	return 0;
}

//执行键盘输入事件,向客户端发送消息
int deal_key_send(fd_set readfds)
{
	char buf[128];
	int size;
	int newfd;
	size=scanf("%d %s",&newfd,buf);//服务端指明要向哪个服务器对应的文件描述符发送数据,发送数据内容
	while(getchar()!='\n');

	if(size!=2)
	{
		printf("输入数据格式错误,请正确输入格式:fd string\n");
		return -1;
	}

	if(newfd<=3 || FD_ISSET(newfd,&readfds)==0)
	{
		printf("请输入合法的文件描述符\n");
		return -1;
	}

	if(send(newfd,buf,sizeof(buf),0)<0)
	{
		ERR_MSG("send");
		return -1;
	}
	
	return 0;
}

//执行连接客户端事件
int deal_CliConnect(int sfd,struct sockaddr_in clinfo[],fd_set *preadfds,int *pmaxfd)
{
	int newfd;
	struct sockaddr_in cin;
	socklen_t addrlen=sizeof(cin);
	newfd=accept(sfd,(struct sockaddr*)&cin,&addrlen);
	if(newfd<0)
	{
		ERR_MSG("accept");
		return -1;
	}
	printf("[%s %d] newfd=%d客户端连接成功\n",inet_ntoa(cin.sin_addr),\
					ntohs(cin.sin_port),newfd);

	clinfo[newfd]=cin;//将获取到的客户端信息存储在对应文件描述符的下标中

	//将连接到的文件描述符添加到select的集合中
	FD_SET(newfd,preadfds);
	
	//更新maxfd
	*pmaxfd = *pmaxfd>newfd?*pmaxfd:newfd;

	return 0;
}


//执行与客户端的交互事件
int deal_CliRecvSend(int newfd,struct sockaddr_in clinfo[],fd_set *preadfds,int *pmaxfd)
{
	char buf[128]="";
	ssize_t res;
	//接收
	bzero(buf,sizeof(buf));
	res=recv(newfd,buf,sizeof(buf),0);
	if(res<0)
	{
		ERR_MSG("recv");
		return -1;
	}
	else if(0==res)
	{
		printf("[%s %d] newfd=%d客户端已下线\n",inet_ntoa(clinfo[newfd].sin_addr),\
				ntohs(clinfo[newfd].sin_port),newfd);
		close(newfd);//下线则关闭对应的文件描述符
		
		//将文件描述符从select集合中删除
		FD_CLR(newfd,preadfds);

		//更新maxfd
		//删除的文件描述符如果不是最大文件描述符,则不需要更新maxfd
		//
		//删除的文件描述符可能是最大文件描述符
		/*
		 *如果是最大文件描述符,则最大文件描述符不在集合中,
		 *依次判断最大文件描述符的前一位是否存在(如删除10,则依次判断9,8,7.....)
		 */
		while(FD_ISSET(*pmaxfd,preadfds)==0 && *pmaxfd-- >0);

		return 0;
	}
	printf("[%s %d] newfd=%d客户端发送的数据:%s\n",inet_ntoa(clinfo[newfd].sin_addr),\
				ntohs(clinfo[newfd].sin_port),newfd,buf);

/*	//发送数据
	strcat(buf,"*-*");
	if(send(newfd,buf,sizeof(buf),0)<0)
	{
		ERR_MSG("send");
		return -1;
	}
	printf("发送数据成功\n");
*/	
	return 0;

}

3.select的TCP客户端

#include <stdio.h>
#include <sys/socket.h>
#include <sys/types.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <string.h>
#include <unistd.h>
#include <sys/select.h>
#include <sys/time.h>

#define PORT 6666
#define IP "192.168.124.124"

#define ERR_MSG(err) do{\
	fprintf(stderr,"%d",__LINE__);\
	perror(err);\
}while(0);

int deal_Send(int cfd);

int deal_Recv(int cfd);

int main(int argc, const char *argv[])
{
	//创建套接字文件
	int cfd=socket(AF_INET,SOCK_STREAM,0);
	if(cfd<0)
	{
		ERR_MSG("socket");
		return -1;
	}
	printf("流式套接字文件创建成功\n");

	//绑定bind,非必须绑定
	

	//手动补充服务器的地址信息结构体
	//ip,端口号,为connect函数使用
	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))<0)
	{
		ERR_MSG("connect");
		return -1;
	}
	printf("连接服务器成功[%s : %d]\n",IP,PORT);


	//发送数据
	fd_set readfd,tempfd;
	FD_ZERO(&readfd);
	FD_ZERO(&tempfd);
	
	FD_SET(0,&readfd);
	FD_SET(cfd,&readfd);

	char buf[128]="";
	while(1)
	{
		tempfd=readfd;
		int num=select(cfd+1,&tempfd,NULL,NULL,NULL);
		if(num<0)
		{
			ERR_MSG("select");
			return -1;
		}
		else if(num==0)
		{
			printf("timeout.....\n");
			break;
		}
	
		if(FD_ISSET(0,&tempfd))		
		{
			printf("触发键盘输入事件,向服务器发送消息\n");
			deal_Send(cfd);
		}
		if(FD_ISSET(cfd,&tempfd))
		{
			printf("接收服务器的数据\n");
			deal_Recv(cfd);
		}
		

	}

	if(close(cfd)<0)
	{
		ERR_MSG("close");
		return -1;
	}
	return 0;
}


int deal_Recv(int cfd)
{
	char buf[128]="";
	bzero(buf,sizeof(buf));
	int	res=recv(cfd,buf,sizeof(buf),0);
	if(res<0)
	{
		ERR_MSG("recv");
		return -1;
	}else if(res==0)
	{
		printf("服务器[%s : %d]已关闭\n",IP,PORT);
		return 0;
	}
	printf("接收服务器[%s: %d]发送的数据:%s\n",IP,PORT,buf);
	return 0;
}


int deal_Send(int cfd)
{
	//发送数据	
	char buf[128]="";
	ssize_t res;
	scanf("%s",buf);
	while(getchar()!='\n');

	if(send(cfd,buf,sizeof(buf),0)<0)
	{
		ERR_MSG("send");
		return -1;
	}

	return 0;
}

  • 2
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值