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