1
#![cfg_attr(docsrs, feature(doc_cfg))]
2
#![doc = include_str!("../README.md")]
3
// @@ begin lint list maintained by maint/add_warning @@
4
#![allow(renamed_and_removed_lints)] // @@REMOVE_WHEN(ci_arti_stable)
5
#![allow(unknown_lints)] // @@REMOVE_WHEN(ci_arti_nightly)
6
#![warn(missing_docs)]
7
#![warn(noop_method_call)]
8
#![warn(unreachable_pub)]
9
#![warn(clippy::all)]
10
#![deny(clippy::await_holding_lock)]
11
#![deny(clippy::cargo_common_metadata)]
12
#![deny(clippy::cast_lossless)]
13
#![deny(clippy::checked_conversions)]
14
#![allow(clippy::cognitive_complexity)] // See arti#2556
15
#![deny(clippy::debug_assert_with_mut_call)]
16
#![deny(clippy::exhaustive_enums)]
17
#![deny(clippy::exhaustive_structs)]
18
#![deny(clippy::expl_impl_clone_on_copy)]
19
#![deny(clippy::fallible_impl_from)]
20
#![deny(clippy::implicit_clone)]
21
#![deny(clippy::large_stack_arrays)]
22
#![warn(clippy::manual_ok_or)]
23
#![deny(clippy::missing_docs_in_private_items)]
24
#![warn(clippy::needless_borrow)]
25
#![warn(clippy::needless_pass_by_value)]
26
#![warn(clippy::option_option)]
27
#![deny(clippy::print_stderr)]
28
#![deny(clippy::print_stdout)]
29
#![warn(clippy::rc_buffer)]
30
#![deny(clippy::ref_option_ref)]
31
#![warn(clippy::semicolon_if_nothing_returned)]
32
#![warn(clippy::trait_duplication_in_bounds)]
33
#![deny(clippy::unchecked_time_subtraction)]
34
#![deny(clippy::unnecessary_wraps)]
35
#![warn(clippy::unseparated_literal_suffix)]
36
#![deny(clippy::unwrap_used)]
37
#![deny(clippy::mod_module_files)]
38
#![allow(clippy::let_unit_value)] // This can reasonably be done for explicitness
39
#![allow(clippy::uninlined_format_args)]
40
#![allow(clippy::significant_drop_in_scrutinee)] // arti/-/merge_requests/588/#note_2812945
41
#![allow(clippy::result_large_err)] // temporary workaround for arti#587
42
#![allow(clippy::needless_raw_string_hashes)] // complained-about code is fine, often best
43
#![allow(clippy::needless_lifetimes)] // See arti#1765
44
#![allow(mismatched_lifetime_syntaxes)] // temporary workaround for arti#2060
45
#![allow(clippy::collapsible_if)] // See arti#2342
46
#![deny(clippy::unused_async)]
47
#![deny(clippy::string_slice)] // See arti#2571
48
#![allow(recursion_depth_exceeding_limit)] // arti#2715, rust/issues/159228
49
//! <!-- @@ end lint list maintained by maint/add_warning @@ -->
50

            
51
// TODO #1645 (either remove this, or decide to have it everywhere)
52
#![cfg_attr(
53
    not(all(feature = "full", feature = "experimental")),
54
    allow(unused, unreachable_pub)
