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
64int
65evutil_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
162int
163evutil_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
179ev_int64_t
180evutil_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
208int
209evutil_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
223int
224evutil_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
234int
235evutil_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
251static int
252evutil_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
270const char *
271evutil_getenv(const char *varname)
272{
273	if (evutil_issetugid())
274		return NULL;
275
276	return getenv(varname);
277}
278