]> git.stg.codes - stg.git/blob - stglibs/ia_auth_c.lib/ia_auth_c.h
0a916f4cb4f86a3bc7c7e6b499d75f753a3d926f
[stg.git] / stglibs / ia_auth_c.lib / ia_auth_c.h
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 1, or (at your option)
5 ** 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., 675 Mass Ave, Cambridge, MA 02139, USA.
15 */
16
17 /*
18  $Author: faust $
19  $Revision: 1.10 $
20  $Date: 2010/03/15 12:57:24 $
21 */
22
23 /*
24 * Author : Boris Mikhailenko <stg34@stargazer.dp.ua>
25 */
26 //---------------------------------------------------------------------------
27 #ifndef IA_AUTH_C_H
28 #define IA_AUTH_C_H
29
30 #ifndef WIN32
31 #include <netinet/in.h>
32 #include <pthread.h>
33 #else
34 #include <winsock2.h>
35 #endif
36
37 #include <string>
38 #include <vector>
39 #include <map>
40
41 #include "stg/blowfish.h"
42 #include "stg/ia_packets.h"
43
44 #define IA_BIND_ERROR           (1)
45 #define IA_SERVER_ERROR         (2)
46 #define IA_FCNTL_ERROR          (3)
47 #define IA_GETHOSTBYNAME_ERROR  (4)
48
49 #define IA_PROTO_VER            (8)
50 #define IA_PROTO_PROXY_VER      (101)
51
52 using namespace std;
53
54 typedef void (*tpStatusChangedCb)(int status, void * data);
55 typedef void (*tpStatChangedCb)(const LOADSTAT & stat, void * data);
56 typedef void (*tpCallBackInfoFn)(const string & message, int infoType, int showTime, int sendTime, void * data);
57 typedef void (*tpCallBackErrorFn)(const string & message, int netError, void * data);
58 typedef void (*tpCallBackDirNameFn)(const vector<string> & dirName, void * data);
59
60 //---------------------------------------------------------------------------
61 class IA_CLIENT_PROT
62 {
63 #ifdef WIN32
64 friend unsigned long WINAPI RunW(void * data);
65 #else
66 friend void * RunL(void * data);
67 #endif
68
69 public:
70     IA_CLIENT_PROT(const string & sn, uint16_t p, uint16_t localPort = 0);
71     ~IA_CLIENT_PROT();
72
73     void        Start();
74     void        Stop();
75     void        GetStat(LOADSTAT * ls);
76
77     void        SetServer(const string & sn, unsigned short port);
78     void        SetLogin(const string & login);
79     void        SetPassword(const string & password);
80     void        SetEnabledDirs(const bool * selectedDirs);
81
82     void        SetStatusChangedCb(tpStatusChangedCb p, void * data);
83     void        SetStatChangedCb(tpStatChangedCb p, void * data);
84     void        SetInfoCb(tpCallBackInfoFn p, void * data);
85     void        SetErrorCb(tpCallBackErrorFn p, void * data);
86     void        SetDirNameCb(tpCallBackDirNameFn p, void * data);
87
88     int         Connect();
89     int         Disconnect();
90     int         GetAuthorized() const { return phase == 3 || phase == 4; };
91     int         GetPhase() const { return phase; };
92     int         GetStatus() const;
93     int         GetReconnect() const { return reconnect; };
94     void        SetReconnect(int r) { reconnect = r; };
95     char        GetProtoVer() const { return proxyMode ? IA_PROTO_PROXY_VER : IA_PROTO_VER; };
96     void        GetMessageText(string * text) const { *text = messageText; };
97     void        GetInfoText(string * text) const { *text = infoText; };
98     int         GetStrError(string * error) const;
99
100     void        SetProxyMode(bool on) { proxyMode = on; };
101     bool        GetProxyMode() const { return proxyMode; };
102
103     void        SetIP(uint32_t ip) { IA_CLIENT_PROT::ip = ip; };
104     uint32_t    GetIP() const { return ip; };
105
106 private:
107     void            Run();
108     int             NetRecv();
109     int             NetSend(int n);
110     bool            GetNonstop() const { return nonstop; };
111     void            PrepareNet();
112     int             DeterminatePacketType(const char * buffer);
113
114     int             Process_CONN_SYN_ACK_8(const char * buffer);
115     int             Process_ALIVE_SYN_8(const char * buffer);
116     int             Process_DISCONN_SYN_ACK_8(const char * buffer);
117     int             Process_FIN_8(const char * buffer);
118     int             Process_INFO_8(const char * buffer);
119     int             Process_ERROR(const char * buffer);
120
121     int             Prepare_CONN_SYN_8(char * buffer);
122     int             Prepare_CONN_ACK_8(char * buffer);
123     int             Prepare_ALIVE_ACK_8(char * buffer);
124     int             Prepare_DISCONN_SYN_8(char * buffer);
125     int             Prepare_DISCONN_ACK_8(char * buffer);
126
127     void            FillHdr8(char * buffer, unsigned long ip);
128     int             Send(char * buffer, int len);
129     int             Recv(char * buffer, int len);
130
131     LOADSTAT        stat;
132     int             action;
133     int             phase;
134     int             phaseTime;
135     string          messageText;
136     string          infoText;
137     mutable string  strError;
138     mutable int     codeError;
139     bool            nonstop;
140     bool            isNetPrepared;
141     bool            proxyMode;
142
143     BLOWFISH_CTX    ctxPass;
144     BLOWFISH_CTX    ctxHdr;
145
146     bool            selectedDirs[DIR_NUM];
147
148     string          password;
149     string          login;
150
151     #ifdef WIN32
152     WSADATA wsaData;
153     #else
154     pthread_t thread;
155     #endif
156
157     string          serverName;
158     uint16_t        port;
159     uint32_t        ip;
160     uint32_t        localPort;
161
162     struct sockaddr_in  servAddr;
163
164     bool            firstConnect;
165     int             reconnect;
166     int             sockr;
167     int             protNum;
168     int             userTimeout;
169     int             aliveTimeout;
170     unsigned int    rnd;
171
172     tpStatusChangedCb   pStatusChangedCb;
173     tpStatChangedCb     pStatChangedCb;
174     tpCallBackInfoFn    pInfoCb;
175     tpCallBackErrorFn   pErrorCb;
176     tpCallBackDirNameFn pDirNameCb;
177
178     void              * statusChangedCbData;
179     void              * statChangedCbData;
180     void              * infoCbData;
181     void              * errorCbData;
182     void              * dirNameCbData;
183
184     map<string, int>    packetTypes;
185
186     CONN_SYN_8        * connSyn8;
187     CONN_SYN_ACK_8    * connSynAck8;
188     CONN_ACK_8        * connAck8;
189     ALIVE_SYN_8       * aliveSyn8;
190     ALIVE_ACK_8       * aliveAck8;
191     DISCONN_SYN_8     * disconnSyn8;
192     DISCONN_SYN_ACK_8 * disconnSynAck8;
193     DISCONN_ACK_8     * disconnAck8;
194     INFO_8            * info;
195 };
196 //---------------------------------------------------------------------------
197 #ifdef WIN32
198 unsigned long WINAPI RunW(void *);
199 #else
200 void * RunW(void *);
201 #endif
202
203 //---------------------------------------------------------------------------
204 #endif //IA_AUTH_C_H