1use std::io::Read;
32
33use ssh_key::{HashAlg, LineEnding, PrivateKey, PublicKey, SshSig};
34
35use crate::agent::client::Agent;
36use crate::allowed_signers::AllowedSigners;
37use crate::GitwayError;
38
39#[derive(Debug, Clone)]
43pub struct Verified {
44 pub principal: String,
47 pub fingerprint: String,
49}
50
51pub fn sign<R: Read>(
65 data: &mut R,
66 key: &PrivateKey,
67 namespace: &str,
68 hash: HashAlg,
69) -> Result<String, GitwayError> {
70 let mut buf = Vec::new();
71 data.read_to_end(&mut buf)?;
72 let sig = SshSig::sign(key, namespace, hash, &buf)
73 .map_err(|e| GitwayError::signing(format!("sshsig sign failed: {e}")))?;
74 sig.to_pem(LineEnding::LF)
75 .map_err(|e| GitwayError::signing(format!("sshsig armor failed: {e}")))
76}
77
78pub fn sign_with_agent<R: Read>(
93 data: &mut R,
94 agent: &mut Agent,
95 public_key: &PublicKey,
96 namespace: &str,
97 hash: HashAlg,
98) -> Result<String, GitwayError> {
99 let mut buf = Vec::new();
100 data.read_to_end(&mut buf)?;
101 let signed_blob = SshSig::signed_data(namespace, hash, &buf)
102 .map_err(|e| GitwayError::signing(format!("sshsig signed_data failed: {e}")))?;
103 let signature = agent.sign(public_key, &signed_blob)?;
104 let sig = SshSig::new(public_key.key_data().clone(), namespace, hash, signature)
105 .map_err(|e| GitwayError::signing(format!("sshsig wrap failed: {e}")))?;
106 sig.to_pem(LineEnding::LF)
107 .map_err(|e| GitwayError::signing(format!("sshsig armor failed: {e}")))
108}
109
110pub fn verify<R: Read>(
124 data: &mut R,
125 armored_sig: &str,
126 signer_identity: &str,
127 namespace: &str,
128 allowed: &AllowedSigners,
129) -> Result<Verified, GitwayError> {
130 let sig = SshSig::from_pem(armored_sig)
131 .map_err(|e| GitwayError::signature_invalid(format!("malformed signature: {e}")))?;
132
133 if sig.namespace() != namespace {
134 return Err(GitwayError::signature_invalid(format!(
135 "namespace mismatch: signature is {:?}, expected {namespace:?}",
136 sig.namespace()
137 )));
138 }
139
140 let mut buf = Vec::new();
141 data.read_to_end(&mut buf)?;
142
143 let public_key = PublicKey::from(sig.public_key().clone());
144 public_key
145 .verify(namespace, &buf, &sig)
146 .map_err(|e| GitwayError::signature_invalid(format!("cryptographic check failed: {e}")))?;
147
148 if !allowed.is_authorized(signer_identity, &public_key, namespace) {
149 return Err(GitwayError::signature_invalid(format!(
150 "signer {signer_identity:?} is not authorized for namespace {namespace:?} \
151 with key {}",
152 public_key.fingerprint(HashAlg::Sha256)
153 )));
154 }
155
156 Ok(Verified {
157 principal: signer_identity.to_owned(),
158 fingerprint: public_key.fingerprint(HashAlg::Sha256).to_string(),
159 })
160}
161
162pub fn check_novalidate<R: Read>(
172 data: &mut R,
173 armored_sig: &str,
174 namespace: &str,
175) -> Result<(), GitwayError> {
176 let sig = SshSig::from_pem(armored_sig)
177 .map_err(|e| GitwayError::signature_invalid(format!("malformed signature: {e}")))?;
178
179 if sig.namespace() != namespace {
180 return Err(GitwayError::signature_invalid(format!(
181 "namespace mismatch: signature is {:?}, expected {namespace:?}",
182 sig.namespace()
183 )));
184 }
185
186 let mut buf = Vec::new();
187 data.read_to_end(&mut buf)?;
188
189 let public_key = PublicKey::from(sig.public_key().clone());
190 public_key
191 .verify(namespace, &buf, &sig)
192 .map_err(|e| GitwayError::signature_invalid(format!("cryptographic check failed: {e}")))?;
193
194 Ok(())
195}
196
197pub fn find_principals(
209 armored_sig: &str,
210 allowed: &AllowedSigners,
211 namespace: &str,
212) -> Result<Vec<String>, GitwayError> {
213 let sig = SshSig::from_pem(armored_sig)
214 .map_err(|e| GitwayError::signature_invalid(format!("malformed signature: {e}")))?;
215 let public_key = PublicKey::from(sig.public_key().clone());
216 Ok(allowed
217 .find_principals(&public_key, namespace)
218 .iter()
219 .map(|s| (*s).to_owned())
220 .collect())
221}
222
223#[cfg(test)]
226mod tests {
227 use super::*;
228 use std::io::Cursor;
229
230 use crate::keygen::{generate, KeyType};
231
232 fn roundtrip(kind: KeyType, hash: HashAlg) {
233 let key = generate(kind, None, "sign@test").unwrap();
234 let payload = b"the quick brown fox jumps over the lazy dog";
235 let armored = sign(&mut Cursor::new(payload), &key, "git", hash).unwrap();
236 assert!(armored.contains("BEGIN SSH SIGNATURE"));
237
238 check_novalidate(&mut Cursor::new(payload), &armored, "git").unwrap();
240
241 let err = check_novalidate(&mut Cursor::new(payload), &armored, "file").unwrap_err();
243 assert!(err.to_string().contains("namespace"));
244
245 let err = check_novalidate(&mut Cursor::new(b"tampered"), &armored, "git").unwrap_err();
247 assert!(err.to_string().contains("cryptographic"));
248 }
249
250 #[test]
251 fn ed25519_sign_verify_roundtrip() {
252 roundtrip(KeyType::Ed25519, HashAlg::Sha512);
253 }
254
255 #[test]
256 fn ecdsa_p256_sign_verify_roundtrip() {
257 roundtrip(KeyType::EcdsaP256, HashAlg::Sha512);
258 }
259
260 #[test]
265 #[ignore = "RSA SSHSIG path not yet wired up in ssh-key 0.6.7"]
266 fn rsa_sign_verify_roundtrip() {
267 let key = generate(KeyType::Rsa, Some(2048), "rsa-sign@test").unwrap();
268 let payload = b"hello rsa";
269 let armored = sign(&mut Cursor::new(payload), &key, "git", HashAlg::Sha512).unwrap();
270 check_novalidate(&mut Cursor::new(payload), &armored, "git").unwrap();
271 }
272
273 #[test]
274 fn verify_against_allowed_signers_success() {
275 let key = generate(KeyType::Ed25519, None, "alice@test").unwrap();
276 let pubkey_line = key.public_key().to_openssh().unwrap();
277 let allowed_text = format!("alice@example.com {pubkey_line}");
278 let allowed = AllowedSigners::parse(&allowed_text).unwrap();
279
280 let payload = b"signed content";
281 let armored = sign(&mut Cursor::new(payload), &key, "git", HashAlg::Sha512).unwrap();
282
283 let verified = verify(
284 &mut Cursor::new(payload),
285 &armored,
286 "alice@example.com",
287 "git",
288 &allowed,
289 )
290 .unwrap();
291 assert_eq!(verified.principal, "alice@example.com");
292 assert!(verified.fingerprint.starts_with("SHA256:"));
293 }
294
295 #[test]
296 fn verify_against_allowed_signers_rejects_unknown_identity() {
297 let key = generate(KeyType::Ed25519, None, "bob@test").unwrap();
298 let pubkey_line = key.public_key().to_openssh().unwrap();
299 let allowed_text = format!("alice@example.com {pubkey_line}");
300 let allowed = AllowedSigners::parse(&allowed_text).unwrap();
301
302 let payload = b"signed content";
303 let armored = sign(&mut Cursor::new(payload), &key, "git", HashAlg::Sha512).unwrap();
304
305 let err = verify(
306 &mut Cursor::new(payload),
307 &armored,
308 "mallory@example.com",
309 "git",
310 &allowed,
311 )
312 .unwrap_err();
313 assert!(err.to_string().contains("not authorized"));
314 }
315
316 #[test]
317 fn find_principals_returns_matching_entries() {
318 let key = generate(KeyType::Ed25519, None, "carol@test").unwrap();
319 let pubkey_line = key.public_key().to_openssh().unwrap();
320 let allowed_text = format!("carol@example.com,dave@example.com {pubkey_line}");
321 let allowed = AllowedSigners::parse(&allowed_text).unwrap();
322
323 let armored = sign(&mut Cursor::new(b"x"), &key, "git", HashAlg::Sha512).unwrap();
324 let principals = find_principals(&armored, &allowed, "git").unwrap();
325 assert!(principals.iter().any(|p| p == "carol@example.com"));
326 assert!(principals.iter().any(|p| p == "dave@example.com"));
327 }
328}