Logo Search packages:      
Sourcecode: hfsprogs version File versions  Download package

utilities.c

/*
 * Copyright (c) 1999 Apple Computer, Inc. All rights reserved.
 *
 * @APPLE_LICENSE_HEADER_START@
 * 
 * "Portions Copyright (c) 1999 Apple Computer, Inc.  All Rights
 * Reserved.  This file contains Original Code and/or Modifications of
 * Original Code as defined in and that are subject to the Apple Public
 * Source License Version 1.0 (the 'License').  You may not use this file
 * except in compliance with the License.  Please obtain a copy of the
 * License at http://www.apple.com/publicsource and read it before using
 * this file.
 * 
 * The Original Code and all software distributed under the License are
 * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
 * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
 * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE OR NON-INFRINGEMENT.  Please see the
 * License for the specific language governing rights and limitations
 * under the License."
 * 
 * @APPLE_LICENSE_HEADER_END@
 */
/*
 * Copyright (c) 1980, 1986, 1993
 *    The Regents of the University of California.  All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 * 3. All advertising materials mentioning features or use of this software
 *    must display the following acknowledgement:
 *    This product includes software developed by the University of
 *    California, Berkeley and its contributors.
 * 4. Neither the name of the University nor the names of its contributors
 *    may be used to endorse or promote products derived from this software
 *    without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
 * SUCH DAMAGE.
 */


#include <sys/param.h>
#include <sys/time.h>

#include <ufs/ufs/dinode.h>
#include <ufs/ufs/dir.h>
#include <ufs/ffs/fs.h>

#include <ctype.h>
#include <err.h>
#include <string.h>

#include "fsck.h"
#if   REV_ENDIAN_FS
#include "ufs_byte_order.h"
#endif      /* REV_ENDIAN_FS */

long  diskreads, totalreads;  /* Disk cache statistics */

static void rwerror __P((char *mesg, ufs_daddr_t blk));

int
ftypeok(dp)
      struct dinode *dp;
{
      switch (dp->di_mode & IFMT) {

      case IFDIR:
      case IFREG:
      case IFBLK:
      case IFCHR:
      case IFLNK:
      case IFSOCK:
      case IFIFO:
            return (1);

      default:
            if (debug)
                  printf("bad file type 0%o\n", dp->di_mode);
            return (0);
      }
}

int
reply(question)
      char *question;
{
      int persevere;
      char c;

      if (preen)
            pfatal("INTERNAL ERROR: GOT TO reply()");
      persevere = !strcmp(question, "CONTINUE");
      printf("\n");
      if (!persevere && (nflag || fswritefd < 0)) {
            printf("%s? no\n\n", question);
            return (0);
      }
      if (yflag || (persevere && nflag)) {
            printf("%s? yes\n\n", question);
            return (1);
      }
      do    {
            printf("%s? [yn] ", question);
            (void) fflush(stdout);
            c = getc(stdin);
            while (c != '\n' && getc(stdin) != '\n')
                  if (feof(stdin))
                        return (0);
      } while (c != 'y' && c != 'Y' && c != 'n' && c != 'N');
      printf("\n");
      if (c == 'y' || c == 'Y')
            return (1);
      return (0);
}

#if   REV_ENDIAN_FS
void
swapblock(struct bufarea *bp, int writeout)
{
struct fs * fs;
      fs = &sblock; 

      if (!rev_endian) return;
      if (writeout) {
      /* Writing out contents of the block */
            switch (bp->b_type) {
            case SUPERBLOCK:
                  byte_swap_sbout(bp->b_un.b_fs);
                   break;
            case INODEBLOCK:
                byte_swap_dinodes(bp,writeout);
                break;
            case CYLGROUP:
                byte_swap_cgout(bp->b_un.b_cg,&sblock);
                break;
             case INDIRECTBLOCK:
                byte_swap_ints(bp->b_un.b_indir, NINDIR(&sblock));
                break;
            case DIRECTORYBLOCK:
                byte_swap_dir_block_out(bp);
                break;
             default:
                break;
            }
            bp->b_flags &= ~B_SWAPPED;
      } else {
            /*  If Buffer is already swapped; return */
            if (bp->b_flags & B_SWAPPED) 
                  return;
            switch (bp->b_type) {
      /* Reading in contents of the block */
             case SUPERBLOCK:
                byte_swap_sbin(bp->b_un.b_fs);
                break;
            case INODEBLOCK:
                byte_swap_dinodes(bp,writeout);
                break;
             case CYLGROUP:
                byte_swap_cgin(bp->b_un.b_cg,&sblock);
                break;
            case INDIRECTBLOCK:
                byte_swap_ints(bp->b_un.b_indir, NINDIR(&sblock));
                break;
            case DIRECTORYBLOCK:
                byte_swap_dir_block_in(bp);
                break;
             default:
                break;
            }
            bp->b_flags |= B_SWAPPED;
      }
}
#endif      /* REV_ENDIAN_FS */

