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
185
186
187
188
189
190
191
192
193
194
195
196
197
198
|
# 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
# honor SOFTWARE_CENTER_NET_{DIS,}CONNECTED in the environment variables
import os
env_map = {
'SOFTWARE_CENTER_NET_DISCONNECTED': NetState.NM_STATE_DISCONNECTED,
'SOFTWARE_CENTER_NET_CONNECTED': NetState.NM_STATE_CONNECTED_GLOBAL,
}
for envkey, state in env_map.iteritems():
if envkey in os.environ:
NETWORK_STATE = state
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()
|