usr.sbin/makefs/v7fs/v7fs_populate.c

   1 /*      $NetBSD: v7fs_populate.c,v 1.3 2011/08/10 11:31:49 uch 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_populate.c,v 1.3 2011/08/10 11:31:49 uch Exp $");
  39 #endif  /* !__lint */
  40
  41 #include <stdio.h>
  42 #include <stdlib.h>
  43 #include <string.h>
  44 #include <unistd.h>
  45 #include <fcntl.h>
  46 #include <errno.h>
  47 #include <err.h>
  48
  49 #if !HAVE_NBTOOL_CONFIG_H
  50 #include <sys/mount.h>
  51 #endif
  52
  53 #include "makefs.h"
  54 #include "v7fs.h"
  55 #include "v7fs_impl.h"
  56 #include "v7fs_inode.h"
  57 #include "v7fs_superblock.h"
  58 #include "v7fs_datablock.h"
  59 #include "v7fs_endian.h"
  60 #include "v7fs_file.h"
  61 #include "v7fs_makefs.h"
  62 #include "newfs_v7fs.h"
  63
  64 #define VPRINTF(fmt, args...)   { if (v7fs_newfs_verbose) printf(fmt, ##args); }
  65
  66 static void
  67 attr_setup(fsnode *node, struct v7fs_fileattr *attr)
  68 {
  69         struct stat *st = &node->inode->st;
  70
  71         attr->mode = node->type | st->st_mode;
  72         attr->uid = st->st_uid;
  73         attr->gid = st->st_gid;
  74         attr->device = 0;
  75         attr->ctime = st->st_ctime;
  76         attr->atime = st->st_atime;
  77         attr->mtime = st->st_mtime;
  78 }
  79
  80 static int
  81 allocate(struct v7fs_self *fs, struct v7fs_inode *parent_inode, fsnode *node,
  82     dev_t dev, struct v7fs_inode *inode)
  83 {
  84         int error;
  85         v7fs_ino_t ino;
  86         struct v7fs_fileattr attr;
  87
  88         memset(inode, 0, sizeof(*inode));
  89
  90         attr_setup(node, &attr);
  91         attr.device = dev;
  92         if ((error = v7fs_file_allocate(fs, parent_inode, node->name, &attr,
  93             &ino))) {
  94                 errno = error;
  95                 warn("%s", node->name);
  96                 return error;
  97         }
  98         node->inode->ino = ino;
  99         node->inode->flags |= FI_ALLOCATED;
 100         if ((error = v7fs_inode_load(fs, inode, ino))) {
 101                 errno = error;
 102                 warn("%s", node->name);
 103                 return error;
 104         }
 105
 106         return 0;
 107 }
 108
 109 struct copy_arg {
 110         int fd;
 111         uint8_t buf[V7FS_BSIZE];
 112 };
 113
 114 static int
 115 copy_subr(struct v7fs_self *fs, void *ctx, v7fs_daddr_t blk, size_t sz)
 116 {
 117         struct copy_arg *p = ctx;
 118
 119         if (read(p->fd, p->buf, sz) != (ssize_t)sz) {
 120                 return V7FS_ITERATOR_ERROR;
 121         }
 122
 123         if (!fs->io.write(fs->io.cookie, p->buf, blk)) {
 124                 errno = EIO;
 125                 return V7FS_ITERATOR_ERROR;
 126         }
 127         progress(0);
 128
 129         return 0;
 130 }
 131
 132 static int
 133 file_copy(struct v7fs_self *fs, struct v7fs_inode *parent_inode, fsnode *node,
 134         const char *filepath)
 135 {
 136         struct v7fs_inode inode;
 137         const char *errmsg;
 138         fsinode *fnode = node->inode;
 139         int error = 0;
 140         int fd;
 141
 142         /* Check hard-link */
 143         if ((fnode->nlink > 1) && (fnode->flags & FI_ALLOCATED)) {
 144                 if ((error = v7fs_inode_load(fs, &inode, fnode->ino))) {
 145                         errmsg = "inode load";
 146                         goto err_exit;
 147                 }
 148                 if ((error = v7fs_file_link(fs, parent_inode, &inode,
 149                             node->name))) {
 150                         errmsg = "hard link";
 151                         goto err_exit;
 152                 }
 153                 return 0;
 154         }
 155
 156         /* Allocate file */
 157         if ((error = allocate(fs, parent_inode, node, 0, &inode))) {
 158                 errmsg = "file allocate";
 159                 goto err_exit;
 160         }
 161         if ((error = v7fs_datablock_expand(fs, &inode, fnode->st.st_size))) {
 162                 errmsg = "datablock expand";
 163                 goto err_exit;
 164         }
 165
 166         /* Data copy */
 167         if ((fd = open(filepath, O_RDONLY)) == -1) {
 168                 error = errno;
 169                 errmsg = "source file";
 170                 goto err_exit;
 171         }
 172
 173         error = v7fs_datablock_foreach(fs, &inode, copy_subr,
 174             &(struct copy_arg){ .fd = fd });
 175         if (error != V7FS_ITERATOR_END) {
 176                 errmsg = "data copy";
 177                 close(fd);
 178                 goto err_exit;
 179         } else {
 180                 error = 0;
 181         }
 182         close(fd);
 183
 184         return error;
 185
 186 err_exit:
 187         errno = error;
 188         warn("%s %s", node->name, errmsg);
 189
 190         return error;
 191 }
 192
 193 static int
 194 populate_walk(struct v7fs_self *fs, struct v7fs_inode *parent_inode,
 195     fsnode *root, char *dir)
 196 {
 197         fsnode *cur;
 198         char *mydir = dir + strlen(dir);
 199         char srcpath[MAXPATHLEN + 1];
 200         struct v7fs_inode inode;
 201         int error = 0;
 202         bool failed = false;
 203
 204         for (cur = root; cur != NULL; cur = cur->next) {
 205                 switch (cur->type & S_IFMT) {
 206                 default:
 207                         VPRINTF("%x\n", cur->flags & S_IFMT);
 208                         break;
 209                 case S_IFCHR:
 210                         /*FALLTHROUGH*/
 211                 case S_IFBLK:
 212                         if ((error = allocate(fs, parent_inode, cur,
 213                             cur->inode->st.st_rdev, &inode))) {
 214                                 errno = error;
 215                                 warn("%s", cur->name);
 216                         }
 217                         break;
 218                 case S_IFDIR:
 219                         if (!cur->child)        /*'.'*/
 220                                 break;
 221                         /* Allocate this directory. */
 222                         if ((error = allocate(fs, parent_inode, cur, 0,
 223                             &inode))) {
 224                                 errno = error;
 225                                 warn("%s", cur->name);
 226                         } else {
 227                                 /* Populate children. */
 228                                 mydir[0] = '/';
 229                                 strcpy(&mydir[1], cur->name);
 230                                 error = populate_walk(fs, &inode, cur->child,
 231                                     dir);
 232                                 mydir[0] = '\0';
 233                         }
 234                         break;
 235                 case S_IFREG:
 236                         snprintf(srcpath, sizeof(srcpath), "%s/%s", dir,
 237                             cur->name);
 238                         error = file_copy(fs, parent_inode, cur, srcpath);
 239                         break;
 240                 case S_IFLNK:
 241                         if ((error = allocate(fs, parent_inode, cur, 0,
 242                             &inode))) {
 243                                 errno = error;
 244                                 warn("%s", cur->name);
 245                         } else {
 246                                 v7fs_file_symlink(fs, &inode, cur->symlink);
 247                         }
 248                         break;
 249                 }
 250                 if (error)
 251                         failed = true;
 252         }
 253
 254         return failed ? 2 : 0;
 255 }
 256
 257 int
 258 v7fs_populate(const char *dir, fsnode *root, fsinfo_t *fsopts,
 259     const struct v7fs_mount_device *device)
 260 {
 261         v7fs_opt_t *v7fs_opts = fsopts->fs_specific;
 262         static char path[MAXPATHLEN + 1];
 263         struct v7fs_inode root_inode;
 264         struct v7fs_self *fs;
 265         int error;
 266
 267         if ((error = v7fs_io_init(&fs, device, V7FS_BSIZE))) {
 268                 errno = error;
 269                 warn("I/O setup failed.");
 270                 return error;
 271         }
 272         fs->endian = device->endian;
 273         v7fs_endian_init(fs);
 274
 275         if ((error = v7fs_superblock_load(fs))) {
 276                 errno = error;
 277                 warn("Can't load superblock.");
 278                 return error;
 279         }
 280         fsopts->superblock = &fs->superblock;   /* not used. */
 281
 282         if ((error = v7fs_inode_load(fs, &root_inode, V7FS_ROOT_INODE))) {
 283                 errno = error;
 284                 warn("Can't load root inode.");
 285                 return error;
 286         }
 287
 288         progress(&(struct progress_arg){ .label = "populate", .tick =
 289             v7fs_opts->npuredatablk / PROGRESS_BAR_GRANULE });
 290
 291         strncpy(path, dir, sizeof(path));
 292         error = populate_walk(fs, &root_inode, root, path);
 293
 294         v7fs_inode_writeback(fs, &root_inode);
 295         v7fs_superblock_writeback(fs);
 296
 297         return error;
 298 }