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