freebsd-dev/usr.sbin/boot98cfg/boot98cfg.c
2003-02-11 07:29:14 +00:00

296 lines
7.6 KiB
C

/*
* Copyright (c) KATO Takenori, 2000.
*
* All rights reserved. Unpublished rights reserved under the copyright
* laws of Japan.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions
* are met:
*
* 1. Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer as
* the first lines of this file unmodified.
* 2. Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in the
* documentation and/or other materials provided with the distribution.
* 3. The name of the author may not be used to endorse or promote products
* derived from this software without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
* IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
* OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
* IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
* INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
* NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
* THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*
* $FreeBSD$
*/
/*
* Copyright (c) 1999 Robert Nordier
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions
* are met:
* 1. Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
* 2. Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in the
* documentation and/or other materials provided with the distribution.
*
* THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS``AS IS'' AND
* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
* PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS
* BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY,
* OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT
* OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR
* BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
* WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE
* OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE,
* EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*/
#include <sys/param.h>
#include <sys/stat.h>
#include <sys/diskpc98.h>
#include <err.h>
#include <errno.h>
#include <fcntl.h>
#include <paths.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#define BOOTSIZE 0x2000
#define IPLSIZE 512 /* IPL size */
#define BOOTMENUSIZE 7168 /* Max HDD boot menu size */
#define BOOTMENUOFF 0x400
static char *mkrdev(char *);
static void usage(void);
static int read_boot(const char *, u_char *);
static int write_boot(const char *, u_char *);
u_char boot0buf[BOOTSIZE];
u_char ipl[IPLSIZE];
u_char menu[BOOTMENUSIZE];
/*
* Produce a device path for a "canonical" name, where appropriate.
*/
static char *
mkrdev(char *fname)
{
char buf[MAXPATHLEN];
struct stat sb;
char *s;
s = (char *)fname;
if (!strchr(fname, '/')) {
snprintf(buf, sizeof(buf), "%sr%s", _PATH_DEV, fname);
if (stat(buf, &sb))
snprintf(buf, sizeof(buf), "%s%s", _PATH_DEV, fname);
if (!(s = strdup(buf)))
err(1, NULL);
}
return s;
}
static void
usage(void)
{
fprintf(stderr,
"boot98cfg [-B][-i boot0][-m boot0.5][-s secsize][-v version]\n"
" [-f ipl.bak][-F menu.bak] disk\n");
exit(1);
}
static int
read_boot(const char *disk, u_char *boot)
{
int fd, n;
/* Read IPL, partition table and HDD boot menu. */
fd = open(disk, O_RDONLY);
if (fd < 0)
err(1, "%s", disk);
n = read(fd, boot, BOOTSIZE);
if (n != BOOTSIZE)
errx(1, "%s: short read", disk);
close(fd);
return 0;
}
static int
write_boot(const char *disk, u_char *boot)
{
int fd, n, i;
char buf[MAXPATHLEN];
fd = open(disk, O_RDWR);
if (fd != -1) {
if (lseek(fd, 0, SEEK_SET) == -1)
err(1, "%s", disk);
if ((n = write(fd, boot, BOOTSIZE)) < 0)
err(1, "%s", disk);
if (n != BOOTSIZE)
errx(1, "%s: short write", disk);
close(fd);
return 0;
}
for (i = 0; i < NDOSPART; i++) {
snprintf(buf, sizeof(buf), "%ss%d", disk, i + 1);
fd = open(buf, O_RDONLY);
if (fd < 0)
continue;
n = ioctl(fd, DIOCGPC98, boot);
if (n != 0)
err(1, "%s: ioctl DIOCGPC98", disk);
close(fd);
return 0;
}
err(1, "%s", disk);
}
int
main(int argc, char *argv[])
{
char *endptr;
const char *iplpath = "/boot/boot0", *menupath = "/boot/boot0.5";
char *iplbakpath = NULL, *menubakpath = NULL;
char *disk;
int B_flag = 0;
int c;
int fd1;
int n;
int secsize = 512;
int v_flag = 0, version;
while ((c = getopt(argc, argv, "BF:f:i:m:s:v:")) != -1) {
switch (c) {
case 'B':
B_flag = 1;
break;
case 'F':
menubakpath = optarg;
break;
case 'f':
iplbakpath = optarg;
break;
case 'i':
iplpath = optarg;
break;
case 'm':
menupath = optarg;
break;
case 's':
secsize = strtol(optarg, &endptr, 0);
if (errno || *optarg == NULL || *endptr)
errx(1, "%s: Bad argument to -s option",
optarg);
switch (secsize) {
case 256:
case 512:
case 1024:
case 2048:
break;
default:
errx(1, "%s: unsupported sector size", optarg);
break;
}
break;
case 'v':
v_flag = 1;
version = strtol(optarg, &endptr, 0);
if (errno || *optarg == NULL || *endptr ||
version < 0 || version > 255)
errx(1, "%s: Bad argument to -s option",
optarg);
break;
default:
usage();
/* NOTREACHED */
break;
}
}
argc -= optind;
argv += optind;
if (argc != 1)
usage();
disk = mkrdev(*argv);
read_boot(disk, boot0buf);
if (iplbakpath != NULL) {
fd1 = open(iplbakpath, O_WRONLY | O_CREAT | O_TRUNC, 0666);
if (fd1 < 0)
err(1, "%s", iplbakpath);
n = write(fd1, boot0buf, IPLSIZE);
if (n == -1)
err(1, "%s", iplbakpath);
if (n != IPLSIZE)
errx(1, "%s: short write", iplbakpath);
close(fd1);
}
if (menubakpath != NULL) {
fd1 = open(menubakpath, O_WRONLY | O_CREAT | O_TRUNC, 0666);
if (fd1 < 0)
err(1, "%s", menubakpath);
n = write(fd1, boot0buf + BOOTMENUOFF, BOOTMENUSIZE);
if (n == -1)
err(1, "%s", menubakpath);
if (n != BOOTMENUSIZE)
errx(1, "%s: short write", menubakpath);
close(fd1);
}
if (B_flag) {
/* Read IPL (boot0). */
fd1 = open(iplpath, O_RDONLY);
if (fd1 < 0)
err(1, "%s", disk);
n = read(fd1, ipl, IPLSIZE);
if (n < 0)
err(1, "%s", iplpath);
if (n != IPLSIZE)
errx(1, "%s: invalid file", iplpath);
close(fd1);
/* Read HDD boot menu (boot0.5). */
fd1 = open(menupath, O_RDONLY);
if (fd1 < 0)
err(1, "%s", disk);
n = read(fd1, menu, BOOTMENUSIZE);
if (n < 0)
err(1, "%s", menupath);
if (n != BOOTMENUSIZE)
errx(1, "%s: invalid file", menupath);
close(fd1);
memcpy(boot0buf, ipl, IPLSIZE);
memcpy(boot0buf + BOOTMENUOFF, menu, BOOTMENUSIZE);
}
/* Set version number field. */
if (v_flag)
*(boot0buf + secsize - 4) = (u_char)version;
if (B_flag || v_flag)
write_boot(disk, boot0buf);
return 0;
}