2005-04-01 22:15:56 +02:00
|
|
|
/* Copyright 2003-2004 Roger Dingledine
|
2006-02-09 06:46:49 +01:00
|
|
|
* Copyright 2004-2006 Roger Dingledine, Nick Mathewson */
|
2004-11-01 21:41:47 +01:00
|
|
|
/* See LICENSE for licensing information */
|
|
|
|
/* $Id$ */
|
2005-12-14 21:40:40 +01:00
|
|
|
const char compat_c_id[] =
|
|
|
|
"$Id$";
|
2004-11-01 21:41:47 +01:00
|
|
|
|
2005-06-11 07:31:17 +02:00
|
|
|
/**
|
|
|
|
* \file compat.c
|
|
|
|
* \brief Wrappers to make calls more portable. This code defines
|
|
|
|
* functions such as tor_malloc, tor_snprintf, get/set various data types,
|
|
|
|
* renaming, setting socket options, switching user IDs. It is basically
|
|
|
|
* where the non-portable items are conditionally included depending on
|
|
|
|
* the platform.
|
|
|
|
**/
|
|
|
|
|
2005-06-18 04:17:11 +02:00
|
|
|
/* This is required on rh7 to make strptime not complain.
|
|
|
|
* We also need it to make memmem get defined (where available)
|
|
|
|
*/
|
2004-11-01 21:41:47 +01:00
|
|
|
#define _GNU_SOURCE
|
|
|
|
|
|
|
|
#include "orconfig.h"
|
2004-11-04 05:01:19 +01:00
|
|
|
#include "compat.h"
|
2004-11-01 21:41:47 +01:00
|
|
|
|
|
|
|
#ifdef MS_WINDOWS
|
|
|
|
#include <process.h>
|
2006-08-04 20:32:43 +02:00
|
|
|
#include <windows.h>
|
2004-11-01 21:41:47 +01:00
|
|
|
#endif
|
2006-08-04 20:32:43 +02:00
|
|
|
|
2004-11-01 21:41:47 +01:00
|
|
|
#ifdef HAVE_UNAME
|
|
|
|
#include <sys/utsname.h>
|
|
|
|
#endif
|
|
|
|
#ifdef HAVE_SYS_TIME_H
|
|
|
|
#include <sys/time.h>
|
|
|
|
#endif
|
|
|
|
#ifdef HAVE_UNISTD_H
|
|
|
|
#include <unistd.h>
|
|
|
|
#endif
|
|
|
|
#ifdef HAVE_SYS_FCNTL_H
|
|
|
|
#include <sys/fcntl.h>
|
|
|
|
#endif
|
|
|
|
#ifdef HAVE_PWD_H
|
|
|
|
#include <pwd.h>
|
|
|
|
#endif
|
|
|
|
#ifdef HAVE_GRP_H
|
|
|
|
#include <grp.h>
|
|
|
|
#endif
|
|
|
|
#ifdef HAVE_FCNTL_H
|
|
|
|
#include <fcntl.h>
|
|
|
|
#endif
|
|
|
|
#ifdef HAVE_SYS_RESOURCE_H
|
|
|
|
#include <sys/resource.h>
|
|
|
|
#endif
|
|
|
|
#ifdef HAVE_ERRNO_H
|
|
|
|
#include <errno.h>
|
|
|
|
#endif
|
|
|
|
#ifdef HAVE_NETINET_IN_H
|
|
|
|
#include <netinet/in.h>
|
|
|
|
#endif
|
|
|
|
#ifdef HAVE_ARPA_INET_H
|
|
|
|
#include <arpa/inet.h>
|
|
|
|
#endif
|
|
|
|
#ifndef HAVE_GETTIMEOFDAY
|
|
|
|
#ifdef HAVE_FTIME
|
|
|
|
#include <sys/timeb.h>
|
|
|
|
#endif
|
|
|
|
#endif
|
|
|
|
#ifdef HAVE_SYS_SOCKET_H
|
|
|
|
#include <sys/socket.h>
|
|
|
|
#endif
|
2004-12-22 06:29:06 +01:00
|
|
|
#ifdef HAVE_NETDB_H
|
|
|
|
#include <netdb.h>
|
|
|
|
#endif
|
2004-11-01 21:41:47 +01:00
|
|
|
#ifdef HAVE_SYS_PARAM_H
|
|
|
|
#include <sys/param.h> /* FreeBSD needs this to know what version it is */
|
|
|
|
#endif
|
|
|
|
#include <stdarg.h>
|
|
|
|
#include <stdio.h>
|
|
|
|
#include <stdlib.h>
|
|
|
|
#include <string.h>
|
2004-11-03 08:29:03 +01:00
|
|
|
#include <assert.h>
|
2005-01-27 23:34:48 +01:00
|
|
|
#ifdef HAVE_PTHREAD_H
|
|
|
|
#include <pthread.h>
|
|
|
|
#endif
|
2005-09-15 01:27:52 +02:00
|
|
|
#ifdef HAVE_UTIME_H
|
|
|
|
#include <utime.h>
|
|
|
|
#endif
|
|
|
|
#ifdef HAVE_SYS_UTIME_H
|
|
|
|
#include <sys/utime.h>
|
|
|
|
#endif
|
2006-05-28 18:54:39 +02:00
|
|
|
#ifdef HAVE_SYS_MMAN_H
|
|
|
|
#include <sys/mman.h>
|
|
|
|
#endif
|
2004-11-01 21:41:47 +01:00
|
|
|
|
|
|
|
#include "log.h"
|
|
|
|
#include "util.h"
|
|
|
|
|
2004-11-09 02:24:10 +01:00
|
|
|
/* Inline the strl functions if the platform doesn't have them. */
|
2004-11-02 20:25:52 +01:00
|
|
|
#ifndef HAVE_STRLCPY
|
|
|
|
#include "strlcpy.c"
|
|
|
|
#endif
|
|
|
|
#ifndef HAVE_STRLCAT
|
|
|
|
#include "strlcat.c"
|
|
|
|
#endif
|
|
|
|
|
2004-12-22 06:29:06 +01:00
|
|
|
/* used by inet_addr, not defined on solaris anywhere!? */
|
|
|
|
#ifndef INADDR_NONE
|
|
|
|
#define INADDR_NONE ((unsigned long) -1)
|
|
|
|
#endif
|
|
|
|
|
2006-08-04 20:32:43 +02:00
|
|
|
#ifdef HAVE_SYS_MMAN_H
|
|
|
|
tor_mmap_t *
|
2006-08-05 19:53:21 +02:00
|
|
|
tor_mmap_file(const char *filename)
|
2006-05-28 18:54:39 +02:00
|
|
|
{
|
|
|
|
int fd; /* router file */
|
|
|
|
char *string;
|
|
|
|
int page_size;
|
2006-08-04 20:32:43 +02:00
|
|
|
tor_mmap_t *res;
|
|
|
|
size_t size;
|
2006-05-28 18:54:39 +02:00
|
|
|
|
|
|
|
tor_assert(filename);
|
|
|
|
|
|
|
|
fd = open(filename, O_RDONLY, 0);
|
|
|
|
if (fd<0) {
|
|
|
|
log_warn(LD_FS,"Could not open \"%s\" for mmap().",filename);
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
2006-08-04 20:32:43 +02:00
|
|
|
size = lseek(fd, 0, SEEK_END);
|
2006-05-28 18:54:39 +02:00
|
|
|
lseek(fd, 0, SEEK_SET);
|
|
|
|
/* ensure page alignment */
|
|
|
|
page_size = getpagesize();
|
2006-08-04 20:32:43 +02:00
|
|
|
size += (page_size + (page_size-(size%page_size)));
|
2006-05-28 18:54:39 +02:00
|
|
|
|
2006-08-04 20:32:43 +02:00
|
|
|
string = mmap(0, size, PROT_READ, MAP_PRIVATE, fd, 0);
|
2006-06-03 23:41:14 +02:00
|
|
|
if (string == MAP_FAILED) {
|
2006-08-05 19:53:32 +02:00
|
|
|
close(fd);
|
2006-05-28 18:54:39 +02:00
|
|
|
log_warn(LD_FS,"Could not mmap file \"%s\": %s", filename,
|
|
|
|
strerror(errno));
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
close(fd);
|
|
|
|
|
2006-08-04 20:32:43 +02:00
|
|
|
res = tor_malloc_zero(sizeof(tor_mmap_t));
|
2006-08-05 19:53:21 +02:00
|
|
|
res->data = string;
|
|
|
|
res->size = size;
|
2006-08-04 20:32:43 +02:00
|
|
|
|
|
|
|
return res;
|
|
|
|
}
|
|
|
|
void
|
|
|
|
tor_munmap_file(tor_mmap_t *handle)
|
|
|
|
{
|
2006-08-05 19:53:08 +02:00
|
|
|
munmap((char*)handle->data, handle->size);
|
2006-05-28 18:54:39 +02:00
|
|
|
}
|
2006-08-04 20:32:43 +02:00
|
|
|
#elif defined(MS_WINDOWS)
|
2006-08-05 19:53:08 +02:00
|
|
|
typedef struct win_mmap_t {
|
|
|
|
tor_mmap_t base;
|
2006-08-04 20:32:43 +02:00
|
|
|
HANDLE file_handle;
|
|
|
|
HANDLE mmap_handle;
|
2006-08-05 19:53:08 +02:00
|
|
|
} tor_mmap_impl_t;
|
2006-08-04 20:32:43 +02:00
|
|
|
tor_mmap_t *
|
2006-08-05 19:53:21 +02:00
|
|
|
tor_mmap_file(const char *filename)
|
2006-08-04 20:32:43 +02:00
|
|
|
{
|
2006-08-05 19:53:08 +02:00
|
|
|
win_mmap_t *res = tor_malloc_zero(sizeof(win_mmap_t));
|
2006-08-04 20:32:43 +02:00
|
|
|
res->mmap_handle = res->file_handle = INVALID_HANDLE_VALUE;
|
|
|
|
|
2006-08-04 21:03:40 +02:00
|
|
|
res->file_handle = CreateFileForMapping(filename,
|
|
|
|
GENERIC_READ,
|
|
|
|
0, NULL,
|
|
|
|
OPEN_EXISTING,
|
|
|
|
FILE_ATTRIBUTE_NORMAL, 0);
|
2006-08-05 19:53:08 +02:00
|
|
|
res->base.size = GetFileSize(res->file_handle, NULL);
|
2006-08-04 20:32:43 +02:00
|
|
|
|
|
|
|
res->mmap_handle = CreateFileMapping(res->file_handle,
|
|
|
|
NULL,
|
|
|
|
PAGE_READONLY,
|
2006-08-04 21:03:40 +02:00
|
|
|
(size >> 32),
|
|
|
|
(size & 0xfffffffful),
|
|
|
|
NULL);
|
2006-08-04 20:32:43 +02:00
|
|
|
if (res->mmap_handle != INVALID_HANDLE_VALUE)
|
|
|
|
goto err;
|
2006-08-05 19:53:08 +02:00
|
|
|
res->base.data = (char*) MapViewOfFile(res->mmap_handle,
|
|
|
|
access,
|
|
|
|
0, 0, 0);
|
2006-08-04 20:32:43 +02:00
|
|
|
if (!res->data)
|
|
|
|
goto err;
|
|
|
|
|
2006-08-05 19:53:08 +02:00
|
|
|
return &(res->base);
|
2006-08-04 20:32:43 +02:00
|
|
|
err:
|
|
|
|
tor_munmap_file(res);
|
|
|
|
return NULL;
|
|
|
|
}
|
2006-05-28 18:54:39 +02:00
|
|
|
void
|
2006-08-04 20:32:43 +02:00
|
|
|
tor_munmap_file(tor_mmap_t *handle)
|
2006-05-28 18:54:39 +02:00
|
|
|
{
|
2006-08-05 19:53:08 +02:00
|
|
|
win_mmap_t *h = (win_mmap_t*)
|
|
|
|
(((char*)handle) - STRUCT_OFFSET(win_mmap_t, base));
|
2006-08-04 20:32:43 +02:00
|
|
|
if (handle->data)
|
|
|
|
UnmapViewOfFile(handle->data);
|
|
|
|
if (res->mmap_handle != INVALID_HANDLE_VALUE)
|
|
|
|
CloseHandle(res->mmap_handle);
|
|
|
|
if (res->file_handle != INVALID_HANDLE_VALUE)
|
|
|
|
CloseHandle(self->file_handle);
|
|
|
|
tor_free(res);
|
2006-05-28 18:54:39 +02:00
|
|
|
}
|
|
|
|
#else
|
2006-08-04 20:32:43 +02:00
|
|
|
tor_mmap_t *
|
2006-08-05 19:53:21 +02:00
|
|
|
tor_mmap_file(const char *filename)
|
2006-05-28 18:54:39 +02:00
|
|
|
{
|
|
|
|
char *res = read_file_to_str(filename, 1);
|
2006-08-04 20:32:43 +02:00
|
|
|
tor_mmap_t *handle;
|
2006-08-05 19:53:08 +02:00
|
|
|
if (! res)
|
|
|
|
return NULL;
|
2006-08-04 20:32:43 +02:00
|
|
|
handle = tor_malloc_zero(sizeof(tor_mmap_t));
|
2006-08-05 19:53:21 +02:00
|
|
|
handle->data = res;
|
|
|
|
handle->size = strlen(res) + 1;
|
2006-08-04 20:32:43 +02:00
|
|
|
return handle;
|
2006-05-28 18:54:39 +02:00
|
|
|
}
|
|
|
|
void
|
2006-08-04 20:32:43 +02:00
|
|
|
tor_munmap_file(tor_mmap_t *handle)
|
2006-05-28 18:54:39 +02:00
|
|
|
{
|
2006-08-05 19:53:08 +02:00
|
|
|
char *d = (char*)handle->data;
|
|
|
|
tor_free(d);
|
|
|
|
memset(handle, sizeof(tor_mmap_t), 0);
|
|
|
|
tor_free(handle);
|
2006-05-28 18:54:39 +02:00
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
2004-11-01 21:41:47 +01:00
|
|
|
/** Replacement for snprintf. Differs from platform snprintf in two
|
|
|
|
* ways: First, always NUL-terminates its output. Second, always
|
|
|
|
* returns -1 if the result is truncated. (Note that this return
|
|
|
|
* behavior does <i>not</i> conform to C99; it just happens to be the
|
|
|
|
* easiest to emulate "return -1" with conformant implementations than
|
|
|
|
* it is to emulate "return number that would be written" with
|
|
|
|
* non-conformant implementations.) */
|
2005-06-11 20:52:12 +02:00
|
|
|
int
|
|
|
|
tor_snprintf(char *str, size_t size, const char *format, ...)
|
2004-11-01 21:41:47 +01:00
|
|
|
{
|
|
|
|
va_list ap;
|
|
|
|
int r;
|
|
|
|
va_start(ap,format);
|
|
|
|
r = tor_vsnprintf(str,size,format,ap);
|
|
|
|
va_end(ap);
|
|
|
|
return r;
|
|
|
|
}
|
|
|
|
|
2004-11-29 07:49:04 +01:00
|
|
|
/** Replacement for vsnprintf; behavior differs as tor_snprintf differs from
|
2004-11-01 21:41:47 +01:00
|
|
|
* snprintf.
|
|
|
|
*/
|
2005-06-11 20:52:12 +02:00
|
|
|
int
|
|
|
|
tor_vsnprintf(char *str, size_t size, const char *format, va_list args)
|
2004-11-01 21:41:47 +01:00
|
|
|
{
|
|
|
|
int r;
|
2004-11-29 07:49:04 +01:00
|
|
|
if (size == 0)
|
|
|
|
return -1; /* no place for the NUL */
|
2004-12-02 05:33:01 +01:00
|
|
|
if (size > SIZE_T_CEILING)
|
|
|
|
return -1;
|
2004-11-01 21:41:47 +01:00
|
|
|
#ifdef MS_WINDOWS
|
|
|
|
r = _vsnprintf(str, size, format, args);
|
|
|
|
#else
|
|
|
|
r = vsnprintf(str, size, format, args);
|
|
|
|
#endif
|
|
|
|
str[size-1] = '\0';
|
|
|
|
if (r < 0 || ((size_t)r) >= size)
|
|
|
|
return -1;
|
|
|
|
return r;
|
|
|
|
}
|
|
|
|
|
2005-06-18 04:17:11 +02:00
|
|
|
/** Given <b>hlen</b> bytes at <b>haystack</b> and <b>nlen</b> bytes at
|
2005-06-18 04:23:11 +02:00
|
|
|
* <b>needle</b>, return a pointer to the first occurrence of the needle
|
2005-06-18 04:17:11 +02:00
|
|
|
* within the haystack, or NULL if there is no such occurrence.
|
|
|
|
*
|
|
|
|
* Requires that nlen be greater than zero.
|
|
|
|
*/
|
|
|
|
const void *
|
2005-12-14 21:40:40 +01:00
|
|
|
tor_memmem(const void *_haystack, size_t hlen,
|
|
|
|
const void *_needle, size_t nlen)
|
2005-06-18 04:17:11 +02:00
|
|
|
{
|
|
|
|
#if defined(HAVE_MEMMEM) && (!defined(__GNUC__) || __GNUC__ >= 2)
|
|
|
|
tor_assert(nlen);
|
2005-06-28 01:40:17 +02:00
|
|
|
return memmem(_haystack, hlen, _needle, nlen);
|
2005-06-18 04:17:11 +02:00
|
|
|
#else
|
2005-12-14 21:40:40 +01:00
|
|
|
/* This isn't as fast as the GLIBC implementation, but it doesn't need to
|
|
|
|
* be. */
|
2005-06-28 01:35:04 +02:00
|
|
|
const char *p, *end;
|
|
|
|
const char *haystack = (const char*)_haystack;
|
|
|
|
const char *needle = (const char*)_needle;
|
2005-06-18 04:17:11 +02:00
|
|
|
char first;
|
|
|
|
tor_assert(nlen);
|
|
|
|
|
|
|
|
p = haystack;
|
|
|
|
end = haystack + hlen;
|
|
|
|
first = *(const char*)needle;
|
|
|
|
while ((p = memchr(p, first, end-p))) {
|
2005-08-22 02:34:23 +02:00
|
|
|
if (p+nlen > end)
|
2005-06-18 04:17:11 +02:00
|
|
|
return NULL;
|
|
|
|
if (!memcmp(p, needle, nlen))
|
|
|
|
return p;
|
|
|
|
++p;
|
|
|
|
}
|
|
|
|
return NULL;
|
|
|
|
#endif
|
|
|
|
}
|
|
|
|
|
2005-10-25 09:05:03 +02:00
|
|
|
#ifdef MS_WINDOWS
|
2004-12-22 03:32:26 +01:00
|
|
|
/** Take a filename and return a pointer to its final element. This
|
|
|
|
* function is called on __FILE__ to fix a MSVC nit where __FILE__
|
|
|
|
* contains the full path to the file. This is bad, because it
|
|
|
|
* confuses users to find the home directory of the person who
|
|
|
|
* compiled the binary in their warrning messages.
|
|
|
|
*/
|
|
|
|
const char *
|
2005-10-25 09:05:03 +02:00
|
|
|
tor_fix_source_file(const char *fname)
|
2004-12-22 03:32:26 +01:00
|
|
|
{
|
2004-12-22 03:46:28 +01:00
|
|
|
const char *cp1, *cp2, *r;
|
2004-12-22 03:32:26 +01:00
|
|
|
cp1 = strrchr(fname, '/');
|
|
|
|
cp2 = strrchr(fname, '\\');
|
|
|
|
if (cp1 && cp2) {
|
2004-12-22 03:46:28 +01:00
|
|
|
r = (cp1<cp2)?(cp2+1):(cp1+1);
|
2004-12-22 03:32:26 +01:00
|
|
|
} else if (cp1) {
|
2004-12-22 03:46:28 +01:00
|
|
|
r = cp1+1;
|
2004-12-22 03:32:26 +01:00
|
|
|
} else if (cp2) {
|
2004-12-22 03:46:28 +01:00
|
|
|
r = cp2+1;
|
2004-12-22 03:32:26 +01:00
|
|
|
} else {
|
2004-12-22 03:46:28 +01:00
|
|
|
r = fname;
|
2004-12-22 03:32:26 +01:00
|
|
|
}
|
2004-12-22 03:46:28 +01:00
|
|
|
return r;
|
2004-12-22 03:32:26 +01:00
|
|
|
}
|
2005-10-25 09:05:03 +02:00
|
|
|
#endif
|
2004-12-22 03:32:26 +01:00
|
|
|
|
2004-11-01 21:41:47 +01:00
|
|
|
/**
|
2004-12-01 04:48:14 +01:00
|
|
|
* Read a 16-bit value beginning at <b>cp</b>. Equivalent to
|
2004-11-01 21:41:47 +01:00
|
|
|
* *(uint16_t*)(cp), but will not cause segfaults on platforms that forbid
|
|
|
|
* unaligned memory access.
|
|
|
|
*/
|
2005-06-11 20:52:12 +02:00
|
|
|
uint16_t
|
|
|
|
get_uint16(const char *cp)
|
2004-11-01 21:41:47 +01:00
|
|
|
{
|
|
|
|
uint16_t v;
|
|
|
|
memcpy(&v,cp,2);
|
|
|
|
return v;
|
|
|
|
}
|
|
|
|
/**
|
2004-12-01 04:48:14 +01:00
|
|
|
* Read a 32-bit value beginning at <b>cp</b>. Equivalent to
|
2004-11-01 21:41:47 +01:00
|
|
|
* *(uint32_t*)(cp), but will not cause segfaults on platforms that forbid
|
|
|
|
* unaligned memory access.
|
|
|
|
*/
|
2005-06-11 20:52:12 +02:00
|
|
|
uint32_t
|
|
|
|
get_uint32(const char *cp)
|
2004-11-01 21:41:47 +01:00
|
|
|
{
|
|
|
|
uint32_t v;
|
|
|
|
memcpy(&v,cp,4);
|
|
|
|
return v;
|
|
|
|
}
|
|
|
|
/**
|
|
|
|
* Set a 16-bit value beginning at <b>cp</b> to <b>v</b>. Equivalent to
|
|
|
|
* *(uint16_t)(cp) = v, but will not cause segfaults on platforms that forbid
|
|
|
|
* unaligned memory access. */
|
2005-06-11 20:52:12 +02:00
|
|
|
void
|
|
|
|
set_uint16(char *cp, uint16_t v)
|
2004-11-01 21:41:47 +01:00
|
|
|
{
|
|
|
|
memcpy(cp,&v,2);
|
|
|
|
}
|
|
|
|
/**
|
|
|
|
* Set a 32-bit value beginning at <b>cp</b> to <b>v</b>. Equivalent to
|
|
|
|
* *(uint32_t)(cp) = v, but will not cause segfaults on platforms that forbid
|
|
|
|
* unaligned memory access. */
|
2005-06-11 20:52:12 +02:00
|
|
|
void
|
|
|
|
set_uint32(char *cp, uint32_t v)
|
2004-11-01 21:41:47 +01:00
|
|
|
{
|
|
|
|
memcpy(cp,&v,4);
|
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
2005-06-12 06:33:26 +02:00
|
|
|
* Rename the file <b>from</b> to the file <b>to</b>. On unix, this is
|
|
|
|
* the same as rename(2). On windows, this removes <b>to</b> first if
|
|
|
|
* it already exists.
|
2004-11-01 21:41:47 +01:00
|
|
|
* Returns 0 on success. Returns -1 and sets errno on failure.
|
|
|
|
*/
|
2005-06-11 20:52:12 +02:00
|
|
|
int
|
|
|
|
replace_file(const char *from, const char *to)
|
2004-11-01 21:41:47 +01:00
|
|
|
{
|
|
|
|
#ifndef MS_WINDOWS
|
|
|
|
return rename(from,to);
|
|
|
|
#else
|
2004-11-28 10:05:49 +01:00
|
|
|
switch (file_status(to))
|
2004-11-01 21:41:47 +01:00
|
|
|
{
|
|
|
|
case FN_NOENT:
|
|
|
|
break;
|
|
|
|
case FN_FILE:
|
|
|
|
if (unlink(to)) return -1;
|
|
|
|
break;
|
|
|
|
case FN_ERROR:
|
|
|
|
return -1;
|
|
|
|
case FN_DIR:
|
|
|
|
errno = EISDIR;
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
return rename(from,to);
|
|
|
|
#endif
|
|
|
|
}
|
|
|
|
|
2005-09-15 01:27:52 +02:00
|
|
|
/** Change <b>fname</b>'s modification time to now. */
|
|
|
|
int
|
|
|
|
touch_file(const char *fname)
|
|
|
|
{
|
|
|
|
if (utime(fname, NULL)!=0)
|
|
|
|
return -1;
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2004-11-01 21:41:47 +01:00
|
|
|
/** Turn <b>socket</b> into a nonblocking socket.
|
|
|
|
*/
|
2005-06-11 20:52:12 +02:00
|
|
|
void
|
|
|
|
set_socket_nonblocking(int socket)
|
2004-11-01 21:41:47 +01:00
|
|
|
{
|
|
|
|
#ifdef MS_WINDOWS
|
2004-11-28 06:57:18 +01:00
|
|
|
int nonblocking = 1;
|
2004-11-01 21:41:47 +01:00
|
|
|
ioctlsocket(socket, FIONBIO, (unsigned long*) &nonblocking);
|
|
|
|
#else
|
|
|
|
fcntl(socket, F_SETFL, O_NONBLOCK);
|
|
|
|
#endif
|
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
|
|
* Allocate a pair of connected sockets. (Like socketpair(family,
|
|
|
|
* type,protocol,fd), but works on systems that don't have
|
|
|
|
* socketpair.)
|
|
|
|
*
|
|
|
|
* Currently, only (AF_UNIX, SOCK_STREAM, 0 ) sockets are supported.
|
|
|
|
*
|
|
|
|
* Note that on systems without socketpair, this call will fail if
|
|
|
|
* localhost is inaccessible (for example, if the networking
|
|
|
|
* stack is down). And even if it succeeds, the socket pair will not
|
|
|
|
* be able to read while localhost is down later (the socket pair may
|
|
|
|
* even close, depending on OS-specific timeouts).
|
2005-06-26 07:03:21 +02:00
|
|
|
*
|
2005-06-30 09:17:38 +02:00
|
|
|
* Returns 0 on success and -errno on failure; do not rely on the value
|
|
|
|
* of errno or WSAGetLastSocketError().
|
2004-11-01 21:41:47 +01:00
|
|
|
**/
|
2005-06-30 09:17:38 +02:00
|
|
|
/* It would be nicer just to set errno, but that won't work for windows. */
|
2004-11-01 21:41:47 +01:00
|
|
|
int
|
|
|
|
tor_socketpair(int family, int type, int protocol, int fd[2])
|
|
|
|
{
|
|
|
|
#ifdef HAVE_SOCKETPAIR
|
2005-06-30 09:17:38 +02:00
|
|
|
int r;
|
|
|
|
r = socketpair(family, type, protocol, fd);
|
|
|
|
return r < 0 ? -errno : r;
|
2004-11-01 21:41:47 +01:00
|
|
|
#else
|
|
|
|
/* This socketpair does not work when localhost is down. So
|
|
|
|
* it's really not the same thing at all. But it's close enough
|
|
|
|
* for now, and really, when localhost is down sometimes, we
|
|
|
|
* have other problems too.
|
|
|
|
*/
|
|
|
|
int listener = -1;
|
|
|
|
int connector = -1;
|
|
|
|
int acceptor = -1;
|
|
|
|
struct sockaddr_in listen_addr;
|
|
|
|
struct sockaddr_in connect_addr;
|
|
|
|
int size;
|
2005-06-30 09:17:38 +02:00
|
|
|
int saved_errno = -1;
|
2004-11-01 21:41:47 +01:00
|
|
|
|
|
|
|
if (protocol
|
|
|
|
#ifdef AF_UNIX
|
|
|
|
|| family != AF_UNIX
|
|
|
|
#endif
|
|
|
|
) {
|
|
|
|
#ifdef MS_WINDOWS
|
2005-06-30 09:17:38 +02:00
|
|
|
return -WSAEAFNOSUPPORT;
|
2004-11-01 21:41:47 +01:00
|
|
|
#else
|
2005-06-30 09:17:38 +02:00
|
|
|
return -EAFNOSUPPORT;
|
2004-11-01 21:41:47 +01:00
|
|
|
#endif
|
|
|
|
}
|
|
|
|
if (!fd) {
|
2005-06-30 09:17:38 +02:00
|
|
|
return -EINVAL;
|
2004-11-01 21:41:47 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
listener = socket(AF_INET, type, 0);
|
2006-06-05 11:08:10 +02:00
|
|
|
if (listener < 0)
|
2005-06-30 09:17:38 +02:00
|
|
|
return -tor_socket_errno(-1);
|
2004-11-23 00:28:26 +01:00
|
|
|
memset(&listen_addr, 0, sizeof(listen_addr));
|
2004-11-01 21:41:47 +01:00
|
|
|
listen_addr.sin_family = AF_INET;
|
2004-11-23 00:28:26 +01:00
|
|
|
listen_addr.sin_addr.s_addr = htonl(INADDR_LOOPBACK);
|
2004-12-01 04:48:14 +01:00
|
|
|
listen_addr.sin_port = 0; /* kernel chooses port. */
|
2004-11-01 21:41:47 +01:00
|
|
|
if (bind(listener, (struct sockaddr *) &listen_addr, sizeof (listen_addr))
|
|
|
|
== -1)
|
2005-06-30 09:17:38 +02:00
|
|
|
goto tidy_up_and_fail;
|
2004-11-01 21:41:47 +01:00
|
|
|
if (listen(listener, 1) == -1)
|
2005-06-30 09:17:38 +02:00
|
|
|
goto tidy_up_and_fail;
|
2004-11-01 21:41:47 +01:00
|
|
|
|
|
|
|
connector = socket(AF_INET, type, 0);
|
2006-06-05 11:08:10 +02:00
|
|
|
if (connector < 0)
|
2005-06-30 09:17:38 +02:00
|
|
|
goto tidy_up_and_fail;
|
2004-11-01 21:41:47 +01:00
|
|
|
/* We want to find out the port number to connect to. */
|
2004-11-23 00:28:26 +01:00
|
|
|
size = sizeof(connect_addr);
|
2004-11-01 21:41:47 +01:00
|
|
|
if (getsockname(listener, (struct sockaddr *) &connect_addr, &size) == -1)
|
2005-06-30 09:17:38 +02:00
|
|
|
goto tidy_up_and_fail;
|
2004-11-01 21:41:47 +01:00
|
|
|
if (size != sizeof (connect_addr))
|
2005-06-30 09:17:38 +02:00
|
|
|
goto abort_tidy_up_and_fail;
|
2004-11-01 21:41:47 +01:00
|
|
|
if (connect(connector, (struct sockaddr *) &connect_addr,
|
2004-11-23 00:28:26 +01:00
|
|
|
sizeof(connect_addr)) == -1)
|
2005-06-30 09:17:38 +02:00
|
|
|
goto tidy_up_and_fail;
|
2004-11-01 21:41:47 +01:00
|
|
|
|
2004-11-23 00:28:26 +01:00
|
|
|
size = sizeof(listen_addr);
|
2004-11-01 21:41:47 +01:00
|
|
|
acceptor = accept(listener, (struct sockaddr *) &listen_addr, &size);
|
2006-06-05 11:08:10 +02:00
|
|
|
if (acceptor < 0)
|
2005-06-30 09:17:38 +02:00
|
|
|
goto tidy_up_and_fail;
|
2004-11-01 21:41:47 +01:00
|
|
|
if (size != sizeof(listen_addr))
|
2005-06-30 09:17:38 +02:00
|
|
|
goto abort_tidy_up_and_fail;
|
2004-11-01 21:41:47 +01:00
|
|
|
tor_close_socket(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)
|
2005-06-30 09:17:38 +02:00
|
|
|
goto tidy_up_and_fail;
|
2004-11-01 21:41:47 +01:00
|
|
|
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) {
|
2005-06-30 09:17:38 +02:00
|
|
|
goto abort_tidy_up_and_fail;
|
2004-11-01 21:41:47 +01:00
|
|
|
}
|
|
|
|
fd[0] = connector;
|
|
|
|
fd[1] = acceptor;
|
2005-01-03 20:07:25 +01:00
|
|
|
|
2004-11-01 21:41:47 +01:00
|
|
|
return 0;
|
|
|
|
|
|
|
|
abort_tidy_up_and_fail:
|
|
|
|
#ifdef MS_WINDOWS
|
2005-06-30 09:17:38 +02:00
|
|
|
saved_errno = WSAECONNABORTED;
|
2004-11-01 21:41:47 +01:00
|
|
|
#else
|
2005-06-30 09:17:38 +02:00
|
|
|
saved_errno = ECONNABORTED; /* I hope this is portable and appropriate. */
|
2004-11-01 21:41:47 +01:00
|
|
|
#endif
|
|
|
|
tidy_up_and_fail:
|
2005-06-30 09:17:38 +02:00
|
|
|
if (saved_errno < 0)
|
|
|
|
saved_errno = errno;
|
|
|
|
if (listener != -1)
|
|
|
|
tor_close_socket(listener);
|
|
|
|
if (connector != -1)
|
|
|
|
tor_close_socket(connector);
|
|
|
|
if (acceptor != -1)
|
|
|
|
tor_close_socket(acceptor);
|
|
|
|
return -saved_errno;
|
2004-11-01 21:41:47 +01:00
|
|
|
#endif
|
|
|
|
}
|
|
|
|
|
2005-02-23 00:52:44 +01:00
|
|
|
#define ULIMIT_BUFFER 32 /* keep 32 extra fd's beyond _ConnLimit */
|
|
|
|
|
2006-02-09 03:59:37 +01:00
|
|
|
/** Learn the maximum allowed number of file descriptors. (Some systems
|
|
|
|
* have a low soft limit.
|
|
|
|
*
|
|
|
|
* We compute this by finding the largest number between <b>limit</b>
|
|
|
|
* and <b>cap</b> that we can use. If we can't find a number greater
|
|
|
|
* than or equal to <b>limit</b>, then we fail: return -1.
|
|
|
|
*
|
|
|
|
* Otherwise, return the number minus some buffer to allow for other
|
|
|
|
* file descriptors we'll want available for ordinary use. */
|
2005-03-14 04:28:46 +01:00
|
|
|
int
|
2005-09-30 03:09:52 +02:00
|
|
|
set_max_file_descriptors(unsigned long limit, unsigned long cap)
|
|
|
|
{
|
2004-11-01 21:41:47 +01:00
|
|
|
#ifndef HAVE_GETRLIMIT
|
2005-10-18 23:58:19 +02:00
|
|
|
log_fn(LOG_INFO, LD_NET,
|
|
|
|
"This platform is missing getrlimit(). Proceeding.");
|
2006-01-24 00:39:13 +01:00
|
|
|
if (limit < cap) {
|
2006-02-13 09:01:59 +01:00
|
|
|
log_info(LD_CONFIG, "ConnLimit must be at most %d. Using that.", cap);
|
2005-02-23 00:52:44 +01:00
|
|
|
limit = cap;
|
|
|
|
}
|
2004-11-01 21:41:47 +01:00
|
|
|
#else
|
|
|
|
struct rlimit rlim;
|
2005-03-14 04:28:46 +01:00
|
|
|
unsigned long most;
|
|
|
|
tor_assert(limit > 0);
|
|
|
|
tor_assert(cap > 0);
|
2004-11-01 21:41:47 +01:00
|
|
|
|
|
|
|
if (getrlimit(RLIMIT_NOFILE, &rlim) != 0) {
|
2006-02-13 09:01:59 +01:00
|
|
|
log_warn(LD_NET, "Could not get maximum number of file descriptors: %s",
|
|
|
|
strerror(errno));
|
2004-11-01 21:41:47 +01:00
|
|
|
return -1;
|
|
|
|
}
|
2005-02-23 00:52:44 +01:00
|
|
|
if (rlim.rlim_max < limit) {
|
2006-02-13 09:01:59 +01:00
|
|
|
log_warn(LD_CONFIG,"We need %lu file descriptors available, and we're "
|
|
|
|
"limited to %lu. Please change your ulimit -n.",
|
|
|
|
limit, (unsigned long)rlim.rlim_max);
|
2004-11-01 21:41:47 +01:00
|
|
|
return -1;
|
|
|
|
}
|
2005-03-14 04:28:46 +01:00
|
|
|
most = (rlim.rlim_max > cap) ? cap : (unsigned) rlim.rlim_max;
|
2005-02-23 00:52:44 +01:00
|
|
|
if (most > rlim.rlim_cur) {
|
2006-02-13 09:01:59 +01:00
|
|
|
log_info(LD_NET,"Raising max file descriptors from %lu to %lu.",
|
|
|
|
(unsigned long)rlim.rlim_cur, most);
|
2004-11-01 21:41:47 +01:00
|
|
|
}
|
2005-02-23 00:52:44 +01:00
|
|
|
rlim.rlim_cur = most;
|
2004-11-01 21:41:47 +01:00
|
|
|
if (setrlimit(RLIMIT_NOFILE, &rlim) != 0) {
|
2006-02-13 09:01:59 +01:00
|
|
|
log_warn(LD_CONFIG, "Could not set maximum number of file descriptors: %s",
|
|
|
|
strerror(errno));
|
2004-11-01 21:41:47 +01:00
|
|
|
return -1;
|
|
|
|
}
|
2005-02-23 00:52:44 +01:00
|
|
|
/* leave some overhead for logs, etc, */
|
|
|
|
limit = most;
|
2004-11-01 21:41:47 +01:00
|
|
|
#endif
|
2005-02-23 00:52:44 +01:00
|
|
|
|
|
|
|
if (limit < ULIMIT_BUFFER) {
|
2006-02-13 09:01:59 +01:00
|
|
|
log_warn(LD_CONFIG,
|
|
|
|
"ConnLimit must be at least %d. Failing.", ULIMIT_BUFFER);
|
2005-02-23 00:52:44 +01:00
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
return limit - ULIMIT_BUFFER;
|
2004-11-01 21:41:47 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
/** Call setuid and setgid to run as <b>user</b>:<b>group</b>. Return 0 on
|
|
|
|
* success. On failure, log and return -1.
|
|
|
|
*/
|
2005-06-11 20:52:12 +02:00
|
|
|
int
|
2005-09-30 03:09:52 +02:00
|
|
|
switch_id(char *user, char *group)
|
|
|
|
{
|
2004-11-01 21:41:47 +01:00
|
|
|
#ifndef MS_WINDOWS
|
|
|
|
struct passwd *pw = NULL;
|
|
|
|
struct group *gr = NULL;
|
|
|
|
|
|
|
|
if (user) {
|
|
|
|
pw = getpwnam(user);
|
|
|
|
if (pw == NULL) {
|
2006-03-26 08:47:08 +02:00
|
|
|
log_warn(LD_CONFIG,"User '%s' not found.", user);
|
2004-11-01 21:41:47 +01:00
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/* switch the group first, while we still have the privileges to do so */
|
|
|
|
if (group) {
|
|
|
|
gr = getgrnam(group);
|
|
|
|
if (gr == NULL) {
|
2006-03-26 08:47:08 +02:00
|
|
|
log_warn(LD_CONFIG,"Group '%s' not found.", group);
|
2004-11-01 21:41:47 +01:00
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (setgid(gr->gr_gid) != 0) {
|
2006-03-26 08:47:08 +02:00
|
|
|
log_warn(LD_GENERAL,"Error setting GID: %s", strerror(errno));
|
2004-11-01 21:41:47 +01:00
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
} else if (user) {
|
|
|
|
if (setgid(pw->pw_gid) != 0) {
|
2006-03-26 08:47:08 +02:00
|
|
|
log_warn(LD_GENERAL,"Error setting GID: %s", strerror(errno));
|
2004-11-01 21:41:47 +01:00
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/* now that the group is switched, we can switch users and lose
|
|
|
|
privileges */
|
|
|
|
if (user) {
|
|
|
|
if (setuid(pw->pw_uid) != 0) {
|
2006-03-26 08:47:08 +02:00
|
|
|
log_warn(LD_GENERAL,"Error setting UID: %s", strerror(errno));
|
2004-11-01 21:41:47 +01:00
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
#endif
|
|
|
|
|
2006-03-26 08:47:08 +02:00
|
|
|
log_warn(LD_CONFIG,
|
|
|
|
"User or group specified, but switching users is not supported.");
|
2004-11-01 21:41:47 +01:00
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
2004-11-10 15:23:31 +01:00
|
|
|
#ifdef HAVE_PWD_H
|
|
|
|
/** Allocate and return a string containing the home directory for the
|
|
|
|
* user <b>username</b>. Only works on posix-like systems */
|
|
|
|
char *
|
|
|
|
get_user_homedir(const char *username)
|
|
|
|
{
|
|
|
|
struct passwd *pw;
|
|
|
|
tor_assert(username);
|
|
|
|
|
|
|
|
if (!(pw = getpwnam(username))) {
|
2006-02-13 09:01:59 +01:00
|
|
|
log_err(LD_CONFIG,"User \"%s\" not found.", username);
|
2004-11-10 15:23:31 +01:00
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
return tor_strdup(pw->pw_dir);
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
2004-11-01 21:41:47 +01:00
|
|
|
/** Set *addr to the IP address (in dotted-quad notation) stored in c.
|
|
|
|
* Return 1 on success, 0 if c is badly formatted. (Like inet_aton(c,addr),
|
|
|
|
* but works on Windows and Solaris.)
|
|
|
|
*/
|
2005-06-11 20:52:12 +02:00
|
|
|
int
|
|
|
|
tor_inet_aton(const char *c, struct in_addr* addr)
|
2004-11-01 21:41:47 +01:00
|
|
|
{
|
|
|
|
#ifdef HAVE_INET_ATON
|
|
|
|
return inet_aton(c, addr);
|
|
|
|
#else
|
|
|
|
uint32_t r;
|
|
|
|
tor_assert(c);
|
|
|
|
tor_assert(addr);
|
|
|
|
if (strcmp(c, "255.255.255.255") == 0) {
|
|
|
|
addr->s_addr = 0xFFFFFFFFu;
|
|
|
|
return 1;
|
|
|
|
}
|
|
|
|
r = inet_addr(c);
|
|
|
|
if (r == INADDR_NONE)
|
|
|
|
return 0;
|
|
|
|
addr->s_addr = r;
|
|
|
|
return 1;
|
|
|
|
#endif
|
|
|
|
}
|
|
|
|
|
2004-12-22 06:29:06 +01:00
|
|
|
/** Similar behavior to Unix gethostbyname: resolve <b>name</b>, and set
|
|
|
|
* *addr to the proper IP address, in network byte order. Returns 0
|
|
|
|
* on success, -1 on failure; 1 on transient failure.
|
|
|
|
*
|
|
|
|
* (This function exists because standard windows gethostbyname
|
|
|
|
* doesn't treat raw IP addresses properly.)
|
|
|
|
*/
|
2005-06-11 20:52:12 +02:00
|
|
|
int
|
|
|
|
tor_lookup_hostname(const char *name, uint32_t *addr)
|
2004-12-22 06:29:06 +01:00
|
|
|
{
|
|
|
|
/* Perhaps eventually this should be replaced by a tor_getaddrinfo or
|
|
|
|
* something.
|
|
|
|
*/
|
|
|
|
struct in_addr iaddr;
|
2005-08-26 09:37:07 +02:00
|
|
|
tor_assert(name);
|
2004-12-22 06:29:06 +01:00
|
|
|
tor_assert(addr);
|
|
|
|
if (!*name) {
|
|
|
|
/* Empty address is an error. */
|
|
|
|
return -1;
|
|
|
|
} else if (tor_inet_aton(name, &iaddr)) {
|
|
|
|
/* It's an IP. */
|
|
|
|
memcpy(addr, &iaddr.s_addr, 4);
|
|
|
|
return 0;
|
|
|
|
} else {
|
2005-02-03 22:31:04 +01:00
|
|
|
#ifdef HAVE_GETADDRINFO
|
|
|
|
int err;
|
2005-06-07 15:45:58 +02:00
|
|
|
struct addrinfo *res=NULL, *res_p;
|
2005-02-03 22:31:04 +01:00
|
|
|
struct addrinfo hints;
|
2005-06-07 15:45:58 +02:00
|
|
|
int result = -1;
|
2005-02-03 22:31:04 +01:00
|
|
|
memset(&hints, 0, sizeof(hints));
|
|
|
|
hints.ai_family = PF_INET;
|
|
|
|
hints.ai_socktype = SOCK_STREAM;
|
2006-08-16 20:47:19 +02:00
|
|
|
err = getaddrinfo(name, NULL, &hints, &res);
|
2005-02-03 22:31:04 +01:00
|
|
|
if (!err) {
|
|
|
|
for (res_p = res; res_p; res_p = res_p->ai_next) {
|
2005-02-04 00:13:36 +01:00
|
|
|
if (res_p->ai_family == AF_INET) {
|
|
|
|
struct sockaddr_in *sin = (struct sockaddr_in *)res_p->ai_addr;
|
|
|
|
memcpy(addr, &sin->sin_addr, 4);
|
2005-06-07 15:45:58 +02:00
|
|
|
result = 0;
|
|
|
|
break;
|
2005-02-03 22:31:04 +01:00
|
|
|
}
|
|
|
|
}
|
2005-06-07 15:45:58 +02:00
|
|
|
freeaddrinfo(res);
|
|
|
|
return result;
|
2005-02-03 22:31:04 +01:00
|
|
|
}
|
|
|
|
return (err == EAI_AGAIN) ? 1 : -1;
|
|
|
|
#else
|
|
|
|
struct hostent *ent;
|
2005-02-22 05:50:31 +01:00
|
|
|
int err;
|
|
|
|
#ifdef HAVE_GETHOSTBYNAME_R_6_ARG
|
|
|
|
char buf[2048];
|
|
|
|
struct hostent hostent;
|
|
|
|
int r;
|
|
|
|
r = gethostbyname_r(name, &hostent, buf, sizeof(buf), &ent, &err);
|
2005-02-22 05:55:19 +01:00
|
|
|
#elif defined(HAVE_GETHOSTBYNAME_R_5_ARG)
|
2005-02-22 05:50:31 +01:00
|
|
|
char buf[2048];
|
|
|
|
struct hostent hostent;
|
|
|
|
ent = gethostbyname_r(name, &hostent, buf, sizeof(buf), &err);
|
2005-02-22 05:55:19 +01:00
|
|
|
#elif defined(HAVE_GETHOSTBYNAME_R_3_ARG)
|
2005-02-22 05:50:31 +01:00
|
|
|
struct hostent_data data;
|
|
|
|
struct hostent hent;
|
|
|
|
memset(&data, 0, sizeof(data));
|
|
|
|
err = gethostbyname_r(name, &hent, &data);
|
|
|
|
ent = err ? NULL : &hent;
|
2005-02-03 22:31:04 +01:00
|
|
|
#else
|
2005-02-22 03:51:55 +01:00
|
|
|
ent = gethostbyname(name);
|
2005-02-22 05:50:31 +01:00
|
|
|
#ifdef MS_WINDOWS
|
|
|
|
err = WSAGetLastError();
|
|
|
|
#else
|
|
|
|
err = h_errno;
|
|
|
|
#endif
|
2005-02-03 22:31:04 +01:00
|
|
|
#endif
|
2004-12-22 06:29:06 +01:00
|
|
|
if (ent) {
|
|
|
|
/* break to remind us if we move away from IPv4 */
|
|
|
|
tor_assert(ent->h_length == 4);
|
|
|
|
memcpy(addr, ent->h_addr, 4);
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
memset(addr, 0, 4);
|
|
|
|
#ifdef MS_WINDOWS
|
2005-02-22 05:50:31 +01:00
|
|
|
return (err == WSATRY_AGAIN) ? 1 : -1;
|
2004-12-22 06:29:06 +01:00
|
|
|
#else
|
2005-02-22 05:50:31 +01:00
|
|
|
return (err == TRY_AGAIN) ? 1 : -1;
|
2005-02-03 22:31:04 +01:00
|
|
|
#endif
|
2004-12-22 06:29:06 +01:00
|
|
|
#endif
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2005-06-12 06:33:26 +02:00
|
|
|
/** Hold the result of our call to <b>uname</b>. */
|
2004-11-01 21:41:47 +01:00
|
|
|
static char uname_result[256];
|
2005-06-12 06:33:26 +02:00
|
|
|
/** True iff uname_result is set. */
|
2004-11-01 21:41:47 +01:00
|
|
|
static int uname_result_is_set = 0;
|
|
|
|
|
2005-06-12 06:33:26 +02:00
|
|
|
/** Return a pointer to a description of our platform.
|
2004-11-01 21:41:47 +01:00
|
|
|
*/
|
|
|
|
const char *
|
|
|
|
get_uname(void)
|
|
|
|
{
|
|
|
|
#ifdef HAVE_UNAME
|
|
|
|
struct utsname u;
|
|
|
|
#endif
|
|
|
|
if (!uname_result_is_set) {
|
|
|
|
#ifdef HAVE_UNAME
|
|
|
|
if (uname(&u) != -1) {
|
|
|
|
/* (linux says 0 is success, solaris says 1 is success) */
|
2005-04-06 23:11:05 +02:00
|
|
|
tor_snprintf(uname_result, sizeof(uname_result), "%s %s",
|
|
|
|
u.sysname, u.machine);
|
2004-11-01 21:41:47 +01:00
|
|
|
} else
|
|
|
|
#endif
|
|
|
|
{
|
2005-01-20 00:10:16 +01:00
|
|
|
#ifdef MS_WINDOWS
|
2005-10-06 00:04:45 +02:00
|
|
|
OSVERSIONINFOEX info;
|
2005-01-20 00:10:16 +01:00
|
|
|
int i;
|
2005-10-24 20:37:09 +02:00
|
|
|
unsigned int leftover_mask;
|
2005-01-20 00:10:16 +01:00
|
|
|
const char *plat = NULL;
|
|
|
|
static struct {
|
|
|
|
int major; int minor; const char *version;
|
|
|
|
} win_version_table[] = {
|
2005-10-06 00:04:45 +02:00
|
|
|
{ 6, 0, "Windows \"Longhorn\"" },
|
2005-01-20 00:10:16 +01:00
|
|
|
{ 5, 2, "Windows Server 2003" },
|
|
|
|
{ 5, 1, "Windows XP" },
|
|
|
|
{ 5, 0, "Windows 2000" },
|
|
|
|
/* { 4, 0, "Windows NT 4.0" }, */
|
|
|
|
{ 4, 90, "Windows Me" },
|
|
|
|
{ 4, 10, "Windows 98" },
|
|
|
|
/* { 4, 0, "Windows 95" } */
|
|
|
|
{ 3, 51, "Windows NT 3.51" },
|
|
|
|
{ -1, -1, NULL }
|
|
|
|
};
|
2006-07-10 00:29:12 +02:00
|
|
|
#ifdef VER_SUITE_BACKOFFICE
|
2005-10-24 20:37:09 +02:00
|
|
|
static struct {
|
|
|
|
unsigned int mask; const char *str;
|
|
|
|
} win_mask_table[] = {
|
|
|
|
{ VER_SUITE_BACKOFFICE, " {backoffice}" },
|
|
|
|
{ VER_SUITE_BLADE, " {\"blade\" (2003, web edition)}" },
|
|
|
|
{ VER_SUITE_DATACENTER, " {datacenter}" },
|
|
|
|
{ VER_SUITE_ENTERPRISE, " {enterprise}" },
|
|
|
|
{ VER_SUITE_EMBEDDEDNT, " {embedded}" },
|
|
|
|
{ VER_SUITE_PERSONAL, " {personal}" },
|
2005-12-14 21:40:40 +01:00
|
|
|
{ VER_SUITE_SINGLEUSERTS,
|
|
|
|
" {terminal services, single user}" },
|
2005-10-24 20:37:09 +02:00
|
|
|
{ VER_SUITE_SMALLBUSINESS, " {small business}" },
|
2005-12-14 21:40:40 +01:00
|
|
|
{ VER_SUITE_SMALLBUSINESS_RESTRICTED,
|
|
|
|
" {small business, restricted}" },
|
2005-10-24 20:37:09 +02:00
|
|
|
{ VER_SUITE_TERMINAL, " {terminal services}" },
|
|
|
|
{ 0, NULL },
|
|
|
|
};
|
2006-07-10 00:29:12 +02:00
|
|
|
#endif
|
2006-03-24 06:26:33 +01:00
|
|
|
memset(&info, 0, sizeof(info));
|
2005-01-20 00:10:16 +01:00
|
|
|
info.dwOSVersionInfoSize = sizeof(info);
|
2006-03-24 06:26:33 +01:00
|
|
|
if (! GetVersionEx((LPOSVERSIONINFO)&info)) {
|
|
|
|
int err = GetLastError();
|
|
|
|
strlcpy(uname_result, "Bizarre version of Windows where GetVersionEx"
|
|
|
|
" doesn't work.", sizeof(uname_result));
|
|
|
|
uname_result_is_set = 1;
|
|
|
|
return uname_result;
|
|
|
|
}
|
2005-01-20 00:10:16 +01:00
|
|
|
if (info.dwMajorVersion == 4 && info.dwMinorVersion == 0) {
|
|
|
|
if (info.dwPlatformId == VER_PLATFORM_WIN32_NT)
|
|
|
|
plat = "Windows NT 4.0";
|
|
|
|
else
|
|
|
|
plat = "Windows 95";
|
|
|
|
} else {
|
|
|
|
for (i=0; win_version_table[i].major>=0; ++i) {
|
|
|
|
if (win_version_table[i].major == info.dwMajorVersion &&
|
2005-01-20 20:03:09 +01:00
|
|
|
win_version_table[i].minor == info.dwMinorVersion) {
|
2005-01-20 00:10:16 +01:00
|
|
|
plat = win_version_table[i].version;
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
if (plat) {
|
2005-10-07 21:33:47 +02:00
|
|
|
tor_snprintf(uname_result, sizeof(uname_result), "%s %s",
|
2005-10-06 00:04:45 +02:00
|
|
|
plat, info.szCSDVersion);
|
2005-01-20 00:10:16 +01:00
|
|
|
} else {
|
2005-10-06 00:04:45 +02:00
|
|
|
if (info.dwMajorVersion > 6 ||
|
|
|
|
(info.dwMajorVersion==6 && info.dwMinorVersion>0))
|
2005-01-20 20:03:09 +01:00
|
|
|
tor_snprintf(uname_result, sizeof(uname_result),
|
2005-12-14 21:40:40 +01:00
|
|
|
"Very recent version of Windows [major=%d,minor=%d] %s",
|
|
|
|
(int)info.dwMajorVersion,(int)info.dwMinorVersion,
|
|
|
|
info.szCSDVersion);
|
2005-01-20 00:10:16 +01:00
|
|
|
else
|
2005-01-20 20:03:09 +01:00
|
|
|
tor_snprintf(uname_result, sizeof(uname_result),
|
2005-12-14 21:40:40 +01:00
|
|
|
"Unrecognized version of Windows [major=%d,minor=%d] %s",
|
|
|
|
(int)info.dwMajorVersion,(int)info.dwMinorVersion,
|
|
|
|
info.szCSDVersion);
|
2005-01-20 00:10:16 +01:00
|
|
|
}
|
2006-07-10 00:29:12 +02:00
|
|
|
#ifdef VER_SUITE_BACKOFFICE
|
2005-10-07 21:33:47 +02:00
|
|
|
if (info.wProductType == VER_NT_DOMAIN_CONTROLLER) {
|
2005-10-06 00:04:45 +02:00
|
|
|
strlcat(uname_result, " [domain controller]", sizeof(uname_result));
|
2005-10-07 21:33:47 +02:00
|
|
|
} else if (info.wProductType == VER_NT_SERVER) {
|
2005-10-06 00:04:45 +02:00
|
|
|
strlcat(uname_result, " [server]", sizeof(uname_result));
|
2005-10-07 21:33:47 +02:00
|
|
|
} else if (info.wProductType == VER_NT_WORKSTATION) {
|
2005-10-06 00:04:45 +02:00
|
|
|
strlcat(uname_result, " [workstation]", sizeof(uname_result));
|
|
|
|
}
|
2005-10-24 20:37:09 +02:00
|
|
|
leftover_mask = info.wSuiteMask;
|
|
|
|
for (i = 0; win_mask_table[i].mask; ++i) {
|
2005-11-15 04:05:23 +01:00
|
|
|
if (info.wSuiteMask & win_mask_table[i].mask) {
|
2005-10-24 20:37:09 +02:00
|
|
|
strlcat(uname_result, win_mask_table[i].str, sizeof(uname_result));
|
|
|
|
leftover_mask &= ~win_mask_table[i].mask;
|
|
|
|
}
|
2005-10-06 00:04:45 +02:00
|
|
|
}
|
2005-10-24 20:37:09 +02:00
|
|
|
if (leftover_mask) {
|
|
|
|
size_t len = strlen(uname_result);
|
|
|
|
tor_snprintf(uname_result+len, sizeof(uname_result)-len,
|
|
|
|
" {0x%x}", info.wSuiteMask);
|
2005-10-06 00:04:45 +02:00
|
|
|
}
|
2006-07-10 00:29:12 +02:00
|
|
|
#endif
|
2005-01-20 00:10:16 +01:00
|
|
|
#else
|
2004-11-01 21:41:47 +01:00
|
|
|
strlcpy(uname_result, "Unknown platform", sizeof(uname_result));
|
2005-01-20 00:10:16 +01:00
|
|
|
#endif
|
2004-11-01 21:41:47 +01:00
|
|
|
}
|
|
|
|
uname_result_is_set = 1;
|
|
|
|
}
|
|
|
|
return uname_result;
|
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Process control
|
|
|
|
*/
|
|
|
|
|
2005-01-27 23:34:48 +01:00
|
|
|
#if defined(USE_PTHREADS)
|
2005-10-06 06:33:40 +02:00
|
|
|
/** Wraps a an int (*)(void*) function and its argument so we can
|
|
|
|
* invoke them in a way pthreads would expect.
|
|
|
|
*/
|
2005-07-22 23:12:10 +02:00
|
|
|
typedef struct tor_pthread_data_t {
|
2005-01-27 23:34:48 +01:00
|
|
|
int (*func)(void *);
|
|
|
|
void *data;
|
2005-07-22 23:12:10 +02:00
|
|
|
} tor_pthread_data_t;
|
2005-01-27 23:34:48 +01:00
|
|
|
static void *
|
|
|
|
tor_pthread_helper_fn(void *_data)
|
|
|
|
{
|
2005-07-22 23:12:10 +02:00
|
|
|
tor_pthread_data_t *data = _data;
|
2005-01-27 23:34:48 +01:00
|
|
|
int (*func)(void*);
|
|
|
|
void *arg;
|
|
|
|
func = data->func;
|
|
|
|
arg = data->data;
|
|
|
|
tor_free(_data);
|
|
|
|
func(arg);
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
2004-11-01 21:41:47 +01:00
|
|
|
/** Minimalist interface to run a void function in the background. On
|
|
|
|
* unix calls fork, on win32 calls beginthread. Returns -1 on failure.
|
|
|
|
* func should not return, but rather should call spawn_exit.
|
2005-01-03 20:07:25 +01:00
|
|
|
*
|
|
|
|
* NOTE: if <b>data</b> is used, it should not be allocated on the stack,
|
|
|
|
* since in a multithreaded environment, there is no way to be sure that
|
|
|
|
* the caller's stack will still be around when the called function is
|
|
|
|
* running.
|
2004-11-01 21:41:47 +01:00
|
|
|
*/
|
2004-11-23 00:28:26 +01:00
|
|
|
int
|
|
|
|
spawn_func(int (*func)(void *), void *data)
|
2004-11-01 21:41:47 +01:00
|
|
|
{
|
2005-01-27 23:34:48 +01:00
|
|
|
#if defined(USE_WIN32_THREADS)
|
2004-11-01 21:41:47 +01:00
|
|
|
int rv;
|
|
|
|
rv = _beginthread(func, 0, data);
|
|
|
|
if (rv == (unsigned long) -1)
|
|
|
|
return -1;
|
|
|
|
return 0;
|
2005-01-27 23:34:48 +01:00
|
|
|
#elif defined(USE_PTHREADS)
|
|
|
|
pthread_t thread;
|
2005-07-22 23:12:10 +02:00
|
|
|
tor_pthread_data_t *d;
|
|
|
|
d = tor_malloc(sizeof(tor_pthread_data_t));
|
2005-01-27 23:34:48 +01:00
|
|
|
d->data = data;
|
|
|
|
d->func = func;
|
|
|
|
if (pthread_create(&thread,NULL,tor_pthread_helper_fn,d))
|
|
|
|
return -1;
|
|
|
|
if (pthread_detach(thread))
|
|
|
|
return -1;
|
|
|
|
return 0;
|
2004-11-01 21:41:47 +01:00
|
|
|
#else
|
|
|
|
pid_t pid;
|
|
|
|
pid = fork();
|
|
|
|
if (pid<0)
|
|
|
|
return -1;
|
|
|
|
if (pid==0) {
|
|
|
|
/* Child */
|
|
|
|
func(data);
|
|
|
|
tor_assert(0); /* Should never reach here. */
|
|
|
|
return 0; /* suppress "control-reaches-end-of-non-void" warning. */
|
|
|
|
} else {
|
|
|
|
/* Parent */
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
}
|
|
|
|
|
|
|
|
/** End the current thread/process.
|
|
|
|
*/
|
2005-06-11 20:52:12 +02:00
|
|
|
void
|
2005-09-30 00:59:17 +02:00
|
|
|
spawn_exit(void)
|
2004-11-01 21:41:47 +01:00
|
|
|
{
|
2005-01-27 23:34:48 +01:00
|
|
|
#if defined(USE_WIN32_THREADS)
|
2004-11-01 21:41:47 +01:00
|
|
|
_endthread();
|
2005-01-27 23:34:48 +01:00
|
|
|
#elif defined(USE_PTHREADS)
|
|
|
|
pthread_exit(NULL);
|
2004-11-01 21:41:47 +01:00
|
|
|
#else
|
2005-02-03 07:40:06 +01:00
|
|
|
/* http://www.erlenstar.demon.co.uk/unix/faq_2.html says we should
|
|
|
|
* call _exit, not exit, from child processes. */
|
|
|
|
_exit(0);
|
2004-11-01 21:41:47 +01:00
|
|
|
#endif
|
|
|
|
}
|
|
|
|
|
|
|
|
/** Set *timeval to the current time of day. On error, log and terminate.
|
|
|
|
* (Same as gettimeofday(timeval,NULL), but never returns -1.)
|
|
|
|
*/
|
2005-06-11 20:52:12 +02:00
|
|
|
void
|
|
|
|
tor_gettimeofday(struct timeval *timeval)
|
|
|
|
{
|
2005-08-03 18:28:39 +02:00
|
|
|
#ifdef MS_WINDOWS
|
|
|
|
/* Epoch bias copied from perl: number of units between windows epoch and
|
|
|
|
* unix epoch. */
|
|
|
|
#define EPOCH_BIAS U64_LITERAL(116444736000000000)
|
|
|
|
#define UNITS_PER_SEC U64_LITERAL(10000000)
|
|
|
|
#define USEC_PER_SEC U64_LITERAL(1000000)
|
|
|
|
#define UNITS_PER_USEC U64_LITERAL(10)
|
|
|
|
union {
|
|
|
|
uint64_t ft_64;
|
|
|
|
FILETIME ft_ft;
|
|
|
|
} ft;
|
|
|
|
/* number of 100-nsec units since Jan 1, 1601 */
|
|
|
|
GetSystemTimeAsFileTime(&ft.ft_ft);
|
|
|
|
if (ft.ft_64 < EPOCH_BIAS) {
|
2006-02-13 09:01:59 +01:00
|
|
|
log_err(LD_GENERAL,"System time is before 1970; failing.");
|
2005-08-03 18:28:39 +02:00
|
|
|
exit(1);
|
|
|
|
}
|
|
|
|
ft.ft_64 -= EPOCH_BIAS;
|
2005-08-03 19:16:48 +02:00
|
|
|
timeval->tv_sec = (unsigned) (ft.ft_64 / UNITS_PER_SEC);
|
|
|
|
timeval->tv_usec = (unsigned) ((ft.ft_64 / UNITS_PER_USEC) % USEC_PER_SEC);
|
2005-08-03 18:28:39 +02:00
|
|
|
#elif defined(HAVE_GETTIMEOFDAY)
|
2004-11-01 21:41:47 +01:00
|
|
|
if (gettimeofday(timeval, NULL)) {
|
2006-02-13 09:01:59 +01:00
|
|
|
log_err(LD_GENERAL,"gettimeofday failed.");
|
2004-11-01 21:41:47 +01:00
|
|
|
/* If gettimeofday dies, we have either given a bad timezone (we didn't),
|
|
|
|
or segfaulted.*/
|
|
|
|
exit(1);
|
|
|
|
}
|
|
|
|
#elif defined(HAVE_FTIME)
|
|
|
|
struct timeb tb;
|
|
|
|
ftime(&tb);
|
|
|
|
timeval->tv_sec = tb.time;
|
|
|
|
timeval->tv_usec = tb.millitm * 1000;
|
|
|
|
#else
|
|
|
|
#error "No way to get time."
|
|
|
|
#endif
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
2005-02-22 08:09:17 +01:00
|
|
|
#if defined(TOR_IS_MULTITHREADED) && !defined(MS_WINDOWS)
|
|
|
|
#define TIME_FNS_NEED_LOCKS
|
|
|
|
#endif
|
|
|
|
|
2005-02-22 08:03:03 +01:00
|
|
|
#ifndef HAVE_LOCALTIME_R
|
2005-02-22 08:09:17 +01:00
|
|
|
#ifdef TIME_FNS_NEED_LOCKS
|
2005-06-11 20:52:12 +02:00
|
|
|
struct tm *
|
|
|
|
tor_localtime_r(const time_t *timep, struct tm *result)
|
2005-02-22 08:03:03 +01:00
|
|
|
{
|
|
|
|
struct tm *r;
|
|
|
|
static tor_mutex_t *m=NULL;
|
|
|
|
if (!m) { m=tor_mutex_new(); }
|
|
|
|
tor_assert(result);
|
|
|
|
tor_mutex_acquire(m);
|
|
|
|
r = localtime(timep);
|
|
|
|
memcpy(result, r, sizeof(struct tm));
|
|
|
|
tor_mutex_release(m);
|
|
|
|
return result;
|
|
|
|
}
|
2005-02-22 08:09:17 +01:00
|
|
|
#else
|
2005-06-11 20:52:12 +02:00
|
|
|
struct tm *
|
|
|
|
tor_localtime_r(const time_t *timep, struct tm *result)
|
2005-02-22 08:09:17 +01:00
|
|
|
{
|
|
|
|
struct tm *r;
|
|
|
|
tor_assert(result);
|
|
|
|
r = localtime(timep);
|
|
|
|
memcpy(result, r, sizeof(struct tm));
|
|
|
|
return result;
|
|
|
|
}
|
|
|
|
#endif
|
2005-02-22 08:03:03 +01:00
|
|
|
#endif
|
|
|
|
|
|
|
|
#ifndef HAVE_GMTIME_R
|
2005-02-22 08:09:17 +01:00
|
|
|
#ifdef TIME_FNS_NEED_LOCKS
|
2005-06-11 20:52:12 +02:00
|
|
|
struct tm *
|
|
|
|
tor_gmtime_r(const time_t *timep, struct tm *result)
|
2005-02-22 08:03:03 +01:00
|
|
|
{
|
|
|
|
struct tm *r;
|
|
|
|
static tor_mutex_t *m=NULL;
|
|
|
|
if (!m) { m=tor_mutex_new(); }
|
|
|
|
tor_assert(result);
|
|
|
|
tor_mutex_acquire(m);
|
|
|
|
r = gmtime(timep);
|
|
|
|
memcpy(result, r, sizeof(struct tm));
|
|
|
|
tor_mutex_release(m);
|
|
|
|
return result;
|
|
|
|
}
|
2005-02-22 08:09:17 +01:00
|
|
|
#else
|
2005-06-11 20:52:12 +02:00
|
|
|
struct tm *
|
|
|
|
tor_gmtime_r(const time_t *timep, struct tm *result)
|
2005-02-22 08:09:17 +01:00
|
|
|
{
|
|
|
|
struct tm *r;
|
|
|
|
tor_assert(result);
|
|
|
|
r = gmtime(timep);
|
|
|
|
memcpy(result, r, sizeof(struct tm));
|
|
|
|
return result;
|
|
|
|
}
|
|
|
|
#endif
|
2005-02-22 08:03:03 +01:00
|
|
|
#endif
|
|
|
|
|
2005-01-27 23:34:48 +01:00
|
|
|
#ifdef USE_WIN32_THREADS
|
2005-10-06 06:33:40 +02:00
|
|
|
/** A generic lock structure for multithreaded builds. */
|
2004-11-01 21:41:47 +01:00
|
|
|
struct tor_mutex_t {
|
|
|
|
HANDLE handle;
|
|
|
|
};
|
2005-06-11 20:52:12 +02:00
|
|
|
tor_mutex_t *
|
|
|
|
tor_mutex_new(void)
|
2004-11-01 21:41:47 +01:00
|
|
|
{
|
|
|
|
tor_mutex_t *m;
|
|
|
|
m = tor_malloc_zero(sizeof(tor_mutex_t));
|
|
|
|
m->handle = CreateMutex(NULL, FALSE, NULL);
|
|
|
|
tor_assert(m->handle != NULL);
|
|
|
|
return m;
|
|
|
|
}
|
2005-06-11 20:52:12 +02:00
|
|
|
void
|
|
|
|
tor_mutex_free(tor_mutex_t *m)
|
2004-11-01 21:41:47 +01:00
|
|
|
{
|
|
|
|
CloseHandle(m->handle);
|
|
|
|
tor_free(m);
|
|
|
|
}
|
2005-06-11 20:52:12 +02:00
|
|
|
void
|
|
|
|
tor_mutex_acquire(tor_mutex_t *m)
|
2004-11-01 21:41:47 +01:00
|
|
|
{
|
|
|
|
DWORD r;
|
|
|
|
r = WaitForSingleObject(m->handle, INFINITE);
|
|
|
|
switch (r) {
|
|
|
|
case WAIT_ABANDONED: /* holding thread exited. */
|
2005-07-13 19:24:55 +02:00
|
|
|
case WAIT_OBJECT_0: /* we got the mutex normally. */
|
2004-11-01 21:41:47 +01:00
|
|
|
break;
|
|
|
|
case WAIT_TIMEOUT: /* Should never happen. */
|
2005-07-13 19:24:55 +02:00
|
|
|
tor_assert(0);
|
2004-11-01 21:41:47 +01:00
|
|
|
break;
|
2005-07-13 19:24:55 +02:00
|
|
|
case WAIT_FAILED:
|
2006-02-13 09:01:59 +01:00
|
|
|
log_warn(LD_GENERAL, "Failed to acquire mutex: %d", GetLastError());
|
2004-11-01 21:41:47 +01:00
|
|
|
}
|
|
|
|
}
|
2005-06-11 20:52:12 +02:00
|
|
|
void
|
|
|
|
tor_mutex_release(tor_mutex_t *m)
|
2004-11-01 21:41:47 +01:00
|
|
|
{
|
|
|
|
BOOL r;
|
|
|
|
r = ReleaseMutex(m->handle);
|
|
|
|
if (!r) {
|
2006-02-13 09:01:59 +01:00
|
|
|
log_warn(LD_GENERAL, "Failed to release mutex: %d", GetLastError());
|
2004-11-01 21:41:47 +01:00
|
|
|
}
|
|
|
|
}
|
2005-02-13 23:32:25 +01:00
|
|
|
unsigned long
|
|
|
|
tor_get_thread_id(void)
|
|
|
|
{
|
|
|
|
return (unsigned long)GetCurrentThreadId();
|
|
|
|
}
|
2005-01-27 23:34:48 +01:00
|
|
|
#elif defined(USE_PTHREADS)
|
2005-10-06 06:33:40 +02:00
|
|
|
/** A generic lock structure for multithreaded builds. */
|
2005-01-27 23:34:48 +01:00
|
|
|
struct tor_mutex_t {
|
|
|
|
pthread_mutex_t mutex;
|
|
|
|
};
|
2005-06-11 20:52:12 +02:00
|
|
|
tor_mutex_t *
|
|
|
|
tor_mutex_new(void)
|
2005-01-27 23:34:48 +01:00
|
|
|
{
|
|
|
|
tor_mutex_t *mutex = tor_malloc_zero(sizeof(tor_mutex_t));
|
|
|
|
pthread_mutex_init(&mutex->mutex, NULL);
|
|
|
|
return mutex;
|
|
|
|
}
|
2005-06-11 20:52:12 +02:00
|
|
|
void
|
|
|
|
tor_mutex_acquire(tor_mutex_t *m)
|
2005-01-27 23:34:48 +01:00
|
|
|
{
|
|
|
|
tor_assert(m);
|
|
|
|
pthread_mutex_lock(&m->mutex);
|
|
|
|
}
|
2005-06-11 20:52:12 +02:00
|
|
|
void
|
|
|
|
tor_mutex_release(tor_mutex_t *m)
|
2005-01-27 23:34:48 +01:00
|
|
|
{
|
|
|
|
tor_assert(m);
|
|
|
|
pthread_mutex_unlock(&m->mutex);
|
|
|
|
}
|
2005-06-11 20:52:12 +02:00
|
|
|
void
|
|
|
|
tor_mutex_free(tor_mutex_t *m)
|
2005-01-27 23:34:48 +01:00
|
|
|
{
|
|
|
|
tor_assert(m);
|
|
|
|
pthread_mutex_destroy(&m->mutex);
|
|
|
|
tor_free(m);
|
|
|
|
}
|
2005-02-13 23:32:25 +01:00
|
|
|
unsigned long
|
|
|
|
tor_get_thread_id(void)
|
|
|
|
{
|
2005-09-14 22:59:25 +02:00
|
|
|
union {
|
|
|
|
pthread_t thr;
|
|
|
|
unsigned long id;
|
|
|
|
} r;
|
|
|
|
r.thr = pthread_self();
|
|
|
|
return r.id;
|
2005-02-13 23:32:25 +01:00
|
|
|
}
|
2005-01-27 23:34:48 +01:00
|
|
|
#else
|
2005-10-06 06:33:40 +02:00
|
|
|
/** A generic lock structure for multithreaded builds. */
|
2005-01-27 23:34:48 +01:00
|
|
|
struct tor_mutex_t {
|
2005-02-03 20:59:10 +01:00
|
|
|
int _unused;
|
2005-01-27 23:34:48 +01:00
|
|
|
};
|
2004-11-01 21:41:47 +01:00
|
|
|
#endif
|
|
|
|
|
|
|
|
/**
|
|
|
|
* On Windows, WSAEWOULDBLOCK is not always correct: when you see it,
|
|
|
|
* you need to ask the socket for its actual errno. Also, you need to
|
|
|
|
* get your errors from WSAGetLastError, not errno. (If you supply a
|
|
|
|
* socket of -1, we check WSAGetLastError, but don't correct
|
|
|
|
* WSAEWOULDBLOCKs.)
|
2004-11-28 06:48:02 +01:00
|
|
|
*
|
|
|
|
* The upshot of all of this is that when a socket call fails, you
|
|
|
|
* should call tor_socket_errno <em>at most once</em> on the failing
|
|
|
|
* socket to get the error.
|
2004-11-01 21:41:47 +01:00
|
|
|
*/
|
|
|
|
#ifdef MS_WINDOWS
|
2005-09-30 03:09:52 +02:00
|
|
|
int
|
|
|
|
tor_socket_errno(int sock)
|
2004-11-01 21:41:47 +01:00
|
|
|
{
|
|
|
|
int optval, optvallen=sizeof(optval);
|
|
|
|
int err = WSAGetLastError();
|
|
|
|
if (err == WSAEWOULDBLOCK && sock >= 0) {
|
|
|
|
if (getsockopt(sock, SOL_SOCKET, SO_ERROR, (void*)&optval, &optvallen))
|
|
|
|
return err;
|
|
|
|
if (optval)
|
|
|
|
return optval;
|
|
|
|
}
|
|
|
|
return err;
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
|
|
|
#ifdef MS_WINDOWS
|
|
|
|
#define E(code, s) { code, (s " [" #code " ]") }
|
|
|
|
struct { int code; const char *msg; } windows_socket_errors[] = {
|
|
|
|
E(WSAEINTR, "Interrupted function call"),
|
|
|
|
E(WSAEACCES, "Permission denied"),
|
|
|
|
E(WSAEFAULT, "Bad address"),
|
|
|
|
E(WSAEINVAL, "Invalid argument"),
|
|
|
|
E(WSAEMFILE, "Too many open files"),
|
|
|
|
E(WSAEWOULDBLOCK, "Resource temporarily unavailable"),
|
|
|
|
E(WSAEINPROGRESS, "Operation now in progress"),
|
|
|
|
E(WSAEALREADY, "Operation already in progress"),
|
|
|
|
E(WSAENOTSOCK, "Socket operation on nonsocket"),
|
|
|
|
E(WSAEDESTADDRREQ, "Destination address required"),
|
|
|
|
E(WSAEMSGSIZE, "Message too long"),
|
|
|
|
E(WSAEPROTOTYPE, "Protocol wrong for socket"),
|
|
|
|
E(WSAENOPROTOOPT, "Bad protocol option"),
|
|
|
|
E(WSAEPROTONOSUPPORT, "Protocol not supported"),
|
|
|
|
E(WSAESOCKTNOSUPPORT, "Socket type not supported"),
|
|
|
|
/* What's the difference between NOTSUPP and NOSUPPORT? :) */
|
|
|
|
E(WSAEOPNOTSUPP, "Operation not supported"),
|
|
|
|
E(WSAEPFNOSUPPORT, "Protocol family not supported"),
|
|
|
|
E(WSAEAFNOSUPPORT, "Address family not supported by protocol family"),
|
|
|
|
E(WSAEADDRINUSE, "Address already in use"),
|
|
|
|
E(WSAEADDRNOTAVAIL, "Cannot assign requested address"),
|
|
|
|
E(WSAENETDOWN, "Network is down"),
|
|
|
|
E(WSAENETUNREACH, "Network is unreachable"),
|
|
|
|
E(WSAENETRESET, "Network dropped connection on reset"),
|
|
|
|
E(WSAECONNABORTED, "Software caused connection abort"),
|
|
|
|
E(WSAECONNRESET, "Connection reset by peer"),
|
2004-12-01 04:48:14 +01:00
|
|
|
E(WSAENOBUFS, "No buffer space available"),
|
2004-11-01 21:41:47 +01:00
|
|
|
E(WSAEISCONN, "Socket is already connected"),
|
|
|
|
E(WSAENOTCONN, "Socket is not connected"),
|
|
|
|
E(WSAESHUTDOWN, "Cannot send after socket shutdown"),
|
|
|
|
E(WSAETIMEDOUT, "Connection timed out"),
|
|
|
|
E(WSAECONNREFUSED, "Connection refused"),
|
|
|
|
E(WSAEHOSTDOWN, "Host is down"),
|
|
|
|
E(WSAEHOSTUNREACH, "No route to host"),
|
|
|
|
E(WSAEPROCLIM, "Too many processes"),
|
|
|
|
/* Yes, some of these start with WSA, not WSAE. No, I don't know why. */
|
|
|
|
E(WSASYSNOTREADY, "Network subsystem is unavailable"),
|
|
|
|
E(WSAVERNOTSUPPORTED, "Winsock.dll out of range"),
|
|
|
|
E(WSANOTINITIALISED, "Successful WSAStartup not yet performed"),
|
|
|
|
E(WSAEDISCON, "Graceful shutdown now in progress"),
|
|
|
|
#ifdef WSATYPE_NOT_FOUND
|
|
|
|
E(WSATYPE_NOT_FOUND, "Class type not found"),
|
|
|
|
#endif
|
|
|
|
E(WSAHOST_NOT_FOUND, "Host not found"),
|
|
|
|
E(WSATRY_AGAIN, "Nonauthoritative host not found"),
|
|
|
|
E(WSANO_RECOVERY, "This is a nonrecoverable error"),
|
|
|
|
E(WSANO_DATA, "Valid name, no data record of requested type)"),
|
|
|
|
|
|
|
|
/* There are some more error codes whose numeric values are marked
|
|
|
|
* <b>OS dependent</b>. They start with WSA_, apparently for the same
|
|
|
|
* reason that practitioners of some craft traditions deliberately
|
|
|
|
* introduce imperfections into their baskets and rugs "to allow the
|
|
|
|
* evil spirits to escape." If we catch them, then our binaries
|
|
|
|
* might not report consistent results across versions of Windows.
|
|
|
|
* Thus, I'm going to let them all fall through.
|
|
|
|
*/
|
|
|
|
{ -1, NULL },
|
|
|
|
};
|
|
|
|
/** There does not seem to be a strerror equivalent for winsock errors.
|
|
|
|
* Naturally, we have to roll our own.
|
|
|
|
*/
|
2005-06-11 20:52:12 +02:00
|
|
|
const char *
|
|
|
|
tor_socket_strerror(int e)
|
2004-11-01 21:41:47 +01:00
|
|
|
{
|
|
|
|
int i;
|
|
|
|
for (i=0; windows_socket_errors[i].code >= 0; ++i) {
|
|
|
|
if (e == windows_socket_errors[i].code)
|
|
|
|
return windows_socket_errors[i].msg;
|
|
|
|
}
|
|
|
|
return strerror(e);
|
|
|
|
}
|
|
|
|
#endif
|
2004-12-22 06:29:06 +01:00
|
|
|
|
|
|
|
/** Called before we make any calls to network-related functions.
|
|
|
|
* (Some operating systems require their network libraries to be
|
|
|
|
* initialized.) */
|
2005-06-11 20:52:12 +02:00
|
|
|
int
|
|
|
|
network_init(void)
|
2004-12-22 06:29:06 +01:00
|
|
|
{
|
|
|
|
#ifdef MS_WINDOWS
|
2005-06-12 06:33:26 +02:00
|
|
|
/* This silly exercise is necessary before windows will allow
|
|
|
|
* gethostbyname to work. */
|
2004-12-22 06:29:06 +01:00
|
|
|
WSADATA WSAData;
|
|
|
|
int r;
|
|
|
|
r = WSAStartup(0x101,&WSAData);
|
|
|
|
if (r) {
|
2006-02-13 09:01:59 +01:00
|
|
|
log_warn(LD_NET,"Error initializing windows network layer: code was %d",r);
|
2004-12-22 06:29:06 +01:00
|
|
|
return -1;
|
|
|
|
}
|
2005-03-28 03:11:02 +02:00
|
|
|
/* WSAData.iMaxSockets might show the max sockets we're allowed to use.
|
|
|
|
* We might use it to complain if we're trying to be a server but have
|
|
|
|
* too few sockets available. */
|
2004-12-22 06:29:06 +01:00
|
|
|
#endif
|
|
|
|
return 0;
|
|
|
|
}
|
2005-06-09 21:03:31 +02:00
|
|
|
|