hpc · reidpr · Jul 29, 2024 · Jul 9, 2022 · Feb 14, 2024 · Feb 14, 2024
diff --git a/bin/ch-checkns.c b/bin/ch-checkns.c
@@ -71,7 +71,7 @@ void fatal_(const char *file, int line, int errno_, const char *str)
    char *url = "https://github.com/hpc/charliecloud/blob/master/bin/ch-checkns.c";
    printf("error: %s: %d: %s\n", file, line, str);
    printf("errno: %d\nsee: %s\n", errno_, url);
-   exit(1);
+   exit(ERR_CHRUN);
 }
 
 int main(int argc, char *argv[])

diff --git a/bin/ch-completion.bash b/bin/ch-completion.bash
@@ -259,14 +259,16 @@ _ch_convert_complete () {
 _image_build_opts="-b --bind --build-arg -f --file --force
                    --force-cmd -n --dry-run --parse-only -t --tag"
 
+_image_modify_opts="-c -S --shell"
+
 _image_common_opts="-a --arch --always-download --auth --break
                     --cache --cache-large --dependencies -h --help
                     --no-cache --no-lock --no-xattrs --profile
                     --rebuild --password-many -q --quiet -s --storage
                     --tls-no-verify -v --verbose --version --xattrs"
 
-_image_subcommands="build build-cache delete gestalt
-                    import list pull push reset undelete"
+_image_subcommands="build build-cache delete gestalt import
+                    list modify pull push reset undelete"
 
 # archs taken from ARCH_MAP in charliecloud.py
 _archs="amd64 arm/v5 arm/v6 arm/v7 arm64/v8 386 mips64le ppc64le s390x"
@@ -374,20 +376,26 @@ _ch_image_complete () {
     build-cache)
         COMPREPLY=( $(compgen -W "--reset --gc --tree --dot" -- "$cur") )
         ;;
-    delete|list)
-        if [[ "$sub_cmd" == "list" ]]; then
+    delete|list|modify)
+        case "$sub_cmd" in
+        list)
             if [[ "$prev" == "--undeletable" || "$prev" == "--undeleteable" || "$prev" == "-u" ]]; then
                 COMPREPLY=( $(compgen -W "$(_ch_undelete_list "$strg_dir")" -- "$cur") )
                 return 0
             fi
-            extras+="$extras -l --long -u --undeletable"
+            extras="$extras -l --long -u --undeletable"
             # If “cur” starts with “--undelete,” add “--undeleteable” (the less
             # correct version of “--undeletable”) to the list of possible
             # completions.
             if [[ ${cur::10} == "--undelete" ]]; then
                 extras="$extras --undeleteable"
             fi
-        fi
+            ;;
+        modify)
+            # FIXME: Implement
+            extras="$extras $_image_modify_opts"
+            ;;
+        esac
         COMPREPLY=( $(compgen -W "$(_ch_list_images "$strg_dir") $extras" -- "$cur") )
         __ltrim_colon_completions "$cur"
         ;;

diff --git a/bin/ch-image.py.in b/bin/ch-image.py.in
@@ -13,6 +13,7 @@ import charliecloud as ch
 import build
 import build_cache as bu
 import filesystem as fs
+import image as im
 import misc
 import pull
 import push
@@ -278,6 +279,16 @@ def main():
    sp.add_argument("image_ref", metavar="IMAGE_REF", nargs="?",
                    help="print details of this image only")
 
+   # modify
+   sp = ap.add_parser("modify", "foo")
+   add_opts(sp, build.modify, deps_check=True, stog_init=True)
+   sp.add_argument("-c", metavar="CMD", action="append", default=[], nargs=1, help="foo")
+   sp.add_argument("-S", "--shell", metavar="shell", help="foo")
+   sp.add_argument("image_ref", metavar="IMAGE_REF", help="image to modify")
+   sp.add_argument("out_image", metavar="OUT_IMAGE", help="destination of modified image")
+   # Optional positional argument? https://stackoverflow.com/a/4480202
+   sp.add_argument("script", metavar="SCRIPT", help="foo", nargs='?')
+
    # pull
    sp = ap.add_parser("pull",
                       "copy image from remote repository to local filesystem")

diff --git a/bin/ch-run.c b/bin/ch-run.c
@@ -189,7 +189,7 @@ int main(int argc, char *argv[])
 
    if (arg_next >= argc - 1) {
       printf("usage: ch-run [OPTION...] IMAGE -- COMMAND [ARG...]\n");
-      FATAL("IMAGE and/or COMMAND not specified");
+      FATAL(0, "IMAGE and/or COMMAND not specified");
    }
    args.c.img_ref = argv[arg_next++];
    args.c.newroot = realpath_(args.c.newroot, true);
