tor_proto/crypto/handshake.rs
1//! Circuit extension handshake for Tor.
2//!
3//! Tor circuit handshakes all implement a one-way-authenticated key
4//! exchange, where a client that knows a public "onion key" for a
5//! relay sends a "client onionskin" to extend to a relay, and receives a
6//! "relay onionskin" in response. When the handshake is successful,
7//! both the client and relay share a set of session keys, and the
8//! client knows that nobody _else_ shares those keys unless they
9//! relay's private onion key.
10//!
11//! Currently, this module implements only the "ntor" handshake used
12//! for circuits on today's Tor.
13pub(crate) mod fast;
14#[cfg(feature = "hs-common")]
15pub mod hs_ntor;
16pub(crate) mod ntor;
17pub(crate) mod ntor_v3;
18
19use std::borrow::Borrow;
20
21use crate::Result;
22//use zeroize::Zeroizing;
23use rand_core::{CryptoRng, RngCore};
24use tor_bytes::SecretBuf;
25
26/// A ClientHandshake is used to generate a client onionskin and
27/// handle a relay onionskin.
28pub(crate) trait ClientHandshake {
29 /// The type for the onion key.
30 type KeyType;
31 /// The type for the state that the client holds while waiting for a reply.
32 type StateType;
33 /// A type that is returned and used to generate session keys.x
34 type KeyGen;
35 /// Type of extra data sent from client (without forward secrecy).
36 type ClientAuxData: ?Sized;
37 /// Type of extra data returned by server (without forward secrecy).
38 type ServerAuxData;
39 /// Generate a new client onionskin for a relay with a given onion key,
40 /// including `client_aux_data` to be sent without forward secrecy.
41 ///
42 /// On success, return a state object that will be used to
43 /// complete the handshake, along with the message to send.
44 fn client1<R: RngCore + CryptoRng, M: Borrow<Self::ClientAuxData>>(
45 rng: &mut R,
46 key: &Self::KeyType,
47 client_aux_data: &M,
48 ) -> Result<(Self::StateType, Vec<u8>)>;
49 /// Handle an onionskin from a relay, and produce aux data returned
50 /// from the server, and a key generator.
51 ///
52 /// The state object must match the one that was used to make the
53 /// client onionskin that the server is replying to.
54 fn client2<T: AsRef<[u8]>>(
55 state: Self::StateType,
56 msg: T,
57 ) -> Result<(Self::ServerAuxData, Self::KeyGen)>;
58}
59
60/// Trait for an object that handles incoming auxiliary data and
61/// returns the server's auxiliary data to be included in the reply.
62///
63/// This is implemented for `FnMut(&H::ClientAuxData) -> Option<H::ServerAuxData>` automatically.
64pub(crate) trait AuxDataReply<H>
65where
66 H: ServerHandshake + ?Sized,
67{
68 /// Given a list of extensions received from a client, decide
69 /// what extensions to send in reply.
70 ///
71 /// Return None if the handshake should fail.
72 fn reply(&mut self, msg: &H::ClientAuxData) -> Option<H::ServerAuxData>;
73}
74
75impl<F, H> AuxDataReply<H> for F
76where
77 H: ServerHandshake + ?Sized,
78 F: FnMut(&H::ClientAuxData) -> Option<H::ServerAuxData>,
79{
80 fn reply(&mut self, msg: &H::ClientAuxData) -> Option<H::ServerAuxData> {
81 self(msg)
82 }
83}
84
85/// A ServerHandshake is used to handle a client onionskin and generate a
86/// server onionskin.
87pub(crate) trait ServerHandshake {
88 /// The type for the onion key. This is a private key type.
89 type KeyType;
90 /// The returned key generator type.
91 type KeyGen;
92 /// Type of extra data sent from client (without forward secrecy).
93 type ClientAuxData: ?Sized;
94 /// Type of extra data returned by server (without forward secrecy).
95 type ServerAuxData;
96
97 /// Perform the server handshake. Take as input a strong PRNG in `rng`, a
98 /// function for processing requested extensions, a slice of all our private
99 /// onion keys, and the client's message.
100 ///
101 /// On success, return a key generator and a server handshake message
102 /// to send in reply.
103 #[allow(dead_code)] // TODO #1383 ????
104 fn server<R: RngCore + CryptoRng, REPLY: AuxDataReply<Self>, T: AsRef<[u8]>>(
105 rng: &mut R,
106 reply_fn: &mut REPLY,
107 key: &[Self::KeyType],
108 msg: T,
109 ) -> RelayHandshakeResult<(Self::KeyGen, Vec<u8>)>;
110}
111
112/// A KeyGenerator is returned by a handshake, and used to generate
113/// session keys for the protocol.
114///
115/// Typically, it wraps a KDF function, and some seed key material.
116///
117/// It can only be used once.
118#[allow(unreachable_pub)] // This is only exported depending on enabled features.
119pub trait KeyGenerator {
120 /// Consume the key
121 fn expand(self, keylen: usize) -> Result<SecretBuf>;
122}
123
124/// Generates keys based on the KDF-TOR function.
125///
126/// This is deprecated and shouldn't be used for new keys.
127pub(crate) struct TapKeyGenerator {
128 /// Seed for the TAP KDF.
129 seed: SecretBuf,
130}
131
132impl TapKeyGenerator {
133 /// Create a key generator based on a provided seed
134 pub(crate) fn new(seed: SecretBuf) -> Self {
135 TapKeyGenerator { seed }
136 }
137}
138
139impl KeyGenerator for TapKeyGenerator {
140 fn expand(self, keylen: usize) -> Result<SecretBuf> {
141 use crate::crypto::ll::kdf::{Kdf, LegacyKdf};
142 LegacyKdf::new(1).derive(&self.seed[..], keylen)
143 }
144}
145
146/// Generates keys based on SHAKE-256.
147pub(crate) struct ShakeKeyGenerator {
148 /// Seed for the key generator
149 seed: SecretBuf,
150}
151
152impl ShakeKeyGenerator {
153 /// Create a key generator based on a provided seed
154 #[allow(dead_code)] // We'll construct these for v3 onion services
155 pub(crate) fn new(seed: SecretBuf) -> Self {
156 ShakeKeyGenerator { seed }
157 }
158}
159
160impl KeyGenerator for ShakeKeyGenerator {
161 fn expand(self, keylen: usize) -> Result<SecretBuf> {
162 use crate::crypto::ll::kdf::{Kdf, ShakeKdf};
163 ShakeKdf::new().derive(&self.seed[..], keylen)
164 }
165}
166
167/// An error produced by a Relay's attempt to handle a client's onion handshake.
168#[derive(Clone, Debug, thiserror::Error)]
169pub(crate) enum RelayHandshakeError {
170 /// An error in parsing a handshake message.
171 #[error("Problem decoding onion handshake")]
172 Fmt(#[from] tor_bytes::Error),
173 /// The client asked for a key we didn't have.
174 #[error("Client asked for a key or ID that we don't have")]
175 MissingKey,
176 /// The client did something wrong with their handshake or cryptography.
177 #[error("Bad handshake from client")]
178 BadClientHandshake,
179 /// An internal error.
180 #[error("Internal error")]
181 Internal(#[from] tor_error::Bug),
182}
183
184/// Type alias for results from a relay's attempt to handle a client's onion
185/// handshake.
186pub(crate) type RelayHandshakeResult<T> = std::result::Result<T, RelayHandshakeError>;