山寨：不是最好的，是最适合我们的！欢迎体验山寨 中文版MSDN

Blog @ Blog

当华美的叶片落尽，生命的脉络才历历可见。 －－ 聂鲁达

常用链接

统计

积分与排名

BBS

Blog

Web

最新评论

连接服务流程代码

void CTCPUDPDlg::OnBtnConnect() //连接服务

bool CSocketComm::ConnectTo(LPCTSTR strDestination, LPCTSTR strServiceName, int nProtocol, int nType)

BOOL CSocketComm::WatchClientSocket()//启动Socket线程

UINT WINAPI CSocketComm::SocketClientThreadProc(LPVOID pParam)//客户端线程


void CSocketComm::ClientRun()//Client主线程函数

//附件：
//连接服务: socket();bind();connect()
bool CSocketComm::ConnectTo(LPCTSTR strDestination, LPCTSTR strServiceName, int nProtocol, int nType)
{
    // 如果socket已经打开
    if ( IsOpen() )
        return false;

    SOCKADDR_IN sockAddr = {0};
    //创建一个SOCKET
    SOCKET sock  = ::socket(nProtocol, nType, 0);
    if(sock != INVALID_SOCKET)
    {
        // 让socket绑定一个地址
        TCHAR strHost[HOSTNAME_SIZE] = { 0 };
        if (false == CSocketComm::GetLocalName( strHost, sizeof(strHost)/sizeof(TCHAR)))
        {
            closesocket( sock );
            return false;
        }
        //获取本机IP
        sockAddr.sin_addr.s_addr = htonl( CSocketComm::GetIPAddress( strHost ) );
        sockAddr.sin_family = nProtocol;

        if ( SOCKET_ERROR == bind(sock, (LPSOCKADDR)&sockAddr, sizeof(SOCKADDR_IN)))
        {
            closesocket( sock );
            return false;
        }
        // 获得目标地址
        if ( strDestination[0])
        {
            sockAddr.sin_addr.s_addr = htonl(CSocketComm::GetIPAddress( strDestination ) );
        }
        // 获得端口
        sockAddr.sin_port = htons( GetPortNumber( strServiceName ));
        if ( 0 != sockAddr.sin_port )
        {
            // 连接服务器
            if (SOCKET_ERROR == connect( sock, (LPSOCKADDR)&sockAddr, sizeof(SOCKADDR_IN)))
            {
                closesocket( sock );
                return false;
            }

            // 保存socket
            m_hSocket = (HANDLE) sock;
            return true;
        }
    }
    else
    {
        CString strError;
        strError.Format("Create Socket Error Code: %d",GetLastError());
        AfxMessageBox(strError,MB_ICONERROR);
    }
    return false;
}
//启动Socket线程
BOOL CSocketComm::WatchClientSocket()
{
    //首先判断线程是否启动
    if(!IsStart())
    {
        //判断是否打开通信，即socket是否成功创建
        if(IsOpen())
        {
            HANDLE hThread;
            UINT uiThreadId = 0;
            //启动线程，使用_beginthreadex
            hThread = (HANDLE)_beginthreadex(NULL,    // 安全参数
                                                0,    // 堆栈
                                SocketClientThreadProc,    // 线程程序
                                            this,    // 线程参数
                                CREATE_SUSPENDED,    //创建模式
                                    &uiThreadId);    // 线程ID
            //如果线程不为空
            if(hThread != NULL)
            {
                //继续线程
                ::ResumeThread(hThread);
                m_hThread = hThread;
                return TRUE;
            }
        }
    }
    return false;
}

//客户端线程
UINT WINAPI CSocketComm::SocketClientThreadProc(LPVOID pParam)
{
    //reinterpret_cast用于各种指针的转化
    CSocketComm* pThis = reinterpret_cast<CSocketComm*>(pParam);
    _ASSERTE( pThis != NULL );

    pThis->ClientRun();

    return 1L;
}

