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