1 /* GIO - GLib Input, Output and Streaming Library
2 *
3 * Copyright 2014-2018 Jan-Michael Brummer <jan.brummer@tabos.org>
4 *
5 * This library is free software; you can redistribute it and/or
6 * modify it under the terms of the GNU Lesser General Public
7 * License as published by the Free Software Foundation; either
8 * version 2 of the License, or (at your option) any later version.
9 *
10 * This library is distributed in the hope that it will be useful,
11 * but WITHOUT ANY WARRANTY; without even the implied warranty of
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
13 * Lesser General Public License for more details.
14 *
15 * You should have received a copy of the GNU Lesser General
16 * Public License along with this library; if not, write to the
17 * Free Software Foundation, Inc., 59 Temple Place, Suite 330,
18 * Boston, MA 02111-1307, USA.
19 */
20
21 #include "config.h"
22
23 #include <errno.h>
24
25 #ifdef HAVE_UNISTD_H
26 #include <unistd.h>
27 #endif
28
29 #include <winsock2.h>
30 #include <ws2tcpip.h>
31 #include <iphlpapi.h>
32 #include <stdio.h>
33
34 #include "gwin32networkmonitor.h"
35 #include "ginetaddress.h"
36 #include "ginetaddressmask.h"
37 #include "ginitable.h"
38 #include "giomodule-priv.h"
39 #include "glibintl.h"
40 #include "glib/gstdio.h"
41 #include "gnetworkingprivate.h"
42 #include "gsocket.h"
43 #include "gnetworkmonitor.h"
44 #include "gioerror.h"
45
46 static GInitableIface *initable_parent_iface;
47 static void g_win32_network_monitor_iface_init (GNetworkMonitorInterface *iface);
48 static void g_win32_network_monitor_initable_iface_init (GInitableIface *iface);
49
50 struct _GWin32NetworkMonitorPrivate
51 {
52 gboolean initialized;
53 GError *init_error;
54 GMainContext *main_context;
55 GSource *route_change_source;
56 HANDLE handle;
57 };
58
59 #define g_win32_network_monitor_get_type _g_win32_network_monitor_get_type
60 G_DEFINE_TYPE_WITH_CODE (GWin32NetworkMonitor, g_win32_network_monitor, G_TYPE_NETWORK_MONITOR_BASE,
61 G_ADD_PRIVATE (GWin32NetworkMonitor)
62 G_IMPLEMENT_INTERFACE (G_TYPE_NETWORK_MONITOR,
63 g_win32_network_monitor_iface_init)
64 G_IMPLEMENT_INTERFACE (G_TYPE_INITABLE,
65 g_win32_network_monitor_initable_iface_init)
66 _g_io_modules_ensure_extension_points_registered ();
67 g_io_extension_point_implement (G_NETWORK_MONITOR_EXTENSION_POINT_NAME,
68 g_define_type_id,
69 "win32",
70 20))
71
72 static void
g_win32_network_monitor_init(GWin32NetworkMonitor * win)73 g_win32_network_monitor_init (GWin32NetworkMonitor *win)
74 {
75 win->priv = g_win32_network_monitor_get_instance_private (win);
76 }
77
78 static gboolean
win_network_monitor_get_ip_info(IP_ADDRESS_PREFIX prefix,GSocketFamily * family,const guint8 ** dest,gsize * len)79 win_network_monitor_get_ip_info (IP_ADDRESS_PREFIX prefix,
80 GSocketFamily *family,
81 const guint8 **dest,
82 gsize *len)
83 {
84 switch (prefix.Prefix.si_family)
85 {
86 case AF_UNSPEC:
87 /* Fall-through: AF_UNSPEC deliveres both IPV4 and IPV6 infos, let`s stick with IPV4 here */
88 case AF_INET:
89 *family = G_SOCKET_FAMILY_IPV4;
90 *dest = (guint8 *) &prefix.Prefix.Ipv4.sin_addr;
91 *len = prefix.PrefixLength;
92 break;
93 case AF_INET6:
94 *family = G_SOCKET_FAMILY_IPV6;
95 *dest = (guint8 *) &prefix.Prefix.Ipv6.sin6_addr;
96 *len = prefix.PrefixLength;
97 break;
98 default:
99 return FALSE;
100 }
101
102 return TRUE;
103 }
104
105 static GInetAddressMask *
get_network_mask(GSocketFamily family,const guint8 * dest,gsize len)106 get_network_mask (GSocketFamily family,
107 const guint8 *dest,
108 gsize len)
109 {
110 GInetAddressMask *network;
111 GInetAddress *dest_addr;
112
113 if (dest != NULL)
114 dest_addr = g_inet_address_new_from_bytes (dest, family);
115 else
116 dest_addr = g_inet_address_new_any (family);
117
118 network = g_inet_address_mask_new (dest_addr, len, NULL);
119 g_object_unref (dest_addr);
120
121 return network;
122 }
123
124 static gboolean
win_network_monitor_process_table(GWin32NetworkMonitor * win,GError ** error)125 win_network_monitor_process_table (GWin32NetworkMonitor *win,
126 GError **error)
127 {
128 DWORD ret = 0;
129 GPtrArray *networks;
130 gsize i;
131 MIB_IPFORWARD_TABLE2 *routes = NULL;
132 MIB_IPFORWARD_ROW2 *route;
133
134 ret = GetIpForwardTable2 (AF_UNSPEC, &routes);
135 if (ret != ERROR_SUCCESS)
136 {
137 g_set_error (error, G_IO_ERROR, G_IO_ERROR_FAILED,
138 "GetIpForwardTable2 () failed: %ld", ret);
139
140 return FALSE;
141 }
142
143 networks = g_ptr_array_new_full (routes->NumEntries, g_object_unref);
144 for (i = 0; i < routes->NumEntries; i++)
145 {
146 GInetAddressMask *network;
147 const guint8 *dest;
148 gsize len;
149 GSocketFamily family;
150
151 route = routes->Table + i;
152
153 if (!win_network_monitor_get_ip_info (route->DestinationPrefix, &family, &dest, &len))
154 continue;
155
156 network = get_network_mask (family, dest, len);
157 if (network == NULL)
158 continue;
159
160 g_ptr_array_add (networks, network);
161 }
162
163 g_network_monitor_base_set_networks (G_NETWORK_MONITOR_BASE (win),
164 (GInetAddressMask **) networks->pdata,
165 networks->len);
166
167 return TRUE;
168 }
169
170 static void
add_network(GWin32NetworkMonitor * win,GSocketFamily family,const guint8 * dest,gsize dest_len)171 add_network (GWin32NetworkMonitor *win,
172 GSocketFamily family,
173 const guint8 *dest,
174 gsize dest_len)
175 {
176 GInetAddressMask *network;
177
178 network = get_network_mask (family, dest, dest_len);
179 if (network != NULL)
180 {
181 g_network_monitor_base_add_network (G_NETWORK_MONITOR_BASE (win), network);
182 g_object_unref (network);
183 }
184 }
185
186 static void
remove_network(GWin32NetworkMonitor * win,GSocketFamily family,const guint8 * dest,gsize dest_len)187 remove_network (GWin32NetworkMonitor *win,
188 GSocketFamily family,
189 const guint8 *dest,
190 gsize dest_len)
191 {
192 GInetAddressMask *network;
193
194 network = get_network_mask (family, dest, dest_len);
195 if (network != NULL)
196 {
197 g_network_monitor_base_remove_network (G_NETWORK_MONITOR_BASE (win), network);
198 g_object_unref (network);
199 }
200 }
201
202 typedef struct {
203 PMIB_IPFORWARD_ROW2 route;
204 MIB_NOTIFICATION_TYPE type;
205 GWin32NetworkMonitor *win;
206 } RouteData;
207
208 static gboolean
win_network_monitor_invoke_route_changed(gpointer user_data)209 win_network_monitor_invoke_route_changed (gpointer user_data)
210 {
211 GSocketFamily family;
212 RouteData *route_data = user_data;
213 const guint8 *dest;
214 gsize len;
215
216 switch (route_data->type)
217 {
218 case MibDeleteInstance:
219 if (!win_network_monitor_get_ip_info (route_data->route->DestinationPrefix, &family, &dest, &len))
220 break;
221
222 remove_network (route_data->win, family, dest, len);
223 break;
224 case MibAddInstance:
225 if (!win_network_monitor_get_ip_info (route_data->route->DestinationPrefix, &family, &dest, &len))
226 break;
227
228 add_network (route_data->win, family, dest, len);
229 break;
230 case MibInitialNotification:
231 default:
232 break;
233 }
234
235 return G_SOURCE_REMOVE;
236 }
237
238 static VOID WINAPI
win_network_monitor_route_changed_cb(PVOID context,PMIB_IPFORWARD_ROW2 route,MIB_NOTIFICATION_TYPE type)239 win_network_monitor_route_changed_cb (PVOID context,
240 PMIB_IPFORWARD_ROW2 route,
241 MIB_NOTIFICATION_TYPE type)
242 {
243 GWin32NetworkMonitor *win = context;
244 RouteData *route_data;
245
246 route_data = g_new0 (RouteData, 1);
247 route_data->route = route;
248 route_data->type = type;
249 route_data->win = win;
250
251 win->priv->route_change_source = g_idle_source_new ();
252 g_source_set_priority (win->priv->route_change_source, G_PRIORITY_DEFAULT);
253 g_source_set_callback (win->priv->route_change_source,
254 win_network_monitor_invoke_route_changed,
255 route_data,
256 g_free);
257
258 g_source_attach (win->priv->route_change_source, win->priv->main_context);
259 }
260
261 static gboolean
g_win32_network_monitor_initable_init(GInitable * initable,GCancellable * cancellable,GError ** error)262 g_win32_network_monitor_initable_init (GInitable *initable,
263 GCancellable *cancellable,
264 GError **error)
265 {
266 GWin32NetworkMonitor *win = G_WIN32_NETWORK_MONITOR (initable);
267 NTSTATUS status;
268 gboolean read;
269
270 if (!win->priv->initialized)
271 {
272 win->priv->main_context = g_main_context_ref_thread_default ();
273
274 /* Read current IP routing table. */
275 read = win_network_monitor_process_table (win, &win->priv->init_error);
276 if (read)
277 {
278 /* Register for IPv4 and IPv6 route updates. */
279 status = NotifyRouteChange2 (AF_UNSPEC, (PIPFORWARD_CHANGE_CALLBACK) win_network_monitor_route_changed_cb, win, FALSE, &win->priv->handle);
280 if (status != NO_ERROR)
281 g_set_error (&win->priv->init_error, G_IO_ERROR, G_IO_ERROR_FAILED,
282 "NotifyRouteChange2() error: %ld", status);
283 }
284
285 win->priv->initialized = TRUE;
286 }
287
288 /* Forward the results. */
289 if (win->priv->init_error != NULL)
290 {
291 g_propagate_error (error, g_error_copy (win->priv->init_error));
292 return FALSE;
293 }
294
295 return initable_parent_iface->init (initable, cancellable, error);
296 }
297
298 static void
g_win32_network_monitor_finalize(GObject * object)299 g_win32_network_monitor_finalize (GObject *object)
300 {
301 GWin32NetworkMonitor *win = G_WIN32_NETWORK_MONITOR (object);
302
303 /* Cancel notification event */
304 if (win->priv->handle)
305 CancelMibChangeNotify2 (win->priv->handle);
306
307 g_clear_error (&win->priv->init_error);
308
309 if (win->priv->route_change_source != NULL)
310 {
311 g_source_destroy (win->priv->route_change_source);
312 g_source_unref (win->priv->route_change_source);
313 }
314
315 g_main_context_unref (win->priv->main_context);
316
317 G_OBJECT_CLASS (g_win32_network_monitor_parent_class)->finalize (object);
318 }
319
320 static void
g_win32_network_monitor_class_init(GWin32NetworkMonitorClass * win_class)321 g_win32_network_monitor_class_init (GWin32NetworkMonitorClass *win_class)
322 {
323 GObjectClass *gobject_class = G_OBJECT_CLASS (win_class);
324
325 gobject_class->finalize = g_win32_network_monitor_finalize;
326 }
327
328 static void
g_win32_network_monitor_iface_init(GNetworkMonitorInterface * monitor_iface)329 g_win32_network_monitor_iface_init (GNetworkMonitorInterface *monitor_iface)
330 {
331 }
332
333 static void
g_win32_network_monitor_initable_iface_init(GInitableIface * iface)334 g_win32_network_monitor_initable_iface_init (GInitableIface *iface)
335 {
336 initable_parent_iface = g_type_interface_peek_parent (iface);
337
338 iface->init = g_win32_network_monitor_initable_init;
339 }
340