You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
64 lines
2.5 KiB
64 lines
2.5 KiB
/****************************************************************************** |
|
* |
|
* Copyright (C) 2014 Google, Inc. |
|
* |
|
* 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, |
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
|
* See the License for the specific language governing permissions and |
|
* limitations under the License. |
|
* |
|
******************************************************************************/ |
|
|
|
#include <gtest/gtest.h> |
|
|
|
#include "osi/include/allocation_tracker.h" |
|
|
|
void allocation_tracker_uninit(void); |
|
|
|
static const allocator_id_t allocator_id = 5; |
|
|
|
TEST(AllocationTrackerTest, test_uninit_no_bad_effects) { |
|
void* dummy_allocation = malloc(4); |
|
|
|
// Ensure uninitialized state (previous tests may have called init) |
|
allocation_tracker_uninit(); |
|
|
|
EXPECT_EQ(4U, allocation_tracker_resize_for_canary(4)); |
|
allocation_tracker_notify_alloc(allocator_id, dummy_allocation, 4); |
|
EXPECT_EQ(0U, allocation_tracker_expect_no_allocations()); // should not have |
|
// registered an |
|
// allocation |
|
allocation_tracker_notify_free(allocator_id, dummy_allocation); |
|
EXPECT_EQ(0U, allocation_tracker_expect_no_allocations()); |
|
|
|
free(dummy_allocation); |
|
} |
|
|
|
TEST(AllocationTrackerTest, test_canaries_on) { |
|
allocation_tracker_uninit(); |
|
allocation_tracker_init(); |
|
|
|
size_t with_canary_size = allocation_tracker_resize_for_canary(4); |
|
EXPECT_TRUE(with_canary_size > 4); |
|
|
|
void* dummy_allocation = malloc(with_canary_size); |
|
void* useable_ptr = |
|
allocation_tracker_notify_alloc(allocator_id, dummy_allocation, 4); |
|
EXPECT_TRUE(useable_ptr > dummy_allocation); |
|
EXPECT_EQ(4U, allocation_tracker_expect_no_allocations()); // should have |
|
// registered the |
|
// allocation |
|
void* freeable_ptr = |
|
allocation_tracker_notify_free(allocator_id, useable_ptr); |
|
EXPECT_EQ(dummy_allocation, freeable_ptr); |
|
EXPECT_EQ(0U, allocation_tracker_expect_no_allocations()); |
|
|
|
free(dummy_allocation); |
|
}
|
|
|