Mirrors/gallery-dl
1
0
Fork 0
mirror of https://github.com/mikf/gallery-dl.git synced 2024-11-26 04:32:51 +01:00
Code Issues Releases Wiki Activity

[downloader:http] add signature check for .mp4 files

Browse Source
This commit is contained in:
Mike Fährmann 2022-11-16 21:45:26 +01:00
parent 7897f68225
commit f87cfa5f66
No known key found for this signature in database
GPG Key ID: 5680CA389D365A88
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
gallery_dl/downloader/http.py
View File

@ -390,11 +390,13 @@ SIGNATURE_CHECKS = {
"bmp" : lambda s: s[0:2] == b"BM", "bmp" : lambda s: s[0:2] == b"BM",
"webp": lambda s: (s[0:4] == b"RIFF" and "webp": lambda s: (s[0:4] == b"RIFF" and
s[8:12] == b"WEBP"), s[8:12] == b"WEBP"),
"avif": lambda s: s[4:12] == b"ftypavif", "avif": lambda s: s[4:11] == b"ftypavi" and s[11] in b"fs",
"svg" : lambda s: s[0:5] == b"<?xml", "svg" : lambda s: s[0:5] == b"<?xml",
"ico" : lambda s: s[0:4] == b"\x00\x00\x01\x00", "ico" : lambda s: s[0:4] == b"\x00\x00\x01\x00",
"cur" : lambda s: s[0:4] == b"\x00\x00\x02\x00", "cur" : lambda s: s[0:4] == b"\x00\x00\x02\x00",
"psd" : lambda s: s[0:4] == b"8BPS", "psd" : lambda s: s[0:4] == b"8BPS",
"mp4" : lambda s: (s[4:8] == b"ftyp" and s[8:11] in (
b"mp4", b"avc", b"iso", b"M4V")),
"webm": lambda s: s[0:4] == b"\x1A\x45\xDF\xA3", "webm": lambda s: s[0:4] == b"\x1A\x45\xDF\xA3",
"ogg" : lambda s: s[0:4] == b"OggS", "ogg" : lambda s: s[0:4] == b"OggS",
"wav" : lambda s: (s[0:4] == b"RIFF" and "wav" : lambda s: (s[0:4] == b"RIFF" and
@ -402,7 +404,7 @@ SIGNATURE_CHECKS = {
"mp3" : lambda s: (s[0:3] == b"ID3" or "mp3" : lambda s: (s[0:3] == b"ID3" or
s[0:2] in (b"\xFF\xFB", b"\xFF\xF3", b"\xFF\xF2")), s[0:2] in (b"\xFF\xFB", b"\xFF\xF3", b"\xFF\xF2")),
"zip" : lambda s: s[0:4] in (b"PK\x03\x04", b"PK\x05\x06", b"PK\x07\x08"), "zip" : lambda s: s[0:4] in (b"PK\x03\x04", b"PK\x05\x06", b"PK\x07\x08"),
"rar" : lambda s: s[0:6] == b"\x52\x61\x72\x21\x1A\x07", "rar" : lambda s: s[0:6] == b"Rar!\x1A\x07",
"7z" : lambda s: s[0:6] == b"\x37\x7A\xBC\xAF\x27\x1C", "7z" : lambda s: s[0:6] == b"\x37\x7A\xBC\xAF\x27\x1C",
"pdf" : lambda s: s[0:5] == b"%PDF-", "pdf" : lambda s: s[0:5] == b"%PDF-",
"swf" : lambda s: s[0:3] in (b"CWS", b"FWS"), "swf" : lambda s: s[0:3] in (b"CWS", b"FWS"),