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