switch-coreboot/util/lar/stream.c

/*
 * lar - lightweight archiver
 *
 * This includes code from previous versions of the LAR utility,
 * including create.c, list.c, extract.c and bootblock.c
 *
 * Copyright (C) 2006-2007 coresystems GmbH
 * (Written by Stefan Reinauer <stepan@coresystems.de> for coresystems GmbH)
 * Copyright (C) 2007 Patrick Georgi <patrick@georgi-clan.de>
 * Copyright (C) 2007 Advanced Micro Devices, Inc.
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; version 2 of the License.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA, 02110-1301 USA
 */

#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include <fcntl.h>
#include <sys/stat.h>
#include <sys/mman.h>
#include <netinet/in.h>
#include <libgen.h>

#include "elf.h"
#include "lar.h"
#include "lib.h"

/**
 * \def err(fmt,args...)
 * Abstract outputing error strings to avoid repetition
 */
#define err(fmt,args...) fprintf(stderr, fmt, ##args)

extern enum compalgo algo;

/* ELF processing */
/**
 * Given a ptr to data, determine if the data is an ELF image. 
 * @param filebuf pointer to the data
 * @return 1 if ELF, 0 if not
 */
int iself(char *filebuf)
{
	Elf32_Ehdr *ehdr;
	/* validate elf header */
	ehdr = (Elf32_Ehdr *)filebuf;
	if (memcmp(ehdr->e_ident, ELFMAG, 4))
		return 0;
	return 1;
}

/**
 * Output all the ELF segments for a given file
 * @param lar The LAR archive
 * @param name The LAR name
 * @param filebuf The ELF file
 * @param filelen Size of the ELF file
 * @param algo The recommend compression algorithm
 * Return 0 on success, -1 on failure
 */
