| OLD | NEW |
| (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 #include "remoting/host/security_key/gnubby_auth_handler.h" | |
| 6 | |
| 7 #include <cstdint> | |
| 8 #include <map> | |
| 9 #include <memory> | |
| 10 #include <string> | |
| 11 | |
| 12 #include "base/bind.h" | |
| 13 #include "base/location.h" | |
| 14 #include "base/logging.h" | |
| 15 #include "base/memory/weak_ptr.h" | |
| 16 #include "base/strings/stringprintf.h" | |
| 17 #include "base/strings/utf_string_conversions.h" | |
| 18 #include "base/threading/thread_checker.h" | |
| 19 #include "base/threading/thread_task_runner_handle.h" | |
| 20 #include "base/time/time.h" | |
| 21 #include "base/timer/timer.h" | |
| 22 #include "base/win/win_util.h" | |
| 23 #include "ipc/ipc_channel.h" | |
| 24 #include "ipc/ipc_listener.h" | |
| 25 #include "ipc/ipc_message.h" | |
| 26 #include "ipc/ipc_message_macros.h" | |
| 27 #include "remoting/base/logging.h" | |
| 28 #include "remoting/host/chromoting_messages.h" | |
| 29 #include "remoting/host/client_session_details.h" | |
| 30 #include "remoting/host/ipc_util.h" | |
| 31 #include "remoting/host/security_key/remote_security_key_ipc_constants.h" | |
| 32 #include "remoting/host/security_key/remote_security_key_ipc_server.h" | |
| 33 | |
| 34 namespace { | |
| 35 | |
| 36 // The timeout used to disconnect a client from the IPC Server channel if it | |
| 37 // forgets to do so. This ensures the server channel is not blocked forever. | |
| 38 const int kInitialRequestTimeoutSeconds = 5; | |
| 39 | |
| 40 // This value represents the amount of time to wait for a gnubby request from | |
| 41 // the client before terminating the connection. | |
| 42 const int kGnubbyRequestTimeoutSeconds = 60; | |
| 43 | |
| 44 } // namespace | |
| 45 | |
| 46 namespace remoting { | |
| 47 | |
| 48 // Creates an IPC server channel which services IPC clients that want to start | |
| 49 // a security key forwarding session. Once an IPC Client connects to the | |
| 50 // server, the GnubbyAuthHandlerWin class will create a new | |
| 51 // RemoteSecurityKeyIpcServer instance that will service that request. The new | |
| 52 // instance will exist for the lifetime of the security key request and will be | |
| 53 // assigned a unique IPC channel name and connection id. The channel name is | |
| 54 // sent to the client which should disconnect the IPC server channel and | |
| 55 // connect to the security key forwarding session IPC channel to send/receive | |
| 56 // security key messages. The IPC server channel will then be reset so it can | |
| 57 // can service the next client/request. This system allows multiple security | |
| 58 // key forwarding sessions to occur concurrently. | |
| 59 // TODO(joedow): Update GnubbyAuthHandler impls to run on a separate IO thread | |
| 60 // instead of the thread it was created on: crbug.com/591739 | |
| 61 class GnubbyAuthHandlerWin : public GnubbyAuthHandler, public IPC::Listener { | |
| 62 public: | |
| 63 explicit GnubbyAuthHandlerWin(ClientSessionDetails* client_session_details); | |
| 64 ~GnubbyAuthHandlerWin() override; | |
| 65 | |
| 66 private: | |
| 67 typedef std::map<int, std::unique_ptr<RemoteSecurityKeyIpcServer>> | |
| 68 ActiveChannels; | |
| 69 | |
| 70 // GnubbyAuthHandler interface. | |
| 71 void CreateGnubbyConnection() override; | |
| 72 bool IsValidConnectionId(int gnubby_connection_id) const override; | |
| 73 void SendClientResponse(int gnubby_connection_id, | |
| 74 const std::string& response) override; | |
| 75 void SendErrorAndCloseConnection(int gnubby_connection_id) override; | |
| 76 void SetSendMessageCallback(const SendMessageCallback& callback) override; | |
| 77 size_t GetActiveConnectionCountForTest() const override; | |
| 78 void SetRequestTimeoutForTest(base::TimeDelta timeout) override; | |
| 79 | |
| 80 // IPC::Listener implementation. | |
| 81 bool OnMessageReceived(const IPC::Message& message) override; | |
| 82 void OnChannelConnected(int32_t peer_pid) override; | |
| 83 void OnChannelError() override; | |
| 84 | |
| 85 // Creates the IPC server channel and waits for a connection using | |
| 86 // |ipc_server_channel_name_|. | |
| 87 void StartIpcServerChannel(); | |
| 88 | |
| 89 // Restarts the IPC server channel to prepare for another connection. | |
| 90 void RecreateIpcServerChannel(); | |
| 91 | |
| 92 // Closes the IPC channel created for a security key forwarding session. | |
| 93 void CloseSecurityKeyRequestIpcChannel(int connection_id); | |
| 94 | |
| 95 // Returns the IPC Channel instance created for |connection_id|. | |
| 96 ActiveChannels::const_iterator GetChannelForConnectionId( | |
| 97 int connection_id) const; | |
| 98 | |
| 99 // Creates a unique name based on the well-known IPC channel name. | |
| 100 std::string GenerateUniqueChannelName(); | |
| 101 | |
| 102 // Represents the last id assigned to a new security key request IPC channel. | |
| 103 int last_connection_id_ = 0; | |
| 104 | |
| 105 // Sends a gnubby extension messages to the remote client when called. | |
| 106 SendMessageCallback send_message_callback_; | |
| 107 | |
| 108 // Interface which provides details about the client session. | |
| 109 ClientSessionDetails* client_session_details_ = nullptr; | |
| 110 | |
| 111 // Tracks the IPC channel created for each security key forwarding session. | |
| 112 ActiveChannels active_channels_; | |
| 113 | |
| 114 // The amount of time to wait for a client to process the connection details | |
| 115 // message and disconnect from the IPC server channel before disconnecting it. | |
| 116 base::TimeDelta disconnect_timeout_; | |
| 117 | |
| 118 // Used to recreate the IPC server channel if a client forgets to disconnect. | |
| 119 base::OneShotTimer timer_; | |
| 120 | |
| 121 // IPC Clients connect to this channel first to receive their own IPC | |
| 122 // channel to start a security key forwarding session on. | |
| 123 std::unique_ptr<IPC::Channel> ipc_server_channel_; | |
| 124 | |
| 125 // Ensures GnubbyAuthHandlerWin methods are called on the same thread. | |
| 126 base::ThreadChecker thread_checker_; | |
| 127 | |
| 128 base::WeakPtrFactory<GnubbyAuthHandlerWin> weak_factory_; | |
| 129 | |
| 130 DISALLOW_COPY_AND_ASSIGN(GnubbyAuthHandlerWin); | |
| 131 }; | |
| 132 | |
| 133 std::unique_ptr<GnubbyAuthHandler> GnubbyAuthHandler::Create( | |
| 134 ClientSessionDetails* client_session_details, | |
| 135 const SendMessageCallback& send_message_callback) { | |
| 136 std::unique_ptr<GnubbyAuthHandler> auth_handler( | |
| 137 new GnubbyAuthHandlerWin(client_session_details)); | |
| 138 auth_handler->SetSendMessageCallback(send_message_callback); | |
| 139 return auth_handler; | |
| 140 } | |
| 141 | |
| 142 GnubbyAuthHandlerWin::GnubbyAuthHandlerWin( | |
| 143 ClientSessionDetails* client_session_details) | |
| 144 : client_session_details_(client_session_details), | |
| 145 disconnect_timeout_( | |
| 146 base::TimeDelta::FromSeconds(kInitialRequestTimeoutSeconds)), | |
| 147 weak_factory_(this) { | |
| 148 DCHECK(client_session_details_); | |
| 149 } | |
| 150 | |
| 151 GnubbyAuthHandlerWin::~GnubbyAuthHandlerWin() {} | |
| 152 | |
| 153 void GnubbyAuthHandlerWin::CreateGnubbyConnection() { | |
| 154 DCHECK(thread_checker_.CalledOnValidThread()); | |
| 155 StartIpcServerChannel(); | |
| 156 } | |
| 157 | |
| 158 bool GnubbyAuthHandlerWin::IsValidConnectionId(int connection_id) const { | |
| 159 DCHECK(thread_checker_.CalledOnValidThread()); | |
| 160 return (GetChannelForConnectionId(connection_id) != active_channels_.end()); | |
| 161 } | |
| 162 | |
| 163 void GnubbyAuthHandlerWin::SendClientResponse( | |
| 164 int connection_id, | |
| 165 const std::string& response_data) { | |
| 166 DCHECK(thread_checker_.CalledOnValidThread()); | |
| 167 | |
| 168 ActiveChannels::const_iterator iter = | |
| 169 GetChannelForConnectionId(connection_id); | |
| 170 if (iter == active_channels_.end()) { | |
| 171 HOST_LOG << "Invalid gnubby connection ID received: " << connection_id; | |
| 172 return; | |
| 173 } | |
| 174 | |
| 175 if (!iter->second->SendResponse(response_data)) { | |
| 176 CloseSecurityKeyRequestIpcChannel(connection_id); | |
| 177 } | |
| 178 } | |
| 179 | |
| 180 void GnubbyAuthHandlerWin::SendErrorAndCloseConnection(int connection_id) { | |
| 181 DCHECK(thread_checker_.CalledOnValidThread()); | |
| 182 | |
| 183 SendClientResponse(connection_id, kRemoteSecurityKeyConnectionError); | |
| 184 CloseSecurityKeyRequestIpcChannel(connection_id); | |
| 185 } | |
| 186 | |
| 187 void GnubbyAuthHandlerWin::SetSendMessageCallback( | |
| 188 const SendMessageCallback& callback) { | |
| 189 DCHECK(thread_checker_.CalledOnValidThread()); | |
| 190 send_message_callback_ = callback; | |
| 191 } | |
| 192 | |
| 193 size_t GnubbyAuthHandlerWin::GetActiveConnectionCountForTest() const { | |
| 194 return active_channels_.size(); | |
| 195 } | |
| 196 | |
| 197 void GnubbyAuthHandlerWin::SetRequestTimeoutForTest(base::TimeDelta timeout) { | |
| 198 disconnect_timeout_ = timeout; | |
| 199 } | |
| 200 | |
| 201 void GnubbyAuthHandlerWin::StartIpcServerChannel() { | |
| 202 DCHECK(thread_checker_.CalledOnValidThread()); | |
| 203 | |
| 204 // Create a named pipe owned by the current user (the LocalService account | |
| 205 // (SID: S-1-5-19) when running in the network process) which is available to | |
| 206 // all authenticated users. | |
| 207 // presubmit: allow wstring | |
| 208 std::wstring user_sid; | |
| 209 CHECK(base::win::GetUserSidString(&user_sid)); | |
| 210 std::string user_sid_utf8 = base::WideToUTF8(user_sid); | |
| 211 std::string security_descriptor = base::StringPrintf( | |
| 212 "O:%sG:%sD:(A;;GA;;;AU)", user_sid_utf8.c_str(), user_sid_utf8.c_str()); | |
| 213 | |
| 214 base::win::ScopedHandle pipe; | |
| 215 CHECK(CreateIpcChannel(remoting::GetRemoteSecurityKeyIpcChannelName(), | |
| 216 security_descriptor, &pipe)); | |
| 217 ipc_server_channel_ = | |
| 218 IPC::Channel::CreateNamedServer(IPC::ChannelHandle(pipe.Get()), this); | |
| 219 CHECK(ipc_server_channel_->Connect()); | |
| 220 } | |
| 221 | |
| 222 void GnubbyAuthHandlerWin::RecreateIpcServerChannel() { | |
| 223 DCHECK(thread_checker_.CalledOnValidThread()); | |
| 224 | |
| 225 timer_.Stop(); | |
| 226 ipc_server_channel_.reset(); | |
| 227 | |
| 228 StartIpcServerChannel(); | |
| 229 } | |
| 230 | |
| 231 void GnubbyAuthHandlerWin::CloseSecurityKeyRequestIpcChannel( | |
| 232 int connection_id) { | |
| 233 active_channels_.erase(connection_id); | |
| 234 } | |
| 235 | |
| 236 GnubbyAuthHandlerWin::ActiveChannels::const_iterator | |
| 237 GnubbyAuthHandlerWin::GetChannelForConnectionId(int connection_id) const { | |
| 238 return active_channels_.find(connection_id); | |
| 239 } | |
| 240 | |
| 241 std::string GnubbyAuthHandlerWin::GenerateUniqueChannelName() { | |
| 242 return GetRemoteSecurityKeyIpcChannelName() + "." + | |
| 243 IPC::Channel::GenerateUniqueRandomChannelID(); | |
| 244 } | |
| 245 | |
| 246 bool GnubbyAuthHandlerWin::OnMessageReceived(const IPC::Message& message) { | |
| 247 DCHECK(thread_checker_.CalledOnValidThread()); | |
| 248 // This class does handle any IPC messages sent by the client. | |
| 249 return false; | |
| 250 } | |
| 251 | |
| 252 void GnubbyAuthHandlerWin::OnChannelConnected(int32_t peer_pid) { | |
| 253 DCHECK(thread_checker_.CalledOnValidThread()); | |
| 254 | |
| 255 timer_.Start(FROM_HERE, disconnect_timeout_, | |
| 256 base::Bind(&GnubbyAuthHandlerWin::OnChannelError, | |
| 257 base::Unretained(this))); | |
| 258 | |
| 259 // Verify the IPC connection attempt originated from the session we are | |
| 260 // currently remoting. We don't want to service requests from arbitrary | |
| 261 // Windows sessions. | |
| 262 bool close_connection = false; | |
| 263 DWORD peer_session_id; | |
| 264 if (!ProcessIdToSessionId(peer_pid, &peer_session_id)) { | |
| 265 PLOG(ERROR) << "ProcessIdToSessionId() failed"; | |
| 266 close_connection = true; | |
| 267 } else if (peer_session_id != client_session_details_->desktop_session_id()) { | |
| 268 LOG(INFO) << "Ignoring connection attempt from outside remoted session."; | |
| 269 close_connection = true; | |
| 270 } | |
| 271 if (close_connection) { | |
| 272 base::ThreadTaskRunnerHandle::Get()->PostTask( | |
| 273 FROM_HERE, base::Bind(&GnubbyAuthHandlerWin::OnChannelError, | |
| 274 weak_factory_.GetWeakPtr())); | |
| 275 return; | |
| 276 } | |
| 277 | |
| 278 int new_connection_id = ++last_connection_id_; | |
| 279 std::unique_ptr<RemoteSecurityKeyIpcServer> ipc_server( | |
| 280 RemoteSecurityKeyIpcServer::Create( | |
| 281 new_connection_id, peer_session_id, disconnect_timeout_, | |
| 282 send_message_callback_, | |
| 283 base::Bind(&GnubbyAuthHandlerWin::CloseSecurityKeyRequestIpcChannel, | |
| 284 base::Unretained(this), new_connection_id))); | |
| 285 | |
| 286 std::string unique_channel_name = GenerateUniqueChannelName(); | |
| 287 if (ipc_server->CreateChannel( | |
| 288 unique_channel_name, | |
| 289 base::TimeDelta::FromSeconds(kGnubbyRequestTimeoutSeconds))) { | |
| 290 active_channels_[new_connection_id] = std::move(ipc_server); | |
| 291 ipc_server_channel_->Send( | |
| 292 new ChromotingNetworkToRemoteSecurityKeyMsg_ConnectionDetails( | |
| 293 unique_channel_name)); | |
| 294 } | |
| 295 } | |
| 296 | |
| 297 void GnubbyAuthHandlerWin::OnChannelError() { | |
| 298 DCHECK(thread_checker_.CalledOnValidThread()); | |
| 299 | |
| 300 // Could be an error, most likely the client disconnected though. Either way | |
| 301 // we should restart the server to prepare for the next connection. | |
| 302 RecreateIpcServerChannel(); | |
| 303 } | |
| 304 | |
| 305 } // namespace remoting | |
| OLD | NEW |