AIL/ail-framework
8
0
Fork 0
mirror of https://github.com/ail-project/ail-framework.git synced 2025-01-18 08:26:15 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

fix: [exifs module] Mp4 UnidentifiedImageError

Browse source
This commit is contained in:
terrtia 2024-12-11 10:13:11 +01:00
parent f4584f7e98
commit 6f964d7f8b
No known key found for this signature in database
GPG key ID: 1E1B1F50D84613D0
1 changed files with 20 additions and 17 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

37
bin/modules/Exif.py
View file

@ -12,7 +12,7 @@ The Exif Module
import os
import sys
from PIL import Image, ExifTags
from PIL import Image, ExifTags, UnidentifiedImageError
sys.path.append(os.environ['AIL_BIN'])
##################################
@ -38,22 +38,25 @@ class Exif(AbstractModule):
def compute(self, message):
image = self.get_obj()
print(image)
img = Image.open(image.get_filepath())
img_exif = img.getexif()
print(img_exif)
if img_exif:
self.logger.critical(f'Exif: {self.get_obj().id}')
gps = img_exif.get(34853)
print(gps)
self.logger.critical(f'gps: {gps}')
for key, val in img_exif.items():
if key in ExifTags.TAGS:
print(f'{ExifTags.TAGS[key]}:{val}')
self.logger.critical(f'{ExifTags.TAGS[key]}:{val}')
else:
print(f'{key}:{val}')
self.logger.critical(f'{key}:{val}')
sys.exit(0)
try:
img = Image.open(image.get_filepath())
img_exif = img.getexif()
print(img_exif)
if img_exif:
self.logger.critical(f'Exif: {self.get_obj().id}')
gps = img_exif.get(34853)
print(gps)
self.logger.critical(f'gps: {gps}')
for key, val in img_exif.items():
if key in ExifTags.TAGS:
print(f'{ExifTags.TAGS[key]}:{val}')
self.logger.critical(f'{ExifTags.TAGS[key]}:{val}')
else:
print(f'{key}:{val}')
self.logger.critical(f'{key}:{val}')
sys.exit(0)
except UnidentifiedImageError:
self.logger.info(f'Invalid image: {image.get_filepath()}')
# tag = 'infoleak:automatic-detection="cve"'
# Send to Tags Queue