int output_elf_segments(struct lar *lar, char *name, char *filebuf,
			int filelen, enum compalgo algo)
{
	int ret;
	Elf32_Phdr *phdr;
	Elf32_Ehdr *ehdr;
	Elf32_Shdr *shdr;
	u32 entry;
	int i;
	int size;
	int segment = 0;
	char *header;
	char ename[MAX_PATHLEN];
	int headers;
	char *temp;
	enum compalgo thisalgo;
	u32 complen;

	/* Allocate a temporary buffer to compress into - this is unavoidable,
	 * because we need to make sure that the compressed data will fit in
	 * the LAR, and we won't know the size of the compressed data until
	 * we actually compress it.
	 * FIXME: In case the compressed file is bigger than the original,
	 * we corrupt memory. Compute maximum size increase and allocate that
	 * on top of the file length.
	 */

	temp = calloc(filelen, 1);

	if (temp == NULL) {
		err("Out of memory.\n");
		return -1;
	}

	/* validate elf header */
	ehdr = (Elf32_Ehdr *)filebuf;
	headers = ehdr->e_phnum;
	header  = (char *)ehdr;
	if (verbose())
		fprintf(stderr, "Type %d machine %d version %d entry %p phoff %d shoff %d flags %#x hsize %d phentsize %d phnum %d s_hentsize %d s_shnum %d \n", 
		ehdr->e_type,
		ehdr->e_machine,
		ehdr->e_version,
		(void *)(unsigned long)ehdr->e_entry,
		ehdr->e_phoff,
		ehdr->e_shoff,
		ehdr->e_flags,
		ehdr->e_ehsize,
		ehdr->e_phentsize,
		ehdr->e_phnum,
		ehdr->e_shentsize,
		ehdr->e_shnum);
	phdr = (Elf32_Phdr *)&(header[ehdr->e_phoff]);
	shdr = (Elf32_Shdr *)&(header[ehdr->e_shoff]);

	if (verbose())
		fprintf(stderr, "%s: header %p #headers %d\n", __func__, ehdr, headers);

	entry = ehdr->e_entry;
	/* bss segments are special. They are in the section headers, 
	 * not program headers. So, sadly, we have to look at section headers. 
	 */

	for(i = 0; i < ehdr->e_shnum; i++) {
		char *p, *q;
		/* Ignore data that I don't need to handle */
		if (shdr[i].sh_type != SHT_NOBITS) {
			if (verbose())
				fprintf(stderr, "Dropping non SHT_NOBITS section\n");
			continue;
		}
		/* might need to test flags with SHF_ALLOC */
		if (shdr[i].sh_size == 0) {
			if (verbose())
				fprintf(stderr, "Dropping empty section\n");
			continue;
		}
		thisalgo = ALGO_ZEROES;
		if (verbose())
			fprintf(stderr,  "New section addr %#x size %#x\n",
			(u32)shdr[i].sh_addr, (u32)shdr[i].sh_size);
		/* Clean up the values */
		size = shdr[i].sh_size;
		if (verbose()) {
			fprintf(stderr, "(cleaned up) New section addr %p size 0x%#x\n",
				(void *)(unsigned long)shdr[i].sh_addr, (u32)shdr[i].sh_size);
		}
			/* ok, copy it out */
		sprintf(ename, "%s/segment%d", name, segment++);
		/* just allocate a bunch of zeros */
		p = calloc(sizeof(*p), size);
		q = calloc(sizeof(*q), size);
		complen = lar_compress(p, size, q, &thisalgo);
		ret = lar_add_entry(lar, ename, q, complen, size, 
				    shdr[i].sh_addr, entry, thisalgo);
		free(p);
		free(q);
	}

	for(i = 0; i < headers; i++) {
		/* Ignore data that I don't need to handle */
		if (phdr[i].p_type != PT_LOAD) {
			if (verbose())
				fprintf(stderr, "Dropping non PT_LOAD segment\n");
			continue;
		}
		if (phdr[i].p_memsz == 0) {
			if (verbose())
				fprintf(stderr, "Dropping empty segment\n");
			continue;
		}
		thisalgo = algo;
		if (verbose())
			fprintf(stderr,  "New segment addr 0x%ulx size 0x%ulx offset 0x%ulx filesize 0x%ulx\n",
			(u32)phdr[i].p_paddr, (u32)phdr[i].p_memsz, 
			(u32)phdr[i].p_offset, (u32)phdr[i].p_filesz);
		/* Clean up the values */
		size = phdr[i].p_filesz;
		if (phdr[i].p_filesz > phdr[i].p_memsz)  {
			size = phdr[i].p_memsz;
		}
		if (verbose()) {
			fprintf(stderr, "(cleaned up) New segment addr %p size 0x%#x offset 0x%x\n",
				(void *)(unsigned long)phdr[i].p_paddr, size, phdr[i].p_offset);
			fprintf(stderr, "Copy to %p from %p for %d bytes\n", 
				(unsigned char *)(unsigned long)phdr[i].p_paddr, 
				&header[phdr[i].p_offset], size);
			fprintf(stderr, "entry %ux loadaddr %ux\n", 
				entry,  phdr[i].p_paddr);
		}
			/* ok, copy it out */
		sprintf(ename, "%s/segment%d", name, segment++);
		complen = lar_compress(&header[phdr[i].p_offset], size, temp, &thisalgo);
		ret = lar_add_entry(lar, ename, temp, complen, size, 
				    phdr[i].p_paddr, entry, thisalgo);
	}
	return 0;
out:
	return -1;
}

/**
 * Given a size, return the offset of the bootblock (including the
 * header)
 * @param size Size of the LAR archive
 * @return The offset of the bootblock header
 */
static inline u32 get_bootblock_offset(u32 size)
{
	return size - (BOOTBLOCK_SIZE + sizeof(struct lar_header)
		       + BOOTBLOCK_NAME_LEN);
}

/**
 * Return the expected offset of the next LAR header after the given one
 * @param header The current LAR header
 * @return The offset of the next possible LAR header
 */
static inline u32 get_next_offset(struct lar_header *header) {
	return ((ntohl(header->len) + ntohl(header->offset) - 1) & 0xFFFFFFF0)
		+ 16;
}

/**
 * Read the size of the LAR archive from the size embedded in the bootblock
 * @param lar The LAR archive to read from
 * @return The size as read from the bootblock header
 */
