15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright (c) 2012 The Chromium Authors. All rights reserved. 25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be 35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// found in the LICENSE file. 45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 55821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "remoting/host/win/elevated_controller.h" 65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/file_version_info.h" 81320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci#include "base/files/file_util.h" 95821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/json/json_reader.h" 105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/json/json_writer.h" 11868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)#include "base/logging.h" 125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/memory/scoped_ptr.h" 135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/path_service.h" 14a3f7b4e666c476898878fa745f637129375cd889Ben Murdoch#include "base/process/memory.h" 15868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)#include "base/strings/utf_string_conversions.h" 165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/values.h" 175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/win/scoped_handle.h" 185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "remoting/host/branding.h" 191675a649fd7a8b3cb80ffddae2dc181f122353c5Ben Murdoch#include "remoting/host/host_config.h" 205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "remoting/host/usage_stats_consent.h" 215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "remoting/host/verify_config_window_win.h" 222a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "remoting/host/win/core_resource.h" 232a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "remoting/host/win/security_descriptor.h" 242a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 252a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)namespace remoting { 265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace { 285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// The maximum size of the configuration file. "1MB ought to be enough" for any 305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// reasonable configuration we will ever need. 1MB is low enough to make 315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// the probability of out of memory situation fairly low. OOM is still possible 325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// and we will crash if it occurs. 335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const size_t kMaxConfigFileSize = 1024 * 1024; 345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// The host configuration file name. 362a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)const base::FilePath::CharType kConfigFileName[] = FILE_PATH_LITERAL("host.json"); 375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// The unprivileged configuration file name. 392a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)const base::FilePath::CharType kUnprivilegedConfigFileName[] = 405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) FILE_PATH_LITERAL("host_unprivileged.json"); 415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// The extension for the temporary file. 432a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)const base::FilePath::CharType kTempFileExtension[] = FILE_PATH_LITERAL("json~"); 445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// The host configuration file security descriptor that enables full access to 465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Local System and built-in administrators only. 472a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)const char kConfigFileSecurityDescriptor[] = 482a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) "O:BAG:BAD:(A;;GA;;;SY)(A;;GA;;;BA)"; 495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 502a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)const char kUnprivilegedConfigFileSecurityDescriptor[] = 512a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) "O:BAG:BAD:(A;;GA;;;SY)(A;;GA;;;BA)(A;;GR;;;AU)"; 525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Configuration keys. 545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// The configuration keys that cannot be specified in UpdateConfig(). 561675a649fd7a8b3cb80ffddae2dc181f122353c5Ben Murdochconst char* const kReadonlyKeys[] = { 571675a649fd7a8b3cb80ffddae2dc181f122353c5Ben Murdoch kHostIdConfigPath, kHostOwnerConfigPath, kHostOwnerEmailConfigPath, 581675a649fd7a8b3cb80ffddae2dc181f122353c5Ben Murdoch kXmppLoginConfigPath }; 595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// The configuration keys whose values may be read by GetConfig(). 611675a649fd7a8b3cb80ffddae2dc181f122353c5Ben Murdochconst char* const kUnprivilegedConfigKeys[] = { 621675a649fd7a8b3cb80ffddae2dc181f122353c5Ben Murdoch kHostIdConfigPath, kXmppLoginConfigPath }; 635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Determines if the client runs in the security context that allows performing 655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// administrative tasks (i.e. the user belongs to the adminstrators group and 665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// the client runs elevated). 675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool IsClientAdmin() { 685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) HRESULT hr = CoImpersonateClient(); 695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (FAILED(hr)) { 705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return false; 715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SID_IDENTIFIER_AUTHORITY nt_authority = SECURITY_NT_AUTHORITY; 745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) PSID administrators_group = NULL; 755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) BOOL result = AllocateAndInitializeSid(&nt_authority, 765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2, 775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SECURITY_BUILTIN_DOMAIN_RID, 785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DOMAIN_ALIAS_RID_ADMINS, 795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 0, 0, 0, 0, 0, 0, 805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) &administrators_group); 815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (result) { 825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!CheckTokenMembership(NULL, administrators_group, &result)) { 835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) result = false; 845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) FreeSid(administrators_group); 865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) hr = CoRevertToSelf(); 895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CHECK(SUCCEEDED(hr)); 905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return !!result; 925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Reads and parses the configuration file up to |kMaxConfigFileSize| in 955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// size. 962a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)HRESULT ReadConfig(const base::FilePath& filename, 975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_ptr<base::DictionaryValue>* config_out) { 985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Read raw data from the configuration file. 1005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::win::ScopedHandle file( 1015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CreateFileW(filename.value().c_str(), 1025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) GENERIC_READ, 1035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) FILE_SHARE_READ | FILE_SHARE_WRITE, 1045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) NULL, 1055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) OPEN_EXISTING, 1065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) FILE_FLAG_SEQUENTIAL_SCAN, 1075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) NULL)); 1085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!file.IsValid()) { 1105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DWORD error = GetLastError(); 111cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) PLOG(ERROR) << "Failed to open '" << filename.value() << "'"; 1125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return HRESULT_FROM_WIN32(error); 1135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 115c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) scoped_ptr<char[]> buffer(new char[kMaxConfigFileSize]); 1165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DWORD size = kMaxConfigFileSize; 1171320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci if (!::ReadFile(file.Get(), &buffer[0], size, &size, NULL)) { 1185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DWORD error = GetLastError(); 119cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) PLOG(ERROR) << "Failed to read '" << filename.value() << "'"; 1205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return HRESULT_FROM_WIN32(error); 1215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Parse the JSON configuration, expecting it to contain a dictionary. 1245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::string file_content(buffer.get(), size); 1255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_ptr<base::Value> value( 1265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::JSONReader::Read(file_content, base::JSON_ALLOW_TRAILING_COMMAS)); 1275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::DictionaryValue* dictionary; 1295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (value.get() == NULL || !value->GetAsDictionary(&dictionary)) { 1305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) LOG(ERROR) << "Failed to read '" << filename.value() << "'."; 1315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return E_FAIL; 1325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) value.release(); 1355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) config_out->reset(dictionary); 1365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return S_OK; 1375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1392a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)base::FilePath GetTempLocationFor(const base::FilePath& filename) { 1405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return filename.ReplaceExtension(kTempFileExtension); 1415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Writes a config file to a temporary location. 1442a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)HRESULT WriteConfigFileToTemp(const base::FilePath& filename, 1452a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const char* security_descriptor, 1465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const char* content, 1475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) size_t length) { 1482a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // Create the security descriptor for the configuration file. 1492a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ScopedSd sd = ConvertSddlToSd(security_descriptor); 1502a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (!sd) { 1515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DWORD error = GetLastError(); 152cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) PLOG(ERROR) 153cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) << "Failed to create a security descriptor for the configuration file"; 1545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return HRESULT_FROM_WIN32(error); 1555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1572a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) SECURITY_ATTRIBUTES security_attributes = {0}; 1582a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) security_attributes.nLength = sizeof(security_attributes); 1592a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) security_attributes.lpSecurityDescriptor = sd.get(); 1602a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) security_attributes.bInheritHandle = FALSE; 1612a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Create a temporary file and write configuration to it. 1632a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::FilePath tempname = GetTempLocationFor(filename); 1645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::win::ScopedHandle file( 1655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CreateFileW(tempname.value().c_str(), 1665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) GENERIC_WRITE, 1675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 0, 1685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) &security_attributes, 1695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CREATE_ALWAYS, 1705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) FILE_FLAG_SEQUENTIAL_SCAN, 1715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) NULL)); 1725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!file.IsValid()) { 1745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DWORD error = GetLastError(); 175cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) PLOG(ERROR) << "Failed to create '" << filename.value() << "'"; 1765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return HRESULT_FROM_WIN32(error); 1775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DWORD written; 1801320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci if (!WriteFile(file.Get(), content, static_cast<DWORD>(length), &written, 1811320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci NULL)) { 1825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DWORD error = GetLastError(); 183cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) PLOG(ERROR) << "Failed to write to '" << filename.value() << "'"; 1845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return HRESULT_FROM_WIN32(error); 1855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return S_OK; 1885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Moves a config file from its temporary location to its permanent location. 1912a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)HRESULT MoveConfigFileFromTemp(const base::FilePath& filename) { 1925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Now that the configuration is stored successfully replace the actual 1935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // configuration file. 1942a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::FilePath tempname = GetTempLocationFor(filename); 1955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!MoveFileExW(tempname.value().c_str(), 1965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) filename.value().c_str(), 1975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MOVEFILE_REPLACE_EXISTING)) { 1985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DWORD error = GetLastError(); 199cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) PLOG(ERROR) << "Failed to rename '" << tempname.value() << "' to '" 200cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) << filename.value() << "'"; 2015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return HRESULT_FROM_WIN32(error); 2025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 2035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return S_OK; 2055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Writes the configuration file up to |kMaxConfigFileSize| in size. 2085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)HRESULT WriteConfig(const char* content, size_t length, HWND owner_window) { 2095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (length > kMaxConfigFileSize) { 2105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return E_FAIL; 2115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 2125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Extract the configuration data that the user will verify. 2145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_ptr<base::Value> config_value(base::JSONReader::Read(content)); 2155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!config_value.get()) { 2165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return E_FAIL; 2175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 2185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::DictionaryValue* config_dict = NULL; 2195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!config_value->GetAsDictionary(&config_dict)) { 2205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return E_FAIL; 2215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 22268043e1e95eeb07d5cae7aca370b26518b0867d6Torne (Richard Coles) std::string email; 2231675a649fd7a8b3cb80ffddae2dc181f122353c5Ben Murdoch if (!config_dict->GetString(kHostOwnerEmailConfigPath, &email)) { 2241675a649fd7a8b3cb80ffddae2dc181f122353c5Ben Murdoch if (!config_dict->GetString(kHostOwnerConfigPath, &email)) { 2251675a649fd7a8b3cb80ffddae2dc181f122353c5Ben Murdoch if (!config_dict->GetString(kXmppLoginConfigPath, &email)) { 2261675a649fd7a8b3cb80ffddae2dc181f122353c5Ben Murdoch return E_FAIL; 2271675a649fd7a8b3cb80ffddae2dc181f122353c5Ben Murdoch } 22868043e1e95eeb07d5cae7aca370b26518b0867d6Torne (Richard Coles) } 22968043e1e95eeb07d5cae7aca370b26518b0867d6Torne (Richard Coles) } 23068043e1e95eeb07d5cae7aca370b26518b0867d6Torne (Richard Coles) std::string host_id, host_secret_hash; 2311675a649fd7a8b3cb80ffddae2dc181f122353c5Ben Murdoch if (!config_dict->GetString(kHostIdConfigPath, &host_id) || 2321675a649fd7a8b3cb80ffddae2dc181f122353c5Ben Murdoch !config_dict->GetString(kHostSecretHashConfigPath, &host_secret_hash)) { 2335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return E_FAIL; 2345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 2355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Ask the user to verify the configuration (unless the client is admin 2375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // already). 2385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!IsClientAdmin()) { 2395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) remoting::VerifyConfigWindowWin verify_win(email, host_id, 2405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) host_secret_hash); 2415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DWORD error = verify_win.DoModal(owner_window); 2425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (error != ERROR_SUCCESS) { 2435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return HRESULT_FROM_WIN32(error); 2445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 2455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 2465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Extract the unprivileged fields from the configuration. 2485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::DictionaryValue unprivileged_config_dict; 2495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (int i = 0; i < arraysize(kUnprivilegedConfigKeys); ++i) { 2505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const char* key = kUnprivilegedConfigKeys[i]; 2515d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) base::string16 value; 2525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (config_dict->GetString(key, &value)) { 2535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) unprivileged_config_dict.SetString(key, value); 2545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 2555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 2565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::string unprivileged_config_str; 2575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::JSONWriter::Write(&unprivileged_config_dict, &unprivileged_config_str); 2585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Write the full configuration file to a temporary location. 2602a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::FilePath full_config_file_path = 2615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) remoting::GetConfigDir().Append(kConfigFileName); 2625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) HRESULT hr = WriteConfigFileToTemp(full_config_file_path, 2635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) kConfigFileSecurityDescriptor, 2645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) content, 2655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) length); 2665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (FAILED(hr)) { 2675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return hr; 2685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 2695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Write the unprivileged configuration file to a temporary location. 2712a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::FilePath unprivileged_config_file_path = 2725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) remoting::GetConfigDir().Append(kUnprivilegedConfigFileName); 2735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) hr = WriteConfigFileToTemp(unprivileged_config_file_path, 2745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) kUnprivilegedConfigFileSecurityDescriptor, 2755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) unprivileged_config_str.data(), 2765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) unprivileged_config_str.size()); 2775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (FAILED(hr)) { 2785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return hr; 2795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 2805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Move the full configuration file to its permanent location. 2825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) hr = MoveConfigFileFromTemp(full_config_file_path); 2835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (FAILED(hr)) { 2845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return hr; 2855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 2865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Move the unprivileged configuration file to its permanent location. 2885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) hr = MoveConfigFileFromTemp(unprivileged_config_file_path); 2895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (FAILED(hr)) { 2905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return hr; 2915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 2925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return S_OK; 2945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace 2975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)ElevatedController::ElevatedController() : owner_window_(NULL) { 2995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 3005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)HRESULT ElevatedController::FinalConstruct() { 3025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return S_OK; 3035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 3045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void ElevatedController::FinalRelease() { 3065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 3075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)STDMETHODIMP ElevatedController::GetConfig(BSTR* config_out) { 3092a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::FilePath config_dir = remoting::GetConfigDir(); 3105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Read the unprivileged part of host configuration. 3125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_ptr<base::DictionaryValue> config; 3135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) HRESULT hr = ReadConfig(config_dir.Append(kUnprivilegedConfigFileName), 3145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) &config); 3155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (FAILED(hr)) { 3165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return hr; 3175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 3185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Convert the config back to a string and return it to the caller. 3205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::string file_content; 3215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::JSONWriter::Write(config.get(), &file_content); 3225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3235d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) *config_out = ::SysAllocString(base::UTF8ToUTF16(file_content).c_str()); 3245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (config_out == NULL) { 3255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return E_OUTOFMEMORY; 3265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 3275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return S_OK; 3295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 3305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)STDMETHODIMP ElevatedController::GetVersion(BSTR* version_out) { 3325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Report the product version number of the daemon controller binary as 3335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // the host version. 3345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) HMODULE binary = base::GetModuleFromAddress( 3355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) reinterpret_cast<void*>(&ReadConfig)); 3365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_ptr<FileVersionInfo> version_info( 3375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) FileVersionInfo::CreateFileVersionInfoForModule(binary)); 3385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3395d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) base::string16 version; 3405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (version_info.get()) { 3415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) version = version_info->product_version(); 3425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 3435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) *version_out = ::SysAllocString(version.c_str()); 3455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (version_out == NULL) { 3465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return E_OUTOFMEMORY; 3475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 3485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return S_OK; 3505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 3515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)STDMETHODIMP ElevatedController::SetConfig(BSTR config) { 3535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Determine the config directory path and create it if necessary. 3542a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::FilePath config_dir = remoting::GetConfigDir(); 355a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) if (!base::CreateDirectory(config_dir)) { 3565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return HRESULT_FROM_WIN32(ERROR_ACCESS_DENIED); 3575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 3585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3595d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) std::string file_content = base::UTF16ToUTF8( 3605d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) base::string16(static_cast<base::char16*>(config), ::SysStringLen(config))); 3615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return WriteConfig(file_content.c_str(), file_content.size(), owner_window_); 3635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 3645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)STDMETHODIMP ElevatedController::SetOwnerWindow(LONG_PTR window_handle) { 3665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) owner_window_ = reinterpret_cast<HWND>(window_handle); 3675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return S_OK; 3685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 3695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)STDMETHODIMP ElevatedController::StartDaemon() { 3715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ScopedScHandle service; 3725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) HRESULT hr = OpenService(&service); 3735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (FAILED(hr)) { 3745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return hr; 3755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 3765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Change the service start type to 'auto'. 3781320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci if (!::ChangeServiceConfigW(service.Get(), 3795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SERVICE_NO_CHANGE, 3805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SERVICE_AUTO_START, 3815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SERVICE_NO_CHANGE, 3825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) NULL, 3835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) NULL, 3845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) NULL, 3855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) NULL, 3865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) NULL, 3875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) NULL, 3885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) NULL)) { 3895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DWORD error = GetLastError(); 390cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) PLOG(ERROR) << "Failed to change the '" << kWindowsServiceName 391cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) << "'service start type to 'auto'"; 3925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return HRESULT_FROM_WIN32(error); 3935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 3945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Start the service. 3961320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci if (!StartService(service.Get(), 0, NULL)) { 3975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DWORD error = GetLastError(); 3985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (error != ERROR_SERVICE_ALREADY_RUNNING) { 399cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) PLOG(ERROR) << "Failed to start the '" << kWindowsServiceName 400cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) << "'service"; 4015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return HRESULT_FROM_WIN32(error); 4035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 4045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 4055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return S_OK; 4075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 4085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)STDMETHODIMP ElevatedController::StopDaemon() { 4105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ScopedScHandle service; 4115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) HRESULT hr = OpenService(&service); 4125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (FAILED(hr)) { 4135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return hr; 4145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 4155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Change the service start type to 'manual'. 4171320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci if (!::ChangeServiceConfigW(service.Get(), 4185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SERVICE_NO_CHANGE, 4195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SERVICE_DEMAND_START, 4205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SERVICE_NO_CHANGE, 4215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) NULL, 4225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) NULL, 4235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) NULL, 4245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) NULL, 4255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) NULL, 4265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) NULL, 4275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) NULL)) { 4285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DWORD error = GetLastError(); 429cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) PLOG(ERROR) << "Failed to change the '" << kWindowsServiceName 430cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) << "'service start type to 'manual'"; 4315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return HRESULT_FROM_WIN32(error); 4325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 4335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Stop the service. 4355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SERVICE_STATUS status; 4361320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci if (!ControlService(service.Get(), SERVICE_CONTROL_STOP, &status)) { 4375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DWORD error = GetLastError(); 4385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (error != ERROR_SERVICE_NOT_ACTIVE) { 439cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) PLOG(ERROR) << "Failed to stop the '" << kWindowsServiceName 440cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) << "'service"; 4415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return HRESULT_FROM_WIN32(error); 4425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 4435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 4445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return S_OK; 4465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 4475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)STDMETHODIMP ElevatedController::UpdateConfig(BSTR config) { 4495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Parse the config. 4505d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) std::string config_str = base::UTF16ToUTF8( 4515d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) base::string16(static_cast<base::char16*>(config), ::SysStringLen(config))); 4525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_ptr<base::Value> config_value(base::JSONReader::Read(config_str)); 4535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!config_value.get()) { 4545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return E_FAIL; 4555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 4565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::DictionaryValue* config_dict = NULL; 4575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!config_value->GetAsDictionary(&config_dict)) { 4585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return E_FAIL; 4595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 4605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Check for bad keys. 4615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (int i = 0; i < arraysize(kReadonlyKeys); ++i) { 4625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (config_dict->HasKey(kReadonlyKeys[i])) { 4635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return HRESULT_FROM_WIN32(ERROR_ACCESS_DENIED); 4645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 4655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 4665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Get the old config. 4672a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::FilePath config_dir = remoting::GetConfigDir(); 4685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_ptr<base::DictionaryValue> config_old; 4695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) HRESULT hr = ReadConfig(config_dir.Append(kConfigFileName), &config_old); 4705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (FAILED(hr)) { 4715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return hr; 4725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 4735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Merge items from the given config into the old config. 4745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) config_old->MergeDictionary(config_dict); 4755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Write the updated config. 4765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::string config_updated_str; 4775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::JSONWriter::Write(config_old.get(), &config_updated_str); 4785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return WriteConfig(config_updated_str.c_str(), config_updated_str.size(), 4795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) owner_window_); 4805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 4815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)STDMETHODIMP ElevatedController::GetUsageStatsConsent(BOOL* allowed, 4835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) BOOL* set_by_policy) { 4845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bool local_allowed; 4855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bool local_set_by_policy; 4865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (remoting::GetUsageStatsConsent(&local_allowed, &local_set_by_policy)) { 4875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) *allowed = local_allowed; 4885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) *set_by_policy = local_set_by_policy; 4895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return S_OK; 4905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } else { 4915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return E_FAIL; 4925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 4935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 4945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)STDMETHODIMP ElevatedController::SetUsageStatsConsent(BOOL allowed) { 4965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (remoting::SetUsageStatsConsent(!!allowed)) { 4975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return S_OK; 4985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } else { 4995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return E_FAIL; 5005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 5015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 5025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)HRESULT ElevatedController::OpenService(ScopedScHandle* service_out) { 5045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DWORD error; 5055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ScopedScHandle scmanager( 5075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ::OpenSCManagerW(NULL, SERVICES_ACTIVE_DATABASE, 5085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SC_MANAGER_CONNECT | SC_MANAGER_ENUMERATE_SERVICE)); 5095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!scmanager.IsValid()) { 5105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) error = GetLastError(); 511cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) PLOG(ERROR) << "Failed to connect to the service control manager"; 5125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return HRESULT_FROM_WIN32(error); 5145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 5155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DWORD desired_access = SERVICE_CHANGE_CONFIG | SERVICE_QUERY_STATUS | 5175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SERVICE_START | SERVICE_STOP; 5185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ScopedScHandle service( 5191320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci ::OpenServiceW(scmanager.Get(), kWindowsServiceName, desired_access)); 5205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!service.IsValid()) { 5215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) error = GetLastError(); 522cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) PLOG(ERROR) << "Failed to open to the '" << kWindowsServiceName 523cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) << "' service"; 5245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return HRESULT_FROM_WIN32(error); 5265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 5275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) service_out->Set(service.Take()); 5295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return S_OK; 5305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 5315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace remoting 533