xref: /netbsd-src/usr.sbin/makefs/ffs.c (revision 7022656d8ab2142ec648610e6c63c908b926ed5b)
1 /*	$NetBSD: ffs.c,v 1.76 2024/06/17 23:53:42 riastradh Exp $	*/
2 
3 /*
4  * Copyright (c) 2001 Wasabi Systems, Inc.
5  * All rights reserved.
6  *
7  * Written by Luke Mewburn for Wasabi Systems, Inc.
8  *
9  * Redistribution and use in source and binary forms, with or without
10  * modification, are permitted provided that the following conditions
11  * are met:
12  * 1. Redistributions of source code must retain the above copyright
13  *    notice, this list of conditions and the following disclaimer.
14  * 2. Redistributions in binary form must reproduce the above copyright
15  *    notice, this list of conditions and the following disclaimer in the
16  *    documentation and/or other materials provided with the distribution.
17  * 3. All advertising materials mentioning features or use of this software
18  *    must display the following acknowledgement:
19  *      This product includes software developed for the NetBSD Project by
20  *      Wasabi Systems, Inc.
21  * 4. The name of Wasabi Systems, Inc. may not be used to endorse
22  *    or promote products derived from this software without specific prior
23  *    written permission.
24  *
25  * THIS SOFTWARE IS PROVIDED BY WASABI SYSTEMS, INC. ``AS IS'' AND
26  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
27  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
28  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL WASABI SYSTEMS, INC
29  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
30  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
31  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
32  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
33  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
34  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
35  * POSSIBILITY OF SUCH DAMAGE.
36  */
37 /*
38  * Copyright (c) 1982, 1986, 1989, 1993
39  *	The Regents of the University of California.  All rights reserved.
40  *
41  * Redistribution and use in source and binary forms, with or without
42  * modification, are permitted provided that the following conditions
43  * are met:
44  * 1. Redistributions of source code must retain the above copyright
45  *    notice, this list of conditions and the following disclaimer.
46  * 2. Redistributions in binary form must reproduce the above copyright
47  *    notice, this list of conditions and the following disclaimer in the
48  *    documentation and/or other materials provided with the distribution.
49  * 3. Neither the name of the University nor the names of its contributors
50  *    may be used to endorse or promote products derived from this software
51  *    without specific prior written permission.
52  *
53  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
54  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
55  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
56  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
57  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
58  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
59  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
60  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
61  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
62  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
63  * SUCH DAMAGE.
64  *
65  *	@(#)ffs_alloc.c	8.19 (Berkeley) 7/13/95
66  */
67 
68 #if HAVE_NBTOOL_CONFIG_H
69 #include "nbtool_config.h"
70 #endif
71 
72 #include <sys/cdefs.h>
73 #if defined(__RCSID) && !defined(__lint)
74 __RCSID("$NetBSD: ffs.c,v 1.76 2024/06/17 23:53:42 riastradh Exp $");
75 #endif	/* !__lint */
76 
77 #include <sys/param.h>
78 
79 #if !HAVE_NBTOOL_CONFIG_H
80 #include <sys/mount.h>
81 #endif
82 
83 #include <assert.h>
84 #include <errno.h>
85 #include <fcntl.h>
86 #include <stdarg.h>
87 #include <stdio.h>
88 #include <stdlib.h>
89 #include <string.h>
90 #include <unistd.h>
91 #include <util.h>
92 
93 #include "makefs.h"
94 #include "ffs.h"
95 
96 #if HAVE_STRUCT_STATVFS_F_IOSIZE && HAVE_FSTATVFS
97 #include <sys/statvfs.h>
98 #endif
99 
100 #include <ufs/ufs/dinode.h>
101 #include <ufs/ufs/dir.h>
102 #include <ufs/ffs/fs.h>
103 #include <ufs/ufs/ufs_bswap.h>
104 
105 #include "ffs/ufs_inode.h"
106 #include "ffs/newfs_extern.h"
107 #include "ffs/ffs_extern.h"
108 
109 #undef DIP
110 #define DIP(dp, field) \
111 	((ffs_opts->version == 1) ? \
112 	(dp)->ffs1_din.di_##field : (dp)->ffs2_din.di_##field)
113 
114 /*
115  * Various file system defaults (cribbed from newfs(8)).
116  */
117 #define	DFL_FRAGSIZE		1024		/* fragment size */
118 #define	DFL_BLKSIZE		8192		/* block size */
119 #define	DFL_SECSIZE		512		/* sector size */
120 #define	DFL_CYLSPERGROUP	65536		/* cylinders per group */
121 #define	DFL_FRAGSPERINODE	4		/* fragments per inode */
122 #define	DFL_ROTDELAY		0		/* rotational delay */
123 #define	DFL_NRPOS		1		/* rotational positions */
124 #define	DFL_RPM			3600		/* rpm of disk */
125 #define	DFL_NSECTORS		64		/* # of sectors */
126 #define	DFL_NTRACKS		16		/* # of tracks */
127 
128 
129 typedef struct {
130 	u_char		*buf;		/* buf for directory */
131 	doff_t		size;		/* full size of buf */
132 	doff_t		cur;		/* offset of current entry */
133 } dirbuf_t;
134 
135 
136 static	int	ffs_create_image(const char *, fsinfo_t *);
137 static	void	ffs_dump_fsinfo(fsinfo_t *);
138 static	void	ffs_dump_dirbuf(dirbuf_t *, const char *, int);
139 static	void	ffs_make_dirbuf(dirbuf_t *, const char *, fsnode *, int);
140 static	int	ffs_populate_dir(const char *, fsnode *, fsinfo_t *);
141 static	void	ffs_size_dir(fsnode *, fsinfo_t *);
142 static	void	ffs_validate(const char *, fsnode *, fsinfo_t *);
143 static	void	ffs_write_file(union dinode *, uint32_t, void *, fsinfo_t *);
144 static	void	ffs_write_inode(union dinode *, uint32_t, const fsinfo_t *);
145 static  void	*ffs_build_dinode1(struct ufs1_dinode *, dirbuf_t *, fsnode *,
146 				 fsnode *, fsinfo_t *);
147 static  void	*ffs_build_dinode2(struct ufs2_dinode *, dirbuf_t *, fsnode *,
148 				 fsnode *, fsinfo_t *);
149 
150 
151 
152 	/* publicly visible functions */
153 void
ffs_prep_opts(fsinfo_t * fsopts)154 ffs_prep_opts(fsinfo_t *fsopts)
155 {
156 	ffs_opt_t *ffs_opts = ecalloc(1, sizeof(*ffs_opts));
157 
158 	const option_t ffs_options[] = {
159 	    { 'b', "bsize", &ffs_opts->bsize, OPT_INT32,
160 	      1, INT_MAX, "block size" },
161 	    { 'f', "fsize", &ffs_opts->fsize, OPT_INT32,
162 	      1, INT_MAX, "fragment size" },
163 	    { 'd', "density", &ffs_opts->density, OPT_INT32,
164 	      1, INT_MAX, "bytes per inode" },
165 	    { 'm', "minfree", &ffs_opts->minfree, OPT_INT32,
166 	      0, 99, "minfree" },
167 	    { 'M', "maxbpg", &ffs_opts->maxbpg, OPT_INT32,
168 	      1, INT_MAX, "max blocks per file in a cg" },
169 	    { 'a', "avgfilesize", &ffs_opts->avgfilesize, OPT_INT32,
170 	      1, INT_MAX, "expected average file size" },
171 	    { 'n', "avgfpdir", &ffs_opts->avgfpdir, OPT_INT32,
172 	      1, INT_MAX, "expected # of files per directory" },
173 	    { 'x', "extent", &ffs_opts->maxbsize, OPT_INT32,
174 	      1, INT_MAX, "maximum # extent size" },
175 	    { 'g', "maxbpcg", &ffs_opts->maxblkspercg, OPT_INT32,
176 	      1, INT_MAX, "max # of blocks per group" },
177 	    { 'v', "version", &ffs_opts->version, OPT_INT32,
178 	      1, 2, "UFS version" },
179 	    { 'o', "optimization", NULL, OPT_STRBUF,
180 	      0, 0, "Optimization (time|space)" },
181 	    { 'l', "label", ffs_opts->label, OPT_STRARRAY,
182 	      1, sizeof(ffs_opts->label), "UFS label" },
183 	    { 'e', "extattr", &ffs_opts->extattr, OPT_INT32,
184 	      0, 1, "extattr support" },
185 	    { .name = NULL }
186 	};
187 
188 	ffs_opts->bsize= -1;
189 	ffs_opts->fsize= -1;
190 	ffs_opts->cpg= -1;
191 	ffs_opts->density= -1;
192 	ffs_opts->minfree= -1;
193 	ffs_opts->optimization= -1;
194 	ffs_opts->maxcontig= -1;
195 	ffs_opts->maxbpg= -1;
196 	ffs_opts->avgfilesize= -1;
197 	ffs_opts->avgfpdir= -1;
198 	ffs_opts->version = 1;
199 	ffs_opts->extattr = 1;
200 
201 	fsopts->fs_specific = ffs_opts;
202 	fsopts->fs_options = copy_opts(ffs_options);
203 }
204 
205 void
ffs_cleanup_opts(fsinfo_t * fsopts)206 ffs_cleanup_opts(fsinfo_t *fsopts)
207 {
208 	free(fsopts->fs_specific);
209 	free(fsopts->fs_options);
210 }
211 
212 int
ffs_parse_opts(const char * option,fsinfo_t * fsopts)213 ffs_parse_opts(const char *option, fsinfo_t *fsopts)
214 {
215 	ffs_opt_t	*ffs_opts = fsopts->fs_specific;
216 	option_t *ffs_options = fsopts->fs_options;
217 	char buf[1024];
218 
219 	int	rv;
220 
221 	assert(option != NULL);
222 	assert(fsopts != NULL);
223 	assert(ffs_opts != NULL);
224 
225 	if (debug & DEBUG_FS_PARSE_OPTS)
226 		printf("ffs_parse_opts: got `%s'\n", option);
227 
228 	rv = set_option(ffs_options, option, buf, sizeof(buf));
229 	if (rv == -1)
230 		return 0;
231 
232 	if (ffs_options[rv].name == NULL)
233 		abort();
234 
235 	switch (ffs_options[rv].letter) {
236 	case 'o':
237 		if (strcmp(buf, "time") == 0) {
238 			ffs_opts->optimization = FS_OPTTIME;
239 		} else if (strcmp(buf, "space") == 0) {
240 			ffs_opts->optimization = FS_OPTSPACE;
241 		} else {
242 			warnx("Invalid optimization `%s'", buf);
243 			return 0;
244 		}
245 		break;
246 	default:
247 		break;
248 	}
249 	return 1;
250 }
251 
252 
253 void
ffs_makefs(const char * image,const char * dir,fsnode * root,fsinfo_t * fsopts)254 ffs_makefs(const char *image, const char *dir, fsnode *root, fsinfo_t *fsopts)
255 {
256 	struct fs	*superblock;
257 	struct timeval	start;
258 
259 	assert(image != NULL);
260 	assert(dir != NULL);
261 	assert(root != NULL);
262 	assert(fsopts != NULL);
263 
264 	if (debug & DEBUG_FS_MAKEFS)
265 		printf("ffs_makefs: image %s directory %s root %p\n",
266 		    image, dir, root);
267 
268 		/* validate tree and options */
269 	TIMER_START(start);
270 	ffs_validate(dir, root, fsopts);
271 	TIMER_RESULTS(start, "ffs_validate");
272 
273 	printf("Calculated size of `%s': %lld bytes, %lld inodes\n",
274 	    image, (long long)fsopts->size, (long long)fsopts->inodes);
275 
276 		/* create image */
277 	TIMER_START(start);
278 	if (ffs_create_image(image, fsopts) == -1)
279 		errx(EXIT_FAILURE, "Image file `%s' not created.", image);
280 	TIMER_RESULTS(start, "ffs_create_image");
281 
282 	fsopts->curinode = UFS_ROOTINO;
283 
284 	if (debug & DEBUG_FS_MAKEFS)
285 		putchar('\n');
286 
287 		/* populate image */
288 	printf("Populating `%s'\n", image);
289 	TIMER_START(start);
290 	if (! ffs_populate_dir(dir, root, fsopts))
291 		errx(EXIT_FAILURE, "Image file `%s' not populated.", image);
292 	TIMER_RESULTS(start, "ffs_populate_dir");
293 
294 		/* ensure no outstanding buffers remain */
295 	if (debug & DEBUG_FS_MAKEFS)
296 		bcleanup();
297 
298 		/* update various superblock parameters */
299 	superblock = fsopts->superblock;
300 	superblock->fs_fmod = 0;
301 	superblock->fs_old_cstotal.cs_ndir   = superblock->fs_cstotal.cs_ndir;
302 	superblock->fs_old_cstotal.cs_nbfree = superblock->fs_cstotal.cs_nbfree;
303 	superblock->fs_old_cstotal.cs_nifree = superblock->fs_cstotal.cs_nifree;
304 	superblock->fs_old_cstotal.cs_nffree = superblock->fs_cstotal.cs_nffree;
305 
306 		/* write out superblock; image is now complete */
307 	ffs_write_superblock(fsopts->superblock, fsopts);
308 	if (close(fsopts->fd) == -1)
309 		err(EXIT_FAILURE, "Closing `%s'", image);
310 	fsopts->fd = -1;
311 	printf("Image `%s' complete\n", image);
312 }
313 
314 	/* end of public functions */
315 
316 
317 static void
ffs_validate(const char * dir,fsnode * root,fsinfo_t * fsopts)318 ffs_validate(const char *dir, fsnode *root, fsinfo_t *fsopts)
319 {
320 	int32_t	ncg = 1;
321 #if notyet
322 	int32_t	spc, nspf, ncyl, fssize;
323 #endif
324 	ffs_opt_t	*ffs_opts = fsopts->fs_specific;
325 
326 	assert(dir != NULL);
327 	assert(root != NULL);
328 	assert(fsopts != NULL);
329 	assert(ffs_opts != NULL);
330 
331 	if (debug & DEBUG_FS_VALIDATE) {
332 		printf("ffs_validate: before defaults set:\n");
333 		ffs_dump_fsinfo(fsopts);
334 	}
335 
336 		/* set FFS defaults */
337 	if (fsopts->sectorsize == -1)
338 		fsopts->sectorsize = DFL_SECSIZE;
339 	if (ffs_opts->fsize == -1)
340 		ffs_opts->fsize = MAX(DFL_FRAGSIZE, fsopts->sectorsize);
341 	if (ffs_opts->bsize == -1)
342 		ffs_opts->bsize = MIN(DFL_BLKSIZE, 8 * ffs_opts->fsize);
343 	if (ffs_opts->cpg == -1)
344 		ffs_opts->cpg = DFL_CYLSPERGROUP;
345 	else
346 		ffs_opts->cpgflg = 1;
347 				/* fsopts->density is set below */
348 	if (ffs_opts->nsectors == -1)
349 		ffs_opts->nsectors = DFL_NSECTORS;
350 	if (ffs_opts->minfree == -1)
351 		ffs_opts->minfree = MINFREE;
352 	if (ffs_opts->optimization == -1)
353 		ffs_opts->optimization = DEFAULTOPT;
354 	if (ffs_opts->maxcontig == -1)
355 		ffs_opts->maxcontig =
356 		    MAX(1, MIN(MAXPHYS, FFS_MAXBSIZE) / ffs_opts->bsize);
357 	/* XXX ondisk32 */
358 	if (ffs_opts->maxbpg == -1)
359 		ffs_opts->maxbpg = ffs_opts->bsize / sizeof(int32_t);
360 	if (ffs_opts->avgfilesize == -1)
361 		ffs_opts->avgfilesize = AVFILESIZ;
362 	if (ffs_opts->avgfpdir == -1)
363 		ffs_opts->avgfpdir = AFPDIR;
364 
365 		/* calculate size of tree */
366 	ffs_size_dir(root, fsopts);
367 	fsopts->inodes += UFS_ROOTINO;		/* include first two inodes */
368 
369 	if (debug & DEBUG_FS_VALIDATE)
370 		printf("ffs_validate: size of tree: %lld bytes, %lld inodes\n",
371 		    (long long)fsopts->size, (long long)fsopts->inodes);
372 
373 		/* add requested slop */
374 	fsopts->size += fsopts->freeblocks;
375 	fsopts->inodes += fsopts->freefiles;
376 	if (fsopts->freefilepc > 0)
377 		fsopts->inodes =
378 		    fsopts->inodes * (100 + fsopts->freefilepc) / 100;
379 	if (fsopts->freeblockpc > 0)
380 		fsopts->size =
381 		    fsopts->size * (100 + fsopts->freeblockpc) / 100;
382 
383 		/* add space needed for superblocks */
384 	/*
385 	 * The old SBOFF (SBLOCK_UFS1) is used here because makefs is
386 	 * typically used for small filesystems where space matters.
387 	 * XXX make this an option.
388 	 */
389 	fsopts->size += (SBLOCK_UFS1 + SBLOCKSIZE) * ncg;
390 		/* add space needed to store inodes, x3 for blockmaps, etc */
391 	if (ffs_opts->version == 1)
392 		fsopts->size += ncg * DINODE1_SIZE *
393 		    roundup(fsopts->inodes / ncg,
394 			ffs_opts->bsize / DINODE1_SIZE);
395 	else
396 		fsopts->size += ncg * DINODE2_SIZE *
397 		    roundup(fsopts->inodes / ncg,
398 			ffs_opts->bsize / DINODE2_SIZE);
399 
400 		/* add minfree */
401 	if (ffs_opts->minfree > 0)
402 		fsopts->size =
403 		    fsopts->size * (100 + ffs_opts->minfree) / 100;
404 	/*
405 	 * XXX	any other fs slop to add, such as csum's, bitmaps, etc ??
406 	 */
407 
408 	if (fsopts->size < fsopts->minsize)	/* ensure meets minimum size */
409 		fsopts->size = fsopts->minsize;
410 
411 		/* round up to the next block */
412 	fsopts->size = roundup(fsopts->size, ffs_opts->bsize);
413 
414 		/* calculate density if necessary */
415 	if (ffs_opts->density == -1)
416 		ffs_opts->density = fsopts->size / fsopts->inodes + 1;
417 
418 	if (debug & DEBUG_FS_VALIDATE) {
419 		printf("ffs_validate: after defaults set:\n");
420 		ffs_dump_fsinfo(fsopts);
421 		printf("ffs_validate: dir %s; %lld bytes, %lld inodes\n",
422 		    dir, (long long)fsopts->size, (long long)fsopts->inodes);
423 	}
424 		/* now check calculated sizes vs requested sizes */
425 	if (fsopts->maxsize > 0 && fsopts->size > fsopts->maxsize) {
426 		errx(EXIT_FAILURE,
427 		    "`%s' size of %lld is larger than the maxsize of %lld.",
428 		    dir, (long long)fsopts->size, (long long)fsopts->maxsize);
429 	}
430 }
431 
432 
433 static void
ffs_dump_fsinfo(fsinfo_t * f)434 ffs_dump_fsinfo(fsinfo_t *f)
435 {
436 
437 	ffs_opt_t	*fs = f->fs_specific;
438 
439 	printf("fsopts at %p\n", f);
440 
441 	printf("\tsize %lld, inodes %lld, curinode %u\n",
442 	    (long long)f->size, (long long)f->inodes, f->curinode);
443 
444 	printf("\tminsize %lld, maxsize %lld\n",
445 	    (long long)f->minsize, (long long)f->maxsize);
446 	printf("\tfree files %lld, freefile %% %d\n",
447 	    (long long)f->freefiles, f->freefilepc);
448 	printf("\tfree blocks %lld, freeblock %% %d\n",
449 	    (long long)f->freeblocks, f->freeblockpc);
450 	printf("\tneedswap %d, sectorsize %d\n", f->needswap, f->sectorsize);
451 
452 	printf("\tbsize %d, fsize %d, cpg %d, density %d\n",
453 	    fs->bsize, fs->fsize, fs->cpg, fs->density);
454 	printf("\tnsectors %d, rpm %d, minfree %d\n",
455 	    fs->nsectors, fs->rpm, fs->minfree);
456 	printf("\tmaxcontig %d, maxbpg %d\n",
457 	    fs->maxcontig, fs->maxbpg);
458 	printf("\toptimization %s\n",
459 	    fs->optimization == FS_OPTSPACE ? "space" : "time");
460 }
461 
462 
463 static int
ffs_create_image(const char * image,fsinfo_t * fsopts)464 ffs_create_image(const char *image, fsinfo_t *fsopts)
465 {
466 #if HAVE_STRUCT_STATVFS_F_IOSIZE && HAVE_FSTATVFS
467 	struct statvfs	sfs;
468 #endif
469 	struct fs	*fs;
470 	char	*buf;
471 	int	i, bufsize;
472 	off_t	bufrem;
473 	time_t	tstamp;
474 	int	oflags = O_RDWR | O_CREAT;
475 
476 	assert (image != NULL);
477 	assert (fsopts != NULL);
478 
479 		/* create image */
480 	if (fsopts->offset == 0)
481 		oflags |= O_TRUNC;
482 	if ((fsopts->fd = open(image, oflags, 0666)) == -1) {
483 		warn("Can't open `%s' for writing", image);
484 		return (-1);
485 	}
486 
487 		/* zero image */
488 #if HAVE_STRUCT_STATVFS_F_IOSIZE && HAVE_FSTATVFS
489 	if (fstatvfs(fsopts->fd, &sfs) == -1) {
490 #endif
491 		bufsize = 8192;
492 #if HAVE_STRUCT_STATVFS_F_IOSIZE && HAVE_FSTATVFS
493 		warn("can't fstatvfs `%s', using default %d byte chunk",
494 		    image, bufsize);
495 	} else
496 		bufsize = sfs.f_iosize;
497 #endif
498 	bufrem = fsopts->size;
499 
500 	if (fsopts->sparse) {
501 		if (ftruncate(fsopts->fd, bufrem) == -1) {
502 			printf ("ERROR in truncate. Sparse option disabled\n");
503 			fsopts->sparse = 0;
504 		} else {
505 			bufrem = 0; /* File truncated at bufrem. Remaining is 0 */
506 			buf = NULL;
507 		}
508 	}
509 
510 	if (fsopts->offset != 0)
511 		if (lseek(fsopts->fd, fsopts->offset, SEEK_SET) == -1) {
512 			warn("can't seek");
513 			return -1;
514 		}
515 
516 	if ((debug & DEBUG_FS_CREATE_IMAGE) && fsopts->sparse == 0)
517 		printf(
518 		    "zero-ing image `%s', %lld sectors, using %d byte chunks\n",
519 		    image, (long long)bufrem, bufsize);
520 	if (bufrem > 0)
521 		buf = ecalloc(1, bufsize);
522 	while (bufrem > 0) {
523 		i = write(fsopts->fd, buf, MIN(bufsize, bufrem));
524 		if (i == -1) {
525 			warn("zeroing image, %lld bytes to go",
526 			    (long long)bufrem);
527 			free(buf);
528 			return (-1);
529 		}
530 		bufrem -= i;
531 	}
532 	if (buf)
533 		free(buf);
534 
535 		/* make the file system */
536 	if (debug & DEBUG_FS_CREATE_IMAGE)
537 		printf("calling mkfs(\"%s\", ...)\n", image);
538 
539 	if (stampst.st_ino)
540 		tstamp = stampst.st_ctime;
541 	else
542 		tstamp = start_time.tv_sec;
543 
544 	srandom(tstamp);
545 
546 	fs = ffs_mkfs(image, fsopts, tstamp);
547 	fsopts->superblock = (void *)fs;
548 	if (debug & DEBUG_FS_CREATE_IMAGE) {
549 		time_t t;
550 
551 		t = (time_t)((struct fs *)fsopts->superblock)->fs_time;
552 		printf("mkfs returned %p; fs_time %s",
553 		    fsopts->superblock, ctime(&t));
554 		printf("fs totals: nbfree %lld, nffree %lld, nifree %lld, ndir %lld\n",
555 		    (long long)fs->fs_cstotal.cs_nbfree,
556 		    (long long)fs->fs_cstotal.cs_nffree,
557 		    (long long)fs->fs_cstotal.cs_nifree,
558 		    (long long)fs->fs_cstotal.cs_ndir);
559 	}
560 
561 	if ((off_t)(fs->fs_cstotal.cs_nifree + UFS_ROOTINO) < fsopts->inodes) {
562 		warnx(
563 		"Image file `%s' has %lld free inodes; %lld are required.",
564 		    image,
565 		    (long long)(fs->fs_cstotal.cs_nifree + UFS_ROOTINO),
566 		    (long long)fsopts->inodes);
567 		return (-1);
568 	}
569 	return (fsopts->fd);
570 }
571 
572 
573 static void
ffs_size_dir(fsnode * root,fsinfo_t * fsopts)574 ffs_size_dir(fsnode *root, fsinfo_t *fsopts)
575 {
576 	struct direct	tmpdir;
577 	fsnode *	node;
578 	int		curdirsize, this;
579 	ffs_opt_t	*ffs_opts = fsopts->fs_specific;
580 
581 	/* node may be NULL (empty directory) */
582 	assert(fsopts != NULL);
583 	assert(ffs_opts != NULL);
584 
585 	if (debug & DEBUG_FS_SIZE_DIR)
586 		printf("ffs_size_dir: entry: bytes %lld inodes %lld\n",
587 		    (long long)fsopts->size, (long long)fsopts->inodes);
588 
589 #define	ADDDIRENT(e) do {						\
590 	tmpdir.d_namlen = strlen((e));					\
591 	this = UFS_DIRSIZ(0, &tmpdir, 0);				\
592 	if (debug & DEBUG_FS_SIZE_DIR_ADD_DIRENT)			\
593 		printf("ADDDIRENT: was: %s (%d) this %d cur %d\n",	\
594 		    e, tmpdir.d_namlen, this, curdirsize);		\
595 	if (this + curdirsize > roundup(curdirsize, UFS_DIRBLKSIZ))	\
596 		curdirsize = roundup(curdirsize, UFS_DIRBLKSIZ);	\
597 	curdirsize += this;						\
598 	if (debug & DEBUG_FS_SIZE_DIR_ADD_DIRENT)			\
599 		printf("ADDDIRENT: now: %s (%d) this %d cur %d\n",	\
600 		    e, tmpdir.d_namlen, this, curdirsize);		\
601 } while (0);
602 
603 	/*
604 	 * XXX	this needs to take into account extra space consumed
605 	 *	by indirect blocks, etc.
606 	 */
607 #define	ADDSIZE(x) do {							\
608 	fsopts->size += roundup((x), ffs_opts->fsize);			\
609 } while (0);
610 
611 	curdirsize = 0;
612 	for (node = root; node != NULL; node = node->next) {
613 		ADDDIRENT(node->name);
614 		if (node == root) {			/* we're at "." */
615 			assert(strcmp(node->name, ".") == 0);
616 			ADDDIRENT("..");
617 		} else if ((node->inode->flags & FI_SIZED) == 0) {
618 				/* don't count duplicate names */
619 			node->inode->flags |= FI_SIZED;
620 			if (debug & DEBUG_FS_SIZE_DIR_NODE)
621 				printf("ffs_size_dir: `%s' size %lld\n",
622 				    node->name,
623 				    (long long)node->inode->st.st_size);
624 			fsopts->inodes++;
625 			if (node->type == S_IFREG)
626 				ADDSIZE(node->inode->st.st_size);
627 			if (node->type == S_IFLNK) {
628 				size_t	slen;
629 
630 				slen = strlen(node->symlink) + 1;
631 				if (slen >= (ffs_opts->version == 1 ?
632 						UFS1_MAXSYMLINKLEN :
633 						UFS2_MAXSYMLINKLEN))
634 					ADDSIZE(slen);
635 			}
636 		}
637 		if (node->type == S_IFDIR)
638 			ffs_size_dir(node->child, fsopts);
639 	}
640 	ADDSIZE(curdirsize);
641 
642 	if (debug & DEBUG_FS_SIZE_DIR)
643 		printf("ffs_size_dir: exit: size %lld inodes %lld\n",
644 		    (long long)fsopts->size, (long long)fsopts->inodes);
645 }
646 
647 static void *
ffs_build_dinode1(struct ufs1_dinode * dinp,dirbuf_t * dbufp,fsnode * cur,fsnode * root,fsinfo_t * fsopts)648 ffs_build_dinode1(struct ufs1_dinode *dinp, dirbuf_t *dbufp, fsnode *cur,
649 		 fsnode *root, fsinfo_t *fsopts)
650 {
651 	size_t slen;
652 	void *membuf;
653 	struct stat *st = stampst.st_ino ? &stampst : &cur->inode->st;
654 
655 	memset(dinp, 0, sizeof(*dinp));
656 	dinp->di_mode = cur->inode->st.st_mode;
657 	dinp->di_nlink = cur->inode->nlink;
658 	dinp->di_size = cur->inode->st.st_size;
659 #if HAVE_STRUCT_STAT_ST_FLAGS
660 	dinp->di_flags = cur->inode->st.st_flags;
661 #endif
662 #if HAVE_STRUCT_STAT_ST_GEN
663 	dinp->di_gen = cur->inode->st.st_gen;
664 #endif
665 	dinp->di_uid = cur->inode->st.st_uid;
666 	dinp->di_gid = cur->inode->st.st_gid;
667 
668 	dinp->di_atime = st->st_atime;
669 	dinp->di_mtime = st->st_mtime;
670 	dinp->di_ctime = st->st_ctime;
671 #if HAVE_STRUCT_STAT_ST_MTIMENSEC
672 	dinp->di_atimensec = st->st_atimensec;
673 	dinp->di_mtimensec = st->st_mtimensec;
674 	dinp->di_ctimensec = st->st_ctimensec;
675 #endif
676 		/* not set: di_db, di_ib, di_blocks, di_spare */
677 
678 	membuf = NULL;
679 	if (cur == root) {			/* "."; write dirbuf */
680 		membuf = dbufp->buf;
681 		dinp->di_size = dbufp->size;
682 	} else if (S_ISBLK(cur->type) || S_ISCHR(cur->type)) {
683 		dinp->di_size = 0;	/* a device */
684 		dinp->di_rdev =
685 		    ufs_rw32(cur->inode->st.st_rdev, fsopts->needswap);
686 	} else if (S_ISLNK(cur->type)) {	/* symlink */
687 		slen = strlen(cur->symlink);
688 		if (slen < UFS1_MAXSYMLINKLEN) {	/* short link */
689 			memcpy(dinp->di_db, cur->symlink, slen);
690 		} else
691 			membuf = cur->symlink;
692 		dinp->di_size = slen;
693 	}
694 	return membuf;
695 }
696 
697 static void *
ffs_build_dinode2(struct ufs2_dinode * dinp,dirbuf_t * dbufp,fsnode * cur,fsnode * root,fsinfo_t * fsopts)698 ffs_build_dinode2(struct ufs2_dinode *dinp, dirbuf_t *dbufp, fsnode *cur,
699 		 fsnode *root, fsinfo_t *fsopts)
700 {
701 	size_t slen;
702 	void *membuf;
703 	struct stat *st = stampst.st_ino ? &stampst : &cur->inode->st;
704 
705 	memset(dinp, 0, sizeof(*dinp));
706 	dinp->di_mode = cur->inode->st.st_mode;
707 	dinp->di_nlink = cur->inode->nlink;
708 	dinp->di_size = cur->inode->st.st_size;
709 #if HAVE_STRUCT_STAT_ST_FLAGS
710 	dinp->di_flags = cur->inode->st.st_flags;
711 #endif
712 #if HAVE_STRUCT_STAT_ST_GEN
713 	dinp->di_gen = cur->inode->st.st_gen;
714 #endif
715 	dinp->di_uid = cur->inode->st.st_uid;
716 	dinp->di_gid = cur->inode->st.st_gid;
717 
718 	dinp->di_atime = st->st_atime;
719 	dinp->di_mtime = st->st_mtime;
720 	dinp->di_ctime = st->st_ctime;
721 #if HAVE_STRUCT_STAT_ST_MTIMENSEC
722 	dinp->di_atimensec = st->st_atimensec;
723 	dinp->di_mtimensec = st->st_mtimensec;
724 	dinp->di_ctimensec = st->st_ctimensec;
725 #endif
726 #if HAVE_STRUCT_STAT_BIRTHTIME
727 	dinp->di_birthtime = st->st_birthtime;
728 	dinp->di_birthnsec = st->st_birthtimensec;
729 #endif
730 		/* not set: di_db, di_ib, di_blocks, di_spare */
731 
732 	membuf = NULL;
733 	if (cur == root) {			/* "."; write dirbuf */
734 		membuf = dbufp->buf;
735 		dinp->di_size = dbufp->size;
736 	} else if (S_ISBLK(cur->type) || S_ISCHR(cur->type)) {
737 		dinp->di_size = 0;	/* a device */
738 		dinp->di_rdev =
739 		    ufs_rw64(cur->inode->st.st_rdev, fsopts->needswap);
740 	} else if (S_ISLNK(cur->type)) {	/* symlink */
741 		slen = strlen(cur->symlink);
742 		if (slen < UFS2_MAXSYMLINKLEN) {	/* short link */
743 			memcpy(dinp->di_db, cur->symlink, slen);
744 		} else
745 			membuf = cur->symlink;
746 		dinp->di_size = slen;
747 	}
748 	return membuf;
749 }
750 
751 static int
ffs_populate_dir(const char * dir,fsnode * root,fsinfo_t * fsopts)752 ffs_populate_dir(const char *dir, fsnode *root, fsinfo_t *fsopts)
753 {
754 	fsnode		*cur;
755 	dirbuf_t	dirbuf;
756 	union dinode	din;
757 	void		*membuf;
758 	char		path[MAXPATHLEN + 1];
759 	ffs_opt_t	*ffs_opts = fsopts->fs_specific;
760 
761 	assert(dir != NULL);
762 	assert(root != NULL);
763 	assert(fsopts != NULL);
764 	assert(ffs_opts != NULL);
765 
766 	(void)memset(&dirbuf, 0, sizeof(dirbuf));
767 
768 	if (debug & DEBUG_FS_POPULATE)
769 		printf("ffs_populate_dir: PASS 1  dir %s node %p\n", dir, root);
770 
771 		/*
772 		 * pass 1: allocate inode numbers, build directory `file'
773 		 */
774 	for (cur = root; cur != NULL; cur = cur->next) {
775 		if ((cur->inode->flags & FI_ALLOCATED) == 0) {
776 			cur->inode->flags |= FI_ALLOCATED;
777 			if (cur == root && cur->parent != NULL)
778 				cur->inode->ino = cur->parent->inode->ino;
779 			else {
780 				cur->inode->ino = fsopts->curinode;
781 				fsopts->curinode++;
782 			}
783 		}
784 		ffs_make_dirbuf(&dirbuf, cur->name, cur, fsopts->needswap);
785 		if (cur == root) {		/* we're at "."; add ".." */
786 			ffs_make_dirbuf(&dirbuf, "..",
787 			    cur->parent == NULL ? cur : cur->parent->first,
788 			    fsopts->needswap);
789 			root->inode->nlink++;	/* count my parent's link */
790 		} else if (cur->child != NULL)
791 			root->inode->nlink++;	/* count my child's link */
792 
793 		/*
794 		 * XXX	possibly write file and long symlinks here,
795 		 *	ensuring that blocks get written before inodes?
796 		 *	otoh, this isn't a real filesystem, so who
797 		 *	cares about ordering? :-)
798 		 */
799 	}
800 	if (debug & DEBUG_FS_POPULATE_DIRBUF)
801 		ffs_dump_dirbuf(&dirbuf, dir, fsopts->needswap);
802 
803 		/*
804 		 * pass 2: write out dirbuf, then non-directories at this level
805 		 */
806 	if (debug & DEBUG_FS_POPULATE)
807 		printf("ffs_populate_dir: PASS 2  dir %s\n", dir);
808 	for (cur = root; cur != NULL; cur = cur->next) {
809 		if (cur->inode->flags & FI_WRITTEN)
810 			continue;		/* skip hard-linked entries */
811 		cur->inode->flags |= FI_WRITTEN;
812 
813 		if ((size_t)snprintf(path, sizeof(path), "%s/%s/%s", cur->root,
814 		    cur->path, cur->name) >= sizeof(path))
815 			errx(EXIT_FAILURE, "Pathname too long.");
816 
817 		if (cur->child != NULL)
818 			continue;		/* child creates own inode */
819 
820 				/* build on-disk inode */
821 		if (ffs_opts->version == 1)
822 			membuf = ffs_build_dinode1(&din.ffs1_din, &dirbuf, cur,
823 			    root, fsopts);
824 		else
825 			membuf = ffs_build_dinode2(&din.ffs2_din, &dirbuf, cur,
826 			    root, fsopts);
827 
828 		if (debug & DEBUG_FS_POPULATE_NODE) {
829 			printf("ffs_populate_dir: writing ino %lld, %s",
830 			    (long long)cur->inode->ino, inode_type(cur->type));
831 			if (cur->inode->nlink > 1)
832 				printf(", nlink %d", cur->inode->nlink);
833 			putchar('\n');
834 		}
835 
836 		if (membuf != NULL) {
837 			ffs_write_file(&din, cur->inode->ino, membuf, fsopts);
838 		} else if (S_ISREG(cur->type)) {
839 			ffs_write_file(&din, cur->inode->ino, path, fsopts);
840 		} else {
841 			assert (! S_ISDIR(cur->type));
842 			ffs_write_inode(&din, cur->inode->ino, fsopts);
843 		}
844 	}
845 
846 		/*
847 		 * pass 3: write out sub-directories
848 		 */
849 	if (debug & DEBUG_FS_POPULATE)
850 		printf("ffs_populate_dir: PASS 3  dir %s\n", dir);
851 	for (cur = root; cur != NULL; cur = cur->next) {
852 		if (cur->child == NULL)
853 			continue;
854 		if ((size_t)snprintf(path, sizeof(path), "%s/%s", dir,
855 		    cur->name) >= sizeof(path))
856 			errx(EXIT_FAILURE, "Pathname too long.");
857 		if (! ffs_populate_dir(path, cur->child, fsopts))
858 			return (0);
859 	}
860 
861 	if (debug & DEBUG_FS_POPULATE)
862 		printf("ffs_populate_dir: DONE dir %s\n", dir);
863 
864 		/* cleanup */
865 	if (dirbuf.buf != NULL)
866 		free(dirbuf.buf);
867 	return (1);
868 }
869 
870 
871 static void
ffs_write_file(union dinode * din,uint32_t ino,void * buf,fsinfo_t * fsopts)872 ffs_write_file(union dinode *din, uint32_t ino, void *buf, fsinfo_t *fsopts)
873 {
874 	int 	isfile, ffd;
875 	char	*fbuf, *p;
876 	off_t	bufleft, chunk, offset;
877 	ssize_t nread;
878 	struct inode	in;
879 	struct buf *	bp;
880 	ffs_opt_t	*ffs_opts = fsopts->fs_specific;
881 	struct vnode vp = { fsopts, NULL };
882 
883 	assert (din != NULL);
884 	assert (buf != NULL);
885 	assert (fsopts != NULL);
886 	assert (ffs_opts != NULL);
887 
888 	isfile = S_ISREG(DIP(din, mode));
889 	fbuf = NULL;
890 	ffd = -1;
891 	p = NULL;
892 
893 	in.i_fs = (struct fs *)fsopts->superblock;
894 	in.i_devvp = &vp;
895 
896 	if (debug & DEBUG_FS_WRITE_FILE) {
897 		printf(
898 		    "ffs_write_file: ino %u, din %p, isfile %d, %s, size %lld",
899 		    ino, din, isfile, inode_type(DIP(din, mode) & S_IFMT),
900 		    (long long)DIP(din, size));
901 		if (isfile)
902 			printf(", file '%s'\n", (char *)buf);
903 		else
904 			printf(", buffer %p\n", buf);
905 	}
906 
907 	in.i_number = ino;
908 	in.i_size = DIP(din, size);
909 	if (ffs_opts->version == 1)
910 		memcpy(&in.i_din.ffs1_din, &din->ffs1_din,
911 		    sizeof(in.i_din.ffs1_din));
912 	else
913 		memcpy(&in.i_din.ffs2_din, &din->ffs2_din,
914 		    sizeof(in.i_din.ffs2_din));
915 
916 	if (DIP(din, size) == 0)
917 		goto write_inode_and_leave;		/* mmm, cheating */
918 
919 	if (isfile) {
920 		fbuf = emalloc(ffs_opts->bsize);
921 		if ((ffd = open((char *)buf, O_RDONLY, 0444)) == -1) {
922 			warn("Can't open `%s' for reading", (char *)buf);
923 			goto leave_ffs_write_file;
924 		}
925 	} else {
926 		p = buf;
927 	}
928 
929 	chunk = 0;
930 	for (bufleft = DIP(din, size); bufleft > 0; bufleft -= chunk) {
931 		chunk = MIN(bufleft, ffs_opts->bsize);
932 		if (!isfile)
933 			;
934 		else if ((nread = read(ffd, fbuf, chunk)) == -1)
935 			err(EXIT_FAILURE, "Reading `%s', %lld bytes to go",
936 			    (char *)buf, (long long)bufleft);
937 		else if (nread != chunk)
938 			errx(EXIT_FAILURE, "Reading `%s', %lld bytes to go, "
939 			    "read %zd bytes, expected %ju bytes, does "
940 			    "metalog size= attribute mismatch source size?",
941 			    (char *)buf, (long long)bufleft, nread,
942 			    (uintmax_t)chunk);
943 		else
944 			p = fbuf;
945 		offset = DIP(din, size) - bufleft;
946 		if (debug & DEBUG_FS_WRITE_FILE_BLOCK)
947 			printf(
948 		"ffs_write_file: write %p offset %lld size %lld left %lld\n",
949 			    p, (long long)offset,
950 			    (long long)chunk, (long long)bufleft);
951 	/*
952 	 * XXX	if holey support is desired, do the check here
953 	 *
954 	 * XXX	might need to write out last bit in fragroundup
955 	 *	sized chunk. however, ffs_balloc() handles this for us
956 	 */
957 		errno = ffs_balloc(&in, offset, chunk, &bp);
958  bad_ffs_write_file:
959 		if (errno != 0)
960 			err(EXIT_FAILURE,
961 			    "Writing inode %d (%s), bytes %lld + %lld",
962 			    ino,
963 			    isfile ? (char *)buf :
964 			      inode_type(DIP(din, mode) & S_IFMT),
965 			    (long long)offset, (long long)chunk);
966 		memcpy(bp->b_data, p, chunk);
967 		errno = bwrite(bp);
968 		if (errno != 0)
969 			goto bad_ffs_write_file;
970 		if (!isfile)
971 			p += chunk;
972 	}
973 
974  write_inode_and_leave:
975 	ffs_write_inode(&in.i_din, in.i_number, fsopts);
976 
977  leave_ffs_write_file:
978 	if (fbuf)
979 		free(fbuf);
980 	if (ffd != -1)
981 		close(ffd);
982 }
983 
984 
985 static void
ffs_dump_dirbuf(dirbuf_t * dbuf,const char * dir,int needswap)986 ffs_dump_dirbuf(dirbuf_t *dbuf, const char *dir, int needswap)
987 {
988 	doff_t		i;
989 	struct direct	*de;
990 	uint16_t	reclen;
991 
992 	assert (dbuf != NULL);
993 	assert (dir != NULL);
994 	printf("ffs_dump_dirbuf: dir %s size %d cur %d\n",
995 	    dir, dbuf->size, dbuf->cur);
996 
997 	for (i = 0; i < dbuf->size; ) {
998 		de = (struct direct *)(dbuf->buf + i);
999 		reclen = ufs_rw16(de->d_reclen, needswap);
1000 		printf(
1001 	    " inode %4d %7s offset %4d reclen %3d namlen %3d name %s\n",
1002 		    ufs_rw32(de->d_fileno, needswap),
1003 		    inode_type(DTTOIF(de->d_type)), i, reclen,
1004 		    de->d_namlen, de->d_name);
1005 		i += reclen;
1006 		assert(reclen > 0);
1007 	}
1008 }
1009 
1010 static void
ffs_make_dirbuf(dirbuf_t * dbuf,const char * name,fsnode * node,int needswap)1011 ffs_make_dirbuf(dirbuf_t *dbuf, const char *name, fsnode *node, int needswap)
1012 {
1013 	struct direct	de, *dp;
1014 	uint16_t	llen, reclen;
1015 	u_char		*newbuf;
1016 
1017 	assert (dbuf != NULL);
1018 	assert (name != NULL);
1019 	assert (node != NULL);
1020 					/* create direct entry */
1021 	(void)memset(&de, 0, sizeof(de));
1022 	de.d_fileno = ufs_rw32(node->inode->ino, needswap);
1023 	de.d_type = IFTODT(node->type);
1024 	de.d_namlen = (uint8_t)strlen(name);
1025 	strcpy(de.d_name, name);
1026 	reclen = UFS_DIRSIZ(0, &de, needswap);
1027 	de.d_reclen = ufs_rw16(reclen, needswap);
1028 
1029 	dp = (struct direct *)(dbuf->buf + dbuf->cur);
1030 	llen = 0;
1031 	if (dp != NULL)
1032 		llen = UFS_DIRSIZ(0, dp, needswap);
1033 
1034 	if (debug & DEBUG_FS_MAKE_DIRBUF)
1035 		printf(
1036 		    "ffs_make_dirbuf: dbuf siz %d cur %d lastlen %d\n"
1037 		    "  ino %d type %d reclen %d namlen %d name %.30s\n",
1038 		    dbuf->size, dbuf->cur, llen,
1039 		    ufs_rw32(de.d_fileno, needswap), de.d_type, reclen,
1040 		    de.d_namlen, de.d_name);
1041 
1042 	if (reclen + dbuf->cur + llen > roundup(dbuf->size, UFS_DIRBLKSIZ)) {
1043 		if (debug & DEBUG_FS_MAKE_DIRBUF)
1044 			printf("ffs_make_dirbuf: growing buf to %d\n",
1045 			    dbuf->size + UFS_DIRBLKSIZ);
1046 		newbuf = erealloc(dbuf->buf, dbuf->size + UFS_DIRBLKSIZ);
1047 		dbuf->buf = newbuf;
1048 		dbuf->size += UFS_DIRBLKSIZ;
1049 		memset(dbuf->buf + dbuf->size - UFS_DIRBLKSIZ, 0, UFS_DIRBLKSIZ);
1050 		dbuf->cur = dbuf->size - UFS_DIRBLKSIZ;
1051 	} else if (dp) {			/* shrink end of previous */
1052 		dp->d_reclen = ufs_rw16(llen,needswap);
1053 		dbuf->cur += llen;
1054 	}
1055 	dp = (struct direct *)(dbuf->buf + dbuf->cur);
1056 	memcpy(dp, &de, reclen);
1057 	dp->d_reclen = ufs_rw16(dbuf->size - dbuf->cur, needswap);
1058 }
1059 
1060 /*
1061  * cribbed from sys/ufs/ffs/ffs_alloc.c
1062  */
1063 static void
ffs_write_inode(union dinode * dp,uint32_t ino,const fsinfo_t * fsopts)1064 ffs_write_inode(union dinode *dp, uint32_t ino, const fsinfo_t *fsopts)
1065 {
1066 	char 		*buf;
1067 	struct ufs1_dinode *dp1;
1068 	struct ufs2_dinode *dp2, *dip;
1069 	struct cg	*cgp;
1070 	struct fs	*fs;
1071 	uint32_t	cg, cgino, i;
1072 	daddr_t		d;
1073 	char		sbbuf[FFS_MAXBSIZE];
1074 	uint32_t	initediblk;
1075 	ffs_opt_t	*ffs_opts = fsopts->fs_specific;
1076 
1077 	assert (dp != NULL);
1078 	assert (ino > 0);
1079 	assert (fsopts != NULL);
1080 	assert (ffs_opts != NULL);
1081 
1082 	fs = (struct fs *)fsopts->superblock;
1083 	cg = ino_to_cg(fs, ino);
1084 	cgino = ino % fs->fs_ipg;
1085 	if (debug & DEBUG_FS_WRITE_INODE)
1086 		printf("ffs_write_inode: din %p ino %u cg %d cgino %d\n",
1087 		    dp, ino, cg, cgino);
1088 
1089 	ffs_rdfs(FFS_FSBTODB(fs, cgtod(fs, cg)), (int)fs->fs_cgsize, &sbbuf,
1090 	    fsopts);
1091 	cgp = (struct cg *)sbbuf;
1092 	if (!cg_chkmagic(cgp, fsopts->needswap))
1093 		errx(EXIT_FAILURE,
1094 		    "ffs_write_inode: cg %d: bad magic number", cg);
1095 
1096 	assert (isclr(cg_inosused(cgp, fsopts->needswap), cgino));
1097 
1098 	buf = emalloc(fs->fs_bsize);
1099 	dp1 = (struct ufs1_dinode *)buf;
1100 	dp2 = (struct ufs2_dinode *)buf;
1101 
1102 	if (fs->fs_cstotal.cs_nifree == 0)
1103 		errx(EXIT_FAILURE,
1104 		    "ffs_write_inode: fs out of inodes for ino %u", ino);
1105 	if (fs->fs_cs(fs, cg).cs_nifree == 0)
1106 		errx(EXIT_FAILURE,
1107 		    "ffs_write_inode: cg %d out of inodes for ino %u",
1108 		    cg, ino);
1109 	setbit(cg_inosused(cgp, fsopts->needswap), cgino);
1110 	ufs_add32(cgp->cg_cs.cs_nifree, -1, fsopts->needswap);
1111 	fs->fs_cstotal.cs_nifree--;
1112 	fs->fs_cs(fs, cg).cs_nifree--;
1113 	if (S_ISDIR(DIP(dp, mode))) {
1114 		ufs_add32(cgp->cg_cs.cs_ndir, 1, fsopts->needswap);
1115 		fs->fs_cstotal.cs_ndir++;
1116 		fs->fs_cs(fs, cg).cs_ndir++;
1117 	}
1118 
1119 	/*
1120 	 * Initialize inode blocks on the fly for UFS2.
1121 	 */
1122 	initediblk = ufs_rw32(cgp->cg_initediblk, fsopts->needswap);
1123 	while (ffs_opts->version == 2 &&
1124 	    (uint32_t)(cgino + FFS_INOPB(fs)) > initediblk &&
1125 	    initediblk < ufs_rw32(cgp->cg_niblk, fsopts->needswap)) {
1126 		memset(buf, 0, fs->fs_bsize);
1127 		dip = (struct ufs2_dinode *)buf;
1128 		for (i = 0; i < FFS_INOPB(fs); i++) {
1129 			dip->di_gen = random() / 2 + 1;
1130 			dip++;
1131 		}
1132 		ffs_wtfs(FFS_FSBTODB(fs, ino_to_fsba(fs,
1133 				  cg * fs->fs_ipg + initediblk)),
1134 		    fs->fs_bsize, buf, fsopts);
1135 		initediblk += FFS_INOPB(fs);
1136 		cgp->cg_initediblk = ufs_rw32(initediblk, fsopts->needswap);
1137 	}
1138 
1139 
1140 	ffs_wtfs(FFS_FSBTODB(fs, cgtod(fs, cg)), (int)fs->fs_cgsize, &sbbuf,
1141 	    fsopts);
1142 
1143 					/* now write inode */
1144 	d = FFS_FSBTODB(fs, ino_to_fsba(fs, ino));
1145 	ffs_rdfs(d, fs->fs_bsize, buf, fsopts);
1146 	if (fsopts->needswap) {
1147 		if (ffs_opts->version == 1)
1148 			ffs_dinode1_swap(&dp->ffs1_din,
1149 			    &dp1[ino_to_fsbo(fs, ino)]);
1150 		else
1151 			ffs_dinode2_swap(&dp->ffs2_din,
1152 			    &dp2[ino_to_fsbo(fs, ino)]);
1153 	} else {
1154 		if (ffs_opts->version == 1)
1155 			dp1[ino_to_fsbo(fs, ino)] = dp->ffs1_din;
1156 		else
1157 			dp2[ino_to_fsbo(fs, ino)] = dp->ffs2_din;
1158 	}
1159 	ffs_wtfs(d, fs->fs_bsize, buf, fsopts);
1160 	free(buf);
1161 }
1162 
1163 void
panic(const char * fmt,...)1164 panic(const char *fmt, ...)
1165 {
1166 	va_list ap;
1167 
1168 	va_start(ap, fmt);
1169 	vwarnx(fmt, ap);
1170 	va_end(ap);
1171 	exit(EXIT_FAILURE);
1172 }
1173