ACPR/android_system_core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Merge "init/epoll: Fix a potential use-after-free"

Browse source
This commit is contained in:
Treehugger Robot 2022-10-20 00:38:06 +00:00 • committed by Gerrit Code Review
commit 9c3113ab46
1 changed files with 6 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
init/epoll.cpp
View file

@ -54,9 +54,7 @@ Result<void> Epoll::RegisterHandler(int fd, Handler handler, uint32_t events) {
} }
epoll_event ev; epoll_event ev;
ev.events = events; ev.events = events;
// std::map's iterators do not get invalidated until erased, so we use the ev.data.fd = fd;
// pointer to the std::function in the map directly for epoll_ctl.
ev.data.ptr = reinterpret_cast<void*>(&it->second);
if (epoll_ctl(epoll_fd_, EPOLL_CTL_ADD, fd, &ev) == -1) { if (epoll_ctl(epoll_fd_, EPOLL_CTL_ADD, fd, &ev) == -1) {
Result<void> result = ErrnoError() << "epoll_ctl failed to add fd"; Result<void> result = ErrnoError() << "epoll_ctl failed to add fd";
epoll_handlers_.erase(fd); epoll_handlers_.erase(fd);
@ -96,7 +94,11 @@ Result<std::vector<std::shared_ptr<Epoll::Handler>>> Epoll::Wait(
} }
std::vector<std::shared_ptr<Handler>> pending_functions; std::vector<std::shared_ptr<Handler>> pending_functions;
for (int i = 0; i < num_events; ++i) { for (int i = 0; i < num_events; ++i) {
auto& info = *reinterpret_cast<Info*>(ev[i].data.ptr); const auto it = epoll_handlers_.find(ev[i].data.fd);
if (it == epoll_handlers_.end()) {
continue;
}
const Info& info = it->second;
if ((info.events & (EPOLLIN | EPOLLPRI)) == (EPOLLIN | EPOLLPRI) && if ((info.events & (EPOLLIN | EPOLLPRI)) == (EPOLLIN | EPOLLPRI) &&
(ev[i].events & EPOLLIN) != ev[i].events) { (ev[i].events & EPOLLIN) != ev[i].events) {
// This handler wants to know about exception events, and just got one. // This handler wants to know about exception events, and just got one.