Some work on the network devices

1 files changed, 467 insertions, 0 deletions

diff --git a/shell/hardware/nmclient.py b/shell/hardware/nmclient.py
new file mode 100644
index 0000000..9a2c5a8
--- /dev/null
+++ b/shell/hardware/nmclient.py
@@ -0,0 +1,467 @@
+#
+# Copyright (C) 2006, Red Hat, Inc.
+#
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation; either version 2 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program; if not, write to the Free Software
+# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
+
+import logging
+import os
+
+import dbus
+import dbus.glib
+import dbus.decorators
+import gobject
+
+from hardware import nminfo
+
+IW_AUTH_ALG_OPEN_SYSTEM = 0x00000001
+IW_AUTH_ALG_SHARED_KEY  = 0x00000002
+
+NM_DEVICE_STAGE_STRINGS=("Unknown",
+    "Prepare",
+    "Config",
+    "Need Users Key",
+    "IP Config",
+    "IP Config Get",
+    "IP Config Commit",
+    "Activated",
+    "Failed",
+    "Cancled"
+)
+
+NM_SERVICE = 'org.freedesktop.NetworkManager'
+NM_IFACE = 'org.freedesktop.NetworkManager'
+NM_IFACE_DEVICES = 'org.freedesktop.NetworkManager.Devices'
+NM_PATH = '/org/freedesktop/NetworkManager'
+
+DEVICE_TYPE_UNKNOWN = 0
+DEVICE_TYPE_802_3_ETHERNET = 1
+DEVICE_TYPE_802_11_WIRELESS = 2
+
+NM_DEVICE_CAP_NONE = 0x00000000
+NM_DEVICE_CAP_NM_SUPPORTED = 0x00000001
+NM_DEVICE_CAP_CARRIER_DETECT = 0x00000002
+NM_DEVICE_CAP_WIRELESS_SCAN = 0x00000004
+
+sys_bus = dbus.SystemBus()
+
+NM_802_11_CAP_NONE            = 0x00000000
+NM_802_11_CAP_PROTO_NONE      = 0x00000001
+NM_802_11_CAP_PROTO_WEP       = 0x00000002
+NM_802_11_CAP_PROTO_WPA       = 0x00000004
+NM_802_11_CAP_PROTO_WPA2      = 0x00000008
+NM_802_11_CAP_KEY_MGMT_PSK    = 0x00000040
+NM_802_11_CAP_KEY_MGMT_802_1X = 0x00000080
+NM_802_11_CAP_CIPHER_WEP40    = 0x00001000
+NM_802_11_CAP_CIPHER_WEP104   = 0x00002000
+NM_802_11_CAP_CIPHER_TKIP     = 0x00004000
+NM_802_11_CAP_CIPHER_CCMP     = 0x00008000
+
+
+class Network(gobject.GObject):
+    __gsignals__ = {
+        'init-failed': (gobject.SIGNAL_RUN_FIRST, gobject.TYPE_NONE, ([]))
+    }
+
+    def __init__(self, op):
+        gobject.GObject.__init__(self)
+        self._op = op
+        self._ssid = None
+        self._mode = None
+        self._strength = 0
+        self._valid = False
+
+        obj = sys_bus.get_object(NM_SERVICE, self._op)
+        net = dbus.Interface(obj, NM_IFACE_DEVICES)
+        net.getProperties(reply_handler=self._update_reply_cb,
+                error_handler=self._update_error_cb)
+
+    def _update_reply_cb(self, *props):
+        self._ssid = props[1]
+        self._strength = props[3]
+        self._mode = props[6]
+        caps = props[7]
+        if caps & NM_802_11_CAP_PROTO_WPA or caps & NM_802_11_CAP_PROTO_WPA2:
+            # We do not support WPA at this time, so don't show
+            # WPA-enabled access points in the menu
+            logging.debug("Net(%s): ssid '%s' dropping because WPA[2] unsupported" % (self._op,
+                    self._ssid))
+            self._valid = False
+            self.emit('init-failed')
+        else:
+            self._valid = True
+            logging.debug("Net(%s): ssid '%s', mode %d, strength %d" % (self._op,
+                    self._ssid, self._mode, self._strength))
+
+    def _update_error_cb(self, err):
+        logging.debug("Net(%s): failed to update. (%s)" % (self._op, err))
+        self._valid = False
+        self.emit('init-failed')
+
+    def get_ssid(self):
+        return self._ssid
+
+    def get_op(self):
+        return self._op
+
+    def get_strength(self):
+        return self._strength
+
+    def set_strength(self, strength):
+        self._strength = strength
+
+    def is_valid(self):
+        return self._valid
+
+class Device(gobject.GObject):
+    __gsignals__ = {
+        'init-failed': (gobject.SIGNAL_RUN_FIRST, gobject.TYPE_NONE, ([])),
+        'activated': (gobject.SIGNAL_RUN_FIRST, gobject.TYPE_NONE, ([])),
+        'strength-changed': (gobject.SIGNAL_RUN_FIRST, gobject.TYPE_NONE,
+                            ([gobject.TYPE_PYOBJECT]))
+    }
+
+    def __init__(self, op):
+        gobject.GObject.__init__(self)
+        self._op = op
+        self._iface = None
+        self._type = DEVICE_TYPE_UNKNOWN
+        self._udi = None
+        self._active = False
+        self._strength = 0
+        self._link = False
+        self._valid = False
+        self._networks = {}
+        self._active_net = None
+        self._caps = 0
+
+        obj = sys_bus.get_object(NM_SERVICE, self._op)
+        dev = dbus.Interface(obj, NM_IFACE_DEVICES)
+        dev.getProperties(reply_handler=self._update_reply_cb,
+                error_handler=self._update_error_cb)
+
+    def _update_reply_cb(self, *props):
+        self._iface = props[1]
+        self._type = props[2]
+        self._udi = props[3]
+        self._active = props[4]
+        if self._active:
+            self.emit('activated')
+        self._link = props[15]
+        self._caps = props[17]
+
+        if self._type == DEVICE_TYPE_802_11_WIRELESS:
+            old_strength = self._strength
+            self._strength = props[14]
+            if self._strength != old_strength:
+                self.emit('strength-changed', self._strength)
+            self._update_networks(props[20], props[19])
+
+        self._valid = True
+
+    def _update_networks(self, net_ops, active_op):
+        for op in net_ops:
+            net = Network(op)
+            self._networks[op] = net
+            net.connect('init-failed', self._net_init_failed)
+            if op == active_op:
+                self._active_net = op
+
+    def _update_error_cb(self, err):
+        logging.debug("Device(%s): failed to update. (%s)" % (self._op, err))
+        self._valid = False
+        self.emit('init-failed')
+
+    def _net_init_failed(self, net):
+        net_op = net.get_op()
+        if not self._networks.has_key(net_op):
+            return
+        if net_op == self._active_net:
+            self._active_net = None
+        del self._networks[net_op]
+
+    def get_op(self):
+        return self._op
+
+    def get_network(self, op):
+        if self._networks.has_key(op):
+            return self._networks[op]
+        return None
+
+    def get_network_ops(self):
+        return self._networks.keys()
+
+    def get_strength(self):
+        return self._strength
+
+    def set_strength(self, strength):
+        if strength == self._strength:
+            return False
+
+        if strength >= 0 and strength <= 100:
+            self._strength = strength
+        else:
+            self._strength = 0
+
+        self.emit('strength-changed', self._strength)
+
+    def network_appeared(self, network):
+        if self._networks.has_key(network):
+            return
+        net = Network(network)
+        self._networks[network] = net
+        net.connect('init-failed', self._net_init_failed)
+
+    def network_disappeared(self, network):
+        if not self._networks.has_key(network):
+            return
+        if network == self._active_net:
+            self._active_net = None
+        del self._networks[network]
+
+    def get_active(self):
+        return self._active
+
+    def set_active(self, active, ssid=None):
+        self._active = active
+        if self._type == DEVICE_TYPE_802_11_WIRELESS:
+            if not ssid:
+                self._active_net = None
+            else:
+                for (op, net) in self._networks.items():
+                    if net.get_ssid() == ssid:
+                        self._active_net = op
+
+    def get_type(self):
+        return self._type
+
+    def is_valid(self):
+        return self._valid
+
+    def set_carrier(self, on):
+        self._link = on
+
+    def get_capabilities(self):
+        return self._caps
+
+NM_STATE_UNKNOWN = 0
+NM_STATE_ASLEEP = 1
+NM_STATE_CONNECTING = 2
+NM_STATE_CONNECTED = 3
+NM_STATE_DISCONNECTED = 4
+
+class NMClient(object):
+    def __init__(self):
+        self.nminfo = None
+        self._nm_present = False
+        self._nm_state = NM_STATE_UNKNOWN
+        self._update_timer = 0
+        self._active_device = None
+        self._devices = {}
+
+        try:
+            self.nminfo = nminfo.NMInfo(self)
+        except RuntimeError:
+            pass
+        self._setup_dbus()
+        if self._nm_present:
+            self._get_nm_state()
+            self._get_initial_devices()
+
+    def get_devices(self):
+        return self._devices
+
+    def _get_nm_state(self):
+        # Grab NM's state
+        self._nm_obj.state(reply_handler=self._get_state_reply_cb, \
+                error_handler=self._get_state_error_cb)
+
+    def _get_state_reply_cb(self, state):
+        self._nm_state = state
+
+    def _get_state_error_cb(self, err):
+        logging.debug("Failed to get NetworkManager state! %s" % err)
+
+    def _get_initial_devices_reply_cb(self, ops):
+        for op in ops:
+            self._add_device(op)
+
+    def _dev_init_failed_cb(self, dev):
+        # Device failed to initialize, likely due to dbus errors or something
+        op = dev.get_op()
+        self._remove_device(op)
+
+    def _get_initial_devices_error_cb(self, err):
+        logging.debug("Error updating devices (%s)" % err)
+
+    def _get_initial_devices(self):
+        self._nm_obj.getDevices(reply_handler=self._get_initial_devices_reply_cb, \
+                error_handler=self._get_initial_devices_error_cb)
+
+    def _add_device(self, dev_op):
+        if self._devices.has_key(dev_op):
+            return
+        dev = Device(dev_op)
+        self._devices[dev_op] = dev
+        dev.connect('init-failed', self._dev_init_failed_cb)
+        dev.connect('activated', self._dev_activated_cb)
+        dev.connect('strength-changed', self._dev_strength_changed_cb)
+
+    def _remove_device(self, dev_op):
+        if not self._devices.has_key(dev_op):
+            return
+        if self._active_device == dev_op:
+            self._active_device = None
+        dev = self._devices[dev_op]
+        dev.disconnect('activated')
+        dev.disconnect('init-failed')
+        dev.disconnect('strength-changed')
+        del self._devices[dev_op]
+
+    def _dev_activated_cb(self, dev):
+        op = dev.get_op()
+        if not self._devices.has_key(op):
+            return
+        if not dev.get_active():
+            return
+        self._active_device = op
+
+    def _dev_strength_changed_cb(self, dev, strength):
+        op = dev.get_op()
+        if not self._devices.has_key(op):
+            return
+        if not dev.get_active():
+            return
+
+    def get_device(self, dev_op):
+        if not self._devices.has_key(dev_op):
+            return None
+        return self._devices[dev_op]
+
+    def _setup_dbus(self):
+        self._sig_handlers = {
+            'StateChange': self.state_change_sig_handler,
+            'DeviceAdded': self.device_added_sig_handler,
+            'DeviceRemoved': self.device_removed_sig_handler,
+            'DeviceActivationStage': self.device_activation_stage_sig_handler,
+            'DeviceActivating': self.device_activating_sig_handler,
+            'DeviceNowActive': self.device_now_active_sig_handler,
+            'DeviceNoLongerActive': self.device_no_longer_active_sig_handler,
+            'DeviceCarrierOn': self.device_carrier_on_sig_handler,
+            'DeviceCarrierOff': self.device_carrier_off_sig_handler,
+            'DeviceStrengthChanged': self.wireless_device_strength_changed_sig_handler,
+            'WirelessNetworkAppeared': self.wireless_network_appeared_sig_handler,
+            'WirelessNetworkDisappeared': self.wireless_network_disappeared_sig_handler,
+            'WirelessNetworkStrengthChanged': self.wireless_network_strength_changed_sig_handler
+        }
+
+        self._nm_proxy = sys_bus.get_object(NM_SERVICE, NM_PATH)
+        self._nm_obj = dbus.Interface(self._nm_proxy, NM_IFACE)
+
+        sys_bus.add_signal_receiver(self.name_owner_changed_sig_handler,
+                                         signal_name="NameOwnerChanged",
+                                         dbus_interface="org.freedesktop.DBus")
+
+        for (signal, handler) in self._sig_handlers.items():
+            sys_bus.add_signal_receiver(handler, signal_name=signal, dbus_interface=NM_IFACE)
+
+        # Find out whether or not NM is running
+        try:
+            bus_object = sys_bus.get_object('org.freedesktop.DBus', '/org/freedesktop/DBus')
+            name = bus_object.GetNameOwner("org.freedesktop.NetworkManagerInfo", \
+                    dbus_interface='org.freedesktop.DBus')
+            if name:
+                self._nm_present = True
+        except dbus.DBusException:
+            pass
+
+    def get_key_for_network(self, net, async_cb, async_err_cb):
+		pass
+
+    def device_activation_stage_sig_handler(self, device, stage):
+        logging.debug('Device Activation Stage "%s" for device %s' % (NM_DEVICE_STAGE_STRINGS[stage], device))
+
+    def state_change_sig_handler(self, state):
+        self._nm_state = state
+
+    def device_activating_sig_handler(self, device):
+        self._active_device = device
+
+    def device_now_active_sig_handler(self, device, ssid=None):
+        if not self._devices.has_key(device):
+            return
+        self._active_device = device
+        self._devices[device].set_active(True, ssid)
+
+    def device_no_longer_active_sig_handler(self, device):
+        if not self._devices.has_key(device):
+            return
+        if self._active_device == device:
+            self._active_device = None
+        self._devices[device].set_active(False)
+
+    def name_owner_changed_sig_handler(self, name, old, new):
+        if name != NM_SERVICE:
+            return
+        if (old and len(old)) and (not new and not len(new)):
+            # NM went away
+            self._nm_present = False
+            for op in self._devices.keys():
+                del self._devices[op]
+            self._devices = {}
+            self._active_device = None
+            self._nm_state = NM_STATE_UNKNOWN
+        elif (not old and not len(old)) and (new and len(new)):
+            # NM started up
+            self._nm_present = True
+            self._get_nm_state()
+            self._get_initial_devices()
+
+    def device_added_sig_handler(self, device):
+        self._add_device(device)
+
+    def device_removed_sig_handler(self, device):
+        self._remove_device(device)
+
+    def wireless_network_appeared_sig_handler(self, device, network):
+        if not self._devices.has_key(device):
+            return
+        self._devices[device].network_appeared(network)
+
+    def wireless_network_disappeared_sig_handler(self, device, network):
+        if not self._devices.has_key(device):
+            return
+        self._devices[device].network_disappeared(network)
+
+    def wireless_device_strength_changed_sig_handler(self, device, strength):
+        if not self._devices.has_key(device):
+            return
+        self._devices[device].set_strength(strength)
+
+    def wireless_network_strength_changed_sig_handler(self, device, network, strength):
+        if not self._devices.has_key(device):
+            return
+        net = self._devices[device].get_network(network)
+        if net:
+            net.set_strength(strength)
+
+    def device_carrier_on_sig_handler(self, device):
+        if not self._devices.has_key(device):
+            return
+        self._devices[device].set_carrier(True)
+
+    def device_carrier_off_sig_handler(self, device):
+        if not self._devices.has_key(device):
+            return
+        self._devices[device].set_carrier(False)