Home | History | Annotate | Download | only in tests 
      1 //===-- sanitizer_ioctl_test.cc -------------------------------------------===//
      2 //
      3 //                     The LLVM Compiler Infrastructure
      4 //
      5 // This file is distributed under the University of Illinois Open Source
      6 // License. See LICENSE.TXT for details.
      7 //
      8 //===----------------------------------------------------------------------===//
      9 //
     10 // Tests for ioctl interceptor implementation in sanitizer_common.
     11 //
     12 //===----------------------------------------------------------------------===//
     13 
     14 #include "sanitizer_common/sanitizer_platform.h"
     15 #if SANITIZER_LINUX
     16 
     17 #include <linux/input.h>
     18 #include <vector>
     19 
     20 #include "interception/interception.h"
     21 #include "sanitizer_test_utils.h"
     22 #include "sanitizer_common/sanitizer_platform_limits_posix.h"
     23 #include "sanitizer_common/sanitizer_common.h"
     24 #include "gtest/gtest.h"
     25 
     26 
     27 using namespace __sanitizer;
     28 
     29 #define COMMON_INTERCEPTOR_READ_RANGE(ctx, ptr, sz) \
     30   do {                                              \
     31     (void) ctx;                                     \
     32     (void) ptr;                                     \
     33     (void) sz;                                      \
     34   } while (0)
     35 #define COMMON_INTERCEPTOR_WRITE_RANGE(ctx, ptr, sz) \
     36   do {                                               \
     37     (void) ctx;                                      \
     38     (void) ptr;                                      \
     39     (void) sz;                                       \
     40   } while (0)
     41 
     42 #include "sanitizer_common/sanitizer_common_interceptors_ioctl.inc"
     43 
     44 static struct IoctlInit {
     45   IoctlInit() {
     46     ioctl_init();
     47     // Avoid unused function warnings.
     48     (void)&ioctl_common_pre;
     49     (void)&ioctl_common_post;
     50     (void)&ioctl_decode;
     51   }
     52 } ioctl_static_initializer;
     53 
     54 TEST(SanitizerIoctl, Fixup) {
     55   EXPECT_EQ((unsigned)FIONBIO, ioctl_request_fixup(FIONBIO));
     56 
     57   EXPECT_EQ(EVIOCGBIT(0, 0), ioctl_request_fixup(EVIOCGBIT(0, 16)));
     58   EXPECT_EQ(EVIOCGBIT(0, 0), ioctl_request_fixup(EVIOCGBIT(1, 16)));
     59   EXPECT_EQ(EVIOCGBIT(0, 0), ioctl_request_fixup(EVIOCGBIT(1, 17)));
     60   EXPECT_EQ(EVIOCGBIT(0, 0), ioctl_request_fixup(EVIOCGBIT(31, 16)));
     61   EXPECT_NE(EVIOCGBIT(0, 0), ioctl_request_fixup(EVIOCGBIT(32, 16)));
     62 
     63   EXPECT_EQ(EVIOCGABS(0), ioctl_request_fixup(EVIOCGABS(0)));
     64   EXPECT_EQ(EVIOCGABS(0), ioctl_request_fixup(EVIOCGABS(5)));
     65   EXPECT_EQ(EVIOCGABS(0), ioctl_request_fixup(EVIOCGABS(63)));
     66   EXPECT_NE(EVIOCGABS(0), ioctl_request_fixup(EVIOCGABS(64)));
     67 
     68   EXPECT_EQ(EVIOCSABS(0), ioctl_request_fixup(EVIOCSABS(0)));
     69   EXPECT_EQ(EVIOCSABS(0), ioctl_request_fixup(EVIOCSABS(5)));
     70   EXPECT_EQ(EVIOCSABS(0), ioctl_request_fixup(EVIOCSABS(63)));
     71   EXPECT_NE(EVIOCSABS(0), ioctl_request_fixup(EVIOCSABS(64)));
     72 
     73   const ioctl_desc *desc = ioctl_lookup(EVIOCGKEY(16));
     74   EXPECT_NE((void *)0, desc);
     75   EXPECT_EQ(EVIOCGKEY(0), desc->req);
     76 }
     77 
     78 // Test decoding KVM ioctl numbers.
     79 TEST(SanitizerIoctl, KVM_GET_MP_STATE) {
     80   ioctl_desc desc;
     81   unsigned int desc_value = SANITIZER_MIPS ? 0x4004ae98U : 0x8004ae98U;
     82   bool res = ioctl_decode(desc_value, &desc);
     83   EXPECT_TRUE(res);
     84   EXPECT_EQ(ioctl_desc::WRITE, desc.type);
     85   EXPECT_EQ(4U, desc.size);
     86 }
     87 
     88 TEST(SanitizerIoctl, KVM_GET_LAPIC) {
     89   ioctl_desc desc;
     90   unsigned int desc_value = SANITIZER_MIPS ? 0x4400ae8eU : 0x8400ae8eU;
     91   bool res = ioctl_decode(desc_value, &desc);
     92   EXPECT_TRUE(res);
     93   EXPECT_EQ(ioctl_desc::WRITE, desc.type);
     94   EXPECT_EQ(1024U, desc.size);
     95 }
     96 
     97 TEST(SanitizerIoctl, KVM_GET_MSR_INDEX_LIST) {
     98   ioctl_desc desc;
     99   bool res = ioctl_decode(0xc004ae02U, &desc);
    100   EXPECT_TRUE(res);
    101   EXPECT_EQ(ioctl_desc::READWRITE, desc.type);
    102   EXPECT_EQ(4U, desc.size);
    103 }
    104 
    105 #endif // SANITIZER_LINUX
    106