2013-02-26 19:02:50 +00:00
|
|
|
/*
|
|
|
|
* Copyright © 2012,2013 Naim A.
|
|
|
|
*
|
|
|
|
* This file is part of UDPT.
|
|
|
|
*
|
|
|
|
* UDPT 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 3 of the License, or
|
|
|
|
* (at your option) any later version.
|
|
|
|
*
|
|
|
|
* UDPT is distributed in the hope that it will be useful,
|
|
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
|
|
* GNU General Public License for more details.
|
|
|
|
*
|
|
|
|
* You should have received a copy of the GNU General Public License
|
|
|
|
* along with UDPT. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
*/
|
|
|
|
|
|
|
|
#include <iostream>
|
|
|
|
|
2013-08-10 20:12:08 +01:00
|
|
|
#include "logging.h"
|
2013-02-26 19:02:50 +00:00
|
|
|
#include "multiplatform.h"
|
|
|
|
#include "udpTracker.hpp"
|
|
|
|
#include "settings.hpp"
|
2013-03-08 00:42:02 +00:00
|
|
|
#include "http/httpserver.hpp"
|
2013-03-21 00:44:10 +00:00
|
|
|
#include "http/webapp.hpp"
|
2013-03-22 11:54:50 +00:00
|
|
|
#include <cstdlib> // atoi
|
2013-08-16 15:06:09 +01:00
|
|
|
#include <csignal> // signal
|
2013-02-26 19:02:50 +00:00
|
|
|
|
|
|
|
using namespace std;
|
|
|
|
using namespace UDPT;
|
2013-03-10 21:18:37 +00:00
|
|
|
using namespace UDPT::Server;
|
2013-02-26 19:02:50 +00:00
|
|
|
|
2013-08-10 20:12:08 +01:00
|
|
|
Logger *logger;
|
2013-08-16 16:56:27 +01:00
|
|
|
static struct {
|
|
|
|
Settings *settings;
|
|
|
|
UDPTracker *usi;
|
|
|
|
WebApp *wa;
|
|
|
|
HTTPServer *httpserver;
|
|
|
|
} Instance;
|
2013-08-10 20:12:08 +01:00
|
|
|
|
2013-02-26 19:02:50 +00:00
|
|
|
static void _print_usage ()
|
|
|
|
{
|
|
|
|
cout << "Usage: udpt [<configuration file>]" << endl;
|
|
|
|
}
|
|
|
|
|
2013-03-22 00:20:49 +00:00
|
|
|
static void _doAPIStart (Settings *settings, WebApp **wa, HTTPServer **srv, DatabaseDriver *drvr)
|
|
|
|
{
|
|
|
|
if (settings == NULL)
|
|
|
|
return;
|
|
|
|
Settings::SettingClass *sc = settings->getClass("apiserver");
|
|
|
|
if (sc == NULL)
|
|
|
|
return; // no settings set!
|
|
|
|
|
|
|
|
if (sc->get("enable") != "1")
|
|
|
|
{
|
|
|
|
cerr << "API Server not enabled." << endl;
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
string s_port = sc->get("port");
|
|
|
|
string s_threads = sc->get("threads");
|
|
|
|
|
|
|
|
uint16_t port = (s_port == "" ? 6969 : atoi (s_port.c_str()));
|
|
|
|
uint16_t threads = (s_threads == "" ? 1 : atoi (s_threads.c_str()));
|
|
|
|
|
|
|
|
if (threads <= 0)
|
|
|
|
threads = 1;
|
|
|
|
|
|
|
|
try {
|
2013-08-16 16:56:27 +01:00
|
|
|
*srv = Instance.httpserver = new HTTPServer (port, threads);
|
|
|
|
*wa = Instance.wa = new WebApp (*srv, drvr, settings);
|
2013-03-22 00:20:49 +00:00
|
|
|
(*wa)->deploy();
|
|
|
|
} catch (ServerException &e)
|
|
|
|
{
|
|
|
|
cerr << "ServerException #" << e.getErrorCode() << ": " << e.getErrorMsg() << endl;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2013-08-15 22:52:36 +01:00
|
|
|
/**
|
|
|
|
* Sets current working directory to executables directory.
|
|
|
|
*/
|
|
|
|
static void _setCWD (char *argv0)
|
|
|
|
{
|
|
|
|
#ifdef WIN32
|
|
|
|
wchar_t strFileName [MAX_PATH];
|
|
|
|
DWORD r, i;
|
|
|
|
r = GetModuleFileNameW(NULL, strFileName, MAX_PATH);
|
|
|
|
for (i = r;i >= 0;i--)
|
|
|
|
{
|
|
|
|
if (strFileName[i] == '\\')
|
|
|
|
{
|
|
|
|
strFileName[i] = '\0';
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
SetCurrentDirectoryW(strFileName);
|
|
|
|
|
|
|
|
#elif defined(linux)
|
|
|
|
int len, i;
|
|
|
|
char *strFN;
|
|
|
|
if (argv0 != NULL)
|
|
|
|
{
|
|
|
|
len = strlen (argv0);
|
|
|
|
strFN = new char [len + 1];
|
|
|
|
|
|
|
|
for (i = len;i >= 0;i--)
|
|
|
|
{
|
|
|
|
if (strFN[i] == '/')
|
|
|
|
{
|
|
|
|
strFN = '\0';
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
chdir (strFN);
|
|
|
|
delete [] strFN;
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
|
|
|
}
|
|
|
|
|
2013-08-16 16:56:27 +01:00
|
|
|
/**
|
|
|
|
* Releases resources before exit.
|
|
|
|
*/
|
|
|
|
static void _doCleanup ()
|
|
|
|
{
|
|
|
|
delete Instance.wa;
|
|
|
|
delete Instance.httpserver;
|
|
|
|
delete Instance.usi;
|
|
|
|
delete Instance.settings;
|
|
|
|
delete logger;
|
|
|
|
|
|
|
|
memset (&Instance, 0, sizeof(Instance));
|
|
|
|
logger = NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
static void _signal_handler (int sig)
|
|
|
|
{
|
|
|
|
stringstream ss;
|
|
|
|
ss << "Signal " << sig << " raised. Terminating...";
|
|
|
|
logger->log(Logger::LL_INFO, ss.str());
|
|
|
|
_doCleanup();
|
|
|
|
}
|
|
|
|
|
2013-02-26 19:02:50 +00:00
|
|
|
int main(int argc, char *argv[])
|
|
|
|
{
|
2013-08-16 16:56:27 +01:00
|
|
|
Settings *settings;
|
|
|
|
UDPTracker *usi;
|
2013-02-26 19:02:50 +00:00
|
|
|
string config_file;
|
|
|
|
int r;
|
|
|
|
|
|
|
|
#ifdef WIN32
|
|
|
|
WSADATA wsadata;
|
|
|
|
WSAStartup(MAKEWORD(2, 2), &wsadata);
|
|
|
|
#endif
|
|
|
|
|
|
|
|
cout << "UDP Tracker (UDPT) " << VERSION << endl;
|
|
|
|
cout << "Copyright 2012,2013 Naim Abda <naim94a@gmail.com>\n\tReleased under the GPLv3 License." << endl;
|
|
|
|
cout << "Build Date: " << __DATE__ << endl << endl;
|
|
|
|
|
|
|
|
config_file = "udpt.conf";
|
2013-08-16 16:56:27 +01:00
|
|
|
memset(&Instance, 0, sizeof(Instance));
|
|
|
|
|
|
|
|
#ifdef SIGBREAK
|
|
|
|
signal(SIGBREAK, &_signal_handler);
|
|
|
|
#endif
|
|
|
|
#ifdef SIGTERM
|
|
|
|
signal(SIGTERM, &_signal_handler);
|
|
|
|
#endif
|
|
|
|
#ifdef SIGABRT
|
|
|
|
signal(SIGABRT, &_signal_handler);
|
|
|
|
#endif
|
|
|
|
#ifdef SIGINT
|
|
|
|
signal(SIGINT, &_signal_handler);
|
|
|
|
#endif
|
2013-02-26 19:02:50 +00:00
|
|
|
|
|
|
|
if (argc <= 1)
|
|
|
|
{
|
2013-08-15 22:52:36 +01:00
|
|
|
// set current directory when no filename is present.
|
|
|
|
_setCWD(argv[0]);
|
|
|
|
|
2013-02-26 19:02:50 +00:00
|
|
|
_print_usage ();
|
|
|
|
}
|
2013-08-10 20:12:08 +01:00
|
|
|
else if (argc >= 2)
|
|
|
|
{
|
|
|
|
config_file = argv[1]; // reported in issue #5.
|
|
|
|
}
|
2013-02-26 19:02:50 +00:00
|
|
|
|
2013-08-16 16:56:27 +01:00
|
|
|
settings = Instance.settings = new Settings (config_file);
|
2013-02-26 19:02:50 +00:00
|
|
|
|
|
|
|
if (!settings->load())
|
|
|
|
{
|
|
|
|
const char strDATABASE[] = "database";
|
|
|
|
const char strTRACKER[] = "tracker";
|
2013-03-22 00:20:49 +00:00
|
|
|
const char strAPISRV [] = "apiserver";
|
2013-02-26 19:02:50 +00:00
|
|
|
// set default settings:
|
|
|
|
|
|
|
|
settings->set (strDATABASE, "driver", "sqlite3");
|
|
|
|
settings->set (strDATABASE, "file", "tracker.db");
|
|
|
|
|
2013-03-23 19:54:25 +00:00
|
|
|
settings->set (strTRACKER, "is_dynamic", "0");
|
2013-03-22 00:20:49 +00:00
|
|
|
settings->set (strTRACKER, "port", "6969"); // UDP PORT
|
2013-02-26 19:02:50 +00:00
|
|
|
settings->set (strTRACKER, "threads", "5");
|
2013-03-23 19:54:25 +00:00
|
|
|
settings->set (strTRACKER, "allow_remotes", "1");
|
|
|
|
settings->set (strTRACKER, "allow_iana_ips", "1");
|
2013-02-26 19:02:50 +00:00
|
|
|
settings->set (strTRACKER, "announce_interval", "1800");
|
|
|
|
settings->set (strTRACKER, "cleanup_interval", "120");
|
|
|
|
|
2013-03-22 00:20:49 +00:00
|
|
|
settings->set (strAPISRV, "enable", "1");
|
|
|
|
settings->set (strAPISRV, "threads", "1");
|
|
|
|
settings->set (strAPISRV, "port", "6969"); // TCP PORT
|
|
|
|
|
2013-02-26 19:02:50 +00:00
|
|
|
settings->save();
|
2013-03-22 00:20:49 +00:00
|
|
|
cerr << "Failed to read from '" << config_file.c_str() << "'. Using default settings." << endl;
|
2013-02-26 19:02:50 +00:00
|
|
|
}
|
|
|
|
|
2013-08-16 15:06:09 +01:00
|
|
|
logger = new Logger (settings);
|
2013-08-16 16:56:27 +01:00
|
|
|
usi = Instance.usi = new UDPTracker (settings);
|
2013-02-26 19:02:50 +00:00
|
|
|
|
2013-03-10 21:18:37 +00:00
|
|
|
HTTPServer *apiSrv = NULL;
|
2013-03-21 00:44:10 +00:00
|
|
|
WebApp *wa = NULL;
|
2013-03-08 00:42:02 +00:00
|
|
|
|
2013-02-26 19:02:50 +00:00
|
|
|
r = usi->start();
|
|
|
|
if (r != UDPTracker::START_OK)
|
|
|
|
{
|
2013-03-22 00:20:49 +00:00
|
|
|
cerr << "Error While trying to start server." << endl;
|
2013-02-26 19:02:50 +00:00
|
|
|
switch (r)
|
|
|
|
{
|
|
|
|
case UDPTracker::START_ESOCKET_FAILED:
|
2013-03-22 00:20:49 +00:00
|
|
|
cerr << "Failed to create socket." << endl;
|
2013-02-26 19:02:50 +00:00
|
|
|
break;
|
|
|
|
case UDPTracker::START_EBIND_FAILED:
|
2013-03-22 00:20:49 +00:00
|
|
|
cerr << "Failed to bind socket." << endl;
|
2013-02-26 19:02:50 +00:00
|
|
|
break;
|
|
|
|
default:
|
2013-03-22 00:20:49 +00:00
|
|
|
cerr << "Unknown Error" << endl;
|
2013-02-26 19:02:50 +00:00
|
|
|
break;
|
|
|
|
}
|
|
|
|
goto cleanup;
|
|
|
|
}
|
|
|
|
|
2013-03-22 00:20:49 +00:00
|
|
|
_doAPIStart(settings, &wa, &apiSrv, usi->conn);
|
2013-03-08 00:42:02 +00:00
|
|
|
|
2013-08-16 16:56:27 +01:00
|
|
|
cout << "Hit Control-C to exit." << endl;
|
2013-02-26 19:02:50 +00:00
|
|
|
|
2013-08-16 16:56:27 +01:00
|
|
|
usi->wait();
|
2013-02-26 19:02:50 +00:00
|
|
|
|
|
|
|
cleanup:
|
|
|
|
cout << endl << "Goodbye." << endl;
|
|
|
|
|
|
|
|
#ifdef WIN32
|
|
|
|
WSACleanup();
|
|
|
|
#endif
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|