-#include <sys/time.h>
#ifdef WIN32
#include <winsock2.h>
#else
#include <sys/socket.h>
+#include <sys/time.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <netdb.h>
#endif
#include "netbase.h"
+#include "util.h"
extern int GetRandInt(int nMax);
u = n - 0x83aa7e80; // 2208988800 1970 - 1900 in seconds
}
-inline void HTONL_FP(l_fp *h, l_fp *n)
-{
- (n)->Ul_i.Xl_ui = htonl((h)->Ul_i.Xl_ui);
- (n)->Ul_f.Xl_uf = htonl((h)->Ul_f.Xl_uf);
-}
-
inline void ntohl_fp(l_fp *n, l_fp *h)
{
(h)->Ul_i.Xl_ui = ntohl((n)->Ul_i.Xl_ui);
// ... To be continued
};
-#ifdef WIN32
-bool InitWithRandom(SOCKET &sockfd, int &servlen, struct sockaddr *pcliaddr)
-#else
-bool InitWithRandom(int &sockfd, socklen_t &servlen, struct sockaddr *pcliaddr)
-#endif
+bool InitWithRandom(SOCKET &sockfd, socklen_t &servlen, struct sockaddr *pcliaddr)
{
int nAttempt = 0;
sockfd = socket(AF_INET, SOCK_DGRAM, 0);
- if (sockfd == -1)
+ if (sockfd == INVALID_SOCKET)
continue; // socket initialization error
if (connect(sockfd, (struct sockaddr *) &servaddr, sizeof(servaddr)) == -1 )
return false;
}
-#ifdef WIN32
-bool InitWithHost(std::string &strHostName, SOCKET &sockfd, int &servlen, struct sockaddr *pcliaddr)
-#else
-bool InitWithHost(std::string &strHostName, int &sockfd, socklen_t &servlen, struct sockaddr *pcliaddr)
-#endif
+bool InitWithHost(std::string &strHostName, SOCKET &sockfd, socklen_t &servlen, struct sockaddr *pcliaddr)
{
sockfd = -1;
sockfd = socket(AF_INET, SOCK_DGRAM, 0);
- if (sockfd == -1)
+ if (sockfd == INVALID_SOCKET)
return false; // socket initialization error
if (connect(sockfd, (struct sockaddr *) &servaddr, sizeof(servaddr)) == -1 )
return false; // "connection" error
}
+
*pcliaddr = *((struct sockaddr *) &servaddr);
servlen = sizeof(servaddr);
}
+int64_t DoReq(SOCKET sockfd, socklen_t servlen, struct sockaddr cliaddr)
+{
+
#ifdef WIN32
-int64_t DoReq(SOCKET sockfd, int servlen, struct sockaddr cliaddr)
+ u_long nOne = 1;
+ if (ioctlsocket(sockfd, FIONBIO, &nOne) == SOCKET_ERROR)
+ {
+ printf("ConnectSocket() : ioctlsocket non-blocking setting failed, error %d\n", WSAGetLastError());
#else
-int64_t DoReq(int sockfd, socklen_t servlen, struct sockaddr cliaddr)
+ if (fcntl(sockfd, F_SETFL, O_NONBLOCK) == SOCKET_ERROR)
+ {
+ printf("ConnectSocket() : fcntl non-blocking setting failed, error %d\n", errno);
#endif
-{
+ return -2;
+ }
+
struct pkt *msg = new pkt;
struct pkt *prt = new pkt;
msg->xmt.Ul_f.Xl_f=0;
int len=48;
- sendto(sockfd, (char *) msg, len, 0, &cliaddr, servlen);
- int n = recvfrom(sockfd, (char *) msg, len, 0, NULL, NULL);
+ int retcode = sendto(sockfd, (char *) msg, len, 0, &cliaddr, servlen);
+ if (retcode < 0) {
+ printf("sendto() failed: %d", retcode);
+ return -3;
+ }
+
+ fd_set fdset;
+ struct timeval timeout = {5, 0};
+ FD_ZERO(&fdset);
+ FD_SET(sockfd, &fdset);
+
+ retcode = select(sockfd + 1, &fdset, NULL, NULL, &timeout);
+ if (retcode <= 0)
+ {
+ printf("recvfrom() error");
+ return -4;
+ }
+
+ recvfrom(sockfd, (char *) msg, len, 0, NULL, NULL);
ntohl_fp(&msg->rec, &prt->rec);
ntohl_fp(&msg->xmt, &prt->xmt);
{
struct sockaddr cliaddr;
-#ifdef WIN32
SOCKET sockfd;
- int servlen;
-#else
- int sockfd;
socklen_t servlen;
-#endif
if (!InitWithRandom(sockfd, servlen, &cliaddr))
return -1;
int64_t nTime = DoReq(sockfd, servlen, cliaddr);
-#ifdef WIN32
closesocket(sockfd);
-#else
- close(sockfd);
-#endif
return nTime;
}
{
struct sockaddr cliaddr;
-#ifdef WIN32
SOCKET sockfd;
- int servlen;
-#else
- int sockfd;
socklen_t servlen;
-#endif
if (!InitWithHost(strHostName, sockfd, servlen, &cliaddr))
return -1;
int64_t nTime = DoReq(sockfd, servlen, cliaddr);
-#ifdef WIN32
closesocket(sockfd);
-#else
- close(sockfd);
-#endif
return nTime;
}