usr.sbin/makefs/chfs/chfs_mkfs.c

   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
  47 #if defined(__minix)
  48 #include <unistd.h>
  49 #endif
  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         memset(buf, 0xFF, opts->pagesize);
 132
 133         ebhdr.ec_hdr.magic = htole32(CHFS_MAGIC_BITMASK);
 134         ebhdr.ec_hdr.erase_cnt = htole32(1);
 135         ebhdr.ec_hdr.crc_ec = htole32(crc32(0,
 136             (uint8_t *)&ebhdr.ec_hdr + 8, 4));
 137
 138         memcpy(buf, &ebhdr.ec_hdr, CHFS_EB_EC_HDR_SIZE);
 139
 140         buf_write(fsopts, buf, opts->pagesize);
 141
 142         memset(buf, 0xFF, opts->pagesize);
 143
 144         if (opts->mediatype == TYPE_NAND) {
 145                 ebhdr.u.nand_hdr.lid = htole32(lebnumber++);
 146                 ebhdr.u.nand_hdr.serial = htole64(++(max_serial));
 147                 ebhdr.u.nand_hdr.crc = htole32(crc32(0,
 148                     (uint8_t *)&ebhdr.u.nand_hdr + 4,
 149                     CHFS_EB_HDR_NAND_SIZE - 4));
 150                 memcpy(buf, &ebhdr.u.nand_hdr, CHFS_EB_HDR_NAND_SIZE);
 151         } else {
 152                 ebhdr.u.nor_hdr.lid = htole32(lebnumber++);
 153                 ebhdr.u.nor_hdr.crc = htole32(crc32(0,
 154                     (uint8_t *)&ebhdr.u.nor_hdr + 4,
 155                     CHFS_EB_HDR_NOR_SIZE - 4));
 156                 memcpy(buf, &ebhdr.u.nor_hdr, CHFS_EB_HDR_NOR_SIZE);
 157         }
 158
 159         buf_write(fsopts, buf, opts->pagesize);
 160         free(buf);
 161 }
 162
 163 void
 164 write_vnode(fsinfo_t *fsopts, fsnode *node)
 165 {
 166         struct chfs_flash_vnode fvnode;
 167         memset(&fvnode, 0, sizeof(fvnode));
 168
 169         fvnode.magic = htole16(CHFS_FS_MAGIC_BITMASK);
 170         fvnode.type = htole16(CHFS_NODETYPE_VNODE);
 171         fvnode.length = htole32(CHFS_PAD(sizeof(fvnode)));
 172         fvnode.hdr_crc = htole32(crc32(0, (uint8_t *)&fvnode,
 173             CHFS_NODE_HDR_SIZE - 4));
 174         fvnode.vno = htole64(node->inode->ino);
 175         fvnode.version = htole64(version++);
 176         fvnode.mode = htole32(node->inode->st.st_mode);
 177         fvnode.dn_size = htole32(node->inode->st.st_size);
 178         fvnode.atime = htole32(node->inode->st.st_atime);
 179         fvnode.ctime = htole32(node->inode->st.st_ctime);
 180         fvnode.mtime = htole32(node->inode->st.st_mtime);
 181         fvnode.gid = htole32(node->inode->st.st_uid);
 182         fvnode.uid = htole32(node->inode->st.st_gid);
 183         fvnode.node_crc = htole32(crc32(0, (uint8_t *)&fvnode,
 184             sizeof(fvnode) - 4));
 185
 186         pad_block_if_less_than(fsopts, sizeof(fvnode));
 187         buf_write(fsopts, &fvnode, sizeof(fvnode));
 188         padword(fsopts);
 189 }
 190
 191 void
 192 write_dirent(fsinfo_t *fsopts, fsnode *node)
 193 {
 194         struct chfs_flash_dirent_node fdirent;
 195         char *name;
 196
 197         name = emalloc(strlen(node->name));
 198         memcpy(name, node->name, strlen(node->name));
 199
 200         memset(&fdirent, 0, sizeof(fdirent));
 201         fdirent.magic = htole16(CHFS_FS_MAGIC_BITMASK);
 202         fdirent.type = htole16(CHFS_NODETYPE_DIRENT);
 203         fdirent.length = htole32(CHFS_PAD(sizeof(fdirent) + strlen(name)));
 204         fdirent.hdr_crc = htole32(crc32(0, (uint8_t *)&fdirent,
 205             CHFS_NODE_HDR_SIZE - 4));
 206         fdirent.vno = htole64(node->inode->ino);
 207         if (node->parent != NULL) {
 208                 fdirent.pvno = htole64(node->parent->inode->ino);
 209         } else {
 210                 fdirent.pvno = htole64(node->inode->ino);
 211         }
 212
 213         fdirent.version = htole64(version++);
 214         fdirent.mctime = 0;
 215         fdirent.nsize = htole32(strlen(name));
 216         fdirent.dtype = htole32(IFTOCHT(node->type & S_IFMT));
 217         fdirent.name_crc = htole32(crc32(0, (uint8_t *)name, fdirent.nsize));
 218         fdirent.node_crc = htole32(crc32(0, (uint8_t *)&fdirent,
 219             sizeof(fdirent) - 4));
 220
 221         pad_block_if_less_than(fsopts, sizeof(fdirent) + fdirent.nsize);
 222         buf_write(fsopts, &fdirent, sizeof(fdirent));
 223         buf_write(fsopts, name, fdirent.nsize);
 224         padword(fsopts);
 225 }
 226
 227 void
 228 write_file(fsinfo_t *fsopts, fsnode *node, const char *dir)
 229 {
 230         int fd;
 231         ssize_t len;
 232         char *name = node->name;
 233         chfs_opt_t *opts;
 234         unsigned char *buf;
 235         uint32_t fileofs = 0;
 236
 237         opts = fsopts->fs_specific;
 238         buf = emalloc(opts->pagesize);
 239         if (node->type == S_IFREG || node->type == S_IFSOCK) {
 240                 char *longname;
 241                 if (asprintf(&longname, "%s/%s", dir, name) == 1)
 242                         goto out;
 243
 244                 fd = open(longname, O_RDONLY, 0444);
 245                 if (fd == -1)
 246                         err(EXIT_FAILURE, "Cannot open `%s'", longname);
 247
 248                 while ((len = read(fd, buf, opts->pagesize))) {
 249                         if (len < 0) {
 250                                 warn("ERROR while reading %s", longname);
 251                                 free(longname);
 252                                 free(buf);
 253                                 close(fd);
 254                                 return;
 255                         }
 256
 257                         write_data(fsopts, node, buf, len, fileofs);
 258                         fileofs += len;
 259                 }
 260                 free(longname);
 261                 close(fd);
 262         } else if (node->type == S_IFLNK) {
 263                 len = strlen(node->symlink);
 264                 memcpy(buf, node->symlink, len);
 265                 write_data(fsopts, node, buf, len, 0);
 266         } else if (node->type == S_IFCHR || node->type == S_IFBLK ||
 267                 node->type == S_IFIFO) {
 268                 len = sizeof(dev_t);
 269                 memcpy(buf, &(node->inode->st.st_rdev), len);
 270                 write_data(fsopts, node, buf, len, 0);
 271         }
 272
 273         free(buf);
 274         return;
 275 out:
 276         err(EXIT_FAILURE, "Memory allocation failed");
 277 }
 278
 279 void
 280 write_data(fsinfo_t *fsopts, fsnode *node, unsigned char *buf, size_t len,
 281     uint32_t ofs)
 282 {
 283         struct chfs_flash_data_node fdata;
 284
 285         memset(&fdata, 0, sizeof(fdata));
 286         if (len == 0) {
 287                 return;
 288         }
 289
 290         pad_block_if_less_than(fsopts, sizeof(fdata) + len);
 291
 292         fdata.magic = htole16(CHFS_FS_MAGIC_BITMASK);
 293         fdata.type = htole16(CHFS_NODETYPE_DATA);
 294         fdata.length = htole32(CHFS_PAD(sizeof(fdata) + len));
 295         fdata.hdr_crc = htole32(crc32(0, (uint8_t *)&fdata,
 296             CHFS_NODE_HDR_SIZE - 4));
 297         fdata.vno = htole64(node->inode->ino);
 298         fdata.data_length = htole32(len);
 299         fdata.offset = htole32(ofs);
 300         fdata.data_crc = htole32(crc32(0, (uint8_t *)buf, len));
 301         fdata.node_crc = htole32(crc32(0,
 302             (uint8_t *)&fdata, sizeof(fdata) - 4));
 303
 304         buf_write(fsopts, &fdata, sizeof(fdata));
 305         buf_write(fsopts, buf, len);
 306         padword(fsopts);
 307 }