1// Copyright 2013 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 "chrome/browser/local_discovery/service_discovery_shared_client.h"
6
7#include "content/public/browser/browser_thread.h"
8
9#if defined(OS_WIN)
10#include "base/files/file_path.h"
11#include "base/metrics/histogram.h"
12#include "base/path_service.h"
13#include "base/timer/elapsed_timer.h"
14#include "chrome/installer/util/browser_distribution.h"
15#include "chrome/installer/util/firewall_manager_win.h"
16#endif  // OS_WIN
17
18#if defined(OS_MACOSX)
19#include "chrome/browser/local_discovery/service_discovery_client_mac_factory.h"
20#endif
21
22#if defined(ENABLE_MDNS)
23#include "chrome/browser/local_discovery/service_discovery_client_mdns.h"
24#include "chrome/browser/local_discovery/service_discovery_client_utility.h"
25#endif  // ENABLE_MDNS
26
27namespace {
28
29#if defined(OS_WIN)
30
31bool g_is_firewall_ready = false;
32bool g_is_firewall_state_reported = false;
33
34void ReportFirewallStats() {
35  if (g_is_firewall_state_reported)
36    return;
37  g_is_firewall_state_reported = true;
38  base::FilePath exe_path;
39  if (!PathService::Get(base::FILE_EXE, &exe_path))
40    return;
41  base::ElapsedTimer timer;
42  scoped_ptr<installer::FirewallManager> manager =
43      installer::FirewallManager::Create(BrowserDistribution::GetDistribution(),
44                                         exe_path);
45  if (!manager)
46    return;
47  g_is_firewall_ready = manager->CanUseLocalPorts();
48  UMA_HISTOGRAM_TIMES("LocalDiscovery.FirewallAccessTime", timer.Elapsed());
49  UMA_HISTOGRAM_BOOLEAN("LocalDiscovery.IsFirewallReady", g_is_firewall_ready);
50}
51#endif  // OS_WIN
52
53}  // namespace
54
55
56namespace local_discovery {
57
58using content::BrowserThread;
59
60namespace {
61ServiceDiscoverySharedClient* g_service_discovery_client = NULL;
62}  // namespace
63
64ServiceDiscoverySharedClient::ServiceDiscoverySharedClient() {
65  DCHECK(!g_service_discovery_client);
66  g_service_discovery_client = this;
67}
68
69ServiceDiscoverySharedClient::~ServiceDiscoverySharedClient() {
70  DCHECK_EQ(g_service_discovery_client, this);
71  g_service_discovery_client = NULL;
72}
73
74#if defined(ENABLE_MDNS) || defined(OS_MACOSX)
75
76scoped_refptr<ServiceDiscoverySharedClient>
77    ServiceDiscoverySharedClient::GetInstance() {
78  DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
79
80  if (g_service_discovery_client)
81    return g_service_discovery_client;
82
83#if defined(OS_MACOSX)
84  return ServiceDiscoveryClientMacFactory::CreateInstance();
85#else  // OS_MACOSX
86
87  return new ServiceDiscoveryClientMdns();
88#endif  // OS_MACOSX
89}
90
91// static
92void ServiceDiscoverySharedClient::GetInstanceWithoutAlert(
93    const GetInstanceCallback& callback) {
94#if !defined(OS_WIN)
95
96  scoped_refptr<ServiceDiscoverySharedClient> result = GetInstance();
97  return callback.Run(result);
98
99#else   // OS_WIN
100  DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
101  // TODO(vitalybuka): Switch to |ServiceDiscoveryClientMdns| after we find what
102  // to do with firewall for user-level installs. crbug.com/366408
103  scoped_refptr<ServiceDiscoverySharedClient> result =
104      g_service_discovery_client;
105  if (result)
106    return callback.Run(result);
107
108  if (!g_is_firewall_state_reported) {
109    BrowserThread::PostTaskAndReply(
110        BrowserThread::FILE,
111        FROM_HERE,
112        base::Bind(&ReportFirewallStats),
113        base::Bind(&ServiceDiscoverySharedClient::GetInstanceWithoutAlert,
114                   callback));
115    return;
116  }
117
118  result =
119      g_is_firewall_ready ? GetInstance() : new ServiceDiscoveryClientUtility();
120  callback.Run(result);
121#endif  // OS_WIN
122}
123
124#else
125
126scoped_refptr<ServiceDiscoverySharedClient>
127    ServiceDiscoverySharedClient::GetInstance() {
128  DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
129  NOTIMPLEMENTED();
130  return NULL;
131}
132
133#endif  // ENABLE_MDNS
134
135}  // namespace local_discovery
136