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_cost::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
92488
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
64
92488
        write!(f, "${}", hex::encode(&self.id.as_ref()[..]))
65
92488
    }
66
}
67
impl fmt::Debug for RsaIdentity {
68
49544
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
69
49544
        write!(f, "RsaIdentity {{ {} }}", self)
70
49544
    }
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
18308
    fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
87
18308
    where
88
18308
        S: serde::Serializer,
89
    {
90
18308
        if serializer.is_human_readable() {
91
18306
            serializer.serialize_str(&hex::encode(&self.id.as_ref()[..]))
92
        } else {
93
2
            serializer.serialize_bytes(&self.id.as_ref()[..])
94
        }
95
18308
    }
96
}
97

            
98
impl<'de> serde::Deserialize<'de> for RsaIdentity {
99
580
    fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
100
580
    where
101
580
        D: serde::Deserializer<'de>,
102
    {
103
580
        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
576
                fn visit_str<E>(self, s: &str) -> Result<Self::Value, E>
112
576
                where
113
576
                    E: serde::de::Error,
114
                {
115
576
                    RsaIdentity::from_hex(s)
116
576
                        .ok_or_else(|| E::custom("wrong encoding for RSA identity"))
117
576
                }
118
            }
119

            
120
576
            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
580
    }
140
}
141

            
142
impl RsaIdentity {
143
    /// Expose an RsaIdentity as a slice of bytes.
144
911416
    pub fn as_bytes(&self) -> &[u8] {
145
911416
        &self.id.as_ref()[..]
146
911416
    }
147

            
148
    /// Expose an RsaIdentity as an uppercase hexadecimal string.
149
1144
    pub fn as_hex_upper(&self) -> String {
150
1144
        hex::encode_upper(self.as_bytes())
151
1144
    }
152

            
153
    /// Construct an RsaIdentity from a slice of bytes.
154
    ///
155
    /// Returns None if the input is not of the correct length.
156
    ///
157
    /// ```
158
    /// use tor_llcrypto::pk::rsa::RsaIdentity;
159
    ///
160
    /// let bytes = b"xyzzyxyzzyxyzzyxyzzy";
161
    /// let id = RsaIdentity::from_bytes(bytes);
162
    /// assert_eq!(id.unwrap().as_bytes(), bytes);
163
    ///
164
    /// let truncated = b"xyzzy";
165
    /// let id = RsaIdentity::from_bytes(truncated);
166
    /// assert_eq!(id, None);
167
    /// ```
168
22792
    pub fn from_bytes(bytes: &[u8]) -> Option<Self> {
169
        Some(RsaIdentity {
170
22792
            id: CtByteArray::from(<[u8; RSA_ID_LEN]>::try_from(bytes).ok()?),
171
        })
172
22792
    }
173
    /// Decode an `RsaIdentity` from a hexadecimal string.
174
    ///
175
    /// The string must have no spaces, or any extra characters.
176
4636808
    pub fn from_hex(s: &str) -> Option<Self> {
177
4636808
        let mut array = [0_u8; 20];
178
4636808
        match hex::decode_to_slice(s, &mut array) {
179
5104
            Err(_) => None,
180
4631704
            Ok(()) => Some(RsaIdentity::from(array)),
181
        }
182
4636808
    }
183

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

            
196
impl From<[u8; 20]> for RsaIdentity {
197
5985584
    fn from(id: [u8; 20]) -> RsaIdentity {
198
5985584
        RsaIdentity { id: id.into() }
199
5985584
    }
200
}
201

            
202
/// An RSA public key.
203
///
204
/// This implementation is a simple wrapper so that we can define new
205
/// methods and traits on the type.
206
#[derive(Clone, Debug, Eq, PartialEq)]
207
pub struct PublicKey(rsa::RsaPublicKey);
208

            
209
/// An RSA private key.
210
///
211
/// This is not so useful at present, since Arti currently only has
212
/// client support, and Tor clients never actually need RSA private
213
/// keys.
214
pub struct KeyPair(rsa::RsaPrivateKey);
215

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

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

            
306
19540
        let mut e = self.0.e().to_bytes_be();
307
        // prepend 0 if high bit is 1 to ensure correct signed encoding
308
19540
        if e[0] & 0b10000000 != 0 {
309
            e.insert(0, 0_u8);
310
19540
        }
311
19540
        let e = BerObject::from_int_slice(&e);
312

            
313
19540
        let asn1 = BerObject::from_seq(vec![n, e]);
314
19540
        asn1.to_vec().expect("RSA key not encodable as DER")
315
19540
    }
316

            
317
    /// Compute the RsaIdentity for this public key.
318
18656
    pub fn to_rsa_identity(&self) -> RsaIdentity {
319
        use crate::d::Sha1;
320
        use digest::Digest;
321
18656
        let id: [u8; RSA_ID_LEN] = Sha1::digest(self.to_der()).into();
322
18656
        RsaIdentity { id: id.into() }
323
18656
    }
324

            
325
    /// Return a reference to the underlying key type.
326
88
    pub fn as_key(&self) -> &rsa::RsaPublicKey {
327
88
        &self.0
328
88
    }
329
}
330

            
331
impl<'a> From<&'a KeyPair> for PublicKey {
332
88
    fn from(value: &'a KeyPair) -> Self {
333
88
        PublicKey(value.to_public_key().0)
334
88
    }
335
}
336

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

            
343
impl From<rsa::RsaPublicKey> for PublicKey {
344
    fn from(value: rsa::RsaPublicKey) -> Self {
345
        Self(value)
346
    }
347
}
348

            
349
/// An RSA signature plus all the information needed to validate it.
350
pub struct ValidatableRsaSignature {
351
    /// The key that allegedly signed this signature
352
    key: PublicKey,
353
    /// The signature in question
354
    sig: Vec<u8>,
355
    /// The value we expect to find that the signature is a signature of.
356
    expected_hash: Vec<u8>,
357
}
358

            
359
impl ValidatableRsaSignature {
360
    /// Construct a new ValidatableRsaSignature.
361
11440
    pub fn new(key: &PublicKey, sig: &[u8], expected_hash: &[u8]) -> Self {
362
11440
        ValidatableRsaSignature {
363
11440
            key: key.clone(),
364
11440
            sig: sig.into(),
365
11440
            expected_hash: expected_hash.into(),
366
11440
        }
367
11440
    }
368
}
369

            
370
impl super::ValidatableSignature for ValidatableRsaSignature {
371
7568
    fn is_valid(&self) -> bool {
372
7568
        self.key
373
7568
            .verify(&self.expected_hash[..], &self.sig[..])
374
7568
            .is_ok()
375
7568
    }
376
}