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_bytes(&buf).expect("base64 is always valid HeaderValue");
22    header.set_sensitive(true);
23    header
24}
25
26// xor-shift
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    use std::num::Wrapping;
33
34    thread_local! {
35        static RNG: Cell<Wrapping<u64>> = Cell::new(Wrapping(seed()));
36    }
37
38    fn seed() -> u64 {
39        let seed = RandomState::new();
40
41        let mut out = 0;
42        let mut cnt = 0;
43        while out == 0 {
44            cnt += 1;
45            let mut hasher = seed.build_hasher();
46            hasher.write_usize(cnt);
47            out = hasher.finish();
48        }
49        out
50    }
51
52    RNG.with(|rng| {
53        let mut n = rng.get();
54        debug_assert_ne!(n.0, 0);
55        n ^= n >> 12;
56        n ^= n << 25;
57        n ^= n >> 27;
58        rng.set(n);
59        n.0.wrapping_mul(0x2545_f491_4f6c_dd1d)
60    })
61}
62
63pub(crate) fn replace_headers(dst: &mut HeaderMap, src: HeaderMap) {
64    // IntoIter of HeaderMap yields (Option<HeaderName>, HeaderValue).
65    // The first time a name is yielded, it will be Some(name), and if
66    // there are more values with the same name, the next yield will be
67    // None.
68
69    let mut prev_entry: Option<OccupiedEntry<_>> = None;
70    for (key, value) in src {
71        match key {
72            Some(key) => match dst.entry(key) {
73                Entry::Occupied(mut e) => {
74                    e.insert(value);
75                    prev_entry = Some(e);
76                }
77                Entry::Vacant(e) => {
78                    let e = e.insert_entry(value);
79                    prev_entry = Some(e);
80                }
81            },
82            None => match prev_entry {
83                Some(ref mut entry) => {
84                    entry.append(value);
85                }
86                None => unreachable!("HeaderMap::into_iter yielded None first"),
87            },
88        }
89    }
90}
91
92#[cfg(feature = "cookies")]
93#[cfg(not(target_arch = "wasm32"))]
94pub(crate) fn add_cookie_header(
95    headers: &mut HeaderMap,
96    cookie_store: &dyn crate::cookie::CookieStore,
97    url: &url::Url,
98) {
99    if let Some(header) = cookie_store.cookies(url) {
100        headers.insert(crate::header::COOKIE, header);
101    }
102}
103
104pub(crate) struct Escape<'a>(&'a [u8]);
105
106#[cfg(not(target_arch = "wasm32"))]
107impl<'a> Escape<'a> {
108    pub(crate) fn new(bytes: &'a [u8]) -> Self {
109        Escape(bytes)
110    }
111}
112
113impl fmt::Debug for Escape<'_> {
114    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
115        write!(f, "b\"{}\"", self)?;
116        Ok(())
117    }
118}
119
120impl fmt::Display for Escape<'_> {
121    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
122        for &c in self.0 {
123            // https://doc.rust-lang.org/reference.html#byte-escapes
124            if c == b'\n' {
125                write!(f, "\\n")?;
126            } else if c == b'\r' {
127                write!(f, "\\r")?;
128            } else if c == b'\t' {
129                write!(f, "\\t")?;
130            } else if c == b'\\' || c == b'"' {
131                write!(f, "\\{}", c as char)?;
132            } else if c == b'\0' {
133                write!(f, "\\0")?;
134            // ASCII printable
135            } else if c >= 0x20 && c < 0x7f {
136                write!(f, "{}", c as char)?;
137            } else {
138                write!(f, "\\x{c:02x}")?;
139            }
140        }
141        Ok(())
142    }
143}
reqwest/util.rs

reqwest/
util.rs
