components/nacl/loader/nonsfi/irt_icache_unittest.cc

   1 // Copyright 2014 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #if defined(__arm__)  // nacl_irt_icache is supported only on ARM.
   6
   7 #include <errno.h>
   8 #include <sys/mman.h>
   9
  10 #include "components/nacl/loader/nonsfi/irt_interfaces.h"
  11 #include "testing/gtest/include/gtest/gtest.h"
  12
  13 TEST(NaClIrtIcacheTest, ClearCache) {
  14   int (*irt_clear_cache)(void*, size_t) =
  15       nacl::nonsfi::kIrtIcache.clear_cache;
  16   // Create code for a function to return 0x01.
  17   static const uint32_t code_template[] = {
  18     0xe3000001,  // movw r0, #0x1
  19     0xe12fff1e,  // bx lr
  20   };
  21   void* start =
  22       mmap(NULL, sizeof(code_template), PROT_READ | PROT_WRITE | PROT_EXEC,
  23            MAP_PRIVATE | MAP_ANONYMOUS, -1, 0);
  24   EXPECT_NE(MAP_FAILED, start);
  25   memcpy(start, code_template, sizeof(code_template));
  26   size_t size = sizeof(code_template);
  27   EXPECT_EQ(0, irt_clear_cache(start, size));
  28   typedef int (*TestFunc)(void);
  29   TestFunc func = reinterpret_cast<TestFunc>(start);
  30   EXPECT_EQ(0x1, func());
  31   // Modify the function to return 0x11.
  32   *reinterpret_cast<uint32_t*>(start) = 0xe3000011;  // movw r0, #0x11
  33   // In most cases 0x1 is still returned because the cached code is executed
  34   // although the CPU is not obliged to do so.
  35   // Uncomment the following line to see if the cached code is executed.
  36   // EXPECT_EQ(0x1, func());
  37   EXPECT_EQ(0, irt_clear_cache(start, size));
  38   // Now it is ensured that 0x11 is returned because I-cache was invalidated
  39   // and updated with the new code.
  40   EXPECT_EQ(0x11, func());
  41   EXPECT_EQ(0, munmap(start, sizeof(code_template)));
  42 }
  43
  44 TEST(NaClIrtIcacheTest, ClearCacheInvalidArg) {
  45   int (*irt_clear_cache)(void*, size_t) =
  46       nacl::nonsfi::kIrtIcache.clear_cache;
  47   const size_t mem_size = 256;
  48   void* start =
  49       mmap(NULL, mem_size, PROT_READ | PROT_WRITE | PROT_EXEC,
  50            MAP_PRIVATE | MAP_ANONYMOUS, -1, 0);
  51   EXPECT_EQ(0, irt_clear_cache(start, mem_size));
  52   EXPECT_EQ(EINVAL, irt_clear_cache(start, 0));
  53   EXPECT_EQ(EINVAL, irt_clear_cache(NULL, mem_size));
  54   EXPECT_EQ(0, munmap(start, mem_size));
  55 }
  56
  57 #endif