| OLD | NEW |
| 1 // Copyright (c) 2011 The Chromium OS Authors. All rights reserved. | 1 // Copyright (c) 2011 The Chromium OS Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #ifndef CHROMEOS_PROCESS_H_ | 5 #ifndef CHROMEOS_PROCESS_H_ |
| 6 #define CHROMEOS_PROCESS_H_ | 6 #define CHROMEOS_PROCESS_H_ |
| 7 | 7 |
| 8 #include <sys/types.h> | 8 #include <sys/types.h> |
| 9 | 9 |
| 10 #include <map> | 10 #include <map> |
| (...skipping 33 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 44 } | 44 } |
| 45 | 45 |
| 46 // Redirects stderr and stdout to |output_file|. | 46 // Redirects stderr and stdout to |output_file|. |
| 47 virtual void RedirectOutput(const std::string& output_file) = 0; | 47 virtual void RedirectOutput(const std::string& output_file) = 0; |
| 48 | 48 |
| 49 // Indicates we want to redirect |child_fd| in the child process's | 49 // Indicates we want to redirect |child_fd| in the child process's |
| 50 // file table to a pipe. |child_fd| will be available for reading | 50 // file table to a pipe. |child_fd| will be available for reading |
| 51 // from child process's perspective iff |is_input|. | 51 // from child process's perspective iff |is_input|. |
| 52 virtual void RedirectUsingPipe(int child_fd, bool is_input) = 0; | 52 virtual void RedirectUsingPipe(int child_fd, bool is_input) = 0; |
| 53 | 53 |
| 54 // Set the real/effective/saved user ID of the child process. |
| 55 virtual void SetUid(uid_t uid) = 0; |
| 56 |
| 57 // Set the real/effective/saved group ID of the child process. |
| 58 virtual void SetGid(gid_t gid) = 0; |
| 59 |
| 54 // Gets the pipe file descriptor mapped to the process's |child_fd|. | 60 // Gets the pipe file descriptor mapped to the process's |child_fd|. |
| 55 virtual int GetPipe(int child_fd) = 0; | 61 virtual int GetPipe(int child_fd) = 0; |
| 56 | 62 |
| 57 // Starts this process, returning true if successful. | 63 // Starts this process, returning true if successful. |
| 58 virtual bool Start() = 0; | 64 virtual bool Start() = 0; |
| 59 | 65 |
| 60 // Waits for this process to finish. Returns the process's exit | 66 // Waits for this process to finish. Returns the process's exit |
| 61 // status if it exited normally, or otherwise returns -1. | 67 // status if it exited normally, or otherwise returns -1. |
| 62 virtual int Wait() = 0; | 68 virtual int Wait() = 0; |
| 63 | 69 |
| (...skipping 29 matching lines...) Expand all Loading... |
| 93 }; | 99 }; |
| 94 | 100 |
| 95 class ProcessImpl : public Process { | 101 class ProcessImpl : public Process { |
| 96 public: | 102 public: |
| 97 ProcessImpl(); | 103 ProcessImpl(); |
| 98 virtual ~ProcessImpl(); | 104 virtual ~ProcessImpl(); |
| 99 | 105 |
| 100 virtual void AddArg(const std::string& arg); | 106 virtual void AddArg(const std::string& arg); |
| 101 virtual void RedirectOutput(const std::string& output_file); | 107 virtual void RedirectOutput(const std::string& output_file); |
| 102 virtual void RedirectUsingPipe(int child_fd, bool is_input); | 108 virtual void RedirectUsingPipe(int child_fd, bool is_input); |
| 109 virtual void SetUid(uid_t uid); |
| 110 virtual void SetGid(gid_t gid); |
| 103 virtual int GetPipe(int child_fd); | 111 virtual int GetPipe(int child_fd); |
| 104 virtual bool Start(); | 112 virtual bool Start(); |
| 105 virtual int Wait(); | 113 virtual int Wait(); |
| 106 virtual int Run(); | 114 virtual int Run(); |
| 107 virtual pid_t pid(); | 115 virtual pid_t pid(); |
| 108 virtual bool Kill(int signal, int timeout); | 116 virtual bool Kill(int signal, int timeout); |
| 109 virtual void Reset(pid_t pid); | 117 virtual void Reset(pid_t pid); |
| 110 virtual bool ResetPidByFile(const std::string& pid_file); | 118 virtual bool ResetPidByFile(const std::string& pid_file); |
| 111 virtual pid_t Release(); | 119 virtual pid_t Release(); |
| 112 | 120 |
| 113 protected: | 121 protected: |
| 114 struct PipeInfo { | 122 struct PipeInfo { |
| 115 PipeInfo() : parent_fd_(-1), child_fd_(-1), is_input_(false) {} | 123 PipeInfo() : parent_fd_(-1), child_fd_(-1), is_input_(false) {} |
| 116 // Parent (our) side of the pipe to the the child process. | 124 // Parent (our) side of the pipe to the the child process. |
| 117 int parent_fd_; | 125 int parent_fd_; |
| 118 // Child's side of the pipe to the parent. | 126 // Child's side of the pipe to the parent. |
| 119 int child_fd_; | 127 int child_fd_; |
| 120 // Is this an input or output pipe from child's perspective. | 128 // Is this an input or output pipe from child's perspective. |
| 121 bool is_input_; | 129 bool is_input_; |
| 122 }; | 130 }; |
| 123 typedef std::map<int, PipeInfo> PipeMap; | 131 typedef std::map<int, PipeInfo> PipeMap; |
| 124 | 132 |
| 125 std::string output_file_; | |
| 126 std::vector<std::string> arguments_; | |
| 127 // Map of child target file descriptors (first) to information about | |
| 128 // pipes created (second). | |
| 129 PipeMap pipe_map_; | |
| 130 | |
| 131 void UpdatePid(pid_t new_pid); | 133 void UpdatePid(pid_t new_pid); |
| 132 bool PopulatePipeMap(); | 134 bool PopulatePipeMap(); |
| 133 | 135 |
| 134 private: | 136 private: |
| 135 // Pid of currently managed process or 0 if no currently managed | 137 // Pid of currently managed process or 0 if no currently managed |
| 136 // process. pid must not be modified except by calling | 138 // process. pid must not be modified except by calling |
| 137 // UpdatePid(new_pid). | 139 // UpdatePid(new_pid). |
| 138 pid_t pid_; | 140 pid_t pid_; |
| 141 std::string output_file_; |
| 142 std::vector<std::string> arguments_; |
| 143 // Map of child target file descriptors (first) to information about |
| 144 // pipes created (second). |
| 145 PipeMap pipe_map_; |
| 146 uid_t uid_; |
| 147 gid_t gid_; |
| 139 }; | 148 }; |
| 140 | 149 |
| 141 } // namespace chromeos | 150 } // namespace chromeos |
| 142 | 151 |
| 143 #endif // CHROMEOS_PROCESS_H | 152 #endif // CHROMEOS_PROCESS_H |
| OLD | NEW |