123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248 |
- /* Copyright 2003 Roger Dingledine */
- /* See LICENSE for licensing information */
- /* $Id$ */
- #include "../or/or.h"
- #ifdef MS_WINDOWS
- #include <io.h>
- #include <limits.h>
- #include <process.h>
- #endif
- #include "util.h"
- #include "log.h"
- void *tor_malloc(size_t size) {
- void *result;
- result = malloc(size);
- if(!result) {
- log_fn(LOG_ERR, "Out of memory. Dying.");
- exit(1);
- }
- return result;
- }
- void
- my_gettimeofday(struct timeval *timeval)
- {
- #ifdef HAVE_GETTIMEOFDAY
- if (gettimeofday(timeval, NULL)) {
- log_fn(LOG_ERR, "gettimeofday failed.");
- /* If gettimeofday dies, we have either given a bad timezone (we didn't),
- or segfaulted.*/
- exit(1);
- }
- #elif defined(HAVE_FTIME)
- ftime(timeval);
- #else
- #error "No way to get time."
- #endif
- return;
- }
- long
- tv_udiff(struct timeval *start, struct timeval *end)
- {
- long udiff;
- long end_usec = end->tv_usec;
- long secdiff = end->tv_sec - start->tv_sec;
- if (secdiff+1 > LONG_MAX/1000000) {
- log_fn(LOG_NOTICE, "comparing times too far apart.");
- return LONG_MAX;
- }
- udiff = secdiff*1000000L + (end_usec - start->tv_usec);
- if(udiff < 0) {
- log_fn(LOG_NOTICE, "start is after end. Returning 0.");
- return 0;
- }
- return udiff;
- }
- int tv_cmp(struct timeval *a, struct timeval *b) {
- if (a->tv_sec > b->tv_sec)
- return 1;
- if (a->tv_sec < b->tv_sec)
- return -1;
- if (a->tv_usec > b->tv_usec)
- return 1;
- if (a->tv_usec < b->tv_usec)
- return -1;
- return 0;
- }
- void tv_add(struct timeval *a, struct timeval *b) {
- a->tv_usec += b->tv_usec;
- a->tv_sec += b->tv_sec + (a->tv_usec / 1000000);
- a->tv_usec %= 1000000;
- }
- void tv_addms(struct timeval *a, long ms) {
- a->tv_usec += (ms * 1000) % 1000000;
- a->tv_sec += ((ms * 1000) / 1000000) + (a->tv_usec / 1000000);
- a->tv_usec %= 1000000;
- }
- void set_socket_nonblocking(int socket)
- {
- #ifdef MS_WINDOWS
- /* Yes means no and no means yes. Do you not want to be nonblocking? */
- int nonblocking = 0;
- ioctlsocket(socket, FIONBIO, (unsigned long*) &nonblocking);
- #else
- fcntl(socket, F_SETFL, O_NONBLOCK);
- #endif
- }
- int spawn_func(int (*func)(void *), void *data)
- {
- #ifdef MS_WINDOWS
- int rv;
- rv = _beginthread(func, 0, data);
- if (rv == (unsigned long) -1)
- return -1;
- return 0;
- #else
- pid_t pid;
- pid = fork();
- if (pid<0)
- return -1;
- if (pid==0) {
- /* Child */
- func(data);
- assert(0); /* Should never reach here. */
- return 0; /* suppress "control-reaches-end-of-non-void" warning. */
- } else {
- /* Parent */
- return 0;
- }
- #endif
- }
- void spawn_exit()
- {
- #ifdef MS_WINDOWS
- _endthread();
- #else
- exit(0);
- #endif
- }
- /* Fake socket pair over TCP. Code adapted from perl 5.8.0's util.c */
- int
- tor_socketpair(int family, int type, int protocol, int fd[2])
- {
- #ifdef HAVE_SOCKETPAIR_XXXX
- /* For testing purposes, we never fall back to real socketpairs. */
- return socketpair(family, type, protocol, fd);
- #else
- int listener = -1;
- int connector = -1;
- int acceptor = -1;
- struct sockaddr_in listen_addr;
- struct sockaddr_in connect_addr;
- size_t size;
-
- if (protocol
- #ifdef AF_UNIX
- || family != AF_UNIX
- #endif
- ) {
- #ifdef MS_WINDOWS
- errno = WSAEAFNOSUPPORT;
- #else
- errno = EAFNOSUPPORT;
- #endif
- return -1;
- }
- if (!fd) {
- errno = EINVAL;
- return -1;
- }
- listener = socket(AF_INET, type, 0);
- if (listener == -1)
- return -1;
- memset (&listen_addr, 0, sizeof (listen_addr));
- listen_addr.sin_family = AF_INET;
- listen_addr.sin_addr.s_addr = htonl (INADDR_LOOPBACK);
- listen_addr.sin_port = 0; /* kernel choses port. */
- if (bind(listener, (struct sockaddr *) &listen_addr, sizeof (listen_addr))
- == -1)
- goto tidy_up_and_fail;
- if (listen(listener, 1) == -1)
- goto tidy_up_and_fail;
- connector = socket(AF_INET, type, 0);
- if (connector == -1)
- goto tidy_up_and_fail;
- /* We want to find out the port number to connect to. */
- size = sizeof (connect_addr);
- if (getsockname(listener, (struct sockaddr *) &connect_addr, &size) == -1)
- goto tidy_up_and_fail;
- if (size != sizeof (connect_addr))
- goto abort_tidy_up_and_fail;
- if (connect(connector, (struct sockaddr *) &connect_addr,
- sizeof (connect_addr)) == -1)
- goto tidy_up_and_fail;
- size = sizeof (listen_addr);
- acceptor = accept(listener, (struct sockaddr *) &listen_addr, &size);
- if (acceptor == -1)
- goto tidy_up_and_fail;
- if (size != sizeof(listen_addr))
- goto abort_tidy_up_and_fail;
- close(listener);
- /* Now check we are talking to ourself by matching port and host on the
- two sockets. */
- if (getsockname(connector, (struct sockaddr *) &connect_addr, &size) == -1)
- goto tidy_up_and_fail;
- if (size != sizeof (connect_addr)
- || listen_addr.sin_family != connect_addr.sin_family
- || listen_addr.sin_addr.s_addr != connect_addr.sin_addr.s_addr
- || listen_addr.sin_port != connect_addr.sin_port) {
- goto abort_tidy_up_and_fail;
- }
- fd[0] = connector;
- fd[1] = acceptor;
- return 0;
- abort_tidy_up_and_fail:
- #ifdef MS_WINDOWS
- errno = WSAECONNABORTED;
- #else
- errno = ECONNABORTED; /* I hope this is portable and appropriate. */
- #endif
- tidy_up_and_fail:
- {
- int save_errno = errno;
- if (listener != -1)
- close(listener);
- if (connector != -1)
- close(connector);
- if (acceptor != -1)
- close(acceptor);
- errno = save_errno;
- return -1;
- }
- #endif
- }
- #ifdef MS_WINDOWS
- int correct_socket_errno(int s)
- {
- int optval, optvallen=sizeof(optval);
- assert(errno == WSAEWOULDBLOCK);
- if (getsockopt(s, SOL_SOCKET, SO_ERROR, (void*)&optval, &optvallen))
- return errno;
- if (optval)
- return optval;
- return WSAEWOULDBLOCK;
- }
- #endif
|