/*
 * Malloc buffers and set up cache.
 */
void
bufinit()
{
      register struct bufarea *bp;
      long bufcnt, i;
      char *bufp;

      pbp = pdirbp = (struct bufarea *)0;
      bufp = malloc((unsigned int)sblock.fs_bsize);
      if (bufp == 0)
            errx(EEXIT, "cannot allocate buffer pool");
      cgblk.b_un.b_buf = bufp;
      initbarea(&cgblk);
#if   REV_ENDIAN_FS
      cgblk.b_type = CYLGROUP;
#endif      /* REV_ENDIAN_FS */
      bufhead.b_next = bufhead.b_prev = &bufhead;
      bufcnt = MAXBUFSPACE / sblock.fs_bsize;
      if (bufcnt < MINBUFS)
            bufcnt = MINBUFS;
      for (i = 0; i < bufcnt; i++) {
            bp = (struct bufarea *)malloc(sizeof(struct bufarea));
            bufp = malloc((unsigned int)sblock.fs_bsize);
            if (bp == NULL || bufp == NULL) {
                  if (i >= MINBUFS)
                        break;
                  errx(EEXIT, "cannot allocate buffer pool");
            }
            bp->b_un.b_buf = bufp;
            bp->b_prev = &bufhead;
            bp->b_next = bufhead.b_next;
            bufhead.b_next->b_prev = bp;
            bufhead.b_next = bp;
            initbarea(bp);
      }
      bufhead.b_size = i;     /* save number of buffers */
}

/*
 * Manage a cache of directory blocks.
 */
struct bufarea *
getdatablk(blkno, size)
      ufs_daddr_t blkno;
      long size;
{
      register struct bufarea *bp;

      for (bp = bufhead.b_next; bp != &bufhead; bp = bp->b_next)
            if (bp->b_bno == fsbtodb(&sblock, blkno))
                  goto foundit;
      for (bp = bufhead.b_prev; bp != &bufhead; bp = bp->b_prev)
            if ((bp->b_flags & B_INUSE) == 0)
                  break;
      if (bp == &bufhead)
            errx(EEXIT, "deadlocked buffer pool");
      getblk(bp, blkno, size);
      /* fall through */
foundit:
      totalreads++;
      bp->b_prev->b_next = bp->b_next;
      bp->b_next->b_prev = bp->b_prev;
      bp->b_prev = &bufhead;
      bp->b_next = bufhead.b_next;
      bufhead.b_next->b_prev = bp;
      bufhead.b_next = bp;
      bp->b_flags |= B_INUSE;
      return (bp);
}

void
getblk(bp, blk, size)
      register struct bufarea *bp;
      ufs_daddr_t blk;
      long size;
{
      ufs_daddr_t dblk;

      dblk = fsbtodb(&sblock, blk);
      if (bp->b_bno != dblk) {
            flush(fswritefd, bp);
            diskreads++;
            bp->b_errs = bread(fsreadfd, bp->b_un.b_buf, dblk, size);
            bp->b_bno = dblk;
            bp->b_size = size;
#if REV_ENDIAN_FS
            bp->b_flags &= ~B_SWAPPED;
#endif
      }
}

