From 961d0814849c2e4626a01806531ad322e978ee7a Mon Sep 17 00:00:00 2001
From: dbarzin <dbarzin@gmail.com>
Date: Thu, 30 Jun 2022 10:44:21 +0200
Subject: [PATCH] integrate new mouse click

---
 pandora-box.py | 24 +++++++++++++++++-------
 1 file changed, 17 insertions(+), 7 deletions(-)

diff --git a/pandora-box.py b/pandora-box.py
index 74d392d..0358d22 100755
--- a/pandora-box.py
+++ b/pandora-box.py
@@ -76,6 +76,16 @@ def display_image(status):
         return
     os.system("convert -resize %s -background black -gravity center -extent %s %s bgra:/dev/fb0" % (SCREEN_SIZE, SCREEN_SIZE, image))
 
+# -----------------------------------------------------------
+
+def waitMouseClick():
+    mouse = open( "/dev/input/mice", "rb" )
+    while True:
+        buf = mouse.read(3)
+        if ((buf[0] & 0x1)==1):
+            break;
+    mouse.close()
+
 # -----------------------------------------------------------
 # CURSES Screen
 # -----------------------------------------------------------
@@ -83,13 +93,12 @@ def display_image(status):
 """Initialise curses"""
 def init_curses():
     global screen
-    screen = curses.initscr()
-    screen.keypad(1)
-    curses.mousemask(curses.ALL_MOUSE_EVENTS | curses.REPORT_MOUSE_POSITION)
-    curses.flushinp()
-    curses.noecho()
     if CURSES:
-        # remove blinking cursor
+        screen = curses.initscr()
+        screen.keypad(1)
+        curses.mousemask(curses.ALL_MOUSE_EVENTS | curses.REPORT_MOUSE_POSITION)
+        curses.flushinp()
+        curses.noecho()
         curses.curs_set(0)
     else: 
         display_image("WAIT")
@@ -334,9 +343,10 @@ def device_loop():
                         log('%d infected files found !' % len(infected_files))
                         if not CURSES:
                             display_image("BAD")
+                            waitMouseClick()
                         else:
                             log('PRESS KEY TO CLEAN')
-                        screen.getch()
+                            screen.getch()
                         # Remove infected files
                         for file in infected_files:
                             try :