1
0
mirror of https://git.FreeBSD.org/src.git synced 2024-12-21 11:13:30 +00:00
freebsd/sbin/sysinstall/mbr.c
Poul-Henning Kamp d97ce60ae9 Some cleanup done.
Include bteasy, bootsd, sdboot and termcap entries using file2.c

Remove all traces of "termcap.small".

The policy in this program regarding termcap is:

| If $TERM is set
| 	do nothing special, rely on usual termcap.
| else
| 	use compiled in (via file2c) termcap entries
1995-01-29 02:31:38 +00:00

481 lines
12 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 <stdio.h>
#include <unistd.h>
#include <dialog.h>
#include <fcntl.h>
#include <sys/types.h>
#include <sys/disklabel.h>
#include <sys/uio.h>
#include "mbr.h"
#include "sysinstall.h"
extern struct mbr *mbr;
extern int inst_part;
extern int whole_disk;
struct part_type part_types[] = PARTITION_TYPES
char *
part_type(int type)
{
int num_types = (sizeof(part_types)/sizeof(struct part_type));
int next_type = 0;
struct part_type *ptr = part_types;
while (next_type < num_types) {
if(ptr->type == type)
return(ptr->name);
ptr++;
next_type++;
}
return("Unknown");
}
void
read_dospart(int fd, struct dos_partition *dp)
{
u_char buf[512];
if (lseek(fd, 0, SEEK_SET) == -1)
AskAbort("Couldn't seek for master boot record read\n");
if (read(fd, buf, 512) != 512) {
AskAbort("Failed to read master boot record\n");
}
memcpy(dp, buf+DOSPARTOFF, sizeof(*dp)*NDOSPART);
}
void
write_dospart(int fd, struct dos_partition *dp)
{
u_char buf[512];
if (lseek(fd, 0, SEEK_SET) == -1)
AskAbort("Couldn't seek for master boot record read\n");
if (read(fd, buf, 512) != 512) {
AskAbort("Failed to read master boot record\n");
}
memcpy(buf+DOSPARTOFF, dp, sizeof(*dp)*NDOSPART);
buf[510] = 0x55;
buf[511] = 0xaa;
if (lseek(fd, 0, SEEK_SET) == -1)
AskAbort("Couldn't seek for master boot record write\n");
enable_label(fd);
if (write(fd, buf, 512) != 512)
AskAbort("Failed to write master boot record\n");
disable_label(fd);
}
void
write_bootcode(int fd)
{
u_char buf[512];
extern const char boot0[];
if (lseek(fd, 0, SEEK_SET) == -1)
AskAbort("Couldn't seek for master boot record read\n");
if (read(fd, buf, 512) != 512) {
AskAbort("Failed to read master boot record\n");
}
memcpy(buf, boot0, DOSPARTOFF);
buf[510] = 0x55;
buf[511] = 0xaa;
if (lseek(fd, 0, SEEK_SET) == -1)
AskAbort("Couldn't seek for master boot record write\n");
enable_label(fd);
if (write(fd, buf, 512) != 512)
AskAbort("Failed to write master boot record\n");
disable_label(fd);
}
int
WriteBootblock(int dfd,struct disklabel *label,struct dos_partition *dospart)
{
off_t of = label->d_partitions[OURPART].p_offset;
u_char bootblocks[BBSIZE];
extern const char boot1[];
extern const char boot2[];
memcpy(bootblocks, boot1, MBRSIZE);
memcpy(&bootblocks[MBRSIZE], boot2, (int)(label->d_bbsize - MBRSIZE));
bcopy(dospart, &bootblocks[DOSPARTOFF],
sizeof(struct dos_partition) * NDOSPART);
label->d_checksum = 0;
label->d_checksum = dkcksum(label);
bcopy(label, &bootblocks[(LABELSECTOR * label->d_secsize) + LABELOFFSET],
sizeof *label);
Debug("Seeking to byte %ld ", of * label->d_secsize);
if (lseek(dfd, (of * label->d_secsize), SEEK_SET) < 0) {
Fatal("Couldn't seek to start of partition\n");
}
enable_label(dfd);
if (write(dfd, bootblocks, label->d_bbsize) != label->d_bbsize) {
Fatal("Failed to write bootblocks (%p,%d) %d %s\n",
bootblocks, label->d_bbsize,
errno, strerror(errno)
);
}
disable_label(dfd);
return(0);
}
static int
FillIn(struct dos_partition *dp, int sec, int hd)
{
u_long l2,l3=0,sect,c,s,h;
sect = dp->dp_start;
l2 = sect / (sec*hd);
sect -= l2*sec*hd;
if(l2>1023) l2 = 1023;
c = (l2 & 0xff);
s = (l2 >> 2) & 0xc0;
l2 = sect / sec;
h = l2;
sect -= l2*sec;
s |= (sect+1) & 0x3f;
#define NIC(a,b) if (a != b) {a = b; l3++;}
NIC(dp->dp_ssect, s);
NIC(dp->dp_scyl, c);
NIC(dp->dp_shd, h);
sect = dp->dp_start + dp->dp_size-1;
l2 = sect / (sec*hd);
sect -= l2*sec*hd;
if(l2>1023) l2 = 1023;
c = (l2 & 0xff);
s = (l2 >> 2) & 0xc0;
l2 = sect / sec;
h = l2;
sect -= l2*sec;
s |= (sect+1) & 0x3f;
NIC(dp->dp_esect, s);
NIC(dp->dp_ecyl, c);
NIC(dp->dp_ehd, h);
#undef NIC
return l2;
}
void
Fdisk()
{
int i, j, done=0, diskno, flag;
char buf[128];
struct dos_partition dp[NDOSPART];
struct disklabel *lbl;
u_long cyl, hd, sec, tsec;
u_long l, l1, l2, l3, l4;
int changed = 0;
char *grumble = NULL;
*buf = 0;
i = AskEm(stdscr, "Enter number of disk to Fdisk> ", buf, 2);
printf("%d", i);
if(i != '\n' && i != '\r') return;
diskno = atoi(buf);
if(!(diskno >= 0 && diskno < MAX_NO_DISKS && Dname[diskno])) return;
lbl = Dlbl[diskno];
lbl->d_bbsize = 8192;
hd = lbl->d_ntracks;
sec = lbl->d_nsectors;
tsec = Dlbl[diskno]->d_partitions[RAWPART].p_size;
cyl = tsec/(hd*sec);
read_dospart(Dfd[diskno], dp);
while(!done) {
clear(); standend();
j = 0;
mvprintw(j++, 0, "%s -- Diskspace editor -- FDISK", TITLE);
j++;
mvprintw(j++, 0,
"Disk: %s Geometry: %lu Cyl * %lu Hd * %lu Sect",
Dname[diskno], cyl, hd, sec);
printw(" = %luMb = %lu Sect", (tsec+1024)/2048, tsec);
j++;
for(i=0;i<NDOSPART;i++, j+=4) {
mvprintw(j, 0, "%d ", i+1);
#if 0
printw("[%02x %02x %02x %02x %02x %02x %02x %02x %08lx %08lx]\n",
dp[i].dp_flag, dp[i].dp_shd, dp[i].dp_ssect, dp[i].dp_scyl,
dp[i].dp_typ, dp[i].dp_ehd, dp[i].dp_esect, dp[i].dp_ecyl,
dp[i].dp_start, dp[i].dp_size);
#endif
if(!dp[i].dp_size) {
printw("Unused");
continue;
}
printw("Boot?=%s", dp[i].dp_flag == 0x80 ? "Yes" : "No ");
printw(" Type=%s\n", part_type(dp[i].dp_typ));
printw(" Phys=(c%d/h%d/s%d..c%d/h%d/s%d)",
DPCYL(dp[i].dp_scyl, dp[i].dp_ssect), dp[i].dp_shd,
DPSECT(dp[i].dp_ssect),
DPCYL(dp[i].dp_ecyl, dp[i].dp_esect), dp[i].dp_ehd,
DPSECT(dp[i].dp_esect));
printw(" Sector=(%lu..%lu)\n",
dp[i].dp_start, dp[i].dp_size + dp[i].dp_start-1);
printw(" Size=%lu MB, %lu Cylinders", (dp[i].dp_size+1024L)/2048L,
dp[i].dp_size/lbl->d_secpercyl);
l = dp[i].dp_size%lbl->d_secpercyl;
if(l) {
printw(" + %lu Tracks", l/lbl->d_nsectors);
l = l % lbl->d_nsectors;
if(l) {
printw(" + %lu Sectors", l);
}
}
}
mvprintw(20, 0, "Commands available: ");
mvprintw(21, 0, "(H)elp (T)utorial (D)elete (E)dit (R)eread (W)rite MBR (Q)uit");
mvprintw(22, 0, "(U)se entire disk for FreeBSD (G)eometry use (B)oot manager");
if (grumble) {
standout();
mvprintw(24, 0, grumble);
standend();
grumble = NULL;
} else if (changed) {
standout();
mvprintw(24, 0, "Use (W)rite to save changes to disk");
standend();
}
mvprintw(23, 0, "Enter Command> ");
i=getch();
switch(i) {
case 'h': case 'H':
clear();
mvprintw(0, 0,
"%s -- Diskspace editor -- FDISK -- Command Help
Basic commands:
(H)elp - This screen
(T)utorial - A more detailed discussion of MBR's, disklabels, etc.
(D)elete - Delete an existing partition
(E)dit - Edit an existing partition
(R)eread - Read fdisk information from disk again, abandoning changes
(W)rite MBR - Write modified fdisk information to disk
(Q)uit - Exit the FDISK editor
Advanced commands:
(U)se entire disk for FreeBSD - Assign ALL disk space on current drive
(G)eometry - Edit the default disk geometry settings
Write (B)oot manager - Install multi-OS bootmanager.
Press any key to return to FDISK editor...
", TITLE);
getch();
break;
case 't': case 'T':
ShowFile(HELPME_FILE,"Help file for disklayout");
break;
case 'r': case 'R':
read_dospart(Dfd[diskno], dp);
changed=0;
break;
case 'b': case 'B':
grumble = 0;
for(i=0;i<NDOSPART;i++) {
if(dp[i].dp_start == 0 && dp[i].dp_typ== MBR_PTYPE_FreeBSD) {
grumble = "Boot manager not needed.";
break;
}
}
if (!grumble)
write_bootcode(Dfd[diskno]);
grumble = "Wrote boot manager";
break;
case 'e': case 'E':
*buf = 0;
i = AskEm(stdscr, "Edit which Slice> ", buf, 2);
if(i != '\n' && i != '\r') break;
l = strtol(buf, 0, 0);
if(l < 1 || l > NDOSPART) break;
l1=sec; l2=tsec;
for(i=0;i<NDOSPART;i++) {
if((i+1) == l) continue;
if(!dp[i].dp_size) continue;
if(dp[i].dp_start > l2) continue;
if((dp[i].dp_start + dp[i].dp_size) <= l1) continue;
if(dp[i].dp_start > l1)
l3 = dp[i].dp_start - l1;
else
l3 = 0;
if(l2 > (dp[i].dp_start + dp[i].dp_size))
l4 = l2 - (dp[i].dp_start + dp[i].dp_size);
else
l4 = 0;
if(l3 >= l4)
l2 = dp[i].dp_start;
else
l1 = dp[i].dp_start + dp[i].dp_size;
}
sprintf(buf, "%lu", (l2-l1+1024L)/2048L);
i = AskEm(stdscr, "Size of slice in MB> ", buf, 10);
l3=strtol(buf, 0, 0) * 2048L;
if(!l3) break;
if(l3 > l2-l1)
l3 = l2-l1;
if((l1+l3) % lbl->d_secpercyl) { /* Special for cyl==0 */
l3 += lbl->d_secpercyl - ((l1+l3) % lbl->d_secpercyl);
}
if(l3+l1 > tsec)
l3 = tsec - l1;
changed=1;
dp[l-1].dp_start=l1;
dp[l-1].dp_size=l3;
FillIn(&dp[l-1],sec,hd);
l4 = dp[l-1].dp_typ;
if(!l4) l4 = MBR_PTYPE_FreeBSD;
sprintf(buf, "0x%lx", l4);
i = AskEm(stdscr, "Type of slice (0xa5=FreeBSD)> ", buf, 5);
l3 = strtol(buf, 0, 0);
if(l3 == MBR_PTYPE_FreeBSD) {
for(i=0;i<NDOSPART;i++)
if(i != (l-1) && dp[i].dp_typ== MBR_PTYPE_FreeBSD)
memset(&dp[i], 0, sizeof dp[i]);
sprintf(buf, "0x80");
} else {
sprintf(buf, "0");
}
dp[l-1].dp_typ=l3;
i = AskEm(stdscr, "Bootflag (0x80 for YES)> ", buf, 5);
dp[l-1].dp_flag=strtol(buf, 0, 0);
if(dp[l-1].dp_flag)
for(i=0;i<NDOSPART;i++)
if(i != (l-1))
dp[i].dp_flag = 0;
break;
case 'u': case 'U':
memset(&dp[0], 0, sizeof dp);
changed=1;
dp[0].dp_start = 0;
dp[0].dp_size = tsec;
FillIn(&dp[0],sec,hd);
dp[0].dp_typ = MBR_PTYPE_FreeBSD;
dp[0].dp_flag = 0x80;
break;
case 'g': case 'G':
sprintf(buf,"%lu",sec);
i = AskEm(stdscr, "Number of Sectors> ",buf,7);
l2 = strtoul(buf,0,0);
if(l2 != sec)
changed++;
sec=l2;
sprintf(buf,"%lu",hd);
i = AskEm(stdscr, "Number of Heads> ",buf,7);
l2 = strtoul(buf,0,0);
if(l2 != hd)
changed++;
hd=l2;
cyl = tsec/(hd*sec);
sprintf(buf,"%lu",cyl);
i = AskEm(stdscr, "Number of Cylinders> ",buf,7);
l2 = strtoul(buf,0,0);
if(l2 != cyl)
changed++;
cyl=l2;
for (l=0;l<NDOSPART;l++) {
if (!dp[l].dp_typ || !dp[l].dp_size)
continue;
changed += FillIn(&dp[l], sec, hd);
}
break;
case 'd': case 'D':
*buf = 0;
i = AskEm(stdscr, "Delete which Slice> ", buf, 2);
if(i != '\n' && i != '\r') break;
l = strtol(buf, 0, 0);
if(l < 1 || l > NDOSPART) break;
memset(&dp[l-1], 0, sizeof dp[l-1]);
changed=1;
break;
case 'w': case 'W':
write_dospart(Dfd[diskno], dp);
Dlbl[diskno]->d_partitions[OURPART].p_offset = 0;
Dlbl[diskno]->d_partitions[OURPART].p_size = 0;
for(i=0;i<NDOSPART;i++) {
if(dp[i].dp_typ == MBR_PTYPE_FreeBSD) {
Dlbl[diskno]->d_partitions[OURPART].p_offset =
dp[i].dp_start;
Dlbl[diskno]->d_partitions[OURPART].p_size =
dp[i].dp_size;
goto wok;
}
}
grumble = "No FreeBSD slice, cannot write.";
break;
wok:
Dlbl[diskno]->d_ntracks = hd;
Dlbl[diskno]->d_nsectors = sec;
Dlbl[diskno]->d_ncylinders = cyl;
Dlbl[diskno]->d_secpercyl = hd*sec;
Dlbl[diskno]->d_magic = DISKMAGIC;
Dlbl[diskno]->d_magic2 = DISKMAGIC;
Dlbl[diskno]->d_checksum = 0;
Dlbl[diskno]->d_checksum = dkcksum(Dlbl[diskno]);
flag=1;
enable_label(Dfd[diskno]);
if(ioctl(Dfd[diskno], DIOCSDINFO, Dlbl[diskno]) == -1)
AskAbort("Couldn't set label: %s", strerror(errno));
if(ioctl(Dfd[diskno], DIOCWDINFO, Dlbl[diskno]) == -1)
AskAbort("Couldn't write label: %s", strerror(errno));
flag=0;
disable_label(Dfd[diskno]);
changed=0;
if (Dlbl[diskno]->d_partitions[OURPART].p_size) {
WriteBootblock(Dfd[diskno], lbl, dp);
grumble = "Wrote MBR and disklabel to disk";
} else {
grumble = "Wrote MBR to disk";
}
break;
case 'q': case 'Q':
return;
break;
default:
beep();
break;
}
}
}