From 4dbda5975a387a7eb32205dc6a770386eac2155a Mon Sep 17 00:00:00 2001 From: terrtia Date: Wed, 2 Oct 2024 16:41:23 +0200 Subject: [PATCH] fix: [qrcode extractor] fix image/screenshot tag --- bin/modules/QrCodeReader.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/bin/modules/QrCodeReader.py b/bin/modules/QrCodeReader.py index 43ff3836..a83a7f92 100755 --- a/bin/modules/QrCodeReader.py +++ b/bin/modules/QrCodeReader.py @@ -60,6 +60,7 @@ class QrCodeReader(AbstractModule): try: decodeds = decode(image) for decoded in decodeds: + qr_codes = True if decoded.data: contents.append(decoded.data.decode()) except ValueError as e: @@ -76,6 +77,7 @@ class QrCodeReader(AbstractModule): data_qr, box, qrcode_image = detector.detectAndDecode(image) if data_qr: contents.append(data_qr) + qr_codes = True if qr_codes and not contents: # # # # 0.5s per image @@ -84,6 +86,7 @@ class QrCodeReader(AbstractModule): decoded_text = qreader.detect_and_decode(image=image) for d in decoded_text: contents.append(d) + qr_codes = True except ValueError as e: self.logger.error(f'{e}: {self.obj.get_global_id()}') return qr_codes, contents @@ -122,7 +125,7 @@ class QrCodeReader(AbstractModule): # TODO only if new ??? self.add_message_to_queue(obj=qr_code, queue='Item') - if is_qrcode: + if is_qrcode or contents: tag = 'infoleak:automatic-detection="qrcode"' self.add_message_to_queue(obj=self.obj, message=tag, queue='Tags')