mirror of
https://github.com/jayofelony/pwnagotchi.git
synced 2025-07-01 18:37:27 -04:00
moved BtS to plugins repo, edited defaults.toml
Signed-off-by: Jeroen Oudshoorn <oudshoorn.jeroen@gmail.com>
This commit is contained in:
Jeroen Oudshoorn
@ -57,13 +57,6 @@ main.plugins.gps.enabled = false
|
||||
main.plugins.gps.speed = 19200
|
||||
main.plugins.gps.device = "/dev/ttyUSB0"
|
||||
|
||||
main.plugins.bluetoothsniffer.enabled = false
|
||||
main.plugins.bluetoothsniffer.timer = 45 # On how may seconds to scan for bluetooth devices
|
||||
main.plugins.bluetoothsniffer.devices_file = "/root/handshakes/bluetooth_devices.json" # Path to the JSON file with bluetooth devices
|
||||
main.plugins.bluetoothsniffer.count_interval = 86400 # On how may seconds to update count bluetooth devices
|
||||
main.plugins.bluetoothsniffer.bt_x_coord = 160
|
||||
main.plugins.bluetoothsniffer.bt_y_coord = 66
|
||||
|
||||
main.plugins.webgpsmap.enabled = false
|
||||
|
||||
main.plugins.onlinehashcrack.enabled = false
|
||||
|
||||
@ -1,185 +0,0 @@
|
||||
import logging
|
||||
import os
|
||||
import subprocess
|
||||
import json
|
||||
import time
|
||||
|
||||
import pwnagotchi.plugins as plugins
|
||||
import pwnagotchi.ui.fonts as fonts
|
||||
from pwnagotchi.ui.components import LabeledValue
|
||||
from pwnagotchi.ui.view import BLACK
|
||||
from datetime import datetime
|
||||
|
||||
|
||||
class BluetoothSniffer(plugins.Plugin):
|
||||
__author__ = 'diytechtinker, fixed by Jayofelony'
|
||||
__version__ = '0.1.3'
|
||||
__license__ = 'GPL3'
|
||||
__description__ = 'A plugin that sniffs Bluetooth devices and saves their MAC addresses, name and counts to a JSON file'
|
||||
|
||||
def __init__(self):
|
||||
# Defining the instance variables
|
||||
self.options = {
|
||||
'timer': 45,
|
||||
'devices_file': '/root/handshakes/bluetooth_devices.json',
|
||||
'count_interval': 86400,
|
||||
'bt_x_coord': 160,
|
||||
'bt_y_coord': 66
|
||||
}
|
||||
self.data = {}
|
||||
self.last_scan_time = 0
|
||||
|
||||
def on_loaded(self):
|
||||
logging.info("[BtS] bluetoothsniffer plugin loaded.")
|
||||
logging.info("[BtS] Bluetooth devices file location: %s", self.options['devices_file'])
|
||||
# Creating the device file path if it does not exist
|
||||
if not os.path.exists(os.path.dirname(self.options['devices_file'])):
|
||||
os.makedirs(os.path.dirname(self.options['devices_file']))
|
||||
|
||||
# Creating the device file if it does not exist
|
||||
if not os.path.exists(self.options['devices_file']):
|
||||
with open(self.options['devices_file'], 'w') as f:
|
||||
json.dump({}, f)
|
||||
|
||||
# Loading the data from the device file
|
||||
with open(self.options['devices_file'], 'r') as f:
|
||||
self.data = json.load(f)
|
||||
|
||||
def on_ui_setup(self, ui):
|
||||
with ui._lock:
|
||||
ui.add_element('BtS', LabeledValue(color=BLACK,
|
||||
label='BT SNFD',
|
||||
value=" ",
|
||||
position=(int(self.options["bt_x_coord"]),
|
||||
int(self.options["bt_y_coord"])),
|
||||
label_font=fonts.Small,
|
||||
text_font=fonts.Small))
|
||||
|
||||
def on_unload(self, ui):
|
||||
with ui._lock:
|
||||
ui.remove_element('BtS')
|
||||
|
||||
def on_ui_update(self, ui):
|
||||
current_time = time.time()
|
||||
# Checking the time elapsed since last scan
|
||||
if current_time - self.last_scan_time >= self.options['timer']:
|
||||
self.last_scan_time = current_time
|
||||
# logging.info("[BtS] Bluetooth sniffed: %s", str(self.bt_sniff_info()))
|
||||
ui.set('BtS', str(self.bt_sniff_info()))
|
||||
self.scan(ui)
|
||||
|
||||
# Method for scanning the nearby bluetooth devices
|
||||
def scan(self, display):
|
||||
logging.info("[BtS] Scanning for bluetooth devices...")
|
||||
current_time = time.time()
|
||||
changed = False
|
||||
device_class = None
|
||||
|
||||
# Running the system command hcitool to scan nearby bluetooth devices
|
||||
cmd_inq = "hcitool inq --flush"
|
||||
try:
|
||||
inq_output = subprocess.check_output(cmd_inq.split())
|
||||
for line in inq_output.splitlines()[1:]:
|
||||
fields = line.split()
|
||||
mac_address = fields[0].decode()
|
||||
for i in range(len(fields)):
|
||||
if fields[i].decode() == "class:" and i + 1 < len(fields):
|
||||
device_class = fields[i + 1].decode()
|
||||
logging.info("[BtS] Found bluetooth %s", mac_address)
|
||||
|
||||
# Update the count, first_seen, and last_seen time of the device
|
||||
if mac_address in self.data and len(self.data) > 0:
|
||||
if 'Unknown' == self.data[mac_address]['name']:
|
||||
name = self.get_device_name(mac_address)
|
||||
self.data[mac_address]['name'] = name
|
||||
self.data[mac_address]['new_info'] = 2
|
||||
logging.info("[BtS] Updated bluetooth name: %s", name)
|
||||
changed = True
|
||||
|
||||
if 'Unknown' == self.data[mac_address]['manufacturer']:
|
||||
manufacturer = self.get_device_manufacturer(mac_address)
|
||||
self.data[mac_address]['manufacturer'] = manufacturer
|
||||
self.data[mac_address]['new_info'] = 2
|
||||
logging.info("[BtS] Updated bluetooth manufacturer: %s", manufacturer)
|
||||
changed = True
|
||||
|
||||
if device_class != self.data[mac_address]['class']:
|
||||
self.data[mac_address]['class'] = device_class
|
||||
self.data[mac_address]['new_info'] = 2
|
||||
logging.info("[BtS] Updated bluetooth class: %s", device_class)
|
||||
changed = True
|
||||
|
||||
last_seen_time = int(
|
||||
datetime.strptime(self.data[mac_address]['last_seen'], '%H:%M:%S %d-%m-%Y').timestamp())
|
||||
if current_time - last_seen_time >= self.options['count_interval']:
|
||||
self.data[mac_address]['count'] += 1
|
||||
self.data[mac_address]['last_seen'] = time.strftime('%H:%M:%S %d-%m-%Y',
|
||||
time.localtime(current_time))
|
||||
self.data[mac_address]['new_info'] = 2
|
||||
logging.info("[BtS] Updated bluetooth count.")
|
||||
changed = True
|
||||
else:
|
||||
name = self.get_device_name(mac_address)
|
||||
manufacturer = self.get_device_manufacturer(mac_address)
|
||||
self.data[mac_address] = {'name': name, 'count': 1, 'class': device_class,
|
||||
'manufacturer': manufacturer,
|
||||
'first_seen': time.strftime('%H:%M:%S %d-%m-%Y',
|
||||
time.localtime(current_time)),
|
||||
'last_seen': time.strftime('%H:%M:%S %d-%m-%Y',
|
||||
time.localtime(current_time)),
|
||||
'new_info': True}
|
||||
logging.info("[BtS] Added new bluetooth device %s with MAC: %s", name, mac_address)
|
||||
changed = True
|
||||
|
||||
except subprocess.CalledProcessError as e:
|
||||
logging.error("[BtS] Error running command: %s", e)
|
||||
|
||||
# Save the updated devices to the JSON file
|
||||
if changed:
|
||||
with open(self.options['devices_file'], 'w') as f:
|
||||
logging.info("[BtS] Saving bluetooths %s into json.", name)
|
||||
json.dump(self.data, f)
|
||||
display.set('status', 'Bluetooth sniffed and stored!').update(force=True)
|
||||
|
||||
# Method to get the device name
|
||||
def get_device_name(self, mac_address):
|
||||
logging.info("[BtS] Trying to get name for %s", mac_address)
|
||||
name = 'Unknown'
|
||||
hcitool_process = subprocess.Popen(["hcitool", "name", mac_address], stdout=subprocess.PIPE)
|
||||
output, error = hcitool_process.communicate()
|
||||
if output.decode().strip() != '':
|
||||
name = output.decode().strip()
|
||||
logging.info("[BtS] Got name %s for %s", name, mac_address)
|
||||
return name
|
||||
|
||||
# Method to get the device manufacturer
|
||||
def get_device_manufacturer(self, mac_address):
|
||||
manufacturer = 'Unknown'
|
||||
cmd_info = f"hcitool info {mac_address} | grep 'Manufacturer:' | cut -d ' ' -f 2-"
|
||||
try:
|
||||
logging.info("[BtS] Trying to get manufacturer for %s", mac_address)
|
||||
start_time = time.time()
|
||||
process = subprocess.Popen(cmd_info, stdout=subprocess.PIPE, stderr=subprocess.PIPE, shell=True)
|
||||
while process.poll() is None:
|
||||
time.sleep(0.1)
|
||||
if time.time() - start_time > 7:
|
||||
logging.info("[BtS] Timeout while trying to get manufacturer for %s", mac_address)
|
||||
process.kill()
|
||||
return manufacturer
|
||||
output, error = process.communicate(timeout=1)
|
||||
if output.decode().strip() != '':
|
||||
manufacturer = output.decode().strip()
|
||||
logging.info("[BtS] Got manufacturer %s for %s", manufacturer, mac_address)
|
||||
except Exception as e:
|
||||
logging.info("[BtS] Error while trying to get manufacturer for %s: %s", mac_address, str(e))
|
||||
return manufacturer
|
||||
|
||||
def bt_sniff_info(self):
|
||||
num_devices = len(self.data)
|
||||
if num_devices > 0:
|
||||
num_unknown = sum(1 for device in self.data.values() if device['name'] == 'Unknown' or device['manufacturer'] == 'Unknown')
|
||||
num_known = num_devices - num_unknown
|
||||
return_text = "%s|%s" % (num_devices, num_known)
|
||||
else:
|
||||
return_text = "0|0"
|
||||
return return_text
|
||||
Reference in New Issue
Block a user