/* $XFree86: xc/programs/Xserver/hw/xfree86/loader/os2funcs.c,v 1.7 2003/03/25 04:18:22 dawes Exp $ */
/*
 * (c) Copyright 1997 by Sebastien Marineau
 *                      <marineau@genie.uottawa.ca>
 *
 * Permission is hereby granted, free of charge, to any person obtaining a
 * copy of this software and associated documentation files (the "Software"),
 * to deal in the Software without restriction, including without limitation
 * the rights to use, copy, modify, merge, publish, distribute, sublicense,
 * and/or sell copies of the Software, and to permit persons to whom the
 * Software is furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
 * SEBASTIEN MARINEAU BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
 * WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF
 * OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 * SOFTWARE.
 *
 * Except as contained in this notice, the name of Sebastien Marineau shall not be
 * used in advertising or otherwise to promote the sale, use or other dealings
 * in this Software without prior written authorization from Sebastien Marineau.
 *
 */


/* Implements some OS/2 memory allocation functions to allow 
 * execute permissions for modules. We allocate some mem using DosAllocMem
 * and then use the EMX functions to create a heap from which we allocate
 * the requests. We create a heap of 2 megs, hopefully enough for now.
 */

#define I_NEED_OS2_H
#define INCL_DOSMEMMGR
#include <os2.h>
#include <sys/types.h>
#include <umalloc.h>
#include "os.h"
#undef ADDRESS
#include "xf86str.h"

#define RESERVED_BLOCKS 4096  /* reserve 16MB memory for modules */

void *os2ldAddToHeap(Heap_t, size_t *, int *);
void os2ldRemoveFromHeap(Heap_t, void *, size_t);

PVOID os2ldCommitedTop;
PVOID os2ldBase;
Heap_t os2ldHeap;
int os2ldTotalCommitedBlocks;
static BOOL FirstTime = TRUE;

void *os2ldcalloc(size_t num_elem, size_t size_elem) {
	APIRET rc;
	int ret;

	if (FirstTime) {
		if ((rc=DosAllocMem(&os2ldBase,RESERVED_BLOCKS * 4096, 
				PAG_READ | PAG_WRITE | PAG_EXECUTE)) != 0) {
			xf86Msg(X_ERROR,
				"OS2LD: DosAllocMem failed, rc=%d\n",
				rc);
			return NULL;
		}

		/* Now commit the first 128Kb, the rest will 
		 * be done dynamically */
		if ((rc=DosSetMem(os2ldBase,
				  32*4096,
				  PAG_DEFAULT | PAG_COMMIT)) != 0) {
			xf86Msg(X_ERROR,
				"OS2LD: DosSetMem failed, rc=%d\n",rc);
			DosFreeMem(os2ldBase);
			return NULL;
		}
	        os2ldCommitedTop = os2ldBase + 32*4096;
		os2ldTotalCommitedBlocks = 32;
#ifdef DEBUG
		xf86Msg(X_INFO,
			"OS2LD: Initial heap at addr=%p\n",
			os2ldBase);
#endif

		if ((os2ldHeap=_ucreate(os2ldBase,
					32*4096, _BLOCK_CLEAN,
					_HEAP_REGULAR, os2ldAddToHeap, 
					os2ldRemoveFromHeap)) == NULL) {
			xf86Msg(X_ERROR,
				"OS2LD: heap creation failed\n");
			DosFreeMem(os2ldBase);
			return NULL;
		}
	
		if ((ret=_uopen(os2ldHeap)) != 0) {
			xf86Msg(X_ERROR,
				"OS2LD: heap open failed\n");
			ret = _udestroy(os2ldHeap,_FORCE);
			DosFreeMem(os2ldBase);
			return(NULL);
		}

		FirstTime = FALSE;

#ifdef DEBUG
		xf86Msg(X_INFO,"OS2LD: Created module heap at addr=%p\n",
			os2ldHeap);
#endif
	}

	return _ucalloc(os2ldHeap,num_elem,size_elem);
}

void *os2ldAddToHeap(Heap_t H, size_t *new_size, int *PCLEAN)
{
	PVOID NewBase;
	long adjusted_size;
	long blocks;
	APIRET rc;

	if (H != os2ldHeap) {
		xf86Msg(X_ERROR,
			"OS2LD: Heap corruption in GrowHeap, p=%08x\n",H);
		return NULL;
	}
	NewBase = os2ldCommitedTop;
	adjusted_size = (*new_size/65536) * 65536;
	if ((*new_size % 65536) > 0) 
		adjusted_size += 65536;
	blocks = adjusted_size / 4096;

	if ((os2ldTotalCommitedBlocks + blocks) > RESERVED_BLOCKS) {
		xf86Msg(X_ERROR,
			"OS2LD: Out of memory in GrowHeap\n");
		xf86Msg(X_ERROR,
			"OS2LD: Max available memory is of %ld bytes\n",
			RESERVED_BLOCKS * 4096);
		return NULL;
	}
	if ((rc=DosSetMem(NewBase, adjusted_size, PAG_DEFAULT | PAG_COMMIT)) != 0) {
		xf86Msg(X_ERROR,
			"OS2LD: DosSetMem failed in GrowHeap, size req'd=%d, rc=%d\n",
			adjusted_size,
			rc);
		return NULL;
	}

	os2ldCommitedTop += adjusted_size;
	os2ldTotalCommitedBlocks += blocks;
	*PCLEAN = _BLOCK_CLEAN;
	*new_size = adjusted_size;
#ifdef DEBUG
	xf86Msg(X_INFO,"OS2LD: Heap extended by %d bytes, addr=%p\n",
		adjusted_size, NewBase);
#endif
	return NewBase;
}

void os2ldRemoveFromHeap(Heap_t H, void *memory, size_t size)
{
	if (H != os2ldHeap) {
		xf86Msg(X_ERROR,
			"OS2LD: Heap corruption in ShrinkHeap, p=%08x\n",H);
		return;
	}

	/* Currently we do nothing, as we do not keep track of the
	 * commited memory */


	/* Only handle it if it is the base address */
	if (memory == os2ldBase) {
		DosFreeMem(os2ldBase);
#ifdef DEBUG
		xf86Msg(X_INFO,"OS2LD: total heap area deallocated\n");
#endif
		os2ldBase = 0;
		FirstTime = TRUE;
	}
}
