Add container information of a given kernel in an appropriate second stage file

tools/emile.c

@@ -0,0 +1,157 @@
+/*
+ *
+ * (c) 2004 Laurent Vivier <LaurentVivier@wanadoo.fr>
+ *
+ */
+
+#include <sys/types.h>
+#include <sys/stat.h>
+#include <fcntl.h>
+#include <errno.h>
+#include <string.h>
+#include <stdio.h>
+#include <unistd.h>
+#include <stdlib.h>
+
+#include "blocks.h"
+#include "emile.h"
+
+int main(int argc, char **argv)
+{
+	int fd;
+	int fd_kernel;
+	int ret;
+	emile_l2_header_t header;
+	off_t container_offset;
+	struct emile_container *container;
+	int max_blocks;
+	int i;
+	unsigned long kernel_image_size;
+
+	fd = open(argv[1], O_RDWR);
+	if (fd == -1)
+	{
+		perror("Cannot open second stage");
+		return 1;
+	}
+
+	ret = read(fd, &header, sizeof(header));
+	if (ret != sizeof(header))
+	{
+		close(fd);
+		return 2;
+	}
+
+	if (!EMILE_COMPAT(EMILE_04_SIGNATURE, read_long(&header.signature)))
+	{
+		fprintf(stderr, "Bad Header signature\n");
+		close(fd);
+		return 3;
+	}
+
+	container_offset = read_long(&header.kernel_image_offset);
+	if (container_offset == 0)
+	{
+		fprintf(stderr, "Corrupted Header, kernel_image_offset must not be 0\n");
+		close(fd);
+		return 5;
+	}
+
+	printf("Container found at offset %ld\n", container_offset);
+
+	ret = lseek(fd, container_offset, SEEK_SET);
+	if (ret != container_offset) 
+	{
+		perror("lseek");
+		close(fd);
+		return 5;
+	}
+
+	container = (struct emile_container*)
+					malloc(sizeof(struct emile_container));
+	if (container == NULL) {
+		perror("malloc");
+		return 6;
+	}
+
+	ret = read(fd, container, sizeof(struct emile_container));
+	if (ret != sizeof(struct emile_container))
+	{
+		close(fd);
+		return 7;
+	}
+
+	max_blocks = container->max_blocks;
+	printf("Max blocks: %d\n", max_blocks);
+
+	free(container);
+	container = (struct emile_container*)
+				malloc(sizeof(struct emile_container) 
+				     + max_blocks * sizeof(struct emile_block));
+	if (container == NULL) {
+		perror("malloc");
+		return 8;
+	}
+	container->max_blocks = max_blocks;
+	fd_kernel = open(argv[2], O_RDONLY);
+	if (fd_kernel == -1)
+	{
+		perror("open kernel");
+		return 9;
+	}
+	ret = scsi_create_container(fd_kernel, container);
+	if (ret != 0)
+		return 10;
+	close(fd_kernel);
+
+	kernel_image_size = 0;
+	for(i = 0; i < max_blocks; i++)
+	{
+		if (container->blocks[i].count == 0)
+			break;
+		printf("(%d, %d) ", container->blocks[i].offset, 
+				    container->blocks[i].count);
+		kernel_image_size += container->blocks[i].count;
+	}
+	putchar('\n');
+	kernel_image_size *= container->block_size;
+	printf("Kernel image size: %ld bytes\n", kernel_image_size);
+
+	ret = lseek(fd, container_offset, SEEK_SET);
+	if (ret != container_offset) 
+	{
+		perror("lseek");
+		close(fd);
+		return 11;
+	}
+
+	ret = write(fd, container, sizeof(struct emile_container)
+                                     + max_blocks * sizeof(struct emile_block));
+	if (ret != sizeof(struct emile_container)
+                                     + max_blocks * sizeof(struct emile_block))
+	{
+		perror("write");
+		return 12;
+	}
+
+	ret = lseek(fd, 0, SEEK_SET);
+	if (ret != 0) 
+	{
+		perror("lseek");
+		close(fd);
+		return 11;
+	}
+	header.kernel_image_size = kernel_image_size;
+
+	ret = write(fd, &header, sizeof(header));
+	if (ret != sizeof(header))
+	{
+		perror("write");
+		close(fd);
+		return 12;
+	}
+	
+
+	close(fd);
+	return 0;
+}