void
flush(fd, bp)
      int fd;
      register struct bufarea *bp;
{
      register int i, j;

      if (!bp->b_dirty)
            return;
      if (bp->b_errs != 0)
            pfatal("WRITING %sZERO'ED BLOCK %d TO DISK\n",
                (bp->b_errs == bp->b_size / dev_bsize) ? "" : "PARTIALLY ",
                bp->b_bno);
      bp->b_dirty = 0;
      bp->b_errs = 0;
#if   REV_ENDIAN_FS
      swapblock(bp, 1);
#endif      /* REV_ENDIAN_FS */
      bwrite(fd, bp->b_un.b_buf, bp->b_bno, (long)bp->b_size);
#if   REV_ENDIAN_FS
      swapblock(bp, 0);
#endif      /* REV_ENDIAN_FS */
      if (bp != &sblk)
            return;
      for (i = 0, j = 0; i < sblock.fs_cssize; i += sblock.fs_bsize, j++) {
#if   REV_ENDIAN_FS
            int size = sblock.fs_cssize - i < sblock.fs_bsize ?
                     sblock.fs_cssize - i : sblock.fs_bsize;
            if (rev_endian)
                  byte_swap_ints((int *) ((char *)sblock.fs_csp + i), 
                        size / sizeof(int));
            bwrite(fswritefd, (char *)sblock.fs_csp + i,
                fsbtodb(&sblock, sblock.fs_csaddr + j * sblock.fs_frag),
                size);
            if (rev_endian)
                  byte_swap_ints((int *) ((char *)sblock.fs_csp + i), 
                        size / sizeof(int));
#else /* REV_ENDIAN_FS */
            bwrite(fswritefd, (char *)sblock.fs_csp + i,
                fsbtodb(&sblock, sblock.fs_csaddr + j * sblock.fs_frag),
                sblock.fs_cssize - i < sblock.fs_bsize ?
                sblock.fs_cssize - i : sblock.fs_bsize);
#endif      /* REV_ENDIAN_FS */
      }
}

static void
rwerror(mesg, blk)
      char *mesg;
      ufs_daddr_t blk;
{

      if (preen == 0)
            printf("\n");
      pfatal("CANNOT %s: BLK %ld", mesg, blk);
      if (reply("CONTINUE") == 0)
            exit(EEXIT);
}

void
ckfini(markclean)
      int markclean;
{
      register struct bufarea *bp, *nbp;
      int ofsmodified, cnt = 0;

      if (fswritefd < 0) {
            (void)close(fsreadfd);
            return;
      }
      flush(fswritefd, &sblk);
      if (havesb && sblk.b_bno != SBOFF / dev_bsize &&
          !preen && reply("UPDATE STANDARD SUPERBLOCK")) {
            sblk.b_bno = SBOFF / dev_bsize;
            sbdirty();
            flush(fswritefd, &sblk);
      }
      flush(fswritefd, &cgblk);
      free(cgblk.b_un.b_buf);
      for (bp = bufhead.b_prev; bp && bp != &bufhead; bp = nbp) {
            cnt++;
            flush(fswritefd, bp);
            nbp = bp->b_prev;
            free(bp->b_un.b_buf);
            free((char *)bp);
      }
      if (bufhead.b_size != cnt)
            errx(EEXIT, "Panic: lost %d buffers", bufhead.b_size - cnt);
      pbp = pdirbp = (struct bufarea *)0;
      if (markclean && sblock.fs_clean == 0) {
            sblock.fs_clean = 1;
            sbdirty();
            ofsmodified = fsmodified;
            flush(fswritefd, &sblk);
            fsmodified = ofsmodified;
            if (!preen)
                  printf("\n***** FILE SYSTEM MARKED CLEAN *****\n");
      }
      if (debug)
            printf("cache missed %ld of %ld (%d%%)\n", diskreads,
                totalreads, (int)(diskreads * 100 / totalreads));
      (void)close(fsreadfd);
      (void)close(fswritefd);
}

