must zero the buffer
[unix-history] / usr / src / sbin / fsck / setup.c
index fcdc4c4..8a39dce 100644 (file)
@@ -5,27 +5,32 @@
  */
 
 #ifndef lint
  */
 
 #ifndef lint
-static char sccsid[] = "@(#)setup.c    5.3 (Berkeley) %G%";
+static char sccsid[] = "@(#)setup.c    5.6 (Berkeley) %G%";
 #endif not lint
 
 #endif not lint
 
+#define DKTYPENAMES
 #include <sys/param.h>
 #include <sys/inode.h>
 #include <sys/fs.h>
 #include <sys/stat.h>
 #include <sys/param.h>
 #include <sys/inode.h>
 #include <sys/fs.h>
 #include <sys/stat.h>
+#include <sys/ioctl.h>
+#include <sys/disklabel.h>
+#include <sys/file.h>
+#include <ctype.h>
 #include "fsck.h"
 
 #include "fsck.h"
 
+#define POWEROF2(num)  (((num) & ((num) - 1)) == 0)
+
 char   *calloc();
 char   *calloc();
+char   *index();
 
 setup(dev)
        char *dev;
 {
        dev_t rootdev;
 
 setup(dev)
        char *dev;
 {
        dev_t rootdev;
+       long cg, ncg, size, i, j;
        struct stat statb;
        struct stat statb;
-       daddr_t super = bflag ? bflag : SBLOCK;
-       int i, j;
-       long size;
-       BUFAREA asblk;
-#      define altsblock asblk.b_un.b_fs
+       struct fs proto;
 
        if (stat("/", &statb) < 0)
                errexit("Can't stat root\n");
 
        if (stat("/", &statb) < 0)
                errexit("Can't stat root\n");
@@ -45,13 +50,13 @@ setup(dev)
        }
        if (rootdev == statb.st_rdev)
                hotroot++;
        }
        if (rootdev == statb.st_rdev)
                hotroot++;
-       if ((dfile.rfdes = open(dev, 0)) < 0) {
+       if ((dfile.rfdes = open(dev, O_RDONLY)) < 0) {
                printf("Can't open %s\n", dev);
                return (0);
        }
        if (preen == 0)
                printf("** %s", dev);
                printf("Can't open %s\n", dev);
                return (0);
        }
        if (preen == 0)
                printf("** %s", dev);
-       if (nflag || (dfile.wfdes = open(dev, 1)) < 0) {
+       if (nflag || (dfile.wfdes = open(dev, O_WRONLY)) < 0) {
                dfile.wfdes = -1;
                if (preen)
                        pfatal("NO WRITE ACCESS");
                dfile.wfdes = -1;
                if (preen)
                        pfatal("NO WRITE ACCESS");
@@ -65,28 +70,33 @@ setup(dev)
        initbarea(&fileblk);
        initbarea(&inoblk);
        initbarea(&cgblk);
        initbarea(&fileblk);
        initbarea(&inoblk);
        initbarea(&cgblk);
-       initbarea(&asblk);
-       /*
-        * Read in the super block and its summary info.
-        */
-       if (bread(&dfile, (char *)&sblock, super, (long)SBSIZE) != 0)
-               return (0);
-       sblk.b_bno = super;
-       sblk.b_size = SBSIZE;
        /*
        /*
-        * run a few consistency checks of the super block
+        * Read in the superblock, looking for alternates if necessary
         */
         */
-       if (sblock.fs_magic != FS_MAGIC)
-               { badsb("MAGIC NUMBER WRONG"); return (0); }
-       if (sblock.fs_ncg < 1)
-               { badsb("NCG OUT OF RANGE"); return (0); }
-       if (sblock.fs_cpg < 1 || sblock.fs_cpg > MAXCPG)
-               { badsb("CPG OUT OF RANGE"); return (0); }
-       if (sblock.fs_ncg * sblock.fs_cpg < sblock.fs_ncyl ||
-           (sblock.fs_ncg - 1) * sblock.fs_cpg >= sblock.fs_ncyl)
-               { badsb("NCYL DOES NOT JIVE WITH NCG*CPG"); return (0); }
-       if (sblock.fs_sbsize > SBSIZE)
-               { badsb("SIZE PREPOSTEROUSLY LARGE"); return (0); }
+       if (readsb(1) == 0) {
+               if (bflag || calcsb(dev, dfile.rfdes, &proto) == 0 || preen)
+                       return(0);
+               if (reply("LOOK FOR ALTERNATE SUPERBLOCKS") == 0)
+                       return (0);
+               for (cg = 0; cg < proto.fs_ncg; cg++) {
+                       bflag = fsbtodb(&proto, cgsblock(&proto, cg));
+                       if (readsb(0) != 0)
+                               break;
+               }
+               if (cg >= proto.fs_ncg) {
+                       printf("%s %s\n%s %s\n%s %s\n",
+                               "SEARCH FOR ALTERNATE SUPER-BLOCK",
+                               "FAILED. YOU MUST USE THE",
+                               "-b OPTION TO FSCK TO SPECIFY THE",
+                               "LOCATION OF AN ALTERNATE",
+                               "SUPER-BLOCK TO SUPPLY NEEDED",
+                               "INFORMATION; SEE fsck(8).");
+                       return(0);
+               }
+               pwarn("USING ALTERNATE SUPERBLOCK AT %d\n", bflag);
+       }
+       fmax = sblock.fs_size;
+       imax = sblock.fs_ncg * sblock.fs_ipg;
        /*
         * Check and potentially fix certain fields in the super block.
         */
        /*
         * Check and potentially fix certain fields in the super block.
         */
@@ -105,39 +115,22 @@ setup(dev)
                        sbdirty();
                }
        }
                        sbdirty();
                }
        }
