/*      $nEtBSD: makefs.h,v 1.38 2022/04/09 10:05:35 riastradh Exp $    */

/*
* Copyright (c) 2001 Wasabi Systems, Inc.
* All rights reserved.
*
* Written by Luke Mewburn for Wasabi Systems, Inc.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions
* are met:
* 1. Redistributions of source code must retain the above copyright
*    notice, this list of conditions and the following disclaimer.
* 2. Redistributions in binary form must reproduce the above copyright
*    notice, this list of conditions and the following disclaimer in the
*    documentation and/or other materials provided with the distribution.
* 3. All advertising materials mentioning features or use of this software
*    must display the following acknowledgement:
*      This product includes software developed for the NetBSD Project by
*      Wasabi Systems, Inc.
* 4. The name of Wasabi Systems, Inc. may not be used to endorse
*    or promote products derived from this software without specific prior
*    written permission.
*
* THIS SOFTWARE IS PROVIDED BY WASABI SYSTEMS, INC. ``AS IS'' AND
* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
* TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
* PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL WASABI SYSTEMS, INC
* BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
* CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
* SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
* INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
* CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
* ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
* POSSIBILITY OF SUCH DAMAGE.
*/

#ifndef _MAKEFS_H
#define _MAKEFS_H

#if HAVE_NBTOOL_CONFIG_H
#include "nbtool_config.h"
#else
#define HAVE_STRUCT_STAT_ST_FLAGS 1
#define HAVE_STRUCT_STAT_ST_GEN 1
#define HAVE_STRUCT_STAT_ST_MTIMENSEC 1
#define HAVE_STRUCT_STATVFS_F_IOSIZE 1
#define HAVE_STRUCT_STAT_BIRTHTIME 1
#define HAVE_FSTATVFS 1
#endif

#include <stdio.h>
#include <sys/stat.h>
#include <err.h>

/*
* fsnode -
*      a component of the tree; contains a filename, a pointer to
*      fsinode, optional symlink name, and tree pointers
*
* fsinode -
*      equivalent to an inode, containing target file system inode number,
*      refcount (nlink), and stat buffer
*
* A tree of fsnodes looks like this:
*
*      name    "."             "bin"           "netbsd"
*      type    S_IFDIR         S_IFDIR         S_IFREG
*      next      >               >             NULL
*      parent  NULL            NULL            NULL
*      child   NULL              v
*
*      name                    "."             "ls"
*      type                    S_IFDIR         S_IFREG
*      next                      >             NULL
*      parent                    ^             ^ (to "bin")
*      child                   NULL            NULL
*
* Notes:
*      -   first always points to first entry, at current level, which
*          must be "." when the tree has been built; during build it may
*          not be if "." hasn't yet been found by readdir(2).
*/

enum fi_flags {
       FI_SIZED =      1<<0,           /* inode sized */
       FI_ALLOCATED =  1<<1,           /* fsinode->ino allocated */
       FI_WRITTEN =    1<<2,           /* inode written */
};

typedef struct {
       uint64_t         ino;           /* inode number used on target fs */
       uint32_t         nlink;         /* number of links to this entry */
       enum fi_flags    flags;         /* flags used by fs specific code */
       struct stat      st;            /* stat entry */
       void            *fsuse;         /* for storing FS dependent info */
#if !HAVE_STRUCT_STAT_ST_FLAGS
       uint32_t         st_flags;      /* stand-in for st.st_flags */
#endif
} fsinode;

#if HAVE_STRUCT_STAT_ST_FLAGS
#define FSINODE_ST_FLAGS(inode) (inode).st.st_flags
#else
#define FSINODE_ST_FLAGS(inode) (inode).st_flags
#endif

typedef struct _fsnode {
       struct _fsnode  *parent;        /* parent (NULL if root) */
       struct _fsnode  *child;         /* child (if type == S_IFDIR) */
       struct _fsnode  *next;          /* next */
       struct _fsnode  *first;         /* first node of current level (".") */
       uint32_t         type;          /* type of entry */
       fsinode         *inode;         /* actual inode data */
       char            *symlink;       /* symlink target */
       const char      *root;          /* root path */
       char            *path;          /* directory name */
       char            *name;          /* file name */
       int             flags;          /* misc flags */
} fsnode;

