mirror of
https://github.com/vanhoefm/fragattacks.git
synced 2024-11-25 00:38:24 -05:00
79 lines
2.0 KiB
Python
79 lines
2.0 KiB
Python
|
#!/usr/bin/python
|
||
|
|
|
||
|
|
import dbus
|
||
|
|
import sys, os
|
||
|
|
import time
|
||
|
|
import gobject
|
||
|
|
from dbus.mainloop.glib import DBusGMainLoop
|
||
|
|
|
||
|
|
WPAS_DBUS_SERVICE = "fi.w1.wpa_supplicant1"
|
||
|
|
WPAS_DBUS_INTERFACE = "fi.w1.wpa_supplicant1"
|
||
|
|
WPAS_DBUS_OPATH = "/fi/w1/wpa_supplicant1"
|
||
|
|
|
||
|
|
WPAS_DBUS_INTERFACES_INTERFACE = "fi.w1.wpa_supplicant1.Interface"
|
||
|
|
WPAS_DBUS_WPS_INTERFACE = "fi.w1.wpa_supplicant1.Interface.WPS"
|
||
|
|
|
||
|
|
def stateChanged(newState, oldState):
|
||
|
|
print "StateChanged(%s -> %s)" % (oldState, newState)
|
||
|
|
|
||
|
|
def scanDone(success):
|
||
|
|
print "Scan done: success=%s" % success
|
||
|
|
|
||
|
|
def bssAdded(bss, properties):
|
||
|
|
print "BSS added: %s" % (bss)
|
||
|
|
|
||
|
|
def bssRemoved(bss):
|
||
|
|
print "BSS removed: %s" % (bss)
|
||
|
|
|
||
|
|
def wpsEvent(name, args):
|
||
|
|
print "WPS event: %s" % (name)
|
||
|
|
print args
|
||
|
|
|
||
|
|
def credentials(cred):
|
||
|
|
print "WPS credentials: %s" % (cred)
|
||
|
|
|
||
|
|
def main():
|
||
|
|
dbus.mainloop.glib.DBusGMainLoop(set_as_default=True)
|
||
|
|
global bus
|
||
|
|
bus = dbus.SystemBus()
|
||
|
|
wpas_obj = bus.get_object(WPAS_DBUS_SERVICE, WPAS_DBUS_OPATH)
|
||
|
|
|
||
|
|
if len(sys.argv) != 2:
|
||
|
|
print "Missing ifname argument"
|
||
|
|
os._exit(1)
|
||
|
|
|
||
|
|
wpas = dbus.Interface(wpas_obj, WPAS_DBUS_INTERFACE)
|
||
|
|
bus.add_signal_receiver(scanDone,
|
||
|
|
dbus_interface=WPAS_DBUS_INTERFACES_INTERFACE,
|
||
|
|
signal_name="ScanDone")
|
||
|
|
bus.add_signal_receiver(bssAdded,
|
||
|
|
dbus_interface=WPAS_DBUS_INTERFACES_INTERFACE,
|
||
|
|
signal_name="BSSAdded")
|
||
|
|
bus.add_signal_receiver(bssRemoved,
|
||
|
|
dbus_interface=WPAS_DBUS_INTERFACES_INTERFACE,
|
||
|
|
signal_name="BSSRemoved")
|
||
|
|
bus.add_signal_receiver(stateChanged,
|
||
|
|
dbus_interface=WPAS_DBUS_INTERFACES_INTERFACE,
|
||
|
|
signal_name="StateChanged")
|
||
|
|
bus.add_signal_receiver(wpsEvent,
|
||
|
|
dbus_interface=WPAS_DBUS_WPS_INTERFACE,
|
||
|
|
signal_name="Event")
|
||
|
|
bus.add_signal_receiver(credentials,
|
||
|
|
dbus_interface=WPAS_DBUS_WPS_INTERFACE,
|
||
|
|
signal_name="Credentials")
|
||
|
|
|
||
|
|
ifname = sys.argv[1]
|
||
|
|
|
||
|
|
path = wpas.GetInterface(ifname)
|
||
|
|
global if_obj
|
||
|
|
if_obj = bus.get_object(WPAS_DBUS_SERVICE, path)
|
||
|
|
global iface
|
||
|
|
wps = dbus.Interface(if_obj, WPAS_DBUS_WPS_INTERFACE)
|
||
|
|
wps.Start({'Role': 'enrollee', 'Type': 'pbc'})
|
||
|
|
|
||
|
|
gobject.MainLoop().run()
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
main()
|
||
|
|
|