int
bread(fd, buf, blk, size)
      int fd;
      char *buf;
      ufs_daddr_t blk;
      long size;
{
      char *cp;
      int i, errs;
      off_t offset;

      offset = blk;
      offset *= dev_bsize;
      if (lseek(fd, offset, 0) < 0)
            rwerror("SEEK", blk);
      else if (read(fd, buf, (int)size) == size)
            return (0);
      rwerror("READ", blk);
      if (lseek(fd, offset, 0) < 0)
            rwerror("SEEK", blk);
      errs = 0;
      memset(buf, 0, (size_t)size);
      printf("THE FOLLOWING DISK SECTORS COULD NOT BE READ:");
      for (cp = buf, i = 0; i < size; i += secsize, cp += secsize) {
            if (read(fd, cp, (int)secsize) != secsize) {
                  (void)lseek(fd, offset + i + secsize, 0);
                  if (secsize != dev_bsize && dev_bsize != 1)
                        printf(" %ld (%ld),",
                            (blk * dev_bsize + i) / secsize,
                            blk + i / dev_bsize);
                  else
                        printf(" %ld,", blk + i / dev_bsize);
                  errs++;
            }
      }
      printf("\n");
      return (errs);
}

void
bwrite(fd, buf, blk, size)
      int fd;
      char *buf;
      ufs_daddr_t blk;
      long size;
{
      int i;
      char *cp;
      off_t offset;

      if (fd < 0)
            return;
      offset = blk;
      offset *= dev_bsize;
      if (lseek(fd, offset, 0) < 0)
            rwerror("SEEK", blk);
      else if (write(fd, buf, (int)size) == size) {
            fsmodified = 1;
            return;
      }
      rwerror("WRITE", blk);
      if (lseek(fd, offset, 0) < 0)
            rwerror("SEEK", blk);
      printf("THE FOLLOWING SECTORS COULD NOT BE WRITTEN:");
      for (cp = buf, i = 0; i < size; i += dev_bsize, cp += dev_bsize)
            if (write(fd, cp, (int)dev_bsize) != dev_bsize) {
                  (void)lseek(fd, offset + i + dev_bsize, 0);
                  printf(" %ld,", blk + i / dev_bsize);
            }
      printf("\n");
      return;
}

/*
 * allocate a data block with the specified number of fragments
 */
ufs_daddr_t
allocblk(frags)
      long frags;
{
      register int i, j, k;

      if (frags <= 0 || frags > sblock.fs_frag)
            return (0);
      for (i = 0; i < maxfsblock - sblock.fs_frag; i += sblock.fs_frag) {
            for (j = 0; j <= sblock.fs_frag - frags; j++) {
                  if (testbmap(i + j))
                        continue;
                  for (k = 1; k < frags; k++)
                        if (testbmap(i + j + k))
                              break;
                  if (k < frags) {
                        j += k;
                        continue;
                  }
                  for (k = 0; k < frags; k++)
                        setbmap(i + j + k);
                  n_blks += frags;
                  return (i + j);
            }
      }
      return (0);
}

/*
 * Free a previously allocated block
 */
void
freeblk(blkno, frags)
      ufs_daddr_t blkno;
      long frags;
{
      struct inodesc idesc;

      idesc.id_blkno = blkno;
      idesc.id_numfrags = frags;
      (void)pass4check(&idesc);
}

/*
 * Find a pathname
 */
void
getpathname(namebuf, curdir, ino)
      char *namebuf;
      ino_t curdir, ino;
{
      int len;
      register char *cp;
      struct inodesc idesc;
      static int busy = 0;
      extern int findname();

      if (curdir == ino && ino == ROOTINO) {
            (void)strcpy(namebuf, "/");
            return;
      }
      if (busy ||
          (statemap[curdir] != DSTATE && statemap[curdir] != DFOUND)) {
            (void)strcpy(namebuf, "?");
            return;
      }
      busy = 1;
      memset(&idesc, 0, sizeof(struct inodesc));
      idesc.id_type = DATA;
      idesc.id_fix = IGNORE;
      cp = &namebuf[MAXPATHLEN - 1];
      *cp = '\0';
      if (curdir != ino) {
            idesc.id_parent = curdir;
            goto namelookup;
      }
      while (ino != ROOTINO) {
            idesc.id_number = ino;
            idesc.id_func = findino;
            idesc.id_name = "..";
            if ((ckinode(ginode(ino), &idesc) & FOUND) == 0)
                  break;
      namelookup:
            idesc.id_number = idesc.id_parent;
            idesc.id_parent = ino;
            idesc.id_func = findname;
            idesc.id_name = namebuf;
            if ((ckinode(ginode(idesc.id_number), &idesc)&FOUND) == 0)
                  break;
            len = strlen(namebuf);
            cp -= len;
            memmove(cp, namebuf, (size_t)len);
            *--cp = '/';
            if (cp < &namebuf[MAXNAMLEN])
                  break;
            ino = idesc.id_number;
      }
      busy = 0;
      if (ino != ROOTINO)
            *--cp = '?';
      memmove(namebuf, cp, (size_t)(&namebuf[MAXPATHLEN] - cp));
}