static int lar_read_size(struct lar *lar)
{
	u32 *ptr = (u32 *) (lar->map + (lar->size - 12));
	return ptr[0];
}

/**
 * Add the LAR archive size to the bootblock, and clean up some other params
 * in what we're loosely calling the "bootblockh header"
 * @param ptr Pointer to the start of the bootblock
 * @param size The size value to write to the bootblock header
 */
static void annotate_bootblock(u8 *ptr, u32 size)
{
	int i;
	u32 *p;

	memset(ptr + (BOOTBLOCK_SIZE - 13), 0, 13);
	p = (u32 *) (ptr + BOOTBLOCK_SIZE - 12);
	p[0] = size;
}

/**
 * Add a bootblock file to the LAR - the bootblock must be of a consistant
 * size, and always gets put in a special location in the LAR
 * @param lar The LAR archive to write to
 * @param bootblock The name of the bootblock file to insert
 * @return 0 on success, or -1 on failure
 */
int lar_add_bootblock(struct lar *lar, const char *bootblock)
{
	u8 *offset;
	struct lar_header *header;
	int ret;

	offset = lar->map + get_bootblock_offset(lar->size);
	header = (struct lar_header *) offset;

	memcpy(header->magic, MAGIC, 8);
	header->reallen = htonl(BOOTBLOCK_SIZE);
	header->len = htonl(BOOTBLOCK_SIZE);
	header->offset = htonl(sizeof(struct lar_header) + BOOTBLOCK_NAME_LEN);

	offset += sizeof(struct lar_header);
	strcpy((char *) offset, BOOTBLOCK_NAME);

	offset += BOOTBLOCK_NAME_LEN;

	/* If a file waas specified, then load it, and read it directly
	 * into place */

	if (bootblock != NULL) {
		struct stat s;
		int fd = open(bootblock, O_RDONLY);

		if (fd == -1) {
			err("Unable to read bootblock file %s\n", bootblock);
			return -1;
		}

	        ret = fstat(fd, &s);

		if (ret == -1) {
			err("Unable to stat %s\n", bootblock);
			close(fd);
			return -1;
		}

		if (s.st_size != BOOTBLOCK_SIZE) {
			err("Bootblock %s does not appear to be a bootblock.\n",
			    bootblock);
			err("It is the wrong size; it should be %d bytes and it is %ld bytes\n", 
				BOOTBLOCK_SIZE, s.st_size);
			close(fd);
			return -1;
		}

		ret = read(fd, offset, BOOTBLOCK_SIZE);
		close(fd);

		if (ret != BOOTBLOCK_SIZE) {
			err("Unable to read all the bytes in the bootblock file.\n");
			return -1;
		}
	}

	annotate_bootblock(offset, lar->size);
	return 0;
}

/**
 * mmap() the LAR archive.
 *
 * @param lar The LAR archive information to map.
 * @param size Size of the mapped region.
 * @return 0 on success, or -1 on failure.
 */
static int _map_lar(struct lar *lar, u32 size)
{
	if (lar == NULL)
		return -1;

	lar->map = mmap(0, size, PROT_READ | PROT_WRITE, MAP_SHARED,
			lar->fd, 0);

	lar->size = size;

	if (lar->map == MAP_FAILED)
		return -1;

	return 0;
}

/**
 * Close the LAR archive and free all structures
 * @param lar The LAR archive to close
 */
void lar_close_archive(struct lar *lar)
{
	if (lar == NULL)
		return;

	if (lar->map != MAP_FAILED)
		munmap(lar->map, lar->size);

	if (lar->fd >= 0)
		close(lar->fd);

	free(lar);
}

/**
 * Create a new LAR archive - the archive must not exist on disk.
 * @param archive The filename of the new archive
 * @param size The intended size of the new archive
 * @return A LAR archive structure for the new archive
 */
struct lar * lar_new_archive(const char *archive, u32 size)
{
	struct lar *lar;
	int i;

