src/login/logind-button.c

   1 /* SPDX-License-Identifier: LGPL-2.1+ */
   2
   3 #include <errno.h>
   4 #include <fcntl.h>
   5 #include <string.h>
   6 #include <sys/ioctl.h>
   7 #include <unistd.h>
   8 #include <linux/input.h>
   9
  10 #include "sd-messages.h"
  11
  12 #include "alloc-util.h"
  13 #include "fd-util.h"
  14 #include "logind-button.h"
  15 #include "missing_input.h"
  16 #include "string-util.h"
  17 #include "util.h"
  18
  19 #define CONST_MAX4(a, b, c, d) CONST_MAX(CONST_MAX(a, b), CONST_MAX(c, d))
  20
  21 #define ULONG_BITS (sizeof(unsigned long)*8)
  22
  23 static bool bitset_get(const unsigned long *bits, unsigned i) {
  24         return (bits[i / ULONG_BITS] >> (i % ULONG_BITS)) & 1UL;
  25 }
  26
  27 static void bitset_put(unsigned long *bits, unsigned i) {
  28         bits[i / ULONG_BITS] |= (unsigned long) 1 << (i % ULONG_BITS);
  29 }
  30
  31 Button* button_new(Manager *m, const char *name) {
  32         Button *b;
  33
  34         assert(m);
  35         assert(name);
  36
  37         b = new0(Button, 1);
  38         if (!b)
  39                 return NULL;
  40
  41         b->name = strdup(name);
  42         if (!b->name)
  43                 return mfree(b);
  44
  45         if (hashmap_put(m->buttons, b->name, b) < 0) {
  46                 free(b->name);
  47                 return mfree(b);
  48         }
  49
  50         b->manager = m;
  51         b->fd = -1;
  52
  53         return b;
  54 }
  55
  56 void button_free(Button *b) {
  57         assert(b);
  58
  59         hashmap_remove(b->manager->buttons, b->name);
  60
  61         sd_event_source_unref(b->io_event_source);
  62         sd_event_source_unref(b->check_event_source);
  63
  64         if (b->fd >= 0)
  65                 /* If the device has been unplugged close() returns
  66                  * ENODEV, let's ignore this, hence we don't use
  67                  * safe_close() */
  68                 (void) close(b->fd);
  69
  70         free(b->name);
  71         free(b->seat);
  72         free(b);
  73 }
  74
  75 int button_set_seat(Button *b, const char *sn) {
  76         char *s;
  77
  78         assert(b);
  79         assert(sn);
  80
  81         s = strdup(sn);
  82         if (!s)
  83                 return -ENOMEM;
  84
  85         free(b->seat);
  86         b->seat = s;
  87
  88         return 0;
  89 }
  90
  91 static void button_lid_switch_handle_action(Manager *manager, bool is_edge) {
  92         HandleAction handle_action;
  93
  94         assert(manager);
  95
  96         /* If we are docked or on external power, handle the lid switch
  97          * differently */
  98         if (manager_is_docked_or_external_displays(manager))
  99                 handle_action = manager->handle_lid_switch_docked;
 100         else if (manager->handle_lid_switch_ep != _HANDLE_ACTION_INVALID &&
 101                  manager_is_on_external_power())
 102                 handle_action = manager->handle_lid_switch_ep;
 103         else
 104                 handle_action = manager->handle_lid_switch;
 105
 106         manager_handle_action(manager, INHIBIT_HANDLE_LID_SWITCH, handle_action, manager->lid_switch_ignore_inhibited, is_edge);
 107 }
 108
 109 static int button_recheck(sd_event_source *e, void *userdata) {
 110         Button *b = userdata;
 111
 112         assert(b);
 113         assert(b->lid_closed);
 114
 115         button_lid_switch_handle_action(b->manager, false);
 116         return 1;
 117 }
 118
 119 static int button_install_check_event_source(Button *b) {
 120         int r;
 121         assert(b);
 122
 123         /* Install a post handler, so that we keep rechecking as long as the lid is closed. */
 124
 125         if (b->check_event_source)
 126                 return 0;
 127
 128         r = sd_event_add_post(b->manager->event, &b->check_event_source, button_recheck, b);
 129         if (r < 0)
 130                 return r;
 131
 132         return sd_event_source_set_priority(b->check_event_source, SD_EVENT_PRIORITY_IDLE+1);
 133 }
 134
 135 static int button_dispatch(sd_event_source *s, int fd, uint32_t revents, void *userdata) {
 136         Button *b = userdata;
 137         struct input_event ev;
 138         ssize_t l;
 139
 140         assert(s);
 141         assert(fd == b->fd);
 142         assert(b);
 143
 144         l = read(b->fd, &ev, sizeof(ev));
 145         if (l < 0)
 146                 return errno != EAGAIN ? -errno : 0;
 147         if ((size_t) l < sizeof(ev))
 148                 return -EIO;
 149
 150         if (ev.type == EV_KEY && ev.value > 0) {
 151
 152                 switch (ev.code) {
 153
 154                 case KEY_POWER:
 155                 case KEY_POWER2:
 156                         log_struct(LOG_INFO,
 157                                    LOG_MESSAGE("Power key pressed."),
 158                                    "MESSAGE_ID=" SD_MESSAGE_POWER_KEY_STR);
 159
 160                         manager_handle_action(b->manager, INHIBIT_HANDLE_POWER_KEY, b->manager->handle_power_key, b->manager->power_key_ignore_inhibited, true);
 161                         break;
 162
 163                 /* The kernel is a bit confused here:
 164
 165                    KEY_SLEEP   = suspend-to-ram, which everybody else calls "suspend"
 166                    KEY_SUSPEND = suspend-to-disk, which everybody else calls "hibernate"
 167                 */
 168
 169                 case KEY_SLEEP:
 170                         log_struct(LOG_INFO,
 171                                    LOG_MESSAGE("Suspend key pressed."),
 172                                    "MESSAGE_ID=" SD_MESSAGE_SUSPEND_KEY_STR);
 173
 174                         manager_handle_action(b->manager, INHIBIT_HANDLE_SUSPEND_KEY, b->manager->handle_suspend_key, b->manager->suspend_key_ignore_inhibited, true);
 175                         break;
 176
 177                 case KEY_SUSPEND:
 178                         log_struct(LOG_INFO,
 179                                    LOG_MESSAGE("Hibernate key pressed."),
 180                                    "MESSAGE_ID=" SD_MESSAGE_HIBERNATE_KEY_STR);
 181
 182                         manager_handle_action(b->manager, INHIBIT_HANDLE_HIBERNATE_KEY, b->manager->handle_hibernate_key, b->manager->hibernate_key_ignore_inhibited, true);
 183                         break;
 184                 }
 185
 186         } else if (ev.type == EV_SW && ev.value > 0) {
 187
 188                 if (ev.code == SW_LID) {
 189                         log_struct(LOG_INFO,
 190                                    LOG_MESSAGE("Lid closed."),
 191                                    "MESSAGE_ID=" SD_MESSAGE_LID_CLOSED_STR);
 192
 193                         b->lid_closed = true;
 194                         button_lid_switch_handle_action(b->manager, true);
 195                         button_install_check_event_source(b);
 196
 197                 } else if (ev.code == SW_DOCK) {
 198                         log_struct(LOG_INFO,
 199                                    LOG_MESSAGE("System docked."),
 200                                    "MESSAGE_ID=" SD_MESSAGE_SYSTEM_DOCKED_STR);
 201
 202                         b->docked = true;
 203                 }
 204
 205         } else if (ev.type == EV_SW && ev.value == 0) {
 206
 207                 if (ev.code == SW_LID) {
 208                         log_struct(LOG_INFO,
 209                                    LOG_MESSAGE("Lid opened."),
 210                                    "MESSAGE_ID=" SD_MESSAGE_LID_OPENED_STR);
 211
 212                         b->lid_closed = false;
 213                         b->check_event_source = sd_event_source_unref(b->check_event_source);
 214
 215                 } else if (ev.code == SW_DOCK) {
 216                         log_struct(LOG_INFO,
 217                                    LOG_MESSAGE("System undocked."),
 218                                    "MESSAGE_ID=" SD_MESSAGE_SYSTEM_UNDOCKED_STR);
 219
 220                         b->docked = false;
 221                 }
 222         }
 223
 224         return 0;
 225 }
 226
 227 static int button_suitable(Button *b) {
 228         unsigned long types[CONST_MAX(EV_KEY, EV_SW)/ULONG_BITS+1];
 229
 230         assert(b);
 231         assert(b->fd);
 232
 233         if (ioctl(b->fd, EVIOCGBIT(EV_SYN, sizeof(types)), types) < 0)
 234                 return -errno;
 235
 236         if (bitset_get(types, EV_KEY)) {
 237                 unsigned long keys[CONST_MAX4(KEY_POWER, KEY_POWER2, KEY_SLEEP, KEY_SUSPEND)/ULONG_BITS+1];
 238
 239                 if (ioctl(b->fd, EVIOCGBIT(EV_KEY, sizeof(keys)), keys) < 0)
 240                         return -errno;
 241
 242                 if (bitset_get(keys, KEY_POWER) ||
 243                     bitset_get(keys, KEY_POWER2) ||
 244                     bitset_get(keys, KEY_SLEEP) ||
 245                     bitset_get(keys, KEY_SUSPEND))
 246                         return true;
 247         }
 248
 249         if (bitset_get(types, EV_SW)) {
 250                 unsigned long switches[CONST_MAX(SW_LID, SW_DOCK)/ULONG_BITS+1];
 251
 252                 if (ioctl(b->fd, EVIOCGBIT(EV_SW, sizeof(switches)), switches) < 0)
 253                         return -errno;
 254
 255                 if (bitset_get(switches, SW_LID) ||
 256                     bitset_get(switches, SW_DOCK))
 257                         return true;
 258         }
 259
 260         return false;
 261 }
 262
 263 static int button_set_mask(Button *b) {
 264         unsigned long
 265                 types[CONST_MAX(EV_KEY, EV_SW)/ULONG_BITS+1] = {},
 266                 keys[CONST_MAX4(KEY_POWER, KEY_POWER2, KEY_SLEEP, KEY_SUSPEND)/ULONG_BITS+1] = {},
 267                 switches[CONST_MAX(SW_LID, SW_DOCK)/ULONG_BITS+1] = {};
 268         struct input_mask mask;
 269
 270         assert(b);
 271         assert(b->fd >= 0);
 272
 273         bitset_put(types, EV_KEY);
 274         bitset_put(types, EV_SW);
 275
 276         mask = (struct input_mask) {
 277                 .type = EV_SYN,
 278                 .codes_size = sizeof(types),
 279                 .codes_ptr = PTR_TO_UINT64(types),
 280         };
 281
 282         if (ioctl(b->fd, EVIOCSMASK, &mask) < 0)
 283                 /* Log only at debug level if the kernel doesn't do EVIOCSMASK yet */
 284                 return log_full_errno(IN_SET(errno, ENOTTY, EOPNOTSUPP, EINVAL) ? LOG_DEBUG : LOG_WARNING,
 285                                       errno, "Failed to set EV_SYN event mask on /dev/input/%s: %m", b->name);
 286
 287         bitset_put(keys, KEY_POWER);
 288         bitset_put(keys, KEY_POWER2);
 289         bitset_put(keys, KEY_SLEEP);
 290         bitset_put(keys, KEY_SUSPEND);
 291
 292         mask = (struct input_mask) {
 293                 .type = EV_KEY,
 294                 .codes_size = sizeof(keys),
 295                 .codes_ptr = PTR_TO_UINT64(keys),
 296         };
 297
 298         if (ioctl(b->fd, EVIOCSMASK, &mask) < 0)
 299                 return log_warning_errno(errno, "Failed to set EV_KEY event mask on /dev/input/%s: %m", b->name);
 300
 301         bitset_put(switches, SW_LID);
 302         bitset_put(switches, SW_DOCK);
 303
 304         mask = (struct input_mask) {
 305                 .type = EV_SW,
 306                 .codes_size = sizeof(switches),
 307                 .codes_ptr = PTR_TO_UINT64(switches),
 308         };
 309
 310         if (ioctl(b->fd, EVIOCSMASK, &mask) < 0)
 311                 return log_warning_errno(errno, "Failed to set EV_SW event mask on /dev/input/%s: %m", b->name);
 312
 313         return 0;
 314 }
 315
 316 int button_open(Button *b) {
 317         char *p, name[256];
 318         int r;
 319
 320         assert(b);
 321
 322         b->fd = safe_close(b->fd);
 323
 324         p = strjoina("/dev/input/", b->name);
 325
 326         b->fd = open(p, O_RDWR|O_CLOEXEC|O_NOCTTY|O_NONBLOCK);
 327         if (b->fd < 0)
 328                 return log_warning_errno(errno, "Failed to open %s: %m", p);
 329
 330         r = button_suitable(b);
 331         if (r < 0)
 332                 return log_warning_errno(r, "Failed to determine whether input device is relevant to us: %m");
 333         if (r == 0)
 334                 return log_debug_errno(SYNTHETIC_ERRNO(EADDRNOTAVAIL),
 335                                        "Device %s does not expose keys or switches relevant to us, ignoring.",
 336                                        p);
 337
 338         if (ioctl(b->fd, EVIOCGNAME(sizeof(name)), name) < 0) {
 339                 r = log_error_errno(errno, "Failed to get input name: %m");
 340                 goto fail;
 341         }
 342
 343         (void) button_set_mask(b);
 344
 345         r = sd_event_add_io(b->manager->event, &b->io_event_source, b->fd, EPOLLIN, button_dispatch, b);
 346         if (r < 0) {
 347                 log_error_errno(r, "Failed to add button event: %m");
 348                 goto fail;
 349         }
 350
 351         log_info("Watching system buttons on /dev/input/%s (%s)", b->name, name);
 352
 353         return 0;
 354
 355 fail:
 356         b->fd = safe_close(b->fd);
 357         return r;
 358 }
 359
 360 int button_check_switches(Button *b) {
 361         unsigned long switches[CONST_MAX(SW_LID, SW_DOCK)/ULONG_BITS+1] = {};
 362         assert(b);
 363
 364         if (b->fd < 0)
 365                 return -EINVAL;
 366
 367         if (ioctl(b->fd, EVIOCGSW(sizeof(switches)), switches) < 0)
 368                 return -errno;
 369
 370         b->lid_closed = bitset_get(switches, SW_LID);
 371         b->docked = bitset_get(switches, SW_DOCK);
 372
 373         if (b->lid_closed)
 374                 button_install_check_event_source(b);
 375
 376         return 0;
 377 }