X-Git-Url: https://git.stg.codes/stg.git/blobdiff_plain/8c6fa3fbaccc22127280bf77a48fab5a3ee0716e..46b0747592074017ff0ea4b33d4a7194235886e5:/projects/stargazer/plugins/configuration/sgconfig/conn.h diff --git a/projects/stargazer/plugins/configuration/sgconfig/conn.h b/projects/stargazer/plugins/configuration/sgconfig/conn.h deleted file mode 100644 index ed2e64d1..00000000 --- a/projects/stargazer/plugins/configuration/sgconfig/conn.h +++ /dev/null @@ -1,146 +0,0 @@ -/* - * This program is free software; you can redistribute it and/or modify - * it under the terms of the GNU General Public License as published by - * the Free Software Foundation; either version 2 of the License, or - * (at your option) any later version. - * - * This program is distributed in the hope that it will be useful, - * but WITHOUT ANY WARRANTY; without even the implied warranty of - * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the - * GNU General Public License for more details. - * - * You should have received a copy of the GNU General Public License - * along with this program; if not, write to the Free Software - * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA - */ - -/* - * Author : Maxim Mamontov - */ - -#ifndef __STG_SGCONFIG_CONN_H__ -#define __STG_SGCONFIG_CONN_H__ - -#include "parser.h" - -#include "dumphelpers.h" - -#include "stg/os_int.h" -#include "stg/const.h" - -#include -#include - -#include - -#include - -class SETTINGS; -class ADMINS; -class USERS; -class TARIFFS; -class ADMIN; -class BASE_PARSER; -class PLUGIN_LOGGER; - -namespace STG -{ - -class DECRYPT_STREAM; - -class Conn -{ - public: - struct Error : public std::runtime_error - { - explicit Error(const std::string& message) : runtime_error(message.c_str()) {} - }; - - Conn(const BASE_PARSER::REGISTRY & registry, - ADMINS & admins, int sock, const sockaddr_in& addr, - PLUGIN_LOGGER & logger); - ~Conn(); - - int Sock() const { return m_sock; } - uint32_t IP() const { return *(uint32_t *)(&m_addr.sin_addr); } - uint16_t Port() const { return ntohs(m_addr.sin_port); } - - std::string endpoint() const { return inet_ntostring(IP()) + ":" + x2str(Port()); } - - bool Read(); - - bool IsOk() const { return m_state != ERROR; } - bool IsDone() const { return m_state == DONE; } - bool IsKeepAlive() const { return m_keepAlive; } - - void SetKeepAlive() { m_keepAlive = true; } - - private: - - static const char STG_HEADER[5]; - static const char OK_HEADER[5]; - static const char ERR_HEADER[5]; - static const char OK_LOGIN[5]; - static const char ERR_LOGIN[5]; - static const char OK_LOGINS[5]; - static const char ERR_LOGINS[5]; - - const BASE_PARSER::REGISTRY & m_registry; - - ADMINS & m_admins; - - ADMIN * m_admin; - - int m_sock; - sockaddr_in m_addr; - bool m_keepAlive; - - BASE_PARSER * m_parser; - - XML_Parser m_xmlParser; - - enum { HEADER, LOGIN, CRYPTO_LOGIN, DATA, DONE, ERROR } m_state; - - void * m_buffer; - size_t m_bufferSize; - char m_header[sizeof(STG_HEADER) - 1]; // Without \0 - char m_login[ADM_LOGIN_LEN]; // Without \0 - char m_cryptoLogin[ADM_LOGIN_LEN]; // Without \0 - char m_data[1024]; - STG::DECRYPT_STREAM * m_stream; - PLUGIN_LOGGER & m_logger; - - BASE_PARSER * GetParser(const std::string & tag) const; - - bool HandleBuffer(size_t size); - - bool HandleHeader(); - bool HandleLogin(); - bool HandleCryptoLogin(); - bool HandleData(size_t size); - - bool WriteAnswer(const void* buffer, size_t size); - bool WriteResponse(); - - void Log(const char * file, const std::string & message); - - struct DataState - { - DataState(bool f, Conn & c) : final(f), conn(c) {} - bool final; - Conn & conn; - } m_dataState; - -#ifdef DUMPCRYPTO - Dumper m_dumper; -#endif - - static bool DataCallback(const void * block, size_t size, void * data); - static void ParseXMLStart(void * data, const char * el, const char ** attr); - static void ParseXMLEnd(void * data, const char * el); - static bool WriteCallback(const void * block, size_t size, void * data); -}; - -} - -#endif