#define FSNODE_F_HASSPEC        0x01    /* fsnode has a spec entry */

/*
* option_t - contains option name, description, pointer to location to store
* result, and range checks for the result. Used to simplify fs specific
* option setting
*/
typedef enum {
       OPT_STRARRAY,
       OPT_STRPTR,
       OPT_STRBUF,
       OPT_BOOL,
       OPT_INT8,
       OPT_INT16,
       OPT_INT32,
       OPT_INT64
} opttype_t;

typedef struct {
       char            letter;         /* option letter NUL for none */
       const char      *name;          /* option name */
       void            *value;         /* where to stuff the value */
       opttype_t       type;           /* type of entry */
       long long       minimum;        /* minimum for value */
       long long       maximum;        /* maximum for value */
       const char      *desc;          /* option description */
} option_t;

/*
* fsinfo_t - contains various settings and parameters pertaining to
* the image, including current settings, global options, and fs
* specific options
*/
typedef struct makefs_fsinfo {
               /* current settings */
       off_t   size;           /* total size */
       off_t   inodes;         /* number of inodes */
       uint32_t curinode;      /* current inode */

               /* image settings */
       int     fd;             /* file descriptor of image */
       void    *superblock;    /* superblock */
       int     onlyspec;       /* only add entries in specfile */


               /* global options */
       off_t   minsize;        /* minimum size image should be */
       off_t   maxsize;        /* maximum size image can be */
       off_t   freefiles;      /* free file entries to leave */
       off_t   freeblocks;     /* free blocks to leave */
       off_t   offset;         /* offset from start of file */
       int     freefilepc;     /* free file % */
       int     freeblockpc;    /* free block % */
       int     needswap;       /* non-zero if byte swapping needed */
       int     sectorsize;     /* sector size */
       int     sparse;         /* sparse image, don't fill it with zeros */
       int     replace;        /* replace files when merging */
       int     follow;         /* follow symlinks */

       void    *fs_specific;   /* File system specific additions. */
       option_t *fs_options;   /* File system specific options */
} fsinfo_t;




void            apply_specfile(const char *, const char *, fsnode *, int);
void            dump_fsnodes(fsnode *);
const char *    inode_type(mode_t);
int             set_option(const option_t *, const char *, char *, size_t);
void            print_options(FILE *, const option_t *);
int             set_option_var(const option_t *, const char *, const char *,
   char *, size_t);
fsnode *        walk_dir(const char *, const char *, fsnode *, fsnode *, int,
   int);
void            free_fsnodes(fsnode *);
option_t *      copy_opts(const option_t *);

#define DECLARE_FUN(fs)                                                 \
void            fs ## _prep_opts(fsinfo_t *);                           \
int             fs ## _parse_opts(const char *, fsinfo_t *);            \
void            fs ## _cleanup_opts(fsinfo_t *);                        \
void            fs ## _makefs(const char *, const char *, fsnode *, fsinfo_t *)

DECLARE_FUN(ffs);
DECLARE_FUN(cd9660);
DECLARE_FUN(chfs);
DECLARE_FUN(v7fs);
DECLARE_FUN(msdos);
DECLARE_FUN(udf);

extern  u_int           debug;
extern  struct timespec start_time;
extern  struct stat stampst;

/*
* If -x is specified, we want to exclude nodes which do not appear
* in the spec file.
*/
#define FSNODE_EXCLUDE_P(opts, fsnode)  \
       ((opts)->onlyspec != 0 && ((fsnode)->flags & FSNODE_F_HASSPEC) == 0)

#define DEBUG_TIME                      0x00000001
               /* debug bits 1..3 unused at this time */
