1
//! Re-exporting RSA implementations.
2
//!
3
//! This module can currently handle public keys and signature
4
//! verification used in the Tor directory protocol and
5
//! similar places.
6
//!
7
//! Currently, that means validating PKCSv1 signatures, and encoding
8
//! and decoding RSA public keys from DER.
9
//!
10
//! # Limitations:
11
//!
12
//! Currently missing are support for signing and RSA-OEAP.  In Tor,
13
//! RSA signing is only needed for relays and authorities, and
14
//! RSA-OAEP padding is only needed for the (obsolete) TAP protocol.
15
//!
16
//! This module should expose RustCrypto trait-based wrappers,
17
//! but the [`rsa`] crate didn't support them as of initial writing.
18
use rsa::pkcs1::{DecodeRsaPrivateKey, DecodeRsaPublicKey};
19
use std::fmt;
20
use subtle::{Choice, ConstantTimeEq};
21

            
22
#[cfg(feature = "memquota-memcost")]
23
use {derive_deftly::Deftly, tor_memquota::derive_deftly_template_HasMemoryCost};
24

            
25
use crate::util::{ct::CtByteArray, rng::RngCompat};
26

            
27
pub use rsa::Error;
28

            
29
/// How many bytes are in an "RSA ID"?  (This is a legacy tor
30
/// concept, and refers to identifying a relay by a SHA1 digest
31
/// of its RSA public identity key.)
32
pub const RSA_ID_LEN: usize = 20;
33

            
34
/// An identifier for an RSA key, based on SHA1 and DER.
35
///
36
/// These are used (for legacy purposes) all over the Tor protocol.
37
///
38
/// This object is an "identity" in the sense that it identifies (up to) one RSA
39
/// key.  It may also represent the identity for a particular entity, such as a
40
/// relay or a directory authority.
41
///
42
/// Note that for modern purposes, you should almost always identify a relay by
43
/// its [`Ed25519Identity`](crate::pk::ed25519::Ed25519Identity) instead of by
44
/// this kind of identity key.
45
#[derive(Clone, Copy, Hash, Ord, PartialOrd, Eq, PartialEq)]
46
#[cfg_attr(
47
    feature = "memquota-memcost",
48
    derive(Deftly),
49
    derive_deftly(HasMemoryCost)
50
)]
51
pub struct RsaIdentity {
52
    /// SHA1 digest of a DER encoded public key.
53
    id: CtByteArray<RSA_ID_LEN>,
54
}
55

            
56
impl ConstantTimeEq for RsaIdentity {
57
    fn ct_eq(&self, other: &Self) -> Choice {
58
        self.id.ct_eq(&other.id)
59
    }
60
}
61

            
62
impl fmt::Display for RsaIdentity {
63
77544
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
64
77544
        write!(f, "${}", hex::encode(&self.id.as_ref()[..]))
65
77544
    }
66
}
67
impl fmt::Debug for RsaIdentity {
68
40608
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
69
40608
        write!(f, "RsaIdentity {{ {} }}", self)
70
40608
    }
