xref: /freebsd/usr.sbin/makefs/ffs/buf.c (revision b3e7694832e81d7a904a10f525f8797b753bf0d3)
1 /*	$NetBSD: buf.c,v 1.13 2004/06/20 22:20:18 jmc Exp $	*/
2 
3 /*-
4  * SPDX-License-Identifier: BSD-4-Clause
5  *
6  * Copyright (c) 2001 Wasabi Systems, Inc.
7  * All rights reserved.
8  *
9  * Written by Luke Mewburn for Wasabi Systems, Inc.
10  *
11  * Redistribution and use in source and binary forms, with or without
12  * modification, are permitted provided that the following conditions
13  * are met:
14  * 1. Redistributions of source code must retain the above copyright
15  *    notice, this list of conditions and the following disclaimer.
16  * 2. Redistributions in binary form must reproduce the above copyright
17  *    notice, this list of conditions and the following disclaimer in the
18  *    documentation and/or other materials provided with the distribution.
19  * 3. All advertising materials mentioning features or use of this software
20  *    must display the following acknowledgement:
21  *      This product includes software developed for the NetBSD Project by
22  *      Wasabi Systems, Inc.
23  * 4. The name of Wasabi Systems, Inc. may not be used to endorse
24  *    or promote products derived from this software without specific prior
25  *    written permission.
26  *
27  * THIS SOFTWARE IS PROVIDED BY WASABI SYSTEMS, INC. ``AS IS'' AND
28  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
29  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
30  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL WASABI SYSTEMS, INC
31  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
32  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
33  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
34  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
35  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
36  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
37  * POSSIBILITY OF SUCH DAMAGE.
38  */
39 
40 #include <sys/cdefs.h>
41 __FBSDID("$FreeBSD$");
42 
43 #include <sys/param.h>
44 #include <sys/time.h>
45 
46 #include <assert.h>
47 #include <errno.h>
48 #include <stdio.h>
49 #include <stdlib.h>
50 #include <string.h>
51 #include <unistd.h>
52 #include <util.h>
53 
54 #include "makefs.h"
55 #include "buf.h"
56 
57 static TAILQ_HEAD(buftailhead, m_buf) buftail;
58 
59 int
60 bread(struct m_vnode *vp, daddr_t blkno, int size, struct ucred *u1 __unused,
61     struct m_buf **bpp)
62 {
63 	off_t	offset;
64 	ssize_t	rv;
65 	fsinfo_t *fs = vp->fs;
66 
67 	assert (bpp != NULL);
68 
69 	if (debug & DEBUG_BUF_BREAD)
70 		printf("%s: blkno %lld size %d\n", __func__, (long long)blkno,
71 		    size);
72 	*bpp = getblk(vp, blkno, size, 0, 0, 0);
73 	offset = (off_t)(*bpp)->b_blkno * fs->sectorsize + fs->offset;
74 	if (debug & DEBUG_BUF_BREAD)
75 		printf("%s: blkno %lld offset %lld bcount %ld\n", __func__,
76 		    (long long)(*bpp)->b_blkno, (long long) offset,
77 		    (*bpp)->b_bcount);
78 	if (lseek((*bpp)->b_fs->fd, offset, SEEK_SET) == -1)
79 		err(1, "%s: lseek %lld (%lld)", __func__,
80 		    (long long)(*bpp)->b_blkno, (long long)offset);
81 	rv = read((*bpp)->b_fs->fd, (*bpp)->b_data, (size_t)(*bpp)->b_bcount);
82 	if (debug & DEBUG_BUF_BREAD)
83 		printf("%s: read %ld (%lld) returned %d\n", __func__,
84 		    (*bpp)->b_bcount, (long long)offset, (int)rv);
85 	if (rv == -1)				/* read error */
86 		err(1, "%s: read %ld (%lld) returned %d", __func__,
87 		    (*bpp)->b_bcount, (long long)offset, (int)rv);
88 	else if (rv != (*bpp)->b_bcount)	/* short read */
89 		err(1, "%s: read %ld (%lld) returned %d", __func__,
90 		    (*bpp)->b_bcount, (long long)offset, (int)rv);
91 	else
92 		return (0);
93 }
94 
95 void
96 brelse(struct m_buf *bp)
97 {
98 
99 	assert (bp != NULL);
100 	assert (bp->b_data != NULL);
101 
102 	if (bp->b_lblkno < 0) {
103 		/*
104 		 * XXX	don't remove any buffers with negative logical block
105 		 *	numbers (lblkno), so that we retain the mapping
106 		 *	of negative lblkno -> real blkno that ffs_balloc()
107 		 *	sets up.
108 		 *
109 		 *	if we instead released these buffers, and implemented
110 		 *	ufs_strategy() (and ufs_bmaparray()) and called those
111 		 *	from bread() and bwrite() to convert the lblkno to
112 		 *	a real blkno, we'd add a lot more code & complexity
113 		 *	and reading off disk, for little gain, because this
114 		 *	simple hack works for our purpose.
115 		 */
116 		bp->b_bcount = 0;
117 		return;
118 	}
119 
120 	TAILQ_REMOVE(&buftail, bp, b_tailq);
121 	free(bp->b_data);
122 	free(bp);
123 }
124 
125 int
126 bwrite(struct m_buf *bp)
127 {
128 	off_t	offset;
129 	ssize_t	rv;
130 	size_t	bytes;
131 	int	e;
132 	fsinfo_t *fs = bp->b_fs;
133 
134 	assert (bp != NULL);
135 	offset = (off_t)bp->b_blkno * fs->sectorsize + fs->offset;
136 	bytes = (size_t)bp->b_bcount;
137 	if (debug & DEBUG_BUF_BWRITE)
138 		printf("%s: blkno %lld offset %lld bcount %zu\n", __func__,
139 		    (long long)bp->b_blkno, (long long) offset, bytes);
140 	if (lseek(bp->b_fs->fd, offset, SEEK_SET) == -1) {
141 		brelse(bp);
142 		return (errno);
143 	}
144 	rv = write(bp->b_fs->fd, bp->b_data, bytes);
145 	e = errno;
146 	if (debug & DEBUG_BUF_BWRITE)
147 		printf("%s: write %ld (offset %lld) returned %lld\n", __func__,
148 		    bp->b_bcount, (long long)offset, (long long)rv);
149 	brelse(bp);
150 	if (rv == (ssize_t)bytes)
151 		return (0);
152 	if (rv == -1)		/* write error */
153 		return (e);
154 	return (EAGAIN);
155 }
156 
157 void
158 bcleanup(void)
159 {
160 	struct m_buf *bp;
161 
162 	/*
163 	 * XXX	this really shouldn't be necessary, but i'm curious to
164 	 *	know why there's still some buffers lying around that
165 	 *	aren't brelse()d
166 	 */
167 
168 	if (TAILQ_EMPTY(&buftail))
169 		return;
170 
171 	printf("%s: unflushed buffers:\n", __func__);
172 	TAILQ_FOREACH(bp, &buftail, b_tailq) {
173 		printf("\tlblkno %10lld  blkno %10lld  count %6ld  bufsize %6ld\n",
174 		    (long long)bp->b_lblkno, (long long)bp->b_blkno,
175 		    bp->b_bcount, bp->b_bufsize);
176 	}
177 	printf("%s: done\n", __func__);
178 }
179 
180 struct m_buf *
181 getblk(struct m_vnode *vp, daddr_t blkno, int size, int u1 __unused,
182     int u2 __unused, int u3 __unused)
183 {
184 	static int buftailinitted;
185 	struct m_buf *bp;
186 	void *n;
187 
188 	if (debug & DEBUG_BUF_GETBLK)
189 		printf("%s: blkno %lld size %d\n", __func__, (long long)blkno,
190 		    size);
191 
192 	bp = NULL;
193 	if (!buftailinitted) {
194 		if (debug & DEBUG_BUF_GETBLK)
195 			printf("%s: initialising tailq\n", __func__);
196 		TAILQ_INIT(&buftail);
197 		buftailinitted = 1;
198 	} else {
199 		TAILQ_FOREACH(bp, &buftail, b_tailq) {
200 			if (bp->b_lblkno != blkno)
201 				continue;
202 			break;
203 		}
204 	}
205 	if (bp == NULL) {
206 		bp = ecalloc(1, sizeof(*bp));
207 		bp->b_bufsize = 0;
208 		bp->b_blkno = bp->b_lblkno = blkno;
209 		bp->b_fs = vp->fs;
210 		bp->b_data = NULL;
211 		TAILQ_INSERT_HEAD(&buftail, bp, b_tailq);
212 	}
213 	bp->b_bcount = size;
214 	if (bp->b_data == NULL || bp->b_bcount > bp->b_bufsize) {
215 		n = erealloc(bp->b_data, (size_t)size);
216 		memset(n, 0, (size_t)size);
217 		bp->b_data = n;
218 		bp->b_bufsize = size;
219 	}
220 
221 	return (bp);
222 }
223