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.
65 lines
2.5 KiB
65 lines
2.5 KiB
/******************************************************************************
|
|
*
|
|
* Copyright 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);
|
|
}
|