projects/stargazer/plugins/authorization/inetaccess/inetaccess.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.79 $
  23  $Date: 2010/03/25 15:18:48 $
  24  $Author: faust $
  25  */
  26
  27 #ifndef _GNU_SOURCE
  28 #define _GNU_SOURCE
  29 #endif
  30
  31 #include <sys/types.h>
  32 #include <sys/socket.h>
  33 #include <csignal>
  34 #include <cstdlib>
  35 #include <algorithm>
  36
  37 #include "inetaccess.h"
  38 #include "common.h"
  39
  40 extern volatile const time_t stgTime;
  41
  42 //-----------------------------------------------------------------------------
  43 class IA_CREATOR
  44 {
  45 private:
  46     AUTH_IA * ia;
  47
  48 public:
  49     IA_CREATOR()
  50         : ia(new AUTH_IA())
  51         {
  52         };
  53     ~IA_CREATOR()
  54         {
  55         delete ia;
  56         };
  57
  58     AUTH_IA * GetPlugin()
  59     {
  60         return ia;
  61     };
  62 };
  63 //-----------------------------------------------------------------------------
  64 //-----------------------------------------------------------------------------
  65 //-----------------------------------------------------------------------------
  66 IA_CREATOR iac;
  67 //-----------------------------------------------------------------------------
  68 //-----------------------------------------------------------------------------
  69 //-----------------------------------------------------------------------------
  70 BASE_PLUGIN * GetPlugin()
  71 {
  72 return iac.GetPlugin();
  73 }
  74 //-----------------------------------------------------------------------------
  75 //-----------------------------------------------------------------------------
  76 //-----------------------------------------------------------------------------
  77 AUTH_IA_SETTINGS::AUTH_IA_SETTINGS()
  78     : userDelay(0),
  79       userTimeout(0),
  80       port(0),
  81       freeMbShowType(freeMbCash)
  82 {
  83 }
  84 //-----------------------------------------------------------------------------
  85 int AUTH_IA_SETTINGS::ParseIntInRange(const string & str, int min, int max, int * val)
  86 {
  87 if (str2x(str.c_str(), *val))
  88     {
  89     errorStr = "Incorrect value \'" + str + "\'.";
  90     return -1;
  91     }
  92 if (*val < min || *val > max)
  93     {
  94     errorStr = "Value \'" + str + "\' out of range.";
  95     return -1;
  96     }
  97 return 0;
  98 }
  99 //-----------------------------------------------------------------------------
 100 int AUTH_IA_SETTINGS::ParseSettings(const MODULE_SETTINGS & s)
 101 {
 102 int p;
 103 PARAM_VALUE pv;
 104 vector<PARAM_VALUE>::const_iterator pvi;
 105 ///////////////////////////
 106 pv.param = "Port";
 107 pvi = find(s.moduleParams.begin(), s.moduleParams.end(), pv);
 108 if (pvi == s.moduleParams.end())
 109     {
 110     errorStr = "Parameter \'Port\' not found.";
 111     printfd(__FILE__, "Parameter 'Port' not found\n");
 112     return -1;
 113     }
 114 if (ParseIntInRange(pvi->value[0], 2, 65535, &p))
 115     {
 116     errorStr = "Cannot parse parameter \'Port\': " + errorStr;
 117     printfd(__FILE__, "Cannot parse parameter 'Port'\n");
 118     return -1;
 119     }
 120 port = p;
 121 ///////////////////////////
 122 pv.param = "UserDelay";
 123 pvi = find(s.moduleParams.begin(), s.moduleParams.end(), pv);
 124 if (pvi == s.moduleParams.end())
 125     {
 126     errorStr = "Parameter \'UserDelay\' not found.";
 127     printfd(__FILE__, "Parameter 'UserDelay' not found\n");
 128     return -1;
 129     }
 130
 131 if (ParseIntInRange(pvi->value[0], 5, 600, &userDelay))
 132     {
 133     errorStr = "Cannot parse parameter \'UserDelay\': " + errorStr;
 134     printfd(__FILE__, "Cannot parse parameter 'UserDelay'\n");
 135     return -1;
 136     }
 137 ///////////////////////////
 138 pv.param = "UserTimeout";
 139 pvi = find(s.moduleParams.begin(), s.moduleParams.end(), pv);
 140 if (pvi == s.moduleParams.end())
 141     {
 142     errorStr = "Parameter \'UserTimeout\' not found.";
 143     printfd(__FILE__, "Parameter 'UserTimeout' not found\n");
 144     return -1;
 145     }
 146
 147 if (ParseIntInRange(pvi->value[0], 15, 1200, &userTimeout))
 148     {
 149     errorStr = "Cannot parse parameter \'UserTimeout\': " + errorStr;
 150     printfd(__FILE__, "Cannot parse parameter 'UserTimeout'\n");
 151     return -1;
 152     }
 153 /////////////////////////////////////////////////////////////
 154 string freeMbType;
 155 int n = 0;
 156 pv.param = "FreeMb";
 157 pvi = find(s.moduleParams.begin(), s.moduleParams.end(), pv);
 158 if (pvi == s.moduleParams.end())
 159     {
 160     errorStr = "Parameter \'FreeMb\' not found.";
 161     printfd(__FILE__, "Parameter 'FreeMb' not found\n");
 162     return -1;
 163     }
 164 freeMbType = pvi->value[0];
 165
 166 if (strcasecmp(freeMbType.c_str(), "cash") == 0)
 167     {
 168     freeMbShowType = freeMbCash;
 169     }
 170 else if (strcasecmp(freeMbType.c_str(), "none") == 0)
 171     {
 172     freeMbShowType = freeMbNone;
 173     }
 174 else if (!str2x(freeMbType.c_str(), n))
 175     {
 176     if (n < 0 || n >= DIR_NUM)
 177         {
 178         errorStr = "Incorrect parameter \'" + freeMbType + "\'.";
 179         printfd(__FILE__, "%s\n", errorStr.c_str());
 180         return -1;
 181         }
 182     freeMbShowType = (FREEMB)(freeMb0 + n);
 183     }
 184 else
 185     {
 186     errorStr = "Incorrect parameter \'" + freeMbType + "\'.";
 187     printfd(__FILE__, "%s\n", errorStr.c_str());
 188     return -1;
 189     }
 190 /////////////////////////////////////////////////////////////
 191 return 0;
 192 }
 193 //-----------------------------------------------------------------------------
 194 //-----------------------------------------------------------------------------
 195 //-----------------------------------------------------------------------------
 196 #ifdef IA_PHASE_DEBUG
 197 IA_PHASE::IA_PHASE()
 198     : phase(1),
 199       flog(NULL)
 200 {
 201 gettimeofday(&phaseTime, NULL);
 202 }
 203 #else
 204 IA_PHASE::IA_PHASE()
 205     : phase(1)
 206 {
 207 gettimeofday(&phaseTime, NULL);
 208 }
 209 #endif
 210 //-----------------------------------------------------------------------------
 211 IA_PHASE::~IA_PHASE()
 212 {
 213 #ifdef IA_PHASE_DEBUG
 214 flog = fopen(log.c_str(), "at");
 215 if (flog)
 216     {
 217     fprintf(flog, "IA %s D\n", login.c_str());
 218     fclose(flog);
 219     }
 220 #endif
 221 }
 222 //-----------------------------------------------------------------------------
 223 #ifdef IA_PHASE_DEBUG
 224 void IA_PHASE::SetLogFileName(const string & logFileName)
 225 {
 226 log = logFileName + ".ia.log";
 227 }
 228 //-----------------------------------------------------------------------------
 229 void IA_PHASE::SetUserLogin(const string & login)
 230 {
 231 IA_PHASE::login = login;
 232 }
 233 //-----------------------------------------------------------------------------
 234 void IA_PHASE::WritePhaseChange(int newPhase)
 235 {
 236 UTIME newPhaseTime;
 237 gettimeofday(&newPhaseTime, NULL);
 238 flog = fopen(log.c_str(), "at");
 239 /*int64_t tn = newPhaseTime.GetSec()*1000000 + newPhaseTime.GetUSec();
 240 int64_t to = phaseTime.GetSec()*1000000 + phaseTime.GetUSec();*/
 241 if (flog)
 242     {
 243     string action = newPhase == phase ? "U" : "C";
 244     double delta = newPhaseTime.GetSec() - phaseTime.GetSec();
 245     delta += (newPhaseTime.GetUSec() - phaseTime.GetUSec()) * 1.0e-6;
 246     fprintf(flog, "IA %s %s oldPhase = %d, newPhase = %d. dt = %.6f\n",
 247             login.c_str(),
 248             action.c_str(),
 249             phase,
 250             newPhase,
 251             delta);
 252     fclose(flog);
 253     }
 254 }
 255 #endif
 256 //-----------------------------------------------------------------------------
 257 void IA_PHASE::SetPhase1()
 258 {
 259 #ifdef IA_PHASE_DEBUG
 260 WritePhaseChange(1);
 261 #endif
 262 phase = 1;
 263 gettimeofday(&phaseTime, NULL);
 264 }
 265 //-----------------------------------------------------------------------------
 266 void IA_PHASE::SetPhase2()
 267 {
 268 #ifdef IA_PHASE_DEBUG
 269 WritePhaseChange(2);
 270 #endif
 271 phase = 2;
 272 gettimeofday(&phaseTime, NULL);
 273 }
 274 //-----------------------------------------------------------------------------
 275 void IA_PHASE::SetPhase3()
 276 {
 277 #ifdef IA_PHASE_DEBUG
 278 WritePhaseChange(3);
 279 #endif
 280 phase = 3;
 281 gettimeofday(&phaseTime, NULL);
 282 }
 283 //-----------------------------------------------------------------------------
 284 void IA_PHASE::SetPhase4()
 285 {
 286 #ifdef IA_PHASE_DEBUG
 287 WritePhaseChange(4);
 288 #endif
 289 phase = 4;
 290 gettimeofday(&phaseTime, NULL);
 291 }
 292 //-----------------------------------------------------------------------------
 293 void IA_PHASE::SetPhase5()
 294 {
 295 #ifdef IA_PHASE_DEBUG
 296 WritePhaseChange(5);
 297 #endif
 298 phase = 5;
 299 gettimeofday(&phaseTime, NULL);
 300 }
 301 //-----------------------------------------------------------------------------
 302 int IA_PHASE::GetPhase() const
 303 {
 304 return phase;
 305 }
 306 //-----------------------------------------------------------------------------
 307 void IA_PHASE::UpdateTime()
 308 {
 309 #ifdef IA_PHASE_DEBUG
 310 WritePhaseChange(phase);
 311 #endif
 312 gettimeofday(&phaseTime, NULL);
 313 }
 314 //-----------------------------------------------------------------------------
 315 const UTIME & IA_PHASE::GetTime() const
 316 {
 317 return phaseTime;
 318 }
 319 //-----------------------------------------------------------------------------
 320 //-----------------------------------------------------------------------------
 321 //-----------------------------------------------------------------------------
 322 AUTH_IA::AUTH_IA()
 323     : nonstop(false),
 324       isRunningRun(false),
 325       isRunningRunTimeouter(false),
 326       WriteServLog(GetStgLogger()),
 327       enabledDirs(0xFFffFFff),
 328       onDelUserNotifier(*this)
 329 {
 330 InitEncrypt(&ctxS, "pr7Hhen");
 331
 332 pthread_mutexattr_t attr;
 333 pthread_mutexattr_init(&attr);
 334 pthread_mutexattr_settype(&attr, PTHREAD_MUTEX_RECURSIVE);
 335 pthread_mutex_init(&mutex, &attr);
 336
 337 memset(&connSynAck6, 0, sizeof(CONN_SYN_ACK_6));
 338 memset(&connSynAck8, 0, sizeof(CONN_SYN_ACK_8));
 339 memset(&disconnSynAck6, 0, sizeof(DISCONN_SYN_ACK_6));
 340 memset(&disconnSynAck8, 0, sizeof(DISCONN_SYN_ACK_8));
 341 memset(&aliveSyn6, 0, sizeof(ALIVE_SYN_6));
 342 memset(&aliveSyn8, 0, sizeof(ALIVE_SYN_8));
 343 memset(&fin6, 0, sizeof(FIN_6));
 344 memset(&fin8, 0, sizeof(FIN_8));
 345
 346 printfd(__FILE__, "sizeof(CONN_SYN_6) = %d %d\n",           sizeof(CONN_SYN_6),     Min8(sizeof(CONN_SYN_6)));
 347 printfd(__FILE__, "sizeof(CONN_SYN_8) = %d %d\n",           sizeof(CONN_SYN_8),     Min8(sizeof(CONN_SYN_8)));
 348 printfd(__FILE__, "sizeof(CONN_SYN_ACK_6) = %d %d\n",       sizeof(CONN_SYN_ACK_6), Min8(sizeof(CONN_SYN_ACK_6)));
 349 printfd(__FILE__, "sizeof(CONN_SYN_ACK_8) = %d %d\n",       sizeof(CONN_SYN_ACK_8), Min8(sizeof(CONN_SYN_ACK_8)));
 350 printfd(__FILE__, "sizeof(CONN_ACK_6) = %d %d\n",           sizeof(CONN_ACK_6),     Min8(sizeof(CONN_ACK_6)));
 351 printfd(__FILE__, "sizeof(ALIVE_SYN_6) = %d %d\n",          sizeof(ALIVE_SYN_6),    Min8(sizeof(ALIVE_SYN_6)));
 352 printfd(__FILE__, "sizeof(ALIVE_SYN_8) = %d %d\n",          sizeof(ALIVE_SYN_8),    Min8(sizeof(ALIVE_SYN_8)));
 353 printfd(__FILE__, "sizeof(ALIVE_ACK_6) = %d %d\n",          sizeof(ALIVE_ACK_6),    Min8(sizeof(ALIVE_ACK_6)));
 354 printfd(__FILE__, "sizeof(DISCONN_SYN_6) = %d %d\n",        sizeof(DISCONN_SYN_6),  Min8(sizeof(DISCONN_SYN_6)));
 355 printfd(__FILE__, "sizeof(DISCONN_SYN_ACK_6) = %d %d\n",    sizeof(DISCONN_SYN_ACK_6), Min8(sizeof(DISCONN_SYN_ACK_6)));
 356 printfd(__FILE__, "sizeof(DISCONN_SYN_ACK_8) = %d %d\n",    sizeof(DISCONN_SYN_ACK_8), Min8(sizeof(DISCONN_SYN_ACK_8)));
 357 printfd(__FILE__, "sizeof(DISCONN_ACK_6) = %d %d\n",        sizeof(DISCONN_ACK_6),  Min8(sizeof(DISCONN_ACK_6)));
 358 printfd(__FILE__, "sizeof(FIN_6) = %d %d\n",                sizeof(FIN_6),          Min8(sizeof(FIN_6)));
 359 printfd(__FILE__, "sizeof(FIN_8) = %d %d\n",                sizeof(FIN_8),          Min8(sizeof(FIN_8)));
 360 printfd(__FILE__, "sizeof(ERR) = %d %d\n",                  sizeof(ERR),            Min8(sizeof(ERR)));
 361 printfd(__FILE__, "sizeof(INFO_6) = %d %d\n",               sizeof(INFO_6),         Min8(sizeof(INFO_6)));
 362 printfd(__FILE__, "sizeof(INFO_7) = %d %d\n",               sizeof(INFO_7),         Min8(sizeof(INFO_7)));
 363 printfd(__FILE__, "sizeof(INFO_8) = %d %d\n",               sizeof(INFO_8),         Min8(sizeof(INFO_8)));
 364
 365 packetTypes["CONN_SYN"] = CONN_SYN_N;
 366 packetTypes["CONN_SYN_ACK"] = CONN_SYN_ACK_N;
 367 packetTypes["CONN_ACK"] = CONN_ACK_N;
 368 packetTypes["ALIVE_SYN"] = ALIVE_SYN_N;
 369 packetTypes["ALIVE_ACK"] = ALIVE_ACK_N;
 370 packetTypes["DISCONN_SYN"] = DISCONN_SYN_N;
 371 packetTypes["DISCONN_SYN_ACK"] = DISCONN_SYN_ACK_N;
 372 packetTypes["DISCONN_ACK"] = DISCONN_ACK_N;
 373 packetTypes["FIN"] = FIN_N;
 374 packetTypes["ERR"] = ERROR_N;
 375 }
 376 //-----------------------------------------------------------------------------
 377 AUTH_IA::~AUTH_IA()
 378 {
 379 pthread_mutex_destroy(&mutex);
 380 }
 381 //-----------------------------------------------------------------------------
 382 int AUTH_IA::Start()
 383 {
 384 users->AddNotifierUserDel(&onDelUserNotifier);
 385 nonstop = true;
 386
 387 if (PrepareNet())
 388     {
 389     return -1;
 390     }
 391
 392 if (!isRunningRun)
 393     {
 394     if (pthread_create(&recvThread, NULL, Run, this))
 395         {
 396         errorStr = "Cannot create thread.";
 397         printfd(__FILE__, "Cannot create recv thread\n");
 398         return -1;
 399         }
 400     }
 401
 402 if (!isRunningRunTimeouter)
 403     {
 404     if (pthread_create(&timeouterThread, NULL, RunTimeouter, this))
 405         {
 406         errorStr = "Cannot create thread.";
 407         printfd(__FILE__, "Cannot create timeouter thread\n");
 408         return -1;
 409         }
 410     }
 411 errorStr = "";
 412 return 0;
 413 }
 414 //-----------------------------------------------------------------------------
 415 int AUTH_IA::Stop()
 416 {
 417 if (!IsRunning())
 418     return 0;
 419
 420 nonstop = false;
 421
 422 std::for_each(
 423         ip2user.begin(),
 424         ip2user.end(),
 425         UnauthorizeUser(this)
 426         );
 427
 428 if (isRunningRun)
 429     {
 430     //5 seconds to thread stops itself
 431     for (int i = 0; i < 25 && isRunningRun; i++)
 432         {
 433         usleep(200000);
 434         }
 435
 436     //after 5 seconds waiting thread still running. now killing it
 437     if (isRunningRun)
 438         {
 439         //TODO pthread_cancel()
 440         if (pthread_kill(recvThread, SIGINT))
 441             {
 442             errorStr = "Cannot kill thread.";
 443             printfd(__FILE__, "Cannot kill thread\n");
 444             return -1;
 445             }
 446         for (int i = 0; i < 25 && isRunningRun; ++i)
 447             usleep(200000);
 448         if (isRunningRun)
 449             {
 450             printfd(__FILE__, "Failed to stop recv thread\n");
 451             }
 452         else
 453             {
 454             pthread_join(recvThread, NULL);
 455             }
 456         printfd(__FILE__, "AUTH_IA killed Run\n");
 457         }
 458     }
 459
 460 FinalizeNet();
 461
 462 if (isRunningRunTimeouter)
 463     {
 464     //5 seconds to thread stops itself
 465     for (int i = 0; i < 25 && isRunningRunTimeouter; i++)
 466         {
 467         usleep(200000);
 468         }
 469
 470     //after 5 seconds waiting thread still running. now killing it
 471     if (isRunningRunTimeouter)
 472         {
 473         //TODO pthread_cancel()
 474         if (pthread_kill(timeouterThread, SIGINT))
 475             {
 476             errorStr = "Cannot kill thread.";
 477             return -1;
 478             }
 479         for (int i = 0; i < 25 && isRunningRunTimeouter; ++i)
 480             usleep(200000);
 481         if (isRunningRunTimeouter)
 482             {
 483             printfd(__FILE__, "Failed to stop timeouter thread\n");
 484             }
 485         else
 486             {
 487             pthread_join(timeouterThread, NULL);
 488             }
 489         printfd(__FILE__, "AUTH_IA killed Timeouter\n");
 490         }
 491     }
 492 printfd(__FILE__, "AUTH_IA::Stoped successfully.\n");
 493 users->DelNotifierUserDel(&onDelUserNotifier);
 494 return 0;
 495 }
 496 //-----------------------------------------------------------------------------
 497 void * AUTH_IA::Run(void * d)
 498 {
 499 AUTH_IA * ia = static_cast<AUTH_IA *>(d);
 500
 501 ia->isRunningRun = true;
 502
 503 char buffer[512];
 504
 505 time_t touchTime = stgTime - MONITOR_TIME_DELAY_SEC;
 506
 507 while (ia->nonstop)
 508     {
 509     ia->RecvData(buffer, sizeof(buffer));
 510     if ((touchTime + MONITOR_TIME_DELAY_SEC <= stgTime) && ia->stgSettings->GetMonitoring())
 511         {
 512         touchTime = stgTime;
 513         string monFile = ia->stgSettings->GetMonitorDir() + "/inetaccess_r";
 514         TouchFile(monFile.c_str());
 515         }
 516     }
 517
 518 ia->isRunningRun = false;
 519 return NULL;
 520 }
 521 //-----------------------------------------------------------------------------
 522 void * AUTH_IA::RunTimeouter(void * d)
 523 {
 524 AUTH_IA * ia = static_cast<AUTH_IA *>(d);
 525
 526 ia->isRunningRunTimeouter = true;
 527
 528 int a = -1;
 529 string monFile = ia->stgSettings->GetMonitorDir() + "/inetaccess_t";
 530 while (ia->nonstop)
 531     {
 532     usleep(20000);
 533     ia->Timeouter();
 534     // TODO cahange counter to timer and MONITOR_TIME_DELAY_SEC
 535     if (++a % (50*60) == 0 && ia->stgSettings->GetMonitoring())
 536         {
 537         TouchFile(monFile.c_str());
 538         }
 539     }
 540
 541 ia->isRunningRunTimeouter = false;
 542 return NULL;
 543 }
 544 //-----------------------------------------------------------------------------
 545 int AUTH_IA::ParseSettings()
 546 {
 547 int ret = iaSettings.ParseSettings(settings);
 548 if (ret)
 549     errorStr = iaSettings.GetStrError();
 550 return ret;
 551 }
 552 //-----------------------------------------------------------------------------
 553 int AUTH_IA::PrepareNet()
 554 {
 555 struct sockaddr_in listenAddr;
 556
 557 listenSocket = socket(AF_INET, SOCK_DGRAM, 0);
 558
 559 if (listenSocket < 0)
 560     {
 561     errorStr = "Cannot create socket.";
 562     return -1;
 563     }
 564
 565 listenAddr.sin_family = AF_INET;
 566 listenAddr.sin_port = htons(iaSettings.GetUserPort());
 567 listenAddr.sin_addr.s_addr = inet_addr("0.0.0.0");
 568
 569 if (bind(listenSocket, (struct sockaddr*)&listenAddr, sizeof(listenAddr)) < 0)
 570     {
 571     errorStr = "AUTH_IA: Bind failed.";
 572     return -1;
 573     }
 574
 575 /*int buffLen;
 576 if (getsockopt(listenSocket, SOL_SOCKET, SO_SNDBUF, &buffLen, sizeof(buffLen)) < 0)
 577     {
 578
 579     errorStr = "Getsockopt failed. " + string(strerror(errno));
 580     return -1;
 581     }*/
 582
 583 //WriteServLog("buffLen = %d", buffLen);
 584
 585 return 0;
 586 }
 587 //-----------------------------------------------------------------------------
 588 int AUTH_IA::FinalizeNet()
 589 {
 590 close(listenSocket);
 591 return 0;
 592 }
 593 //-----------------------------------------------------------------------------
 594 int AUTH_IA::RecvData(char * buffer, int bufferSize)
 595 {
 596 if (!WaitPackets(listenSocket)) // Timeout
 597     {
 598     return 0;
 599     }
 600
 601 struct sockaddr_in outerAddr;
 602 socklen_t outerAddrLen(sizeof(outerAddr));
 603 int dataLen = recvfrom(listenSocket, buffer, bufferSize, 0, (struct sockaddr *)&outerAddr, &outerAddrLen);
 604
 605 if (!dataLen) // EOF
 606     {
 607     return 0;
 608     }
 609
 610 if (dataLen <= 0) // Error
 611     {
 612     if (errno != EINTR)
 613         {
 614         printfd(__FILE__, "recvfrom res=%d, error: '%s'\n", dataLen, strerror(errno));
 615         return -1;
 616         }
 617     return 0;
 618     }
 619
 620 if (dataLen > 256)
 621     return -1;
 622
 623 int protoVer;
 624 if (CheckHeader(buffer, &protoVer))
 625     return -1;
 626
 627 char login[PASSWD_LEN];  //TODO why PASSWD_LEN ?
 628 memset(login, 0, PASSWD_LEN);
 629
 630 Decrypt(&ctxS, login, buffer + 8, PASSWD_LEN / 8);
 631
 632 uint32_t sip = *((uint32_t*)&outerAddr.sin_addr);
 633 uint16_t sport = htons(outerAddr.sin_port);
 634
 635 user_iter user;
 636 if (users->FindByName(login, &user) == 0)
 637     {
 638     printfd(__FILE__, "User %s FOUND!\n", user->GetLogin().c_str());
 639     PacketProcessor(buffer, dataLen, sip, sport, protoVer, &user);
 640     }
 641 else
 642     {
 643     WriteServLog("User\'s connect failed:: user \'%s\' not found. IP \'%s\'",
 644                  login,
 645                  inet_ntostring(sip).c_str());
 646     printfd(__FILE__, "User %s NOT found!\n", login);
 647     SendError(sip, sport, protoVer, "îÅÐÒÁ×ÉÌØÎÙÊ ÌÏÇÉÎ ÉÌÉ ÐÁÒÏÌØ!");
 648     }
 649
 650 return 0;
 651
 652 }
 653 //-----------------------------------------------------------------------------
 654 int AUTH_IA::CheckHeader(const char * buffer, int * protoVer)
 655 {
 656 if (strncmp(IA_ID, buffer, strlen(IA_ID)) != 0)
 657     {
 658     //SendError(userIP, updateMsg);
 659     printfd(__FILE__, "update needed - IA_ID\n");
 660     //SendError(userIP, "Incorrect header!");
 661     return -1;
 662     }
 663
 664 if (buffer[6] != 0) //proto[0] shoud be 0
 665     {
 666     printfd(__FILE__, "update needed - PROTO major: %d\n", buffer[6]);
 667     //SendError(userIP, updateMsg);
 668     return -1;
 669     }
 670
 671 if (buffer[7] < 6)
 672     {
 673     // need update
 674     //SendError(userIP, updateMsg);
 675     printfd(__FILE__, "update needed - PROTO minor: %d\n", buffer[7]);
 676     return -1;
 677     }
 678 else
 679     {
 680     *protoVer = buffer[7];
 681     }
 682 return 0;
 683 }
 684 //-----------------------------------------------------------------------------
 685 int AUTH_IA::Timeouter()
 686 {
 687 STG_LOCKER lock(&mutex, __FILE__, __LINE__);
 688
 689 map<uint32_t, IA_USER>::iterator it;
 690 it = ip2user.begin();
 691 uint32_t sip;
 692
 693 while (it != ip2user.end())
 694     {
 695     sip = it->first;
 696
 697     static UTIME currTime;
 698     gettimeofday(&currTime, NULL);
 699
 700     if ((it->second.phase.GetPhase() == 2)
 701         && (currTime - it->second.phase.GetTime()) > iaSettings.GetUserDelay())
 702         {
 703         it->second.phase.SetPhase1();
 704         printfd(__FILE__, "Phase changed from 2 to 1. Reason: timeout\n");
 705         }
 706
 707     if (it->second.phase.GetPhase() == 3)
 708         {
 709         if (!it->second.messagesToSend.empty())
 710             {
 711             if (it->second.protoVer == 6)
 712                 RealSendMessage6(*it->second.messagesToSend.begin(), sip, it->second);
 713
 714             if (it->second.protoVer == 7)
 715                 RealSendMessage7(*it->second.messagesToSend.begin(), sip, it->second);
 716
 717             if (it->second.protoVer == 8)
 718                 RealSendMessage8(*it->second.messagesToSend.begin(), sip, it->second);
 719
 720             it->second.messagesToSend.erase(it->second.messagesToSend.begin());
 721             }
 722
 723         if((currTime - it->second.lastSendAlive) > iaSettings.GetUserDelay())
 724             {
 725             switch (it->second.protoVer)
 726                 {
 727                 case 6:
 728                     Send_ALIVE_SYN_6(&(it->second), sip);
 729                     break;
 730                 case 7:
 731                     Send_ALIVE_SYN_7(&(it->second), sip);
 732                     break;
 733                 case 8:
 734                     Send_ALIVE_SYN_8(&(it->second), sip);
 735                     break;
 736                 }
 737
 738             gettimeofday(&it->second.lastSendAlive, NULL);
 739             }
 740
 741         if ((currTime - it->second.phase.GetTime()) > iaSettings.GetUserTimeout())
 742             {
 743             it->second.user->Unauthorize(this);
 744             ip2user.erase(it++);
 745             continue;
 746             }
 747         }
 748
 749     if ((it->second.phase.GetPhase() == 4)
 750         && ((currTime - it->second.phase.GetTime()) > iaSettings.GetUserDelay()))
 751         {
 752         it->second.phase.SetPhase3();
 753         printfd(__FILE__, "Phase changed from 4 to 3. Reason: timeout\n");
 754         }
 755
 756     ++it;
 757     }
 758
 759 return 0;
 760 }
 761 //-----------------------------------------------------------------------------
 762 int AUTH_IA::PacketProcessor(char * buff, int dataLen, uint32_t sip, uint16_t sport, int protoVer, user_iter * user)
 763 {
 764 STG_LOCKER lock(&mutex, __FILE__, __LINE__);
 765 // ôÕÔ ÓÏÂÒÁÎÙ ÏÂÒÁÂÏÔÞÉËÉ ÒÁÚÎÙÈ ÐÁËÅÔÏ×
 766 int pn = -1;
 767 const int offset = LOGIN_LEN + 2 + 6; // LOGIN_LEN + sizeOfMagic + sizeOfVer;
 768
 769 IA_USER * iaUser = NULL;
 770
 771 CONN_SYN_6 * connSyn6;
 772 CONN_SYN_7 * connSyn7;
 773 CONN_SYN_8 * connSyn8;
 774
 775 CONN_ACK_6 * connAck;
 776 ALIVE_ACK_6 * aliveAck;
 777 DISCONN_SYN_6 * disconnSyn;
 778 DISCONN_ACK_6 * disconnAck;
 779
 780 map<uint32_t, IA_USER>::iterator it;
 781 it = ip2user.find(sip);
 782
 783 if (it == ip2user.end() || (*user)->GetID() != it->second.user->GetID())
 784     {
 785     // åÝÅ ÎÅ ÂÙÌÏ ÚÁÐÒÏÓÏ× Ó ÜÔÏÇÏ IP
 786     printfd(__FILE__, "Add new user\n");
 787     ip2user[sip].protoVer = protoVer;
 788     ip2user[sip].user = *user;
 789     ip2user[sip].port = sport;
 790     #ifdef IA_PHASE_DEBUG
 791     ip2user[sip].phase.SetLogFileName(stgSettings->GetLogFileName());
 792     ip2user[sip].phase.SetUserLogin((*user)->GetLogin());
 793     #endif
 794
 795
 796     it = ip2user.find(sip); //TODO
 797     if (it == ip2user.end())
 798         {
 799         printfd(__FILE__, "+++ ERROR +++\n");
 800         return -1;
 801         }
 802     }
 803
 804 iaUser = &(it->second);
 805
 806 if (iaUser->port != sport)
 807     iaUser->port = sport;
 808
 809 if (iaUser->password != (*user)->property.password.Get())
 810     {
 811     InitEncrypt(&iaUser->ctx, (*user)->property.password.Get());
 812     iaUser->password = (*user)->property.password.Get();
 813     }
 814
 815 buff += offset;
 816 Decrypt(&iaUser->ctx, buff, buff, (dataLen - offset) / 8);
 817
 818 char packetName[IA_MAX_TYPE_LEN];
 819 strncpy(packetName,  buff + 4, IA_MAX_TYPE_LEN);
 820 packetName[IA_MAX_TYPE_LEN - 1] = 0;
 821
 822 map<string, int>::iterator pi;
 823 pi = packetTypes.find(packetName);
 824 if (pi == packetTypes.end())
 825     {
 826     SendError(sip, sport, protoVer, "îÅÐÒÁ×ÉÌØÎÙÊ ÌÏÇÉÎ ÉÌÉ ÐÁÒÏÌØ!");
 827     printfd(__FILE__, "Login or password is wrong!\n");
 828     WriteServLog("User's connect failed. IP \'%s\'. Wrong login or password", inet_ntostring(sip).c_str());
 829     return 0;
 830     }
 831 else
 832     {
 833     pn = pi->second;
 834     }
 835
 836 if ((*user)->property.disabled.Get())
 837     {
 838     SendError(sip, sport, protoVer, "õÞÅÔÎÁÑ ÚÁÐÉÓØ ÚÁÂÌÏËÉÒÏ×ÁÎÁ");
 839     return 0;
 840     }
 841
 842 if ((*user)->property.passive.Get())
 843     {
 844     SendError(sip, sport, protoVer, "õÞÅÔÎÁÑ ÚÁÐÉÓØ ÚÁÍÏÒÏÖÅÎÁ");
 845     return 0;
 846     }
 847
 848 if ((*user)->GetAuthorized() && (*user)->GetCurrIP() != sip)
 849     {
 850     printfd(__FILE__, "Login %s alredy in use. IP \'%s\'\n", (*user)->GetLogin().c_str(), inet_ntostring(sip).c_str());
 851     WriteServLog("Login %s alredy in use. IP \'%s\'", (*user)->GetLogin().c_str(), inet_ntostring(sip).c_str());
 852     SendError(sip, sport, protoVer, "÷ÁÛ ÌÏÇÉÎ ÕÖÅ ÉÓÐÏÌØÚÕÅÔÓÑ!");
 853     return 0;
 854     }
 855
 856 user_iter u;
 857 if (users->FindByIPIdx(sip, &u) == 0 && u->GetLogin() != (*user)->GetLogin())
 858     {
 859     printfd(__FILE__, "IP address alredy in use. IP \'%s\'", inet_ntostring(sip).c_str());
 860     WriteServLog("IP address alredy in use. IP \'%s\'", inet_ntostring(sip).c_str());
 861     SendError(sip, sport, protoVer, "÷ÁÛ IP ÁÄÒÅÓ ÕÖÅ ÉÓÐÏÌØÚÕÅÔÓÑ!");
 862     return 0;
 863     }
 864
 865 // ôÅÐÅÒØ ÍÙ ÄÏÌÖÎÙ ÐÒÏ×ÅÒÉÔØ, ÍÏÖÅÔ ÌÉ ÐÏÌØÚÏ×ÁÔÅÌØ ÐÏÄËÌÀÞÉÔÓÑ Ó ÜÔÏÇÏ ÁÄÒÅÓÁ.
 866 int ipFound = (*user)->property.ips.Get().IsIPInIPS(sip);
 867 if (!ipFound)
 868     {
 869     printfd(__FILE__, "User %s. IP address is incorrect. IP \'%s\'\n", (*user)->GetLogin().c_str(), inet_ntostring(sip).c_str());
 870     WriteServLog("User %s. IP address is incorrect. IP \'%s\'", (*user)->GetLogin().c_str(), inet_ntostring(sip).c_str());
 871     SendError(sip, sport, protoVer, "ðÏÌØÚÏ×ÁÔÅÌØ ÎÅ ÏÐÏÚÎÁÎ! ðÒÏ×ÅÒØÔÅ IP ÁÄÒÅÓ.");
 872     return 0;
 873     }
 874
 875 int ret = -1;
 876
 877 switch (pn)
 878     {
 879     case CONN_SYN_N:
 880         switch (protoVer)
 881             {
 882             case 6:
 883                 connSyn6 = (CONN_SYN_6*)(buff - offset);
 884                 ret = Process_CONN_SYN_6(connSyn6, &(it->second), user, sip);
 885                 break;
 886             case 7:
 887                 connSyn7 = (CONN_SYN_7*)(buff - offset);
 888                 ret = Process_CONN_SYN_7(connSyn7, &(it->second), user, sip);
 889                 break;
 890             case 8:
 891                 connSyn8 = (CONN_SYN_8*)(buff - offset);
 892                 ret = Process_CONN_SYN_8(connSyn8, &(it->second), user, sip);
 893                 break;
 894             }
 895
 896         if (ret != 0)
 897             {
 898             return 0;
 899             }
 900         switch (protoVer)
 901             {
 902             case 6:
 903                 Send_CONN_SYN_ACK_6(iaUser, user, sip);
 904                 break;
 905             case 7:
 906                 Send_CONN_SYN_ACK_7(iaUser, user, sip);
 907                 break;
 908             case 8:
 909                 Send_CONN_SYN_ACK_8(iaUser, user, sip);
 910                 break;
 911             }
 912         break;
 913
 914     case CONN_ACK_N:
 915         connAck = (CONN_ACK_6*)(buff - offset);
 916         switch (protoVer)
 917             {
 918             case 6:
 919                 ret = Process_CONN_ACK_6(connAck, iaUser, user, sip);
 920                 break;
 921             case 7:
 922                 ret = Process_CONN_ACK_7(connAck, iaUser, user, sip);
 923                 break;
 924             case 8:
 925                 ret = Process_CONN_ACK_8((CONN_ACK_8*)(buff - offset), iaUser, user, sip);
 926                 break;
 927             }
 928
 929         if (ret != 0)
 930             {
 931             SendError(sip, sport, protoVer, errorStr);
 932             return 0;
 933             }
 934
 935         switch (protoVer)
 936             {
 937             case 6:
 938                 Send_ALIVE_SYN_6(iaUser, sip);
 939                 break;
 940             case 7:
 941                 Send_ALIVE_SYN_7(iaUser, sip);
 942                 break;
 943             case 8:
 944                 Send_ALIVE_SYN_8(iaUser, sip);
 945                 break;
 946             }
 947
 948         break;
 949
 950         // ðÒÉÂÙÌ ÏÔ×ÅÔ Ó ÐÏÄÔ×ÅÒÖÄÅÎÉÅÍ ALIVE
 951     case ALIVE_ACK_N:
 952         aliveAck = (ALIVE_ACK_6*)(buff - offset);
 953         switch (protoVer)
 954             {
 955             case 6:
 956                 ret = Process_ALIVE_ACK_6(aliveAck, iaUser, user, sip);
 957                 break;
 958             case 7:
 959                 ret = Process_ALIVE_ACK_7(aliveAck, iaUser, user, sip);
 960                 break;
 961             case 8:
 962                 ret = Process_ALIVE_ACK_8((ALIVE_ACK_8*)(buff - offset), iaUser, user, sip);
 963                 break;
 964             }
 965         break;
 966
 967         // úÁÐÒÏÓ ÎÁ ÏÔËÌÀÞÅÎÉÅ
 968     case DISCONN_SYN_N:
 969
 970         disconnSyn = (DISCONN_SYN_6*)(buff - offset);
 971         switch (protoVer)
 972             {
 973             case 6:
 974                 ret = Process_DISCONN_SYN_6(disconnSyn, iaUser, user, sip);
 975                 break;
 976             case 7:
 977                 ret = Process_DISCONN_SYN_7(disconnSyn, iaUser, user, sip);
 978                 break;
 979             case 8:
 980                 ret = Process_DISCONN_SYN_8((DISCONN_SYN_8*)(buff - offset), iaUser, user, sip);
 981                 break;
 982             }
 983
 984         if (ret != 0)
 985             return 0;
 986
 987         switch (protoVer)
 988             {
 989             case 6:
 990                 Send_DISCONN_SYN_ACK_6(iaUser, sip);
 991                 break;
 992             case 7:
 993                 Send_DISCONN_SYN_ACK_7(iaUser, sip);
 994                 break;
 995             case 8:
 996                 Send_DISCONN_SYN_ACK_8(iaUser, sip);
 997                 break;
 998             }
 999         break;
1000
1001     case DISCONN_ACK_N:
1002         disconnAck = (DISCONN_ACK_6*)(buff - offset);
1003
1004         switch (protoVer)
1005             {
1006             case 6:
1007                 ret = Process_DISCONN_ACK_6(disconnAck, iaUser, user, sip, it);
1008                 break;
1009             case 7:
1010                 ret = Process_DISCONN_ACK_7(disconnAck, iaUser, user, sip, it);
1011                 break;
1012             case 8:
1013                 ret = Process_DISCONN_ACK_8((DISCONN_ACK_8*)(buff - offset), iaUser, user, sip, it);
1014                 break;
1015             }
1016
1017         switch (protoVer)
1018             {
1019             case 6:
1020                 Send_FIN_6(iaUser, sip, it);
1021                 break;
1022             case 7:
1023                 Send_FIN_7(iaUser, sip, it);
1024                 break;
1025             case 8:
1026                 Send_FIN_8(iaUser, sip, it);
1027                 break;
1028             }
1029         break;
1030     }
1031
1032 return 0;
1033 }
1034 //-----------------------------------------------------------------------------
1035 void AUTH_IA::DelUser(user_iter u)
1036 {
1037 STG_LOCKER lock(&mutex, __FILE__, __LINE__);
1038
1039 uint32_t ip = u->GetCurrIP();
1040
1041 if (!ip)
1042     return;
1043
1044 map<uint32_t, IA_USER>::iterator it;
1045 it = ip2user.find(ip);
1046 if (it == ip2user.end())
1047     {
1048     //Nothing to delete
1049     printfd(__FILE__, "Nothing to delete\n");
1050     return;
1051     }
1052
1053 if (it->second.user == u)
1054     {
1055     printfd(__FILE__, "User removed!\n");
1056     it->second.user->Unauthorize(this);
1057     ip2user.erase(it);
1058     }
1059 }
1060 //-----------------------------------------------------------------------------
1061 int AUTH_IA::SendError(uint32_t ip, uint16_t port, int protoVer, const string & text)
1062 {
1063 struct sockaddr_in sendAddr;
1064 switch (protoVer)
1065     {
1066     int res;
1067     case 6:
1068     case 7:
1069         ERR err;
1070         memset(&err, 0, sizeof(ERR));
1071
1072         sendAddr.sin_family = AF_INET;
1073         sendAddr.sin_port = htons(port);
1074
1075         sendAddr.sin_addr.s_addr = ip;// IP ÐÏÌØÚÏ×ÁÔÅÌÑ
1076
1077         err.len = 1;
1078         strncpy((char*)err.type, "ERR", 16);
1079         strncpy((char*)err.text, text.c_str(), MAX_MSG_LEN);
1080
1081         #ifdef ARCH_BE
1082         SwapBytes(err.len);
1083         #endif
1084
1085         res = sendto(listenSocket, &err, sizeof(err), 0, (struct sockaddr*)&sendAddr, sizeof(sendAddr));
1086         printfd(__FILE__, "SendError %d bytes sent\n", res);
1087         break;
1088
1089     case 8:
1090         ERR_8 err8;
1091         memset(&err8, 0, sizeof(ERR_8));
1092
1093         sendAddr.sin_family = AF_INET;
1094         sendAddr.sin_port = htons(port);
1095
1096         sendAddr.sin_addr.s_addr = ip;// IP ÐÏÌØÚÏ×ÁÔÅÌÑ
1097
1098         err8.len = 256;
1099         strncpy((char*)err8.type, "ERR", 16);
1100         strncpy((char*)err8.text, text.c_str(), MAX_MSG_LEN);
1101
1102         #ifdef ARCH_BE
1103         SwapBytes(err8.len);
1104         #endif
1105
1106         res = sendto(listenSocket, &err8, sizeof(err8), 0, (struct sockaddr*)&sendAddr, sizeof(sendAddr));
1107         printfd(__FILE__, "SendError_8 %d bytes sent\n", res);
1108         break;
1109     }
1110
1111 return 0;
1112 }
1113 //-----------------------------------------------------------------------------
1114 int AUTH_IA::Send(uint32_t ip, uint16_t port, const char * buffer, int len)
1115 {
1116 struct sockaddr_in sendAddr;
1117 int res;
1118
1119 sendAddr.sin_family = AF_INET;
1120 sendAddr.sin_port = htons(port);
1121 sendAddr.sin_addr.s_addr = ip;
1122
1123 res = sendto(listenSocket, buffer, len, 0, (struct sockaddr*)&sendAddr, sizeof(sendAddr));
1124
1125 static struct timeval tv;
1126 gettimeofday(&tv, NULL);
1127
1128 return res;
1129 }
1130 //-----------------------------------------------------------------------------
1131 void AUTH_IA::InitEncrypt(BLOWFISH_CTX * ctx, const string & password)
1132 {
1133 unsigned char keyL[PASSWD_LEN];  // ðÁÒÏÌØ ÄÌÑ ÛÉÆÒÏ×ËÉ
1134 memset(keyL, 0, PASSWD_LEN);
1135 strncpy((char *)keyL, password.c_str(), PASSWD_LEN);
1136 Blowfish_Init(ctx, keyL, PASSWD_LEN);
1137 }
1138 //-----------------------------------------------------------------------------
1139 void AUTH_IA::Decrypt(BLOWFISH_CTX * ctx, char * dst, const char * src, int len8)
1140 {
1141 // len8 - ÄÌÉÎÁ × 8-ÍÉ ÂÁÊÔÏ×ÙÈ ÂÌÏËÁÈ
1142
1143 for (int i = 0; i < len8; i++)
1144     DecodeString(dst + i * 8, src + i * 8, ctx);
1145 }
1146 //-----------------------------------------------------------------------------
1147 void AUTH_IA::Encrypt(BLOWFISH_CTX * ctx, char * dst, const char * src, int len8)
1148 {
1149 // len8 - ÄÌÉÎÁ × 8-ÍÉ ÂÁÊÔÏ×ÙÈ ÂÌÏËÁÈ
1150
1151 for (int i = 0; i < len8; i++)
1152     EncodeString(dst + i * 8, src + i * 8, ctx);
1153 }
1154 //-----------------------------------------------------------------------------
1155 int AUTH_IA::SendMessage(const STG_MSG & msg, uint32_t ip) const
1156 {
1157 STG_LOCKER lock(&mutex, __FILE__, __LINE__);
1158
1159 printfd(__FILE__, "SendMessage userIP=%s\n", inet_ntostring(ip).c_str());
1160
1161 map<uint32_t, IA_USER>::iterator it;
1162 it = ip2user.find(ip);
1163 if (it == ip2user.end())
1164     {
1165     errorStr = "Unknown user.";
1166     return -1;
1167     }
1168 it->second.messagesToSend.push_back(msg);
1169 return 0;
1170 }
1171 //-----------------------------------------------------------------------------
1172 int AUTH_IA::RealSendMessage6(const STG_MSG & msg, uint32_t ip, IA_USER & user)
1173 {
1174 printfd(__FILE__, "RealSendMessage 6 user=%s\n", user.user->GetLogin().c_str());
1175
1176 char buffer[256];
1177 INFO_6 info;
1178
1179 memset(&info, 0, sizeof(INFO_6));
1180
1181 info.len = 256;
1182 strncpy((char*)info.type, "INFO", 16);
1183 info.infoType = 'I';
1184 strncpy((char*)info.text, msg.text.c_str(), 235);
1185 info.text[234] = 0;
1186
1187 size_t len = info.len;
1188 #ifdef ARCH_BE
1189 SwapBytes(info.len);
1190 #endif
1191
1192 memcpy(buffer, &info, sizeof(INFO_6));
1193 Encrypt(&user.ctx, buffer, buffer, len / 8);
1194 Send(ip, iaSettings.GetUserPort(), buffer, len);
1195
1196 return 0;
1197 }
1198 //-----------------------------------------------------------------------------
1199 int AUTH_IA::RealSendMessage7(const STG_MSG & msg, uint32_t ip, IA_USER & user)
1200 {
1201 printfd(__FILE__, "RealSendMessage 7 user=%s\n", user.user->GetLogin().c_str());
1202
1203 char buffer[300];
1204 INFO_7 info;
1205
1206 memset(&info, 0, sizeof(INFO_7));
1207
1208 info.len = 264;
1209 strncpy((char*)info.type, "INFO_7", 16);
1210 info.infoType = msg.header.type;
1211 info.showTime = msg.header.showTime;
1212
1213 info.sendTime = msg.header.creationTime;
1214
1215 size_t len = info.len;
1216 #ifdef ARCH_BE
1217 SwapBytes(info.len);
1218 SwapBytes(info.sendTime);
1219 #endif
1220
1221 strncpy((char*)info.text, msg.text.c_str(), MAX_MSG_LEN - 1);
1222 info.text[MAX_MSG_LEN - 1] = 0;
1223
1224 memcpy(buffer, &info, sizeof(INFO_7));
1225
1226 Encrypt(&user.ctx, buffer, buffer, len / 8);
1227 Send(ip, iaSettings.GetUserPort(), buffer, len);
1228
1229 return 0;
1230 }
1231 //-----------------------------------------------------------------------------
1232 int AUTH_IA::RealSendMessage8(const STG_MSG & msg, uint32_t ip, IA_USER & user)
1233 {
1234 printfd(__FILE__, "RealSendMessage 8 user=%s\n", user.user->GetLogin().c_str());
1235
1236 char buffer[1500];
1237 memset(buffer, 0, sizeof(buffer));
1238
1239 INFO_8 info;
1240
1241 memset(&info, 0, sizeof(INFO_8));
1242
1243 info.len = 1056;
1244 strncpy((char*)info.type, "INFO_8", 16);
1245 info.infoType = msg.header.type;
1246 info.showTime = msg.header.showTime;
1247 info.sendTime = msg.header.creationTime;
1248
1249 strncpy((char*)info.text, msg.text.c_str(), IA_MAX_MSG_LEN_8 - 1);
1250 info.text[IA_MAX_MSG_LEN_8 - 1] = 0;
1251
1252 size_t len = info.len;
1253 #ifdef ARCH_BE
1254 SwapBytes(info.len);
1255 SwapBytes(info.sendTime);
1256 #endif
1257
1258 memcpy(buffer, &info, sizeof(INFO_8));
1259
1260 Encrypt(&user.ctx, buffer, buffer, len / 8);
1261 Send(ip, user.port, buffer, len);
1262
1263 return 0;
1264 }
1265 //-----------------------------------------------------------------------------
1266 int AUTH_IA::Process_CONN_SYN_6(CONN_SYN_6 *, IA_USER * iaUser, user_iter *, uint32_t)
1267 {
1268 if (!(iaUser->phase.GetPhase() == 1 || iaUser->phase.GetPhase() == 3))
1269     return -1;
1270
1271 enabledDirs = 0xFFffFFff;
1272
1273 iaUser->phase.SetPhase2();
1274 printfd(__FILE__, "Phase changed from %d to 2. Reason: CONN_SYN_6\n", iaUser->phase.GetPhase());
1275 return 0;
1276 }
1277 //-----------------------------------------------------------------------------
1278 int AUTH_IA::Process_CONN_SYN_7(CONN_SYN_7 * connSyn, IA_USER * iaUser, user_iter * user, uint32_t sip)
1279 {
1280 return Process_CONN_SYN_6(connSyn, iaUser, user, sip);
1281 }
1282 //-----------------------------------------------------------------------------
1283 int AUTH_IA::Process_CONN_SYN_8(CONN_SYN_8 * connSyn, IA_USER * iaUser, user_iter * user, uint32_t sip)
1284 {
1285 #ifdef ARCH_BE
1286 SwapBytes(connSyn->dirs);
1287 #endif
1288 int ret = Process_CONN_SYN_6((CONN_SYN_6*)connSyn, iaUser, user, sip);
1289 enabledDirs = connSyn->dirs;
1290 return ret;
1291 }
1292 //-----------------------------------------------------------------------------
1293 int AUTH_IA::Process_CONN_ACK_6(CONN_ACK_6 * connAck, IA_USER * iaUser, user_iter *, uint32_t sip)
1294 {
1295 #ifdef ARCH_BE
1296 SwapBytes(connAck->len);
1297 SwapBytes(connAck->rnd);
1298 #endif
1299 printfd( __FILE__, "CONN_ACK_6 %s\n", connAck->type);
1300 // ÕÓÔÁÎÏ×ÉÔØ ÎÏ×ÕÀ ÆÁÚÕ É ×ÒÅÍÑ É ÒÁÚÒÅÛÉÔØ ÉÎÅÔ
1301 if ((iaUser->phase.GetPhase() == 2) && (connAck->rnd == iaUser->rnd + 1))
1302     {
1303     iaUser->phase.UpdateTime();
1304
1305     iaUser->lastSendAlive = iaUser->phase.GetTime();
1306     if (iaUser->user->Authorize(sip, "", enabledDirs, this) == 0)
1307         {
1308         iaUser->phase.SetPhase3();
1309         printfd(__FILE__, "Phase changed from 2 to 3. Reason: CONN_ACK_6\n");
1310         return 0;
1311         }
1312     else
1313         {
1314         errorStr = iaUser->user->GetStrError();
1315         iaUser->phase.SetPhase1();
1316         printfd(__FILE__, "Phase changed from 2 to 1. Reason: failed to authorize user\n");
1317         return -1;
1318         }
1319     }
1320 printfd(__FILE__, "Invalid phase or control number. Phase: %d. Control number: %d\n", iaUser->phase.GetPhase(), connAck->rnd);
1321 return -1;
1322 }
1323 //-----------------------------------------------------------------------------
1324 int AUTH_IA::Process_CONN_ACK_7(CONN_ACK_7 * connAck, IA_USER * iaUser, user_iter * user, uint32_t sip)
1325 {
1326 return Process_CONN_ACK_6(connAck, iaUser, user, sip);
1327 }
1328 //-----------------------------------------------------------------------------
1329 int AUTH_IA::Process_CONN_ACK_8(CONN_ACK_8 * connAck, IA_USER * iaUser, user_iter *, uint32_t sip)
1330 {
1331 #ifdef ARCH_BE
1332 SwapBytes(connAck->len);
1333 SwapBytes(connAck->rnd);
1334 #endif
1335 printfd( __FILE__, "CONN_ACK_8 %s\n", connAck->type);
1336
1337 if ((iaUser->phase.GetPhase() == 2) && (connAck->rnd == iaUser->rnd + 1))
1338     {
1339     iaUser->phase.UpdateTime();
1340     iaUser->lastSendAlive = iaUser->phase.GetTime();
1341     if (iaUser->user->Authorize(sip, "", enabledDirs, this) == 0)
1342         {
1343         iaUser->phase.SetPhase3();
1344         printfd(__FILE__, "Phase changed from 2 to 3. Reason: CONN_ACK_8\n");
1345         return 0;
1346         }
1347     else
1348         {
1349         errorStr = iaUser->user->GetStrError();
1350         iaUser->phase.SetPhase1();
1351         printfd(__FILE__, "Phase changed from 2 to 1. Reason: failed to authorize user\n");
1352         return -1;
1353         }
1354     }
1355 printfd(__FILE__, "Invalid phase or control number. Phase: %d. Control number: %d\n", iaUser->phase.GetPhase(), connAck->rnd);
1356 return -1;
1357 }
1358 //-----------------------------------------------------------------------------
1359 int AUTH_IA::Process_ALIVE_ACK_6(ALIVE_ACK_6 * aliveAck, IA_USER * iaUser, user_iter *, uint32_t)
1360 {
1361 #ifdef ARCH_BE
1362 SwapBytes(aliveAck->len);
1363 SwapBytes(aliveAck->rnd);
1364 #endif
1365 printfd(__FILE__, "ALIVE_ACK_6\n");
1366 if ((iaUser->phase.GetPhase() == 3) && (aliveAck->rnd == iaUser->rnd + 1))
1367     {
1368     iaUser->phase.UpdateTime();
1369     #ifdef IA_DEBUG
1370     iaUser->aliveSent = false;
1371     #endif
1372     }
1373 return 0;
1374 }
1375 //-----------------------------------------------------------------------------
1376 int AUTH_IA::Process_ALIVE_ACK_7(ALIVE_ACK_7 * aliveAck, IA_USER * iaUser, user_iter * user, uint32_t sip)
1377 {
1378 return Process_ALIVE_ACK_6(aliveAck, iaUser, user, sip);
1379 }
1380 //-----------------------------------------------------------------------------
1381 int AUTH_IA::Process_ALIVE_ACK_8(ALIVE_ACK_8 * aliveAck, IA_USER * iaUser, user_iter *, uint32_t)
1382 {
1383 #ifdef ARCH_BE
1384 SwapBytes(aliveAck->len);
1385 SwapBytes(aliveAck->rnd);
1386 #endif
1387 printfd(__FILE__, "ALIVE_ACK_8\n");
1388 if ((iaUser->phase.GetPhase() == 3) && (aliveAck->rnd == iaUser->rnd + 1))
1389     {
1390     iaUser->phase.UpdateTime();
1391     #ifdef IA_DEBUG
1392     iaUser->aliveSent = false;
1393     #endif
1394     }
1395 return 0;
1396 }
1397 //-----------------------------------------------------------------------------
1398 int AUTH_IA::Process_DISCONN_SYN_6(DISCONN_SYN_6 *, IA_USER * iaUser, user_iter *, uint32_t)
1399 {
1400 printfd(__FILE__, "DISCONN_SYN_6\n");
1401 if (iaUser->phase.GetPhase() != 3)
1402     {
1403     printfd(__FILE__, "Invalid phase. Expected 3, actual %d\n", iaUser->phase.GetPhase());
1404     errorStr = "Incorrect request DISCONN_SYN";
1405     return -1;
1406     }
1407
1408 iaUser->phase.SetPhase4();
1409 printfd(__FILE__, "Phase changed from 3 to 4. Reason: DISCONN_SYN_6\n");
1410
1411 return 0;
1412 }
1413 //-----------------------------------------------------------------------------
1414 int AUTH_IA::Process_DISCONN_SYN_7(DISCONN_SYN_7 * disconnSyn, IA_USER * iaUser, user_iter * user, uint32_t sip)
1415 {
1416 return Process_DISCONN_SYN_6(disconnSyn, iaUser, user, sip);
1417 }
1418 //-----------------------------------------------------------------------------
1419 int AUTH_IA::Process_DISCONN_SYN_8(DISCONN_SYN_8 *, IA_USER * iaUser, user_iter *, uint32_t)
1420 {
1421 if (iaUser->phase.GetPhase() != 3)
1422     {
1423     errorStr = "Incorrect request DISCONN_SYN";
1424     printfd(__FILE__, "Invalid phase. Expected 3, actual %d\n", iaUser->phase.GetPhase());
1425     return -1;
1426     }
1427
1428 iaUser->phase.SetPhase4();
1429 printfd(__FILE__, "Phase changed from 3 to 4. Reason: DISCONN_SYN_6\n");
1430
1431 return 0;
1432 }
1433 //-----------------------------------------------------------------------------
1434 int AUTH_IA::Process_DISCONN_ACK_6(DISCONN_ACK_6 * disconnAck,
1435                                    IA_USER * iaUser,
1436                                    user_iter *,
1437                                    uint32_t,
1438                                    map<uint32_t, IA_USER>::iterator)
1439 {
1440 #ifdef ARCH_BE
1441 SwapBytes(disconnAck->len);
1442 SwapBytes(disconnAck->rnd);
1443 #endif
1444 printfd(__FILE__, "DISCONN_ACK_6\n");
1445 if (!((iaUser->phase.GetPhase() == 4) && (disconnAck->rnd == iaUser->rnd + 1)))
1446     {
1447     printfd(__FILE__, "Invalid phase or control number. Phase: %d. Control number: %d\n", iaUser->phase.GetPhase(), disconnAck->rnd);
1448     return -1;
1449     }
1450
1451 return 0;
1452 }
1453 //-----------------------------------------------------------------------------
1454 int AUTH_IA::Process_DISCONN_ACK_7(DISCONN_ACK_7 * disconnAck, IA_USER * iaUser, user_iter * user, uint32_t sip, map<uint32_t, IA_USER>::iterator it)
1455 {
1456 return Process_DISCONN_ACK_6(disconnAck, iaUser, user, sip, it);
1457 }
1458 //-----------------------------------------------------------------------------
1459 int AUTH_IA::Process_DISCONN_ACK_8(DISCONN_ACK_8 * disconnAck, IA_USER * iaUser, user_iter *, uint32_t, map<uint32_t, IA_USER>::iterator)
1460 {
1461 #ifdef ARCH_BE
1462 SwapBytes(disconnAck->len);
1463 SwapBytes(disconnAck->rnd);
1464 #endif
1465 printfd(__FILE__, "DISCONN_ACK_8\n");
1466 if (!((iaUser->phase.GetPhase() == 4) && (disconnAck->rnd == iaUser->rnd + 1)))
1467     {
1468     printfd(__FILE__, "Invalid phase or control number. Phase: %d. Control number: %d\n", iaUser->phase.GetPhase(), disconnAck->rnd);
1469     return -1;
1470     }
1471
1472 return 0;
1473 }
1474 //-----------------------------------------------------------------------------
1475 int AUTH_IA::Send_CONN_SYN_ACK_6(IA_USER * iaUser, user_iter *, uint32_t sip)
1476 {
1477 //+++ Fill static data in connSynAck +++
1478 // TODO Move this code. It must be executed only once
1479 connSynAck6.len = Min8(sizeof(CONN_SYN_ACK_6));
1480 strcpy((char*)connSynAck6.type, "CONN_SYN_ACK");
1481 for (int j = 0; j < DIR_NUM; j++)
1482     {
1483     strncpy((char*)connSynAck6.dirName[j],
1484             stgSettings->GetDirName(j).c_str(),
1485             sizeof(string16));
1486
1487     connSynAck6.dirName[j][sizeof(string16) - 1] = 0;
1488     }
1489 //--- Fill static data in connSynAck ---
1490
1491 iaUser->rnd = random();
1492 connSynAck6.rnd = iaUser->rnd;
1493
1494 connSynAck6.userTimeOut = iaSettings.GetUserTimeout();
1495 connSynAck6.aliveDelay = iaSettings.GetUserDelay();
1496
1497 #ifdef ARCH_BE
1498 SwapBytes(connSynAck6.len);
1499 SwapBytes(connSynAck6.rnd);
1500 SwapBytes(connSynAck6.userTimeOut);
1501 SwapBytes(connSynAck6.aliveDelay);
1502 #endif
1503
1504 Encrypt(&iaUser->ctx, (char*)&connSynAck6, (char*)&connSynAck6, Min8(sizeof(CONN_SYN_ACK_6))/8);
1505 return Send(sip, iaSettings.GetUserPort(), (char*)&connSynAck6, Min8(sizeof(CONN_SYN_ACK_6)));;
1506 }
1507 //-----------------------------------------------------------------------------
1508 int AUTH_IA::Send_CONN_SYN_ACK_7(IA_USER * iaUser, user_iter * user, uint32_t sip)
1509 {
1510 return Send_CONN_SYN_ACK_6(iaUser, user, sip);
1511 }
1512 //-----------------------------------------------------------------------------
1513 int AUTH_IA::Send_CONN_SYN_ACK_8(IA_USER * iaUser, user_iter *, uint32_t sip)
1514 {
1515 strcpy((char*)connSynAck8.hdr.magic, IA_ID);
1516 connSynAck8.hdr.protoVer[0] = 0;
1517 connSynAck8.hdr.protoVer[1] = 8;
1518
1519 //+++ Fill static data in connSynAck +++
1520 // TODO Move this code. It must be executed only once
1521 connSynAck8.len = Min8(sizeof(CONN_SYN_ACK_8));
1522 strcpy((char*)connSynAck8.type, "CONN_SYN_ACK");
1523 for (int j = 0; j < DIR_NUM; j++)
1524     {
1525     strncpy((char*)connSynAck8.dirName[j],
1526             stgSettings->GetDirName(j).c_str(),
1527             sizeof(string16));
1528
1529     connSynAck8.dirName[j][sizeof(string16) - 1] = 0;
1530     }
1531 //--- Fill static data in connSynAck ---
1532
1533 iaUser->rnd = random();
1534 connSynAck8.rnd = iaUser->rnd;
1535
1536 connSynAck8.userTimeOut = iaSettings.GetUserTimeout();
1537 connSynAck8.aliveDelay = iaSettings.GetUserDelay();
1538
1539 #ifdef ARCH_BE
1540 SwapBytes(connSynAck8.len);
1541 SwapBytes(connSynAck8.rnd);
1542 SwapBytes(connSynAck8.userTimeOut);
1543 SwapBytes(connSynAck8.aliveDelay);
1544 #endif
1545
1546 Encrypt(&iaUser->ctx, (char*)&connSynAck8, (char*)&connSynAck8, Min8(sizeof(CONN_SYN_ACK_8))/8);
1547 return Send(sip, iaUser->port, (char*)&connSynAck8, Min8(sizeof(CONN_SYN_ACK_8)));
1548 //return Send(sip, iaUser->port, (char*)&connSynAck8, 384);
1549 }
1550 //-----------------------------------------------------------------------------
1551 int AUTH_IA::Send_ALIVE_SYN_6(IA_USER * iaUser, uint32_t sip)
1552 {
1553 aliveSyn6.len = Min8(sizeof(ALIVE_SYN_6));
1554 aliveSyn6.rnd = iaUser->rnd = random();
1555
1556 strcpy((char*)aliveSyn6.type, "ALIVE_SYN");
1557
1558 for (int i = 0; i < DIR_NUM; i++)
1559     {
1560     aliveSyn6.md[i] = iaUser->user->property.down.Get()[i];
1561     aliveSyn6.mu[i] = iaUser->user->property.up.Get()[i];
1562
1563     aliveSyn6.sd[i] = iaUser->user->GetSessionDownload()[i];
1564     aliveSyn6.su[i] = iaUser->user->GetSessionUpload()[i];
1565     }
1566
1567 //TODO
1568 int dn = iaSettings.GetFreeMbShowType();
1569 const TARIFF * tf = iaUser->user->GetTariff();
1570
1571 if (dn < DIR_NUM)
1572     {
1573     double p = tf->GetPriceWithTraffType(aliveSyn6.mu[dn],
1574                                          aliveSyn6.md[dn],
1575                                          dn,
1576                                          stgTime);
1577     p *= (1024 * 1024);
1578     if (p == 0)
1579         {
1580         snprintf((char*)aliveSyn6.freeMb, IA_FREEMB_LEN, "---");
1581         }
1582     else
1583         {
1584         double fmb = iaUser->user->property.freeMb;
1585         fmb = fmb < 0 ? 0 : fmb;
1586         snprintf((char*)aliveSyn6.freeMb, IA_FREEMB_LEN, "%.3f", fmb / p);
1587         }
1588     }
1589 else
1590     {
1591     if (freeMbNone == iaSettings.GetFreeMbShowType())
1592         {
1593         aliveSyn6.freeMb[0] = 0;
1594         }
1595     else
1596         {
1597         double fmb = iaUser->user->property.freeMb;
1598         fmb = fmb < 0 ? 0 : fmb;
1599         snprintf((char*)aliveSyn6.freeMb, IA_FREEMB_LEN, "C%.3f", fmb);
1600         }
1601     }
1602
1603 #ifdef IA_DEBUG
1604 if (iaUser->aliveSent)
1605     {
1606     printfd(__FILE__, "========= ALIVE_ACK_6(7) TIMEOUT !!! %s =========\n", iaUser->user->GetLogin().c_str());
1607     }
1608 iaUser->aliveSent = true;
1609 #endif
1610
1611 aliveSyn6.cash =(int64_t) (iaUser->user->property.cash.Get() * 1000.0);
1612 if (!stgSettings->GetShowFeeInCash())
1613     aliveSyn6.cash -= (int64_t)(tf->GetFee() * 1000.0);
1614
1615 #ifdef ARCH_BE
1616 SwapBytes(aliveSyn6.len);
1617 SwapBytes(aliveSyn6.rnd);
1618 SwapBytes(aliveSyn6.cash);
1619 for (int i = 0; i < DIR_NUM; ++i)
1620     {
1621     SwapBytes(aliveSyn6.mu[i]);
1622     SwapBytes(aliveSyn6.md[i]);
1623     SwapBytes(aliveSyn6.su[i]);
1624     SwapBytes(aliveSyn6.sd[i]);
1625     }
1626 #endif
1627
1628 Encrypt(&(iaUser->ctx), (char*)&aliveSyn6, (char*)&aliveSyn6, Min8(sizeof(aliveSyn6))/8);
1629 return Send(sip, iaSettings.GetUserPort(), (char*)&aliveSyn6, Min8(sizeof(aliveSyn6)));
1630 }
1631 //-----------------------------------------------------------------------------
1632 int AUTH_IA::Send_ALIVE_SYN_7(IA_USER * iaUser, uint32_t sip)
1633 {
1634 return Send_ALIVE_SYN_6(iaUser, sip);
1635 }
1636 //-----------------------------------------------------------------------------
1637 int AUTH_IA::Send_ALIVE_SYN_8(IA_USER * iaUser, uint32_t sip)
1638 {
1639 strcpy((char*)aliveSyn8.hdr.magic, IA_ID);
1640 aliveSyn8.hdr.protoVer[0] = 0;
1641 aliveSyn8.hdr.protoVer[1] = 8;
1642
1643 aliveSyn8.len = Min8(sizeof(ALIVE_SYN_8));
1644 aliveSyn8.rnd = iaUser->rnd = random();
1645
1646 strcpy((char*)aliveSyn8.type, "ALIVE_SYN");
1647
1648 for (int i = 0; i < DIR_NUM; i++)
1649     {
1650     aliveSyn8.md[i] = iaUser->user->property.down.Get()[i];
1651     aliveSyn8.mu[i] = iaUser->user->property.up.Get()[i];
1652
1653     aliveSyn8.sd[i] = iaUser->user->GetSessionDownload()[i];
1654     aliveSyn8.su[i] = iaUser->user->GetSessionUpload()[i];
1655     }
1656
1657 //TODO
1658 int dn = iaSettings.GetFreeMbShowType();
1659
1660 if (dn < DIR_NUM)
1661     {
1662     const TARIFF * tf = iaUser->user->GetTariff();
1663     double p = tf->GetPriceWithTraffType(aliveSyn8.mu[dn],
1664                                          aliveSyn8.md[dn],
1665                                          dn,
1666                                          stgTime);
1667     p *= (1024 * 1024);
1668     if (p == 0)
1669         {
1670         snprintf((char*)aliveSyn8.freeMb, IA_FREEMB_LEN, "---");
1671         }
1672     else
1673         {
1674         double fmb = iaUser->user->property.freeMb;
1675         fmb = fmb < 0 ? 0 : fmb;
1676         snprintf((char*)aliveSyn8.freeMb, IA_FREEMB_LEN, "%.3f", fmb / p);
1677         }
1678     }
1679 else
1680     {
1681     if (freeMbNone == iaSettings.GetFreeMbShowType())
1682         {
1683         aliveSyn8.freeMb[0] = 0;
1684         }
1685     else
1686         {
1687         double fmb = iaUser->user->property.freeMb;
1688         fmb = fmb < 0 ? 0 : fmb;
1689         snprintf((char*)aliveSyn8.freeMb, IA_FREEMB_LEN, "C%.3f", fmb);
1690         }
1691     }
1692
1693 #ifdef IA_DEBUG
1694 if (iaUser->aliveSent)
1695     {
1696     printfd(__FILE__, "========= ALIVE_ACK_8 TIMEOUT !!! =========\n");
1697     }
1698 iaUser->aliveSent = true;
1699 #endif
1700
1701 const TARIFF * tf = iaUser->user->GetTariff();
1702
1703 aliveSyn8.cash =(int64_t) (iaUser->user->property.cash.Get() * 1000.0);
1704 if (!stgSettings->GetShowFeeInCash())
1705     aliveSyn8.cash -= (int64_t)(tf->GetFee() * 1000.0);
1706
1707 #ifdef ARCH_BE
1708 SwapBytes(aliveSyn8.len);
1709 SwapBytes(aliveSyn8.rnd);
1710 SwapBytes(aliveSyn8.cash);
1711 SwapBytes(aliveSyn8.status);
1712 for (int i = 0; i < DIR_NUM; ++i)
1713     {
1714     SwapBytes(aliveSyn8.mu[i]);
1715     SwapBytes(aliveSyn8.md[i]);
1716     SwapBytes(aliveSyn8.su[i]);
1717     SwapBytes(aliveSyn8.sd[i]);
1718     }
1719 #endif
1720
1721 Encrypt(&(iaUser->ctx), (char*)&aliveSyn8, (char*)&aliveSyn8, Min8(sizeof(aliveSyn8))/8);
1722 return Send(sip, iaUser->port, (char*)&aliveSyn8, Min8(sizeof(aliveSyn8)));
1723 }
1724 //-----------------------------------------------------------------------------
1725 int AUTH_IA::Send_DISCONN_SYN_ACK_6(IA_USER * iaUser, uint32_t sip)
1726 {
1727 disconnSynAck6.len = Min8(sizeof(DISCONN_SYN_ACK_6));
1728 strcpy((char*)disconnSynAck6.type, "DISCONN_SYN_ACK");
1729 disconnSynAck6.rnd = iaUser->rnd = random();
1730
1731 #ifdef ARCH_BE
1732 SwapBytes(disconnSynAck6.len);
1733 SwapBytes(disconnSynAck6.rnd);
1734 #endif
1735
1736 Encrypt(&iaUser->ctx, (char*)&disconnSynAck6, (char*)&disconnSynAck6, Min8(sizeof(disconnSynAck6))/8);
1737 return Send(sip, iaSettings.GetUserPort(), (char*)&disconnSynAck6, Min8(sizeof(disconnSynAck6)));
1738 }
1739 //-----------------------------------------------------------------------------
1740 int AUTH_IA::Send_DISCONN_SYN_ACK_7(IA_USER * iaUser, uint32_t sip)
1741 {
1742 return Send_DISCONN_SYN_ACK_6(iaUser, sip);
1743 }
1744 //-----------------------------------------------------------------------------
1745 int AUTH_IA::Send_DISCONN_SYN_ACK_8(IA_USER * iaUser, uint32_t sip)
1746 {
1747 strcpy((char*)disconnSynAck8.hdr.magic, IA_ID);
1748 disconnSynAck8.hdr.protoVer[0] = 0;
1749 disconnSynAck8.hdr.protoVer[1] = 8;
1750
1751 disconnSynAck8.len = Min8(sizeof(DISCONN_SYN_ACK_8));
1752 strcpy((char*)disconnSynAck8.type, "DISCONN_SYN_ACK");
1753 disconnSynAck8.rnd = iaUser->rnd = random();
1754
1755 #ifdef ARCH_BE
1756 SwapBytes(disconnSynAck8.len);
1757 SwapBytes(disconnSynAck8.rnd);
1758 #endif
1759
1760 Encrypt(&iaUser->ctx, (char*)&disconnSynAck8, (char*)&disconnSynAck8, Min8(sizeof(disconnSynAck8))/8);
1761 return Send(sip, iaUser->port, (char*)&disconnSynAck8, Min8(sizeof(disconnSynAck8)));
1762 }
1763 //-----------------------------------------------------------------------------
1764 int AUTH_IA::Send_FIN_6(IA_USER * iaUser, uint32_t sip, map<uint32_t, IA_USER>::iterator it)
1765 {
1766 fin6.len = Min8(sizeof(FIN_6));
1767 strcpy((char*)fin6.type, "FIN");
1768 strcpy((char*)fin6.ok, "OK");
1769
1770 #ifdef ARCH_BE
1771 SwapBytes(fin6.len);
1772 #endif
1773
1774 Encrypt(&iaUser->ctx, (char*)&fin6, (char*)&fin6, Min8(sizeof(fin6))/8);
1775
1776 iaUser->user->Unauthorize(this);
1777
1778 int ret = Send(sip, iaSettings.GetUserPort(), (char*)&fin6, Min8(sizeof(fin6)));
1779 ip2user.erase(it);
1780 return ret;
1781 }
1782 //-----------------------------------------------------------------------------
1783 int AUTH_IA::Send_FIN_7(IA_USER * iaUser, uint32_t sip, map<uint32_t, IA_USER>::iterator it)
1784 {
1785 return Send_FIN_6(iaUser, sip, it);
1786 }
1787 //-----------------------------------------------------------------------------
1788 int AUTH_IA::Send_FIN_8(IA_USER * iaUser, uint32_t sip, map<uint32_t, IA_USER>::iterator it)
1789 {
1790 strcpy((char*)fin8.hdr.magic, IA_ID);
1791 fin8.hdr.protoVer[0] = 0;
1792 fin8.hdr.protoVer[1] = 8;
1793
1794 fin8.len = Min8(sizeof(FIN_8));
1795 strcpy((char*)fin8.type, "FIN");
1796 strcpy((char*)fin8.ok, "OK");
1797
1798 #ifdef ARCH_BE
1799 SwapBytes(fin8.len);
1800 #endif
1801
1802 Encrypt(&iaUser->ctx, (char*)&fin8, (char*)&fin8, Min8(sizeof(fin8))/8);
1803
1804 iaUser->user->Unauthorize(this);
1805
1806 int ret = Send(sip, iaUser->port, (char*)&fin8, Min8(sizeof(fin8)));
1807 ip2user.erase(it);
1808 return ret;
1809 }
1810 //-----------------------------------------------------------------------------
1811 bool AUTH_IA::WaitPackets(int sd) const
1812 {
1813 fd_set rfds;
1814 FD_ZERO(&rfds);
1815 FD_SET(sd, &rfds);
1816
1817 struct timeval tv;
1818 tv.tv_sec = 0;
1819 tv.tv_usec = 500000;
1820
1821 int res = select(sd + 1, &rfds, NULL, NULL, &tv);
1822 if (res == -1) // Error
1823     {
1824     if (errno != EINTR)
1825         {
1826         printfd(__FILE__, "Error on select: '%s'\n", strerror(errno));
1827         }
1828     return false;
1829     }
1830
1831 if (res == 0) // Timeout
1832     {
1833     return false;
1834     }
1835
1836 return true;
1837 }