15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright (c) 2006-2010 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)#ifndef SANDBOX_SRC_WIN_UTILS_H_ 65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#define SANDBOX_SRC_WIN_UTILS_H_ 75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 85821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <windows.h> 95821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <string> 10d57369da7c6519fef57db42085f7b42d4c8845c1Torne (Richard Coles) 115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/basictypes.h" 12d57369da7c6519fef57db42085f7b42d4c8845c1Torne (Richard Coles)#include "base/strings/string16.h" 135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace sandbox { 155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Prefix for path used by NT calls. 175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const wchar_t kNTPrefix[] = L"\\??\\"; 185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const size_t kNTPrefixLen = arraysize(kNTPrefix) - 1; 195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const wchar_t kNTObjManPrefix[] = L"\\Device\\"; 215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const size_t kNTObjManPrefixLen = arraysize(kNTObjManPrefix) - 1; 225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Automatically acquires and releases a lock when the object is 245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// is destroyed. 255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class AutoLock { 265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public: 275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Acquires the lock. 285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) explicit AutoLock(CRITICAL_SECTION *lock) : lock_(lock) { 295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ::EnterCriticalSection(lock); 305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) }; 315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Releases the lock; 335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ~AutoLock() { 345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ::LeaveCriticalSection(lock_); 355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) }; 365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) private: 385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CRITICAL_SECTION *lock_; 395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DISALLOW_IMPLICIT_CONSTRUCTORS(AutoLock); 405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Basic implementation of a singleton which calls the destructor 435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// when the exe is shutting down or the DLL is being unloaded. 445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)template <typename Derived> 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class SingletonBase { 465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public: 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) static Derived* GetInstance() { 485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) static Derived* instance = NULL; 495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (NULL == instance) { 505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) instance = new Derived(); 515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Microsoft CRT extension. In an exe this this called after 525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // winmain returns, in a dll is called in DLL_PROCESS_DETACH 535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) _onexit(OnExit); 545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return instance; 565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) private: 595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // this is the function that gets called by the CRT when the 605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // process is shutting down. 615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) static int __cdecl OnExit() { 625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) delete GetInstance(); 635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return 0; 645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Convert a short path (C:\path~1 or \\??\\c:\path~1) to the long version of 685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// the path. If the path is not a valid filesystem path, the function returns 695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// false and the output parameter is not modified. 70d57369da7c6519fef57db42085f7b42d4c8845c1Torne (Richard Coles)bool ConvertToLongPath(const base::string16& short_path, 71d57369da7c6519fef57db42085f7b42d4c8845c1Torne (Richard Coles) base::string16* long_path); 725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Sets result to true if the path contains a reparse point. The return value 745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// is ERROR_SUCCESS when the function succeeds or the appropriate error code 755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// when the function fails. 765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// This function is not smart. It looks for each element in the path and 775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// returns true if any of them is a reparse point. 78d57369da7c6519fef57db42085f7b42d4c8845c1Torne (Richard Coles)DWORD IsReparsePoint(const base::string16& full_path, bool* result); 795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Returns true if the handle corresponds to the object pointed by this path. 815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool SameObject(HANDLE handle, const wchar_t* full_path); 825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Resolves a handle to an nt path. Returns true if the handle can be resolved. 84d57369da7c6519fef57db42085f7b42d4c8845c1Torne (Richard Coles)bool GetPathFromHandle(HANDLE handle, base::string16* path); 855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Resolves a win32 path to an nt path using GetPathFromHandle. The path must 875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// exist. Returs true if the translation was succesful. 88d57369da7c6519fef57db42085f7b42d4c8845c1Torne (Richard Coles)bool GetNtPathFromWin32Path(const base::string16& path, 89d57369da7c6519fef57db42085f7b42d4c8845c1Torne (Richard Coles) base::string16* nt_path); 905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Translates a reserved key name to its handle. 925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// For example "HKEY_LOCAL_MACHINE" returns HKEY_LOCAL_MACHINE. 935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Returns NULL if the name does not represent any reserved key name. 94d57369da7c6519fef57db42085f7b42d4c8845c1Torne (Richard Coles)HKEY GetReservedKeyFromName(const base::string16& name); 955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Resolves a user-readable registry path to a system-readable registry path. 975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// For example, HKEY_LOCAL_MACHINE\\Software\\microsoft is translated to 985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// \\registry\\machine\\software\\microsoft. Returns false if the path 995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// cannot be resolved. 100d57369da7c6519fef57db42085f7b42d4c8845c1Torne (Richard Coles)bool ResolveRegistryName(base::string16 name, base::string16* resolved_name); 1015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Writes |length| bytes from the provided |buffer| into the address space of 1035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// |child_process|, at the specified |address|, preserving the original write 1045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// protection attributes. Returns true on success. 1055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool WriteProtectedChildMemory(HANDLE child_process, void* address, 1065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const void* buffer, size_t length); 1075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace sandbox 1095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Resolves a function name in NTDLL to a function pointer. The second parameter 1115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// is a pointer to the function pointer. 1125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void ResolveNTFunctionPtr(const char* name, void* ptr); 1135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#endif // SANDBOX_SRC_WIN_UTILS_H_ 115