Archives
/
extract_otp_secret_keys
mirror of https://github.com/scito/extract_otp_secret_keys
2
0
Fork 0
Code Issues Projects Releases Wiki Activity

yolo steps

Browse Source
cv2_segfault_repro_1
scito 1 year ago
parent 4c3f016cd6
commit 2e250684a0
2 changed files with 180 additions and 43 deletions
Show Stats Download Patch File Download Diff File

3
.github/workflows/ci.yml vendored
Unescape Escape View File

@ -60,10 +60,9 @@ jobs:
pytest -vvv -s tests/extract_otp_secrets_test.py::test_cv2_segfault_4 pytest -vvv -s tests/extract_otp_secrets_test.py::test_cv2_segfault_4
pytest -vvv -s tests/extract_otp_secrets_test.py::test_cv2_segfault_5 pytest -vvv -s tests/extract_otp_secrets_test.py::test_cv2_segfault_5
pytest -vvv -s tests/extract_otp_secrets_test.py::test_cv2_segfault_6 pytest -vvv -s tests/extract_otp_secrets_test.py::test_cv2_segfault_6
pytest -vvv -s tests/extract_otp_secrets_test.py::test_cv2_segfault_6_d0
pytest -vvv -s tests/extract_otp_secrets_test.py::test_cv2_segfault_6_d1 pytest -vvv -s tests/extract_otp_secrets_test.py::test_cv2_segfault_6_d1
pytest -vvv -s tests/extract_otp_secrets_test.py::test_cv2_segfault_6_d2 pytest -vvv -s tests/extract_otp_secrets_test.py::test_cv2_segfault_6_d2
pytest -vvv -s tests/extract_otp_secrets_test.py::test_cv2_segfault_6_m
pytest -vvv -s tests/extract_otp_secrets_test.py::test_cv2_segfault_6_n
pytest -vvv -s tests/extract_otp_secrets_test.py::test_cv2_segfault_7 pytest -vvv -s tests/extract_otp_secrets_test.py::test_cv2_segfault_7
pytest -vvv -s tests/extract_otp_secrets_test.py pytest -vvv -s tests/extract_otp_secrets_test.py
if: (matrix.python-version != '3.x' || matrix.platform != 'ubuntu-latest') if: (matrix.python-version != '3.x' || matrix.platform != 'ubuntu-latest')

220
tests/extract_otp_secrets_test.py
Unescape Escape View File

