third_party/libevent/evutil.c

   1 /*
   2  * Copyright (c) 2007 Niels Provos <provos@citi.umich.edu>
   3  * All rights reserved.
   4  *
   5  * Redistribution and use in source and binary forms, with or without
   6  * modification, are permitted provided that the following conditions
   7  * are met:
   8  * 1. Redistributions of source code must retain the above copyright
   9  *    notice, this list of conditions and the following disclaimer.
  10  * 2. Redistributions in binary form must reproduce the above copyright
  11  *    notice, this list of conditions and the following disclaimer in the
  12  *    documentation and/or other materials provided with the distribution.
  13  * 3. The name of the author may not be used to endorse or promote products
  14  *    derived from this software without specific prior written permission.
  15  *
  16  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
  17  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
  18  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
  19  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
  20  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
  21  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  22  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  23  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  24  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
  25  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  26  */
  27 #ifdef HAVE_CONFIG_H
  28 #include "config.h"
  29 #endif
  30
  31 #ifdef WIN32
  32 #include <winsock2.h>
  33 #define WIN32_LEAN_AND_MEAN
  34 #include <windows.h>
  35 #undef WIN32_LEAN_AND_MEAN
  36 #endif
  37
  38 #include <sys/types.h>
  39 #ifdef HAVE_SYS_SOCKET_H
  40 #include <sys/socket.h>
  41 #endif
  42 #ifdef HAVE_UNISTD_H
  43 #include <unistd.h>
  44 #endif
  45 #ifdef HAVE_FCNTL_H
  46 #include <fcntl.h>
  47 #endif
  48 #ifdef HAVE_STDLIB_H
  49 #include <stdlib.h>
  50 #endif
  51 #include <errno.h>
  52 #if defined WIN32 && !defined(HAVE_GETTIMEOFDAY_H)
  53 #include <sys/timeb.h>
  54 #endif
  55 #include <stdio.h>
  56 #include <signal.h>
  57
  58 #include <sys/queue.h>
  59 #include "event.h"
  60 #include "event-internal.h"
  61 #include "evutil.h"
  62 #include "log.h"
  63
  64 int
  65 evutil_socketpair(int family, int type, int protocol, int fd[2])
  66 {
  67 #ifndef WIN32
  68         return socketpair(family, type, protocol, fd);
  69 #else
  70         /* This code is originally from Tor.  Used with permission. */
  71
  72         /* This socketpair does not work when localhost is down. So
  73          * it's really not the same thing at all. But it's close enough
  74          * for now, and really, when localhost is down sometimes, we
  75          * have other problems too.
  76          */
  77         int listener = -1;
  78         int connector = -1;
  79         int acceptor = -1;
  80         struct sockaddr_in listen_addr;
  81         struct sockaddr_in connect_addr;
  82         int size;
  83         int saved_errno = -1;
  84
  85         if (protocol
  86 #ifdef AF_UNIX
  87                 || family != AF_UNIX
  88 #endif
  89                 ) {
  90                 EVUTIL_SET_SOCKET_ERROR(WSAEAFNOSUPPORT);
  91                 return -1;
  92         }
  93         if (!fd) {
  94                 EVUTIL_SET_SOCKET_ERROR(WSAEINVAL);
  95                 return -1;
  96         }
  97
  98         listener = socket(AF_INET, type, 0);
  99         if (listener < 0)
 100                 return -1;
 101         memset(&listen_addr, 0, sizeof(listen_addr));
 102         listen_addr.sin_family = AF_INET;
 103         listen_addr.sin_addr.s_addr = htonl(INADDR_LOOPBACK);
 104         listen_addr.sin_port = 0;       /* kernel chooses port.  */
 105         if (bind(listener, (struct sockaddr *) &listen_addr, sizeof (listen_addr))
 106                 == -1)
 107                 goto tidy_up_and_fail;
 108         if (listen(listener, 1) == -1)
 109                 goto tidy_up_and_fail;
 110
 111         connector = socket(AF_INET, type, 0);
 112         if (connector < 0)
 113                 goto tidy_up_and_fail;
 114         /* We want to find out the port number to connect to.  */
 115         size = sizeof(connect_addr);
 116         if (getsockname(listener, (struct sockaddr *) &connect_addr, &size) == -1)
 117                 goto tidy_up_and_fail;
 118         if (size != sizeof (connect_addr))
 119                 goto abort_tidy_up_and_fail;
 120         if (connect(connector, (struct sockaddr *) &connect_addr,
 121                                 sizeof(connect_addr)) == -1)
 122                 goto tidy_up_and_fail;
 123
 124         size = sizeof(listen_addr);
 125         acceptor = accept(listener, (struct sockaddr *) &listen_addr, &size);
 126         if (acceptor < 0)
 127                 goto tidy_up_and_fail;
 128         if (size != sizeof(listen_addr))
 129                 goto abort_tidy_up_and_fail;
 130         EVUTIL_CLOSESOCKET(listener);
 131         /* Now check we are talking to ourself by matching port and host on the
 132            two sockets.  */
 133         if (getsockname(connector, (struct sockaddr *) &connect_addr, &size) == -1)
 134                 goto tidy_up_and_fail;
 135         if (size != sizeof (connect_addr)
 136                 || listen_addr.sin_family != connect_addr.sin_family
 137                 || listen_addr.sin_addr.s_addr != connect_addr.sin_addr.s_addr
 138                 || listen_addr.sin_port != connect_addr.sin_port)
 139                 goto abort_tidy_up_and_fail;
 140         fd[0] = connector;
 141         fd[1] = acceptor;
 142
 143         return 0;
 144
 145  abort_tidy_up_and_fail:
 146         saved_errno = WSAECONNABORTED;
 147  tidy_up_and_fail:
 148         if (saved_errno < 0)
 149                 saved_errno = WSAGetLastError();
 150         if (listener != -1)
 151                 EVUTIL_CLOSESOCKET(listener);
 152         if (connector != -1)
 153                 EVUTIL_CLOSESOCKET(connector);
 154         if (acceptor != -1)
 155                 EVUTIL_CLOSESOCKET(acceptor);
 156
 157         EVUTIL_SET_SOCKET_ERROR(saved_errno);
 158         return -1;
 159 #endif
 160 }
 161
 162 int
 163 evutil_make_socket_nonblocking(int fd)
 164 {
 165 #ifdef WIN32
 166         {
 167                 unsigned long nonblocking = 1;
 168                 ioctlsocket(fd, FIONBIO, (unsigned long*) &nonblocking);
 169         }
 170 #else
 171         if (fcntl(fd, F_SETFL, O_NONBLOCK) == -1) {
 172                 event_warn("fcntl(O_NONBLOCK)");
 173                 return -1;
 174 }
 175 #endif
 176         return 0;
 177 }
 178
 179 ev_int64_t
 180 evutil_strtoll(const char *s, char **endptr, int base)
 181 {
 182 #ifdef HAVE_STRTOLL
 183         return (ev_int64_t)strtoll(s, endptr, base);
 184 #elif SIZEOF_LONG == 8
 185         return (ev_int64_t)strtol(s, endptr, base);
 186 #elif defined(WIN32) && defined(_MSC_VER) && _MSC_VER < 1300
 187         /* XXXX on old versions of MS APIs, we only support base
 188          * 10. */
 189         ev_int64_t r;
 190         if (base != 10)
 191                 return 0;
 192         r = (ev_int64_t) _atoi64(s);
 193         while (isspace(*s))
 194                 ++s;
 195         while (isdigit(*s))
 196                 ++s;
 197         if (endptr)
 198                 *endptr = (char*) s;
 199         return r;
 200 #elif defined(WIN32)
 201         return (ev_int64_t) _strtoi64(s, endptr, base);
 202 #else
 203 #error "I don't know how to parse 64-bit integers."
 204 #endif
 205 }
 206
 207 #ifndef _EVENT_HAVE_GETTIMEOFDAY
 208 int
 209 evutil_gettimeofday(struct timeval *tv, struct timezone *tz)
 210 {
 211         struct _timeb tb;
 212
 213         if(tv == NULL)
 214                 return -1;
 215
 216         _ftime(&tb);
 217         tv->tv_sec = (long) tb.time;
 218         tv->tv_usec = ((int) tb.millitm) * 1000;
 219         return 0;
 220 }
 221 #endif
 222
 223 int
 224 evutil_snprintf(char *buf, size_t buflen, const char *format, ...)
 225 {
 226         int r;
 227         va_list ap;
 228         va_start(ap, format);
 229         r = evutil_vsnprintf(buf, buflen, format, ap);
 230         va_end(ap);
 231         return r;
 232 }
 233
 234 int
 235 evutil_vsnprintf(char *buf, size_t buflen, const char *format, va_list ap)
 236 {
 237 #ifdef _MSC_VER
 238         int r = _vsnprintf(buf, buflen, format, ap);
 239         buf[buflen-1] = '\0';
 240         if (r >= 0)
 241                 return r;
 242         else
 243                 return _vscprintf(format, ap);
 244 #else
 245         int r = vsnprintf(buf, buflen, format, ap);
 246         buf[buflen-1] = '\0';
 247         return r;
 248 #endif
 249 }
 250
 251 static int
 252 evutil_issetugid(void)
 253 {
 254 #ifdef _EVENT_HAVE_ISSETUGID
 255         return issetugid();
 256 #else
 257
 258 #ifdef _EVENT_HAVE_GETEUID
 259         if (getuid() != geteuid())
 260                 return 1;
 261 #endif
 262 #ifdef _EVENT_HAVE_GETEGID
 263         if (getgid() != getegid())
 264                 return 1;
 265 #endif
 266         return 0;
 267 #endif
 268 }
 269
 270 const char *
 271 evutil_getenv(const char *varname)
 272 {
 273         if (evutil_issetugid())
 274                 return NULL;
 275
 276         return getenv(varname);
 277 }