From 824ce64e66d0959df40ee4a16cecacb99e507505 Mon Sep 17 00:00:00 2001
From: SomberNight <somber.night@protonmail.com>
Date: Mon, 2 Jan 2023 13:38:05 +0000
Subject: [PATCH] qt gui: display nice error if QR code data overflows

there is existing handler-code at e.g.
https://github.com/spesmilo/electrum/blob/1a7634e615156f6f9b4f799e0d549af05446bc69/electrum/gui/qt/transaction_dialog.py#L309
but we should make sure setData() always raises the exc when needed,
as paintEvent() is too late for nice handling.

closes https://github.com/spesmilo/electrum/issues/4288
closes https://github.com/spesmilo/electrum/issues/4280
---
 electrum/gui/qt/qrcodewidget.py | 22 ++++++++++++++++------
 1 file changed, 16 insertions(+), 6 deletions(-)

diff --git a/electrum/gui/qt/qrcodewidget.py b/electrum/gui/qt/qrcodewidget.py
index 35f251afb..01d865934 100644
--- a/electrum/gui/qt/qrcodewidget.py
+++ b/electrum/gui/qt/qrcodewidget.py
@@ -1,6 +1,7 @@
 from typing import Optional
 
 import qrcode
+import qrcode.exceptions
 
 from PyQt5.QtGui import QColor, QPen
 import PyQt5.QtGui as QtGui
@@ -16,6 +17,10 @@ from electrum.simple_config import SimpleConfig
 from .util import WindowModalDialog, WWLabel, getSaveFileName
 
 
+class QrCodeDataOverflow(qrcode.exceptions.DataOverflowError):
+    pass
+
+
 class QRCodeWidget(QWidget):
 
     def __init__(self, data=None, *, manual_size: bool = False):
@@ -27,20 +32,25 @@ class QRCodeWidget(QWidget):
         self.setData(data)
 
     def setData(self, data):
-        if self.data != data:
-            self.data = data
-        if self.data:
-            self.qr = qrcode.QRCode(
+        if data:
+            qr = qrcode.QRCode(
                 error_correction=qrcode.constants.ERROR_CORRECT_L,
                 box_size=10,
                 border=0,
             )
-            self.qr.add_data(self.data)
+            try:
+                qr.add_data(data)
+                qr_matrix = qr.get_matrix()  # test that data fits in QR code
+            except (ValueError, qrcode.exceptions.DataOverflowError) as e:
+                raise QrCodeDataOverflow() from e
+            self.qr = qr
+            self.data = data
             if not self._manual_size:
-                k = len(self.qr.get_matrix())
+                k = len(qr_matrix)
                 self.setMinimumSize(k * 5, k * 5)
         else:
             self.qr = None
+            self.data = None
 
         self.update()