ferencd@0: // ferencd@0: // VMime library (http://www.vmime.org) ferencd@0: // Copyright (C) 2002-2013 Vincent Richard 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: ferencd@0: VMIME_TEST_SUITE_BEGIN(bodyPartTest) ferencd@0: ferencd@0: VMIME_TEST_LIST_BEGIN ferencd@0: VMIME_TEST(testParse) ferencd@0: VMIME_TEST(testGenerate) ferencd@0: VMIME_TEST(testParseGuessBoundary) ferencd@0: VMIME_TEST(testParseGuessBoundaryWithTransportPadding) ferencd@0: VMIME_TEST(testParseMissingLastBoundary) ferencd@0: VMIME_TEST(testPrologEpilog) ferencd@0: VMIME_TEST(testPrologEncoding) ferencd@0: VMIME_TEST(testSuccessiveBoundaries) ferencd@0: VMIME_TEST(testTransportPaddingInBoundary) ferencd@0: VMIME_TEST(testGenerate7bit) ferencd@0: VMIME_TEST(testTextUsageForQPEncoding) ferencd@0: VMIME_TEST(testParseVeryBigMessage) ferencd@0: VMIME_TEST_LIST_END ferencd@0: ferencd@0: ferencd@0: static const vmime::string extractComponentString ferencd@0: (const vmime::string& buffer, const vmime::component& c) ferencd@0: { ferencd@0: return vmime::string(buffer.begin() + c.getParsedOffset(), ferencd@0: buffer.begin() + c.getParsedOffset() + c.getParsedLength()); ferencd@0: } ferencd@0: ferencd@0: static const vmime::string extractContents(const vmime::shared_ptr cts) ferencd@0: { ferencd@0: std::ostringstream oss; ferencd@0: vmime::utility::outputStreamAdapter os(oss); ferencd@0: ferencd@0: cts->extract(os); ferencd@0: ferencd@0: return oss.str(); ferencd@0: } ferencd@0: ferencd@0: ferencd@0: void testParse() ferencd@0: { ferencd@0: vmime::string str1 = "HEADER\r\n\r\nBODY"; ferencd@0: vmime::bodyPart p1; ferencd@0: p1.parse(str1); ferencd@0: ferencd@0: VASSERT_EQ("1", "HEADER\r\n\r\n", extractComponentString(str1, *p1.getHeader())); ferencd@0: VASSERT_EQ("2", "BODY", extractComponentString(str1, *p1.getBody())); ferencd@0: ferencd@0: vmime::string str2 = "HEADER\n\nBODY"; ferencd@0: vmime::bodyPart p2; ferencd@0: p2.parse(str2); ferencd@0: ferencd@0: VASSERT_EQ("3", "HEADER\n\n", extractComponentString(str2, *p2.getHeader())); ferencd@0: VASSERT_EQ("4", "BODY", extractComponentString(str2, *p2.getBody())); ferencd@0: ferencd@0: vmime::string str3 = "HEADER\r\n\nBODY"; ferencd@0: vmime::bodyPart p3; ferencd@0: p3.parse(str3); ferencd@0: ferencd@0: VASSERT_EQ("5", "HEADER\r\n\n", extractComponentString(str3, *p3.getHeader())); ferencd@0: VASSERT_EQ("6", "BODY", extractComponentString(str3, *p3.getBody())); ferencd@0: } ferencd@0: ferencd@0: void testParseMissingLastBoundary() ferencd@0: { ferencd@0: vmime::string str = ferencd@0: "Content-Type: multipart/mixed; boundary=\"MY-BOUNDARY\"" ferencd@0: "\r\n\r\n" ferencd@0: "--MY-BOUNDARY\r\nHEADER1\r\n\r\nBODY1\r\n" ferencd@0: "--MY-BOUNDARY\r\nHEADER2\r\n\r\nBODY2"; ferencd@0: ferencd@0: vmime::bodyPart p; ferencd@0: p.parse(str); ferencd@0: ferencd@0: VASSERT_EQ("count", 2, p.getBody()->getPartCount()); ferencd@0: ferencd@0: VASSERT_EQ("part1-body", "BODY1", extractContents(p.getBody()->getPartAt(0)->getBody()->getContents())); ferencd@0: VASSERT_EQ("part2-body", "BODY2", extractContents(p.getBody()->getPartAt(1)->getBody()->getContents())); ferencd@0: } ferencd@0: ferencd@0: void testGenerate() ferencd@0: { ferencd@0: vmime::bodyPart p1; ferencd@0: p1.getHeader()->getField("Foo")->setValue(vmime::string("bar")); ferencd@0: p1.getBody()->setContents(vmime::make_shared ("Baz")); ferencd@0: ferencd@0: VASSERT_EQ("1", "Foo: bar\r\n\r\nBaz", p1.generate()); ferencd@0: } ferencd@0: ferencd@0: void testPrologEpilog() ferencd@0: { ferencd@0: const char testMail[] = ferencd@0: "To: test@vmime.org\r\n" ferencd@0: "From: test@vmime.org\r\n" ferencd@0: "Subject: Prolog and epilog test\r\n" ferencd@0: "Content-Type: multipart/mixed; \r\n" ferencd@0: " boundary=\"=_boundary\"\r\n" ferencd@0: "\r\n" ferencd@0: "Prolog text\r\n" ferencd@0: "--=_boundary\r\n" ferencd@0: "Content-Type: text/plain\r\n" ferencd@0: "\r\n" ferencd@0: "Part1\r\n" ferencd@0: "--=_boundary--\r\n" ferencd@0: "Epilog text"; ferencd@0: ferencd@0: vmime::bodyPart part; ferencd@0: part.parse(testMail); ferencd@0: ferencd@0: VASSERT_EQ("prolog", "Prolog text", part.getBody()->getPrologText()); ferencd@0: VASSERT_EQ("epilog", "Epilog text", part.getBody()->getEpilogText()); ferencd@0: } ferencd@0: ferencd@0: // Test for bug fix: prolog should not be encoded ferencd@0: // http://sourceforge.net/tracker/?func=detail&atid=525568&aid=3174903&group_id=69724 ferencd@0: void testPrologEncoding() ferencd@0: { ferencd@0: const char testmail[] = ferencd@0: "To: test@vmime.org\r\n" ferencd@0: "From: test@vmime.org\r\n" ferencd@0: "Subject: Prolog encoding test\r\n" ferencd@0: "Content-Type: multipart/mixed; \r\n" ferencd@0: " boundary=\"=_+ZWjySayKqSf2CyrfnNpaAcO6-G1HpoXdHZ4YyswAWqEY39Q\"\r\n" ferencd@0: "\r\n" ferencd@0: "This is a multi-part message in MIME format. Your mail reader does not\r\n" ferencd@0: "understand MIME message format.\r\n" ferencd@0: "--=_+ZWjySayKqSf2CyrfnNpaAcO6-G1HpoXdHZ4YyswAWqEY39Q\r\n" ferencd@0: "Content-Type: text/html; charset=windows-1251\r\n" ferencd@0: "Content-Transfer-Encoding: quoted-printable\r\n" ferencd@0: "\r\n" ferencd@0: "=DD=F2=EE =F2=E5=EA=F1=F2=EE=E2=E0=FF =F7=E0=F1=F2=FC =F1=EB=EE=E6=ED=EE=E3=\r\n" ferencd@0: "=EE =F1=EE=EE=E1=F9=E5=ED=E8=FF\r\n" ferencd@0: "--=_+ZWjySayKqSf2CyrfnNpaAcO6-G1HpoXdHZ4YyswAWqEY39Q\r\n" ferencd@0: "Content-Type: application/octet-stream; charset=windows-1251\r\n" ferencd@0: "Content-Disposition: attachment; filename=FNS.zip\r\n" ferencd@0: "Content-Transfer-Encoding: base64\r\n" ferencd@0: "\r\n" ferencd@0: "UEsDBB...snap...EEAAAAAA==\r\n" ferencd@0: "--=_+ZWjySayKqSf2CyrfnNpaAcO6-G1HpoXdHZ4YyswAWqEY39Q--\r\n" ferencd@0: "Epilog text"; ferencd@0: ferencd@0: vmime::shared_ptr msg = vmime::make_shared(); ferencd@0: ferencd@0: std::string istr(testmail); ferencd@0: ferencd@0: std::string ostr; ferencd@0: vmime::utility::outputStreamStringAdapter out(ostr); ferencd@0: ferencd@0: for (int i = 0 ; i < 10 ; ++i) ferencd@0: { ferencd@0: ostr.clear(); ferencd@0: ferencd@0: msg->parse(istr); ferencd@0: msg->generate(out); ferencd@0: ferencd@0: istr = ostr; ferencd@0: } ferencd@0: ferencd@0: VASSERT_EQ("prolog", "This is a multi-part message in MIME format. Your mail reader" ferencd@0: " does not understand MIME message format.", msg->getBody()->getPrologText()); ferencd@0: VASSERT_EQ("epilog", "Epilog text", msg->getBody()->getEpilogText()); ferencd@0: } ferencd@0: ferencd@0: void testSuccessiveBoundaries() ferencd@0: { ferencd@0: vmime::string str = ferencd@0: "Content-Type: multipart/mixed; boundary=\"MY-BOUNDARY\"" ferencd@0: "\r\n\r\n" ferencd@0: "--MY-BOUNDARY\r\nHEADER1\r\n\r\nBODY1\r\n" ferencd@0: "--MY-BOUNDARY\r\n" ferencd@0: "--MY-BOUNDARY--\r\n"; ferencd@0: ferencd@0: vmime::bodyPart p; ferencd@0: p.parse(str); ferencd@0: ferencd@0: VASSERT_EQ("count", 2, p.getBody()->getPartCount()); ferencd@0: ferencd@0: VASSERT_EQ("part1-body", "BODY1", extractContents(p.getBody()->getPartAt(0)->getBody()->getContents())); ferencd@0: VASSERT_EQ("part2-body", "", extractContents(p.getBody()->getPartAt(1)->getBody()->getContents())); ferencd@0: } ferencd@0: ferencd@0: void testTransportPaddingInBoundary() ferencd@0: { ferencd@0: vmime::string str = ferencd@0: "Content-Type: multipart/mixed; boundary=\"MY-BOUNDARY\"" ferencd@0: "\r\n\r\n" ferencd@0: "-- \t MY-BOUNDARY\r\nHEADER1\r\n\r\nBODY1\r\n" ferencd@0: "--MY-BOUNDARY\r\n" ferencd@0: "-- MY-BOUNDARY--\r\n"; ferencd@0: ferencd@0: vmime::bodyPart p; ferencd@0: p.parse(str); ferencd@0: ferencd@0: VASSERT_EQ("count", 2, p.getBody()->getPartCount()); ferencd@0: ferencd@0: VASSERT_EQ("part1-body", "BODY1", extractContents(p.getBody()->getPartAt(0)->getBody()->getContents())); ferencd@0: VASSERT_EQ("part2-body", "", extractContents(p.getBody()->getPartAt(1)->getBody()->getContents())); ferencd@0: } ferencd@0: ferencd@0: /** Ensure '7bit' encoding is used when body is 7-bit only. */ ferencd@0: void testGenerate7bit() ferencd@0: { ferencd@0: vmime::shared_ptr p1 = vmime::make_shared (); ferencd@0: p1->setText(vmime::make_shared ("Part1 is US-ASCII only.")); ferencd@0: ferencd@0: vmime::shared_ptr msg = vmime::make_shared (); ferencd@0: p1->generateIn(msg, msg); ferencd@0: ferencd@0: vmime::shared_ptr header1 = msg->getBody()->getPartAt(0)->getHeader(); ferencd@0: VASSERT_EQ("1", "7bit", header1->ContentTransferEncoding()->getValue()->generate()); ferencd@0: } ferencd@0: ferencd@0: void testTextUsageForQPEncoding() ferencd@0: { ferencd@0: vmime::shared_ptr part = vmime::make_shared (); ferencd@0: part->setText(vmime::make_shared ("Part1-line1\r\nPart1-line2\r\n\x89")); ferencd@0: ferencd@0: vmime::shared_ptr msg = vmime::make_shared (); ferencd@0: part->generateIn(msg, msg); ferencd@0: ferencd@0: vmime::shared_ptr body = msg->getBody()->getPartAt(0)->getBody(); ferencd@0: vmime::shared_ptr header = msg->getBody()->getPartAt(0)->getHeader(); ferencd@0: ferencd@0: std::ostringstream oss; ferencd@0: vmime::utility::outputStreamAdapter os(oss); ferencd@0: body->generate(os, 80); ferencd@0: ferencd@0: VASSERT_EQ("1", "quoted-printable", header->ContentTransferEncoding()->getValue()->generate()); ferencd@0: ferencd@0: // This should *NOT* be: ferencd@0: // Part1-line1=0D=0APart1-line2=0D=0A=89 ferencd@0: VASSERT_EQ("2", "Part1-line1\r\nPart1-line2\r\n=89", oss.str()); ferencd@0: } ferencd@0: ferencd@0: void testParseGuessBoundary() ferencd@0: { ferencd@0: // Boundary is not specified in "Content-Type" field ferencd@0: // Parser will try to guess it from message contents. ferencd@0: ferencd@0: vmime::string str = ferencd@0: "Content-Type: multipart/mixed" ferencd@0: "\r\n\r\n" ferencd@0: "--UNKNOWN-BOUNDARY\r\nHEADER1\r\n\r\nBODY1\r\n" ferencd@0: "--UNKNOWN-BOUNDARY\r\nHEADER2\r\n\r\nBODY2\r\n" ferencd@0: "--UNKNOWN-BOUNDARY--"; ferencd@0: ferencd@0: vmime::bodyPart p; ferencd@0: p.parse(str); ferencd@0: ferencd@0: VASSERT_EQ("count", 2, p.getBody()->getPartCount()); ferencd@0: ferencd@0: VASSERT_EQ("part1-body", "BODY1", extractContents(p.getBody()->getPartAt(0)->getBody()->getContents())); ferencd@0: VASSERT_EQ("part2-body", "BODY2", extractContents(p.getBody()->getPartAt(1)->getBody()->getContents())); ferencd@0: } ferencd@0: ferencd@0: void testParseGuessBoundaryWithTransportPadding() ferencd@0: { ferencd@0: // Boundary is not specified in "Content-Type" field ferencd@0: // Parser will try to guess it from message contents. ferencd@0: // Transport padding white spaces should be ignored. ferencd@0: ferencd@0: vmime::string str = ferencd@0: "Content-Type: multipart/mixed" ferencd@0: "\r\n\r\n" ferencd@0: "-- \t UNKNOWN-BOUNDARY\r\nHEADER1\r\n\r\nBODY1\r\n" ferencd@0: "--UNKNOWN-BOUNDARY\r\nHEADER2\r\n\r\nBODY2\r\n" ferencd@0: "--UNKNOWN-BOUNDARY--"; ferencd@0: ferencd@0: vmime::bodyPart p; ferencd@0: p.parse(str); ferencd@0: ferencd@0: VASSERT_EQ("count", 2, p.getBody()->getPartCount()); ferencd@0: ferencd@0: VASSERT_EQ("part1-body", "BODY1", extractContents(p.getBody()->getPartAt(0)->getBody()->getContents())); ferencd@0: VASSERT_EQ("part2-body", "BODY2", extractContents(p.getBody()->getPartAt(1)->getBody()->getContents())); ferencd@0: } ferencd@0: ferencd@0: void testParseVeryBigMessage() ferencd@0: { ferencd@0: // When parsing from a seekable input stream, body contents should not ferencd@0: // be kept in memory in a "stringContentHandler" object. Instead, content ferencd@0: // should be accessible via a "streamContentHandler" object. ferencd@0: ferencd@0: static const std::string BODY1_BEGIN = "BEGIN1BEGIN1BEGIN1"; ferencd@0: static const std::string BODY1_LINE = "BODY1BODY1BODY1BODY1BODY1BODY1BODY1BODY1BODY1BODY1BODY1BODY1BODY1"; ferencd@0: static const std::string BODY1_END = "END1END1"; ferencd@0: static const unsigned int BODY1_REPEAT = 35000; ferencd@0: static const unsigned int BODY1_LENGTH = ferencd@0: BODY1_BEGIN.length() + BODY1_LINE.length() * BODY1_REPEAT + BODY1_END.length(); ferencd@0: ferencd@0: static const std::string BODY2_LINE = "BODY2BODY2BODY2BODY2BODY2BODY2BODY2BODY2BODY2BODY2BODY2BODY2BODY2"; ferencd@0: static const unsigned int BODY2_REPEAT = 20000; ferencd@0: ferencd@0: std::ostringstream oss; ferencd@0: oss << "Content-Type: multipart/mixed; boundary=\"MY-BOUNDARY\"" ferencd@0: << "\r\n\r\n" ferencd@0: << "--MY-BOUNDARY\r\n" ferencd@0: << "HEADER1\r\n" ferencd@0: << "\r\n"; ferencd@0: ferencd@0: oss << BODY1_BEGIN; ferencd@0: ferencd@0: for (unsigned int i = 0 ; i < BODY1_REPEAT ; ++i) ferencd@0: oss << BODY1_LINE; ferencd@0: ferencd@0: oss << BODY1_END; ferencd@0: ferencd@0: oss << "\r\n" ferencd@0: << "--MY-BOUNDARY\r\n" ferencd@0: << "HEADER2\r\n" ferencd@0: << "\r\n"; ferencd@0: ferencd@0: for (unsigned int i = 0 ; i < BODY2_REPEAT ; ++i) ferencd@0: oss << BODY2_LINE; ferencd@0: ferencd@0: oss << "\r\n" ferencd@0: << "--MY-BOUNDARY--\r\n"; ferencd@0: ferencd@0: vmime::shared_ptr is = ferencd@0: vmime::make_shared (oss.str()); ferencd@0: ferencd@0: vmime::shared_ptr msg = vmime::make_shared (); ferencd@0: msg->parse(is, oss.str().length()); ferencd@0: ferencd@0: vmime::shared_ptr body1 = msg->getBody()->getPartAt(0)->getBody(); ferencd@0: vmime::shared_ptr body1Cts = body1->getContents(); ferencd@0: ferencd@0: vmime::shared_ptr body2 = msg->getBody()->getPartAt(1)->getBody(); ferencd@0: vmime::shared_ptr body2Cts = body2->getContents(); ferencd@0: ferencd@0: vmime::string body1CtsExtracted; ferencd@0: vmime::utility::outputStreamStringAdapter body1CtsExtractStream(body1CtsExtracted); ferencd@0: body1Cts->extract(body1CtsExtractStream); ferencd@0: ferencd@0: VASSERT_EQ("1.1", BODY1_LENGTH, body1Cts->getLength()); ferencd@0: VASSERT("1.2", vmime::dynamicCast (body1Cts) != NULL); ferencd@0: VASSERT_EQ("1.3", BODY1_LENGTH, body1CtsExtracted.length()); ferencd@0: VASSERT_EQ("1.4", BODY1_BEGIN, body1CtsExtracted.substr(0, BODY1_BEGIN.length())); ferencd@0: VASSERT_EQ("1.5", BODY1_END, body1CtsExtracted.substr(BODY1_LENGTH - BODY1_END.length(), BODY1_END.length())); ferencd@0: ferencd@0: VASSERT_EQ("2.1", BODY2_LINE.length() * BODY2_REPEAT, body2Cts->getLength()); ferencd@0: VASSERT("2.2", vmime::dynamicCast (body2Cts) != NULL); ferencd@0: } ferencd@0: ferencd@0: VMIME_TEST_SUITE_END ferencd@0: