Home | History | Annotate | Download | only in host
      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 
     25 CurtainModeWin::CurtainModeWin() {
     26 }
     27 
     28 bool 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
     46 scoped_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