chrissewell/pwnagotchi
1
0
Fork 0
mirror of https://github.com/jayofelony/pwnagotchi.git synced 2025-07-01 18:37:27 -04:00
Code Issues Packages Projects Releases Wiki Activity

changes to support plugins requesting parts of session, and get notice of events

Browse Source
This commit is contained in:
Sniffleupagus
2023-04-04 14:11:23 -07:00
parent 544314b88c
commit 70bffe5200
2 changed files with 10 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
pwnagotchi/agent.py
View File

@ -324,6 +324,12 @@ class Agent(Client, Automata, AsyncAdvertiser, AsyncTrainer):
found_handshake = False found_handshake = False
jmsg = json.loads(msg) jmsg = json.loads(msg)
# give plugins access to the events
try:
plugins.on('bcap_%s' % re.sub(r"[^a-z0-9_]+", "_", jmsg['tag'].lower()), self, jmsg)
except Exception as err:
logging.error("Processing event: %s" % err)
if jmsg['tag'] == 'wifi.client.handshake': if jmsg['tag'] == 'wifi.client.handshake':
filename = jmsg['data']['file'] filename = jmsg['data']['file']
sta_mac = jmsg['data']['station'] sta_mac = jmsg['data']['station']

6
pwnagotchi/bettercap.py
View File

@ -31,8 +31,10 @@ class Client(object):
self.websocket = "ws://%s:%s@%s:%d/api" % (username, password, hostname, port) self.websocket = "ws://%s:%s@%s:%d/api" % (username, password, hostname, port)
self.auth = HTTPBasicAuth(username, password) self.auth = HTTPBasicAuth(username, password)
def session(self): # session takes optional argument to pull a sub-dictionary
r = requests.get("%s/session" % self.url, auth=self.auth) # ex.: "session/wifi", "session/ble"
def session(self, sess="session"):
r = requests.get("%s/%s" % (self.url, sess), auth=self.auth)
return decode(r) return decode(r)
async def start_websocket(self, consumer): async def start_websocket(self, consumer):