projects/stargazer/plugins/store/files/file_store.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  *    Author : Boris Mikhailenko <stg34@stargazer.dp.ua>
  19  */
  20
  21 /*
  22  $Revision: 1.67 $
  23  $Date: 2010/10/07 19:53:11 $
  24  $Author: faust $
  25  */
  26
  27 #ifndef _GNU_SOURCE
  28 #define _GNU_SOURCE
  29 #endif
  30
  31 #include <pwd.h>
  32 #include <grp.h>
  33 #include <sys/stat.h>
  34 #include <unistd.h>
  35 #include <sys/time.h>
  36 #include <fcntl.h>
  37 #include <dirent.h>
  38
  39 #include <cstdio>
  40 #include <ctime>
  41 #include <cerrno>
  42 #include <cstring>
  43 #include <sstream>
  44 #include <algorithm>
  45
  46 #include "stg/common.h"
  47 #include "stg/user_ips.h"
  48 #include "stg/user_conf.h"
  49 #include "stg/user_stat.h"
  50 #include "stg/const.h"
  51 #include "stg/blowfish.h"
  52 #include "stg/logger.h"
  53 #include "stg/locker.h"
  54 #include "stg/plugin_creator.h"
  55 #include "file_store.h"
  56
  57 #define DELETED_USERS_DIR   "deleted_users"
  58
  59 #define adm_enc_passwd "cjeifY8m3"
  60
  61 int GetFileList(std::vector<std::string> * fileList, const std::string & directory, mode_t mode, const std::string & ext);
  62
  63 const int pt_mega = 1024 * 1024;
  64 //-----------------------------------------------------------------------------
  65 //-----------------------------------------------------------------------------
  66 //-----------------------------------------------------------------------------
  67 namespace
  68 {
  69 PLUGIN_CREATOR<FILES_STORE> fsc;
  70 }
  71
  72 extern "C" STORE * GetStore();
  73 //-----------------------------------------------------------------------------
  74 //-----------------------------------------------------------------------------
  75 //-----------------------------------------------------------------------------
  76 STORE * GetStore()
  77 {
  78 return fsc.GetPlugin();
  79 }
  80 //-----------------------------------------------------------------------------
  81 FILES_STORE_SETTINGS::FILES_STORE_SETTINGS()
  82     : settings(NULL),
  83       errorStr(),
  84       workDir(),
  85       usersDir(),
  86       adminsDir(),
  87       tariffsDir(),
  88       statMode(0),
  89       statUID(0),
  90       statGID(0),
  91       confMode(0),
  92       confUID(0),
  93       confGID(0),
  94       userLogMode(0),
  95       userLogUID(0),
  96       userLogGID(0),
  97       removeBak(true),
  98       readBak(true)
  99 {
 100 }
 101 //-----------------------------------------------------------------------------
 102 int FILES_STORE_SETTINGS::ParseOwner(const std::vector<PARAM_VALUE> & moduleParams, const std::string & owner, uid_t * uid)
 103 {
 104 PARAM_VALUE pv;
 105 pv.param = owner;
 106 std::vector<PARAM_VALUE>::const_iterator pvi;
 107 pvi = find(moduleParams.begin(), moduleParams.end(), pv);
 108 if (pvi == moduleParams.end())
 109     {
 110     errorStr = "Parameter \'" + owner + "\' not found.";
 111     printfd(__FILE__, "%s\n", errorStr.c_str());
 112     return -1;
 113     }
 114 if (User2UID(pvi->value[0].c_str(), uid) < 0)
 115     {
 116     errorStr = "Parameter \'" + owner + "\': Unknown user \'" + pvi->value[0] + "\'";
 117     printfd(__FILE__, "%s\n", errorStr.c_str());
 118     return -1;
 119     }
 120 return 0;
 121 }
 122 //-----------------------------------------------------------------------------
 123 int FILES_STORE_SETTINGS::ParseGroup(const std::vector<PARAM_VALUE> & moduleParams, const std::string & group, gid_t * gid)
 124 {
 125 PARAM_VALUE pv;
 126 pv.param = group;
 127 std::vector<PARAM_VALUE>::const_iterator pvi;
 128 pvi = find(moduleParams.begin(), moduleParams.end(), pv);
 129 if (pvi == moduleParams.end())
 130     {
 131     errorStr = "Parameter \'" + group + "\' not found.";
 132     printfd(__FILE__, "%s\n", errorStr.c_str());
 133     return -1;
 134     }
 135 if (Group2GID(pvi->value[0].c_str(), gid) < 0)
 136     {
 137     errorStr = "Parameter \'" + group + "\': Unknown group \'" + pvi->value[0] + "\'";
 138     printfd(__FILE__, "%s\n", errorStr.c_str());
 139     return -1;
 140     }
 141 return 0;
 142 }
 143 //-----------------------------------------------------------------------------
 144 int FILES_STORE_SETTINGS::ParseYesNo(const std::string & value, bool * val)
 145 {
 146 if (0 == strcasecmp(value.c_str(), "yes"))
 147     {
 148     *val = true;
 149     return 0;
 150     }
 151 if (0 == strcasecmp(value.c_str(), "no"))
 152     {
 153     *val = false;
 154     return 0;
 155     }
 156
 157 errorStr = "Incorrect value \'" + value + "\'.";
 158 return -1;
 159 }
 160 //-----------------------------------------------------------------------------
 161 int FILES_STORE_SETTINGS::ParseMode(const std::vector<PARAM_VALUE> & moduleParams, const std::string & modeStr, mode_t * mode)
 162 {
 163 PARAM_VALUE pv;
 164 pv.param = modeStr;
 165 std::vector<PARAM_VALUE>::const_iterator pvi;
 166 pvi = find(moduleParams.begin(), moduleParams.end(), pv);
 167 if (pvi == moduleParams.end())
 168     {
 169     errorStr = "Parameter \'" + modeStr + "\' not found.";
 170     printfd(__FILE__, "%s\n", errorStr.c_str());
 171     return -1;
 172     }
 173 if (Str2Mode(pvi->value[0].c_str(), mode) < 0)
 174     {
 175     errorStr = "Parameter \'" + modeStr + "\': Incorrect mode \'" + pvi->value[0] + "\'";
 176     printfd(__FILE__, "%s\n", errorStr.c_str());
 177     return -1;
 178     }
 179 return 0;
 180 }
 181 //-----------------------------------------------------------------------------
 182 int FILES_STORE_SETTINGS::ParseSettings(const MODULE_SETTINGS & s)
 183 {
 184 if (ParseOwner(s.moduleParams, "StatOwner", &statUID) < 0)
 185     return -1;
 186 if (ParseGroup(s.moduleParams, "StatGroup", &statGID) < 0)
 187     return -1;
 188 if (ParseMode(s.moduleParams, "StatMode", &statMode) < 0)
 189     return -1;
 190
 191 if (ParseOwner(s.moduleParams, "ConfOwner", &confUID) < 0)
 192     return -1;
 193 if (ParseGroup(s.moduleParams, "ConfGroup", &confGID) < 0)
 194     return -1;
 195 if (ParseMode(s.moduleParams, "ConfMode", &confMode) < 0)
 196     return -1;
 197
 198 if (ParseOwner(s.moduleParams, "UserLogOwner", &userLogUID) < 0)
 199     return -1;
 200 if (ParseGroup(s.moduleParams, "UserLogGroup", &userLogGID) < 0)
 201     return -1;
 202 if (ParseMode(s.moduleParams, "UserLogMode", &userLogMode) < 0)
 203     return -1;
 204
 205 std::vector<PARAM_VALUE>::const_iterator pvi;
 206 PARAM_VALUE pv;
 207 pv.param = "RemoveBak";
 208 pvi = find(s.moduleParams.begin(), s.moduleParams.end(), pv);
 209 if (pvi == s.moduleParams.end())
 210     {
 211     removeBak = true;
 212     }
 213 else
 214     {
 215     if (ParseYesNo(pvi->value[0], &removeBak))
 216         {
 217         printfd(__FILE__, "Cannot parse parameter 'RemoveBak'\n");
 218         return -1;
 219         }
 220     }
 221
 222 pv.param = "ReadBak";
 223 pvi = find(s.moduleParams.begin(), s.moduleParams.end(), pv);
 224 if (pvi == s.moduleParams.end())
 225     {
 226     readBak = false;
 227     }
 228 else
 229     {
 230     if (ParseYesNo(pvi->value[0], &readBak))
 231         {
 232         printfd(__FILE__, "Cannot parse parameter 'ReadBak'\n");
 233         return -1;
 234         }
 235     }
 236
 237 pv.param = "WorkDir";
 238 pvi = find(s.moduleParams.begin(), s.moduleParams.end(), pv);
 239 if (pvi == s.moduleParams.end())
 240     {
 241     errorStr = "Parameter \'WorkDir\' not found.";
 242     printfd(__FILE__, "Parameter 'WorkDir' not found\n");
 243     return -1;
 244     }
 245
 246 workDir = pvi->value[0];
 247 if (workDir.size() && workDir[workDir.size() - 1] == '/')
 248     {
 249     workDir.resize(workDir.size() - 1);
 250     }
 251 usersDir = workDir + "/users/";
 252 tariffsDir = workDir + "/tariffs/";
 253 adminsDir = workDir + "/admins/";
 254
 255 return 0;
 256 }
 257 //-----------------------------------------------------------------------------
 258 const std::string & FILES_STORE_SETTINGS::GetStrError() const
 259 {
 260 return errorStr;
 261 }
 262 //-----------------------------------------------------------------------------
 263 int FILES_STORE_SETTINGS::User2UID(const char * user, uid_t * uid)
 264 {
 265 struct passwd * pw;
 266 pw = getpwnam(user);
 267 if (!pw)
 268     {
 269     errorStr = std::string("User \'") + std::string(user) + std::string("\' not found in system.");
 270     printfd(__FILE__, "%s\n", errorStr.c_str());
 271     return -1;
 272     }
 273
 274 *uid = pw->pw_uid;
 275 return 0;
 276 }
 277 //-----------------------------------------------------------------------------
 278 int FILES_STORE_SETTINGS::Group2GID(const char * gr, gid_t * gid)
 279 {
 280 struct group * grp;
 281 grp = getgrnam(gr);
 282 if (!grp)
 283     {
 284     errorStr = std::string("Group \'") + std::string(gr) + std::string("\' not found in system.");
 285     printfd(__FILE__, "%s\n", errorStr.c_str());
 286     return -1;
 287     }
 288
 289 *gid = grp->gr_gid;
 290 return 0;
 291 }
 292 //-----------------------------------------------------------------------------
 293 int FILES_STORE_SETTINGS::Str2Mode(const char * str, mode_t * mode)
 294 {
 295 char a;
 296 char b;
 297 char c;
 298 if (strlen(str) > 3)
 299     {
 300     errorStr = std::string("Error parsing mode \'") + str + std::string("\'");
 301     printfd(__FILE__, "%s\n", errorStr.c_str());
 302     return -1;
 303     }
 304
 305 for (int i = 0; i < 3; i++)
 306     if (str[i] > '7' || str[i] < '0')
 307         {
 308         errorStr = std::string("Error parsing mode \'") + str + std::string("\'");
 309         printfd(__FILE__, "%s\n", errorStr.c_str());
 310         return -1;
 311         }
 312
 313 a = str[0] - '0';
 314 b = str[1] - '0';
 315 c = str[2] - '0';
 316
 317 *mode = ((mode_t)c) + ((mode_t)b << 3) + ((mode_t)a << 6);
 318
 319 return 0;
 320 }
 321 //-----------------------------------------------------------------------------
 322 mode_t FILES_STORE_SETTINGS::GetStatModeDir() const
 323 {
 324 mode_t mode = statMode;
 325 if (statMode & S_IRUSR) mode |= S_IXUSR;
 326 if (statMode & S_IRGRP) mode |= S_IXGRP;
 327 if (statMode & S_IROTH) mode |= S_IXOTH;
 328 return mode;
 329 }
 330 //-----------------------------------------------------------------------------
 331 mode_t FILES_STORE_SETTINGS::GetConfModeDir() const
 332 {
 333 mode_t mode = confMode;
 334 if (confMode & S_IRUSR) mode |= S_IXUSR;
 335 if (confMode & S_IRGRP) mode |= S_IXGRP;
 336 if (confMode & S_IROTH) mode |= S_IXOTH;
 337 return mode;
 338 }
 339 //-----------------------------------------------------------------------------
 340 //-----------------------------------------------------------------------------
 341 //-----------------------------------------------------------------------------
 342 FILES_STORE::FILES_STORE()
 343     : errorStr(),
 344       version("file_store v.1.04"),
 345       storeSettings(),
 346       settings(),
 347       mutex(),
 348       logger(GetPluginLogger(GetStgLogger(), "store_files"))
 349 {
 350 pthread_mutexattr_t attr;
 351 pthread_mutexattr_init(&attr);
 352 pthread_mutexattr_settype(&attr, PTHREAD_MUTEX_RECURSIVE);
 353 pthread_mutex_init(&mutex, &attr);
 354 }
 355 //-----------------------------------------------------------------------------
 356 int FILES_STORE::ParseSettings()
 357 {
 358 int ret = storeSettings.ParseSettings(settings);
 359 if (ret)
 360     {
 361     STG_LOCKER lock(&mutex);
 362     errorStr = storeSettings.GetStrError();
 363     }
 364 return ret;
 365 }
 366 //-----------------------------------------------------------------------------
 367 int FILES_STORE::GetUsersList(std::vector<std::string> * userList) const
 368 {
 369 std::vector<std::string> files;
 370
 371 if (GetFileList(&files, storeSettings.GetUsersDir(), S_IFDIR, ""))
 372     {
 373     STG_LOCKER lock(&mutex);
 374     errorStr = "Failed to open '" + storeSettings.GetUsersDir() + "': " + std::string(strerror(errno));
 375     return -1;
 376     }
 377
 378 STG_LOCKER lock(&mutex);
 379
 380 userList->swap(files);
 381
 382 return 0;
 383 }
 384 //-----------------------------------------------------------------------------
 385 int FILES_STORE::GetAdminsList(std::vector<std::string> * adminList) const
 386 {
 387 std::vector<std::string> files;
 388
 389 if (GetFileList(&files, storeSettings.GetAdminsDir(), S_IFREG, ".adm"))
 390     {
 391     STG_LOCKER lock(&mutex);
 392     errorStr = "Failed to open '" + storeSettings.GetAdminsDir() + "': " + std::string(strerror(errno));
 393     return -1;
 394     }
 395
 396 STG_LOCKER lock(&mutex);
 397
 398 adminList->swap(files);
 399
 400 return 0;
 401 }
 402 //-----------------------------------------------------------------------------
 403 int FILES_STORE::GetTariffsList(std::vector<std::string> * tariffList) const
 404 {
 405 std::vector<std::string> files;
 406
 407 if (GetFileList(&files, storeSettings.GetTariffsDir(), S_IFREG, ".tf"))
 408     {
 409     STG_LOCKER lock(&mutex);
 410     errorStr = "Failed to open '" + storeSettings.GetTariffsDir() + "': " + std::string(strerror(errno));
 411     return -1;
 412     }
 413
 414 STG_LOCKER lock(&mutex);
 415
 416 tariffList->swap(files);
 417
 418 return 0;
 419 }
 420 //-----------------------------------------------------------------------------
 421 int FILES_STORE::RemoveDir(const char * path) const
 422 {
 423 DIR * d = opendir(path);
 424
 425 if (!d)
 426     {
 427     errorStr = "failed to open dir. Message: '";
 428     errorStr += strerror(errno);
 429     errorStr += "'";
 430     printfd(__FILE__, "FILE_STORE::RemoveDir() - Failed to open dir '%s': '%s'\n", path, strerror(errno));
 431     return -1;
 432     }
 433
 434 dirent * entry;
 435 while ((entry = readdir(d)))
 436     {
 437     if (!(strcmp(entry->d_name, ".") && strcmp(entry->d_name, "..")))
 438         continue;
 439
 440     std::string str = path;
 441     str += "/" + std::string(entry->d_name);
 442
 443     struct stat st;
 444     if (stat(str.c_str(), &st))
 445         continue;
 446
 447     if ((st.st_mode & S_IFREG))
 448         {
 449         if (unlink(str.c_str()))
 450             {
 451             STG_LOCKER lock(&mutex);
 452             errorStr = "unlink failed. Message: '";
 453             errorStr += strerror(errno);
 454             errorStr += "'";
 455             printfd(__FILE__, "FILES_STORE::RemoveDir() - unlink failed. Message: '%s'\n", strerror(errno));
 456             closedir(d);
 457             return -1;
 458             }
 459         }
 460
 461     if (!(st.st_mode & S_IFDIR))
 462         {
 463         if (RemoveDir(str.c_str()))
 464             {
 465             closedir(d);
 466             return -1;
 467             }
 468
 469         }
 470     }
 471
 472 closedir(d);
 473
 474 if (rmdir(path))
 475     {
 476     STG_LOCKER lock(&mutex);
 477     errorStr = "rmdir failed. Message: '";
 478     errorStr += strerror(errno);
 479     errorStr += "'";
 480     printfd(__FILE__, "FILES_STORE::RemoveDir() - rmdir failed. Message: '%s'\n", strerror(errno));
 481     return -1;
 482     }
 483
 484 return 0;
 485 }
 486 //-----------------------------------------------------------------------------
 487 int FILES_STORE::AddUser(const std::string & login) const
 488 {
 489 std::string fileName;
 490
 491 strprintf(&fileName, "%s%s", storeSettings.GetUsersDir().c_str(), login.c_str());
 492
 493 if (mkdir(fileName.c_str(), S_IRWXU | S_IRGRP | S_IXGRP | S_IROTH | S_IXOTH) == -1)
 494     {
 495     STG_LOCKER lock(&mutex);
 496     errorStr = std::string("mkdir failed. Message: '") + strerror(errno) + "'";
 497     printfd(__FILE__, "FILES_STORE::AddUser - mkdir failed. Message: '%s'\n", strerror(errno));
 498     return -1;
 499     }
 500
 501 strprintf(&fileName, "%s%s/conf", storeSettings.GetUsersDir().c_str(), login.c_str());
 502 if (Touch(fileName))
 503     {
 504     STG_LOCKER lock(&mutex);
 505     errorStr = "Cannot create file \"" + fileName + "\'";
 506     printfd(__FILE__, "FILES_STORE::AddUser - fopen failed. Message: '%s'\n", strerror(errno));
 507     return -1;
 508     }
 509
 510 strprintf(&fileName, "%s%s/stat", storeSettings.GetUsersDir().c_str(), login.c_str());
 511 if (Touch(fileName))
 512     {
 513     STG_LOCKER lock(&mutex);
 514     errorStr = "Cannot create file \"" + fileName + "\'";
 515     printfd(__FILE__, "FILES_STORE::AddUser - fopen failed. Message: '%s'\n", strerror(errno));
 516     return -1;
 517     }
 518 return 0;
 519 }
 520 //-----------------------------------------------------------------------------
 521 int FILES_STORE::DelUser(const std::string & login) const
 522 {
 523 std::string dirName;
 524 std::string dirName1;
 525
 526 strprintf(&dirName, "%s/%s", storeSettings.GetWorkDir().c_str(), DELETED_USERS_DIR);
 527 if (access(dirName.c_str(), F_OK) != 0)
 528     {
 529     if (mkdir(dirName.c_str(), 0700) != 0)
 530         {
 531         STG_LOCKER lock(&mutex);
 532         errorStr = "Directory '" + dirName + "' cannot be created.";
 533         printfd(__FILE__, "FILES_STORE::DelUser - mkdir failed. Message: '%s'\n", strerror(errno));
 534         return -1;
 535         }
 536     }
 537
 538 if (access(dirName.c_str(), F_OK) == 0)
 539     {
 540     strprintf(&dirName, "%s/%s/%s.%lu", storeSettings.GetWorkDir().c_str(), DELETED_USERS_DIR, login.c_str(), time(NULL));
 541     strprintf(&dirName1, "%s/%s", storeSettings.GetUsersDir().c_str(), login.c_str());
 542     if (rename(dirName1.c_str(), dirName.c_str()))
 543         {
 544         STG_LOCKER lock(&mutex);
 545         errorStr = "Error moving dir from " + dirName1 + " to " + dirName;
 546         printfd(__FILE__, "FILES_STORE::DelUser - rename failed. Message: '%s'\n", strerror(errno));
 547         return -1;
 548         }
 549     }
 550 else
 551     {
 552     strprintf(&dirName, "%s/%s", storeSettings.GetUsersDir().c_str(), login.c_str());
 553     if (RemoveDir(dirName.c_str()))
 554         {
 555         return -1;
 556         }
 557     }
 558 return 0;
 559 }
 560 //-----------------------------------------------------------------------------
 561 int FILES_STORE::RestoreUserConf(USER_CONF * conf, const std::string & login) const
 562 {
 563 std::string fileName;
 564 fileName = storeSettings.GetUsersDir() + "/" + login + "/conf";
 565 if (RestoreUserConf(conf, login, fileName))
 566     {
 567     if (!storeSettings.GetReadBak())
 568         {
 569         return -1;
 570         }
 571     return RestoreUserConf(conf, login, fileName + ".bak");
 572     }
 573 return 0;
 574 }
 575 //-----------------------------------------------------------------------------
 576 int FILES_STORE::RestoreUserConf(USER_CONF * conf, const std::string & login, const std::string & fileName) const
 577 {
 578 CONFIGFILE cf(fileName);
 579 int e = cf.Error();
 580
 581 if (e)
 582     {
 583     STG_LOCKER lock(&mutex);
 584     errorStr = "User \'" + login + "\' data not read.";
 585     printfd(__FILE__, "FILES_STORE::RestoreUserConf - conf read failed for user '%s'\n", login.c_str());
 586     return -1;
 587     }
 588
 589 if (cf.ReadString("Password", &conf->password, "") < 0)
 590     {
 591     STG_LOCKER lock(&mutex);
 592     errorStr = "User \'" + login + "\' data not read. Parameter Password.";
 593     printfd(__FILE__, "FILES_STORE::RestoreUserConf - password read failed for user '%s'\n", login.c_str());
 594     return -1;
 595     }
 596 if (conf->password.empty())
 597     {
 598     STG_LOCKER lock(&mutex);
 599     errorStr = "User \'" + login + "\' password is blank.";
 600     printfd(__FILE__, "FILES_STORE::RestoreUserConf - password is blank for user '%s'\n", login.c_str());
 601     return -1;
 602     }
 603
 604 if (cf.ReadString("tariff", &conf->tariffName, "") < 0)
 605     {
 606     STG_LOCKER lock(&mutex);
 607     errorStr = "User \'" + login + "\' data not read. Parameter Tariff.";
 608     printfd(__FILE__, "FILES_STORE::RestoreUserConf - tariff read failed for user '%s'\n", login.c_str());
 609     return -1;
 610     }
 611 if (conf->tariffName.empty())
 612     {
 613     STG_LOCKER lock(&mutex);
 614     errorStr = "User \'" + login + "\' tariff is blank.";
 615     printfd(__FILE__, "FILES_STORE::RestoreUserConf - tariff is blank for user '%s'\n", login.c_str());
 616     return -1;
 617     }
 618
 619 std::string ipStr;
 620 cf.ReadString("IP", &ipStr, "?");
 621 USER_IPS ips;
 622 try
 623     {
 624     ips = StrToIPS(ipStr);
 625     }
 626 catch (const std::string & s)
 627     {
 628     STG_LOCKER lock(&mutex);
 629     errorStr = "User \'" + login + "\' data not read. Parameter IP address. " + s;
 630     printfd(__FILE__, "FILES_STORE::RestoreUserConf - ip read failed for user '%s'\n", login.c_str());
 631     return -1;
 632     }
 633 conf->ips = ips;
 634
 635 if (cf.ReadInt("alwaysOnline", &conf->alwaysOnline, 0) != 0)
 636     {
 637     STG_LOCKER lock(&mutex);
 638     errorStr = "User \'" + login + "\' data not read. Parameter AlwaysOnline.";
 639     printfd(__FILE__, "FILES_STORE::RestoreUserConf - alwaysonline read failed for user '%s'\n", login.c_str());
 640     return -1;
 641     }
 642
 643 if (cf.ReadInt("down", &conf->disabled, 0) != 0)
 644     {
 645     STG_LOCKER lock(&mutex);
 646     errorStr = "User \'" + login + "\' data not read. Parameter Down.";
 647     printfd(__FILE__, "FILES_STORE::RestoreUserConf - down read failed for user '%s'\n", login.c_str());
 648     return -1;
 649     }
 650
 651 if (cf.ReadInt("passive", &conf->passive, 0) != 0)
 652     {
 653     STG_LOCKER lock(&mutex);
 654     errorStr = "User \'" + login + "\' data not read. Parameter Passive.";
 655     printfd(__FILE__, "FILES_STORE::RestoreUserConf - passive read failed for user '%s'\n", login.c_str());
 656     return -1;
 657     }
 658
 659 cf.ReadInt("DisabledDetailStat", &conf->disabledDetailStat, 0);
 660 cf.ReadTime("CreditExpire", &conf->creditExpire, 0);
 661 cf.ReadString("TariffChange", &conf->nextTariff, "");
 662 cf.ReadString("Group", &conf->group, "");
 663 cf.ReadString("RealName", &conf->realName, "");
 664 cf.ReadString("Address", &conf->address, "");
 665 cf.ReadString("Phone", &conf->phone, "");
 666 cf.ReadString("Note", &conf->note, "");
 667 cf.ReadString("email", &conf->email, "");
 668
 669 char userdataName[12];
 670 for (int i = 0; i < USERDATA_NUM; i++)
 671     {
 672     snprintf(userdataName, 12, "Userdata%d", i);
 673     cf.ReadString(userdataName, &conf->userdata[i], "");
 674     }
 675
 676 if (cf.ReadDouble("Credit", &conf->credit, 0) != 0)
 677     {
 678     STG_LOCKER lock(&mutex);
 679     errorStr = "User \'" + login + "\' data not read. Parameter Credit.";
 680     printfd(__FILE__, "FILES_STORE::RestoreUserConf - credit read failed for user '%s'\n", login.c_str());
 681     return -1;
 682     }
 683
 684 return 0;
 685 }
 686 //-----------------------------------------------------------------------------
 687 int FILES_STORE::RestoreUserStat(USER_STAT * stat, const std::string & login) const
 688 {
 689 std::string fileName;
 690 fileName = storeSettings.GetUsersDir() + "/" + login + "/stat";
 691
 692 if (RestoreUserStat(stat, login, fileName))
 693     {
 694     if (!storeSettings.GetReadBak())
 695         {
 696         return -1;
 697         }
 698     return RestoreUserStat(stat, login, fileName + ".bak");
 699     }
 700 return 0;
 701 }
 702 //-----------------------------------------------------------------------------
 703 int FILES_STORE::RestoreUserStat(USER_STAT * stat, const std::string & login, const std::string & fileName) const
 704 {
 705 CONFIGFILE cf(fileName);
 706
 707 int e = cf.Error();
 708
 709 if (e)
 710     {
 711     STG_LOCKER lock(&mutex);
 712     errorStr = "User \'" + login + "\' stat not read. Cannot open file " + fileName + ".";
 713     printfd(__FILE__, "FILES_STORE::RestoreUserStat - stat read failed for user '%s'\n", login.c_str());
 714     return -1;
 715     }
 716
 717 char s[22];
 718
 719 for (int i = 0; i < DIR_NUM; i++)
 720     {
 721     uint64_t traff;
 722     snprintf(s, 22, "D%d", i);
 723     if (cf.ReadULongLongInt(s, &traff, 0) != 0)
 724         {
 725         STG_LOCKER lock(&mutex);
 726         errorStr = "User \'" + login + "\' stat not read. Parameter " + std::string(s);
 727         printfd(__FILE__, "FILES_STORE::RestoreUserStat - download stat read failed for user '%s'\n", login.c_str());
 728         return -1;
 729         }
 730     stat->monthDown[i] = traff;
 731
 732     snprintf(s, 22, "U%d", i);
 733     if (cf.ReadULongLongInt(s, &traff, 0) != 0)
 734         {
 735         STG_LOCKER lock(&mutex);
 736         errorStr =   "User \'" + login + "\' stat not read. Parameter " + std::string(s);
 737         printfd(__FILE__, "FILES_STORE::RestoreUserStat - upload stat read failed for user '%s'\n", login.c_str());
 738         return -1;
 739         }
 740     stat->monthUp[i] = traff;
 741     }
 742
 743 if (cf.ReadDouble("Cash", &stat->cash, 0) != 0)
 744     {
 745     STG_LOCKER lock(&mutex);
 746     errorStr =   "User \'" + login + "\' stat not read. Parameter Cash";
 747     printfd(__FILE__, "FILES_STORE::RestoreUserStat - cash read failed for user '%s'\n", login.c_str());
 748     return -1;
 749     }
 750
 751 if (cf.ReadDouble("FreeMb", &stat->freeMb, 0) != 0)
 752     {
 753     STG_LOCKER lock(&mutex);
 754     errorStr =   "User \'" + login + "\' stat not read. Parameter FreeMb";
 755     printfd(__FILE__, "FILES_STORE::RestoreUserStat - freemb read failed for user '%s'\n", login.c_str());
 756     return -1;
 757     }
 758
 759 if (cf.ReadTime("LastCashAddTime", &stat->lastCashAddTime, 0) != 0)
 760     {
 761     STG_LOCKER lock(&mutex);
 762     errorStr =   "User \'" + login + "\' stat not read. Parameter LastCashAddTime";
 763     printfd(__FILE__, "FILES_STORE::RestoreUserStat - lastcashaddtime read failed for user '%s'\n", login.c_str());
 764     return -1;
 765     }
 766
 767 if (cf.ReadTime("PassiveTime", &stat->passiveTime, 0) != 0)
 768     {
 769     STG_LOCKER lock(&mutex);
 770     errorStr =   "User \'" + login + "\' stat not read. Parameter PassiveTime";
 771     printfd(__FILE__, "FILES_STORE::RestoreUserStat - passivetime read failed for user '%s'\n", login.c_str());
 772     return -1;
 773     }
 774
 775 if (cf.ReadDouble("LastCashAdd", &stat->lastCashAdd, 0) != 0)
 776     {
 777     STG_LOCKER lock(&mutex);
 778     errorStr =   "User \'" + login + "\' stat not read. Parameter LastCashAdd";
 779     printfd(__FILE__, "FILES_STORE::RestoreUserStat - lastcashadd read failed for user '%s'\n", login.c_str());
 780     return -1;
 781     }
 782
 783 if (cf.ReadTime("LastActivityTime", &stat->lastActivityTime, 0) != 0)
 784     {
 785     STG_LOCKER lock(&mutex);
 786     errorStr =   "User \'" + login + "\' stat not read. Parameter LastActivityTime";
 787     printfd(__FILE__, "FILES_STORE::RestoreUserStat - lastactivitytime read failed for user '%s'\n", login.c_str());
 788     return -1;
 789     }
 790
 791 return 0;
 792 }
 793 //-----------------------------------------------------------------------------
 794 int FILES_STORE::SaveUserConf(const USER_CONF & conf, const std::string & login) const
 795 {
 796 std::string fileName;
 797 fileName = storeSettings.GetUsersDir() + "/" + login + "/conf";
 798
 799 CONFIGFILE cfstat(fileName, true);
 800
 801 int e = cfstat.Error();
 802
 803 if (e)
 804     {
 805     STG_LOCKER lock(&mutex);
 806     errorStr = std::string("User \'") + login + "\' conf not written\n";
 807     printfd(__FILE__, "FILES_STORE::SaveUserConf - conf write failed for user '%s'\n", login.c_str());
 808     return -1;
 809     }
 810
 811 e = chmod(fileName.c_str(), storeSettings.GetConfMode());
 812 e += chown(fileName.c_str(), storeSettings.GetConfUID(), storeSettings.GetConfGID());
 813
 814 if (e)
 815     {
 816     STG_LOCKER lock(&mutex);
 817     printfd(__FILE__, "FILES_STORE::SaveUserConf - chmod/chown failed for user '%s'. Error: '%s'\n", login.c_str(), strerror(errno));
 818     }
 819
 820 cfstat.WriteString("Password",     conf.password);
 821 cfstat.WriteInt   ("Passive",      conf.passive);
 822 cfstat.WriteInt   ("Down",         conf.disabled);
 823 cfstat.WriteInt("DisabledDetailStat", conf.disabledDetailStat);
 824 cfstat.WriteInt   ("AlwaysOnline", conf.alwaysOnline);
 825 cfstat.WriteString("Tariff",       conf.tariffName);
 826 cfstat.WriteString("Address",      conf.address);
 827 cfstat.WriteString("Phone",        conf.phone);
 828 cfstat.WriteString("Email",        conf.email);
 829 cfstat.WriteString("Note",         conf.note);
 830 cfstat.WriteString("RealName",     conf.realName);
 831 cfstat.WriteString("Group",        conf.group);
 832 cfstat.WriteDouble("Credit",       conf.credit);
 833 cfstat.WriteString("TariffChange", conf.nextTariff);
 834
 835 char userdataName[12];
 836 for (int i = 0; i < USERDATA_NUM; i++)
 837     {
 838     snprintf(userdataName, 12, "Userdata%d", i);
 839     cfstat.WriteString(userdataName, conf.userdata[i]);
 840     }
 841 cfstat.WriteInt("CreditExpire",    conf.creditExpire);
 842
 843 std::ostringstream ipStr;
 844 ipStr << conf.ips;
 845 cfstat.WriteString("IP", ipStr.str());
 846
 847 return 0;
 848 }
 849 //-----------------------------------------------------------------------------
 850 int FILES_STORE::SaveUserStat(const USER_STAT & stat, const std::string & login) const
 851 {
 852 std::string fileName;
 853 fileName = storeSettings.GetUsersDir() + "/" + login + "/stat";
 854
 855     {
 856     CONFIGFILE cfstat(fileName, true);
 857     int e = cfstat.Error();
 858
 859     if (e)
 860         {
 861         STG_LOCKER lock(&mutex);
 862         errorStr = std::string("User \'") + login + "\' stat not written\n";
 863         printfd(__FILE__, "FILES_STORE::SaveUserStat - stat write failed for user '%s'\n", login.c_str());
 864         return -1;
 865         }
 866
 867     for (int i = 0; i < DIR_NUM; i++)
 868         {
 869         char s[22];
 870         snprintf(s, 22, "D%d", i);
 871         cfstat.WriteInt(s, stat.monthDown[i]);
 872         snprintf(s, 22, "U%d", i);
 873         cfstat.WriteInt(s, stat.monthUp[i]);
 874         }
 875
 876     cfstat.WriteDouble("Cash", stat.cash);
 877     cfstat.WriteDouble("FreeMb", stat.freeMb);
 878     cfstat.WriteDouble("LastCashAdd", stat.lastCashAdd);
 879     cfstat.WriteInt("LastCashAddTime", stat.lastCashAddTime);
 880     cfstat.WriteInt("PassiveTime", stat.passiveTime);
 881     cfstat.WriteInt("LastActivityTime", stat.lastActivityTime);
 882     }
 883
 884 int e = chmod(fileName.c_str(), storeSettings.GetStatMode());
 885 e += chown(fileName.c_str(), storeSettings.GetStatUID(), storeSettings.GetStatGID());
 886
 887 if (e)
 888     {
 889     STG_LOCKER lock(&mutex);
 890     printfd(__FILE__, "FILES_STORE::SaveUserStat - chmod/chown failed for user '%s'. Error: '%s'\n", login.c_str(), strerror(errno));
 891     }
 892
 893 return 0;
 894 }
 895 //-----------------------------------------------------------------------------
 896 int FILES_STORE::WriteLogString(const std::string & str, const std::string & login) const
 897 {
 898 FILE * f;
 899 time_t tm = time(NULL);
 900 std::string fileName;
 901 fileName = storeSettings.GetUsersDir() + "/" + login + "/log";
 902 f = fopen(fileName.c_str(), "at");
 903
 904 if (f)
 905     {
 906     fprintf(f, "%s", LogDate(tm));
 907     fprintf(f, " -- ");
 908     fprintf(f, "%s", str.c_str());
 909     fprintf(f, "\n");
 910     fclose(f);
 911     }
 912 else
 913     {
 914     STG_LOCKER lock(&mutex);
 915     errorStr = "Cannot open \'" + fileName + "\'";
 916     printfd(__FILE__, "FILES_STORE::WriteLogString - log write failed for user '%s'\n", login.c_str());
 917     return -1;
 918     }
 919
 920 int e = chmod(fileName.c_str(), storeSettings.GetLogMode());
 921 e += chown(fileName.c_str(), storeSettings.GetLogUID(), storeSettings.GetLogGID());
 922
 923 if (e)
 924     {
 925     STG_LOCKER lock(&mutex);
 926     printfd(__FILE__, "FILES_STORE::WriteLogString - chmod/chown failed for user '%s'. Error: '%s'\n", login.c_str(), strerror(errno));
 927     }
 928
 929 return 0;
 930 }
 931 //-----------------------------------------------------------------------------
 932 int FILES_STORE::WriteLog2String(const std::string & str, const std::string & login) const
 933 {
 934 FILE * f;
 935 time_t tm = time(NULL);
 936 std::string fileName;
 937 fileName = storeSettings.GetUsersDir() + "/" + login + "/log2";
 938 f = fopen(fileName.c_str(), "at");
 939
 940 if (f)
 941     {
 942     fprintf(f, "%s", LogDate(tm));
 943     fprintf(f, " -- ");
 944     fprintf(f, "%s", str.c_str());
 945     fprintf(f, "\n");
 946     fclose(f);
 947     }
 948 else
 949     {
 950     STG_LOCKER lock(&mutex);
 951     errorStr = "Cannot open \'" + fileName + "\'";
 952     printfd(__FILE__, "FILES_STORE::WriteLogString - log write failed for user '%s'\n", login.c_str());
 953     return -1;
 954     }
 955
 956 int e = chmod(fileName.c_str(), storeSettings.GetLogMode());
 957 e += chown(fileName.c_str(), storeSettings.GetLogUID(), storeSettings.GetLogGID());
 958
 959 if (e)
 960     {
 961     STG_LOCKER lock(&mutex);
 962     printfd(__FILE__, "FILES_STORE::WriteLogString - chmod/chown failed for user '%s'. Error: '%s'\n", login.c_str(), strerror(errno));
 963     }
 964
 965 return 0;
 966 }
 967 //-----------------------------------------------------------------------------
 968 int FILES_STORE::WriteUserChgLog(const std::string & login,
 969                                  const std::string & admLogin,
 970                                  uint32_t       admIP,
 971                                  const std::string & paramName,
 972                                  const std::string & oldValue,
 973                                  const std::string & newValue,
 974                                  const std::string & message) const
 975 {
 976 std::string userLogMsg = "Admin \'" + admLogin + "\', " + inet_ntostring(admIP) + ": \'"
 977     + paramName + "\' parameter changed from \'" + oldValue +
 978     "\' to \'" + newValue + "\'. " + message;
 979
 980 return WriteLogString(userLogMsg, login);
 981 }
 982 //-----------------------------------------------------------------------------
 983 int FILES_STORE::WriteUserConnect(const std::string & login, uint32_t ip) const
 984 {
 985 std::string logStr = "Connect, " + inet_ntostring(ip);
 986 if (WriteLogString(logStr, login))
 987     return -1;
 988 return WriteLog2String(logStr, login);
 989 }
 990 //-----------------------------------------------------------------------------
 991 int FILES_STORE::WriteUserDisconnect(const std::string & login,
 992                                      const DIR_TRAFF & monthUp,
 993                                      const DIR_TRAFF & monthDown,
 994                                      const DIR_TRAFF & sessionUp,
 995                                      const DIR_TRAFF & sessionDown,
 996                                      double cash,
 997                                      double freeMb,
 998                                      const std::string & reason) const
 999 {
1000 std::ostringstream logStr;
1001 logStr << "Disconnect, "
1002        << " session upload: \'"
1003        << sessionUp
1004        << "\' session download: \'"
1005        << sessionDown
1006        << "\' month upload: \'"
1007        << monthUp
1008        << "\' month download: \'"
1009        << monthDown
1010        << "\' cash: \'"
1011        << cash
1012        << "\'";
1013
1014 if (WriteLogString(logStr.str(), login))
1015     return -1;
1016
1017 logStr << " freeMb: \'"
1018        << freeMb
1019        << "\'"
1020        << " reason: \'"
1021        << reason
1022        << "\'";
1023
1024 return WriteLog2String(logStr.str(), login);
1025 }
1026 //-----------------------------------------------------------------------------
1027 int FILES_STORE::SaveMonthStat(const USER_STAT & stat, int month, int year, const std::string & login) const
1028 {
1029 // Classic stats
1030 std::string stat1;
1031 strprintf(&stat1,"%s/%s/stat.%d.%02d",
1032         storeSettings.GetUsersDir().c_str(), login.c_str(), year + 1900, month + 1);
1033
1034 CONFIGFILE s(stat1, true);
1035
1036 if (s.Error())
1037     {
1038     STG_LOCKER lock(&mutex);
1039     errorStr = "Cannot create file '" + stat1 + "'";
1040     printfd(__FILE__, "FILES_STORE::SaveMonthStat - month stat write failed for user '%s'\n", login.c_str());
1041     return -1;
1042     }
1043
1044 // New stats
1045 std::string stat2;
1046 strprintf(&stat2,"%s/%s/stat2.%d.%02d",
1047         storeSettings.GetUsersDir().c_str(), login.c_str(), year + 1900, month + 1);
1048
1049 CONFIGFILE s2(stat2, true);
1050
1051 if (s2.Error())
1052     {
1053     STG_LOCKER lock(&mutex);
1054     errorStr = "Cannot create file '" + stat2 + "'";
1055     printfd(__FILE__, "FILES_STORE::SaveMonthStat - month stat write failed for user '%s'\n", login.c_str());
1056     return -1;
1057     }
1058
1059 for (size_t i = 0; i < DIR_NUM; i++)
1060     {
1061     char dirName[3];
1062     snprintf(dirName, 3, "U%llu", (unsigned long long)i);
1063     s.WriteInt(dirName, stat.monthUp[i]); // Classic
1064     s2.WriteInt(dirName, stat.monthUp[i]); // New
1065     snprintf(dirName, 3, "D%llu", (unsigned long long)i);
1066     s.WriteInt(dirName, stat.monthDown[i]); // Classic
1067     s2.WriteInt(dirName, stat.monthDown[i]); // New
1068     }
1069
1070 // Classic
1071 s.WriteDouble("cash", stat.cash);
1072
1073 // New
1074 s2.WriteDouble("Cash", stat.cash);
1075 s2.WriteDouble("FreeMb", stat.freeMb);
1076 s2.WriteDouble("LastCashAdd", stat.lastCashAdd);
1077 s2.WriteInt("LastCashAddTime", stat.lastCashAddTime);
1078 s2.WriteInt("PassiveTime", stat.passiveTime);
1079 s2.WriteInt("LastActivityTime", stat.lastActivityTime);
1080
1081 return 0;
1082 }
1083 //-----------------------------------------------------------------------------*/
1084 int FILES_STORE::AddAdmin(const std::string & login) const
1085 {
1086 std::string fileName;
1087 strprintf(&fileName, "%s/%s.adm", storeSettings.GetAdminsDir().c_str(), login.c_str());
1088
1089 if (Touch(fileName))
1090     {
1091     STG_LOCKER lock(&mutex);
1092     errorStr = "Cannot create file " + fileName;
1093     printfd(__FILE__, "FILES_STORE::AddAdmin - failed to add admin '%s'\n", login.c_str());
1094     return -1;
1095     }
1096
1097 return 0;
1098 }
1099 //-----------------------------------------------------------------------------*/
1100 int FILES_STORE::DelAdmin(const std::string & login) const
1101 {
1102 std::string fileName;
1103 strprintf(&fileName, "%s/%s.adm", storeSettings.GetAdminsDir().c_str(), login.c_str());
1104 if (unlink(fileName.c_str()))
1105     {
1106     STG_LOCKER lock(&mutex);
1107     errorStr = "unlink failed. Message: '";
1108     errorStr += strerror(errno);
1109     errorStr += "'";
1110     printfd(__FILE__, "FILES_STORE::DelAdmin - unlink failed. Message: '%s'\n", strerror(errno));
1111     }
1112 return 0;
1113 }
1114 //-----------------------------------------------------------------------------*/
1115 int FILES_STORE::SaveAdmin(const ADMIN_CONF & ac) const
1116 {
1117 std::string fileName;
1118
1119 strprintf(&fileName, "%s/%s.adm", storeSettings.GetAdminsDir().c_str(), ac.login.c_str());
1120
1121     {
1122     CONFIGFILE cf(fileName, true);
1123
1124     int e = cf.Error();
1125
1126     if (e)
1127         {
1128         STG_LOCKER lock(&mutex);
1129         errorStr = "Cannot write admin " + ac.login + ". " + fileName;
1130         printfd(__FILE__, "FILES_STORE::SaveAdmin - failed to save admin '%s'\n", ac.login.c_str());
1131         return -1;
1132         }
1133
1134     char pass[ADM_PASSWD_LEN + 1];
1135     memset(pass, 0, sizeof(pass));
1136
1137     char adminPass[ADM_PASSWD_LEN + 1];
1138     memset(adminPass, 0, sizeof(adminPass));
1139
1140     BLOWFISH_CTX ctx;
1141     EnDecodeInit(adm_enc_passwd, strlen(adm_enc_passwd), &ctx);
1142
1143     strncpy(adminPass, ac.password.c_str(), ADM_PASSWD_LEN);
1144     adminPass[ADM_PASSWD_LEN - 1] = 0;
1145
1146     for (int i = 0; i < ADM_PASSWD_LEN/8; i++)
1147         {
1148         EncodeString(pass + 8*i, adminPass + 8*i, &ctx);
1149         }
1150
1151     pass[ADM_PASSWD_LEN - 1] = 0;
1152     char passwordE[2 * ADM_PASSWD_LEN + 2];
1153     Encode12(passwordE, pass, ADM_PASSWD_LEN);
1154
1155     cf.WriteString("password", passwordE);
1156     cf.WriteInt("ChgConf",     ac.priv.userConf);
1157     cf.WriteInt("ChgPassword", ac.priv.userPasswd);
1158     cf.WriteInt("ChgStat",     ac.priv.userStat);
1159     cf.WriteInt("ChgCash",     ac.priv.userCash);
1160     cf.WriteInt("UsrAddDel",   ac.priv.userAddDel);
1161     cf.WriteInt("ChgTariff",   ac.priv.tariffChg);
1162     cf.WriteInt("ChgAdmin",    ac.priv.adminChg);
1163     cf.WriteInt("ChgService",  ac.priv.serviceChg);
1164     cf.WriteInt("ChgCorp",     ac.priv.corpChg);
1165     }
1166
1167 return 0;
1168 }
1169 //-----------------------------------------------------------------------------
1170 int FILES_STORE::RestoreAdmin(ADMIN_CONF * ac, const std::string & login) const
1171 {
1172 std::string fileName;
1173 strprintf(&fileName, "%s/%s.adm", storeSettings.GetAdminsDir().c_str(), login.c_str());
1174 CONFIGFILE cf(fileName);
1175 char pass[ADM_PASSWD_LEN + 1];
1176 char password[ADM_PASSWD_LEN + 1];
1177 char passwordE[2 * ADM_PASSWD_LEN + 2];
1178 BLOWFISH_CTX ctx;
1179
1180 std::string p;
1181
1182 if (cf.Error())
1183     {
1184     STG_LOCKER lock(&mutex);
1185     errorStr = "Cannot open " + fileName;
1186     printfd(__FILE__, "FILES_STORE::RestoreAdmin - failed to restore admin '%s'\n", ac->login.c_str());
1187     return -1;
1188     }
1189
1190 if (cf.ReadString("password", &p, "*"))
1191     {
1192     STG_LOCKER lock(&mutex);
1193     errorStr = "Error in parameter password";
1194     printfd(__FILE__, "FILES_STORE::RestoreAdmin - password read failed for admin '%s'\n", ac->login.c_str());
1195     return -1;
1196     }
1197
1198 memset(passwordE, 0, sizeof(passwordE));
1199 strncpy(passwordE, p.c_str(), 2*ADM_PASSWD_LEN);
1200
1201 memset(pass, 0, sizeof(pass));
1202
1203 if (passwordE[0] != 0)
1204     {
1205     Decode21(pass, passwordE);
1206     EnDecodeInit(adm_enc_passwd, strlen(adm_enc_passwd), &ctx);
1207
1208     for (int i = 0; i < ADM_PASSWD_LEN/8; i++)
1209         {
1210         DecodeString(password + 8*i, pass + 8*i, &ctx);
1211         }
1212     }
1213 else
1214     {
1215     password[0] = 0;
1216     }
1217
1218 ac->password = password;
1219
1220 uint16_t a;
1221
1222 if (cf.ReadUShortInt("ChgConf", &a, 0) == 0)
1223     ac->priv.userConf = a;
1224 else
1225     {
1226     STG_LOCKER lock(&mutex);
1227     errorStr = "Error in parameter ChgConf";
1228     printfd(__FILE__, "FILES_STORE::RestoreAdmin - chgconf read failed for admin '%s'\n", ac->login.c_str());
1229     return -1;
1230     }
1231
1232 if (cf.ReadUShortInt("ChgPassword", &a, 0) == 0)
1233     ac->priv.userPasswd = a;
1234 else
1235     {
1236     STG_LOCKER lock(&mutex);
1237     errorStr = "Error in parameter ChgPassword";
1238     printfd(__FILE__, "FILES_STORE::RestoreAdmin - chgpassword read failed for admin '%s'\n", ac->login.c_str());
1239     return -1;
1240     }
1241
1242 if (cf.ReadUShortInt("ChgStat", &a, 0) == 0)
1243     ac->priv.userStat = a;
1244 else
1245     {
1246     STG_LOCKER lock(&mutex);
1247     errorStr = "Error in parameter ChgStat";
1248     printfd(__FILE__, "FILES_STORE::RestoreAdmin - chgstat read failed for admin '%s'\n", ac->login.c_str());
1249     return -1;
1250     }
1251
1252 if (cf.ReadUShortInt("ChgCash", &a, 0) == 0)
1253     ac->priv.userCash = a;
1254 else
1255     {
1256     STG_LOCKER lock(&mutex);
1257     errorStr = "Error in parameter ChgCash";
1258     printfd(__FILE__, "FILES_STORE::RestoreAdmin - chgcash read failed for admin '%s'\n", ac->login.c_str());
1259     return -1;
1260     }
1261
1262 if (cf.ReadUShortInt("UsrAddDel", &a, 0) == 0)
1263     ac->priv.userAddDel = a;
1264 else
1265     {
1266     STG_LOCKER lock(&mutex);
1267     errorStr = "Error in parameter UsrAddDel";
1268     printfd(__FILE__, "FILES_STORE::RestoreAdmin - usradddel read failed for admin '%s'\n", ac->login.c_str());
1269     return -1;
1270     }
1271
1272 if (cf.ReadUShortInt("ChgAdmin", &a, 0) == 0)
1273     ac->priv.adminChg = a;
1274 else
1275     {
1276     STG_LOCKER lock(&mutex);
1277     errorStr = "Error in parameter ChgAdmin";
1278     printfd(__FILE__, "FILES_STORE::RestoreAdmin - chgadmin read failed for admin '%s'\n", ac->login.c_str());
1279     return -1;
1280     }
1281
1282 if (cf.ReadUShortInt("ChgTariff", &a, 0) == 0)
1283     ac->priv.tariffChg = a;
1284 else
1285     {
1286     STG_LOCKER lock(&mutex);
1287     errorStr = "Error in parameter ChgTariff";
1288     printfd(__FILE__, "FILES_STORE::RestoreAdmin - chgtariff read failed for admin '%s'\n", ac->login.c_str());
1289     return -1;
1290     }
1291
1292 if (cf.ReadUShortInt("ChgService", &a, 0) == 0)
1293     ac->priv.serviceChg = a;
1294 else
1295     ac->priv.serviceChg = 0;
1296
1297 if (cf.ReadUShortInt("ChgCorp", &a, 0) == 0)
1298     ac->priv.corpChg = a;
1299 else
1300     ac->priv.corpChg = 0;
1301
1302 return 0;
1303 }
1304 //-----------------------------------------------------------------------------
1305 int FILES_STORE::AddTariff(const std::string & name) const
1306 {
1307 std::string fileName;
1308 strprintf(&fileName, "%s/%s.tf", storeSettings.GetTariffsDir().c_str(), name.c_str());
1309 if (Touch(fileName))
1310     {
1311     STG_LOCKER lock(&mutex);
1312     errorStr = "Cannot create file " + fileName;
1313     printfd(__FILE__, "FILES_STORE::AddTariff - failed to add tariff '%s'\n", name.c_str());
1314     return -1;
1315     }
1316 return 0;
1317 }
1318 //-----------------------------------------------------------------------------
1319 int FILES_STORE::DelTariff(const std::string & name) const
1320 {
1321 std::string fileName;
1322 strprintf(&fileName, "%s/%s.tf", storeSettings.GetTariffsDir().c_str(), name.c_str());
1323 if (unlink(fileName.c_str()))
1324     {
1325     STG_LOCKER lock(&mutex);
1326     errorStr = "unlink failed. Message: '";
1327     errorStr += strerror(errno);
1328     errorStr += "'";
1329     printfd(__FILE__, "FILES_STORE::DelTariff - unlink failed. Message: '%s'\n", strerror(errno));
1330     }
1331 return 0;
1332 }
1333 //-----------------------------------------------------------------------------
1334 int FILES_STORE::RestoreTariff(TARIFF_DATA * td, const std::string & tariffName) const
1335 {
1336 std::string fileName = storeSettings.GetTariffsDir() + "/" + tariffName + ".tf";
1337 CONFIGFILE conf(fileName);
1338 std::string str;
1339 td->tariffConf.name = tariffName;
1340
1341 if (conf.Error() != 0)
1342     {
1343     STG_LOCKER lock(&mutex);
1344     errorStr = "Cannot read file " + fileName;
1345     printfd(__FILE__, "FILES_STORE::RestoreTariff - failed to read tariff '%s'\n", tariffName.c_str());
1346     return -1;
1347     }
1348
1349 std::string param;
1350 for (int i = 0; i<DIR_NUM; i++)
1351     {
1352     strprintf(&param, "Time%d", i);
1353     if (conf.ReadString(param, &str, "00:00-00:00") < 0)
1354         {
1355         STG_LOCKER lock(&mutex);
1356         errorStr = "Cannot read tariff " + tariffName + ". Parameter " + param;
1357         printfd(__FILE__, "FILES_STORE::RestoreTariff - time%d read failed for tariff '%s'\n", i, tariffName.c_str());
1358         return -1;
1359         }
1360
1361     ParseTariffTimeStr(str.c_str(),
1362                        td->dirPrice[i].hDay,
1363                        td->dirPrice[i].mDay,
1364                        td->dirPrice[i].hNight,
1365                        td->dirPrice[i].mNight);
1366
1367     strprintf(&param, "PriceDayA%d", i);
1368     if (conf.ReadDouble(param, &td->dirPrice[i].priceDayA, 0.0) < 0)
1369         {
1370         STG_LOCKER lock(&mutex);
1371         errorStr = "Cannot read tariff " + tariffName + ". Parameter " + param;
1372         printfd(__FILE__, "FILES_STORE::RestoreTariff - pricedaya read failed for tariff '%s'\n", tariffName.c_str());
1373         return -1;
1374         }
1375     td->dirPrice[i].priceDayA /= (1024*1024);
1376
1377     strprintf(&param, "PriceDayB%d", i);
1378     if (conf.ReadDouble(param, &td->dirPrice[i].priceDayB, 0.0) < 0)
1379         {
1380         STG_LOCKER lock(&mutex);
1381         errorStr = "Cannot read tariff " + tariffName + ". Parameter " + param;
1382         printfd(__FILE__, "FILES_STORE::RestoreTariff - pricedayb read failed for tariff '%s'\n", tariffName.c_str());
1383         return -1;
1384         }
1385     td->dirPrice[i].priceDayB /= (1024*1024);
1386
1387     strprintf(&param, "PriceNightA%d", i);
1388     if (conf.ReadDouble(param, &td->dirPrice[i].priceNightA, 0.0) < 0)
1389         {
1390         STG_LOCKER lock(&mutex);
1391         errorStr = "Cannot read tariff " + tariffName + ". Parameter " + param;
1392         printfd(__FILE__, "FILES_STORE::RestoreTariff - pricenighta read failed for tariff '%s'\n", tariffName.c_str());
1393         return -1;
1394         }
1395     td->dirPrice[i].priceNightA /= (1024*1024);
1396
1397     strprintf(&param, "PriceNightB%d", i);
1398     if (conf.ReadDouble(param, &td->dirPrice[i].priceNightB, 0.0) < 0)
1399         {
1400         STG_LOCKER lock(&mutex);
1401         errorStr = "Cannot read tariff " + tariffName + ". Parameter " + param;
1402         printfd(__FILE__, "FILES_STORE::RestoreTariff - pricenightb read failed for tariff '%s'\n", tariffName.c_str());
1403         return -1;
1404         }
1405     td->dirPrice[i].priceNightB /= (1024*1024);
1406
1407     strprintf(&param, "Threshold%d", i);
1408     if (conf.ReadInt(param, &td->dirPrice[i].threshold, 0) < 0)
1409         {
1410         STG_LOCKER lock(&mutex);
1411         errorStr = "Cannot read tariff " + tariffName + ". Parameter " + param;
1412         printfd(__FILE__, "FILES_STORE::RestoreTariff - threshold read failed for tariff '%s'\n", tariffName.c_str());
1413         return -1;
1414         }
1415
1416     strprintf(&param, "SinglePrice%d", i);
1417     if (conf.ReadInt(param, &td->dirPrice[i].singlePrice, 0) < 0)
1418         {
1419         STG_LOCKER lock(&mutex);
1420         errorStr = "Cannot read tariff " + tariffName + ". Parameter " + param;
1421         printfd(__FILE__, "FILES_STORE::RestoreTariff - singleprice read failed for tariff '%s'\n", tariffName.c_str());
1422         return -1;
1423         }
1424
1425     strprintf(&param, "NoDiscount%d", i);
1426     if (conf.ReadInt(param, &td->dirPrice[i].noDiscount, 0) < 0)
1427         {
1428         STG_LOCKER lock(&mutex);
1429         errorStr = "Cannot read tariff " + tariffName + ". Parameter " + param;
1430         printfd(__FILE__, "FILES_STORE::RestoreTariff - nodiscount read failed for tariff '%s'\n", tariffName.c_str());
1431         return -1;
1432         }
1433     }
1434
1435 if (conf.ReadDouble("Fee", &td->tariffConf.fee, 0) < 0)
1436     {
1437     STG_LOCKER lock(&mutex);
1438     errorStr = "Cannot read tariff " + tariffName + ". Parameter Fee";
1439     printfd(__FILE__, "FILES_STORE::RestoreTariff - fee read failed for tariff '%s'\n", tariffName.c_str());
1440     return -1;
1441     }
1442
1443 if (conf.ReadDouble("Free", &td->tariffConf.free, 0) < 0)
1444     {
1445     STG_LOCKER lock(&mutex);
1446     errorStr = "Cannot read tariff " + tariffName + ". Parameter Free";
1447     printfd(__FILE__, "FILES_STORE::RestoreTariff - free read failed for tariff '%s'\n", tariffName.c_str());
1448     return -1;
1449     }
1450
1451 if (conf.ReadDouble("PassiveCost", &td->tariffConf.passiveCost, 0) < 0)
1452     {
1453     STG_LOCKER lock(&mutex);
1454     errorStr = "Cannot read tariff " + tariffName + ". Parameter PassiveCost";
1455     printfd(__FILE__, "FILES_STORE::RestoreTariff - passivecost read failed for tariff '%s'\n", tariffName.c_str());
1456     return -1;
1457     }
1458
1459 if (conf.ReadString("TraffType", &str, "") < 0)
1460     {
1461     STG_LOCKER lock(&mutex);
1462     errorStr = "Cannot read tariff " + tariffName + ". Parameter TraffType";
1463     printfd(__FILE__, "FILES_STORE::RestoreTariff - trafftype read failed for tariff '%s'\n", tariffName.c_str());
1464     return -1;
1465     }
1466
1467 if (!strcasecmp(str.c_str(), "up"))
1468     td->tariffConf.traffType = TRAFF_UP;
1469 else
1470     if (!strcasecmp(str.c_str(), "down"))
1471         td->tariffConf.traffType = TRAFF_DOWN;
1472     else
1473         if (!strcasecmp(str.c_str(), "up+down"))
1474             td->tariffConf.traffType = TRAFF_UP_DOWN;
1475         else
1476             if (!strcasecmp(str.c_str(), "max"))
1477                 td->tariffConf.traffType = TRAFF_MAX;
1478             else
1479                 {
1480                 STG_LOCKER lock(&mutex);
1481                 errorStr = "Cannot read tariff " + tariffName + ". Parameter TraffType incorrect";
1482                 printfd(__FILE__, "FILES_STORE::RestoreTariff - invalid trafftype for tariff '%s'\n", tariffName.c_str());
1483                 return -1;
1484                 }
1485
1486 if (conf.ReadString("Period", &str, "month") < 0)
1487     td->tariffConf.period = TARIFF::MONTH;
1488 else
1489     td->tariffConf.period = TARIFF::StringToPeriod(str);
1490 return 0;
1491 }
1492 //-----------------------------------------------------------------------------
1493 int FILES_STORE::SaveTariff(const TARIFF_DATA & td, const std::string & tariffName) const
1494 {
1495 std::string fileName = storeSettings.GetTariffsDir() + "/" + tariffName + ".tf";
1496
1497     {
1498     CONFIGFILE cf(fileName, true);
1499
1500     int e = cf.Error();
1501
1502     if (e)
1503         {
1504         STG_LOCKER lock(&mutex);
1505         errorStr = "Error writing tariff " + tariffName;
1506         printfd(__FILE__, "FILES_STORE::RestoreTariff - failed to save tariff '%s'\n", tariffName.c_str());
1507         return e;
1508         }
1509
1510     std::string param;
1511     for (int i = 0; i < DIR_NUM; i++)
1512         {
1513         strprintf(&param, "PriceDayA%d", i);
1514         cf.WriteDouble(param, td.dirPrice[i].priceDayA * pt_mega);
1515
1516         strprintf(&param, "PriceDayB%d", i);
1517         cf.WriteDouble(param, td.dirPrice[i].priceDayB * pt_mega);
1518
1519         strprintf(&param, "PriceNightA%d", i);
1520         cf.WriteDouble(param, td.dirPrice[i].priceNightA * pt_mega);
1521
1522         strprintf(&param, "PriceNightB%d", i);
1523         cf.WriteDouble(param, td.dirPrice[i].priceNightB * pt_mega);
1524
1525         strprintf(&param, "Threshold%d", i);
1526         cf.WriteInt(param, td.dirPrice[i].threshold);
1527
1528         std::string s;
1529         strprintf(&param, "Time%d", i);
1530
1531         strprintf(&s, "%0d:%0d-%0d:%0d",
1532                 td.dirPrice[i].hDay,
1533                 td.dirPrice[i].mDay,
1534                 td.dirPrice[i].hNight,
1535                 td.dirPrice[i].mNight);
1536
1537         cf.WriteString(param, s);
1538
1539         strprintf(&param, "NoDiscount%d", i);
1540         cf.WriteInt(param, td.dirPrice[i].noDiscount);
1541
1542         strprintf(&param, "SinglePrice%d", i);
1543         cf.WriteInt(param, td.dirPrice[i].singlePrice);
1544         }
1545
1546     cf.WriteDouble("PassiveCost", td.tariffConf.passiveCost);
1547     cf.WriteDouble("Fee", td.tariffConf.fee);
1548     cf.WriteDouble("Free", td.tariffConf.free);
1549
1550     switch (td.tariffConf.traffType)
1551         {
1552         case TRAFF_UP:
1553             cf.WriteString("TraffType", "up");
1554             break;
1555         case TRAFF_DOWN:
1556             cf.WriteString("TraffType", "down");
1557             break;
1558         case TRAFF_UP_DOWN:
1559             cf.WriteString("TraffType", "up+down");
1560             break;
1561         case TRAFF_MAX:
1562             cf.WriteString("TraffType", "max");
1563             break;
1564         }
1565
1566     cf.WriteString("Period", TARIFF::PeriodToString(td.tariffConf.period));
1567     }
1568
1569 return 0;
1570 }
1571 //-----------------------------------------------------------------------------
1572 int FILES_STORE::WriteDetailedStat(const std::map<IP_DIR_PAIR, STAT_NODE> & statTree,
1573                                    time_t lastStat,
1574                                    const std::string & login) const
1575 {
1576 char fn[FN_STR_LEN];
1577 char dn[FN_STR_LEN];
1578 FILE * statFile;
1579 time_t t;
1580 tm * lt;
1581
1582 t = time(NULL);
1583
1584 snprintf(dn, FN_STR_LEN, "%s/%s/detail_stat", storeSettings.GetUsersDir().c_str(), login.c_str());
1585 if (access(dn, F_OK) != 0)
1586     {
1587     if (mkdir(dn, 0700) != 0)
1588         {
1589         STG_LOCKER lock(&mutex);
1590         errorStr = "Directory \'" + std::string(dn) + "\' cannot be created.";
1591         printfd(__FILE__, "FILES_STORE::WriteDetailStat - mkdir failed. Message: '%s'\n", strerror(errno));
1592         return -1;
1593         }
1594     }
1595
1596 int e = chown(dn, storeSettings.GetStatUID(), storeSettings.GetStatGID());
1597 e += chmod(dn, storeSettings.GetStatModeDir());
1598
1599 if (e)
1600     {
1601     STG_LOCKER lock(&mutex);
1602     printfd(__FILE__, "FILES_STORE::WriteDetailStat - chmod/chown failed for user '%s'. Error: '%s'\n", login.c_str(), strerror(errno));
1603     }
1604
1605 lt = localtime(&t);
1606
1607 if (lt->tm_hour == 0 && lt->tm_min <= 5)
1608     {
1609     t -= 3600 * 24;
1610     lt = localtime(&t);
1611     }
1612
1613 snprintf(dn, FN_STR_LEN, "%s/%s/detail_stat/%d",
1614          storeSettings.GetUsersDir().c_str(),
1615          login.c_str(),
1616          lt->tm_year+1900);
1617
1618 if (access(dn, F_OK) != 0)
1619     {
1620     if (mkdir(dn, 0700) != 0)
1621         {
1622         STG_LOCKER lock(&mutex);
1623         errorStr = "Directory \'" + std::string(dn) + "\' cannot be created.";
1624         printfd(__FILE__, "FILES_STORE::WriteDetailStat - mkdir failed. Message: '%s'\n", strerror(errno));
1625         return -1;
1626         }
1627     }
1628
1629 e = chown(dn, storeSettings.GetStatUID(), storeSettings.GetStatGID());
1630 e += chmod(dn, storeSettings.GetStatModeDir());
1631
1632 if (e)
1633     {
1634     STG_LOCKER lock(&mutex);
1635     printfd(__FILE__, "FILES_STORE::WriteDetailStat - chmod/chown failed for user '%s'. Error: '%s'\n", login.c_str(), strerror(errno));
1636     }
1637
1638 snprintf(dn, FN_STR_LEN, "%s/%s/detail_stat/%d/%s%d",
1639          storeSettings.GetUsersDir().c_str(),
1640          login.c_str(),
1641          lt->tm_year+1900,
1642          lt->tm_mon+1 < 10 ? "0" : "",
1643          lt->tm_mon+1);
1644 if (access(dn, F_OK) != 0)
1645     {
1646     if (mkdir(dn, 0700) != 0)
1647         {
1648         STG_LOCKER lock(&mutex);
1649         errorStr = "Directory \'" + std::string(dn) + "\' cannot be created.";
1650         printfd(__FILE__, "FILES_STORE::WriteDetailStat - mkdir failed. Message: '%s'\n", strerror(errno));
1651         return -1;
1652         }
1653     }
1654
1655 e = chown(dn, storeSettings.GetStatUID(), storeSettings.GetStatGID());
1656 e += chmod(dn, storeSettings.GetStatModeDir());
1657
1658 if (e)
1659     {
1660     STG_LOCKER lock(&mutex);
1661     printfd(__FILE__, "FILES_STORE::WriteDetailStat - chmod/chown failed for user '%s'. Error: '%s'\n", login.c_str(), strerror(errno));
1662     }
1663
1664 snprintf(fn, FN_STR_LEN, "%s/%s%d", dn, lt->tm_mday < 10 ? "0" : "", lt->tm_mday);
1665
1666 statFile = fopen (fn, "at");
1667
1668 if (!statFile)
1669     {
1670     STG_LOCKER lock(&mutex);
1671     errorStr = "File \'" + std::string(fn) + "\' cannot be written.";
1672     printfd(__FILE__, "FILES_STORE::WriteDetailStat - fopen failed. Message: '%s'\n", strerror(errno));
1673     return -1;
1674     }
1675
1676 struct tm * lt1;
1677 struct tm * lt2;
1678
1679 lt1 = localtime(&lastStat);
1680
1681 int h1, m1, s1;
1682 int h2, m2, s2;
1683
1684 h1 = lt1->tm_hour;
1685 m1 = lt1->tm_min;
1686 s1 = lt1->tm_sec;
1687
1688 lt2 = localtime(&t);
1689
1690 h2 = lt2->tm_hour;
1691 m2 = lt2->tm_min;
1692 s2 = lt2->tm_sec;
1693
1694 if (fprintf(statFile, "-> %02d.%02d.%02d - %02d.%02d.%02d\n",
1695             h1, m1, s1, h2, m2, s2) < 0)
1696     {
1697     STG_LOCKER lock(&mutex);
1698     errorStr = std::string("fprint failed. Message: '") + strerror(errno) + "'";
1699     printfd(__FILE__, "FILES_STORE::WriteDetailStat - fprintf failed. Message: '%s'\n", strerror(errno));
1700     fclose(statFile);
1701     return -1;
1702     }
1703
1704 std::map<IP_DIR_PAIR, STAT_NODE>::const_iterator stIter;
1705 stIter = statTree.begin();
1706
1707 while (stIter != statTree.end())
1708     {
1709     std::string u, d;
1710     x2str(stIter->second.up, u);
1711     x2str(stIter->second.down, d);
1712     #ifdef TRAFF_STAT_WITH_PORTS
1713     if (fprintf(statFile, "%17s:%hu\t%15d\t%15s\t%15s\t%f\n",
1714                 inet_ntostring(stIter->first.ip).c_str(),
1715                 stIter->first.port,
1716                 stIter->first.dir,
1717                 d.c_str(),
1718                 u.c_str(),
1719                 stIter->second.cash) < 0)
1720         {
1721         STG_LOCKER lock(&mutex);
1722         errorStr = "fprint failed. Message: '";
1723         errorStr += strerror(errno);
1724         errorStr += "'";
1725         printfd(__FILE__, "FILES_STORE::WriteDetailStat - fprintf failed. Message: '%s'\n", strerror(errno));
1726         fclose(statFile);
1727         return -1;
1728         }
1729     #else
1730     if (fprintf(statFile, "%17s\t%15d\t%15s\t%15s\t%f\n",
1731                 inet_ntostring(stIter->first.ip).c_str(),
1732                 stIter->first.dir,
1733                 d.c_str(),
1734                 u.c_str(),
1735                 stIter->second.cash) < 0)
1736         {
1737         STG_LOCKER lock(&mutex);
1738         errorStr = std::string("fprint failed. Message: '");
1739         errorStr += strerror(errno);
1740         errorStr += "'";
1741         printfd(__FILE__, "FILES_STORE::WriteDetailStat - fprintf failed. Message: '%s'\n", strerror(errno));
1742         fclose(statFile);
1743         return -1;
1744         }
1745     #endif
1746
1747     ++stIter;
1748     }
1749
1750 fclose(statFile);
1751
1752 e = chown(fn, storeSettings.GetStatUID(), storeSettings.GetStatGID());
1753 e += chmod(fn, storeSettings.GetStatMode());
1754
1755 if (e)
1756     {
1757     STG_LOCKER lock(&mutex);
1758     printfd(__FILE__, "FILES_STORE::WriteDetailStat - chmod/chown failed for user '%s'. Error: '%s'\n", login.c_str(), strerror(errno));
1759     }
1760
1761 return 0;
1762 }
1763 //-----------------------------------------------------------------------------
1764 int FILES_STORE::AddMessage(STG_MSG * msg, const std::string & login) const
1765 {
1766 std::string fn;
1767 std::string dn;
1768 struct timeval tv;
1769
1770 strprintf(&dn, "%s/%s/messages", storeSettings.GetUsersDir().c_str(), login.c_str());
1771 if (access(dn.c_str(), F_OK) != 0)
1772     {
1773     if (mkdir(dn.c_str(), 0700) != 0)
1774         {
1775         STG_LOCKER lock(&mutex);
1776         errorStr = "Directory \'";
1777         errorStr += dn;
1778         errorStr += "\' cannot be created.";
1779         printfd(__FILE__, "FILES_STORE::AddMessage - mkdir failed. Message: '%s'\n", strerror(errno));
1780         return -1;
1781         }
1782     }
1783
1784 chmod(dn.c_str(), storeSettings.GetConfModeDir());
1785
1786 gettimeofday(&tv, NULL);
1787
1788 msg->header.id = ((long long)tv.tv_sec) * 1000000 + ((long long)tv.tv_usec);
1789 strprintf(&fn, "%s/%lld", dn.c_str(), msg->header.id);
1790
1791 if (Touch(fn))
1792     {
1793     STG_LOCKER lock(&mutex);
1794     errorStr = "File \'";
1795     errorStr += fn;
1796     errorStr += "\' cannot be writen.";
1797     printfd(__FILE__, "FILES_STORE::AddMessage - fopen failed. Message: '%s'\n", strerror(errno));
1798     return -1;
1799     }
1800
1801 return EditMessage(*msg, login);
1802 }
1803 //-----------------------------------------------------------------------------
1804 int FILES_STORE::EditMessage(const STG_MSG & msg, const std::string & login) const
1805 {
1806 std::string fileName;
1807
1808 FILE * msgFile;
1809 strprintf(&fileName, "%s/%s/messages/%lld", storeSettings.GetUsersDir().c_str(), login.c_str(), msg.header.id);
1810
1811 if (access(fileName.c_str(), F_OK) != 0)
1812     {
1813     std::string idstr;
1814     x2str(msg.header.id, idstr);
1815     STG_LOCKER lock(&mutex);
1816     errorStr = "Message for user \'";
1817     errorStr += login + "\' with ID \'";
1818     errorStr += idstr + "\' does not exist.";
1819     printfd(__FILE__, "FILES_STORE::EditMessage - %s\n", errorStr.c_str());
1820     return -1;
1821     }
1822
1823 Touch(fileName + ".new");
1824
1825 msgFile = fopen((fileName + ".new").c_str(), "wt");
1826 if (!msgFile)
1827     {
1828     STG_LOCKER lock(&mutex);
1829     errorStr = "File \'" + fileName + "\' cannot be writen.";
1830     printfd(__FILE__, "FILES_STORE::EditMessage - fopen failed. Message: '%s'\n", strerror(errno));
1831     return -1;
1832     }
1833
1834 bool res = true;
1835 res &= (fprintf(msgFile, "%u\n", msg.header.type) >= 0);
1836 res &= (fprintf(msgFile, "%u\n", msg.header.lastSendTime) >= 0);
1837 res &= (fprintf(msgFile, "%u\n", msg.header.creationTime) >= 0);
1838 res &= (fprintf(msgFile, "%u\n", msg.header.showTime) >= 0);
1839 res &= (fprintf(msgFile, "%d\n", msg.header.repeat) >= 0);
1840 res &= (fprintf(msgFile, "%u\n", msg.header.repeatPeriod) >= 0);
1841 res &= (fprintf(msgFile, "%s", msg.text.c_str()) >= 0);
1842
1843 if (!res)
1844     {
1845     STG_LOCKER lock(&mutex);
1846     errorStr = std::string("fprintf failed. Message: '") + strerror(errno) + "'";
1847     printfd(__FILE__, "FILES_STORE::EditMessage - fprintf failed. Message: '%s'\n", strerror(errno));
1848     fclose(msgFile);
1849     return -1;
1850     }
1851
1852 fclose(msgFile);
1853
1854 chmod((fileName + ".new").c_str(), storeSettings.GetConfMode());
1855
1856 if (rename((fileName + ".new").c_str(), fileName.c_str()) < 0)
1857     {
1858     STG_LOCKER lock(&mutex);
1859     errorStr = "Error moving dir from " + fileName + ".new to " + fileName;
1860     printfd(__FILE__, "FILES_STORE::EditMessage - rename failed. Message: '%s'\n", strerror(errno));
1861     return -1;
1862     }
1863
1864 return 0;
1865 }
1866 //-----------------------------------------------------------------------------
1867 int FILES_STORE::GetMessage(uint64_t id, STG_MSG * msg, const std::string & login) const
1868 {
1869 std::string fn;
1870 strprintf(&fn, "%s/%s/messages/%lld", storeSettings.GetUsersDir().c_str(), login.c_str(), id);
1871 msg->header.id = id;
1872 return ReadMessage(fn, &msg->header, &msg->text);
1873 }
1874 //-----------------------------------------------------------------------------
1875 int FILES_STORE::DelMessage(uint64_t id, const std::string & login) const
1876 {
1877 std::string fn;
1878 strprintf(&fn, "%s/%s/messages/%lld", storeSettings.GetUsersDir().c_str(), login.c_str(), id);
1879
1880 return unlink(fn.c_str());
1881 }
1882 //-----------------------------------------------------------------------------
1883 int FILES_STORE::GetMessageHdrs(std::vector<STG_MSG_HDR> * hdrsList, const std::string & login) const
1884 {
1885 std::string dn(storeSettings.GetUsersDir() + "/" + login + "/messages/");
1886
1887 if (access(dn.c_str(), F_OK) != 0)
1888     {
1889     return 0;
1890     }
1891
1892 std::vector<std::string> messages;
1893 GetFileList(&messages, dn, S_IFREG, "");
1894
1895 for (unsigned i = 0; i < messages.size(); i++)
1896     {
1897     unsigned long long id = 0;
1898
1899     if (str2x(messages[i].c_str(), id))
1900         {
1901         if (unlink((dn + messages[i]).c_str()))
1902             {
1903             STG_LOCKER lock(&mutex);
1904             errorStr = std::string("unlink failed. Message: '") + strerror(errno) + "'";
1905             printfd(__FILE__, "FILES_STORE::GetMessageHdrs - unlink failed. Message: '%s'\n", strerror(errno));
1906             return -1;
1907             }
1908         continue;
1909         }
1910
1911     STG_MSG_HDR hdr;
1912     if (ReadMessage(dn + messages[i], &hdr, NULL))
1913         {
1914         return -1;
1915         }
1916
1917     if (hdr.repeat < 0)
1918         {
1919         if (unlink((dn + messages[i]).c_str()))
1920             {
1921             STG_LOCKER lock(&mutex);
1922             errorStr = std::string("unlink failed. Message: '") + strerror(errno) + "'";
1923             printfd(__FILE__, "FILES_STORE::GetMessageHdrs - unlink failed. Message: '%s'\n", strerror(errno));
1924             return -1;
1925             }
1926         continue;
1927         }
1928
1929     hdr.id = id;
1930     hdrsList->push_back(hdr);
1931     }
1932 return 0;
1933 }
1934 //-----------------------------------------------------------------------------
1935 int FILES_STORE::ReadMessage(const std::string & fileName,
1936                              STG_MSG_HDR * hdr,
1937                              std::string * text) const
1938 {
1939 FILE * msgFile;
1940 msgFile = fopen(fileName.c_str(), "rt");
1941 if (!msgFile)
1942     {
1943     STG_LOCKER lock(&mutex);
1944     errorStr = "File \'";
1945     errorStr += fileName;
1946     errorStr += "\' cannot be openned.";
1947     printfd(__FILE__, "FILES_STORE::ReadMessage - fopen failed. Message: '%s'\n", strerror(errno));
1948     return -1;
1949     }
1950 char p[20];
1951 unsigned * d[6];
1952 d[0] = &hdr->type;
1953 d[1] = &hdr->lastSendTime;
1954 d[2] = &hdr->creationTime;
1955 d[3] = &hdr->showTime;
1956 d[4] = (unsigned*)(&hdr->repeat);
1957 d[5] = &hdr->repeatPeriod;
1958
1959 memset(p, 0, sizeof(p));
1960
1961 for (int pos = 0; pos < 6; pos++)
1962     {
1963     if (fgets(p, sizeof(p) - 1, msgFile) == NULL) {
1964         STG_LOCKER lock(&mutex);
1965         errorStr = "Cannot read file \'";
1966         errorStr += fileName;
1967         errorStr += "\'. Missing data.";
1968         printfd(__FILE__, "FILES_STORE::ReadMessage - cannot read file (missing data)\n");
1969         printfd(__FILE__, "FILES_STORE::ReadMessage - position: %d\n", pos);
1970         fclose(msgFile);
1971         return -1;
1972     }
1973
1974     char * ep;
1975     ep = strrchr(p, '\r');
1976     if (ep) *ep = 0;
1977     ep = strrchr(p, '\n');
1978     if (ep) *ep = 0;
1979
1980     if (feof(msgFile))
1981         {
1982         STG_LOCKER lock(&mutex);
1983         errorStr = "Cannot read file \'";
1984         errorStr += fileName;
1985         errorStr += "\'. Missing data.";
1986         printfd(__FILE__, "FILES_STORE::ReadMessage - cannot read file (feof)\n");
1987         printfd(__FILE__, "FILES_STORE::ReadMessage - position: %d\n", pos);
1988         fclose(msgFile);
1989         return -1;
1990         }
1991
1992     if (str2x(p, *(d[pos])))
1993         {
1994         STG_LOCKER lock(&mutex);
1995         errorStr = "Cannot read file \'";
1996         errorStr += fileName;
1997         errorStr += "\'. Incorrect value. \'";
1998         errorStr += p;
1999         errorStr += "\'";
2000         printfd(__FILE__, "FILES_STORE::ReadMessage - incorrect value\n");
2001         fclose(msgFile);
2002         return -1;
2003         }
2004     }
2005
2006 char txt[2048];
2007 memset(txt, 0, sizeof(txt));
2008 if (text)
2009     {
2010     text->erase(text->begin(), text->end());
2011     while (!feof(msgFile))
2012         {
2013         txt[0] = 0;
2014         if (fgets(txt, sizeof(txt) - 1, msgFile) == NULL) {
2015             break;
2016         }
2017
2018         (*text) += txt;
2019         }
2020     }
2021 fclose(msgFile);
2022 return 0;
2023 }
2024 //-----------------------------------------------------------------------------
2025 int FILES_STORE::Touch(const std::string & path) const
2026 {
2027 FILE * f = fopen(path.c_str(), "wb");
2028 if (f)
2029     {
2030     fclose(f);
2031     return 0;
2032     }
2033 return -1;
2034 }
2035 //-----------------------------------------------------------------------------
2036 int GetFileList(std::vector<std::string> * fileList, const std::string & directory, mode_t mode, const std::string & ext)
2037 {
2038 DIR * d = opendir(directory.c_str());
2039
2040 if (!d)
2041     {
2042     printfd(__FILE__, "GetFileList - Failed to open dir '%s': '%s'\n", directory.c_str(), strerror(errno));
2043     return -1;
2044     }
2045
2046 dirent * entry;
2047 while ((entry = readdir(d)))
2048     {
2049     if (!(strcmp(entry->d_name, ".") && strcmp(entry->d_name, "..")))
2050         continue;
2051
2052     std::string str = directory + "/" + std::string(entry->d_name);
2053
2054     struct stat st;
2055     if (stat(str.c_str(), &st))
2056         continue;
2057
2058     if (!(st.st_mode & mode)) // Filter by mode
2059         continue;
2060
2061     if (!ext.empty())
2062         {
2063         // Check extension
2064         size_t d_nameLen = strlen(entry->d_name);
2065         if (d_nameLen <= ext.size())
2066             continue;
2067
2068         if (ext == entry->d_name + (d_nameLen - ext.size()))
2069             {
2070             entry->d_name[d_nameLen - ext.size()] = 0;
2071             fileList->push_back(entry->d_name);
2072             }
2073         }
2074     else
2075         {
2076         fileList->push_back(entry->d_name);
2077         }
2078     }
2079
2080 closedir(d);
2081
2082 return 0;
2083 }
2084 //-----------------------------------------------------------------------------