Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update socket.h

Browse Source
This commit is contained in:
Tianqi Chen 2014-11-23 22:46:57 -08:00
parent f53be2884a
commit cdcfa5687a
1 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
src/utils/socket.h
View File

@ -69,7 +69,7 @@ struct SockAddr {
#ifdef _WIN32
const char *s = inet_ntop(AF_INET, (PVOID)&addr.sin_addr, &buf[0], buf.length());
#else
const char *s = inet_ntop(AF_INET, &addr.sin_addr, &buf[0], buf.length());
const char *s = inet_ntop(AF_INET, &addr.sin_addr, &buf[0], buf.length());
#endif
Assert(s != NULL, "cannot decode address");
return std::string(s);
@ -111,12 +111,12 @@ class TCPSocket {
*/
inline static void Startup(void) {
#ifdef _WIN32
WSADATA wsa_data;
WSADATA wsa_data;
if (WSAStartup(MAKEWORD(2, 2), &wsa_data) != -1) {
SockError("Startup");
}
SockError("Startup");
}
if (LOBYTE(wsa_data.wVersion) != 2 || HIBYTE(wsa_data.wVersion) != 2) {
WSACleanup();
WSACleanup();
utils::Error("Could not find a usable version of Winsock.dll\n");
}
#endif
@ -136,10 +136,10 @@ class TCPSocket {
*/
inline void SetNonBlock(bool non_block) {
#ifdef _WIN32
u_long mode = non_block ? 1 : 0;
if (ioctlsocket(sockfd, FIONBIO, &mode) != NO_ERROR) {
u_long mode = non_block ? 1 : 0;
if (ioctlsocket(sockfd, FIONBIO, &mode) != NO_ERROR) {
SockError("SetNonBlock");
}
}
#else
int flag = fcntl(sockfd, F_GETFL, 0);
if (flag == -1) {