diff options
Diffstat (limited to 'src/lib-mail/test-qp-decoder.c')
| -rw-r--r-- | src/lib-mail/test-qp-decoder.c | 188 |
1 files changed, 188 insertions, 0 deletions
diff --git a/src/lib-mail/test-qp-decoder.c b/src/lib-mail/test-qp-decoder.c new file mode 100644 index 0000000..9b4d827 --- /dev/null +++ b/src/lib-mail/test-qp-decoder.c @@ -0,0 +1,188 @@ +/* Copyright (c) 2007-2018 Dovecot authors, see the included COPYING file */ + +#include "lib.h" +#include "str.h" +#include "qp-decoder.h" +#include "test-common.h" + +struct test_quoted_printable_decode_data { + const char *input; + const char *output; + size_t error_pos; + int ret; +}; + +static void test_qp_decoder(void) +{ +#define WHITESPACE10 " \t \t \t" +#define WHITESPACE70 WHITESPACE10 WHITESPACE10 WHITESPACE10 WHITESPACE10 WHITESPACE10 WHITESPACE10 WHITESPACE10 + static struct test_quoted_printable_decode_data tests[] = { + { "foo \r\nbar=\n", "foo\r\nbar", 0, 0 }, + { "foo\t=\nbar", "foo\tbar", 0, 0 }, + { "foo = \n=01", "foo \001", 0, 0 }, + { "foo =\t\r\nbar", "foo bar", 0, 0 }, + { "foo =\r\n=01", "foo \001", 0, 0 }, + { "foo \nbar=\r\n", "foo\r\nbar", 0, 0 }, + { "=0A=0D ", "\n\r", 0, 0 }, + { "foo_bar", "foo_bar", 0, 0 }, + { "\n\n", "\r\n\r\n", 0, 0 }, + { "\r\n\n\n\r\n", "\r\n\r\n\r\n\r\n", 0, 0 }, + + { "foo=", "foo=", 4, -1 }, + { "foo= =66", "foo= f", 5, -1 }, + { "foo= \t", "foo= \t", 6, -1 }, + { "foo= \r", "foo= \r", 6, -1 }, + { "foo= \r bar", "foo= \r bar", 6, -1 }, + { "foo=A", "foo=A", 5, -1 }, + { "foo=Ax", "foo=Ax", 5, -1 }, + { "foo=Ax=xy", "foo=Ax=xy", 5, -1 }, + + /* above 76 whitespaces is invalid and gets truncated + (at 77th whitespace because of the current implementation) */ + { WHITESPACE70" 7\n", WHITESPACE70" 7\r\n", 0, 0 }, + { WHITESPACE70" 8\n", WHITESPACE70" 8\r\n", 77, -1 }, + { WHITESPACE70" 9\n", WHITESPACE70" 9\r\n", 78, -1 }, + { WHITESPACE70" 0\n", WHITESPACE70" 0\r\n", 79, -1 }, + /* Expect extra whitespace to be truncated */ + { WHITESPACE70" 7\n"WHITESPACE10"", WHITESPACE70" 7\r\n", 0, 0 }, + { WHITESPACE70" 7=\r\n"WHITESPACE10, WHITESPACE70" 7", 0, 0 }, + /* Unnecessarily encoded */ + { "=66=6f=6f=42=61=72", "fooBar", 0, 0 }, + /* Expected to be encoded but not */ + { "\xc3\x9c""berm=c3=a4\xc3\x9figer Gebrauch", "\xc3\x9c""berm\xc3\xa4\xc3\x9figer Gebrauch", 0, 0 }, + /* Decode control characters */ + { "=0C=07", "\x0C\x07", 0, 0 }, + /* Data */ + { "=DE=AD=BE=EF", "\xDE\xAD\xBE\xEF", 0, 0 }, + /* Non hex data */ + { "=FJ=X1", "=FJ=X1", 2, -1 }, + /* No content allowed after Soft Line Break */ + { "=C3=9C = ","\xc3\x9c"" = ", 9, -1 }, + /* Boundary delimiter */ + { "=C3=9C=\r\n-------","\xc3\x9c""-------", 0, 0 }, + { "=----------- =C3=9C","=----------- \xc3\x9c""", 1, -1 }, + { "=___________ =C3=9C","=___________ \xc3\x9c""", 1, -1 }, + { "___________ =C3=9C","___________ \xc3\x9c""", 0, 0 }, + { "=2D=2D=2D=2D=2D=2D =C3=9C","------ \xc3\x9c""", 0, 0 }, + { "=FC=83=BF=BF=BF=BF", "\xFC\x83\xBF\xBF\xBF\xBF", 0, 0 }, + { "=FE=FE=FF=FF", "\xFE\xFE\xFF\xFF", 0, 0 }, + { "\xFF=C3=9C\xFE\xFF""foobar", "\xFF\xc3\x9c""\xFE\xFF""foobar", 0, 0 }, + /* Unnecessarily encoded and trailing whitespace */ + { + "=66=6f=6f=42=61=72 ", + "fooBar", 0, 0 + }, + /* Indicate error if encoded line is longer then 76 */ + { + WHITESPACE70" =C3=9C\n", + WHITESPACE70" \xc3\x9c""\r\n", 77, -1 + }, + /* Soft Line Break example from the RFC */ + { + "Now's the time =\r\nfor all folk to come=\r\n to the" + " aid of their country.", + "Now's the time for all folk to come to the aid of " + "their country.", 0, 0 + }, + { + "=C3=9Cberm=C3=A4=C3=9Figer Gebrauch", + "\xc3\x9c""berm\xc3\xa4\xc3\x9figer Gebrauch", 0, 0 + }, + /* Softlinebreak without following content */ + { + "=C3=9Cberm=C3=A4=C3=9Figer Gebrauch=", + "\xc3\x9c""berm\xc3\xa4\xc3\x9figer Gebrauch=", 36, -1 + }, + /* Lowercase formally illegal but allowed for robustness */ + { + "=c3=9cberm=c3=a4=c3=9figer Gebrauch", + "\xc3\x9c""berm\xc3\xa4\xc3\x9figer Gebrauch", 0, 0 + }, + /* Control characters in input */ + { + "=c3=9c=10berm=c3=a4=c3=9figer Geb=0Frauch", + "\xc3\x9c\x10""berm\xc3\xa4\xc3\x9figer Geb\x0Frauch", 0, 0 + }, + /* Trailing whitespace */ + { + "Trailing Whitesp=C3=A4ce =\r\n ", + "Trailing Whitesp\xc3\xa4""ce ", 0 ,0 + }, + { + "Trailing Whitesp=C3=A4ce ", + "Trailing Whitesp\xc3\xa4""ce", 0 ,0 + }, + { + "=54=65=73=74=20=6D=65=73=73=61=67=65", + "Test message", 0 , 0 + }, + { + "=E3=81=93=E3=82=8C=E3=81=AF=E5=A2\r\n=83=E7=95=8C=E3" + "=81=AE=E3=81=82=E3=82=8B=E3=83=A1=E3=83=83=E3=82=BB=" + "E3=83=BC=E3=82=B8=E3=81=A7=E3=81=99", + "\xE3\x81\x93\xE3\x82\x8C\xE3\x81\xAF\xE5\xA2\r\n\x83" + "\xE7\x95\x8C\xE3\x81\xAE\xE3\x81\x82\xE3\x82\x8B\xE3" + "\x83\xA1\xE3\x83\x83\xE3\x82\xBB\xE3\x83\xBC\xE3\x82" + "\xB8\xE3\x81\xA7\xE3\x81\x99", 0, 0 + }, + { + "=E3=81\xc3\xf1=93=E3=82=8\xff""C=E3=81=AF=E5=A2", + "\xE3\x81\xc3\xf1\x93\xE3\x82=8\xff""C\xE3\x81\xAF\xE5\xA2", + 19, -1 + }, + { + "\x77Hello\x76=20 \x20 =E3=81\xc3\xf1=93=E3=82", + "wHellov \xE3\x81\xc3\xf1\x93\xE3\x82", + 0, 0 + }, + }; + string_t *str; + unsigned int i, j; + + test_begin("qp-decoder"); + str = t_str_new(128); + for (i = 0; i < N_ELEMENTS(tests); i++) { + const char *input = tests[i].input; + struct qp_decoder *qp = qp_decoder_init(str); + size_t error_pos; + const char *error; + int ret; + + /* try all at once */ + ret = qp_decoder_more(qp, (const void *)input, strlen(input), + &error_pos, &error); + if (qp_decoder_finish(qp, &error) < 0 && ret == 0) { + error_pos = strlen(input); + ret = -1; + } + test_assert_idx(ret == tests[i].ret, i); + test_assert_idx(ret == 0 || error_pos == tests[i].error_pos, i); + test_assert_strcmp_idx(str_c(str), tests[i].output, i); + + /* try in small pieces */ + str_truncate(str, 0); + ret = 0; + for (j = 0; input[j] != '\0'; j++) { + unsigned char c = (unsigned char)input[j]; + if (qp_decoder_more(qp, &c, 1, &error_pos, &error) < 0) + ret = -1; + } + if (qp_decoder_finish(qp, &error) < 0) + ret = -1; + test_assert_idx(ret == tests[i].ret, i); + test_assert_strcmp_idx(str_c(str), tests[i].output, i); + + qp_decoder_deinit(&qp); + str_truncate(str, 0); + } + test_end(); +} + +int main(void) +{ + static void (*const test_functions[])(void) = { + test_qp_decoder, + NULL + }; + return test_run(test_functions); +} |