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
121
122
123
124
use crate::{parsers::MessageStream, HeaderValue};
impl<'x> MessageStream<'x> {
pub fn parse_raw(&mut self) -> HeaderValue<'x> {
let mut token_start: usize = 0;
let mut token_end: usize = 0;
while let Some(ch) = self.next() {
match ch {
b'\n' => {
if !self.try_next_is_space() {
return if token_start > 0 {
HeaderValue::Text(String::from_utf8_lossy(
self.bytes(token_start - 1..token_end),
))
} else {
HeaderValue::Empty
};
} else {
continue;
}
}
b' ' | b'\t' | b'\r' => continue,
_ => (),
}
if token_start == 0 {
token_start = self.offset();
}
token_end = self.offset();
}
HeaderValue::Empty
}
pub fn parse_and_ignore(&mut self) {
while let Some(&ch) = self.next() {
if ch == b'\n' && !self.try_next_is_space() {
break;
}
}
}
}
#[cfg(test)]
mod tests {
use crate::{parsers::MessageStream, Message};
#[test]
fn parse_raw_text() {
let inputs = [
("Saying Hello\nMessage-Id", "Saying Hello"),
("Re: Saying Hello\r\n \r\nFrom:", "Re: Saying Hello"),
(
concat!(
" from x.y.test\n by example.net\n via TCP\n",
" with ESMTP\n id ABC12345\n ",
"for <mary@example.net>; 21 Nov 1997 10:05:43 -0600\n"
),
concat!(
"from x.y.test\n by example.net\n via TCP\n",
" with ESMTP\n id ABC12345\n ",
"for <mary@example.net>; 21 Nov 1997 10:05:43 -0600"
),
),
];
for input in inputs {
let str = input.0.to_string();
assert_eq!(
MessageStream::new(str.as_bytes()).parse_raw().unwrap_text(),
input.1,
"Failed for '{:?}'",
input.0
);
}
}
#[test]
fn ordered_raw_headers() {
let input = br#"From: Art Vandelay <art@vandelay.com>
To: jane@example.com
Date: Sat, 20 Nov 2021 14:22:01 -0800
Subject: Why not both importing AND exporting? =?utf-8?b?4pi6?=
Content-Type: multipart/mixed; boundary="festivus";
Here's a message body.
"#;
let message = Message::parse(input).unwrap();
let mut iter = message.headers_raw();
assert_eq!(
iter.next().unwrap(),
("From", " Art Vandelay <art@vandelay.com>\n")
);
assert_eq!(iter.next().unwrap(), ("To", " jane@example.com\n"));
assert_eq!(
iter.next().unwrap(),
("Date", " Sat, 20 Nov 2021 14:22:01 -0800\n")
);
assert_eq!(
iter.next().unwrap(),
(
"Subject",
" Why not both importing AND exporting? =?utf-8?b?4pi6?=\n"
)
);
assert_eq!(
iter.next().unwrap(),
("Content-Type", " multipart/mixed; boundary=\"festivus\";\n")
);
}
}