1 // Copyright 2021 Ian Jackson and contributors to Hippotat
2 // SPDX-License-Identifier: GPL-3.0-or-later
3 // There is NO WARRANTY.
9 pub ic: Arc<InstanceConfig>,
10 pub web: mpsc::Sender<WebRequest>,
11 pub route: mpsc::Sender<RoutedPacket>,
14 pub async fn run(global: Arc<Global>,
15 ic: Arc<InstanceConfig>,
16 mut web: mpsc::Receiver<WebRequest>,
17 mut routed: mpsc::Receiver<RoutedPacket>)
21 reply_to: oneshot::Sender<WebResponse>,
25 struct OutstandingInner {
27 target_requests_outstanding: u32,
30 let mut outstanding: VecDeque<Outstanding> = default();
31 let mut downbound: PacketQueue<RoutedPacketData> = default();
33 let try_send_response = |
34 reply_to: oneshot::Sender<WebResponse>,
37 reply_to.send(response)
38 .unwrap_or_else(|_: WebResponse| () /* oh dear */ /* xxx trace? */);
42 let eff_max_batch_down = outstanding
44 .map(|o| o.oi.max_batch_down)
46 .unwrap_or(ic.max_batch_down)
50 if ! downbound.is_empty() {
51 outstanding.pop_front()
52 } else if let Some((i,_)) = outstanding.iter().enumerate().find({
53 |(_,o)| outstanding.len() > o.oi.target_requests_outstanding.sat()
54 // xxx need timeout-based return too
56 Some(outstanding.remove(i).unwrap())
61 let mut build: FrameQueueBuf = default();
64 let next = if let Some(n) = downbound.peek_front() { n }
66 // Don't add 1 for the ESC since we will strip one
67 if build.len() + next.len() >= eff_max_batch_down { break }
68 build.esc_push(downbound.pop_front().unwrap());
70 if ! build.is_empty() {
75 let response = WebResponse {
80 try_send_response(req.reply_to, response);
83 let max = usize::saturating_mul(
84 ic.max_requests_outstanding.sat(),
86 ).saturating_add(1 /* one boundary SLIP_ESC which we'll trim */);
88 while downbound.total_len() > max {
89 let _ = downbound.pop_front();
95 data = routed.recv() =>
97 let data = data.ok_or_else(|| anyhow!("routers shut down!"))?;
98 downbound.push_back(data.data);
104 initial, initial_remaining, length_hint, mut body,
106 reply_to, conn, mut warnings, may_route,
107 } = req.ok_or_else(|| anyhow!("webservers all shut down!"))?;
111 let initial_used = initial.len() - initial_remaining;
113 let whole_request = read_limited_bytes(
114 ic.max_batch_up.sat(),
118 ).await.context("read request body")?;
120 let (meta, mut comps) =
121 multipart::ComponentIterator::resume_mid_component(
122 &whole_request[initial_used..],
124 ).context("resume parsing body, after auth checks")?;
126 let mut meta = MetadataFieldIterator::new(&meta);
129 { $v:ident, ( $( $badcmp:tt )? ), $ret:expr,
130 let $server:ident, $client:ident $($code:tt)*
134 let $client $($code)*
136 if $client $badcmp $server {
137 throw!(anyhow!("mismatch: client={:?} {} server={:?}",
138 $client, stringify!($badcmp), $server));
142 })().context(stringify!($v))?;
147 target_requests_outstanding, ( != ), client,
148 let server, client: u32 = meta.need_parse()?;
151 http_timeout, ( > ), client,
152 let server, client = Duration::from_secs(meta.need_parse()?);
156 let server, client: u32 = meta.parse()?.unwrap_or(server);
159 max_batch_down, (), min(client, server),
160 let server, client: u32 = meta.parse()?.unwrap_or(server);
163 max_batch_up, ( > ), client,
164 let server, client = meta.parse()?.unwrap_or(server);
166 let _ = max_batch_up; // we don't use this further
168 while let Some(comp) = comps.next(&mut warnings, PartName::d)? {
169 if comp.name != PartName::d {
170 warnings.add(&format_args!("unexpected part {:?}", comp.name))?;
172 slip::processn(Mime2Slip, mtu, comp.payload, |header| {
173 let saddr = ip_packet_addr::<false>(header)?;
174 if saddr != ic.link.client.0 { throw!(PE::Src(saddr)) }
175 let daddr = ip_packet_addr::<true>(header)?;
177 }, |(daddr,packet)| route_packet(
178 &global, &conn, Some(&ic.link.client), daddr,
179 packet, may_route.clone(),
181 |e| Ok::<_,SlipFramesError<_>>({ warnings.add(&e)?; })
185 let deadline = Instant::now() + http_timeout;
187 let oi = OutstandingInner {
188 target_requests_outstanding,
194 Ok(oi) => outstanding.push_back(Outstanding { reply_to, oi }),
196 try_send_response(reply_to, WebResponse {
205 //Err(anyhow!("xxx"))