TCP/IP linux下多线程通信(测试成功,客户端的接收和发送分别在两个线程)

经过多次尝试,终于测试成功linux环境下TCP/IP的多线程通信,与网上大多数代码的区别在于,网上的很多代码都是把发送或者接收的功能之一放在主线程中,这样就导致如果要在主线程撰写一些具体的操作就会造成卡死,或者响应不及时。查阅了资料,用了合适的方法把接收和发送分离在两个子线程。难点在于退出的时候如何结束线程,方法是在发送的线程里,接收到结束命令时,先把接收的线程结束掉pthread_cancle(),再自己结束掉线程pthread_exit(),然后在主线程里面通过pthread_join()等待子线程结束的命令,运行效果良好,现分享给大家。注意,服务器这里的接收和发送其中一个功能仍然在主线程里面,因为本人在服务器的操作命令不多,所以没有加以修改,如果有人需要服务器进行多种操作,也可以仿照client.cpp的形式修改,欢迎留言讨论。
client.cpp

#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include  //atoi()
#include 
#include 
#include 
#include 
using std::string;
using std::cout;
static pthread_t send_thread;
//客户端的发送功能要放在子线程里面
void * recv_msg(void *arg);//接收消息函数声明
void * send_msg(void *arg);
std::vector split(std::string str, std::string pattern);
struct myData
{
	float x[10];
	float y[10];
	float z[10];
	float w[10];
	
};
	static struct myData mData; 
int main(int argc, char *argv[])
{
	//判断参数个数是否匹配

	int port = 7777;//从命令行接收参数
	if( port<1025 || port>65535 )//0~1024一般给系统使用,一共可以分配到65535
	{
		printf("端口号范围应为1025~65535");
		return -1;
	}
 
	//1 创建tcp通信socket
	int socket_fd = socket(AF_INET, SOCK_STREAM, 0);
	if(socket_fd == -1)
	{
		perror("socket failed!\n");
	}
 
	//2 连接服务器
	struct sockaddr_in server_addr = {0};//服务器的地址信息
	server_addr.sin_family = AF_INET;//IPv4协议
	server_addr.sin_port = htons(port);//服务器端口号
	server_addr.sin_addr.s_addr = inet_addr("10.40.128.128");//设置服务器IP
	int ret = connect(socket_fd, (struct sockaddr *)&server_addr, sizeof(server_addr));
	if(ret == -1)
	{
		perror("connect failed!\n");
	}
	else
	{
		printf("connect server successful!\n");
	}
 
	//开启接收线程
	pthread_t recv_thread;//存放线程id       recv_msg:线程执行的函数,将通信socket:new_socket_fd传递进去
	ret = pthread_create(&recv_thread, NULL, recv_msg, (void*)&socket_fd);
	if(ret != 0)
	{
		printf("开启接收线程失败\n");
	}
 	ret= pthread_create(&send_thread,NULL,send_msg,(void*)&socket_fd);
	if(ret != 0)
	{
		printf("开启发送线程失败\n");
	}
	//3 发送消息
/*	while(1)
	{
		char buf[1024] = {0};//存放要发送的消息
		strcpy(buf,"0.5;-0.5;1.0;1.0;-2;0.5;1;7;8;9;10;11;-1;-2;-3;");
		write(socket_fd,buf,strlen(buf));
		if(strcmp(buf, "exit") == 0 || strcmp(buf, "") == 0)
		{
			ret = pthread_cancel(recv_thread);//取消线程
			
			break;
		}
	}
*/
 	pthread_join(send_thread,NULL);
	//pthread_join(send_thread,NULL);
	//4 关闭通信socket
	close(socket_fd);
 
	return 0;
}
 
