net/protocols/
blob.rs

1/* This Source Code Form is subject to the terms of the Mozilla Public
2 * License, v. 2.0. If a copy of the MPL was not distributed with this
3 * file, You can obtain one at https://mozilla.org/MPL/2.0/. */
4
5use std::future::{Future, ready};
6use std::pin::Pin;
7
8use headers::{HeaderMapExt, Range};
9use http::Method;
10use log::debug;
11use net_traits::blob_url_store::{BlobURLStoreError, parse_blob_url};
12use net_traits::http_status::HttpStatus;
13use net_traits::request::Request;
14use net_traits::response::{Response, ResponseBody};
15use net_traits::{NetworkError, ResourceFetchTiming};
16use tokio::sync::mpsc::unbounded_channel;
17
18use crate::fetch::methods::{Data, DoneChannel, FetchContext};
19use crate::protocols::{ProtocolHandler, partial_content, range_not_satisfiable_error};
20
21#[derive(Default)]
22pub struct BlobProtocolHander {}
23
24impl ProtocolHandler for BlobProtocolHander {
25    fn load(
26        &self,
27        request: &mut Request,
28        done_chan: &mut DoneChannel,
29        context: &FetchContext,
30    ) -> Pin<Box<dyn Future<Output = Response> + Send>> {
31        let url = request.current_url();
32        debug!("Loading blob {}", url.as_str());
33
34        // Step 2.
35        if request.method != Method::GET {
36            return Box::pin(ready(Response::network_error(NetworkError::InvalidMethod)));
37        }
38
39        let range_header = request.headers.typed_get::<Range>();
40        let is_range_request = range_header.is_some();
41
42        let (id, origin) = match parse_blob_url(&url) {
43            Ok((id, origin)) => (id, origin),
44            Err(error) => {
45                return Box::pin(ready(Response::network_error(
46                    NetworkError::ResourceLoadError(format!("Invalid blob URL ({error})")),
47                )));
48            },
49        };
50
51        let mut response = Response::new(url, ResourceFetchTiming::new(request.timing_type()));
52        response.status = HttpStatus::default();
53
54        if is_range_request {
55            response.range_requested = true;
56            partial_content(&mut response);
57        }
58
59        let (mut done_sender, done_receiver) = unbounded_channel();
60        *done_chan = Some((done_sender.clone(), done_receiver));
61        *response.body.lock() = ResponseBody::Receiving(vec![]);
62
63        if let Err(err) = context.filemanager.fetch_file(
64            &mut done_sender,
65            context.cancellation_listener.clone(),
66            id,
67            &context.file_token,
68            origin,
69            &mut response,
70            range_header,
71        ) {
72            let _ = done_sender.send(Data::Done);
73            let err = match err {
74                BlobURLStoreError::InvalidRange => {
75                    range_not_satisfiable_error(&mut response);
76                    return Box::pin(ready(response));
77                },
78                _ => format!("{:?}", err),
79            };
80            return Box::pin(ready(Response::network_error(
81                NetworkError::BlobURLStoreError(err),
82            )));
83        };
84
85        Box::pin(ready(response))
86    }
87}
net/protocols/blob.rs

net/protocols/
blob.rs
