From 60b279ce48bb210b59eb1a0e9bb4eb2c73f2825f Mon Sep 17 00:00:00 2001 From: Colin Walters Date: Wed, 7 Jan 2015 16:12:55 -0500 Subject: [PATCH] compose: Move the passwd/group migration code to passwd-util Pure code motion; no functional changes. Trying to get all of the passwd/group code in the same place so I can fix bugs in the interaction between them more easily. --- src/rpmostree-passwd-util.c | 159 ++++++++++++++++++++++++++++++++ src/rpmostree-passwd-util.h | 12 +++ src/rpmostree-postprocess.c | 174 ++---------------------------------- 3 files changed, 177 insertions(+), 168 deletions(-) diff --git a/src/rpmostree-passwd-util.c b/src/rpmostree-passwd-util.c index 0d299ffc..74561f83 100644 --- a/src/rpmostree-passwd-util.c +++ b/src/rpmostree-passwd-util.c @@ -668,6 +668,165 @@ rpmostree_check_groups (OstreeRepo *repo, treedata, cancellable, error); } +static FILE * +gfopen (const char *path, + const char *mode, + GCancellable *cancellable, + GError **error) +{ + FILE *ret = NULL; + + ret = fopen (path, mode); + if (!ret) + { + _rpmostree_set_prefix_error_from_errno (error, errno, "fopen(%s): ", path); + return NULL; + } + return ret; +} + +static gboolean +gfflush (FILE *f, + GCancellable *cancellable, + GError **error) +{ + if (fflush (f) != 0) + { + _rpmostree_set_prefix_error_from_errno (error, errno, "fflush: "); + return FALSE; + } + return TRUE; +} + +/* + * This function is taking the /etc/passwd generated in the install + * root, and splitting it into two streams: a new /etc/passwd that + * just contains the root entry, and /usr/lib/passwd which contains + * everything else. + * + * The implementation is kind of horrible because I wanted to avoid + * duplicating the user/group code. + */ +gboolean +rpmostree_passwd_migrate_except_root (GFile *rootfs, + RpmOstreePasswdMigrateKind kind, + GHashTable *preserve, + GCancellable *cancellable, + GError **error) +{ + gboolean ret = FALSE; + const char *name = kind == RPM_OSTREE_PASSWD_MIGRATE_PASSWD ? "passwd" : "group"; + gs_free char *src_path = g_strconcat (gs_file_get_path_cached (rootfs), "/etc/", name, NULL); + gs_free char *etctmp_path = g_strconcat (gs_file_get_path_cached (rootfs), "/etc/", name, ".tmp", NULL); + gs_free char *usrdest_path = g_strconcat (gs_file_get_path_cached (rootfs), "/usr/lib/", name, NULL); + FILE *src_stream = NULL; + FILE *etcdest_stream = NULL; + FILE *usrdest_stream = NULL; + + src_stream = gfopen (src_path, "r", cancellable, error); + if (!src_stream) + goto out; + + etcdest_stream = gfopen (etctmp_path, "w", cancellable, error); + if (!etcdest_stream) + goto out; + + usrdest_stream = gfopen (usrdest_path, "a", cancellable, error); + if (!usrdest_stream) + goto out; + + errno = 0; + while (TRUE) + { + struct passwd *pw = NULL; + struct group *gr = NULL; + FILE *deststream; + int r; + guint32 id; + const char *name; + + if (kind == RPM_OSTREE_PASSWD_MIGRATE_PASSWD) + pw = fgetpwent (src_stream); + else + gr = fgetgrent (src_stream); + + if (!(pw || gr)) + { + if (errno != 0 && errno != ENOENT) + { + _rpmostree_set_prefix_error_from_errno (error, errno, "fgetpwent: "); + goto out; + } + else + break; + } + + + if (pw) + { + id = pw->pw_uid; + name = pw->pw_name; + } + else + { + id = gr->gr_gid; + name = gr->gr_name; + } + + if (id == 0) + deststream = etcdest_stream; + else + deststream = usrdest_stream; + + if (pw) + r = putpwent (pw, deststream); + else + r = putgrent (gr, deststream); + + /* If it's marked in the preserve group, we need to write to + * *both* /etc and /usr/lib in order to preserve semantics for + * upgraded systems from before we supported the preserve + * concept. + */ + if (preserve && g_hash_table_contains (preserve, name)) + { + /* We should never be trying to preserve the root entry, it + * should always be only in /etc. + */ + g_assert (deststream == usrdest_stream); + if (pw) + r = putpwent (pw, etcdest_stream); + else + r = putgrent (gr, etcdest_stream); + } + + if (r == -1) + { + _rpmostree_set_prefix_error_from_errno (error, errno, "putpwent: "); + goto out; + } + } + + if (!gfflush (etcdest_stream, cancellable, error)) + goto out; + if (!gfflush (usrdest_stream, cancellable, error)) + goto out; + + if (rename (etctmp_path, src_path) != 0) + { + _rpmostree_set_prefix_error_from_errno (error, errno, "rename(%s, %s): ", + etctmp_path, src_path); + goto out; + } + + ret = TRUE; + out: + if (src_stream) (void) fclose (src_stream); + if (etcdest_stream) (void) fclose (etcdest_stream); + if (usrdest_stream) (void) fclose (usrdest_stream); + return ret; +} + static gboolean concat_passwd_file (GFile *yumroot, GFile *previous_commit, diff --git a/src/rpmostree-passwd-util.h b/src/rpmostree-passwd-util.h index d7ac49e9..5469024a 100644 --- a/src/rpmostree-passwd-util.h +++ b/src/rpmostree-passwd-util.h @@ -40,6 +40,18 @@ rpmostree_check_groups (OstreeRepo *repo, GCancellable *cancellable, GError **error); +typedef enum { + RPM_OSTREE_PASSWD_MIGRATE_PASSWD, + RPM_OSTREE_PASSWD_MIGRATE_GROUP +} RpmOstreePasswdMigrateKind; + +gboolean +rpmostree_passwd_migrate_except_root (GFile *rootfs, + RpmOstreePasswdMigrateKind kind, + GHashTable *preserve, + GCancellable *cancellable, + GError **error); + gboolean rpmostree_generate_passwd_from_previous (OstreeRepo *repo, GFile *yumroot, diff --git a/src/rpmostree-postprocess.c b/src/rpmostree-postprocess.c index 5d52765d..b7ca0c45 100644 --- a/src/rpmostree-postprocess.c +++ b/src/rpmostree-postprocess.c @@ -34,6 +34,7 @@ #include #include "rpmostree-postprocess.h" +#include "rpmostree-passwd-util.h" #include "rpmostree-libcontainer.h" #include "rpmostree-cleanup.h" #include "rpmostree-json-parsing.h" @@ -593,170 +594,6 @@ workaround_selinux_cross_labeling (GFile *rootfs, return ret; } -static FILE * -gfopen (const char *path, - const char *mode, - GCancellable *cancellable, - GError **error) -{ - FILE *ret = NULL; - - ret = fopen (path, mode); - if (!ret) - { - _rpmostree_set_prefix_error_from_errno (error, errno, "fopen(%s): ", path); - return NULL; - } - return ret; -} - -static gboolean -gfflush (FILE *f, - GCancellable *cancellable, - GError **error) -{ - if (fflush (f) != 0) - { - _rpmostree_set_prefix_error_from_errno (error, errno, "fflush: "); - return FALSE; - } - return TRUE; -} - -typedef enum { - MIGRATE_PASSWD, - MIGRATE_GROUP -} MigrateKind; - -/* - * This function is taking the /etc/passwd generated in the install - * root, and splitting it into two streams: a new /etc/passwd that - * just contains the root entry, and /usr/lib/passwd which contains - * everything else. - * - * The implementation is kind of horrible because I wanted to avoid - * duplicating the user/group code. - */ -static gboolean -migrate_passwd_file_except_root (GFile *rootfs, - MigrateKind kind, - GHashTable *preserve, - GCancellable *cancellable, - GError **error) -{ - gboolean ret = FALSE; - const char *name = kind == MIGRATE_PASSWD ? "passwd" : "group"; - gs_free char *src_path = g_strconcat (gs_file_get_path_cached (rootfs), "/etc/", name, NULL); - gs_free char *etctmp_path = g_strconcat (gs_file_get_path_cached (rootfs), "/etc/", name, ".tmp", NULL); - gs_free char *usrdest_path = g_strconcat (gs_file_get_path_cached (rootfs), "/usr/lib/", name, NULL); - FILE *src_stream = NULL; - FILE *etcdest_stream = NULL; - FILE *usrdest_stream = NULL; - - src_stream = gfopen (src_path, "r", cancellable, error); - if (!src_stream) - goto out; - - etcdest_stream = gfopen (etctmp_path, "w", cancellable, error); - if (!etcdest_stream) - goto out; - - usrdest_stream = gfopen (usrdest_path, "a", cancellable, error); - if (!usrdest_stream) - goto out; - - errno = 0; - while (TRUE) - { - struct passwd *pw = NULL; - struct group *gr = NULL; - FILE *deststream; - int r; - guint32 id; - const char *name; - - if (kind == MIGRATE_PASSWD) - pw = fgetpwent (src_stream); - else - gr = fgetgrent (src_stream); - - if (!(pw || gr)) - { - if (errno != 0 && errno != ENOENT) - { - _rpmostree_set_prefix_error_from_errno (error, errno, "fgetpwent: "); - goto out; - } - else - break; - } - - - if (pw) - { - id = pw->pw_uid; - name = pw->pw_name; - } - else - { - id = gr->gr_gid; - name = gr->gr_name; - } - - if (id == 0) - deststream = etcdest_stream; - else - deststream = usrdest_stream; - - if (pw) - r = putpwent (pw, deststream); - else - r = putgrent (gr, deststream); - - /* If it's marked in the preserve group, we need to write to - * *both* /etc and /usr/lib in order to preserve semantics for - * upgraded systems from before we supported the preserve - * concept. - */ - if (preserve && g_hash_table_contains (preserve, name)) - { - /* We should never be trying to preserve the root entry, it - * should always be only in /etc. - */ - g_assert (deststream == usrdest_stream); - if (pw) - r = putpwent (pw, etcdest_stream); - else - r = putgrent (gr, etcdest_stream); - } - - if (r == -1) - { - _rpmostree_set_prefix_error_from_errno (error, errno, "putpwent: "); - goto out; - } - } - - if (!gfflush (etcdest_stream, cancellable, error)) - goto out; - if (!gfflush (usrdest_stream, cancellable, error)) - goto out; - - if (rename (etctmp_path, src_path) != 0) - { - _rpmostree_set_prefix_error_from_errno (error, errno, "rename(%s, %s): ", - etctmp_path, src_path); - goto out; - } - - ret = TRUE; - out: - if (src_stream) (void) fclose (src_stream); - if (etcdest_stream) (void) fclose (etcdest_stream); - if (usrdest_stream) (void) fclose (usrdest_stream); - return ret; -} - static gboolean replace_nsswitch (GFile *target_usretc, GCancellable *cancellable, @@ -908,8 +745,8 @@ create_rootfs_from_yumroot_content (GFile *targetroot, goto out; g_print ("Migrating /etc/passwd to /usr/lib/\n"); - if (!migrate_passwd_file_except_root (yumroot, MIGRATE_PASSWD, NULL, - cancellable, error)) + if (!rpmostree_passwd_migrate_except_root (yumroot, RPM_OSTREE_PASSWD_MIGRATE_PASSWD, NULL, + cancellable, error)) goto out; if (json_object_has_member (treefile, "etc-group-members")) @@ -919,8 +756,9 @@ create_rootfs_from_yumroot_content (GFile *targetroot, } g_print ("Migrating /etc/group to /usr/lib/\n"); - if (!migrate_passwd_file_except_root (yumroot, MIGRATE_GROUP, preserve_groups_set, - cancellable, error)) + if (!rpmostree_passwd_migrate_except_root (yumroot, RPM_OSTREE_PASSWD_MIGRATE_GROUP, + preserve_groups_set, + cancellable, error)) goto out; /* NSS configuration to look at the new files */