Commit 9b785217 authored by Robbie McElrath's avatar Robbie McElrath Committed by Commit Bot

Migrate PrivetTrafficDetector to use NetworkConnectionTracker.

This migrates PrivetTrafficDetector from using
net::NetworkChangeNotifier to content::NetworkConnectionTracker, which
works with the network service enabled.

Bug: 859134
Change-Id: I06f2ecbe142bbed9a8772f69ab109b0c0dc32581
Reviewed-on: https://chromium-review.googlesource.com/1123681Reviewed-by: default avatarJulian Pastarmov <pastarmovj@chromium.org>
Reviewed-by: default avatarLei Zhang <thestig@chromium.org>
Commit-Queue: Robbie McElrath <rmcelrath@chromium.org>
Cr-Commit-Position: refs/heads/master@{#574700}
parent b20bb048
...@@ -14,8 +14,10 @@ ...@@ -14,8 +14,10 @@
#include "base/task_scheduler/post_task.h" #include "base/task_scheduler/post_task.h"
#include "base/threading/thread_restrictions.h" #include "base/threading/thread_restrictions.h"
#include "base/threading/thread_task_runner_handle.h" #include "base/threading/thread_task_runner_handle.h"
#include "chrome/browser/browser_process.h"
#include "net/base/ip_address.h" #include "net/base/ip_address.h"
#include "net/base/net_errors.h" #include "net/base/net_errors.h"
#include "net/base/network_change_notifier.h"
#include "net/base/network_interfaces.h" #include "net/base/network_interfaces.h"
#include "net/dns/dns_protocol.h" #include "net/dns/dns_protocol.h"
#include "net/dns/dns_response.h" #include "net/dns/dns_response.h"
...@@ -83,20 +85,22 @@ void PrivetTrafficDetector::Start() { ...@@ -83,20 +85,22 @@ void PrivetTrafficDetector::Start() {
PrivetTrafficDetector::~PrivetTrafficDetector() { PrivetTrafficDetector::~PrivetTrafficDetector() {
DCHECK_CURRENTLY_ON(content::BrowserThread::IO); DCHECK_CURRENTLY_ON(content::BrowserThread::IO);
net::NetworkChangeNotifier::RemoveNetworkChangeObserver(this); g_browser_process->network_connection_tracker()
->RemoveNetworkConnectionObserver(this);
} }
void PrivetTrafficDetector::StartOnIOThread() { void PrivetTrafficDetector::StartOnIOThread() {
DCHECK_CURRENTLY_ON(content::BrowserThread::IO); DCHECK_CURRENTLY_ON(content::BrowserThread::IO);
net::NetworkChangeNotifier::AddNetworkChangeObserver(this); g_browser_process->network_connection_tracker()->AddNetworkConnectionObserver(
this);
ScheduleRestart(); ScheduleRestart();
} }
void PrivetTrafficDetector::OnNetworkChanged( void PrivetTrafficDetector::OnConnectionChanged(
net::NetworkChangeNotifier::ConnectionType type) { network::mojom::ConnectionType type) {
DCHECK_CURRENTLY_ON(content::BrowserThread::IO); DCHECK_CURRENTLY_ON(content::BrowserThread::IO);
restart_attempts_ = kMaxRestartAttempts; restart_attempts_ = kMaxRestartAttempts;
if (type != net::NetworkChangeNotifier::CONNECTION_NONE) if (type != network::mojom::ConnectionType::CONNECTION_NONE)
ScheduleRestart(); ScheduleRestart();
} }
......
...@@ -11,9 +11,9 @@ ...@@ -11,9 +11,9 @@
#include "base/cancelable_callback.h" #include "base/cancelable_callback.h"
#include "base/macros.h" #include "base/macros.h"
#include "content/public/browser/browser_thread.h" #include "content/public/browser/browser_thread.h"
#include "content/public/browser/network_connection_tracker.h"
#include "net/base/address_family.h" #include "net/base/address_family.h"
#include "net/base/ip_endpoint.h" #include "net/base/ip_endpoint.h"
#include "net/base/network_change_notifier.h"
namespace net { namespace net {
class DatagramServerSocket; class DatagramServerSocket;
...@@ -28,8 +28,9 @@ namespace cloud_print { ...@@ -28,8 +28,9 @@ namespace cloud_print {
// When traffic is detected, class fires callback and shutdowns itself. // When traffic is detected, class fires callback and shutdowns itself.
class PrivetTrafficDetector class PrivetTrafficDetector
: public base::RefCountedThreadSafe< : public base::RefCountedThreadSafe<
PrivetTrafficDetector, content::BrowserThread::DeleteOnIOThread>, PrivetTrafficDetector,
private net::NetworkChangeNotifier::NetworkChangeObserver { content::BrowserThread::DeleteOnIOThread>,
private content::NetworkConnectionTracker::NetworkConnectionObserver {
public: public:
PrivetTrafficDetector(net::AddressFamily address_family, PrivetTrafficDetector(net::AddressFamily address_family,
const base::Closure& on_traffic_detected); const base::Closure& on_traffic_detected);
...@@ -42,9 +43,8 @@ class PrivetTrafficDetector ...@@ -42,9 +43,8 @@ class PrivetTrafficDetector
friend class base::DeleteHelper<PrivetTrafficDetector>; friend class base::DeleteHelper<PrivetTrafficDetector>;
~PrivetTrafficDetector() override; ~PrivetTrafficDetector() override;
// net::NetworkChangeNotifier::NetworkChangeObserver implementation. // content::NetworkConnectionTracker::NetworkConnectionObserver:
void OnNetworkChanged( void OnConnectionChanged(network::mojom::ConnectionType type) override;
net::NetworkChangeNotifier::ConnectionType type) override;
void StartOnIOThread(); void StartOnIOThread();
void ScheduleRestart(); void ScheduleRestart();
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment