/*
 * Copyright 2025, Haiku, Inc. All rights reserved.
 * Distributed under the terms of the MIT License.
 */

#include <OS.h>
#include <stdio.h>
#include <syscalls.h>


int
reservation_follows_resize_test()
{
	size_t reserveSize = 1 * 1024 * 1024;
	addr_t newAreaBase;
	status_t status = _kern_reserve_address_range(&newAreaBase,
		B_RANDOMIZED_ANY_ADDRESS, reserveSize);
	if (status < B_OK) {
		printf("reserve_address_range 1 unexpectedly failed!\n");
		return status;
	}

	area_id area = create_area("test area", (void**)&newAreaBase,
		B_EXACT_ADDRESS, B_PAGE_SIZE, B_NO_LOCK, B_READ_AREA | B_WRITE_AREA);
	if (area < B_OK) {
		printf("create_area 1 unexpectedly failed!\n");
		return area;
	}

	// Resize the area larger (shriking the reservation)
	status = resize_area(area, B_PAGE_SIZE * 2);
	if (status < B_OK)
		return status;

	// Make sure all is as expected
	addr_t past = (newAreaBase + B_PAGE_SIZE * 2);
	status = _kern_reserve_address_range(&past, B_EXACT_ADDRESS, B_PAGE_SIZE);
	if (status == B_OK) {
		printf("reserve_address_range 2 unexpectedly succeeded!\n");
		return status;
	}

	status = area_for((void*)(newAreaBase + B_PAGE_SIZE * 1));
	if (status != area)
		return status;

	// Now resize the area smaller again
	status = resize_area(area, B_PAGE_SIZE * 1);
	if (status < B_OK)
		return status;

	// And check that the reservation resized with it
	past = (newAreaBase + B_PAGE_SIZE * 1);
	status = _kern_reserve_address_range(&past, B_EXACT_ADDRESS, B_PAGE_SIZE);
	if (status == B_OK) {
		printf("reservation did not resize downwards!\n");
		return status;
	}

	// And check that the reservation still extends to where we expect
	past = (newAreaBase + reserveSize - B_PAGE_SIZE);
	status = _kern_reserve_address_range(&past, B_EXACT_ADDRESS, B_PAGE_SIZE);
	if (status == B_OK) {
		printf("reservation has wrong size!\n");
		return status;
	}

	past = (newAreaBase + reserveSize);
	status = _kern_reserve_address_range(&past, B_EXACT_ADDRESS, B_PAGE_SIZE);
	if (status != B_OK) {
		printf("reservation has wrong size!\n");
		return status;
	}

	delete_area(area);
	return _kern_unreserve_address_range(newAreaBase, reserveSize);
}


int
resize_past_reservation_test()
{
	size_t reserveSize = 4 * B_PAGE_SIZE;
	addr_t newAreaBase;
	status_t status = _kern_reserve_address_range(&newAreaBase,
		B_RANDOMIZED_ANY_ADDRESS, reserveSize * 2);
	if (status < B_OK) {
		printf("reserve_address_range 2 unexpectedly failed!\n");
		return status;
	}

	// Shrink the reservation partially
	status = _kern_unreserve_address_range(newAreaBase + reserveSize, reserveSize);
	if (status < B_OK)
		return status;

	// Make sure it's gone
	status = area_for((void*)(newAreaBase + reserveSize));
	if (status != B_ERROR)
		return -1;

	area_id area = create_area("test area", (void**)&newAreaBase,
		B_EXACT_ADDRESS, B_PAGE_SIZE, B_NO_LOCK, B_READ_AREA | B_WRITE_AREA);
	if (area < B_OK) {
		printf("create_area 2 unexpectedly failed!\n");
		return area;
	}

	// Resize the area larger, past the reservation entirely
	status = resize_area(area, reserveSize * 2);
	if (status < B_OK) {
		printf("failed to resize area past first reservation!\n");
		return status;
	}

	// Shrink again, put two new reservations there
	status = resize_area(area, reserveSize);
	if (status < B_OK)
		return status;

	addr_t reserve = newAreaBase + reserveSize + B_PAGE_SIZE;
	status = _kern_reserve_address_range(&reserve, B_EXACT_ADDRESS, B_PAGE_SIZE);
	if (status < B_OK)
		return status;

	reserve += B_PAGE_SIZE;
	status = _kern_reserve_address_range(&reserve, B_EXACT_ADDRESS, B_PAGE_SIZE);
	if (status < B_OK)
		return status;

	// Resize the area larger again, past both reservations
	status = resize_area(area, reserveSize * 2);
	if (status < B_OK) {
		printf("failed to resize area past second and third reservations!\n");
		return status;
	}

	return 0;
}


int
main()
{
	status_t status;

	if ((status = reservation_follows_resize_test()) != 0)
		return status;

	if ((status = resize_past_reservation_test()) != 0)
		return status;

	return 0;
}