/* lilo.c - LILO command-line parameter processing */

/* Copyright 1992-1998 Werner Almesberger. See file COPYING for details. */

#include <stdlib.h>

#include <unistd.h>

#include <stdio.h>

#include <string.h>

#include <fcntl.h>

#include <errno.h>

#include <sys/stat.h>

#include "config.h"

#include "common.h"

#include "lilo.h"

#include "temp.h"

#include "device.h"

#include "geometry.h"

#include "map.h"

#include "bsect.h"

#include "cfg.h"

#include "identify.h"

#include "partition.h"

#include "md-int.h"

#define S2(x) #x

#define S(x) S2(x)

#define MAX_TOKEN 200

static void show_other(int fd)

{

BOOT_SECTOR buf[SETUPSECS-1];

const unsigned char *drvmap;

const unsigned char *prtmap;

if (read(fd,buf,sizeof(buf)) != sizeof(buf))

die("Read on map file failed (access conflict ?)");

if (!strncmp(buf[0].par_c.signature-4,"LILO",4)) {

printf(" Pre-21 signature (0x%02x,0x%02x,0x%02x,0x%02x)\n",

buf[0].par_c.signature[0],buf[0].par_c.signature[1],

buf[0].par_c.signature[2],buf[0].par_c.signature[3]);

return;

}

if (strncmp(buf[0].par_c.signature,"LILO",4)) {

printf(" Bad signature (0x%02x,0x%02x,0x%02x,0x%02x)\n",

buf[0].par_c.signature[0],buf[0].par_c.signature[1],

buf[0].par_c.signature[2],buf[0].par_c.signature[3]);

return;

}

drvmap = ((unsigned char *) buf+buf[0].par_c.drvmap);

prtmap = drvmap+2*(DRVMAP_SIZE+1);

while (drvmap[0] && drvmap[1]) {

printf(" BIOS drive 0x%02x is mapped to 0x%02x\n",drvmap[0],

drvmap[1]);

drvmap += 2;

}

while (prtmap[0] && prtmap[1]) {

printf(" BIOS drive 0x%02x, offset 0x%x: 0x%02x -> 0x%02x\n",

prtmap[0],prtmap[1]+PART_TABLE_OFFSET,prtmap[2],prtmap[3]);

prtmap += 4;

}

}

static void show_images(char *map_file)

