coreboot/util/lar/create.c

/*
 * lar - LinuxBIOS archiver
 *
 * Copyright (C) 2006-2007 coresystems GmbH
 * Written by Stefan Reinauer <stepan@coresystems.de> for coresystems GmbH.
 *
 * 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 <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <fcntl.h>
#include <errno.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <sys/mman.h>
#include <netinet/in.h>

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

int create_lar(const char *archivename, struct file *files)
{
	int i, ret;
	int diff = 0;
	FILE *archive, *source;
	char *tempmem;
	char *filebuf;
	char *pathname;
	u32 *walk;
	u32 csum;
	int pathlen, entrylen, filelen;
	long currentsize = 0;
	struct lar_header *header;
	struct stat statbuf;

	if (!files) {
		fprintf(stderr, "No files for archive %s\n", archivename);
		exit(1);
	}

	if (verbose())
		printf("Opening %s\n", archivename);

	archive = fopen(archivename, "w");
	if (!archive) {
		fprintf(stderr, "Could not open archive %s for writing\n",
			archivename);
		exit(1);
	}

	while (files) {
		char *name = files->name;

		/* skip ./ if available */
		if (name[0] == '.' && name[1] == '/')
			name += 2;

		if (verbose())
			printf("  Adding %s to archive\n", name);

		ret = stat(name, &statbuf);
		if (ret) {
			fprintf(stderr, "No such file %s\n", name);
			exit(1);
		}
		filelen = statbuf.st_size;

		tempmem = malloc(sizeof(struct lar_header) + MAX_PATHLEN
				 + filelen + 16);
		if (!tempmem) {
			fprintf(stderr, "Out of memory.\n");
			return (1);
		}
		memset(tempmem, 0, sizeof(struct lar_header) + MAX_PATHLEN
		       + filelen + 16);

		header = (struct lar_header *)tempmem;
		pathname = tempmem + sizeof(struct lar_header);
		pathlen = snprintf(pathname, MAX_PATHLEN - 1, name) + 1;
		pathlen = (pathlen + 15) & 0xfffffff0;/* Align to 16 bytes. */

		/* Read file into memory. */
		filebuf = pathname + pathlen;
		source = fopen(name, "r");
		if (!source) {
			fprintf(stderr, "No such file %s\n", name);
			exit(1);
		}
		fread(filebuf, statbuf.st_size, 1, source);
		fclose(source);

		/* Create correct header. */
		memcpy(header, MAGIC, 8);
		header->len = htonl(statbuf.st_size);
		header->offset = htonl(sizeof(struct lar_header) + pathlen);

		/* Calculate checksum. */
		csum = 0;
		for (walk = (u32 *) tempmem;
		     walk < (u32 *) (tempmem + statbuf.st_size +
				     sizeof(struct lar_header) + pathlen);
		     walk++) {
			csum += ntohl(*walk);
		}
		header->checksum = htonl(csum);

		/* Write out entry to archive. */
		entrylen = (filelen + pathlen + sizeof(struct lar_header) +
			    15) & 0xfffffff0;

		fwrite(tempmem, entrylen, 1, archive);

		free(tempmem);

		/* size counter */
		currentsize += entrylen;

		files = files->next;
	}

	/* Calculate difference, if a size has been specified.
	 * Otherwise diff stays 0 and no action is taken below.
	 */
	if (get_larsize())
		diff = get_larsize() - currentsize;

	if (diff < 0) {
		fprintf(stderr,
			"Error: LAR archive exceeded size (%ld > %ld)\n",
			currentsize, get_larsize());

		/* Open files can not be deleted. */
		fclose(archive);
		/* File is too big, delete it. */
		unlink(archivename);
		return -1;
	}

	if (diff > 0) {
		char *padding;
		/* generate padding (0xff is flash friendly) */
		padding = malloc(diff);
		if (!padding) {
			fprintf(stderr, "Out of memory.\n");
			exit(1);
		}
		memset(padding, 0xff, diff);
		fwrite(padding, diff, 1, archive);
		free(padding);
	}

	fclose(archive);

	if (verbose())
		printf("done.\n");

	return 0;
}