add paranoia checks on setuid/setgid

Signed-off-by: Denys Vlasenko <vda.linux@googlemail.com>

1 files changed, 12 insertions, 0 deletions

diff --git a/lib/Utils/xfuncs.cpp b/lib/Utils/xfuncs.cpp
index 0e57639..8621b5f 100644
--- a/lib/Utils/xfuncs.cpp
+++ b/lib/Utils/xfuncs.cpp
@@ -362,6 +362,18 @@ bool string_to_bool(const char *s)
 	return false;
 }
 
+void xsetreuid(uid_t ruid, uid_t euid)
+{
+	if (setreuid(ruid, euid) != 0)
+		perror_msg_and_die("can't set %cid %d", 'u', (int)ruid);
+}
+
+void xsetregid(gid_t rgid, uid_t egid)
+{
+	if (setregid(rgid, egid) != 0)
+		perror_msg_and_die("can't set %cid %d", 'g', (int)rgid);
+}
+
 uid_t getuidbyname(const char* login)
 {
     struct passwd* pwd = getpwnam(login);