1
use crate::gpg::{check_signature, MissingKeyError, PgpSignature};
2
use chrono::{DateTime, FixedOffset, NaiveDateTime, TimeZone};
3
use git2::Repository;
4
use sequoia_openpgp::{Cert, KeyHandle};
5
use serde::Serialize;
6

            
7
type Result<T> = anyhow::Result<T>;
8

            
9
const BEGIN_PGP_SIGNATURE: &[u8] = b"-----BEGIN PGP SIGNATURE-----\n";
10
const END_PGP_SIGNATURE: &[u8] = b"-----END PGP SIGNATURE-----";
11
const BEGIN_PGP_MESSAGE: &[u8] = b"-----BEGIN PGP MESSAGE-----\n";
12
const END_PGP_MESSAGE: &[u8] = b"-----END PGP MESSAGE-----";
13

            
14
168
fn find_subsequence(haystack: &[u8], needle: &[u8]) -> Option<usize> {
15
168
    haystack
16
168
        .windows(needle.len())
17
32568
        .position(|window| window == needle)
18
168
}
19

            
20
2904
fn find_begin_end<'a>(
21
2904
    msg: &'a [u8],
22
2904
    begin: &'static [u8],
23
2904
    end: &'static [u8],
24
2904
) -> Option<(&'a [u8], &'a [u8])> {
25
2904
    if msg.len() < begin.len() + end.len() + 1 {
26
        return None;
27
2904
    }
28
2904
    let maybe_end = if msg[msg.len() - 1] == b'\n' {
29
2904
        &msg[msg.len() - end.len() - 1..msg.len() - 1]
30
    } else {
31
        &msg[msg.len() - end.len()..]
32
    };
33
2904
    if maybe_end != end {
34
2736
        return None;
35
168
    }
36
168
    if let Some(first) = find_subsequence(msg, begin) {
37
168
        let signature = &msg[first..];
38
168
        return Some((&msg[..first], signature));
39
    }
40
    None
41
2904
}
42

            
43
1536
fn get_pgp_signature(msg: Option<&[u8]>) -> Option<(&[u8], &[u8])> {
44
1536
    let msg = msg?;
45
1536
    if let Some(r) = find_begin_end(msg, BEGIN_PGP_SIGNATURE, END_PGP_SIGNATURE) {
46
168
        return Some(r);
47
1368
    }
48
1368
    if let Some(r) = find_begin_end(msg, BEGIN_PGP_MESSAGE, END_PGP_MESSAGE) {
49
        return Some(r);
50
1368
    }
51
1368
    None
52
1536
}
53

            
54
struct PartialGitSignature {
55
    oid: String,
56
    parent: String,
57
    name: Option<String>,
58
    email: Option<String>,
59
    datetime: Option<DateTime<FixedOffset>>,
60
}
61

            
62
432
fn strip_prefix<'a>(bytes: &'a [u8], prefix: &[u8]) -> Option<&'a [u8]> {
63
432
    if bytes.len() >= prefix.len() && &bytes[..prefix.len()] == prefix {
64
432
        return Some(&bytes[prefix.len()..]);
65
    }
66
    None