void
catch(sig)
      int sig;
{
      if (!doinglevel2)
            ckfini(0);
      exit(12);
}

/*
 * When preening, allow a single quit to signal
 * a special exit after filesystem checks complete
 * so that reboot sequence may be interrupted.
 */
void
catchquit(sig)
      int sig;
{
      extern int returntosingle;

      printf("returning to single-user after filesystem check\n");
      returntosingle = 1;
      (void)signal(SIGQUIT, SIG_DFL);
}

/*
 * Ignore a single quit signal; wait and flush just in case.
 * Used by child processes in preen.
 */
void
voidquit(sig)
      int sig;
{

      sleep(1);
      (void)signal(SIGQUIT, SIG_IGN);
      (void)signal(SIGQUIT, SIG_DFL);
}

/*
 * determine whether an inode should be fixed.
 */
int
dofix(idesc, msg)
      register struct inodesc *idesc;
      char *msg;
{

      switch (idesc->id_fix) {

      case DONTKNOW:
            if (idesc->id_type == DATA)
                  direrror(idesc->id_number, msg);
            else
                  pwarn(msg);
            if (preen) {
                  printf(" (SALVAGED)\n");
                  idesc->id_fix = FIX;
                  return (ALTERED);
            }
            if (reply("SALVAGE") == 0) {
                  idesc->id_fix = NOFIX;
                  return (0);
            }
            idesc->id_fix = FIX;
            return (ALTERED);

      case FIX:
            return (ALTERED);

      case NOFIX:
      case IGNORE:
            return (0);

      default:
            errx(EEXIT, "UNKNOWN INODESC FIX MODE %d", idesc->id_fix);
      }
      /* NOTREACHED */
      return (0);
}

#if __STDC__
#include <stdarg.h>
#else
#include <varargs.h>
#endif

/*
 * An unexpected inconsistency occured.
 * Die if preening, otherwise just print message and continue.
 */
void
#if __STDC__
pfatal(const char *fmt, ...)
#else
pfatal(fmt, va_alist)
      char *fmt;
      va_dcl
#endif
{
      va_list ap;
#if __STDC__
      va_start(ap, fmt);
#else
      va_start(ap);
#endif
      if (!preen) {
            (void)vfprintf(stderr, fmt, ap);
            va_end(ap);
            return;
      }
      (void)fprintf(stderr, "%s: ", cdevname);
      (void)vfprintf(stderr, fmt, ap);
      (void)fprintf(stderr,
          "\n%s: UNEXPECTED INCONSISTENCY; RUN fsck MANUALLY.\n",
          cdevname);
      exit(EEXIT);
}

/*
 * Pwarn just prints a message when not preening,
 * or a warning (preceded by filename) when preening.
 */
void
#if __STDC__
pwarn(const char *fmt, ...)
#else
pwarn(fmt, va_alist)
      char *fmt;
      va_dcl
#endif
{
      va_list ap;
#if __STDC__
      va_start(ap, fmt);
#else
      va_start(ap);
#endif
      if (preen)
            (void)fprintf(stderr, "%s: ", cdevname);
      (void)vfprintf(stderr, fmt, ap);
      va_end(ap);
}

/*
 * Stub for routines from kernel.
 */
#ifdef __APPLE__
__private_extern__
#endif
void
#if __STDC__
panic(const char *fmt, ...)
#else
panic(fmt, va_alist)
      char *fmt;
      va_dcl
#endif
{
      va_list ap;
#if __STDC__
      va_start(ap, fmt);
#else
      va_start(ap);
#endif
      pfatal("INTERNAL INCONSISTENCY:");
      (void)vfprintf(stderr, fmt, ap);
      va_end(ap);
      exit(EEXIT);
}

Generated by  Doxygen 1.6.0   Back to index