6dd3a38fff
Could do with some cosmetic tuning regarding placement and things. Fixed some dialog code (from Andrew). Pass mountpoints onto stage2 in a struct fstab *mounts[] Fix all the field connections to conform to the new L&F document.
164 lines
3.7 KiB
C
164 lines
3.7 KiB
C
/*
|
|
* 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 <fstab.h>
|
|
#include <unistd.h>
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
#include <dialog.h>
|
|
|
|
#include <sys/param.h>
|
|
#include <sys/types.h>
|
|
#include <sys/errno.h>
|
|
#include <sys/disklabel.h>
|
|
#include <sys/ioctl.h>
|
|
#include <sys/fcntl.h>
|
|
#include <sys/uio.h>
|
|
#include <ufs/ffs/fs.h>
|
|
|
|
#include "disk.h"
|
|
#include "sysinstall.h"
|
|
|
|
char boot1[] = BOOT1;
|
|
char boot2[] = BOOT2;
|
|
|
|
int
|
|
enable_label(int fd)
|
|
{
|
|
int flag = 1;
|
|
if (ioctl(fd, DIOCWLABEL, &flag) < 0)
|
|
return (-1);
|
|
return (0);
|
|
}
|
|
|
|
int
|
|
disable_label(int fd)
|
|
{
|
|
int flag = 0;
|
|
if (ioctl(fd, DIOCWLABEL, &flag) < 0)
|
|
return (-1);
|
|
return (0);
|
|
}
|
|
|
|
int
|
|
write_bootblocks(int disk)
|
|
{
|
|
int fd;
|
|
off_t offset;
|
|
int part = disk_list[disk].inst_part;
|
|
struct disklabel *lbl = &disk_list[disk].lbl;
|
|
unsigned char bootblocks[BBSIZE];
|
|
|
|
/* Load MBR boot code */
|
|
|
|
if ((fd = open(boot1, O_RDONLY)) == -1) {
|
|
sprintf(errmsg, "Couldn't open boot file %s for bootblocks\n%s\n",
|
|
boot1, strerror(errno));
|
|
return (-1);
|
|
}
|
|
|
|
if (read(fd, bootblocks, MBRSIZE) < 0) {
|
|
sprintf(errmsg, "Couldn't load boot file %s into bootblocks\n%s\n",
|
|
boot1, strerror(errno));
|
|
return (-1);
|
|
}
|
|
|
|
if (close(fd) == -1) {
|
|
sprintf(errmsg, "Couldn't close boot file %s\n%s\n", boot1,
|
|
strerror(errno));
|
|
return (-1);
|
|
}
|
|
|
|
/* Load second level boot code */
|
|
|
|
if ((fd = open(boot2, O_RDONLY)) == -1) {
|
|
sprintf(errmsg, "Couldn't open boot file %s for bootblocks\n%s\n",
|
|
boot2, strerror(errno));
|
|
return (-1);
|
|
}
|
|
|
|
if (read(fd, &bootblocks[MBRSIZE], (int)(lbl->d_bbsize - MBRSIZE)) < 0) {
|
|
sprintf(errmsg, "Couldn't load boot file %s into bootblocks\n%s\n",
|
|
boot2, strerror(errno));
|
|
return (-1);
|
|
}
|
|
|
|
if (close(fd) == -1) {
|
|
sprintf(errmsg, "Couldn't close boot file %s\n%s\n", boot2,
|
|
strerror(errno));
|
|
return (-1);
|
|
}
|
|
|
|
|
|
/* Copy the current MBR table into the boot blocks */
|
|
|
|
bcopy(&disk_list[disk].mbr.dospart, &bootblocks[DOSPARTOFF],
|
|
sizeof(struct dos_partition) * NDOSPART);
|
|
|
|
/* Set checksum */
|
|
lbl->d_checksum = 0;
|
|
lbl->d_checksum = dkcksum(lbl);
|
|
|
|
/* Copy disklabel into bootblocks */
|
|
|
|
bcopy(lbl, &bootblocks[(LABELSECTOR * lbl->d_secsize) + LABELOFFSET],
|
|
sizeof *lbl);
|
|
|
|
/* Calculate offset to start of MBR partition we're installing to */
|
|
|
|
offset = disk_list[disk].mbr.dospart[part].dp_start;
|
|
offset *= lbl->d_secsize;
|
|
|
|
/* Write the boot blocks out to the raw disk */
|
|
|
|
if ((fd = open(diskname(disk), O_RDWR)) == -1) {
|
|
sprintf(errmsg, "Couldn't open %s to write bootblocks\n%s\n",
|
|
scratch,strerror(errno));
|
|
return (-1);
|
|
}
|
|
|
|
if (lseek(fd, offset, SEEK_SET) < 0) {
|
|
sprintf(errmsg, "Couldn't seek to bootblocks area %s\n%s\n",
|
|
scratch, strerror(errno));
|
|
return (-1);
|
|
}
|
|
|
|
/* Update the in-core label too if possible */
|
|
|
|
if (ioctl(fd, DIOCSDINFO, lbl) < 0) {
|
|
sprintf(errmsg, "Couldn't change in-core disklabel for %s\n\n%s",
|
|
scratch, strerror(errno));
|
|
return (-1);
|
|
}
|
|
|
|
if (enable_label(fd) == -1)
|
|
return (-1);
|
|
|
|
if (write(fd, bootblocks, lbl->d_bbsize) != lbl->d_bbsize) {
|
|
sprintf(errmsg, "Failed to write out bootblocks to %s\n%s\n",
|
|
scratch, strerror(errno));
|
|
return (-1);
|
|
}
|
|
|
|
if (disable_label(fd) == -1)
|
|
return (-1);
|
|
|
|
if (close(fd) == -1) {
|
|
sprintf(errmsg, "Couldn't close device %s\n\n%s",
|
|
scratch, strerror(errno));
|
|
return (-1);
|
|
}
|
|
return(0);
|
|
}
|