340 lines
8.2 KiB
C
340 lines
8.2 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/stat.h>
|
|
#include <ufs/ffs/fs.h>
|
|
#include <machine/console.h>
|
|
|
|
#include "mbr.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;
|
|
|
|
int no_disks = 0;
|
|
int inst_disk = 0;
|
|
int inst_part = 0;
|
|
int whole_disk = 0;
|
|
int custom_install = 0;
|
|
int dialog_active = 0;
|
|
|
|
void exit_sysinstall();
|
|
void exit_prompt();
|
|
extern char *part_type(int);
|
|
extern int disk_size(struct disklabel *);
|
|
|
|
char selection[30];
|
|
char boot1[] = BOOT1;
|
|
char boot2[] = BOOT2;
|
|
|
|
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);
|
|
|
|
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);
|
|
}
|
|
|
|
|
|
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(&avail_disklabels[i]),avail_disknames[i]);
|
|
}
|
|
|
|
if (dialog_menu("FreeBSD Installation", scratch, 10, 75, 5, no_disks, options, selection)) {
|
|
dialog_clear();
|
|
sprintf(scratch,"\n\n\nYou did not select a valid disk.\n\n");
|
|
AskAbort(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;
|
|
whole_disk = 0;
|
|
|
|
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(&avail_disklabels[inst_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");
|
|
dialog_clear();
|
|
AskAbort(scratch);
|
|
valid = 0;
|
|
}
|
|
dialog_clear();
|
|
choice = atoi(selection) - 1;
|
|
if (choice == -1) {
|
|
whole_disk = 1;
|
|
choice = 0;
|
|
if (dialog_yesno(TITLE, "\n\nInstalling to the whole disk will erase all its current data.\n\nAre you sure you want to do this?", 10, 75)) {
|
|
valid = 0;
|
|
whole_disk = 0;
|
|
}
|
|
}
|
|
dialog_clear();
|
|
} while (!valid);
|
|
|
|
return(choice);
|
|
}
|
|
|
|
void
|
|
stage1()
|
|
{
|
|
int i;
|
|
int ok = 0;
|
|
int ready = 0;
|
|
|
|
query_disks();
|
|
|
|
while (!ready) {
|
|
ready = 1;
|
|
|
|
inst_disk = select_disk();
|
|
|
|
if (read_mbr(avail_fds[inst_disk], mbr) == -1) {
|
|
sprintf(scratch, "The following error occured while trying\nto read the master boot record:\n\n%s\nIn order to install FreeBSD a new master boot record\nwill have to be written which will mean all current\ndata on the hard disk will be lost.", errmsg);
|
|
ok = 0;
|
|
while (!ok) {
|
|
AskAbort(scratch);
|
|
if (!dialog_yesno(TITLE, "Are you sure you wish to proceed?",
|
|
10, 75)) {
|
|
dialog_clear();
|
|
if (clear_mbr(mbr, boot1) == -1) {
|
|
sprintf(scratch, "\n\nCouldn't create new master boot record.\n\n%s", errmsg);
|
|
Fatal(scratch);;
|
|
}
|
|
ok = 1;
|
|
}
|
|
dialog_clear();
|
|
}
|
|
}
|
|
|
|
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]);
|
|
}
|
|
|
|
dialog_clear();
|
|
inst_part = select_partition(inst_disk);
|
|
|
|
ok = 0;
|
|
while (!ok) {
|
|
if (build_mbr(mbr, boot1, &avail_disklabels[inst_disk]) != -1) {
|
|
ready = 1;
|
|
ok = 1;
|
|
} else {
|
|
sprintf(scratch, "The DOS partition table is inconsistent.\n\n%s\nDo you wish to edit it by hand?", errmsg);
|
|
if (!dialog_yesno(TITLE, scratch, 10, 75)) {
|
|
dialog_clear();
|
|
edit_mbr(mbr, &avail_disklabels[inst_disk]);
|
|
} else {
|
|
dialog_clear();
|
|
AskAbort("Installation cannot proceed without\na valid master boot record\n");
|
|
ok = 1;
|
|
ready = 0;
|
|
}
|
|
}
|
|
dialog_clear();
|
|
}
|
|
|
|
if (ready) {
|
|
default_disklabel(&avail_disklabels[inst_disk],
|
|
mbr->dospart[inst_part].dp_size,
|
|
mbr->dospart[inst_part].dp_start);
|
|
dialog_msgbox(TITLE, "This is an experimental disklabel configuration\nmenu. It doesn't perform any validation of the entries\nas yet so BE SURE YOU TYPE THINGS CORRECTLY.\n\n Hit escape to quit the editor.\n\nThere may be some delay exiting because of a dialog bug", 20,70,1);
|
|
dialog_clear();
|
|
edit_disklabel(&avail_disklabels[inst_disk]);
|
|
build_disklabel(&avail_disklabels[inst_disk]);
|
|
if (build_bootblocks(&avail_disklabels[inst_disk]) == -1)
|
|
Fatal(errmsg);
|
|
}
|
|
|
|
/* ready could have been reset above */
|
|
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)) {
|
|
dialog_clear();
|
|
AskAbort("Do you want to quit?");
|
|
ready = 0;
|
|
}
|
|
dialog_clear();
|
|
}
|
|
}
|
|
|
|
/* Write master boot record and bootblocks */
|
|
if (write_mbr(avail_fds[inst_disk], mbr) == -1)
|
|
Fatal(errmsg);
|
|
if (write_bootblocks(avail_fds[inst_disk],
|
|
mbr->dospart[inst_part].dp_start,
|
|
avail_disklabels[inst_disk].d_bbsize) == -1)
|
|
Fatal(errmsg);
|
|
|
|
/* 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);
|
|
}
|
|
}
|