Mercurial > thymian

annotate 3rdparty/vmime/tests/utility/filteredStreamTest.cpp @ 0:a4671277546c tip

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
created the repository for the thymian project
author ferencd
date Tue, 17 Aug 2021 11:19:54 +0200
parents
children
rev   line source
ferencd@0 1 //
ferencd@0 2 // VMime library (http://www.vmime.org)
ferencd@0 3 // Copyright (C) 2002-2013 Vincent Richard <vincent@vmime.org>
ferencd@0 4 //
ferencd@0 5 // This program is free software; you can redistribute it and/or
ferencd@0 6 // modify it under the terms of the GNU General Public License as
ferencd@0 7 // published by the Free Software Foundation; either version 3 of
ferencd@0 8 // the License, or (at your option) any later version.
ferencd@0 9 //
ferencd@0 10 // This program is distributed in the hope that it will be useful,
ferencd@0 11 // but WITHOUT ANY WARRANTY; without even the implied warranty of
ferencd@0 12 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
ferencd@0 13 // General Public License for more details.
ferencd@0 14 //
ferencd@0 15 // You should have received a copy of the GNU General Public License along
ferencd@0 16 // with this program; if not, write to the Free Software Foundation, Inc.,
ferencd@0 17 // 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
ferencd@0 18 //
ferencd@0 19 // Linking this library statically or dynamically with other modules is making
ferencd@0 20 // a combined work based on this library. Thus, the terms and conditions of
ferencd@0 21 // the GNU General Public License cover the whole combination.
ferencd@0 22 //
ferencd@0 23
ferencd@0 24 #include "tests/testUtils.hpp"
ferencd@0 25
ferencd@0 26 #include "vmime/utility/filteredStream.hpp"
ferencd@0 27 #include "vmime/utility/stringUtils.hpp"
ferencd@0 28
ferencd@0 29
ferencd@0 30 VMIME_TEST_SUITE_BEGIN(filteredStreamTest)
ferencd@0 31
ferencd@0 32 VMIME_TEST_LIST_BEGIN
ferencd@0 33 VMIME_TEST(testDotFilteredInputStream)
ferencd@0 34 VMIME_TEST(testDotFilteredOutputStream)
ferencd@0 35 VMIME_TEST(testCRLFToLFFilteredOutputStream)
ferencd@0 36 VMIME_TEST(testStopSequenceFilteredInputStream1)
ferencd@0 37 VMIME_TEST(testStopSequenceFilteredInputStreamN_2)
ferencd@0 38 VMIME_TEST(testStopSequenceFilteredInputStreamN_3)
ferencd@0 39 VMIME_TEST(testLFToCRLFFilteredOutputStream_Global)
ferencd@0 40 VMIME_TEST(testLFToCRLFFilteredOutputStream_Edge)
ferencd@0 41 VMIME_TEST_LIST_END
ferencd@0 42
ferencd@0 43
ferencd@0 44 class chunkInputStream : public vmime::utility::inputStream
ferencd@0 45 {
ferencd@0 46 private:
ferencd@0 47
ferencd@0 48 std::vector <std::string> m_chunks;
ferencd@0 49 size_t m_index;
ferencd@0 50
ferencd@0 51 public:
ferencd@0 52
ferencd@0 53 chunkInputStream() : m_index(0) { }
ferencd@0 54
ferencd@0 55 void addChunk(const std::string& chunk) { m_chunks.push_back(chunk); }
ferencd@0 56
ferencd@0 57 bool eof() const { return (m_index >= m_chunks.size()); }
ferencd@0 58 void reset() { m_index = 0; }
ferencd@0 59
ferencd@0 60 vmime::size_t read(vmime::byte_t* const data, const vmime::size_t /* count */)
ferencd@0 61 {
ferencd@0 62 if (eof())
ferencd@0 63 return 0;
ferencd@0 64
ferencd@0 65 const std::string chunk = m_chunks[m_index];
ferencd@0 66
ferencd@0 67 // Warning: 'count' should be larger than chunk length.
ferencd@0 68 // This is OK for our tests.
ferencd@0 69 std::copy(chunk.begin(), chunk.end(), data);
ferencd@0 70
ferencd@0 71 ++m_index;
ferencd@0 72
ferencd@0 73 return chunk.length();
ferencd@0 74 }
ferencd@0 75
ferencd@0 76 vmime::size_t skip(const vmime::size_t /* count */)
ferencd@0 77 {
ferencd@0 78 // Not supported
ferencd@0 79 return 0;
ferencd@0 80 }
ferencd@0 81 };
ferencd@0 82
ferencd@0 83
ferencd@0 84 const std::string readWhole(vmime::utility::inputStream& is)
ferencd@0 85 {
ferencd@0 86 vmime::byte_t buffer[256];
ferencd@0 87 std::string whole;
ferencd@0 88
ferencd@0 89 while (!is.eof())
ferencd@0 90 {
ferencd@0 91 const vmime::size_t read = is.read(buffer, sizeof(buffer));
ferencd@0 92
ferencd@0 93 whole += vmime::utility::stringUtils::makeStringFromBytes(buffer, read);
ferencd@0 94 }
ferencd@0 95
ferencd@0 96 return (whole);
ferencd@0 97 }
ferencd@0 98
ferencd@0 99
ferencd@0 100 // dotFilteredInputStream
ferencd@0 101
ferencd@0 102 void testDotFilteredInputStreamHelper
ferencd@0 103 (const std::string& number, const std::string& expected,
ferencd@0 104 const std::string& c1, const std::string& c2 = "",
ferencd@0 105 const std::string& c3 = "", const std::string& c4 = "")
ferencd@0 106 {
ferencd@0 107 chunkInputStream cis;
ferencd@0 108 cis.addChunk(c1);
ferencd@0 109 if (!c2.empty()) cis.addChunk(c2);
ferencd@0 110 if (!c3.empty()) cis.addChunk(c3);
ferencd@0 111 if (!c4.empty()) cis.addChunk(c4);
ferencd@0 112
ferencd@0 113 vmime::utility::dotFilteredInputStream is(cis);
ferencd@0 114
ferencd@0 115 std::ostringstream oss;
ferencd@0 116 vmime::utility::outputStreamAdapter os(oss);
ferencd@0 117
ferencd@0 118 vmime::utility::bufferedStreamCopy(is, os);
ferencd@0 119
ferencd@0 120 VASSERT_EQ(number, expected, oss.str());
ferencd@0 121 }
ferencd@0 122
ferencd@0 123 void testDotFilteredInputStream()
ferencd@0 124 {
ferencd@0 125 testDotFilteredInputStreamHelper("1", "foo\n.bar", "foo\n..bar");
ferencd@0 126 testDotFilteredInputStreamHelper("2", "foo\n.bar", "foo\n", "..bar");
ferencd@0 127 testDotFilteredInputStreamHelper("3", "foo\n.bar", "foo\n.", ".bar");
ferencd@0 128 testDotFilteredInputStreamHelper("4", "foo\n.bar", "foo\n..", "bar");
ferencd@0 129 testDotFilteredInputStreamHelper("5", "foo\n.bar", "foo\n", ".", ".bar");
ferencd@0 130 testDotFilteredInputStreamHelper("6", "foo\n.bar", "foo\n", ".", ".", "bar");
ferencd@0 131 }
ferencd@0 132
ferencd@0 133 // dotFilteredOutputStream
ferencd@0 134 // CRLFToLFFilteredOutputStream
ferencd@0 135
ferencd@0 136 template <typename FILTER>
ferencd@0 137 void testFilteredOutputStreamHelper
ferencd@0 138 (const std::string& number, const std::string& expected,
ferencd@0 139 const std::string& c1, const std::string& c2 = "",
ferencd@0 140 const std::string& c3 = "", const std::string& c4 = "")
ferencd@0 141 {
ferencd@0 142 std::ostringstream oss;
ferencd@0 143 vmime::utility::outputStreamAdapter os(oss);
ferencd@0 144
ferencd@0 145 FILTER fos(os);
ferencd@0 146
ferencd@0 147 fos.write(c1.data(), c1.length());
ferencd@0 148 if (!c2.empty()) fos.write(c2.data(), c2.length());
ferencd@0 149 if (!c3.empty()) fos.write(c3.data(), c3.length());
ferencd@0 150 if (!c4.empty()) fos.write(c4.data(), c4.length());
ferencd@0 151
ferencd@0 152 VASSERT_EQ(number, expected, oss.str());
ferencd@0 153 }
ferencd@0 154
ferencd@0 155 void testDotFilteredOutputStream()
ferencd@0 156 {
ferencd@0 157 typedef vmime::utility::dotFilteredOutputStream FILTER;
ferencd@0 158
ferencd@0 159 testFilteredOutputStreamHelper<FILTER>("1", "foo\n..bar", "foo\n.bar");
ferencd@0 160 testFilteredOutputStreamHelper<FILTER>("2", "foo\n..bar", "foo\n", ".bar");
ferencd@0 161 testFilteredOutputStreamHelper<FILTER>("3", "foo\n..bar", "foo", "\n.bar");
ferencd@0 162 testFilteredOutputStreamHelper<FILTER>("4", "foo\n..bar", "foo", "\n", ".bar");
ferencd@0 163 testFilteredOutputStreamHelper<FILTER>("5", "foo\n..bar", "foo", "\n", ".", "bar");
ferencd@0 164
ferencd@0 165 testFilteredOutputStreamHelper<FILTER>("6", "..\nfoobar", ".\nfoobar");
ferencd@0 166 testFilteredOutputStreamHelper<FILTER>("7", "..\r\nfoobar", ".\r\nfoobar");
ferencd@0 167 testFilteredOutputStreamHelper<FILTER>("8", "..\r\nfoobar", ".\r", "\nfoobar");
ferencd@0 168 testFilteredOutputStreamHelper<FILTER>("9", ".foobar", ".foobar");
ferencd@0 169 testFilteredOutputStreamHelper<FILTER>("10", ".foobar", ".", "foobar");
ferencd@0 170 }
ferencd@0 171
ferencd@0 172 void testCRLFToLFFilteredOutputStream()
ferencd@0 173 {
ferencd@0 174 typedef vmime::utility::CRLFToLFFilteredOutputStream FILTER;
ferencd@0 175
ferencd@0 176 testFilteredOutputStreamHelper<FILTER>("1", "foo\nbar", "foo\r\nbar");
ferencd@0 177 testFilteredOutputStreamHelper<FILTER>("2", "foo\nbar", "foo\r\n", "bar");
ferencd@0 178 testFilteredOutputStreamHelper<FILTER>("3", "foo\nbar", "foo\r", "\nbar");
ferencd@0 179 testFilteredOutputStreamHelper<FILTER>("4", "foo\nbar", "foo", "\r\nbar");
ferencd@0 180 testFilteredOutputStreamHelper<FILTER>("5", "foo\nbar", "foo", "\r", "\nbar");
ferencd@0 181 testFilteredOutputStreamHelper<FILTER>("6", "foo\nbar", "foo", "\r", "\n", "bar");
ferencd@0 182 testFilteredOutputStreamHelper<FILTER>("7", "foo\nba\nr", "foo\r", "\nba\r\nr");
ferencd@0 183 }
ferencd@0 184
ferencd@0 185 // stopSequenceFilteredInputStream
ferencd@0 186
ferencd@0 187 template <int N>
ferencd@0 188 void testStopSequenceFISHelper
ferencd@0 189 (const std::string& number, const std::string& sequence,
ferencd@0 190 const std::string& expected, const std::string& c1,
ferencd@0 191 const std::string& c2 = "", const std::string& c3 = "",
ferencd@0 192 const std::string& c4 = "", const std::string& c5 = "")
ferencd@0 193 {
ferencd@0 194 chunkInputStream cis;
ferencd@0 195 cis.addChunk(c1);
ferencd@0 196 if (!c2.empty()) cis.addChunk(c2);
ferencd@0 197 if (!c3.empty()) cis.addChunk(c3);
ferencd@0 198 if (!c4.empty()) cis.addChunk(c4);
ferencd@0 199 if (!c5.empty()) cis.addChunk(c5);
ferencd@0 200
ferencd@0 201 vmime::utility::stopSequenceFilteredInputStream <N> is(cis, sequence.data());
ferencd@0 202
ferencd@0 203 VASSERT_EQ(number, expected, readWhole(is));
ferencd@0 204 }
ferencd@0 205
ferencd@0 206 void testStopSequenceFilteredInputStream1()
ferencd@0 207 {
ferencd@0 208 testStopSequenceFISHelper <1>("1", "x", "foo", "fooxbar");
ferencd@0 209 testStopSequenceFISHelper <1>("2", "x", "foo", "foox", "bar");
ferencd@0 210 testStopSequenceFISHelper <1>("3", "x", "foo", "foo", "x", "bar");
ferencd@0 211 testStopSequenceFISHelper <1>("4", "x", "foo", "fo", "o", "x", "bar");
ferencd@0 212 testStopSequenceFISHelper <1>("5", "x", "foo", "fo", "o", "x", "b", "ar");
ferencd@0 213
ferencd@0 214 testStopSequenceFISHelper <1>("6", "x", "foobar", "fo", "o", "b", "ar");
ferencd@0 215 testStopSequenceFISHelper <1>("7", "x", "foobar", "foo", "bar");
ferencd@0 216 testStopSequenceFISHelper <1>("8", "x", "foobar", "foo", "b", "ar");
ferencd@0 217
ferencd@0 218 testStopSequenceFISHelper <1>("9", "x", "foobar", "foobar");
ferencd@0 219 testStopSequenceFISHelper <1>("10", "x", "foobar", "foobarx");
ferencd@0 220
ferencd@0 221 testStopSequenceFISHelper <1>("11", "x", "", "");
ferencd@0 222 testStopSequenceFISHelper <1>("12", "x", "", "x");
ferencd@0 223 testStopSequenceFISHelper <1>("13", "x", "", "", "x");
ferencd@0 224 }
ferencd@0 225
ferencd@0 226 void testStopSequenceFilteredInputStreamN_2()
ferencd@0 227 {
ferencd@0 228 testStopSequenceFISHelper <2>("1", "xy", "foo", "fooxybar");
ferencd@0 229 testStopSequenceFISHelper <2>("2", "xy", "foo", "foox", "ybar");
ferencd@0 230 testStopSequenceFISHelper <2>("3", "xy", "foo", "foox", "y", "bar");
ferencd@0 231 testStopSequenceFISHelper <2>("4", "xy", "foo", "foo", "x", "ybar");
ferencd@0 232 testStopSequenceFISHelper <2>("5", "xy", "foo", "foo", "xy", "bar");
ferencd@0 233 testStopSequenceFISHelper <2>("6", "xy", "foo", "foo", "x", "y", "bar");
ferencd@0 234
ferencd@0 235 testStopSequenceFISHelper <2>("7", "xy", "fooxbar", "foox", "bar");
ferencd@0 236 testStopSequenceFISHelper <2>("8", "xy", "fooxbar", "foo", "xbar");
ferencd@0 237 testStopSequenceFISHelper <2>("9", "xy", "fooxbar", "foo", "x", "bar");
ferencd@0 238 testStopSequenceFISHelper <2>("10", "xy", "foobarx", "foo", "barx");
ferencd@0 239
ferencd@0 240 testStopSequenceFISHelper <2>("11", "xy", "foobar", "foobarxy");
ferencd@0 241 testStopSequenceFISHelper <2>("12", "xy", "foobar", "foo", "barxy");
ferencd@0 242 testStopSequenceFISHelper <2>("13", "xy", "foobar", "foo", "bar", "xy");
ferencd@0 243
ferencd@0 244 testStopSequenceFISHelper <2>("14", "xy", "", "");
ferencd@0 245 testStopSequenceFISHelper <2>("15", "xy", "x", "x");
ferencd@0 246 testStopSequenceFISHelper <2>("16", "xy", "", "xy");
ferencd@0 247 testStopSequenceFISHelper <2>("17", "xy", "", "x", "y");
ferencd@0 248 }
ferencd@0 249
ferencd@0 250 void testStopSequenceFilteredInputStreamN_3()
ferencd@0 251 {
ferencd@0 252 testStopSequenceFISHelper <3>("1", "xyz", "foo", "fooxyzbar");
ferencd@0 253 testStopSequenceFISHelper <3>("2", "xyz", "foo", "foox", "yzbar");
ferencd@0 254 testStopSequenceFISHelper <3>("3", "xyz", "foo", "foox", "y", "zbar");
ferencd@0 255 testStopSequenceFISHelper <3>("4", "xyz", "foo", "foox", "yz", "bar");
ferencd@0 256 testStopSequenceFISHelper <3>("5", "xyz", "foo", "foo", "xyz", "bar");
ferencd@0 257 testStopSequenceFISHelper <3>("6", "xyz", "foo", "foo", "xy", "zbar");
ferencd@0 258 testStopSequenceFISHelper <3>("7", "xyz", "foo", "foo", "x", "y", "zbar");
ferencd@0 259 testStopSequenceFISHelper <3>("8", "xyz", "foo", "foo", "x", "y", "z", "bar");
ferencd@0 260 testStopSequenceFISHelper <3>("9", "xyz", "foo", "fooxy", "z", "bar");
ferencd@0 261
ferencd@0 262 testStopSequenceFISHelper <3>("10", "xyz", "fooxybar", "foox", "y", "bar");
ferencd@0 263 testStopSequenceFISHelper <3>("11", "xyz", "fooxybar", "fooxy", "bar");
ferencd@0 264 testStopSequenceFISHelper <3>("12", "xyz", "fooxybar", "fo", "ox", "y", "bar");
ferencd@0 265 testStopSequenceFISHelper <3>("13", "xyz", "fooxybar", "fo", "o", "x", "y", "bar");
ferencd@0 266 testStopSequenceFISHelper <3>("14", "xyz", "fooxybar", "foo", "x", "ybar");
ferencd@0 267 testStopSequenceFISHelper <3>("15", "xyz", "fooxybar", "foo", "xybar");
ferencd@0 268
ferencd@0 269 testStopSequenceFISHelper <3>("16", "xyz", "xfoxoxybxar", "xfoxo", "xybxar");
ferencd@0 270 testStopSequenceFISHelper <3>("17", "xyz", "xfoxoxybxarx", "xfoxo", "xybxarx");
ferencd@0 271 testStopSequenceFISHelper <3>("18", "xyz", "xfoxoxybxarxy", "xfoxo", "xybxarxy");
ferencd@0 272
ferencd@0 273 testStopSequenceFISHelper <3>("19", "xyz", "", "");
ferencd@0 274 testStopSequenceFISHelper <3>("20", "xyz", "x", "x");
ferencd@0 275 testStopSequenceFISHelper <3>("21", "xyz", "xy", "xy");
ferencd@0 276 testStopSequenceFISHelper <3>("22", "xyz", "", "xyz");
ferencd@0 277 testStopSequenceFISHelper <3>("23", "xyz", "", "x", "yz");
ferencd@0 278 testStopSequenceFISHelper <3>("24", "xyz", "", "x", "y", "z");
ferencd@0 279 }
ferencd@0 280
ferencd@0 281
ferencd@0 282 // LFToCRLFFilteredOutputStream
ferencd@0 283
ferencd@0 284 void testLFToCRLFFilteredOutputStream_Global()
ferencd@0 285 {
ferencd@0 286 typedef vmime::utility::LFToCRLFFilteredOutputStream FILTER;
ferencd@0 287
ferencd@0 288 testFilteredOutputStreamHelper<FILTER>("1", "ABC\r\nDEF", "ABC\nDEF");
ferencd@0 289 testFilteredOutputStreamHelper<FILTER>("2", "ABC\r\nDEF", "ABC\rDEF");
ferencd@0 290 testFilteredOutputStreamHelper<FILTER>("3", "\r\n\r\nAB\r\n\r\nA\r\nB\r\n", "\n\nAB\n\nA\nB\n");
ferencd@0 291 testFilteredOutputStreamHelper<FILTER>("4", "ABCDE\r\nF", "ABCDE\nF");
ferencd@0 292 testFilteredOutputStreamHelper<FILTER>("5", "ABCDE\r\nF", "ABCDE\r\nF");
ferencd@0 293 testFilteredOutputStreamHelper<FILTER>("6", "\r\n\r\n\r\n", "\n\n\n");
ferencd@0 294 testFilteredOutputStreamHelper<FILTER>("7", "\r\n\r\n\r\n", "\r\r\n\n");
ferencd@0 295 testFilteredOutputStreamHelper<FILTER>("8", "\r\n\r\n\r\n\r\n", "\r\r\r\r");
ferencd@0 296 testFilteredOutputStreamHelper<FILTER>("9", "\r\n\r\n\r\n\r\n", "\n\n\n\n");
ferencd@0 297 testFilteredOutputStreamHelper<FILTER>("10", "\r\n\r\n\r\n", "\r\n\n\n");
ferencd@0 298 testFilteredOutputStreamHelper<FILTER>("11", "\r\n\r\n\r\n\r\n", "\n\n\n\r\n");
ferencd@0 299 }
ferencd@0 300
ferencd@0 301 void testLFToCRLFFilteredOutputStream_Edge()
ferencd@0 302 {
ferencd@0 303 typedef vmime::utility::LFToCRLFFilteredOutputStream FILTER;
ferencd@0 304
ferencd@0 305 testFilteredOutputStreamHelper<FILTER>("1", "\r\n\r\n", "\r", "\r");
ferencd@0 306 testFilteredOutputStreamHelper<FILTER>("2", "\r\n\r\n", "\r", "\n\r");
ferencd@0 307 testFilteredOutputStreamHelper<FILTER>("3", "ABC\r\n\r\n", "ABC\r", "\n\r");
ferencd@0 308 testFilteredOutputStreamHelper<FILTER>("4", "ABC\r\n\r\n\r\n", "ABC\r", "\n\r", "\n\n");
ferencd@0 309 testFilteredOutputStreamHelper<FILTER>("5", "\r\n\r\n", "\n", "\n");
ferencd@0 310 testFilteredOutputStreamHelper<FILTER>("6", "\r\n\r\n", "\r\n\r\n");
ferencd@0 311 testFilteredOutputStreamHelper<FILTER>("7", "\r\n\r\n", "\r\n\r", "\n");
ferencd@0 312
ferencd@0 313 testFilteredOutputStreamHelper<FILTER>("8", "A\r\nB\r\nC\r\nD", "A\rB", "\nC\r\nD");
ferencd@0 314 testFilteredOutputStreamHelper<FILTER>("9", "\r\nA\r\nB\r\nC\r\nD", "\rA\r", "B\nC\r\nD");
ferencd@0 315 testFilteredOutputStreamHelper<FILTER>("10", "\r\nA\r\nB\r\nC\r\nD", "\nA\r", "B\nC\r\nD");
ferencd@0 316 testFilteredOutputStreamHelper<FILTER>("11", "\r\nA\r\nB\r\nC\r\nD\r\n", "\nA\rB", "\nC\r\nD\r");
ferencd@0 317 }
ferencd@0 318
ferencd@0 319 VMIME_TEST_SUITE_END
ferencd@0 320