blob: 168c586a82dc0ce7e86ca02557bf1c1ee9a9ffc0 [file] [log] [blame]
// 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.
#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 {
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 {
ProcessOutputWatcher(int out_fd, const ProcessOutputCallback& callback);
~ProcessOutputWatcher() override;
void Start();
// 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_;
} // namespace chromeos