@ -206,7 +206,7 @@ def test_cv2_segfault_6(qr_mode: str) -> None:
print('Done') print('Done')
def test_cv2_segfault_6_d1(qr_mode: str) -> None: def test_cv2_segfault_6_d0(qr_mode: str) -> None:
print(f'QRmode: {qr_mode}') print(f'QRmode: {qr_mode}')
print('cv2.imread') print('cv2.imread')
@ -232,8 +232,13 @@ def test_cv2_segfault_6_d1(qr_mode: str) -> None:
print('QReader()') print('QReader()')
qreader = QReader() qreader = QReader()
print('yolo.detect') print('cv2.dnn.blobFromImage')
yolo.detect(img) _INPUT_SIZE = (416, 416)
_CONF_THRESHOLD = 0.5
blob = cv2.dnn.blobFromImage(img, 1 / 255, _INPUT_SIZE, swapRB=False, crop=False)
# self.yolo_v3_QR_detector.setInput(blob=blob)
# output = self.yolo_v3_QR_detector.forward(self.output_layer_name)
elif qr_mode_2 == extract_otp_secrets.QRMode.CV2: elif qr_mode_2 == extract_otp_secrets.QRMode.CV2:
cv2.QRCodeDetector().detectAndDecode(img) cv2.QRCodeDetector().detectAndDecode(img)
@ -245,7 +250,7 @@ def test_cv2_segfault_6_d1(qr_mode: str) -> None:
print('Done') print('Done')
def test_cv2_segfault_6_d2(qr_mode: str) -> None: def test_cv2_segfault_6_d1(qr_mode: str) -> None:
print(f'QRmode: {qr_mode}') print(f'QRmode: {qr_mode}')
print('cv2.imread') print('cv2.imread')
@ -271,11 +276,15 @@ def test_cv2_segfault_6_d2(qr_mode: str) -> None:
print('QReader()') print('QReader()')
qreader = QReader() qreader = QReader()
print('yolo.detect') print('cv2.dnn.blobFromImage')
yolo.detect(img) _INPUT_SIZE = (416, 416)
_CONF_THRESHOLD = 0.5
blob = cv2.dnn.blobFromImage(img, 1 / 255, _INPUT_SIZE, swapRB=False, crop=False)
print('yolo.detect') print('yolo.yolo_v3_QR_detector.setInput')
cv2_qr.detect(img) yolo.yolo_v3_QR_detector.setInput(blob=blob)
# output = self.yolo_v3_QR_detector.forward(self.output_layer_name)
elif qr_mode_2 == extract_otp_secrets.QRMode.CV2: elif qr_mode_2 == extract_otp_secrets.QRMode.CV2:
cv2.QRCodeDetector().detectAndDecode(img) cv2.QRCodeDetector().detectAndDecode(img)
@ -287,7 +296,7 @@ def test_cv2_segfault_6_d2(qr_mode: str) -> None:
print('Done') print('Done')
def test_cv2_segfault_6_m(qr_mode: str) -> None: def test_cv2_segfault_6_d2(qr_mode: str) -> None:
print(f'QRmode: {qr_mode}') print(f'QRmode: {qr_mode}')
print('cv2.imread') print('cv2.imread')
@ -299,13 +308,13 @@ def test_cv2_segfault_6_m(qr_mode: str) -> None:
if qr_mode_2 in [extract_otp_secrets.QRMode.QREADER, extract_otp_secrets.QRMode.DEEP_QREADER]: if qr_mode_2 in [extract_otp_secrets.QRMode.QREADER, extract_otp_secrets.QRMode.DEEP_QREADER]:
print('cv.QRCodeDetector()') print('cv.QRCodeDetector()')
cv2.QRCodeDetector() cv2_qr =cv2.QRCodeDetector()
print('from __yolo_v3_qr_detector.yolov3_qr_detector import _YoloV3QRDetector') print('from __yolo_v3_qr_detector.yolov3_qr_detector import _YoloV3QRDetector')
from __yolo_v3_qr_detector.yolov3_qr_detector import _YoloV3QRDetector from __yolo_v3_qr_detector.yolov3_qr_detector import _YoloV3QRDetector
print('_YoloV3QRDetector()') print('_YoloV3QRDetector()')
_YoloV3QRDetector() yolo = _YoloV3QRDetector()
print('from qreader import QReader') print('from qreader import QReader')
from qreader import QReader from qreader import QReader
@ -313,8 +322,17 @@ def test_cv2_segfault_6_m(qr_mode: str) -> None:
print('QReader()') print('QReader()')
qreader = QReader() qreader = QReader()
print('qreader.detect(img)') print('cv2.dnn.blobFromImage')
found, bbox = qreader.detect(img) _INPUT_SIZE = (416, 416)
_CONF_THRESHOLD = 0.5
blob = cv2.dnn.blobFromImage(img, 1 / 255, _INPUT_SIZE, swapRB=False, crop=False)
print('yolo.yolo_v3_QR_detector.setInput')
yolo.yolo_v3_QR_detector.setInput(blob=blob)
print('yolo.yolo_v3_QR_detector.forward')
output = yolo.yolo_v3_QR_detector.forward(yolo.output_layer_name)
elif qr_mode_2 == extract_otp_secrets.QRMode.CV2: elif qr_mode_2 == extract_otp_secrets.QRMode.CV2:
cv2.QRCodeDetector().detectAndDecode(img) cv2.QRCodeDetector().detectAndDecode(img)
@ -326,46 +344,166 @@ def test_cv2_segfault_6_m(qr_mode: str) -> None:
print('Done') print('Done')
def test_cv2_segfault_6_n(qr_mode: str) -> None: # def test_cv2_segfault_6_d1(qr_mode: str) -> None:
print(f'QRmode: {qr_mode}') # print(f'QRmode: {qr_mode}')
print('cv2.imread') # print('cv2.imread')
img = cv2.imread('tests/data/test_googleauth_export.png') # img = cv2.imread('tests/data/test_googleauth_export.png')
qr_mode_2 = extract_otp_secrets.QRMode[qr_mode] # qr_mode_2 = extract_otp_secrets.QRMode[qr_mode]
print(f'detect and decode for qr_mode {qr_mode_2}') # print(f'detect and decode for qr_mode {qr_mode_2}')
if qr_mode_2 in [extract_otp_secrets.QRMode.QREADER, extract_otp_secrets.QRMode.DEEP_QREADER]: # if qr_mode_2 in [extract_otp_secrets.QRMode.QREADER, extract_otp_secrets.QRMode.DEEP_QREADER]:
print('cv.QRCodeDetector()') # print('cv.QRCodeDetector()')
cv2.QRCodeDetector() # cv2_qr =cv2.QRCodeDetector()
print('from __yolo_v3_qr_detector.yolov3_qr_detector import _YoloV3QRDetector') # print('from __yolo_v3_qr_detector.yolov3_qr_detector import _YoloV3QRDetector')
from __yolo_v3_qr_detector.yolov3_qr_detector import _YoloV3QRDetector # from __yolo_v3_qr_detector.yolov3_qr_detector import _YoloV3QRDetector
print('_YoloV3QRDetector()') # print('_YoloV3QRDetector()')
_YoloV3QRDetector() # yolo = _YoloV3QRDetector()
print('from qreader import QReader') # print('from qreader import QReader')
from qreader import QReader # from qreader import QReader
print('QReader()') # print('QReader()')
qreader = QReader() # qreader = QReader()
print('qreader.detect(img)') # print('yolo.detect')
found, bbox = qreader.detect(img) # yolo.detect(img)
print('qreader.decode(img, bbox)') # elif qr_mode_2 == extract_otp_secrets.QRMode.CV2:
qreader.decode(img, bbox) # cv2.QRCodeDetector().detectAndDecode(img)
# elif qr_mode_2 == extract_otp_secrets.QRMode.WECHAT:
# cv2.wechat_qrcode.WeChatQRCode().detectAndDecode(img)
# elif qr_mode_2 == extract_otp_secrets.QRMode.ZBAR:
# zbar.decode(img)
elif qr_mode_2 == extract_otp_secrets.QRMode.CV2: # print('Done')
cv2.QRCodeDetector().detectAndDecode(img)
elif qr_mode_2 == extract_otp_secrets.QRMode.WECHAT:
cv2.wechat_qrcode.WeChatQRCode().detectAndDecode(img)
elif qr_mode_2 == extract_otp_secrets.QRMode.ZBAR:
zbar.decode(img)
print('Done')
# def test_cv2_segfault_6_d2(qr_mode: str) -> None:
# print(f'QRmode: {qr_mode}')
# print('cv2.imread')
# img = cv2.imread('tests/data/test_googleauth_export.png')
# qr_mode_2 = extract_otp_secrets.QRMode[qr_mode]
# print(f'detect and decode for qr_mode {qr_mode_2}')
# if qr_mode_2 in [extract_otp_secrets.QRMode.QREADER, extract_otp_secrets.QRMode.DEEP_QREADER]:
# print('cv.QRCodeDetector()')
# cv2_qr =cv2.QRCodeDetector()
# print('from __yolo_v3_qr_detector.yolov3_qr_detector import _YoloV3QRDetector')
# from __yolo_v3_qr_detector.yolov3_qr_detector import _YoloV3QRDetector
# print('_YoloV3QRDetector()')
# yolo = _YoloV3QRDetector()
# print('from qreader import QReader')
# from qreader import QReader
# print('QReader()')
# qreader = QReader()
# print('yolo.detect')
# yolo.detect(img)
# print('yolo.detect')
# cv2_qr.detect(img)
# elif qr_mode_2 == extract_otp_secrets.QRMode.CV2:
# cv2.QRCodeDetector().detectAndDecode(img)
# elif qr_mode_2 == extract_otp_secrets.QRMode.WECHAT:
# cv2.wechat_qrcode.WeChatQRCode().detectAndDecode(img)
# elif qr_mode_2 == extract_otp_secrets.QRMode.ZBAR:
# zbar.decode(img)
# print('Done')
# def test_cv2_segfault_6_m(qr_mode: str) -> None:
# print(f'QRmode: {qr_mode}')
# print('cv2.imread')
# img = cv2.imread('tests/data/test_googleauth_export.png')
# qr_mode_2 = extract_otp_secrets.QRMode[qr_mode]
# print(f'detect and decode for qr_mode {qr_mode_2}')
# if qr_mode_2 in [extract_otp_secrets.QRMode.QREADER, extract_otp_secrets.QRMode.DEEP_QREADER]:
# print('cv.QRCodeDetector()')
# cv2.QRCodeDetector()
# print('from __yolo_v3_qr_detector.yolov3_qr_detector import _YoloV3QRDetector')
# from __yolo_v3_qr_detector.yolov3_qr_detector import _YoloV3QRDetector
# print('_YoloV3QRDetector()')
# _YoloV3QRDetector()
# print('from qreader import QReader')
# from qreader import QReader
# print('QReader()')
# qreader = QReader()
# print('qreader.detect(img)')
# found, bbox = qreader.detect(img)
# elif qr_mode_2 == extract_otp_secrets.QRMode.CV2:
# cv2.QRCodeDetector().detectAndDecode(img)
# elif qr_mode_2 == extract_otp_secrets.QRMode.WECHAT:
# cv2.wechat_qrcode.WeChatQRCode().detectAndDecode(img)
# elif qr_mode_2 == extract_otp_secrets.QRMode.ZBAR:
# zbar.decode(img)
# print('Done')
# def test_cv2_segfault_6_n(qr_mode: str) -> None:
# print(f'QRmode: {qr_mode}')
# print('cv2.imread')
# img = cv2.imread('tests/data/test_googleauth_export.png')
# qr_mode_2 = extract_otp_secrets.QRMode[qr_mode]
# print(f'detect and decode for qr_mode {qr_mode_2}')
# if qr_mode_2 in [extract_otp_secrets.QRMode.QREADER, extract_otp_secrets.QRMode.DEEP_QREADER]:
# print('cv.QRCodeDetector()')
# cv2.QRCodeDetector()
# print('from __yolo_v3_qr_detector.yolov3_qr_detector import _YoloV3QRDetector')
# from __yolo_v3_qr_detector.yolov3_qr_detector import _YoloV3QRDetector
# print('_YoloV3QRDetector()')
# _YoloV3QRDetector()
# print('from qreader import QReader')
# from qreader import QReader
# print('QReader()')
# qreader = QReader()
# print('qreader.detect(img)')
# found, bbox = qreader.detect(img)
# print('qreader.decode(img, bbox)')
# qreader.decode(img, bbox)
# elif qr_mode_2 == extract_otp_secrets.QRMode.CV2:
# cv2.QRCodeDetector().detectAndDecode(img)
# elif qr_mode_2 == extract_otp_secrets.QRMode.WECHAT:
# cv2.wechat_qrcode.WeChatQRCode().detectAndDecode(img)
# elif qr_mode_2 == extract_otp_secrets.QRMode.ZBAR:
# zbar.decode(img)
# print('Done')
def test_cv2_segfault_7(qr_mode: str) -> None: def test_cv2_segfault_7(qr_mode: str) -> None:

Write Preview
Loading…
Cancel
Save