支持ipV6和ipV4的客戶端編程

   ipv4和ipv6在socket初始化的時候是不一樣的。

  ipv4 socket初始化：

int CClient::InitSocket(CString strIP, short portNum)
{
    WSADATA wsd; 
    SOCKADDR_IN servAddr; 
    int retVal; 
    //初始化套結字動態庫  
    if (WSAStartup(MAKEWORD(2, 2), &wsd) != 0)
    {
        return -1;
    }
    //創建套接字  
    m_sHost = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
    if (INVALID_SOCKET == m_sHost)
    {
        WSACleanup();//釋放套接字資源  
        return  -1;
    }

    //設置服務器地址  
    servAddr.sin_family = AF_INET;
    servAddr.sin_addr.s_addr = inet_addr((LPCTSTR)strIP);
    servAddr.sin_port = htons(portNum);
    int nServAddlen = sizeof(servAddr);

    //設置非阻塞方式連接
    unsigned long ul = 1;
    retVal = ioctlsocket(m_sHost, FIONBIO, (unsigned long*)&ul);
    if (retVal == SOCKET_ERROR) return 0;

    //連接服務器  
    retVal = connect(m_sHost, (LPSOCKADDR)&servAddr, sizeof(servAddr));

    if (retVal == 0)
    {

    }
    else
    {
        int iErr = GetLastError();
        if (WSAEWOULDBLOCK == iErr)
        {
            struct timeval timeout;
            fd_set r;

            FD_ZERO(&r);
            FD_SET(m_sHost, &r);
            timeout.tv_sec = 1; 
            timeout.tv_usec = 0;
            retVal = select(m_sHost + 1, NULL, &r, NULL, &timeout);
            if (retVal <= 0)
            {
                closesocket(m_sHost); //關閉套接字  
                WSACleanup(); //釋放套接字資源  
                return -1;
            }
        }
    }

    //退出  
    return 0;
}

 

ipv6 socket初始化：

int CClient::InitSocket(CString strIP, short portNum)
{
    WSADATA wsd;
    struct sockaddr_in6 servAddr_in6; 
    int retVal; 
    int nServAddlen;
    
    //初始化套結字動態庫  
    if (WSAStartup(MAKEWORD(2, 2), &wsd) != 0)
    {
        return -1;
    }

    sprintf_s(m_szIP, sizeof(m_szIP), strIP);

    memset(&servAddr_in6, 0, sizeof(struct sockaddr_in6));

    //創建套接字  
    m_sHost = socket(AF_INET6, SOCK_STREAM, IPPROTO_TCP);
    if (INVALID_SOCKET == m_sHost)
    {
        return  -1;
    }

    //設置服務器地址  
    servAddr_in6.sin6_family = AF_INET6;

    if (!inet_pton(AF_INET6, m_szIP, &servAddr_in6.sin6_addr.s6_addr))
    {
        closesocket(m_sHost); //關閉套接字 
        WSACleanup(); //釋放套接字資源 
        return  -1;
    }
    
    //設置非阻塞方式連接
    unsigned long ul = 1;
    retVal = ioctlsocket(m_sHost, FIONBIO, (unsigned long*)&ul);
    if (retVal == SOCKET_ERROR) return 0;
    
    servAddr_in6.sin6_port = htons(portNum);
    nServAddlen = sizeof(servAddr_in6);
    //連接服務器  
    retVal = connect(m_sHost, (LPSOCKADDR)&servAddr_in6, (socklen_t)nServAddlen);

   if (retVal == 0)
    {

    }
    else
    {
        int iErr = GetLastError();
        if (WSAEWOULDBLOCK == iErr)
        {
            struct timeval timeout;
            fd_set r;

            FD_ZERO(&r);
            FD_SET(m_sHost, &r);
            timeout.tv_sec = 1; 
            timeout.tv_usec = 0;
            retVal = select(m_sHost + 1, NULL, &r, NULL, &timeout);
            if (retVal <= 0)
            {
                closesocket(m_sHost); //關閉套接字  
                WSACleanup(); //釋放套接字資源  
                return -1;
            }
        }
    }

    //退出  
    return 0;
}

