1 /* 2 * Copyright (c) 1983, 1988 Regents of the University of California. 3 * All rights reserved. 4 * 5 * Redistribution and use in source and binary forms, with or without 6 * modification, are permitted provided that the following conditions 7 * are met: 8 * 1. Redistributions of source code must retain the above copyright 9 * notice, this list of conditions and the following disclaimer. 10 * 2. Redistributions in binary form must reproduce the above copyright 11 * notice, this list of conditions and the following disclaimer in the 12 * documentation and/or other materials provided with the distribution. 13 * 3. All advertising materials mentioning features or use of this software 14 * must display the following acknowledgement: 15 * This product includes software developed by the University of 16 * California, Berkeley and its contributors. 17 * 4. Neither the name of the University nor the names of its contributors 18 * may be used to endorse or promote products derived from this software 19 * without specific prior written permission. 20 * 21 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND 22 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 23 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 24 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE 25 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 26 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 27 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 28 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 29 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 30 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 31 * SUCH DAMAGE. 32 */ 33 34 #ifndef lint 35 char copyright[] = 36 "@(#) Copyright (c) 1983, 1988 Regents of the University of California.\n\ 37 All rights reserved.\n"; 38 #endif /* not lint */ 39 40 #ifndef lint 41 static char sccsid[] = "@(#)diskpart.c 5.11 (Berkeley) 6/1/90"; 42 #endif /* not lint */ 43 44 /* 45 * Program to calculate standard disk partition sizes. 46 */ 47 #include <sys/param.h> 48 #define DKTYPENAMES 49 #include <sys/disklabel.h> 50 51 #include <stdio.h> 52 #include <ctype.h> 53 54 #define for_now /* show all of `c' partition for disklabel */ 55 #define NPARTITIONS 8 56 #define PART(x) (x - 'a') 57 58 /* 59 * Default partition sizes, where they exist. 60 */ 61 #define NDEFAULTS 4 62 int defpart[NDEFAULTS][NPARTITIONS] = { 63 { 15884, 66880, 0, 15884, 307200, 0, 0, 291346 }, /* ~ 356+ Mbytes */ 64 { 15884, 33440, 0, 15884, 55936, 0, 0, 291346 }, /* ~ 206-355 Mbytes */ 65 { 15884, 33440, 0, 15884, 55936, 0, 0, 0 }, /* ~ 61-205 Mbytes */ 66 { 15884, 10032, 0, 15884, 0, 0, 0, 0 }, /* ~ 20-60 Mbytes */ 67 }; 68 69 /* 70 * Each array defines a layout for a disk; 71 * that is, the collection of partitions totally 72 * covers the physical space on a disk. 73 */ 74 #define NLAYOUTS 3 75 char layouts[NLAYOUTS][NPARTITIONS] = { 76 { 'a', 'b', 'h', 'g' }, 77 { 'a', 'b', 'h', 'd', 'e', 'f' }, 78 { 'c' }, 79 }; 80 81 /* 82 * Default disk block and disk block fragment 83 * sizes for each file system. Those file systems 84 * with zero block and frag sizes are special cases 85 * (e.g. swap areas or for access to the entire device). 86 */ 87 struct partition defparam[NPARTITIONS] = { 88 { 0, 0, 1024, FS_UNUSED, 8, 0 }, /* a */ 89 { 0, 0, 1024, FS_SWAP, 8, 0 }, /* b */ 90 { 0, 0, 1024, FS_UNUSED, 8, 0 }, /* c */ 91 { 0, 0, 512, FS_UNUSED, 8, 0 }, /* d */ 92 { 0, 0, 1024, FS_UNUSED, 8, 0 }, /* e */ 93 { 0, 0, 1024, FS_UNUSED, 8, 0 }, /* f */ 94 { 0, 0, 1024, FS_UNUSED, 8, 0 }, /* g */ 95 { 0, 0, 1024, FS_UNUSED, 8, 0 } /* h */ 96 }; 97 98 /* 99 * Each disk has some space reserved for a bad sector 100 * forwarding table. DEC standard 144 uses the first 101 * 5 even numbered sectors in the last track of the 102 * last cylinder for replicated storage of the bad sector 103 * table; another 126 sectors past this is needed as a 104 * pool of replacement sectors. 105 */ 106 int badsecttable = 126; /* # sectors */ 107 108 int pflag; /* print device driver partition tables */ 109 int dflag; /* print disktab entry */ 110 111 struct disklabel *promptfordisk(); 112 113 main(argc, argv) 114 int argc; 115 char *argv[]; 116 { 117 struct disklabel *dp; 118 register int curcyl, spc, def, part, layout, j; 119 int threshhold, numcyls[NPARTITIONS], startcyl[NPARTITIONS]; 120 int totsize = 0; 121 char *lp, *tyname; 122 123 argc--, argv++; 124 if (argc < 1) { 125 fprintf(stderr, 126 "usage: diskpart [ -p ] [ -d ] [ -s size ] disk-type\n"); 127 exit(1); 128 } 129 if (argc > 0 && strcmp(*argv, "-p") == 0) { 130 pflag++; 131 argc--, argv++; 132 } 133 if (argc > 0 && strcmp(*argv, "-d") == 0) { 134 dflag++; 135 argc--, argv++; 136 } 137 if (argc > 1 && strcmp(*argv, "-s") == 0) { 138 totsize = atoi(argv[1]); 139 argc += 2, argv += 2; 140 } 141 dp = getdiskbyname(*argv); 142 if (dp == NULL) { 143 if (isatty(0)) 144 dp = promptfordisk(*argv); 145 if (dp == NULL) { 146 fprintf(stderr, "%s: unknown disk type\n", *argv); 147 exit(2); 148 } 149 } else { 150 if (dp->d_flags & D_REMOVABLE) 151 tyname = "removable"; 152 else if (dp->d_flags & D_RAMDISK) 153 tyname = "simulated"; 154 else 155 tyname = "winchester"; 156 } 157 spc = dp->d_secpercyl; 158 /* 159 * Bad sector table contains one track for the replicated 160 * copies of the table and enough full tracks preceding 161 * the last track to hold the pool of free blocks to which 162 * bad sectors are mapped. 163 * If disk size was specified explicitly, use specified size. 164 */ 165 if (dp->d_type == DTYPE_SMD && dp->d_flags & D_BADSECT && 166 totsize == 0) { 167 badsecttable = dp->d_nsectors + 168 roundup(badsecttable, dp->d_nsectors); 169 threshhold = howmany(spc, badsecttable); 170 } else { 171 badsecttable = 0; 172 threshhold = 0; 173 } 174 /* 175 * If disk size was specified, recompute number of cylinders 176 * that may be used, and set badsecttable to any remaining 177 * fraction of the last cylinder. 178 */ 179 if (totsize != 0) { 180 dp->d_ncylinders = howmany(totsize, spc); 181 badsecttable = spc * dp->d_ncylinders - totsize; 182 } 183 184 /* 185 * Figure out if disk is large enough for 186 * expanded swap area and 'd', 'e', and 'f' 187 * partitions. Otherwise, use smaller defaults 188 * based on RK07. 189 */ 190 for (def = 0; def < NDEFAULTS; def++) { 191 curcyl = 0; 192 for (part = PART('a'); part < NPARTITIONS; part++) 193 curcyl += howmany(defpart[def][part], spc); 194 if (curcyl < dp->d_ncylinders - threshhold) 195 break; 196 } 197 if (def >= NDEFAULTS) { 198 fprintf(stderr, "%s: disk too small, calculate by hand\n", 199 *argv); 200 exit(3); 201 } 202 203 /* 204 * Calculate number of cylinders allocated to each disk 205 * partition. We may waste a bit of space here, but it's 206 * in the interest of (very backward) compatibility 207 * (for mixed disk systems). 208 */ 209 for (curcyl = 0, part = PART('a'); part < NPARTITIONS; part++) { 210 numcyls[part] = 0; 211 if (defpart[def][part] != 0) { 212 numcyls[part] = howmany(defpart[def][part], spc); 213 curcyl += numcyls[part]; 214 } 215 } 216 numcyls[PART('f')] = dp->d_ncylinders - curcyl; 217 numcyls[PART('g')] = 218 numcyls[PART('d')] + numcyls[PART('e')] + numcyls[PART('f')]; 219 numcyls[PART('c')] = dp->d_ncylinders; 220 defpart[def][PART('f')] = numcyls[PART('f')] * spc - badsecttable; 221 defpart[def][PART('g')] = numcyls[PART('g')] * spc - badsecttable; 222 defpart[def][PART('c')] = numcyls[PART('c')] * spc; 223 #ifndef for_now 224 if (totsize || !pflag) 225 #else 226 if (totsize) 227 #endif 228 defpart[def][PART('c')] -= badsecttable; 229 230 /* 231 * Calculate starting cylinder number for each partition. 232 * Note the 'h' partition is physically located before the 233 * 'g' or 'd' partition. This is reflected in the layout 234 * arrays defined above. 235 */ 236 for (layout = 0; layout < NLAYOUTS; layout++) { 237 curcyl = 0; 238 for (lp = layouts[layout]; *lp != 0; lp++) { 239 startcyl[PART(*lp)] = curcyl; 240 curcyl += numcyls[PART(*lp)]; 241 } 242 } 243 244 if (pflag) { 245 printf("}, %s_sizes[%d] = {\n", dp->d_typename, NPARTITIONS); 246 for (part = PART('a'); part < NPARTITIONS; part++) { 247 if (numcyls[part] == 0) { 248 printf("\t0,\t0,\n"); 249 continue; 250 } 251 if (dp->d_type != DTYPE_MSCP) { 252 printf("\t%d,\t%d,\t\t/* %c=cyl %d thru %d */\n", 253 defpart[def][part], startcyl[part], 254 'A' + part, startcyl[part], 255 startcyl[part] + numcyls[part] - 1); 256 continue; 257 } 258 printf("\t%d,\t%d,\t\t/* %c=sectors %d thru %d */\n", 259 defpart[def][part], spc * startcyl[part], 260 'A' + part, spc * startcyl[part], 261 spc * startcyl[part] + defpart[def][part] - 1); 262 } 263 exit(0); 264 } 265 if (dflag) { 266 int nparts; 267 268 /* 269 * In case the disk is in the ``in-between'' range 270 * where the 'g' partition is smaller than the 'h' 271 * partition, reverse the frag sizes so the /usr partition 272 * is always set up with a frag size larger than the 273 * user's partition. 274 */ 275 if (defpart[def][PART('g')] < defpart[def][PART('h')]) { 276 int temp; 277 278 temp = defparam[PART('h')].p_fsize; 279 defparam[PART('h')].p_fsize = 280 defparam[PART('g')].p_fsize; 281 defparam[PART('g')].p_fsize = temp; 282 } 283 printf("%s:\\\n", dp->d_typename); 284 printf("\t:ty=%s:ns#%d:nt#%d:nc#%d:", tyname, 285 dp->d_nsectors, dp->d_ntracks, dp->d_ncylinders); 286 if (dp->d_secpercyl != dp->d_nsectors * dp->d_ntracks) 287 printf("sc#%d:", dp->d_secpercyl); 288 if (dp->d_type == DTYPE_SMD && dp->d_flags & D_BADSECT) 289 printf("sf:"); 290 printf("\\\n\t:dt=%s:", dktypenames[dp->d_type]); 291 for (part = NDDATA - 1; part >= 0; part--) 292 if (dp->d_drivedata[part]) 293 break; 294 for (j = 0; j <= part; j++) 295 printf("d%d#%d:", j, dp->d_drivedata[j]); 296 printf("\\\n"); 297 for (nparts = 0, part = PART('a'); part < NPARTITIONS; part++) 298 if (defpart[def][part] != 0) 299 nparts++; 300 for (part = PART('a'); part < NPARTITIONS; part++) { 301 if (defpart[def][part] == 0) 302 continue; 303 printf("\t:p%c#%d:", 'a' + part, defpart[def][part]); 304 printf("o%c#%d:b%c#%d:f%c#%d:", 305 'a' + part, spc * startcyl[part], 306 'a' + part, 307 defparam[part].p_frag * defparam[part].p_fsize, 308 'a' + part, defparam[part].p_fsize); 309 if (defparam[part].p_fstype == FS_SWAP) 310 printf("t%c=swap:", 'a' + part); 311 nparts--; 312 printf("%s\n", nparts > 0 ? "\\" : ""); 313 } 314 #ifdef for_now 315 defpart[def][PART('c')] -= badsecttable; 316 part = PART('c'); 317 printf("#\t:p%c#%d:", 'a' + part, defpart[def][part]); 318 printf("o%c#%d:b%c#%d:f%c#%d:\n", 319 'a' + part, spc * startcyl[part], 320 'a' + part, 321 defparam[part].p_frag * defparam[part].p_fsize, 322 'a' + part, defparam[part].p_fsize); 323 #endif 324 exit(0); 325 } 326 printf("%s: #sectors/track=%d, #tracks/cylinder=%d #cylinders=%d\n", 327 dp->d_typename, dp->d_nsectors, dp->d_ntracks, 328 dp->d_ncylinders); 329 printf("\n Partition\t Size\t Offset\t Range\n"); 330 for (part = PART('a'); part < NPARTITIONS; part++) { 331 printf("\t%c\t", 'a' + part); 332 if (numcyls[part] == 0) { 333 printf(" unused\n"); 334 continue; 335 } 336 printf("%7d\t%7d\t%4d - %d%s\n", 337 defpart[def][part], startcyl[part] * spc, 338 startcyl[part], startcyl[part] + numcyls[part] - 1, 339 defpart[def][part] % spc ? "*" : ""); 340 } 341 } 342 343 struct disklabel disk; 344 345 struct field { 346 char *f_name; 347 char *f_defaults; 348 u_long *f_location; 349 } fields[] = { 350 { "sector size", "512", &disk.d_secsize }, 351 { "#sectors/track", 0, &disk.d_nsectors }, 352 { "#tracks/cylinder", 0, &disk.d_ntracks }, 353 { "#cylinders", 0, &disk.d_ncylinders }, 354 { 0, 0, 0 }, 355 }; 356 357 struct disklabel * 358 promptfordisk(name) 359 char *name; 360 { 361 register struct disklabel *dp = &disk; 362 register struct field *fp; 363 register i; 364 char buf[BUFSIZ], **tp, *cp, *gets(); 365 366 strncpy(dp->d_typename, name, sizeof(dp->d_typename)); 367 fprintf(stderr, 368 "%s: unknown disk type, want to supply parameters (y/n)? ", 369 name); 370 (void) gets(buf); 371 if (*buf != 'y') 372 return ((struct disklabel *)0); 373 for (;;) { 374 fprintf(stderr, "Disk/controller type (%s)? ", dktypenames[1]); 375 (void) gets(buf); 376 if (buf[0] == 0) 377 dp->d_type = 1; 378 else 379 dp->d_type = gettype(buf, dktypenames); 380 if (dp->d_type >= 0) 381 break; 382 fprintf(stderr, "%s: unrecognized controller type\n", buf); 383 fprintf(stderr, "use one of:\n", buf); 384 for (tp = dktypenames; *tp; tp++) 385 if (index(*tp, ' ') == 0) 386 fprintf(stderr, "\t%s\n", *tp); 387 } 388 gettype: 389 dp->d_flags = 0; 390 fprintf(stderr, "type (winchester|removable|simulated)? "); 391 (void) gets(buf); 392 if (strcmp(buf, "removable") == 0) 393 dp->d_flags = D_REMOVABLE; 394 else if (strcmp(buf, "simulated") == 0) 395 dp->d_flags = D_RAMDISK; 396 else if (strcmp(buf, "winchester")) { 397 fprintf(stderr, "%s: bad disk type\n", buf); 398 goto gettype; 399 } 400 strncpy(dp->d_typename, buf, sizeof(dp->d_typename)); 401 fprintf(stderr, "(type <cr> to get default value, if only one)\n"); 402 if (dp->d_type == DTYPE_SMD) 403 fprintf(stderr, "Do %ss support bad144 bad block forwarding (yes)? ", 404 dp->d_typename); 405 (void) gets(buf); 406 if (*buf != 'n') 407 dp->d_flags |= D_BADSECT; 408 for (fp = fields; fp->f_name != NULL; fp++) { 409 again: 410 fprintf(stderr, "%s ", fp->f_name); 411 if (fp->f_defaults != NULL) 412 fprintf(stderr, "(%s)", fp->f_defaults); 413 fprintf(stderr, "? "); 414 cp = gets(buf); 415 if (*cp == '\0') { 416 if (fp->f_defaults == NULL) { 417 fprintf(stderr, "no default value\n"); 418 goto again; 419 } 420 cp = fp->f_defaults; 421 } 422 *fp->f_location = atol(cp); 423 if (*fp->f_location == 0) { 424 fprintf(stderr, "%s: bad value\n", cp); 425 goto again; 426 } 427 } 428 fprintf(stderr, "sectors/cylinder (%d)? ", 429 dp->d_nsectors * dp->d_ntracks); 430 (void) gets(buf); 431 if (buf[0] == 0) 432 dp->d_secpercyl = dp->d_nsectors * dp->d_ntracks; 433 else 434 dp->d_secpercyl = atol(buf); 435 fprintf(stderr, "Drive-type-specific parameters, <cr> to terminate:\n"); 436 for (i = 0; i < NDDATA; i++) { 437 fprintf(stderr, "d%d? ", i); 438 (void) gets(buf); 439 if (buf[0] == 0) 440 break; 441 dp->d_drivedata[i] = atol(buf); 442 } 443 return (dp); 444 } 445 446 gettype(t, names) 447 char *t; 448 char **names; 449 { 450 register char **nm; 451 452 for (nm = names; *nm; nm++) 453 if (ustrcmp(t, *nm) == 0) 454 return (nm - names); 455 if (isdigit(*t)) 456 return (atoi(t)); 457 return (-1); 458 } 459 460 ustrcmp(s1, s2) 461 register char *s1, *s2; 462 { 463 #define lower(c) (islower(c) ? (c) : tolower(c)) 464 465 for (; *s1; s1++, s2++) { 466 if (*s1 == *s2) 467 continue; 468 if (isalpha(*s1) && isalpha(*s2) && 469 lower(*s1) == lower(*s2)) 470 continue; 471 return (*s2 - *s1); 472 } 473 return (0); 474 } 475