Compare commits
3 commits
3cb95a9f04
...
f7431ca6a4
Author | SHA1 | Date | |
---|---|---|---|
f7431ca6a4 | |||
67aa009746 | |||
8a09da764e |
1 changed files with 56 additions and 54 deletions
|
@ -86,7 +86,15 @@ pub fn decrypt(
|
|||
ContentType::Image => DecryptedData::Image(blob, container.len()),
|
||||
ContentType::Audio => DecryptedData::Audio(blob),
|
||||
ContentType::Video => DecryptedData::Video(blob),
|
||||
ContentType::ZipArchive => {
|
||||
ContentType::ZipArchive => handle_zip_archive(blob, container),
|
||||
ContentType::Gzip => handle_gzip(blob, container),
|
||||
ContentType::Unknown => DecryptedData::Blob(blob),
|
||||
};
|
||||
|
||||
Ok((data, MimeType(mime_type.to_owned())))
|
||||
}
|
||||
|
||||
fn handle_zip_archive(blob: Arc<Blob>, container: Vec<u8>) -> DecryptedData {
|
||||
let mut entries = vec![];
|
||||
let cursor = Cursor::new(container);
|
||||
if let Ok(mut zip) = zip::ZipArchive::new(cursor) {
|
||||
|
@ -111,14 +119,14 @@ pub fn decrypt(
|
|||
entries.sort_by(|a, b| a.name.cmp(&b.name));
|
||||
DecryptedData::Archive(blob, entries)
|
||||
}
|
||||
ContentType::GzipArchive => {
|
||||
|
||||
fn handle_gzip(blob: Arc<Blob>, container: Vec<u8>) -> DecryptedData {
|
||||
let mut entries = vec![];
|
||||
let cursor = Cursor::new(container);
|
||||
let gzip_dec = flate2::read::GzDecoder::new(cursor);
|
||||
let mut archive = tar::Archive::new(gzip_dec);
|
||||
if let Ok(files) = archive.entries() {
|
||||
for file in files {
|
||||
if let Ok(file) = file {
|
||||
for file in files.flatten() {
|
||||
let file_path = if let Ok(file_path) = file.path() {
|
||||
file_path.display().to_string()
|
||||
} else {
|
||||
|
@ -130,17 +138,11 @@ pub fn decrypt(
|
|||
});
|
||||
}
|
||||
}
|
||||
}
|
||||
if entries.len() > 0 {
|
||||
DecryptedData::Archive(blob, entries)
|
||||
} else {
|
||||
if entries.is_empty() {
|
||||
DecryptedData::Blob(blob)
|
||||
} else {
|
||||
DecryptedData::Archive(blob, entries)
|
||||
}
|
||||
},
|
||||
ContentType::Unknown => DecryptedData::Blob(blob),
|
||||
};
|
||||
|
||||
Ok((data, MimeType(mime_type.to_owned())))
|
||||
}
|
||||
|
||||
#[derive(Clone, Copy, Debug, PartialEq, Eq, PartialOrd, Ord)]
|
||||
|
@ -150,7 +152,7 @@ enum ContentType {
|
|||
Audio,
|
||||
Video,
|
||||
ZipArchive,
|
||||
GzipArchive,
|
||||
Gzip,
|
||||
Unknown,
|
||||
}
|
||||
|
||||
|
@ -188,7 +190,7 @@ impl<T: AsRef<[u8]>> ContentTypeExt for T {
|
|||
} else if mime_type == "application/zip" {
|
||||
ContentType::ZipArchive
|
||||
} else if mime_type == "application/gzip" {
|
||||
ContentType::GzipArchive
|
||||
ContentType::Gzip
|
||||
} else {
|
||||
ContentType::Unknown
|
||||
}
|
||||
|
@ -221,7 +223,7 @@ mod content_type {
|
|||
test_content_type!(mp4_is_video, "movie.mp4", ContentType::Video);
|
||||
test_content_type!(mkv_is_video, "movie.mkv", ContentType::Video);
|
||||
test_content_type!(zip_is_zip, "archive.zip", ContentType::ZipArchive);
|
||||
test_content_type!(gzip_is_gzip, "image.png.gz", ContentType::GzipArchive);
|
||||
test_content_type!(gzip_is_gzip, "image.png.gz", ContentType::Gzip);
|
||||
test_content_type!(binary_is_unknown, "omegaupload", ContentType::Unknown);
|
||||
test_content_type!(pgp_is_text, "text.pgp", ContentType::Text);
|
||||
}
|
||||
|
|
Loading…
Reference in a new issue