67
432
}
68

            
69
fn split_at_newline(bytes: &[u8]) -> Option<(&[u8], &[u8])> {
70
19224
    if let Some(pos) = bytes.iter().position(|b| *b == b'\n') {
71
432
        return Some((&bytes[..pos], &bytes[pos + 1..]));
72
    }
73
    None
74
432
}
75

            
76
432
fn get_line_value<'a>(bytes: &'a [u8], prefix: &[u8]) -> Option<(&'a [u8], &'a [u8])> {
77
432
    let value = strip_prefix(bytes, prefix)?;
78
432
    split_at_newline(value)
79
432
}
80

            
81
/// Parse a value of the form
82
/// 'First Last <flast@kernel.org> 1493772623 +0430'
83
144
fn parse_tagger(tagger: &[u8]) -> Option<(String, String, DateTime<FixedOffset>)> {
84
2628
    let gt_pos = tagger.iter().position(|b| *b == b'<')?;
85
144
    if gt_pos == 0 || tagger[gt_pos - 1] != b' ' {
86
        return None;
87
144
    }
88
144
    let name = &tagger[..gt_pos - 1];
89
144
    let rest = &tagger[gt_pos + 1..];
90
2196
    let lt_pos = rest.iter().position(|b| *b == b'>')?;
91
144
    if lt_pos + 1 == rest.len() || rest[lt_pos + 1] != b' ' {
92
        return None;
93
144
    }
94
144
    let email = &rest[..lt_pos];
95
144
    let rest = &rest[lt_pos + 2..];
96
1584
    let space_pos = rest.iter().position(|b| *b == b' ')?;
97
144
    let seconds = &rest[..space_pos];
98
144
    let tz = &rest[space_pos + 1..];
99
144
    let name = String::from_utf8(name.to_vec()).ok()?;
100
144
    let email = String::from_utf8(email.to_vec()).ok()?;
101
144
    let unix_timestamp_seconds: i64 = std::str::from_utf8(seconds)
102
144
        .ok()
103
144
        .and_then(|s| s.parse::<i64>().ok())?;
104
144
    if tz.len() != 5 {
105
        return None;
106
144
    }
107
144
    let offset_hours = std::str::from_utf8(&tz[1..2])
108
144
        .ok()
109
144
        .and_then(|s| s.parse::<i32>().ok())?;
110
144
    let offset_minutes = std::str::from_utf8(&tz[3..4])
111
144
        .ok()
112
144
        .and_then(|s| s.parse::<i32>().ok())?;
113
144
    let offset_seconds = offset_hours * 3600 + offset_minutes * 60;
114
144
    let dt = NaiveDateTime::from_timestamp_opt(unix_timestamp_seconds, 0)?;
115
144
    let offset = if tz[0] == b'-' {
116
        FixedOffset::west_opt(offset_seconds)
117
    } else {
118
144
        FixedOffset::east_opt(offset_seconds)
119
    }?;
120
144
    let datetime = offset.from_utc_datetime(&dt);
121
144
    Some((name, email, datetime))
122
144
}
123

            
124
/// obtain name, email, date, time and tag from the text of the commit
125
144
fn parse_payload(payload: &[u8]) -> Option<PartialGitSignature> {
126
144
    let (tree, remainder) = get_line_value(payload, b"tree ")?;
127
144
    let oid = String::from_utf8(tree.to_vec()).ok()?;
128
144
    let (parent, remainder) = get_line_value(remainder, b"parent ")?;
129
144
    let poid = String::from_utf8(parent.to_vec()).ok()?;
130
144
    let mut name = None;
131
144
    let mut email = None;
132
144
    let mut datetime = None;
133
144
    if let Some((author, _)) = get_line_value(remainder, b"author ") {
134
144
        if let Some((n, e, dt)) = parse_tagger(author) {
135
144
            name = Some(n);
136
144
            email = Some(e);
137
144
            datetime = Some(dt);
138
144
        }
139
    };
140
144
    Some(PartialGitSignature {
141
144
        oid,
142
144
        parent: poid,
143
144
        name,
144
144
        email,
145
144
        datetime,
146
144
    })
147
144
}
148

            
149
/// Retrieve the signature of the commit if it has one.
150
1512
fn get_commit_signature(commit: &git2::Commit) -> Option<TagSignature> {
151
1512
    let msg = commit.raw_header().map(|b| b.as_bytes());
152
1512
    parse_signature(msg)
153
1512
}
154

            
155
1512
fn parse_signature(msg: Option<&[u8]>) -> Option<TagSignature> {
156
1512
    let (message, signature) = get_pgp_signature(msg)?;
157
144
    if let Some(partial) = parse_payload(message) {
158
144
        Some(TagSignature {
159
144
            tag: partial.parent,
160
144
            oid: partial.oid,
161
144
            name: partial.name,
162
144
            email: partial.email,
163
144
            datetime: partial.datetime,
164
144
            signature: PgpSignature {
165
144
                payload: message.to_vec(),
166
144
                signature: signature.to_vec(),
167
144
            },
168
144
        })
169
    } else {
170
        /*
171
        println!(
172
            "could not parse message '{}'",
173
            String::from_utf8_lossy(message)
174
        );
175
        */
176
        None
177
    }
178
1512
}
179

            
180
/// Retrieve the signature of the tag if it has one.
181
24
fn get_tag_signature(tag: &git2::Tag) -> Option<TagSignature> {
182
24
    let (message, signature) = get_pgp_signature(tag.message_bytes())?;
183
24
    let tag_name = tag.name()?;
184
    // create a payload for checking the digital signature
185
    // see https://git-scm.com/docs/signature-format
186
24
    let mut payload: Vec<u8> = Vec::with_capacity(256 + message.len());
187
24
    payload.extend_from_slice(b"object ");
188
24
    let id = format!("{}", tag.target_id());
189
24
    payload.extend_from_slice(id.as_bytes());
190
24
    payload.extend_from_slice(b"\ntype commit\ntag ");
191
24
    payload.extend_from_slice(tag.name_bytes());
192
24
    payload.push(b'\n');
193
24
    let mut name = None;
194
24
    let mut email = None;
195
24
    let mut datetime = None;
196
24
    if let Some(tagger) = tag.tagger() {
197
24
        name = tagger.name().map(ToString::to_string);
198
24
        email = tagger.email().map(ToString::to_string);
199
24
        let when = tagger.when();
200
24
        let unix_timestamp_seconds = when.seconds();
201
24
        let offset_hours = when.offset_minutes() / 60;
202
24
        let offset_minutes = when.offset_minutes() % 60;
203
24
        let tagger = format!(
204
24
            "tagger {} {} {:+03}{:02}\n",
205
24
            tagger, unix_timestamp_seconds, offset_hours, offset_minutes
206
24
        );
207
24
        payload.extend_from_slice(tagger.as_bytes());
208
24
        if let Some(dt) = NaiveDateTime::from_timestamp_opt(unix_timestamp_seconds, 0) {
209
24
            datetime =
210
24
                FixedOffset::east_opt(when.offset_minutes() * 60).map(|o| o.from_utc_datetime(&dt));
211
24
        }
212
    }
213
24
    payload.push(b'\n');
214
24
    payload.extend_from_slice(message);
215
24
    let oid = format!("{}", tag.id());
216
24
    Some(TagSignature {
217
24
        tag: tag_name.to_string(),
218
24
        oid,
219
24
        name,
220
24
        email,
221
24
        datetime,
222
24
        signature: PgpSignature {
223
24
            payload,
224
24
            signature: signature.to_vec(),
225
24
        },
226
24
    })
227
24
}
228

            
229
4
#[derive(Serialize, PartialEq)]
230
pub struct TagSignature {
231
    tag: String,
232
    oid: String,
233
    #[serde(skip_serializing_if = "Option::is_none")]
234
    name: Option<String>,
235
    #[serde(skip_serializing_if = "Option::is_none")]
236
    email: Option<String>,
237
    #[serde(skip_serializing_if = "Option::is_none")]
238
    datetime: Option<DateTime<FixedOffset>>,
239
    pub signature: PgpSignature,
240
}
241

            
242
24
pub fn get_signatures(repo: &Repository) -> Result<Vec<TagSignature>> {
243
24
    let mut signatures = Vec::new();
244
180
    for reference in repo.references()? {
245
180
        let reference = reference?;
246
180
        if let Ok(commit) = reference.peel_to_commit() {
247
180
            if let Some(signature) = get_commit_signature(&commit) {
248
48
                signatures.push(signature);
249
132
            }
250
        }
251
180
        if let Ok(tag) = reference.peel_to_tag() {
252
24
            if let Some(signature) = get_tag_signature(&tag) {
253
24
                signatures.push(signature);
254
24
            }
255
156
        }
256
    }
257
24
    let mut revwalk = repo.revwalk()?;
258
24
    revwalk.push_head()?;
259
1356
    for oid in revwalk {
260
1332
        if let Ok(commit) = oid.and_then(|o| repo.find_commit(o)) {
261
1332
            if let Some(signature) = get_commit_signature(&commit) {
262
96
                signatures.push(signature);
263
1236
            }
264
        }
265
    }
266
24
    Ok(signatures)
267
24
}
268

            
269
/// Check the digital signature on the tag.
270
/// Return the ids of the keys with which the signature was created.
271
/// Return an empty vector if the signature could not be checked due to
272
/// missing keys. The missing keys are added to the supplied vector.
273
/// Return an error if the check failed.
274
168
fn check_tag_signature(
275
168
    signature: &TagSignature,
276
168
    certs: &[Cert],
277
168
    missing_keys: &mut Vec<KeyHandle>,
278
168
) -> Result<Vec<(KeyHandle, Cert)>> {
279
168
    match check_signature(certs, &signature.signature) {
280
168
        Ok(validated_signature) => {
281
324
            for e in &validated_signature.errors {
282
156
                if let Some(source) = e.downcast_ref::<MissingKeyError>() {
283
144
                    if let Some(missing_key) = source.missing_key() {
284
144
                        if !missing_keys.contains(&missing_key) {
285
36
                            missing_keys.push(missing_key);
286
108
                        }
287
                    }
288
12
                }
289
            }
290
168
            Ok(validated_signature.certificates)
291
        }
292
        Err(e) => Err(e),
293
    }
294
168
}
295

            
296
pub struct CheckSignaturesResult {
297
    pub valid_signatures: Vec<TagSignature>,
298
    pub found_keys: Vec<(KeyHandle, Cert)>,
299
    pub missing_keys: Vec<KeyHandle>,
300
}
301

            
302
24
pub fn check_signatures(
303
24
    certs: &[Cert],
304
24
    signatures: Vec<TagSignature>,
305
24
) -> Result<CheckSignaturesResult> {
306
24
    let mut missing_keys = Vec::new();
307
24
    let mut found_keys = Vec::new();
308
24
    let mut valid_signatures = Vec::new();
309
192
    for signature in signatures {
310
168
        if let Ok(mut key_ids) = check_tag_signature(&signature, certs, &mut missing_keys) {
311
168
            let checked = !key_ids.is_empty();
312
168
            if checked {
313
24
                valid_signatures.push(signature);
314
24
                found_keys.append(&mut key_ids);
315
144
            }
316
        }
317
    }
318
24
    Ok(CheckSignaturesResult {
319
24
        valid_signatures,
320
24
        found_keys,
321
24
        missing_keys,
322
24
    })
323
24
}