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
|
// Copyright 2016 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 "components/arc/standalone/service_helper.h"
#include "base/files/file_util.h"
#include "base/logging.h"
#include "base/message_loop/message_loop.h"
#include "base/posix/eintr_wrapper.h"
namespace arc {
// static
ServiceHelper* ServiceHelper::self_ = nullptr;
ServiceHelper::ServiceHelper() {
}
ServiceHelper::~ServiceHelper() {
// Best efforts clean up, ignore the return values.
sigaction(SIGINT, &old_sigint_, NULL);
sigaction(SIGTERM, &old_sigterm_, NULL);
}
void ServiceHelper::Init(const base::Closure& closure) {
CHECK(!ServiceHelper::self_);
ServiceHelper::self_ = this;
closure_ = closure;
// Initialize pipe
int pipe_fd[2];
CHECK_EQ(0, pipe(pipe_fd));
read_fd_.reset(pipe_fd[0]);
write_fd_.reset(pipe_fd[1]);
CHECK(base::SetNonBlocking(write_fd_.get()));
CHECK(base::MessageLoopForIO::current()->WatchFileDescriptor(
read_fd_.get(),
true, /* persistent */
base::MessageLoopForIO::WATCH_READ,
&watcher_,
this));
struct sigaction action = {};
CHECK_EQ(0, sigemptyset(&action.sa_mask));
action.sa_handler = ServiceHelper::TerminationHandler;
action.sa_flags = 0;
CHECK_EQ(0, sigaction(SIGINT, &action, &old_sigint_));
CHECK_EQ(0, sigaction(SIGTERM, &action, &old_sigterm_));
}
// static
void ServiceHelper::TerminationHandler(int /* signum */) {
if (HANDLE_EINTR(write(self_->write_fd_.get(), "1", 1)) < 0) {
_exit(2); // We still need to exit the program, but in a brute force way.
}
}
void ServiceHelper::OnFileCanReadWithoutBlocking(int fd) {
CHECK_EQ(read_fd_.get(), fd);
char c;
// We don't really care about the return value, since it indicates closing.
HANDLE_EINTR(read(read_fd_.get(), &c, 1));
closure_.Run();
}
void ServiceHelper::OnFileCanWriteWithoutBlocking(int fd) {
NOTREACHED();
}
} // namespace arc
|
