]> git.stg.codes - stg.git/blob - libs/ia/include/stg/ia.h
11030a7eeeda9c65f2daa6abc72fb8e01d7d9aba
[stg.git] / libs / ia / include / stg / 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 #pragma once
28
29 #include "stg/blowfish.h"
30 #include "stg/ia_packets.h"
31
32 #include <string>
33 #include <vector>
34 #include <map>
35
36 #ifndef WIN32
37 #include <sys/types.h>
38 #include <sys/socket.h>
39 #include <netinet/in.h>
40 #include <pthread.h>
41 #else
42 #include <winsock2.h>
43 #endif
44
45 #define IA_BIND_ERROR           (1)
46 #define IA_SERVER_ERROR         (2)
47 #define IA_FCNTL_ERROR          (3)
48 #define IA_GETHOSTBYNAME_ERROR  (4)
49
50 #define IA_PROTO_VER            (8)
51 #define IA_PROTO_PROXY_VER      (101)
52
53 typedef void (*tpStatusChangedCb)(int status, void * data);
54 typedef void (*tpStatChangedCb)(const LOADSTAT & stat, void * data);
55 typedef void (*tpCallBackInfoFn)(const std::string & message, int infoType, int showTime, int sendTime, void * data);
56 typedef void (*tpCallBackErrorFn)(const std::string & message, int netError, void * data);
57 typedef void (*tpCallBackDirNameFn)(const std::vector<std::string> & dirName, void * data);
58
59 //---------------------------------------------------------------------------
60 class IA_CLIENT_PROT
61 {
62 #ifdef WIN32
63 friend unsigned long WINAPI RunW(void * data);
64 #else
65 friend void * RunL(void * data);
66 #endif
67
68     public:
69         IA_CLIENT_PROT(const std::string & sn, uint16_t p, const std::string & localName = "", uint16_t localPort = 0);
70         ~IA_CLIENT_PROT();
71
72         void        Start();
73         void        Stop();
74         void        GetStat(LOADSTAT * ls);
75
76         void        SetServer(const std::string & sn, unsigned short port);
77         void        SetLogin(const std::string & login);
78         void        SetPassword(const std::string & password);
79         void        SetEnabledDirs(const bool * selectedDirs);
80
81         void        SetStatusChangedCb(tpStatusChangedCb p, void * data);
82         void        SetStatChangedCb(tpStatChangedCb p, void * data);
83         void        SetInfoCb(tpCallBackInfoFn p, void * data);
84         void        SetErrorCb(tpCallBackErrorFn p, void * data);
85         void        SetDirNameCb(tpCallBackDirNameFn p, void * data);
86
87         int         Connect();
88         int         Disconnect();
89         int         GetAuthorized() const { return m_phase == 3 || m_phase == 4; };
90         int         GetPhase() const { return m_phase; };
91         int         GetStatus() const;
92         int         GetReconnect() const { return m_reconnect; };
93         void        SetReconnect(int r) { m_reconnect = r; };
94         char        GetProtoVer() const { return m_proxyMode ? IA_PROTO_PROXY_VER : IA_PROTO_VER; };
95         void        GetMessageText(std::string * text) const { *text = m_messageText; };
96         void        GetInfoText(std::string * text) const { *text = m_infoText; };
97         int         GetStrError(std::string * error) const;
98
99         void        SetProxyMode(bool on) { m_proxyMode = on; };
100         bool        GetProxyMode() const { return m_proxyMode; };
101
102         void        SetIP(uint32_t ip) { m_ip = ip; };
103         uint32_t    GetIP() const { return m_ip; };
104
105     private:
106         void            Run();
107         int             NetRecv();
108         int             NetSend(int n);
109         bool            GetNonstop() const { return m_nonstop; };
110         void            PrepareNet();
111         int             DeterminatePacketType(const char * buffer);
112
113         int             Process_CONN_SYN_ACK_8(const void* buffer);
114         int             Process_ALIVE_SYN_8(const void* buffer);
115         int             Process_DISCONN_SYN_ACK_8(const void* buffer);
116         int             Process_FIN_8(const void* buffer);
117         int             Process_INFO_8(const void* buffer);
118         int             Process_ERROR(const void* buffer);
119
120         int             Prepare_CONN_SYN_8(void* buffer);
121         int             Prepare_CONN_ACK_8(void* buffer);
122         int             Prepare_ALIVE_ACK_8(void* buffer);
123         int             Prepare_DISCONN_SYN_8(void* buffer);
124         int             Prepare_DISCONN_ACK_8(void* buffer);
125
126         void            FillHdr8(char* buffer, unsigned long ip);
127         int             Send(char * buffer, int len);
128         int             Recv(char * buffer, int len);
129
130         LOADSTAT        m_stat;
131         int             m_action;
132         int             m_phase;
133         int             m_phaseTime;
134         std::string     m_messageText;
135         std::string     m_infoText;
136         mutable std::string m_strError;
137         mutable int     m_codeError;
138         bool            m_nonstop;
139         bool            m_isNetPrepared;
140         bool            m_proxyMode;
141
142         BLOWFISH_CTX    m_ctxPass;
143         BLOWFISH_CTX    m_ctxHdr;
144
145         bool            m_selectedDirs[DIR_NUM];
146
147         std::string     m_password;
148         std::string     m_login;
149
150         #ifdef WIN32
151         WSADATA m_wsaData;
152         #else
153         pthread_t m_thread;
154         #endif
155
156         std::string     m_serverName;
157         uint16_t        m_port;
158         uint32_t        m_ip;
159         std::string     m_localName;
160         uint32_t        m_localIP;
161         uint32_t        m_localPort;
162
163         struct sockaddr_in  m_servAddr;
164
165         bool            m_firstConnect;
166         int             m_reconnect;
167         int             m_sockr;
168         int             m_protNum;
169         int             m_userTimeout;
170         int             m_aliveTimeout;
171         unsigned int    m_rnd;
172
173         tpStatusChangedCb   m_pStatusChangedCb;
174         tpStatChangedCb     m_pStatChangedCb;
175         tpCallBackInfoFn    m_pInfoCb;
176         tpCallBackErrorFn   m_pErrorCb;
177         tpCallBackDirNameFn m_pDirNameCb;
178
179         void              * m_statusChangedCbData;
180         void              * m_statChangedCbData;
181         void              * m_infoCbData;
182         void              * m_errorCbData;
183         void              * m_dirNameCbData;
184
185         std::map<std::string, int> m_packetTypes;
186
187         CONN_SYN_8        * m_connSyn8;
188         const CONN_SYN_ACK_8    * m_connSynAck8;
189         CONN_ACK_8        * m_connAck8;
190         const ALIVE_SYN_8       * m_aliveSyn8;
191         ALIVE_ACK_8       * m_aliveAck8;
192         DISCONN_SYN_8     * m_disconnSyn8;
193         const DISCONN_SYN_ACK_8 * m_disconnSynAck8;
194         DISCONN_ACK_8     * m_disconnAck8;
195         const INFO_8            * m_info;
196 };
197 //---------------------------------------------------------------------------
198 #ifdef WIN32
199 unsigned long WINAPI RunW(void *);
200 #else
201 void * RunW(void *);
202 #endif