55
)]
56

            
57
#[cfg(feature = "bench")]
58
pub mod bench_utils;
59
pub mod channel;
60
pub mod circuit;
61
pub mod client;
62
pub(crate) mod conflux;
63
mod congestion;
64
mod crypto;
65
pub mod memquota;
66
pub mod peer;
67
pub mod stream;
68
pub(crate) mod streammap;
69
pub(crate) mod tunnel;
70
mod util;
71

            
72
#[cfg(feature = "relay")]
73
pub mod relay;
74
#[cfg(feature = "relay")]
75
pub use relay::channel::{RelayChannelAuthMaterial, RelayChannelBuilder};
76

            
77
pub use util::err::{Error, ResolveError};
78
pub use util::skew::ClockSkew;
79

            
80
pub use channel::params::ChannelPaddingInstructions;
81
pub use client::{ClientTunnel, HopLocation, TargetHop, channel::ClientChannelBuilder};
82
pub use congestion::params as ccparams;
83
pub use crypto::cell::{HopNum, HopNumDisplay};
84
pub use stream::flow_ctrl::params::{CellCount, FlowCtrlParameters};
85
#[cfg(feature = "send-control-msg")]
86
pub use {
87
    crate::client::Conversation,
88
    crate::client::msghandler::{MsgHandler, UserMsgHandler},
89
    crate::client::reactor::MetaCellDisposition,
90
};
91

            
92
/// A Result type for this crate.
93
pub type Result<T> = std::result::Result<T, Error>;
94

            
95
use std::fmt::Debug;
96
use tor_memquota::{
97
    HasMemoryCost,
98
    mq_queue::{self, ChannelSpec as _},
99
};
100
use tor_rtcompat::DynTimeProvider;
101

            
102
#[doc(hidden)]
103
pub use {derive_deftly, tor_memquota};
104

            
105
/// Timestamp object that we update whenever we get incoming traffic.
106
///
107
/// Used to implement [`time_since_last_incoming_traffic`]
108
static LAST_INCOMING_TRAFFIC: util::ts::AtomicOptTimestamp = util::ts::AtomicOptTimestamp::new();
109

            
110
/// Called whenever we receive incoming traffic.
111
///
112
/// Used to implement [`time_since_last_incoming_traffic`]
113
#[inline]
114
642
pub(crate) fn note_incoming_traffic() {
115
642
    LAST_INCOMING_TRAFFIC.update();
116
642
}
117

            
118
/// Return the amount of time since we last received "incoming traffic".
119
///
120
/// This is a global counter, and is subject to interference from
121
/// other users of the `tor_proto`.  Its only permissible use is for
122
/// checking how recently we have been definitely able to receive
123
/// incoming traffic.
124
///
125
/// When enabled, this timestamp is updated whenever we receive a valid
126
/// cell, and whenever we complete a channel handshake.
127
///
128
/// Returns `None` if we never received "incoming traffic".
129
649587
pub fn time_since_last_incoming_traffic() -> Option<std::time::Duration> {
130
649587
    LAST_INCOMING_TRAFFIC.time_since_update().map(Into::into)
131
649587
}
132

            
133
/// Make an MPSC queue, of any type, that participates in memquota, but a fake one for testing
134
#[cfg(any(test, feature = "testing"))] // Used by Channel::new_fake which is also feature=testing
135
2440
pub(crate) fn fake_mpsc<T: HasMemoryCost + Debug + Send>(
136
2440
    buffer: usize,
137
2440
) -> (
138
2440
    mq_queue::Sender<T, mq_queue::MpscSpec>,
139
2440
    mq_queue::Receiver<T, mq_queue::MpscSpec>,
140
2440
) {
141
2440
    mq_queue::MpscSpec::new(buffer)
142
2440
        .new_mq(
143
            // The fake Account doesn't care about the data ages, so this will do.
144
            //
145
            // Thiw would be wrong to use generally in tests, where we might want to mock time,
146
            // since we end up, here with totally *different* mocked time.
147
            // But it's OK here, and saves passing a runtime parameter into this function.
148
2440
            DynTimeProvider::new(tor_rtmock::MockRuntime::default()),
149
2440
            &tor_memquota::Account::new_noop(),
150
        )
151
2440
        .expect("create fake mpsc")
152
2440
}
153

            
154
/// Return a list of the protocols [supported](tor_protover::doc_supported)
155
/// by this crate, running as a client.
156
10916
pub fn supported_client_protocols() -> tor_protover::Protocols {
157
    use tor_protover::named::*;
158
    // WARNING: REMOVING ELEMENTS FROM THIS LIST CAN BE DANGEROUS!
159
    // SEE [`tor_protover::doc_changing`]
160
10916
    let protocols = vec![
161
        LINK_V4,
162
        LINK_V5,
163
        LINKAUTH_ED25519_SHA256_EXPORTER,
164
        FLOWCTRL_AUTH_SENDME,
165
        FLOWCTRL_CC,
166
        RELAY_NTOR,
167
        RELAY_EXTEND_IPv6,
168
        RELAY_NTORV3,
169
        RELAY_NEGOTIATE_SUBPROTO,
170
        RELAY_CRYPT_CGO,
171
    ];
172

            
173
10916
    protocols.into_iter().collect()
174
10916
}
175

            
176
#[cfg(test)]
177
mod test {
178
    // @@ begin test lint list maintained by maint/add_warning @@
179
    #![allow(clippy::bool_assert_comparison)]
180
    #![allow(clippy::clone_on_copy)]
181
    #![allow(clippy::dbg_macro)]
182
    #![allow(clippy::mixed_attributes_style)]
183
    #![allow(clippy::print_stderr)]
184
    #![allow(clippy::print_stdout)]
185
    #![allow(clippy::single_char_pattern)]
186
    #![allow(clippy::unwrap_used)]
187
    #![allow(clippy::unchecked_time_subtraction)]
188
    #![allow(clippy::useless_vec)]
189
    #![allow(clippy::needless_pass_by_value)]
190
    #![allow(clippy::string_slice)] // See arti#2571
191
    //! <!-- @@ end test lint list maintained by maint/add_warning @@ -->
192

            
193
    use super::*;
194

            
195
    #[test]
196
    fn protocols() {
197
        let pr = supported_client_protocols();
198
        let expected = "FlowCtrl=1-2 Link=4-5 LinkAuth=3 Relay=2-6"
199
            .parse()
200
            .unwrap();
201
        assert_eq!(pr, expected);
202
    }
203
}