diff --git a/pwnagotchi/agent.py b/pwnagotchi/agent.py index 00d4f964..5916ce05 100644 --- a/pwnagotchi/agent.py +++ b/pwnagotchi/agent.py @@ -21,11 +21,12 @@ RECOVERY_DATA_FILE = '/root/.pwnagotchi-recovery' class Agent(Client, Automata, AsyncAdvertiser, AsyncTrainer): def __init__(self, view, config, keypair): - Client.__init__(self, config['bettercap']['hostname'], - config['bettercap']['scheme'], - config['bettercap']['port'], - config['bettercap']['username'], - config['bettercap']['password']) + Client.__init__(self, + "127.0.0.1" if "hostname" not in config['bettercap'] else config['bettercap']['hostname'], + "http" if "scheme" not in config['bettercap'] else config['bettercap']['scheme'], + 8081 if "port" not in config['bettercap'] else config['bettercap']['port'], + "pwnagotchi" if "username" not in config['bettercap'] else config['bettercap']['username'], + "pwnagotchi" if "password" not in config['bettercap'] else config['bettercap']['password']) Automata.__init__(self, config, view) AsyncAdvertiser.__init__(self, config, view, keypair) AsyncTrainer.__init__(self, config) @@ -178,7 +179,8 @@ class Agent(Client, Automata, AsyncAdvertiser, AsyncTrainer): for ap in s['wifi']['aps']: if ap['encryption'] == '' or ap['encryption'] == 'OPEN': continue - elif ap['hostname'] in whitelist or ap['mac'][:13].lower() in whitelist or ap['mac'].lower() in whitelist: + elif ap['hostname'] in whitelist or ap['mac'][:13].lower() in whitelist or ap[ + 'mac'].lower() in whitelist: continue else: aps.append(ap) @@ -362,7 +364,8 @@ class Agent(Client, Automata, AsyncAdvertiser, AsyncTrainer): plugins.on('handshake', self, filename, ap_mac, sta_mac) else: (ap, sta) = ap_and_station - self._last_pwnd = ap['hostname'] if ap['hostname'] != '' and ap['hostname'] != '' else ap_mac + self._last_pwnd = ap['hostname'] if ap['hostname'] != '' and ap[ + 'hostname'] != '' else ap_mac logging.warning( "!!! captured new handshake on channel %d, %d dBm: %s (%s) -> %s [%s (%s)] !!!", ap['channel'], ap['rssi'], sta['mac'], sta['vendor'], ap['hostname'], ap['mac'], ap['vendor']) diff --git a/pwnagotchi/defaults.toml b/pwnagotchi/defaults.toml index 3f7b0534..b175fff1 100644 --- a/pwnagotchi/defaults.toml +++ b/pwnagotchi/defaults.toml @@ -198,11 +198,6 @@ ui.display.enabled = false ui.display.rotation = 180 ui.display.type = "waveshare_4" -bettercap.scheme = "http" -bettercap.hostname = "localhost" -bettercap.port = 8081 -bettercap.username = "pwnagotchi" -bettercap.password = "pwnagotchi" bettercap.handshakes = "/root/handshakes" bettercap.silence = [ "ble.device.new",