@@ -210,11 +210,11 @@ int main(int argc, char *argv[])
       break;
    case IMG_SQUASH:
 #ifndef HAVE_LIBSQUASHFUSE
-      FATAL("this ch-run does not support internal SquashFS mounts");
+      FATAL(0, "this ch-run does not support internal SquashFS mounts");
 #endif
       break;
    case IMG_NONE:
-      FATAL("unknown image type: %s", args.c.img_ref);
+      FATAL(0, "unknown image type: %s", args.c.img_ref);
       break;
    }
 
@@ -445,7 +445,7 @@ static error_t parse_opt(int key, char *arg, struct argp_state *state)
 #ifdef HAVE_FNM_EXTMATCH
          exit(0);
 #else
-         exit(1);
+         exit(ERR_CHRUN);
 #endif
       } else if (!strcmp(arg, "overlayfs")) {
 #ifdef HAVE_OVERLAYFS
@@ -457,13 +457,13 @@ static error_t parse_opt(int key, char *arg, struct argp_state *state)
 #ifdef HAVE_SECCOMP
          exit(0);
 #else
-         exit(1);
+         exit(ERR_CHRUN);
 #endif
       } else if (!strcmp(arg, "squash")) {
 #ifdef HAVE_LIBSQUASHFUSE
          exit(0);
 #else
-         exit(1);
+         exit(ERR_CHRUN);
 #endif
       } else if (!strcmp(arg, "tmpfs-xattrs")) {
 #ifdef HAVE_TMPFS_XATTRS
@@ -473,7 +473,7 @@ static error_t parse_opt(int key, char *arg, struct argp_state *state)
 #endif
       }
       else
-         FATAL("unknown feature: %s", arg);
+         FATAL(0, "unknown feature: %s", arg);
       break;
    case -12: // --home
       Tf (args->c.host_home = getenv("HOME"), "--home failed: $HOME not set");
@@ -504,7 +504,7 @@ static error_t parse_opt(int key, char *arg, struct argp_state *state)
       else if (!strcmp(arg, "log-fail"))
          test_logging(true);
       else
