1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
|
/* Copyright (c) 2015-2018 Dovecot authors, see the included COPYING file */
#include "lib.h"
#include "str.h"
#include "istream.h"
#include "mail-html2text.h"
#include "test-common.h"
static const struct {
const char *input;
const char *output;
} tests[] = {
{ "&&aaaaaaaaaa", "" },
{ "a&<♣>b",
"a&<\xE2\x99\xA3>b" },
{ "&", "" },
{ "&", "" },
{ "a<style>stylesheet is ignored</style>b",
"a b" },
{ "a<stylea>b</stylea>c",
"a b c" },
{ "a<!--x <p foo=\"bar\">commented tags ignored also</p> y-->b",
"ab" },
{ "a<script>javascript <p>foo</p> ignored</script>b",
"a b" },
{ "a<scripta>b</scripta>c",
"a b c" },
{ "a<blockquote><blockquote>second level</blockquote>ignored</blockquote>b",
"a b" },
{ "a<![CDATA[<style>]] >b</style>]]>c",
"a<style>]] >b</style>c" },
{ "a<foo", "a" },
{ "a<blockquote", "a" },
{ "a<blockquote>foo</blockquote", "a " },
{ "a<", "a" },
{ "a<![CDATA[b", "ab" },
{ "a<![CDATA[b]]", "ab" },
{ "aä", "a\xC3\xA4" },
{ "aä", "a\xC3\xA4" },
{ "€", "\xE2\x82\xAC" },
{ "&#deee;", "" }, // invalid codepoint
};
static const char *test_blockquote_input[] = {
"a<blockquote>b<blockquote><blockquote>c</blockquote>d</blockquote>e</blockquote>f",
"a&<blockquote>b&<blockquote>&<blockquote>&c</blockquote>d&</blockquote>&e</blockquote>f&",
NULL
};
static const char *test_blockquote_output[] = {
"a\n> b\n> \n> c\n> d\n> e\nf",
"a&\n> b&\n> &\n> &c\n> d&\n> &e\nf&",
NULL
};
static void test_mail_html2text(void)
{
string_t *str = t_str_new(128);
struct mail_html2text *ht;
unsigned int i, j;
test_begin("mail_html2text()");
for (i = 0; i < N_ELEMENTS(tests); i++) {
ht = mail_html2text_init(MAIL_HTML2TEXT_FLAG_SKIP_QUOTED);
for (j = 0; tests[i].input[j] != '\0'; j++) {
unsigned char c = tests[i].input[j];
mail_html2text_more(ht, &c, 1, str);
}
test_assert_idx(strcmp(str_c(str), tests[i].output) == 0, i);
mail_html2text_deinit(&ht);
str_truncate(str, 0);
}
/* test without skipping quoted */
for (unsigned int i = 0; test_blockquote_input[i] != NULL; i++) {
str_truncate(str, 0);
ht = mail_html2text_init(0);
mail_html2text_more(ht, (const void *)test_blockquote_input[i],
strlen(test_blockquote_input[i]), str);
test_assert_idx(strcmp(str_c(str), test_blockquote_output[i]) == 0, i);
mail_html2text_deinit(&ht);
}
test_end();
}
static void test_mail_html2text_random(void)
{
string_t *str = t_str_new(128);
struct mail_html2text *ht;
test_begin("mail_html2text() random");
for (unsigned int i = 0; i < 1000; i++) {
char valid_chars[] = { '0', 'a', '<', '>', '&', ';', '\\', '\'', '"', '/' };
unsigned char s[2];
ht = mail_html2text_init(0);
for (unsigned int i = 0; i < 100; i++) {
s[0] = valid_chars[i_rand_limit(N_ELEMENTS(valid_chars))];
s[1] = valid_chars[i_rand_limit(N_ELEMENTS(valid_chars))];
mail_html2text_more(ht, s, i_rand_minmax(1, 2), str);
}
mail_html2text_deinit(&ht);
str_truncate(str, 0);
}
test_end();
}
int main(void)
{
static void (*const test_functions[])(void) = {
test_mail_html2text,
test_mail_html2text_random,
NULL
};
return test_run(test_functions);
}
|
