xref: /netbsd-src/usr.sbin/makefs/chfs/chfs_mkfs.c (revision bf2ad7248906fbda5c0a68bac7ee290dcd2d2b38)
1 /*-
2  * Copyright (c) 2012 Department of Software Engineering,
3  *		      University of Szeged, Hungary
4  * Copyright (c) 2012 Tamas Toth <ttoth@inf.u-szeged.hu>
5  * All rights reserved.
6  *
7  * This code is derived from software contributed to The NetBSD Foundation
8  * by the Department of Software Engineering, University of Szeged, Hungary
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 AUTHOR ``AS IS'' AND ANY EXPRESS OR
20  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
21  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
22  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
23  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
24  * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
25  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
26  * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
27  * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
28  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
29  * SUCH DAMAGE.
30  */
31 
32 #if HAVE_NBTOOL_CONFIG_H
33 #include "nbtool_config.h"
34 #endif
35 
36 #include <sys/param.h>
37 #include <sys/stat.h>
38 
39 #include <assert.h>
40 #include <fcntl.h>
41 #include <stdio.h>
42 #include <stdlib.h>
43 #include <string.h>
44 #include <zlib.h>
45 #include <util.h>
46 #include <unistd.h>
47 
48 #include "makefs.h"
49 #include "chfs_makefs.h"
50 
51 #include "media.h"
52 #include "ebh.h"
53 
54 #include "chfs/chfs_mkfs.h"
55 
56 static uint32_t img_ofs = 0;
57 static uint64_t version = 0;
58 static uint64_t max_serial = 0;
59 static int lebnumber = 0;
60 
61 static const unsigned char ffbuf[16] = {
62 	0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff,
63 	0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff
64 };
65 
66 static void
buf_write(fsinfo_t * fsopts,const void * buf,size_t len)67 buf_write(fsinfo_t *fsopts, const void *buf, size_t len)
68 {
69 	ssize_t retval;
70 	const char *charbuf = buf;
71 
72 	while (len > 0) {
73 		retval = write(fsopts->fd, charbuf, len);
74 
75 		if (retval == -1) {
76 			err(EXIT_FAILURE, "ERROR while writing");
77 		}
78 
79 		len -= retval;
80 		charbuf += retval;
81 		img_ofs += retval;
82 	}
83 }
84 
85 void
padblock(fsinfo_t * fsopts)86 padblock(fsinfo_t *fsopts)
87 {
88 	chfs_opt_t *chfs_opts = fsopts->fs_specific;
89 	while (img_ofs % chfs_opts->eraseblock) {
90 		buf_write(fsopts, ffbuf, MIN(sizeof(ffbuf),
91 		    chfs_opts->eraseblock - (img_ofs % chfs_opts->eraseblock)));
92 	}
93 }
94 
95 static void
padword(fsinfo_t * fsopts)96 padword(fsinfo_t *fsopts)
97 {
98 	if (img_ofs % 4) {
99 		buf_write(fsopts, ffbuf, 4 - img_ofs % 4);
100 	}
101 }
102 
103 static void
pad_block_if_less_than(fsinfo_t * fsopts,int req)104 pad_block_if_less_than(fsinfo_t *fsopts, int req)
105 {
106 	chfs_opt_t *chfs_opts = fsopts->fs_specific;
107 	if ((img_ofs % chfs_opts->eraseblock) + req >
108 	    (uint32_t)chfs_opts->eraseblock) {
109 		padblock(fsopts);
110 		write_eb_header(fsopts);
111 	}
112 }
113 
114 void
write_eb_header(fsinfo_t * fsopts)115 write_eb_header(fsinfo_t *fsopts)
116 {
117 	chfs_opt_t *opts;
118 	struct chfs_eb_hdr ebhdr;
119 	char *buf;
120 
121 	opts = fsopts->fs_specific;
122 
123 #define MINSIZE MAX(MAX(CHFS_EB_EC_HDR_SIZE, CHFS_EB_HDR_NOR_SIZE), \
124     CHFS_EB_HDR_NAND_SIZE)
125 	if ((uint32_t)opts->pagesize < MINSIZE)
126 		errx(EXIT_FAILURE, "pagesize cannot be less than %zu", MINSIZE);
127 	buf = emalloc(opts->pagesize);
128 
129 	ebhdr.ec_hdr.magic = htole32(CHFS_MAGIC_BITMASK);
130 	ebhdr.ec_hdr.erase_cnt = htole32(1);
131 	ebhdr.ec_hdr.crc_ec = htole32(crc32(0,
132 	    (uint8_t *)&ebhdr.ec_hdr + 8, 4));
133 
134 	memcpy(buf, &ebhdr.ec_hdr, CHFS_EB_EC_HDR_SIZE);
135 	memset(buf + CHFS_EB_EC_HDR_SIZE, 0xFF,
136 	    opts->pagesize - CHFS_EB_EC_HDR_SIZE);
137 
138 	buf_write(fsopts, buf, opts->pagesize);
139 
140 	memset(buf, 0xFF, opts->pagesize);
141 
142 	if (opts->mediatype == TYPE_NAND) {
143 		ebhdr.u.nand_hdr.lid = htole32(lebnumber++);
144 		ebhdr.u.nand_hdr.serial = htole64(++(max_serial));
145 		ebhdr.u.nand_hdr.crc = htole32(crc32(0,
146 		    (uint8_t *)&ebhdr.u.nand_hdr + 4,
147 		    CHFS_EB_HDR_NAND_SIZE - 4));
148 		memcpy(buf, &ebhdr.u.nand_hdr, CHFS_EB_HDR_NAND_SIZE);
149 	} else {
150 		ebhdr.u.nor_hdr.lid = htole32(lebnumber++);
151 		ebhdr.u.nor_hdr.crc = htole32(crc32(0,
152 		    (uint8_t *)&ebhdr.u.nor_hdr + 4,
153 		    CHFS_EB_HDR_NOR_SIZE - 4));
154 		memcpy(buf, &ebhdr.u.nor_hdr, CHFS_EB_HDR_NOR_SIZE);
155 	}
156 
157 	buf_write(fsopts, buf, opts->pagesize);
158 	free(buf);
159 }
160 
161 void
write_vnode(fsinfo_t * fsopts,fsnode * node)162 write_vnode(fsinfo_t *fsopts, fsnode *node)
163 {
164 	struct chfs_flash_vnode fvnode;
165 	memset(&fvnode, 0, sizeof(fvnode));
166 
167 	fvnode.magic = htole16(CHFS_FS_MAGIC_BITMASK);
168 	fvnode.type = htole16(CHFS_NODETYPE_VNODE);
169 	fvnode.length = htole32(CHFS_PAD(sizeof(fvnode)));
170 	fvnode.hdr_crc = htole32(crc32(0, (uint8_t *)&fvnode,
171 	    CHFS_NODE_HDR_SIZE - 4));
172 	fvnode.vno = htole64(node->inode->ino);
173 	fvnode.version = htole64(version++);
174 	fvnode.mode = htole32(node->inode->st.st_mode);
175 	fvnode.dn_size = htole32(node->inode->st.st_size);
176 	fvnode.atime = htole32(node->inode->st.st_atime);
177 	fvnode.ctime = htole32(node->inode->st.st_ctime);
178 	fvnode.mtime = htole32(node->inode->st.st_mtime);
179 	fvnode.gid = htole32(node->inode->st.st_uid);
180 	fvnode.uid = htole32(node->inode->st.st_gid);
181 	fvnode.node_crc = htole32(crc32(0, (uint8_t *)&fvnode,
182 	    sizeof(fvnode) - 4));
183 
184 	pad_block_if_less_than(fsopts, sizeof(fvnode));
185 	buf_write(fsopts, &fvnode, sizeof(fvnode));
186 	padword(fsopts);
187 }
188 
189 void
write_dirent(fsinfo_t * fsopts,fsnode * node)190 write_dirent(fsinfo_t *fsopts, fsnode *node)
191 {
192 	struct chfs_flash_dirent_node fdirent;
193 	char *name;
194 
195 	name = emalloc(strlen(node->name));
196 	memcpy(name, node->name, strlen(node->name));
197 
198 	memset(&fdirent, 0, sizeof(fdirent));
199 	fdirent.magic = htole16(CHFS_FS_MAGIC_BITMASK);
200 	fdirent.type = htole16(CHFS_NODETYPE_DIRENT);
201 	fdirent.length = htole32(CHFS_PAD(sizeof(fdirent) + strlen(name)));
202 	fdirent.hdr_crc = htole32(crc32(0, (uint8_t *)&fdirent,
203 	    CHFS_NODE_HDR_SIZE - 4));
204 	fdirent.vno = htole64(node->inode->ino);
205 	if (node->parent != NULL) {
206 		fdirent.pvno = htole64(node->parent->inode->ino);
207 	} else {
208 		fdirent.pvno = htole64(node->inode->ino);
209 	}
210 
211 	fdirent.version = htole64(version++);
212 	fdirent.mctime = 0;
213 	fdirent.nsize = htole32(strlen(name));
214 	fdirent.dtype = htole32(IFTOCHT(node->type & S_IFMT));
215 	fdirent.name_crc = htole32(crc32(0, (uint8_t *)name, fdirent.nsize));
216 	fdirent.node_crc = htole32(crc32(0, (uint8_t *)&fdirent,
217 	    sizeof(fdirent) - 4));
218 
219 	pad_block_if_less_than(fsopts, sizeof(fdirent) + fdirent.nsize);
220 	buf_write(fsopts, &fdirent, sizeof(fdirent));
221 	buf_write(fsopts, name, fdirent.nsize);
222 	padword(fsopts);
223 }
224 
225 void
write_file(fsinfo_t * fsopts,fsnode * node,const char * dir)226 write_file(fsinfo_t *fsopts, fsnode *node, const char *dir)
227 {
228 	int fd;
229 	ssize_t len;
230 	char *name = node->name;
231 	chfs_opt_t *opts;
232 	unsigned char *buf;
233 	uint32_t fileofs = 0;
234 
235 	opts = fsopts->fs_specific;
236 	buf = emalloc(opts->pagesize);
237 	if (node->type == S_IFREG || node->type == S_IFSOCK) {
238 		char *longname;
239 		if (asprintf(&longname, "%s/%s", dir, name) == 1)
240 			goto out;
241 
242 		fd = open(longname, O_RDONLY, 0444);
243 		if (fd == -1)
244 			err(EXIT_FAILURE, "Cannot open `%s'", longname);
245 
246 		while ((len = read(fd, buf, opts->pagesize))) {
247 			if (len < 0) {
248 				warn("ERROR while reading %s", longname);
249 				free(longname);
250 				free(buf);
251 				close(fd);
252 				return;
253 			}
254 
255 			write_data(fsopts, node, buf, len, fileofs);
256 			fileofs += len;
257 		}
258 		free(longname);
259 		close(fd);
260 	} else if (node->type == S_IFLNK) {
261 		len = strlen(node->symlink);
262 		memcpy(buf, node->symlink, len);
263 		write_data(fsopts, node, buf, len, 0);
264 	} else if (node->type == S_IFCHR || node->type == S_IFBLK ||
265 		node->type == S_IFIFO) {
266 		len = sizeof(dev_t);
267 		memcpy(buf, &(node->inode->st.st_rdev), len);
268 		write_data(fsopts, node, buf, len, 0);
269 	}
270 
271 	free(buf);
272 	return;
273 out:
274 	err(EXIT_FAILURE, "Memory allocation failed");
275 }
276 
277 void
write_data(fsinfo_t * fsopts,fsnode * node,unsigned char * buf,size_t len,uint32_t ofs)278 write_data(fsinfo_t *fsopts, fsnode *node, unsigned char *buf, size_t len,
279     uint32_t ofs)
280 {
281 	struct chfs_flash_data_node fdata;
282 
283 	memset(&fdata, 0, sizeof(fdata));
284 	if (len == 0) {
285 		return;
286 	}
287 
288 	pad_block_if_less_than(fsopts, sizeof(fdata) + len);
289 
290 	fdata.magic = htole16(CHFS_FS_MAGIC_BITMASK);
291 	fdata.type = htole16(CHFS_NODETYPE_DATA);
292 	fdata.length = htole32(CHFS_PAD(sizeof(fdata) + len));
293 	fdata.hdr_crc = htole32(crc32(0, (uint8_t *)&fdata,
294 	    CHFS_NODE_HDR_SIZE - 4));
295 	fdata.vno = htole64(node->inode->ino);
296 	fdata.data_length = htole32(len);
297 	fdata.offset = htole32(ofs);
298 	fdata.data_crc = htole32(crc32(0, (uint8_t *)buf, len));
299 	fdata.node_crc = htole32(crc32(0,
300 	    (uint8_t *)&fdata, sizeof(fdata) - 4));
301 
302 	buf_write(fsopts, &fdata, sizeof(fdata));
303 	buf_write(fsopts, buf, len);
304 	padword(fsopts);
305 }
306