mirror of
https://github.com/vivier/EMILE.git
synced 2024-07-31 23:29:05 +00:00
use getopt() and emile_get_uncompressed_size()
This commit is contained in:
Laurent Vivier
parent
964f5809ac
commit
24baafee89
@ -11,129 +11,170 @@
|
||||
#include <fcntl.h>
|
||||
#include <stdlib.h>
|
||||
#include <string.h>
|
||||
#include <getopt.h>
|
||||
|
||||
#include "libemile.h"
|
||||
|
||||
enum {
|
||||
ARG_NONE = 0,
|
||||
ARG_HELP ='h',
|
||||
ARG_VERBOSE = 'v',
|
||||
ARG_FIRST = 'f',
|
||||
ARG_SECOND = 's',
|
||||
ARG_KERNEL = 'k',
|
||||
ARG_RAMDISK = 'r',
|
||||
ARG_BUFFER ='b',
|
||||
};
|
||||
|
||||
static struct option long_options[] =
|
||||
{
|
||||
{"help", 0, NULL, ARG_HELP },
|
||||
{"verbose", 0, NULL, ARG_VERBOSE },
|
||||
{"first", 1, NULL, ARG_FIRST },
|
||||
{"second", 1, NULL, ARG_SECOND },
|
||||
{"kernel", 1, NULL, ARG_KERNEL },
|
||||
{"ramdisk", 1, NULL, ARG_RAMDISK },
|
||||
{"buffer", 1, NULL, ARG_BUFFER },
|
||||
{NULL, 0, NULL, 0 },
|
||||
};
|
||||
|
||||
static void usage(int argc, char** argv)
|
||||
{
|
||||
fprintf(stderr, "Usage: %s -f <first level> -s <second level> -i <kernel image> -r <ramdisk> -b <buffer size> <image>\n", argv[0]);
|
||||
fprintf(stderr, "Usage: %s [OPTION] <image>\n", argv[0]);
|
||||
fprintf(stderr, "Create an EMILE bootable floppy disk\n");
|
||||
fprintf(stderr, "EMILE allows you to boot linux from a floppy disk\n");
|
||||
fprintf(stderr, " -h, --help display this text\n");
|
||||
fprintf(stderr, " -v, --verbose verbose mode\n");
|
||||
fprintf(stderr, " -f, --first first level to copy to floppy\n");
|
||||
fprintf(stderr, " -s, --second second level to copy to floppy\n");
|
||||
fprintf(stderr, " -k, --kernel kernel to copy to floppy\n");
|
||||
fprintf(stderr, " -r, --ramdisk ramdisk to copy to floppy\n");
|
||||
fprintf(stderr, " -b, --buffer buffer size to decompress kernel\n");
|
||||
fprintf(stderr, "\nbuild: \n%s\n", SIGNATURE);
|
||||
}
|
||||
|
||||
int main(int argc, char** argv)
|
||||
{
|
||||
int verbose = 0;
|
||||
int option_index = 0;
|
||||
char* first_level = NULL;
|
||||
char* second_level = NULL;
|
||||
char* kernel_image = NULL;
|
||||
char* ramdisk = NULL;
|
||||
unsigned long buffer_size = 0;
|
||||
char* image = NULL;
|
||||
int i;
|
||||
int c;
|
||||
int ret;
|
||||
|
||||
if ((argc != 12) && (argc != 10) && (argc != 8) )
|
||||
while(1)
|
||||
{
|
||||
c = getopt_long(argc, argv, "hvf:s:k:r:b:", long_options,
|
||||
&option_index);
|
||||
if (c == -1)
|
||||
break;
|
||||
switch(c)
|
||||
{
|
||||
case ARG_HELP:
|
||||
usage(argc, argv);
|
||||
return 0;
|
||||
case ARG_VERBOSE:
|
||||
verbose = 1;
|
||||
break;
|
||||
case ARG_FIRST:
|
||||
first_level = optarg;
|
||||
break;
|
||||
case ARG_SECOND:
|
||||
second_level = optarg;
|
||||
break;
|
||||
case ARG_KERNEL:
|
||||
kernel_image = optarg;
|
||||
break;
|
||||
case ARG_RAMDISK:
|
||||
ramdisk = optarg;
|
||||
break;
|
||||
case ARG_BUFFER:
|
||||
buffer_size = atol(optarg);
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
if (optind < argc)
|
||||
image = argv[optind];
|
||||
|
||||
if (image == NULL)
|
||||
{
|
||||
fprintf(stderr,
|
||||
"ERROR: you must provide an image file or a block device.\n");
|
||||
usage(argc, argv);
|
||||
return 1;
|
||||
}
|
||||
|
||||
for (i = 1; i < argc; i++)
|
||||
if (first_level == NULL)
|
||||
first_level = PREFIX "/lib/emile/first_floppy";
|
||||
|
||||
if (second_level == NULL)
|
||||
second_level = PREFIX "/lib/emile/second_floppy";
|
||||
|
||||
if (kernel_image == NULL)
|
||||
kernel_image = PREFIX "/boot/vmlinuz";
|
||||
|
||||
if (buffer_size == 0)
|
||||
{
|
||||
if (strcmp("-f", argv[i]) == 0)
|
||||
buffer_size = emile_get_uncompressed_size(kernel_image);
|
||||
if (buffer_size == -1)
|
||||
{
|
||||
if (first_level != NULL)
|
||||
{
|
||||
fprintf(stderr, "Duplicate definition of -f\n");
|
||||
return 1;
|
||||
}
|
||||
i++;
|
||||
if (i < argc)
|
||||
first_level = argv[i];
|
||||
else
|
||||
{
|
||||
fprintf(stderr, "Missing argument for -f\n");
|
||||
return 1;
|
||||
}
|
||||
}
|
||||
else if (strcmp("-s", argv[i]) == 0)
|
||||
{
|
||||
if (second_level != NULL)
|
||||
{
|
||||
fprintf(stderr, "Duplicate definition of -s\n");
|
||||
return 1;
|
||||
}
|
||||
i++;
|
||||
if (i < argc)
|
||||
second_level = argv[i];
|
||||
else
|
||||
{
|
||||
fprintf(stderr, "Missing argument for -s\n");
|
||||
return 1;
|
||||
}
|
||||
}
|
||||
else if (strcmp("-i", argv[i]) == 0)
|
||||
{
|
||||
if (kernel_image != NULL)
|
||||
{
|
||||
fprintf(stderr, "Duplicate definition of -i\n");
|
||||
return 1;
|
||||
}
|
||||
i++;
|
||||
if (i < argc)
|
||||
kernel_image = argv[i];
|
||||
else
|
||||
{
|
||||
fprintf(stderr, "Missing argument for -i\n");
|
||||
return 1;
|
||||
}
|
||||
}
|
||||
else if (strcmp("-b", argv[i]) == 0)
|
||||
{
|
||||
if (buffer_size != 0)
|
||||
{
|
||||
fprintf(stderr, "Duplicate definition of -b\n");
|
||||
return 1;
|
||||
}
|
||||
i++;
|
||||
if (i < argc)
|
||||
buffer_size = atol(argv[i]);
|
||||
else
|
||||
{
|
||||
fprintf(stderr, "Missing argument for -b\n");
|
||||
return 1;
|
||||
}
|
||||
}
|
||||
else if (strcmp("-r", argv[i]) == 0)
|
||||
{
|
||||
if (ramdisk != 0)
|
||||
{
|
||||
fprintf(stderr, "Duplicate definition of -r\n");
|
||||
return 1;
|
||||
}
|
||||
i++;
|
||||
if (i < argc)
|
||||
ramdisk = argv[i];
|
||||
else
|
||||
{
|
||||
fprintf(stderr, "Missing argument for -r\n");
|
||||
return 1;
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
if (image != NULL)
|
||||
{
|
||||
fprintf(stderr, "Unknown parameter %s\n",
|
||||
argv[i]);
|
||||
usage(argc, argv);
|
||||
return 1;
|
||||
}
|
||||
image = argv[i];
|
||||
fprintf(stderr,
|
||||
"ERROR: cannot compute size of uncompressed kernel\n");
|
||||
fprintf(stderr,
|
||||
" use \"--buffer <size>\" to set it or set path of gzip in PATH\n");
|
||||
fprintf(stderr,
|
||||
" or check \"%s\" can be read\n", kernel_image);
|
||||
return 2;
|
||||
}
|
||||
}
|
||||
|
||||
if (verbose)
|
||||
{
|
||||
printf("first: %s\n", first_level);
|
||||
printf("second: %s\n", second_level);
|
||||
printf("kernel: %s\n", kernel_image);
|
||||
printf("buffer: %ld\n", buffer_size);
|
||||
printf("ramdisk: %s\n", ramdisk);
|
||||
printf("image: %s\n", image);
|
||||
}
|
||||
|
||||
ret = emile_floppy_create_image(first_level, second_level,
|
||||
kernel_image, ramdisk, buffer_size,
|
||||
image);
|
||||
switch(ret)
|
||||
{
|
||||
case 0:
|
||||
break;
|
||||
case EEMILE_CANNOT_WRITE_FIRST:
|
||||
fprintf(stderr, "ERROR: cannot write first\n");
|
||||
break;
|
||||
case EEMILE_CANNOT_WRITE_SECOND:
|
||||
fprintf(stderr, "ERROR: cannot write second\n");
|
||||
break;
|
||||
case EEMILE_CANNOT_WRITE_KERNEL:
|
||||
fprintf(stderr, "ERROR: cannot write kernel\n");
|
||||
break;
|
||||
case EEMILE_CANNOT_WRITE_RAMDISK:
|
||||
fprintf(stderr, "ERROR: cannot write ramdisk\n");
|
||||
break;
|
||||
case EEMILE_CANNOT_WRITE_PAD:
|
||||
fprintf(stderr, "ERROR: cannot write padding\n");
|
||||
break;
|
||||
case EEMILE_CANNOT_CREATE_IMAGE:
|
||||
fprintf(stderr, "ERROR: cannot create image\n");
|
||||
break;
|
||||
case EEMILE_CANNOT_OPEN_FILE:
|
||||
fprintf(stderr, "ERROR: cannot open one of provided files\n");
|
||||
break;
|
||||
default:
|
||||
fprintf(stderr, "ERROR: unknown error :-P\n");
|
||||
break;
|
||||
}
|
||||
|
||||
return ret;
|
||||
}
|
||||
|
||||
Loading…
Reference in New Issue
Block a user