lib/libufs/block.c

   1 /*
   2  * Copyright (c) 2002 Juli Mallett.  All rights reserved.
   3  *
   4  * This software was written by Juli Mallett <jmallett@FreeBSD.org> for the
   5  * FreeBSD project.  Redistribution and use in source and binary forms, with
   6  * or without modification, are permitted provided that the following
   7  * conditions are met:
   8  *
   9  * 1. Redistribution of source code must retain the above copyright notice,
  10  *    this list of conditions and the following disclaimer.
  11  * 2. Redistribution in binary form must reproduce the above copyright
  12  *    notice, this list of conditions and the following disclaimer in the
  13  *    documentation and/or other materials provided with the distribution.
  14  *
  15  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
  16  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
  17  * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
  18  * DISCLAIMED. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT,
  19  * INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
  20  * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
  21  * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  22  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
  23  * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING
  24  * IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
  25  * POSSIBILITY OF SUCH DAMAGE.
  26  */
  27
  28 #include <sys/cdefs.h>
  29 __FBSDID("$FreeBSD$");
  30
  31 #include <sys/param.h>
  32 #include <sys/mount.h>
  33 #include <sys/disk.h>
  34 #include <sys/disklabel.h>
  35 #include <sys/stat.h>
  36
  37 #include <ufs/ufs/ufsmount.h>
  38 #include <ufs/ufs/dinode.h>
  39 #include <ufs/ffs/fs.h>
  40
  41 #include <errno.h>
  42 #include <fcntl.h>
  43 #include <stdio.h>
  44 #include <stdlib.h>
  45 #include <string.h>
  46 #include <unistd.h>
  47
  48 #include <libufs.h>
  49
  50 ssize_t
  51 bread(struct uufsd *disk, ufs2_daddr_t blockno, void *data, size_t size)
  52 {
  53         void *p2;
  54         ssize_t cnt;
  55
  56         ERROR(disk, NULL);
  57
  58         p2 = data;
  59         /*
  60          * XXX: various disk controllers require alignment of our buffer
  61          * XXX: which is stricter than struct alignment.
  62          * XXX: Bounce the buffer if not 64 byte aligned.
  63          * XXX: this can be removed if/when the kernel is fixed
  64          */
  65         if (((intptr_t)data) & 0x3f) {
  66                 p2 = malloc(size);
  67                 if (p2 == NULL) {
  68                         ERROR(disk, "allocate bounce buffer");
  69                         goto fail;
  70                 }
  71         }
  72         cnt = pread(disk->d_fd, p2, size, (off_t)(blockno * disk->d_bsize));
  73         if (cnt == -1) {
  74                 ERROR(disk, "read error from block device");
  75                 goto fail;
  76         }
  77         if (cnt == 0) {
  78                 ERROR(disk, "end of file from block device");
  79                 goto fail;
  80         }
  81         if ((size_t)cnt != size) {
  82                 ERROR(disk, "short read or read error from block device");
  83                 goto fail;
  84         }
  85         if (p2 != data) {
  86                 memcpy(data, p2, size);
  87                 free(p2);
  88         }
  89         return (cnt);
  90 fail:   memset(data, 0, size);
  91         if (p2 != data) {
  92                 free(p2);
  93         }
  94         return (-1);
  95 }
  96
  97 ssize_t
  98 bwrite(struct uufsd *disk, ufs2_daddr_t blockno, const void *data, size_t size)
  99 {
 100         ssize_t cnt;
 101         int rv;
 102         void *p2 = NULL;
 103
 104         ERROR(disk, NULL);
 105
 106         rv = ufs_disk_write(disk);
 107         if (rv == -1) {
 108                 ERROR(disk, "failed to open disk for writing");
 109                 return (-1);
 110         }
 111
 112         /*
 113          * XXX: various disk controllers require alignment of our buffer
 114          * XXX: which is stricter than struct alignment.
 115          * XXX: Bounce the buffer if not 64 byte aligned.
 116          * XXX: this can be removed if/when the kernel is fixed
 117          */
 118         if (((intptr_t)data) & 0x3f) {
 119                 p2 = malloc(size);
 120                 if (p2 == NULL) {
 121                         ERROR(disk, "allocate bounce buffer");
 122                         return (-1);
 123                 }
 124                 memcpy(p2, data, size);
 125                 data = p2;
 126         }
 127         cnt = pwrite(disk->d_fd, data, size, (off_t)(blockno * disk->d_bsize));
 128         if (p2 != NULL)
 129                 free(p2);
 130         if (cnt == -1) {
 131                 ERROR(disk, "write error to block device");
 132                 return (-1);
 133         }
 134         if ((size_t)cnt != size) {
 135                 ERROR(disk, "short write to block device");
 136                 return (-1);
 137         }
 138
 139         return (cnt);
 140 }
 141
 142 int
 143 berase(struct uufsd *disk, ufs2_daddr_t blockno, ufs2_daddr_t size)
 144 {
 145         off_t ioarg[2];
 146         int rv;
 147
 148         ERROR(disk, NULL);
 149         rv = ufs_disk_write(disk);
 150         if (rv == -1) {
 151                 ERROR(disk, "failed to open disk for writing");
 152                 return(rv);
 153         }
 154         ioarg[0] = blockno * disk->d_bsize;
 155         ioarg[1] = size;
 156         rv = ioctl(disk->d_fd, DIOCGDELETE, ioarg);
 157         return (rv);
 158 }