reqwest/
util.rs

1use crate::header::{Entry, HeaderMap, HeaderValue, OccupiedEntry};
2use std::fmt;
3
4pub fn basic_auth<U, P>(username: U, password: Option<P>) -> HeaderValue
5where
6    U: std::fmt::Display,
7    P: std::fmt::Display,
8{
9    use base64::prelude::BASE64_STANDARD;
10    use base64::write::EncoderWriter;
11    use std::io::Write;
12
13    let mut buf = b"Basic ".to_vec();
14    {
15        let mut encoder = EncoderWriter::new(&mut buf, &BASE64_STANDARD);
16        let _ = write!(encoder, "{username}:");
17        if let Some(password) = password {
18            let _ = write!(encoder, "{password}");
19        }
20    }
21    let mut header = HeaderValue::from_maybe_shared(bytes::Bytes::from(buf))
22        .expect("base64 is always valid HeaderValue");
23    header.set_sensitive(true);
24    header
25}
26
27#[cfg(not(target_arch = "wasm32"))]
28pub(crate) fn fast_random() -> u64 {
29    use std::cell::Cell;
30    use std::collections::hash_map::RandomState;
31    use std::hash::{BuildHasher, Hasher};
32
33    thread_local! {
34        static KEY: RandomState = RandomState::new();
35        static COUNTER: Cell<u64> = Cell::new(0);
36    }
37
38    KEY.with(|key| {
39        COUNTER.with(|ctr| {
40            let n = ctr.get().wrapping_add(1);
41            ctr.set(n);
42
43            let mut h = key.build_hasher();
44            h.write_u64(n);
45            h.finish()
46        })
47    })
48}
49
50pub(crate) fn replace_headers(dst: &mut HeaderMap, src: HeaderMap) {
51    // IntoIter of HeaderMap yields (Option<HeaderName>, HeaderValue).
52    // The first time a name is yielded, it will be Some(name), and if
53    // there are more values with the same name, the next yield will be
54    // None.
55
56    let mut prev_entry: Option<OccupiedEntry<_>> = None;
57    for (key, value) in src {
58        match key {
59            Some(key) => match dst.entry(key) {
60                Entry::Occupied(mut e) => {
61                    e.insert(value);
62                    prev_entry = Some(e);
63                }
64                Entry::Vacant(e) => {
65                    let e = e.insert_entry(value);
66                    prev_entry = Some(e);
67                }
68            },
69            None => match prev_entry {
70                Some(ref mut entry) => {
71                    entry.append(value);
72                }
73                None => unreachable!("HeaderMap::into_iter yielded None first"),
74            },
75        }
76    }
77}
78
79#[cfg(feature = "cookies")]
80#[cfg(not(target_arch = "wasm32"))]
81pub(crate) fn add_cookie_header(
82    headers: &mut HeaderMap,
83    cookie_store: &dyn crate::cookie::CookieStore,
84    url: &url::Url,
85) {
86    if let Some(header) = cookie_store.cookies(url) {
87        headers.insert(crate::header::COOKIE, header);
88    }
89}
90
91pub(crate) struct Escape<'a>(&'a [u8]);
92
93#[cfg(not(target_arch = "wasm32"))]
94impl<'a> Escape<'a> {
95    pub(crate) fn new(bytes: &'a [u8]) -> Self {
96        Escape(bytes)
97    }
98}
99
100impl fmt::Debug for Escape<'_> {
101    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
102        write!(f, "b\"{}\"", self)?;
103        Ok(())
104    }
105}
106
107impl fmt::Display for Escape<'_> {
108    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
109        for &c in self.0 {
110            // https://doc.rust-lang.org/reference.html#byte-escapes
111            if c == b'\n' {
112                write!(f, "\\n")?;
113            } else if c == b'\r' {
114                write!(f, "\\r")?;
115            } else if c == b'\t' {
116                write!(f, "\\t")?;
117            } else if c == b'\\' || c == b'"' {
118                write!(f, "\\{}", c as char)?;
119            } else if c == b'\0' {
120                write!(f, "\\0")?;
121            // ASCII printable
122            } else if c >= 0x20 && c < 0x7f {
123                write!(f, "{}", c as char)?;
124            } else {
125                write!(f, "\\x{c:02x}")?;
126            }
127        }
128        Ok(())
129    }
130}
reqwest/util.rs

reqwest/
util.rs
