1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184
|
# Copyright (C) 2009 Canonical
#
# Authors:
# Matthew McGowan
# Michael Vogt
#
# 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; version 3.
#
# 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 Street, Fifth Floor, Boston, MA 02110-1301 USA
import dbus
import logging
from urlparse import urlparse
from dbus.mainloop.glib import DBusGMainLoop
from gi.repository import GObject
LOG=logging.getLogger(__name__)
# enums
class NetState(object):
""" enums for network manager status """
# Old enum values are for NM 0.7
# The NetworkManager daemon is in an unknown state.
NM_STATE_UNKNOWN = 0
NM_STATE_UNKNOWN_LIST = [NM_STATE_UNKNOWN]
# The NetworkManager daemon is asleep and all interfaces managed by it are inactive.
NM_STATE_ASLEEP_OLD = 1
NM_STATE_ASLEEP = 10
NM_STATE_ASLEEP_LIST = [NM_STATE_ASLEEP_OLD,
NM_STATE_ASLEEP]
# The NetworkManager daemon is connecting a device.
NM_STATE_CONNECTING_OLD = 2
NM_STATE_CONNECTING = 40
NM_STATE_CONNECTING_LIST = [NM_STATE_CONNECTING_OLD,
NM_STATE_CONNECTING]
# The NetworkManager daemon is connected.
NM_STATE_CONNECTED_OLD = 3
NM_STATE_CONNECTED_LOCAL = 50
NM_STATE_CONNECTED_SITE = 60
NM_STATE_CONNECTED_GLOBAL = 70
NM_STATE_CONNECTED_LIST = [NM_STATE_CONNECTED_OLD,
NM_STATE_CONNECTED_LOCAL,
NM_STATE_CONNECTED_SITE,
NM_STATE_CONNECTED_GLOBAL]
# The NetworkManager daemon is disconnecting.
NM_STATE_DISCONNECTING = 30
NM_STATE_DISCONNECTING_LIST = [NM_STATE_DISCONNECTING]
# The NetworkManager daemon is disconnected.
NM_STATE_DISCONNECTED_OLD = 4
NM_STATE_DISCONNECTED = 20
NM_STATE_DISCONNECTED_LIST = [NM_STATE_DISCONNECTED_OLD,
NM_STATE_DISCONNECTED]
class NetworkStatusWatcher(GObject.GObject):
""" simple watcher which notifys subscribers to network events..."""
__gsignals__ = {'changed':(GObject.SIGNAL_RUN_FIRST,
GObject.TYPE_NONE,
(int,)),
}
def __init__(self):
GObject.GObject.__init__(self)
return
# internal helper
NETWORK_STATE = 0
def __connection_state_changed_handler(state):
global NETWORK_STATE
NETWORK_STATE = int(state)
__WATCHER__.emit("changed", NETWORK_STATE)
return
# init network state
def __init_network_state():
global NETWORK_STATE
# check is SOFTWARE_CENTER_NET_DISCONNECTED is in the environment variables
# if so force the network status to be disconnected
import os
if ("SOFTWARE_CENTER_NET_DISCONNECTED" in os.environ and
os.environ["SOFTWARE_CENTER_NET_DISCONNECTED"] == 1):
NETWORK_STATE = NetState.NM_STATE_DISCONNECTED
print('forced netstate into disconnected mode...')
return
dbus_loop = DBusGMainLoop()
try:
bus = dbus.SystemBus(mainloop=dbus_loop)
nm = bus.get_object('org.freedesktop.NetworkManager',
'/org/freedesktop/NetworkManager')
NETWORK_STATE = nm.state(dbus_interface='org.freedesktop.NetworkManager')
bus.add_signal_receiver(__connection_state_changed_handler,
dbus_interface="org.freedesktop.NetworkManager",
signal_name="StateChanged")
return
except Exception as e:
logging.warn("failed to init network state watcher '%s'" % e)
NETWORK_STATE = NetState.NM_STATE_UNKNOWN
# test ping to check if there is internet connectivity despite
# NetworkManager not being available
import threading
thread = threading.Thread(target=test_ping, name='test_ping')
thread.start()
return
#helper
def test_ping():
global NETWORK_STATE
import subprocess
# ping the main deb repository from the sources.list
import aptsources
source_list = aptsources.apt_pkg.SourceList()
source_list.read_main_list()
if not source_list.list:
LOG.warn("apt sourcelist had no sources!!!")
NETWORK_STATE = NetState.NM_STATE_DISCONNECTED
else:
# get a host to ping
host = urlparse(source_list.list[0].uri)[1]
msg = ("Attempting one time ping of %s to test if internet "
"connectivity exists." % host)
logging.info(msg)
ping = subprocess.Popen(
["ping", "-c", "1", host],
stdout=subprocess.PIPE,
stderr=subprocess.PIPE
)
out, error = ping.communicate()
if len(error.splitlines()):
NETWORK_STATE = NetState.NM_STATE_DISCONNECTED
msg = "Could not detect an internet connection\n%s" % error
else:
NETWORK_STATE = NetState.NM_STATE_CONNECTED_GLOBAL
msg = "Internet connection available!\n%s" % out
__WATCHER__.emit("changed", NETWORK_STATE)
logging.info("ping output: '%s'" % msg)
return
# global watcher
__WATCHER__ = NetworkStatusWatcher()
def get_network_watcher():
return __WATCHER__
# simply query
def get_network_state():
""" get the NetState state """
global NETWORK_STATE
return NETWORK_STATE
# simply query even more
def network_state_is_connected():
""" get bool if we are connected """
# unkown because in doubt, just assume we have network
return get_network_state() in NetState.NM_STATE_UNKNOWN_LIST + \
NetState.NM_STATE_CONNECTED_LIST
# init it once
__init_network_state()
if __name__ == '__main__':
loop = GObject.MainLoop()
loop.run()
|