lifegpc/pixiv_downloader
1
0
Fork 0
mirror of https://github.com/lifegpc/pixiv_downloader.git synced 2026-07-08 01:32:41 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix create_download_tasks_simple may panic

Browse Source
This commit is contained in:
lifegpc
2022-06-08 13:34:39 +00:00
committed by GitHub
parent 3c204aa2fb
commit 9912210fe2
1 changed files with 23 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

42
src/downloader/tasks.rs
View File

@@ -26,28 +26,32 @@ pub async fn create_download_tasks_simple<T: Seek + Write + Send + Sync>(d: Arc<
result = d.client.get(d.url.deref().clone(), d.headers.deref().clone()).try_err(gettext("Failed to get url."))?;
status = result.status();
} else if status == 206 {
let range = ContentRange::parse_bytes(result.headers()["Content-Range"].as_bytes());
let need_reget = match range {
ContentRange::Bytes(b) => {
if file_size != 0 && b.complete_length != file_size {
true
} else if start != b.first_byte {
true
} else {
false
let headers = result.headers();
let need_reget = if headers.contains_key("Content-Range") {
match ContentRange::parse_bytes(headers["Content-Range"].as_bytes()) {
ContentRange::Bytes(b) => {
if file_size != 0 && b.complete_length != file_size {
true
} else if start != b.first_byte {
true
} else {
false
}
}
}
ContentRange::UnboundBytes(b) => {
if start != b.first_byte {
true
} else {
false
ContentRange::UnboundBytes(b) => {
if start != b.first_byte {
true
} else {
false
}
}
ContentRange::Unknown => {
true
}
_ => { false }
}
ContentRange::Unknown => {
true
}
_ => { false }
} else {
true
};
if need_reget {
d.pd.clear()?;