Make SyzyAsan support the allocation > 1GB

syzygy/agent/asan/heap_managers/block_heap_manager_unittest.cc

 // Copyright 2014 Google Inc. All Rights Reserved.
 //
 // Licensed under the Apache License, Version 2.0 (the "License");
 // you may not use this file except in compliance with the License.
 // You may obtain a copy of the License at
 //
 //     http://www.apache.org/licenses/LICENSE-2.0
 //
 // Unless required by applicable law or agreed to in writing, software
 // distributed under the License is distributed on an "AS IS" BASIS,
@@ skipping 26 matching lines @@
 #include "syzygy/agent/asan/unittest_util.h"
 #include "syzygy/agent/asan/heaps/internal_heap.h"
 #include "syzygy/agent/asan/heaps/large_block_heap.h"
 #include "syzygy/agent/asan/heaps/simple_block_heap.h"
 #include "syzygy/agent/asan/heaps/win_heap.h"
 #include "syzygy/agent/asan/heaps/zebra_block_heap.h"
 #include "syzygy/agent/asan/memory_notifiers/null_memory_notifier.h"
 #include "syzygy/assm/assembler.h"
 #include "syzygy/assm/buffer_serializer.h"
 #include "syzygy/common/asan_parameters.h"
+#include "syzygy/testing/laa.h"
 
 namespace agent {
 namespace asan {
 namespace heap_managers {
 
 namespace {
 
 using heaps::ZebraBlockHeap;
 using testing::IsAccessible;
 using testing::IsNotAccessible;
@@ skipping 548 matching lines @@
   const size_t kAllocSize = 17;
   HeapId heap_id = heap_manager_->CreateHeap();
   EXPECT_NE(0u, heap_id);
   void* alloc = heap_manager_->Allocate(heap_id, kAllocSize);
   EXPECT_NE(static_cast<void*>(nullptr), alloc);
   EXPECT_LE(kAllocSize, heap_manager_->Size(heap_id, alloc));
   EXPECT_TRUE(heap_manager_->Free(heap_id, alloc));
   EXPECT_TRUE(heap_manager_->DestroyHeap(heap_id));
 }
 
+TEST_F(BlockHeapManagerTest, AllocAndFreeLargeBlock) {
+  TEST_ONLY_SUPPORTS_4G();
+
+  const size_t kAllocSize = 0x7000001c;
+  HeapId heap_id = heap_manager_->CreateHeap();
+  EXPECT_NE(0u, heap_id);
+  void* alloc = heap_manager_->Allocate(heap_id, kAllocSize);
+  EXPECT_NE(static_cast<void*>(nullptr), alloc);
+  EXPECT_LE(kAllocSize, heap_manager_->Size(heap_id, alloc));
+  EXPECT_TRUE(heap_manager_->Free(heap_id, alloc));
+  EXPECT_TRUE(heap_manager_->DestroyHeap(heap_id));
+}
+
+TEST_F(BlockHeapManagerTest, AllocLargeBlockFail) {
+  const size_t kAllocSize = 0x80000000;
+  HeapId heap_id = heap_manager_->CreateHeap();
+  EXPECT_NE(0u, heap_id);
+  void* alloc = heap_manager_->Allocate(heap_id, kAllocSize);
+  EXPECT_EQ(static_cast<void*>(nullptr), alloc);
+  EXPECT_TRUE(heap_manager_->DestroyHeap(heap_id));
+}
+
 TEST_F(BlockHeapManagerTest, FreeNullPointer) {
   HeapId heap_id = heap_manager_->CreateHeap();
   EXPECT_NE(0u, heap_id);
   EXPECT_TRUE(heap_manager_->Free(heap_id, static_cast<void*>(nullptr)));
   EXPECT_TRUE(heap_manager_->DestroyHeap(heap_id));
 }
 
 TEST_F(BlockHeapManagerTest, FreeUnguardedAlloc) {
   const size_t kAllocSize = 100;
   ::common::AsanParameters params = heap_manager_->parameters();
@@ skipping 1272 matching lines @@
       bi.header->alloc_stack->relative_stack_id());
 
   // Clear the heap ID and delete the block, expecting this to succeed.
   bi.trailer->heap_id = 0;
   EXPECT_TRUE(heap_manager_->Free(wh, alloc));
 }
 
 }  // namespace heap_managers
 }  // namespace asan
 }  // namespace agent