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