]> git.stg.codes - stg.git/blobdiff - libs/ia/include/stg/ia.h
Fix build on Darwin.
[stg.git] / libs / ia / include / stg / ia.h
index c37f4394a6aa67190990fb5c762b6eabc7ef68b3..48baed66568658e2d816d35211cd5ff2104b4964 100644 (file)
 * Author : Boris Mikhailenko <stg34@stargazer.dp.ua>
 */
 //---------------------------------------------------------------------------
 * Author : Boris Mikhailenko <stg34@stargazer.dp.ua>
 */
 //---------------------------------------------------------------------------
-#ifndef IA_AUTH_C_H
-#define IA_AUTH_C_H
+#pragma once
+
+#include "stg/blowfish.h"
+#include "stg/ia_packets.h"
+
+#include <string>
+#include <vector>
+#include <map>
+#pragma GCC diagnostic push
+#pragma GCC diagnostic ignored "-Wshadow"
+#include <jthread.hpp>
+#pragma GCC diagnostic pop
 
 #ifndef WIN32
 #include <sys/types.h>
 #include <sys/socket.h>
 #include <netinet/in.h>
 
 #ifndef WIN32
 #include <sys/types.h>
 #include <sys/socket.h>
 #include <netinet/in.h>
-#include <pthread.h>
 #else
 #include <winsock2.h>
 #endif
 
 #else
 #include <winsock2.h>
 #endif
 
-#include <string>
-#include <vector>
-#include <map>
-
-#include "stg/blowfish.h"
-#include "stg/ia_packets.h"
-
 #define IA_BIND_ERROR           (1)
 #define IA_SERVER_ERROR         (2)
 #define IA_FCNTL_ERROR          (3)
 #define IA_BIND_ERROR           (1)
 #define IA_SERVER_ERROR         (2)
 #define IA_FCNTL_ERROR          (3)
