ferencd@0: //
ferencd@0: // VMime library (http://www.vmime.org)
ferencd@0: // Copyright (C) 2002-2013 Vincent Richard <vincent@vmime.org>
ferencd@0: //
ferencd@0: // This program is free software; you can redistribute it and/or
ferencd@0: // modify it under the terms of the GNU General Public License as
ferencd@0: // published by the Free Software Foundation; either version 3 of
ferencd@0: // the License, or (at your option) any later version.
ferencd@0: //
ferencd@0: // This program is distributed in the hope that it will be useful,
ferencd@0: // but WITHOUT ANY WARRANTY; without even the implied warranty of
ferencd@0: // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
ferencd@0: // General Public License for more details.
ferencd@0: //
ferencd@0: // You should have received a copy of the GNU General Public License along
ferencd@0: // with this program; if not, write to the Free Software Foundation, Inc.,
ferencd@0: // 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
ferencd@0: //
ferencd@0: // Linking this library statically or dynamically with other modules is making
ferencd@0: // a combined work based on this library.  Thus, the terms and conditions of
ferencd@0: // the GNU General Public License cover the whole combination.
ferencd@0: //
ferencd@0: 
ferencd@0: #include "tests/testUtils.hpp"
ferencd@0: 
ferencd@0: #include "vmime/utility/filteredStream.hpp"
ferencd@0: #include "vmime/utility/stringUtils.hpp"
ferencd@0: 
ferencd@0: 
ferencd@0: VMIME_TEST_SUITE_BEGIN(filteredStreamTest)
ferencd@0: 
ferencd@0: 	VMIME_TEST_LIST_BEGIN
ferencd@0: 		VMIME_TEST(testDotFilteredInputStream)
ferencd@0: 		VMIME_TEST(testDotFilteredOutputStream)
ferencd@0: 		VMIME_TEST(testCRLFToLFFilteredOutputStream)
ferencd@0: 		VMIME_TEST(testStopSequenceFilteredInputStream1)
ferencd@0: 		VMIME_TEST(testStopSequenceFilteredInputStreamN_2)
ferencd@0: 		VMIME_TEST(testStopSequenceFilteredInputStreamN_3)
ferencd@0: 		VMIME_TEST(testLFToCRLFFilteredOutputStream_Global)
ferencd@0: 		VMIME_TEST(testLFToCRLFFilteredOutputStream_Edge)
ferencd@0: 	VMIME_TEST_LIST_END
ferencd@0: 
ferencd@0: 
ferencd@0: 	class chunkInputStream : public vmime::utility::inputStream
ferencd@0: 	{
ferencd@0: 	private:
ferencd@0: 
ferencd@0: 		std::vector <std::string> m_chunks;
ferencd@0: 		size_t m_index;
ferencd@0: 
ferencd@0: 	public:
ferencd@0: 
ferencd@0: 		chunkInputStream() : m_index(0) { }
ferencd@0: 
ferencd@0: 		void addChunk(const std::string& chunk) { m_chunks.push_back(chunk); }
ferencd@0: 
ferencd@0: 		bool eof() const { return (m_index >= m_chunks.size()); }
ferencd@0: 		void reset() { m_index = 0; }
ferencd@0: 
ferencd@0: 		vmime::size_t read(vmime::byte_t* const data, const vmime::size_t /* count */)
ferencd@0: 		{
ferencd@0: 			if (eof())
ferencd@0: 				return 0;
ferencd@0: 
ferencd@0: 			const std::string chunk = m_chunks[m_index];
ferencd@0: 
ferencd@0: 			// Warning: 'count' should be larger than chunk length.
ferencd@0: 			// This is OK for our tests.
ferencd@0: 			std::copy(chunk.begin(), chunk.end(), data);
ferencd@0: 
ferencd@0: 			++m_index;
ferencd@0: 
ferencd@0: 			return chunk.length();
ferencd@0: 		}
ferencd@0: 
ferencd@0: 		vmime::size_t skip(const vmime::size_t /* count */)
ferencd@0: 		{
ferencd@0: 			// Not supported
ferencd@0: 			return 0;
ferencd@0: 		}
ferencd@0: 	};
ferencd@0: 
ferencd@0: 
ferencd@0: 	const std::string readWhole(vmime::utility::inputStream& is)
ferencd@0: 	{
ferencd@0: 		vmime::byte_t buffer[256];
ferencd@0: 		std::string whole;
ferencd@0: 
ferencd@0: 		while (!is.eof())
ferencd@0: 		{
ferencd@0: 			const vmime::size_t read = is.read(buffer, sizeof(buffer));
ferencd@0: 
ferencd@0: 			whole += vmime::utility::stringUtils::makeStringFromBytes(buffer, read);
ferencd@0: 		}
ferencd@0: 
ferencd@0: 		return (whole);
ferencd@0: 	}
ferencd@0: 
ferencd@0: 
ferencd@0: 	// dotFilteredInputStream
ferencd@0: 
ferencd@0: 	void testDotFilteredInputStreamHelper
ferencd@0: 		(const std::string& number, const std::string& expected,
ferencd@0: 		 const std::string& c1, const std::string& c2 = "",
ferencd@0: 		 const std::string& c3 = "", const std::string& c4 = "")
ferencd@0: 	{
ferencd@0: 		chunkInputStream cis;
ferencd@0: 		cis.addChunk(c1);
ferencd@0: 		if (!c2.empty()) cis.addChunk(c2);
ferencd@0: 		if (!c3.empty()) cis.addChunk(c3);
ferencd@0: 		if (!c4.empty()) cis.addChunk(c4);
ferencd@0: 
ferencd@0: 		vmime::utility::dotFilteredInputStream is(cis);
ferencd@0: 
ferencd@0: 		std::ostringstream oss;
ferencd@0: 		vmime::utility::outputStreamAdapter os(oss);
ferencd@0: 
ferencd@0: 		vmime::utility::bufferedStreamCopy(is, os);
ferencd@0: 
ferencd@0: 		VASSERT_EQ(number, expected, oss.str());
ferencd@0: 	}
ferencd@0: 
ferencd@0: 	void testDotFilteredInputStream()
ferencd@0: 	{
ferencd@0: 		testDotFilteredInputStreamHelper("1", "foo\n.bar", "foo\n..bar");
ferencd@0: 		testDotFilteredInputStreamHelper("2", "foo\n.bar", "foo\n", "..bar");
ferencd@0: 		testDotFilteredInputStreamHelper("3", "foo\n.bar", "foo\n.", ".bar");
ferencd@0: 		testDotFilteredInputStreamHelper("4", "foo\n.bar", "foo\n..", "bar");
ferencd@0: 		testDotFilteredInputStreamHelper("5", "foo\n.bar", "foo\n", ".", ".bar");
ferencd@0: 		testDotFilteredInputStreamHelper("6", "foo\n.bar", "foo\n", ".", ".", "bar");
ferencd@0: 	}
ferencd@0: 
ferencd@0: 	// dotFilteredOutputStream
ferencd@0: 	// CRLFToLFFilteredOutputStream
ferencd@0: 
ferencd@0: 	template <typename FILTER>
ferencd@0: 	void testFilteredOutputStreamHelper
ferencd@0: 		(const std::string& number, const std::string& expected,
ferencd@0: 		 const std::string& c1, const std::string& c2 = "",
ferencd@0: 		 const std::string& c3 = "", const std::string& c4 = "")
ferencd@0: 	{
ferencd@0: 		std::ostringstream oss;
ferencd@0: 		vmime::utility::outputStreamAdapter os(oss);
ferencd@0: 
ferencd@0: 		FILTER fos(os);
ferencd@0: 
ferencd@0: 		fos.write(c1.data(), c1.length());
ferencd@0: 		if (!c2.empty()) fos.write(c2.data(), c2.length());
ferencd@0: 		if (!c3.empty()) fos.write(c3.data(), c3.length());
ferencd@0: 		if (!c4.empty()) fos.write(c4.data(), c4.length());
ferencd@0: 
ferencd@0: 		VASSERT_EQ(number, expected, oss.str());
ferencd@0: 	}
ferencd@0: 
ferencd@0: 	void testDotFilteredOutputStream()
ferencd@0: 	{
ferencd@0: 		typedef vmime::utility::dotFilteredOutputStream FILTER;
ferencd@0: 
ferencd@0: 		testFilteredOutputStreamHelper<FILTER>("1", "foo\n..bar", "foo\n.bar");
ferencd@0: 		testFilteredOutputStreamHelper<FILTER>("2", "foo\n..bar", "foo\n", ".bar");
ferencd@0: 		testFilteredOutputStreamHelper<FILTER>("3", "foo\n..bar", "foo", "\n.bar");
ferencd@0: 		testFilteredOutputStreamHelper<FILTER>("4", "foo\n..bar", "foo", "\n", ".bar");
ferencd@0: 		testFilteredOutputStreamHelper<FILTER>("5", "foo\n..bar", "foo", "\n", ".", "bar");
ferencd@0: 
ferencd@0: 		testFilteredOutputStreamHelper<FILTER>("6", "..\nfoobar", ".\nfoobar");
ferencd@0: 		testFilteredOutputStreamHelper<FILTER>("7", "..\r\nfoobar", ".\r\nfoobar");
ferencd@0: 		testFilteredOutputStreamHelper<FILTER>("8", "..\r\nfoobar", ".\r", "\nfoobar");
ferencd@0: 		testFilteredOutputStreamHelper<FILTER>("9", ".foobar", ".foobar");
ferencd@0: 		testFilteredOutputStreamHelper<FILTER>("10", ".foobar", ".", "foobar");
ferencd@0: 	}
ferencd@0: 
ferencd@0: 	void testCRLFToLFFilteredOutputStream()
ferencd@0: 	{
ferencd@0: 		typedef vmime::utility::CRLFToLFFilteredOutputStream FILTER;
ferencd@0: 
ferencd@0: 		testFilteredOutputStreamHelper<FILTER>("1", "foo\nbar", "foo\r\nbar");
ferencd@0: 		testFilteredOutputStreamHelper<FILTER>("2", "foo\nbar", "foo\r\n", "bar");
ferencd@0: 		testFilteredOutputStreamHelper<FILTER>("3", "foo\nbar", "foo\r", "\nbar");
ferencd@0: 		testFilteredOutputStreamHelper<FILTER>("4", "foo\nbar", "foo", "\r\nbar");
ferencd@0: 		testFilteredOutputStreamHelper<FILTER>("5", "foo\nbar", "foo", "\r", "\nbar");
ferencd@0: 		testFilteredOutputStreamHelper<FILTER>("6", "foo\nbar", "foo", "\r", "\n", "bar");
ferencd@0: 		testFilteredOutputStreamHelper<FILTER>("7", "foo\nba\nr", "foo\r", "\nba\r\nr");
ferencd@0: 	}
ferencd@0: 
ferencd@0: 	// stopSequenceFilteredInputStream
ferencd@0: 
ferencd@0: 	template <int N>
ferencd@0: 	void testStopSequenceFISHelper
ferencd@0: 		(const std::string& number, const std::string& sequence,
ferencd@0: 		 const std::string& expected, const std::string& c1,
ferencd@0: 		 const std::string& c2 = "", const std::string& c3 = "",
ferencd@0: 		 const std::string& c4 = "", const std::string& c5 = "")
ferencd@0: 	{
ferencd@0: 		chunkInputStream cis;
ferencd@0: 		cis.addChunk(c1);
ferencd@0: 		if (!c2.empty()) cis.addChunk(c2);
ferencd@0: 		if (!c3.empty()) cis.addChunk(c3);
ferencd@0: 		if (!c4.empty()) cis.addChunk(c4);
ferencd@0: 		if (!c5.empty()) cis.addChunk(c5);
ferencd@0: 
ferencd@0: 		vmime::utility::stopSequenceFilteredInputStream <N> is(cis, sequence.data());
ferencd@0: 
ferencd@0: 		VASSERT_EQ(number, expected, readWhole(is));
ferencd@0: 	}
ferencd@0: 
ferencd@0: 	void testStopSequenceFilteredInputStream1()
ferencd@0: 	{
ferencd@0: 		testStopSequenceFISHelper <1>("1", "x", "foo", "fooxbar");
ferencd@0: 		testStopSequenceFISHelper <1>("2", "x", "foo", "foox", "bar");
ferencd@0: 		testStopSequenceFISHelper <1>("3", "x", "foo", "foo", "x", "bar");
ferencd@0: 		testStopSequenceFISHelper <1>("4", "x", "foo", "fo", "o", "x", "bar");
ferencd@0: 		testStopSequenceFISHelper <1>("5", "x", "foo", "fo", "o", "x", "b", "ar");
ferencd@0: 
ferencd@0: 		testStopSequenceFISHelper <1>("6", "x", "foobar", "fo", "o", "b", "ar");
ferencd@0: 		testStopSequenceFISHelper <1>("7", "x", "foobar", "foo", "bar");
ferencd@0: 		testStopSequenceFISHelper <1>("8", "x", "foobar", "foo", "b", "ar");
ferencd@0: 
ferencd@0: 		testStopSequenceFISHelper <1>("9", "x", "foobar", "foobar");
ferencd@0: 		testStopSequenceFISHelper <1>("10", "x", "foobar", "foobarx");
ferencd@0: 
ferencd@0: 		testStopSequenceFISHelper <1>("11", "x", "", "");
ferencd@0: 		testStopSequenceFISHelper <1>("12", "x", "", "x");
ferencd@0: 		testStopSequenceFISHelper <1>("13", "x", "", "", "x");
ferencd@0: 	}
ferencd@0: 
ferencd@0: 	void testStopSequenceFilteredInputStreamN_2()
ferencd@0: 	{
ferencd@0: 		testStopSequenceFISHelper <2>("1", "xy", "foo", "fooxybar");
ferencd@0: 		testStopSequenceFISHelper <2>("2", "xy", "foo", "foox", "ybar");
ferencd@0: 		testStopSequenceFISHelper <2>("3", "xy", "foo", "foox", "y", "bar");
ferencd@0: 		testStopSequenceFISHelper <2>("4", "xy", "foo", "foo", "x", "ybar");
ferencd@0: 		testStopSequenceFISHelper <2>("5", "xy", "foo", "foo", "xy", "bar");
ferencd@0: 		testStopSequenceFISHelper <2>("6", "xy", "foo", "foo", "x", "y", "bar");
ferencd@0: 
ferencd@0: 		testStopSequenceFISHelper <2>("7", "xy", "fooxbar", "foox", "bar");
ferencd@0: 		testStopSequenceFISHelper <2>("8", "xy", "fooxbar", "foo", "xbar");
ferencd@0: 		testStopSequenceFISHelper <2>("9", "xy", "fooxbar", "foo", "x", "bar");
ferencd@0: 		testStopSequenceFISHelper <2>("10", "xy", "foobarx", "foo", "barx");
ferencd@0: 
ferencd@0: 		testStopSequenceFISHelper <2>("11", "xy", "foobar", "foobarxy");
ferencd@0: 		testStopSequenceFISHelper <2>("12", "xy", "foobar", "foo", "barxy");
ferencd@0: 		testStopSequenceFISHelper <2>("13", "xy", "foobar", "foo", "bar", "xy");
ferencd@0: 
ferencd@0: 		testStopSequenceFISHelper <2>("14", "xy", "", "");
ferencd@0: 		testStopSequenceFISHelper <2>("15", "xy", "x", "x");
ferencd@0: 		testStopSequenceFISHelper <2>("16", "xy", "", "xy");
ferencd@0: 		testStopSequenceFISHelper <2>("17", "xy", "", "x", "y");
ferencd@0: 	}
ferencd@0: 
ferencd@0: 	void testStopSequenceFilteredInputStreamN_3()
ferencd@0: 	{
ferencd@0: 		testStopSequenceFISHelper <3>("1", "xyz", "foo", "fooxyzbar");
ferencd@0: 		testStopSequenceFISHelper <3>("2", "xyz", "foo", "foox", "yzbar");
ferencd@0: 		testStopSequenceFISHelper <3>("3", "xyz", "foo", "foox", "y", "zbar");
ferencd@0: 		testStopSequenceFISHelper <3>("4", "xyz", "foo", "foox", "yz", "bar");
ferencd@0: 		testStopSequenceFISHelper <3>("5", "xyz", "foo", "foo", "xyz", "bar");
ferencd@0: 		testStopSequenceFISHelper <3>("6", "xyz", "foo", "foo", "xy", "zbar");
ferencd@0: 		testStopSequenceFISHelper <3>("7", "xyz", "foo", "foo", "x", "y", "zbar");
ferencd@0: 		testStopSequenceFISHelper <3>("8", "xyz", "foo", "foo", "x", "y", "z", "bar");
ferencd@0: 		testStopSequenceFISHelper <3>("9", "xyz", "foo", "fooxy", "z", "bar");
ferencd@0: 
ferencd@0: 		testStopSequenceFISHelper <3>("10", "xyz", "fooxybar", "foox", "y", "bar");
ferencd@0: 		testStopSequenceFISHelper <3>("11", "xyz", "fooxybar", "fooxy", "bar");
ferencd@0: 		testStopSequenceFISHelper <3>("12", "xyz", "fooxybar", "fo", "ox", "y", "bar");
ferencd@0: 		testStopSequenceFISHelper <3>("13", "xyz", "fooxybar", "fo", "o", "x", "y", "bar");
ferencd@0: 		testStopSequenceFISHelper <3>("14", "xyz", "fooxybar", "foo", "x", "ybar");
ferencd@0: 		testStopSequenceFISHelper <3>("15", "xyz", "fooxybar", "foo", "xybar");
ferencd@0: 
ferencd@0: 		testStopSequenceFISHelper <3>("16", "xyz", "xfoxoxybxar", "xfoxo", "xybxar");
ferencd@0: 		testStopSequenceFISHelper <3>("17", "xyz", "xfoxoxybxarx", "xfoxo", "xybxarx");
ferencd@0: 		testStopSequenceFISHelper <3>("18", "xyz", "xfoxoxybxarxy", "xfoxo", "xybxarxy");
ferencd@0: 
ferencd@0: 		testStopSequenceFISHelper <3>("19", "xyz", "", "");
ferencd@0: 		testStopSequenceFISHelper <3>("20", "xyz", "x", "x");
ferencd@0: 		testStopSequenceFISHelper <3>("21", "xyz", "xy", "xy");
ferencd@0: 		testStopSequenceFISHelper <3>("22", "xyz", "", "xyz");
ferencd@0: 		testStopSequenceFISHelper <3>("23", "xyz", "", "x", "yz");
ferencd@0: 		testStopSequenceFISHelper <3>("24", "xyz", "", "x", "y", "z");
ferencd@0: 	}
ferencd@0: 
ferencd@0: 
ferencd@0: 	// LFToCRLFFilteredOutputStream
ferencd@0: 
ferencd@0: 	void testLFToCRLFFilteredOutputStream_Global()
ferencd@0: 	{
ferencd@0: 		typedef vmime::utility::LFToCRLFFilteredOutputStream FILTER;
ferencd@0: 
ferencd@0: 		testFilteredOutputStreamHelper<FILTER>("1",  "ABC\r\nDEF",       "ABC\nDEF");
ferencd@0: 		testFilteredOutputStreamHelper<FILTER>("2",  "ABC\r\nDEF",       "ABC\rDEF");
ferencd@0: 		testFilteredOutputStreamHelper<FILTER>("3",  "\r\n\r\nAB\r\n\r\nA\r\nB\r\n", "\n\nAB\n\nA\nB\n");
ferencd@0: 		testFilteredOutputStreamHelper<FILTER>("4",  "ABCDE\r\nF",       "ABCDE\nF");
ferencd@0: 		testFilteredOutputStreamHelper<FILTER>("5",  "ABCDE\r\nF",       "ABCDE\r\nF");
ferencd@0: 		testFilteredOutputStreamHelper<FILTER>("6",  "\r\n\r\n\r\n",     "\n\n\n");
ferencd@0: 		testFilteredOutputStreamHelper<FILTER>("7",  "\r\n\r\n\r\n",     "\r\r\n\n");
ferencd@0: 		testFilteredOutputStreamHelper<FILTER>("8",  "\r\n\r\n\r\n\r\n", "\r\r\r\r");
ferencd@0: 		testFilteredOutputStreamHelper<FILTER>("9",  "\r\n\r\n\r\n\r\n", "\n\n\n\n");
ferencd@0: 		testFilteredOutputStreamHelper<FILTER>("10", "\r\n\r\n\r\n",     "\r\n\n\n");
ferencd@0: 		testFilteredOutputStreamHelper<FILTER>("11", "\r\n\r\n\r\n\r\n", "\n\n\n\r\n");
ferencd@0: 	}
ferencd@0: 
ferencd@0: 	void testLFToCRLFFilteredOutputStream_Edge()
ferencd@0: 	{
ferencd@0: 		typedef vmime::utility::LFToCRLFFilteredOutputStream FILTER;
ferencd@0: 
ferencd@0: 		testFilteredOutputStreamHelper<FILTER>("1",  "\r\n\r\n",         "\r", "\r");
ferencd@0: 		testFilteredOutputStreamHelper<FILTER>("2",  "\r\n\r\n",         "\r", "\n\r");
ferencd@0: 		testFilteredOutputStreamHelper<FILTER>("3",  "ABC\r\n\r\n",      "ABC\r", "\n\r");
ferencd@0: 		testFilteredOutputStreamHelper<FILTER>("4",  "ABC\r\n\r\n\r\n",  "ABC\r", "\n\r", "\n\n");
ferencd@0: 		testFilteredOutputStreamHelper<FILTER>("5",  "\r\n\r\n",         "\n", "\n");
ferencd@0: 		testFilteredOutputStreamHelper<FILTER>("6",  "\r\n\r\n",         "\r\n\r\n");
ferencd@0: 		testFilteredOutputStreamHelper<FILTER>("7",  "\r\n\r\n",         "\r\n\r", "\n");
ferencd@0: 
ferencd@0: 		testFilteredOutputStreamHelper<FILTER>("8",  "A\r\nB\r\nC\r\nD",         "A\rB", "\nC\r\nD");
ferencd@0: 		testFilteredOutputStreamHelper<FILTER>("9",  "\r\nA\r\nB\r\nC\r\nD",     "\rA\r", "B\nC\r\nD");
ferencd@0: 		testFilteredOutputStreamHelper<FILTER>("10", "\r\nA\r\nB\r\nC\r\nD",     "\nA\r", "B\nC\r\nD");
ferencd@0: 		testFilteredOutputStreamHelper<FILTER>("11", "\r\nA\r\nB\r\nC\r\nD\r\n", "\nA\rB", "\nC\r\nD\r");
ferencd@0: 	}
ferencd@0: 
ferencd@0: VMIME_TEST_SUITE_END
ferencd@0: