freebsd-skq/sbin/sysinstall/sysinstall.c

832 lines
21 KiB
C

/*
#define DEBUG
* Copyright (c) 1994, Paul Richards.
*
* All rights reserved.
*
* This software may be used, modified, copied, distributed, and
* sold, in both source and binary form provided that the above
* copyright and these terms are retained, verbatim, as the first
* lines of this file. Under no circumstances is the author
* responsible for the proper functioning of this software, nor does
* the author assume any responsibility for damages incurred with
* its use.
*/
#include <dialog.h>
#include <fcntl.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <stdarg.h>
#include <sys/types.h>
#include <sys/errno.h>
#include <sys/disklabel.h>
#include <sys/ioctl.h>
#include <sys/param.h>
#include <sys/mount.h>
#include <sys/reboot.h>
#include <sys/stat.h>
#include <sys/utsname.h>
#include <sys/wait.h>
#include <ufs/ffs/fs.h>
#include <machine/console.h>
#include "mbr.h"
#include "bootarea.h"
#include "sysinstall.h"
struct disklabel *avail_disklabels;
int *avail_fds;
unsigned char **options;
unsigned char **avail_disknames;
unsigned char *scratch;
unsigned char *errmsg;
unsigned char *bootblocks;
struct mbr *mbr;
struct utsname utsname;
unsigned char *title = utsname.sysname;
struct sysinstall *sysinstall;
struct sysinstall *sequence;
int no_disks = 0;
int inst_disk = 0;
int inst_part = 0;
int custom_install;
int dialog_active = 0;
void exit_sysinstall();
void abort_installation(char *);
void exit_prompt();
void fatal(char *);
extern char *part_type(int);
extern int disk_size(int);
/* To make the binary as small as possible these should be malloc'd */
char selection[30];
void
abort_installation(char *prompt)
{
strcpy(scratch, prompt);
strcat(scratch,"\n\n Do you wish to abort the installation ?");
if (!dialog_yesno("Abort installation ?",scratch,10,75))
exit_prompt();
dialog_clear();
}
void
exit_prompt()
{
sprintf(scratch,"Are you sure you want to exit sysinstall?");
if (!dialog_yesno("Exit sysinstall", scratch, 10, 75))
exit_sysinstall();
dialog_clear();
}
int
alloc_memory()
{
int i;
scratch = (char *)calloc(SCRATCHSIZE, sizeof(char));
if (!scratch)
return(-1);
errmsg = (char *)calloc(ERRMSGSIZE, sizeof(char));
if (!errmsg)
return(-1);
avail_disklabels =
(struct disklabel *)calloc(MAX_NO_DISKS, sizeof(struct disklabel));
if (!avail_disklabels)
return(-1);
avail_fds = (int *) calloc(MAX_NO_DISKS, sizeof(int));
if (!avail_fds)
return(-1);
avail_disknames = (unsigned char **) calloc(MAX_NO_DISKS, sizeof(char *));
if (!avail_disknames)
return(-1);
for (i = 0; i < MAX_NO_DISKS; i++) {
avail_disknames[i] = (char *)calloc(15, sizeof(char));
if (!avail_disknames[i])
return(-1);
}
options = (unsigned char **)calloc(MAX_NO_DISKS, sizeof(char *));
if (!options)
return(-1);
for (i = 0; i < MAX_NO_DISKS; i++) {
options[i] = (char *)calloc(100, sizeof(char));
if (!options[i])
return(-1);
}
mbr = (struct mbr *)malloc(sizeof(struct mbr));
if (!mbr)
return(-1);
bootblocks = (char *)malloc(BBSIZE);
if (!bootblocks)
return(-1);
sysinstall = (struct sysinstall *)malloc(sizeof(struct sysinstall));
if (!sysinstall)
return(-1);
sequence = (struct sysinstall *)malloc(sizeof(struct sysinstall));
if (!sequence)
return(-1);
return(0);
}
void
free_memory()
{
int i;
free(scratch);
free(errmsg);
free(avail_disklabels);
free(avail_fds);
for (i = 0; i < MAX_NO_DISKS; i++)
free(avail_disknames[i]);
free(avail_disknames);
for (i = 0; i < MAX_NO_DISKS; i++)
free(options[i]);
free(options);
free(mbr);
free(bootblocks);
free(sysinstall);
free(sequence);
}
void
exit_sysinstall()
{
if (getpid() == 1) {
if (reboot(RB_AUTOBOOT) == -1)
if (dialog_active)
while (1)
dialog_msgbox("Exit sysinstall",
"Reboot failed -- hit reset",
10, 75, 20);
else {
fprintf(stderr, "Reboot failed -- hit reset");
while (1);
}
} else {
free_memory();
if (dialog_active)
end_dialog();
exit(0);
}
}
void
fatal(char *errmsg)
{
if (dialog_active)
dialog_msgbox("Fatal Error -- Aborting installation",
errmsg, 10, 75, 20);
else
fprintf(stderr, "Fatal Error -- Aborting installation:\n%s\n", errmsg);
exit_sysinstall();
}
void
query_disks()
{
int i;
char disk[15];
char diskname[5];
struct stat st;
int fd;
no_disks = 0;
for (i = 0; i < 10; i++) {
sprintf(diskname,"wd%d",i);
sprintf(disk,"/dev/r%sd",diskname);
if ((stat(disk, &st) == 0) && (st.st_mode & S_IFCHR))
if ((fd = open(disk, O_RDWR)) != -1) {
avail_fds[no_disks] = fd;
bcopy(diskname, avail_disknames[no_disks], strlen(diskname));
if (ioctl(fd, DIOCGDINFO, &avail_disklabels[no_disks++]) == -1)
no_disks--;
}
}
for (i=0;i<10;i++) {
sprintf(diskname,"sd%d",i);
sprintf(disk,"/dev/r%sd",diskname);
if ((stat(disk, &st) == 0) && (st.st_mode & S_IFCHR))
if ((fd = open(disk, O_RDWR)) != -1) {
avail_fds[no_disks] = fd;
bcopy(diskname, avail_disknames[no_disks], strlen(diskname));
if (ioctl(fd, DIOCGDINFO, &avail_disklabels[no_disks++]) == -1)
no_disks--;
}
}
}
int
select_disk()
{
int i;
int valid;
do {
valid = 1;
sprintf(scratch,"There are %d disks available for installation: ",no_disks);
for (i=0;i<no_disks;i++) {
sprintf(options[(i*2)], "%d",i+1);
sprintf(options[(i*2)+1], "%s, (%dMb) -> %s",avail_disklabels[i].d_typename,disk_size(i),avail_disknames[i]);
}
if (dialog_menu("FreeBSD Installation", scratch, 10, 75, 5, no_disks, options, selection)) {
sprintf(scratch,"You did not select a valid disk");
abort_installation(scratch);
valid = 0;
}
dialog_clear();
} while (!valid);
return(atoi(selection) - 1);
}
int
select_partition(int disk)
{
int valid;
int i;
int choice;
do {
valid = 1;
sprintf(scratch,"Select one of the following areas to install to:");
sprintf(options[0], "%d", 0);
sprintf(options[1], "%s, (%dMb)", "Install to entire disk",
disk_size(disk));
for (i=0; i < NDOSPART; i++) {
sprintf(options[(i*2)+2], "%d",i+1);
sprintf(options[(i*2)+3], "%s, (%ldMb)",
part_type(mbr->dospart[i].dp_typ),
mbr->dospart[i].dp_size * 512 / (1024 * 1024));
}
if (dialog_menu(title,
scratch, 10, 75, 5, 5, options, selection)) {
sprintf(scratch,"You did not select a valid partition");
abort_installation(scratch);
valid = 0;
}
dialog_clear();
choice = atoi(selection);
if (!choice)
if (dialog_yesno(title, "Installing to the whole disk will erase all its present data.\n\nAre you sure you want to do this?", 10, 75))
valid = 0;
dialog_clear();
} while (!valid);
return(atoi(selection) - 1);
}
int
exec(char *cmd, char *args, ...)
{
int pid, w, status;
char **argv = NULL;
int arg = 0;
int no_args = 0;
va_list ap;
struct stat dummy;
if (stat(cmd, &dummy) == -1) {
sprintf(errmsg, "Executable %s does not exist\n", cmd);
return(-1);
}
va_start(ap, args);
do {
if (arg == no_args) {
no_args += 10;
if (!(argv = realloc(argv, no_args * sizeof(char *)))) {
sprintf(errmsg, "Failed to allocate memory during exec of %s\n", cmd);
return(-1);
}
if (arg == 0)
argv[arg++] = (char *)args;
}
} while ((argv[arg++] = va_arg(ap, char *)));
va_end(ap);
if ((pid = fork()) == 0) {
execv(cmd, argv);
exit(1);
}
while ((w = wait(&status)) != pid && w != -1)
;
free(argv);
if (w == -1) {
sprintf(errmsg, "Child process %s terminated abnormally\n", cmd);
return(-1);
}
return(0);
}
int
set_termcap()
{
char *term;
term = getenv("TERM");
if (term == NULL) {
int color_display;
if (setenv("TERMCAP", "/etc/termcap.small", 1) < 0)
return -1;
if (ioctl(STDERR_FILENO, GIO_COLOR, &color_display) < 0) {
char buf[64];
int len;
/* serial console */
fprintf(stderr, "Enter your terminal type (must be present in /etc/termcap.small): ");
if (fgets(buf, sizeof(buf), stdin) == NULL)
return -1;
len = strlen(buf);
if (len > 0 && buf[len-1] == '\n')
buf[len-1] = '\0';
if (setenv("TERM", buf, 1) < 0)
return -1;
} else if (color_display) {
/* color console */
if (setenv("TERM", "cons25", 1) < 0)
return -1;
} else {
/* mono console */
if (setenv("TERM", "cons25-m", 1) < 0)
return -1;
}
}
return 0;
}
int
read_status(char *file, struct sysinstall *sysinstall)
{
FILE *fd;
if (!(fd = fopen(file, "r"))) {
sprintf(errmsg, "Couldn't open status file %s for reading\n", file);
return(-1);
}
if (fscanf(fd, "Root device: %s\n", sysinstall->root_dev) == -1) {
sprintf(errmsg, "Failed to read root device from file %s\n", file);
return(-1);
}
if (fscanf(fd, "Installation media: %s\n", sysinstall->media) == -1) {
sprintf(errmsg, "Failed to read installation media from file %s\n", file);
return(-1);
}
if (fscanf(fd, "Installation status: %d\n", &sysinstall->status) == -1) {
sprintf(errmsg, "Status file %s has invalid format\n", file);
return(-1);
}
if (fscanf(fd, "Sequence name: %s\n", sysinstall->seq_name) == -1) {
sprintf(errmsg, "Failed to read sequence name from file %s\n", file);
return(-1);
}
if (fscanf(fd, "Sequence number: %d of %d\n",
&sysinstall->seq_no, &sysinstall->seq_size) == -1) {
sprintf(errmsg, "Failed to read sequence information from file %s\n", file);
return(-1);
}
if (fscanf(fd, "Archive: %s\n", sysinstall->archive) == -1) {
sprintf(errmsg, "Failed to read archive name from file %s\n", file);
return(-1);
}
if (fclose(fd) != 0) {
sprintf(errmsg, "Couldn't close file %s after reading status\n", file);
return(-1);
}
return(0);
}
int
write_status(char *file, struct sysinstall *sysinstall)
{
FILE *fd;
if (!(fd = fopen(file, "w"))) {
sprintf(errmsg, "Couldn't open status file %s for writing\n", file);
return(-1);
}
if (fprintf(fd, "Root device: %s\n", sysinstall->root_dev) == -1) {
sprintf(errmsg, "Failed to write root device to file %s\n", file);
return(-1);
}
if (fprintf(fd, "Installation media: %s\n", sysinstall->media) == -1) {
sprintf(errmsg, "Failed to write installation media to file %s\n", file);
return(-1);
}
if (fprintf(fd, "Installation status: %d\n", sysinstall->status) == -1) {
sprintf(errmsg, "Failed to write status information to file %s\n", file);
return(-1);
}
if (fprintf(fd, "Sequence name: %s\n", sysinstall->seq_name) == -1) {
sprintf(errmsg, "Failed to write sequence name to file %s\n", file);
return(-1);
}
if (fprintf(fd, "Sequence number: %d of %d\n",
sysinstall->seq_no, sysinstall->seq_size) == -1) {
sprintf(errmsg, "Failed to write sequence information to file %s\n", file);
return(-1);
}
if (fprintf(fd, "Archive: %s\n", sysinstall->archive) == -1) {
sprintf(errmsg, "Failed to write archive name to file %s\n", file);
return(-1);
}
if (fclose(fd) != 0) {
sprintf(errmsg, "Couldn't close status file %s after status update\n", file);
return(-1);
}
return(0);
}
int
load_floppy(char *device, int seq_no)
{
struct ufs_args ufsargs;
ufsargs.fspec = device;
if (mount(MOUNT_UFS,"/mnt", 0, (caddr_t) &ufsargs) == -1) {
sprintf(errmsg, "Failed to mount floppy %s: %s\n",scratch, strerror(errno));
return(-1);
}
strcpy(scratch, "/mnt/");
strcat(scratch, STATUSFILE);
if (read_status(scratch, sequence) == -1) {
if (unmount("/mnt", 0) == -1) {
strcat(errmsg, "Error unmounting floppy: ");
strcat(errmsg, strerror(errno));
fatal(errmsg);
}
return(-1);
}
if ((bcmp(sequence->seq_name, sysinstall->seq_name,
sizeof(sequence->seq_name)) != 0) || (sequence->seq_no != seq_no)) {
sprintf(errmsg, "Mounted floppy is not the one expected\n");
if (unmount("/mnt", 0) == -1) {
strcat(errmsg, "Error unmounting floppy: ");
strcat(errmsg, strerror(errno));
fatal(errmsg);
}
return(-1);
}
return(0);
}
void
stage1()
{
int i;
int ok = 0;
int ready = 0;
struct ufs_args ufsargs;
while (!ready) {
ready = 1;
query_disks();
inst_disk = select_disk();
#ifdef DEBUG
read_mbr(avail_fds[inst_disk], mbr);
show_mbr(mbr);
#endif
if (read_mbr(avail_fds[inst_disk], mbr) == -1) {
sprintf(scratch, "The following error occured will trying to read the master boot record:\n\n%s\n\nIn order to install FreeBSD a new master boot record will have to be written which will mean all current data on the hard disk will be lost.", errmsg);
ok = 0;
while (!ok) {
abort_installation(scratch);
if (!dialog_yesno(title, "Are you sure you wish to proceed?",
10, 75)) {
dialog_clear();
clear_mbr(mbr);
ok = 1;
}
}
}
if (custom_install)
if (!dialog_yesno(title, "Do you wish to edit the DOS partition table?",
10, 75)) {
dialog_clear();
edit_mbr(mbr, &avail_disklabels[inst_disk]);
}
inst_part = select_partition(inst_disk);
ok = 0;
while (!ok) {
if (build_mbr(mbr, &avail_disklabels[inst_disk]))
ok = 1;
else {
sprintf(scratch, "The DOS partition table is inconsistent.\n\n%s\n\nDo you wish to edit it by hand?", errmsg);
if (!dialog_yesno(title, scratch, 10, 75)) {
edit_mbr(mbr, &avail_disklabels[inst_disk]);
dialog_clear();
} else {
abort_installation("");
ok = 1;
ready = 0;
}
}
}
default_disklabel(&avail_disklabels[inst_disk],
mbr->dospart[inst_part].dp_size,
mbr->dospart[inst_part].dp_start);
build_bootblocks(&avail_disklabels[inst_disk]);
if (ready) {
if (dialog_yesno(title, "We are now ready to format the hard disk for FreeBSD.\n\nSome or all of the disk will be overwritten during this process.\n\nAre you sure you wish to proceed ?", 10, 75)) {
abort_installation("");
ready = 0;
}
dialog_clear();
}
}
/* Write master boot record and bootblocks */
write_mbr(avail_fds[inst_disk], mbr);
write_bootblocks(avail_fds[inst_disk],
mbr->dospart[inst_part].dp_start,
avail_disklabels[inst_disk].d_bbsize);
#ifdef DEBUG
read_mbr(avail_fds[inst_disk], mbr);
show_mbr(mbr);
#endif
/* close all the open disks */
for (i=0; i < no_disks; i++)
if (close(avail_fds[i]) == -1) {
sprintf(errmsg, "Error on closing file descriptors: %s\n",
strerror(errno));
fatal(errmsg);
}
/* newfs the root partition */
strcpy(scratch, avail_disknames[inst_disk]);
strcat(scratch, "a");
if (exec("/sbin/newfs","/sbin/newfs", scratch, 0) == -1)
fatal(errmsg);
/* newfs the /usr partition */
strcpy(scratch, avail_disknames[inst_disk]);
strcat(scratch, "e");
if (exec("/sbin/newfs", "/sbin/newfs", scratch, 0) == -1)
fatal(errmsg);
strcpy(scratch, "/dev/");
strcat(scratch, avail_disknames[inst_disk]);
strcat(scratch, "a");
ufsargs.fspec = scratch;
if (mount(MOUNT_UFS,"/mnt", 0, (caddr_t) &ufsargs) == -1) {
sprintf(errmsg, "Error mounting %s: %s\n",scratch, strerror(errno));
fatal(errmsg);
}
if (mkdir("/mnt/usr",S_IRWXU) == -1) {
sprintf(errmsg, "Couldn't create directory /mnt/usr: %s\n",
strerror(errno));
fatal(errmsg);
}
if (mkdir("/mnt/mnt",S_IRWXU) == -1) {
sprintf(errmsg, "Couldn't create directory /mnt/mnt: %s\n",
strerror(errno));
fatal(errmsg);
}
strcpy(scratch, "/dev/");
strcat(scratch, avail_disknames[inst_disk]);
strcat(scratch, "e");
ufsargs.fspec = scratch;
if (mount(MOUNT_UFS,"/mnt/usr", 0, (caddr_t) &ufsargs) == -1) {
sprintf(errmsg, "Error mounting %s: %s\n",scratch, strerror(errno));
fatal(errmsg);
}
if (exec("/bin/cp","/bin/cp","/kernel","/mnt", 0) == -1) {
sprintf(errmsg, "Couldn't copy /kernel to /mnt: %s\n",strerror(errno));
fatal(errmsg);
}
if (exec("/bin/cp","/bin/cp","/sysinstall","/mnt", 0) == -1) {
sprintf(errmsg, "Couldn't copy /sysinstall to /mnt: %s\n",
strerror(errno));
fatal(errmsg);
}
if (exec("/bin/cp","/bin/cp","-R","/etc","/mnt", 0) == -1) {
sprintf(errmsg, "Couldn't copy /etc to /mnt: %s\n",strerror(errno));
fatal(errmsg);
}
if (exec("/bin/cp","/bin/cp","-R","/sbin","/mnt", 0) == -1) {
sprintf(errmsg, "Couldn't copy /sbin to /mnt: %s\n",strerror(errno));
fatal(errmsg);
}
if (exec("/bin/cp","/bin/cp","-R","/bin","/mnt", 0) == -1) {
sprintf(errmsg, "Couldn't copy /bin to /mnt: %s\n",strerror(errno));
fatal(errmsg);
}
if (exec("/bin/cp","/bin/cp","-R","/dev","/mnt", 0) == -1) {
sprintf(errmsg, "Couldn't copy /dev to /mnt: %s\n",strerror(errno));
fatal(errmsg);
}
if (exec("/bin/cp","/bin/cp","-R","/usr","/mnt", 0) == -1) {
sprintf(errmsg, "Couldn't copy /usr to /mnt: %s\n",strerror(errno));
fatal(errmsg);
}
sysinstall->status = DISK_READY;
bcopy(avail_disknames[inst_disk], sysinstall->root_dev,
strlen(avail_disknames[inst_disk]));
sprintf(scratch, "/mnt/etc/%s", STATUSFILE);
if (write_status(scratch, sysinstall) == -1)
fatal(errmsg);
if (unmount("/mnt/usr", 0) == -1) {
sprintf(errmsg, "Error unmounting /mnt/usr: %s\n", strerror(errno));
fatal(errmsg);
}
if (unmount("/mnt", 0) == -1) {
sprintf(errmsg, "Error unmounting /mnt: %s\n", strerror(errno));
fatal(errmsg);
}
}
void
stage2()
{
int i;
struct ufs_args ufsargs;
ufsargs.fspec = sysinstall->root_dev;
if (mount(MOUNT_UFS,"/", 0, (caddr_t) &ufsargs) == -1) {
sprintf(errmsg, "Failed to mount root read/write: %s\n%s", strerror(errno), ufsargs.fspec);
fatal(errmsg);
}
sprintf(scratch, "Insert floppy %d in drive\n", sysinstall->seq_no + 1);
dialog_msgbox("Stage 2 installation", scratch, 10, 75, 1);
i = load_floppy(sysinstall->media, sysinstall->seq_no + 1);
while (i == -1) {
dialog_msgbox("Stage 2 installation",errmsg, 10, 75, 1);
sprintf(scratch, "Please insert installation floppy %d in the boot drive", sysinstall->seq_no + 1);
dialog_msgbox("Stage 2 installation",scratch, 10, 75, 1);
i = load_floppy(sysinstall->media, sysinstall->seq_no + 1);
};
if (exec("/bin/cp","/bin/cp","/mnt/pax","/bin", 0) == -1) {
sprintf(errmsg, "Couldn't copy /mnt/pax to /bin %s\n",strerror(errno));
fatal(errmsg);
}
if (exec("/bin/pax", "/bin/pax", "-r", "-f", sequence->archive, 0) == -1) {
sprintf(errmsg, "Failed to extract from archive file %s\n", sequence->archive);
fatal(errmsg);
}
sysinstall->status = INSTALLED_BASE;
sprintf(scratch, "/etc/%s", STATUSFILE);
if (write_status(scratch, sysinstall) == -1)
fatal(errmsg);
if (unmount("/mnt", 0) == -1) {
strcat(errmsg, "Error unmounting floppy: ");
strcat(errmsg, strerror(errno));
fatal(errmsg);
}
}
/*
* This is the overall plan: (phk's version)
*
* If (pid == 1)
* reopen stdin, stdout, stderr, and do various other magic.
*
* If (file exists /this_is_boot.flp)
* stage0:
* present /COPYRIGHT
* present /README
* stage1:
* Ask about diskallocation and do the fdisk/disklabel stunt.
* stage2:
* Do newfs, mount and copy over a minimal world.
* make /mnt/etc/fstab. Install ourself as /mnt/sbin/init
* Else
* stage3:
* Read cpio.flp and fiddle around with the bits a bit.
* stage4:
* Read bin-tarballs:
* Using ftp
* Using NFS (?)
* Using floppy
* Using tape
* Using shell-prompt
* stage5:
* Extract bin-tarballs
* stage6:
* Ask various questions and collect answers into system-config
* files.
* stage7:
* execl("/sbin/init");
*/
void
main(int argc, char **argv)
{
int i;
/* phk's main */
if (argc > 1 && !strcmp(argv[1],"phk")) {
return Xmain(argc,argv);
}
/* paul's main */
/* Are we running as init? */
if (getpid() == 1) {
close(0); open("/dev/console", O_RDWR);
close(1); dup(0);
close(2); dup(0);
i = 1;
ioctl(0,TIOCSPGRP,&i);
setlogin("root");
}
if (set_termcap() == -1)
fatal("Can't find terminal entry\n");
if (alloc_memory() == -1)
fatal("Couldn't allocate memory\n");
if (uname(&utsname) == -1) {
/* Fake uname entry */
bcopy("FreeBSD", utsname.sysname, strlen("FreeBSD"));
}
/* XXX - libdialog has particularly bad return value checking */
init_dialog();
/* If we haven't crashed I guess dialog is running ! */
dialog_active = 1;
strcpy(scratch, "/etc/");
strcat(scratch, STATUSFILE);
if (read_status(scratch, sysinstall) == -1) {
fatal(errmsg);
}
switch(sysinstall->status) {
case NOT_INSTALLED:
stage1();
dialog_msgbox("Stage 1 complete",
"Remove all floppy disks from the drives and hit return to reboot from the hard disk",
10, 75, 1);
if (reboot(RB_AUTOBOOT) == -1)
fatal("Reboot failed");
break;
case DISK_READY:
dialog_msgbox("Stage 2 install", "Hi!", 10, 75, 1);
stage2();
dialog_msgbox("Stage 2 complete",
"Well, this is as far as it goes so far :-)\n",
10, 75, 1);
break;
case INSTALLED_BASE:
break;
default:
fatal("Unknown installation status");
}
exit_sysinstall();
}