//Client主线程函数ReadComm():
void CSocketComm::ClientRun()
{
    BYTE buffer[BUFFER_SIZE];
    DWORD dwBytes = 0L;

    HANDLE hThread = ::GetCurrentThread();
    DWORD dwTimeout = DEFAULT_TIMEOUT;
    
    //如果SOCKET 已经创建
    while(IsOpen())
    {
        //采用阻塞式SOCKET，等待事件通知
        dwBytes = ReadComm(buffer,sizeof(buffer),dwTimeout);
        // 如果有错误发生
        if (dwBytes == (DWORD)-1)
        {
            // 如果要关闭，则不发送事件
            if(IsOpen())
                OnEvent( EVT_CONDROP,ServerSockAddr); // 失去连接
            break;
        }

        // 是否有数据收到
        if (IsSmartAddressing() && dwBytes == sizeof(SOCKADDR_IN))
            OnEvent( EVT_ZEROLENGTH,ServerSockAddr);
        else if(dwBytes > 0L)
        {
            OnDataReceived(buffer, dwBytes);
        }

        Sleep(0);
    }

}

//读数据函数select();recvfrom();recv()
DWORD CSocketComm::ReadComm(LPBYTE lpBuffer, DWORD dwSize, DWORD dwTimeout)
{
    _ASSERTE(IsOpen());
    _ASSERTE(lpBuffer != NULL);
    if(lpBuffer == NULL || dwSize<1L)
        return 0L;

    fd_set fdRead = {0};
    TIMEVAL stTime;
    TIMEVAL *pstTime = NULL;

    if(INFINITE != dwTimeout)
    {
        stTime.tv_sec=0;
        stTime.tv_usec=dwTimeout*1000;
        pstTime = &stTime;
    }
    SOCKET s=(SOCKET)m_hSocket;
    //设定描述符
    if(!FD_ISSET(s,&fdRead))
        FD_SET(s,&fdRead);

    //选择函数，设定超时时间
    DWORD dwBytesRead = 0L;
    int res = select(s+1,&fdRead,NULL,NULL,pstTime);
    if(res > 0)
    {
        if(IsBroadCast() || IsSmartAddressing())
        {
            SOCKADDR_IN sockAddr = {0};//获得地址
            int nOffset = IsSmartAddressing()?sizeof(sockAddr):0;
            int nLen = sizeof(sockAddr);
            if(dwSize < (DWORD)nLen)//缓冲区太小
            {
                ::SetLastError(ERROR_INVALID_USER_BUFFER);
                return -1L;
            }
            //获得数据
            res = ::recvfrom(s,(LPSTR)&lpBuffer[nOffset],dwSize,0,(LPSOCKADDR)&sockAddr,&nLen);
            
            memset(&sockAddr.sin_addr,0,sizeof(sockAddr.sin_zero));
            if(res>=0)
            {
                LockList();//锁定地址列表
                //删除调重复地址
                SockAddrIn sockin;
                sockin.SetAddr(&sockAddr);
                m_AddrList.remove(sockin);
                m_AddrList.insert(m_AddrList.end(),sockin);

                if(IsSmartAddressing())
                {
                    memcpy(lpBuffer,&sockAddr,sizeof(sockAddr));
                    res+=sizeof(sockAddr);
                }
                UnlockList();//解开地址列表
            }
        }
        else
        {
            res = recv(s,(LPSTR)lpBuffer,dwSize,0);
        }
        dwBytesRead = (DWORD)((res>0)?(res):(-1));
    }
    
     return dwBytesRead;//返回记录长度
}

posted on 2008-05-02 10:10 isabc 阅读(690) 评论(1)  编辑 收藏 引用 所属分类: 分享代码

评论

# re: 连接服务流程代码[未登录] 2008-11-18 10:39 1

11111  回复  更多评论   


只有注册用户登录后才能发表评论。
网站导航: 博客园   IT新闻   BlogJava   知识库   博问   管理


广告信息(免费广告联系)

中文版MSDN：
欢迎体验