{

DESCR_SECTORS descrs;

BOOT_SECTOR boot;

GEOMETRY geo;

SECTOR_ADDR addr[4];

char buffer[SECTOR_SIZE];

char *name;

int fd,image,i;

int tsecs;

unsigned short checksum,flags;

fd = geo_open(&geo,map_file,O_RDONLY);

if (read(fd,buffer,SECTOR_SIZE) != SECTOR_SIZE)

die("read %s: %s",map_file,strerror(errno));

if (read(fd,(char *) &descrs,sizeof(descrs)) != sizeof(descrs))

die("read %s: %s",map_file,strerror(errno));

if (verbose > 0) {

bsect_read(cfg_get_strg(cf_options,"boot"),&boot);

printf("Global settings:\n");

tsecs = (boot.par_1.delay*55+50)/100;

printf(" Delay before booting: %d.%d seconds\n",tsecs/10,tsecs % 10);

if (boot.par_1.timeout == 0xffff) printf(" No command-line timeout\n");

else {

tsecs = (boot.par_1.timeout*55+50)/100;

printf(" Command-line timeout: %d.%d seconds\n",tsecs/10,

tsecs % 10);

}

if (boot.par_1.prompt) printf(" Always enter boot prompt\n");

else printf(" Enter boot prompt only on demand\n");

if (!boot.par_1.port) printf(" Serial line access is disabled\n");

else printf(" Boot prompt can be accessed from COM%d\n",

boot.par_1.port);

if (!boot.par_1.msg_len) printf(" No message for boot prompt\n");

else printf(" Boot prompt message is %d bytes\n",boot.par_1.msg_len);

if (*(unsigned short *) buffer != DC_MAGIC || !buffer[2])

printf(" No default boot command line\n");

else printf(" Default boot command line: \"%s\"\n",buffer+2);

printf("Images:\n");

}

for (image = 0; image < MAX_IMAGES; image++)

if (*(name = descrs.d.descr[image].name)) {

printf("%s%-" S(MAX_IMAGE_NAME) "s%s",verbose > 0 ? " " : "",name,

image ? " " : " *");

if (verbose > 1) {

printf(" <dev=0x%02x,hd=%d,cyl=%d,sct=%d>",

descrs.d.descr[image].start.device,

descrs.d.descr[image].start.head,

descrs.d.descr[image].start.track,

descrs.d.descr[image].start.sector);

}

printf("\n");

if (verbose > 0) {

flags = descrs.d.descr[image].flags;

if (!*descrs.d.descr[image].password)

printf(" No password\n");

else printf(" Password is required for %s\n",flags &

FLAG_RESTR ? "specifying options" : "booting this image");

printf(" Boot command-line %s be locked\n",flags &

FLAG_LOCK ? "WILL" : "won't");

printf(" %single-key activation\n",flags & FLAG_SINGLE ?

"S" : "No s");

if (flags & FLAG_KERNEL) {

#ifdef NORMAL_VGA

if (!(flags & FLAG_VGA))

printf(" VGA mode is taken from boot image\n");

else {

printf(" VGA mode: ");

switch (descrs.d.descr[image].vga_mode) {

case NORMAL_VGA:

printf("NORMAL\n");

break;

case EXTENDED_VGA:

printf("EXTENDED\n");

break;

case ASK_VGA:

printf("ASK\n");

break;

default:

printf("%d (0x%04x)\n",

descrs.d.descr[image].vga_mode,

descrs.d.descr[image].vga_mode);

}

}

#endif

if (!descrs.d.descr[image].start_page)

printf(" Kernel is loaded \"low\"\n");

else printf(" Kernel is loaded \"high\", at 0x%08lx\n",

(unsigned long) descrs.d.descr[image].start_page*

getpagesize());

if (!*(unsigned long *) descrs.d.descr[image].rd_size)

printf(" No initial RAM disk\n");

else printf(" Initial RAM disk is %ld bytes\n",

*(unsigned long *) descrs.d.descr[image].rd_size);

}

if (!geo_find(&geo,descrs.d.descr[image].start)) {

printf(" Map sector not found\n");

continue;

}

if (read(fd,addr,4*sizeof(SECTOR_ADDR)) !=

4*sizeof(SECTOR_ADDR))

die("Read on map file failed (access conflict ?)");

if (!geo_find(&geo,addr[0]))

printf(" Fallback sector not found\n");

else {

if (read(fd,buffer,SECTOR_SIZE) != SECTOR_SIZE)

die("Read on map file failed (access conflict ?)");

if (*(unsigned short *) buffer != DC_MAGIC)

printf(" No fallback\n");

else printf(" Fallback: \"%s\"\n",buffer+2);

}

if (flags & FLAG_KERNEL)

if (!geo_find(&geo,addr[1]))

printf(" Options sector not found\n");

else {

if (read(fd,buffer,SECTOR_SIZE) != SECTOR_SIZE)

die("Read on map file failed (access conflict ?)");

printf(" Options: \"%s\"\n",buffer);

}

else if (geo_find(&geo,addr[3])) show_other(fd);

else printf(" Image data not found\n");

}

}

(void) close(fd);

checksum = INIT_CKS;

for (i = 0; i < sizeof(descrs)/sizeof(unsigned short); i++)

checksum ^= ((unsigned short *) &descrs)[i];

if (!checksum) exit(0);

fflush(stdout);

fprintf(stderr,"Checksum error\n");

exit(1);

}

static void usage(char *name)

