| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "remoting/host/sas_injector.h" | 5 #include "remoting/host/sas_injector.h" |
| 6 | 6 |
| 7 #include <windows.h> | 7 #include <windows.h> |
| 8 #include <string> | 8 #include <string> |
| 9 | 9 |
| 10 #include "base/logging.h" | 10 #include "base/logging.h" |
| (...skipping 16 matching lines...) Expand all Loading... |
| 27 | 27 |
| 28 // The prototype of SendSAS(). | 28 // The prototype of SendSAS(). |
| 29 typedef VOID (WINAPI *SendSasFunc)(BOOL); | 29 typedef VOID (WINAPI *SendSasFunc)(BOOL); |
| 30 | 30 |
| 31 // The registry key and value holding the policy controlling software SAS | 31 // The registry key and value holding the policy controlling software SAS |
| 32 // generation. | 32 // generation. |
| 33 const wchar_t kSystemPolicyKeyName[] = | 33 const wchar_t kSystemPolicyKeyName[] = |
| 34 L"Software\\Microsoft\\Windows\\CurrentVersion\\Policies\\System"; | 34 L"Software\\Microsoft\\Windows\\CurrentVersion\\Policies\\System"; |
| 35 const wchar_t kSoftwareSasValueName[] = L"SoftwareSASGeneration"; | 35 const wchar_t kSoftwareSasValueName[] = L"SoftwareSASGeneration"; |
| 36 | 36 |
| 37 const DWORD kEnableSoftwareSasByApps = 2; | 37 const DWORD kEnableSoftwareSasByServices = 1; |
| 38 | 38 |
| 39 // Toggles the default software SAS generation policy to enable SAS generation | 39 // Toggles the default software SAS generation policy to enable SAS generation |
| 40 // by applications. Non-default policy is not changed. | 40 // by services. Non-default policy is not changed. |
| 41 class ScopedSoftwareSasPolicy { | 41 class ScopedSoftwareSasPolicy { |
| 42 public: | 42 public: |
| 43 ScopedSoftwareSasPolicy(); | 43 ScopedSoftwareSasPolicy(); |
| 44 ~ScopedSoftwareSasPolicy(); | 44 ~ScopedSoftwareSasPolicy(); |
| 45 | 45 |
| 46 bool Apply(); | 46 bool Apply(); |
| 47 | 47 |
| 48 private: | 48 private: |
| 49 // The handle of the registry key were SoftwareSASGeneration policy is stored. | 49 // The handle of the registry key were SoftwareSASGeneration policy is stored. |
| 50 base::win::RegKey system_policy_; | 50 base::win::RegKey system_policy_; |
| (...skipping 31 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 82 LOG_GETLASTERROR(ERROR) << "Failed to open 'HKLM\\" | 82 LOG_GETLASTERROR(ERROR) << "Failed to open 'HKLM\\" |
| 83 << kSystemPolicyKeyName << "'"; | 83 << kSystemPolicyKeyName << "'"; |
| 84 return false; | 84 return false; |
| 85 } | 85 } |
| 86 | 86 |
| 87 bool custom_policy = system_policy_.HasValue(kSoftwareSasValueName); | 87 bool custom_policy = system_policy_.HasValue(kSoftwareSasValueName); |
| 88 | 88 |
| 89 // Override the default policy (i.e. there is no value in the registry) only. | 89 // Override the default policy (i.e. there is no value in the registry) only. |
| 90 if (!custom_policy) { | 90 if (!custom_policy) { |
| 91 result = system_policy_.WriteValue(kSoftwareSasValueName, | 91 result = system_policy_.WriteValue(kSoftwareSasValueName, |
| 92 kEnableSoftwareSasByApps); | 92 kEnableSoftwareSasByServices); |
| 93 if (result != ERROR_SUCCESS) { | 93 if (result != ERROR_SUCCESS) { |
| 94 SetLastError(result); | 94 SetLastError(result); |
| 95 LOG_GETLASTERROR(ERROR) | 95 LOG_GETLASTERROR(ERROR) |
| 96 << "Failed to enable software SAS generation by services"; | 96 << "Failed to enable software SAS generation by services"; |
| 97 return false; | 97 return false; |
| 98 } else { | 98 } else { |
| 99 restore_policy_ = true; | 99 restore_policy_ = true; |
| 100 } | 100 } |
| 101 } | 101 } |
| 102 | 102 |
| 103 return true; | 103 return true; |
| 104 } | 104 } |
| 105 | 105 |
| 106 } // namespace | 106 } // namespace |
| 107 | 107 |
| 108 // Sends the Secure Attention Sequence using the SendSAS() function from | 108 // Sends Secure Attention Sequence using the SendSAS() function from sas.dll. |
| 109 // sas.dll. This library is shipped starting from Win7/W2K8 R2 only. However | 109 // This library is shipped starting from Win7/W2K8 R2 only. However Win7 SDK |
| 110 // Win7 SDK includes a redistributable verion of the same library that works on | 110 // includes a redistributable verion of the same library that works on |
| 111 // Vista/W2K8. We install the latter along with our binaries. | 111 // Vista/W2K8. We install the latter along with our binaries. |
| 112 class SasInjectorWin : public SasInjector { | 112 class SasInjectorWin : public SasInjector { |
| 113 public: | 113 public: |
| 114 SasInjectorWin(); | 114 SasInjectorWin(); |
| 115 virtual ~SasInjectorWin(); | 115 virtual ~SasInjectorWin(); |
| 116 | 116 |
| 117 // SasInjector implementation. | 117 // SasInjector implementation. |
| 118 virtual bool InjectSas() OVERRIDE; | 118 virtual bool InjectSas() OVERRIDE; |
| 119 | 119 |
| 120 private: | 120 private: |
| (...skipping 42 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 163 send_sas_ = static_cast<SendSasFunc>( | 163 send_sas_ = static_cast<SendSasFunc>( |
| 164 sas_dll_.GetFunctionPointer(kSendSasName)); | 164 sas_dll_.GetFunctionPointer(kSendSasName)); |
| 165 } | 165 } |
| 166 if (send_sas_ == NULL) { | 166 if (send_sas_ == NULL) { |
| 167 LOG(ERROR) << "Failed to retrieve the address of '" << kSendSasName | 167 LOG(ERROR) << "Failed to retrieve the address of '" << kSendSasName |
| 168 << "()'"; | 168 << "()'"; |
| 169 return false; | 169 return false; |
| 170 } | 170 } |
| 171 | 171 |
| 172 // Enable software SAS generation by services and send SAS. SAS can still fail | 172 // Enable software SAS generation by services and send SAS. SAS can still fail |
| 173 // if the policy does not allow applications to generate software SAS. | 173 // if the policy does not allow services to generate software SAS. |
| 174 ScopedSoftwareSasPolicy enable_sas; | 174 ScopedSoftwareSasPolicy enable_sas; |
| 175 if (!enable_sas.Apply()) | 175 if (!enable_sas.Apply()) |
| 176 return false; | 176 return false; |
| 177 | 177 |
| 178 (*send_sas_)(FALSE); | 178 (*send_sas_)(FALSE); |
| 179 return true; | 179 return true; |
| 180 } | 180 } |
| 181 | 181 |
| 182 SasInjectorXp::SasInjectorXp() { | 182 SasInjectorXp::SasInjectorXp() { |
| 183 } | 183 } |
| (...skipping 28 matching lines...) Expand all Loading... |
| 212 | 212 |
| 213 scoped_ptr<SasInjector> SasInjector::Create() { | 213 scoped_ptr<SasInjector> SasInjector::Create() { |
| 214 if (base::win::GetVersion() < base::win::VERSION_VISTA) { | 214 if (base::win::GetVersion() < base::win::VERSION_VISTA) { |
| 215 return scoped_ptr<SasInjector>(new SasInjectorXp()); | 215 return scoped_ptr<SasInjector>(new SasInjectorXp()); |
| 216 } else { | 216 } else { |
| 217 return scoped_ptr<SasInjector>(new SasInjectorWin()); | 217 return scoped_ptr<SasInjector>(new SasInjectorWin()); |
| 218 } | 218 } |
| 219 } | 219 } |
| 220 | 220 |
| 221 } // namespace remoting | 221 } // namespace remoting |
| OLD | NEW |