mirror of
https://github.com/vivier/EMILE.git
synced 2024-09-08 19:54:38 +00:00
247 lines
5.4 KiB
C
247 lines
5.4 KiB
C
/*
|
|
*
|
|
* (c) 2004 Laurent Vivier <Laurent@lvivier.info>
|
|
*
|
|
*/
|
|
|
|
#include <stdio.h>
|
|
#include <unistd.h>
|
|
#include <stdlib.h>
|
|
#include <sys/types.h>
|
|
#include <sys/stat.h>
|
|
#include <fcntl.h>
|
|
#include <string.h>
|
|
#include <getopt.h>
|
|
|
|
#include "emile.h"
|
|
#include "libemile.h"
|
|
|
|
#define BLOCK_SIZE 512
|
|
|
|
enum {
|
|
ARG_NONE = 0,
|
|
ARG_HELP ='h',
|
|
ARG_DRIVE = 'd',
|
|
ARG_OFFSET ='o',
|
|
ARG_SIZE = 's',
|
|
ARG_PATH = 'p',
|
|
ARG_SCSI = 'i',
|
|
};
|
|
|
|
static struct option long_options[] =
|
|
{
|
|
{"help", 0, NULL, ARG_HELP },
|
|
{"drive", 1, NULL, ARG_DRIVE },
|
|
{"offset", 1, NULL, ARG_OFFSET },
|
|
{"size", 1, NULL, ARG_SIZE },
|
|
{"path", 1, NULL, ARG_PATH },
|
|
{"scsi", 0, NULL, ARG_SCSI },
|
|
{NULL, 0, NULL, 0 },
|
|
};
|
|
|
|
static void usage(int argc, char** argv)
|
|
{
|
|
fprintf(stderr, "Usage: %s [-i][-d <drive>][-o <offset>][-s <size>] <image>\n", argv[0]);
|
|
fprintf(stderr, "Usage: %s [-p <path>|-b <id> <start> <length>] <image>\n", argv[0]);
|
|
fprintf(stderr, "Set EMILE first level boot block info (floppy or scsi):\n");
|
|
fprintf(stderr, " -d, --drive <drive> set the drive number (default 1)\n");
|
|
fprintf(stderr, " -o, --offset <offset> set offset of second level in bytes\n");
|
|
fprintf(stderr, " -s, --size <size> set size of second level in bytes\n");
|
|
fprintf(stderr, " -i, --scsi specify scsi first level format (offset is a block number)\n");
|
|
fprintf(stderr, "Set EMILE first level boot block info (scsi):\n");
|
|
fprintf(stderr, " -p, --path <path> set path of second level\n");
|
|
fprintf(stderr, "Display current values if no flags provided\n");
|
|
fprintf(stderr, "\nbuild: \n%s\n", SIGNATURE);
|
|
}
|
|
|
|
int first_tune( char* image, unsigned short tune_mask, int drive_num,
|
|
int second_offset, int second_size)
|
|
{
|
|
int fd;
|
|
int ret;
|
|
|
|
fd = open(image, O_RDWR);
|
|
if (fd == -1)
|
|
{
|
|
perror("Cannot open image file");
|
|
return 2;
|
|
}
|
|
if (tune_mask == 0)
|
|
{
|
|
ret = emile_first_get_param(fd, &drive_num, &second_offset,
|
|
&second_size);
|
|
if (ret == 0)
|
|
{
|
|
printf("EMILE boot block identified\n\n");
|
|
printf("Drive number: %d\n", drive_num);
|
|
printf("Second level offset: %d\n", second_offset);
|
|
printf("Second level size: %d\n", second_size);
|
|
}
|
|
else
|
|
printf("EMILE is not installed in this bootblock\n");
|
|
|
|
return 0;
|
|
}
|
|
|
|
ret = emile_first_set_param(fd, tune_mask, drive_num, second_offset, second_size);
|
|
|
|
close(fd);
|
|
return 0;
|
|
}
|
|
|
|
int first_tune_path( char* image, char *second_path)
|
|
{
|
|
int fd;
|
|
int ret;
|
|
|
|
fd = open(image, O_RDWR);
|
|
if (fd == -1)
|
|
{
|
|
perror("Cannot open image file");
|
|
return 2;
|
|
}
|
|
|
|
ret = emile_first_set_param_scsi(fd, second_path);
|
|
|
|
close(fd);
|
|
return ret;
|
|
}
|
|
|
|
int first_tune_scsi( char* image, int drive_num, int second_offset, int size)
|
|
{
|
|
int fd;
|
|
int ret;
|
|
char first[1024];
|
|
int current;
|
|
unsigned short max_blocks;
|
|
|
|
fd = open(image, O_RDWR);
|
|
if (fd == -1)
|
|
{
|
|
perror("Cannot open image file");
|
|
return 2;
|
|
}
|
|
|
|
ret = read(fd, first, 1024);
|
|
if (ret == -1)
|
|
return EEMILE_CANNOT_READ_FIRST;
|
|
|
|
max_blocks = read_short((u_int16_t*)&first[1022]) / 6;
|
|
|
|
write_short((u_int16_t*)&first[1014], BLOCK_SIZE);
|
|
write_short((u_int16_t*)&first[1016], drive_num);
|
|
|
|
write_long((u_int32_t*)&first[1018], 0);
|
|
current = 1014;
|
|
|
|
current -= 2;
|
|
write_short((u_int16_t*)&first[current], (size + BLOCK_SIZE - 1) / BLOCK_SIZE);
|
|
current -= 4;
|
|
write_long((u_int32_t*)&first[current], second_offset);
|
|
|
|
/* mark end of blocks list */
|
|
current -= 2;
|
|
write_short((u_int16_t*)(&first[current]), 0);
|
|
/* set second level size */
|
|
write_long((u_int32_t*)&first[1018], (size + BLOCK_SIZE - 1) / BLOCK_SIZE * BLOCK_SIZE);
|
|
|
|
ret = lseek(fd, 0, SEEK_SET);
|
|
if (ret != 0)
|
|
return EEMILE_CANNOT_WRITE_FIRST;
|
|
|
|
ret = write(fd, first, 1024);
|
|
if (ret == -1)
|
|
return EEMILE_CANNOT_WRITE_FIRST;
|
|
|
|
close(fd);
|
|
return 0;
|
|
}
|
|
|
|
int main(int argc, char** argv)
|
|
{
|
|
int ret;
|
|
int option_index;
|
|
int c;
|
|
char* image = NULL;
|
|
char* path = NULL;
|
|
unsigned short tune_mask = 0;
|
|
int drive_num, second_offset, second_size;
|
|
int use_scsi = 0;
|
|
|
|
while(1)
|
|
{
|
|
c = getopt_long(argc, argv, "hd:o:s:p:i", long_options,
|
|
&option_index);
|
|
if (c == EOF)
|
|
break;
|
|
|
|
switch(c)
|
|
{
|
|
case ARG_HELP:
|
|
usage(argc, argv);
|
|
return 0;
|
|
case ARG_DRIVE:
|
|
tune_mask |= EMILE_FIRST_TUNE_DRIVE;
|
|
drive_num = atoi(optarg);
|
|
break;
|
|
case ARG_OFFSET:
|
|
tune_mask |= EMILE_FIRST_TUNE_OFFSET;
|
|
second_offset = atoi(optarg);
|
|
break;
|
|
case ARG_SIZE:
|
|
tune_mask |= EMILE_FIRST_TUNE_SIZE;
|
|
second_size = atoi(optarg);
|
|
break;
|
|
case ARG_PATH:
|
|
path = optarg;
|
|
break;
|
|
case ARG_SCSI:
|
|
use_scsi = 1;
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (optind < argc)
|
|
image = argv[optind];
|
|
|
|
if (path && tune_mask)
|
|
{
|
|
fprintf(stderr, "ERROR: you cannot supply second path and size, offset or drive number\n");
|
|
usage(argc, argv);
|
|
return 1;
|
|
}
|
|
|
|
if (image == NULL)
|
|
{
|
|
fprintf(stderr, "ERROR: Missing filename to apply tuning\n");
|
|
usage(argc, argv);
|
|
return 1;
|
|
}
|
|
|
|
if (path)
|
|
ret = first_tune_path( image, path);
|
|
else if (use_scsi)
|
|
ret = first_tune_scsi( image, drive_num, second_offset, second_size);
|
|
else {
|
|
second_offset = (second_offset + 0x1FF) & 0xFFFFFE00;
|
|
second_size = (second_size + 0x1FF) & 0xFFFFFE00;
|
|
ret = first_tune( image, tune_mask, drive_num, second_offset, second_size);
|
|
}
|
|
switch(ret)
|
|
{
|
|
case 0:
|
|
break;
|
|
case EEMILE_CANNOT_WRITE_FIRST:
|
|
fprintf(stderr, "ERROR: cannot write to file\n");
|
|
break;
|
|
case EEMILE_UNKNOWN_FIRST:
|
|
fprintf(stderr, "ERROR: unknown file format\n");
|
|
break;
|
|
default:
|
|
fprintf(stderr, "ERROR: unknowm error :-P\n");
|
|
break;
|
|
}
|
|
|
|
return ret;
|
|
}
|