projects/stargazer/plugins/store/postgresql/postgresql_store_users.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 : Maxim Mamontov <faust@stargazer.dp.ua>
  19  */
  20
  21 /*
  22  *  User manipulation methods
  23  *
  24  *  $Revision: 1.14 $
  25  *  $Date: 2010/05/07 07:26:36 $
  26  *
  27  */
  28
  29 #include <string>
  30 #include <vector>
  31 #include <sstream>
  32 #include <ctime>
  33
  34 #include <libpq-fe.h>
  35
  36 #include "stg/common.h"
  37 #include "stg/const.h"
  38 #include "stg/locker.h"
  39 #include "../../../stg_timer.h"
  40 #include "postgresql_store.h"
  41
  42 //-----------------------------------------------------------------------------
  43 int POSTGRESQL_STORE::GetUsersList(std::vector<std::string> * usersList) const
  44 {
  45 STG_LOCKER lock(&mutex);
  46
  47 if (PQstatus(connection) != CONNECTION_OK)
  48     {
  49     printfd(__FILE__, "POSTGRESQL_STORE::GetUsersList(): 'Connection lost. Trying to reconnect...'\n", strError.c_str());
  50     if (Reset())
  51         {
  52         strError = "Connection lost";
  53         printfd(__FILE__, "POSTGRESQL_STORE::GetUsersList(): '%s'\n", strError.c_str());
  54         return -1;
  55         }
  56     }
  57
  58 PGresult * result;
  59
  60 if (StartTransaction())
  61     {
  62     printfd(__FILE__, "POSTGRESQL_STORE::GetUsersList(): 'Failed to start transaction'\n");
  63     return -1;
  64     }
  65
  66 result = PQexec(connection, "SELECT name FROM tb_users");
  67
  68 if (PQresultStatus(result) != PGRES_TUPLES_OK)
  69     {
  70     strError = PQresultErrorMessage(result);
  71     PQclear(result);
  72     printfd(__FILE__, "POSTGRESQL_STORE::GetUsersList(): '%s'\n", strError.c_str());
  73     if (RollbackTransaction())
  74         {
  75         printfd(__FILE__, "POSTGRESQL_STORE::GetUsersList(): 'Failed to rollback transaction'\n");
  76         }
  77     return -1;
  78     }
  79
  80 int tuples = PQntuples(result);
  81
  82 for (int i = 0; i < tuples; ++i)
  83     {
  84     usersList->push_back(PQgetvalue(result, i, 0));
  85     }
  86
  87 PQclear(result);
  88
  89 if (CommitTransaction())
  90     {
  91     printfd(__FILE__, "POSTGRESQL_STORE::GetUsersList(): 'Failed to commit transaction'\n");
  92     return -1;
  93     }
  94
  95 return 0;
  96 }
  97
  98 //-----------------------------------------------------------------------------
  99 int POSTGRESQL_STORE::AddUser(const std::string & name) const
 100 {
 101 STG_LOCKER lock(&mutex);
 102
 103 if (PQstatus(connection) != CONNECTION_OK)
 104     {
 105     printfd(__FILE__, "POSTGRESQL_STORE::AddUser(): 'Connection lost. Trying to reconnect...'\n", strError.c_str());
 106     if (Reset())
 107         {
 108         strError = "Connection lost";
 109         printfd(__FILE__, "POSTGRESQL_STORE::AddUser(): '%s'\n", strError.c_str());
 110         return -1;
 111         }
 112     }
 113
 114 PGresult * result;
 115
 116 if (StartTransaction())
 117     {
 118     printfd(__FILE__, "POSTGRESQL_STORE::AddUser(): 'Failed to start transaction'\n");
 119     return -1;
 120     }
 121
 122 std::string elogin = name;
 123
 124 if (EscapeString(elogin))
 125     {
 126     printfd(__FILE__, "POSTGRESQL_STORE::AddUser(): 'Failed to escape login'\n");
 127     if (RollbackTransaction())
 128         {
 129         printfd(__FILE__, "POSTGRESQL_STORE::AddUser(): 'Failed to rollback transaction'\n");
 130         }
 131     return -1;
 132     }
 133
 134 std::ostringstream query;
 135 query << "SELECT sp_add_user('" << elogin << "')";
 136
 137 result = PQexec(connection, query.str().c_str());
 138
 139 if (PQresultStatus(result) != PGRES_TUPLES_OK)
 140     {
 141     strError = PQresultErrorMessage(result);
 142     PQclear(result);
 143     printfd(__FILE__, "POSTGRESQL_STORE::AddUser(): '%s'\n", strError.c_str());
 144     if (RollbackTransaction())
 145         {
 146         printfd(__FILE__, "POSTGRESQL_STORE::AddUser(): 'Failed to rollback transaction'\n");
 147         }
 148     return -1;
 149     }
 150
 151 PQclear(result);
 152
 153 if (CommitTransaction())
 154     {
 155     printfd(__FILE__, "POSTGRESQL_STORE::AddUser(): 'Failed to commit transaction'\n");
 156     return -1;
 157     }
 158
 159 return 0;
 160 }
 161
 162 //-----------------------------------------------------------------------------
 163 int POSTGRESQL_STORE::DelUser(const std::string & login) const
 164 {
 165 STG_LOCKER lock(&mutex);
 166
 167 if (PQstatus(connection) != CONNECTION_OK)
 168     {
 169     printfd(__FILE__, "POSTGRESQL_STORE::DelUser(): 'Connection lost. Trying to reconnect...'\n", strError.c_str());
 170     if (Reset())
 171         {
 172         strError = "Connection lost";
 173         printfd(__FILE__, "POSTGRESQL_STORE::DelUser(): '%s'\n", strError.c_str());
 174         return -1;
 175         }
 176     }
 177
 178 PGresult * result;
 179
 180 if (StartTransaction())
 181     {
 182     printfd(__FILE__, "POSTGRESQL_STORE::DelUser(): 'Failed to start transaction'\n");
 183     return -1;
 184     }
 185
 186 std::string elogin = login;
 187
 188 if (EscapeString(elogin))
 189     {
 190     printfd(__FILE__, "POSTGRESQL_STORE::DelUser(): 'Failed to escape login'\n");
 191     if (RollbackTransaction())
 192         {
 193         printfd(__FILE__, "POSTGRESQL_STORE::DelUser(): 'Failed to rollback transaction'\n");
 194         }
 195     return -1;
 196     }
 197
 198 std::ostringstream query;
 199 query << "DELETE FROM tb_users WHERE name = '" << elogin << "'";
 200
 201 result = PQexec(connection, query.str().c_str());
 202
 203 if (PQresultStatus(result) != PGRES_COMMAND_OK)
 204     {
 205     strError = PQresultErrorMessage(result);
 206     PQclear(result);
 207     printfd(__FILE__, "POSTGRESQL_STORE::DelUser(): '%s'\n", strError.c_str());
 208     if (RollbackTransaction())
 209         {
 210         printfd(__FILE__, "POSTGRESQL_STORE::DelUser(): 'Failed to rollback transaction'\n");
 211         }
 212     return -1;
 213     }
 214
 215 PQclear(result);
 216
 217 if (CommitTransaction())
 218     {
 219     printfd(__FILE__, "POSTGRESQL_STORE::DelUser(): 'Failed to commit transaction'\n");
 220     return -1;
 221     }
 222
 223 return 0;
 224 }
 225 //-----------------------------------------------------------------------------
 226 int POSTGRESQL_STORE::SaveUserStat(const USER_STAT & stat,
 227                                    const std::string & login) const
 228 {
 229 STG_LOCKER lock(&mutex);
 230
 231 return SaveStat(stat, login);
 232 }
 233 //-----------------------------------------------------------------------------
 234 int POSTGRESQL_STORE::SaveStat(const USER_STAT & stat,
 235                                const std::string & login,
 236                                int year,
 237                                int month) const
 238 {
 239 if (PQstatus(connection) != CONNECTION_OK)
 240     {
 241     printfd(__FILE__, "POSTGRESQL_STORE::SaveStat(): 'Connection lost. Trying to reconnect...'\n", strError.c_str());
 242     if (Reset())
 243         {
 244         strError = "Connection lost";
 245         printfd(__FILE__, "POSTGRESQL_STORE::SaveStat(): '%s'\n", strError.c_str());
 246         return -1;
 247         }
 248     }
 249
 250 PGresult * result;
 251
 252 if (StartTransaction())
 253     {
 254     printfd(__FILE__, "POSTGRESQL_STORE::SaveStat(): 'Failed to start transaction'\n");
 255     return -1;
 256     }
 257
 258 std::string elogin = login;
 259
 260 if (EscapeString(elogin))
 261     {
 262     printfd(__FILE__, "POSTGRESQL_STORE::SaveStat(): 'Failed to escape login'\n");
 263     if (RollbackTransaction())
 264         {
 265         printfd(__FILE__, "POSTGRESQL_STORE::SaveStat(): 'Failed to rollback transaction'\n");
 266         }
 267     return -1;
 268     }
 269
 270 std::ostringstream query;
 271 query << "UPDATE tb_users SET "
 272             "cash = " << stat.cash << ", "
 273             "free_mb = " << stat.freeMb << ", "
 274             "last_activity_time = CAST('" << formatTime(stat.lastActivityTime) << "' AS TIMESTAMP), "
 275             "last_cash_add = " << stat.lastCashAdd << ", "
 276             "last_cash_add_time = CAST('" << formatTime(stat.lastCashAddTime) << "' AS TIMESTAMP), "
 277             "passive_time = " << stat.passiveTime << " "
 278          "WHERE name = '" << elogin << "'";
 279
 280 result = PQexec(connection, query.str().c_str());
 281
 282 if (PQresultStatus(result) != PGRES_COMMAND_OK)
 283     {
 284     strError = PQresultErrorMessage(result);
 285     PQclear(result);
 286     printfd(__FILE__, "POSTGRESQL_STORE::SaveStat(): '%s'\n", strError.c_str());
 287     if (RollbackTransaction())
 288         {
 289         printfd(__FILE__, "POSTGRESQL_STORE::SaveStat(): 'Failed to rollback transaction'\n");
 290         }
 291     return -1;
 292     }
 293
 294 PQclear(result);
 295
 296 std::string date;
 297
 298 MakeDate(date, year, month);
 299
 300 for (int dir = 0; dir < DIR_NUM; ++dir)
 301     {
 302     query.str("");
 303     query << "SELECT sp_add_stats_traffic ("
 304                 "'" << elogin << "', "
 305                 "CAST('" << date << "' AS DATE), "
 306                 "CAST(" << dir << " AS SMALLINT), "
 307                 "CAST(" << stat.monthUp[dir] << " AS BIGINT), "
 308                 "CAST(" << stat.monthDown[dir] << " AS BIGINT))";
 309
 310     result = PQexec(connection, query.str().c_str());
 311
 312     if (PQresultStatus(result) != PGRES_TUPLES_OK)
 313         {
 314         strError = PQresultErrorMessage(result);
 315         PQclear(result);
 316         printfd(__FILE__, "POSTGRESQL_STORE::SaveStat(): '%s'\n", strError.c_str());
 317         if (RollbackTransaction())
 318             {
 319             printfd(__FILE__, "POSTGRESQL_STORE::SaveStat(): 'Failed to rollback transaction'\n");
 320             }
 321         return -1;
 322         }
 323
 324     PQclear(result);
 325     }
 326
 327 if (CommitTransaction())
 328     {
 329     printfd(__FILE__, "POSTGRESQL_STORE::SaveStat(): 'Failed to commit transaction'\n");
 330     return -1;
 331     }
 332
 333 return 0;
 334 }
 335
 336 //-----------------------------------------------------------------------------
 337 int POSTGRESQL_STORE::SaveUserConf(const USER_CONF & conf,
 338                                  const std::string & login) const
 339 {
 340 STG_LOCKER lock(&mutex);
 341
 342 if (PQstatus(connection) != CONNECTION_OK)
 343     {
 344     printfd(__FILE__, "POSTGRESQL_STORE::SaveUserConf(): 'Connection lost. Trying to reconnect...'\n", strError.c_str());
 345     if (Reset())
 346         {
 347         strError = "Connection lost";
 348         printfd(__FILE__, "POSTGRESQL_STORE::SaveUserConf(): '%s'\n", strError.c_str());
 349         return -1;
 350         }
 351     }
 352
 353 PGresult * result;
 354
 355 if (StartTransaction())
 356     {
 357     printfd(__FILE__, "POSTGRESQL_STORE::SaveUserConf(): 'Failed to start transaction'\n");
 358     return -1;
 359     }
 360
 361 std::string elogin = login;
 362
 363 if (EscapeString(elogin))
 364     {
 365     printfd(__FILE__, "POSTGRESQL_STORE::SaveUserConf(): 'Failed to escape login'\n");
 366     if (RollbackTransaction())
 367         {
 368         printfd(__FILE__, "POSTGRESQL_STORE::SaveUserConf(): 'Failed to rollback transaction'\n");
 369         }
 370     return -1;
 371     }
 372
 373 std::ostringstream query;
 374 query << "SELECT pk_user FROM tb_users WHERE name = '" << elogin << "'";
 375
 376 result = PQexec(connection, query.str().c_str());
 377
 378 if (PQresultStatus(result) != PGRES_TUPLES_OK)
 379     {
 380     strError = PQresultErrorMessage(result);
 381     PQclear(result);
 382     printfd(__FILE__, "POSTGRESQL_STORE::SaveUserConf(): '%s'\n", strError.c_str());
 383     if (RollbackTransaction())
 384         {
 385         printfd(__FILE__, "POSTGRESQL_STORE::SaveUserConf(): 'Failed to rollback transaction'\n");
 386         }
 387     return -1;
 388     }
 389
 390 int tuples = PQntuples(result);
 391
 392 if (tuples != 1)
 393     {
 394     strError = "Failed to fetch user's ID";
 395     printfd(__FILE__, "POSTGRESQL_STORE::SaveUserConf(): 'Invalid number of tuples. Wanted 1, actulally %d'\n", tuples);
 396     PQclear(result);
 397     if (RollbackTransaction())
 398         {
 399         printfd(__FILE__, "POSTGRESQL_STORE::SaveUserConf(): 'Failed to rollback transaction'\n");
 400         }
 401     return -1;
 402     }
 403
 404 uint32_t uid;
 405
 406     {
 407     std::stringstream tuple;
 408     tuple << PQgetvalue(result, 0, 0);
 409
 410     PQclear(result);
 411
 412     tuple >> uid;
 413     }
 414
 415 std::string eaddress = conf.address;
 416 std::string eemail = conf.email;
 417 std::string egroup = conf.group;
 418 std::string enote = conf.note;
 419 std::string epassword = conf.password;
 420 std::string ephone = conf.phone;
 421 std::string erealname = conf.realName;
 422 std::string etariffname = conf.tariffName;
 423 std::string enexttariff = conf.nextTariff;
 424 std::string ecorporation = conf.corp;
 425
 426 if (EscapeString(eaddress))
 427     {
 428     printfd(__FILE__, "POSTGRESQL_STORE::SaveUserConf(): 'Failed to escape address'\n");
 429     if (RollbackTransaction())
 430         {
 431         printfd(__FILE__, "POSTGRESQL_STORE::SaveUserConf(): 'Failed to rollback transaction'\n");
 432         }
 433     return -1;
 434     }
 435
 436 if (EscapeString(eemail))
 437     {
 438     printfd(__FILE__, "POSTGRESQL_STORE::SaveUserConf(): 'Failed to escape email'\n");
 439     if (RollbackTransaction())
 440         {
 441         printfd(__FILE__, "POSTGRESQL_STORE::SaveUserConf(): 'Failed to rollback transaction'\n");
 442         }
 443     return -1;
 444     }
 445
 446 if (EscapeString(egroup))
 447     {
 448     printfd(__FILE__, "POSTGRESQL_STORE::SaveUserConf(): 'Failed to escape group'\n");
 449     if (RollbackTransaction())
 450         {
 451         printfd(__FILE__, "POSTGRESQL_STORE::SaveUserConf(): 'Failed to rollback transaction'\n");
 452         }
 453     return -1;
 454     }
 455
 456 if (EscapeString(enote))
 457     {
 458     printfd(__FILE__, "POSTGRESQL_STORE::SaveUserConf(): 'Failed to escape note'\n");
 459     if (RollbackTransaction())
 460         {
 461         printfd(__FILE__, "POSTGRESQL_STORE::SaveUserConf(): 'Failed to rollback transaction'\n");
 462         }
 463     return -1;
 464     }
 465
 466 if (EscapeString(epassword))
 467     {
 468     printfd(__FILE__, "POSTGRESQL_STORE::SaveUserConf(): 'Failed to escape password'\n");
 469     if (RollbackTransaction())
 470         {
 471         printfd(__FILE__, "POSTGRESQL_STORE::SaveUserConf(): 'Failed to rollback transaction'\n");
 472         }
 473     return -1;
 474     }
 475
 476 if (EscapeString(ephone))
 477     {
 478     printfd(__FILE__, "POSTGRESQL_STORE::SaveUserConf(): 'Failed to escape phone'\n");
 479     if (RollbackTransaction())
 480         {
 481         printfd(__FILE__, "POSTGRESQL_STORE::SaveUserConf(): 'Failed to rollback transaction'\n");
 482         }
 483     return -1;
 484     }
 485
 486 if (EscapeString(erealname))
 487     {
 488     printfd(__FILE__, "POSTGRESQL_STORE::SaveUserConf(): 'Failed to escape real name'\n");
 489     if (RollbackTransaction())
 490         {
 491         printfd(__FILE__, "POSTGRESQL_STORE::SaveUserConf(): 'Failed to rollback transaction'\n");
 492         }
 493     return -1;
 494     }
 495
 496 if (EscapeString(etariffname))
 497     {
 498     printfd(__FILE__, "POSTGRESQL_STORE::SaveUserConf(): 'Failed to escape tariff name'\n");
 499     if (RollbackTransaction())
 500         {
 501         printfd(__FILE__, "POSTGRESQL_STORE::SaveUserConf(): 'Failed to rollback transaction'\n");
 502         }
 503     return -1;
 504     }
 505
 506 if (EscapeString(enexttariff))
 507     {
 508     printfd(__FILE__, "POSTGRESQL_STORE::SaveUserConf(): 'Failed to escape next tariff name'\n");
 509     if (RollbackTransaction())
 510         {
 511         printfd(__FILE__, "POSTGRESQL_STORE::SaveUserConf(): 'Failed to rollback transaction'\n");
 512         }
 513     return -1;
 514     }
 515
 516 if (EscapeString(ecorporation))
 517     {
 518     printfd(__FILE__, "POSTGRESQL_STORE::SaveUserConf(): 'Failed to escape corporation name'\n");
 519     if (RollbackTransaction())
 520         {
 521         printfd(__FILE__, "POSTGRESQL_STORE::SaveUserConf(): 'Failed to rollback transaction'\n");
 522         }
 523     return -1;
 524     }
 525
 526 query.str("");
 527 query << "UPDATE tb_users SET "
 528              "address = '" << eaddress << "', "
 529              "always_online = " << (conf.alwaysOnline ? "'t'" : "'f'") << ", "
 530              "credit = " << conf.credit << ", "
 531              "credit_expire = CAST('" << formatTime(conf.creditExpire) << "' AS TIMESTAMP), "
 532              "disabled = " << (conf.disabled ? "'t'" : "'f'") << ", "
 533              "disabled_detail_stat = " << (conf.disabledDetailStat ? "'t'" : "'f'") << ", "
 534              "email = '" << eemail << "', "
 535              "grp = '" << egroup << "', "
 536              "note = '" << enote << "', "
 537              "passive = " << (conf.passive ? "'t'" : "'f'") << ", "
 538              "passwd = '" << epassword << "', "
 539              "phone = '" << ephone << "', "
 540              "real_name = '" << erealname << "', "
 541              "fk_tariff = (SELECT pk_tariff "
 542                    "FROM tb_tariffs "
 543                    "WHERE name = '" << etariffname << "'), "
 544              "fk_tariff_change = (SELECT pk_tariff "
 545                    "FROM tb_tariffs "
 546                    "WHERE name = '" << enexttariff << "'), "
 547              "fk_corporation = (SELECT pk_corporation "
 548                    "FROM tb_corporations "
 549                    "WHERE name = '" << ecorporation << "') "
 550          "WHERE pk_user = " << uid;
 551
 552 result = PQexec(connection, query.str().c_str());
 553
 554 if (PQresultStatus(result) != PGRES_COMMAND_OK)
 555     {
 556     strError = PQresultErrorMessage(result);
 557     PQclear(result);
 558     printfd(__FILE__, "POSTGRESQL_STORE::SaveUserConf(): '%s'\n", strError.c_str());
 559     if (RollbackTransaction())
 560         {
 561         printfd(__FILE__, "POSTGRESQL_STORE::SaveUserConf(): 'Failed to rollback transaction'\n");
 562         }
 563     return -1;
 564     }
 565
 566 PQclear(result);
 567
 568 if (SaveUserServices(uid, conf.services))
 569     {
 570     printfd(__FILE__, "POSTGRESQL_STORE::SaveUserConf(): 'Failed to save user's services'\n");
 571     if (RollbackTransaction())
 572         {
 573         printfd(__FILE__, "POSTGRESQL_STORE::SaveUserConf(): 'Failed to rollback transaction'\n");
 574         }
 575     return -1;
 576     }
 577
 578 if (SaveUserData(uid, conf.userdata))
 579     {
 580     printfd(__FILE__, "POSTGRESQL_STORE::SaveUserConf(): 'Failed to save user's data'\n");
 581     if (RollbackTransaction())
 582         {
 583         printfd(__FILE__, "POSTGRESQL_STORE::SaveUserConf(): 'Failed to rollback transaction'\n");
 584         }
 585     return -1;
 586     }
 587
 588 if (SaveUserIPs(uid, conf.ips))
 589     {
 590     printfd(__FILE__, "POSTGRESQL_STORE::SaveUserConf(): 'Failed to save user's IPs'\n");
 591     if (RollbackTransaction())
 592         {
 593         printfd(__FILE__, "POSTGRESQL_STORE::SaveUserConf(): 'Failed to rollback transaction'\n");
 594         }
 595     return -1;
 596     }
 597
 598 if (CommitTransaction())
 599     {
 600     printfd(__FILE__, "POSTGRESQL_STORE::SaveUserConf(): 'Failed to commit transaction'\n");
 601     return -1;
 602     }
 603
 604 return 0;
 605 }
 606
 607 //-----------------------------------------------------------------------------
 608 int POSTGRESQL_STORE::RestoreUserStat(USER_STAT * stat,
 609                                     const std::string & login) const
 610 {
 611 STG_LOCKER lock(&mutex);
 612
 613 if (PQstatus(connection) != CONNECTION_OK)
 614     {
 615     printfd(__FILE__, "POSTGRESQL_STORE::RestoreUserStat(): 'Connection lost. Trying to reconnect...'\n", strError.c_str());
 616     if (Reset())
 617         {
 618         strError = "Connection lost";
 619         printfd(__FILE__, "POSTGRESQL_STORE::RestoreUserStat(): '%s'\n", strError.c_str());
 620         return -1;
 621         }
 622     }
 623
 624 PGresult * result;
 625
 626 if (StartTransaction())
 627     {
 628     printfd(__FILE__, "POSTGRESQL_STORE::RestoreUserStat(): 'Failed to start transaction'\n");
 629     return -1;
 630     }
 631
 632 std::string elogin = login;
 633
 634 if (EscapeString(elogin))
 635     {
 636     printfd(__FILE__, "POSTGRESQL_STORE::RestoreUserStat(): 'Failed to escape login'\n");
 637     if (RollbackTransaction())
 638         {
 639         printfd(__FILE__, "POSTGRESQL_STORE::RestoreUserStat(): 'Failed to rollback transaction'\n");
 640         }
 641     return -1;
 642     }
 643
 644     {
 645     std::ostringstream query;
 646     query << "SELECT cash, free_mb, "
 647                 "last_activity_time, last_cash_add, "
 648                 "last_cash_add_time, passive_time "
 649              "FROM tb_users "
 650              "WHERE name = '" << elogin << "'";
 651
 652     result = PQexec(connection, query.str().c_str());
 653     }
 654
 655 if (PQresultStatus(result) != PGRES_TUPLES_OK)
 656     {
 657     strError = PQresultErrorMessage(result);
 658     printfd(__FILE__, "POSTGRESQL_STORE::RestoreUserStat(): '%s'\n", strError.c_str());
 659     PQclear(result);
 660     if (RollbackTransaction())
 661         {
 662         printfd(__FILE__, "POSTGRESQL_STORE::RestoreUserStat(): 'Failed to rollback transaction'\n");
 663         }
 664     return -1;
 665     }
 666
 667 int tuples = PQntuples(result);
 668
 669 if (tuples != 1)
 670     {
 671     strError = "Failed to fetch user's stat";
 672     printfd(__FILE__, "POSTGRESQL_STORE::RestoreUserStat(): 'Invalid number of tuples. Wanted 1, actulally %d'\n", tuples);
 673     PQclear(result);
 674     if (RollbackTransaction())
 675         {
 676         printfd(__FILE__, "POSTGRESQL_STORE::RestoreUserStat(): 'Failed to rollback transaction'\n");
 677         }
 678     return -1;
 679     }
 680
 681     {
 682     std::stringstream tuple;
 683     tuple << PQgetvalue(result, 0, 0) << " ";
 684     tuple << PQgetvalue(result, 0, 1) << " ";
 685     stat->lastActivityTime = readTime(PQgetvalue(result, 0, 2));
 686     tuple << PQgetvalue(result, 0, 3) << " ";
 687     stat->lastCashAddTime = readTime(PQgetvalue(result, 0, 4));
 688     tuple << PQgetvalue(result, 0, 5) << " ";
 689
 690     PQclear(result);
 691
 692     tuple >> stat->cash
 693           >> stat->freeMb
 694           >> stat->lastCashAdd
 695           >> stat->passiveTime;
 696     }
 697
 698     {
 699     std::ostringstream query;
 700     query << "SELECT dir_num, upload, download "
 701              "FROM tb_stats_traffic "
 702              "WHERE fk_user IN (SELECT pk_user FROM tb_users WHERE name = '" << elogin << "') AND "
 703                    "DATE_TRUNC('month', stats_date) = DATE_TRUNC('month', CAST('" << formatTime(stgTime) << "' AS TIMESTAMP))";
 704
 705     result = PQexec(connection, query.str().c_str());
 706     }
 707
 708 if (PQresultStatus(result) != PGRES_TUPLES_OK)
 709     {
 710     strError = PQresultErrorMessage(result);
 711     printfd(__FILE__, "POSTGRESQL_STORE::RestoreUserStat(): '%s'\n", strError.c_str());
 712     PQclear(result);
 713     if (RollbackTransaction())
 714         {
 715         printfd(__FILE__, "POSTGRESQL_STORE::RestoreUserStat(): 'Failed to rollback transaction'\n");
 716         }
 717     return -1;
 718     }
 719
 720 tuples = PQntuples(result);
 721
 722 for (int i = 0; i < tuples; ++i)
 723     {
 724     std::stringstream tuple;
 725     tuple << PQgetvalue(result, i, 0) << " ";
 726     tuple << PQgetvalue(result, i, 1) << " ";
 727     tuple << PQgetvalue(result, i, 2) << " ";
 728
 729     int dir;
 730
 731     tuple >> dir;
 732     tuple >> stat->monthUp[dir];
 733     tuple >> stat->monthDown[dir];
 734     }
 735
 736 PQclear(result);
 737
 738 if (CommitTransaction())
 739     {
 740     printfd(__FILE__, "POSTGRESQL_STORE::RestoreUserStat(): 'Failed to commit transaction'\n");
 741     return -1;
 742     }
 743
 744 return 0;
 745 }
 746
 747 //-----------------------------------------------------------------------------
 748 int POSTGRESQL_STORE::RestoreUserConf(USER_CONF * conf,
 749                                     const std::string & login) const
 750 {
 751 STG_LOCKER lock(&mutex);
 752
 753 if (PQstatus(connection) != CONNECTION_OK)
 754     {
 755     printfd(__FILE__, "POSTGRESQL_STORE::RestoreUserConf(): 'Connection lost. Trying to reconnect...'\n", strError.c_str());
 756     if (Reset())
 757         {
 758         strError = "Connection lost";
 759         printfd(__FILE__, "POSTGRESQL_STORE::RestoreUserConf(): '%s'\n", strError.c_str());
 760         return -1;
 761         }
 762     }
 763
 764 PGresult * result;
 765
 766 if (StartTransaction())
 767     {
 768     printfd(__FILE__, "POSTGRESQL_STORE::RestoreUserStat(): 'Failed to start transaction'\n");
 769     return -1;
 770     }
 771
 772 std::string elogin = login;
 773
 774 if (EscapeString(elogin))
 775     {
 776     printfd(__FILE__, "POSTGRESQL_STORE::RestoreUserStat(): 'Failed to escape login'\n");
 777     if (RollbackTransaction())
 778         {
 779         printfd(__FILE__, "POSTGRESQL_STORE::RestoreUserStat(): 'Failed to rollback transaction'\n");
 780         }
 781     return -1;
 782     }
 783
 784     {
 785     std::ostringstream query;
 786     query << "SELECT tb_users.pk_user, tb_users.address, tb_users.always_online, "
 787                     "tb_users.credit, tb_users.credit_expire, tb_users.disabled, "
 788                     "tb_users.disabled_detail_stat, tb_users.email, tb_users.grp, "
 789                     "tb_users.note, tb_users.passive, tb_users.passwd, tb_users.phone, "
 790                     "tb_users.real_name, tf1.name, tf2.name, tb_corporations.name "
 791              "FROM tb_users LEFT JOIN tb_tariffs AS tf1 "
 792                                 "ON tf1.pk_tariff = tb_users.fk_tariff "
 793                            "LEFT JOIN tb_tariffs AS tf2 "
 794                                 "ON tf2.pk_tariff = tb_users.fk_tariff_change "
 795                            "LEFT JOIN tb_corporations "
 796                                 "ON tb_corporations.pk_corporation = tb_users.fk_corporation "
 797              "WHERE tb_users.name = '" << elogin << "'";
 798
 799     result = PQexec(connection, query.str().c_str());
 800     }
 801
 802 if (PQresultStatus(result) != PGRES_TUPLES_OK)
 803     {
 804     strError = PQresultErrorMessage(result);
 805     printfd(__FILE__, "POSTGRESQL_STORE::RestoreUserConf(): '%s'\n", strError.c_str());
 806     PQclear(result);
 807     if (RollbackTransaction())
 808         {
 809         printfd(__FILE__, "POSTGRESQL_STORE::RestoreUserConf(): 'Failed to rollback transaction'\n");
 810         }
 811     return -1;
 812     }
 813
 814 int tuples = PQntuples(result);
 815
 816 if (tuples != 1)
 817     {
 818     strError = "Failed to fetch user's stat";
 819     printfd(__FILE__, "POSTGRESQL_STORE::RestoreUserConf(): 'Invalid number of tuples. Wanted 1, actulally %d'\n", tuples);
 820     PQclear(result);
 821     if (RollbackTransaction())
 822         {
 823         printfd(__FILE__, "POSTGRESQL_STORE::RestoreUserConf(): 'Failed to rollback transaction'\n");
 824         }
 825     return -1;
 826     }
 827
 828 uint32_t uid;
 829
 830     {
 831     std::stringstream tuple;
 832     tuple << PQgetvalue(result, 0, 0) << " ";               // uid
 833     conf->address = PQgetvalue(result, 0, 1);               // address
 834     conf->alwaysOnline = !strncmp(PQgetvalue(result, 0, 2), "t", 1);
 835     tuple << PQgetvalue(result, 0, 3) << " ";               // credit
 836     conf->creditExpire = readTime(PQgetvalue(result, 0, 4));  // creditExpire
 837     conf->disabled = !strncmp(PQgetvalue(result, 0, 5), "t", 1);
 838     conf->disabledDetailStat = !strncmp(PQgetvalue(result, 0, 6), "t", 1);
 839     conf->email = PQgetvalue(result, 0, 7);                 // email
 840     conf->group = PQgetvalue(result, 0, 8);                 // group
 841     conf->note = PQgetvalue(result, 0, 9);                  // note
 842     conf->passive = !strncmp(PQgetvalue(result, 0, 10), "t", 1);
 843     conf->password = PQgetvalue(result, 0, 11);             // password
 844     conf->phone = PQgetvalue(result, 0, 12);                // phone
 845     conf->realName = PQgetvalue(result, 0, 13);             // realName
 846     conf->tariffName = PQgetvalue(result, 0, 14);           // tariffName
 847     conf->nextTariff = PQgetvalue(result, 0, 15);           // nextTariff
 848     conf->corp = PQgetvalue(result, 0, 16);                 // corp
 849
 850     PQclear(result);
 851
 852     if (conf->tariffName == "")
 853         conf->tariffName = NO_TARIFF_NAME;
 854     if (conf->corp == "")
 855         conf->corp = NO_CORP_NAME;
 856
 857     tuple >> uid
 858           >> conf->credit;
 859     }
 860
 861     {
 862     std::ostringstream query;
 863     query << "SELECT name FROM tb_services "
 864              "WHERE pk_service IN (SELECT fk_service "
 865                                   "FROM tb_users_services "
 866                                   "WHERE fk_user = " << uid << ")";
 867
 868     result = PQexec(connection, query.str().c_str());
 869     }
 870
 871 if (PQresultStatus(result) != PGRES_TUPLES_OK)
 872     {
 873     strError = PQresultErrorMessage(result);
 874     printfd(__FILE__, "POSTGRESQL_STORE::RestoreUserConf(): '%s'\n", strError.c_str());
 875     PQclear(result);
 876     if (RollbackTransaction())
 877         {
 878         printfd(__FILE__, "POSTGRESQL_STORE::RestoreUserConf(): 'Failed to rollback transaction'\n");
 879         }
 880     return -1;
 881     }
 882
 883 tuples = PQntuples(result);
 884
 885 for (int i = 0; i < tuples; ++i)
 886     {
 887     conf->services.push_back(PQgetvalue(result, i, 0));
 888     }
 889
 890 PQclear(result);
 891
 892     {
 893     std::ostringstream query;
 894     query << "SELECT num, data "
 895              "FROM tb_users_data "
 896              "WHERE fk_user = " << uid;
 897
 898     result = PQexec(connection, query.str().c_str());
 899     }
 900
 901 if (PQresultStatus(result) != PGRES_TUPLES_OK)
 902     {
 903     strError = PQresultErrorMessage(result);
 904     printfd(__FILE__, "POSTGRESQL_STORE::RestoreUserConf(): '%s'\n", strError.c_str());
 905     PQclear(result);
 906     if (RollbackTransaction())
 907         {
 908         printfd(__FILE__, "POSTGRESQL_STORE::RestoreUserConf(): 'Failed to rollback transaction'\n");
 909         }
 910     return -1;
 911     }
 912
 913 tuples = PQntuples(result);
 914
 915 for (int i = 0; i < tuples; ++i)
 916     {
 917     int num;
 918     if (str2x(PQgetvalue(result, i, 0), num))
 919         {
 920         printfd(__FILE__, "POSTGRESQL_STORE::RestoreUserConf(): 'Failed to convert string to int'\n");
 921         }
 922     else
 923         {
 924         if (num < USERDATA_NUM &&
 925             num >= 0)
 926             {
 927             conf->userdata[num] = PQgetvalue(result, i, 1);
 928             }
 929         }
 930     }
 931
 932 PQclear(result);
 933
 934     {
 935     std::ostringstream query;
 936     query << "SELECT host(ip), masklen(ip) "
 937              "FROM tb_allowed_ip "
 938              "WHERE fk_user = " << uid;
 939
 940     result = PQexec(connection, query.str().c_str());
 941     }
 942
 943 if (PQresultStatus(result) != PGRES_TUPLES_OK)
 944     {
 945     strError = PQresultErrorMessage(result);
 946     printfd(__FILE__, "POSTGRESQL_STORE::RestoreUserConf(): '%s'\n", strError.c_str());
 947     PQclear(result);
 948     if (RollbackTransaction())
 949         {
 950         printfd(__FILE__, "POSTGRESQL_STORE::RestoreUserConf(): 'Failed to rollback transaction'\n");
 951         }
 952     return -1;
 953     }
 954
 955 tuples = PQntuples(result);
 956
 957 USER_IPS ips;
 958 for (int i = 0; i < tuples; ++i)
 959     {
 960     IP_MASK im;
 961
 962     im.ip = inet_strington(PQgetvalue(result, i, 0));
 963
 964     if (str2x(PQgetvalue(result, i, 1), im.mask))
 965         {
 966         printfd(__FILE__, "POSTGRESQL_STORE::RestoreUserConf(): 'Failed to fetch mask'\n");
 967         continue;
 968         }
 969
 970     ips.Add(im);
 971     }
 972 conf->ips = ips;
 973
 974 PQclear(result);
 975
 976 if (CommitTransaction())
 977     {
 978     printfd(__FILE__, "POSTGRESQL_STORE::RestoreUserConf(): 'Failed to commit transaction'\n");
 979     return -1;
 980     }
 981
 982 return 0;
 983 }
 984
 985 //-----------------------------------------------------------------------------
 986 int POSTGRESQL_STORE::WriteUserChgLog(const std::string & login,
 987                                     const std::string & admLogin,
 988                                     uint32_t admIP,
 989                                     const std::string & paramName,
 990                                     const std::string & oldValue,
 991                                     const std::string & newValue,
 992                                     const std::string & message = "") const
 993 {
 994 STG_LOCKER lock(&mutex);
 995
 996 if (PQstatus(connection) != CONNECTION_OK)
 997     {
 998     printfd(__FILE__, "POSTGRESQL_STORE::WriteUserChgLog(): 'Connection lost. Trying to reconnect...'\n", strError.c_str());
 999     if (Reset())
1000         {
1001         strError = "Connection lost";
1002         printfd(__FILE__, "POSTGRESQL_STORE::WriteUserChgLog(): '%s'\n", strError.c_str());
1003         return -1;
1004         }
1005     }
1006
1007 PGresult * result;
1008
1009 if (StartTransaction())
1010     {
1011     printfd(__FILE__, "POSTGRESQL_STORE::WriteUserChgLog(): 'Failed to start transaction'\n");
1012     return -1;
1013     }
1014
1015 std::string elogin(login);
1016 std::string eadminLogin(admLogin);
1017 std::string eparam(paramName);
1018 std::string eold(oldValue);
1019 std::string enew(newValue);
1020 std::string emessage(message);
1021
1022 if (EscapeString(elogin))
1023     {
1024     printfd(__FILE__, "POSTGRESQL_STORE::WriteUserChgLog(): 'Failed to escape login'\n");
1025     if (RollbackTransaction())
1026         {
1027         printfd(__FILE__, "POSTGRESQL_STORE::WriteUserChgLog(): 'Failed to rollback transaction'\n");
1028         }
1029     return -1;
1030     }
1031
1032 if (EscapeString(eadminLogin))
1033     {
1034     printfd(__FILE__, "POSTGRESQL_STORE::WriteUserChgLog(): 'Failed to escape admin's login'\n");
1035     if (RollbackTransaction())
1036         {
1037         printfd(__FILE__, "POSTGRESQL_STORE::WriteUserChgLog(): 'Failed to rollback transaction'\n");
1038         }
1039     return -1;
1040     }
1041
1042 if (EscapeString(eparam))
1043     {
1044     printfd(__FILE__, "POSTGRESQL_STORE::WriteUserChgLog(): 'Failed to escape param's name'\n");
1045     if (RollbackTransaction())
1046         {
1047         printfd(__FILE__, "POSTGRESQL_STORE::WriteUserChgLog(): 'Failed to rollback transaction'\n");
1048         }
1049     return -1;
1050     }
1051
1052 if (EscapeString(eold))
1053     {
1054     printfd(__FILE__, "POSTGRESQL_STORE::WriteUserChgLog(): 'Failed to escape old value'\n");
1055     if (RollbackTransaction())
1056         {
1057         printfd(__FILE__, "POSTGRESQL_STORE::WriteUserChgLog(): 'Failed to rollback transaction'\n");
1058         }
1059     return -1;
1060     }
1061
1062 if (EscapeString(enew))
1063     {
1064     printfd(__FILE__, "POSTGRESQL_STORE::WriteUserChgLog(): 'Failed to escape new value'\n");
1065     if (RollbackTransaction())
1066         {
1067         printfd(__FILE__, "POSTGRESQL_STORE::WriteUserChgLog(): 'Failed to rollback transaction'\n");
1068         }
1069     return -1;
1070     }
1071
1072 std::ostringstream query;
1073 query << "SELECT sp_add_param_log_entry("
1074             "'" << elogin << "', "
1075             "'" << eadminLogin << "', CAST('"
1076             << inet_ntostring(admIP) << "/32' AS INET), "
1077             "'" << eparam << "', "
1078             "CAST('" << formatTime(stgTime) << "' AS TIMESTAMP), "
1079             "'" << eold << "', "
1080             "'" << enew << "', "
1081             "'" << emessage << "')";
1082
1083 result = PQexec(connection, query.str().c_str());
1084
1085 if (PQresultStatus(result) != PGRES_TUPLES_OK)
1086     {
1087     strError = PQresultErrorMessage(result);
1088     PQclear(result);
1089     printfd(__FILE__, "POSTGRESQL_STORE::WriteUserChgLog(): '%s'\n", strError.c_str());
1090     if (RollbackTransaction())
1091         {
1092         printfd(__FILE__, "POSTGRESQL_STORE::WriteUserChgLog(): 'Failed to rollback transaction'\n");
1093         }
1094     return -1;
1095     }
1096
1097 PQclear(result);
1098
1099 if (CommitTransaction())
1100     {
1101     printfd(__FILE__, "POSTGRESQL_STORE::WriteUserChgLog(): 'Failed to commit transaction'\n");
1102     return -1;
1103     }
1104
1105 return 0;
1106 }
1107
1108 //-----------------------------------------------------------------------------
1109 int POSTGRESQL_STORE::WriteUserConnect(const std::string & login, uint32_t ip) const
1110 {
1111 STG_LOCKER lock(&mutex);
1112
1113 if (PQstatus(connection) != CONNECTION_OK)
1114     {
1115     printfd(__FILE__, "POSTGRESQL_STORE::WriteUserConnect(): 'Connection lost. Trying to reconnect...'\n", strError.c_str());
1116     if (Reset())
1117         {
1118         strError = "Connection lost";
1119         printfd(__FILE__, "POSTGRESQL_STORE::WriteUserConnect(): '%s'\n", strError.c_str());
1120         return -1;
1121         }
1122     }
1123
1124 PGresult * result;
1125
1126 if (StartTransaction())
1127     {
1128     printfd(__FILE__, "POSTGRESQL_STORE::WriteUserConnect(): 'Failed to start transaction'\n");
1129     return -1;
1130     }
1131
1132 std::string elogin(login);
1133
1134 if (EscapeString(elogin))
1135     {
1136     printfd(__FILE__, "POSTGRESQL_STORE::WriteUserConnect(): 'Failed to escape login'\n");
1137     if (RollbackTransaction())
1138         {
1139         printfd(__FILE__, "POSTGRESQL_STORE::WriteUserConnect(): 'Failed to rollback transaction'\n");
1140         }
1141     return -1;
1142     }
1143
1144 std::ostringstream query;
1145 if (version < 6)
1146     {
1147     query << "SELECT sp_add_session_log_entry("
1148                  "'" << elogin << "', "
1149                  "CAST('" << formatTime(stgTime) << "' AS TIMESTAMP), "
1150                  "'c', CAST('"
1151                  << inet_ntostring(ip) << "/32' AS INET), 0)";
1152     }
1153 else
1154     {
1155     query << "SELECT sp_add_session_log_entry("
1156                  "'" << elogin << "', "
1157                  "CAST('" << formatTime(stgTime) << "' AS TIMESTAMP), "
1158                  "'c', CAST('"
1159                  << inet_ntostring(ip) << "/32' AS INET), 0, 0, '')";
1160     }
1161
1162 result = PQexec(connection, query.str().c_str());
1163
1164 if (PQresultStatus(result) != PGRES_TUPLES_OK)
1165     {
1166     strError = PQresultErrorMessage(result);
1167     PQclear(result);
1168     printfd(__FILE__, "POSTGRESQL_STORE::WriteUserConnect(): '%s'\n", strError.c_str());
1169     if (RollbackTransaction())
1170         {
1171         printfd(__FILE__, "POSTGRESQL_STORE::WriteUserConnect(): 'Failed to rollback transaction'\n");
1172         }
1173     return -1;
1174     }
1175
1176 PQclear(result);
1177
1178 if (CommitTransaction())
1179     {
1180     printfd(__FILE__, "POSTGRESQL_STORE::WriteUserConnect(): 'Failed to commit transaction'\n");
1181     return -1;
1182     }
1183
1184 return 0;
1185 }
1186
1187 //-----------------------------------------------------------------------------
1188 int POSTGRESQL_STORE::WriteUserDisconnect(const std::string & login,
1189                     const DIR_TRAFF & monthUp,
1190                     const DIR_TRAFF & monthDown,
1191                     const DIR_TRAFF & sessionUp,
1192                     const DIR_TRAFF & sessionDown,
1193                     double cash,
1194                     double freeMb,
1195                     const std::string & reason) const
1196 {
1197 STG_LOCKER lock(&mutex);
1198
1199 if (PQstatus(connection) != CONNECTION_OK)
1200     {
1201     printfd(__FILE__, "POSTGRESQL_STORE::WriteUserDisconnect(): 'Connection lost. Trying to reconnect...'\n", strError.c_str());
1202     if (Reset())
1203         {
1204         strError = "Connection lost";
1205         printfd(__FILE__, "POSTGRESQL_STORE::WriteUserDisconnect(): '%s'\n", strError.c_str());
1206         return -1;
1207         }
1208     }
1209
1210 PGresult * result;
1211
1212 if (StartTransaction())
1213     {
1214     printfd(__FILE__, "POSTGRESQL_STORE::WriteUserDisconnect(): 'Failed to start transaction'\n");
1215     return -1;
1216     }
1217
1218 std::string elogin(login);
1219
1220 if (EscapeString(elogin))
1221     {
1222     printfd(__FILE__, "POSTGRESQL_STORE::WriteUserDisconnect(): 'Failed to escape login'\n");
1223     if (RollbackTransaction())
1224         {
1225         printfd(__FILE__, "POSTGRESQL_STORE::WriteUserDisconnect(): 'Failed to rollback transaction'\n");
1226         }
1227     return -1;
1228     }
1229
1230 std::string ereason(reason);
1231
1232 if (EscapeString(ereason))
1233     {
1234     printfd(__FILE__, "POSTGRESQL_STORE::WriteUserDisconnect(): 'Failed to escape reason'\n");
1235     if (RollbackTransaction())
1236         {
1237         printfd(__FILE__, "POSTGRESQL_STORE::WriteUserDisconnect(): 'Failed to rollback transaction'\n");
1238         }
1239     return -1;
1240     }
1241
1242     {
1243     std::ostringstream query;
1244     if (version < 6)
1245         {
1246         // Old database version - no freeMb logging support
1247         query << "SELECT sp_add_session_log_entry("
1248                     "'" << elogin << "', "
1249                     "CAST('" << formatTime(stgTime) << "' AS TIMESTAMP), "
1250                     "'d', CAST('0.0.0.0/0' AS INET), "
1251                     << cash << ")";
1252         }
1253     else
1254         {
1255         query << "SELECT sp_add_session_log_entry("
1256                     "'" << elogin << "', "
1257                     "CAST('" << formatTime(stgTime) << "' AS TIMESTAMP), "
1258                     "'d', CAST('0.0.0.0/0' AS INET), "
1259                     << cash << ", " << freeMb << ", '" << ereason << "')";
1260         }
1261
1262     result = PQexec(connection, query.str().c_str());
1263     }
1264
1265 if (PQresultStatus(result) != PGRES_TUPLES_OK)
1266     {
1267     strError = PQresultErrorMessage(result);
1268     PQclear(result);
1269     printfd(__FILE__, "POSTGRESQL_STORE::WriteUserDisconnect(): '%s'\n", strError.c_str());
1270     if (RollbackTransaction())
1271         {
1272         printfd(__FILE__, "POSTGRESQL_STORE::WriteUserDisconnect(): 'Failed to rollback transaction'\n");
1273         }
1274     return -1;
1275     }
1276
1277 int tuples = PQntuples(result);
1278
1279 if (tuples != 1)
1280     {
1281     strError = "Failed to fetch session's log ID";
1282     printfd(__FILE__, "POSTGRESQL_STORE::WriteUserDisconnect(): 'Invalid number of tuples. Wanted 1, actulally %d'\n", tuples);
1283     PQclear(result);
1284     if (RollbackTransaction())
1285         {
1286         printfd(__FILE__, "POSTGRESQL_STORE::WriteUserDisconnect(): 'Failed to rollback transaction'\n");
1287         }
1288     return -1;
1289     }
1290
1291 uint32_t lid;
1292
1293 if (str2x(PQgetvalue(result, 0, 0), lid))
1294     {
1295     strError = "Failed to convert string to int";
1296     printfd(__FILE__, "POSTGRESQL_STORE::WriteUserDisconnect(): '%s'\n", strError.c_str());
1297     PQclear(result);
1298     if (RollbackTransaction())
1299         {
1300         printfd(__FILE__, "POSTGRESQL_STORE::WriteUserDisconnect(): 'Failed to rollback transaction'\n");
1301         }
1302     return -1;
1303     }
1304
1305 PQclear(result);
1306
1307 for (int i = 0; i < DIR_NUM; ++i)
1308     {
1309     std::ostringstream query;
1310     query << "INSERT INTO tb_sessions_data "
1311                 "(fk_session_log, "
1312                  "dir_num, "
1313                  "session_upload, "
1314                  "session_download, "
1315                  "month_upload, "
1316                  "month_download)"
1317              "VALUES ("
1318                 << lid << ", "
1319                 << i << ", "
1320                 << sessionUp[i] << ", "
1321                 << sessionDown[i] << ", "
1322                 << monthUp[i] << ", "
1323                 << monthDown[i] << ")";
1324
1325     result = PQexec(connection, query.str().c_str());
1326
1327     if (PQresultStatus(result) != PGRES_COMMAND_OK)
1328         {
1329         strError = PQresultErrorMessage(result);
1330         PQclear(result);
1331         printfd(__FILE__, "POSTGRESQL_STORE::WriteUserDisconnect(): '%s'\n", strError.c_str());
1332         if (RollbackTransaction())
1333             {
1334             printfd(__FILE__, "POSTGRESQL_STORE::WriteUserDisconnect(): 'Failed to rollback transaction'\n");
1335             }
1336         return -1;
1337         }
1338
1339     PQclear(result);
1340     }
1341
1342 if (CommitTransaction())
1343     {
1344     printfd(__FILE__, "POSTGRESQL_STORE::WriteUserDisconnect(): 'Failed to commit transaction'\n");
1345     return -1;
1346     }
1347
1348 return 0;
1349 }
1350
1351 //-----------------------------------------------------------------------------
1352 int POSTGRESQL_STORE::WriteDetailedStat(const std::map<IP_DIR_PAIR, STAT_NODE> & statTree,
1353                                       time_t lastStat,
1354                                       const std::string & login) const
1355 {
1356 STG_LOCKER lock(&mutex);
1357
1358 if (PQstatus(connection) != CONNECTION_OK)
1359     {
1360     printfd(__FILE__, "POSTGRESQL_STORE::WriteDetailedStat(): 'Connection lost. Trying to reconnect...'\n", strError.c_str());
1361     if (Reset())
1362         {
1363         strError = "Connection lost";
1364         printfd(__FILE__, "POSTGRESQL_STORE::WriteDetailedStat(): '%s'\n", strError.c_str());
1365         return -1;
1366         }
1367     }
1368
1369 if (StartTransaction())
1370     {
1371     printfd(__FILE__, "POSTGRESQL_STORE::WriteDetailedStat(): 'Failed to start transaction'\n");
1372     return -1;
1373     }
1374
1375 std::string elogin(login);
1376
1377 if (EscapeString(elogin))
1378     {
1379     printfd(__FILE__, "POSTGRESQL_STORE::WriteDetailedStat(): 'Failed to escape login'\n");
1380     if (RollbackTransaction())
1381         {
1382         printfd(__FILE__, "POSTGRESQL_STORE::WriteDetailedStat(): 'Failed to rollback transaction'\n");
1383         }
1384     return -1;
1385     }
1386
1387 std::map<IP_DIR_PAIR, STAT_NODE>::const_iterator it;
1388 time_t currTime = time(NULL);
1389
1390 for (it = statTree.begin(); it != statTree.end(); ++it)
1391     {
1392     std::ostringstream query;
1393     query << "INSERT INTO tb_detail_stats "
1394                 "(till_time, from_time, fk_user, "
1395                  "dir_num, ip, download, upload, cost) "
1396              "VALUES ("
1397                 "CAST('" << formatTime(currTime) << "' AS TIMESTAMP), "
1398                 "CAST('" << formatTime(lastStat) << "' AS TIMESTAMP), "
1399                 "(SELECT pk_user FROM tb_users WHERE name = '" << elogin << "'), "
1400                 << it->first.dir << ", "
1401                 << "CAST('" << inet_ntostring(it->first.ip) << "' AS INET), "
1402                 << it->second.down << ", "
1403                 << it->second.up << ", "
1404                 << it->second.cash << ")";
1405
1406     PGresult * result = PQexec(connection, query.str().c_str());
1407
1408     if (PQresultStatus(result) != PGRES_COMMAND_OK)
1409         {
1410         strError = PQresultErrorMessage(result);
1411         PQclear(result);
1412         printfd(__FILE__, "POSTGRESQL_STORE::WriteDetailedStat(): '%s'\n", strError.c_str());
1413         if (RollbackTransaction())
1414             {
1415             printfd(__FILE__, "POSTGRESQL_STORE::WriteDetailedStat(): 'Failed to rollback transaction'\n");
1416             }
1417         return -1;
1418         }
1419
1420     PQclear(result);
1421     }
1422
1423 if (CommitTransaction())
1424     {
1425     printfd(__FILE__, "POSTGRESQL_STORE::WriteDetailedStat(): 'Failed to commit transaction'\n");
1426     return -1;
1427     }
1428
1429 return 0;
1430 }
1431
1432 //-----------------------------------------------------------------------------
1433 int POSTGRESQL_STORE::SaveMonthStat(const USER_STAT & stat, int month, int year, const std::string & login) const
1434 {
1435 STG_LOCKER lock(&mutex);
1436
1437 return SaveStat(stat, login, year, month);
1438 }
1439
1440 //-----------------------------------------------------------------------------
1441 int POSTGRESQL_STORE::SaveUserServices(uint32_t uid,
1442                                        const std::vector<std::string> & services) const
1443 {
1444 PGresult * result;
1445
1446     {
1447     std::ostringstream query;
1448     query << "DELETE FROM tb_users_services WHERE fk_user = " << uid;
1449
1450     result = PQexec(connection, query.str().c_str());
1451
1452     if (PQresultStatus(result) != PGRES_COMMAND_OK)
1453         {
1454         strError = PQresultErrorMessage(result);
1455         PQclear(result);
1456         printfd(__FILE__, "POSTGRESQL_STORE::SaveUserServices(): '%s'\n", strError.c_str());
1457         return -1;
1458         }
1459
1460     PQclear(result);
1461     }
1462
1463 std::vector<std::string>::const_iterator it;
1464
1465 for (it = services.begin(); it != services.end(); ++it)
1466     {
1467     std::string ename = *it;
1468
1469     if (EscapeString(ename))
1470         {
1471         printfd(__FILE__, "POSTGRESQL_STORE::SaveUserServices(): 'Failed to escape service name'\n");
1472         return -1;
1473         }
1474
1475     std::ostringstream query;
1476     query << "INSERT INTO tb_users_services "
1477                 "(fk_user, fk_service) "
1478              "VALUES "
1479                 "(" << uid << ", "
1480                   "(SELECT pk_service "
1481                    "FROM tb_services "
1482                    "WHERE name = '" << ename << "'))";
1483
1484     result = PQexec(connection, query.str().c_str());
1485
1486     if (PQresultStatus(result) != PGRES_COMMAND_OK)
1487         {
1488         strError = PQresultErrorMessage(result);
1489         PQclear(result);
1490         printfd(__FILE__, "POSTGRESQL_STORE::SaveUserServices(): '%s'\n", strError.c_str());
1491         return -1;
1492         }
1493
1494     PQclear(result);
1495     }
1496
1497 return 0;
1498 }
1499
1500 //-----------------------------------------------------------------------------
1501 int POSTGRESQL_STORE::SaveUserIPs(uint32_t uid,
1502                                   const USER_IPS & ips) const
1503 {
1504 PGresult * result;
1505
1506     {
1507     std::ostringstream query;
1508     query << "DELETE FROM tb_allowed_ip WHERE fk_user = " << uid;
1509
1510     result = PQexec(connection, query.str().c_str());
1511     }
1512
1513 if (PQresultStatus(result) != PGRES_COMMAND_OK)
1514     {
1515     strError = PQresultErrorMessage(result);
1516     PQclear(result);
1517     printfd(__FILE__, "POSTGRESQL_STORE::SaveUserIPs(): '%s'\n", strError.c_str());
1518     return -1;
1519     }
1520
1521 PQclear(result);
1522
1523 for (size_t i = 0; i < ips.Count(); ++i)
1524     {
1525     std::ostringstream query;
1526     query << "INSERT INTO tb_allowed_ip "
1527                 "(fk_user, ip) "
1528              "VALUES "
1529                 "(" << uid << ", CAST('"
1530                     << inet_ntostring(ips[i].ip) << "/"
1531                     << static_cast<int>(ips[i].mask) << "' AS INET))";
1532
1533     result = PQexec(connection, query.str().c_str());
1534
1535     if (PQresultStatus(result) != PGRES_COMMAND_OK)
1536         {
1537         strError = PQresultErrorMessage(result);
1538         PQclear(result);
1539         printfd(__FILE__, "POSTGRESQL_STORE::SaveUserIPs(): '%s'\n", strError.c_str());
1540         return -1;
1541         }
1542
1543     PQclear(result);
1544     }
1545
1546 return 0;
1547 }
1548
1549 //-----------------------------------------------------------------------------
1550 int POSTGRESQL_STORE::SaveUserData(uint32_t uid,
1551                                    const std::vector<std::string> & data) const
1552 {
1553 for (unsigned i = 0; i < data.size(); ++i)
1554     {
1555     std::string edata = data[i];
1556
1557     if (EscapeString(edata))
1558         {
1559         printfd(__FILE__, "POSTGRESQL_STORE::SaveUserData(): 'Failed to escape userdata field'\n");
1560         return -1;
1561         }
1562
1563     PGresult * result;
1564
1565     std::ostringstream query;
1566     query << "SELECT sp_set_user_data("
1567                 << uid << ", "
1568                 << "CAST(" << i << " AS SMALLINT), "
1569                 << "'" << edata << "')";
1570
1571     result = PQexec(connection, query.str().c_str());
1572
1573     if (PQresultStatus(result) != PGRES_TUPLES_OK)
1574         {
1575         strError = PQresultErrorMessage(result);
1576         PQclear(result);
1577         printfd(__FILE__, "POSTGRESQL_STORE::SaveUserData(): '%s'\n", strError.c_str());
1578         return -1;
1579         }
1580
1581     PQclear(result);
1582     }
1583
1584 return 0;
1585 }
1586