c++ TCP Socket

最新推荐文章于 2024-08-19 13:25:45 发布

dxm809

最新推荐文章于 2024-08-19 13:25:45 发布

阅读量232

点赞数

分类专栏： C++

本文链接：https://blog.csdn.net/dxm809/article/details/115473549

版权

C++ 专栏收录该内容

334 篇文章 4 订阅

订阅专栏

该博客详细介绍了如何使用C++实现TCP套接字的初始化、连接、关闭、发送命令和接收消息功能。通过`CTCPSocket`类展示了套接字的基本操作，包括设置超时、错误检查等关键步骤，适用于网络通信的基础开发。

摘要由CSDN通过智能技术生成

#pragma once
#include <string>
using namespace std;

class CTCPSocket
{
public:
    CTCPSocket();
    ~CTCPSocket();

    bool SocketInit();
    bool Connect(CString strIp, int nPort);
    bool Close();
    bool SendCmd(CString cmd, int iTimeOut=1);
    bool ReceiveMsg(CString& strRecvMsg, int iTimeOut=2);
private:
    SOCKET m_Scoket;
    WORD m_dwVserion;
    WSADATA m_WsData;
    int m_iError;
    bool m_isSocketFlag;
};

#include "pch.h"
#include "CTCPSocket.h"
#include <string>

#pragma comment(lib, "ws2_32.lib")

CTCPSocket::CTCPSocket()
{
    m_dwVserion = 0;
    m_isSocketFlag = true;
    memset(&m_WsData, 0, sizeof(WSADATA));
}


CTCPSocket::~CTCPSocket()
{
    /*closesocket(m_Scoket);
    WSACleanup();*/
}

bool CTCPSocket::SocketInit()
{
    m_dwVserion = MAKEWORD(1, 1);
    m_iError = WSAStartup(m_dwVserion, &m_WsData);
    if (m_iError != 0)
    {
        m_isSocketFlag = false;
    }
    if (LOBYTE(m_WsData.wVersion) != 1 && HIBYTE(m_WsData.wVersion) != 1)
    {
        WSACleanup();
        m_isSocketFlag = false;
    }
    return m_isSocketFlag;
}

bool CTCPSocket::Connect(CString strIp, int nPort)
{
    if (!m_isSocketFlag)
        return false;
    m_Scoket = socket(AF_INET, SOCK_STREAM, 0);
    SOCKADDR_IN addr;
    addr.sin_addr.S_un.S_addr = inet_addr(strIp);
    addr.sin_family = AF_INET;
    addr.sin_port = htons(nPort);

    m_iError = connect(m_Scoket, (SOCKADDR*)&addr, sizeof(SOCKADDR));
    if (m_iError != 0)
        m_isSocketFlag = false;
    return m_isSocketFlag;
}

bool CTCPSocket::Close()
{
   closesocket(m_Scoket); 
   return true;
}

bool CTCPSocket::SendCmd(CString cmd,int iTimeOut)
{
    if (!m_isSocketFlag)
        return false;
    int timeOut = iTimeOut * 1000;        //秒
    setsockopt(m_Scoket, SOL_SOCKET, SO_SNDTIMEO, (char*)&timeOut, sizeof(timeOut));  
    int iRet = send(m_Scoket, cmd, cmd.GetLength(), 0);
    if (iRet == 0)
    {
        return false;
    }
   
    return true;
}

bool CTCPSocket::ReceiveMsg(CString& strRecvMsg, int iTimeOut)
{
    if (!m_isSocketFlag)
        return false;
    int timeOut = iTimeOut * 1000;        //秒
    setsockopt(m_Scoket, SOL_SOCKET, SO_RCVTIMEO, (char*)&timeOut, sizeof(timeOut));
    char recvBuf[1024] = { 0 };
    int iRet = recv(m_Scoket, recvBuf, sizeof(recvBuf), 0);
    if (iRet == -1)
    {
        return false;
    }
    string recvMsg = string(recvBuf);

    strRecvMsg = recvMsg.c_str();
}