X-Git-Url: https://git.stg.codes/stg.git/blobdiff_plain/8c6fa3fbaccc22127280bf77a48fab5a3ee0716e..46b0747592074017ff0ea4b33d4a7194235886e5:/projects/stargazer/plugins/configuration/sgconfig/conn.cpp diff --git a/projects/stargazer/plugins/configuration/sgconfig/conn.cpp b/projects/stargazer/plugins/configuration/sgconfig/conn.cpp deleted file mode 100644 index 71de7356..00000000 --- a/projects/stargazer/plugins/configuration/sgconfig/conn.cpp +++ /dev/null @@ -1,307 +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 - */ - -#include "conn.h" - -#include "stg/admins.h" -#include "stg/admin.h" -#include "stg/logger.h" -#include "stg/blowfish.h" -#include "stg/bfstream.h" -#include "stg/common.h" - -#include -#include -#include - -#include -#include - -using STG::Conn; - -const char Conn::STG_HEADER[] = "SG04"; -const char Conn::OK_HEADER[] = "OKHD"; -const char Conn::ERR_HEADER[] = "ERHD"; -const char Conn::OK_LOGIN[] = "OKLG"; -const char Conn::ERR_LOGIN[] = "ERLG"; -const char Conn::OK_LOGINS[] = "OKLS"; -const char Conn::ERR_LOGINS[] = "ERLS"; - -Conn::Conn(const BASE_PARSER::REGISTRY & registry, - ADMINS & admins, int sock, const sockaddr_in& addr, - PLUGIN_LOGGER & logger) - : m_registry(registry), - m_admins(admins), - m_admin(NULL), - m_sock(sock), - m_addr(addr), - m_keepAlive(false), - m_parser(NULL), - m_xmlParser(XML_ParserCreate(NULL)), - m_state(HEADER), - m_buffer(m_header), - m_bufferSize(sizeof(m_header)), - m_stream(NULL), - m_logger(logger), -#ifdef DUMPCRYPTO - m_dataState(false, *this), - m_dumper(endpoint()) -#else - m_dataState(false, *this) -#endif -{ - if (m_xmlParser == NULL) - throw Error("Failed to create XML parser."); - - XML_ParserReset(m_xmlParser, NULL); - XML_SetElementHandler(m_xmlParser, ParseXMLStart, ParseXMLEnd); - XML_SetUserData(m_xmlParser, this); -} - -Conn::~Conn() -{ - shutdown(m_sock, SHUT_RDWR); - close(m_sock); - - XML_ParserFree(m_xmlParser); - - delete m_stream; - delete m_parser; -} - -bool Conn::Read() -{ - ssize_t res = read(m_sock, m_buffer, m_bufferSize); - if (res < 0) - { - m_state = ERROR; - Log(__FILE__, "Failed to read data from " + endpoint() + ". Reason: '" + strerror(errno) + "'"); - return false; - } - if (res == 0 && m_state != DATA) // EOF is ok for data. - { - m_state = ERROR; - Log(__FILE__, "Failed to read data from " + endpoint() + ". Unexpected EOF."); - return false; - } -#ifdef DUMPCRYPTO - m_dumper.write(m_buffer, res); -#endif - m_bufferSize -= res; - m_buffer = static_cast(m_buffer) + res; - return HandleBuffer(res); -} - -bool Conn::WriteAnswer(const void* buffer, size_t size) -{ - ssize_t res = write(m_sock, buffer, size); - if (res < 0) - { - m_state = ERROR; - Log(__FILE__, "Failed to write data to " + endpoint() + ". Reason: '" + strerror(errno) + "'."); - return false; - } - return true; -} - -BASE_PARSER * Conn::GetParser(const std::string & tag) const -{ - BASE_PARSER::REGISTRY::const_iterator it = m_registry.find(ToLower(tag)); - if (it == m_registry.end()) - return NULL; - return it->second->create(*m_admin); -} - -bool Conn::HandleBuffer(size_t size) -{ - if (m_state == DATA) - return HandleData(size); - - if (m_bufferSize > 0) - return true; - - switch (m_state) - { - case HEADER: return HandleHeader(); - case LOGIN: return HandleLogin(); - case CRYPTO_LOGIN: return HandleCryptoLogin(); - default: return true; - } - - return true; -} - -bool Conn::HandleHeader() -{ - if (strncmp(m_header, STG_HEADER, sizeof(m_header)) != 0) - { - Log(__FILE__, "Received invalid header from " + endpoint() + "."); - WriteAnswer(ERR_HEADER, sizeof(ERR_HEADER) - 1); // Without \0 - m_state = ERROR; - return false; - } - m_state = LOGIN; - m_buffer = m_login; - m_bufferSize = sizeof(m_login); - return WriteAnswer(OK_HEADER, sizeof(OK_HEADER) - 1); // Without \0 -} - -bool Conn::HandleLogin() -{ - if (m_admins.Find(m_login, &m_admin)) // ADMINS::Find returns true on error. - { - std::string login(m_login, strnlen(m_login, sizeof(m_login))); - Log(__FILE__, "Received invalid login '" + ToPrintable(login) + "' from " + endpoint() + "."); - WriteAnswer(ERR_LOGIN, sizeof(ERR_LOGIN) - 1); // Without \0 - m_state = ERROR; - return false; - } - m_admin->SetIP(IP()); - m_state = CRYPTO_LOGIN; - m_buffer = m_cryptoLogin; - m_bufferSize = sizeof(m_cryptoLogin); - return WriteAnswer(OK_LOGIN, sizeof(OK_LOGIN) - 1); // Without \0 -} - -bool Conn::HandleCryptoLogin() -{ - char login[ADM_LOGIN_LEN + 1]; - BLOWFISH_CTX ctx; - InitContext(m_admin->GetPassword().c_str(), ADM_PASSWD_LEN, &ctx); - DecryptString(login, m_cryptoLogin, ADM_LOGIN_LEN, &ctx); - - if (strncmp(m_login, login, sizeof(login)) != 0) - { - Log(__FILE__, "Attempt to connect with wrong password from " + m_admin->GetLogin() + "@" + endpoint() + "."); - WriteAnswer(ERR_LOGINS, sizeof(ERR_LOGINS) - 1); // Without \0 - m_state = ERROR; - return false; - } - - m_state = DATA; - m_buffer = m_data; - m_bufferSize = sizeof(m_data); - m_stream = new STG::DECRYPT_STREAM(m_admin->GetPassword(), DataCallback, &m_dataState); - return WriteAnswer(OK_LOGINS, sizeof(OK_LOGINS) - 1); // Without \0 -} - -bool Conn::HandleData(size_t size) -{ - m_stream->Put(m_data, size, size == 0 || memchr(m_data, 0, size) != NULL); - m_buffer = m_data; - return m_stream->IsOk(); -} - -bool Conn::DataCallback(const void * block, size_t size, void * data) -{ - assert(data != NULL); - DataState& state = *static_cast(data); - - const char * xml = static_cast(block); - size_t length = strnlen(xml, size); - - state.final = state.final || length < size || size == 0; - - if (XML_Parse(state.conn.m_xmlParser, xml, length, state.final) == XML_STATUS_ERROR) - { - state.conn.Log(__FILE__, "Received invalid XML from " + state.conn.m_admin->GetLogin() + "@" + state.conn.endpoint() + "."); - printfd(__FILE__, "XML parse error at line %d, %d: %s. Is final: %d\n", - static_cast(XML_GetCurrentLineNumber(state.conn.m_xmlParser)), - static_cast(XML_GetCurrentColumnNumber(state.conn.m_xmlParser)), - XML_ErrorString(XML_GetErrorCode(state.conn.m_xmlParser)), (int)state.final); - printfd(__FILE__, "Data block: '%s' of size %d\n", xml, length); - state.conn.m_state = ERROR; - return false; - } - - if (state.final) - { - if (!state.conn.WriteResponse()) - { - state.conn.Log(__FILE__, "Failed to write response to " + state.conn.m_admin->GetLogin() + "@" + state.conn.endpoint() + "."); - state.conn.m_state = ERROR; - return false; - } - state.conn.m_state = DONE; - } - - return true; -} - -void Conn::ParseXMLStart(void * data, const char * el, const char ** attr) -{ - assert(data != NULL); - Conn & conn = *static_cast(data); - - if (conn.m_parser == NULL) - conn.m_parser = conn.GetParser(el); - - if (conn.m_parser == NULL) - { - conn.Log(__FILE__, "Received unknown command '" + std::string(el) + "' from " + conn.m_admin->GetLogin() + "@" + conn.endpoint() + "."); - conn.m_state = ERROR; - return; - } - - conn.m_parser->Start(data, el, attr); -} - -void Conn::ParseXMLEnd(void * data, const char * el) -{ - assert(data != NULL); - Conn & conn = *static_cast(data); - - if (conn.m_parser == NULL) - { - // No need to log it. - conn.m_state = ERROR; - return; - } - - conn.m_parser->End(data, el); -} - -bool Conn::WriteResponse() -{ - STG::ENCRYPT_STREAM stream(m_admin->GetPassword(), WriteCallback, this); - std::string answer; - if (m_parser != NULL) - answer = m_parser->GetAnswer(); - else - answer = ""; - delete m_parser; - m_parser = NULL; - printfd(__FILE__, "Writing %d bytes of answer.\n", answer.length()); - stream.Put(answer.c_str(), answer.length() + 1 /* including \0 */, true /* final */); - return stream.IsOk(); -} - -bool Conn::WriteCallback(const void * block, size_t size, void * data) -{ - assert(data != NULL); - Conn & conn = *static_cast(data); - return WriteAll(conn.m_sock, block, size);; -} - -void Conn::Log(const char * file, const std::string & message) -{ - printfd(file, "%s\n", message.c_str()); - m_logger(message); -}