wpa_supplicant/examples/wpas-dbus-new.py

   1 #!/usr/bin/python
   2
   3 import dbus
   4 import sys, os
   5 import time
   6 import gobject
   7 from dbus.mainloop.glib import DBusGMainLoop
   8
   9 WPAS_DBUS_SERVICE = "fi.w1.wpa_supplicant1"
  10 WPAS_DBUS_INTERFACE = "fi.w1.wpa_supplicant1"
  11 WPAS_DBUS_OPATH = "/fi/w1/wpa_supplicant1"
  12
  13 WPAS_DBUS_INTERFACES_INTERFACE = "fi.w1.wpa_supplicant1.Interface"
  14 WPAS_DBUS_INTERFACES_OPATH = "/fi/w1/wpa_supplicant1/Interfaces"
  15 WPAS_DBUS_BSS_INTERFACE = "fi.w1.wpa_supplicant1.Interface.BSS"
  16
  17 def byte_array_to_string(s):
  18         import urllib
  19         r = ""
  20         for c in s:
  21                 if c >= 32 and c < 127:
  22                         r += "%c" % c
  23                 else:
  24                         r += urllib.quote(chr(c))
  25         return r
  26
  27 def list_interfaces(wpas_obj):
  28         ifaces = wpas_obj.Interfaces
  29         for i in ifaces:
  30                 print "%s" (i)
  31
  32 def stateChanged(newState, oldState):
  33         print "StateChanged(%s -> %s)" % (oldState, newState)
  34
  35 def showBss(bss):
  36         net_obj = bus.get_object(WPAS_DBUS_SERVICE, bss)
  37         net = dbus.Interface(net_obj, WPAS_DBUS_BSS_INTERFACE)
  38
  39         # Convert the byte-array for SSID and BSSID to printable strings
  40         val = net_obj.Get(WPAS_DBUS_BSS_INTERFACE, 'BSSID',
  41                           dbus_interface=dbus.PROPERTIES_IFACE)
  42         bssid = ""
  43         for item in val:
  44                 bssid = bssid + ":%02x" % item
  45         bssid = bssid[1:]
  46         val = net_obj.Get(WPAS_DBUS_BSS_INTERFACE, 'SSID',
  47                           dbus_interface=dbus.PROPERTIES_IFACE)
  48         ssid = byte_array_to_string(val)
  49
  50         val = net_obj.Get(WPAS_DBUS_BSS_INTERFACE, 'WPAIE',
  51                           dbus_interface=dbus.PROPERTIES_IFACE)
  52         wpa = "no"
  53         if val != None:
  54                 wpa = "yes"
  55         val = net_obj.Get(WPAS_DBUS_BSS_INTERFACE, 'RSNIE',
  56                           dbus_interface=dbus.PROPERTIES_IFACE)
  57         wpa2 = "no"
  58         if val != None:
  59                 wpa2 = "yes"
  60         freq = net_obj.Get(WPAS_DBUS_BSS_INTERFACE, 'Frequency',
  61                            dbus_interface=dbus.PROPERTIES_IFACE)
  62         signal = net_obj.Get(WPAS_DBUS_BSS_INTERFACE, 'Signal',
  63                              dbus_interface=dbus.PROPERTIES_IFACE)
  64         val = net_obj.Get(WPAS_DBUS_BSS_INTERFACE, 'MaxRate',
  65                           dbus_interface=dbus.PROPERTIES_IFACE)
  66         maxrate = val / 1000000
  67
  68         print "  %s  ::  ssid='%s'  wpa=%s  wpa2=%s  signal=%d  rate=%d  freq=%d" % (bssid, ssid, wpa, wpa2, signal, maxrate, freq)
  69
  70 def scanDone(success):
  71         gobject.MainLoop().quit()
  72         print "Scan done: success=%s" % success
  73
  74         res = if_obj.Get(WPAS_DBUS_INTERFACES_INTERFACE, 'BSSs',
  75                          dbus_interface=dbus.PROPERTIES_IFACE)
  76
  77         print "Scanned wireless networks:"
  78         for opath in res:
  79                 print opath
  80                 showBss(opath)
  81
  82 def bssAdded(bss, properties):
  83         print "BSS added: %s" % (bss)
  84         showBss(bss)
  85
  86 def bssRemoved(bss):
  87         print "BSS removed: %s" % (bss)
  88
  89 def main():
  90         dbus.mainloop.glib.DBusGMainLoop(set_as_default=True)
  91         global bus
  92         bus = dbus.SystemBus()
  93         wpas_obj = bus.get_object(WPAS_DBUS_SERVICE, WPAS_DBUS_OPATH)
  94         wpas = dbus.Interface(wpas_obj, WPAS_DBUS_INTERFACE)
  95         bus.add_signal_receiver(scanDone,
  96                                 dbus_interface=WPAS_DBUS_INTERFACES_INTERFACE,
  97                                 signal_name="ScanDone")
  98         bus.add_signal_receiver(bssAdded,
  99                                 dbus_interface=WPAS_DBUS_INTERFACES_INTERFACE,
 100                                 signal_name="BSSAdded")
 101         bus.add_signal_receiver(bssRemoved,
 102                                 dbus_interface=WPAS_DBUS_INTERFACES_INTERFACE,
 103                                 signal_name="BSSRemoved")
 104         bus.add_signal_receiver(stateChanged,
 105                                 dbus_interface=WPAS_DBUS_INTERFACES_INTERFACE,
 106                                 signal_name="StateChanged")
 107
 108         if len(sys.argv) != 2:
 109                 list_interfaces(wpas_obj)
 110                 os._exit(1)
 111
 112         ifname = sys.argv[1]
 113
 114         # See if wpa_supplicant already knows about this interface
 115         path = None
 116         try:
 117                 path = wpas.GetInterface(ifname)
 118         except dbus.DBusException, exc:
 119                 if not str(exc).startswith("fi.w1.wpa_supplicant1.InterfaceUnknown:"):
 120                         raise exc
 121                 try:
 122                         path = wpas.CreateInterface({'Ifname': ifname, 'Driver': 'test'})
 123                         time.sleep(1)
 124
 125                 except dbus.DBusException, exc:
 126                         if not str(exc).startswith("fi.w1.wpa_supplicant1.InterfaceExists:"):
 127                                 raise exc
 128
 129         global if_obj
 130         if_obj = bus.get_object(WPAS_DBUS_SERVICE, path)
 131         global iface
 132         iface = dbus.Interface(if_obj, WPAS_DBUS_INTERFACES_INTERFACE)
 133         iface.Scan({'Type': 'active'})
 134
 135         gobject.MainLoop().run()
 136
 137         wpas.RemoveInterface(dbus.ObjectPath(path))
 138
 139 if __name__ == "__main__":
 140         main()
 141