tor/src/feature/api/tor_api.c

168 lines
4.2 KiB
C
Raw Normal View History

/* Copyright (c) 2001 Matej Pfajfar.
* Copyright (c) 2001-2004, Roger Dingledine.
* Copyright (c) 2004-2006, Roger Dingledine, Nick Mathewson.
2019-01-16 18:33:22 +01:00
* Copyright (c) 2007-2019, The Tor Project, Inc. */
/* See LICENSE for licensing information */
/**
* \file tor_api.c
**/
#ifdef _WIN32
#include <winsock2.h>
#include <windows.h>
#endif
#include "feature/api/tor_api.h"
// Include this after the above headers, to insure that they don't
// depend on anything else.
#include "orconfig.h"
#include "lib/cc/torint.h"
#include "feature/api/tor_api_internal.h"
#include "lib/cc/compat_compiler.h"
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
// We don't want to use tor_malloc and tor_free here, since this needs
// to run before anything is initialized at all, and ought to run when
// we're not linked to anything at all.
#define raw_malloc malloc
#define raw_free free
#define raw_realloc realloc
#define raw_strdup strdup
#ifdef _WIN32
#include "lib/net/socketpair.h"
#define raw_socketpair tor_ersatz_socketpair
#define raw_closesocket closesocket
#define snprintf _snprintf
#else /* !defined(_WIN32) */
#define raw_socketpair socketpair
#define raw_closesocket close
2019-06-05 15:33:35 +02:00
#endif /* defined(_WIN32) */
#ifdef HAVE_UNISTD_H
#include <unistd.h>
#endif
/**
* Helper: Add a copy of <b>arg</b> to the owned arguments of <b>cfg</b>.
* Return 0 on success, -1 on failure.
*/
static int
cfg_add_owned_arg(tor_main_configuration_t *cfg, const char *arg)
{
/* We aren't using amortized realloc here, because libc should do it for us,
* and because this function is not critical-path. */
char **new_argv = raw_realloc(cfg->argv_owned,
sizeof(char*) * (cfg->argc_owned+1));
if (new_argv == NULL)
return -1;
cfg->argv_owned = new_argv;
if (NULL == (cfg->argv_owned[cfg->argc_owned] = raw_strdup(arg)))
return -1;
++cfg->argc_owned;
return 0;
}
tor_main_configuration_t *
tor_main_configuration_new(void)
{
static const char *fake_argv[] = { "tor" };
tor_main_configuration_t *cfg = raw_malloc(sizeof(*cfg));
if (cfg == NULL)
return NULL;
memset(cfg, 0, sizeof(*cfg));
cfg->argc = 1;
cfg->argv = (char **) fake_argv;
cfg->owning_controller_socket = TOR_INVALID_SOCKET;
return cfg;
}
int
tor_main_configuration_set_command_line(tor_main_configuration_t *cfg,
int argc, char *argv[])
{
if (cfg == NULL)
return -1;
cfg->argc = argc;
cfg->argv = argv;
return 0;
}
tor_control_socket_t
tor_main_configuration_setup_control_socket(tor_main_configuration_t *cfg)
{
if (SOCKET_OK(cfg->owning_controller_socket))
return INVALID_TOR_CONTROL_SOCKET;
tor_socket_t fds[2];
if (raw_socketpair(AF_UNIX, SOCK_STREAM, 0, fds) < 0) {
return INVALID_TOR_CONTROL_SOCKET;
}
char buf[32];
snprintf(buf, sizeof(buf), "%"PRIu64, (uint64_t)fds[1]);
cfg_add_owned_arg(cfg, "__OwningControllerFD");
cfg_add_owned_arg(cfg, buf);
cfg->owning_controller_socket = fds[1];
return fds[0];
}
void
tor_main_configuration_free(tor_main_configuration_t *cfg)
{
if (cfg == NULL)
return;
if (cfg->argv_owned) {
2018-09-04 17:04:55 +02:00
for (int i = 0; i < cfg->argc_owned; ++i) {
raw_free(cfg->argv_owned[i]);
}
raw_free(cfg->argv_owned);
}
if (SOCKET_OK(cfg->owning_controller_socket)) {
raw_closesocket(cfg->owning_controller_socket);
}
raw_free(cfg);
}
const char *
tor_api_get_provider_version(void)
{
return "tor " VERSION;
}
/* Main entry point for the Tor process. Called from main().
*
* This function is distinct from main() only so we can link main.c into
* the unittest binary without conflicting with the unittests' main.
*
* Some embedders have historically called this function; but that usage is
* deprecated: they should use tor_run_main() instead.
*/
int
tor_main(int argc, char *argv[])
{
tor_main_configuration_t *cfg = tor_main_configuration_new();
if (!cfg) {
puts("INTERNAL ERROR: Allocation failure. Cannot proceed");
return 1;
}
if (tor_main_configuration_set_command_line(cfg, argc, argv) < 0) {
puts("INTERNAL ERROR: Can't set command line. Cannot proceed.");
return 1;
}
int rv = tor_run_main(cfg);
tor_main_configuration_free(cfg);
return rv;
}