int crypto_init() {
if (-1 == sodium_init())
return 1;
randombytes_set_implementation(&randombytes_salsa20_implementation);
randombytes_stir();
return 0;
}
int
crypto_init(const char *password) {
if (sodium_init() == -1) {
return 1;
}
randombytes_set_implementation(&randombytes_salsa20_implementation);
randombytes_stir();
return crypto_generichash(secret_key, sizeof secret_key, (uint8_t*)password, strlen(password), NULL, 0);
}
static int impl_tests(void)
{
#ifndef __native_client__
randombytes_implementation impl = randombytes_sysrandom_implementation;
#else
randombytes_implementation impl = randombytes_nativeclient_implementation;
#endif
uint32_t v = randombytes_random();
impl.uniform = randombytes_uniform_impl;
randombytes_close();
randombytes_set_implementation(&impl);
assert(randombytes_uniform(v) == v);
assert(randombytes_uniform(v) == v);
assert(randombytes_uniform(v) == v);
assert(randombytes_uniform(v) == v);
randombytes_close();
impl.close = NULL;
randombytes_close();
return 0;
}
int wmain(int argc, wchar_t* argv[])
{
//Windows XP with SP3 support
#if (defined(PLATFORM_WIN32) && !defined(PLATFORM_WIN64))
GetFunctionPointer(FUNCTION_GETTICKCOUNT64);
GetFunctionPointer(FUNCTION_INET_NTOP);
#endif
#elif (defined(PLATFORM_LINUX) || defined(PLATFORM_MACX))
int main(int argc, char *argv[])
{
#endif
//Get commands.
if (argc > 0)
{
if (!ReadCommand(argc, argv))
return EXIT_SUCCESS;
}
else {
return EXIT_FAILURE;
}
//Read configuration file and WinPcap or LibPcap initialization.
if (!ReadParameter())
{
WSACleanup();
return EXIT_FAILURE;
}
//Mark Local DNS address to PTR Records.
std::thread NetworkInformationMonitorThread(NetworkInformationMonitor);
NetworkInformationMonitorThread.detach();
//Read IPFilter and Hosts.
if (Parameter.OperationMode == LISTEN_MODE_CUSTOM || Parameter.BlacklistCheck || Parameter.LocalRouting)
{
std::thread IPFilterThread(ReadIPFilter);
IPFilterThread.detach();
}
std::thread HostsThread(ReadHosts);
HostsThread.detach();
//DNSCurve initialization
#if defined(ENABLE_LIBSODIUM)
if (Parameter.DNSCurve && DNSCurveParameter.IsEncryption)
{
randombytes_set_implementation(&randombytes_salsa20_implementation);
randombytes_stir();
DNSCurveInit();
}
#endif
#if defined(PLATFORM_WIN)
//Service initialization and start service.
SERVICE_TABLE_ENTRYW ServiceTable[] = {{DEFAULT_LOCAL_SERVICE_NAME, (LPSERVICE_MAIN_FUNCTIONW)ServiceMain}, {nullptr, nullptr}};
if (!StartServiceCtrlDispatcherW(ServiceTable))
{
Parameter.Console = true;
wprintf_s(L"System Error: Service start error, error code is %lu.\n", GetLastError());
wprintf_s(L"System Error: Program will continue to run in console mode.\n");
wprintf_s(L"Please ignore those error messages if you want to run in console mode.\n");
//Handle the system signal and start all monitors.
SetConsoleCtrlHandler((PHANDLER_ROUTINE)CtrlHandler, TRUE);
MonitorInit();
}
#elif (defined(PLATFORM_LINUX) || defined(PLATFORM_MACX))
MonitorInit();
#endif
WSACleanup();
return EXIT_SUCCESS;
}
//Read commands from main program
#if defined(PLATFORM_WIN)
bool __fastcall ReadCommand(int argc, wchar_t* argv[])
#elif (defined(PLATFORM_LINUX) || defined(PLATFORM_MACX))
bool ReadCommand(int argc, char *argv[])
#endif
{
//Path initialization
#if defined(PLATFORM_WIN)
if (!FileNameInit(argv[0]))
return false;
#elif (defined(PLATFORM_LINUX) || defined(PLATFORM_MACX))
std::shared_ptr<char> FileName(new char[PATH_MAX + 1U]());
memset(FileName.get(), 0, PATH_MAX + 1U);
if (getcwd(FileName.get(), PATH_MAX) == nullptr)
{
wprintf(L"Path initialization error.\n");
return false;
}
if (!FileNameInit(FileName.get()))
return false;
FileName.reset();
#endif
#if defined(PLATFORM_WIN)
//Winsock initialization
std::shared_ptr<WSAData> WSAInitialization(new WSAData());
if (WSAStartup(MAKEWORD(WINSOCK_VERSION_HIGH, WINSOCK_VERSION_LOW), WSAInitialization.get()) != 0 ||
LOBYTE(WSAInitialization->wVersion) != WINSOCK_VERSION_LOW || HIBYTE(WSAInitialization->wVersion) != WINSOCK_VERSION_HIGH)
{
wprintf_s(L"Winsock initialization error, error code is %d.\n", WSAGetLastError());
PrintError(LOG_ERROR_NETWORK, L"Winsock initialization error", WSAGetLastError(), nullptr, 0);
WSACleanup();
return false;
}
//Read commands.
std::wstring Commands;
#elif (defined(PLATFORM_LINUX) || defined(PLATFORM_MACX))
std::string Commands;
#endif
for (size_t Index = 1U;(SSIZE_T)Index < argc;++Index)
{
Commands = argv[Index];
//Flush DNS Cache from user.
if (Commands == COMMAND_FLUSH_DNS)
{
#if defined(PLATFORM_WIN)
FlushDNSMailSlotSender();
#elif (defined(PLATFORM_LINUX) || defined(PLATFORM_MACX))
FlushDNSFIFOSender();
#endif
WSACleanup();
return false;
}
//Windows Firewall Test in first start.
#if defined(PLATFORM_WIN)
else if (Commands == COMMAND_FIREWALL_TEST)
{
if (!FirewallTest(AF_INET6) && !FirewallTest(AF_INET))
{
wprintf_s(L"Windows Firewall Test error.\n");
PrintError(LOG_ERROR_NETWORK, L"Windows Firewall Test error", WSAGetLastError(), nullptr, 0);
}
WSACleanup();
return false;
}
#endif
//Set system daemon.
#if defined(PLATFORM_LINUX)
else if (Commands == COMMAND_DISABLE_DAEMON)
{
Parameter.Daemon = false;
}
#endif
//Print current version.
else if (Commands == COMMAND_LONG_PRINT_VERSION || Commands == COMMAND_SHORT_PRINT_VERSION)
{
wprintf_s(L"Pcap_DNSProxy ");
wprintf_s(FULL_VERSION);
wprintf_s(L"\n");
WSACleanup();
return false;
}
//Print help messages.
else if (Commands == COMMAND_LONG_HELP || Commands == COMMAND_SHORT_HELP)
{
wprintf_s(L"Usage: Please see ReadMe... files in Documents folder.\n");
WSACleanup();
return false;
}
//Set working directory from commands.
else if (Commands == COMMAND_LONG_SET_PATH || Commands == COMMAND_SHORT_SET_PATH)
{
//Commands check
if ((SSIZE_T)Index + 1 >= argc)
{
wprintf_s(L"Commands error.\n");
PrintError(LOG_ERROR_SYSTEM, L"Commands error", 0, nullptr, 0);
WSACleanup();
return false;
}
else {
++Index;
Commands = argv[Index];
//Path check.
if (Commands.length() > MAX_PATH)
{
wprintf_s(L"Commands error.\n");
PrintError(LOG_ERROR_SYSTEM, L"Commands error", 0, nullptr, 0);
WSACleanup();
return false;
}
else {
if (!FileNameInit(Commands.c_str()))
return false;
}
}
}
}
//Set system daemon.
#if defined(PLATFORM_LINUX)
if (Parameter.Daemon && daemon(0, 0) == RETURN_ERROR)
{
PrintError(LOG_ERROR_SYSTEM, L"Set system daemon error", 0, nullptr, 0);
return false;
}
#endif
return true;
}
//Get path of program from the main function parameter and Winsock initialization
#if defined(PLATFORM_WIN)
bool __fastcall FileNameInit(const wchar_t *OriginalPath)
#elif (defined(PLATFORM_LINUX) || defined(PLATFORM_MACX))
bool FileNameInit(const char *OriginalPath)
#endif
{
//Path process
#if defined(PLATFORM_WIN)
Parameter.Path_Global->clear();
Parameter.Path_Global->push_back(OriginalPath);
Parameter.Path_Global->front().erase(Parameter.Path_Global->front().rfind(L"\\") + 1U);
for (size_t Index = 0;Index < Parameter.Path_Global->front().length();++Index)
{
if ((Parameter.Path_Global->front()).at(Index) == L'\\')
{
Parameter.Path_Global->front().insert(Index, L"\\");
++Index;
}
}
#elif (defined(PLATFORM_LINUX) || defined(PLATFORM_MACX))
Parameter.sPath_Global->clear();
Parameter.sPath_Global->push_back(OriginalPath);
Parameter.sPath_Global->front().append("/");
std::wstring StringTemp;
MBSToWCSString(StringTemp, OriginalPath);
StringTemp.append(L"/");
Parameter.Path_Global->clear();
Parameter.Path_Global->push_back(StringTemp);
StringTemp.clear();
#endif
//Get path of error/running status log file and mark start time.
Parameter.Path_ErrorLog->clear();
*Parameter.Path_ErrorLog = Parameter.Path_Global->front();
Parameter.Path_ErrorLog->append(L"Error.log");
#if (defined(PLATFORM_LINUX) || defined(PLATFORM_MACX))
Parameter.sPath_ErrorLog->clear();
*Parameter.sPath_ErrorLog = Parameter.sPath_Global->front();
Parameter.sPath_ErrorLog->append("Error.log");
#endif
Parameter.PrintError = true;
time(&StartTime);
return true;
}
static int randombytes_tests(void)
{
unsigned int f = 0U;
unsigned int i;
uint32_t n;
#ifdef __EMSCRIPTEN__
assert(strcmp(randombytes_implementation_name(), "sysrandom"));
#else
assert(strcmp(randombytes_implementation_name(), "js"));
#endif
randombytes(x, 1U);
do {
n = randombytes_random();
f |= ((n >> 24) > 1);
f |= ((n >> 16) > 1) << 1;
f |= ((n >> 8) > 1) << 2;
f |= ((n ) > 1) << 3;
f |= (n > 0x7fffffff) << 4;
} while (f != 0x1f);
randombytes_close();
for (i = 0; i < 256; ++i) {
freq[i] = 0;
}
for (i = 0; i < 65536; ++i) {
++freq[randombytes_uniform(256)];
}
for (i = 0; i < 256; ++i) {
if (!freq[i]) {
printf("randombytes_uniform() test failed\n");
}
}
assert(randombytes_uniform(1U) == 0U);
randombytes_close();
#ifndef __EMSCRIPTEN__
randombytes_set_implementation(&randombytes_salsa20_implementation);
assert(strcmp(randombytes_implementation_name(), "salsa20") == 0);
#endif
randombytes_stir();
for (i = 0; i < 256; ++i) {
freq[i] = 0;
}
for (i = 0; i < 65536; ++i) {
++freq[randombytes_uniform(256)];
}
for (i = 0; i < 256; ++i) {
if (!freq[i]) {
printf("randombytes_uniform() test failed\n");
}
}
memset(x, 0, sizeof x);
randombytes_buf(x, sizeof x);
for (i = 0; i < 256; ++i) {
freq[i] = 0;
}
for (i = 0; i < sizeof x; ++i) {
++freq[255 & (int)x[i]];
}
for (i = 0; i < 256; ++i) {
if (!freq[i]) {
printf("randombytes_buf() test failed\n");
}
}
assert(randombytes_uniform(1U) == 0U);
randombytes_close();
randombytes(x, 1U);
randombytes_close();
return 0;
}
int
dnscrypt_proxy_main(int argc, char *argv[])
{
ProxyContext proxy_context;
setvbuf(stdout, NULL, _IOLBF, BUFSIZ);
stack_trace_on_crash();
#ifdef PLUGINS
if ((app_context.dcps_context = plugin_support_context_new()) == NULL) {
logger_noformat(NULL, LOG_ERR, "Unable to setup plugin support");
exit(2);
}
#endif
if (proxy_context_init(&proxy_context, argc, argv) != 0) {
logger_noformat(NULL, LOG_ERR, "Unable to start the proxy");
exit(1);
}
#ifdef USE_ONLY_PORTABLE_IMPLEMENTATIONS
randombytes_stir();
#else
logger_noformat(&proxy_context, LOG_INFO,
"Initializing libsodium for optimal performance");
if (sodium_init() != 0) {
exit(1);
}
#endif
randombytes_set_implementation(&randombytes_salsa20_implementation);
#ifdef PLUGINS
if (plugin_support_context_load(app_context.dcps_context) != 0) {
logger_noformat(NULL, LOG_ERR, "Unable to load plugins");
exit(2);
}
#endif
app_context.proxy_context = &proxy_context;
logger_noformat(&proxy_context, LOG_INFO, "Generating a new key pair");
dnscrypt_client_init_with_new_key_pair(&proxy_context.dnscrypt_client);
logger_noformat(&proxy_context, LOG_INFO, "Done");
if (cert_updater_init(&proxy_context) != 0 ||
udp_listener_bind(&proxy_context) != 0 ||
tcp_listener_bind(&proxy_context) != 0) {
exit(1);
}
#ifdef SIGPIPE
signal(SIGPIPE, SIG_IGN);
#endif
revoke_privileges(&proxy_context);
if (cert_updater_start(&proxy_context) != 0) {
exit(1);
}
if (skip_dispatch == 0) {
event_base_dispatch(proxy_context.event_loop);
}
logger_noformat(&proxy_context, LOG_INFO, "Stopping proxy");
cert_updater_free(&proxy_context);
udp_listener_stop(&proxy_context);
tcp_listener_stop(&proxy_context);
event_base_free(proxy_context.event_loop);
#ifdef PLUGINS
plugin_support_context_free(app_context.dcps_context);
#endif
proxy_context_free(&proxy_context);
app_context.proxy_context = NULL;
randombytes_close();
return 0;
}
//The Main function of program
int main(int argc, char *argv[])
{
#ifdef _DEBUG
//Handle the system signal.
SetConsoleCtrlHandler((PHANDLER_ROUTINE)CtrlHandler, TRUE);
#endif
if (argc > 0)
{
std::shared_ptr<wchar_t> wPath(new wchar_t[MAX_PATH]());
//Path initialization and Winsock initialization.
MultiByteToWideChar(CP_ACP, NULL, argv[0], MBSTOWCS_NULLTERMINATE, wPath.get(), MAX_PATH);
if (FileInit(wPath.get()) == EXIT_FAILURE)
return EXIT_FAILURE;
wPath.reset();
//Windows Firewall Test in first start.
if (argc > 1 && strlen(argv[1U]) == strlen("--FirstStart") && memcmp(argv[1], ("--FirstStart"), strlen("--FirstStart")) == 0)
{
if (FirewallTest(AF_INET6) == EXIT_FAILURE && FirewallTest(AF_INET) == EXIT_FAILURE)
{
PrintError(WINSOCK_ERROR, L"Windows Firewall Test error", NULL, nullptr, NULL);
WSACleanup();
return EXIT_FAILURE;
}
else {
return EXIT_SUCCESS;
}
}
}
else {
return EXIT_FAILURE;
}
//Read configuration file and WinPcap initialization.
if (Parameter.ReadParameter() == EXIT_FAILURE)
{
WSACleanup();
return EXIT_FAILURE;
}
std::thread CaptureInitializationThread(CaptureInit);
CaptureInitializationThread.detach();
//Get Localhost DNS PTR Records.
std::thread IPv6LocalAddressThread(LocalAddressToPTR, AF_INET6);
std::thread IPv4LocalAddressThread(LocalAddressToPTR, AF_INET);
IPv6LocalAddressThread.detach();
IPv4LocalAddressThread.detach();
//DNSCurve initialization
if (Parameter.DNSCurve && DNSCurveParameter.Encryption)
{
randombytes_set_implementation(&randombytes_salsa20_implementation);
DNSCurveInit();
}
//Read IPFilter, start DNS Cache monitor(Timer type) and read Hosts.
if (Parameter.FileRefreshTime > 0)
{
if (Parameter.OperationMode == LISTEN_CUSTOMMODE)
{
std::thread IPFilterThread(&Configuration::ReadIPFilter, std::ref(Parameter));
IPFilterThread.detach();
}
if (Parameter.CacheType != 0)
{
std::thread DNSCacheTimerThread(DNSCacheTimerMonitor, Parameter.CacheType);
DNSCacheTimerThread.detach();
}
std::thread HostsThread(&Configuration::ReadHosts, std::ref(Parameter));
HostsThread.detach();
}
//Service initialization and start service.
SERVICE_TABLE_ENTRYW ServiceTable[] = {{LOCAL_SERVICENAME, (LPSERVICE_MAIN_FUNCTIONW)ServiceMain}, {nullptr, NULL}};
if (!StartServiceCtrlDispatcherW(ServiceTable))
{
PrintError(SYSTEM_ERROR, L"Service start error", GetLastError(), nullptr, NULL);
WSACleanup();
return EXIT_FAILURE;
}
WSACleanup();
return EXIT_SUCCESS;
}
