Socket 服务器客户端多线程异步例程

2014年3月29日 2866点热度 0人点赞 0条评论

下面这个是客户端的头文件,解释够清楚把

#pragma once

typedef void (*CallFun_Client) (LPVOID/*传递值*/,int/*代码*/,char*/*数据*/);/*回调函数指针*/

class CClientSocket
{
public:
    CClientSocket();//构造函数
    ~CClientSocket();//析构函数
    BOOL Initialize(CallFun_Client call/*回调函数*/,LPVOID lpParameter/*传递值*/);//初始化
    int Send(char* chBuf/*发送给服务器的数据*/);//发送
    BOOL Connect(char* chAddr/*服务器IP*/, int nPort/*服务器端口*/);//连接服务器
    BOOL Close();//关闭连接
public:
    SOCKET m_hSock;//Socket句柄
    CallFun_Client m_pCall;//回调函数指针
    LPVOID m_lpParameter;//传递值
    HANDLE m_hThread;//线程句柄
};

下面是它的实现文件

构造函数,调用 2.2版本的Socket库

CClientSocket::CClientSocket()
{
    WSADATA wsaData;
    WSAStartup(MAKEWORD(2,2),&wsaData);
}

析构

CClientSocket::~CClientSocket()
{
    closesocket(m_hSock);//关闭Socket句柄
    WSACleanup();//释放Socket
    CloseHandle(m_hThread);//关闭线程句柄
}

初始化.关闭 Initialize Close

BOOL CClientSocket::Initialize(CallFun_Client call,LPVOID lpParameter)
{
    m_pCall=call;
    m_lpParameter=lpParameter;
    return TRUE;
}
BOOL CClientSocket::Close()
{
    if(closesocket(m_hSock)!=0)
        return FALSE;
    CloseHandle(m_hThread);
    m_hSock=0;
    m_hThread=0;
    return TRUE;
}

客户端发送 Send

int CClientSocket::Send(char* chBuf)
{
    return send(m_hSock,chBuf,strlen(chBuf),0);
}

连接服务器 Connect

BOOL CClientSocket::Connect(char* chAddr, int nPort)
{
    int nError;
    m_hSock=socket(AF_INET,SOCK_STREAM,IPPROTO_TCP);//创建套接字
    SOCKADDR_IN addrSrv;
    addrSrv.sin_addr.S_un.S_addr=inet_addr(chAddr);//服务器的IP
    addrSrv.sin_family=AF_INET;
    addrSrv.sin_port=htons(nPort);//服务器的端口
    nError=connect(m_hSock,(SOCKADDR*)&addrSrv,sizeof(SOCKADDR));//连接
    if(!nError)
        return FALSE;
    m_hThread=CreateThread(NULL,0,ThreadProc_Client,this,0,NULL);//启动线程
    return TRUE;
}

线程处理:

DWORD _stdcall ThreadProc_Client(LPVOID lpParameter)
{
    CClientSocket* pClient=(CClientSocket*)lpParameter;
    int nError;
    while (true)
    {
        char chBuf[1024]={};
        nError=recv(pClient->m_hSock,chBuf,sizeof(chBuf),0);//接受
        if(nError<=0)//如果返回<= 的数,就证明已经断开服务器,或者Socket关闭
        {
            pClient->m_pCall(pClient->m_lpParameter,-1,NULL);//-1代表断开服务器
            break;//跳出循环
        }
        pClient->m_pCall(pClient->m_lpParameter,1,chBuf);//通过回调函数,告诉数据,代码1表示收到数据
    }
    return TRUE;
}

客户端完毕

服务器的头文件:

#pragma once

typedef void (*CallFun_Sever) (LPVOID/*传递值*/,SOCKET/*客户端Socket句柄*/,int/*代码*/,char*/*数据*/);/*回调函数指针*/

class CSeverSocket
{
public:
    CSeverSocket(void);
    ~CSeverSocket(void);
    BOOL Initialize(CallFun_Sever call/*回调函数*/,LPVOID lpParameter/*传递值*/);//初始化
    BOOL Close();//停止服务器
    BOOL Start(int nPort);//启动服务器
    int Send(SOCKET ClientSock/*客户端Socket*/,char* chBuf/*数据*/);//发送数据给客户端

public:
    SOCKET m_hSock;//Socket句柄
    CallFun_Sever m_pCall;//回调函数指针
    LPVOID m_lpParameter;//传递值
    HANDLE m_hThread;//线程句柄
};

实现文件,构造函数,析构函数,Initialize,Close都和客户端差不多

Send 发送

int CSeverSocket::Send(SOCKET ClientSock,char* chBuf)
{
    return send(ClientSock,chBuf,strlen(chBuf),0);
}

Start 启动服务器

BOOL CSeverSocket::Start(int nPort)
{
    m_hSock=socket(AF_INET,SOCK_STREAM,IPPROTO_TCP);//创建套接字
    SOCKADDR_IN addrSrv;
    addrSrv.sin_addr.S_un.S_addr=htonl(INADDR_ANY);
    addrSrv.sin_family=AF_INET;
    addrSrv.sin_port=htons(nPort);
    if(bind(m_hSock,(SOCKADDR*)&addrSrv,sizeof(SOCKADDR))!=0)//给未命名套接字赋一个本地名
        return FALSE;
    if(listen(m_hSock,1)!=0)//在指定套接字上监听外来连接
        return FALSE;
    m_hThread=CreateThread(NULL,0,ThreadProc_Sever,this,0,NULL);//启动线程
    return TRUE;
}

线程处理:

DWORD _stdcall ThreadProc_Sever(LPVOID lpParameter)
{
    CSeverSocket* pSever=(CSeverSocket*)lpParameter;
    fd_set fdSocket;
    FD_ZERO(&fdSocket);
    FD_SET(pSever->m_hSock,&fdSocket);
    while (true)
    {
        fd_set fdRead=fdSocket;
        if(select(0,&fdRead,NULL,NULL,NULL)<=0)
            break;
        for(int i=0; i<(int)fdSocket.fd_count; i++)
        {
            if(!FD_ISSET(fdSocket.fd_array[i],&fdRead))
                continue;
            if(fdSocket.fd_array[i]==pSever->m_hSock)
            {
                //来新的客户
                sockaddr_in addrRemote;
                int nAddrLen=sizeof(addrRemote);
                SOCKET sNew=accept(pSever->m_hSock,(SOCKADDR*)&addrRemote, &nAddrLen);
                pSever->m_pCall(pSever->m_lpParameter,sNew,0,NULL);
                FD_SET(sNew,&fdSocket);
            }
            else
            {
                void* chBuf[1024];
                memset(chBuf,0x00,sizeof(chBuf)/sizeof(chBuf[0]));
                if(recv(fdSocket.fd_array[i],(char*)chBuf,sizeof(chBuf)/sizeof(chBuf[0]),0)>0)
                {
                    //收到新数据
                    pSever->m_pCall(pSever->m_lpParameter,fdSocket.fd_array[i],1,(char*)chBuf);
                }
                else
                {
                    //客户断开服务器
                    closesocket(fdSocket.fd_array[i]);
                    pSever->m_pCall(pSever->m_lpParameter,fdSocket.fd_array[i],-1,NULL);
                    FD_CLR(fdSocket.fd_array[i], &fdSocket);
                }
            }
        }

    }
    return TRUE;
}

源码下载,有什么缺陷什么的希望大家能提出来,我好改正

Socket 服务器 客户端 多线程 异步例程

文章评论

Socket 服务器客户端多线程异步例程