platform_system_core/init/keychords.cpp

294 lines
8.5 KiB
C++

/*
* Copyright (C) 2010 The Android Open Source Project
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#include "keychords.h"
#include <dirent.h>
#include <fcntl.h>
#include <linux/input.h>
#include <sys/cdefs.h>
#include <sys/inotify.h>
#include <sys/ioctl.h>
#include <sys/types.h>
#include <unistd.h>
#include <algorithm>
#include <functional>
#include <map>
#include <memory>
#include <string>
#include <vector>
#include <android-base/logging.h>
namespace android {
namespace init {
Keychords::Keychords() : epoll_(nullptr), inotify_fd_(-1) {}
Keychords::~Keychords() noexcept {
if (inotify_fd_ >= 0) {
epoll_->UnregisterHandler(inotify_fd_);
::close(inotify_fd_);
}
while (!registration_.empty()) GeteventCloseDevice(registration_.begin()->first);
}
Keychords::Mask::Mask(size_t bit) : bits_((bit + sizeof(mask_t) - 1) / sizeof(mask_t), 0) {}
void Keychords::Mask::SetBit(size_t bit, bool value) {
auto idx = bit / (kBitsPerByte * sizeof(mask_t));
if (idx >= bits_.size()) return;
if (value) {
bits_[idx] |= mask_t(1) << (bit % (kBitsPerByte * sizeof(mask_t)));
} else {
bits_[idx] &= ~(mask_t(1) << (bit % (kBitsPerByte * sizeof(mask_t))));
}
}
bool Keychords::Mask::GetBit(size_t bit) const {
auto idx = bit / (kBitsPerByte * sizeof(mask_t));
return bits_[idx] & (mask_t(1) << (bit % (kBitsPerByte * sizeof(mask_t))));
}
size_t Keychords::Mask::bytesize() const {
return bits_.size() * sizeof(mask_t);
}
void* Keychords::Mask::data() {
return bits_.data();
}
size_t Keychords::Mask::size() const {
return bits_.size() * sizeof(mask_t) * kBitsPerByte;
}
void Keychords::Mask::resize(size_t bit) {
auto idx = bit / (kBitsPerByte * sizeof(mask_t));
if (idx >= bits_.size()) {
bits_.resize(idx + 1, 0);
}
}
Keychords::Mask::operator bool() const {
for (size_t i = 0; i < bits_.size(); ++i) {
if (bits_[i]) return true;
}
return false;
}
Keychords::Mask Keychords::Mask::operator&(const Keychords::Mask& rval) const {
auto len = std::min(bits_.size(), rval.bits_.size());
Keychords::Mask ret;
ret.bits_.resize(len);
for (size_t i = 0; i < len; ++i) {
ret.bits_[i] = bits_[i] & rval.bits_[i];
}
return ret;
}
void Keychords::Mask::operator|=(const Keychords::Mask& rval) {
auto len = rval.bits_.size();
bits_.resize(len);
for (size_t i = 0; i < len; ++i) {
bits_[i] |= rval.bits_[i];
}
}
Keychords::Entry::Entry() : notified(false) {}
void Keychords::LambdaCheck() {
for (auto& [keycodes, entry] : entries_) {
auto found = true;
for (auto& code : keycodes) {
if (!current_.GetBit(code)) {
entry.notified = false;
found = false;
break;
}
}
if (!found) continue;
if (entry.notified) continue;
entry.notified = true;
handler_(keycodes);
}
}
void Keychords::LambdaHandler(int fd) {
input_event event;
auto res = TEMP_FAILURE_RETRY(::read(fd, &event, sizeof(event)));
if ((res != sizeof(event)) || (event.type != EV_KEY)) return;
current_.SetBit(event.code, event.value);
LambdaCheck();
}
bool Keychords::GeteventEnable(int fd) {
// Make sure it is an event channel, should pass this ioctl call
int version;
if (::ioctl(fd, EVIOCGVERSION, &version)) return false;
#ifdef EVIOCSMASK
static auto EviocsmaskSupported = true;
if (EviocsmaskSupported) {
Keychords::Mask mask(EV_KEY);
mask.SetBit(EV_KEY);
input_mask msg = {};
msg.type = EV_SYN;
msg.codes_size = mask.bytesize();
msg.codes_ptr = reinterpret_cast<uintptr_t>(mask.data());
if (::ioctl(fd, EVIOCSMASK, &msg) == -1) {
PLOG(WARNING) << "EVIOCSMASK not supported";
EviocsmaskSupported = false;
}
}
#endif
Keychords::Mask mask;
for (auto& [keycodes, entry] : entries_) {
for (auto& code : keycodes) {
mask.resize(code);
mask.SetBit(code);
}
}
current_.resize(mask.size());
Keychords::Mask available(mask.size());
auto res = ::ioctl(fd, EVIOCGBIT(EV_KEY, available.bytesize()), available.data());
if (res == -1) return false;
if (!(available & mask)) return false;
#ifdef EVIOCSMASK
if (EviocsmaskSupported) {
input_mask msg = {};
msg.type = EV_KEY;
msg.codes_size = mask.bytesize();
msg.codes_ptr = reinterpret_cast<uintptr_t>(mask.data());
::ioctl(fd, EVIOCSMASK, &msg);
}
#endif
Keychords::Mask set(mask.size());
res = ::ioctl(fd, EVIOCGKEY(res), set.data());
if (res > 0) {
current_ |= mask & available & set;
LambdaCheck();
}
if (auto result = epoll_->RegisterHandler(fd, [this, fd]() { this->LambdaHandler(fd); });
!result.ok()) {
LOG(WARNING) << "Could not register keychord epoll handler: " << result.error();
return false;
}
return true;
}
void Keychords::GeteventOpenDevice(const std::string& device) {
if (registration_.count(device)) return;
auto fd = TEMP_FAILURE_RETRY(::open(device.c_str(), O_RDONLY | O_CLOEXEC));
if (fd == -1) {
PLOG(ERROR) << "Can not open " << device;
return;
}
if (!GeteventEnable(fd)) {
::close(fd);
} else {
registration_.emplace(device, fd);
}
}
void Keychords::GeteventCloseDevice(const std::string& device) {
auto it = registration_.find(device);
if (it == registration_.end()) return;
auto fd = (*it).second;
epoll_->UnregisterHandler(fd);
registration_.erase(it);
::close(fd);
}
void Keychords::InotifyHandler() {
unsigned char buf[512]; // History shows 32-64 bytes typical
auto res = TEMP_FAILURE_RETRY(::read(inotify_fd_, buf, sizeof(buf)));
if (res < 0) {
PLOG(WARNING) << "could not get event";
return;
}
auto event_buf = buf;
while (static_cast<size_t>(res) >= sizeof(inotify_event)) {
auto event = reinterpret_cast<inotify_event*>(event_buf);
auto event_size = sizeof(inotify_event) + event->len;
if (static_cast<size_t>(res) < event_size) break;
if (event->len) {
std::string devname(kDevicePath);
devname += '/';
devname += event->name;
if (event->mask & IN_CREATE) {
GeteventOpenDevice(devname);
} else {
GeteventCloseDevice(devname);
}
}
res -= event_size;
event_buf += event_size;
}
}
void Keychords::GeteventOpenDevice() {
inotify_fd_ = ::inotify_init1(IN_NONBLOCK | IN_CLOEXEC);
if (inotify_fd_ < 0) {
PLOG(WARNING) << "Could not instantiate inotify for " << kDevicePath;
} else if (::inotify_add_watch(inotify_fd_, kDevicePath, IN_DELETE | IN_CREATE | IN_ONLYDIR) <
0) {
PLOG(WARNING) << "Could not add watch for " << kDevicePath;
::close(inotify_fd_);
inotify_fd_ = -1;
}
std::unique_ptr<DIR, decltype(&closedir)> device(opendir(kDevicePath), closedir);
if (device) {
dirent* entry;
while ((entry = readdir(device.get()))) {
if (entry->d_name[0] == '.') continue;
std::string devname(kDevicePath);
devname += '/';
devname += entry->d_name;
GeteventOpenDevice(devname);
}
}
if (inotify_fd_ >= 0) {
if (auto result =
epoll_->RegisterHandler(inotify_fd_, [this]() { this->InotifyHandler(); });
!result.ok()) {
LOG(WARNING) << "Could not register keychord epoll handler: " << result.error();
}
}
}
void Keychords::Register(const std::vector<int>& keycodes) {
if (keycodes.empty()) return;
entries_.try_emplace(keycodes, Entry());
}
void Keychords::Start(Epoll* epoll, std::function<void(const std::vector<int>&)> handler) {
epoll_ = epoll;
handler_ = std::move(handler);
if (entries_.size()) GeteventOpenDevice();
}
} // namespace init
} // namespace android