Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(238)

Side by Side Diff: remoting/host/security_key/fake_remote_security_key_ipc_server.h

Issue 2162083003: Renaming Gnubby and RemoteSecurityKey files/classes/members (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@master
Patch Set: Fixing a GYP build error Created 4 years, 5 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
OLDNEW
(Empty)
1 // Copyright 2016 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 #ifndef REMOTING_HOST_SECURITY_KEY_FAKE_SECURITY_KEY_IPC_SERVER_H_
6 #define REMOTING_HOST_SECURITY_KEY_FAKE_SECURITY_KEY_IPC_SERVER_H_
7
8 #include "remoting/host/security_key/remote_security_key_ipc_server.h"
9
10 #include <cstdint>
11 #include <map>
12 #include <memory>
13 #include <string>
14
15 #include "base/callback_forward.h"
16 #include "base/macros.h"
17 #include "base/memory/weak_ptr.h"
18 #include "ipc/ipc_listener.h"
19
20 namespace IPC {
21 class Channel;
22 class Message;
23 } // IPC
24
25 namespace remoting {
26
27 // Used to send/receive security key messages for testing. It provides a
28 // WeakPtr reference to itself which allows tests to verify its lifetime is
29 // managed properly without interfering with it.
30 class FakeRemoteSecurityKeyIpcServer : public RemoteSecurityKeyIpcServer,
31 public IPC::Listener {
32 public:
33 FakeRemoteSecurityKeyIpcServer(
34 int connection_id,
35 uint32_t peer_session_id,
36 base::TimeDelta initial_connect_timeout,
37 const GnubbyAuthHandler::SendMessageCallback& send_message_callback,
38 const base::Closure& channel_closed_callback);
39 ~FakeRemoteSecurityKeyIpcServer() override;
40
41 // RemoteSecurityKeyIpcServer interface.
42 bool CreateChannel(const std::string& channel_name,
43 base::TimeDelta request_timeout) override;
44 bool SendResponse(const std::string& message_data) override;
45
46 // Simulates receipt of a security key request message.
47 void SendRequest(const std::string& message_data);
48
49 // Simulates the IPC channel being closed.
50 void CloseChannel();
51
52 // Returns a WeakPtr reference to this instance.
53 base::WeakPtr<FakeRemoteSecurityKeyIpcServer> AsWeakPtr();
54
55 // Returns the payload for the last message received.
56 const std::string& last_message_received() const {
57 return last_message_received_;
58 }
59
60 // The name of the IPC channel created by this instance.
61 const std::string& channel_name() const { return channel_name_; }
62
63 // Signaled when a security key response message is received.
64 // NOTE: Ths callback will be used instead of the IPC channel for response
65 // notifications if it is set.
66 void set_send_response_callback(const base::Closure& send_response_callback) {
67 send_response_callback_ = send_response_callback;
68 }
69
70 private:
71 // IPC::Listener interface.
72 bool OnMessageReceived(const IPC::Message& message) override;
73 void OnChannelConnected(int32_t peer_pid) override;
74 void OnChannelError() override;
75
76 // The id assigned to this IPC connection.
77 int connection_id_;
78
79 // Name of the IPC channel this instance was told to connect to.
80 std::string channel_name_;
81
82 // The payload for the last message received.
83 std::string last_message_received_;
84
85 // Used to forward security key requests to the remote client.
86 GnubbyAuthHandler::SendMessageCallback send_message_callback_;
87
88 // Signaled when the IPC channel is closed.
89 base::Closure channel_closed_callback_;
90
91 // Signaled when a security key response message is received.
92 base::Closure send_response_callback_;
93
94 // Used for sending/receiving security key messages between processes.
95 std::unique_ptr<IPC::Channel> ipc_channel_;
96
97 // NOTE: Weak pointers must be invalidated before all other member variables.
98 base::WeakPtrFactory<FakeRemoteSecurityKeyIpcServer> weak_factory_;
99
100 DISALLOW_COPY_AND_ASSIGN(FakeRemoteSecurityKeyIpcServer);
101 };
102
103 // Used to create FakeRemoteSecurityKeyIpcServer instances for testing.
104 // Provides a method which will return a WeakPtr reference to each instance
105 // this factory creates. This allows tests to inject/retrieve messages and
106 // verify the backing instance is destroyed at the appropriate time.
107 class FakeRemoteSecurityKeyIpcServerFactory
108 : public RemoteSecurityKeyIpcServerFactory {
109 public:
110 FakeRemoteSecurityKeyIpcServerFactory();
111 ~FakeRemoteSecurityKeyIpcServerFactory() override;
112
113 // RemoteSecurityKeyIpcServerFactory implementation.
114 std::unique_ptr<RemoteSecurityKeyIpcServer> Create(
115 int connection_id,
116 uint32_t peer_session_id,
117 base::TimeDelta initial_connect_timeout,
118 const GnubbyAuthHandler::SendMessageCallback& message_callback,
119 const base::Closure& done_callback) override;
120
121 // Provide a WeakPtr reference to the FakeRemoteSecurityKeyIpcServer object
122 // created for the |connection_id| IPC channel.
123 base::WeakPtr<FakeRemoteSecurityKeyIpcServer> GetIpcServerObject(
124 int connection_id);
125
126 private:
127 // Tracks each FakeRemoteSecurityKeyIpcServer instance created by this
128 // factory which allows them to be retrieved and queried for tests.
129 std::map<int, base::WeakPtr<FakeRemoteSecurityKeyIpcServer>> ipc_server_map_;
130
131 DISALLOW_COPY_AND_ASSIGN(FakeRemoteSecurityKeyIpcServerFactory);
132 };
133
134 } // namespace remoting
135
136 #endif // REMOTING_HOST_SECURITY_KEY_FAKE_SECURITY_KEY_IPC_SERVER_H_
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698