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