Enable chrome.mdns for Chrome Apps

chrome/browser/extensions/api/mdns/mdns_api.cc

Index: chrome/browser/extensions/api/mdns/mdns_api.cc
diff --git a/chrome/browser/extensions/api/mdns/mdns_api.cc b/chrome/browser/extensions/api/mdns/mdns_api.cc
index 572ae1eec7b08e50906f96f94aeda0314b75d1c0..4e712861cabadae4f9828eb2f9a922f381e8950a 100644
--- a/chrome/browser/extensions/api/mdns/mdns_api.cc
+++ b/chrome/browser/extensions/api/mdns/mdns_api.cc
@@ -10,6 +10,7 @@
 #include "base/profiler/scoped_profile.h"
 #include "chrome/browser/extensions/extension_service.h"
 #include "chrome/common/extensions/api/mdns.h"
+#include "extensions/browser/extension_registry.h"
 
 namespace extensions {
 
@@ -32,8 +33,9 @@ bool IsServiceTypeWhitelisted(const std::string& service_type) {
 
 MDnsAPI::MDnsAPI(content::BrowserContext* context) : browser_context_(context) {
   DCHECK(browser_context_);
-  EventRouter::Get(context)
-      ->RegisterObserver(this, mdns::OnServiceList::kEventName);
+  extensions::EventRouter* event_router = EventRouter::Get(context);
+  DCHECK(event_router);
+  event_router->RegisterObserver(this, mdns::OnServiceList::kEventName);
 }
 
 MDnsAPI::~MDnsAPI() {
@@ -58,6 +60,8 @@ BrowserContextKeyedAPIFactory<MDnsAPI>* MDnsAPI::GetFactoryInstance() {
 void MDnsAPI::SetDnsSdRegistryForTesting(
     scoped_ptr<DnsSdRegistry> dns_sd_registry) {
   dns_sd_registry_ = dns_sd_registry.Pass();
+  if (dns_sd_registry_.get())
+    dns_sd_registry_.get()->AddObserver(this);
 }
 
 DnsSdRegistry* MDnsAPI::dns_sd_registry() {
@@ -86,7 +90,7 @@ void MDnsAPI::OnListenerRemoved(const EventListenerInfo& details) {
 void MDnsAPI::UpdateMDnsListeners(const EventListenerInfo& details) {
   std::set<std::string> new_service_types;
 
-  // Check all listeners for service type filers.
+  // Check all listeners for service type filters.
   const EventListenerMap::ListenerList& listeners =
       extensions::EventRouter::Get(browser_context_)
           ->listeners()
@@ -99,6 +103,19 @@ void MDnsAPI::UpdateMDnsListeners(const EventListenerInfo& details) {
     filter->GetStringASCII(kEventFilterServiceTypeKey, &filter_value);
     if (filter_value.empty())
       continue;
+
+    const Extension* extension = ExtensionRegistry::Get(browser_context_)->
+        enabled_extensions().GetByID((*it)->extension_id());
+    // Don't listen for services associated only with disabled extensions.
+    if (!extension)
+      continue;
+
+    // Platform apps may query for all services; other types of extensions are
+    // restricted to a whitelist.
+    if (!extension->is_platform_app() &&
+        !IsServiceTypeWhitelisted(filter_value))
+      continue;
+
     new_service_types.insert(filter_value);
   }
 
@@ -114,15 +131,12 @@ void MDnsAPI::UpdateMDnsListeners(const EventListenerInfo& details) {
   DnsSdRegistry* registry = dns_sd_registry();
   for (std::set<std::string>::iterator it = added_service_types.begin();

[Vitaly Buka (NO REVIEWS), 2015/03/18 19:07:00]

maybe: for (const auto& srv: added_service_types)

[Red Daly, 2015/03/27 20:06:08]

Done.

        it != added_service_types.end(); ++it) {
-    if (IsServiceTypeWhitelisted(*it))
-      registry->RegisterDnsSdListener(*it);
+    registry->RegisterDnsSdListener(*it);
   }
   for (std::set<std::string>::iterator it = removed_service_types.begin();
        it != removed_service_types.end(); ++it) {
-    if (IsServiceTypeWhitelisted(*it))
-      registry->UnregisterDnsSdListener(*it);
+    registry->UnregisterDnsSdListener(*it);
   }
-
   service_types_ = new_service_types;
 }
 
@@ -148,10 +162,10 @@ void MDnsAPI::OnDnsSdEvent(const std::string& service_type,
   event->restrict_to_browser_context = browser_context_;
   event->filter_info.SetServiceType(service_type);
 
-  VLOG(1) << "Broadcasting OnServiceList event: " << event.get();
-
   // TODO(justinlin): To avoid having listeners without filters getting all
   // events, modify API to have this event require filters.
+  // TODO(reddaly): If event isn't on whitelist, ensure it does not get
+  // broadcast to extensions.
   extensions::EventRouter::Get(browser_context_)->BroadcastEvent(event.Pass());
 }