*/
#include <unistd.h>
-#include <sys/socket.h>
-#include <netinet/in.h>
-#include <arpa/inet.h>
#include <csignal>
-#include <cstdio>
-#include <cstring>
#include <iostream>
-#include <vector>
#include "stg/ia.h"
#include "stg/common.h"
-#include "web.h"
-#include "settings_impl.h"
-int mes;
-char infoText[256];
-char messageText[256];
-
-const int winKOI = 0;
-
-IA_CLIENT_PROT * clnp;
-WEB * web = NULL;
-
-using namespace std;
+#include "settings.h"
time_t stgTime;
+bool running;
//-----------------------------------------------------------------------------
//-----------------------------------------------------------------------------
//-----------------------------------------------------------------------------
void Usage()
{
-printf("sgauth <path_to_config>\n");
+std::cout << "sgauth <path_to_config>" << std::endl;
}
//-----------------------------------------------------------------------------
-void SetDirName(const vector<string> & dn, void *)
+void StatUpdate(const LOADSTAT &, void *)
{
-for (int j = 0; j < DIR_NUM; j++)
- {
- if (winKOI)
- {
- string dir;
- KOIToWin(dn[j], &dir);
- if (web)
- web->SetDirName(dir, j);
- }
- else
- {
- if (web)
- web->SetDirName(dn[j], j);
- }
- }
-}
-//-----------------------------------------------------------------------------
-void StatUpdate(const LOADSTAT & ls, void *)
-{
-if (web)
- web->UpdateStat(ls);
}
//-----------------------------------------------------------------------------
void StatusChanged(int, void *)
{
}
//-----------------------------------------------------------------------------
-void ShowMessage(const string & message, int i, int, int, void *)
-{
-if (web)
- web->AddMessage(message, i);
-}
-//-----------------------------------------------------------------------------
-void ShowError(const string & message, int, void *)
+void ShowMessage(const string &, int, int, int, void *)
{
-if (web)
- web->AddMessage(message, 0);
}
//-----------------------------------------------------------------------------
-void CatchUSR1(int)
+void ShowError(const string &, int, void *)
{
-if (clnp->GetAuthorized())
- {
- cout << "Connect" << endl;
- clnp->Connect();
- }
-}
-//-----------------------------------------------------------------------------
-void CatchUSR2(int)
-{
-cout << "Disconnect" << endl;
-clnp->Disconnect();
}
//-----------------------------------------------------------------------------
void CatchTERM(int)
{
-cout << "Terminated" << endl;
-clnp->Disconnect();
-sleep(2);
-exit(0);
+running = false;
}
//-----------------------------------------------------------------------------
static void SetSignalHandlers()
newsa.sa_flags = 0;
sigaction(SIGINT, &newsa, &oldsa);
-sigemptyset(&sigmask);
-sigaddset(&sigmask, SIGUSR1);
-newsa.sa_handler = CatchUSR1;
-newsa.sa_mask = sigmask;
-newsa.sa_flags = 0;
-sigaction(SIGUSR1, &newsa, &oldsa);
-
-sigemptyset(&sigmask);
-sigaddset(&sigmask, SIGUSR2);
-newsa.sa_handler = CatchUSR2;
-newsa.sa_mask = sigmask;
-newsa.sa_flags = 0;
-sigaction(SIGUSR2, &newsa, &oldsa);
-
return;
}
//-----------------------------------------------------------------------------
int main(int argc, char *argv[])
{
-SETTINGS_IMPL settings;
+SETTINGS settings;
if (argc == 2)
{
settings.SetConfFile(argv[1]);
}
-else
- {
- // Usage
- }
if (settings.ReadSettings())
{
- printf("ReadSettingsError\n");
- printf("%s\n", settings.GetStrError().c_str());
- exit(-1);
+ std::cerr << "Failed to read settings\n"
+ << settings.GetStrError() << std::endl;
+ Usage();
+ return -1;
}
-settings.Print();
-
-if (settings.GetDaemon())
- {
- switch (fork())
- {
- case -1:
- exit(1);
- break;
- case 0:
- setsid();
- break;
-
- default:
- exit(0);
- break;
- }
- }
-
-clnp = new IA_CLIENT_PROT(settings.GetServerName(), settings.GetServerPort(), settings.GetLocalPort());
-
-if (!settings.GetNoWeb())
- {
- web = new WEB();
- web->SetRefreshPagePeriod(settings.GetRefreshPeriod());
- web->SetListenAddr(settings.GetListenWebIP());
- web->Start();
- }
-
-clnp->SetLogin(settings.GetLogin());
-clnp->SetPassword(settings.GetPassword());
-
-clnp->SetStatusChangedCb(StatusChanged, NULL);
-clnp->SetInfoCb(ShowMessage, NULL);
-clnp->SetErrorCb(ShowError, NULL);
-clnp->SetDirNameCb(SetDirName, NULL);
-clnp->SetStatChangedCb(StatUpdate, NULL);
-clnp->SetReconnect(settings.GetReconnect());
+SetSignalHandlers();
-clnp->Start();
+IA_CLIENT_PROT proto(settings.GetServerName(), settings.GetServerPort(), settings.GetLocalPort());
-SetSignalHandlers();
+proto.SetLogin(settings.GetLogin());
+proto.SetPassword(settings.GetPassword());
-#ifdef LINUX
-for (int i = 1; i < argc; i++)
- memset(argv[i], 0, strlen(argv[i]));
+proto.SetStatusChangedCb(StatusChanged, NULL);
+proto.SetInfoCb(ShowMessage, NULL);
+proto.SetErrorCb(ShowError, NULL);
+proto.SetStatChangedCb(StatUpdate, NULL);
-if(argc > 1)
- strcpy(argv[1], "Connecting...");
-#endif
+proto.Start();
-#ifdef FREEBSD
-setproctitle("Connecting...");
-#endif
-clnp->Connect();
+proto.Connect();
-while (1)
+running = true;
+while (running)
{
usleep(200000);
+ }
- char state[20];
-
- if (clnp->GetAuthorized())
- {
- if (settings.GetShowPid())
- sprintf(state, "On %d", getpid());
- else
- strcpy(state, "Online");
- }
- else
- {
- if (settings.GetShowPid())
- sprintf(state, "Off %d", getpid());
- else
- strcpy(state, "Offline");
- }
-
- #ifdef LINUX
- for (int i = 1; i < argc; i++)
- memset(argv[i], 0, strlen(argv[i]));
- if(argc > 1)
- strcpy(argv[1], state);
- #endif
-
- #ifdef FREEBSD
- setproctitle(state);
- #endif
+proto.Disconnect();
- #ifdef FREEBSD_5
- setproctitle(state);
- #endif
- }
+proto.Stop();
return 0;
}
{
}
//-----------------------------------------------------------------------------
-int ParseYesNo(const string & value, bool * val)
+int ParseYesNo(const std::string & value, bool * val)
{
if (0 == strcasecmp(value.c_str(), "yes"))
{
return -1;
}
//-----------------------------------------------------------------------------
-int ParseInt(const string & value, int * val)
+int ParseInt(const std::string & value, int * val)
{
if (str2x<int>(value, *val))
return -1;
return 0;
}
//-----------------------------------------------------------------------------
-int ParseUnsigned(const string & value, unsigned * val)
+int ParseUnsigned(const std::string & value, unsigned * val)
{
if (str2x<unsigned>(value, *val))
return -1;
return 0;
}
//-----------------------------------------------------------------------------
-int ParseIntInRange(const string & value, int min, int max, int * val)
+int ParseIntInRange(const std::string & value, int min, int max, int * val)
{
if (ParseInt(value, val) != 0)
return -1;
return 0;
}
//-----------------------------------------------------------------------------
-int ParseUnsignedInRange(const string & value, unsigned min, unsigned max, unsigned * val)
+int ParseUnsignedInRange(const std::string & value, unsigned min, unsigned max, unsigned * val)
{
if (ParseUnsigned(value, val) != 0)
return -1;
PARAM_VALUE pv;
pv.param = childNode->getName();
int i = 0;
+ const char * value;
while ((value = childNode->getValue(i++)) != NULL)
pv.value.push_back(value);
params->push_back(pv);
"Login",
"Password",
"ServerName",
- "ServerPort"
+ "ServerPort",
NULL
};
else if (strcasecmp(node->getName(), "ServerName") == 0)
serverName = node->getValue(0);
else if (strcasecmp(node->getName(), "ServerPort") == 0)
+ {
if (ParseIntInRange(node->getValue(0), 1, 65535, &port))
{
strError = "Parameter 'ServerPort' is not valid.";
printfd(__FILE__, "SETTINGS::ReadSettings() - %s\n", strError.c_str());
return -1;
}
+ }
else if (strcasecmp(node->getName(), "LocalPort") == 0)
+ {
if (ParseIntInRange(node->getValue(0), 0, 65535, &localPort))
{
strError = "Parameter 'LocalPort' is not valid.";
printfd(__FILE__, "SETTINGS::ReadSettings() - %s\n", strError.c_str());
return -1;
}
+ }
else if (strcasecmp(node->getName(), "StoreModule") == 0)
{
if (node->getValue(1))