1 /*-
2 * SPDX-License-Identifier: BSD-2-Clause
3 *
4 * Copyright (c) 2003 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/param.h>
31 #include <sys/mount.h>
32 #include <sys/disklabel.h>
33 #include <sys/stat.h>
34
35 #include <ufs/ufs/extattr.h>
36 #include <ufs/ufs/quota.h>
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 ufs2_daddr_t
cgballoc(struct uufsd * disk)51 cgballoc(struct uufsd *disk)
52 {
53 u_int8_t *blksfree;
54 struct cg *cgp;
55 struct fs *fs;
56 long bno;
57
58 fs = &disk->d_fs;
59 cgp = &disk->d_cg;
60 blksfree = cg_blksfree(cgp);
61 for (bno = 0; bno < fs->fs_fpg / fs->fs_frag; bno++)
62 if (ffs_isblock(fs, blksfree, bno))
63 goto gotit;
64 return (0);
65 gotit:
66 fs->fs_cs(fs, cgp->cg_cgx).cs_nbfree--;
67 ffs_clrblock(fs, blksfree, (long)bno);
68 ffs_clusteracct(fs, cgp, bno, -1);
69 cgp->cg_cs.cs_nbfree--;
70 fs->fs_cstotal.cs_nbfree--;
71 fs->fs_fmod = 1;
72 return (cgbase(fs, cgp->cg_cgx) + blkstofrags(fs, bno));
73 }
74
75 int
cgbfree(struct uufsd * disk,ufs2_daddr_t bno,long size)76 cgbfree(struct uufsd *disk, ufs2_daddr_t bno, long size)
77 {
78 u_int8_t *blksfree;
79 struct fs *fs;
80 struct cg *cgp;
81 ufs1_daddr_t fragno, cgbno;
82 int i, cg, blk, frags, bbase;
83
84 fs = &disk->d_fs;
85 cg = dtog(fs, bno);
86 if (cgread1(disk, cg) != 1)
87 return (-1);
88 cgp = &disk->d_cg;
89 cgbno = dtogd(fs, bno);
90 blksfree = cg_blksfree(cgp);
91 if (size == fs->fs_bsize) {
92 fragno = fragstoblks(fs, cgbno);
93 ffs_setblock(fs, blksfree, fragno);
94 ffs_clusteracct(fs, cgp, fragno, 1);
95 cgp->cg_cs.cs_nbfree++;
96 fs->fs_cstotal.cs_nbfree++;
97 fs->fs_cs(fs, cg).cs_nbfree++;
98 } else {
99 bbase = cgbno - fragnum(fs, cgbno);
100 /*
101 * decrement the counts associated with the old frags
102 */
103 blk = blkmap(fs, blksfree, bbase);
104 ffs_fragacct(fs, blk, cgp->cg_frsum, -1);
105 /*
106 * deallocate the fragment
107 */
108 frags = numfrags(fs, size);
109 for (i = 0; i < frags; i++)
110 setbit(blksfree, cgbno + i);
111 cgp->cg_cs.cs_nffree += i;
112 fs->fs_cstotal.cs_nffree += i;
113 fs->fs_cs(fs, cg).cs_nffree += i;
114 /*
115 * add back in counts associated with the new frags
116 */
117 blk = blkmap(fs, blksfree, bbase);
118 ffs_fragacct(fs, blk, cgp->cg_frsum, 1);
119 /*
120 * if a complete block has been reassembled, account for it
121 */
122 fragno = fragstoblks(fs, bbase);
123 if (ffs_isblock(fs, blksfree, fragno)) {
124 cgp->cg_cs.cs_nffree -= fs->fs_frag;
125 fs->fs_cstotal.cs_nffree -= fs->fs_frag;
126 fs->fs_cs(fs, cg).cs_nffree -= fs->fs_frag;
127 ffs_clusteracct(fs, cgp, fragno, 1);
128 cgp->cg_cs.cs_nbfree++;
129 fs->fs_cstotal.cs_nbfree++;
130 fs->fs_cs(fs, cg).cs_nbfree++;
131 }
132 }
133 return cgwrite(disk);
134 }
135
136 ino_t
cgialloc(struct uufsd * disk)137 cgialloc(struct uufsd *disk)
138 {
139 struct ufs2_dinode *dp2;
140 u_int8_t *inosused;
141 struct cg *cgp;
142 struct fs *fs;
143 ino_t ino;
144 int i;
145
146 fs = &disk->d_fs;
147 cgp = &disk->d_cg;
148 inosused = cg_inosused(cgp);
149 for (ino = 0; ino < fs->fs_ipg; ino++)
150 if (isclr(inosused, ino))
151 goto gotit;
152 return (0);
153 gotit:
154 if (fs->fs_magic == FS_UFS2_MAGIC &&
155 ino + INOPB(fs) > cgp->cg_initediblk &&
156 cgp->cg_initediblk < cgp->cg_niblk) {
157 char block[MAXBSIZE];
158 bzero(block, (int)fs->fs_bsize);
159 dp2 = (struct ufs2_dinode *)█
160 for (i = 0; i < INOPB(fs); i++) {
161 dp2->di_gen = arc4random();
162 dp2++;
163 }
164 if (bwrite(disk, ino_to_fsba(fs,
165 cgp->cg_cgx * fs->fs_ipg + cgp->cg_initediblk),
166 block, fs->fs_bsize))
167 return (0);
168 cgp->cg_initediblk += INOPB(fs);
169 }
170
171 setbit(inosused, ino);
172 cgp->cg_irotor = ino;
173 cgp->cg_cs.cs_nifree--;
174 fs->fs_cstotal.cs_nifree--;
175 fs->fs_cs(fs, cgp->cg_cgx).cs_nifree--;
176 fs->fs_fmod = 1;
177
178 return (ino + (cgp->cg_cgx * fs->fs_ipg));
179 }
180
181 int
cgread(struct uufsd * disk)182 cgread(struct uufsd *disk)
183 {
184
185 if (disk->d_ccg >= disk->d_fs.fs_ncg)
186 return (0);
187 return (cgread1(disk, disk->d_ccg++));
188 }
189
190 /* Short read/write error messages from cgget()/cgput() */
191 static const char *failmsg;
192
193 int
cgread1(struct uufsd * disk,int c)194 cgread1(struct uufsd *disk, int c)
195 {
196
197 if (cgget(disk->d_fd, &disk->d_fs, c, &disk->d_cg) == 0) {
198 disk->d_lcg = c;
199 return (1);
200 }
201 ERROR(disk, NULL);
202 if (failmsg != NULL) {
203 ERROR(disk, failmsg);
204 return (-1);
205 }
206 switch (errno) {
207 case EINTEGRITY:
208 ERROR(disk, "cylinder group checks failed");
209 break;
210 case EIO:
211 ERROR(disk, "read error from block device");
212 break;
213 default:
214 ERROR(disk, strerror(errno));
215 break;
216 }
217 return (-1);
218 }
219
220 int
cgget(int devfd,struct fs * fs,int cg,struct cg * cgp)221 cgget(int devfd, struct fs *fs, int cg, struct cg *cgp)
222 {
223 uint32_t cghash, calchash;
224 size_t cnt;
225
226 failmsg = NULL;
227 if ((cnt = pread(devfd, cgp, fs->fs_cgsize,
228 fsbtodb(fs, cgtod(fs, cg)) * (fs->fs_fsize / fsbtodb(fs,1)))) < 0)
229 return (-1);
230 if (cnt == 0) {
231 failmsg = "end of file from block device";
232 errno = EIO;
233 return (-1);
234 }
235 if (cnt != fs->fs_cgsize) {
236 failmsg = "short read from block device";
237 errno = EIO;
238 return (-1);
239 }
240 calchash = cgp->cg_ckhash;
241 if ((fs->fs_metackhash & CK_CYLGRP) != 0) {
242 cghash = cgp->cg_ckhash;
243 cgp->cg_ckhash = 0;
244 calchash = calculate_crc32c(~0L, (void *)cgp, fs->fs_cgsize);
245 cgp->cg_ckhash = cghash;
246 }
247 if (cgp->cg_ckhash != calchash || !cg_chkmagic(cgp) ||
248 cgp->cg_cgx != cg) {
249 errno = EINTEGRITY;
250 return (-1);
251 }
252 return (0);
253 }
254
255 int
cgwrite(struct uufsd * disk)256 cgwrite(struct uufsd *disk)
257 {
258
259 return (cgwrite1(disk, disk->d_cg.cg_cgx));
260 }
261
262 int
cgwrite1(struct uufsd * disk,int cg)263 cgwrite1(struct uufsd *disk, int cg)
264 {
265 static char errmsg[BUFSIZ];
266
267 if (cg == disk->d_cg.cg_cgx) {
268 if (ufs_disk_write(disk) == -1) {
269 ERROR(disk, "failed to open disk for writing");
270 return (-1);
271 }
272 if (cgput(disk->d_fd, &disk->d_fs, &disk->d_cg) == 0)
273 return (0);
274 ERROR(disk, NULL);
275 if (failmsg != NULL) {
276 ERROR(disk, failmsg);
277 return (-1);
278 }
279 switch (errno) {
280 case EIO:
281 ERROR(disk, "unable to write cylinder group");
282 break;
283 default:
284 ERROR(disk, strerror(errno));
285 break;
286 }
287 return (-1);
288 }
289 snprintf(errmsg, BUFSIZ, "Cylinder group %d in buffer does not match "
290 "the cylinder group %d that cgwrite1 requested",
291 disk->d_cg.cg_cgx, cg);
292 ERROR(disk, errmsg);
293 errno = EDOOFUS;
294 return (-1);
295 }
296
297 int
cgput(int devfd,struct fs * fs,struct cg * cgp)298 cgput(int devfd, struct fs *fs, struct cg *cgp)
299 {
300 size_t cnt;
301
302 if ((fs->fs_metackhash & CK_CYLGRP) != 0) {
303 cgp->cg_ckhash = 0;
304 cgp->cg_ckhash =
305 calculate_crc32c(~0L, (void *)cgp, fs->fs_cgsize);
306 }
307 failmsg = NULL;
308 if ((cnt = pwrite(devfd, cgp, fs->fs_cgsize,
309 fsbtodb(fs, cgtod(fs, cgp->cg_cgx)) *
310 (fs->fs_fsize / fsbtodb(fs,1)))) < 0)
311 return (-1);
312 if (cnt != fs->fs_cgsize) {
313 failmsg = "short write to block device";
314 return (-1);
315 }
316 return (0);
317 }
318