blob: 858e588172257ed66f66491aab2d0a987c0f02cd (
plain)
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
|
// Copyright 2015 The Chromium OS 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 "chromeos/process_reaper.h"
#include <sys/signalfd.h>
#include <sys/types.h>
#include <sys/wait.h>
#include <base/bind.h>
#include <base/posix/eintr_wrapper.h>
#include <chromeos/asynchronous_signal_handler.h>
#include <chromeos/daemons/daemon.h>
#include <chromeos/location_logging.h>
namespace chromeos {
ProcessReaper::~ProcessReaper() {
Unregister();
}
void ProcessReaper::Register(
AsynchronousSignalHandlerInterface* async_signal_handler) {
CHECK(!async_signal_handler_);
async_signal_handler_ = async_signal_handler;
async_signal_handler->RegisterHandler(
SIGCHLD,
base::Bind(&ProcessReaper::HandleSIGCHLD, base::Unretained(this)));
}
void ProcessReaper::Unregister() {
if (!async_signal_handler_)
return;
async_signal_handler_->UnregisterHandler(SIGCHLD);
async_signal_handler_ = nullptr;
}
bool ProcessReaper::WatchForChild(const tracked_objects::Location& from_here,
pid_t pid,
const ChildCallback& callback) {
if (watched_processes_.find(pid) != watched_processes_.end())
return false;
watched_processes_.emplace(pid, WatchedProcess{from_here, callback});
return true;
}
bool ProcessReaper::HandleSIGCHLD(const struct signalfd_siginfo& sigfd_info) {
// One SIGCHLD may correspond to multiple terminated children, so ignore
// sigfd_info and reap any available children.
while (true) {
siginfo_t info;
info.si_pid = 0;
int rc = HANDLE_EINTR(waitid(P_ALL, 0, &info, WNOHANG | WEXITED));
if (rc == -1) {
if (errno != ECHILD) {
PLOG(ERROR) << "waitid failed";
}
break;
}
if (info.si_pid == 0) {
break;
}
auto proc = watched_processes_.find(info.si_pid);
if (proc == watched_processes_.end()) {
LOG(INFO) << "Untracked process " << info.si_pid
<< " terminated with status " << info.si_status
<< " (code = " << info.si_code << ")";
} else {
DVLOG_LOC(proc->second.location, 1)
<< "Process " << info.si_pid << " terminated with status "
<< info.si_status << " (code = " << info.si_code << ")";
ChildCallback callback = std::move(proc->second.callback);
watched_processes_.erase(proc);
callback.Run(info);
}
}
// Return false to indicate that our handler should not be uninstalled.
return false;
}
} // namespace chromeos
|
