Home | History | Annotate | Line # | Download | only in chfs
chfs_mkfs.c revision 1.7
      1 /*-
      2  * Copyright (c) 2012 Department of Software Engineering,
      3  *		      University of Szeged, Hungary
      4  * Copyright (c) 2012 Tamas Toth <ttoth (at) 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 
     47 #include "makefs.h"
     48 #include "chfs_makefs.h"
     49 
     50 #include "media.h"
     51 #include "ebh.h"
     52 
     53 #include "chfs/chfs_mkfs.h"
     54 
     55 static uint32_t img_ofs = 0;
     56 static uint64_t version = 0;
     57 static uint64_t max_serial = 0;
     58 static int lebnumber = 0;
     59 
     60 static const unsigned char ffbuf[16] = {
     61 	0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff,
     62 	0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff
     63 };
     64 
     65 static void
     66 buf_write(fsinfo_t *fsopts, const void *buf, size_t len)
     67 {
     68 	ssize_t retval;
     69 	const char *charbuf = buf;
     70 
     71 	while (len > 0) {
     72 		retval = write(fsopts->fd, charbuf, len);
     73 
     74 		if (retval == -1) {
     75 			err(EXIT_FAILURE, "ERROR while writing");
     76 		}
     77 
     78 		len -= retval;
     79 		charbuf += retval;
     80 		img_ofs += retval;
     81 	}
     82 }
     83 
     84 void
     85 padblock(fsinfo_t *fsopts)
     86 {
     87 	chfs_opt_t *chfs_opts = fsopts->fs_specific;
     88 	while (img_ofs % chfs_opts->eraseblock) {
     89 		buf_write(fsopts, ffbuf, MIN(sizeof(ffbuf),
     90 		    chfs_opts->eraseblock - (img_ofs % chfs_opts->eraseblock)));
     91 	}
     92 }
     93 
     94 static void
     95 padword(fsinfo_t *fsopts)
     96 {
     97 	if (img_ofs % 4) {
     98 		buf_write(fsopts, ffbuf, 4 - img_ofs % 4);
     99 	}
    100 }
    101 
    102 static void
    103 pad_block_if_less_than(fsinfo_t *fsopts, int req)
    104 {
    105 	chfs_opt_t *chfs_opts = fsopts->fs_specific;
    106 	if ((img_ofs % chfs_opts->eraseblock) + req >
    107 	    (uint32_t)chfs_opts->eraseblock) {
    108 		padblock(fsopts);
    109 		write_eb_header(fsopts);
    110 	}
    111 }
    112 
    113 void
    114 write_eb_header(fsinfo_t *fsopts)
    115 {
    116 	chfs_opt_t *opts;
    117 	struct chfs_eb_hdr ebhdr;
    118 	char *buf;
    119 
    120 	opts = fsopts->fs_specific;
    121 
    122 #define MINSIZE MAX(MAX(CHFS_EB_EC_HDR_SIZE, CHFS_EB_HDR_NOR_SIZE), \
    123     CHFS_EB_HDR_NAND_SIZE)
    124 	if ((uint32_t)opts->pagesize < MINSIZE)
    125 		errx(EXIT_FAILURE, "pagesize cannot be less than %zu", MINSIZE);
    126 	buf = emalloc(opts->pagesize);
    127 
    128 	ebhdr.ec_hdr.magic = htole32(CHFS_MAGIC_BITMASK);
    129 	ebhdr.ec_hdr.erase_cnt = htole32(1);
    130 	ebhdr.ec_hdr.crc_ec = htole32(crc32(0,
    131 	    (uint8_t *)&ebhdr.ec_hdr + 8, 4));
    132 
    133 	memcpy(buf, &ebhdr.ec_hdr, CHFS_EB_EC_HDR_SIZE);
    134 	memset(buf + CHFS_EB_EC_HDR_SIZE, 0xFF,
    135 	    opts->pagesize - CHFS_EB_EC_HDR_SIZE);
    136 
    137 	buf_write(fsopts, buf, opts->pagesize);
    138 
    139 	memset(buf, 0xFF, opts->pagesize);
    140 
    141 	if (opts->mediatype == TYPE_NAND) {
    142 		ebhdr.u.nand_hdr.lid = htole32(lebnumber++);
    143 		ebhdr.u.nand_hdr.serial = htole64(++(max_serial));
    144 		ebhdr.u.nand_hdr.crc = htole32(crc32(0,
    145 		    (uint8_t *)&ebhdr.u.nand_hdr + 4,
    146 		    CHFS_EB_HDR_NAND_SIZE - 4));
    147 		memcpy(buf, &ebhdr.u.nand_hdr, CHFS_EB_HDR_NAND_SIZE);
    148 	} else {
    149 		ebhdr.u.nor_hdr.lid = htole32(lebnumber++);
    150 		ebhdr.u.nor_hdr.crc = htole32(crc32(0,
    151 		    (uint8_t *)&ebhdr.u.nor_hdr + 4,
    152 		    CHFS_EB_HDR_NOR_SIZE - 4));
    153 		memcpy(buf, &ebhdr.u.nor_hdr, CHFS_EB_HDR_NOR_SIZE);
    154 	}
    155 
    156 	buf_write(fsopts, buf, opts->pagesize);
    157 	free(buf);
    158 }
    159 
    160 void
    161 write_vnode(fsinfo_t *fsopts, fsnode *node)
    162 {
    163 	struct chfs_flash_vnode fvnode;
    164 	memset(&fvnode, 0, sizeof(fvnode));
    165 
    166 	fvnode.magic = htole16(CHFS_FS_MAGIC_BITMASK);
    167 	fvnode.type = htole16(CHFS_NODETYPE_VNODE);
    168 	fvnode.length = htole32(CHFS_PAD(sizeof(fvnode)));
    169 	fvnode.hdr_crc = htole32(crc32(0, (uint8_t *)&fvnode,
    170 	    CHFS_NODE_HDR_SIZE - 4));
    171 	fvnode.vno = htole64(node->inode->ino);
    172 	fvnode.version = htole64(version++);
    173 	fvnode.mode = htole32(node->inode->st.st_mode);
    174 	fvnode.dn_size = htole32(node->inode->st.st_size);
    175 	fvnode.atime = htole32(node->inode->st.st_atime);
    176 	fvnode.ctime = htole32(node->inode->st.st_ctime);
    177 	fvnode.mtime = htole32(node->inode->st.st_mtime);
    178 	fvnode.gid = htole32(node->inode->st.st_uid);
    179 	fvnode.uid = htole32(node->inode->st.st_gid);
    180 	fvnode.node_crc = htole32(crc32(0, (uint8_t *)&fvnode,
    181 	    sizeof(fvnode) - 4));
    182 
    183 	pad_block_if_less_than(fsopts, sizeof(fvnode));
    184 	buf_write(fsopts, &fvnode, sizeof(fvnode));
    185 	padword(fsopts);
    186 }
    187 
    188 void
    189 write_dirent(fsinfo_t *fsopts, fsnode *node)
    190 {
    191 	struct chfs_flash_dirent_node fdirent;
    192 	char *name;
    193 
    194 	name = emalloc(strlen(node->name));
    195 	memcpy(name, node->name, strlen(node->name));
    196 
    197 	memset(&fdirent, 0, sizeof(fdirent));
    198 	fdirent.magic = htole16(CHFS_FS_MAGIC_BITMASK);
    199 	fdirent.type = htole16(CHFS_NODETYPE_DIRENT);
    200 	fdirent.length = htole32(CHFS_PAD(sizeof(fdirent) + strlen(name)));
    201 	fdirent.hdr_crc = htole32(crc32(0, (uint8_t *)&fdirent,
    202 	    CHFS_NODE_HDR_SIZE - 4));
    203 	fdirent.vno = htole64(node->inode->ino);
    204 	if (node->parent != NULL) {
    205 		fdirent.pvno = htole64(node->parent->inode->ino);
    206 	} else {
    207 		fdirent.pvno = htole64(node->inode->ino);
    208 	}
    209 
    210 	fdirent.version = htole64(version++);
    211 	fdirent.mctime = 0;
    212 	fdirent.nsize = htole32(strlen(name));
    213 	fdirent.dtype = htole32(IFTOCHT(node->type & S_IFMT));
    214 	fdirent.name_crc = htole32(crc32(0, (uint8_t *)name, fdirent.nsize));
    215 	fdirent.node_crc = htole32(crc32(0, (uint8_t *)&fdirent,
    216 	    sizeof(fdirent) - 4));
    217 
    218 	pad_block_if_less_than(fsopts, sizeof(fdirent) + fdirent.nsize);
    219 	buf_write(fsopts, &fdirent, sizeof(fdirent));
    220 	buf_write(fsopts, name, fdirent.nsize);
    221 	padword(fsopts);
    222 }
    223 
    224 void
    225 write_file(fsinfo_t *fsopts, fsnode *node, const char *dir)
    226 {
    227 	int fd;
    228 	ssize_t len;
    229 	char *name = node->name;
    230 	chfs_opt_t *opts;
    231 	unsigned char *buf;
    232 	uint32_t fileofs = 0;
    233 
    234 	opts = fsopts->fs_specific;
    235 	buf = emalloc(opts->pagesize);
    236 	if (node->type == S_IFREG || node->type == S_IFSOCK) {
    237 		char *longname;
    238 		if (asprintf(&longname, "%s/%s", dir, name) == 1)
    239 			goto out;
    240 
    241 		fd = open(longname, O_RDONLY, 0444);
    242 		if (fd == -1)
    243 			err(EXIT_FAILURE, "Cannot open `%s'", longname);
    244 
    245 		while ((len = read(fd, buf, opts->pagesize))) {
    246 			if (len < 0) {
    247 				warn("ERROR while reading %s", longname);
    248 				free(longname);
    249 				free(buf);
    250 				close(fd);
    251 				return;
    252 			}
    253 
    254 			write_data(fsopts, node, buf, len, fileofs);
    255 			fileofs += len;
    256 		}
    257 		free(longname);
    258 		close(fd);
    259 	} else if (node->type == S_IFLNK) {
    260 		len = strlen(node->symlink);
    261 		memcpy(buf, node->symlink, len);
    262 		write_data(fsopts, node, buf, len, 0);
    263 	} else if (node->type == S_IFCHR || node->type == S_IFBLK ||
    264 		node->type == S_IFIFO) {
    265 		len = sizeof(dev_t);
    266 		memcpy(buf, &(node->inode->st.st_rdev), len);
    267 		write_data(fsopts, node, buf, len, 0);
    268 	}
    269 
    270 	free(buf);
    271 	return;
    272 out:
    273 	err(EXIT_FAILURE, "Memory allocation failed");
    274 }
    275 
    276 void
    277 write_data(fsinfo_t *fsopts, fsnode *node, unsigned char *buf, size_t len,
    278     uint32_t ofs)
    279 {
    280 	struct chfs_flash_data_node fdata;
    281 
    282 	memset(&fdata, 0, sizeof(fdata));
    283 	if (len == 0) {
    284 		return;
    285 	}
    286 
    287 	pad_block_if_less_than(fsopts, sizeof(fdata) + len);
    288 
    289 	fdata.magic = htole16(CHFS_FS_MAGIC_BITMASK);
    290 	fdata.type = htole16(CHFS_NODETYPE_DATA);
    291 	fdata.length = htole32(CHFS_PAD(sizeof(fdata) + len));
    292 	fdata.hdr_crc = htole32(crc32(0, (uint8_t *)&fdata,
    293 	    CHFS_NODE_HDR_SIZE - 4));
    294 	fdata.vno = htole64(node->inode->ino);
    295 	fdata.data_length = htole32(len);
    296 	fdata.offset = htole32(ofs);
    297 	fdata.data_crc = htole32(crc32(0, (uint8_t *)buf, len));
    298 	fdata.node_crc = htole32(crc32(0,
    299 	    (uint8_t *)&fdata, sizeof(fdata) - 4));
    300 
    301 	buf_write(fsopts, &fdata, sizeof(fdata));
    302 	buf_write(fsopts, buf, len);
    303 	padword(fsopts);
    304 }
    305