FediMovies/fedimovies
1
1
Fork 0
Code Issues 3 Pull requests Projects Releases Packages Activity

Increase max download size

Browse source
This commit is contained in:
silverpill 2022-09-01 12:21:25 +00:00
parent 26ae1036de
commit 0d7325f8f6
1 changed files with 7 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
src/activitypub/fetcher/fetchers.rs
View file

@ -79,7 +79,7 @@ async fn send_request(
Ok(data) Ok(data)
} }
const FILE_MAX_SIZE: usize = 1024 * 1024 * 10; const FILE_MAX_SIZE: u64 = 1024 * 1024 * 20;
pub async fn fetch_file( pub async fn fetch_file(
url: &str, url: &str,
@ -87,8 +87,13 @@ pub async fn fetch_file(
) -> Result<(String, Option<String>), FetchError> { ) -> Result<(String, Option<String>), FetchError> {
let client = build_client()?; let client = build_client()?;
let response = client.get(url).send().await?; let response = client.get(url).send().await?;
if let Some(file_size) = response.content_length() {
if file_size > FILE_MAX_SIZE {
return Err(FetchError::OtherError("file is too large"));
};
};
let file_data = response.bytes().await?; let file_data = response.bytes().await?;
if file_data.len() > FILE_MAX_SIZE { if file_data.len() > FILE_MAX_SIZE as usize {
return Err(FetchError::OtherError("file is too large")); return Err(FetchError::OtherError("file is too large"));
}; };
let (file_name, media_type) = save_file(file_data.to_vec(), output_dir, None)?; let (file_name, media_type) = save_file(file_data.to_vec(), output_dir, None)?;