gio/tests/network-monitor.c

   1 /* GIO - GLib Input, Output and Streaming Library
   2  *
   3  * Copyright 2011 Red Hat, Inc.
   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 "gio.h"
  22
  23 /* hack */
  24 #define GIO_COMPILATION
  25 #include "gnetworkmonitorbase.h"
  26
  27 #include <string.h>
  28
  29 /* Test data; the GInetAddresses and GInetAddressMasks get filled in
  30  * in main(). Each address in a TestAddress matches the mask in its
  31  * corresponding TestMask, and none of them match any of the other
  32  * masks. The addresses in unmatched don't match any of the masks.
  33  */
  34
  35 typedef struct {
  36   const char *string;
  37   GInetAddress *address;
  38 } TestAddress;
  39
  40 typedef struct {
  41   const char *mask_string;
  42   GInetAddressMask *mask;
  43   TestAddress *addresses;
  44 } TestMask;
  45
  46 TestAddress net127addrs[] = {
  47   { "127.0.0.1", NULL },
  48   { "127.0.0.2", NULL },
  49   { "127.0.0.255", NULL },
  50   { "127.0.1.0", NULL },
  51   { "127.0.255.0", NULL },
  52   { "127.0.255.0", NULL },
  53   { "127.255.255.255", NULL },
  54   { NULL, NULL }
  55 };
  56 TestMask net127 = { "127.0.0.0/8", NULL, net127addrs };
  57
  58 TestAddress net10addrs[] = {
  59   { "10.0.0.1", NULL },
  60   { "10.0.0.2", NULL },
  61   { "10.0.0.255", NULL },
  62   { NULL, NULL }
  63 };
  64 TestMask net10 = { "10.0.0.0/24", NULL, net10addrs };
  65
  66 TestAddress net192addrs[] = {
  67   { "192.168.0.1", NULL },
  68   { "192.168.0.2", NULL },
  69   { "192.168.0.255", NULL },
  70   { "192.168.1.0", NULL },
  71   { "192.168.15.0", NULL },
  72   { NULL, NULL }
  73 };
  74 TestMask net192 = { "192.168.0.0/20", NULL, net192addrs };
  75
  76 TestAddress netlocal6addrs[] = {
  77   { "::1", NULL },
  78   { NULL, NULL }
  79 };
  80 TestMask netlocal6 = { "::1/128", NULL, netlocal6addrs };
  81
  82 TestAddress netfe80addrs[] = {
  83   { "fe80::", NULL },
  84   { "fe80::1", NULL },
  85   { "fe80::21b:77ff:fea2:972a", NULL },
  86   { NULL, NULL }
  87 };
  88 TestMask netfe80 = { "fe80::/64", NULL, netfe80addrs };
  89
  90 TestAddress unmatched[] = {
  91   { "10.0.1.0", NULL },
  92   { "10.0.255.0", NULL },
  93   { "10.255.255.255", NULL },
  94   { "192.168.16.0", NULL },
  95   { "192.168.255.0", NULL },
  96   { "192.169.0.0", NULL },
  97   { "192.255.255.255", NULL },
  98   { "::2", NULL },
  99   { "1::1", NULL },
 100   { "fe80::1:0:0:0:0", NULL },
 101   { "fe80:8000::0:0:0:0", NULL },
 102   { NULL, NULL }
 103 };
 104
 105 GInetAddressMask *ip4_default, *ip6_default;
 106
 107 static void
 108 notify_handler (GObject    *object,
 109                 GParamSpec *pspec,
 110                 gpointer    user_data)
 111 {
 112   gboolean *emitted = user_data;
 113
 114   *emitted = TRUE;
 115 }
 116
 117 static void
 118 network_changed_handler (GNetworkMonitor *monitor,
 119                          gboolean         available,
 120                          gpointer         user_data)
 121 {
 122   gboolean *emitted = user_data;
 123
 124   *emitted = TRUE;
 125 }
 126
 127 static void
 128 assert_signals (GNetworkMonitor *monitor,
 129                 gboolean         should_emit_notify,
 130                 gboolean         should_emit_network_changed,
 131                 gboolean         expected_network_available)
 132 {
 133   gboolean emitted_notify = FALSE, emitted_network_changed = FALSE;
 134   guint h1, h2;
 135
 136   h1 = g_signal_connect (monitor, "notify::network-available",
 137                          G_CALLBACK (notify_handler),
 138                          &emitted_notify);
 139   h2 = g_signal_connect (monitor, "network-changed",
 140                          G_CALLBACK (network_changed_handler),
 141                          &emitted_network_changed);
 142
 143   g_main_context_iteration (NULL, FALSE);
 144
 145   g_signal_handler_disconnect (monitor, h1);
 146   g_signal_handler_disconnect (monitor, h2);
 147
 148   g_assert (emitted_notify == should_emit_notify);
 149   g_assert (emitted_network_changed == should_emit_network_changed);
 150
 151   g_assert (g_network_monitor_get_network_available (monitor) == expected_network_available);
 152 }
 153
 154 typedef struct {
 155   GNetworkMonitor *monitor;
 156   GMainLoop       *loop;
 157   GSocketAddress  *sockaddr;
 158   gboolean         should_be_reachable;
 159 } CanReachData;
 160
 161 static void
 162 reach_cb (GObject      *source,
 163           GAsyncResult *res,
 164           gpointer      user_data)
 165 {
 166   GError *error = NULL;
 167   gboolean reachable;
 168   CanReachData *data = user_data;
 169
 170   reachable = g_network_monitor_can_reach_finish (data->monitor, res, &error);
 171
 172   if (data->should_be_reachable)
 173     g_assert_no_error (error);
 174   else
 175     {
 176       g_assert (error != NULL);
 177       g_clear_error (&error);
 178     }
 179   g_assert (reachable == data->should_be_reachable);
 180
 181   g_main_loop_quit (data->loop);
 182 }
 183
 184 static gboolean
 185 test_reach_async (gpointer user_data)
 186 {
 187   CanReachData *data = user_data;
 188
 189   g_network_monitor_can_reach_async (data->monitor,
 190                                      G_SOCKET_CONNECTABLE (data->sockaddr),
 191                                      NULL,
 192                                      reach_cb,
 193                                      data);
 194   return G_SOURCE_REMOVE;
 195 }
 196
 197 static void
 198 run_tests (GNetworkMonitor *monitor,
 199            TestAddress     *addresses,
 200            gboolean         should_be_reachable)
 201 {
 202   GError *error = NULL;
 203   int i;
 204   gboolean reachable;
 205   GSocketAddress *sockaddr;
 206   CanReachData data;
 207
 208   data.monitor = monitor;
 209   data.loop = g_main_loop_new (NULL, FALSE);
 210
 211   for (i = 0; addresses[i].address; i++)
 212     {
 213       sockaddr = g_inet_socket_address_new (addresses[i].address, 0);
 214       reachable = g_network_monitor_can_reach (monitor,
 215                                                G_SOCKET_CONNECTABLE (sockaddr),
 216                                                NULL, &error);
 217       data.sockaddr = sockaddr;
 218       data.should_be_reachable = should_be_reachable;
 219
 220       g_idle_add (test_reach_async, &data);
 221       g_main_loop_run (data.loop);
 222
 223       g_object_unref (sockaddr);
 224       g_assert_cmpint (reachable, ==, should_be_reachable);
 225       if (should_be_reachable)
 226         g_assert_no_error (error);
 227       else
 228         {
 229           g_assert (error != NULL);
 230           g_clear_error (&error);
 231         }
 232     }
 233 }
 234
 235 static void
 236 test_default (void)
 237 {
 238   GNetworkMonitor *monitor, *m;
 239   GError *error = NULL;
 240
 241   m = g_network_monitor_get_default ();
 242   g_assert (G_IS_NETWORK_MONITOR (m));
 243
 244   monitor = g_initable_new (G_TYPE_NETWORK_MONITOR_BASE, NULL, &error, NULL);
 245   g_assert_no_error (error);
 246
 247   /* In the default configuration, all addresses are reachable */
 248   run_tests (monitor, net127.addresses, TRUE);
 249   run_tests (monitor, net10.addresses, TRUE);
 250   run_tests (monitor, net192.addresses, TRUE);
 251   run_tests (monitor, netlocal6.addresses, TRUE);
 252   run_tests (monitor, netfe80.addresses, TRUE);
 253   run_tests (monitor, unmatched, TRUE);
 254
 255   assert_signals (monitor, FALSE, FALSE, TRUE);
 256
 257   g_object_unref (monitor);
 258 }
 259
 260 static void
 261 test_remove_default (void)
 262 {
 263   GNetworkMonitor *monitor;
 264   GError *error = NULL;
 265
 266   monitor = g_initable_new (G_TYPE_NETWORK_MONITOR_BASE, NULL, &error, NULL);
 267   g_assert_no_error (error);
 268   assert_signals (monitor, FALSE, FALSE, TRUE);
 269
 270   g_network_monitor_base_remove_network (G_NETWORK_MONITOR_BASE (monitor),
 271                                          ip4_default);
 272   assert_signals (monitor, FALSE, TRUE, TRUE);
 273   g_network_monitor_base_remove_network (G_NETWORK_MONITOR_BASE (monitor),
 274                                          ip6_default);
 275   assert_signals (monitor, TRUE, TRUE, FALSE);
 276
 277   /* Now nothing should be reachable */
 278   run_tests (monitor, net127.addresses, FALSE);
 279   run_tests (monitor, net10.addresses, FALSE);
 280   run_tests (monitor, net192.addresses, FALSE);
 281   run_tests (monitor, netlocal6.addresses, FALSE);
 282   run_tests (monitor, netfe80.addresses, FALSE);
 283   run_tests (monitor, unmatched, FALSE);
 284
 285   g_object_unref (monitor);
 286 }
 287
 288 static void
 289 test_add_networks (void)
 290 {
 291   GNetworkMonitor *monitor;
 292   GError *error = NULL;
 293
 294   monitor = g_initable_new (G_TYPE_NETWORK_MONITOR_BASE, NULL, &error, NULL);
 295   g_assert_no_error (error);
 296   assert_signals (monitor, FALSE, FALSE, TRUE);
 297
 298   g_network_monitor_base_remove_network (G_NETWORK_MONITOR_BASE (monitor),
 299                                          ip4_default);
 300   assert_signals (monitor, FALSE, TRUE, TRUE);
 301   g_network_monitor_base_remove_network (G_NETWORK_MONITOR_BASE (monitor),
 302                                          ip6_default);
 303   assert_signals (monitor, TRUE, TRUE, FALSE);
 304
 305   /* Now add the masks one by one */
 306
 307   g_network_monitor_base_add_network (G_NETWORK_MONITOR_BASE (monitor),
 308                                       net127.mask);
 309   assert_signals (monitor, FALSE, TRUE, FALSE);
 310
 311   run_tests (monitor, net127.addresses, TRUE);
 312   run_tests (monitor, net10.addresses, FALSE);
 313   run_tests (monitor, net192.addresses, FALSE);
 314   run_tests (monitor, netlocal6.addresses, FALSE);
 315   run_tests (monitor, netfe80.addresses, FALSE);
 316   run_tests (monitor, unmatched, FALSE);
 317
 318   g_network_monitor_base_add_network (G_NETWORK_MONITOR_BASE (monitor),
 319                                       net10.mask);
 320   assert_signals (monitor, FALSE, TRUE, FALSE);
 321   run_tests (monitor, net127.addresses, TRUE);
 322   run_tests (monitor, net10.addresses, TRUE);
 323   run_tests (monitor, net192.addresses, FALSE);
 324   run_tests (monitor, netlocal6.addresses, FALSE);
 325   run_tests (monitor, netfe80.addresses, FALSE);
 326   run_tests (monitor, unmatched, FALSE);
 327
 328   g_network_monitor_base_add_network (G_NETWORK_MONITOR_BASE (monitor),
 329                                       net192.mask);
 330   assert_signals (monitor, FALSE, TRUE, FALSE);
 331   run_tests (monitor, net127.addresses, TRUE);
 332   run_tests (monitor, net10.addresses, TRUE);
 333   run_tests (monitor, net192.addresses, TRUE);
 334   run_tests (monitor, netlocal6.addresses, FALSE);
 335   run_tests (monitor, netfe80.addresses, FALSE);
 336   run_tests (monitor, unmatched, FALSE);
 337
 338   g_network_monitor_base_add_network (G_NETWORK_MONITOR_BASE (monitor),
 339                                       netlocal6.mask);
 340   assert_signals (monitor, FALSE, TRUE, FALSE);
 341   run_tests (monitor, net127.addresses, TRUE);
 342   run_tests (monitor, net10.addresses, TRUE);
 343   run_tests (monitor, net192.addresses, TRUE);
 344   run_tests (monitor, netlocal6.addresses, TRUE);
 345   run_tests (monitor, netfe80.addresses, FALSE);
 346   run_tests (monitor, unmatched, FALSE);
 347
 348   g_network_monitor_base_add_network (G_NETWORK_MONITOR_BASE (monitor),
 349                                       netfe80.mask);
 350   assert_signals (monitor, FALSE, TRUE, FALSE);
 351   run_tests (monitor, net127.addresses, TRUE);
 352   run_tests (monitor, net10.addresses, TRUE);
 353   run_tests (monitor, net192.addresses, TRUE);
 354   run_tests (monitor, netlocal6.addresses, TRUE);
 355   run_tests (monitor, netfe80.addresses, TRUE);
 356   run_tests (monitor, unmatched, FALSE);
 357
 358   g_object_unref (monitor);
 359 }
 360
 361 static void
 362 test_remove_networks (void)
 363 {
 364   GNetworkMonitor *monitor;
 365   GError *error = NULL;
 366
 367   monitor = g_initable_new (G_TYPE_NETWORK_MONITOR_BASE, NULL, &error, NULL);
 368   g_assert_no_error (error);
 369   assert_signals (monitor, FALSE, FALSE, TRUE);
 370
 371   g_network_monitor_base_remove_network (G_NETWORK_MONITOR_BASE (monitor),
 372                                          ip4_default);
 373   assert_signals (monitor, FALSE, TRUE, TRUE);
 374   g_network_monitor_base_remove_network (G_NETWORK_MONITOR_BASE (monitor),
 375                                          ip6_default);
 376   assert_signals (monitor, TRUE, TRUE, FALSE);
 377
 378   /* First add them */
 379   g_network_monitor_base_add_network (G_NETWORK_MONITOR_BASE (monitor),
 380                                       net127.mask);
 381   assert_signals (monitor, FALSE, TRUE, FALSE);
 382   g_network_monitor_base_add_network (G_NETWORK_MONITOR_BASE (monitor),
 383                                       net10.mask);
 384   assert_signals (monitor, FALSE, TRUE, FALSE);
 385   g_network_monitor_base_add_network (G_NETWORK_MONITOR_BASE (monitor),
 386                                       net192.mask);
 387   assert_signals (monitor, FALSE, TRUE, FALSE);
 388   g_network_monitor_base_add_network (G_NETWORK_MONITOR_BASE (monitor),
 389                                       netlocal6.mask);
 390   assert_signals (monitor, FALSE, TRUE, FALSE);
 391   g_network_monitor_base_add_network (G_NETWORK_MONITOR_BASE (monitor),
 392                                       netfe80.mask);
 393   assert_signals (monitor, FALSE, TRUE, FALSE);
 394
 395   run_tests (monitor, net127.addresses, TRUE);
 396   run_tests (monitor, net10.addresses, TRUE);
 397   run_tests (monitor, net192.addresses, TRUE);
 398   run_tests (monitor, netlocal6.addresses, TRUE);
 399   run_tests (monitor, netfe80.addresses, TRUE);
 400   run_tests (monitor, unmatched, FALSE);
 401
 402   /* Now remove them one by one */
 403   g_network_monitor_base_remove_network (G_NETWORK_MONITOR_BASE (monitor),
 404                                          net127.mask);
 405   assert_signals (monitor, FALSE, TRUE, FALSE);
 406   run_tests (monitor, net127.addresses, FALSE);
 407   run_tests (monitor, net10.addresses, TRUE);
 408   run_tests (monitor, net192.addresses, TRUE);
 409   run_tests (monitor, netlocal6.addresses, TRUE);
 410   run_tests (monitor, netfe80.addresses, TRUE);
 411   run_tests (monitor, unmatched, FALSE);
 412
 413   g_network_monitor_base_remove_network (G_NETWORK_MONITOR_BASE (monitor),
 414                                          net10.mask);
 415   assert_signals (monitor, FALSE, TRUE, FALSE);
 416   run_tests (monitor, net127.addresses, FALSE);
 417   run_tests (monitor, net10.addresses, FALSE);
 418   run_tests (monitor, net192.addresses, TRUE);
 419   run_tests (monitor, netlocal6.addresses, TRUE);
 420   run_tests (monitor, netfe80.addresses, TRUE);
 421   run_tests (monitor, unmatched, FALSE);
 422
 423   g_network_monitor_base_remove_network (G_NETWORK_MONITOR_BASE (monitor),
 424                                          net192.mask);
 425   assert_signals (monitor, FALSE, TRUE, FALSE);
 426   run_tests (monitor, net127.addresses, FALSE);
 427   run_tests (monitor, net10.addresses, FALSE);
 428   run_tests (monitor, net192.addresses, FALSE);
 429   run_tests (monitor, netlocal6.addresses, TRUE);
 430   run_tests (monitor, netfe80.addresses, TRUE);
 431   run_tests (monitor, unmatched, FALSE);
 432
 433   g_network_monitor_base_remove_network (G_NETWORK_MONITOR_BASE (monitor),
 434                                          netlocal6.mask);
 435   assert_signals (monitor, FALSE, TRUE, FALSE);
 436   run_tests (monitor, net127.addresses, FALSE);
 437   run_tests (monitor, net10.addresses, FALSE);
 438   run_tests (monitor, net192.addresses, FALSE);
 439   run_tests (monitor, netlocal6.addresses, FALSE);
 440   run_tests (monitor, netfe80.addresses, TRUE);
 441   run_tests (monitor, unmatched, FALSE);
 442
 443   g_network_monitor_base_remove_network (G_NETWORK_MONITOR_BASE (monitor),
 444                                          netfe80.mask);
 445   assert_signals (monitor, FALSE, TRUE, FALSE);
 446   run_tests (monitor, net127.addresses, FALSE);
 447   run_tests (monitor, net10.addresses, FALSE);
 448   run_tests (monitor, net192.addresses, FALSE);
 449   run_tests (monitor, netlocal6.addresses, FALSE);
 450   run_tests (monitor, netfe80.addresses, FALSE);
 451   run_tests (monitor, unmatched, FALSE);
 452
 453   g_object_unref (monitor);
 454 }
 455
 456
 457 static void
 458 init_test (TestMask *test)
 459 {
 460   GError *error = NULL;
 461   int i;
 462
 463   test->mask = g_inet_address_mask_new_from_string (test->mask_string, &error);
 464   g_assert_no_error (error);
 465
 466   for (i = 0; test->addresses[i].string; i++)
 467     {
 468       test->addresses[i].address = g_inet_address_new_from_string (test->addresses[i].string);
 469       if (strchr (test->addresses[i].string, ':'))
 470         g_assert_cmpint (g_inet_address_get_family (test->addresses[i].address), ==, G_SOCKET_FAMILY_IPV6);
 471       else
 472         g_assert_cmpint (g_inet_address_get_family (test->addresses[i].address), ==, G_SOCKET_FAMILY_IPV4);
 473     }
 474 }
 475
 476 static void
 477 watch_network_changed (GNetworkMonitor *monitor,
 478                        gboolean         available,
 479                        gpointer         user_data)
 480 {
 481   g_print ("Network is %s\n", available ? "up" : "down");
 482 }
 483
 484 static void
 485 do_watch_network (void)
 486 {
 487   GNetworkMonitor *monitor = g_network_monitor_get_default ();
 488   GMainLoop *loop;
 489
 490   g_print ("Monitoring via %s\n", g_type_name_from_instance ((GTypeInstance *) monitor));
 491
 492   g_signal_connect (monitor, "network-changed",
 493                     G_CALLBACK (watch_network_changed), NULL);
 494   watch_network_changed (monitor, g_network_monitor_get_network_available (monitor), NULL);
 495
 496   loop = g_main_loop_new (NULL, FALSE);
 497   g_main_loop_run (loop);
 498 }
 499
 500 int
 501 main (int argc, char **argv)
 502 {
 503   g_type_init ();
 504
 505   if (argc == 2 && !strcmp (argv[1], "--watch"))
 506     {
 507       do_watch_network ();
 508       return 0;
 509     }
 510
 511   g_test_init (&argc, &argv, NULL);
 512
 513   init_test (&net127);
 514   init_test (&net10);
 515   init_test (&net192);
 516   init_test (&netlocal6);
 517   init_test (&netfe80);
 518   ip4_default = g_inet_address_mask_new_from_string ("0.0.0.0/0", NULL);
 519   ip6_default = g_inet_address_mask_new_from_string ("::/0", NULL);
 520
 521   g_test_add_func ("/network-monitor/default", test_default);
 522   g_test_add_func ("/network-monitor/remove_default", test_remove_default);
 523   g_test_add_func ("/network-monitor/add_networks", test_add_networks);
 524   g_test_add_func ("/network-monitor/remove_networks", test_remove_networks);
 525
 526   return g_test_run ();
 527 }