	if (!access(archive, F_OK)) {
		err("Archive file %s already exists\n", archive);
		return NULL;
	}

	lar = calloc(sizeof(*lar), 1);

	if (lar == NULL) {
		err("Unable to allocate memory.\n");
		return NULL;
	}

	lar->fd = open(archive, O_RDWR | O_CREAT, S_IRUSR | S_IWUSR);

	if (lar->fd == -1) {
		err("Couldn't open the archive %s\n", archive);

		free(lar);
		return NULL;
	}

	/* Expand the file to the correct size */

	if (lseek(lar->fd, size - 1, SEEK_SET) == -1) {
		err("Unable to write the archive %s\n", archive);
		goto err;
	}

	if (write(lar->fd, "", 1) != 1) {
		err("Unable to write the file %s\n", archive);
		goto err;
	}

	if (_map_lar(lar, size)) {
		err("Unable to map the archive %s\n", archive);
		goto err;
	}

	/* Fill the whole thing with flash friendly 0xFFs */
	memset(lar->map, 0xFF, lar->size);

	/* Make a dummy bootblock */

	if (lar_add_bootblock(lar, NULL)) {
		err("Couldn't add a bootblock to %s\n", archive);
		goto err;
	}

	return lar;
err:
	lar_close_archive(lar);

	/* Don't leave a halfbaked LAR laying around */

	unlink(archive);
	return NULL;
}

/**
 * Open an existing LAR archive
 * @param archive The archive filename to open
 * @return A LAR archive structure for the opened archive
 */
struct lar * lar_open_archive(const char *archive)
{
	struct lar *lar;
	int ret;
	u32 romlen;
	struct stat s;

	lar = calloc(sizeof(*lar), 1);

	if (lar == NULL) {
		err("Unable to allocate memory.\n");
		return NULL;
	}

	lar->fd = open(archive, O_RDWR);

	ret = fstat(lar->fd, &s);

	if (ret == -1) {
		err("Unable to stat %s\n", archive);
		goto err;
	}

	if (_map_lar(lar, s.st_size)) {
		err("Unable to map the archive %s\n", archive);
		goto err;
	}

	/* Sanity check - make sure the bootblock header is the same length
	 * as the LAR archive
	 */

	romlen = lar_read_size(lar);

	if (romlen != s.st_size) {
		err("Size mismatch - the header says %d but the file is %d bytes long.\n",
		    romlen, (int) s.st_size);
		goto err;
	}

	return lar;

err:
	lar_close_archive(lar);
	return NULL;
}

/**
 * Return the offset of the first chunk of empty space in the LAR
 * @param lar the LAR archive structure
 * @return The offset of the first chunk of empty space
 */
static s64 lar_empty_offset(struct lar *lar)
{
	u32 offset = 0;
	struct lar_header *header;

	while (offset < get_bootblock_offset(lar->size)) {
		header = (struct lar_header *) (lar->map + offset);

		/* We interpet the absence of the magic as empty space */

		if (memcmp(header->magic, MAGIC, 8))
			break;

		offset += get_next_offset(header);
	}

	if (offset >= get_bootblock_offset(lar->size)) {
		err("No empty space found!\n");	
		return -1;
	}

	return (s64)offset;
}

/**
 * Return a 1 if filename is in the list of files - if files is NULL, then
 * act as if all files are in the list
 * @param files A list of files to check
 * @param filename The filename to check against the list
 * @return 1 if the file is in the list or if the list is NULL, 0 otherwise
 */
static int file_in_list(struct file *files, char *filename)
{
	struct file *p;

	if (files == NULL)
		return 1;

	for(p = files ; p; p = p->next) {
		if (!strncmp(p->name, filename, strlen(p->name)))
			return 1;
	}

	return 0;
}

/**
 * List the files in a LAR archive
 * @param lar The LAR archive to list
 * @param files A list of files to display- if this is NULL, then the default
 * is to list all files
 */
