diff --git a/src/keylogger_detector.py b/src/keylogger_detector.py index 5676eb9..c8739a8 100755 --- a/src/keylogger_detector.py +++ b/src/keylogger_detector.py @@ -131,7 +131,7 @@ def detect_keyloggers(): pids = [] for device_file in keyboard_device_files: pids.append(get_pids_using_file(device_file)) - pids = sorted(list(set(pids))) + pids = sorted(list(set(int(pid) for pid in pids))) if verbose_option: print('[Verbose] Process IDs using keyboard device files:', pids) diff --git a/src/utils.py b/src/utils.py index b7cb085..1f953af 100755 --- a/src/utils.py +++ b/src/utils.py @@ -46,7 +46,7 @@ def get_keyboard_device_files(names): keyboard_device_files = [] for root, dirs, files in os.walk('/dev/input/by-path'): for file in files: - if any(name in files for name in names): + if any(name in file for name in names): keyboard_device_files.append(get_real_path(os.path.join(root, file))) return keyboard_device_files