Mirror/app-MAIL-temp
1
0
Fork 0
mirror of https://github.com/simple-login/app.git synced 2024-11-16 00:48:32 +01:00
Code Issues Projects Releases Packages Wiki Activity

Do not allow SVG image uploads (#1198)

Browse source
This commit is contained in:
Carlos Quintana 2022-07-29 08:52:51 +02:00 committed by GitHub
parent 54466389c5
commit a04152a37f
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
app/image_validation.py
View file

@ -5,7 +5,6 @@ class ImageFormat(Enum):
Png = 1
Jpg = 2
Webp = 3
Svg = 4
Unknown = 9
@ -13,7 +12,6 @@ magic_numbers = {
ImageFormat.Png: bytes([0x89, 0x50, 0x4E, 0x47, 0x0D, 0x0A, 0x1A, 0x0A]),
ImageFormat.Jpg: bytes([0xFF, 0xD8, 0xFF, 0xE0]),
ImageFormat.Webp: bytes([0x52, 0x49, 0x46, 0x46]),
ImageFormat.Svg: bytes([0x3C, 0x3F, 0x78, 0x6D, 0x6C]), # <?xml
}
@ -22,7 +20,6 @@ def detect_image_format(image: bytes) -> ImageFormat:
for fmt, header in magic_numbers.items():
if image.startswith(header):
return fmt
# Detect if is svg
# We don't know the type
return ImageFormat.Unknown

4
tests/test_image_validation.py
View file

@ -42,6 +42,6 @@ def test_webp_file_is_detected():
assert detect_image_format(contents) is ImageFormat.Webp
def test_svg_file_is_detected():
def test_svg_file_is_not_detected():
contents = read_static_file_contents("icon.svg")
assert detect_image_format(contents) is ImageFormat.Svg
assert detect_image_format(contents) is ImageFormat.Unknown