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