1 /****************************************************************************** 2 * 3 * Copyright 2014 Google, Inc. 4 * 5 * Licensed under the Apache License, Version 2.0 (the "License"); 6 * you may not use this file except in compliance with the License. 7 * You may obtain a copy of the License at: 8 * 9 * http://www.apache.org/licenses/LICENSE-2.0 10 * 11 * Unless required by applicable law or agreed to in writing, software 12 * distributed under the License is distributed on an "AS IS" BASIS, 13 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 14 * See the License for the specific language governing permissions and 15 * limitations under the License. 16 * 17 ******************************************************************************/ 18 19 #include <gtest/gtest.h> 20 21 #include "osi/include/allocation_tracker.h" 22 23 void allocation_tracker_uninit(void); 24 25 static const allocator_id_t allocator_id = 5; 26 27 TEST(AllocationTrackerTest, test_uninit_no_bad_effects) { 28 void* dummy_allocation = malloc(4); 29 30 // Ensure uninitialized state (previous tests may have called init) 31 allocation_tracker_uninit(); 32 33 EXPECT_EQ(4U, allocation_tracker_resize_for_canary(4)); 34 allocation_tracker_notify_alloc(allocator_id, dummy_allocation, 4); 35 EXPECT_EQ(0U, allocation_tracker_expect_no_allocations()); // should not have 36 // registered an 37 // allocation 38 allocation_tracker_notify_free(allocator_id, dummy_allocation); 39 EXPECT_EQ(0U, allocation_tracker_expect_no_allocations()); 40 41 free(dummy_allocation); 42 } 43 44 TEST(AllocationTrackerTest, test_canaries_on) { 45 allocation_tracker_uninit(); 46 allocation_tracker_init(); 47 48 size_t with_canary_size = allocation_tracker_resize_for_canary(4); 49 EXPECT_TRUE(with_canary_size > 4); 50 51 void* dummy_allocation = malloc(with_canary_size); 52 void* useable_ptr = 53 allocation_tracker_notify_alloc(allocator_id, dummy_allocation, 4); 54 EXPECT_TRUE(useable_ptr > dummy_allocation); 55 EXPECT_EQ(4U, allocation_tracker_expect_no_allocations()); // should have 56 // registered the 57 // allocation 58 void* freeable_ptr = 59 allocation_tracker_notify_free(allocator_id, useable_ptr); 60 EXPECT_EQ(dummy_allocation, freeable_ptr); 61 EXPECT_EQ(0U, allocation_tracker_expect_no_allocations()); 62 63 free(dummy_allocation); 64 } 65