#define DEBUG_WALK_DIR                  0x00000010
#define DEBUG_WALK_DIR_NODE             0x00000020
#define DEBUG_WALK_DIR_LINKCHECK        0x00000040
#define DEBUG_DUMP_FSNODES              0x00000080
#define DEBUG_DUMP_FSNODES_VERBOSE      0x00000100
#define DEBUG_FS_PARSE_OPTS             0x00000200
#define DEBUG_FS_MAKEFS                 0x00000400
#define DEBUG_FS_VALIDATE               0x00000800
#define DEBUG_FS_CREATE_IMAGE           0x00001000
#define DEBUG_FS_SIZE_DIR               0x00002000
#define DEBUG_FS_SIZE_DIR_NODE          0x00004000
#define DEBUG_FS_SIZE_DIR_ADD_DIRENT    0x00008000
#define DEBUG_FS_POPULATE               0x00010000
#define DEBUG_FS_POPULATE_DIRBUF        0x00020000
#define DEBUG_FS_POPULATE_NODE          0x00040000
#define DEBUG_FS_WRITE_FILE             0x00080000
#define DEBUG_FS_WRITE_FILE_BLOCK       0x00100000
#define DEBUG_FS_MAKE_DIRBUF            0x00200000
#define DEBUG_FS_WRITE_INODE            0x00400000
#define DEBUG_BUF_BREAD                 0x00800000
#define DEBUG_BUF_BWRITE                0x01000000
#define DEBUG_BUF_GETBLK                0x02000000
#define DEBUG_APPLY_SPECFILE            0x04000000
#define DEBUG_APPLY_SPECENTRY           0x08000000
#define DEBUG_APPLY_SPECONLY            0x10000000

#define DEBUG_STRINGS \
       { "time",       DEBUG_TIME }, \
       { "walk_dir",   DEBUG_WALK_DIR }, \
       { "walk_dir_node",      DEBUG_WALK_DIR_NODE }, \
       { "walk_dir_linkcheck", DEBUG_WALK_DIR_LINKCHECK }, \
       { "dump_fsnodes",       DEBUG_DUMP_FSNODES }, \
       { "dump_fsnodes_verbose",       DEBUG_DUMP_FSNODES_VERBOSE }, \
       { "fs_parse_opts",      DEBUG_FS_PARSE_OPTS }, \
       { "fs_makefs",  DEBUG_FS_MAKEFS }, \
       { "fs_validate",        DEBUG_FS_VALIDATE }, \
       { "fs_create_image",    DEBUG_FS_CREATE_IMAGE }, \
       { "fs_size_dir",        DEBUG_FS_SIZE_DIR }, \
       { "fs_size_dir_node",   DEBUG_FS_SIZE_DIR_NODE }, \
       { "fs_size_dir_add_dirent",     DEBUG_FS_SIZE_DIR_ADD_DIRENT }, \
       { "fs_populate",        DEBUG_FS_POPULATE }, \
       { "fs_populate_dirbuf", DEBUG_FS_POPULATE_DIRBUF }, \
       { "fs_populate_node",   DEBUG_FS_POPULATE_NODE }, \
       { "fs_write_file",      DEBUG_FS_WRITE_FILE }, \
       { "fs_write_file_block",        DEBUG_FS_WRITE_FILE_BLOCK }, \
       { "fs_make_dirbuf",     DEBUG_FS_MAKE_DIRBUF }, \
       { "fs_write_inode",     DEBUG_FS_WRITE_INODE }, \
       { "buf_bread",  DEBUG_BUF_BREAD }, \
       { "buf_bwrite", DEBUG_BUF_BWRITE }, \
       { "buf_getblk", DEBUG_BUF_GETBLK }, \
       { "apply_specfile",     DEBUG_APPLY_SPECFILE }, \
       { "apply_specentry",    DEBUG_APPLY_SPECENTRY }, \
       { "apply_speconly",     DEBUG_APPLY_SPECONLY },

#define TIMER_START(x)                          \
       if (debug & DEBUG_TIME)                 \
               gettimeofday(&(x), NULL)

#define TIMER_RESULTS(x,d)                              \
       if (debug & DEBUG_TIME) {                       \
               struct timeval end, td;                 \
               gettimeofday(&end, NULL);               \
               timersub(&end, &(x), &td);              \
               printf("%s took %lld.%06ld seconds\n",  \
                   (d), (long long)td.tv_sec,          \
                   (long)td.tv_usec);                  \
       }


#ifndef DEFAULT_FSTYPE
#define DEFAULT_FSTYPE  "ffs"
#endif


/*
*      ffs specific settings
*      ---------------------
*/

#define FFS_EI          /* for opposite endian support in ffs headers */


#endif  /* _MAKEFS_H */