net/base/network_change_notifier_win.cc

   1 // Copyright (c) 2009 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 #include "net/base/network_change_notifier_win.h"
   6
   7 #include <iphlpapi.h>
   8 #include <windows.h>
   9 #include <winsock2.h>
  10
  11 #include "base/basictypes.h"
  12 #include "base/compiler_specific.h"
  13 #include "base/logging.h"
  14 #include "base/object_watcher.h"
  15
  16 namespace net {
  17
  18 class NetworkChangeNotifierWin::Impl
  19     : public base::ObjectWatcher::Delegate {
  20  public:
  21   explicit Impl(NetworkChangeNotifierWin* notifier);
  22   virtual ~Impl();
  23
  24   void WatchForAddressChange();
  25
  26   // ObjectWatcher::Delegate methods:
  27
  28   virtual void OnObjectSignaled(HANDLE object);
  29
  30  private:
  31   NetworkChangeNotifierWin* const notifier_;
  32   base::ObjectWatcher addr_watcher_;
  33   OVERLAPPED addr_overlapped_;
  34
  35   DISALLOW_COPY_AND_ASSIGN(Impl);
  36 };
  37
  38 NetworkChangeNotifierWin::Impl::Impl(NetworkChangeNotifierWin* notifier)
  39     : notifier_(notifier) {
  40   memset(&addr_overlapped_, 0, sizeof(addr_overlapped_));
  41   addr_overlapped_.hEvent = WSACreateEvent();
  42 }
  43
  44 NetworkChangeNotifierWin::Impl::~Impl() {
  45   CancelIPChangeNotify(&addr_overlapped_);
  46   addr_watcher_.StopWatching();
  47   WSACloseEvent(addr_overlapped_.hEvent);
  48   memset(&addr_overlapped_, 0, sizeof(addr_overlapped_));
  49 }
  50
  51 void NetworkChangeNotifierWin::Impl::WatchForAddressChange() {
  52   HANDLE handle = NULL;
  53   DWORD ret = NotifyAddrChange(&handle, &addr_overlapped_);
  54   CHECK(ret == ERROR_IO_PENDING);
  55   addr_watcher_.StartWatching(addr_overlapped_.hEvent, this);
  56 }
  57
  58 void NetworkChangeNotifierWin::Impl::OnObjectSignaled(HANDLE object) {
  59   notifier_->OnIPAddressChanged();
  60
  61   // Start watching for further address changes.
  62   WatchForAddressChange();
  63 }
  64
  65 NetworkChangeNotifierWin::NetworkChangeNotifierWin()
  66     : impl_(new Impl(ALLOW_THIS_IN_INITIALIZER_LIST(this))) {
  67   impl_->WatchForAddressChange();
  68 }
  69
  70 NetworkChangeNotifierWin::~NetworkChangeNotifierWin() {}
  71
  72 }  // namespace net