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

Blog @ Blog

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

常用链接

统计

积分与排名

BBS

Blog

Web

最新评论

启动服务流程代码

//主对话框启动服务
void CTCPUDPDlg::OnBtnStartServer() 

//Server对话框继承类CSocketComm的CreateSocket
bool CServer::CreateSock(LPCTSTR strServerName, int nProtocol, int nType, UINT uOptions)

//启动服务，主要实现启动套接字（socket();setsockopt();bind()(UDP服务端需要);listen()(TCP服务端需要祯听);closesocket()）
bool CSocketComm::CreateSocket(LPCTSTR strServerName, int nProtocol, int nType, UINT uOptions)

//启动Socket线程
bool CSocketComm::WatchServerSocket()

UINT WINAPI CSocketComm::SocketServerThreadProc(LPVOID pParam)//服务端线程

void CSocketComm::ServerRun()//Server主线程函数

附主要函数流程具体代码
bool CSocketComm::CreateSocket(LPCTSTR strServerName, int nProtocol, int nType, UINT uOptions)
{
    // 如果已经打开，则返回false
    if ( IsOpen() )
        return false;

    SOCKADDR_IN sockAddr = { 0 };
    //地址设定 //SOCKET
    SOCKET sock = socket(nProtocol, nType, 0);//unsigned int 0~4294967295 
    if (INVALID_SOCKET != sock)
    {
        sockAddr.sin_port = htons( GetPortNumber( strServerName ) );
        if ( 0 != sockAddr.sin_port)
        {
            sockAddr.sin_addr.s_addr = htonl( INADDR_ANY );
            sockAddr.sin_family = nProtocol;
            if (uOptions & SO_REUSEADDR) //设定相关选项
            {
                BOOL optval = TRUE;
                if (SOCKET_ERROR == setsockopt( sock, SOL_SOCKET, SO_REUSEADDR,(char *) &optval,sizeof( BOOL )))
                {
                    closesocket( sock );
                    return false;
                }
            }
            if (nType == SOCK_DGRAM) //如果是UDP协议
            {
                if (uOptions & SO_BROADCAST)//如果允许广播
                {
                    BOOL optval = TRUE;
                    if (SOCKET_ERROR == setsockopt(sock,SOL_SOCKET,SO_BROADCAST,(char *) &optval, sizeof(BOOL)))
                    {
                        closesocket( sock );
                        return false;
                    }
                    m_bBroadcast = true;// 设定变量
                }
                m_hMutex = CreateMutex(NULL, FALSE, NULL);// 如果需要广播，则需要设定进程互斥
                if (NULL == m_hMutex)
                {
                    closesocket( sock );
                    return false;
                }
            }
            if ( SOCKET_ERROR == bind(sock, (LPSOCKADDR)&sockAddr, sizeof(SOCKADDR_IN)))// 绑定一个本地地址
            {
                closesocket( sock );
                m_bBroadcast = false;
                if (NULL != m_hMutex)
                    CloseHandle( m_hMutex );
                m_hMutex = NULL;
                return false;
            }
            if (SOCK_STREAM == nType)// 如果是TCP连接
            {
                if ( SOCKET_ERROR == listen(sock, SOMAXCONN))
                {
                    closesocket( sock );
                    return false;
                }
            }
            m_hSocket = (HANDLE) sock;// 保存socket
            return true;
        }
        else
            SetLastError(ERROR_INVALID_PARAMETER); //不能找到端口
        closesocket( sock );// 删除socket
        return false;
    }
    else//创建失败
    {
        CString strError;
        strError.Format("Create Socket Error Code : %d",GetLastError());//获取错误代码
        return false;
    }
}

//启动Socket线程
BOOL CSocketComm::WatchServerSocket()
{
    //首先判断线程是否启动
    if(!IsStart())
    {
        //判断是否打开通信，即socket是否成功创建
        if(IsOpen())
        {
            HANDLE hThread;
            UINT uiThreadId = 0;
            //启动线程，使用_beginthreadex
            hThread = (HANDLE)_beginthreadex(NULL,    // 安全参数
                                                0,    // 堆栈
                                SocketServerThreadProc,    // 线程程序
                                            this,    // 线程参数
                                CREATE_SUSPENDED,    //创建模式
                                    &uiThreadId);    // 线程ID
            //如果线程不为空
            if(hThread != NULL)
            {
                //继续线程
                ::ResumeThread(hThread);
                m_hThread = hThread;
                return TRUE;
            }
        }
    }
    return false;
}

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

    pThis->ServerRun();

    return 1L;
}

//Server主线程函数，WaitForConnection()实际引用;
void CSocketComm::ServerRun()
{
    BYTE buffer[BUFFER_SIZE];
    DWORD dwBytes = 0L;

    HANDLE hThread = ::GetCurrentThread();
    DWORD dwTimeout = DEFAULT_TIMEOUT;

    struct sockaddr_in ClientSockAddr; 
    int addrlen = sizeof(ClientSockAddr); 
    
    //int FAR* addrlen       
    //是否广播模式
    if(!IsBroadCast())
    {
        SOCKET sock = (SOCKET) m_hSocket;
        sock = WaitForConnection(sock,(struct sockaddr *)&ClientSockAddr,&addrlen);//accept()
        //等待新的连接
        if(sock != INVALID_SOCKET)
        {
            //关闭连接
            ShutDownConnection((SOCKET)m_hSocket);    //closesocket()
            m_hSocket = (HANDLE) sock;
            OnEvent(EVT_CONSUCCESS,ClientSockAddr);//connect
        }
        else
        {
            //如果已经关闭则不发送事件，否则发送
            if(IsOpen())
                OnEvent(EVT_CONFAILURE,ClientSockAddr);//等待失败
            return;
        }
    }
    //如果SOCKET 已经创建
    while(IsOpen())
    {
        //采用阻塞式SOCKET，等待事件通知
        dwBytes = ReadComm(buffer,sizeof(buffer),dwTimeout);//recvfrom() or recv()
        // 如果有错误发生
        if (dwBytes == (DWORD)-1)
        {
            // 如果要关闭，则不发送事件
            if (IsOpen())
                OnEvent( EVT_CONDROP,ClientSockAddr ); // 失去连接
            break;
        }
        // 是否有数据收到
        if (IsSmartAddressing() && dwBytes == sizeof(SOCKADDR_IN))
            OnEvent( EVT_ZEROLENGTH ,ClientSockAddr);
        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:06 isabc 阅读(695) 评论(0)  编辑 收藏 引用


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


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

中文版MSDN：
欢迎体验