@@ -60,147 +62,131 @@ typedef void (*tpCallBackDirNameFn)(const std::vector<std::string> & dirName, vo
 //---------------------------------------------------------------------------
 class IA_CLIENT_PROT
 {
 //---------------------------------------------------------------------------
 class IA_CLIENT_PROT
 {
-#ifdef WIN32
-friend unsigned long WINAPI RunW(void * data);
-#else
-friend void * RunL(void * data);
-#endif
-
-public:
-    IA_CLIENT_PROT(const std::string & sn, uint16_t p, const std::string & localName = "", uint16_t localPort = 0);
-    ~IA_CLIENT_PROT();
-
-    void        Start();
-    void        Stop();
-    void        GetStat(LOADSTAT * ls);
-
-    void        SetServer(const std::string & sn, unsigned short port);
-    void        SetLogin(const std::string & login);
-    void        SetPassword(const std::string & password);
-    void        SetEnabledDirs(const bool * selectedDirs);
-
-    void        SetStatusChangedCb(tpStatusChangedCb p, void * data);
-    void        SetStatChangedCb(tpStatChangedCb p, void * data);
-    void        SetInfoCb(tpCallBackInfoFn p, void * data);
-    void        SetErrorCb(tpCallBackErrorFn p, void * data);
-    void        SetDirNameCb(tpCallBackDirNameFn p, void * data);
-
-    int         Connect();
-    int         Disconnect();
-    int         GetAuthorized() const { return phase == 3 || phase == 4; };
-    int         GetPhase() const { return phase; };
-    int         GetStatus() const;
-    int         GetReconnect() const { return reconnect; };
-    void        SetReconnect(int r) { reconnect = r; };
-    char        GetProtoVer() const { return proxyMode ? IA_PROTO_PROXY_VER : IA_PROTO_VER; };
-    void        GetMessageText(std::string * text) const { *text = messageText; };
-    void        GetInfoText(std::string * text) const { *text = infoText; };
-    int         GetStrError(std::string * error) const;
-
-    void        SetProxyMode(bool on) { proxyMode = on; };
-    bool        GetProxyMode() const { return proxyMode; };
-
-    void        SetIP(uint32_t ip) { IA_CLIENT_PROT::ip = ip; };
-    uint32_t    GetIP() const { return ip; };
-
-private:
-    void            Run();
-    int             NetRecv();
-    int             NetSend(int n);
-    bool            GetNonstop() const { return nonstop; };
-    void            PrepareNet();
-    int             DeterminatePacketType(const char * buffer);
-
-    int             Process_CONN_SYN_ACK_8(const char * buffer);
-    int             Process_ALIVE_SYN_8(const char * buffer);
-    int             Process_DISCONN_SYN_ACK_8(const char * buffer);
-    int             Process_FIN_8(const char * buffer);
-    int             Process_INFO_8(const char * buffer);
-    int             Process_ERROR(const char * buffer);
-
-    int             Prepare_CONN_SYN_8(char * buffer);
-    int             Prepare_CONN_ACK_8(char * buffer);
-    int             Prepare_ALIVE_ACK_8(char * buffer);
-    int             Prepare_DISCONN_SYN_8(char * buffer);
-    int             Prepare_DISCONN_ACK_8(char * buffer);
-
-    void            FillHdr8(char * buffer, unsigned long ip);
-    int             Send(char * buffer, int len);
-    int             Recv(char * buffer, int len);
-
-    LOADSTAT        stat;
-    int             action;
-    int             phase;
-    int             phaseTime;
-    std::string     messageText;
-    std::string     infoText;
-    mutable std::string strError;
-    mutable int     codeError;
-    bool            nonstop;
-    bool            isNetPrepared;
-    bool            proxyMode;
-
-    BLOWFISH_CTX    ctxPass;
-    BLOWFISH_CTX    ctxHdr;
-
-    bool            selectedDirs[DIR_NUM];
-
-    std::string     password;
-    std::string     login;
-
-    #ifdef WIN32
-    WSADATA wsaData;
-    #else
-    pthread_t thread;
-    #endif
-
-    std::string     serverName;
-    uint16_t        port;
-    uint32_t        ip;
-    std::string     localName;
-    uint32_t        localIP;
-    uint32_t        localPort;
-
-    struct sockaddr_in  servAddr;
-
-    bool            firstConnect;
-    int             reconnect;
-    int             sockr;
-    int             protNum;
-    int             userTimeout;
-    int             aliveTimeout;
-    unsigned int    rnd;
-
-    tpStatusChangedCb   pStatusChangedCb;
-    tpStatChangedCb     pStatChangedCb;
-    tpCallBackInfoFn    pInfoCb;
-    tpCallBackErrorFn   pErrorCb;
-    tpCallBackDirNameFn pDirNameCb;
-
-    void              * statusChangedCbData;
-    void              * statChangedCbData;
-    void              * infoCbData;
-    void              * errorCbData;
-    void              * dirNameCbData;
-
-    std::map<std::string, int> packetTypes;
-
-    CONN_SYN_8        * connSyn8;
-    CONN_SYN_ACK_8    * connSynAck8;
-    CONN_ACK_8        * connAck8;
-    ALIVE_SYN_8       * aliveSyn8;
-    ALIVE_ACK_8       * aliveAck8;
-    DISCONN_SYN_8     * disconnSyn8;
-    DISCONN_SYN_ACK_8 * disconnSynAck8;
-    DISCONN_ACK_8     * disconnAck8;
-    INFO_8            * info;
+    public:
+        IA_CLIENT_PROT(const std::string & sn, uint16_t p, const std::string & localName = "", uint16_t localPort = 0);
+        ~IA_CLIENT_PROT();
+
+        void        Start();
+        void        Stop();
+        void        GetStat(LOADSTAT * ls);
+
+        void        SetServer(const std::string & sn, unsigned short port);
+        void        SetLogin(const std::string & login);
+        void        SetPassword(const std::string & password);
+        void        SetEnabledDirs(const bool * selectedDirs);
+
+        void        SetStatusChangedCb(tpStatusChangedCb p, void * data);
+        void        SetStatChangedCb(tpStatChangedCb p, void * data);
+        void        SetInfoCb(tpCallBackInfoFn p, void * data);
+        void        SetErrorCb(tpCallBackErrorFn p, void * data);
+        void        SetDirNameCb(tpCallBackDirNameFn p, void * data);
+
+        int         Connect();
+        int         Disconnect();
+        int         GetAuthorized() const { return m_phase == 3 || m_phase == 4; };
+        int         GetPhase() const { return m_phase; };
+        int         GetStatus() const;
+        int         GetReconnect() const { return m_reconnect; };
+        void        SetReconnect(int r) { m_reconnect = r; };
+        char        GetProtoVer() const { return m_proxyMode ? IA_PROTO_PROXY_VER : IA_PROTO_VER; };
+        void        GetMessageText(std::string * text) const { *text = m_messageText; };
+        void        GetInfoText(std::string * text) const { *text = m_infoText; };
+        int         GetStrError(std::string * error) const;
+
+        void        SetProxyMode(bool on) { m_proxyMode = on; };
+        bool        GetProxyMode() const { return m_proxyMode; };
+
+        void        SetIP(uint32_t ip) { m_ip = ip; };
+        uint32_t    GetIP() const { return m_ip; };
+
+    private:
+        void            Run(std::stop_token token);
+        int             NetRecv();
+        int             NetSend(int n);
+        void            PrepareNet();
+        int             DeterminatePacketType(const char * buffer);
+
+        int             Process_CONN_SYN_ACK_8(const void* buffer);
+        int             Process_ALIVE_SYN_8(const void* buffer);
+        int             Process_DISCONN_SYN_ACK_8(const void* buffer);
+        int             Process_FIN_8(const void* buffer);
+        int             Process_INFO_8(const void* buffer);
+        int             Process_ERROR(const void* buffer);
+
+        int             Prepare_CONN_SYN_8(void* buffer);
+        int             Prepare_CONN_ACK_8(void* buffer);
+        int             Prepare_ALIVE_ACK_8(void* buffer);
+        int             Prepare_DISCONN_SYN_8(void* buffer);
+        int             Prepare_DISCONN_ACK_8(void* buffer);
+
+        void            FillHdr8(char* buffer, unsigned long ip);
+        int             Send(char * buffer, int len);
+        int             Recv(char * buffer, int len);
+
+        LOADSTAT        m_stat;
+        int             m_action;
+        int             m_phase;
+        int             m_phaseTime;
+        std::string     m_messageText;
+        std::string     m_infoText;
+        mutable std::string m_strError;
+        mutable int     m_codeError;
+        bool            m_isNetPrepared;
+        bool            m_proxyMode;
+
+        BLOWFISH_CTX    m_ctxPass;
+        BLOWFISH_CTX    m_ctxHdr;
+
+        bool            m_selectedDirs[DIR_NUM];
+
+        std::string     m_password;
+        std::string     m_login;
+
+        #ifdef WIN32
+        WSADATA m_wsaData;
+        #else
+        std::jthread m_thread;
+        #endif
+
+        std::string     m_serverName;
+        uint16_t        m_port;
+        uint32_t        m_ip;
+        std::string     m_localName;
+        uint32_t        m_localIP;
+        uint32_t        m_localPort;
+
+        struct sockaddr_in  m_servAddr;
+
+        bool            m_firstConnect;
+        int             m_reconnect;
+        int             m_sockr;
+        int             m_protNum;
+        int             m_userTimeout;
+        int             m_aliveTimeout;
+        unsigned int    m_rnd;
+
+        tpStatusChangedCb   m_pStatusChangedCb;
+        tpStatChangedCb     m_pStatChangedCb;
+        tpCallBackInfoFn    m_pInfoCb;
+        tpCallBackErrorFn   m_pErrorCb;
+        tpCallBackDirNameFn m_pDirNameCb;
+
+        void              * m_statusChangedCbData;
+        void              * m_statChangedCbData;
+        void              * m_infoCbData;
+        void              * m_errorCbData;
+        void              * m_dirNameCbData;
+
+        std::map<std::string, int> m_packetTypes;
+
+        CONN_SYN_8        * m_connSyn8;
+        const CONN_SYN_ACK_8 * m_connSynAck8;
+        CONN_ACK_8        * m_connAck8;
+        const ALIVE_SYN_8 * m_aliveSyn8;
+        ALIVE_ACK_8       * m_aliveAck8;
+        DISCONN_SYN_8     * m_disconnSyn8;
+        const DISCONN_SYN_ACK_8 * m_disconnSynAck8;
+        DISCONN_ACK_8     * m_disconnAck8;
+        const INFO_8      * m_info;
 };
 //---------------------------------------------------------------------------
 };
 //---------------------------------------------------------------------------
-#ifdef WIN32
-unsigned long WINAPI RunW(void *);
-#else
-void * RunW(void *);
-#endif
-
-//---------------------------------------------------------------------------
-#endif //IA_AUTH_C_H