71
}
72

            
73
impl safelog::Redactable for RsaIdentity {
74
    /// Warning: This displays 16 bits of the RSA identity, which is
75
    /// enough to narrow down a public relay by a great deal.
76
    fn display_redacted(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
77
        write!(f, "${}…", hex::encode(&self.id.as_ref()[..1]))
78
    }
79

            
80
    fn debug_redacted(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
81
        write!(f, "RsaIdentity {{ {} }}", self.redacted())
82
    }
83
}
84

            
85
impl serde::Serialize for RsaIdentity {
86
17078
    fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
87
17078
    where
88
17078
        S: serde::Serializer,
89
    {
90
17078
        if serializer.is_human_readable() {
91
17076
            serializer.serialize_str(&hex::encode(&self.id.as_ref()[..]))
92
        } else {
93
2
            serializer.serialize_bytes(&self.id.as_ref()[..])
94
        }
95
17078
    }
96
}
97

            
98
impl<'de> serde::Deserialize<'de> for RsaIdentity {
99
570
    fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
100
570
    where
101
570
        D: serde::Deserializer<'de>,
102
    {
103
570
        if deserializer.is_human_readable() {
104
            /// Deserialization helper
105
            struct RsaIdentityVisitor;
106
            impl<'de> serde::de::Visitor<'de> for RsaIdentityVisitor {
107
                type Value = RsaIdentity;
108
                fn expecting(&self, fmt: &mut std::fmt::Formatter<'_>) -> fmt::Result {
109
                    fmt.write_str("hex-encoded RSA identity")
110
                }
111
566
                fn visit_str<E>(self, s: &str) -> Result<Self::Value, E>
112
566
                where
113
566
                    E: serde::de::Error,
114
                {
115
566
                    RsaIdentity::from_hex(s)
116
566
                        .ok_or_else(|| E::custom("wrong encoding for RSA identity"))
117
566
                }
118
            }
119

            
120
566
            deserializer.deserialize_str(RsaIdentityVisitor)
121
        } else {
122
            /// Deserialization helper
123
            struct RsaIdentityVisitor;
124
            impl<'de> serde::de::Visitor<'de> for RsaIdentityVisitor {
125
                type Value = RsaIdentity;
126
                fn expecting(&self, fmt: &mut std::fmt::Formatter<'_>) -> fmt::Result {
127
                    fmt.write_str("RSA identity")
128
                }
129
4
                fn visit_bytes<E>(self, bytes: &[u8]) -> Result<Self::Value, E>
130
4
                where
131
4
                    E: serde::de::Error,
132
                {
133
4
                    RsaIdentity::from_bytes(bytes)
134
4
                        .ok_or_else(|| E::custom("wrong length for RSA identity"))
135
4
                }
136
            }
137
4
            deserializer.deserialize_bytes(RsaIdentityVisitor)
138
        }
139
570
    }
140
}
141

            
142
impl RsaIdentity {
143
    /// Expose an RsaIdentity as a slice of bytes.
144
744048
    pub fn as_bytes(&self) -> &[u8] {
145
744048
        &self.id.as_ref()[..]
146
744048
    }
147
    /// Construct an RsaIdentity from a slice of bytes.
148
    ///
149
    /// Returns None if the input is not of the correct length.
150
    ///
151
    /// ```
152
    /// use tor_llcrypto::pk::rsa::RsaIdentity;
153
    ///
154
    /// let bytes = b"xyzzyxyzzyxyzzyxyzzy";
155
    /// let id = RsaIdentity::from_bytes(bytes);
156
    /// assert_eq!(id.unwrap().as_bytes(), bytes);
157
    ///
158
    /// let truncated = b"xyzzy";
159
    /// let id = RsaIdentity::from_bytes(truncated);
160
    /// assert_eq!(id, None);
161
    /// ```
162
18648
    pub fn from_bytes(bytes: &[u8]) -> Option<Self> {
163
        Some(RsaIdentity {
164
18648
            id: CtByteArray::from(<[u8; RSA_ID_LEN]>::try_from(bytes).ok()?),
165
        })
166
18648
    }
167
    /// Decode an `RsaIdentity` from a hexadecimal string.
168
    ///
169
    /// The string must have no spaces, or any extra characters.
170
3189384
    pub fn from_hex(s: &str) -> Option<Self> {
171
3189384
        let mut array = [0_u8; 20];
172
3189384
        match hex::decode_to_slice(s, &mut array) {
173
4176
            Err(_) => None,
174
3185208
            Ok(()) => Some(RsaIdentity::from(array)),
175
        }
176
3189384
    }
177

            
178
    /// Return true if this `RsaIdentity` is composed entirely of zero-valued
179
    /// bytes.
180
    ///
181
    /// Such all-zero values should not be used internally, since they are not
182
    /// the ID of any valid key.  Instead, they are used in some places in the
183
    /// Tor protocols.
184
432
    pub fn is_zero(&self) -> bool {
185
        // We do a constant-time comparison to avoid side-channels.
186
432
        self.id.ct_eq(&[0; RSA_ID_LEN].into()).into()
187
432
    }
188
}
189

            
190
impl From<[u8; 20]> for RsaIdentity {
191
4292928
    fn from(id: [u8; 20]) -> RsaIdentity {
192
4292928
        RsaIdentity { id: id.into() }
193
4292928
    }
194
}
195

            
196
/// An RSA public key.
197
///
198
/// This implementation is a simple wrapper so that we can define new
199
/// methods and traits on the type.
200
#[derive(Clone, Debug, Eq, PartialEq)]
201
pub struct PublicKey(rsa::RsaPublicKey);
202

            
203
/// An RSA private key.
204
///
205
/// This is not so useful at present, since Arti currently only has
206
/// client support, and Tor clients never actually need RSA private
207
/// keys.
208
pub struct KeyPair(rsa::RsaPrivateKey);
209

            
210
impl KeyPair {
211
    /// Generate a new random RSA keypair.
212
    ///
213
    /// This is hardcoded to generate a 1024-bit keypair, since this only exists to support the RSA
214
    /// keys that we require for backwards compatibility (which are all 1024 bit), and we don't
215
    /// anticipate adding anything new that uses RSA in the future.
216
4
    pub fn generate<R: rand_core::RngCore + rand_core::CryptoRng>(
217
4
        csprng: &mut R,
218
4
    ) -> Result<Self, tor_error::Bug> {
219
        // It's maybe a bit strange to return tor_error::Bug here, but I think it makes sense: The
220
        // only way this call can fail is if we choose invalid values for the size and exponent,
221
        // and those values are fixed. Ideally the `rsa` crate would give us a way to check that at
222
        // compile time and thus have a infallible call, but they don't. I don't think it's
223
        // reasonable to bubble up the underlying rsa::Error type, because I don't expect a caller
224
        // to be able to figure out what to do with a error that we expect to never happen.
225
        // Returning tor_error::Bug clearly indicates that the caller won't be able to do anything
226
        // in particular about this error, without hiding a panic anywhere unexpected.
227
        Ok(Self(
228
4
            rsa::RsaPrivateKey::new(&mut RngCompat::new(csprng), 1024).map_err(|_| {
229
                tor_error::internal!("Generating RSA key failed, despite fixed exponent and size")
230
            })?,
231
        ))
232
4
    }
233
    /// Return the public component of this key.
234
576
    pub fn to_public_key(&self) -> PublicKey {
235
576
        PublicKey(self.0.to_public_key())
236
576
    }
237
    /// Construct a PrivateKey from DER pkcs1 encoding.
238
72
    pub fn from_der(der: &[u8]) -> Option<Self> {
239
72
        Some(KeyPair(rsa::RsaPrivateKey::from_pkcs1_der(der).ok()?))
240
72
    }
241
    /// Return a reference to the underlying key type.
242
72
    pub fn as_key(&self) -> &rsa::RsaPrivateKey {
243
72
        &self.0
244
72
    }
245
    /// Sign a message using this keypair.
246
    ///
247
    /// This uses PKCS#1 v1.5 padding and takes a raw bytes, rather than doing the hashing
248
    /// internally. This is because we use PKCS padding without specifying the hash OID, which is a
249
    /// slightly unusual setup that is understandably not supported by the rsa crate.
250
72
    pub fn sign(&self, message: &[u8]) -> Result<Vec<u8>, rsa::Error> {
251
72
        self.0.sign(rsa::Pkcs1v15Sign::new_unprefixed(), message)
252
72
    }
253
}
254
impl PublicKey {
255
    /// Return true iff the exponent for this key is the same
256
    /// number as 'e'.
257
11092
    pub fn exponent_is(&self, e: u32) -> bool {
258
        use rsa::traits::PublicKeyParts;
259
11092
        *self.0.e() == rsa::BigUint::new(vec![e])
260
11092
    }
261
    /// Return the number of bits in the modulus for this key.
262
11236
    pub fn bits(&self) -> usize {
263
        use rsa::traits::PublicKeyParts;
264
11236
        self.0.n().bits()
265
11236
    }
266
    /// Try to check a signature (as used in Tor.)  The signed hash
267
    /// should be in 'hashed', and the alleged signature in 'sig'.
268
    ///
269
    /// Tor uses RSA-PKCSv1 signatures, with hash algorithm OIDs
270
    /// omitted.
271
8568
    pub fn verify(&self, hashed: &[u8], sig: &[u8]) -> Result<(), signature::Error> {
272
8568
        let padding = rsa::pkcs1v15::Pkcs1v15Sign::new_unprefixed();
273
8568
        self.0
274
8568
            .verify(padding, hashed, sig)
275
8572
            .map_err(|_| signature::Error::new())
276
8568
    }
277
    /// Decode an alleged DER byte string into a PublicKey.
278
    ///
279
    /// Return None  if the DER string does not have a valid PublicKey.
280
    ///
281
    /// (This function expects an RsaPublicKey, as used by Tor.  It
282
    /// does not expect or accept a PublicKeyInfo.)
283
13180
    pub fn from_der(der: &[u8]) -> Option<Self> {
284
13180
        Some(PublicKey(rsa::RsaPublicKey::from_pkcs1_der(der).ok()?))
285
13180
    }
286
    /// Encode this public key into the DER format as used by Tor.
287
    ///
288
    /// The result is an RsaPublicKey, not a PublicKeyInfo.
289
10444
    pub fn to_der(&self) -> Vec<u8> {
290
        use der_parser::ber::BerObject;
291
        use rsa::traits::PublicKeyParts;
292

            
293
10444
        let mut n = self.0.n().to_bytes_be();
294
        // prepend 0 if high bit is 1 to ensure correct signed encoding
295
10444
        if n[0] & 0b10000000 != 0 {
296
10444
            n.insert(0, 0_u8);
297
10444
        }
298
10444
        let n = BerObject::from_int_slice(&n);
299

            
300
10444
        let mut e = self.0.e().to_bytes_be();
301
        // prepend 0 if high bit is 1 to ensure correct signed encoding
302
10444
        if e[0] & 0b10000000 != 0 {
303
            e.insert(0, 0_u8);
304
10444
        }
305
10444
        let e = BerObject::from_int_slice(&e);
306

            
307
10444
        let asn1 = BerObject::from_seq(vec![n, e]);
308
10444
        asn1.to_vec().expect("RSA key not encodable as DER")
309
10444
    }
310

            
311
    /// Compute the RsaIdentity for this public key.
312
10008
    pub fn to_rsa_identity(&self) -> RsaIdentity {
313
        use crate::d::Sha1;
314
        use digest::Digest;
315
10008
        let id: [u8; RSA_ID_LEN] = Sha1::digest(self.to_der()).into();
316
10008
        RsaIdentity { id: id.into() }
317
10008
    }
318

            
319
    /// Return a reference to the underlying key type.
320
72
    pub fn as_key(&self) -> &rsa::RsaPublicKey {
321
72
        &self.0
322
72
    }
323
}
324

            
325
impl<'a> From<&'a KeyPair> for PublicKey {
326
72
    fn from(value: &'a KeyPair) -> Self {
327
72
        PublicKey(value.to_public_key().0)
328
72
    }
329
}
330

            
331
impl From<rsa::RsaPrivateKey> for KeyPair {
332
    fn from(value: rsa::RsaPrivateKey) -> Self {
333
        Self(value)
334
    }
335
}
336

            
337
impl From<rsa::RsaPublicKey> for PublicKey {
338
    fn from(value: rsa::RsaPublicKey) -> Self {
339
        Self(value)
340
    }
341
}
342

            
343
/// An RSA signature plus all the information needed to validate it.
344
pub struct ValidatableRsaSignature {
345
    /// The key that allegedly signed this signature
346
    key: PublicKey,
347
    /// The signature in question
348
    sig: Vec<u8>,
349
    /// The value we expect to find that the signature is a signature of.
350
    expected_hash: Vec<u8>,
351
}
352

            
353
impl ValidatableRsaSignature {
354
    /// Construct a new ValidatableRsaSignature.
355
9360
    pub fn new(key: &PublicKey, sig: &[u8], expected_hash: &[u8]) -> Self {
356
9360
        ValidatableRsaSignature {
357
9360
            key: key.clone(),
358
9360
            sig: sig.into(),
359
9360
            expected_hash: expected_hash.into(),
360
9360
        }
361
9360
    }
362
}
363

            
364
impl super::ValidatableSignature for ValidatableRsaSignature {
365
6192
    fn is_valid(&self) -> bool {
366
6192
        self.key
367
6192
            .verify(&self.expected_hash[..], &self.sig[..])
368
6192
            .is_ok()
369
6192
    }
370
}