pwn-plugins/probeReq.py at main · unitMeasure/pwn-plugins · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
import logging
from PIL import ImageFont
import pwnagotchi.plugins as plugins
from pwnagotchi.ui.components import Text
from pwnagotchi.ui.view import BLACK
from pwnagotchi.bettercap import Client

def load_font():
    """Load the smallest suitable font, with fallbacks."""
    font_paths = [
        "/usr/share/fonts/truetype/dejavu/DejaVuSansMono.ttf",
        "/usr/share/fonts/truetype/dejavu/DejaVuSans.ttf",
        "/usr/share/fonts/truetype/dejavu/DejaVuSerif.ttf",
    ]

    for path in font_paths:
        try:
            font = ImageFont.truetype(path, 8)
            logging.info("[probeReq] Using font: %s", path)
            return font
        except (OSError, IOError) as e:
            logging.debug("[probeReq] Could not load font %s: %s", path, e)

    # Final fallback to Pwnagotchi's built-in font
    logging.warning("[probeReq] No system font found, using pwnagotchi fonts.Small")
    return fonts.Small


_TINY_FONT = load_font() # Load a font directly instead of using fonts.Small, to control  exact pixel size

class probeReq(plugins.Plugin):
    __GitHub__ = "https://github.com/unitMeasure/pwn-plugins/"
    __author__ = "avipars"
    __editor__ = "avipars"
    __version__ = "0.0.5"
    __license__ = "GPL3"
    __description__ = "BETA Listens for Wi-Fi probe requests, displays them on screen and logs them."
    __name__ = "probeReq"
    __defaults__ = {
        "enabled": False,
        "verbose": False,
        "logging": False,
    }

    def __init__(self):
        self.ready = False
        self.title = ""
        self.running = True
        self.pr_status = "Waiting"
        self.pos_x = 0
        self.pos_y = 70
        self.show_verbose = False
        self.log_results = False

    def on_loaded(self):
        self.pr_status = "Waiting."
        self.show_verbose = self.options.get("verbose", False)
        self.log_results = self.options.get("logging", False)

    def on_ready(self, agent):
        self.pr_status = "Waiting.."

    def on_ui_setup(self, ui):
        try:
            if "pos_x" in self.options:
                self.pos_x = int(self.options.get("pos_x", 0))
            if "pos_y" in self.options:
                self.pos_y = int(self.options.get("pos_y", 70))

            logging.info(f"[{self.__class__.__name__}] pos_x {self.pos_x} pos_y {self.pos_y}")

            ui.add_element(
                "pr_status",
                Text(
                    color=BLACK,
                    value="Active",
                    position=(self.pos_x, self.pos_y),
                    font=_TINY_FONT,
                ),)

        except Exception as e:
            logging.debug(f"[{self.__class__.__name__}]: Error on_ui_setup: {e}")

    def on_ui_update(self, ui):
        ui.set("pr_status", "%s" % (self.pr_status))

    def on_bcap_wifi_client_probe(self, agent, event):
        """WIFI CLIENT PROBE REQUEST"""
        if not self.running:
            if self.log_results:
                logging.info(f"[{self.__class__.__name__}]: plugin stopped running")
            return

        probe = event["data"]
        d_name = (probe.get('essid') or '')[:16] # essid truncated

        if self.log_results:
            logging.info(f"[{self.__class__.__name__}]: Probe %s" % (probe))

        stat = "pr:%s" % d_name # limit essid to 20 chars
        if self.show_verbose:
            stat += " rssi:%s" % probe["rssi"]
            vend = probe["vendor"]

            if vend and len(vend) >= 1: # has a vendor
               stat += "\n" + "v:%s" % vend[0:15] # limit vendor to 15 chars

            stat += "\n" + "mac:%s" % probe["mac"] # full mac address

        self.pr_status = stat

    def on_unload(self, ui):
        self.running = False
        with ui._lock:
            try:
                ui.remove_element("pr_status")
                if self.log_results:
                    logging.info(f"[{self.__class__.__name__}] plugin unloaded")
            except Exception as e:
                logging.error(f"[{self.__class__.__name__}] unload: %s" % e)