Index: trunk/src/chrome/browser/extensions/api/mdns/mdns_api.cc |
=================================================================== |
--- trunk/src/chrome/browser/extensions/api/mdns/mdns_api.cc (revision 221667) |
+++ trunk/src/chrome/browser/extensions/api/mdns/mdns_api.cc (working copy) |
@@ -1,154 +0,0 @@ |
-// Copyright 2013 The Chromium Authors. All rights reserved. |
-// Use of this source code is governed by a BSD-style license that can be |
-// found in the LICENSE file. |
- |
-#include "chrome/browser/extensions/api/mdns/mdns_api.h" |
- |
-#include <vector> |
- |
-#include "base/lazy_instance.h" |
-#include "chrome/browser/extensions/extension_service.h" |
-#include "chrome/browser/extensions/extension_system.h" |
-#include "chrome/common/extensions/api/mdns.h" |
- |
-namespace extensions { |
- |
-namespace mdns = api::mdns; |
- |
-namespace { |
- |
-// Whitelisted mDNS service types. |
-const char kCastServiceType[] = "_googlecast._tcp.local"; |
-const char kTestServiceType[] = "_testing._tcp.local"; |
- |
-bool IsServiceTypeWhitelisted(const std::string& service_type) { |
- return service_type == kCastServiceType || |
- service_type == kTestServiceType; |
-} |
- |
-} // namespace |
- |
-MDnsAPI::MDnsAPI(Profile* profile) : profile_(profile) { |
- DCHECK(profile_); |
- ExtensionSystem::Get(profile)->event_router()->RegisterObserver( |
- this, mdns::OnServiceList::kEventName); |
-} |
- |
-MDnsAPI::~MDnsAPI() { |
- if (dns_sd_registry_.get()) { |
- dns_sd_registry_->RemoveObserver(this); |
- } |
-} |
- |
-// static |
-MDnsAPI* MDnsAPI::Get(Profile* profile) { |
- return ProfileKeyedAPIFactory<MDnsAPI>::GetForProfile(profile); |
-} |
- |
-static base::LazyInstance<ProfileKeyedAPIFactory<MDnsAPI> > g_factory = |
- LAZY_INSTANCE_INITIALIZER; |
- |
-// static |
-ProfileKeyedAPIFactory<MDnsAPI>* MDnsAPI::GetFactoryInstance() { |
- return &g_factory.Get(); |
-} |
- |
-void MDnsAPI::SetDnsSdRegistryForTesting( |
- scoped_ptr<DnsSdRegistry> dns_sd_registry) { |
- dns_sd_registry_ = dns_sd_registry.Pass(); |
-} |
- |
-DnsSdRegistry* MDnsAPI::dns_sd_registry() { |
- DCHECK(thread_checker_.CalledOnValidThread()); |
- if (!dns_sd_registry_.get()) { |
- dns_sd_registry_.reset(new extensions::DnsSdRegistry()); |
- dns_sd_registry_->AddObserver(this); |
- } |
- return dns_sd_registry_.get(); |
-} |
- |
-void MDnsAPI::OnListenerAdded(const EventListenerInfo& details) { |
- DCHECK(thread_checker_.CalledOnValidThread()); |
- UpdateMDnsListeners(details); |
-} |
- |
-void MDnsAPI::OnListenerRemoved(const EventListenerInfo& details) { |
- DCHECK(thread_checker_.CalledOnValidThread()); |
- UpdateMDnsListeners(details); |
-} |
- |
-void MDnsAPI::UpdateMDnsListeners(const EventListenerInfo& details) { |
- std::set<std::string> new_service_types; |
- |
- // Check all listeners for service type filers. |
- const EventListenerMap::ListenerList& listeners = |
- extensions::ExtensionSystem::Get(profile_)->event_router()-> |
- listeners().GetEventListenersByName(details.event_name); |
- for (EventListenerMap::ListenerList::const_iterator it = listeners.begin(); |
- it != listeners.end(); ++it) { |
- base::DictionaryValue* filter = ((*it)->filter.get()); |
- for (base::DictionaryValue::Iterator iter(*filter); |
- !iter.IsAtEnd(); iter.Advance()) { |
- } |
- |
- std::string filter_value; |
- filter->GetStringASCII(kEventFilterServiceTypeKey, &filter_value); |
- if (filter_value.empty()) |
- continue; |
- |
- new_service_types.insert(filter_value); |
- } |
- |
- // Find all the added and removed service types since last update. |
- std::set<std::string> added_service_types = |
- base::STLSetDifference<std::set<std::string> >( |
- service_types_, new_service_types); |
- std::set<std::string> removed_service_types = |
- base::STLSetDifference<std::set<std::string> >( |
- new_service_types, service_types_); |
- |
- // Update the registry. |
- DnsSdRegistry* registry = dns_sd_registry(); |
- for (std::set<std::string>::iterator it = added_service_types.begin(); |
- it != added_service_types.end(); ++it) { |
- if (IsServiceTypeWhitelisted(*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); |
- } |
- |
- service_types_ = new_service_types; |
-} |
- |
-void MDnsAPI::OnDnsSdEvent(const std::string& service_type, |
- const DnsSdRegistry::DnsSdServiceList& services) { |
- DCHECK(thread_checker_.CalledOnValidThread()); |
- if (services.size() == 0) |
- return; |
- |
- std::vector<linked_ptr<mdns::MDnsService> > args; |
- for (DnsSdRegistry::DnsSdServiceList::const_iterator it = services.begin(); |
- it != services.end(); ++it) { |
- linked_ptr<mdns::MDnsService> mdns_service = |
- make_linked_ptr(new mdns::MDnsService); |
- // TODO(justinlin, mfoltz): Copy data from service list. |
- mdns_service->service_name = service_type; |
- args.push_back(mdns_service); |
- } |
- |
- scoped_ptr<base::ListValue> results = mdns::OnServiceList::Create(args); |
- scoped_ptr<Event> event( |
- new Event(mdns::OnServiceList::kEventName, results.Pass())); |
- event->restrict_to_profile = profile_; |
- event->filter_info.SetServiceType(service_type); |
- |
- // TODO(justinlin): To avoid having listeners without filters getting all |
- // events, modify API to have this event require filters. |
- extensions::ExtensionSystem::Get(profile_)->event_router()-> |
- BroadcastEvent(event.Pass()); |
-} |
- |
-} // namespace extensions |