-       /*
-        * Set all possible fields that could differ, then do check
-        * of whole super block against an alternate super block.
-        * When an alternate super-block is specified this check is skipped.
-        */
-       if (bflag)
-               goto sbok;
-       getblk(&asblk, cgsblock(&sblock, sblock.fs_ncg - 1), sblock.fs_sbsize);
-       if (asblk.b_errs != NULL)
-               return (0);
-       altsblock.fs_link = sblock.fs_link;
-       altsblock.fs_rlink = sblock.fs_rlink;
-       altsblock.fs_time = sblock.fs_time;
-       altsblock.fs_cstotal = sblock.fs_cstotal;
-       altsblock.fs_cgrotor = sblock.fs_cgrotor;
-       altsblock.fs_fmod = sblock.fs_fmod;
-       altsblock.fs_clean = sblock.fs_clean;
-       altsblock.fs_ronly = sblock.fs_ronly;
-       altsblock.fs_flags = sblock.fs_flags;
-       altsblock.fs_maxcontig = sblock.fs_maxcontig;
-       altsblock.fs_minfree = sblock.fs_minfree;
-       altsblock.fs_optim = sblock.fs_optim;
-       altsblock.fs_rotdelay = sblock.fs_rotdelay;
-       altsblock.fs_maxbpg = sblock.fs_maxbpg;
-       bcopy((char *)sblock.fs_csp, (char *)altsblock.fs_csp,
-               sizeof sblock.fs_csp);
-       bcopy((char *)sblock.fs_fsmnt, (char *)altsblock.fs_fsmnt,
-               sizeof sblock.fs_fsmnt);
-       if (bcmp((char *)&sblock, (char *)&altsblock, (int)sblock.fs_sbsize))
-               { badsb("TRASHED VALUES IN SUPER BLOCK"); return (0); }
-sbok:
-       fmax = sblock.fs_size;
-       imax = sblock.fs_ncg * sblock.fs_ipg;
+       if (sblock.fs_interleave < 1) {
+               pfatal("IMPOSSIBLE INTERLEAVE=%d IN SUPERBLOCK",
+                       sblock.fs_interleave);
+               if (reply("SET TO DEFAULT") == 1) {
+                       sblock.fs_interleave = 1;
+                       sbdirty();
+               }
+       }
+       if (sblock.fs_npsect < sblock.fs_nsect) {
+               pfatal("IMPOSSIBLE NPSECT=%d IN SUPERBLOCK",
+                       sblock.fs_npsect);
+               if (reply("SET TO DEFAULT") == 1) {
+                       sblock.fs_npsect = sblock.fs_nsect;
+                       sbdirty();
+               }
+       }
        /*
         * read in the summary info.
         */
        /*
         * read in the summary info.
         */
@@ -176,16 +169,207 @@ sbok:
 badsb:
        ckfini();
        return (0);
 badsb:
        ckfini();
        return (0);
