// Copyright (c) 2011 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "base/basictypes.h"
#include "chrome/browser/net/quoted_printable.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace {
class QuotedPrintableTest : public testing::Test {
};
const char* const kNormalText[] = {
// Basic sentence with an =.
"If you believe that truth=beauty, then surely mathematics is the most "
"beautiful branch of philosophy.",
// All ASCII chars.
"\x1\x2\x3\x4\x5\x6\x7\x8\x9\xA\xB\xC\xD\xE\xF"
"\x10\x11\x12\x13\x14\x15\x16\x17\x18\x19\x1A\x1B\x1C\x1D\x1E\x1F"
"\x20\x21\x22\x23\x24\x25\x26\x27\x28\x29\x2A\x2B\x2C\x2D\x2E\x2F"
"\x30\x31\x32\x33\x34\x35\x36\x37\x38\x39\x3A\x3B\x3C\x3D\x3E\x3F"
"\x40\x41\x42\x43\x44\x45\x46\x47\x48\x49\x4A\x4B\x4C\x4D\x4E\x4F"
"\x50\x51\x52\x53\x54\x55\x56\x57\x58\x59\x5A\x5B\x5C\x5D\x5E\x5F"
"\x60\x61\x62\x63\x64\x65\x66\x67\x68\x69\x6A\x6B\x6C\x6D\x6E\x6F"
"\x70\x71\x72\x73\x74\x75\x76\x77\x78\x79\x7A\x7B\x7C\x7D\x7E\x7F"
"\x80\x81\x82\x83\x84\x85\x86\x87\x88\x89\x8A\x8B\x8C\x8D\x8E\x8F"
"\x90\x91\x92\x93\x94\x95\x96\x97\x98\x99\x9A\x9B\x9C\x9D\x9E\x9F"
"\xA0\xA1\xA2\xA3\xA4\xA5\xA6\xA7\xA8\xA9\xAA\xAB\xAC\xAD\xAE\xAF"
"\xB0\xB1\xB2\xB3\xB4\xB5\xB6\xB7\xB8\xB9\xBA\xBB\xBC\xBD\xBE\xBF"
"\xC0\xC1\xC2\xC3\xC4\xC5\xC6\xC7\xC8\xC9\xCA\xCB\xCC\xCD\xCE\xCF"
"\xD0\xD1\xD2\xD3\xD4\xD5\xD6\xD7\xD8\xD9\xDA\xDB\xDC\xDD\xDE\xDF"
"\xE0\xE1\xE2\xE3\xE4\xE5\xE6\xE7\xE8\xE9\xEA\xEB\xEC\xED\xEE\xEF"
"\xF0\xF1\xF2\xF3\xF4\xF5\xF6\xF7\xF8\xF9\xFA\xFB\xFC\xFD\xFE\xFF",
// Space right before max char per line.
"This line has a space at the 75 characters mark = ********************** "
"The end.",
// Space on max char per line index.
"This line has a space at the 76 characters mark = *********************** "
"The end.",
// Space at end of line.
"This line ends with a space \r\nThe end.",
// Space at end of input.
"This input ends with a space. ",
// Tab right before max char per line.
"This line has a tab at the 75 characters mark = ************************\t"
"The end.",
// Tab on max char per line index.
"This line has a tab at the 76 characters mark = *************************\t"
"The end.",
// Tab at end of line.
"This line ends with a tab\t\r\nThe end.",
// Tab at end of input.
"This input ends with a tab.\t",
// Various EOLs in input.
"This is a test of having EOLs in the input\r\n"
"Any EOL should be converted \r to \n a CRLF \r\n."
};
const char* const kEncodedText[] = {
"If you believe that truth=3Dbeauty, then surely mathematics is the most "
"bea=\r\nutiful branch of philosophy.",
"=01=02=03=04=05=06=07=08=09\r\n"
"=0B=0C\r\n"
"=0E=0F=10=11=12=13=14=15=16=17=18=19=1A=1B=1C=1D=1E=1F !\"#$%&'()*+,-./01234"
"=\r\n"
"56789:;<=3D>?@ABCDEFGHIJKLMNOPQRSTUVWXYZ[\\]^_`abcdefghijklmnopqrstuvwxyz{|}"
"=\r\n"
"~=7F=80=81=82=83=84=85=86=87=88=89=8A=8B=8C=8D=8E=8F=90=91=92=93=94=95=96="
"\r\n"
"=97=98=99=9A=9B=9C=9D=9E=9F=A0=A1=A2=A3=A4=A5=A6=A7=A8=A9=AA=AB=AC=AD=AE=AF="
"\r\n"
"=B0=B1=B2=B3=B4=B5=B6=B7=B8=B9=BA=BB=BC=BD=BE=BF=C0=C1=C2=C3=C4=C5=C6=C7=C8="
"\r\n"
"=C9=CA=CB=CC=CD=CE=CF=D0=D1=D2=D3=D4=D5=D6=D7=D8=D9=DA=DB=DC=DD=DE=DF=E0=E1="
"\r\n"
"=E2=E3=E4=E5=E6=E7=E8=E9=EA=EB=EC=ED=EE=EF=F0=F1=F2=F3=F4=F5=F6=F7=F8=F9=FA="
"\r\n"
"=FB=FC=FD=FE=FF",
"This line has a space at the 75 characters mark =3D ********************** ="
"\r\nThe end.",
"This line has a space at the 76 characters mark =3D ***********************="
"\r\n The end.",
"This line ends with a space=20\r\nThe end.",
"This input ends with a space.=20",
"This line has a tab at the 75 characters mark =3D ************************\t"
"=\r\nThe end.",
"This line has a tab at the 76 characters mark =3D *************************="
"\r\n\tThe end.",
"This line ends with a tab=09\r\nThe end.",
"This input ends with a tab.=09",
"This is a test of having EOLs in the input\r\n"
"Any EOL should be converted=20\r\n to=20\r\n a CRLF=20\r\n."
};
const char* const kBadEncodedText[] = {
// Invalid finish with =.
"A =3D at the end of the input is bad=",
// Invalid = sequence.
"This line contains a valid =3D sequence and invalid ones =$$ = =\t =1 = 2 "
"==",
};
const char* const kBadEncodedTextDecoded[] = {
"A = at the end of the input is bad=",
"This line contains a valid = sequence and invalid ones =$$ = =\t =1 = 2 ==",
};
// Compares the 2 strings and returns true if they are identical, but for EOLs
// that don't have to be the same (ex: \r\n can match \n).
bool CompareEOLInsensitive(const std::string& s1, const std::string& s2) {
std::string::const_iterator s1_iter = s1.begin();
std::string::const_iterator s2_iter = s2.begin();
while (true) {
if (s1_iter == s1.end() && s2_iter == s2.end())
return true;
if ((s1_iter == s1.end() && s2_iter != s2.end()) ||
(s1_iter != s1.end() && s2_iter == s2.end())) {
return false;
}
int s1_eol = chrome::browser::net::IsEOL(s1_iter, s1);
int s2_eol = chrome::browser::net::IsEOL(s2_iter, s2);
if ((!s1_eol && s2_eol) || (s1_eol && !s2_eol)) {
// Unmatched EOL.
return false;
}
if (s1_eol > 0) {
s1_iter += s1_eol;
s2_iter += s2_eol;
} else {
// Non-EOL char.
if (*s1_iter != *s2_iter)
return false;
s1_iter++;
s2_iter++;
}
}
return true;
}
} // namespace
TEST(QuotedPrintableTest, Encode) {
ASSERT_EQ(arraysize(kNormalText), arraysize(kEncodedText));
for (size_t i = 0; i < arraysize(kNormalText); ++i) {
SCOPED_TRACE(::testing::Message() << "Iteration " << i);
std::string output;
chrome::browser::net::QuotedPrintableEncode(kNormalText[i], &output);
std::string expected(kEncodedText[i]);
EXPECT_EQ(expected, output);
}
}
TEST(QuotedPrintableTest, Decode) {
ASSERT_EQ(arraysize(kNormalText), arraysize(kEncodedText));
for (size_t i = 0; i < arraysize(kNormalText); ++i) {
std::string output;
EXPECT_TRUE(chrome::browser::net::QuotedPrintableDecode(
kEncodedText[i], &output));
std::string expected(kNormalText[i]);
SCOPED_TRACE(::testing::Message() << "Iteration " << i <<
"\n Actual=\n" << output << "\n Expected=\n" <<
expected);
// We cannot test for equality as EOLs won't match the normal text
// (as any EOL is converted to a CRLF during encoding).
EXPECT_TRUE(CompareEOLInsensitive(expected, output));
}
}
// Tests that we return false but still do our best to decode badly encoded
// inputs.
TEST(QuotedPrintableTest, DecodeBadInput) {
ASSERT_EQ(arraysize(kBadEncodedText), arraysize(kBadEncodedTextDecoded));
for (size_t i = 0; i < arraysize(kBadEncodedText); ++i) {
SCOPED_TRACE(::testing::Message() << "Iteration " << i);
std::string output;
EXPECT_FALSE(chrome::browser::net::QuotedPrintableDecode(
kBadEncodedText[i], &output));
std::string expected(kBadEncodedTextDecoded[i]);
EXPECT_EQ(expected, output);
}
}