1 /* $NetBSD: v7fs.c,v 1.8 2013/01/29 15:52:25 christos Exp $ */
2
3 /*-
4 * Copyright (c) 2011 The NetBSD Foundation, Inc.
5 * All rights reserved.
6 *
7 * This code is derived from software contributed to The NetBSD Foundation
8 * by UCHIYAMA Yasushi.
9 *
10 * Redistribution and use in source and binary forms, with or without
11 * modification, are permitted provided that the following conditions
12 * are met:
13 * 1. Redistributions of source code must retain the above copyright
14 * notice, this list of conditions and the following disclaimer.
15 * 2. Redistributions in binary form must reproduce the above copyright
16 * notice, this list of conditions and the following disclaimer in the
17 * documentation and/or other materials provided with the distribution.
18 *
19 * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
20 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
21 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
22 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
23 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
24 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
25 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
26 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
27 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
28 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
29 * POSSIBILITY OF SUCH DAMAGE.
30 */
31
32 #if HAVE_NBTOOL_CONFIG_H
33 #include "nbtool_config.h"
34 #endif
35
36 #include <sys/cdefs.h>
37 #if defined(__RCSID) && !defined(__lint)
38 __RCSID("$NetBSD: v7fs.c,v 1.8 2013/01/29 15:52:25 christos Exp $");
39 #endif /* !__lint */
40
41 #include <stdio.h>
42 #include <stdlib.h>
43 #include <unistd.h>
44 #include <string.h>
45 #include <fcntl.h>
46 #include <util.h>
47
48 #include "makefs.h"
49 #include "v7fs.h"
50 #include "v7fs_impl.h"
51 #include "v7fs_makefs.h"
52 #include "newfs_v7fs.h"
53
54
55 #ifndef HAVE_NBTOOL_CONFIG_H
56 #include "progress.h"
57 static bool progress_bar_enable;
58 #endif
59 int v7fs_newfs_verbose;
60
61 void
v7fs_prep_opts(fsinfo_t * fsopts)62 v7fs_prep_opts(fsinfo_t *fsopts)
63 {
64 v7fs_opt_t *v7fs_opts = ecalloc(1, sizeof(*v7fs_opts));
65 const option_t v7fs_options[] = {
66 { 'p', "pdp", &v7fs_opts->pdp_endian, OPT_INT32, false, true,
67 "PDP endian" },
68 { 'P', "progress", &v7fs_opts->progress, OPT_INT32, false, true,
69 "Progress bar" },
70 { .name = NULL }
71 };
72
73 fsopts->fs_specific = v7fs_opts;
74 fsopts->fs_options = copy_opts(v7fs_options);
75 }
76
77 void
v7fs_cleanup_opts(fsinfo_t * fsopts)78 v7fs_cleanup_opts(fsinfo_t *fsopts)
79 {
80 free(fsopts->fs_specific);
81 free(fsopts->fs_options);
82 }
83
84 int
v7fs_parse_opts(const char * option,fsinfo_t * fsopts)85 v7fs_parse_opts(const char *option, fsinfo_t *fsopts)
86 {
87
88 return set_option_var(fsopts->fs_options, option, "1", NULL, 0) != -1;
89 }
90
91 void
v7fs_makefs(const char * image,const char * dir,fsnode * root,fsinfo_t * fsopts)92 v7fs_makefs(const char *image, const char *dir, fsnode *root, fsinfo_t *fsopts)
93 {
94 struct v7fs_mount_device v7fs_mount;
95 int fd, endian, error = 1;
96 v7fs_opt_t *v7fs_opts = fsopts->fs_specific;
97
98 v7fs_newfs_verbose = debug;
99 #ifndef HAVE_NBTOOL_CONFIG_H
100 if ((progress_bar_enable = v7fs_opts->progress)) {
101 progress_switch(progress_bar_enable);
102 progress_init();
103 progress(&(struct progress_arg){ .cdev = image });
104 }
105 #endif
106
107 /* Determine filesystem image size */
108 v7fs_estimate(dir, root, fsopts);
109 printf("Calculated size of `%s': %lld bytes, %ld inodes\n",
110 image, (long long)fsopts->size, (long)fsopts->inodes);
111
112 if ((fd = open(image, O_RDWR | O_CREAT | O_TRUNC, 0666)) == -1) {
113 err(EXIT_FAILURE, "%s", image);
114 }
115 if (lseek(fd, fsopts->size - 1, SEEK_SET) == -1) {
116 goto err_exit;
117 }
118 if (write(fd, &fd, 1) != 1) {
119 goto err_exit;
120 }
121 if (lseek(fd, 0, SEEK_SET) == -1) {
122 goto err_exit;
123 }
124 fsopts->fd = fd;
125 v7fs_mount.device.fd = fd;
126
127 #if !defined BYTE_ORDER
128 #error
129 #endif
130 #if BYTE_ORDER == LITTLE_ENDIAN
131 if (fsopts->needswap)
132 endian = BIG_ENDIAN;
133 else
134 endian = LITTLE_ENDIAN;
135 #else
136 if (fsopts->needswap)
137 endian = LITTLE_ENDIAN;
138 else
139 endian = BIG_ENDIAN;
140 #endif
141 if (v7fs_opts->pdp_endian) {
142 endian = PDP_ENDIAN;
143 }
144
145 v7fs_mount.endian = endian;
146 v7fs_mount.sectors = fsopts->size >> V7FS_BSHIFT;
147 if (v7fs_newfs(&v7fs_mount, fsopts->inodes) != 0) {
148 goto err_exit;
149 }
150
151 if (v7fs_populate(dir, root, fsopts, &v7fs_mount) != 0) {
152 error = 2; /* some files couldn't add */
153 goto err_exit;
154 }
155
156 close(fd);
157 return;
158
159 err_exit:
160 close(fd);
161 err(error, "%s", image);
162 }
163
164 void
progress(const struct progress_arg * p)165 progress(const struct progress_arg *p)
166 {
167 #ifndef HAVE_NBTOOL_CONFIG_H
168 static struct progress_arg Progress;
169 static char cdev[32];
170 static char label[32];
171
172 if (!progress_bar_enable)
173 return;
174
175 if (p) {
176 Progress = *p;
177 if (p->cdev)
178 strcpy(cdev, p->cdev);
179 if (p->label)
180 strcpy(label, p->label);
181 }
182
183 if (!Progress.tick)
184 return;
185 if (++Progress.cnt > Progress.tick) {
186 Progress.cnt = 0;
187 Progress.total++;
188 progress_bar(cdev, label, Progress.total, PROGRESS_BAR_GRANULE);
189 }
190 #endif
191 }
192