| OLD | NEW |
| (Empty) |
| 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 | |
| OLD | NEW |