將上面的邏輯合二為一：

int CClient::InitSocket(CString strIP, short portNum)
{
    WSADATA wsd;
    struct sockaddr_in6 servAddr_in6;
    SOCKADDR_IN servAddr;
    int retVal;
    int nServAddlen;
    
    //初始化套結字動態庫  
    if (WSAStartup(MAKEWORD(2, 2), &wsd) != 0)
    {
        return -1;
    }

    sprintf_s(m_szIP, sizeof(m_szIP), strIP);

    memset(&servAddr, 0, sizeof(servAddr));
    memset(&servAddr_in6, 0, sizeof(struct sockaddr_in6));

    //設置非阻塞方式連接
    unsigned long ul = 1;
    retVal = ioctlsocket(m_sHost, FIONBIO, (unsigned long*)&ul);
    if (retVal == SOCKET_ERROR) return 0;
    if (TRUE == is_string_ipaddr_ipv4(m_szIP, &servAddr.sin_addr))
    {
        //創建套接字  
        m_sHost = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
        if (INVALID_SOCKET == m_sHost)
        {
            WSACleanup();//釋放套接字資源  
            return  -1;
        }

        //設置服務器地址  
        servAddr.sin_family = AF_INET;
        servAddr.sin_addr.s_addr = inet_addr(m_szIP);
        servAddr.sin_port = htons(portNum);
        nServAddlen = sizeof(servAddr);
        //連接服務器  
        retVal = connect(m_sHost, (LPSOCKADDR)&servAddr, (socklen_t)nServAddlen);
    }
    else
    {
        //創建套接字  
        m_sHost = socket(AF_INET6, SOCK_STREAM, IPPROTO_TCP);
        if (INVALID_SOCKET == m_sHost)
        {
            return  -1;
        }

        //設置服務器地址  
        servAddr_in6.sin6_family = AF_INET6;

        if (!inet_pton(AF_INET6, m_szIP, &servAddr_in6.sin6_addr.s6_addr))
        {
            closesocket(m_sHost); //關閉套接字 
            WSACleanup(); //釋放套接字資源 
            return  -1;
        }
        servAddr_in6.sin6_port = htons(portNum);
        nServAddlen = sizeof(servAddr_in6);
        //連接服務器  
        retVal = connect(m_sHost, (LPSOCKADDR)&servAddr_in6, (socklen_t)nServAddlen);
    }

   if (retVal == 0)
    {

    }
    else
    {
        int iErr = GetLastError();
        if (WSAEWOULDBLOCK == iErr)
        {
            struct timeval timeout;
            fd_set r;

            FD_ZERO(&r);
            FD_SET(m_sHost, &r);
            timeout.tv_sec = 1;
            timeout.tv_usec = 0;
            retVal = select(m_sHost + 1, NULL, &r, NULL, &timeout);
            if (retVal <= 0)
            {
                closesocket(m_sHost); //關閉套接字  
                WSACleanup(); //釋放套接字資源  
                return -1;
            }
        }
    }

    //退出  
    return 0;
}

 

判斷一個ip是不是ipv4的

/*判斷傳入的字符串是否為ipv4格式的
*/
bool CClient::is_string_ipaddr_ipv4(const char* str_ipv4, struct in_addr *ip)
{
    int32 ret = -1;
    struct in_addr tmp_ip;

    memset(&tmp_ip, 0, sizeof(struct in_addr));

    if (NULL == str_ipv4)
    {
        return false;
    }

    if (0 == *str_ipv4)
    {
        return false;
    }

    ret = inet_pton(AF_INET, str_ipv4, &tmp_ip);
    if (0 == ret)
    {
        return false;
    }
    else if (ret < 0)
    {
        return false;
    }
    else
    {
        if (NULL != ip)
        {
            memcpy(ip, (void *)&tmp_ip, sizeof(struct in_addr));
        }

        return true;
    }
}