NetBSD/distrib/utils/sysinst/fdisk.c

143 lines
5.2 KiB
C

/* $NetBSD: fdisk.c,v 1.11 1999/01/23 06:11:51 garbled Exp $ */
/*
* Copyright 1997 Piermont Information Systems Inc.
* All rights reserved.
*
* Written by Philip A. Nelson for Piermont Information Systems Inc.
*
* 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.
* 3. All advertising materials mentioning features or use of this software
* must display the following acknowledgement:
* This product includes software develooped for the NetBSD Project by
* Piermont Information Systems Inc.
* 4. The name of Piermont Information Systems Inc. may not be used to endorse
* or promote products derived from this software without specific prior
* written permission.
*
* THIS SOFTWARE IS PROVIDED BY PIERMONT INFORMATION SYSTEMS INC. ``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 PIERMONT INFORMATION SYSTEMS INC. 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.
*
*/
/* fdisk.c -- routines to deal with /sbin/fdisk ... */
#include <stdio.h>
#include "defs.h"
#include "md.h"
#include "txtwalk.h"
#include "msg_defs.h"
#include "menu_defs.h"
struct lookfor fdiskbuf[] = {
{"DLCYL", "DLCYL=%d", "a $0", &dlcyl, 1, 0, NULL},
{"DLHEAD", "DLHEAD=%d", "a $0", &dlhead, 1, 0, NULL},
{"DLSEC", "DLSEC=%d", "a $0", &dlsec, 1, 0, NULL},
{"BCYL", "BCYL=%d", "a $0", &bcyl, 1, 0, NULL},
{"BHEAD", "BHEAD=%d", "a $0", &bhead, 1, 0, NULL},
{"BSEC", "BSEC=%d", "a $0", &bsec, 1, 0, NULL},
{"PART0ID", "PART0ID=%d", "a $0", &part[0][ID], 1, 0, NULL},
{"PART0SIZE", "PART0SIZE=%d", "a $0", &part[0][SIZE], 1, 0, NULL},
{"PART0START", "PART0START=%d", "a $0", &part[0][START], 1, 0, NULL},
{"PART0FLAG", "PART0FLAG=0x%d", "a $0", &part[0][FLAG], 1, 0, NULL},
{"PART1ID", "PART1ID=%d", "a $0", &part[1][ID], 1, 0, NULL},
{"PART1SIZE", "PART1SIZE=%d", "a $0", &part[1][SIZE], 1, 0, NULL},
{"PART1START", "PART1START=%d", "a $0", &part[1][START], 1, 0, NULL},
{"PART1FLAG", "PART1FLAG=0x%d", "a $0", &part[1][FLAG], 1, 0, NULL},
{"PART2ID", "PART2ID=%d", "a $0", &part[2][ID], 1, 0, NULL},
{"PART2SIZE", "PART2SIZE=%d", "a $0", &part[2][SIZE], 1, 0, NULL},
{"PART2START", "PART2START=%d", "a $0", &part[2][START], 1, 0, NULL},
{"PART2FLAG", "PART2FLAG=0x%d", "a $0", &part[2][FLAG], 1, 0, NULL},
{"PART3ID", "PART3ID=%d", "a $0", &part[3][ID], 1, 0, NULL},
{"PART3SIZE", "PART3SIZE=%d", "a $0", &part[3][SIZE], 1, 0, NULL},
{"PART3START", "PART3START=%d", "a $0", &part[3][START], 1, 0, NULL},
{"PART3FLAG", "PART3FLAG=0x%d", "a $0", &part[3][FLAG], 1, 0, NULL}
};
int numfdiskbuf = sizeof(fdiskbuf) / sizeof(struct lookfor);
/*
* Fetch current MBR from disk into core by parsing /sbin/fdisk output.
*/
void
get_fdisk_info()
{
char *textbuf;
int textsize;
int t1, t2;
/* Get Fdisk information */
textsize = collect(T_OUTPUT, &textbuf,
"/sbin/fdisk -S /dev/r%sd 2>/dev/null", diskdev);
if (textsize < 0) {
(void)fprintf(stderr, "Could not run fdisk.");
if (logging)
(void)fprintf(log, "Aborting: Could not run fdisk.\n");
exit (1);
}
walk(textbuf, textsize, fdiskbuf, numfdiskbuf);
free(textbuf);
/*
* A common failure of fdisk is to get the number of cylinders
* wrong and the number of sectors and heads right. This makes
* a disk look very big. In this case, we can just recompute
* the number of cylinders and things should work just fine.
* Also, fdisk may correctly indentify the settings to include
* a cylinder total > 1024, because translation mode is not used.
* Check for it.
*/
/* XXX should warn user about this, and maybe ask! */
if (bcyl > 1024 && disk->geom[1] == bhead && disk->geom[2] == bsec)
bcyl = 1024;
else if (bcyl > 1024 && bsec < 64) {
t1 = disk->geom[0] * disk->geom[1] * disk->geom[2];
t2 = bhead * bsec;
if (bcyl * t2 > t1) {
t2 = t1 / t2;
if (t2 < 1024)
bcyl = t2;
}
}
}
/*
* Write incore MBR geometry and partition info to disk
* using /sbin/fdisk.
*/
void
set_fdisk_info()
{
int i;
for (i = 0; i < 4; i++)
if (part[i][SET])
run_prog(0, 1, "/sbin/fdisk -u -f -%d -b %d/%d/%d "
"-s %d/%d/%d /dev/r%sd",
i, bcyl, bhead, bsec,
part[i][ID], part[i][START],
part[i][SIZE], diskdev);
if (activepart >= 0)
run_prog(0, 1, "/sbin/fdisk -a -%d -f /dev/r%s",
activepart, diskdev);
}