wpas-dbus-new.py 4.1 KB

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