网络：TCP对客户端的封装

最新推荐文章于 2023-03-13 22:29:17 发布

@布响丸辣

最新推荐文章于 2023-03-13 22:29:17 发布

阅读量118

点赞数

分类专栏：网络文章标签：网络 tcp/ip 网络协议

本文链接：https://blog.csdn.net/m0_69414302/article/details/126702569

版权

网络专栏收录该内容

8 篇文章 0 订阅

订阅专栏

声明：

#ifndef CTCPMAP_H
#define CTCPMAP_H
#include<windows.h>
#include<list>
#include<winsock2.h>
#include<iostream>
#include<map>
using namespace std;
class client
{
public:
    client();
    ~client();
    //初始化网络
    bool InitnewWork();
    //销毁网络
    void DeleteWork();
    //发送数据
    bool sendDate(char*szbuf,int nLen);
    //接收数据
    void recvDate();
    //线程函数
    static DWORD WINAPI ThreadWork(LPVOID lpvoid);
private:
SOCKET m_client;
HANDLE m_hThread;
bool bFlagQuit;
};
 
#endif // CLIENT_H

实现：

#include "client.h"
 
client::client()
{
    m_client=0;
    m_hThread=NULL;
    bFlagQuit=true;
 
}
 
client::~client()
{
 
}
 
bool client::InitnewWork()
{
    WORD wVersionRequested;
    WSADATA wsaData;
    int err;
 
 
    wVersionRequested = MAKEWORD(2, 2);
 
    err = WSAStartup(wVersionRequested, &wsaData);
    if (err != 0) {
 
        printf("WSAStartup failed with error: %d\n", err);
        return false;
    }
 
 
 
    if (LOBYTE(wsaData.wVersion) != 2 || HIBYTE(wsaData.wVersion) != 2) {
 
       DeleteWork();
        return false;
    }
    else
        printf("The Winsock 2.2 dll was found okay\n");
 
 
//创建套接字
    m_client=socket(AF_INET,SOCK_STREAM,0);
    if(m_client==INVALID_SOCKET)
    {
       DeleteWork();
        return false;
    }
 
 
 
//connect()
 
 
    //绑定IP
    sockaddr_in addrserver;
    addrserver.sin_addr.S_un.S_addr=inet_addr("127.0.0.1");
    addrserver.sin_port=htons(8899);
    addrserver.sin_family=AF_INET;
    if(SOCKET_ERROR==connect(m_client,(sockaddr*)&addrserver,sizeof(addrserver)))
    {
           DeleteWork();
            return false;
 
    }
    //创建线程
    m_hThread=CreateThread(0,0,&ThreadWork,this,0,0);
    return  true;
 
 
 
}
 
DWORD WINAPI  client::ThreadWork(LPVOID lpvoid)
{
    client* pthis=(client* ) lpvoid;
 
 
 
 
    while(pthis->bFlagQuit)
    {
 
 
    }
return 0;
}
void client::DeleteWork()
{
 
    bFlagQuit=false;
    if(m_hThread)
    {
        if(WAIT_TIMEOUT==WaitForSingleObject(m_hThread,100))
            TerminateThread(m_hThread,-1);
 
        CloseHandle(m_hThread);
 
        m_hThread=NULL;
 
    }
 
 
    if(m_client)
    {
        closesocket(m_client);
        m_client=0;
 
    }
 
    WSACleanup();
 
}
bool client::sendDate(char*szbuf,int nLen)
{
 
   if(!szbuf||nLen<=0)
       return  false;
   if(send(m_client,(char*)&nLen,sizeof(int),0)<=0)
       return  false;
   if(send(m_client,szbuf,nLen,0)<=0)
       return false;
 
 
   return  true;
 
}
void client::recvDate()
{
    int  nRecvNum;
    int  nPackNum;
    char *szbuf;
    int  noffest;
    while(bFlagQuit)
    {
        nRecvNum=recv(m_client,(char*)&nPackNum,sizeof(int),0);
        if(nRecvNum<=0)
        {
            DeleteWork();
            break;
        }
        if(nPackNum<=0) continue;
        szbuf=new char[nPackNum];
        noffest=0;
        while(nPackNum)
        {
            nRecvNum=recv(m_client,szbuf+noffest,nPackNum,0);
            if(nRecvNum>0)
            {
                nPackNum-=nRecvNum;
                noffest+=nRecvNum;
            }
        }
 
    }
 
 
 
}

主函数测试：

#include <QCoreApplication>
#include<winsock2.h>
#include<iostream>
#include"client.h"
using namespace std;
int main(int argc, char *argv[])
{
    client cli;
    if(cli.InitnewWork())
    {
        cout<<"connect sucess"<<endl;
        char szbuf[1024];
        cin>>szbuf;
        cli.sendDate(szbuf,sizeof (szbuf));
    }
 
 
}