]> git.stg.codes - stg.git/blob - projects/stargazer/plugins/store/files/file_store.cpp
Use separate etst files for subsystems.
[stg.git] / 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     InitContext(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         EncryptBlock(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     InitContext(adm_enc_passwd, strlen(adm_enc_passwd), &ctx);
1207
1208     for (int i = 0; i < ADM_PASSWD_LEN/8; i++)
1209         {
1210         DecryptBlock(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 td->tariffConf.traffType = TARIFF::StringToTraffType(str);
1468
1469 if (conf.ReadString("Period", &str, "month") < 0)
1470     td->tariffConf.period = TARIFF::MONTH;
1471 else
1472     td->tariffConf.period = TARIFF::StringToPeriod(str);
1473 return 0;
1474 }
1475 //-----------------------------------------------------------------------------
1476 int FILES_STORE::SaveTariff(const TARIFF_DATA & td, const std::string & tariffName) const
1477 {
1478 std::string fileName = storeSettings.GetTariffsDir() + "/" + tariffName + ".tf";
1479
1480     {
1481     CONFIGFILE cf(fileName, true);
1482
1483     int e = cf.Error();
1484
1485     if (e)
1486         {
1487         STG_LOCKER lock(&mutex);
1488         errorStr = "Error writing tariff " + tariffName;
1489         printfd(__FILE__, "FILES_STORE::RestoreTariff - failed to save tariff '%s'\n", tariffName.c_str());
1490         return e;
1491         }
1492
1493     std::string param;
1494     for (int i = 0; i < DIR_NUM; i++)
1495         {
1496         strprintf(&param, "PriceDayA%d", i);
1497         cf.WriteDouble(param, td.dirPrice[i].priceDayA * pt_mega);
1498
1499         strprintf(&param, "PriceDayB%d", i);
1500         cf.WriteDouble(param, td.dirPrice[i].priceDayB * pt_mega);
1501
1502         strprintf(&param, "PriceNightA%d", i);
1503         cf.WriteDouble(param, td.dirPrice[i].priceNightA * pt_mega);
1504
1505         strprintf(&param, "PriceNightB%d", i);
1506         cf.WriteDouble(param, td.dirPrice[i].priceNightB * pt_mega);
1507
1508         strprintf(&param, "Threshold%d", i);
1509         cf.WriteInt(param, td.dirPrice[i].threshold);
1510
1511         std::string s;
1512         strprintf(&param, "Time%d", i);
1513
1514         strprintf(&s, "%0d:%0d-%0d:%0d",
1515                 td.dirPrice[i].hDay,
1516                 td.dirPrice[i].mDay,
1517                 td.dirPrice[i].hNight,
1518                 td.dirPrice[i].mNight);
1519
1520         cf.WriteString(param, s);
1521
1522         strprintf(&param, "NoDiscount%d", i);
1523         cf.WriteInt(param, td.dirPrice[i].noDiscount);
1524
1525         strprintf(&param, "SinglePrice%d", i);
1526         cf.WriteInt(param, td.dirPrice[i].singlePrice);
1527         }
1528
1529     cf.WriteDouble("PassiveCost", td.tariffConf.passiveCost);
1530     cf.WriteDouble("Fee", td.tariffConf.fee);
1531     cf.WriteDouble("Free", td.tariffConf.free);
1532     cf.WriteString("TraffType", TARIFF::TraffTypeToString(td.tariffConf.traffType));
1533     cf.WriteString("Period", TARIFF::PeriodToString(td.tariffConf.period));
1534     }
1535
1536 return 0;
1537 }
1538 //-----------------------------------------------------------------------------
1539 int FILES_STORE::WriteDetailedStat(const std::map<IP_DIR_PAIR, STAT_NODE> & statTree,
1540                                    time_t lastStat,
1541                                    const std::string & login) const
1542 {
1543 char fn[FN_STR_LEN];
1544 char dn[FN_STR_LEN];
1545 FILE * statFile;
1546 time_t t;
1547 tm * lt;
1548
1549 t = time(NULL);
1550
1551 snprintf(dn, FN_STR_LEN, "%s/%s/detail_stat", storeSettings.GetUsersDir().c_str(), login.c_str());
1552 if (access(dn, F_OK) != 0)
1553     {
1554     if (mkdir(dn, 0700) != 0)
1555         {
1556         STG_LOCKER lock(&mutex);
1557         errorStr = "Directory \'" + std::string(dn) + "\' cannot be created.";
1558         printfd(__FILE__, "FILES_STORE::WriteDetailStat - mkdir failed. Message: '%s'\n", strerror(errno));
1559         return -1;
1560         }
1561     }
1562
1563 int e = chown(dn, storeSettings.GetStatUID(), storeSettings.GetStatGID());
1564 e += chmod(dn, storeSettings.GetStatModeDir());
1565
1566 if (e)
1567     {
1568     STG_LOCKER lock(&mutex);
1569     printfd(__FILE__, "FILES_STORE::WriteDetailStat - chmod/chown failed for user '%s'. Error: '%s'\n", login.c_str(), strerror(errno));
1570     }
1571
1572 lt = localtime(&t);
1573
1574 if (lt->tm_hour == 0 && lt->tm_min <= 5)
1575     {
1576     t -= 3600 * 24;
1577     lt = localtime(&t);
1578     }
1579
1580 snprintf(dn, FN_STR_LEN, "%s/%s/detail_stat/%d",
1581          storeSettings.GetUsersDir().c_str(),
1582          login.c_str(),
1583          lt->tm_year+1900);
1584
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 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 snprintf(dn, FN_STR_LEN, "%s/%s/detail_stat/%d/%s%d", 
1606          storeSettings.GetUsersDir().c_str(),
1607          login.c_str(),
1608          lt->tm_year+1900,
1609          lt->tm_mon+1 < 10 ? "0" : "",
1610          lt->tm_mon+1);
1611 if (access(dn, F_OK) != 0)
1612     {
1613     if (mkdir(dn, 0700) != 0)
1614         {
1615         STG_LOCKER lock(&mutex);
1616         errorStr = "Directory \'" + std::string(dn) + "\' cannot be created.";
1617         printfd(__FILE__, "FILES_STORE::WriteDetailStat - mkdir failed. Message: '%s'\n", strerror(errno));
1618         return -1;
1619         }
1620     }
1621
1622 e = chown(dn, storeSettings.GetStatUID(), storeSettings.GetStatGID());
1623 e += chmod(dn, storeSettings.GetStatModeDir());
1624
1625 if (e)
1626     {
1627     STG_LOCKER lock(&mutex);
1628     printfd(__FILE__, "FILES_STORE::WriteDetailStat - chmod/chown failed for user '%s'. Error: '%s'\n", login.c_str(), strerror(errno));
1629     }
1630
1631 snprintf(fn, FN_STR_LEN, "%s/%s%d", dn, lt->tm_mday < 10 ? "0" : "", lt->tm_mday);
1632
1633 statFile = fopen (fn, "at");
1634
1635 if (!statFile)
1636     {
1637     STG_LOCKER lock(&mutex);
1638     errorStr = "File \'" + std::string(fn) + "\' cannot be written.";
1639     printfd(__FILE__, "FILES_STORE::WriteDetailStat - fopen failed. Message: '%s'\n", strerror(errno));
1640     return -1;
1641     }
1642
1643 struct tm * lt1;
1644 struct tm * lt2;
1645
1646 lt1 = localtime(&lastStat);
1647
1648 int h1, m1, s1;
1649 int h2, m2, s2;
1650
1651 h1 = lt1->tm_hour;
1652 m1 = lt1->tm_min;
1653 s1 = lt1->tm_sec;
1654
1655 lt2 = localtime(&t);
1656
1657 h2 = lt2->tm_hour;
1658 m2 = lt2->tm_min;
1659 s2 = lt2->tm_sec;
1660
1661 if (fprintf(statFile, "-> %02d.%02d.%02d - %02d.%02d.%02d\n",
1662             h1, m1, s1, h2, m2, s2) < 0)
1663     {
1664     STG_LOCKER lock(&mutex);
1665     errorStr = std::string("fprint failed. Message: '") + strerror(errno) + "'";
1666     printfd(__FILE__, "FILES_STORE::WriteDetailStat - fprintf failed. Message: '%s'\n", strerror(errno));
1667     fclose(statFile);
1668     return -1;
1669     }
1670
1671 std::map<IP_DIR_PAIR, STAT_NODE>::const_iterator stIter;
1672 stIter = statTree.begin();
1673
1674 while (stIter != statTree.end())
1675     {
1676     std::string u, d;
1677     x2str(stIter->second.up, u);
1678     x2str(stIter->second.down, d);
1679     #ifdef TRAFF_STAT_WITH_PORTS
1680     if (fprintf(statFile, "%17s:%hu\t%15d\t%15s\t%15s\t%f\n",
1681                 inet_ntostring(stIter->first.ip).c_str(),
1682                 stIter->first.port,
1683                 stIter->first.dir,
1684                 d.c_str(),
1685                 u.c_str(),
1686                 stIter->second.cash) < 0)
1687         {
1688         STG_LOCKER lock(&mutex);
1689         errorStr = "fprint failed. Message: '";
1690         errorStr += strerror(errno);
1691         errorStr += "'";
1692         printfd(__FILE__, "FILES_STORE::WriteDetailStat - fprintf failed. Message: '%s'\n", strerror(errno));
1693         fclose(statFile);
1694         return -1;
1695         }
1696     #else
1697     if (fprintf(statFile, "%17s\t%15d\t%15s\t%15s\t%f\n",
1698                 inet_ntostring(stIter->first.ip).c_str(),
1699                 stIter->first.dir,
1700                 d.c_str(),
1701                 u.c_str(),
1702                 stIter->second.cash) < 0)
1703         {
1704         STG_LOCKER lock(&mutex);
1705         errorStr = std::string("fprint failed. Message: '");
1706         errorStr += strerror(errno);
1707         errorStr += "'";
1708         printfd(__FILE__, "FILES_STORE::WriteDetailStat - fprintf failed. Message: '%s'\n", strerror(errno));
1709         fclose(statFile);
1710         return -1;
1711         }
1712     #endif
1713
1714     ++stIter;
1715     }
1716
1717 fclose(statFile);
1718
1719 e = chown(fn, storeSettings.GetStatUID(), storeSettings.GetStatGID());
1720 e += chmod(fn, storeSettings.GetStatMode());
1721
1722 if (e)
1723     {
1724     STG_LOCKER lock(&mutex);
1725     printfd(__FILE__, "FILES_STORE::WriteDetailStat - chmod/chown failed for user '%s'. Error: '%s'\n", login.c_str(), strerror(errno));
1726     }
1727
1728 return 0;
1729 }
1730 //-----------------------------------------------------------------------------
1731 int FILES_STORE::AddMessage(STG_MSG * msg, const std::string & login) const
1732 {
1733 std::string fn;
1734 std::string dn;
1735 struct timeval tv;
1736
1737 strprintf(&dn, "%s/%s/messages", storeSettings.GetUsersDir().c_str(), login.c_str());
1738 if (access(dn.c_str(), F_OK) != 0)
1739     {
1740     if (mkdir(dn.c_str(), 0700) != 0)
1741         {
1742         STG_LOCKER lock(&mutex);
1743         errorStr = "Directory \'";
1744         errorStr += dn;
1745         errorStr += "\' cannot be created.";
1746         printfd(__FILE__, "FILES_STORE::AddMessage - mkdir failed. Message: '%s'\n", strerror(errno));
1747         return -1;
1748         }
1749     }
1750
1751 chmod(dn.c_str(), storeSettings.GetConfModeDir());
1752
1753 gettimeofday(&tv, NULL);
1754
1755 msg->header.id = ((long long)tv.tv_sec) * 1000000 + ((long long)tv.tv_usec);
1756 strprintf(&fn, "%s/%lld", dn.c_str(), msg->header.id);
1757
1758 if (Touch(fn))
1759     {
1760     STG_LOCKER lock(&mutex);
1761     errorStr = "File \'";
1762     errorStr += fn;
1763     errorStr += "\' cannot be writen.";
1764     printfd(__FILE__, "FILES_STORE::AddMessage - fopen failed. Message: '%s'\n", strerror(errno));
1765     return -1;
1766     }
1767
1768 return EditMessage(*msg, login);
1769 }
1770 //-----------------------------------------------------------------------------
1771 int FILES_STORE::EditMessage(const STG_MSG & msg, const std::string & login) const
1772 {
1773 std::string fileName;
1774
1775 FILE * msgFile;
1776 strprintf(&fileName, "%s/%s/messages/%lld", storeSettings.GetUsersDir().c_str(), login.c_str(), msg.header.id);
1777
1778 if (access(fileName.c_str(), F_OK) != 0)
1779     {
1780     std::string idstr;
1781     x2str(msg.header.id, idstr);
1782     STG_LOCKER lock(&mutex);
1783     errorStr = "Message for user \'";
1784     errorStr += login + "\' with ID \'";
1785     errorStr += idstr + "\' does not exist.";
1786     printfd(__FILE__, "FILES_STORE::EditMessage - %s\n", errorStr.c_str());
1787     return -1;
1788     }
1789
1790 Touch(fileName + ".new");
1791
1792 msgFile = fopen((fileName + ".new").c_str(), "wt");
1793 if (!msgFile)
1794     {
1795     STG_LOCKER lock(&mutex);
1796     errorStr = "File \'" + fileName + "\' cannot be writen.";
1797     printfd(__FILE__, "FILES_STORE::EditMessage - fopen failed. Message: '%s'\n", strerror(errno));
1798     return -1;
1799     }
1800
1801 bool res = true;
1802 res &= (fprintf(msgFile, "%u\n", msg.header.type) >= 0);
1803 res &= (fprintf(msgFile, "%u\n", msg.header.lastSendTime) >= 0);
1804 res &= (fprintf(msgFile, "%u\n", msg.header.creationTime) >= 0);
1805 res &= (fprintf(msgFile, "%u\n", msg.header.showTime) >= 0);
1806 res &= (fprintf(msgFile, "%d\n", msg.header.repeat) >= 0);
1807 res &= (fprintf(msgFile, "%u\n", msg.header.repeatPeriod) >= 0);
1808 res &= (fprintf(msgFile, "%s", msg.text.c_str()) >= 0);
1809
1810 if (!res)
1811     {
1812     STG_LOCKER lock(&mutex);
1813     errorStr = std::string("fprintf failed. Message: '") + strerror(errno) + "'";
1814     printfd(__FILE__, "FILES_STORE::EditMessage - fprintf failed. Message: '%s'\n", strerror(errno));
1815     fclose(msgFile);
1816     return -1;
1817     }
1818
1819 fclose(msgFile);
1820
1821 chmod((fileName + ".new").c_str(), storeSettings.GetConfMode());
1822
1823 if (rename((fileName + ".new").c_str(), fileName.c_str()) < 0)
1824     {
1825     STG_LOCKER lock(&mutex);
1826     errorStr = "Error moving dir from " + fileName + ".new to " + fileName;
1827     printfd(__FILE__, "FILES_STORE::EditMessage - rename failed. Message: '%s'\n", strerror(errno));
1828     return -1;
1829     }
1830
1831 return 0;
1832 }
1833 //-----------------------------------------------------------------------------
1834 int FILES_STORE::GetMessage(uint64_t id, STG_MSG * msg, const std::string & login) const
1835 {
1836 std::string fn;
1837 strprintf(&fn, "%s/%s/messages/%lld", storeSettings.GetUsersDir().c_str(), login.c_str(), id);
1838 msg->header.id = id;
1839 return ReadMessage(fn, &msg->header, &msg->text);
1840 }
1841 //-----------------------------------------------------------------------------
1842 int FILES_STORE::DelMessage(uint64_t id, const std::string & login) const
1843 {
1844 std::string fn;
1845 strprintf(&fn, "%s/%s/messages/%lld", storeSettings.GetUsersDir().c_str(), login.c_str(), id);
1846
1847 return unlink(fn.c_str());
1848 }
1849 //-----------------------------------------------------------------------------
1850 int FILES_STORE::GetMessageHdrs(std::vector<STG_MSG_HDR> * hdrsList, const std::string & login) const
1851 {
1852 std::string dn(storeSettings.GetUsersDir() + "/" + login + "/messages/");
1853
1854 if (access(dn.c_str(), F_OK) != 0)
1855     {
1856     return 0;
1857     }
1858
1859 std::vector<std::string> messages;
1860 GetFileList(&messages, dn, S_IFREG, "");
1861
1862 for (unsigned i = 0; i < messages.size(); i++)
1863     {
1864     unsigned long long id = 0;
1865
1866     if (str2x(messages[i].c_str(), id))
1867         {
1868         if (unlink((dn + messages[i]).c_str()))
1869             {
1870             STG_LOCKER lock(&mutex);
1871             errorStr = std::string("unlink failed. Message: '") + strerror(errno) + "'";
1872             printfd(__FILE__, "FILES_STORE::GetMessageHdrs - unlink failed. Message: '%s'\n", strerror(errno));
1873             return -1;
1874             }
1875         continue;
1876         }
1877
1878     STG_MSG_HDR hdr;
1879     if (ReadMessage(dn + messages[i], &hdr, NULL))
1880         {
1881         return -1;
1882         }
1883
1884     if (hdr.repeat < 0)
1885         {
1886         if (unlink((dn + messages[i]).c_str()))
1887             {
1888             STG_LOCKER lock(&mutex);
1889             errorStr = std::string("unlink failed. Message: '") + strerror(errno) + "'";
1890             printfd(__FILE__, "FILES_STORE::GetMessageHdrs - unlink failed. Message: '%s'\n", strerror(errno));
1891             return -1;
1892             }
1893         continue;
1894         }
1895
1896     hdr.id = id;
1897     hdrsList->push_back(hdr);
1898     }
1899 return 0;
1900 }
1901 //-----------------------------------------------------------------------------
1902 int FILES_STORE::ReadMessage(const std::string & fileName,
1903                              STG_MSG_HDR * hdr,
1904                              std::string * text) const
1905 {
1906 FILE * msgFile;
1907 msgFile = fopen(fileName.c_str(), "rt");
1908 if (!msgFile)
1909     {
1910     STG_LOCKER lock(&mutex);
1911     errorStr = "File \'";
1912     errorStr += fileName;
1913     errorStr += "\' cannot be openned.";
1914     printfd(__FILE__, "FILES_STORE::ReadMessage - fopen failed. Message: '%s'\n", strerror(errno));
1915     return -1;
1916     }
1917 char p[20];
1918 unsigned * d[6];
1919 d[0] = &hdr->type;
1920 d[1] = &hdr->lastSendTime;
1921 d[2] = &hdr->creationTime;
1922 d[3] = &hdr->showTime;
1923 d[4] = (unsigned*)(&hdr->repeat);
1924 d[5] = &hdr->repeatPeriod;
1925
1926 memset(p, 0, sizeof(p));
1927
1928 for (int pos = 0; pos < 6; pos++)
1929     {
1930     if (fgets(p, sizeof(p) - 1, msgFile) == NULL) {
1931         STG_LOCKER lock(&mutex);
1932         errorStr = "Cannot read file \'";
1933         errorStr += fileName;
1934         errorStr += "\'. Missing data.";
1935         printfd(__FILE__, "FILES_STORE::ReadMessage - cannot read file (missing data)\n");
1936         printfd(__FILE__, "FILES_STORE::ReadMessage - position: %d\n", pos);
1937         fclose(msgFile);
1938         return -1;
1939     }
1940
1941     char * ep;
1942     ep = strrchr(p, '\r');
1943     if (ep) *ep = 0;
1944     ep = strrchr(p, '\n');
1945     if (ep) *ep = 0;
1946
1947     if (feof(msgFile))
1948         {
1949         STG_LOCKER lock(&mutex);
1950         errorStr = "Cannot read file \'";
1951         errorStr += fileName;
1952         errorStr += "\'. Missing data.";
1953         printfd(__FILE__, "FILES_STORE::ReadMessage - cannot read file (feof)\n");
1954         printfd(__FILE__, "FILES_STORE::ReadMessage - position: %d\n", pos);
1955         fclose(msgFile);
1956         return -1;
1957         }
1958
1959     if (str2x(p, *(d[pos])))
1960         {
1961         STG_LOCKER lock(&mutex);
1962         errorStr = "Cannot read file \'";
1963         errorStr += fileName;
1964         errorStr += "\'. Incorrect value. \'";
1965         errorStr += p;
1966         errorStr += "\'";
1967         printfd(__FILE__, "FILES_STORE::ReadMessage - incorrect value\n");
1968         fclose(msgFile);
1969         return -1;
1970         }
1971     }
1972
1973 char txt[2048];
1974 memset(txt, 0, sizeof(txt));
1975 if (text)
1976     {
1977     text->erase(text->begin(), text->end());
1978     while (!feof(msgFile))
1979         {
1980         txt[0] = 0;
1981         if (fgets(txt, sizeof(txt) - 1, msgFile) == NULL) {
1982             break;
1983         }
1984
1985         (*text) += txt;
1986         }
1987     }
1988 fclose(msgFile);
1989 return 0;
1990 }
1991 //-----------------------------------------------------------------------------
1992 int FILES_STORE::Touch(const std::string & path) const
1993 {
1994 FILE * f = fopen(path.c_str(), "wb");
1995 if (f)
1996     {
1997     fclose(f);
1998     return 0;
1999     }
2000 return -1;
2001 }
2002 //-----------------------------------------------------------------------------
2003 int GetFileList(std::vector<std::string> * fileList, const std::string & directory, mode_t mode, const std::string & ext)
2004 {
2005 DIR * d = opendir(directory.c_str());
2006
2007 if (!d)
2008     {
2009     printfd(__FILE__, "GetFileList - Failed to open dir '%s': '%s'\n", directory.c_str(), strerror(errno));
2010     return -1;
2011     }
2012
2013 dirent * entry;
2014 while ((entry = readdir(d)))
2015     {
2016     if (!(strcmp(entry->d_name, ".") && strcmp(entry->d_name, "..")))
2017         continue;
2018
2019     std::string str = directory + "/" + std::string(entry->d_name);
2020
2021     struct stat st;
2022     if (stat(str.c_str(), &st))
2023         continue;
2024
2025     if (!(st.st_mode & mode)) // Filter by mode
2026         continue;
2027
2028     if (!ext.empty())
2029         {
2030         // Check extension
2031         size_t d_nameLen = strlen(entry->d_name);
2032         if (d_nameLen <= ext.size())
2033             continue;
2034
2035         if (ext == entry->d_name + (d_nameLen - ext.size()))
2036             {
2037             entry->d_name[d_nameLen - ext.size()] = 0;
2038             fileList->push_back(entry->d_name);
2039             }
2040         }
2041     else
2042         {
2043         fileList->push_back(entry->d_name);
2044         }
2045     }
2046
2047 closedir(d);
2048
2049 return 0;
2050 }
2051 //-----------------------------------------------------------------------------