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
125
126
127
128
129
130
131
132
use std::fmt;
use http::Uri;
use nom::{
branch::alt,
bytes::complete::{is_not, tag},
character::complete::line_ending,
combinator::map_res,
IResult,
};
use crate::Parse;
#[derive(Clone, Debug, PartialEq)]
pub enum EncryptionKey {
Clear(String),
Base64(String),
Uri(Uri),
Prompt,
}
impl fmt::Display for EncryptionKey {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
match self {
Self::Clear(key) => writeln!(f, "k=clear:{}\r", key),
Self::Base64(key) => writeln!(f, "k=base64:{}\r", key),
Self::Uri(uri) => writeln!(f, "k=uri:{}\r", uri),
Self::Prompt => writeln!(f, "k=prompt\r"),
}
}
}
impl Parse for EncryptionKey {
fn parse(input: &str) -> IResult<&str, Self> {
let (rest, _) = tag("k=")(input)?;
let (rest, key) = alt((parse_clear, parse_base64, parse_uri, parse_prompt))(rest)?;
let (rest, _) = line_ending(rest)?;
Ok((rest, key))
}
}
fn parse_clear(input: &str) -> IResult<&str, EncryptionKey> {
let (rest, _) = tag("clear:")(input)?;
let (rest, key) = is_not("\r\n")(rest)?;
Ok((rest, EncryptionKey::Clear(key.to_owned())))
}
fn parse_base64(input: &str) -> IResult<&str, EncryptionKey> {
let (rest, _) = tag("base64:")(input)?;
let (rest, key) = is_not("\r\n")(rest)?;
Ok((rest, EncryptionKey::Base64(key.to_owned())))
}
fn parse_uri(input: &str) -> IResult<&str, EncryptionKey> {
let (rest, _) = tag("uri:")(input)?;
let (rest, key) = map_res(is_not("\r\n"), str::parse)(rest)?;
Ok((rest, EncryptionKey::Uri(key)))
}
fn parse_prompt(input: &str) -> IResult<&str, EncryptionKey> {
let (rest, _) = tag("prompt")(input)?;
Ok((rest, EncryptionKey::Prompt))
}
#[cfg(test)]
mod tests {
use super::*;
use crate::test_util::{assert_err, assert_parse_display};
#[test]
fn test_clear() {
assert_parse_display(
"k=clear:foo\r\nmore",
"more",
&EncryptionKey::Clear("foo".to_owned()),
"k=clear:foo\r\n",
);
assert_err::<EncryptionKey>("k=clear\r\nmore");
assert_err::<EncryptionKey>("k=clear:\r\nmore");
}
#[test]
fn test_base64() {
assert_parse_display(
"k=base64:foo\r\n\rmore",
"\rmore",
&EncryptionKey::Base64("foo".to_owned()),
"k=base64:foo\r\n",
);
assert_err::<EncryptionKey>("k=base64\r\nmore");
assert_err::<EncryptionKey>("k=base64:\r\nmore");
}
#[test]
fn test_uri() {
let uri_str = "https://example.org/key";
let uri = uri_str.parse().unwrap();
assert_parse_display(
&format!("k=uri:{}\r\n\nmore\r", uri_str),
"\nmore\r",
&EncryptionKey::Uri(uri),
&format!("k=uri:{}\r\n", uri_str),
);
assert_err::<EncryptionKey>("k=uri\r\nmore");
assert_err::<EncryptionKey>("k=uri:\r\nmore");
assert_err::<EncryptionKey>("k=uri:!@#$\r\nmore");
}
#[test]
fn test_prompt() {
assert_parse_display(
"k=prompt\r\nmore",
"more",
&EncryptionKey::Prompt,
"k=prompt\r\n",
);
assert_err::<EncryptionKey>("k=prompt:foo\r\nmore");
assert_err::<EncryptionKey>("k=prompt:\r\nmore");
}
#[test]
fn test_invalid() {
assert_err::<EncryptionKey>("k=foo\r\nmore");
assert_err::<EncryptionKey>("k=foo:\r\nmore");
assert_err::<EncryptionKey>("k=foo:bar\r\nmore");
}
}