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