From 2b84065a0f82e71e3f82fcd3148daa29f35a6597 Mon Sep 17 00:00:00 2001 From: David Oberhollenzer Date: Mon, 1 Nov 2021 12:02:02 +0100 Subject: Add a sort-file option to gensquashfs Signed-off-by: David Oberhollenzer --- bin/gensquashfs/mkfs.c | 16 +++++++++++++++- bin/gensquashfs/mkfs.h | 1 + bin/gensquashfs/options.c | 44 +++++++++++++++++++++++++++++++++++++++++++- 3 files changed, 59 insertions(+), 2 deletions(-) (limited to 'bin/gensquashfs') diff --git a/bin/gensquashfs/mkfs.c b/bin/gensquashfs/mkfs.c index 9e8e2d0..cb891f9 100644 --- a/bin/gensquashfs/mkfs.c +++ b/bin/gensquashfs/mkfs.c @@ -52,7 +52,7 @@ static int pack_files(sqfs_block_processor_t *data, fstree_t *fs, return -1; } - flags = 0; + flags = fi->flags; filesize = file->get_size(file); if (opt->no_tail_packing && filesize > opt->cfg.block_size) @@ -141,6 +141,7 @@ static void override_owner_dfs(const options_t *opt, tree_node_t *n) int main(int argc, char **argv) { int status = EXIT_FAILURE; + istream_t *sortfile = NULL; void *sehnd = NULL; sqfs_writer_t sqfs; options_t opt; @@ -156,6 +157,12 @@ int main(int argc, char **argv) goto out; } + if (opt.sortfile != NULL) { + sortfile = istream_open_file(opt.sortfile); + if (sortfile == NULL) + goto out; + } + if (opt.infile == NULL) { if (fstree_from_dir(&sqfs.fs, sqfs.fs.root, opt.packdir, NULL, NULL, opt.dirscan_flags)) { @@ -179,6 +186,11 @@ int main(int argc, char **argv) } } + if (sortfile != NULL) { + if (fstree_sort_files(&sqfs.fs, sortfile)) + goto out; + } + if (pack_files(sqfs.data, &sqfs.fs, &opt)) goto out; @@ -190,6 +202,8 @@ out: sqfs_writer_cleanup(&sqfs, status); if (sehnd != NULL) selinux_close_context_file(sehnd); + if (sortfile != NULL) + sqfs_destroy(sortfile); free(opt.packdir); return status; } diff --git a/bin/gensquashfs/mkfs.h b/bin/gensquashfs/mkfs.h index c43c10a..4f455bd 100644 --- a/bin/gensquashfs/mkfs.h +++ b/bin/gensquashfs/mkfs.h @@ -43,6 +43,7 @@ typedef struct { unsigned int dirscan_flags; const char *infile; const char *selinux; + const char *sortfile; bool no_tail_packing; /* copied from command line or constructed from infile argument diff --git a/bin/gensquashfs/options.c b/bin/gensquashfs/options.c index 14c37ab..fc2d900 100644 --- a/bin/gensquashfs/options.c +++ b/bin/gensquashfs/options.c @@ -35,12 +35,13 @@ static struct option long_opts[] = { #ifdef WITH_SELINUX { "selinux", required_argument, NULL, 's' }, #endif + { "sort-file", required_argument, NULL, 'S' }, { "version", no_argument, NULL, 'V' }, { "help", no_argument, NULL, 'h' }, { NULL, 0, NULL, 0 }, }; -static const char *short_opts = "F:D:X:c:b:B:d:u:g:j:Q:kxoefqThV" +static const char *short_opts = "F:D:X:c:b:B:d:u:g:j:Q:S:kxoefqThV" #ifdef WITH_SELINUX "s:" #endif @@ -98,6 +99,10 @@ static const char *help_string = " directory entries actually specify.\n" " --all-root A short hand for `--set-uid 0 --set-gid 0`.\n" "\n" +" --sort-file, -S Specify a \"sort file\" that can be used to\n" +" micro manage the order of files during packing\n" +" and behaviour (compression, fragmentation, ..)\n" +"\n" #ifdef WITH_SELINUX " --selinux, -s Specify an SELinux label file to get context\n" " attributes from.\n" @@ -169,6 +174,39 @@ const char *help_details = " glob /usr/lib 0777 0 0 -type l -name \"*.so.*\" ./lib\n" "\n\n"; +const char *sort_details = +"When using a sort file, the order in which regular files are packed can be\n" +"changed, as well as packing behaviour. Each line in the sort file consists\n" +"of a numeric priority (can be negative, lower priority is packed first),\n" +"followed by optional packing flags, followed by the file path.\n" +"\n" +"The fields are white-space separated. The flags are comma seperated and\n" +"contained within two square bracktes. Single line comments with `#` and\n" +"empty lines are allowed.\n" +"\n" +"The default priority for un-listed files is 0. The filename specifies the\n" +"path within the SquashFS image and is normalized. If e.g. spaces at the\n" +"beginning or end are needed, it can be enclosed in double quotes `\"`\n" +"and back-slash can be used for escaping.\n" +"\n" +"The sorting algorithm is stable, so files with the same priority do not\n" +"change place relative to each other in their initial ordering.\n" +"\n" +"Example:\n" +" # Specify a packing order with file globbing\n" +" -8000 [glob] bin/*\n" +" -5000 [glob] lib/*\n" +"\n" +" # glob_no_path means * is allowed to match /\n" +" -1000 [glob_no_path] share/*\n" +"\n" +" # Our boot loader needs this\n" +" -100000 [dont_compress,dont_fragment,nosparse] boot/vmlinuz\n" +"\n" +" # For demonstration, a quoted filename and no flags\n" +" 1337 \"usr/share/my \\\"special\\\" file \"\n" +"\n\n"; + void process_command_line(options_t *opt, int argc, char **argv) { bool have_compressor; @@ -285,10 +323,14 @@ void process_command_line(options_t *opt, int argc, char **argv) opt->selinux = optarg; break; #endif + case 'S': + opt->sortfile = optarg; + break; case 'h': printf(help_string, SQFS_DEFAULT_BLOCK_SIZE, SQFS_DEVBLK_SIZE); fputs(help_details, stdout); + fputs(sort_details, stdout); compressor_print_available(); exit(EXIT_SUCCESS); case 'V': -- cgit v1.2.3