1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
|
// Copyright (c) 2012 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef CHROMEOS_PROCESS_PROXY_PROCESS_PROXY_H_
#define CHROMEOS_PROCESS_PROXY_PROCESS_PROXY_H_
#include <fcntl.h>
#include <signal.h>
#include <cstdio>
#include <string>
#include "base/memory/ref_counted.h"
#include "chromeos/process_proxy/process_output_watcher.h"
namespace base {
class TaskRunner;
class Thread;
} // namespace base
namespace chromeos {
// Proxy to a single ChromeOS process.
// This is refcounted. Note that output watcher, when it gets triggered owns a
// a callback with ref to this, so in order for this to be freed, the watcher
// must be destroyed. This is done in Close.
class ProcessProxy : public base::RefCountedThreadSafe<ProcessProxy> {
public:
ProcessProxy();
// Opens a process using command |command|. |pid| is set to new process' pid.
bool Open(const std::string& command, pid_t* pid);
// Triggers watcher object on |watch_thread|. |watch_thread| gets blocked, so
// it should not be one of commonly used threads. It should be thread created
// specifically for running process output watcher.
bool StartWatchingOnThread(base::Thread* watch_thread,
const ProcessOutputCallback& callback);
// Sends some data to the process.
bool Write(const std::string& text);
// Closes the process.
// Must be called if we want this to be eventually deleted.
void Close();
// Notifies underlaying process of terminal size change.
bool OnTerminalResize(int width, int height);
private:
friend class base::RefCountedThreadSafe<ProcessProxy>;
// We want this be used as ref counted object only.
~ProcessProxy();
// Create master and slave end of pseudo terminal that will be used to
// communicate with process.
// pt_pair[0] -> master, pt_pair[1] -> slave.
// pt_pair must be allocated (to size at least 2).
bool CreatePseudoTerminalPair(int *pt_pair);
bool LaunchProcess(const std::string& command, int slave_fd, pid_t* pid);
// Gets called by output watcher when the process writes something to its
// output streams.
void OnProcessOutput(ProcessOutputType type, const std::string& output);
void CallOnProcessOutputCallback(ProcessOutputType type,
const std::string& output);
bool StopWatching();
// Methods for cleaning up pipes.
void CloseAllFdPairs();
// Expects array of 2 file descripters.
void CloseFdPair(int* pipe);
// Expects pointer to single file descriptor.
void CloseFd(int* fd);
void ClearAllFdPairs();
// Expects array of 2 file descripters.
void ClearFdPair(int* pipe);
bool process_launched_;
pid_t pid_;
bool callback_set_;
ProcessOutputCallback callback_;
scoped_refptr<base::TaskRunner> callback_runner_;
bool watcher_started_;
int pt_pair_[2];
int shutdown_pipe_[2];
DISALLOW_COPY_AND_ASSIGN(ProcessProxy);
};
} // namespace chromeos
#endif // CHROMEOS_PROCESS_PROXY_PROCESS_PROXY_H_
|
