From 1600e54aaf8db566efedef2f2d0859b1f55e25f9 Mon Sep 17 00:00:00 2001
From: Kamil Solecki <46087838+kamil-s-solecki@users.noreply.github.com>
Date: Tue, 1 Aug 2023 22:37:47 +0200
Subject: [PATCH] Warn user about unmet dependencies (#1791)

---
 pwndbg/exception.py | 39 +++++++++++++++++++++++++++++++++++++++
 1 file changed, 39 insertions(+)

diff --git a/pwndbg/exception.py b/pwndbg/exception.py
index e7e2494c9..2ed8dbd82 100644
--- a/pwndbg/exception.py
+++ b/pwndbg/exception.py
@@ -1,10 +1,12 @@
 from __future__ import annotations
 
 import functools
+import os
 import sys
 import traceback
 
 import gdb
+import pkg_resources
 
 import pwndbg.lib.cache
 import pwndbg.lib.stdio
@@ -33,6 +35,40 @@ debug = config.add_param(
 )
 
 
+def inform_unmet_dependencies(errors) -> None:
+    """
+    Informs user about unmet dependencies
+    """
+    msg = message.error("You appear to have unmet dependencies.\n")
+    for e in errors:
+        if isinstance(e, pkg_resources.DistributionNotFound):
+            msg += message.notice(f"- required {e.args[0]}, but not installed\n")
+        else:
+            msg += message.notice(f"- required {e.args[1]}, installed: {e.args[0]}\n")
+    msg += message.notice("Consider running: ")
+    msg += message.hint("`setup.sh`\n")
+    print(msg)
+
+
+@pwndbg.lib.cache.cache_until("forever")
+def check_dependencies():
+    """
+    Checks if there are any unmet dependencies in requirements.txt
+    """
+    project_path = os.path.dirname(os.path.abspath(__file__))
+    requirements_path = os.path.join(project_path, os.pardir, "requirements.txt")
+    with open(requirements_path, "r") as f:
+        errors = []
+        for line in f.readlines():
+            try:
+                pkg_resources.require(line)
+            except (pkg_resources.VersionConflict, pkg_resources.DistributionNotFound) as e:
+                errors.append(e)
+
+        if errors:
+            inform_unmet_dependencies(errors)
+
+
 @pwndbg.lib.cache.cache_until("forever")
 def inform_report_issue(exception_msg) -> None:
     """
@@ -76,6 +112,9 @@ def handle(name="Error"):
         e.__traceback__ = T
         raise e
 
+    # Check dependencies against requirements.txt and warn user
+    check_dependencies()
+
     # Display the error
     if debug or verbose:
         exception_msg = traceback.format_exc()