mirror of
https://git.FreeBSD.org/src.git
synced 2024-12-23 11:18:54 +00:00
7bec978084
- Fix to build w/o the HAVE_GEOM option.
1102 lines
24 KiB
C
1102 lines
24 KiB
C
/*
|
|
* ----------------------------------------------------------------------------
|
|
* "THE BEER-WARE LICENSE" (Revision 42):
|
|
* <phk@FreeBSD.org> wrote this file. As long as you retain this notice you
|
|
* can do whatever you want with this stuff. If we meet some day, and you think
|
|
* this stuff is worth it, you can buy me a beer in return. Poul-Henning Kamp
|
|
* ----------------------------------------------------------------------------
|
|
*/
|
|
|
|
#include <sys/cdefs.h>
|
|
__FBSDID("$FreeBSD$");
|
|
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <unistd.h>
|
|
#include <fcntl.h>
|
|
#include <string.h>
|
|
#include <err.h>
|
|
#include <sys/sysctl.h>
|
|
#include <sys/types.h>
|
|
#include <sys/stat.h>
|
|
#include <sys/ioctl.h>
|
|
#include <sys/disklabel.h>
|
|
#include <sys/diskslice.h>
|
|
#ifndef PC98
|
|
#include <sys/diskmbr.h>
|
|
#endif
|
|
#include <paths.h>
|
|
#include "libdisk.h"
|
|
|
|
#ifndef PC98
|
|
#define HAVE_GEOM
|
|
#endif
|
|
#ifdef HAVE_GEOM
|
|
#include <ctype.h>
|
|
#include <errno.h>
|
|
#include <assert.h>
|
|
#endif /*HAVE_GEOM*/
|
|
|
|
#ifndef PC98
|
|
#define DOSPTYP_EXTENDED 5
|
|
#define DOSPTYP_ONTRACK 84
|
|
#endif
|
|
|
|
#ifdef DEBUG
|
|
#define DPRINT(x) warn x
|
|
#define DPRINTX(x) warnx x
|
|
#else
|
|
#define DPRINT(x)
|
|
#define DPRINTX(x)
|
|
#endif
|
|
|
|
const char *chunk_n[] = {
|
|
"whole",
|
|
"unknown",
|
|
"fat",
|
|
"freebsd",
|
|
"extended",
|
|
"part",
|
|
"unused",
|
|
NULL
|
|
};
|
|
|
|
struct disk *
|
|
Open_Disk(const char *name)
|
|
{
|
|
return Int_Open_Disk(name, 0);
|
|
}
|
|
|
|
#ifndef PC98
|
|
static u_int32_t
|
|
Read_Int32(u_int32_t *p)
|
|
{
|
|
u_int8_t *bp = (u_int8_t *)p;
|
|
return bp[0] | (bp[1] << 8) | (bp[2] << 16) | (bp[3] << 24);
|
|
}
|
|
#endif
|
|
|
|
#ifdef HAVE_GEOM
|
|
/*
|
|
* XXX BEGIN HACK XXX
|
|
* Scan/parse the XML geom data to retrieve what we need to
|
|
* carry out the work of Int_Open_Disk. This is a total hack
|
|
* and should be replaced with a real XML parser.
|
|
*/
|
|
typedef enum {
|
|
XML_MESH,
|
|
XML_MESH_END,
|
|
XML_CLASS,
|
|
XML_CLASS_END,
|
|
XML_GEOM,
|
|
XML_GEOM_END,
|
|
XML_CONFIG,
|
|
XML_CONFIG_END,
|
|
XML_PROVIDER,
|
|
XML_PROVIDER_END,
|
|
XML_NAME,
|
|
XML_NAME_END,
|
|
XML_INDEX,
|
|
XML_INDEX_END,
|
|
XML_SECLENGTH,
|
|
XML_SECLENGTH_END,
|
|
XML_SECOFFSET,
|
|
XML_SECOFFSET_END,
|
|
XML_TYPE,
|
|
XML_TYPE_END,
|
|
XML_MEDIASIZE,
|
|
XML_MEDIASIZE_END,
|
|
XML_SECTORSIZE,
|
|
XML_SECTORSIZE_END,
|
|
XML_FWHEADS,
|
|
XML_FWHEADS_END,
|
|
XML_FWSECTORS,
|
|
XML_FWSECTORS_END,
|
|
|
|
XML_OTHER,
|
|
XML_OTHER_END
|
|
} XMLToken;
|
|
|
|
const struct {
|
|
XMLToken t;
|
|
const char* token;
|
|
const char* name;
|
|
} xmltokens[] = {
|
|
{ XML_MESH, "mesh", "XML_MESH" },
|
|
{ XML_CLASS, "class", "XML_CLASS" },
|
|
{ XML_GEOM, "geom", "XML_GEOM" },
|
|
{ XML_CONFIG, "config", "XML_CONFIG" },
|
|
{ XML_PROVIDER, "provider", "XML_PROVIDE" },
|
|
{ XML_NAME, "name", "XML_NAME" },
|
|
{ XML_INDEX, "index", "XML_INDEX" },
|
|
{ XML_SECLENGTH, "seclength", "XML_SECLENGTH" },
|
|
{ XML_SECOFFSET, "secoffset", "XML_SECOFFSET" },
|
|
{ XML_TYPE, "type", "XML_TYPE" },
|
|
{ XML_FWHEADS, "fwheads", "XML_FWHEADS" },
|
|
{ XML_FWSECTORS, "fwsectors", "XML_FWSECTORS" },
|
|
{ XML_MEDIASIZE, "mediasize", "XML_MEDIASIZE" },
|
|
{ XML_SECTORSIZE, "sectorsize", "XML_SECTORSIZE" },
|
|
/* NB: this must be last */
|
|
{ XML_OTHER, NULL, "XML_OTHER" },
|
|
};
|
|
#define N(x) (sizeof (x) / sizeof (x[0]))
|
|
|
|
#ifdef DEBUG
|
|
static const char*
|
|
xmltokenname(XMLToken t)
|
|
{
|
|
int i;
|
|
|
|
for (i = 0; i < N(xmltokens); i++) {
|
|
if (t == xmltokens[i].t)
|
|
return xmltokens[i].name;
|
|
if ((t-1) == xmltokens[i].t) {
|
|
static char tbuf[80];
|
|
snprintf(tbuf, sizeof (tbuf), "%s_END",
|
|
xmltokens[i].name);
|
|
return tbuf;
|
|
}
|
|
}
|
|
return "???";
|
|
}
|
|
#endif /*DEBUG*/
|
|
|
|
/*
|
|
* Parse the next XML token delimited by <..>. If the token
|
|
* has a "builtin terminator" (<... />) then just skip it and
|
|
* go the next token.
|
|
*/
|
|
static int
|
|
xmltoken(const char *start, const char **next, XMLToken *t)
|
|
{
|
|
const char *cp = start;
|
|
const char *token;
|
|
int i;
|
|
|
|
again:
|
|
while (*cp != '<') {
|
|
if (*cp == '\0') {
|
|
*next = cp;
|
|
DPRINTX(("xmltoken: EOD"));
|
|
return 0;
|
|
}
|
|
cp++;
|
|
}
|
|
token = ++cp;
|
|
for (; *cp && *cp != '>' && !isspace(*cp); cp++)
|
|
;
|
|
if (*cp == '\0') {
|
|
*next = cp;
|
|
DPRINTX(("xmltoken: EOD"));
|
|
return 0;
|
|
}
|
|
*t = (*token == '/');
|
|
if (*t)
|
|
token++;
|
|
for (i = 0; xmltokens[i].token != NULL; i++)
|
|
if (strncasecmp(token, xmltokens[i].token, cp-token) == 0)
|
|
break;
|
|
*t += xmltokens[i].t;
|
|
/* now collect the remainder of the string */
|
|
for (; *cp != '>' && *cp != '\0'; cp++)
|
|
;
|
|
if (*cp == '\0') {
|
|
*next = cp;
|
|
DPRINTX(("xmltoken: EOD"));
|
|
return 0;
|
|
}
|
|
if (cp > token && cp[-1] == '/') {
|
|
/* e.g. <geom ref="0xc1c8c100"/> */
|
|
start = cp+1;
|
|
goto again;
|
|
}
|
|
*next = cp+1;
|
|
DPRINTX(("xmltoken: %s \"%.*s\"", xmltokenname(*t), cp-token, token));
|
|
return 1;
|
|
}
|
|
|
|
/*
|
|
* Parse and discard XML up to the token terminator.
|
|
*/
|
|
static int
|
|
discardxml(const char **next, XMLToken terminator)
|
|
{
|
|
const char *xml = *next;
|
|
XMLToken t;
|
|
|
|
DPRINTX(("discard XML up to %s", xmltokenname(terminator)));
|
|
for (;;) {
|
|
if (xmltoken(xml, next, &t) == 0)
|
|
return EINVAL;
|
|
if (t == terminator)
|
|
break;
|
|
if ((t & 1) == 0) {
|
|
int error = discardxml(next, t+1);
|
|
if (error)
|
|
return error;
|
|
}
|
|
xml = *next;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
/*
|
|
* Parse XML from between a range of markers; e.g. <mesh> ... </mesh>.
|
|
* When the specified class name is located we descend looking for the
|
|
* geometry information given by diskname. Once inside there we process
|
|
* tags calling f back for each useful one. The arg is passed into f
|
|
* for use in storing the parsed data.
|
|
*/
|
|
static int
|
|
parsexmlpair(
|
|
const char *xml,
|
|
const char **next,
|
|
const char *classname,
|
|
XMLToken terminator,
|
|
const char *diskname,
|
|
int (*f)(void *, XMLToken, u_int *, u_int64_t),
|
|
void *arg
|
|
)
|
|
{
|
|
const char *cp;
|
|
XMLToken t;
|
|
int error;
|
|
u_int ix = (u_int) -1;
|
|
|
|
DPRINTX(("parse XML up to %s", xmltokenname(terminator)));
|
|
do {
|
|
if (xmltoken(xml, next, &t) == 0) {
|
|
error = EINVAL;
|
|
break;
|
|
}
|
|
if (t == terminator) {
|
|
error = 0;
|
|
break;
|
|
}
|
|
if (t & 1) { /* </mumble> w/o matching <mumble> */
|
|
DPRINTX(("Unexpected token %s", xmltokenname(t)));
|
|
error = EINVAL;
|
|
break;
|
|
}
|
|
switch ((int) t) {
|
|
case XML_NAME:
|
|
for (cp = *next; *cp && *cp != '<'; cp++)
|
|
;
|
|
if (*cp == '\0') {
|
|
DPRINTX(("parsexmlpair: EOD"));
|
|
error = EINVAL;
|
|
goto done;
|
|
}
|
|
DPRINTX(("parsexmlpair: \"%.*s\"", cp-*next, *next));
|
|
switch ((int) terminator) {
|
|
case XML_CLASS_END:
|
|
if (strncasecmp(*next, classname, cp-*next))
|
|
return discardxml(next, terminator);
|
|
break;
|
|
case XML_GEOM_END:
|
|
if (strncasecmp(*next, diskname, cp-*next))
|
|
return discardxml(next, terminator);
|
|
break;
|
|
}
|
|
break;
|
|
case XML_SECOFFSET:
|
|
case XML_SECLENGTH:
|
|
case XML_TYPE:
|
|
if (ix == (u_int) -1) {
|
|
DPRINTX(("parsexmlpair: slice data w/o "
|
|
"preceding index"));
|
|
error = EINVAL;
|
|
goto done;
|
|
}
|
|
/* fall thru... */
|
|
case XML_INDEX:
|
|
case XML_FWHEADS:
|
|
case XML_FWSECTORS:
|
|
case XML_MEDIASIZE:
|
|
case XML_SECTORSIZE:
|
|
if (terminator != XML_CONFIG_END &&
|
|
terminator != XML_PROVIDER_END) {
|
|
DPRINTX(("parsexmlpair: %s in unexpected "
|
|
"context: terminator %s",
|
|
xmltokenname(t),
|
|
xmltokenname(terminator)));
|
|
error = EINVAL;
|
|
goto done;
|
|
}
|
|
error = (*f)(arg, t, &ix, strtoull(*next, NULL, 10));
|
|
if (error)
|
|
goto done;
|
|
break;
|
|
}
|
|
error = parsexmlpair(*next, &xml, classname,
|
|
t+1, diskname, f, arg);
|
|
} while (error == 0);
|
|
done:
|
|
return error;
|
|
}
|
|
|
|
/*
|
|
* XML parser. Just barely smart enough to handle the
|
|
* gibberish that geom passed back from the kernel.
|
|
*/
|
|
static int
|
|
xmlparse(
|
|
const char *confxml,
|
|
const char *classname,
|
|
const char *diskname,
|
|
int (*f)(void *, XMLToken, u_int *, u_int64_t),
|
|
void *arg
|
|
)
|
|
{
|
|
const char *next;
|
|
XMLToken t;
|
|
int error;
|
|
|
|
next = confxml;
|
|
while (xmltoken(next, &next, &t) && t != XML_MESH)
|
|
;
|
|
if (t == XML_MESH)
|
|
error = parsexmlpair(next, &next, classname, XML_MESH_END, diskname, f, arg);
|
|
else {
|
|
DPRINTX(("xmlparse: expecting mesh token, got %s",
|
|
xmltokenname(t)));
|
|
error = EINVAL;
|
|
}
|
|
|
|
return (error ? -1 : 0);
|
|
}
|
|
|
|
/*
|
|
* Callback to collect slice-related data.
|
|
*/
|
|
static int
|
|
assignToSlice(void *arg, XMLToken t, u_int *slice, u_int64_t v)
|
|
{
|
|
struct diskslices *ds = (struct diskslices *) arg;
|
|
|
|
switch ((int) t) {
|
|
case XML_INDEX:
|
|
*slice = BASE_SLICE + (u_int) v;
|
|
if (*slice >= MAX_SLICES) {
|
|
DPRINTX(("assignToSlice: invalid slice index %u > max %u",
|
|
*slice, MAX_SLICES));
|
|
return EINVAL;
|
|
}
|
|
if (*slice >= ds->dss_nslices)
|
|
ds->dss_nslices = (*slice)+1;
|
|
break;
|
|
case XML_SECOFFSET:
|
|
ds->dss_slices[*slice].ds_offset = (u_long) v;
|
|
break;
|
|
case XML_SECLENGTH:
|
|
ds->dss_slices[*slice].ds_size = (u_long) v;
|
|
break;
|
|
case XML_TYPE:
|
|
ds->dss_slices[*slice].ds_type = (int) v;
|
|
break;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
/*
|
|
* Callback to collect disk-related data.
|
|
*/
|
|
static int
|
|
assignToDisk(void *arg, XMLToken t, const u_int *slice, u_int64_t v)
|
|
{
|
|
struct disklabel *dl = (struct disklabel *) arg;
|
|
|
|
switch ((int) t) {
|
|
case XML_FWHEADS:
|
|
dl->d_ntracks = (u_int32_t) v;
|
|
case XML_FWSECTORS:
|
|
dl->d_nsectors = (u_int32_t) v;
|
|
break;
|
|
case XML_MEDIASIZE:
|
|
/* store this temporarily; it gets moved later */
|
|
dl->d_secpercyl = v >> 32;
|
|
dl->d_secperunit = v & 0xffffffff;
|
|
break;
|
|
case XML_SECTORSIZE:
|
|
dl->d_secsize = (u_int32_t) v;
|
|
break;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
/*
|
|
* Callback to collect partition-related data.
|
|
*/
|
|
static int
|
|
assignToPartition(void *arg, XMLToken t, u_int *part, u_int64_t v)
|
|
{
|
|
struct disklabel *dl = (struct disklabel *) arg;
|
|
|
|
switch ((int) t) {
|
|
case XML_INDEX:
|
|
*part = (u_int) v;
|
|
if (*part >= MAXPARTITIONS) {
|
|
DPRINTX(("assignToPartition: invalid partition index %u > max %u",
|
|
*part, MAXPARTITIONS));
|
|
return EINVAL;
|
|
}
|
|
if (*part >= dl->d_npartitions)
|
|
dl->d_npartitions = (*part)+1;
|
|
break;
|
|
case XML_SECOFFSET:
|
|
dl->d_partitions[*part].p_offset = (u_int32_t) v;
|
|
break;
|
|
case XML_SECLENGTH:
|
|
dl->d_partitions[*part].p_size = (u_int32_t) v;
|
|
break;
|
|
case XML_TYPE:
|
|
dl->d_partitions[*part].p_fstype = (u_int8_t) v;
|
|
break;
|
|
}
|
|
return 0;
|
|
}
|
|
#undef N
|
|
#endif /*HAVE_GEOM*/
|
|
|
|
struct disk *
|
|
Int_Open_Disk(const char *name, u_long size)
|
|
{
|
|
int i;
|
|
int fd = -1;
|
|
struct diskslices ds;
|
|
struct disklabel dl;
|
|
char device[64];
|
|
struct disk *d;
|
|
#ifdef PC98
|
|
unsigned char *p;
|
|
#else
|
|
struct dos_partition *dp;
|
|
void *p;
|
|
#endif
|
|
#ifdef HAVE_GEOM
|
|
char *confxml = NULL;
|
|
size_t xmlsize;
|
|
u_int64_t mediasize;
|
|
int error;
|
|
#else
|
|
u_long sector_size;
|
|
char *buf;
|
|
#endif /*HAVE_GEOM*/
|
|
|
|
strlcpy(device, _PATH_DEV, sizeof(device));
|
|
strlcat(device, name, sizeof(device));
|
|
|
|
d = (struct disk *)malloc(sizeof *d);
|
|
if(!d) return NULL;
|
|
memset(d, 0, sizeof *d);
|
|
|
|
fd = open(device, O_RDONLY);
|
|
if (fd < 0) {
|
|
DPRINT(("open(%s) failed", device));
|
|
goto bad;
|
|
}
|
|
|
|
memset(&dl, 0, sizeof dl);
|
|
memset(&ds, 0, sizeof ds);
|
|
#ifdef HAVE_GEOM
|
|
/*
|
|
* Read and hack-parse the XML that provides the info we need.
|
|
*/
|
|
error = sysctlbyname("kern.geom.confxml", NULL, &xmlsize, NULL, 0);
|
|
if (error) {
|
|
warn("kern.geom.confxml sysctl not available, giving up!");
|
|
goto bad;
|
|
}
|
|
confxml = (char *) malloc(xmlsize+1);
|
|
if (confxml == NULL) {
|
|
DPRINT(("cannot malloc memory for confxml"));
|
|
goto bad;
|
|
}
|
|
error = sysctlbyname("kern.geom.confxml", confxml, &xmlsize, NULL, 0);
|
|
if (error) {
|
|
DPRINT(("error reading kern.geom.confxml from the system"));
|
|
goto bad;
|
|
}
|
|
confxml[xmlsize] = '\0'; /* in case kernel bug is still there */
|
|
|
|
if (xmlparse(confxml, "MBR", name, assignToSlice, &ds) != 0) {
|
|
DPRINTX(("Error parsing MBR geometry specification."));
|
|
goto bad;
|
|
}
|
|
if (xmlparse(confxml, "DISK", name, assignToDisk, &dl) != 0) {
|
|
DPRINTX(("Error parsing DISK geometry specification."));
|
|
goto bad;
|
|
}
|
|
if (dl.d_nsectors == 0) {
|
|
DPRINTX(("No (zero) sector information in DISK geometry"));
|
|
goto bad;
|
|
}
|
|
if (dl.d_ntracks == 0) {
|
|
DPRINTX(("No (zero) track information in DISK geometry"));
|
|
goto bad;
|
|
}
|
|
if (dl.d_secsize == 0) {
|
|
DPRINTX(("No (zero) sector size information in DISK geometry"));
|
|
goto bad;
|
|
}
|
|
if (dl.d_secpercyl == 0 && dl.d_secperunit == 0) {
|
|
DPRINTX(("No (zero) media size information in DISK geometry"));
|
|
goto bad;
|
|
}
|
|
/*
|
|
* Now patch up disklabel and diskslice.
|
|
*/
|
|
d->sector_size = dl.d_secsize;
|
|
/* NB: media size was stashed in two parts while parsing */
|
|
mediasize = (((u_int64_t) dl.d_secpercyl) << 32) + dl.d_secperunit;
|
|
dl.d_secpercyl = 0;
|
|
dl.d_secperunit = 0;
|
|
size = mediasize / d->sector_size;
|
|
dl.d_ncylinders = size / (dl.d_ntracks * dl.d_nsectors);
|
|
/* "whole disk" slice maintained for compatibility */
|
|
ds.dss_slices[WHOLE_DISK_SLICE].ds_size = size;
|
|
#else /* !HAVE_GEOM */
|
|
if (ioctl(fd, DIOCGDINFO, &dl) < 0) {
|
|
DPRINT(("DIOCGDINFO(%s) failed", device));
|
|
goto bad;
|
|
}
|
|
i = ioctl(fd, DIOCGSLICEINFO, &ds);
|
|
if (i < 0) {
|
|
DPRINT(("DIOCGSLICEINFO(%s) failed", device));
|
|
goto bad;
|
|
}
|
|
|
|
#ifdef DEBUG
|
|
for(i = 0; i < ds.dss_nslices; i++)
|
|
if(ds.dss_slices[i].ds_openmask)
|
|
printf(" open(%d)=0x%2x",
|
|
i, ds.dss_slices[i].ds_openmask);
|
|
printf("\n");
|
|
#endif
|
|
|
|
/* XXX --- ds.dss_slice[WHOLE_DISK_SLICE].ds.size of MO disk is wrong!!! */
|
|
#ifdef PC98
|
|
if (!size)
|
|
size = dl.d_ncylinders * dl.d_ntracks * dl.d_nsectors;
|
|
#else
|
|
if (!size)
|
|
size = ds.dss_slices[WHOLE_DISK_SLICE].ds_size;
|
|
#endif
|
|
|
|
/* determine media sector size */
|
|
if ((buf = malloc(MAX_SEC_SIZE)) == NULL)
|
|
return NULL;
|
|
for (sector_size = MIN_SEC_SIZE; sector_size <= MAX_SEC_SIZE; sector_size *= 2) {
|
|
if (read(fd, buf, sector_size) == sector_size) {
|
|
d->sector_size = sector_size;
|
|
break;
|
|
}
|
|
}
|
|
free (buf);
|
|
if (sector_size > MAX_SEC_SIZE) {
|
|
DPRINT(("Int_Open_Disk: could not determine sector size, "
|
|
"calculated %u, max %u\n", sector_size, MAX_SEC_SIZE));
|
|
/* could not determine sector size */
|
|
goto bad;
|
|
}
|
|
#endif /*HAVE_GEOM*/
|
|
|
|
#ifdef PC98
|
|
p = (unsigned char*)read_block(fd, 1, d->sector_size);
|
|
#else
|
|
p = read_block(fd, 0, d->sector_size);
|
|
dp = (struct dos_partition*)(p + DOSPARTOFF);
|
|
for (i = 0; i < NDOSPART; i++) {
|
|
if (Read_Int32(&dp->dp_start) >= size)
|
|
continue;
|
|
if (Read_Int32(&dp->dp_start) + Read_Int32(&dp->dp_size) >= size)
|
|
continue;
|
|
if (!Read_Int32(&dp->dp_size))
|
|
continue;
|
|
}
|
|
free(p);
|
|
#endif
|
|
|
|
d->bios_sect = dl.d_nsectors;
|
|
d->bios_hd = dl.d_ntracks;
|
|
|
|
d->name = strdup(name);
|
|
|
|
|
|
if (dl.d_ntracks && dl.d_nsectors)
|
|
d->bios_cyl = size / (dl.d_ntracks * dl.d_nsectors);
|
|
|
|
if (Add_Chunk(d, 0, size, name, whole, 0, 0, "-"))
|
|
DPRINT(("Failed to add 'whole' chunk"));
|
|
|
|
#ifdef __i386__
|
|
#ifdef PC98
|
|
/* XXX -- Quick Hack!
|
|
* Check MS-DOS MO
|
|
*/
|
|
if ((*p == 0xf0 || *p == 0xf8) &&
|
|
(*(p+1) == 0xff) &&
|
|
(*(p+2) == 0xff)) {
|
|
Add_Chunk(d, 0, size, name, fat, 0xa0a0, 0, name);
|
|
free(p);
|
|
goto pc98_mo_done;
|
|
}
|
|
free(p);
|
|
#endif /* PC98 */
|
|
for(i=BASE_SLICE;i<ds.dss_nslices;i++) {
|
|
char sname[20];
|
|
char pname[20];
|
|
chunk_e ce;
|
|
u_long flags=0;
|
|
int subtype=0;
|
|
int j;
|
|
|
|
if (! ds.dss_slices[i].ds_size)
|
|
continue;
|
|
snprintf(sname, sizeof(sname), "%ss%d", name, i - 1);
|
|
#ifdef PC98
|
|
subtype = ds.dss_slices[i].ds_type |
|
|
ds.dss_slices[i].ds_subtype << 8;
|
|
switch (ds.dss_slices[i].ds_type & 0x7f) {
|
|
case 0x14:
|
|
ce = freebsd;
|
|
break;
|
|
case 0x20:
|
|
case 0x21:
|
|
case 0x22:
|
|
case 0x23:
|
|
case 0x24:
|
|
ce = fat;
|
|
break;
|
|
#else /* IBM-PC */
|
|
subtype = ds.dss_slices[i].ds_type;
|
|
switch (ds.dss_slices[i].ds_type) {
|
|
case 0xa5:
|
|
ce = freebsd;
|
|
break;
|
|
case 0x1:
|
|
case 0x6:
|
|
case 0x4:
|
|
case 0xb:
|
|
case 0xc:
|
|
case 0xe:
|
|
ce = fat;
|
|
break;
|
|
case DOSPTYP_EXTENDED:
|
|
case 0xf:
|
|
ce = extended;
|
|
break;
|
|
#endif
|
|
default:
|
|
ce = unknown;
|
|
break;
|
|
}
|
|
#ifdef PC98
|
|
if (Add_Chunk(d, ds.dss_slices[i].ds_offset,
|
|
ds.dss_slices[i].ds_size, sname, ce, subtype, flags,
|
|
ds.dss_slices[i].ds_name))
|
|
#else
|
|
if (Add_Chunk(d, ds.dss_slices[i].ds_offset,
|
|
ds.dss_slices[i].ds_size, sname, ce, subtype, flags, ""))
|
|
#endif
|
|
DPRINT(("failed to add chunk for slice %d", i - 1));
|
|
|
|
#ifdef PC98
|
|
if ((ds.dss_slices[i].ds_type & 0x7f) != 0x14)
|
|
#else
|
|
if (ds.dss_slices[i].ds_type != 0xa5)
|
|
#endif
|
|
continue;
|
|
#ifdef HAVE_GEOM
|
|
if (xmlparse(confxml, "BSD", sname, assignToPartition, &dl) != 0) {
|
|
DPRINTX(("Error parsing MBR geometry specification."));
|
|
goto bad;
|
|
}
|
|
#else
|
|
{
|
|
struct disklabel dl;
|
|
int k;
|
|
|
|
strlcpy(pname, _PATH_DEV, sizeof(pname));
|
|
strlcat(pname, sname, sizeof(pname));
|
|
j = open(pname, O_RDONLY);
|
|
if (j < 0) {
|
|
DPRINT(("open(%s)", pname));
|
|
continue;
|
|
}
|
|
k = ioctl(j, DIOCGDINFO, &dl);
|
|
if (k < 0) {
|
|
DPRINT(("ioctl(%s, DIOCGDINFO)", pname));
|
|
close(j);
|
|
continue;
|
|
}
|
|
close(j);
|
|
}
|
|
#endif /*HAVE_GEOM*/
|
|
|
|
for(j = 0; j <= dl.d_npartitions; j++) {
|
|
if (j == RAW_PART)
|
|
continue;
|
|
if (j == 3)
|
|
continue;
|
|
if (j == dl.d_npartitions) {
|
|
j = 3;
|
|
dl.d_npartitions = 0;
|
|
}
|
|
if (!dl.d_partitions[j].p_size)
|
|
continue;
|
|
if (dl.d_partitions[j].p_size +
|
|
dl.d_partitions[j].p_offset >
|
|
ds.dss_slices[i].ds_size)
|
|
continue;
|
|
snprintf(pname, sizeof(pname), "%s%c", sname, j + 'a');
|
|
if (Add_Chunk(d,
|
|
dl.d_partitions[j].p_offset +
|
|
ds.dss_slices[i].ds_offset,
|
|
dl.d_partitions[j].p_size,
|
|
pname,part,
|
|
dl.d_partitions[j].p_fstype,
|
|
#ifdef PC98
|
|
0,
|
|
ds.dss_slices[i].ds_name) && j != 3)
|
|
#else
|
|
0, "") && j != 3)
|
|
#endif
|
|
DPRINT((
|
|
"Failed to add chunk for partition %c [%lu,%lu]",
|
|
j + 'a', dl.d_partitions[j].p_offset,
|
|
dl.d_partitions[j].p_size));
|
|
}
|
|
}
|
|
#endif /* __i386__ */
|
|
#ifdef __alpha__
|
|
{
|
|
struct disklabel dl;
|
|
char pname[20];
|
|
int j,k;
|
|
|
|
strlcpy(pname, _PATH_DEV, sizeof(pname));
|
|
strlcat(pname, name, sizeof(pname));
|
|
j = open(pname, O_RDONLY);
|
|
if (j < 0) {
|
|
DPRINT(("open(%s)", pname));
|
|
goto nolabel;
|
|
}
|
|
k = ioctl(j, DIOCGDINFO, &dl);
|
|
if (k < 0) {
|
|
DPRINT(("ioctl(%s, DIOCGDINFO)", pname));
|
|
close(j);
|
|
goto nolabel;
|
|
}
|
|
close(j);
|
|
All_FreeBSD(d, 1);
|
|
|
|
for(j = 0; j <= dl.d_npartitions; j++) {
|
|
if (j == RAW_PART)
|
|
continue;
|
|
if (j == 3)
|
|
continue;
|
|
if (j == dl.d_npartitions) {
|
|
j = 3;
|
|
dl.d_npartitions = 0;
|
|
}
|
|
if (!dl.d_partitions[j].p_size)
|
|
continue;
|
|
if (dl.d_partitions[j].p_size +
|
|
dl.d_partitions[j].p_offset >
|
|
ds.dss_slices[WHOLE_DISK_SLICE].ds_size)
|
|
continue;
|
|
snprintf(pname, sizeof(pname), "%s%c", name, j + 'a');
|
|
if (Add_Chunk(d,
|
|
dl.d_partitions[j].p_offset,
|
|
dl.d_partitions[j].p_size,
|
|
pname,part,
|
|
dl.d_partitions[j].p_fstype,
|
|
0, "") && j != 3)
|
|
DPRINT((
|
|
"Failed to add chunk for partition %c [%lu,%lu]",
|
|
j + 'a', dl.d_partitions[j].p_offset,
|
|
dl.d_partitions[j].p_size));
|
|
}
|
|
nolabel:;
|
|
}
|
|
#endif /* __alpha__ */
|
|
#ifdef PC98
|
|
pc98_mo_done:
|
|
#endif
|
|
close(fd);
|
|
Fixup_Names(d);
|
|
return d;
|
|
bad:
|
|
#ifdef HAVE_GEOM
|
|
if (confxml != NULL)
|
|
free(confxml);
|
|
#endif
|
|
if (fd >= 0)
|
|
close(fd);
|
|
return NULL;
|
|
}
|
|
|
|
void
|
|
Debug_Disk(struct disk *d)
|
|
{
|
|
printf("Debug_Disk(%s)", d->name);
|
|
#if 0
|
|
printf(" real_geom=%lu/%lu/%lu", d->real_cyl, d->real_hd, d->real_sect);
|
|
#endif
|
|
printf(" bios_geom=%lu/%lu/%lu = %lu\n",
|
|
d->bios_cyl, d->bios_hd, d->bios_sect,
|
|
d->bios_cyl * d->bios_hd * d->bios_sect);
|
|
#if defined(PC98)
|
|
printf(" boot1=%p, boot2=%p, bootipl=%p, bootmenu=%p\n",
|
|
d->boot1, d->boot2, d->bootipl, d->bootmenu);
|
|
#elif defined(__i386__)
|
|
printf(" boot1=%p, boot2=%p, bootmgr=%p\n",
|
|
d->boot1, d->boot2, d->bootmgr);
|
|
#elif defined(__alpha__)
|
|
printf(" boot1=%p, bootmgr=%p\n",
|
|
d->boot1, d->bootmgr);
|
|
#endif
|
|
Debug_Chunk(d->chunks);
|
|
}
|
|
|
|
void
|
|
Free_Disk(struct disk *d)
|
|
{
|
|
if(d->chunks) Free_Chunk(d->chunks);
|
|
if(d->name) free(d->name);
|
|
#ifdef PC98
|
|
if(d->bootipl) free(d->bootipl);
|
|
if(d->bootmenu) free(d->bootmenu);
|
|
#else
|
|
if(d->bootmgr) free(d->bootmgr);
|
|
#endif
|
|
if(d->boot1) free(d->boot1);
|
|
#if defined(__i386__)
|
|
if(d->boot2) free(d->boot2);
|
|
#endif
|
|
free(d);
|
|
}
|
|
|
|
#if 0
|
|
void
|
|
Collapse_Disk(struct disk *d)
|
|
{
|
|
|
|
while(Collapse_Chunk(d, d->chunks))
|
|
;
|
|
}
|
|
#endif
|
|
|
|
#ifdef PC98
|
|
static char * device_list[] = {"wd", "aacd", "ad", "da", "afd", "fla", "idad", "mlxd", "amrd", "twed", "ar", "fd", 0};
|
|
#else
|
|
static char * device_list[] = {"aacd", "ad", "da", "afd", "fla", "idad", "mlxd", "amrd", "twed", "ar", "fd", 0};
|
|
#endif
|
|
|
|
int qstrcmp(const void* a, const void* b) {
|
|
|
|
char *str1 = *(char**)a;
|
|
char *str2 = *(char**)b;
|
|
return strcmp(str1, str2);
|
|
|
|
}
|
|
|
|
char **
|
|
Disk_Names()
|
|
{
|
|
int i,j,disk_cnt;
|
|
char disk[25];
|
|
char diskname[25];
|
|
struct stat st;
|
|
struct diskslices ds;
|
|
int fd;
|
|
static char **disks;
|
|
int error;
|
|
size_t listsize;
|
|
char *disklist;
|
|
|
|
disks = malloc(sizeof *disks * (1 + MAX_NO_DISKS));
|
|
if (disks == NULL)
|
|
return NULL;
|
|
memset(disks,0,sizeof *disks * (1 + MAX_NO_DISKS));
|
|
error = sysctlbyname("kern.disks", NULL, &listsize, NULL, 0);
|
|
if (!error) {
|
|
disklist = (char *)malloc(listsize+1);
|
|
if (disklist == NULL) {
|
|
free(disks);
|
|
return NULL;
|
|
}
|
|
memset(disklist, 0, listsize+1);
|
|
error = sysctlbyname("kern.disks", disklist, &listsize, NULL, 0);
|
|
if (error) {
|
|
free(disklist);
|
|
free(disks);
|
|
return NULL;
|
|
}
|
|
for (disk_cnt = 0; disk_cnt < MAX_NO_DISKS; disk_cnt++) {
|
|
disks[disk_cnt] = strsep(&disklist, " ");
|
|
if (disks[disk_cnt] == NULL)
|
|
break;
|
|
}
|
|
} else {
|
|
warn("kern.disks sysctl not available");
|
|
disk_cnt = 0;
|
|
for (j = 0; device_list[j]; j++) {
|
|
if(disk_cnt >= MAX_NO_DISKS)
|
|
break;
|
|
for (i = 0; i < MAX_NO_DISKS; i++) {
|
|
snprintf(diskname, sizeof(diskname), "%s%d",
|
|
device_list[j], i);
|
|
snprintf(disk, sizeof(disk), _PATH_DEV"%s", diskname);
|
|
if (stat(disk, &st) || !(st.st_mode & S_IFCHR))
|
|
continue;
|
|
if ((fd = open(disk, O_RDWR)) == -1)
|
|
continue;
|
|
if (ioctl(fd, DIOCGSLICEINFO, &ds) == -1) {
|
|
DPRINT(("DIOCGSLICEINFO %s", disk));
|
|
close(fd);
|
|
continue;
|
|
}
|
|
close(fd);
|
|
disks[disk_cnt++] = strdup(diskname);
|
|
if(disk_cnt >= MAX_NO_DISKS)
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
qsort(disks, disk_cnt, sizeof(char*), qstrcmp);
|
|
|
|
return disks;
|
|
}
|
|
|
|
#ifdef PC98
|
|
void
|
|
Set_Boot_Mgr(struct disk *d, const u_char *bootipl, const size_t bootipl_size,
|
|
const u_char *bootmenu, const size_t bootmenu_size)
|
|
#else
|
|
void
|
|
Set_Boot_Mgr(struct disk *d, const u_char *b, const size_t s)
|
|
#endif
|
|
{
|
|
#ifdef PC98
|
|
if (bootipl_size % d->sector_size != 0)
|
|
return;
|
|
if (d->bootipl)
|
|
free(d->bootipl);
|
|
if (!bootipl) {
|
|
d->bootipl = NULL;
|
|
} else {
|
|
d->bootipl_size = bootipl_size;
|
|
d->bootipl = malloc(bootipl_size);
|
|
if(!d->bootipl) return;
|
|
memcpy(d->bootipl, bootipl, bootipl_size);
|
|
}
|
|
|
|
if (bootmenu_size % d->sector_size != 0)
|
|
return;
|
|
if (d->bootmenu)
|
|
free(d->bootmenu);
|
|
if (!bootmenu) {
|
|
d->bootmenu = NULL;
|
|
} else {
|
|
d->bootmenu_size = bootmenu_size;
|
|
d->bootmenu = malloc(bootmenu_size);
|
|
if(!d->bootmenu) return;
|
|
memcpy(d->bootmenu, bootmenu, bootmenu_size);
|
|
}
|
|
#else
|
|
if (s % d->sector_size != 0)
|
|
return;
|
|
if (d->bootmgr)
|
|
free(d->bootmgr);
|
|
if (!b) {
|
|
d->bootmgr = NULL;
|
|
} else {
|
|
d->bootmgr_size = s;
|
|
d->bootmgr = malloc(s);
|
|
if(!d->bootmgr) return;
|
|
memcpy(d->bootmgr, b, s);
|
|
}
|
|
#endif
|
|
}
|
|
|
|
int
|
|
Set_Boot_Blocks(struct disk *d, const u_char *b1, const u_char *b2)
|
|
{
|
|
#if defined(__i386__)
|
|
if (d->boot1) free(d->boot1);
|
|
d->boot1 = malloc(512);
|
|
if(!d->boot1) return -1;
|
|
memcpy(d->boot1, b1, 512);
|
|
if (d->boot2) free(d->boot2);
|
|
d->boot2 = malloc(15 * 512);
|
|
if(!d->boot2) return -1;
|
|
memcpy(d->boot2, b2, 15 * 512);
|
|
#elif defined(__alpha__)
|
|
if (d->boot1) free(d->boot1);
|
|
d->boot1 = malloc(15 * 512);
|
|
if(!d->boot1) return -1;
|
|
memcpy(d->boot1, b1, 15 * 512);
|
|
#endif
|
|
return 0;
|
|
}
|
|
|
|
const char *
|
|
slice_type_name( int type, int subtype )
|
|
{
|
|
switch (type) {
|
|
case 0: return "whole";
|
|
#ifndef PC98
|
|
case 1: switch (subtype) {
|
|
case 1: return "fat (12-bit)";
|
|
case 2: return "XENIX /";
|
|
case 3: return "XENIX /usr";
|
|
case 4: return "fat (16-bit,<=32Mb)";
|
|
case 5: return "extended DOS";
|
|
case 6: return "fat (16-bit,>32Mb)";
|
|
case 7: return "NTFS/HPFS/QNX";
|
|
case 8: return "AIX bootable";
|
|
case 9: return "AIX data";
|
|
case 10: return "OS/2 bootmgr";
|
|
case 11: return "fat (32-bit)";
|
|
case 12: return "fat (32-bit,LBA)";
|
|
case 14: return "fat (16-bit,>32Mb,LBA)";
|
|
case 15: return "extended DOS, LBA";
|
|
case 18: return "Compaq Diagnostic";
|
|
case 84: return "OnTrack diskmgr";
|
|
case 100: return "Netware 2.x";
|
|
case 101: return "Netware 3.x";
|
|
case 115: return "SCO UnixWare";
|
|
case 128: return "Minix 1.1";
|
|
case 129: return "Minix 1.5";
|
|
case 130: return "linux_swap";
|
|
case 131: return "ext2fs";
|
|
case 166: return "OpenBSD FFS"; /* 0xA6 */
|
|
case 169: return "NetBSD FFS"; /* 0xA9 */
|
|
case 182: return "OpenBSD"; /* dedicated */
|
|
case 183: return "bsd/os";
|
|
case 184: return "bsd/os swap";
|
|
case 238: return "EFI GPT";
|
|
case 239: return "EFI Sys. Part.";
|
|
default: return "unknown";
|
|
}
|
|
#endif
|
|
case 2: return "fat";
|
|
case 3: switch (subtype) {
|
|
#ifdef PC98
|
|
case 0xc494: return "freebsd";
|
|
#else
|
|
case 165: return "freebsd";
|
|
#endif
|
|
default: return "unknown";
|
|
}
|
|
#ifndef PC98
|
|
case 4: return "extended";
|
|
case 5: return "part";
|
|
case 6: return "unused";
|
|
#endif
|
|
default: return "unknown";
|
|
}
|
|
}
|