#ifndef lint
static char rcsid[] = "$Header$";
#endif

/*
 * Memory manager. Memory is divided into NMEMS pieces. There is a struct
 * for each piece telling where it is, how many bytes are used, and how may
 * are left. If a request for core doesn't fit in the left bytes, an sbrk()
 * is done and pieces after the one that requested the growth are moved up.
 */

#include <stdio.h>
#include "out.h"
#include "const.h"
#include "assert.h"
#include "debug.h"
#include "memory.h"

struct memory	mems[NMEMS];

bool	incore = TRUE;	/* TRUE while everything can be kept in core. */
off_t	core_position = (off_t)0;	/* Index of current module. */

#define AT_LEAST	2	/* See comment about string areas. */

/*
 * Initialize some pieces of core. We hope that this will be our last
 * real allocation, meaning we've made the right choices.
 */
init_core()
{
	FILE			*ledrc;
	int			piece;
	off_t			left;
	register char		*base;
	register off_t		total_size;
	register struct memory	*mem;
	extern char		*sbrk();

#ifndef TJALK
	/*
	 * Read in what should be allocated for each piece initially.
	 * This facilitates testing, but is slower and should not
	 * be done in the final version. XXX
	 */
	incore = (ledrc = fopen(".ledrc", "r")) != (FILE *)0;

	if (incore) {
		while (fscanf(ledrc, "%d %d", &piece, &left) == 2)
			mems[piece].mem_left = left;
		fclose(ledrc);
	}
#else TJALK
	mems[ALLOHEAD].mem_left = 20;	/*XXX*/
	mems[ALLOSECT].mem_left = 60;	/*XXX*/
	mems[ALLOEMIT + 0].mem_left = 65536;
	mems[ALLOEMIT + 1].mem_left = 65536;
	mems[ALLORELO].mem_left = 65536;
	mems[ALLOLOCL].mem_left = 65536;
	mems[ALLOGLOB].mem_left = 65536;
	mems[ALLOLCHR].mem_left = 65536;
	mems[ALLOGCHR].mem_left = 65536;
#ifdef SYMDBUG
	mems[ALLODBUG].mem_left = 65536;
#endif SYMDBUG
	mems[ALLOSYMB].mem_left = 4096;
	mems[ALLOARCH].mem_left = 512;
	mems[ALLOMODL].mem_left = 196608;
	mems[ALLORANL].mem_left = 4096;
#endif TJALK

	total_size = (off_t)0;/* Will accumulate the sizes. */
	base = sbrk(0);		/* First free. */
	for (mem = mems; mem < &mems[NMEMS]; mem++) {
		mem->mem_base = base;
		mem->mem_full = (off_t)0;
		base += mem->mem_left;	/* Each piece will start after prev. */
		total_size += mem->mem_left;
	}
	/*
	 * String areas are special-cased. The first byte is unused as a way to
	 * distinguish a name without string from a name which has the first
	 * string in the string area.
	 */
	if (mems[ALLOLCHR].mem_left == 0)
		total_size += 1;
	else
		mems[ALLOLCHR].mem_left -= 1;
	if (mems[ALLOGCHR].mem_left ==  0)
		total_size += 1;
	else
		mems[ALLOGCHR].mem_left -= 1;
	mems[ALLOLCHR].mem_full = 1;
	mems[ALLOGCHR].mem_full = 1;

	if ((int)sbrk(total_size) == -1) {
		incore = FALSE;	/* In core strategy failed. */
		if ((int)sbrk(AT_LEAST) == -1)
			fatal("no core at all");
	}

}

/*
 * Allocate an extra block of `incr' bytes and move all pieces with index
 * higher than `piece' up with the size of the block. Return whether the
 * allocate succeeded.
 */
static bool
move_up(piece, incr)
	register int		piece;
	register off_t		incr;
{
	register struct memory	*mem;
	extern char		*sbrk();

	debug("move_up(%d, %d)\n", piece, (int)incr, 0, 0);
	if ((int)sbrk(incr) == -1)
		return FALSE;

	for (mem = &mems[NMEMS - 1]; mem > &mems[piece]; mem--)
		copy_up(mem, incr);

	mems[piece].mem_left += incr;
	return TRUE;
}

extern int	passnumber;

/*
 * This routine is called if `piece' needs `incr' bytes and the system won't
 * give them. We first steal the free bytes of all lower pieces and move them
 * and `piece' down. If that doesn't give us enough bytes, we steal the free
 * bytes of all higher pieces and move them up. We return whether we have
 * enough bytes, the first or the second time.
 */
