diff options
Diffstat (limited to 'vmime-master/tests/testUtils.cpp')
-rw-r--r-- | vmime-master/tests/testUtils.cpp | 404 |
1 files changed, 404 insertions, 0 deletions
diff --git a/vmime-master/tests/testUtils.cpp b/vmime-master/tests/testUtils.cpp new file mode 100644 index 0000000..ae75547 --- /dev/null +++ b/vmime-master/tests/testUtils.cpp @@ -0,0 +1,404 @@ +// +// VMime library (http://www.vmime.org) +// Copyright (C) 2002 Vincent Richard <vincent@vmime.org> +// +// 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 3 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 02110-1301 USA. +// +// Linking this library statically or dynamically with other modules is making +// a combined work based on this library. Thus, the terms and conditions of +// the GNU General Public License cover the whole combination. +// + +#include "testUtils.hpp" + +#include "vmime/utility/stringUtils.hpp" + +#include <cstring> +#include <iostream> + + +// Enable to output socket send/receive on standard output +#define DEBUG_SOCKET_IN_OUT 0 + + + +// testSocket + +void testSocket::connect(const vmime::string& address, const vmime::port_t port) { + + m_address = address; + m_port = port; + m_connected = true; + + onConnected(); +} + + +void testSocket::disconnect() { + + m_address.clear(); + m_port = 0; + m_connected = false; +} + + +bool testSocket::isConnected() const { + + return m_connected; +} + + +vmime::size_t testSocket::getBlockSize() const { + + return 16384; +} + + +unsigned int testSocket::getStatus() const { + + return 0; +} + + +const vmime::string testSocket::getPeerName() const { + + return "test.vmime.org"; +} + + +const vmime::string testSocket::getPeerAddress() const { + + return "127.0.0.1"; +} + + +vmime::shared_ptr <vmime::net::timeoutHandler> testSocket::getTimeoutHandler() { + + return vmime::null; +} + + +void testSocket::setTracer(const vmime::shared_ptr <vmime::net::tracer>& /* tracer */) { + +} + + +vmime::shared_ptr <vmime::net::tracer> testSocket::getTracer() { + + return vmime::null; +} + + +bool testSocket::waitForRead(const int /* msecs */) { + + return true; +} + + +bool testSocket::waitForWrite(const int /* msecs */) { + + return true; +} + + +void testSocket::receive(vmime::string& buffer) { + + buffer = m_inBuffer; + m_inBuffer.clear(); +} + + +void testSocket::send(const vmime::string& buffer) { + + m_outBuffer += buffer; + + onDataReceived(); +} + + +void testSocket::send(const char* str) { + + sendRaw(reinterpret_cast <const vmime::byte_t*>(str), strlen(str)); +} + + +vmime::size_t testSocket::receiveRaw(vmime::byte_t* buffer, const size_t count) { + + const size_t n = std::min(count, static_cast <size_t>(m_inBuffer.size())); + + std::copy(m_inBuffer.begin(), m_inBuffer.begin() + n, buffer); + m_inBuffer.erase(m_inBuffer.begin(), m_inBuffer.begin() + n); + + return n; +} + + +void testSocket::sendRaw(const vmime::byte_t* buffer, const size_t count) { + + send(vmime::utility::stringUtils::makeStringFromBytes(buffer, count)); +} + + +vmime::size_t testSocket::sendRawNonBlocking(const vmime::byte_t* buffer, const size_t count) { + + sendRaw(buffer, count); + return count; +} + + +void testSocket::localSend(const vmime::string& buffer) { + + m_inBuffer += buffer; + +#if DEBUG_SOCKET_IN_OUT + std::cout << "> " << vmime::utility::stringUtils::trim(buffer) << std::endl; +#endif // DEBUG_SOCKET_IN_OUT + +} + + +void testSocket::localReceive(vmime::string& buffer) { + + buffer = m_outBuffer; + m_outBuffer.clear(); +} + + +bool testSocket::localReceiveLine(vmime::string& line) { + + vmime::size_t eol; + + if ((eol = m_outBuffer.find('\n')) != vmime::string::npos) { + + line = vmime::string(m_outBuffer.begin(), m_outBuffer.begin() + eol); + + if (!line.empty() && line[line.length() - 1] == '\r') { + line.erase(line.end() - 1, line.end()); + } + + m_outBuffer.erase(m_outBuffer.begin(), m_outBuffer.begin() + eol + 1); + + return true; + } + + return false; +} + + +vmime::size_t testSocket::localReceiveRaw(vmime::byte_t* buffer, const size_t count) { + + const size_t received = std::min(count, static_cast <size_t>(m_outBuffer.size())); + + if (received != 0) { + + if (buffer) { + std::copy(m_outBuffer.begin(), m_outBuffer.begin() + received, buffer); + } + + m_outBuffer.erase(m_outBuffer.begin(), m_outBuffer.begin() + received); + } + + return received; +} + + +void testSocket::onDataReceived() { + + // Override +} + + +void testSocket::onConnected() { + + // Override +} + + +// lineBasedTestSocket + +void lineBasedTestSocket::onDataReceived() { + + vmime::string chunk; + localReceive(chunk); + + m_buffer += chunk; + + vmime::size_t eol; + + while ((eol = m_buffer.find('\n')) != vmime::string::npos) { + + vmime::string line(std::string(m_buffer.begin(), m_buffer.begin() + eol)); + + if (!line.empty() && line[line.length() - 1] == '\r') { + line.erase(line.end() - 1, line.end()); + } + +#if DEBUG_SOCKET_IN_OUT + std::cout << "< " << vmime::utility::stringUtils::trim(line) << std::endl; +#endif // DEBUG_SOCKET_IN_OUT + + m_lines.push_back(line); + m_buffer.erase(m_buffer.begin(), m_buffer.begin() + eol + 1); + } + + while (!m_lines.empty()) { + processCommand(); + } +} + + +const vmime::string lineBasedTestSocket::getNextLine() { + + const vmime::string line = m_lines.front(); + m_lines.erase(m_lines.begin(), m_lines.begin() + 1); + return line; +} + + +bool lineBasedTestSocket::haveMoreLines() const { + + return !m_lines.empty(); +} + + +// testTimeoutHandler + +testTimeoutHandler::testTimeoutHandler(const unsigned long delay) + : m_delay(delay), + m_start(0) { + +} + + +bool testTimeoutHandler::isTimeOut() { + + return (vmime::platform::getHandler()->getUnixTime() - m_start) >= m_delay; +} + + +void testTimeoutHandler::resetTimeOut() { + + m_start = vmime::platform::getHandler()->getUnixTime(); +} + + +bool testTimeoutHandler::handleTimeOut() { + + return false; +} + + +// testTimeoutHandlerFactory : public vmime::net::timeoutHandlerFactory + +vmime::shared_ptr <vmime::net::timeoutHandler> testTimeoutHandlerFactory::create() { + + return vmime::make_shared <testTimeoutHandler>(); +} + + + +// Exception helper +std::ostream& operator<<(std::ostream& os, const vmime::exception& e) { + + os << "* vmime::exceptions::" << e.name() << std::endl; + os << " what = " << e.what() << std::endl; + + // More information for special exceptions + if (dynamic_cast <const vmime::exceptions::command_error*>(&e)) { + + const vmime::exceptions::command_error& cee = + dynamic_cast <const vmime::exceptions::command_error&>(e); + + os << " command = " << cee.command() << std::endl; + os << " response = " << cee.response() << std::endl; + } + + if (dynamic_cast <const vmime::exceptions::invalid_response*>(&e)) { + + const vmime::exceptions::invalid_response& ir = + dynamic_cast <const vmime::exceptions::invalid_response&>(e); + + os << " response = " << ir.response() << std::endl; + } + + if (dynamic_cast <const vmime::exceptions::connection_greeting_error*>(&e)) { + + const vmime::exceptions::connection_greeting_error& cgee = + dynamic_cast <const vmime::exceptions::connection_greeting_error&>(e); + + os << " response = " << cgee.response() << std::endl; + } + + if (dynamic_cast <const vmime::exceptions::authentication_error*>(&e)) { + + const vmime::exceptions::authentication_error& aee = + dynamic_cast <const vmime::exceptions::authentication_error&>(e); + + os << " response = " << aee.response() << std::endl; + } + + if (dynamic_cast <const vmime::exceptions::filesystem_exception*>(&e)) { + + const vmime::exceptions::filesystem_exception& fse = + dynamic_cast <const vmime::exceptions::filesystem_exception&>(e); + + os << " path = " << vmime::platform::getHandler()-> + getFileSystemFactory()->pathToString(fse.path()) << std::endl; + } + + if (e.other()) { + os << *e.other(); + } + + return os; +} + + +const vmime::string toHex(const vmime::string str) { + + static const char hexChars[] = "0123456789abcdef"; + + vmime::string res = "\n"; + + for (size_t i = 0 ; i < str.length() ; i += 16) { + + size_t r = std::min(static_cast <size_t>(16), str.length() - i); + + vmime::string hex; + vmime::string chr; + + for (size_t j = 0 ; j < r ; ++j) { + + const unsigned char c = str[i + j]; + + hex += hexChars[c / 16]; + hex += hexChars[c % 16]; + hex += " "; + + if (c >= 32 && c <= 127) + chr += c; + else + chr += '.'; + } + + for (size_t j = r ; j < 16 ; ++j) { + hex += " "; + } + + res += hex + " " + chr + "\n"; + } + + return res; +} |