proxychains-ng/src/main.c

183 lines
4.6 KiB
C
Raw Normal View History

2011-02-25 09:40:11 +00:00
/***************************************************************************
main.c - description
2011-02-25 09:40:11 +00:00
begin : Tue May 14 2002
copyright : netcreature (C) 2002
email : netcreature@users.sourceforge.net
2011-02-25 09:40:11 +00:00
***************************************************************************/
/* GPL */
/***************************************************************************
* *
* This program is free software; you can redistribute it and/or modify *
* it under the terms of the GNU General Public License as published by *
* the Free Software Foundation; either version 2 of the License, or *
* (at your option) any later version. *
* *
***************************************************************************/
#undef _POSIX_C_SOURCE
#define _POSIX_C_SOURCE 200809L
#undef _XOPEN_SOURCE
#define _XOPEN_SOURCE 700
2011-02-25 09:40:11 +00:00
#include <stdio.h>
#include <unistd.h>
#include <stdlib.h>
#include <string.h>
2011-02-25 09:40:11 +00:00
#include <errno.h>
#include <sys/types.h>
#include <sys/wait.h>
#include "common.h"
static int usage(char **argv) {
printf("\nUsage:\t%s -q -f config_file program_name [arguments]\n"
"\t-q makes proxychains quiet - this overrides the config setting\n"
"\t-t allows to manually specify a configfile to use\n"
"\tfor example : proxychains telnet somehost.com\n" "More help in README file\n\n", argv[0]);
return EXIT_FAILURE;
}
int check_path(char *path) {
if(!path)
return 0;
return access(path, R_OK) != -1;
}
static const char *dll_name = DLL_NAME;
2011-11-06 13:11:36 +00:00
static char own_dir[256];
static const char *dll_dirs[] = {
2011-11-06 13:11:36 +00:00
".",
own_dir,
2011-11-06 13:11:36 +00:00
LIB_DIR,
"/lib",
"/usr/lib",
"/usr/local/lib",
"/lib64",
NULL
};
static void set_own_dir(const char *argv0) {
size_t l = strlen(argv0);
while(l && argv0[l - 1] != '/')
l--;
if(l == 0)
memcpy(own_dir, ".", 2);
else {
memcpy(own_dir, argv0, l - 1);
own_dir[l] = 0;
}
}
2012-01-27 16:50:04 +00:00
#define MAX_COMMANDLINE_FLAGS 2
int main(int argc, char *argv[]) {
char *path = NULL;
char buf[256];
char pbuf[256];
int start_argv = 1;
int quiet = 0;
size_t i;
const char *prefix = NULL;
2012-01-27 16:55:37 +00:00
2012-01-27 16:50:04 +00:00
for(i = 0; i < MAX_COMMANDLINE_FLAGS; i++) {
if(start_argv < argc && argv[start_argv][0] == '-') {
if(argv[start_argv][1] == 'q') {
quiet = 1;
start_argv++;
} else if(argv[start_argv][1] == 'f') {
2012-01-27 16:55:37 +00:00
if(start_argv + 1 < argc)
path = argv[start_argv + 1];
2012-01-27 16:55:37 +00:00
else
return usage(argv);
2012-01-27 16:55:37 +00:00
start_argv += 2;
}
2012-01-27 16:55:37 +00:00
} else
break;
}
if(start_argv >= argc)
return usage(argv);
/* check if path of config file has not been passed via command line */
if(!path) {
// priority 1: env var PROXYCHAINS_CONF_FILE
path = getenv(PROXYCHAINS_CONF_FILE_ENV_VAR);
if(check_path(path))
goto have;
2012-01-27 16:55:37 +00:00
// priority 2; proxychains conf in actual dir
path = getcwd(buf, sizeof(buf));
snprintf(pbuf, sizeof(pbuf), "%s/%s", path, PROXYCHAINS_CONF_FILE);
path = pbuf;
if(check_path(path))
goto have;
2012-01-27 16:55:37 +00:00
// priority 3; $HOME/.proxychains/proxychains.conf
path = getenv("HOME");
snprintf(pbuf, sizeof(pbuf), "%s/.proxychains/%s", path, PROXYCHAINS_CONF_FILE);
path = pbuf;
if(check_path(path))
goto have;
2012-01-27 16:55:37 +00:00
// priority 4: $INSTALL_PREFIX/etc/proxychains.conf
2012-01-28 17:33:40 +00:00
snprintf(pbuf, sizeof(pbuf), "%s/etc/%s", INSTALL_PREFIX, PROXYCHAINS_CONF_FILE);
path = pbuf;
if(check_path(path))
goto have;
// priority 5: /etc/proxychains.conf
path = "/etc/proxychains.conf";
if(check_path(path))
goto have;
perror("couldnt find configuration file");
return 1;
}
2012-01-27 16:55:37 +00:00
have:
if(!quiet)
fprintf(stderr, LOG_PREFIX "config file found: %s\n", path);
/* Set PROXYCHAINS_CONF_FILE to get proxychains lib to use new config file. */
setenv(PROXYCHAINS_CONF_FILE_ENV_VAR, path, 1);
2012-01-27 16:55:37 +00:00
if(quiet)
setenv(PROXYCHAINS_QUIET_MODE_ENV_VAR, "1", 1);
2011-11-06 13:11:36 +00:00
// search DLL
set_own_dir(argv[0]);
2011-11-06 13:11:36 +00:00
while(dll_dirs[i]) {
snprintf(buf, sizeof(buf), "%s/%s", dll_dirs[i], dll_name);
if(access(buf, R_OK) != -1) {
prefix = dll_dirs[i];
break;
}
i++;
}
if(!prefix) {
fprintf(stderr, "couldnt locate %s\n", dll_name);
2011-11-06 13:11:36 +00:00
return EXIT_FAILURE;
}
if(!quiet)
fprintf(stderr, LOG_PREFIX "preloading %s/%s\n", prefix, dll_name);
2012-01-27 16:55:37 +00:00
2012-01-27 19:48:24 +00:00
#ifndef IS_MAC
2011-11-06 13:11:36 +00:00
snprintf(buf, sizeof(buf), "LD_PRELOAD=%s/%s", prefix, dll_name);
putenv(buf);
2012-01-27 19:48:24 +00:00
#else
snprintf(buf, sizeof(buf), "DYLD_INSERT_LIBRARIES=%s/%s", prefix, dll_name);
putenv(buf);
putenv("DYLD_FORCE_FLAT_NAMESPACE=1");
#endif
execvp(argv[start_argv], &argv[start_argv]);
perror("proxychains can't load process....");
return EXIT_FAILURE;
2011-02-25 09:40:11 +00:00
}