static bool
compact(piece, incr)
	register int		piece;
	register off_t		incr;
{
	register off_t		gain;
	register struct memory	*mem;

	debug("compact(%d, %d)\n", piece, (int)incr, 0, 0);
	gain = mems[0].mem_left;
	mems[0].mem_left = (off_t)0;
	for (mem = &mems[1]; mem <= &mems[piece]; mem++) {
		/* Here memory is inserted before a piece. */
		assert(passnumber == FIRST || gain == (off_t)0);
		copy_down(mem, gain);
		gain += mem->mem_left;
		mem->mem_left = (off_t)0;
	}
	/*
	 * Note that we already added the left bytes of the piece we want to
	 * enlarge to `gain'.
	 */
	if (gain < incr) {
		register off_t	up = (off_t)0;

		for (mem = &mems[NMEMS - 1]; mem > &mems[piece]; mem--) {
			/* Here memory is appended after a piece. */
			up += mem->mem_left;
			copy_up(mem, up);
			mem->mem_left = (off_t)0;
		}
		gain += up;
	}
	mems[piece].mem_left = gain;
	return gain >= incr;
}

/*
 * The bytes of `mem' must be moved `dist' down in the address space.
 * We copy the bytes from low to high, because the tail of the new area may
 * overlap with the old area, but we do not want to overwrite them before they
 * are copied.
 */
static
copy_down(mem, dist)
	register struct memory	*mem;
	off_t			dist;
{
	register char		*old;
	register char		*new;
	register off_t		size;

	size = mem->mem_full;
	old = mem->mem_base;
	new = old - dist;
	mem->mem_base = new;
	while (size--)
		*new++ = *old++;
}

/*
 * The bytes of `mem' must be moved `dist' up in the address space.
 * We copy the bytes from high to low, because the tail of the new area may
 * overlap with the old area, but we do not want to overwrite them before they
 * are copied.
 */
static
copy_up(mem, dist)
	register struct memory	*mem;
	off_t			dist;
{
	register char		*old;
	register char		*new;
	register off_t		size;

	size = mem->mem_full;
	old = mem->mem_base + size;
	new = old + dist;
	while (size--)
		*--new = *--old;
	mem->mem_base = new;
}

/*
 * Add `size' bytes to the bytes already allocated for `piece'. If it has no
 * free bytes left, ask them from memory or, if that fails, from the free
 * bytes of other pieces. The offset of the new area is returned. No matter
 * how many times the area is moved, because of another allocate, this offset
 * remains valid.
 */
off_t
alloc(piece, size)
	register int		piece;
	register off_t		size;
{
	register off_t		incr = 0;
	register off_t		left = mems[piece].mem_left;
	register off_t		full = mems[piece].mem_full;

	assert(passnumber == FIRST || (!incore && piece == ALLOMODL));
	if (size == (off_t)0)
		return full;

	while (left + incr < size)
		incr += INCRSIZE;

	if (incr == 0 || move_up(piece, incr) || compact(piece, incr)) {
		mems[piece].mem_full += size;
		mems[piece].mem_left -= size;
		return full;
	} else {
		incore = FALSE;
		return BADOFF;
	}
}

/*
 * Same as alloc() but for a piece which really needs it. If the first
 * attempt fails, release the space occupied by other pieces and try again.
 */
off_t
hard_alloc(piece, size)
	int		piece;
	off_t		size;
{
	off_t		ret;
	register int	i;

	if ((ret = alloc(piece, size)) != BADOFF)
		return ret;

	/*
	 * Deallocate what we don't need.
	 */
	for (i = 0; i < NMEMS; i++) {
		switch (i) {
		case ALLOHEAD:
		case ALLOSECT:
		case ALLOGLOB:
		case ALLOGCHR:
		case ALLOSYMB:
		case ALLOARCH:
		case ALLOMODL:
			break;	/* Do not try to deallocate this. */
		default:
			dealloc(i);
			break;
		}
	}
	free_saved_moduls();

	return alloc(piece, size);
}

/*
 * We don't need the previous modules, so we put the current module
 * at the start of the piece allocated for module contents, thereby
 * overwriting the saved modules, and release its space.
 */
static
free_saved_moduls()
{
	register off_t		size;
	register char		*old, *new;
	register struct memory	*mem = &mems[ALLOMODL];

	size = mem->mem_full - core_position;
	new = mem->mem_base;
	old = new + core_position;
	while (size--)
		*new++ = *old++;
	mem->mem_full -= core_position;
	mem->mem_left += core_position;
	core_position = (off_t)0;
}

/*
 * The piece of memory with index `piece' is no longer needed.
 * We take care that it can be used by compact() later, if needed.
 */
dealloc(piece)
	register int		piece;
{
	/*
	 * Some pieces need their memory throughout the program.
	 */
	assert(piece != ALLOHEAD);
	assert(piece != ALLOSECT);
	assert(piece != ALLOGLOB);
	assert(piece != ALLOGCHR);
	assert(piece != ALLOSYMB);
	assert(piece != ALLOARCH);
	mems[piece].mem_left += mems[piece].mem_full;
	mems[piece].mem_full = (off_t)0;
}

