实现服务器与客户端之间的实时通信

服务器程序：

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

#define ERR_MSG(msg) do{\
	fprintf(stderr,"line:%d\n",__LINE__);\
	perror(msg);\
}while(0)
#define IP "192.168.8.77"
#define PQRT 5555

pthread_t pid_r,pid_w;
char buf[128]="";
void*receve_write(void* arg_w)
{
	int newfd=*((int*)arg_w);
	while(1)
	{
		//服务器发送
		bzero(buf,sizeof(buf));
		fgets(buf,sizeof(buf),stdin);
		buf[strlen(buf)-1]='\0';
		if(strcmp(buf,"quit")==0)
		{
			break;
		}
		printf("服务器发送>>>%s\n",buf);
		if(send(newfd,buf,sizeof*(buf),0)<0)
		{
			ERR_MSG("send");
			return NULL;
		}
	}
	pthread_cancel(pid_r);
	pthread_exit(NULL);
}
void*receve_read(void* arg_r)
{
	ssize_t res=0;
	int newfd=*((int*)arg_r);
	while(1)
	{
		//服务器接收
		bzero(buf,sizeof(buf));
		res=recv(newfd,buf,sizeof(buf),0);
		if(res<0)
		{
			ERR_MSG("recv");
			return NULL;
		}
		else if(res==0)
		{
			break;
		}
		printf("服务器接收>>>%s\n",buf);
	}
	pthread_cancel(pid_w);
	pthread_exit(NULL);
}
int main(int argc, const char *argv[])
{
	//创建流式套接字
	int sfd = socket(AF_INET,SOCK_STREAM,0);
	if(sfd<0)
	{
		ERR_MSG("socket");
		return -1;
	
	}
	//允许端口快速重用
	int reuse=0;
	//填充地址信息结构体
	//根据地址族的不同,地址信息结构体不同
	struct sockaddr_in sin;
	sin.sin_family=AF_INET; //ipv4
	sin.sin_port=htons(PQRT); //端口号网络字节序
	sin.sin_addr.s_addr=inet_addr(IP);//本机IP地址的网络字节序
	//绑定服务器的IP和端口
	if(bind(sfd,(struct sockaddr*)&sin,sizeof(sin))<0)
	{
		ERR_MSG("bind");
		return -1;
	
	}
	printf("bind success\n");
	//将套接字设置为被动监听状态
	if(listen(sfd,128)<0)
	{
		ERR_MSG("listen");
		return -1;
	}
	printf("listen success\n");
	//从已经完成连接的队列中获取一个客户端信息,生成一个新的文件描述符
	//生成的新的文件描述符才是与客户端通信的文件描述符
	int newfd=0;
	newfd=accept(sfd,NULL,NULL);
	if(newfd<0)
	{
		ERR_MSG("accept");
		return -1;
	
	}
	printf("newfd=%d\n",newfd);
	char buf[128]="";
	ssize_t res=0;
	//创建多线程
	if(pthread_create(&pid_w,NULL,receve_write,&newfd)!=0)
	{
		fprintf(stderr,"pthread_create failed\n");
		return -1;
	}
	if(pthread_create(&pid_r,NULL,receve_read,&newfd)!=0)
	{
		fprintf(stderr,"pthread_create failed\n");
		return -1;
	}

	pthread_join(pid_r,NULL);
	pthread_join(pid_w,NULL);

	close(sfd);
	close(newfd);

	return 0;
}

客户端程序：

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

#define ERR_MSG(msg) do{\
	fprintf(stderr,"line:%d\n",__LINE__);\
	perror(msg);\
}while(0)
#define IP "192.168.8.77"
#define PQRT 5555

pthread_t pid_w,pid_r;
char buf[128]="";
void *client_read(void* arg_r)
{
	ssize_t res=0;
	int sfd=*((int*)arg_r);
	while(1)
	{
		//接收
		bzero(buf,sizeof(buf));
		res=recv(sfd,buf,sizeof(buf),0);
		if(res<0)
		{
			ERR_MSG("recv");
			return NULL;
		}
		else if(res==0)
		{
			break;
		}
		printf("客户端接收>>>%s\n",buf);
	}
	pthread_cancel(pid_w);
	pthread_exit(NULL);
}
void *client_write(void* arg_w)
{
	ssize_t res=0;
	int sfd=*((int*)arg_w);
	while(1)
	{
		//发送
		bzero(buf,sizeof(buf));
		fgets(buf,sizeof(buf),stdin);
		buf[strlen(buf)-1]='\0';
		if(strcmp(buf,"quit")==0)
		{
			break;
		}
		printf("客户端发送>>>%s\n",buf);
		if(send(sfd,buf,sizeof(buf),0)<0)
		{
			ERR_MSG("send");
			return NULL;
		
		}
	}
	pthread_cancel(pid_r);
	pthread_exit(NULL);
}
int main(int argc, const char *argv[])
{
	//创建流式套接字
	int sfd = socket(AF_INET,SOCK_STREAM,0);
	if(sfd<0)
	{
		ERR_MSG("socket");
		return -1;
	
	}
	//允许端口快速重用
	int reuse=0;
	//填充地址信息结构体
	//根据地址族的不同,地址信息结构体不同
	struct sockaddr_in sin;
	sin.sin_family=AF_INET; //ipv4
	sin.sin_port=htons(PQRT); //端口号网络字节序
	sin.sin_addr.s_addr=inet_addr(IP);//本机IP地址的网络字节序
	//绑定服务器的IP和端口
	if(connect(sfd,(struct sockaddr*)&sin,sizeof(sin))<0)
	{
		ERR_MSG("connect");
		return -1;
	
	}
	printf("connect success\n");
		//从已经完成连接的队列中获取一个客户端信息,生成一个新的文件描述符
	//生成的新的文件描述符才是与客户端通信的文件描述符
	char buf[128]="";
	ssize_t res=0;
	
	//创建多线程用于收发
	if(pthread_create(&pid_r,NULL,client_read,&sfd)!=0)
	{
		fprintf(stderr,"pthread_create failed\n");
		return -1;
	}
	if(pthread_create(&pid_w,NULL,client_write,&sfd)!=0)
	{
		fprintf(stderr,"pthread_create failed\n");
		return -1;
	}
	pthread_join(pid_r,NULL);
	pthread_join(pid_w,NULL);
	close(sfd);
	
	return 0;
}

运行效果：