我正在为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 */
}
}
答案 0 :(得分:4)
你担心定义SOCKET
类型的变量,以及它的构造函数何时运行?
这不是问题,因为SOCKET
是一个C兼容的普通旧数据整数类型,它包含套接字标识符。它不是一个对象。没有与变量本身相关的非平凡构造或破坏。
答案 1 :(得分:2)
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
的文件提供的任何名称),并自动初始化/清理套接字(以及链接到正确的库)。