From ca84da679a170fcb1b2ab59a59ec45e014530996 Mon Sep 17 00:00:00 2001 From: Luca BRUNO Date: Tue, 26 Oct 2021 12:12:48 +0000 Subject: prepare-root: check return codes for errors when assembling paths This adds checks around all `snprintf` calls in order to detect failures and gracefully abort. --- src/switchroot/ostree-prepare-root.c | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) (limited to 'src') diff --git a/src/switchroot/ostree-prepare-root.c b/src/switchroot/ostree-prepare-root.c index 20b8685d..eb1159f4 100644 --- a/src/switchroot/ostree-prepare-root.c +++ b/src/switchroot/ostree-prepare-root.c @@ -135,7 +135,8 @@ resolve_deploy_path (const char * root_mountpoint) if (!ostree_target) errx (EXIT_FAILURE, "No OSTree target; expected ostree=/ostree/boot.N/..."); - snprintf (destpath, sizeof(destpath), "%s/%s", root_mountpoint, ostree_target); + if (snprintf (destpath, sizeof(destpath), "%s/%s", root_mountpoint, ostree_target) < 0) + err (EXIT_FAILURE, "failed to assemble ostree target path"); if (lstat (destpath, &stbuf) < 0) err (EXIT_FAILURE, "Couldn't find specified OSTree root '%s'", destpath); if (!S_ISLNK (stbuf.st_mode)) @@ -287,12 +288,14 @@ main(int argc, char *argv[]) char srcpath[PATH_MAX]; /* If /boot is on the same partition, use a bind mount to make it visible * at /boot inside the deployment. */ - snprintf (srcpath, sizeof(srcpath), "%s/boot/loader", root_mountpoint); + if (snprintf (srcpath, sizeof(srcpath), "%s/boot/loader", root_mountpoint) < 0) + err (EXIT_FAILURE, "failed to assemble /boot/loader path"); if (lstat (srcpath, &stbuf) == 0 && S_ISLNK (stbuf.st_mode)) { if (lstat ("boot", &stbuf) == 0 && S_ISDIR (stbuf.st_mode)) { - snprintf (srcpath, sizeof(srcpath), "%s/boot", root_mountpoint); + if (snprintf (srcpath, sizeof(srcpath), "%s/boot", root_mountpoint) < 0) + err (EXIT_FAILURE, "failed to assemble /boot path"); if (mount (srcpath, "boot", NULL, MS_BIND | MS_SILENT, NULL) < 0) err (EXIT_FAILURE, "failed to bind mount %s to boot", srcpath); } -- cgit v1.2.1