服务端模板:
#include <iostream>
#include <WinSock2.h>
#include <thread>
#include <string>
#include "mds_sync_tcp.h"
#pragma comment(lib, "ws2_32.lib")
/*套接字*/
SOCKET sockConnA;
/*
发送消息
*/
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(sockConnA, send_len.data(), 5, 0);
send(sockConnA, send_str.data(), send_str.length(), 0);
//std::cout << send_str << std::endl;
}
/*
接收客户端订阅消息
*/
SOCKET 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) {
int len = recv(sockConnA, recv_len, 5, 0); //扫描消息通道
if (len > 0) {
//接收到长度消息
if (recv_len[0] == '0') {
recv(sockConnA, 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 sockConnA;
}
}
}
/*
创建用于监听的套接字,等待客户端连接
*/
void ServerA_Main() {
WSADATA wsaData;
if (WSAStartup(MAKEWORD(2, 2), &wsaData) != 0) {
printf("ServerA Failed to load Winsock");
return;
}
//创建用于监听的套接字
SOCKET sockSrv = socket(AF_INET, SOCK_STREAM, 0);
SOCKADDR_IN addrSrv;
addrSrv.sin_family = AF_INET;
addrSrv.sin_port = htons(5097); //1024以上的端口号
addrSrv.sin_addr.S_un.S_addr = htonl(INADDR_ANY);
int retVal = bind(sockSrv, (LPSOCKADDR)&addrSrv, sizeof(SOCKADDR_IN));
if (retVal == SOCKET_ERROR) {
printf("ServerA Failed bind:%d\n", WSAGetLastError());
return;
}
//最大连接客户数
if (listen(sockSrv, 1) == SOCKET_ERROR) {
printf("ServerA Listen failed:%d", WSAGetLastError());
return;
}
SOCKADDR_IN addrClient;
int len = sizeof(SOCKADDR);
while (1) {
//等待客户请求到来
sockConnA = accept(sockSrv, (SOCKADDR *)&addrClient, &len);
if (sockConnA == SOCKET_ERROR) {
printf("ServerA Accept failed:%d", WSAGetLastError());
return;
}
printf("ServerA connects to the client through port 5097\n"); //已连接客户
//接收通道
std::thread ServerA_recv_thread(ServerA_recv);
//等待通道结束
ServerA_recv_thread.join();
}
closesocket(sockConnA);
closesocket(sockSrv);
WSACleanup();
}
客户端模板:
#define _WINSOCK_DEPRECATED_NO_WARNINGS
#include <WinSock2.h>
#include <stdio.h>
#include <iostream>
#include <thread>
#include <string>
#include "mds_json_to_data.h"
#pragma comment(lib, "ws2_32.lib")
SOCKET 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;
}
/*
接收消息
*/
SOCKET recv_mds() {
char recv_len[6];
char recv_str[4096];
memset(recv_len, 0, 6);
memset(recv_str, 0, 4096); //初始化字符串
while (1) {
int 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 --------------------------
//mds_json_to_data(recv_str);
//----------------------------------------------------------------------
memset(recv_str, 0, 4096); //初始化字符串
}
memset(recv_len, 0, 6); //初始化字符串
}
else {
return mds_sockClient;
}
}
}
/*
初始化函数(只能运行一次)
input:
1. port 套接字端口号
*/
void Mds_Initialize(int port) {
//加载套接字
WSADATA wsaData;
if (WSAStartup(MAKEWORD(2, 2), &wsaData) != 0) {
printf("Failed to load Winsock");
return;
}
SOCKADDR_IN addrSrv;
addrSrv.sin_family = AF_INET;
addrSrv.sin_port = htons(port); //端口号
addrSrv.sin_addr.S_un.S_addr = inet_addr("127.0.0.1"); //服务端IP
mds_sockClient = socket(AF_INET, SOCK_STREAM, 0); //套接字
if (SOCKET_ERROR == mds_sockClient) {
printf("Socket() error:%d", WSAGetLastError());
return;
}
//向服务器发出连接请求
if (connect(mds_sockClient, (struct sockaddr*)&addrSrv, sizeof(addrSrv)) == INVALID_SOCKET) {
printf("Connect failed:%d", WSAGetLastError());
return;
}
printf("Connect to the server......\n"); //已连接服务端
//接收数据
recv_mds();
//关闭套接字
closesocket(mds_sockClient);
WSACleanup();
}
本文提供了一套完整的TCP服务端与客户端通信模板代码,包括了服务端的消息发送与接收处理逻辑,以及客户端的连接建立与数据交互过程。
2045

被折叠的 条评论
为什么被折叠?



