我可以在调用WSAStartup()之前创建套接字实例吗?

时间:2013-03-11 16:31:15

标签: c++ oop winsock2 wsastartup

我正在为TCP / IP网络客户端通信编写一个类。在类头中,我创建了一个SOCKET成员。该类还包含一个调用WSAStartup并检查版本的方法。创建连接的方法首先调用WSAStartup,然后初始化调用socket()函数的SOCKET成员。见下面的代码。 我想知道这是否“正确”,或者是否有更好的方法。

标题文件:

/*network.h*/
public class IPnetwork
{
private:
    WSADATA wsaData ;
    SOCKET hSocket ;
    sockaddr_in socketAddress ;
    static const int SERVER_PORT = 502 ;
    unsigned long int = serverIP ;

public:
    IPnetwork(char* serverIPaddress) ;
    bool Connect() ;
    bool Disconnect() ;
    ~IPnetwork() ;

private:
    bool startWinSock() ;
} ;

源代码:

/*network.cpp*/
IPnetwork::IPnetwork(char* serverIPaddress)
{
    serverIP = inet_addr(serverIPaddress) ;
}

bool IPnetwork::Connect()
{
     /* start winsock */
    if(!startWinSock())
    {
        return false ; /* winsock problem */
    }

    /* Create socket */
    if ((hSocket = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP)) == INVALID_SOCKET)
    {
        return false ; /* could not create socket */
    }

    /* fill socket address structure */
    socketAddress.sin_family = AFINET ;
    socketAddress.sin_port = htons(SERVER_PORT) ;
    socketAddress.sin_addr.S_un.S_addr = serverIP ;

    /* connect */
    if(connect(hSocket,reinterpret_cast<sockaddr*>(&socketAddress), sizeof(sockAddr))!=0)
    {
        return false ; /* could not connect*/
    }

    return true ;

}

bool IPnetwork::startWinSock()
{
    if(WSAStartup(MAKEWORD(REQ_WINSOCK_VER,0),&wsaData)==0)
    {
        /* Check if major version is at least REQ_WINSOCK_VER */
        if (LOBYTE(wsaData.wVersion) < REQ_WINSOCK_VER)
        {
            return false ; /* winsock started but version is too low */
        }
        else
        {
            return true ; /* winsock started with right version*/
        }
    }
    else
    {
        return false ; /* winsock not started */
    }

}

3 个答案:

答案 0 :(得分:4)

你担心定义SOCKET类型的变量,以及它的构造函数何时运行?

这不是问题,因为SOCKET是一个C兼容的普通旧数据整数类型,它包含套接字标识符。它不是一个对象。没有与变量本身相关的非平凡构造或破坏。

答案 1 :(得分:2)

必须先调用{p> WSAStartup()才能调用socket(),否则会返回WSANOTINITIALISED错误。

您可以多次致电WSAStartup()。最好只调用WSAStartup()一次,但WinSock是引用计数,因此如果需要可以多次调用它。每次WSACleanup()成功时,您必须确保每次呼叫WSAStartup(),否则引用计数将不平衡。例如,您可以在构造函数中调用WSAStartup()并在析构函数中调用WSACleanup()

public class IPnetwork
{
private:
    WSADATA wsaData ;
    bool wsaInit;
    ...

public:
    IPnetwork(char* serverIPaddress) ;
    ~IPnetwork() ;
} ;

IPnetwork::IPnetwork(char* serverIPaddress)
{
    wsaInit = (WSAStartup(MAKEWORD(REQ_WINSOCK_VER,0), &wsaData) == 0);
    ...
}

IPnetwork::~IPnetwork()
{
    if (wsaInit) WSACleanup();
}

bool IPnetwork::Connect()
{
    if (!wsaInit)
        return false ; /* winsock problem */
    ...
}

虽然我通常会在一个单独的单例类中调用它们。

答案 2 :(得分:1)

虽然我同意@Ben你正在做的事情很好,但我个人试图从你的代码主线中获得这个丑陋。 IMO,这是创建全局变量的合理时间之一,在这样的标题中:

#ifndef WS_INITIATOR_INC_
#define WS_INITIATOR_INC_

#include <winsock2.h>
#pragma comment(lib, "ws2_32.lib")

#include <stdexcept>

struct bad_version : public std::logic_error {
    bad_version(std::string const &s) : logic_error(s) {}
};

struct winsock { 
    static const int version = 2;
    WSADATA wsaData;

    winsock() { 
        WSAStartup(MAKEWORD(2, 2),&wsaData);
        if (LOBYTE(wsaData.wVersion) != 2 || HIBYTE(wsaData.wVersion) != 2)
            throw bad_version("Could not initialize WinSock 2.2");
    }
    ~winsock() { 
        WSACleanup();
    }
} ws_initiator;

#endif

将此包含在您的main.cpp中(或者您为包含main的文件提供的任何名称),并自动初始化/清理套接字(以及链接到正确的库)。