diff options
Diffstat (limited to 'src')
-rwxr-xr-x | src/connections/wireless-dbus | 137 |
1 files changed, 137 insertions, 0 deletions
diff --git a/src/connections/wireless-dbus b/src/connections/wireless-dbus new file mode 100755 index 0000000..d954577 --- /dev/null +++ b/src/connections/wireless-dbus @@ -0,0 +1,137 @@ +#! /usr/bin/env python +import sys +import dbus +import shlex +import subprocess +from signal import SIGTERM +from os import kill +from time import sleep + +# dbus constants. +WPAS_DBUS_SERVICE = "fi.epitest.hostap.WPASupplicant" +WPAS_DBUS_INTERFACE = "fi.epitest.hostap.WPASupplicant" +WPAS_DBUS_OPATH = "/fi/epitest/hostap/WPASupplicant" + +WPAS_DBUS_INTERFACES_INTERFACE = "fi.epitest.hostap.WPASupplicant.Interface" +WPAS_DBUS_INTERFACES_OPATH = "/fi/epitest/hostap/WPASupplicant/Interfaces" +WPAS_DBUS_BSSID_INTERFACE = "fi.epitest.hostap.WPASupplicant.BSSID" +WPAS_DBUS_NETWORKS_INTERFACE = "fi.epitest.hostap.WPASupplicant.Network" + +def read_config(config): + cfg = shlex.split(open(config, "r").read()) + options = {} + for line in cfg: + (var, delim, value) = line.partition('=') + if not delim: + raise SyntaxError, 'Bad configuration file' + options[var] = value + return options + +def start(profile): + # TODO: Add check if it's even a wireless interface + # Interface up... + try: + subprocess.check_call(["ifconfig", profile['INTERFACE'], "up"]) + except CalledProcessError: + print " - Could not bring interface up" + return False + + # Manually set any iwconfig options, ignoring errors + if profile.has_key('IWCONFIG'): + subprocess.call(["iwconfig", profile['INTERFACE'], profile['IWCONFIG']]) + + + # Base arguments + args=["wpa_supplicant", "-Bu", "-P/var/run/wpa_supplicant.pid"] + + try: + args.append(profile['WPA_OPTS']) + except KeyError: + args.append("-Dwext") + + if profile['SECURITY'] == "wpa-config": + try: + args.append("-c" + profile["WPA_CONF"]) + except KeyError: + args.append("-c/etc/wpa_supplicant.conf") + elif not profile['SECURITY'] in ['yes', 'wpa', 'wep']: + print " - Invalid security chosen" + return False + + # Start wpa_supplicant + supplicant = subprocess.Popen(args,stderr=subprocess.STDOUT,stdout=subprocess.PIPE) + output = supplicant.communicate()[0] + if supplicant.returncode not in [255,0]: + print output + print " - Could not start wpa_supplicant" + return False + + # Connect to wpa_supplicant + bus = dbus.SystemBus() + wpas_obj = bus.get_object(WPAS_DBUS_SERVICE, WPAS_DBUS_OPATH) + wpas = dbus.Interface(wpas_obj, WPAS_DBUS_INTERFACE) + + # Add/Get interface path + try: + driver=profile["WPA_DRIVER"] + except KeyError: + driver="wext" + + try: + path = wpas.getInterface("ipw0") + except dbus.exceptions.DBusException: + path = wpas.addInterface("ipw0", {"driver":dbus.String(driver,variant_level=1)}) + + # Get interface object + if_obj = bus.get_object(WPAS_DBUS_SERVICE, path) + iface = dbus.Interface(if_obj, WPAS_DBUS_INTERFACES_INTERFACE); + + # Connect + if profile['SECURITY'] in ['wpa','yes','wep']: + # Add+select the network + path = iface.addNetwork() + net_obj = bus.get_object(WPAS_DBUS_SERVICE, path) + rnet = dbus.Interface(net_obj, WPAS_DBUS_NETWORKS_INTERFACE) + iface.selectNetwork(rnet) + # Set the options + opts = dbus.Dictionary({"ssid": dbus.ByteArray(profile['ESSID']), "psk": dbus.String(profile['KEY'])}, signature="sv") + rnet.set(opts) + + # Determine timeout + try: + timeout = profile["TIMEOUT"] + except KeyError: + timeout = 15 + + # Check for association + n=0 + while n <= timeout: + n+=1 + sleep(1) + state = iface.state() + if state == "COMPLETED": + break + + if n == timeout: + print " - Association/Authentication failed:", state + return False + + # Run ethernet and get an ip. + try: + subprocess.check_call(["/usr/lib/network/connections/ethernet", "up", sys.argv[2]]) + except CalledProcessError: + return False + return True + +def stop(profile): + subprocess.call(["/usr/lib/network/connections/ethernet", "down", sys.argv[2]]) + kill(int(open("/var/run/wpa_supplicant.pid").read()),SIGTERM) + return True + +if __name__ == "__main__": + profile = read_config("/etc/network.d/"+sys.argv[2]) + + if sys.argv[1] == "up": + sys.exit(not start(profile)) + elif sys.argv[1] == "down": + sys.exit(not stop(profile)) |