1 // Copyright (c) 2012 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/curtain_mode.h" 6 7 #include "base/logging.h" 8 #include "base/single_thread_task_runner.h" 9 #include "base/win/windows_version.h" 10 #include "remoting/host/client_session_control.h" 11 12 namespace remoting { 13 14 class CurtainModeWin : public CurtainMode { 15 public: 16 CurtainModeWin(); 17 18 // Overriden from CurtainMode. 19 virtual bool Activate() OVERRIDE; 20 21 private: 22 DISALLOW_COPY_AND_ASSIGN(CurtainModeWin); 23 }; 24 CurtainModeWin()25CurtainModeWin::CurtainModeWin() { 26 } 27 Activate()28bool CurtainModeWin::Activate() { 29 if (base::win::GetVersion() < base::win::VERSION_VISTA) { 30 LOG(ERROR) << "Curtain mode is not supported on Windows XP/2003"; 31 return false; 32 } 33 34 DWORD session_id; 35 if (!ProcessIdToSessionId(GetCurrentProcessId(), &session_id)) { 36 LOG_GETLASTERROR(ERROR) << "Failed to map the current PID to session ID"; 37 return false; 38 } 39 40 // The current session is curtained if it is not attached to the local 41 // console. 42 return WTSGetActiveConsoleSessionId() != session_id; 43 } 44 45 // static Create(scoped_refptr<base::SingleThreadTaskRunner> caller_task_runner,scoped_refptr<base::SingleThreadTaskRunner> ui_task_runner,base::WeakPtr<ClientSessionControl> client_session_control)46scoped_ptr<CurtainMode> CurtainMode::Create( 47 scoped_refptr<base::SingleThreadTaskRunner> caller_task_runner, 48 scoped_refptr<base::SingleThreadTaskRunner> ui_task_runner, 49 base::WeakPtr<ClientSessionControl> client_session_control) { 50 // |client_session_control| is not used because the client session is 51 // disconnected as soon as the session is re-attached to the local console. 52 // See RdpDesktopSession for more details. 53 return scoped_ptr<CurtainMode>(new CurtainModeWin()); 54 } 55 56 } // namespace remoting 57