service_runtime/sel_memory_unittest.cc

   1 /*
   2  * Copyright 2008, Google Inc.
   3  * All rights reserved.
   4  *
   5  * Redistribution and use in source and binary forms, with or without
   6  * modification, are permitted provided that the following conditions are
   7  * met:
   8  *
   9  *     * Redistributions of source code must retain the above copyright
  10  * notice, this list of conditions and the following disclaimer.
  11  *     * Redistributions in binary form must reproduce the above
  12  * copyright notice, this list of conditions and the following disclaimer
  13  * in the documentation and/or other materials provided with the
  14  * distribution.
  15  *     * Neither the name of Google Inc. nor the names of its
  16  * contributors may be used to endorse or promote products derived from
  17  * this software without specific prior written permission.
  18  *
  19  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  20  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  21  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
  22  * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
  23  * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  24  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  25  * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  26  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  27  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  28  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  29  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  30  */
  31
  32 // Testing NativeClient cross-platfom memory management functions
  33
  34 #include "native_client/include/portability.h"
  35 #include "native_client/include/nacl_platform.h"
  36 #include "native_client/service_runtime/nacl_log.h"
  37
  38 #if NACL_WINDOWS
  39 #include "native_client/service_runtime/win/mman.h"
  40 #endif
  41 #include "native_client/service_runtime/sel_memory.h"
  42
  43 #include "gtest/gtest.h"
  44
  45 class SelMemoryBasic : public testing::Test {
  46 protected:
  47   virtual void SetUp();
  48   virtual void TearDown();
  49   // TODO - do we need a destructor here?
  50 };
  51
  52 void SelMemoryBasic::SetUp() {
  53   NaClLogModuleInit();
  54 }
  55
  56 void SelMemoryBasic::TearDown() {
  57   NaClLogModuleFini();
  58 }
  59
  60
  61 TEST_F(SelMemoryBasic, AllocationTests) {
  62   int res = 0;
  63   void *p = NULL;
  64   int size;
  65
  66   size = 0x2001;  // not power of two - should be supported
  67
  68   res = NaCl_page_alloc(&p, size);
  69   EXPECT_EQ(0, res);
  70   EXPECT_NE((void*)NULL, p);
  71
  72   NaCl_page_free(p, size);
  73   p = NULL;
  74
  75   // Try to allocate large memory block
  76   size = 256 * 1024 * 1024;  // 256M
  77
  78   res = NaCl_page_alloc(&p, size);
  79   EXPECT_EQ(0, res);
  80   EXPECT_NE((void*)NULL, p);
  81
  82   NaCl_page_free(p, size);
  83 }
  84
  85 TEST_F(SelMemoryBasic, mprotect) {
  86   int res = 0;
  87   void *p = NULL;
  88   int size;
  89   char *addr;
  90
  91   size = 0x100000;
  92
  93   res = NaCl_page_alloc(&p, size);
  94   EXPECT_EQ(0, res);
  95   EXPECT_NE((void*)NULL, p);
  96
  97   // TODO - since ASSERT_DEATH is not supported for client projects,
  98   // we cannot use gUnit to test the protection. We might want to add some
  99   // internal processing (based on SEH/signals) at some stage
 100
 101   res = NaCl_mprotect(p, size, PROT_READ |PROT_WRITE);
 102   EXPECT_EQ(0, res);
 103   addr = reinterpret_cast<char*>(p);
 104   addr[0] = '5';
 105
 106   res = NaCl_mprotect(p, size, PROT_READ);
 107   EXPECT_EQ(0, res);
 108   EXPECT_EQ('5', addr[0]);
 109
 110   res = NaCl_mprotect(p, size, PROT_READ|PROT_WRITE|PROT_EXEC);
 111
 112   NaCl_page_free(p, size);
 113 }