00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017
00018
00019
00020
00021 #include "GroupsockHelper.hh"
00022
00023 #if defined(__WIN32__) || defined(_WIN32)
00024 #include <time.h>
00025 extern "C" int initializeWinsockIfNecessary();
00026 #else
00027 #include <stdarg.h>
00028 #include <time.h>
00029 #include <fcntl.h>
00030 #define initializeWinsockIfNecessary() 1
00031 #endif
00032 #include <stdio.h>
00033
00034
00035 netAddressBits SendingInterfaceAddr = INADDR_ANY;
00036 netAddressBits ReceivingInterfaceAddr = INADDR_ANY;
00037
00038 static void socketErr(UsageEnvironment& env, char const* errorMsg) {
00039 env.setResultErrMsg(errorMsg);
00040 }
00041
00042 NoReuse::NoReuse(UsageEnvironment& env)
00043 : fEnv(env) {
00044 groupsockPriv(fEnv)->reuseFlag = 0;
00045 }
00046
00047 NoReuse::~NoReuse() {
00048 groupsockPriv(fEnv)->reuseFlag = 1;
00049 reclaimGroupsockPriv(fEnv);
00050 }
00051
00052
00053 _groupsockPriv* groupsockPriv(UsageEnvironment& env) {
00054 if (env.groupsockPriv == NULL) {
00055 _groupsockPriv* result = new _groupsockPriv;
00056 result->socketTable = NULL;
00057 result->reuseFlag = 1;
00058 env.groupsockPriv = result;
00059 }
00060 return (_groupsockPriv*)(env.groupsockPriv);
00061 }
00062
00063 void reclaimGroupsockPriv(UsageEnvironment& env) {
00064 _groupsockPriv* priv = (_groupsockPriv*)(env.groupsockPriv);
00065 if (priv->socketTable == NULL && priv->reuseFlag == 1) {
00066
00067 delete priv;
00068 env.groupsockPriv = NULL;
00069 }
00070 }
00071
00072 static int createSocket(int type) {
00073
00074
00075 int sock;
00076
00077 #ifdef SOCK_CLOEXEC
00078 sock = socket(AF_INET, type|SOCK_CLOEXEC, 0);
00079 if (sock != -1 || errno != EINVAL) return sock;
00080
00081 #endif
00082
00083 sock = socket(AF_INET, type, 0);
00084 #ifdef FD_CLOEXEC
00085 if (sock != -1) fcntl(sock, F_SETFD, FD_CLOEXEC);
00086 #endif
00087 return sock;
00088 }
00089
00090 int setupDatagramSocket(UsageEnvironment& env, Port port) {
00091 if (!initializeWinsockIfNecessary()) {
00092 socketErr(env, "Failed to initialize 'winsock': ");
00093 return -1;
00094 }
00095
00096 int newSocket = createSocket(SOCK_DGRAM);
00097 if (newSocket < 0) {
00098 socketErr(env, "unable to create datagram socket: ");
00099 return newSocket;
00100 }
00101
00102 int reuseFlag = groupsockPriv(env)->reuseFlag;
00103 reclaimGroupsockPriv(env);
00104 if (setsockopt(newSocket, SOL_SOCKET, SO_REUSEADDR,
00105 (const char*)&reuseFlag, sizeof reuseFlag) < 0) {
00106 socketErr(env, "setsockopt(SO_REUSEADDR) error: ");
00107 closeSocket(newSocket);
00108 return -1;
00109 }
00110
00111 #if defined(__WIN32__) || defined(_WIN32)
00112
00113 #else
00114 #ifdef SO_REUSEPORT
00115 if (setsockopt(newSocket, SOL_SOCKET, SO_REUSEPORT,
00116 (const char*)&reuseFlag, sizeof reuseFlag) < 0) {
00117 socketErr(env, "setsockopt(SO_REUSEPORT) error: ");
00118 closeSocket(newSocket);
00119 return -1;
00120 }
00121 #endif
00122
00123 #ifdef IP_MULTICAST_LOOP
00124 const u_int8_t loop = 1;
00125 if (setsockopt(newSocket, IPPROTO_IP, IP_MULTICAST_LOOP,
00126 (const char*)&loop, sizeof loop) < 0) {
00127 socketErr(env, "setsockopt(IP_MULTICAST_LOOP) error: ");
00128 closeSocket(newSocket);
00129 return -1;
00130 }
00131 #endif
00132 #endif
00133
00134
00135 netAddressBits addr = INADDR_ANY;
00136 #if defined(__WIN32__) || defined(_WIN32)
00137 #else
00138 if (port.num() != 0 || ReceivingInterfaceAddr != INADDR_ANY) {
00139 #endif
00140 if (port.num() == 0) addr = ReceivingInterfaceAddr;
00141 MAKE_SOCKADDR_IN(name, addr, port.num());
00142 if (bind(newSocket, (struct sockaddr*)&name, sizeof name) != 0) {
00143 char tmpBuffer[100];
00144 sprintf(tmpBuffer, "bind() error (port number: %d): ",
00145 ntohs(port.num()));
00146 socketErr(env, tmpBuffer);
00147 closeSocket(newSocket);
00148 return -1;
00149 }
00150 #if defined(__WIN32__) || defined(_WIN32)
00151 #else
00152 }
00153 #endif
00154
00155
00156 if (SendingInterfaceAddr != INADDR_ANY) {
00157 struct in_addr addr;
00158 addr.s_addr = SendingInterfaceAddr;
00159
00160 if (setsockopt(newSocket, IPPROTO_IP, IP_MULTICAST_IF,
00161 (const char*)&addr, sizeof addr) < 0) {
00162 socketErr(env, "error setting outgoing multicast interface: ");
00163 closeSocket(newSocket);
00164 return -1;
00165 }
00166 }
00167
00168 return newSocket;
00169 }
00170
00171 Boolean makeSocketNonBlocking(int sock) {
00172 #if defined(__WIN32__) || defined(_WIN32)
00173 unsigned long arg = 1;
00174 return ioctlsocket(sock, FIONBIO, &arg) == 0;
00175 #elif defined(VXWORKS)
00176 int arg = 1;
00177 return ioctl(sock, FIONBIO, (int)&arg) == 0;
00178 #else
00179 int curFlags = fcntl(sock, F_GETFL, 0);
00180 return fcntl(sock, F_SETFL, curFlags|O_NONBLOCK) >= 0;
00181 #endif
00182 }
00183
00184 Boolean makeSocketBlocking(int sock) {
00185 #if defined(__WIN32__) || defined(_WIN32)
00186 unsigned long arg = 0;
00187 return ioctlsocket(sock, FIONBIO, &arg) == 0;
00188 #elif defined(VXWORKS)
00189 int arg = 0;
00190 return ioctl(sock, FIONBIO, (int)&arg) == 0;
00191 #else
00192 int curFlags = fcntl(sock, F_GETFL, 0);
00193 return fcntl(sock, F_SETFL, curFlags&(~O_NONBLOCK)) >= 0;
00194 #endif
00195 }
00196
00197 int setupStreamSocket(UsageEnvironment& env,
00198 Port port, Boolean makeNonBlocking) {
00199 if (!initializeWinsockIfNecessary()) {
00200 socketErr(env, "Failed to initialize 'winsock': ");
00201 return -1;
00202 }
00203
00204 int newSocket = createSocket(SOCK_STREAM);
00205 if (newSocket < 0) {
00206 socketErr(env, "unable to create stream socket: ");
00207 return newSocket;
00208 }
00209
00210 int reuseFlag = groupsockPriv(env)->reuseFlag;
00211 reclaimGroupsockPriv(env);
00212 if (setsockopt(newSocket, SOL_SOCKET, SO_REUSEADDR,
00213 (const char*)&reuseFlag, sizeof reuseFlag) < 0) {
00214 socketErr(env, "setsockopt(SO_REUSEADDR) error: ");
00215 closeSocket(newSocket);
00216 return -1;
00217 }
00218
00219
00220
00221
00222 #ifdef REUSE_FOR_TCP
00223 #if defined(__WIN32__) || defined(_WIN32)
00224
00225 #else
00226 #ifdef SO_REUSEPORT
00227 if (setsockopt(newSocket, SOL_SOCKET, SO_REUSEPORT,
00228 (const char*)&reuseFlag, sizeof reuseFlag) < 0) {
00229 socketErr(env, "setsockopt(SO_REUSEPORT) error: ");
00230 closeSocket(newSocket);
00231 return -1;
00232 }
00233 #endif
00234 #endif
00235 #endif
00236
00237
00238 #if defined(__WIN32__) || defined(_WIN32)
00239 #else
00240 if (port.num() != 0 || ReceivingInterfaceAddr != INADDR_ANY) {
00241 #endif
00242 MAKE_SOCKADDR_IN(name, ReceivingInterfaceAddr, port.num());
00243 if (bind(newSocket, (struct sockaddr*)&name, sizeof name) != 0) {
00244 char tmpBuffer[100];
00245 sprintf(tmpBuffer, "bind() error (port number: %d): ",
00246 ntohs(port.num()));
00247 socketErr(env, tmpBuffer);
00248 closeSocket(newSocket);
00249 return -1;
00250 }
00251 #if defined(__WIN32__) || defined(_WIN32)
00252 #else
00253 }
00254 #endif
00255
00256 if (makeNonBlocking) {
00257 if (!makeSocketNonBlocking(newSocket)) {
00258 socketErr(env, "failed to make non-blocking: ");
00259 closeSocket(newSocket);
00260 return -1;
00261 }
00262 }
00263
00264 return newSocket;
00265 }
00266
00267 int readSocket(UsageEnvironment& env,
00268 int socket, unsigned char* buffer, unsigned bufferSize,
00269 struct sockaddr_in& fromAddress) {
00270 SOCKLEN_T addressSize = sizeof fromAddress;
00271 int bytesRead = recvfrom(socket, (char*)buffer, bufferSize, 0,
00272 (struct sockaddr*)&fromAddress,
00273 &addressSize);
00274 if (bytesRead < 0) {
00275
00276 int err = env.getErrno();
00277 if (err == 111
00278 #if defined(__WIN32__) || defined(_WIN32)
00279
00280
00281
00282
00283
00284
00285 || err == 0 || err == EWOULDBLOCK
00286 #else
00287 || err == EAGAIN
00288 #endif
00289 || err == 113 ) {
00290 fromAddress.sin_addr.s_addr = 0;
00291 return 0;
00292 }
00293
00294 socketErr(env, "recvfrom() error: ");
00295 }
00296
00297 return bytesRead;
00298 }
00299
00300 Boolean writeSocket(UsageEnvironment& env,
00301 int socket, struct in_addr address, Port port,
00302 u_int8_t ttlArg,
00303 unsigned char* buffer, unsigned bufferSize) {
00304 do {
00305 if (ttlArg != 0) {
00306
00307 #if defined(__WIN32__) || defined(_WIN32)
00308 #define TTL_TYPE int
00309 #else
00310 #define TTL_TYPE u_int8_t
00311 #endif
00312 TTL_TYPE ttl = (TTL_TYPE)ttlArg;
00313 if (setsockopt(socket, IPPROTO_IP, IP_MULTICAST_TTL,
00314 (const char*)&ttl, sizeof ttl) < 0) {
00315 socketErr(env, "setsockopt(IP_MULTICAST_TTL) error: ");
00316 break;
00317 }
00318 }
00319
00320 MAKE_SOCKADDR_IN(dest, address.s_addr, port.num());
00321 int bytesSent = sendto(socket, (char*)buffer, bufferSize, 0,
00322 (struct sockaddr*)&dest, sizeof dest);
00323 if (bytesSent != (int)bufferSize) {
00324 char tmpBuf[100];
00325 sprintf(tmpBuf, "writeSocket(%d), sendTo() error: wrote %d bytes instead of %u: ", socket, bytesSent, bufferSize);
00326 socketErr(env, tmpBuf);
00327 break;
00328 }
00329
00330 return True;
00331 } while (0);
00332
00333 return False;
00334 }
00335
00336 static unsigned getBufferSize(UsageEnvironment& env, int bufOptName,
00337 int socket) {
00338 unsigned curSize;
00339 SOCKLEN_T sizeSize = sizeof curSize;
00340 if (getsockopt(socket, SOL_SOCKET, bufOptName,
00341 (char*)&curSize, &sizeSize) < 0) {
00342 socketErr(env, "getBufferSize() error: ");
00343 return 0;
00344 }
00345
00346 return curSize;
00347 }
00348 unsigned getSendBufferSize(UsageEnvironment& env, int socket) {
00349 return getBufferSize(env, SO_SNDBUF, socket);
00350 }
00351 unsigned getReceiveBufferSize(UsageEnvironment& env, int socket) {
00352 return getBufferSize(env, SO_RCVBUF, socket);
00353 }
00354
00355 static unsigned setBufferTo(UsageEnvironment& env, int bufOptName,
00356 int socket, unsigned requestedSize) {
00357 SOCKLEN_T sizeSize = sizeof requestedSize;
00358 setsockopt(socket, SOL_SOCKET, bufOptName, (char*)&requestedSize, sizeSize);
00359
00360
00361 return getBufferSize(env, bufOptName, socket);
00362 }
00363 unsigned setSendBufferTo(UsageEnvironment& env,
00364 int socket, unsigned requestedSize) {
00365 return setBufferTo(env, SO_SNDBUF, socket, requestedSize);
00366 }
00367 unsigned setReceiveBufferTo(UsageEnvironment& env,
00368 int socket, unsigned requestedSize) {
00369 return setBufferTo(env, SO_RCVBUF, socket, requestedSize);
00370 }
00371
00372 static unsigned increaseBufferTo(UsageEnvironment& env, int bufOptName,
00373 int socket, unsigned requestedSize) {
00374
00375
00376 unsigned curSize = getBufferSize(env, bufOptName, socket);
00377
00378
00379
00380 while (requestedSize > curSize) {
00381 SOCKLEN_T sizeSize = sizeof requestedSize;
00382 if (setsockopt(socket, SOL_SOCKET, bufOptName,
00383 (char*)&requestedSize, sizeSize) >= 0) {
00384
00385 return requestedSize;
00386 }
00387 requestedSize = (requestedSize+curSize)/2;
00388 }
00389
00390 return getBufferSize(env, bufOptName, socket);
00391 }
00392 unsigned increaseSendBufferTo(UsageEnvironment& env,
00393 int socket, unsigned requestedSize) {
00394 return increaseBufferTo(env, SO_SNDBUF, socket, requestedSize);
00395 }
00396 unsigned increaseReceiveBufferTo(UsageEnvironment& env,
00397 int socket, unsigned requestedSize) {
00398 return increaseBufferTo(env, SO_RCVBUF, socket, requestedSize);
00399 }
00400
00401 Boolean socketJoinGroup(UsageEnvironment& env, int socket,
00402 netAddressBits groupAddress){
00403 if (!IsMulticastAddress(groupAddress)) return True;
00404
00405 struct ip_mreq imr;
00406 imr.imr_multiaddr.s_addr = groupAddress;
00407 imr.imr_interface.s_addr = ReceivingInterfaceAddr;
00408 if (setsockopt(socket, IPPROTO_IP, IP_ADD_MEMBERSHIP,
00409 (const char*)&imr, sizeof (struct ip_mreq)) < 0) {
00410 #if defined(__WIN32__) || defined(_WIN32)
00411 if (env.getErrno() != 0) {
00412
00413
00414 #endif
00415 socketErr(env, "setsockopt(IP_ADD_MEMBERSHIP) error: ");
00416 return False;
00417 #if defined(__WIN32__) || defined(_WIN32)
00418 }
00419 #endif
00420 }
00421
00422 return True;
00423 }
00424
00425 Boolean socketLeaveGroup(UsageEnvironment&, int socket,
00426 netAddressBits groupAddress) {
00427 if (!IsMulticastAddress(groupAddress)) return True;
00428
00429 struct ip_mreq imr;
00430 imr.imr_multiaddr.s_addr = groupAddress;
00431 imr.imr_interface.s_addr = ReceivingInterfaceAddr;
00432 if (setsockopt(socket, IPPROTO_IP, IP_DROP_MEMBERSHIP,
00433 (const char*)&imr, sizeof (struct ip_mreq)) < 0) {
00434 return False;
00435 }
00436
00437 return True;
00438 }
00439
00440
00441
00442
00443 #if !defined(IP_ADD_SOURCE_MEMBERSHIP)
00444 struct ip_mreq_source {
00445 struct in_addr imr_multiaddr;
00446 struct in_addr imr_sourceaddr;
00447 struct in_addr imr_interface;
00448 };
00449 #endif
00450
00451 #ifndef IP_ADD_SOURCE_MEMBERSHIP
00452
00453 #ifdef LINUX
00454 #define IP_ADD_SOURCE_MEMBERSHIP 39
00455 #define IP_DROP_SOURCE_MEMBERSHIP 40
00456 #else
00457 #define IP_ADD_SOURCE_MEMBERSHIP 25
00458 #define IP_DROP_SOURCE_MEMBERSHIP 26
00459 #endif
00460
00461 #endif
00462
00463 Boolean socketJoinGroupSSM(UsageEnvironment& env, int socket,
00464 netAddressBits groupAddress,
00465 netAddressBits sourceFilterAddr) {
00466 if (!IsMulticastAddress(groupAddress)) return True;
00467
00468 struct ip_mreq_source imr;
00469 imr.imr_multiaddr.s_addr = groupAddress;
00470 imr.imr_sourceaddr.s_addr = sourceFilterAddr;
00471 imr.imr_interface.s_addr = ReceivingInterfaceAddr;
00472 if (setsockopt(socket, IPPROTO_IP, IP_ADD_SOURCE_MEMBERSHIP,
00473 (const char*)&imr, sizeof (struct ip_mreq_source)) < 0) {
00474 socketErr(env, "setsockopt(IP_ADD_SOURCE_MEMBERSHIP) error: ");
00475 return False;
00476 }
00477
00478 return True;
00479 }
00480
00481 Boolean socketLeaveGroupSSM(UsageEnvironment& , int socket,
00482 netAddressBits groupAddress,
00483 netAddressBits sourceFilterAddr) {
00484 if (!IsMulticastAddress(groupAddress)) return True;
00485
00486 struct ip_mreq_source imr;
00487 imr.imr_multiaddr.s_addr = groupAddress;
00488 imr.imr_sourceaddr.s_addr = sourceFilterAddr;
00489 imr.imr_interface.s_addr = ReceivingInterfaceAddr;
00490 if (setsockopt(socket, IPPROTO_IP, IP_DROP_SOURCE_MEMBERSHIP,
00491 (const char*)&imr, sizeof (struct ip_mreq_source)) < 0) {
00492 return False;
00493 }
00494
00495 return True;
00496 }
00497
00498 static Boolean getSourcePort0(int socket, portNumBits& resultPortNum) {
00499 sockaddr_in test; test.sin_port = 0;
00500 SOCKLEN_T len = sizeof test;
00501 if (getsockname(socket, (struct sockaddr*)&test, &len) < 0) return False;
00502
00503 resultPortNum = ntohs(test.sin_port);
00504 return True;
00505 }
00506
00507 Boolean getSourcePort(UsageEnvironment& env, int socket, Port& port) {
00508 portNumBits portNum = 0;
00509 if (!getSourcePort0(socket, portNum) || portNum == 0) {
00510
00511 MAKE_SOCKADDR_IN(name, INADDR_ANY, 0);
00512 bind(socket, (struct sockaddr*)&name, sizeof name);
00513
00514 if (!getSourcePort0(socket, portNum) || portNum == 0) {
00515 socketErr(env, "getsockname() error: ");
00516 return False;
00517 }
00518 }
00519
00520 port = Port(portNum);
00521 return True;
00522 }
00523
00524 static Boolean badAddressForUs(netAddressBits addr) {
00525
00526 netAddressBits nAddr = htonl(addr);
00527 return (nAddr == 0x7F000001
00528 || nAddr == 0
00529 || nAddr == (netAddressBits)(~0));
00530 }
00531
00532 Boolean loopbackWorks = 1;
00533
00534 netAddressBits ourIPAddress(UsageEnvironment& env) {
00535 static netAddressBits ourAddress = 0;
00536 int sock = -1;
00537 struct in_addr testAddr;
00538
00539 if (ourAddress == 0) {
00540
00541 struct sockaddr_in fromAddr;
00542 fromAddr.sin_addr.s_addr = 0;
00543
00544
00545
00546
00547
00548 do {
00549 loopbackWorks = 0;
00550
00551 testAddr.s_addr = our_inet_addr("228.67.43.91");
00552 Port testPort(15947);
00553
00554 sock = setupDatagramSocket(env, testPort);
00555 if (sock < 0) break;
00556
00557 if (!socketJoinGroup(env, sock, testAddr.s_addr)) break;
00558
00559 unsigned char testString[] = "hostIdTest";
00560 unsigned testStringLength = sizeof testString;
00561
00562 if (!writeSocket(env, sock, testAddr, testPort, 0,
00563 testString, testStringLength)) break;
00564
00565
00566 fd_set rd_set;
00567 FD_ZERO(&rd_set);
00568 FD_SET((unsigned)sock, &rd_set);
00569 const unsigned numFds = sock+1;
00570 struct timeval timeout;
00571 timeout.tv_sec = 5;
00572 timeout.tv_usec = 0;
00573 int result = select(numFds, &rd_set, NULL, NULL, &timeout);
00574 if (result <= 0) break;
00575
00576 unsigned char readBuffer[20];
00577 int bytesRead = readSocket(env, sock,
00578 readBuffer, sizeof readBuffer,
00579 fromAddr);
00580 if (bytesRead != (int)testStringLength
00581 || strncmp((char*)readBuffer, (char*)testString, testStringLength) != 0) {
00582 break;
00583 }
00584
00585 loopbackWorks = 1;
00586 } while (0);
00587
00588 if (sock >= 0) {
00589 socketLeaveGroup(env, sock, testAddr.s_addr);
00590 closeSocket(sock);
00591 }
00592
00593 if (!loopbackWorks) do {
00594
00595
00596 char hostname[100];
00597 hostname[0] = '\0';
00598 int result = gethostname(hostname, sizeof hostname);
00599 if (result != 0 || hostname[0] == '\0') {
00600 env.setResultErrMsg("initial gethostname() failed");
00601 break;
00602 }
00603
00604
00605 NetAddressList addresses(hostname);
00606 NetAddressList::Iterator iter(addresses);
00607 NetAddress const* address;
00608
00609
00610 netAddressBits addr = 0;
00611 while ((address = iter.nextAddress()) != NULL) {
00612 netAddressBits a = *(netAddressBits*)(address->data());
00613 if (!badAddressForUs(a)) {
00614 addr = a;
00615 break;
00616 }
00617 }
00618
00619
00620 fromAddr.sin_addr.s_addr = addr;
00621 } while (0);
00622
00623
00624 netAddressBits from = fromAddr.sin_addr.s_addr;
00625 if (badAddressForUs(from)) {
00626 char tmp[100];
00627 sprintf(tmp, "This computer has an invalid IP address: %s", AddressString(from).val());
00628 env.setResultMsg(tmp);
00629 from = 0;
00630 }
00631
00632 ourAddress = from;
00633
00634
00635
00636 struct timeval timeNow;
00637 gettimeofday(&timeNow, NULL);
00638 unsigned seed = ourAddress^timeNow.tv_sec^timeNow.tv_usec;
00639 our_srandom(seed);
00640 }
00641 return ourAddress;
00642 }
00643
00644 netAddressBits chooseRandomIPv4SSMAddress(UsageEnvironment& env) {
00645
00646 (void) ourIPAddress(env);
00647
00648
00649
00650 netAddressBits const first = 0xE8000100, lastPlus1 = 0xE8FFFFFF;
00651 netAddressBits const range = lastPlus1 - first;
00652
00653 return ntohl(first + ((netAddressBits)our_random())%range);
00654 }
00655
00656 char const* timestampString() {
00657 struct timeval tvNow;
00658 gettimeofday(&tvNow, NULL);
00659
00660 #if !defined(_WIN32_WCE)
00661 static char timeString[9];
00662 char const* ctimeResult = ctime((time_t*)&tvNow.tv_sec);
00663 if (ctimeResult == NULL) {
00664 sprintf(timeString, "??:??:??");
00665 } else {
00666 char const* from = &ctimeResult[11];
00667 int i;
00668 for (i = 0; i < 8; ++i) {
00669 timeString[i] = from[i];
00670 }
00671 timeString[i] = '\0';
00672 }
00673 #else
00674
00675
00676
00677 static char timeString[50];
00678 sprintf(timeString, "%lu.%06ld", tvNow.tv_sec, tvNow.tv_usec);
00679 #endif
00680
00681 return (char const*)&timeString;
00682 }
00683
00684 #if defined(__WIN32__) || defined(_WIN32)
00685
00686 #if !defined(_WIN32_WCE)
00687 #include <sys/timeb.h>
00688 #endif
00689
00690 int gettimeofday(struct timeval* tp, int* ) {
00691 #if defined(_WIN32_WCE)
00692
00693 static const unsigned __int64 epoch = 116444736000000000LL;
00694
00695 FILETIME file_time;
00696 SYSTEMTIME system_time;
00697 ULARGE_INTEGER ularge;
00698
00699 GetSystemTime(&system_time);
00700 SystemTimeToFileTime(&system_time, &file_time);
00701 ularge.LowPart = file_time.dwLowDateTime;
00702 ularge.HighPart = file_time.dwHighDateTime;
00703
00704 tp->tv_sec = (long) ((ularge.QuadPart - epoch) / 10000000L);
00705 tp->tv_usec = (long) (system_time.wMilliseconds * 1000);
00706 #else
00707 static LARGE_INTEGER tickFrequency, epochOffset;
00708
00709
00710
00711 static Boolean isFirstCall = True;
00712
00713 LARGE_INTEGER tickNow;
00714 QueryPerformanceCounter(&tickNow);
00715
00716 if (isFirstCall) {
00717 struct timeb tb;
00718 ftime(&tb);
00719 tp->tv_sec = tb.time;
00720 tp->tv_usec = 1000*tb.millitm;
00721
00722
00723 QueryPerformanceFrequency(&tickFrequency);
00724
00725
00726 epochOffset.QuadPart
00727 = tb.time*tickFrequency.QuadPart + (tb.millitm*tickFrequency.QuadPart)/1000 - tickNow.QuadPart;
00728
00729 isFirstCall = False;
00730 } else {
00731
00732 tickNow.QuadPart += epochOffset.QuadPart;
00733
00734 tp->tv_sec = (long) (tickNow.QuadPart / tickFrequency.QuadPart);
00735 tp->tv_usec = (long) (((tickNow.QuadPart % tickFrequency.QuadPart) * 1000000L) / tickFrequency.QuadPart);
00736 }
00737 #endif
00738 return 0;
00739 }
00740 #endif