forked from chromium/chromium
-
Notifications
You must be signed in to change notification settings - Fork 0
/
process_output_watcher.h
81 lines (60 loc) · 2.5 KB
/
process_output_watcher.h
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
// 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_OUTPUT_WATCHER_H_
#define CHROMEOS_PROCESS_PROXY_PROCESS_OUTPUT_WATCHER_H_
#include <stddef.h>
#include <string>
#include "base/callback.h"
#include "base/files/file.h"
#include "base/macros.h"
#include "base/memory/weak_ptr.h"
#include "base/message_loop/message_loop.h"
#include "chromeos/chromeos_export.h"
namespace chromeos {
enum ProcessOutputType {
PROCESS_OUTPUT_TYPE_OUT,
PROCESS_OUTPUT_TYPE_EXIT
};
typedef base::Callback<void(ProcessOutputType,
const std::string&,
const base::Closure&)> ProcessOutputCallback;
// Observes output on |out_fd| and invokes |callback| when some output is
// detected. It assumes UTF8 output.
class CHROMEOS_EXPORT ProcessOutputWatcher
: public base::MessageLoopForIO::Watcher {
public:
ProcessOutputWatcher(int out_fd, const ProcessOutputCallback& callback);
~ProcessOutputWatcher() override;
void Start();
private:
// MessageLoopForIO::Watcher overrides:
void OnFileCanReadWithoutBlocking(int fd) override;
void OnFileCanWriteWithoutBlocking(int fd) override;
// Listens to output from fd passed to the constructor.
void WatchProcessOutput();
// Reads data from fd and invokes callback |on_read_callback_| with read data.
void ReadFromFd(int fd);
// Checks if the read buffer has any trailing incomplete UTF8 characters and
// returns the read buffer size without them.
size_t OutputSizeWithoutIncompleteUTF8();
// Processes new |read_buffer_| state and notifies observer about new process
// output.
void ReportOutput(ProcessOutputType type,
size_t new_bytes_count,
const base::Closure& callback);
char read_buffer_[4096];
// Maximum read buffer content size.
size_t read_buffer_capacity_;
// Current read bufferi content size.
size_t read_buffer_size_;
// Contains file descsriptor to which watched process output is written.
base::File process_output_file_;
base::MessageLoopForIO::FileDescriptorWatcher output_file_watcher_;
// Callback that will be invoked when some output is detected.
ProcessOutputCallback on_read_callback_;
base::WeakPtrFactory<ProcessOutputWatcher> weak_factory_;
DISALLOW_COPY_AND_ASSIGN(ProcessOutputWatcher);
};
} // namespace chromeos
#endif // CHROMEOS_PROCESS_PROXY_PROCESS_OUTPUT_WATCHER_H_