/*
* Construct a prototype disklabel from /etc/disktab.
*/
static void
makelabel(const char *type, struct disklabel *lp)
{
struct disklabel *dp;
if (strcmp(type, "auto") == 0)
dp = getvirginlabel();
else
dp = getdiskbyname(type);
if (dp == NULL)
errx(1, "%s: unknown disk type", type);
*lp = *dp;
bzero(lp->d_packname, sizeof(lp->d_packname));
}
/*
* Construct a prototype disklabel from /etc/disktab. As a side
* effect, set the names of the primary and secondary boot files
* if specified.
*/
void
makelabel(const char *type, const char *name, struct disklabel32 *lp)
{
struct disklabel32 *dp;
if (strcmp(type, "auto") == 0)
dp = getvirginlabel();
else
dp = getdisklabelfromdisktab(type);
if (dp == NULL)
errx(1, "%s: unknown disk type", type);
*lp = *dp;
/*
* NOTE: boot control files may no longer be specified in disktab.
*/
if (name)
strncpy(lp->d_packname, name, sizeof(lp->d_packname));
}
/*
* Check disklabel for errors and fill in
* derived fields according to supplied values.
*/
static int
checklabel(struct disklabel *lp)
{
struct partition *pp;
int i, errors = 0;
char part;
u_long base_offset, needed, total_size, total_percent, current_offset;
long free_space;
int seen_default_offset;
int hog_part;
int j;
struct partition *pp2;
if (lp == NULL)
lp = &lab;
if (allfields) {
if (lp->d_secsize == 0) {
fprintf(stderr, "sector size 0\n");
return (1);
}
if (lp->d_nsectors == 0) {
fprintf(stderr, "sectors/track 0\n");
return (1);
}
if (lp->d_ntracks == 0) {
fprintf(stderr, "tracks/cylinder 0\n");
return (1);
}
if (lp->d_ncylinders == 0) {
fprintf(stderr, "cylinders/unit 0\n");
errors++;
}
if (lp->d_rpm == 0)
warnx("revolutions/minute 0");
if (lp->d_secpercyl == 0)
lp->d_secpercyl = lp->d_nsectors * lp->d_ntracks;
if (lp->d_secperunit == 0)
lp->d_secperunit = lp->d_secpercyl * lp->d_ncylinders;
if (lp->d_bbsize == 0) {
fprintf(stderr, "boot block size 0\n");
errors++;
} else if (lp->d_bbsize % lp->d_secsize)
warnx("boot block size %% sector-size != 0");
if (lp->d_npartitions > MAXPARTITIONS) {
warnx("number of partitions (%lu) > MAXPARTITIONS (%d)",
(u_long)lp->d_npartitions, MAXPARTITIONS);
errors++;
}
if (lp->d_npartitions < DEFPARTITIONS) {
warnx("number of partitions (%lu) < DEFPARTITIONS (%d)",
(u_long)lp->d_npartitions, DEFPARTITIONS);
errors++;
}
} else {
struct disklabel *vl;
vl = getvirginlabel();
if (lp->d_secsize == 0)
lp->d_secsize = vl->d_secsize;
if (lp->d_nsectors == 0)
lp->d_nsectors = vl->d_nsectors;
if (lp->d_ntracks == 0)
lp->d_ntracks = vl->d_ntracks;
if (lp->d_ncylinders == 0)
lp->d_ncylinders = vl->d_ncylinders;
if (lp->d_rpm == 0)
lp->d_rpm = vl->d_rpm;
if (lp->d_interleave == 0)
lp->d_interleave = vl->d_interleave;
if (lp->d_secpercyl == 0)
lp->d_secpercyl = vl->d_secpercyl;
if (lp->d_secperunit == 0 ||
lp->d_secperunit > vl->d_secperunit)
lp->d_secperunit = vl->d_secperunit;
if (lp->d_bbsize == 0)
lp->d_bbsize = vl->d_bbsize;
if (lp->d_npartitions < DEFPARTITIONS ||
lp->d_npartitions > MAXPARTITIONS)
lp->d_npartitions = vl->d_npartitions;
}
/* first allocate space to the partitions, then offsets */
total_size = 0; /* in sectors */
total_percent = 0; /* in percent */
hog_part = -1;
/* find all fixed partitions */
for (i = 0; i < lp->d_npartitions; i++) {
pp = &lp->d_partitions[i];
if (part_set[i]) {
if (part_size_type[i] == '*') {
if (i == RAW_PART) {
pp->p_size = lp->d_secperunit;
} else {
if (hog_part != -1)
warnx("Too many '*' partitions (%c and %c)",
hog_part + 'a',i + 'a');
else
hog_part = i;
}
} else {
off_t size;
size = pp->p_size;
switch (part_size_type[i]) {
case '%':
total_percent += size;
break;
case 't':
case 'T':
size *= 1024ULL;
/* FALLTHROUGH */
case 'g':
case 'G':
size *= 1024ULL;
/* FALLTHROUGH */
case 'm':
case 'M':
size *= 1024ULL;
/* FALLTHROUGH */
case 'k':
case 'K':
size *= 1024ULL;
break;
case '\0':
break;
default:
warnx("unknown multiplier suffix '%c' for partition %c (should be K, M, G or T)",
part_size_type[i], i + 'a');
break;
}
/* don't count %'s yet */
if (part_size_type[i] != '%') {
/*
* for all not in sectors, convert to
* sectors
*/
if (part_size_type[i] != '\0') {
if (size % lp->d_secsize != 0)
warnx("partition %c not an integer number of sectors",
i + 'a');
size /= lp->d_secsize;
pp->p_size = size;
}
/* else already in sectors */
if (i != RAW_PART)
total_size += size;
}
}
}
}
/* Find out the total free space, excluding the boot block area. */
base_offset = BBSIZE / secsize;
free_space = 0;
for (i = 0; i < lp->d_npartitions; i++) {
pp = &lp->d_partitions[i];
if (!part_set[i] || i == RAW_PART ||
part_size_type[i] == '%' || part_size_type[i] == '*')
continue;
if (pp->p_offset > base_offset)
free_space += pp->p_offset - base_offset;
if (pp->p_offset + pp->p_size > base_offset)
base_offset = pp->p_offset + pp->p_size;
}
if (base_offset < lp->d_secperunit)
free_space += lp->d_secperunit - base_offset;
/* handle % partitions - note %'s don't need to add up to 100! */
if (total_percent != 0) {
if (total_percent > 100) {
fprintf(stderr,"total percentage %lu is greater than 100\n",
total_percent);
errors++;
}
if (free_space > 0) {
for (i = 0; i < lp->d_npartitions; i++) {
pp = &lp->d_partitions[i];
if (part_set[i] && part_size_type[i] == '%') {
/* careful of overflows! and integer roundoff */
pp->p_size = ((double)pp->p_size/100) * free_space;
total_size += pp->p_size;
/* FIX we can lose a sector or so due to roundoff per
partition. A more complex algorithm could avoid that */
}
}
} else {
fprintf(stderr,
"%ld sectors available to give to '*' and '%%' partitions\n",
free_space);
errors++;
/* fix? set all % partitions to size 0? */
}
}
/* give anything remaining to the hog partition */
if (hog_part != -1) {
/*
* Find the range of offsets usable by '*' partitions around
* the hog partition and how much space they need.
*/
needed = 0;
base_offset = BBSIZE / secsize;
for (i = hog_part - 1; i >= 0; i--) {
pp = &lp->d_partitions[i];
if (!part_set[i] || i == RAW_PART)
continue;
if (part_offset_type[i] == '*') {
needed += pp->p_size;
continue;
}
base_offset = pp->p_offset + pp->p_size;
break;
}
current_offset = lp->d_secperunit;
for (i = lp->d_npartitions - 1; i > hog_part; i--) {
pp = &lp->d_partitions[i];
if (!part_set[i] || i == RAW_PART)
continue;
if (part_offset_type[i] == '*') {
needed += pp->p_size;
continue;
}
current_offset = pp->p_offset;
}
if (current_offset - base_offset <= needed) {
fprintf(stderr, "Cannot find space for partition %c\n",
hog_part + 'a');
fprintf(stderr,
"Need more than %lu sectors between %lu and %lu\n",
needed, base_offset, current_offset);
errors++;
lp->d_partitions[hog_part].p_size = 0;
} else {
lp->d_partitions[hog_part].p_size = current_offset -
base_offset - needed;
total_size += lp->d_partitions[hog_part].p_size;
}
}
/* Now set the offsets for each partition */
current_offset = BBSIZE / secsize; /* in sectors */
seen_default_offset = 0;
for (i = 0; i < lp->d_npartitions; i++) {
part = 'a' + i;
pp = &lp->d_partitions[i];
if (part_set[i]) {
if (part_offset_type[i] == '*') {
if (i == RAW_PART) {
pp->p_offset = 0;
} else {
pp->p_offset = current_offset;
seen_default_offset = 1;
}
} else {
/* allow them to be out of order for old-style tables */
if (pp->p_offset < current_offset &&
seen_default_offset && i != RAW_PART &&
pp->p_fstype != FS_VINUM) {
fprintf(stderr,
"Offset %ld for partition %c overlaps previous partition which ends at %lu\n",
(long)pp->p_offset,i+'a',current_offset);
fprintf(stderr,
"Labels with any *'s for offset must be in ascending order by sector\n");
errors++;
} else if (pp->p_offset != current_offset &&
i != RAW_PART && seen_default_offset) {
/*
* this may give unneeded warnings if
* partitions are out-of-order
*/
warnx(
"Offset %ld for partition %c doesn't match expected value %ld",
(long)pp->p_offset, i + 'a', current_offset);
}
}
if (i != RAW_PART)
current_offset = pp->p_offset + pp->p_size;
}
}
for (i = 0; i < lp->d_npartitions; i++) {
part = 'a' + i;
pp = &lp->d_partitions[i];
if (pp->p_size == 0 && pp->p_offset != 0)
warnx("partition %c: size 0, but offset %lu",
part, (u_long)pp->p_offset);
#ifdef notdef
if (pp->p_size % lp->d_secpercyl)
warnx("partition %c: size %% cylinder-size != 0",
part);
if (pp->p_offset % lp->d_secpercyl)
warnx("partition %c: offset %% cylinder-size != 0",
part);
#endif
if (pp->p_offset > lp->d_secperunit) {
fprintf(stderr,
"partition %c: offset past end of unit\n", part);
errors++;
}
if (pp->p_offset + pp->p_size > lp->d_secperunit) {
fprintf(stderr,
"partition %c: partition extends past end of unit\n",
part);
errors++;
}
if (i == RAW_PART) {
if (pp->p_fstype != FS_UNUSED)
warnx("partition %c is not marked as unused!",part);
if (pp->p_offset != 0)
warnx("partition %c doesn't start at 0!",part);
if (pp->p_size != lp->d_secperunit)
warnx("partition %c doesn't cover the whole unit!",part);
if ((pp->p_fstype != FS_UNUSED) || (pp->p_offset != 0) ||
(pp->p_size != lp->d_secperunit)) {
warnx("An incorrect partition %c may cause problems for "
"standard system utilities",part);
}
}
/* check for overlaps */
/* this will check for all possible overlaps once and only once */
for (j = 0; j < i; j++) {
pp2 = &lp->d_partitions[j];
if (j != RAW_PART && i != RAW_PART &&
pp->p_fstype != FS_VINUM &&
pp2->p_fstype != FS_VINUM &&
part_set[i] && part_set[j]) {
if (pp2->p_offset < pp->p_offset + pp->p_size &&
(pp2->p_offset + pp2->p_size > pp->p_offset ||
pp2->p_offset >= pp->p_offset)) {
fprintf(stderr,"partitions %c and %c overlap!\n",
j + 'a', i + 'a');
errors++;
}
}
}
}
for (; i < lp->d_npartitions; i++) {
part = 'a' + i;
pp = &lp->d_partitions[i];
if (pp->p_size || pp->p_offset)
warnx("unused partition %c: size %d offset %lu",
'a' + i, pp->p_size, (u_long)pp->p_offset);
}
return (errors);
}