15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright (c) 2011 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 BASE_MEMORY_SCOPED_OPEN_PROCESS_H_ 65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#define BASE_MEMORY_SCOPED_OPEN_PROCESS_H_ 75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 858e6fbe4ee35d65e14b626c557d37565bf8ad179Ben Murdoch#include "base/process/process_handle.h" 95821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace base { 115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// A class that opens a process from its process id and closes it when the 135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// instance goes out of scope. 145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class ScopedOpenProcess { 155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public: 165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ScopedOpenProcess() : handle_(kNullProcessHandle) { 175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Automatically close the process. 205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ~ScopedOpenProcess() { 215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) Close(); 225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Open a new process by pid. Closes any previously opened process (even if 255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // opening the new one fails). 265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bool Open(ProcessId pid) { 275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) Close(); 285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return OpenProcessHandle(pid, &handle_); 295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Close the previously opened process. 325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void Close() { 335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (handle_ == kNullProcessHandle) 345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return; 355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CloseProcessHandle(handle_); 375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) handle_ = kNullProcessHandle; 385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ProcessHandle handle() const { return handle_; } 415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) private: 435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ProcessHandle handle_; 445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DISALLOW_COPY_AND_ASSIGN(ScopedOpenProcess); 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace base 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#endif // BASE_MEMORY_SCOPED_OPEN_PROCESS_H_ 49