]> git.stg.codes - stg.git/blobdiff - projects/stargazer/main.cpp
git.stg.codes / stg.git / blobdiff
? search:
summary | shortlog | log | commit | commitdiff | tree
raw | inline | side by side
Stargazer (#6)
[stg.git] / projects / stargazer / main.cpp
diff --git a/projects/stargazer/main.cpp b/projects/stargazer/main.cpp
index 3db3b2669ede87b264d6c417108e2fb35342f6bc..acc826f71576374b499f0a2b5c5014859f9b2475 100644 (file)
--- a/projects/stargazer/main.cpp
+++ b/projects/stargazer/main.cpp
@@ -29,7 +29,7 @@
 #include "traffcounter_impl.h"
 #include "settings_impl.h"
 #include "pidfile.h"
 #include "traffcounter_impl.h"
 #include "settings_impl.h"
 #include "pidfile.h"
-#include "eventloop.h"
+#include "async_pool.h"
 #include "stg_timer.h"
 
 #include "stg/user.h"
 #include "stg_timer.h"
 
 #include "stg/user.h"
@@ -41,6 +41,7 @@
 
 #include <fstream>
 #include <vector>
 
 #include <fstream>
 #include <vector>
+#include <string>
 #include <set>
 #include <csignal>
 #include <cerrno>
 #include <set>
 #include <csignal>
 #include <cerrno>
@@ -55,339 +56,378 @@
 #include <sys/stat.h> // S_IRUSR
 #include <fcntl.h> // create
 
 #include <sys/stat.h> // S_IRUSR
 #include <fcntl.h> // create
 
-#ifdef DEBUG
-    #define NO_DAEMON  (1)
-#endif
-
 #define START_FILE "/._ST_ART_ED_"
 
 #define START_FILE "/._ST_ART_ED_"
 
+using STG::SettingsImpl;
+using STG::AdminsImpl;
+using STG::TraffCounterImpl;
+using STG::UsersImpl;
+using STG::TariffsImpl;
+using STG::ServicesImpl;
+using STG::CorporationsImpl;
+using STG::StoreLoader;
+
 namespace
 {
 std::set<pid_t> executers;
 
 void StartTimer();
 namespace
 {
 std::set<pid_t> executers;
 
 void StartTimer();
-int StartScriptExecuter(char * procName, int msgKey, int * msgID, SETTINGS_IMPL * settings);
-int ForkAndWait(const std::string & confDir);
+int StartScriptExecuter(char* procName, int msgKey, int* msgID);
+int ForkAndWait(const std::string& confDir);
 void KillExecuters();
 
 //-----------------------------------------------------------------------------
 void StartTimer()
 {
 void KillExecuters();
 
 //-----------------------------------------------------------------------------
 void StartTimer()
 {
-STG_LOGGER & WriteServLog = GetStgLogger();
+    auto& WriteServLog = STG::Logger::get();
 
 
-if (RunStgTimer())
-    {
-    WriteServLog("Cannot start timer. Fatal.");
-    //printfd(__FILE__, "Cannot start timer. Fatal.\n");
-    exit(1);
-    }
-else
+    if (RunStgTimer())
     {
     {
-    WriteServLog("Timer thread started successfully.");
-    //printfd(__FILE__, "Timer thread started successfully.\n");
+        WriteServLog("Cannot start timer. Fatal.");
+        exit(1);
     }
     }
+    else
+        WriteServLog("Timer thread started successfully.");
 }
 //-----------------------------------------------------------------------------
 #if defined(LINUX) || defined(DARWIN)
 }
 //-----------------------------------------------------------------------------
 #if defined(LINUX) || defined(DARWIN)
-int StartScriptExecuter(char * procName, int msgKey, int * msgID, SETTINGS_IMPL * settings)
+int StartScriptExecuter(char* procName, int msgKey, int* msgID)
 #else
 #else
-int StartScriptExecuter(char *, int msgKey, int * msgID, SETTINGS_IMPL * settings)
+int StartScriptExecuter(char*, int msgKey, int* msgID)
 #endif
 {
 #endif
 {
-STG_LOGGER & WriteServLog = GetStgLogger();
+    auto& WriteServLog = STG::Logger::get();
 
 
-if (*msgID == -11)   // If msgID == -11 - first call. Create queue
+    if (*msgID == -11)   // If msgID == -11 - first call. Create queue
     {
     {
-    for (int i = 0; i < 2; i++)
+        for (int i = 0; i < 2; i++)
         {
         {
-        *msgID = msgget(msgKey, IPC_CREAT | IPC_EXCL | 0600);
+            *msgID = msgget(msgKey, IPC_CREAT | IPC_EXCL | 0600);
 
 
-        if (*msgID == -1)
-            {
-            *msgID = msgget(msgKey, 0);
             if (*msgID == -1)
             if (*msgID == -1)
+            {
+                *msgID = msgget(msgKey, 0);
+                if (*msgID == -1)
                 {
                 {
-                WriteServLog("Message queue not created.");
-                return -1;
-                }
-            else
-                {
-                msgctl(*msgID, IPC_RMID, NULL);
+                    WriteServLog("Message queue not created.");
+                    return -1;
                 }
                 }
+                else
+                    msgctl(*msgID, IPC_RMID, NULL);
             }
             }
-        else
+            else
             {
             {
-            WriteServLog("Message queue created successfully. msgKey=%d msgID=%d", msgKey, *msgID);
-            break;
+                WriteServLog("Message queue created successfully. msgKey=%d msgID=%d", msgKey, *msgID);
+                break;
             }
         }
     }
 
             }
         }
     }
 
-pid_t pid = fork();
+    const auto pid = fork();
 
 
-switch (pid)
+    switch (pid)
     {
     {
-    case -1:
-        WriteServLog("Fork error!");
-        return -1;
+        case -1:
+            WriteServLog("Fork error!");
+            return -1;
 
 
-    case 0:
-        delete settings;
+        case 0:
 #if defined(LINUX) || defined(DARWIN)
 #if defined(LINUX) || defined(DARWIN)
-        Executer(*msgID, pid, procName);
+            Executer(*msgID, pid, procName);
 #else
 #else
-        Executer(*msgID, pid);
+            Executer(*msgID, pid);
 #endif
 #endif
-        return 1;
+            return 1;
 
 
-    default:
-        if (executers.empty()) {
+        default:
+            if (executers.empty()) {
 #if defined(LINUX) || defined(DARWIN)
 #if defined(LINUX) || defined(DARWIN)
-            Executer(*msgID, pid, NULL);
+                Executer(*msgID, pid, NULL);
 #else
 #else
-            Executer(*msgID, pid);
+                Executer(*msgID, pid);
 #endif
 #endif
-        }
-        executers.insert(pid);
+            }
+            executers.insert(pid);
     }
     }
-return 0;
+    return 0;
 }
 //-----------------------------------------------------------------------------
 }
 //-----------------------------------------------------------------------------
-#ifndef NO_DAEMON
-int ForkAndWait(const std::string & confDir)
-#else
-int ForkAndWait(const std::string &)
-#endif
+int ForkAndWait(const std::string& confDir)
 {
 {
-#ifndef NO_DAEMON
-pid_t pid = fork();
-std::string startFile = confDir + START_FILE;
-unlink(startFile.c_str());
+    const auto pid = fork();
+    const auto startFile = confDir + START_FILE;
+    unlink(startFile.c_str());
 
 
-switch (pid)
+    switch (pid)
     {
     {
-    case -1:
-        return -1;
-        break;
+        case -1:
+            return -1;
+            break;
 
 
-    case 0:
-        close(1);
-        close(2);
-        setsid();
-        break;
+        case 0:
+            close(1);
+            close(2);
+            setsid();
+            break;
 
 
-    default:
-        struct timespec ts = {0, 200000000};
-        for (int i = 0; i < 120 * 5; i++)
+        default:
+            struct timespec ts = {0, 200000000};
+            for (int i = 0; i < 120 * 5; i++)
             {
             {
-            if (access(startFile.c_str(), F_OK) == 0)
+                if (access(startFile.c_str(), F_OK) == 0)
                 {
                 {
-                unlink(startFile.c_str());
-                exit(0);
+                    unlink(startFile.c_str());
+                    exit(0);
                 }
 
                 }
 
-            nanosleep(&ts, NULL);
+                nanosleep(&ts, NULL);
             }
             }
-        unlink(startFile.c_str());
-        exit(1);
-        break;
+            unlink(startFile.c_str());
+            exit(1);
+            break;
     }
     }
-#endif
-return 0;
+    return 0;
 }
 }
+
 //-----------------------------------------------------------------------------
 void KillExecuters()
 {
 //-----------------------------------------------------------------------------
 void KillExecuters()
 {
-std::set<pid_t>::iterator pid(executers.begin());
-while (pid != executers.end())
+    auto pid = executers.begin();
+    while (pid != executers.end())
     {
     {
-    printfd(__FILE__, "KillExecuters pid=%d\n", *pid);
-    kill(*pid, SIGUSR1);
-    ++pid;
+        printfd(__FILE__, "KillExecuters pid=%d\n", *pid);
+        kill(*pid, SIGUSR1);
+        ++pid;
     }
 }
     }
 }
+
+void PrintHelp(const std::string& programName)
+{
+    std::cout << "Usage: " << programName << "[-h/--help] [-v/--version] [-f/--foreground] [<conf-dir-path>]\n"
+              << "\t --help, -h            - print this help;\n"
+              << "\t --version, -v         - print version;\n"
+              << "\t --foreground, -f      - do not go into background;\n"
+              << "\t <conf-dir-path>       - path to the directory where the configuration file is located.\n";
+}
+
+void PrintVersion(const std::string& programName)
+{
+    std::cout << programName << "\n"
+              << "Stargazer version" <<  " " << SERVER_VERSION << "\n";
+}
 //-----------------------------------------------------------------------------
 } // namespace anonymous
 //-----------------------------------------------------------------------------
 //-----------------------------------------------------------------------------
 } // namespace anonymous
 //-----------------------------------------------------------------------------
-int main(int argc, char * argv[])
+int main(int argc, char* argv[])
 {
 {
-int msgID = -11;
+    int msgID = -11;
 
 
-GetStgLogger().SetLogFileName("/var/log/stargazer.log");
+    STG::Logger::get().setFileName("/var/log/stargazer.log");
 
 
-if (getuid())
+    if (getuid())
     {
     {
-    printf("You must be root. Exit.\n");
-    return 1;
+        printf("You must be root. Exit.\n");
+        return 1;
     }
 
     }
 
-SETTINGS_IMPL settings(argc == 2 ? argv[1] : "");
+    std::string path;
+    bool noDaemon(false);
 
 
-if (settings.ReadSettings())
+    if (argc == 1)
+        path = "";
+    else
     {
     {
-    STG_LOGGER & WriteServLog = GetStgLogger();
-
-    if (settings.GetLogFileName() != "")
-        WriteServLog.SetLogFileName(settings.GetLogFileName());
-
-    WriteServLog("ReadSettings error. %s", settings.GetStrError().c_str());
-    return -1;
+        for (int i = 1; i < argc; ++i)
+        {
+            const std::string arg(argv[i]);
+            if (arg == "--help" || arg == "-h")
+            {
+                PrintHelp(argv[0]);
+                return 0;
+            }
+            if (arg == "--version" || arg == "-v")
+            {
+                PrintVersion(argv[0]);
+                return 0;
+            }
+            if (arg == "--foreground" || arg == "-f")
+                noDaemon = true;
+            else
+                path = arg;
+        }
     }
 
     }
 
-#ifndef NO_DAEMON
-std::string startFile(settings.GetConfDir() + START_FILE);
-#endif
+    SettingsImpl settings(path);
 
 
-if (ForkAndWait(settings.GetConfDir()) < 0)
+    if (settings.ReadSettings())
     {
     {
-    STG_LOGGER & WriteServLog = GetStgLogger();
-    WriteServLog("Fork error!");
-    return -1;
-    }
+        auto& WriteServLog = STG::Logger::get();
 
 
-STG_LOGGER & WriteServLog = GetStgLogger();
-WriteServLog.SetLogFileName(settings.GetLogFileName());
-WriteServLog("Stg v. %s", SERVER_VERSION);
+        if (settings.GetLogFileName() != "")
+            WriteServLog.setFileName(settings.GetLogFileName());
 
 
-for (size_t i = 0; i < settings.GetExecutersNum(); i++)
-    {
-    int ret = StartScriptExecuter(argv[0], settings.GetExecMsgKey(), &msgID, &settings);
-    if (ret < 0)
-        {
-        STG_LOGGER & WriteServLog = GetStgLogger();
-        WriteServLog("Start Script Executer error!");
+        WriteServLog("ReadSettings error. %s", settings.GetStrError().c_str());
         return -1;
         return -1;
-        }
-    if (ret == 1)
+    }
+
+    if (!noDaemon)
+    {
+        if (ForkAndWait(settings.GetConfDir()) < 0)
         {
         {
-        // Stopping child
-        return 0;
+            STG::Logger::get()("Fork error!");
+            return -1;
         }
     }
 
         }
     }
 
-PIDFile pidFile(settings.GetPIDFileName());
-
-sigset_t signalSet;
-sigfillset(&signalSet);
-pthread_sigmask(SIG_BLOCK, &signalSet, NULL);
+    auto& WriteServLog = STG::Logger::get();
+    WriteServLog.setFileName(settings.GetLogFileName());
+    WriteServLog("Stg v. %s", SERVER_VERSION);
 
 
-StartTimer();
-WaitTimer();
-if (!IsStgTimerRunning())
+    for (size_t i = 0; i < settings.GetExecutersNum(); i++)
     {
     {
-    printfd(__FILE__, "Timer thread not started in 1 sec!\n");
-    WriteServLog("Timer thread not started in 1 sec!");
-    return -1;
+        auto ret = StartScriptExecuter(argv[0], settings.GetExecMsgKey(), &msgID);
+        if (ret < 0)
+        {
+            STG::Logger::get()("Start Script Executer error!");
+            return -1;
+        }
+        if (ret == 1)
+            return 0;
     }
 
     }
 
-EVENT_LOOP & loop(EVENT_LOOP_SINGLETON::GetInstance());
+    PIDFile pidFile(settings.GetPIDFileName());
+
+    struct sigaction sa;
+    memset(&sa, 0, sizeof(sa));
+    sa.sa_handler = SIG_DFL;
+    sigaction(SIGHUP, &sa, NULL); // Apparently FreeBSD ignores SIGHUP by default when launched from rc.d at bot time.
+
+    sigset_t signalSet;
+    sigfillset(&signalSet);
+    pthread_sigmask(SIG_BLOCK, &signalSet, NULL);
 
 
-STORE_LOADER storeLoader(settings);
-if (storeLoader.Load())
+    StartTimer();
+    WaitTimer();
+    if (!IsStgTimerRunning())
     {
     {
-    printfd(__FILE__, "Storage plugin: '%s'\n", storeLoader.GetStrError().c_str());
-    WriteServLog("Storage plugin: '%s'", storeLoader.GetStrError().c_str());
-    return -1;
+        printfd(__FILE__, "Timer thread not started in 1 sec!\n");
+        WriteServLog("Timer thread not started in 1 sec!");
+        return -1;
     }
 
     }
 
-if (loop.Start())
+    STG::AsyncPoolST::start();
+
+    StoreLoader storeLoader(settings);
+    if (storeLoader.load())
     {
     {
-    printfd(__FILE__, "Event loop not started.\n");
-    WriteServLog("Event loop not started.");
-    return -1;
+        printfd(__FILE__, "Storage plugin: '%s'\n", storeLoader.GetStrError().c_str());
+        WriteServLog("Storage plugin: '%s'", storeLoader.GetStrError().c_str());
+        return -1;
     }
 
     }
 
-STORE & store(storeLoader.GetStore());
-WriteServLog("Storage plugin: %s. Loading successfull.", store.GetVersion().c_str());
+    auto& store = storeLoader.get();
+    WriteServLog("Storage plugin: %s. Loading successfull.", store.GetVersion().c_str());
 
 
-ADMINS_IMPL admins(&store);
-TARIFFS_IMPL tariffs(&store);
-SERVICES_IMPL services(&store);
-CORPORATIONS_IMPL corps(&store);
-USERS_IMPL users(&settings, &store, &tariffs, services, admins.GetSysAdmin());
-TRAFFCOUNTER_IMPL traffCnt(&users, settings.GetRulesFileName());
-traffCnt.SetMonitorDir(settings.GetMonitorDir());
+    AdminsImpl admins(store);
+    TariffsImpl tariffs(&store);
+    tariffs.ReadTariffs();
+    ServicesImpl services(&store);
+    CorporationsImpl corps(&store);
+    UsersImpl users(&settings, &store, &tariffs, services, admins.sysAdmin());
+    TraffCounterImpl traffCnt(&users, settings.GetRulesFileName());
+    traffCnt.SetMonitorDir(settings.GetMonitorDir());
 
 
-if (users.Start())
-    return -1;
+    if (users.Start())
+        return -1;
 
 
-WriteServLog("Users started successfully.");
+    WriteServLog("Users started successfully.");
 
 
-if (traffCnt.Start())
-    return -1;
+    if (traffCnt.Start())
+        return -1;
 
 
-WriteServLog("Traffcounter started successfully.");
+    WriteServLog("Traffcounter started successfully.");
 
 
-STG::PluginManager manager(settings, store, admins, tariffs, services, corps, users, traffCnt);
+    STG::PluginManager manager(settings, store, admins, tariffs, services, corps, users, traffCnt);
 
 
-srandom(static_cast<unsigned int>(stgTime));
+    srandom(static_cast<unsigned int>(stgTime));
 
 
-WriteServLog("Stg started successfully.");
-WriteServLog("+++++++++++++++++++++++++++++++++++++++++++++");
+    WriteServLog("Stg started successfully.");
+    WriteServLog("+++++++++++++++++++++++++++++++++++++++++++++");
 
 
-#ifndef NO_DAEMON
-creat(startFile.c_str(), S_IRUSR);
-#endif
+    if (!noDaemon)
+    {
+        const auto startFile = settings.GetConfDir() + START_FILE;
+        creat(startFile.c_str(), S_IRUSR);
+    }
 
 
-bool running = true;
-while (running)
+    bool running = true;
+    while (running)
     {
     {
-    sigfillset(&signalSet);
-    int sig = 0;
-    sigwait(&signalSet, &sig);
-    int status;
-    switch (sig)
+        sigfillset(&signalSet);
+        int sig = 0;
+        sigwait(&signalSet, &sig);
+        int status;
+        switch (sig)
         {
         {
-        case SIGHUP:
-            traffCnt.Reload();
-            manager.reload();
-            break;
-        case SIGTERM:
-            running = false;
-            break;
-        case SIGINT:
-            running = false;
-            break;
-        case SIGPIPE:
-            WriteServLog("Broken pipe!");
-            break;
-        case SIGCHLD:
-            executers.erase(waitpid(-1, &status, WNOHANG));
-            if (executers.empty())
+            case SIGHUP:
+            {
+                SettingsImpl newSettings(settings);
+                if (newSettings.ReadSettings())
+                    WriteServLog("ReadSettings error. %s", newSettings.GetStrError().c_str());
+                else
+                    settings = newSettings;
+                WriteServLog.setFileName(settings.GetLogFileName());
+                traffCnt.Reload();
+                manager.reload(settings);
+                break;
+            }
+            case SIGTERM:
                 running = false;
                 running = false;
-            break;
-        default:
-            WriteServLog("Ignore signal %d", sig);
-            break;
+                break;
+            case SIGINT:
+                running = false;
+                break;
+            case SIGPIPE:
+                WriteServLog("Broken pipe!");
+                break;
+            case SIGCHLD:
+                executers.erase(waitpid(-1, &status, WNOHANG));
+                if (executers.empty())
+                    running = false;
+                break;
+            default:
+                WriteServLog("Ignore signal %d", sig);
+                break;
         }
     }
 
         }
     }
 
-WriteServLog("+++++++++++++++++++++++++++++++++++++++++++++");
+    WriteServLog("+++++++++++++++++++++++++++++++++++++++++++++");
+
+    manager.stop();
 
 
-if (loop.Stop())
-    WriteServLog("Event loop not stopped.");
+    STG::AsyncPoolST::stop();
 
 
-if (!traffCnt.Stop())
-    WriteServLog("Traffcounter: Stop successfull.");
+    if (!traffCnt.Stop())
+        WriteServLog("Traffcounter: Stop successfull.");
 
 
-if (!users.Stop())
-    WriteServLog("Users: Stop successfull.");
+    if (!users.Stop())
+        WriteServLog("Users: Stop successfull.");
 
 
-sleep(1);
-int res = msgctl(msgID, IPC_RMID, NULL);
-if (res)
-    WriteServLog("Queue was not removed. id=%d", msgID);
-else
-    WriteServLog("Queue removed successfully.");
+    sleep(1);
+    int res = msgctl(msgID, IPC_RMID, NULL);
+    if (res)
+        WriteServLog("Queue was not removed. id=%d", msgID);
+    else
+        WriteServLog("Queue removed successfully.");
 
 
-KillExecuters();
+    KillExecuters();
 
 
-StopStgTimer();
-WriteServLog("StgTimer: Stop successfull.");
+    StopStgTimer();
+    WriteServLog("StgTimer: Stop successfull.");
 
 
-WriteServLog("Stg stopped successfully.");
-WriteServLog("---------------------------------------------");
+    WriteServLog("Stg stopped successfully.");
+    WriteServLog("---------------------------------------------");
 
 
-return 0;
+    return 0;
 }
 //-----------------------------------------------------------------------------
 }
 //-----------------------------------------------------------------------------
Stargazer Billing System