/* * Mach Operating System * Copyright (c) 1992 Carnegie Mellon University * All Rights Reserved. * * Permission to use, copy, modify and distribute this software and its * documentation is hereby granted, provided that both the copyright * notice and this permission notice appear in all copies of the * software, derivative works or modified versions, and any portions * thereof, and that both notices appear in supporting documentation. * * CARNEGIE MELLON ALLOWS FREE USE OF THIS SOFTWARE IN ITS "AS IS" * CONDITION. CARNEGIE MELLON DISCLAIMS ANY LIABILITY OF ANY KIND FOR * ANY DAMAGES WHATSOEVER RESULTING FROM THE USE OF THIS SOFTWARE. * * Carnegie Mellon requests users of this software to return to * * Software Distribution Coordinator or Software.Distribution@CS.CMU.EDU * School of Computer Science * Carnegie Mellon University * Pittsburgh PA 15213-3890 * * any improvements or extensions that they make and grant Carnegie Mellon * the rights to redistribute these changes. */ #include #include #include #include #include #include int iotest; #define LBUF 100 static char lbuf[LBUF]; /* * * Ported to 386bsd by Julian Elischer Thu Oct 15 20:26:46 PDT 1992 * * 14-Dec-89 Robert Baron (rvb) at Carnegie-Mellon University * Copyright (c) 1989 Robert. V. Baron * Created. */ #define Decimal(str, ans, tmp) if (decimal(str, &tmp, ans)) ans = tmp #define Hex(str, ans, tmp) if (hex(str, &tmp, ans)) ans = tmp #define String(str, ans, len) {char *z = ans; char **dflt = &z; if (string(str, dflt)) strncpy(ans, *dflt, len); } #define RoundCyl(x) ((((x) + cylsecs - 1) / cylsecs) * cylsecs) #define SECSIZE 512 char *disk = "/dev/rwd0d"; char *name; struct disklabel disklabel; /* disk parameters */ int cyls, sectors, heads, cylsecs, disksecs; struct mboot { unsigned char padding[2]; /* force the longs to be long alligned */ unsigned char bootinst[DOSPARTOFF]; struct dos_partition parts[4]; unsigned short int signature; }; struct mboot mboot; #define ACTIVE 0x80 #define BOOT_MAGIC 0xAA55 int dos_cyls; int dos_heads; int dos_sectors; int dos_cylsecs; #define DOSSECT(s,c) ((s & 0x3f) | ((c >> 2) & 0xc0)) #define DOSCYL(c) (c & 0xff) static int dos(); char *get_type(); static int partition = -1; static int a_flag = 0; /* set active partition */ static int i_flag = 0; /* replace partition data */ static int u_flag = 0; /* update partition data */ static unsigned char bootcode[] = { 0x33, 0xc0, 0xfa, 0x8e, 0xd0, 0xbc, 0x00, 0x7c, 0x8e, 0xc0, 0x8e, 0xd8, 0xfb, 0x8b, 0xf4, 0xbf, 0x00, 0x06, 0xb9, 0x00, 0x02, 0xfc, 0xf3, 0xa4, 0xea, 0x1d, 0x06, 0x00, 0x00, 0xb0, 0x04, 0xbe, 0xbe, 0x07, 0x80, 0x3c, 0x80, 0x74, 0x0c, 0x83, 0xc6, 0x10, 0xfe, 0xc8, 0x75, 0xf4, 0xbe, 0xbd, 0x06, 0xeb, 0x43, 0x8b, 0xfe, 0x8b, 0x14, 0x8b, 0x4c, 0x02, 0x83, 0xc6, 0x10, 0xfe, 0xc8, 0x74, 0x0a, 0x80, 0x3c, 0x80, 0x75, 0xf4, 0xbe, 0xbd, 0x06, 0xeb, 0x2b, 0xbd, 0x05, 0x00, 0xbb, 0x00, 0x7c, 0xb8, 0x01, 0x02, 0xcd, 0x13, 0x73, 0x0c, 0x33, 0xc0, 0xcd, 0x13, 0x4d, 0x75, 0xef, 0xbe, 0x9e, 0x06, 0xeb, 0x12, 0x81, 0x3e, 0xfe, 0x7d, 0x55, 0xaa, 0x75, 0x07, 0x8b, 0xf7, 0xea, 0x00, 0x7c, 0x00, 0x00, 0xbe, 0x85, 0x06, 0x2e, 0xac, 0x0a, 0xc0, 0x74, 0x06, 0xb4, 0x0e, 0xcd, 0x10, 0xeb, 0xf4, 0xfb, 0xeb, 0xfe, 'M', 'i', 's', 's', 'i', 'n', 'g', ' ', 'o', 'p', 'e', 'r', 'a', 't', 'i', 'n', 'g', ' ', 's', 'y', 's', 't', 'e', 'm', 0, 'E', 'r', 'r', 'o', 'r', ' ', 'l', 'o', 'a', 'd', 'i', 'n', 'g', ' ', 'o', 'p', 'e', 'r', 'a', 't', 'i', 'n', 'g', ' ', 's', 'y', 's', 't', 'e', 'm', 0, 'I', 'n', 'v', 'a', 'l', 'i', 'd', ' ', 'p', 'a', 'r', 't', 'i', 't', 'i', 'o', 'n', ' ', 't', 'a', 'b', 'l', 'e', 0, 'A', 'u', 't', 'h', 'o', 'r', ' ', '-', ' ', 'S', 'i', 'e', 'g', 'm', 'a', 'r', ' ', 'S', 'c', 'h', 'm', 'i', 'd', 't', 0,0,0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0 }; struct part_type { unsigned char type; char *name; }part_types[] = { {0x00, "unused"} ,{0x01, "Primary DOS with 12 bit FAT"} ,{0x02, "XENIX / filesystem"} ,{0x03, "XENIX /usr filesystem"} ,{0x04, "Primary DOS with 16 bit FAT"} ,{0x05, "Extended DOS"} ,{0x06, "Primary 'big' DOS (> 32MB)"} ,{0x07, "OS/2 HPFS, QNX or Advanced UNIX"} ,{0x08, "AIX filesystem"} ,{0x09, "AIX boot partition or Coherent"} ,{0x0A, "OS/2 Boot Manager or OPUS"} ,{0x10, "OPUS"} ,{0x40, "VENIX 286"} ,{0x50, "DM"} ,{0x51, "DM"} ,{0x52, "CP/M or Microport SysV/AT"} ,{0x56, "GB"} ,{0x61, "Speed"} ,{0x63, "ISC UNIX, other System V/386, GNU HURD or Mach"} ,{0x64, "Novell Netware 2.xx"} ,{0x65, "Novell Netware 3.xx"} ,{0x75, "PCIX"} ,{0x80, "Minix 1.1 ... 1.4a"} ,{0x81, "Minix 1.4b ... 1.5.10"} ,{0x82, "Linux"} ,{0x93, "Amoeba filesystem"} ,{0x94, "Amoeba bad block table"} ,{0xA5, "386BSD"} ,{0xB7, "BSDI BSD/386 filesystem"} ,{0xB8, "BSDI BSD/386 swap"} ,{0xDB, "Concurrent CPM or C.DOS or CTOS"} ,{0xE1, "Speed"} ,{0xE3, "Speed"} ,{0xE4, "Speed"} ,{0xF1, "Speed"} ,{0xF2, "DOS 3.3+ Secondary"} ,{0xF4, "Speed"} ,{0xFF, "BBT (Bad Blocks Table)"} }; main(argc, argv) char **argv; { int i; name = *argv; {register char *cp = name; while (*cp) if (*cp++ == '/') name = cp; } for ( argv++ ; --argc ; argv++ ) { register char *token = *argv; if (*token++ != '-' || !*token) break; else { register int flag; for ( ; flag = *token++ ; ) { switch (flag) { case '0': partition = 0; break; case '1': partition = 1; break; case '2': partition = 2; break; case '3': partition = 3; break; case 'a': a_flag = 1; break; case 'i': i_flag = 1; case 'u': u_flag = 1; break; default: goto usage; } } } } if (argc > 0) disk = argv[0]; if (open_disk(u_flag) < 0) exit(1); printf("******* Working on device %s *******\n",disk); if(u_flag) { get_params_to_use(); } else { print_params(); } if (read_s0()) init_sector0(1); printf("Warning: BIOS sector numbering starts with sector 1\n"); printf("Information from DOS bootblock is:\n"); if (partition == -1) for (i = 0; i < NDOSPART; i++) change_part(i); else change_part(partition); if (u_flag || a_flag) change_active(partition); if (u_flag || a_flag) { printf("\nWe haven't changed the partition table yet. "); printf("This is your last chance.\n"); print_s0(-1); if (ok("Should we write new partition table?")) write_s0(); } exit(0); usage: printf("fdisk {-a|-i|-r} {disk}\n"); } print_s0(which) { int i; print_params(); printf("Information from DOS bootblock is:\n"); if (which == -1) for (i = 0; i < NDOSPART; i++) printf("%d: ", i), print_part(i); else print_part(which); } static struct dos_partition mtpart = { 0 }; print_part(i) { struct dos_partition *partp = ((struct dos_partition *) &mboot.parts) + i; if (!bcmp(partp, &mtpart, sizeof (struct dos_partition))) { printf("\n"); return; } printf("sysid %d,(%s)\n", partp->dp_typ, get_type(partp->dp_typ)); printf(" start %d, size %d (%d Meg), flag %x\n", partp->dp_start, partp->dp_size, partp->dp_size * 512 / (1024 * 1024), partp->dp_flag); printf("\tbeg: cyl %d/ sector %d/ head %d;\n\tend: cyl %d/ sector %d/ head %d\n" ,DPCYL(partp->dp_scyl, partp->dp_ssect) ,DPSECT(partp->dp_ssect) ,partp->dp_shd ,DPCYL(partp->dp_ecyl, partp->dp_esect) ,DPSECT(partp->dp_esect) ,partp->dp_ehd); } init_sector0(start) { struct dos_partition *partp = (struct dos_partition *) (&mboot.parts[3]); int size = disksecs - start; int rest; memcpy(mboot.bootinst, bootcode, sizeof(bootcode)); mboot.signature = BOOT_MAGIC; partp->dp_typ = DOSPTYP_386BSD; partp->dp_flag = ACTIVE; partp->dp_start = start; partp->dp_size = size; dos(partp->dp_start, &partp->dp_scyl, &partp->dp_ssect, &partp->dp_shd); dos(partp->dp_start+partp->dp_size, &partp->dp_ecyl, &partp->dp_esect, &partp->dp_ehd); } change_part(i) { struct dos_partition *partp = ((struct dos_partition *) &mboot.parts) + i; printf("The data for partition %d is:\n", i); print_part(i); if (u_flag && ok("Do you want to change it?")) { int tmp; if (i_flag) { bzero((char *)partp, sizeof (struct dos_partition)); if (i == 3) { init_sector0(1); printf("\nThe static data for the DOS partition 3 has been reinitialized to:\n"); print_part(i); } } do { Decimal("sysid", partp->dp_typ, tmp); Decimal("start", partp->dp_start, tmp); Decimal("size", partp->dp_size, tmp); if (ok("Explicitly specifiy beg/end address ?")) { int tsec,tcyl,thd; tcyl = DPCYL(partp->dp_scyl,partp->dp_ssect); thd = partp->dp_shd; tsec = DPSECT(partp->dp_ssect); Decimal("beginning cylinder", tcyl, tmp); Decimal("beginning head", thd, tmp); Decimal("beginning sector", tsec, tmp); partp->dp_scyl = DOSCYL(tcyl); partp->dp_ssect = DOSSECT(tsec,tcyl); partp->dp_shd = thd; tcyl = DPCYL(partp->dp_ecyl,partp->dp_esect); thd = partp->dp_ehd; tsec = DPSECT(partp->dp_esect); Decimal("ending cylinder", tcyl, tmp); Decimal("ending head", thd, tmp); Decimal("ending sector", tsec, tmp); partp->dp_ecyl = DOSCYL(tcyl); partp->dp_esect = DOSSECT(tsec,tcyl); partp->dp_ehd = thd; } else { dos(partp->dp_start, &partp->dp_scyl, &partp->dp_ssect, &partp->dp_shd); dos(partp->dp_start+partp->dp_size - 1, &partp->dp_ecyl, &partp->dp_esect, &partp->dp_ehd); } print_part(i); } while (!ok("Are we happy with this entry?")); } } print_params() { printf("parameters extracted from in-core disklabel are:\n"); printf("cylinders=%d heads=%d sectors/track=%d (%d blks/cyl)\n\n" ,cyls,heads,sectors,cylsecs); if((dos_sectors > 63) || (dos_cyls > 1023) || (dos_heads > 255)) printf(" Figures below won't work with BIOS for partitions not in cyl 1\n"); printf("parameters to be used for BIOS calculations are:\n"); printf("cylinders=%d heads=%d sectors/track=%d (%d blks/cyl)\n\n" ,dos_cyls,dos_heads,dos_sectors,dos_cylsecs); } change_active(which) { int i; int active = 3, tmp; struct dos_partition *partp = ((struct dos_partition *) &mboot.parts); if (a_flag && which != -1) active = which; if (ok("Do you want to change the active partition?")) { do Decimal("active partition", active, tmp); while(!ok("Are you happy with this choice")); } for (i = 0; i < NDOSPART; i++) partp[i].dp_flag = 0; partp[active].dp_flag = ACTIVE; } get_params_to_use() { int tmp; print_params(); if (ok("Do you want to change our idea of what BIOS thinks ?")) { do { Decimal("BIOS's idea of #cylinders", dos_cyls, tmp); Decimal("BIOS's idea of #heads", dos_heads, tmp); Decimal("BIOS's idea of #sectors", dos_sectors, tmp); dos_cylsecs = dos_heads * dos_sectors; print_params(); } while(!ok("Are you happy with this choice")); } } /***********************************************\ * Change real numbers into strange dos numbers * \***********************************************/ static dos(sec, c, s, h) int sec; unsigned char *c, *s, *h; { int cy; int hd; cy = sec / ( dos_cylsecs ); sec = sec - cy * ( dos_cylsecs ); hd = sec / dos_sectors; sec = (sec - hd * dos_sectors) + 1; *h = hd; *c = cy & 0xff; *s = (sec & 0x3f) | ( (cy & 0x300) >> 2); } int fd; /* Getting device status */ open_disk(u_flag) { struct stat st; if (stat(disk, &st) == -1) { fprintf(stderr, "%s: Can't get file status of %s\n", name, disk); return -1; } else if ( !(st.st_mode & S_IFCHR) ) { fprintf(stderr,"%s: Device %s is not character special\n", name, disk); return -1; } if ((fd = open(disk, u_flag?O_RDWR:O_RDONLY)) == -1) { fprintf(stderr,"%s: Can't open device %s\n", name, disk); return -1; } if (get_params(0) == -1) { fprintf(stderr, "%s: Can't get disk parameters on %s\n", name, disk); return -1; } return fd; } read_disk(sector, buf) { lseek(fd,(sector * 512), 0); return read(fd, buf, 512); } write_disk(sector, buf) { lseek(fd,(sector * 512), 0); return write(fd, buf, 512); } get_params(verbose) { if (ioctl(fd, DIOCGDINFO, &disklabel) == -1) { return -1; } dos_cyls = cyls = disklabel.d_ncylinders; dos_heads = heads = disklabel.d_ntracks; dos_sectors = sectors = disklabel.d_nsectors; dos_cylsecs = cylsecs = heads * sectors; disksecs = cyls * heads * sectors; return (disksecs); } read_s0() { if (read_disk(0, (char *) mboot.bootinst) == -1) { fprintf(stderr, "%s: Can't read fdisk partition table\n", name); return -1; } if (mboot.signature != BOOT_MAGIC) { fprintf(stderr, "%s: Invalid fdisk partition table found\n", name); /* So should we initialize things */ return -1; } return 0; } write_s0() { int flag; if (iotest) { print_s0(-1); return 0; } /* * write enable label sector before write (if necessary), * disable after writing. * needed if the disklabel protected area also protects * sector 0. (e.g. empty disk) */ flag = 1; if (ioctl(fd, DIOCWLABEL, &flag) < 0) perror("ioctl DIOCWLABEL"); if (write_disk(0, (char *) mboot.bootinst) == -1) { fprintf(stderr, "%s: Can't write fdisk partition table\n", name); return -1; flag = 0; (void) ioctl(fd, DIOCWLABEL, &flag); } } ok(str) char *str; { printf("%s [n] ", str); fgets(lbuf, LBUF, stdin); lbuf[strlen(lbuf)-1] = 0; if (*lbuf && (!strcmp(lbuf, "yes") || !strcmp(lbuf, "YES") || !strcmp(lbuf, "y") || !strcmp(lbuf, "Y"))) return 1; else return 0; } decimal(str, num, deflt) char *str; int *num; { int acc = 0, c; char *cp; while (1) { printf("Supply a decimal value for \"%s\" [%d] ", str, deflt); fgets(lbuf, LBUF, stdin); lbuf[strlen(lbuf)-1] = 0; if (!*lbuf) return 0; cp = lbuf; while ((c = *cp) && (c == ' ' || c == '\t')) cp++; if (!c) return 0; while (c = *cp++) { if (c <= '9' && c >= '0') acc = acc * 10 + c - '0'; else break; } if (c == ' ' || c == '\t') while ((c = *cp) && (c == ' ' || c == '\t')) cp++; if (!c) { *num = acc; return 1; } else printf("%s is an invalid decimal number. Try again\n", lbuf); } } hex(str, num, deflt) char *str; int *num; { int acc = 0, c; char *cp; while (1) { printf("Supply a hex value for \"%s\" [%x] ", str, deflt); fgets(lbuf, LBUF, stdin); lbuf[strlen(lbuf)-1] = 0; if (!*lbuf) return 0; cp = lbuf; while ((c = *cp) && (c == ' ' || c == '\t')) cp++; if (!c) return 0; while (c = *cp++) { if (c <= '9' && c >= '0') acc = (acc << 4) + c - '0'; else if (c <= 'f' && c >= 'a') acc = (acc << 4) + c - 'a' + 10; else if (c <= 'F' && c >= 'A') acc = (acc << 4) + c - 'A' + 10; else break; } if (c == ' ' || c == '\t') while ((c = *cp) && (c == ' ' || c == '\t')) cp++; if (!c) { *num = acc; return 1; } else printf("%s is an invalid hex number. Try again\n", lbuf); } } string(str, ans) char *str; char **ans; { int c; char *cp = lbuf; while (1) { printf("Supply a string value for \"%s\" [%s] ", str, *ans); fgets(lbuf, LBUF, stdin); lbuf[strlen(lbuf)-1] = 0; if (!*lbuf) return 0; while ((c = *cp) && (c == ' ' || c == '\t')) cp++; if (c == '"') { c = *++cp; *ans = cp; while ((c = *cp) && c != '"') cp++; } else { *ans = cp; while ((c = *cp) && c != ' ' && c != '\t') cp++; } if (c) *cp = 0; return 1; } } char *get_type(type) int type; { int numentries = (sizeof(part_types)/sizeof(struct part_type)); int counter = 0; struct part_type *ptr = part_types; while(counter < numentries) { if(ptr->type == type) { return(ptr->name); } ptr++; counter++; } return("unknown"); }