char *
core_alloc(piece, size)
	register int	piece;
	register off_t	size;
{
	register off_t	off;

	if ((off = alloc(piece, size)) == BADOFF)
		return (char *)0;
	return address(piece, off);
}

/*
 * Reset index into piece of memory for modules and
 * take care that the allocated pieces will not be moved.
 */
freeze_core()
{
	register int	i;

	core_position = (off_t)0;

	if (incore)
		return;

	for (i = 0; i < NMEMS; i++) {
		switch (i) {
		case ALLOHEAD:
		case ALLOSECT:
		case ALLOGLOB:
		case ALLOGCHR:
		case ALLOSYMB:
		case ALLOARCH:
			break;	/* Do not try to deallocate this. */
		default:
			dealloc(i);
			break;
		}
	}
	compact(NMEMS - 1, (off_t)0);
}

/* ------------------------------------------------------------------------- */

extern bool	bytes_reversed;
extern bool	words_reversed;

/*
 * To transform the various pieces of the output in core to the file format,
 * we must order the bytes in the ushorts and longs as ACK prescribes.
 */
write_bytes()
{
	register struct outhead	*head;
	ushort		nsect, nrelo;
	long		offchar;
	int		fd;
	register int	piece;
	extern ushort	NLocals, NGlobals;
	extern long	NLChars, NGChars;
	extern int	flagword;
	extern char	*outputname;

	head = (struct outhead *)mems[ALLOHEAD].mem_base;
	nsect = head->oh_nsect;
	nrelo = head->oh_nrelo;
	offchar = OFF_CHAR(*head);

	if (bytes_reversed || words_reversed) {
		headswap();
		sectswap(nsect);
		reloswap(nrelo);
	}
	/*
	 * We allocated two areas: one for local and one for global names.
	 * Also, we used another kind of on_foff than on file.
	 * At the end of the global area we have put the section names.
	 */
	if (!(flagword & SFLAG)) {
		namecpy((struct outname *)mems[ALLOLOCL].mem_base,
			NLocals,
			offchar
		);
		namecpy((struct outname *)mems[ALLOGLOB].mem_base,
			NGlobals + nsect,
			offchar + NLChars
		);
	}
	if ((fd = creat(outputname, 0666)) < 0)
		fatal("can't create %s", outputname);
	/*
	 * These pieces must always be written.
	 */
	for (piece = ALLOHEAD; piece < ALLORELO; piece++)
		writelong(fd, mems[piece].mem_base, mems[piece].mem_full);
	/*
	 * The rest depends on the flags.
	 */
	if (flagword & RFLAG)
		writelong(fd, mems[ALLORELO].mem_base, mems[ALLORELO].mem_full);
	if (!(flagword & SFLAG)) {
		writelong(fd, mems[ALLOLOCL].mem_base, mems[ALLOLOCL].mem_full);
		writelong(fd, mems[ALLOGLOB].mem_base, mems[ALLOGLOB].mem_full);
		writelong(fd, mems[ALLOLCHR].mem_base + 1, NLChars);
		writelong(fd, mems[ALLOGCHR].mem_base + 1, NGChars);
#ifdef SYMDBUG
		writelong(fd, mems[ALLODBUG].mem_base, mems[ALLODBUG].mem_size);
#endif SYMDBUG
	}
	close(fd);
}

static
writelong(fd, base, size)
	register int	fd;
	register char	*base;
	register off_t	size;
{
	register int	chunk;

	while (size) {
		chunk = size > (off_t)MAXCHUNK ? MAXCHUNK : size;
		write(fd, base, chunk);
		size -= chunk;
		base += chunk;
	}
}

static
headswap()
{
	register struct outhead	*head;

	head = (struct outhead *)mems[ALLOHEAD].mem_base;
	swap((char *)head, SF_HEAD);
}

static
sectswap(nsect)
	register ushort		nsect;
{
	register struct outsect	*sect;

	sect = (struct outsect *)mems[ALLOSECT].mem_base;
	while (nsect--)
		swap((char *)sect++, SF_SECT);
}

static
reloswap(nrelo)
	register ushort		nrelo;
{
	register struct outrelo	*relo;

	relo = (struct outrelo *)mems[ALLORELO].mem_base;
	while (nrelo--)
		swap((char *)relo++, SF_RELO);
}

static
namecpy(name, nname, offchar)
	register struct outname	*name;
	register ushort		nname;
	register long		offchar;
{
	while (nname--) {
		if (name->on_foff)
			name->on_foff += offchar - 1;
		if (bytes_reversed || words_reversed)
			swap((char *)name, SF_NAME);
		name++;
	}
}