+}
+
+/*
+ * Read in the super block and its summary info.
+ */
+readsb(listerr)
+       int listerr;
+{
+       BUFAREA asblk;
+#      define altsblock asblk.b_un.b_fs
+       daddr_t super = bflag ? bflag : SBLOCK;
+
+       initbarea(&asblk);
+       if (bread(&dfile, (char *)&sblock, super, (long)SBSIZE) != 0)
+               return (0);
+       sblk.b_bno = super;
+       sblk.b_size = SBSIZE;
+       /*
+        * run a few consistency checks of the super block
+        */
+       if (sblock.fs_magic != FS_MAGIC)
+               { badsb(listerr, "MAGIC NUMBER WRONG"); return (0); }
+       if (sblock.fs_ncg < 1)
+               { badsb(listerr, "NCG OUT OF RANGE"); return (0); }
+       if (sblock.fs_cpg < 1 || sblock.fs_cpg > MAXCPG)
+               { badsb(listerr, "CPG OUT OF RANGE"); return (0); }
+       if (sblock.fs_ncg * sblock.fs_cpg < sblock.fs_ncyl ||
+           (sblock.fs_ncg - 1) * sblock.fs_cpg >= sblock.fs_ncyl)
+               { badsb(listerr, "NCYL LESS THAN NCG*CPG"); return (0); }
+       if (sblock.fs_sbsize > SBSIZE)
+               { badsb(listerr, "SIZE PREPOSTEROUSLY LARGE"); return (0); }
+       /*
+        * Set all possible fields that could differ, then do check
+        * of whole super block against an alternate super block.
+        * When an alternate super-block is specified this check is skipped.
+        */
+       if (bflag)
+               return (1);
+       getblk(&asblk, cgsblock(&sblock, sblock.fs_ncg - 1), sblock.fs_sbsize);
+       if (asblk.b_errs != NULL)
+               return (0);
+       altsblock.fs_link = sblock.fs_link;
+       altsblock.fs_rlink = sblock.fs_rlink;
+       altsblock.fs_time = sblock.fs_time;
+       altsblock.fs_cstotal = sblock.fs_cstotal;
+       altsblock.fs_cgrotor = sblock.fs_cgrotor;
+       altsblock.fs_fmod = sblock.fs_fmod;
+       altsblock.fs_clean = sblock.fs_clean;
+       altsblock.fs_ronly = sblock.fs_ronly;
+       altsblock.fs_flags = sblock.fs_flags;
+       altsblock.fs_maxcontig = sblock.fs_maxcontig;
+       altsblock.fs_minfree = sblock.fs_minfree;
+       altsblock.fs_optim = sblock.fs_optim;
+       altsblock.fs_rotdelay = sblock.fs_rotdelay;
+       altsblock.fs_maxbpg = sblock.fs_maxbpg;
+       altsblock.fs_npsect = sblock.fs_npsect;
+       altsblock.fs_interleave = sblock.fs_interleave;
+       bcopy((char *)sblock.fs_csp, (char *)altsblock.fs_csp,
+               sizeof sblock.fs_csp);
+       bcopy((char *)sblock.fs_fsmnt, (char *)altsblock.fs_fsmnt,
+               sizeof sblock.fs_fsmnt);
+       if (bcmp((char *)&sblock, (char *)&altsblock, (int)sblock.fs_sbsize))
+               { badsb(listerr, "TRASHED VALUES IN SUPER BLOCK"); return (0); }
+       return (1);
 #      undef altsblock
 }
 
 #      undef altsblock
 }
 