void lar_list_files(struct lar *lar, struct file *files)
{
	u8 *ptr = lar->map;
	u32 total_size = 0;
	char *filename;

	struct lar_header *header;
	struct file *fp;

	while (ptr < (lar->map + get_bootblock_offset(lar->size))) {
		header = (struct lar_header *) ptr;

		/* We interpret the absence of the magic as empty space */

		if (strncmp(header->magic, MAGIC, 8))
			break;

		filename = (char *) (ptr + sizeof(struct lar_header));

		if (file_in_list(files, filename)) {
			printf("  %s ", filename);
			total_size += ntohl(header->offset);
			total_size += ntohl(header->len);

			if (ntohl(header->compression) == ALGO_NONE) {
				printf("(%d bytes @0x%lx);",
				       ntohl(header->len),
				       (unsigned long)(ptr - lar->map) +
				       ntohl(header->offset));
			} else {
				printf("(%d bytes, %s compressed to %d bytes "
				       "@0x%lx);",
				       ntohl(header->reallen),
				       algo_name[ntohl(header->compression)],
				       ntohl(header->len),
				       (unsigned long)(ptr - lar->map) +
				       ntohl(header->offset));
			}
			printf("loadaddress 0x%#x entry 0x%#x\n", 
					ntohl(header->loadaddress), 
					ntohl(header->entry));
		}

		ptr += get_next_offset(header);
	}

	/* Show the bootblock */

	if (file_in_list(files, BOOTBLOCK_NAME)) {
		header = (struct lar_header *)
			(lar->map + get_bootblock_offset(lar->size));

		printf("  bootblock (%d bytes @0x%x)\n",
		       ntohl(header->len),
		       get_bootblock_offset(lar->size) +
		       ntohl(header->offset));
		total_size += sizeof(struct lar_header) + ntohl(header->len);
	}

	/* Print the total size */
	if (total_size >0)
		printf("Total size = %dB %dKB (0x%x)\n", 
			total_size, total_size/1024, total_size);
	else
		printf("No matching lar entries found.\n");
}

/**
 * Write a buffer to a file - this is used to write blobs from a LAR archive
 * @param filename The file to write
 * @param buffer The source buffer
 * @param len The length of the buffer
 * @return 0 on success , or -1 on failure
 */
static int _write_file(char *filename, u8 *buffer, u32 len)
{
	char *path = strdup(filename);
	int fd, ret;

	if (path == NULL) {
		err("Out of memory.\n");
		return -1;
	}

	mkdirp_below(".", (const char *) dirname(path), 0755);
	free(path);

	fd = open(filename, O_WRONLY | O_CREAT | O_TRUNC, 0644);

	if (fd == -1) {
		err("Error creating file %s\n", filename);
		return -1;
	}

	ret = write(fd, buffer, len);

	if (ret != len)
		err("Error writingthe file %s\n", filename);

	close(fd);
	return (ret == len) ? 0 : -1;
}

/**
 * Extract files from a LAR archive.  If files is not NULL, then only
 * extract the filenames listed within
 * @param lar The LAR archive to extract from
 * @param files A list of files to extract.  If NULL then all files are
 * extracted
 * @return 0 on success, or -1 on failure
 */
int lar_extract_files(struct lar *lar, struct file *files)
{
	u8 *ptr = lar->map;
	char *filename;
	struct lar_header *header;
	int ret = 0;

	while (ptr < (lar->map + get_bootblock_offset(lar->size))) {

		header = (struct lar_header *) ptr;

		if (strncmp(header->magic, MAGIC, 8))
			break;

		filename = (char *) (ptr + sizeof(struct lar_header));

		if (file_in_list(files, filename)) {

			if (ntohl(header->compression) == ALGO_NONE) {
				ret = _write_file(filename,
						  (u8 *) (ptr + ntohl(header->offset)),
						  (u32) ntohl(header->len));
			}
			else {
				char *buf =
					malloc(ntohl(header->reallen));

				if (buf == NULL) {
					err("Out of memory.\n");
					return -1;
				}

				uncompress_functions[ntohl(header->compression)](
					(char*) buf,
					ntohl(header->reallen),
					(char *) ptr + ntohl(header->offset),
					ntohl(header->len));

				ret = _write_file(filename, (u8 *) buf,
						  ntohl(header->reallen));

				free(buf);
			}

			if (ret == -1)
				return -1;
		}

		ptr += get_next_offset(header);
	}

	if (file_in_list(files, BOOTBLOCK_NAME)) {
		header = (struct lar_header *)
			(lar->map + get_bootblock_offset(lar->size));

		ret = _write_file((char *) BOOTBLOCK_NAME,
				  lar->map + (get_bootblock_offset(lar->size) + ntohl(header->offset)),
				  BOOTBLOCK_SIZE);
	}

	return ret;
}