void * send_msg(void *arg)
{
	char buf[1024] = {0};
	strcpy(buf,"0.5;-0.5;1.0;1.0;-2;0.5;1;7;8;9;10;11;-1;-2;-3;");
	int *socket_fd=(int *)arg;
	while(1)
	{
		//写入节点的处理语句
		sleep(5);
		write(*socket_fd,buf,strlen(buf));
		/*
		if(strcmp(buf, "exit") == 0 || strcmp(buf, "") == 0)
		{
			break;//退出
		}
		*/
	}

	pthread_exit(NULL); 
}
//接收线程所要执行的函数 接收消息
void * recv_msg(void *arg)
{
	char buf[1024]={0};
	int *socket_fd=(int *)arg;
	

	std::string myString(buf,1024);
	std::vector v;
	std::string pattern=";";

	while(1)
	{
		
		read(*socket_fd, buf, sizeof(buf));//阻塞,等待接收消息
		//printf("receive msg:%s\n", buf);

		myString.assign(buf,1024);
		//int sign=1;
		v=split(myString,pattern);
		/*std::cout<x[0]< split(std::string str, std::string pattern)
{
    std::string::size_type pos;
    std::vector result;

    str += pattern;//扩展字符串以方便操作
    int size = str.size();

    for (int i = 0; i

server.cpp

#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include  //atoi()
#include 
 
void * recv_msg(void *arg);//接收消息函数声明
 
int main(int argc, char *argv[])
{

	//从命令行获取端口号
	int port = 7777;
	if( port<1025 || port>65535 )//0~1024一般给系统使用,一共可以分配到65535
	{
		printf("端口号范围应为1025~65535");
		return -1;
	}
	
	//1 创建tcp通信socket
	int socket_fd = socket(AF_INET, SOCK_STREAM, 0);
	if(socket_fd == -1)
	{
		perror("socket failed!\n");
		return -1;
	}
 
	//2 绑定socket地址
	struct sockaddr_in server_addr = {0};
	server_addr.sin_family = AF_INET;//AF_INET->IPv4  
	server_addr.sin_port = htons(port);// server port
	server_addr.sin_addr.s_addr = INADDR_ANY;//server ip (auto set by system)
	int ret = bind(socket_fd, (struct sockaddr *) &server_addr, sizeof(server_addr) );
	if(ret == -1)
	{
		perror("bind failed!\n");
		return -1;
	}
 
	//3 设置监听队列,设置为可以同时连接5个客户端
	ret = listen(socket_fd, 5);
	if(ret == -1)
	{
		perror("listen falied!\n");
		return -1;
	}
 
 
	printf("server is running!\n");
 
	struct sockaddr_in client_addr = {0};//用来存放客户端的地址信息
	socklen_t len = sizeof(client_addr);
	int new_socket_fd = -1;//存放与客户端的通信socket
	
	//4 等待客户端连接
	new_socket_fd = accept( socket_fd, (struct sockaddr *)&client_addr, &len);
	if(new_socket_fd == -1)
	{
		perror("accpet error!\n");
	}
	else
	{
		printf("IP:%s, PORT:%d [connected]\n", inet_ntoa(client_addr.sin_addr), ntohs(client_addr.sin_port));
	}
 
	//开启接收线程
	pthread_t recv_thread;//存放线程id       recv_msg:线程执行的函数,将通信socket:new_socket_fd传递进去
	ret = pthread_create(&recv_thread, NULL, recv_msg, (void*)&new_socket_fd);
	if(ret != 0)
	{
		printf("开启线程失败\n");
	}
	
	while(1)
	{
		char buf[1024] = {0};
		strcpy(buf,"0.5;-0.5;1.0;1.0;-2;0.5;1;1.0;8;9;10;11;-1;-2;-3;");
		write(new_socket_fd, buf, sizeof(buf));//发送消息
		
		if(strcmp(buf, "exit") == 0 || strcmp(buf, "") == 0)//退出
		{
			ret = pthread_cancel(recv_thread);//取消线程
			
			break;
		}
	}
 
	//5 关闭通信socket
	close(new_socket_fd);
	close(socket_fd);
 
	return 0;
}
 
//接收线程所要执行的函数 接收消息
void * recv_msg(void *arg)
{
	int *socket_fd = (int *)arg;//通信的socket
	while(1)
	{
		char buf[1024] = {0};
		read(*socket_fd, buf, sizeof(buf));//阻塞,等待接收消息
		printf("receive msg:%s\n", buf);
		if(strncmp(buf, "exit", 4) == 0 || strcmp(buf, "") == 0)
		{
			//通知主线程。。。
			
			break;//退出
		}
	}
	return NULL;
}


你可能感兴趣的:(TCP/IP linux下多线程通信(测试成功,客户端的接收和发送分别在两个线程))