c++在Linux环境下的套接字Tcp通信例子（demo）

demo包括服务端和客户端的通信，发送端发送格式为先发送长度为5的字符串数据，告知对方接下来的数据长度，接收端首先接收到消息长度，再根据消息长度接受接下来的消息。

服务端：

#include <unistd.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <iostream>
#include <thread>
#include <string.h>
#include "mds_sync_tcp.h"

//客户端套接字
int clientA_sockfd;

/*
发送字符串消息
*/
void ServerA_send(std::string *send_str) {
	//计算待发送字符串长度
	std::string send_len = "00000" + std::to_string((*send_str).length());
	send_len = send_len.substr(send_len.length() - 5, 5);

	//发送数据
	send(clientA_sockfd, send_len.data(), 5, 0);
	send(clientA_sockfd, (*send_str).data(), (*send_str).length(), 0);
	//std::cout << send_str << std::endl;
}

/*
接收客户端订阅消息
*/
void ServerA_recv() {
	/*收到消息*/
	char recv_len[6];
	char recv_str[1024];
	//请求数据返回
	std::string select_return;

	//初始化字符串
	memset(recv_len, 0, 6);
	memset(recv_str, 0, 1024);
	while (1) {
		ssize_t len = recv(clientA_sockfd, recv_len, 5, 0); //扫描消息通道
		if (len > 0) {
			//接收到长度消息
			if (recv_len[0] == '0') {
				recv(clientA_sockfd, recv_str, atoi(recv_len), 0);

				//std::cout << recv_str << std::endl;
				//------------------------ 收到消息：recv_str --------------------------
				//if (recv_str[0] == '{') {
				//	json_select(recv_str, &select_return); //查询所需数据
				//	ServerA_send(&select_return); //返回数据
				//}
				//else {
				//	sub_object(recv_str); //订阅
				//}
				//----------------------------------------------------------------------

				memset(recv_str, 0, 1024);	//字符串初始化
			}
			memset(recv_len, 0, 6);
		}
		else {
			printf("ServerA Client has exited\n");
			return;
		}
	}
}

/*
创建用于监听的套接字，等待客户端连接
*/
void ServerA_Main() {
	int server_sockfd;//服务器端套接字
	struct sockaddr_in my_addr;   //服务器网络地址结构体
	struct sockaddr_in remote_addr; //客户端网络地址结构体
	socklen_t sin_size;
	memset(&my_addr, 0, sizeof(my_addr)); //数据初始化--清零
	my_addr.sin_family = AF_INET; //设置为IP通信
	my_addr.sin_addr.s_addr = INADDR_ANY;//服务器IP地址--允许连接到所有本地地址上
	my_addr.sin_port = htons(5097); //服务器端口号

	/*创建服务器端套接字--IPv4协议，面向连接通信，TCP协议*/
	if ((server_sockfd = socket(PF_INET, SOCK_STREAM, 0)) < 0) {
		perror("socket error");
		return;
	}

	/*将套接字绑定到服务器的网络地址上*/
	if (bind(server_sockfd, (struct sockaddr *)&my_addr, sizeof(struct sockaddr)) < 0) {
		perror("bind error");
		return;
	}

	/*监听连接请求--监听队列长度为1*/
	if (listen(server_sockfd, 1) < 0) {
		perror("listen error");
		return;
	};

	sin_size = sizeof(struct sockaddr_in);
	while (1) {
		/*等待客户端连接请求到达*/
		if ((clientA_sockfd = accept(server_sockfd, (struct sockaddr *)&remote_addr, &sin_size)) < 0) {
			perror("accept error");
			return;
		}
		printf("ServerA connects to the client through port 5097\n"); //已连接客户

		//接收通道
		std::thread ServerA_recv_thread(ServerA_recv);
		//等待通道结束
		ServerA_recv_thread.join();
	}

	/*关闭套接字*/
	close(clientA_sockfd);
	close(server_sockfd);
}

客户端：

#define _WINSOCK_DEPRECATED_NO_WARNINGS

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

//客户端套接字
int mds_sockClient;

/*
发送订阅消息
*/
void send_mds(std::string send_str) {
	std::string send_len = "00000" + std::to_string(send_str.length());
	send_len = send_len.substr(send_len.length() - 5, 5);

	send(mds_sockClient, send_len.data(), 5, 0);
	send(mds_sockClient, send_str.data(), send_str.length(), 0);
	//std::cout << send_str << std::endl;
}

/*
接收订阅消息
*/
void recv_mds() {
	char recv_len[6];
	char recv_str[4096]; //数据传送的缓冲区

	memset(recv_len, 0, 6);
	memset(recv_str, 0, 4096); //初始化字符串

	while (1) {
		ssize_t len = recv(mds_sockClient, recv_len, 5, 0);
		if (len > 0) {
			if (recv_len[0] == '0') {
				recv(mds_sockClient, recv_str, atoi(recv_len), 0);

				//std::cout << recv_str << std::endl;
				//------------------------ 收到消息：recv_str --------------------------
				//json_to_data(recv_str);

				//----------------------------------------------------------------------

				memset(recv_str, 0, 4096); //初始化字符串
			}
			memset(recv_len, 0, 6); //初始化字符串
		}
		else {
			return;
		}
	}
}

/*
初始化函数（只能运行一次）
input:
1. port 套接字端口号
instructions：
*/
void Mds_Initialize(uint16_t port) {
	struct sockaddr_in remote_addr; //服务器端网络地址结构体
	memset(&remote_addr, 0, sizeof(remote_addr)); //数据初始化--清零
	remote_addr.sin_family = AF_INET; //设置为IP通信
	remote_addr.sin_addr.s_addr = inet_addr("127.0.0.1");//服务器IP地址
	remote_addr.sin_port = htons(port); //服务器端口号

	/*创建客户端套接字--IPv4协议，面向连接通信，TCP协议*/
	if ((mds_sockClient = socket(PF_INET, SOCK_STREAM, 0)) < 0)
	{
		perror("socket error");
		return;
	}

	/*将套接字绑定到服务器的网络地址上*/
	if (connect(mds_sockClient, (struct sockaddr *)&remote_addr, sizeof(struct sockaddr)) < 0)
	{
		perror("connect error");
		return;
	}
	printf("Connect to the server......\n"); //已连接服务端
	//接收数据
	recv_mds();

	/*关闭套接字*/
	close(mds_sockClient);
}