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