sbin/fsirand/fsirand.c

   1 /*      $OpenBSD: fsirand.c,v 1.9 1997/02/28 00:46:33 millert Exp $     */
   2
   3 /*
   4  * Copyright (c) 1997 Todd C. Miller <Todd.Miller@courtesan.com>
   5  * All rights reserved.
   6  *
   7  * Redistribution and use in source and binary forms, with or without
   8  * modification, are permitted provided that the following conditions
   9  * are met:
  10  * 1. Redistributions of source code must retain the above copyright
  11  *    notice, this list of conditions and the following disclaimer.
  12  * 2. Redistributions in binary form must reproduce the above copyright
  13  *    notice, this list of conditions and the following disclaimer in the
  14  *    documentation and/or other materials provided with the distribution.
  15  * 3. All advertising materials mentioning features or use of this software
  16  *    must display the following acknowledgement:
  17  *      This product includes software developed by Todd C. Miller.
  18  * 4. The name of the author may not be used to endorse or promote products
  19  *    derived from this software without specific prior written permission.
  20  *
  21  * THIS SOFTWARE IS PROVIDED ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES,
  22  * INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY
  23  * AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL
  24  * THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
  25  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
  26  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;
  27  * OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
  28  * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
  29  * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
  30  * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  31  */
  32
  33 #ifndef lint
  34 static const char rcsid[] =
  35   "$FreeBSD$";
  36 #endif /* not lint */
  37
  38 #include <sys/disklabel.h>
  39 #include <sys/param.h>
  40 #include <sys/time.h>
  41 #include <sys/resource.h>
  42
  43 #include <ufs/ffs/fs.h>
  44 #include <ufs/ufs/dinode.h>
  45
  46 #include <err.h>
  47 #include <errno.h>
  48 #include <fcntl.h>
  49 #include <stdio.h>
  50 #include <stdlib.h>
  51 #include <string.h>
  52 #include <unistd.h>
  53
  54 static void usage(void) __dead2;
  55 int fsirand(char *);
  56
  57 int printonly = 0, force = 0, ignorelabel = 0;
  58
  59 int
  60 main(int argc, char *argv[])
  61 {
  62         int n, ex = 0;
  63         struct rlimit rl;
  64
  65         while ((n = getopt(argc, argv, "bfp")) != -1) {
  66                 switch (n) {
  67                 case 'b':
  68                         ignorelabel++;
  69                         break;
  70                 case 'p':
  71                         printonly++;
  72                         break;
  73                 case 'f':
  74                         force++;
  75                         break;
  76                 default:
  77                         usage();
  78                 }
  79         }
  80         if (argc - optind < 1)
  81                 usage();
  82
  83         srandomdev();
  84
  85         /* Increase our data size to the max */
  86         if (getrlimit(RLIMIT_DATA, &rl) == 0) {
  87                 rl.rlim_cur = rl.rlim_max;
  88                 if (setrlimit(RLIMIT_DATA, &rl) < 0)
  89                         warn("can't get resource limit to max data size");
  90         } else
  91                 warn("can't get resource limit for data size");
  92
  93         for (n = optind; n < argc; n++) {
  94                 if (argc - optind != 1)
  95                         (void)puts(argv[n]);
  96                 ex += fsirand(argv[n]);
  97                 if (n < argc - 1)
  98                         putchar('\n');
  99         }
 100
 101         exit(ex);
 102 }
 103
 104 int
 105 fsirand(char *device)
 106 {
 107         static struct dinode *inodebuf;
 108         static size_t oldibufsize;
 109         size_t ibufsize;
 110         struct fs *sblock;
 111         ino_t inumber, maxino;
 112         daddr_t dblk;
 113         char sbuf[SBSIZE], sbuftmp[SBSIZE];
 114         int devfd, n, cg;
 115         u_int32_t bsize = DEV_BSIZE;
 116         struct disklabel label;
 117
 118         if ((devfd = open(device, printonly ? O_RDONLY : O_RDWR)) < 0) {
 119                 warn("can't open %s", device);
 120                 return (1);
 121         }
 122
 123         /* Get block size (usually 512) from disklabel if possible */
 124         if (!ignorelabel) {
 125                 if (ioctl(devfd, DIOCGDINFO, &label) < 0)
 126                         warn("can't read disklabel, using sector size of %d",
 127                             bsize);
 128                 else
 129                         bsize = label.d_secsize;
 130         }
 131
 132         /* Read in master superblock */
 133         (void)memset(&sbuf, 0, sizeof(sbuf));
 134         sblock = (struct fs *)&sbuf;
 135         if (lseek(devfd, SBOFF, SEEK_SET) == -1) {
 136                 warn("can't seek to superblock (%qd) on %s", SBOFF, device);
 137                 return (1);
 138         }
 139         if ((n = read(devfd, (void *)sblock, SBSIZE)) != SBSIZE) {
 140                 warnx("can't read superblock on %s: %s", device,
 141                     (n < SBSIZE) ? "short read" : strerror(errno));
 142                 return (1);
 143         }
 144         maxino = sblock->fs_ncg * sblock->fs_ipg;
 145
 146         /* Simple sanity checks on the superblock */
 147         if (sblock->fs_magic != FS_MAGIC) {
 148                 warnx("bad magic number in superblock");
 149                 return (1);
 150         }
 151         if (sblock->fs_sbsize > SBSIZE) {
 152                 warnx("superblock size is preposterous");
 153                 return (1);
 154         }
 155         if (sblock->fs_postblformat == FS_42POSTBLFMT) {
 156                 warnx("filesystem format is too old, sorry");
 157                 return (1);
 158         }
 159         if (!force && !printonly && sblock->fs_clean != 1) {
 160                 warnx("filesystem is not clean, fsck %s first", device);
 161                 return (1);
 162         }
 163
 164         /* Make sure backup superblocks are sane. */
 165         sblock = (struct fs *)&sbuftmp;
 166         for (cg = 0; cg < sblock->fs_ncg; cg++) {
 167                 dblk = fsbtodb(sblock, cgsblock(sblock, cg));
 168                 if (lseek(devfd, (off_t)dblk * bsize, SEEK_SET) < 0) {
 169                         warn("can't seek to %qd", (off_t)dblk * bsize);
 170                         return (1);
 171                 } else if ((n = write(devfd, (void *)sblock, SBSIZE)) != SBSIZE) {
 172                         warn("can't read backup superblock %d on %s: %s",
 173                             cg + 1, device, (n < SBSIZE) ? "short write"
 174                             : strerror(errno));
 175                         return (1);
 176                 }
 177                 if (sblock->fs_magic != FS_MAGIC) {
 178                         warnx("bad magic number in backup superblock %d on %s",
 179                             cg + 1, device);
 180                         return (1);
 181                 }
 182                 if (sblock->fs_sbsize > SBSIZE) {
 183                         warnx("size of backup superblock %d on %s is preposterous",
 184                             cg + 1, device);
 185                         return (1);
 186                 }
 187         }
 188         sblock = (struct fs *)&sbuf;
 189
 190         /* XXX - should really cap buffer at 512kb or so */
 191         ibufsize = sizeof(struct dinode) * sblock->fs_ipg;
 192         if (oldibufsize < ibufsize) {
 193                 if ((inodebuf = realloc(inodebuf, ibufsize)) == NULL)
 194                         errx(1, "can't allocate memory for inode buffer");
 195                 oldibufsize = ibufsize;
 196         }
 197
 198         if (printonly && (sblock->fs_id[0] || sblock->fs_id[1])) {
 199                 if (sblock->fs_inodefmt >= FS_44INODEFMT && sblock->fs_id[0])
 200                         (void)printf("%s was randomized on %s", device,
 201                             ctime((const time_t *)&(sblock->fs_id[0])));
 202                 (void)printf("fsid: %x %x\n", sblock->fs_id[0],
 203                             sblock->fs_id[1]);
 204         }
 205
 206         /* Randomize fs_id unless old 4.2BSD filesystem */
 207         if ((sblock->fs_inodefmt >= FS_44INODEFMT) && !printonly) {
 208                 /* Randomize fs_id and write out new sblock and backups */
 209                 sblock->fs_id[0] = (u_int32_t)time(NULL);
 210                 sblock->fs_id[1] = random();
 211
 212                 if (lseek(devfd, SBOFF, SEEK_SET) == -1) {
 213                         warn("can't seek to superblock (%qd) on %s", SBOFF,
 214                             device);
 215                         return (1);
 216                 }
 217                 if ((n = write(devfd, (void *)sblock, SBSIZE)) != SBSIZE) {
 218                         warn("can't read superblock on %s: %s", device,
 219                             (n < SBSIZE) ? "short write" : strerror(errno));
 220                         return (1);
 221                 }
 222         }
 223
 224         /* For each cylinder group, randomize inodes and update backup sblock */
 225         for (cg = 0, inumber = 0; cg < sblock->fs_ncg; cg++) {
 226                 /* Update superblock if appropriate */
 227                 if ((sblock->fs_inodefmt >= FS_44INODEFMT) && !printonly) {
 228                         dblk = fsbtodb(sblock, cgsblock(sblock, cg));
 229                         if (lseek(devfd, (off_t)dblk * bsize, SEEK_SET) < 0) {
 230                                 warn("can't seek to %qd", (off_t)dblk * bsize);
 231                                 return (1);
 232                         } else if ((n = write(devfd, (void *)sblock, SBSIZE)) != SBSIZE) {
 233                                 warn("can't read backup superblock %d on %s: %s",
 234                                     cg + 1, device, (n < SBSIZE) ? "short write"
 235                                     : strerror(errno));
 236                                 return (1);
 237                         }
 238                 }
 239
 240                 /* Read in inodes, then print or randomize generation nums */
 241                 dblk = fsbtodb(sblock, ino_to_fsba(sblock, inumber));
 242                 if (lseek(devfd, (off_t)dblk * bsize, SEEK_SET) < 0) {
 243                         warn("can't seek to %qd", (off_t)dblk * bsize);
 244                         return (1);
 245                 } else if ((n = read(devfd, inodebuf, ibufsize)) != ibufsize) {
 246                         warnx("can't read inodes: %s",
 247                              (n < ibufsize) ? "short read" : strerror(errno));
 248                         return (1);
 249                 }
 250
 251                 for (n = 0; n < sblock->fs_ipg; n++, inumber++) {
 252                         if (inumber >= ROOTINO) {
 253                                 if (printonly)
 254                                         (void)printf("ino %d gen %x\n", inumber,
 255                                                      inodebuf[n].di_gen);
 256                                 else
 257                                         inodebuf[n].di_gen = random();
 258                         }
 259                 }
 260
 261                 /* Write out modified inodes */
 262                 if (!printonly) {
 263                         if (lseek(devfd, (off_t)dblk * bsize, SEEK_SET) < 0) {
 264                                 warn("can't seek to %qd",
 265                                     (off_t)dblk * bsize);
 266                                 return (1);
 267                         } else if ((n = write(devfd, inodebuf, ibufsize)) !=
 268                                  ibufsize) {
 269                                 warnx("can't write inodes: %s",
 270                                      (n != ibufsize) ? "short write" :
 271                                      strerror(errno));
 272                                 return (1);
 273                         }
 274                 }
 275         }
 276         (void)close(devfd);
 277
 278         return(0);
 279 }
 280
 281 static void
 282 usage(void)
 283 {
 284         (void)fprintf(stderr,
 285                 "usage: fsirand [-b] [-f] [-p] special [special ...]\n");
 286         exit(1);
 287 }