diff --git a/src/adhocmanager.cpp b/src/adhocmanager.cpp new file mode 100644 index 0000000000000000000000000000000000000000..664b421ed327479000f5a06b9e66f07c1b5596f0 --- /dev/null +++ b/src/adhocmanager.cpp @@ -0,0 +1,163 @@ +/** + * XMPP - libpurple transport + * + * Copyright (C) 2012, Jan Kaluza + * + * 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., 51 Franklin Street, Fifth Floor, Boston, MA 02111-1301 USA + */ + +#include "transport/adhocmanager.h" +#include "transport/adhoccommandfactory.h" +#include "transport/discoitemsresponder.h" +#include "transport/conversation.h" +#include "transport/usermanager.h" +#include "transport/buddy.h" +#include "transport/factory.h" +#include "transport/user.h" +#include "transport/logging.h" + +namespace Transport { + +DEFINE_LOGGER(logger, "AdHocManager"); + +AdHocManager::AdHocManager(Component *component, DiscoItemsResponder *discoItemsResponder) : Swift::Responder(component->getIQRouter()){ + m_component = component; + m_discoItemsResponder = discoItemsResponder; + + m_collectTimer = m_component->getNetworkFactories()->getTimerFactory()->createTimer(20); + m_collectTimer->onTick.connect(boost::bind(&AdHocManager::removeOldSessions, this)); + m_collectTimer->start(); +} + +AdHocManager::~AdHocManager() { + m_collectTimer->stop(); + stop(); +} + +void AdHocManager::start() { + Swift::Responder::start(); +} + +void AdHocManager::stop() { + Swift::Responder::stop(); + + for (SessionsMap::iterator it = m_sessions.begin(); it != m_sessions.end(); it++) { + std::vector candidates; + for (CommandsMap::iterator ct = it->second.begin(); ct != it->second.end(); ct++) { + delete ct->second; + } + } + + m_sessions.clear(); +} + +void AdHocManager::addAdHocCommand(AdHocCommandFactory *factory) { + if (m_factories.find(factory->getNode()) != m_factories.end()) { + LOG4CXX_ERROR(logger, "Command with node " << factory->getNode() << " is already registered. Ignoring this attempt."); + return; + } + + m_factories[factory->getNode()] = factory; + m_discoItemsResponder->addAdHocCommand(factory->getNode(), factory->getName()); +} + +void AdHocManager::removeOldSessions() { + unsigned long removedCommands = 0; + time_t now = time(NULL); + + std::vector toRemove; + for (SessionsMap::iterator it = m_sessions.begin(); it != m_sessions.end(); it++) { + std::vector candidates; + for (CommandsMap::iterator ct = it->second.begin(); ct != it->second.end(); ct++) { + if (now - ct->second->getLastActivity() > 15*60) { + candidates.push_back(it->first); + delete ct->second; + removedCommands++; + } + } + + BOOST_FOREACH(std::string &key, candidates) { + it->second.erase(key); + } + + if (it->second.empty()) { + toRemove.push_back(it->first); + } + } + + BOOST_FOREACH(std::string &key, toRemove) { + m_sessions.erase(key); + } + + if (removedCommands > 0) { + LOG4CXX_INFO(logger, "Removed " << removedCommands << " old commands sessions."); + } +} + +bool AdHocManager::handleGetRequest(const Swift::JID& from, const Swift::JID& to, const std::string& id, boost::shared_ptr payload) { + return false; +} + +bool AdHocManager::handleSetRequest(const Swift::JID& from, const Swift::JID& to, const std::string& id, boost::shared_ptr payload) { + AdHocCommand *command = NULL; + // Try to find AdHocCommand according to 'from' and session_id + if (m_sessions.find(from) != m_sessions.end() && m_sessions[from].find(payload->getSessionID()) != m_sessions[from].end()) { + command = m_sessions[from][payload->getSessionID()]; + } + // Check if we can create command with this node + else if (m_factories.find(payload->getNode()) != m_factories.end()) { + command = m_factories[payload->getNode()]->createAdHocCommand(m_component, from, to); + m_sessions[from][command->getId()] = command; + LOG4CXX_INFO(logger, from.toString() << ": Started new AdHoc command session with node " << payload->getNode()); + } + else { + LOG4CXX_INFO(logger, from.toString() << ": Unknown node " << payload->getNode() << ". Can't start AdHoc command session."); + sendError(from, id, Swift::ErrorPayload::BadRequest, Swift::ErrorPayload::Modify); + return true; + } + + if (!command) { + LOG4CXX_ERROR(logger, from.toString() << ": createAdHocCommand for node " << payload->getNode() << " returned NULL pointer"); + sendError(from, id, Swift::ErrorPayload::BadRequest, Swift::ErrorPayload::Modify); + return true; + } + + boost::shared_ptr response = command->handleRequest(payload); + if (!response) { + LOG4CXX_ERROR(logger, from.toString() << ": handleRequest for node " << payload->getNode() << " returned NULL pointer"); + sendError(from, id, Swift::ErrorPayload::BadRequest, Swift::ErrorPayload::Modify); + return true; + } + + response->setSessionID(command->getId()); + + sendResponse(from, id, response); + + command->refreshLastActivity(); + + // Command completed, so we can remove it now + if (response->getStatus() == Swift::Command::Completed) { + m_sessions[from].erase(command->getId()); + if (m_sessions[from].empty()) { + m_sessions.erase(from); + } + delete command; + } + + + return true; +} + +}