mirror of
https://github.com/gryf/wicd.git
synced 2025-12-29 01:42:33 +01:00
ec37206905
actually use "write=True" instead of just "True" everywhere we do config writes explicitly. Fix the scripts dialog not working for wired connections. Force the monitor to update state after triggering a disconnect or connect. Remove an unneeded Scan call from autoconnect.py
339 lines
12 KiB
Python
Executable File
339 lines
12 KiB
Python
Executable File
#!/usr/bin/env python
|
|
|
|
""" monitor -- connection monitoring process
|
|
|
|
This process is spawned as a child of the daemon, and is responsible
|
|
for monitoring connection status and initiating autoreconnection
|
|
when appropriate.
|
|
|
|
"""
|
|
#
|
|
# Copyright (C) 2007 - 2008 Adam Blackburn
|
|
# Copyright (C) 2007 - 2008 Dan O'Reilly
|
|
#
|
|
# This program is free software; you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License Version 2 as
|
|
# published by the Free Software Foundation.
|
|
#
|
|
# 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, see <http://www.gnu.org/licenses/>.
|
|
#
|
|
|
|
import gobject
|
|
import time
|
|
|
|
from dbus import DBusException
|
|
|
|
from wicd import wpath
|
|
from wicd import misc
|
|
from wicd import dbusmanager
|
|
|
|
misc.RenameProcess("wicd-monitor")
|
|
|
|
if __name__ == '__main__':
|
|
wpath.chdir(__file__)
|
|
|
|
dbusmanager.connect_to_dbus()
|
|
dbus_dict = dbusmanager.get_dbus_ifaces()
|
|
daemon = dbus_dict["daemon"]
|
|
wired = dbus_dict["wired"]
|
|
wireless = dbus_dict["wireless"]
|
|
bus = dbusmanager.get_bus()
|
|
|
|
monitor = to_time = update_callback = None
|
|
|
|
class ConnectionStatus(object):
|
|
""" Class for monitoring the computer's connection status. """
|
|
def __init__(self):
|
|
""" Initialize variables needed for the connection status methods. """
|
|
self.last_strength = -2
|
|
self.displayed_strength = -1
|
|
self.still_wired = False
|
|
self.network = ''
|
|
self.tried_reconnect = False
|
|
self.connection_lost_counter = 0
|
|
self.last_state = misc.NOT_CONNECTED
|
|
self.reconnecting = False
|
|
self.reconnect_tries = 0
|
|
self.last_reconnect_time = time.time()
|
|
self.signal_changed = False
|
|
self.iwconfig = ""
|
|
self.trigger_reconnect = False
|
|
|
|
bus = dbusmanager.get_bus()
|
|
bus.add_signal_receiver(self._force_update_connection_status,
|
|
"UpdateState", "org.wicd.daemon")
|
|
|
|
def check_for_wired_connection(self, wired_ip):
|
|
""" Checks for a wired connection.
|
|
|
|
Checks for two states:
|
|
1) A wired connection is not in use, but a cable is plugged
|
|
in, and the user has chosen to switch to a wired connection
|
|
whenever its available, even if already connected to a
|
|
wireless network
|
|
|
|
2) A wired connection is currently active.
|
|
|
|
"""
|
|
if not wired_ip and daemon.GetPreferWiredNetwork():
|
|
if not daemon.GetForcedDisconnect() and wired.CheckPluggedIn():
|
|
self.trigger_reconnect = True
|
|
|
|
elif wired_ip and wired.CheckPluggedIn():
|
|
# Only change the interface if it's not already set for wired
|
|
if not self.still_wired:
|
|
daemon.SetCurrentInterface(daemon.GetWiredInterface())
|
|
self.still_wired = True
|
|
return True
|
|
# Wired connection isn't active
|
|
self.still_wired = False
|
|
return False
|
|
|
|
def check_for_wireless_connection(self, wireless_ip):
|
|
""" Checks for an active wireless connection.
|
|
|
|
Checks for an active wireless connection. Also notes
|
|
if the signal strength is 0, and if it remains there
|
|
for too long, triggers a wireless disconnect.
|
|
|
|
Returns True if wireless connection is active, and
|
|
False otherwise.
|
|
|
|
"""
|
|
|
|
# Make sure we have an IP before we do anything else.
|
|
if not wireless_ip:
|
|
return False
|
|
|
|
if daemon.NeedsExternalCalls():
|
|
self.iwconfig = wireless.GetIwconfig()
|
|
else:
|
|
self.iwconfig = ''
|
|
# Reset this, just in case.
|
|
self.tried_reconnect = False
|
|
|
|
wifi_signal = self._get_printable_sig_strength()
|
|
if wifi_signal == 0:
|
|
# If we have no signal, increment connection loss counter.
|
|
# If we haven't gotten any signal 4 runs in a row (12 seconds),
|
|
# try to reconnect.
|
|
self.connection_lost_counter += 1
|
|
print self.connection_lost_counter
|
|
if self.connection_lost_counter >= 4:
|
|
wireless.DisconnectWireless()
|
|
self.connection_lost_counter = 0
|
|
return False
|
|
else: # If we have a signal, reset the counter
|
|
self.connection_lost_counter = 0
|
|
|
|
if (wifi_signal != self.last_strength or
|
|
self.network != wireless.GetCurrentNetwork(self.iwconfig)):
|
|
self.last_strength = wifi_signal
|
|
self.signal_changed = True
|
|
daemon.SetCurrentInterface(daemon.GetWirelessInterface())
|
|
|
|
return True
|
|
|
|
def update_connection_status(self):
|
|
""" Updates the tray icon and current connection status.
|
|
|
|
Determines the current connection state and sends a dbus signal
|
|
announcing when the status changes. Also starts the automatic
|
|
reconnection process if necessary.
|
|
|
|
"""
|
|
wired_ip = None
|
|
wifi_ip = None
|
|
|
|
try:
|
|
if daemon.GetSuspend():
|
|
print "Suspended."
|
|
state = misc.SUSPENDED
|
|
self.update_state(state)
|
|
return True
|
|
|
|
# Determine what our current state is.
|
|
# Are we currently connecting?
|
|
if daemon.CheckIfConnecting():
|
|
state = misc.CONNECTING
|
|
self.update_state(state)
|
|
return True
|
|
|
|
daemon.SendConnectResultsIfAvail()
|
|
|
|
# Check for wired.
|
|
wired_ip = wired.GetWiredIP("")
|
|
wired_found = self.check_for_wired_connection(wired_ip)
|
|
# Trigger an AutoConnect if we're plugged in, not connected
|
|
# to a wired network, and the "autoswitch to wired" option
|
|
# is on.
|
|
if self.trigger_reconnect:
|
|
self.trigger_reconnect = False
|
|
wireless.DisconnectWireless()
|
|
daemon.AutoConnect(False, reply_handler=lambda:None,
|
|
error_handler=lambda:None)
|
|
return True
|
|
if wired_found:
|
|
self.update_state(misc.WIRED, wired_ip=wired_ip)
|
|
return True
|
|
|
|
# Check for wireless
|
|
wifi_ip = wireless.GetWirelessIP("")
|
|
self.signal_changed = False
|
|
wireless_found = self.check_for_wireless_connection(wifi_ip)
|
|
if wireless_found:
|
|
self.update_state(misc.WIRELESS, wifi_ip=wifi_ip)
|
|
return True
|
|
|
|
state = misc.NOT_CONNECTED
|
|
if self.last_state == misc.WIRELESS:
|
|
from_wireless = True
|
|
else:
|
|
from_wireless = False
|
|
self.auto_reconnect(from_wireless)
|
|
self.update_state(state)
|
|
except DBusException, e:
|
|
print 'Ignoring DBus Error: ' + str(e)
|
|
return True
|
|
|
|
def _force_update_connection_status(self):
|
|
global update_callback
|
|
gobject.source_remove(update_callback)
|
|
self.update_connection_status()
|
|
add_poll_callback()
|
|
|
|
def update_state(self, state, wired_ip=None, wifi_ip=None):
|
|
""" Set the current connection state. """
|
|
# Set our connection state/info.
|
|
iwconfig = self.iwconfig
|
|
if state == misc.NOT_CONNECTED:
|
|
info = [""]
|
|
elif state == misc.SUSPENDED:
|
|
info = [""]
|
|
elif state == misc.CONNECTING:
|
|
if wired.CheckIfWiredConnecting():
|
|
info = ["wired"]
|
|
else:
|
|
info = ["wireless", str(wireless.GetCurrentNetwork(iwconfig))]
|
|
elif state == misc.WIRELESS:
|
|
self.reconnect_tries = 0
|
|
info = [str(wifi_ip), str(wireless.GetCurrentNetwork(iwconfig)),
|
|
str(self._get_printable_sig_strength()),
|
|
str(wireless.GetCurrentNetworkID(iwconfig))]
|
|
elif state == misc.WIRED:
|
|
self.reconnect_tries = 0
|
|
info = [str(wired_ip)]
|
|
else:
|
|
print 'ERROR: Invalid state!'
|
|
return True
|
|
daemon.SetConnectionStatus(state, info)
|
|
|
|
# Send a D-Bus signal announcing status has changed if necessary.
|
|
if state != self.last_state or \
|
|
(state == misc.WIRELESS and self.signal_changed):
|
|
daemon.EmitStatusChanged(state, info)
|
|
self.last_state = state
|
|
return True
|
|
|
|
def _get_printable_sig_strength(self):
|
|
""" Get the correct signal strength format. """
|
|
try:
|
|
if daemon.GetSignalDisplayType() == 0:
|
|
wifi_signal = int(wireless.GetCurrentSignalStrength(self.iwconfig))
|
|
else:
|
|
wifi_signal = int(wireless.GetCurrentDBMStrength(self.iwconfig))
|
|
except TypeError:
|
|
wifi_signal = 0
|
|
|
|
return wifi_signal
|
|
|
|
def auto_reconnect(self, from_wireless=None):
|
|
""" Automatically reconnects to a network if needed.
|
|
|
|
If automatic reconnection is turned on, this method will
|
|
attempt to first reconnect to the last used wireless network, and
|
|
should that fail will simply run AutoConnect()
|
|
|
|
"""
|
|
if self.reconnecting:
|
|
return
|
|
|
|
# Some checks to keep reconnect retries from going crazy.
|
|
if self.reconnect_tries > 2 and \
|
|
(time.time() - self.last_reconnect_time) < 300:
|
|
return
|
|
|
|
self.reconnecting = True
|
|
daemon.SetCurrentInterface('')
|
|
|
|
if daemon.ShouldAutoReconnect():
|
|
print 'Starting automatic reconnect process'
|
|
self.last_reconnect_time = time.time()
|
|
self.reconnect_tries += 1
|
|
|
|
# If we just lost a wireless connection, try to connect to that
|
|
# network again. Otherwise just call Autoconnect.
|
|
cur_net_id = wireless.GetCurrentNetworkID(self.iwconfig)
|
|
if from_wireless and cur_net_id > -1:
|
|
print 'Trying to reconnect to last used wireless ' + \
|
|
'network'
|
|
wireless.ConnectWireless(cur_net_id)
|
|
else:
|
|
daemon.AutoConnect(True, reply_handler=reply_handle,
|
|
error_handler=err_handle)
|
|
self.reconnecting = False
|
|
|
|
#def rescan_networks(self):
|
|
# """ Calls a wireless scan. """
|
|
# try:
|
|
# if daemon.GetSuspend() or daemon.CheckIfConnecting():
|
|
# return True
|
|
# wireless.Scan()
|
|
# except dbus.exceptions.DBusException, e:
|
|
# print 'dbus exception while attempting rescan: %s' % str(e)
|
|
# finally:
|
|
# return True
|
|
|
|
def reply_handle():
|
|
""" Just a dummy function needed for asynchronous dbus calls. """
|
|
pass
|
|
|
|
def err_handle(error):
|
|
""" Just a dummy function needed for asynchronous dbus calls. """
|
|
pass
|
|
|
|
def add_poll_callback():
|
|
global monitor, to_time, update_callback
|
|
|
|
try:
|
|
update_callback = gobject.timeout_add_seconds(to_time,
|
|
monitor.update_connection_status)
|
|
except:
|
|
update_callback = gobject.timeout_add(to_time * 1000,
|
|
monitor.update_connection_status)
|
|
|
|
def main():
|
|
""" Starts the connection monitor.
|
|
|
|
Starts a ConnectionStatus instance, sets the status to update
|
|
an amount of time determined by the active backend.
|
|
|
|
"""
|
|
global monitor, to_time
|
|
|
|
monitor = ConnectionStatus()
|
|
to_time = daemon.GetBackendUpdateInterval()
|
|
add_poll_callback()
|
|
mainloop = gobject.MainLoop()
|
|
mainloop.run()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|