/*
 * Copyright (c) 1998-2000 Apple Computer, 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@
 */
#define IOKIT_ENABLE_SHARED_PTR
#include <IOKit/IOGuardPageMemoryDescriptor.h>
#include <IOKit/IOMemoryDescriptor.h>
#include <IOKit/IOLib.h>
#include <vm/vm_kern_xnu.h>
#include <mach/mach_vm.h>

#define super IOGeneralMemoryDescriptor

OSDefineMetaClassAndStructorsWithZone(IOGuardPageMemoryDescriptor, IOGeneralMemoryDescriptor, ZC_ZFREE_CLEARMEM);

OSSharedPtr<IOGuardPageMemoryDescriptor>
IOGuardPageMemoryDescriptor::withSize(vm_size_t size)
{
	OSSharedPtr<IOGuardPageMemoryDescriptor> me = OSMakeShared<IOGuardPageMemoryDescriptor>();

	if (me && !me->initWithSize(size)) {
		me.reset();
	}
	return me;
}

bool
IOGuardPageMemoryDescriptor::initWithSize(vm_size_t size)
{
	mach_vm_address_t address;
	kern_return_t kr;
	IOOptionBits  iomdOptions = kIOMemoryTypeVirtual64 | kIOMemoryAsReference | kIODirectionOutIn;

	size = round_page(size);

	_ranges.v64 = IOMallocType(IOAddressRange);
	if (!_ranges.v64) {
		return false;
	}

	kr = mach_vm_allocate_kernel(kernel_map, &address, size,
	    VM_MAP_KERNEL_FLAGS_ANYWHERE(.vm_tag = VM_KERN_MEMORY_IOKIT));
	if (kr != KERN_SUCCESS) {
		return false;
	}


	_ranges.v64->address = address;
	_ranges.v64->length  = size;

	if (!super::initWithOptions(_ranges.v64, 1, 0, kernel_task, iomdOptions, NULL)) {
		return false;
	}

	_size = size;
	_buffer = (vm_offset_t)address;

	return true;
}

void
IOGuardPageMemoryDescriptor::free()
{
	if (_buffer) {
		vm_deallocate(kernel_map, _buffer, _size);
		_buffer = 0;
	}

	if (_ranges.v64) {
		IOFreeType(_ranges.v64, IOAddressRange);
	}

	super::free();
}

IOReturn
IOGuardPageMemoryDescriptor::doMap(vm_map_t           addressMap,
    IOVirtualAddress * atAddress,
    IOOptionBits       options,
    IOByteCount        sourceOffset,
    IOByteCount        length)
{
	IOReturn ret = super::doMap(addressMap, atAddress, options, sourceOffset, length);
	if (ret == kIOReturnSuccess) {
		IOMemoryMap *     mapping = (IOMemoryMap *) *atAddress;
		vm_map_t          map     = mapping->fAddressMap;
		mach_vm_size_t    length  = mapping->fLength;
		mach_vm_address_t address = mapping->fAddress;
		kern_return_t kr = mach_vm_protect(map, address, length, true, VM_PROT_NONE);
		if (kr != KERN_SUCCESS) {
			doUnmap(map, (IOVirtualAddress) mapping, 0);
			return kIOReturnError;
		}
	}
	return ret;
}