Add SCSI support for level 1

tools/emile.c

@@ -16,9 +16,98 @@
 #include "blocks.h"
 #include "emile.h"
 
-int main(int argc, char **argv)
+int patch_first(int fd, char *second_name)
+{
+	int ret;
+	int fd_second;
+	char first[1024];
+	int i;
+	int current;
+	struct emile_container *container;
+	unsigned short max_blocks;
+	unsigned short *first_max_blocks = (unsigned short*)&first[1022];
+	unsigned long *second_size = (unsigned long*)&first[1018];
+	unsigned short *unit_id = (unsigned short*)&first[1016];
+	unsigned short *block_size = (unsigned short*)&first[1014];
+	unsigned short *count;
+	unsigned long *offset;
+
+	ret = read(fd, first, 1024);
+	if (ret == -1) {
+		perror("Cannot read second");
+		return 1;
+	}
+
+	max_blocks = *first_max_blocks / 6;
+
+	printf("Available blocks: %d\n", max_blocks);
+	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_second = open(second_name, O_RDONLY);
+	if (fd_second == -1)
+	{
+		perror("open kernel");
+		return 9;
+	}
+	ret = scsi_create_container(fd_second, container);
+	if (ret != 0)
+		return 10;
+	close(fd_second);
+
+	*unit_id = container->unit_id;
+	*block_size = container->block_size;
+
+	*second_size = 0;
+	current = 1014;
+	for(i = 0; i < max_blocks - 1; i++)
+	{
+		current -= 2;
+		count = (short*)(&first[current]);
+		*count = container->blocks[i].count;
+		if (container->blocks[i].count == 0)
+			break;
+		current -= 4;
+		offset = (long*)(&first[current]);
+		*offset = container->blocks[i].offset;
+		printf("(%d, %d) ", container->blocks[i].offset, 
+				    container->blocks[i].count);
+		(*second_size) += container->blocks[i].count;
+	}
+	putchar('\n');
+	/* mark end of blocks list */
+	current -= 2;
+	count = (short*)(&first[current]);
+	*count = 0;
+	/* set second level size */
+	(*second_size) *= container->block_size;
+	printf("Second size: %ld bytes\n", *second_size);
+
+	ret = lseek(fd, 0, SEEK_SET);
+	if (ret != 0) 
+	{
+		perror("lseek");
+		close(fd);
+		return 11;
+	}
+
+	ret = write(fd, first, 1024);
+	if (ret == -1) {
+		perror("Cannot read second");
+		return 1;
+	}
+	
+	return ret;
+}
+
+int patch_second(int fd, char *kernel_name)
 {
-	int fd;
 	int fd_kernel;
 	int ret;
 	emile_l2_header_t header;
@@ -28,13 +117,6 @@ int main(int argc, char **argv)
 	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))
 	{
@@ -93,7 +175,7 @@ int main(int argc, char **argv)
 		return 8;
 	}
 	container->max_blocks = max_blocks;
-	fd_kernel = open(argv[2], O_RDONLY);
+	fd_kernel = open(kernel_name, O_RDONLY);
 	if (fd_kernel == -1)
 	{
 		perror("open kernel");
@@ -150,6 +232,29 @@ int main(int argc, char **argv)
 		close(fd);
 		return 12;
 	}
+
+	return 0;
+}
+
+int main(int argc, char **argv)
+{
+	int fd;
+	short id;
+	int ret;
+
+	fd = open(argv[1], O_RDWR);
+	if (fd == -1)
+	{
+		perror("Cannot open second stage");
+		return 1;
+	}
+
+	ret = read(fd, &id, 2);
+	lseek(fd, 0, SEEK_SET);
+	if ( (ret == 2) && (id == 0x4C4B) )
+		patch_first(fd, argv[2]);
+	else
+		patch_second(fd, argv[2]);
 	
 
 	close(fd);