-         FATAL("invalid --test argument: %s; see source code", arg);
+         FATAL(0, "invalid --test argument: %s; see source code", arg);
       break;
    case 'b': {  // --bind
          char *src, *dst;

diff --git a/bin/ch_core.c b/bin/ch_core.c
@@ -235,7 +235,7 @@ void bind_mount(const char *src, const char *dst, enum bind_dep dep,
    if (!path_exists(dst_full, NULL, true))
       switch (dep) {
       case BD_REQUIRED:
-         FATAL("can't bind: destination not found: %s", dst_full);
+         FATAL(0, "can't bind: destination not found: %s", dst_full);
          break;
       case BD_OPTIONAL:
          return;
@@ -421,7 +421,7 @@ enum img_type image_type(const char *ref, const char *storage_dir)
       return IMG_SQUASH;
 
    // Well now we’re stumped.
-   FATAL("unknown image type: %s", ref);
+   FATAL(0, "unknown image type: %s", ref);
 }
 
 char *img_name2path(const char *name, const char *storage_dir)
@@ -567,7 +567,8 @@ void run_user_command(char *argv[], const char *initial_dir)
    if (verbose < LL_STDERR)
       T_ (freopen("/dev/null", "w", stderr));
    execvp(argv[0], argv);  // only returns if error
-   Tf (0, "can't execve(2): %s", argv[0]);
+   ERROR(errno, "can't execve(2): %s", argv[0]);
+   exit(ERR_CMD);
 }
 
 /* Set up the fake-syscall seccomp(2) filter. This computes and installs a

diff --git a/bin/ch_fuse.c b/bin/ch_fuse.c
@@ -188,7 +188,7 @@ int sq_loop(void)
 
    // Clean up zombie child if exit signal was SIGCHLD.
    if (!sigchld_received)
-      exit_code = 0;
+      exit_code = ERR_SQUASH;
    else {
       Tf (wait(&child_status) >= 0, "can't wait for child");
       if (WIFEXITED(child_status)) {
@@ -203,7 +203,7 @@ int sq_loop(void)
          //
          // [1]: https://codereview.stackexchange.com/a/109349
          // [2]: https://man7.org/linux/man-pages/man2/wait.2.html
-         exit_code = 1;
+         exit_code = 128 + WTERMSIG(child_status);
          VERBOSE("child terminated by signal %d", WTERMSIG(child_status))
       }
    }
@@ -254,7 +254,7 @@ void sq_mount(const char *img_path, char *mountpt)
                                    &OPS, sizeof(OPS), sq.ll)) {
          break;  // success
       } else if (i <= 0) {
-         FATAL("too many FUSE errors; giving up");
+         FATAL(0, "too many FUSE errors; giving up");
       } else {
          WARNING("FUSE error mounting SquashFS; will retry");
          sleep(1);

diff --git a/bin/ch_misc.c b/bin/ch_misc.c
@@ -449,7 +449,7 @@ void test_logging(bool fail) {
    INFO("info");
    WARNING("warning");
    if (fail)
-      FATAL("the program failed inexplicably (\"log-fail\" specified)");
+      FATAL(0, "the program failed inexplicably (\"log-fail\" specified)");
    exit(0);
 }
 
@@ -586,6 +586,16 @@ void msg(enum log_level level, const char *file, int line, int errno_,
    va_end(ap);
 }
 
+void msg_error(const char *file, int line, int errno_,
+               const char *fmt, ...)
+{
+   va_list ap;
+
+   va_start(ap, fmt);
+   msgv(LL_FATAL, file, line, errno_, fmt, ap);
+   va_end(ap);
+}
+
 noreturn void msg_fatal(const char *file, int line, int errno_,
                        const char *fmt, ...)
 {
@@ -595,7 +605,7 @@ noreturn void msg_fatal(const char *file, int line, int errno_,
    msgv(LL_FATAL, file, line, errno_, fmt, ap);
    va_end(ap);
 
-   exit(EXIT_FAILURE);
+   exit(ERR_CHRUN);
 }
 
 /* va_list form of msg(). */

diff --git a/bin/ch_misc.h b/bin/ch_misc.h
@@ -24,6 +24,11 @@
    don’t need to worry about running out of room. */
 #define WARNINGS_SIZE (4*1024)
 
+/* Exit codes (see also: test/common.bash, lib/build.py). */
+#define ERR_CHRUN 31
+#define ERR_CMD 49
+#define ERR_SQUASH 84
+
 /* Test some value, and if it's not what we expect, exit with a fatal error.
    These are macros so we have access to the file and line number.
 
@@ -66,12 +71,13 @@
 #define Zf(x, ...) if (x)    msg_fatal(__FILE__, __LINE__, errno, __VA_ARGS__)
 #define Ze(x, ...) if (x)    msg_fatal(__FILE__, __LINE__, 0, __VA_ARGS__)
 
-#define FATAL(...)   msg_fatal(      __FILE__, __LINE__, 0, __VA_ARGS__);
-#define WARNING(...) msg(LL_WARNING, __FILE__, __LINE__, 0, __VA_ARGS__);
-#define INFO(...)    msg(LL_INFO,    __FILE__, __LINE__, 0, __VA_ARGS__);
-#define VERBOSE(...) msg(LL_VERBOSE, __FILE__, __LINE__, 0, __VA_ARGS__);
-#define DEBUG(...)   msg(LL_DEBUG,   __FILE__, __LINE__, 0, __VA_ARGS__);
-#define TRACE(...)   msg(LL_TRACE,   __FILE__, __LINE__, 0, __VA_ARGS__);
+#define FATAL(e, ...) msg_fatal(      __FILE__, __LINE__, e, __VA_ARGS__);
+#define ERROR(e, ...) msg_error(      __FILE__, __LINE__, e, __VA_ARGS__);
+#define WARNING(...)  msg(LL_WARNING, __FILE__, __LINE__, 0, __VA_ARGS__);
+#define INFO(...)     msg(LL_INFO,    __FILE__, __LINE__, 0, __VA_ARGS__);
+#define VERBOSE(...)  msg(LL_VERBOSE, __FILE__, __LINE__, 0, __VA_ARGS__);
+#define DEBUG(...)    msg(LL_DEBUG,   __FILE__, __LINE__, 0, __VA_ARGS__);
+#define TRACE(...)    msg(LL_TRACE,   __FILE__, __LINE__, 0, __VA_ARGS__);
 
 
 /** Types **/
@@ -134,6 +140,8 @@ void mkdirs(const char *base, const char *path, char **denylist,
             const char *scratch);
 void msg(enum log_level level, const char *file, int line, int errno_,
          const char *fmt, ...);
+void msg_error(const char *file, int line, int errno_,
+               const char *fmt, ...);
 noreturn void msg_fatal(const char *file, int line, int errno_,
                         const char *fmt, ...);
 bool path_exists(const char *path, struct stat *statbuf, bool follow_symlink);