]> git.stg.codes - stg.git/blob - projects/stargazer/plugins/authorization/inetaccess/inetaccess.cpp
В утилите экранирования строки модуля store_postgresql исправлено
[stg.git] / 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 }