/**
 * Given a pathname in the form [option;]path, determine the file name,
 * LAR path name, and compression algorithm. 
 * @param name name in the [option:][./]path form
 * @param pfilename reference pointer to file name -- this is modified
 * @param ppathname reference pointer to LAR path name -- this is modified
 * @param thisalgo pointer to algorithm, which can be modified by path name
 * @return 0 success, or -1 on failure (i.e. a bad name)
 */
int lar_process_name(char *name, char **pfilename, char **ppathname, 
		     enum compalgo *thisalgo)
{
	char *filename = name, *pathname = name;

	if (!strncmp(name, "nocompress:",11)) {
		filename += 11;
		*thisalgo = ALGO_NONE;
	}

	/* this is dangerous */
	if (filename[0] == '.' && filename[1] == '/')
		filename += 2;

	pathname = strchr(filename, ':');

	if (pathname != NULL) {
		*pathname = '\0';
		pathname++;

		if (!strlen(pathname)) {
			err("Invalid pathname specified.\n");
			return -1;
		}
	}
	else
		pathname = filename;
	*pfilename = filename;
	*ppathname = pathname;
	return 0;
}


/**
 * Given a pathname, open and mmap the file. 
 * @param filename
 * @param size pointer to returned size
 * @return ptr to mmap'ed area on success, or MAP_FAILED on failure
 */
char *mapfile(char *filename, u32 *size)
{
	int fd;
	struct stat s;
	char *ptr;

	/* Open the file */
	fd = open(filename, O_RDONLY);

	if (fd == -1) {
		err("Unable to open %s\n", filename);
		return MAP_FAILED;
	}

	if (fstat(fd, &s)) {
		err("Unable to stat the file %s\n", filename);
		close(fd);
		return MAP_FAILED;
	}

	ptr = mmap(0, s.st_size, PROT_READ, MAP_SHARED, fd, 0);
	close(fd);

	if (ptr == MAP_FAILED) {
		err("Unable to map %s\n", filename);
		return MAP_FAILED;
	}
	*size = s.st_size;
	return ptr;
}

/**
 * Given a name, return the size of the header for that name.  
 * 
 * @param pathname header name
 * @param new_pathlen pointer to the (possibly truncated) pathlen
 * @return header size
 */
int header_len(char *pathname, int *new_pathlen)
{
	int pathlen;
	int len;

	pathlen = strlen(pathname) + 1 > MAX_PATHLEN ? 
		MAX_PATHLEN : strlen(pathname) + 1;
	len = sizeof(struct lar_header) + pathlen;
	len = (len + 15) & 0xFFFFFFF0;
	if (new_pathlen!=NULL)
		*new_pathlen = pathlen;

	return len;
}

/**
 * Return the amount of space left in a lar, given a name for the entry
 * @param lar TODO
 * @param name Name of the entry
 * @return Maximum possible size for the entry
 */
int maxsize(struct lar *lar, char *name)
{
	int size;
	s64 offset;
	int bootblock_size;

	/* Find the beginning of the available space in the LAR */
#warning We should check all chunks of free space in the LAR. Right now we do not return the maximum size, but the size of the first chunk.
	offset = lar_empty_offset(lar);
	if (offset < 0) {
		err("maxsize is negative\n");
		return offset;
	}

	/* Figure out how big our header will be */
	size = get_bootblock_offset(lar->size) - offset - header_len(name,NULL) - 1;

	return size;
}

