projects/stargazer/settings_impl.cpp

   1 /*
   2  *    This program is free software; you can redistribute it and/or modify
   3  *    it under the terms of the GNU General Public License as published by
   4  *    the Free Software Foundation; either version 2 of the License, or
   5  *    (at your option) any later version.
   6  *
   7  *    This program is distributed in the hope that it will be useful,
   8  *    but WITHOUT ANY WARRANTY; without even the implied warranty of
   9  *    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  10  *    GNU General Public License for more details.
  11  *
  12  *    You should have received a copy of the GNU General Public License
  13  *    along with this program; if not, write to the Free Software
  14  *    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  15  */
  16
  17 /*
  18  *    Date: 27.10.2002
  19  */
  20
  21 /*
  22  *    Author : Boris Mikhailenko <stg34@stargazer.dp.ua>
  23  */
  24
  25 /*
  26 $Revision: 1.45 $
  27 $Date: 2010/08/19 13:42:30 $
  28 $Author: faust $
  29 */
  30
  31 #include <cstring>
  32 #include <cerrno>
  33 #include <string>
  34
  35 #include "stg/logger.h"
  36 #include "stg/dotconfpp.h"
  37 #include "settings_impl.h"
  38
  39 //-----------------------------------------------------------------------------
  40 SETTINGS_IMPL::SETTINGS_IMPL(const std::string & cd)
  41     : modulesPath("/usr/lib/stg"),
  42       dirName(DIR_NUM),
  43       confDir(cd.empty() ? "/etc/stargazer" : cd),
  44       scriptsDir(confDir),
  45       rules(confDir + "/rules"),
  46       logFile("/var/log/stargazer.log"),
  47       pidFile("/var/run/stargazer.pid"),
  48       monitorDir("/var/stargazer/monitoring"),
  49       monitoring(false),
  50       detailStatWritePeriod(dsPeriod_1_6),
  51       statWritePeriod(10),
  52       stgExecMsgKey(5555),
  53       executersNum(1),
  54       fullFee(false),
  55       dayFee(0),
  56       dayResetTraff(0),
  57       spreadFee(false),
  58       freeMbAllowInet(false),
  59       dayFeeIsLastDay(false),
  60       stopOnError(true),
  61       writeFreeMbTraffCost(false),
  62       showFeeInCash(true),
  63       messageTimeout(0),
  64       feeChargeType(0),
  65       reconnectOnTariffChange(false),
  66       logger(GetStgLogger())
  67 {
  68 }
  69 //-----------------------------------------------------------------------------
  70 SETTINGS_IMPL::SETTINGS_IMPL(const SETTINGS_IMPL & rval)
  71     : SETTINGS(),
  72       strError(),
  73       modulesPath(rval.modulesPath),
  74       dirName(rval.dirName),
  75       confDir(rval.confDir),
  76       scriptsDir(rval.scriptsDir),
  77       rules(rval.rules),
  78       logFile(rval.logFile),
  79       pidFile(rval.pidFile),
  80       monitorDir(rval.monitorDir),
  81       monitoring(rval.monitoring),
  82       detailStatWritePeriod(rval.detailStatWritePeriod),
  83       statWritePeriod(rval.statWritePeriod),
  84       stgExecMsgKey(rval.stgExecMsgKey),
  85       executersNum(rval.executersNum),
  86       fullFee(rval.fullFee),
  87       dayFee(rval.dayFee),
  88       dayResetTraff(rval.dayResetTraff),
  89       spreadFee(rval.spreadFee),
  90       freeMbAllowInet(rval.freeMbAllowInet),
  91       stopOnError(rval.stopOnError),
  92       dayFeeIsLastDay(rval.dayFeeIsLastDay),
  93       writeFreeMbTraffCost(rval.writeFreeMbTraffCost),
  94       showFeeInCash(rval.showFeeInCash),
  95       messageTimeout(rval.messageTimeout),
  96       feeChargeType(rval.feeChargeType),
  97       reconnectOnTariffChange(rval.reconnectOnTariffChange),
  98       modulesSettings(rval.modulesSettings),
  99       storeModuleSettings(rval.storeModuleSettings),
 100       logger(GetStgLogger())
 101 {
 102 }
 103 //-----------------------------------------------------------------------------
 104 int SETTINGS_IMPL::ParseModuleSettings(const DOTCONFDocumentNode * node, std::vector<PARAM_VALUE> * params)
 105 {
 106 const DOTCONFDocumentNode * childNode;
 107 PARAM_VALUE pv;
 108 const char * value;
 109
 110 pv.param = node->getName();
 111
 112 if (node->getValue(1))
 113     {
 114     strError = "Unexpected value \'" + std::string(node->getValue(1)) + "\'.";
 115     return -1;
 116     }
 117
 118 value = node->getValue(0);
 119
 120 if (!value)
 121     {
 122     strError = "Module name expected.";
 123     return -1;
 124     }
 125
 126 childNode = node->getChildNode();
 127 while (childNode)
 128     {
 129     pv.param = childNode->getName();
 130     int i = 0;
 131     while ((value = childNode->getValue(i++)) != NULL)
 132         {
 133         pv.value.push_back(value);
 134         }
 135     params->push_back(pv);
 136     pv.value.clear();
 137     childNode = childNode->getNextNode();
 138     }
 139
 140 return 0;
 141 }
 142 //-----------------------------------------------------------------------------
 143 void SETTINGS_IMPL::ErrorCallback(void * data, const char * buf)
 144 {
 145     printfd(__FILE__, "SETTINGS_IMPL::ErrorCallback() - %s\n", buf);
 146     SETTINGS_IMPL * settings = static_cast<SETTINGS_IMPL *>(data);
 147     settings->logger("%s", buf);
 148 }
 149 //-----------------------------------------------------------------------------
 150 int SETTINGS_IMPL::ReadSettings()
 151 {
 152 const char * requiredOptions[] = {
 153     "ModulesPath",
 154     "Modules",
 155     "StoreModule",
 156     "Rules",
 157     "LogFile",
 158     "DetailStatWritePeriod",
 159     "DayFee",
 160     "DayResetTraff",
 161     "SpreadFee",
 162     "FreeMbAllowInet",
 163     "DayFeeIsLastDay",
 164     "WriteFreeMbTraffCost",
 165     NULL
 166     };
 167 int storeModulesCount = 0;
 168 modulesSettings.clear();
 169
 170 DOTCONFDocument conf(DOTCONFDocument::CASEINSENSITIVE);
 171 conf.setErrorCallback(SETTINGS_IMPL::ErrorCallback, this);
 172 conf.setRequiredOptionNames(requiredOptions);
 173 std::string confFile = confDir + "/stargazer.conf";
 174
 175 if(conf.setContent(confFile.c_str()) != 0)
 176     {
 177     strError = "Cannot read file " + confFile;
 178     return -1;
 179     }
 180
 181 const DOTCONFDocumentNode * node = conf.getFirstNode();
 182
 183 while (node)
 184     {
 185     if (strcasecmp(node->getName(), "ScriptDir") == 0)
 186         {
 187         scriptsDir = node->getValue(0);
 188         }
 189
 190     if (strcasecmp(node->getName(), "LogFile") == 0)
 191         {
 192         logFile = node->getValue(0);
 193         }
 194
 195     if (strcasecmp(node->getName(), "PIDFile") == 0)
 196         {
 197         pidFile = node->getValue(0);
 198         }
 199
 200     if (strcasecmp(node->getName(), "ModulesPath") == 0)
 201         {
 202         modulesPath = node->getValue(0);
 203         }
 204
 205     if (strcasecmp(node->getName(), "Rules") == 0)
 206         {
 207         rules = node->getValue(0);
 208         }
 209
 210     if (strcasecmp(node->getName(), "DetailStatWritePeriod") == 0)
 211         {
 212         if (ParseDetailStatWritePeriod(node->getValue(0)) != 0)
 213             {
 214             strError = "Incorrect DetailStatWritePeriod value: \'" + std::string(node->getValue(0)) + "\'";
 215             return -1;
 216             }
 217         }
 218
 219     if (strcasecmp(node->getName(), "StatWritePeriod") == 0)
 220         {
 221         if (ParseUnsignedInRange(node->getValue(0), 1, 1440, &statWritePeriod) != 0)
 222             {
 223             strError = "Incorrect StatWritePeriod value: \'" + std::string(node->getValue(0)) + "\'";
 224             return -1;
 225             }
 226         }
 227
 228     if (strcasecmp(node->getName(), "ExecMsgKey") == 0)
 229         {
 230         if (ParseInt(node->getValue(0), &stgExecMsgKey) != 0)
 231             {
 232             strError = "Incorrect ExecMsgKey value: \'" + std::string(node->getValue(0)) + "\'";
 233             return -1;
 234             }
 235         }
 236
 237     if (strcasecmp(node->getName(), "ExecutersNum") == 0)
 238         {
 239         if (ParseUnsignedInRange(node->getValue(0), 1, 1024, &executersNum) != 0)
 240             {
 241             strError = "Incorrect ExecutersNum value: \'" + std::string(node->getValue(0)) + "\'";
 242             return -1;
 243             }
 244         }
 245
 246     if (strcasecmp(node->getName(), "DayFee") == 0)
 247         {
 248         if (ParseUnsignedInRange(node->getValue(0), 0, 31, &dayFee) != 0)
 249             {
 250             strError = "Incorrect DayFee value: \'" + std::string(node->getValue(0)) + "\'";
 251             return -1;
 252             }
 253         }
 254
 255     if (strcasecmp(node->getName(), "FullFee") == 0)
 256         {
 257         if (ParseYesNo(node->getValue(0), &fullFee) != 0)
 258             {
 259             strError = "Incorrect FullFee value: \'" + std::string(node->getValue(0)) + "\'";
 260             return -1;
 261             }
 262         }
 263
 264     if (strcasecmp(node->getName(), "DayResetTraff") == 0)
 265         {
 266         if (ParseUnsignedInRange(node->getValue(0), 0, 31, &dayResetTraff) != 0)
 267             {
 268             strError = "Incorrect DayResetTraff value: \'" + std::string(node->getValue(0)) + "\'";
 269             return -1;
 270             }
 271         }
 272
 273     if (strcasecmp(node->getName(), "SpreadFee") == 0)
 274         {
 275         if (ParseYesNo(node->getValue(0), &spreadFee) != 0)
 276             {
 277             strError = "Incorrect SpreadFee value: \'" + std::string(node->getValue(0)) + "\'";
 278             return -1;
 279             }
 280         }
 281
 282     if (strcasecmp(node->getName(), "FreeMbAllowInet") == 0)
 283         {
 284         if (ParseYesNo(node->getValue(0), &freeMbAllowInet) != 0)
 285             {
 286             strError = "Incorrect FreeMbAllowInet value: \'" + std::string(node->getValue(0)) + "\'";
 287             return -1;
 288             }
 289         }
 290
 291     if (strcasecmp(node->getName(), "DayFeeIsLastDay") == 0)
 292         {
 293         if (ParseYesNo(node->getValue(0), &dayFeeIsLastDay) != 0)
 294             {
 295             strError = "Incorrect DayFeeIsLastDay value: \'" + std::string(node->getValue(0)) + "\'";
 296             return -1;
 297             }
 298         }
 299
 300     if (strcasecmp(node->getName(), "WriteFreeMbTraffCost") == 0)
 301         {
 302         if (ParseYesNo(node->getValue(0), &writeFreeMbTraffCost) != 0)
 303             {
 304             strError = "Incorrect WriteFreeMbTraffCost value: \'" + std::string(node->getValue(0)) + "\'";
 305             return -1;
 306             }
 307         }
 308
 309     if (strcasecmp(node->getName(), "ShowFeeInCash") == 0)
 310         {
 311         if (ParseYesNo(node->getValue(0), &showFeeInCash) != 0)
 312             {
 313             strError = "Incorrect ShowFeeInCash value: \'" + std::string(node->getValue(0)) + "\'";
 314             return -1;
 315             }
 316         }
 317
 318     if (strcasecmp(node->getName(), "MonitorDir") == 0)
 319         {
 320         monitorDir = node->getValue(0);
 321         struct stat stat;
 322         monitoring = false;
 323
 324         if (!lstat(monitorDir.c_str(), &stat) && S_ISDIR(stat.st_mode))
 325             {
 326             monitoring = true;
 327             }
 328         }
 329
 330     if (strcasecmp(node->getName(), "MessageTimeout") == 0)
 331         {
 332         if (ParseUnsigned(node->getValue(0), &messageTimeout) != 0)
 333             {
 334             strError = "Incorrect MessageTimeout value: \'" + std::string(node->getValue(0)) + "\'";
 335             return -1;
 336             }
 337         }
 338
 339     if (strcasecmp(node->getName(), "FeeChargeType") == 0)
 340         {
 341         if (ParseUnsignedInRange(node->getValue(0), 0, 3, &feeChargeType) != 0)
 342             {
 343             strError = "Incorrect FeeChargeType value: \'" + std::string(node->getValue(0)) + "\'";
 344             return -1;
 345             }
 346         }
 347
 348     if (strcasecmp(node->getName(), "ReconnectOnTariffChange") == 0)
 349         {
 350         if (ParseYesNo(node->getValue(0), &reconnectOnTariffChange) != 0)
 351             {
 352             strError = "Incorrect ReconnectOnTariffChange value: \'" + std::string(node->getValue(0)) + "\'";
 353             return -1;
 354             }
 355         }
 356
 357     if (strcasecmp(node->getName(), "DirNames") == 0)
 358         {
 359         const DOTCONFDocumentNode * child = node->getChildNode();
 360         if (child)
 361             {
 362             const DOTCONFDocumentNode * dirNameNode;
 363             dirName.reserve(DIR_NUM);
 364             for (int i = 0; i < DIR_NUM; i++)
 365                 {
 366                 char strDirName[12];
 367                 sprintf(strDirName, "DirName%d", i);
 368                 dirNameNode = conf.findNode(strDirName, node);
 369                 if (dirNameNode && dirNameNode->getValue(0))
 370                     {
 371                     dirName[i] = dirNameNode->getValue(0);
 372                     }
 373                 }
 374             }
 375         }
 376
 377     if (strcasecmp(node->getName(), "StoreModule") == 0)
 378         {
 379         if (node->getValue(1))
 380             {
 381             strError = "Unexpected \'" + std::string(node->getValue(1)) + "\'.";
 382             return -1;
 383             }
 384
 385         if (storeModulesCount)
 386             {
 387             strError = "Should be only one StoreModule.";
 388             return -1;
 389             }
 390         storeModulesCount++;
 391
 392         storeModuleSettings.moduleName = node->getValue(0);
 393         ParseModuleSettings(node, &storeModuleSettings.moduleParams);
 394         }
 395
 396     if (strcasecmp(node->getName(), "Modules") == 0)
 397         {
 398         if (node->getValue(0))
 399             {
 400             strError = "Unexpected \'" + std::string(node->getValue(0)) + "\'.";
 401             return -1;
 402             }
 403         const DOTCONFDocumentNode * child = node->getChildNode();
 404         while (child)
 405             {
 406             if (strcasecmp(child->getName(), "Module") != 0)
 407                 {
 408                 child = child->getNextNode();
 409                 continue;
 410                 }
 411             MODULE_SETTINGS modSettings;
 412             modSettings.moduleParams.clear();
 413             modSettings.moduleName = child->getValue();
 414
 415             ParseModuleSettings(child, &modSettings.moduleParams);
 416
 417             modulesSettings.push_back(modSettings);
 418
 419             child = child->getNextNode();
 420             }
 421         }
 422
 423     if (strcasecmp(node->getName(), "ScriptParams") == 0)
 424         {
 425         for (int i = 0; node->getValue(i) != NULL; ++i)
 426             {
 427             scriptParams.push_back(node->getValue(i));
 428             }
 429         }
 430     node = node->getNextNode();
 431     }
 432
 433 return 0;
 434 }
 435 //-----------------------------------------------------------------------------
 436 int SETTINGS_IMPL::ParseDetailStatWritePeriod(const std::string & detailStatPeriodStr)
 437 {
 438 if (detailStatPeriodStr == "1")
 439     {
 440     detailStatWritePeriod = dsPeriod_1;
 441     return 0;
 442     }
 443 else if (detailStatPeriodStr == "1/2")
 444     {
 445     detailStatWritePeriod = dsPeriod_1_2;
 446     return 0;
 447     }
 448 else if (detailStatPeriodStr == "1/4")
 449     {
 450     detailStatWritePeriod = dsPeriod_1_4;
 451     return 0;
 452     }
 453 else if (detailStatPeriodStr == "1/6")
 454     {
 455     detailStatWritePeriod = dsPeriod_1_6;
 456     return 0;
 457     }
 458
 459 return -1;
 460 }
 461 //-----------------------------------------------------------------------------