tools/testing/selftests/bpf/test_lirc_mode2_user.c

   1 // SPDX-License-Identifier: GPL-2.0
   2 // test ir decoder
   3 //
   4 // Copyright (C) 2018 Sean Young <sean@mess.org>
   5
   6 // A lirc chardev is a device representing a consumer IR (cir) device which
   7 // can receive infrared signals from remote control and/or transmit IR.
   8 //
   9 // IR is sent as a series of pulses and space somewhat like morse code. The
  10 // BPF program can decode this into scancodes so that rc-core can translate
  11 // this into input key codes using the rc keymap.
  12 //
  13 // This test works by sending IR over rc-loopback, so the IR is processed by
  14 // BPF and then decoded into scancodes. The lirc chardev must be the one
  15 // associated with rc-loopback, see the output of ir-keytable(1).
  16 //
  17 // The following CONFIG options must be enabled for the test to succeed:
  18 // CONFIG_RC_CORE=y
  19 // CONFIG_BPF_RAWIR_EVENT=y
  20 // CONFIG_RC_LOOPBACK=y
  21
  22 // Steps:
  23 // 1. Open the /dev/lircN device for rc-loopback (given on command line)
  24 // 2. Attach bpf_lirc_mode2 program which decodes some IR.
  25 // 3. Send some IR to the same IR device; since it is loopback, this will
  26 //    end up in the bpf program
  27 // 4. bpf program should decode IR and report keycode
  28 // 5. We can read keycode from same /dev/lirc device
  29
  30 #include <linux/bpf.h>
  31 #include <linux/input.h>
  32 #include <errno.h>
  33 #include <stdio.h>
  34 #include <stdlib.h>
  35 #include <string.h>
  36 #include <unistd.h>
  37 #include <poll.h>
  38 #include <sys/types.h>
  39 #include <sys/ioctl.h>
  40 #include <sys/stat.h>
  41 #include <fcntl.h>
  42
  43 #include "bpf_util.h"
  44 #include <bpf/bpf.h>
  45 #include <bpf/libbpf.h>
  46
  47 #include "testing_helpers.h"
  48
  49 int main(int argc, char **argv)
  50 {
  51         struct bpf_object *obj;
  52         int ret, lircfd, progfd, inputfd;
  53         int testir1 = 0x1dead;
  54         int testir2 = 0x20101;
  55         u32 prog_ids[10], prog_flags[10], prog_cnt;
  56
  57         if (argc != 3) {
  58                 printf("Usage: %s /dev/lircN /dev/input/eventM\n", argv[0]);
  59                 return 2;
  60         }
  61
  62         ret = bpf_prog_test_load("test_lirc_mode2_kern.o",
  63                                  BPF_PROG_TYPE_LIRC_MODE2, &obj, &progfd);
  64         if (ret) {
  65                 printf("Failed to load bpf program\n");
  66                 return 1;
  67         }
  68
  69         lircfd = open(argv[1], O_RDWR | O_NONBLOCK);
  70         if (lircfd == -1) {
  71                 printf("failed to open lirc device %s: %m\n", argv[1]);
  72                 return 1;
  73         }
  74
  75         /* Let's try detach it before it was ever attached */
  76         ret = bpf_prog_detach2(progfd, lircfd, BPF_LIRC_MODE2);
  77         if (ret != -1 || errno != ENOENT) {
  78                 printf("bpf_prog_detach2 not attached should fail: %m\n");
  79                 return 1;
  80         }
  81
  82         inputfd = open(argv[2], O_RDONLY | O_NONBLOCK);
  83         if (inputfd == -1) {
  84                 printf("failed to open input device %s: %m\n", argv[1]);
  85                 return 1;
  86         }
  87
  88         prog_cnt = 10;
  89         ret = bpf_prog_query(lircfd, BPF_LIRC_MODE2, 0, prog_flags, prog_ids,
  90                              &prog_cnt);
  91         if (ret) {
  92                 printf("Failed to query bpf programs on lirc device: %m\n");
  93                 return 1;
  94         }
  95
  96         if (prog_cnt != 0) {
  97                 printf("Expected nothing to be attached\n");
  98                 return 1;
  99         }
 100
 101         ret = bpf_prog_attach(progfd, lircfd, BPF_LIRC_MODE2, 0);
 102         if (ret) {
 103                 printf("Failed to attach bpf to lirc device: %m\n");
 104                 return 1;
 105         }
 106
 107         /* Write raw IR */
 108         ret = write(lircfd, &testir1, sizeof(testir1));
 109         if (ret != sizeof(testir1)) {
 110                 printf("Failed to send test IR message: %m\n");
 111                 return 1;
 112         }
 113
 114         struct pollfd pfd = { .fd = inputfd, .events = POLLIN };
 115         struct input_event event;
 116
 117         for (;;) {
 118                 poll(&pfd, 1, 100);
 119
 120                 /* Read decoded IR */
 121                 ret = read(inputfd, &event, sizeof(event));
 122                 if (ret != sizeof(event)) {
 123                         printf("Failed to read decoded IR: %m\n");
 124                         return 1;
 125                 }
 126
 127                 if (event.type == EV_MSC && event.code == MSC_SCAN &&
 128                     event.value == 0xdead) {
 129                         break;
 130                 }
 131         }
 132
 133         /* Write raw IR */
 134         ret = write(lircfd, &testir2, sizeof(testir2));
 135         if (ret != sizeof(testir2)) {
 136                 printf("Failed to send test IR message: %m\n");
 137                 return 1;
 138         }
 139
 140         for (;;) {
 141                 poll(&pfd, 1, 100);
 142
 143                 /* Read decoded IR */
 144                 ret = read(inputfd, &event, sizeof(event));
 145                 if (ret != sizeof(event)) {
 146                         printf("Failed to read decoded IR: %m\n");
 147                         return 1;
 148                 }
 149
 150                 if (event.type == EV_REL && event.code == REL_Y &&
 151                     event.value == 1 ) {
 152                         break;
 153                 }
 154         }
 155
 156         prog_cnt = 10;
 157         ret = bpf_prog_query(lircfd, BPF_LIRC_MODE2, 0, prog_flags, prog_ids,
 158                              &prog_cnt);
 159         if (ret) {
 160                 printf("Failed to query bpf programs on lirc device: %m\n");
 161                 return 1;
 162         }
 163
 164         if (prog_cnt != 1) {
 165                 printf("Expected one program to be attached\n");
 166                 return 1;
 167         }
 168
 169         /* Let's try detaching it now it is actually attached */
 170         ret = bpf_prog_detach2(progfd, lircfd, BPF_LIRC_MODE2);
 171         if (ret) {
 172                 printf("bpf_prog_detach2: returned %m\n");
 173                 return 1;
 174         }
 175
 176         return 0;
 177 }