/**
 * Compress an area according to an algorithm. If the area grows, 
 * use no compression.  The size of temp should be at least size bytes.
 * @param ptr data to be compressed
 * @param size size of the data 
 * @param temp destination of compressed data
 * @param thisalgo pointer to algorithm -- this can be modified
 * @return  size of compressed data
 */
int lar_compress(char *ptr, int size, char *temp, enum compalgo *thisalgo)
{
	int complen;
	compress_functions[*thisalgo](ptr, size, temp, &complen);

	if (complen >= size && (*thisalgo != ALGO_NONE)) {
		*thisalgo = ALGO_NONE;
		compress_functions[*thisalgo](ptr, size, temp, &complen);
	}
	return complen;
}

/**
 * Add a new entry to the LAR archive
 * @param lar The LAR archive to write into
 * @param pathname The name of the segment
 * @param data The data for the segment
 * @param complen The compressed length of the segment
 * @param reallen The real (uncompressed) length of the segment
 * @param loadaddress The load address of the segment
 * @param entry The entry point of the segment
 * @param thisalgo The compression algorithm
 * @return  0 on success, or -1 on failure
 */
int lar_add_entry(struct lar *lar, char *pathname, void *data, 
		  u32 complen, u32 reallen, u32 loadaddress, u32 entry, 
		  enum compalgo thisalgo)
{
	struct lar_header *header;
	int ret, hlen;
	int pathlen;
	u32 *walk,  csum;
	s64 offset;

	/* Find the beginning of the available space in the LAR */
	offset = lar_empty_offset(lar);

	/* Figure out how big our header will be */
	hlen = header_len(pathname, &pathlen);

	if ((offset < 0) || 
		(offset + hlen + complen >= get_bootblock_offset(lar->size))) {
		err("Not enough room in the LAR to add the file.\n");
		return -1;
	}

	/* Lets do this thing */

	/* Zero out the header area */
	memset(lar->map + offset, 0, hlen);

	header = (struct lar_header *) (lar->map + offset);

	memcpy(header, MAGIC, 8);
	header->compression = htonl(thisalgo);
	header->reallen = htonl(reallen);
	header->len = htonl(complen);
	header->offset = htonl(hlen);
	header->loadaddress = htonl(loadaddress);
	header->entry = htonl(entry);
	/* Copy the path name */
	strncpy((char *) (lar->map + offset + sizeof(struct lar_header)),
		pathname, pathlen - 1);

	/* Copy in the data */
	memcpy(lar->map + (offset + hlen), data, complen);

	/* Figure out the checksum */

	csum = 0;
	for (walk = (u32 *) (lar->map + offset);
	     walk < (u32 *) (lar->map + complen + hlen);
	     walk++) {
		csum += ntohl(*walk);
	}
	header->checksum = htonl(csum);
	return 0;
}

/**
 * Add a new file to the LAR archive
 * @param lar The LAR archive to write into
 * @param file The name of the file to add (TODO: Fix comment)
 * @return  0 on success, or -1 on failure
 */
int lar_add_file(struct lar *lar, struct file* file)
{
	char *ptr, *temp;

	struct lar_header *header;
	int ret, hlen;
	u32 complen;
	u32 size;

	ptr = mapfile(file->name, &size);

	if (ptr == MAP_FAILED)
		return -1;

	if (elfparse() && iself(ptr)) {
		output_elf_segments(lar, file->pathname, ptr, size, file->algo);
		return 0;
	}

	/* This is legacy stuff. */

	/* Allocate a temporary buffer to compress into - this is unavoidable,
	   because we need to make sure that the compressed data will fit in
	   the LAR, and we won't know the size of the compressed data until
	   we actually compress it */

	temp = calloc(size, 1);

	if (temp == NULL) {
		err("Out of memory.\n");
		munmap(ptr, size);
		return -1;
	}

	complen = lar_compress(ptr, size, temp, &file->algo);

	munmap(ptr, size);

	ret = lar_add_entry(lar, file->pathname, temp, complen, size, 0, 0, file->algo);

	free(temp);
	return ret;
}