/*
* Copyright (c) 2000-2024 Apple Inc. All rights reserved.
*
* @APPLE_OSREFERENCE_LICENSE_HEADER_START@
*
* This file contains Original Code and/or Modifications of Original Code
* as defined in and that are subject to the Apple Public Source License
* Version 2.0 (the 'License'). You may not use this file except in
* compliance with the License. The rights granted to you under the License
* may not be used to create, or enable the creation or redistribution of,
* unlawful or unlicensed copies of an Apple operating system, or to
* circumvent, violate, or enable the circumvention or violation of, any
* terms of an Apple operating system software license agreement.
*
* Please obtain a copy of the License at
* http://www.opensource.apple.com/apsl/ and read it before using this file.
*
* The Original Code and all software distributed under the License are
* distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
* EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
* INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
* Please see the License for the specific language governing rights and
* limitations under the License.
*
* @APPLE_OSREFERENCE_LICENSE_HEADER_END@
*/
#include "std_safe.h"
#include "dt_proxy.h"
#include "unit_test_utils.h"
/* This is an implementation of simple fixed size same-size objects pool */
struct mock_mem_pool {
size_t elem_size;
char *buffer;
char *free_head;
uint32_t free_count;
};
void
mock_mem_init(struct mock_mem_pool* mm, size_t elem_sz, uint32_t count)
{
mm->elem_size = elem_sz;
size_t buf_size = elem_sz * count;
mm->buffer = aligned_alloc(8, buf_size);
PT_QUIET; PT_ASSERT_NOTNULL(mm->buffer, "failed alloc");
memset(mm->buffer, 0, buf_size);
mm->free_head = mm->buffer;
mm->free_count = count;
}
void *
mock_mem_alloc(struct mock_mem_pool* mm)
{
PT_QUIET; PT_ASSERT_NOTNULL(mm->buffer, "mock mem not allocated");
PT_QUIET; PT_ASSERT_TRUE(mm->free_count > 0, "no more space left");
void *ret = mm->free_head;
mm->free_head += mm->elem_size;
mm->free_count--;
return ret;
}
void
mock_mem_free(struct mock_mem_pool* mm, void *ptr)
{
// not implemeted yet rdar://136915968
}
struct mock_mem_pool mm_vm_objects;
// this is used for vm_object and vm_page pointer packing
uintptr_t mock_page_ptr_base;
void
mock_mem_init_vm_objects(void)
{
mock_mem_init(&mm_vm_objects, 256, 100);
mock_page_ptr_base = (uintptr_t)mm_vm_objects.buffer;
}
void *
mock_mem_alloc_vm_object(void)
{
return mock_mem_alloc(&mm_vm_objects);
}