{

char *here;

here = strrchr(name,'/');

if (here) name = here+1;

fprintf(stderr,"usage: %s [ -C config_file ] -q [ -m map_file ] "

"[ -v ... ]\n",name);

fprintf(stderr,"%7s%s [ -C config_file ] [ -b boot_device ] [ -c ] "

"[ -l ]\n","",name);

fprintf(stderr,"%12s[ -i boot_sector ] [ -m map_file ] [ -d delay ]\n","");

fprintf(stderr,"%12s[ -v ... ] [ -t ] [ -s save_file | -S save_file ]\n",

"");

fprintf(stderr,"%12s[ -P fix | -P ignore ] [ -r root_dir ] [ -w ]\n","");

fprintf(stderr,"%7s%s [ -C config_file ] [ -m map_file ] "

"-R [ word ... ]\n","",name);

fprintf(stderr,"%7s%s [ -C config_file ] -I name [ options ]\n","",name);

fprintf(stderr,"%7s%s [ -C config_file ] [ -s save_file ] "

"-u | -U [ boot_device ]\n","",name);

fprintf(stderr,"%7s%s -V\n\n","",name);

exit(1);

}

int main(int argc,char **argv)

{

char *name,*config_file,*reboot_arg,*identify,*ident_opt,*new_root;

char *uninst_dev;

int query,more,version,uninstall,validate,do_md_install,pass;

BOOT_SECTOR dummy;

IMAGE_DESCR dummy2;

struct stat st;

int fd, md_fd;

md_array_info_t md_array_info;

md_disk_info_t md_disk_info;

char md_boot_name[MAX_TOKEN+1];

char md_boot_map[MAX_TOKEN+1];

DT_ENTRY md_disk;

DT_ENTRY *disk;

config_file = DFL_CONFIG;

reboot_arg = identify = ident_opt = new_root = uninst_dev = NULL;

pass = do_md_install = query = version = uninstall = validate = 0;

name = *argv++;

argc--;

cfg_init(cf_options);

while (argc && **argv == '-') {

argc--;

if (argv[0][2]) usage(name);

switch ((*argv++)[1]) {

case 'b':

if (!argc) usage(name);

cfg_set(cf_options,"boot",*argv++,NULL);

argc--;

break;

case 'c':

cfg_set(cf_options,"compact",NULL,NULL);

compact = 1;

break;

case 'd':

if (!argc) usage(name);

cfg_set(cf_options,"delay",*argv++,NULL);

argc--;

break;

case 'D':

if (!argc) usage(name);

cfg_set(cf_options,"default",*argv++,NULL);

argc--;

break;

case 'f':

if (!argc) usage(name);

cfg_set(cf_options,"disktab",*argv++,NULL);

argc--;

break;

case 'l':

cfg_set(cf_options,"linear",NULL,NULL);

linear = 1;

break;

case 'm':

if (!argc) usage(name);

cfg_set(cf_options,"map",*argv++,NULL);

argc--;

break;

case 'i':

if (!argc) usage(name);

cfg_set(cf_options,"install",*argv++,NULL);

argc--;

break;

case 'I':

if (!argc) usage(name);

identify = *argv++;

if (--argc) {

ident_opt = *argv++;

argc--;

}

break;

case 'X':

printf("-DIMAGES=%d -DCODE_START_1=%d -DCODE_START_2=%d "

"-DDESCR_SIZE=%d -DDSC_OFF=%d -DDSC_OFF2=%d -DDFCMD_OFF=%d "

"-DMSG_OFF=%d -DFLAGS_OFF=%d\n",MAX_IMAGES,

sizeof(BOOT_PARAMS_1),sizeof(BOOT_PARAMS_2),

sizeof(IMAGE_DESCR),

(void *) &dummy.par_1.descr[0]-(void *) &dummy,

(void *) &dummy.par_1.descr[1]-(void *) &dummy,

(void *) &dummy.par_1.descr[2]-(void *) &dummy,

(void *) &dummy.par_1.msg_len-(void *) &dummy,

(void *) &dummy2.flags-(void *) &dummy2);

exit(0);

case 'C':

if (!argc) usage(name);

config_file = *argv++;

argc--;

break;

case 'S':

if (!argc) usage(name);

cfg_set(cf_options,"force-backup",*argv++,NULL);

argc--;

break;

case 's':

if (!argc) usage(name);

cfg_set(cf_options,"backup",*argv++,NULL);

argc--;

break;

case 'P':

if (!argc) usage(name);

if (!strcmp(*argv,"fix"))

cfg_set(cf_options,"fix-table",NULL,NULL);

else if (!strcmp(*argv,"ignore"))

cfg_set(cf_options,"ignore-table",NULL,NULL);

else usage(name);

argv++;

argc--;

break;

case 'q':

query = 1;

break;

case 'r':

if (!argc) usage(name);

new_root = *argv++;

argc--;

break;

case 'R':

if (!argc) reboot_arg = "";

else while (argc) {

if (!reboot_arg)

*(reboot_arg = alloc(strlen(*argv)+1)) = 0;

else strcat(reboot_arg = ralloc(reboot_arg,

strlen(reboot_arg)+strlen(*argv)+2)," ");

strcat(reboot_arg,*argv++);

argc--;

}

break;

case 't':

test = 1;

break;

case 'u':

validate = 1;

/* fall through */

case 'U':

uninstall = 1;

if (argc) {

if (argc-- > 1) usage(name);

uninst_dev = *argv;

}

break;

case 'v':

verbose++;

break;

case 'V':

version = 1;

break;

case 'w':

cfg_set(cf_options,"nowarn",NULL,NULL);

nowarn = 1;

break;

default:

usage(name);

}

}

if (!new_root) new_root = getenv("ROOT");

if (new_root && *new_root) {

if (chroot(new_root) < 0) die("chroot %s: %s",new_root,strerror(errno));

if (chdir("/") < 0) die("chdir /: %s",strerror(errno));

}

if (atexit(temp_remove)) die("atexit() failed");

if (verbose > 0 || version) {

printf("LILO version 21%s",version ? "\n" : "");

if (version) return 0;

printf(", Copyright 1992-1998 Werner Almesberger\n\n");

}

#if 0

if (((install || test || boot_device || disktab_file || compact) && !argc)

|| (compact && linear && 0)) usage(name);

#endif

if (!nowarn && compact && linear)

fprintf(stderr,"Warning: COMPACT may conflict with LINEAR on some "

"systems\n");

preload_types();

fd = cfg_open(config_file);

more = cfg_parse(cf_options);

if (!nowarn) {

if (fstat(fd,&st) < 0) {

fprintf(stderr,"fstat %s: %s\n",config_file,strerror(errno));

exit(1);

}

if (S_ISREG(st.st_mode)) {

if (st.st_uid)

fprintf(stderr,"Warning: %s should be owned by root\n",

config_file);

else if (st.st_mode & (S_IWGRP | S_IWOTH))

fprintf(stderr,"Warning: %s should be writable only for "

"root\n",config_file);

else if ((cfg_get_strg(cf_all,"password") || cfg_get_strg(

cf_options,"password")) && (st.st_mode & (S_IRGRP |

S_IROTH)))

fprintf(stderr,"Warning: %s should be readable only "

"for root if using PASSWORD\n",config_file);

}

}

preload_dev_cache();

if (identify) identify_image(identify,ident_opt);

if (strncmp("/dev/md",cfg_get_strg(cf_options,"boot"),7) == 0) {

if ((md_fd=open(cfg_get_strg(cf_options,"boot"),O_NOACCESS)) < 0)

die("Unable to open %s",cfg_get_strg(cf_options,"boot"));

if (fstat(md_fd,&st) < 0)

die("Unable to stat %s",cfg_get_strg(cf_options,"boot"));

if (!S_ISBLK(st.st_mode))

die("%s is not a block device",cfg_get_strg(cf_options,"boot"));

if (ioctl(md_fd,GET_ARRAY_INFO,&md_array_info) < 0)

die("Unable to get RAID info on %s",cfg_get_strg(cf_options,"boot"));

if ((md_array_info.major_version == 0) && (md_array_info.minor_version < 90))

die("Raid versions < 0.90 are not supported");

if (md_array_info.level != 1)

die("Only RAID1 devices are supported as boot devices");

do_md_install = 1;

strcpy(md_boot_name,cfg_get_strg(cf_options,"boot"));

if (cfg_get_strg(cf_options,"map"))

strcpy(md_boot_map,cfg_get_strg(cf_options,"map"));

else

strcpy(md_boot_map,MAP_FILE);

md_disk.device = (MD_MAJOR << 8) | md_array_info.md_minor;

md_disk.bios = 0x80;

md_disk.next = disktab;

disktab = &md_disk;

}

while( (pass == 0) || (do_md_install && (pass < md_array_info.nr_disks)) ) {

if(do_md_install) {

GEOMETRY geo;

DEVICE dev;

int device,disk_fd;

char new_name[MAX_TOKEN+1];

if(pass > 0) {

close(fd);

cfg_init(cf_options);

fd = cfg_open(config_file);

more = cfg_parse(cf_options);

}

md_disk_info.number = pass;

if (ioctl(md_fd,GET_DISK_INFO,&md_disk_info) < 0)

die("main: GET_DISK_INFO: %s", strerror(errno));

device = (md_disk_info.major << 8) | md_disk_info.minor;

disk_fd = dev_open(&dev,device,O_NOACCESS);

if (md_disk_info.state == MD_DISK_FAULTY) {

printf("disk %s marked as faulty, skipping\n",dev.name);

pass++;

continue;

}

geo_query_dev(&geo,device,1);

disk = alloc_t(DT_ENTRY);

disk->bios = 0x80;

disk->device = device & 0xfff0;

disk->sectors = geo.sectors;

disk->heads = geo.heads;

disk->cylinders = geo.cylinders;

disk->start = geo.start;

disk->next = disktab;

disktab = disk;

if (cfg_get_strg(cf_options,"boot")) cfg_unset(cf_options,"boot");

if (cfg_get_strg(cf_options,"map")) cfg_unset(cf_options,"map");

strncpy(new_name,dev.name,8);

new_name[8] = '\0';

cfg_set(cf_options,"boot",new_name,NULL);

snprintf(new_name,MAX_TOKEN,"%s.%04x",md_boot_map,device);

cfg_set(cf_options,"map",new_name,NULL);

printf("boot = %s, map = %s\n", cfg_get_strg(cf_options,"boot"),

cfg_get_strg(cf_options,"map"));

md_disk.sectors = geo.sectors;

md_disk.heads = geo.heads;

md_disk.cylinders = geo.cylinders;

md_disk.start = geo.start;

}

pass++;

if (uninstall)

bsect_uninstall(uninst_dev ? uninst_dev : cfg_get_strg(cf_options,

"boot"),cfg_get_strg(cf_options,"backup"),validate);

compact = cfg_get_flag(cf_options,"compact");

linear = cfg_get_flag(cf_options,"linear");

nowarn = cfg_get_flag(cf_options,"nowarn");

if (cfg_get_strg(cf_options,"verbose"))

verbose += to_number(cfg_get_strg(cf_options,"verbose"));

if (reboot_arg) {

map_patch_first(cfg_get_strg(cf_options,"map") ? cfg_get_strg(

cf_options,"map") : MAP_FILE,reboot_arg);

exit(0);

}

if (argc) usage(name);

geo_init(cfg_get_strg(cf_options,"disktab"));

if (query)

show_images(!cfg_get_strg(cf_options,"map") ? MAP_FILE :

cfg_get_strg(cf_options,"map"));

bsect_open(cfg_get_strg(cf_options,"boot"),cfg_get_strg(cf_options,"map") ?

cfg_get_strg(cf_options,"map") : MAP_FILE,cfg_get_strg(cf_options,

"install"),cfg_get_strg(cf_options,"delay") ? to_number(cfg_get_strg(

cf_options,"delay")) : 0,cfg_get_strg(cf_options,"timeout") ?

to_number(cfg_get_strg(cf_options,"timeout")) : -1);

if (more) {

cfg_init(cf_top);

if (cfg_parse(cf_top)) cfg_error("Syntax error");

}

if (!bsect_number()) die("No images have been defined.");

check_fallback();

if (!test)

if (cfg_get_strg(cf_options,"force-backup"))

bsect_update(cfg_get_strg(cf_options,"force-backup"),1);

else bsect_update(cfg_get_strg(cf_options,"backup"),0);

else {

bsect_cancel();

fprintf(stderr,"The boot sector and the map file have *NOT* been "

"altered.\n");

}

}

return 0;

}