]> git.stg.codes - stg.git/blob - projects/stargazer/plugins/store/files/file_store.cpp
GetPlugin should be extern "C".
[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 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 //-----------------------------------------------------------------------------