]> git.ipfire.org Git - thirdparty/util-linux.git/blobdiff - disk-utils/mkfs.cramfs.c
Make the ways of using output stream consistent in usage()
[thirdparty/util-linux.git] / disk-utils / mkfs.cramfs.c
index bf7f00838745118f9c10627c029f53d9937c06f2..482375351953f55b7b76c03c919c0a025c2949b8 100644 (file)
 
 /* We don't use our include/crc32.h, but crc32 from zlib!
  *
- * The zlib implemenation performs pre/post-conditioning. The util-linux
+ * The zlib implementation performs pre/post-conditioning. The util-linux
  * imlemenation requires post-conditioning (xor) in the applications.
  */
 #include <zlib.h>
 
+#include "blkdev.h"
 #include "c.h"
 #include "cramfs.h"
 #include "md5.h"
@@ -98,7 +99,7 @@ struct entry {
        /* stats */
        unsigned char *name;
        unsigned int mode, size, uid, gid;
-       unsigned char md5sum[MD5LENGTH];
+       unsigned char md5sum[UL_MD5LENGTH];
        unsigned char flags;       /* CRAMFS_EFLAG_* */
 
        /* FS data */
@@ -122,32 +123,31 @@ struct entry {
 #define CRAMFS_GID_WIDTH 8
 #define CRAMFS_OFFSET_WIDTH 26
 
-/* Input status of 0 to print help and exit without an error. */
-static void
-usage(int status) {
-       FILE *stream = status ? stderr : stdout;
-
-       fprintf(stream,
-               _("usage: %s [-h] [-v] [-b blksize] [-e edition] [-N endian] [-i file] "
-                 "[-n name] dirname outfile\n"
-                 " -h         print this help\n"
-                 " -v         be verbose\n"
-                 " -E         make all warnings errors "
-                   "(non-zero exit status)\n"
-                 " -b blksize use this blocksize, must equal page size\n"
-                 " -e edition set edition number (part of fsid)\n"
-                 " -N endian  set cramfs endianness (big|little|host), default host\n"
-                 " -i file    insert a file image into the filesystem "
-                   "(requires >= 2.4.0)\n"
-                 " -n name    set name of cramfs filesystem\n"
-                 " -p         pad by %d bytes for boot code\n"
-                 " -s         sort directory entries (old option, ignored)\n"
-                 " -z         make explicit holes (requires >= 2.3.39)\n"
-                 " dirname    root of the filesystem to be compressed\n"
-                 " outfile    output file\n"),
-               program_invocation_short_name, PAD_SIZE);
-
-       exit(status);
+static void __attribute__((__noreturn__)) usage(void)
+{
+       fputs(USAGE_HEADER, stdout);
+       fprintf(stdout, _(" %s [-h] [-v] [-b blksize] [-e edition] [-N endian] [-i file] [-n name] dirname outfile\n"),
+               program_invocation_short_name);
+       fputs(USAGE_SEPARATOR, stdout);
+       fputs(_("Make compressed ROM file system."), stdout);
+       fputs(USAGE_OPTIONS, stdout);
+       fputs(_(  " -v             be verbose"), stdout);
+       fputs(_(  " -E             make all warnings errors (non-zero exit status)"), stdout);
+       fputs(_(  " -b blksize     use this blocksize, must equal page size"), stdout);
+       fputs(_(  " -e edition     set edition number (part of fsid)"), stdout);
+       fprintf(stdout, _(" -N endian      set cramfs endianness (%s|%s|%s), default %s\n"), "big", "little", "host", "host");
+       fputs(_(  " -i file        insert a file image into the filesystem"), stdout);
+       fputs(_(  " -n name        set name of cramfs filesystem"), stdout);
+       fprintf(stdout, _(" -p             pad by %d bytes for boot code\n"), PAD_SIZE);
+       fputs(_(  " -s             sort directory entries (old option, ignored)"), stdout);
+       fputs(_(  " -z             make explicit holes"), stdout);
+       fputs(_(  " -l[=<mode>]    use exclusive device lock (yes, no or nonblock)"), stdout);
+       fputs(_(  " dirname        root of the filesystem to be compressed"), stdout);
+       fputs(_(  " outfile        output file"), stdout);
+       fputs(USAGE_SEPARATOR, stdout);
+       fprintf(stdout, USAGE_HELP_OPTIONS(16));
+       fprintf(stdout, USAGE_MAN_TAIL("mkfs.cramfs(8)"));
+       exit(MKFS_EX_OK);
 }
 
 static char *
@@ -159,6 +159,8 @@ do_mmap(char *path, unsigned int size, unsigned int mode){
                return NULL;
 
        if (S_ISLNK(mode)) {
+               /* The link buffer is unnecessary to terminate by null as it's
+                * always used as buffer rather than a string */
                start = xmalloc(size);
                if (readlink(path, start, size) < 0) {
                        warn(_("readlink failed: %s"), path);
@@ -196,16 +198,17 @@ do_munmap(char *start, unsigned int size, unsigned int mode){
 /* compute md5sums, so that we do not have to compare every pair of files */
 static void
 mdfile(struct entry *e) {
-       MD5_CTX ctx;
        char *start;
 
        start = do_mmap(e->path, e->size, e->mode);
        if (start == NULL) {
                e->flags |= CRAMFS_EFLAG_INVALID;
        } else {
-               MD5Init(&ctx);
-               MD5Update(&ctx, (unsigned char *) start, e->size);
-               MD5Final(e->md5sum, &ctx);
+               UL_MD5_CTX ctx;
+
+               ul_MD5Init(&ctx);
+               ul_MD5Update(&ctx, (unsigned char *) start, e->size);
+               ul_MD5Final(e->md5sum, &ctx);
 
                do_munmap(start, e->size, e->mode);
 
@@ -257,7 +260,7 @@ static int find_identical_file(struct entry *orig, struct entry *new, loff_t *fs
 
                if ((orig->flags & CRAMFS_EFLAG_MD5) &&
                    (new->flags & CRAMFS_EFLAG_MD5) &&
-                   !memcmp(orig->md5sum, new->md5sum, MD5LENGTH) &&
+                   !memcmp(orig->md5sum, new->md5sum, UL_MD5LENGTH) &&
                    identical_file(orig, new)) {
                        new->same = orig;
                        *fslen_ub -= new->size;
@@ -302,7 +305,7 @@ static unsigned int parse_directory(struct entry *root_entry, const char *name,
        endpath++;
 
        /* read in the directory and sort */
-       dircount = scandir(name, &dirlist, 0, cramsort);
+       dircount = scandir(name, &dirlist, NULL, cramsort);
 
        if (dircount < 0)
                err(MKFS_EX_ERROR, _("could not read directory %s"), name);
@@ -360,7 +363,7 @@ static unsigned int parse_directory(struct entry *root_entry, const char *name,
                        entry->size = parse_directory(root_entry, path, &entry->child, fslen_ub);
                } else if (S_ISREG(st.st_mode)) {
                        entry->path = xstrdup(path);
-                       if (entry->size && entry->size >= (1 << CRAMFS_SIZE_WIDTH)) {
+                       if (entry->size >= (1 << CRAMFS_SIZE_WIDTH)) {
                                warn_size = 1;
                                entry->size = (1 << CRAMFS_SIZE_WIDTH) - 1;
                        }
@@ -412,16 +415,16 @@ static unsigned int write_superblock(struct entry *root, char *base, int size)
        super->size = size;
        memcpy(super->signature, CRAMFS_SIGNATURE, sizeof(super->signature));
 
-       super->fsid.crc = crc32(0L, Z_NULL, 0);
+       super->fsid.crc = crc32(0L, NULL, 0);
        super->fsid.edition = opt_edition;
        super->fsid.blocks = total_blocks;
        super->fsid.files = total_nodes;
 
        memset(super->name, 0x00, sizeof(super->name));
        if (opt_name)
-               strncpy((char *)super->name, opt_name, sizeof(super->name));
+               str2memcpy((char *)super->name, opt_name, sizeof(super->name));
        else
-               strncpy((char *)super->name, "Compressed", sizeof(super->name));
+               str2memcpy((char *)super->name, "Compressed", sizeof(super->name));
 
        super->root.mode = root->mode;
        super->root.uid = root->uid;
@@ -492,7 +495,7 @@ static unsigned int write_directory_structure(struct entry *entry, char *base, u
                        if (entry->child) {
                                if (stack_entries >= stack_size) {
                                        stack_size *= 2;
-                                       entry_stack = xrealloc(entry_stack, stack_size * sizeof(struct entry *));
+                                       entry_stack = xreallocarray(entry_stack, stack_size, sizeof(struct entry *));
                                }
                                entry_stack[stack_entries] = entry;
                                stack_entries++;
@@ -548,9 +551,9 @@ static int is_zero(unsigned char const *begin, unsigned len)
                             (len-- == 0 ||
                              (begin[3] == '\0' &&
                               memcmp(begin, begin + 4, len) == 0))))))));
-       else
-               /* Never create holes. */
-               return 0;
+
+       /* Never create holes. */
+       return 0;
 }
 
 /*
@@ -706,7 +709,8 @@ int main(int argc, char **argv)
        loff_t fslen_ub = sizeof(struct cramfs_super);
        unsigned int fslen_max;
        char const *dirname, *outfile;
-       uint32_t crc = crc32(0L, Z_NULL, 0);
+       char *lockmode = 0;
+       uint32_t crc = crc32(0L, NULL, 0);
        int c;
        cramfs_is_big_endian = HOST_IS_BIG_ENDIAN; /* default is to use host order */
 
@@ -715,13 +719,24 @@ int main(int argc, char **argv)
        setlocale(LC_ALL, "");
        bindtextdomain(PACKAGE, LOCALEDIR);
        textdomain(PACKAGE);
-       atexit(close_stdout);
+       close_stdout_atexit();
+
+       if (argc > 1) {
+               /* first arg may be one of our standard longopts */
+               if (!strcmp(argv[1], "--help"))
+                       usage();
+               if (!strcmp(argv[1], "--version")) {
+                       print_version(EXIT_SUCCESS);
+                       exit(MKFS_EX_OK);
+               }
+       }
+       strutils_set_exitcode(MKFS_EX_USAGE);
 
        /* command line options */
-       while ((c = getopt(argc, argv, "hb:Ee:i:n:N:psVvz")) != EOF) {
+       while ((c = getopt(argc, argv, "hb:Ee:i:n:N:l::psVvz")) != EOF) {
                switch (c) {
                case 'h':
-                       usage(MKFS_EX_OK);
+                       usage();
                case 'b':
                        blksize = strtou32_or_err(optarg, _("invalid blocksize argument"));
                        break;
@@ -749,6 +764,14 @@ int main(int argc, char **argv)
                        image_length = st.st_size; /* may be padded later */
                        fslen_ub += (image_length + 3); /* 3 is for padding */
                        break;
+               case 'l':
+                        lockmode = "1";
+                       if (optarg) {
+                                if (*optarg == '=')
+                                       optarg++;
+                               lockmode = optarg;
+                       }
+                       break;
                case 'n':
                        opt_name = optarg;
                        break;
@@ -760,8 +783,7 @@ int main(int argc, char **argv)
                        /* old option, ignored */
                        break;
                case 'V':
-                       printf(UTIL_LINUX_VERSION);
-                       exit(MKFS_EX_OK);
+                       print_version(MKFS_EX_OK);
                case 'v':
                        verbose = 1;
                        break;
@@ -769,12 +791,14 @@ int main(int argc, char **argv)
                        opt_holes = 1;
                        break;
                default:
-                       usage(FSCK_EX_USAGE);
+                       errtryhelp(MKFS_EX_USAGE);
                }
        }
 
-       if ((argc - optind) != 2)
-               usage(MKFS_EX_USAGE);
+       if ((argc - optind) != 2) {
+               warnx(_("bad usage"));
+               errtryhelp(MKFS_EX_USAGE);
+       }
        dirname = argv[optind];
        outfile = argv[optind + 1];
 
@@ -787,6 +811,9 @@ int main(int argc, char **argv)
        if (fd < 0)
                err(MKFS_EX_USAGE, _("cannot open %s"), outfile);
 
+        if (blkdev_lock(fd, outfile, lockmode) != 0)
+               exit(MKFS_EX_ERROR);
+
        root_entry = xcalloc(1, sizeof(struct entry));
        root_entry->mode = st.st_mode;
        root_entry->uid = st.st_uid;
@@ -910,5 +937,5 @@ int main(int argc, char **argv)
            (warn_namelen|warn_skip|warn_size|warn_uid|warn_gid|warn_dev))
                exit(MKFS_EX_ERROR);
 
-       return EXIT_SUCCESS;
+       return MKFS_EX_OK;
 }