Home | History | Annotate | Download | only in net_watcher 
      1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
      2 // Use of this source code is governed by a BSD-style license that can be
      3 // found in the LICENSE file.
      4 
      5 // This is a small utility that watches for and logs network changes.
      6 
      7 #include <string>
      8 
      9 #include "base/at_exit.h"
     10 #include "base/basictypes.h"
     11 #include "base/command_line.h"
     12 #include "base/compiler_specific.h"
     13 #include "base/json/json_writer.h"
     14 #include "base/logging.h"
     15 #include "base/memory/scoped_ptr.h"
     16 #include "base/message_loop/message_loop.h"
     17 #include "base/values.h"
     18 #include "build/build_config.h"
     19 #include "net/base/network_change_notifier.h"
     20 #include "net/proxy/proxy_config.h"
     21 #include "net/proxy/proxy_config_service.h"
     22 #include "net/proxy/proxy_service.h"
     23 
     24 #if (defined(OS_LINUX) || defined(OS_OPENBSD)) && !defined(OS_CHROMEOS)
     25 #include <glib-object.h>
     26 #endif
     27 
     28 #if defined(OS_MACOSX)
     29 #include "base/mac/scoped_nsautorelease_pool.h"
     30 #endif
     31 
     32 namespace {
     33 
     34 // Conversions from various network-related types to string.
     35 
     36 const char* ConnectionTypeToString(
     37     net::NetworkChangeNotifier::ConnectionType type) {
     38   switch (type) {
     39     case net::NetworkChangeNotifier::CONNECTION_UNKNOWN:
     40       return "CONNECTION_UNKNOWN";
     41     case net::NetworkChangeNotifier::CONNECTION_ETHERNET:
     42       return "CONNECTION_ETHERNET";
     43     case net::NetworkChangeNotifier::CONNECTION_WIFI:
     44       return "CONNECTION_WIFI";
     45     case net::NetworkChangeNotifier::CONNECTION_2G:
     46       return "CONNECTION_2G";
     47     case net::NetworkChangeNotifier::CONNECTION_3G:
     48       return "CONNECTION_3G";
     49     case net::NetworkChangeNotifier::CONNECTION_4G:
     50       return "CONNECTION_4G";
     51     case net::NetworkChangeNotifier::CONNECTION_NONE:
     52       return "CONNECTION_NONE";
     53     default:
     54       return "CONNECTION_UNEXPECTED";
     55   }
     56 }
     57 
     58 std::string ProxyConfigToString(const net::ProxyConfig& config) {
     59   scoped_ptr<base::Value> config_value(config.ToValue());
     60   std::string str;
     61   base::JSONWriter::Write(config_value.get(), &str);
     62   return str;
     63 }
     64 
     65 const char* ConfigAvailabilityToString(
     66     net::ProxyConfigService::ConfigAvailability availability) {
     67   switch (availability) {
     68     case net::ProxyConfigService::CONFIG_PENDING:
     69       return "CONFIG_PENDING";
     70     case net::ProxyConfigService::CONFIG_VALID:
     71       return "CONFIG_VALID";
     72     case net::ProxyConfigService::CONFIG_UNSET:
     73       return "CONFIG_UNSET";
     74     default:
     75       return "CONFIG_UNEXPECTED";
     76   }
     77 }
     78 
     79 // The main observer class that logs network events.
     80 class NetWatcher :
     81       public net::NetworkChangeNotifier::IPAddressObserver,
     82       public net::NetworkChangeNotifier::ConnectionTypeObserver,
     83       public net::NetworkChangeNotifier::DNSObserver,
     84       public net::NetworkChangeNotifier::NetworkChangeObserver,
     85       public net::ProxyConfigService::Observer {
     86  public:
     87   NetWatcher() {}
     88 
     89   virtual ~NetWatcher() {}
     90 
     91   // net::NetworkChangeNotifier::IPAddressObserver implementation.
     92   virtual void OnIPAddressChanged() OVERRIDE {
     93     LOG(INFO) << "OnIPAddressChanged()";
     94   }
     95 
     96   // net::NetworkChangeNotifier::ConnectionTypeObserver implementation.
     97   virtual void OnConnectionTypeChanged(
     98       net::NetworkChangeNotifier::ConnectionType type) OVERRIDE {
     99     LOG(INFO) << "OnConnectionTypeChanged("
    100               << ConnectionTypeToString(type) << ")";
    101   }
    102 
    103   // net::NetworkChangeNotifier::DNSObserver implementation.
    104   virtual void OnDNSChanged() OVERRIDE {
    105     LOG(INFO) << "OnDNSChanged()";
    106   }
    107 
    108   // net::NetworkChangeNotifier::NetworkChangeObserver implementation.
    109   virtual void OnNetworkChanged(
    110       net::NetworkChangeNotifier::ConnectionType type) OVERRIDE {
    111     LOG(INFO) << "OnNetworkChanged("
    112               << ConnectionTypeToString(type) << ")";
    113   }
    114 
    115   // net::ProxyConfigService::Observer implementation.
    116   virtual void OnProxyConfigChanged(
    117       const net::ProxyConfig& config,
    118       net::ProxyConfigService::ConfigAvailability availability) OVERRIDE {
    119     LOG(INFO) << "OnProxyConfigChanged("
    120               << ProxyConfigToString(config) << ", "
    121               << ConfigAvailabilityToString(availability) << ")";
    122   }
    123 
    124  private:
    125   DISALLOW_COPY_AND_ASSIGN(NetWatcher);
    126 };
    127 
    128 }  // namespace
    129 
    130 int main(int argc, char* argv[]) {
    131 #if defined(OS_MACOSX)
    132   base::mac::ScopedNSAutoreleasePool pool;
    133 #endif
    134 #if (defined(OS_LINUX) || defined(OS_OPENBSD)) && !defined(OS_CHROMEOS)
    135   // Needed so ProxyConfigServiceLinux can use gconf.
    136   // Normally handled by BrowserMainLoop::InitializeToolkit().
    137   // From glib version 2.36 onwards, g_type_init is implicitly called and it is
    138   // deprecated.
    139   // TODO(yael) Simplify this once Ubuntu 10.04 is no longer supported.
    140 #if defined(G_GNUC_BEGIN_IGNORE_DEPRECATIONS) && \
    141     defined(G_GNUC_END_IGNORE_DEPRECATIONS)
    142 #define USE_GLIB_DEPRECATIONS_MACROS
    143 #endif
    144 
    145 #if defined(USE_GLIB_DEPRECATIONS_MACROS)
    146 G_GNUC_BEGIN_IGNORE_DEPRECATIONS
    147 #endif
    148   g_type_init();
    149 #if defined(USE_GLIB_DEPRECATIONS_MACROS)
    150 G_GNUC_END_IGNORE_DEPRECATIONS
    151 #endif
    152 #undef USE_GLIB_DEPRECATIONS_MACROS
    153 #endif  // (defined(OS_LINUX) || defined(OS_OPENBSD)) && !defined(OS_CHROMEOS)
    154   base::AtExitManager exit_manager;
    155   CommandLine::Init(argc, argv);
    156   logging::LoggingSettings settings;
    157   settings.logging_dest = logging::LOG_TO_SYSTEM_DEBUG_LOG;
    158   logging::InitLogging(settings);
    159 
    160   // Just make the main message loop the network loop.
    161   base::MessageLoopForIO network_loop;
    162 
    163   NetWatcher net_watcher;
    164 
    165   scoped_ptr<net::NetworkChangeNotifier> network_change_notifier(
    166       net::NetworkChangeNotifier::Create());
    167 
    168   // Use the network loop as the file loop also.
    169   scoped_ptr<net::ProxyConfigService> proxy_config_service(
    170       net::ProxyService::CreateSystemProxyConfigService(
    171           network_loop.message_loop_proxy().get(), &network_loop));
    172 
    173   // Uses |network_change_notifier|.
    174   net::NetworkChangeNotifier::AddIPAddressObserver(&net_watcher);
    175   net::NetworkChangeNotifier::AddConnectionTypeObserver(&net_watcher);
    176   net::NetworkChangeNotifier::AddDNSObserver(&net_watcher);
    177   net::NetworkChangeNotifier::AddNetworkChangeObserver(&net_watcher);
    178 
    179   proxy_config_service->AddObserver(&net_watcher);
    180 
    181   LOG(INFO) << "Initial connection type: "
    182             << ConnectionTypeToString(
    183                 network_change_notifier->GetCurrentConnectionType());
    184 
    185   {
    186     net::ProxyConfig config;
    187     const net::ProxyConfigService::ConfigAvailability availability =
    188         proxy_config_service->GetLatestProxyConfig(&config);
    189     LOG(INFO) << "Initial proxy config: "
    190               << ProxyConfigToString(config) << ", "
    191               << ConfigAvailabilityToString(availability);
    192   }
    193 
    194   LOG(INFO) << "Watching for network events...";
    195 
    196   // Start watching for events.
    197   network_loop.Run();
    198 
    199   proxy_config_service->RemoveObserver(&net_watcher);
    200 
    201   // Uses |network_change_notifier|.
    202   net::NetworkChangeNotifier::RemoveDNSObserver(&net_watcher);
    203   net::NetworkChangeNotifier::RemoveConnectionTypeObserver(&net_watcher);
    204   net::NetworkChangeNotifier::RemoveIPAddressObserver(&net_watcher);
    205   net::NetworkChangeNotifier::RemoveNetworkChangeObserver(&net_watcher);
    206 
    207   return 0;
    208 }
    209