-badsb(s)
+badsb(listerr, s)
+       int listerr;
        char *s;
 {
 
        char *s;
 {
 
+       if (!listerr)
+               return;
        if (preen)
                printf("%s: ", devname);
        if (preen)
                printf("%s: ", devname);
-       printf("BAD SUPER BLOCK: %s\n", s);
-       pwarn("USE -b OPTION TO FSCK TO SPECIFY LOCATION OF AN ALTERNATE\n");
-       pfatal("SUPER-BLOCK TO SUPPLY NEEDED INFORMATION; SEE fsck(8).\n");
+       pfatal("BAD SUPER BLOCK: %s\n", s);
+}
+
+/*
+ * Calculate a prototype superblock based on information in the disk label.
+ * When done the cgsblock macro can be calculated and the fs_ncg field
+ * can be used. Do NOT attempt to use other macros without verifying that
+ * their needed information is available!
+ */
+calcsb(dev, devfd, fs)
+       char *dev;
+       int devfd;
+       register struct fs *fs;
+{
+       register struct disklabel *lp;
+       register struct partition *pp;
+       register char *cp;
+       struct disklabel *getdisklabel();
+       int i;
+
+       cp = index(dev, '\0') - 1;
+       if (cp == (char *)-1 || (*cp < 'a' || *cp > 'h') && !isdigit(*cp)) {
+               pfatal("%s: CANNOT FIGURE OUT FILE SYSTEM PARTITION\n", dev);
+               return (0);
+       }
+       lp = getdisklabel(dev, devfd);
+       if (isdigit(*cp))
+               pp = &lp->d_partitions[0];
+       else
+               pp = &lp->d_partitions[*cp - 'a'];
+       if (pp->p_fstype != FS_BSDFFS) {
+               pfatal("%s: NOT FORMATTED AS A BSD FILE SYSTEM (%s)\n",
+                       dev, pp->p_fstype < FSMAXTYPES ?
+                       fstypenames[pp->p_fstype] : "unknown");
+               return (0);
+       }
+       bzero(fs, sizeof(struct fs));
+       fs->fs_fsize = pp->p_fsize;
+       fs->fs_frag = pp->p_frag;
+       fs->fs_cpg = pp->p_cpg;
+       fs->fs_size = pp->p_size;
+       fs->fs_ntrak = lp->d_ntracks;
+       fs->fs_nsect = lp->d_nsectors;
+       fs->fs_spc = lp->d_secpercyl;
+       fs->fs_nspf = fs->fs_fsize / lp->d_secsize;
+       fs->fs_sblkno = roundup(
+               howmany(lp->d_bbsize + lp->d_sbsize, fs->fs_fsize),
+               fs->fs_frag);
+       fs->fs_cgmask = 0xffffffff;
+       for (i = fs->fs_ntrak; i > 1; i >>= 1)
+               fs->fs_cgmask <<= 1;
+       if (!POWEROF2(fs->fs_ntrak))
+               fs->fs_cgmask <<= 1;
+       fs->fs_cgoffset = roundup(
+               howmany(fs->fs_nsect, NSPF(fs)), fs->fs_frag);
+       fs->fs_fpg = (fs->fs_cpg * fs->fs_spc) / NSPF(fs);
+       fs->fs_ncg = howmany(fs->fs_size / fs->fs_spc, fs->fs_cpg);
+       for (fs->fs_fsbtodb = 0, i = NSPF(fs); i > 1; i >>= 1)
+               fs->fs_fsbtodb++;
+       return (1);
+}
+
+#ifdef byioctl
+struct disklabel *
+getdisklabel(s, fd)
+       char *s;
+       int     fd;
+{
+       static struct disklabel lab;
+
+       if (ioctl(fd, DIOCGDINFO, (char *)&lab) < 0) {
+               perror("ioctl (GDINFO)");
+               fatal("%s: can't read disk label", s);
+       }
+       return (&lab);
+}
+#else byioctl
+char specname[64];
+char boot[BBSIZE];
+
+struct disklabel *
+getdisklabel(s, fd)
+       char *s;
+       int     fd;
+{
+       char *cp;
+       u_long magic = htonl(DISKMAGIC);
+       register struct disklabel *lp;
+       int cfd;
+
+       /*
+        * Make name for 'c' partition.
+        */
+       strcpy(specname, s);
+       cp = specname + strlen(specname) - 1;
+       if (!isdigit(*cp))
+               *cp = 'c';
+       cfd = open(specname, O_RDONLY);
+       if (cfd < 0) {
+               perror(specname);
+               exit(2);
+       }
+
+       if (read(cfd, boot, BBSIZE) < BBSIZE) {
+               perror(specname);
+               exit(2);
+       }
+       close(cfd);
+       for (lp = (struct disklabel *)(boot + LABELOFFSET);
+           lp <= (struct disklabel *)(boot + BBSIZE -
+           sizeof(struct disklabel));
+           lp = (struct disklabel *)((char *)lp + 128))
+               if (lp->d_magic == magic && lp->d_magic2 == magic)
+                       break;
+       if (lp > (struct disklabel *)(boot + BBSIZE -
+           sizeof(struct disklabel)) ||
+           lp->d_magic != magic || lp->d_magic2 != magic ||
+           dkcksum(lp) != 0) {
+               printf("Bad pack magic number %s\n",
+                       "(label is damaged, or pack is unlabeled)");
+               exit(1);
+       }
+#if ENDIAN != BIG
+       swablabel(lp);
+#endif
+       return (lp);
 }
 }
+#endif byioctl