当前位置：首页 > news >正文

C++ 代码实现socket 类使用TCP/IP进行通信（windows 系统）

news 来源：原创 2024/9/21 12:27:55

C++ 代码实现socket 类使用TCP/IP进行通信（windows 系统）

TCP客户端通信常规步骤：

1.初始换socket环境
2.socket()创建TCP套接字。
3.connect()建立到达服务器的连接。
4.与客户端进行通信，recv()/send()接受/发送信息，write()/read() 子进程写入、读取信息
5. close() 关闭客户连接。

具体代码实现如下：

/******************************************************@Copyright (c) 2024, GhY, All rights reserved.*@文件    MySocket.h*@描述    TCP Socket 类封装声明**@作者    GhY*@日期    2024年7月24日*@版本    v1.0.0*****************************************************/
#ifndef __MYSOCKET_H__
#define __MYSOCKET_H__
#include "PublicDefine.h"
#include "sigslot.h"
#include "CIniConfig.h"/**@描述:    Socket 类*@作者:    GhY*@日期:    2024/07/24*@历史:*/
class MySocket : public sigslot::has_slots<>
{
public:typedef sigslot::signal1< Tcp_SendData* > SelectEvent;SelectEvent OnSelectEvent;  // 消息事件
public:MySocket(std::string ip = LOCAL_HOST, int port = PORT, u_long uctl = 1);~MySocket();/**@desc       初始化服务端Ip，Port配置*@param:     ip  port*@return*@author     GhY*@date       2024/07/24*@version    v1.0.0*@history:*/void InitData(const std::string& ip, const int& port);/**@brief    与服务器进行连接*@author   GhY*@date     2024/07/24*/void ClientConnect();/**@brief    关闭socket*@author   GhY*@date     2024/07/24*/void Close();/**@desc       发送数据*@param:     sdata 待发送数据*@return*@author     GhY*@date       2024/07/24*@version    v1.0.0*@history:*/int SendData(const std::string& sdata);/**@brief    接收数据*@author   GhY*@date     2024/07/24*/int ReceiveData();private:/**@brief    初始化WINSOCK*@author   GhY*@date     2024/07/24*/void InitWinsock();private:std::string m_ip;int m_port;u_long m_uctl;  // 阻塞方式：1=非阻塞，0=阻塞 默认1SOCKET m_socket;
};#endif //!__MYSOCKET_H__

/****************************************************
*
*@Copyright (c) 2024, GhY, All rights reserved.
*@文件    MySocket.h
*@描述    TCP Socket 类封装实现
*
*@作者    GhY
*@日期    2024年7月24日
*@版本    v1.0.0
*
****************************************************/
#include "MySocket.h"
#include"winerror.h"
#pragma comment(lib, "ws2_32")
#include <time.h>
#include "CIniConfig.h"using namespace std;MySocket::MySocket(std::string ip, int port, u_long uctl): m_ip(ip), m_port(port), m_uctl(uctl), m_socket(NULL), m_socketClient(NULL)
{
}MySocket::~MySocket()
{Close();
}void MySocket::InitData(const std::string& ip, const int& port)
{if (ip.empty()) {m_ip = LOCAL_HOST;} else {m_ip = ip;}m_port = port;InitWinsock();
}void MySocket::InitWinsock()
{// 初始化WINSOCKWORD wVersionRequested = MAKEWORD(2, 2);WSADATA wsd;if (WSAStartup(wVersionRequested, &wsd) != 0) {std::cout << ("init WSAStartup faild") << std::endl;}if (LOBYTE(wsd.wVersion) != 2 ||HIBYTE(wsd.wVersion) != 2) {WSACleanup();std::cout << "WSACleanup" << std::endl;return;}
}/************************************************************************/
/*                              客户端                                  */
/************************************************************************/void MySocket::ClientConnect()
{if (m_port == 0) {m_port = PORT;}if (m_ip.empty()) {m_ip = LOCAL_HOST; //本地回路地址127，用于一台机器上测试的IP}m_socket = socket(AF_INET, SOCK_STREAM, 0);if (SOCKET_ERROR == m_socket) {std::cout << (" create socket failed !!!") << std::endl;return;}int ret = ioctlsocket(m_socket, FIONBIO, &m_uctl);     //阻塞方式   FIONBIO 允许或者禁止套接字m_socket非阻塞模式  0=阻塞if (SOCKET_ERROR == ret) {std::cout << (" set ioctlsocket faild: model") << m_uctl << std::endl;return ;}SOCKADDR_IN addrSrv;addrSrv.sin_addr.S_un.S_addr = inet_addr(m_ip.c_str());//地址，addrSrv.sin_family = AF_INET;addrSrv.sin_port = htons(m_port);//和服务器端的端口号保持一致connect(m_socket, (SOCKADDR*)&addrSrv, sizeof(SOCKADDR));//连接服务器端（套接字，地址转换，长度）
}void MySocket::Close()
{if (m_socket != NULL) {closesocket(m_socket);  //关闭套接字，释放为这个套接字分配的资源m_socket = NULL;}WSACleanup();//终止对这个套接字库的使用
}int MySocket::SendData(const std::string& sdata)
{char sendBuf[SEND_DATA_LEN];memset(sendBuf, 0, sizeof(sendBuf));std::string strId = g_ConfigPtr.getConfigValueWithKey("base", "id").c_str();std::string strName = g_ConfigPtr.getConfigValueWithKey("base", "name").c_str();int ihead = sizeof(TcpHead);int ibody = sizeof(TcpBody);Tcp_SendData* pData = (Tcp_SendData*)sendBuf;pData->_head._node = strId.empty() ? 0 : atoi(strId.c_str());pData->_head._type = 0;pData->_head._time = time(NULL);TcpBody tBody;tBody._length = sdata.length();memcpy(tBody._data, sdata.c_str(), sdata.length() + 1);memcpy(tBody._srcName, strName.c_str(), strName.length() + 1);memcpy(&sendBuf[ihead], &tBody, ibody);int sendLen = ihead + ibody;int ret = send(m_socket, sendBuf, sendLen, 0);return ret;
}int MySocket::ReceiveData()
{std::string* redata = nullptr;sockaddr_in sin;int slen = sizeof(sin);char reBuf[SEND_DATA_LEN] = {0};int ret = recvfrom(m_socket, reBuf, SEND_DATA_LEN, 0, (sockaddr*)&sin, &slen);if (ret > 0) {Tcp_SendData* pData = (Tcp_SendData*)reBuf;//printf("%s\n", reBuf);OnSelectEvent.emit(pData);}return ret;}