Capability Classes

mojo/shell/public/cpp/lib/connection_impl.h

Index: mojo/shell/public/cpp/lib/connection_impl.h
diff --git a/mojo/shell/public/cpp/lib/connection_impl.h b/mojo/shell/public/cpp/lib/connection_impl.h
index 3247a5a73ea094989b54644deaa8213f3ca9d97c..2c58d86c0cc26324131af561400d8f6d226f385b 100644
--- a/mojo/shell/public/cpp/lib/connection_impl.h
+++ b/mojo/shell/public/cpp/lib/connection_impl.h
@@ -11,6 +11,7 @@
 #include <string>
 
 #include "mojo/public/cpp/bindings/binding.h"
+#include "mojo/shell/public/cpp/capabilities.h"
 #include "mojo/shell/public/cpp/connection.h"
 #include "mojo/shell/public/cpp/identity.h"
 #include "mojo/shell/public/interfaces/connector.mojom.h"
@@ -33,7 +34,7 @@ class ConnectionImpl : public Connection {
                  uint32_t remote_id,
                  shell::mojom::InterfaceProviderPtr remote_interfaces,
                  shell::mojom::InterfaceProviderRequest local_interfaces,
-                 const std::set<std::string>& allowed_interfaces,
+                 const CapabilityRequest& capability_request,
                  State initial_state);
   ~ConnectionImpl() override;
 
@@ -41,6 +42,7 @@ class ConnectionImpl : public Connection {
 
  private:
   // Connection:
+  bool HasCapabilityClass(const std::string& class_name) const override;
   const std::string& GetConnectionName() override;
   const Identity& GetRemoteIdentity() const override;
   void SetConnectionLostClosure(const Closure& handler) override;
@@ -68,7 +70,7 @@ class ConnectionImpl : public Connection {
   InterfaceRegistry local_registry_;
   shell::mojom::InterfaceProviderPtr remote_interfaces_;
 
-  const std::set<std::string> allowed_interfaces_;
+  const CapabilityRequest capability_request_;
   const bool allow_all_interfaces_;
 
   base::WeakPtrFactory<ConnectionImpl> weak_factory_;