/* -*-c-*-
- *
- * $Id: util.c,v 1.3 2004/04/08 01:36:17 mdw Exp $
*
* Utilities for the client and the server
*
* (c) 2001 Straylight/Edgeware
*/
-/*----- Licensing notice --------------------------------------------------*
+/*----- Licensing notice --------------------------------------------------*
*
* This file is part of Trivial IP Encryption (TrIPE).
*
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 2 of the License, or
* (at your option) any later version.
- *
+ *
* TrIPE is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
- *
+ *
* You should have received a copy of the GNU General Public License
* along with TrIPE; if not, write to the Free Software Foundation,
* Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
#include <sys/types.h>
#include <unistd.h>
-#include <fcntl.h>
+
+#include <pwd.h>
+#include <grp.h>
#include <mLib/dstr.h>
+#include <mLib/report.h>
#include "util.h"
-#include <sys/ioctl.h>
-
/*----- Main code ---------------------------------------------------------*/
/* --- @u_quotify@ --- *
}
}
-/* --- @u_detach@ --- *
+/* --- @u_getuser@ --- *
*
- * Arguments: ---
+ * Arguments: @const char *name@ = user name or id requested
+ * @gid_t *gg@ = where to store corresponding gid
*
- * Returns: ---
+ * Returns: Corresponding uid.
*
- * Use: Detaches from the current terminal and ensures it can never
- * acquire a new one. Calls @fork@.
+ * Use: Resolves a user name into a uid. Dies on failure; suitable
+ * for use in argument parsing.
*/
-void u_detach(void)
+uid_t u_getuser(const char *name, gid_t *gg)
{
-#ifdef TIOCNOTTY
- {
- int fd;
- if ((fd = open("/dev/tty", O_RDONLY)) >= 0) {
- ioctl(fd, TIOCNOTTY);
- close(fd);
- }
- }
-#endif
- setsid();
- if (fork() > 0)
- _exit(0);
+ struct passwd *pw;
+ char *p;
+ unsigned long i = strtoul(name, &p, 0);
+
+ if (!*p)
+ pw = getpwuid(i);
+ else
+ pw = getpwnam(name);
+ if (!pw)
+ die(EXIT_FAILURE, "user `%s' not found", name);
+ if (gg && *gg == -1)
+ *gg = pw->pw_gid;
+ return (pw->pw_uid);
}
-/* --- @u_daemon@ --- *
+/* --- @u_getgroup@ --- *
*
- * Arguments: ---
+ * Arguments: @const char *name@ = user name or id requested
*
- * Returns: Zero if OK, nonzero on failure.
+ * Returns: Corresponding gid.
*
- * Use: Becomes a daemon.
+ * Use: Resolves a group name into a gid. Dies on failure; suitable
+ * for use in argument parsing.
*/
-int u_daemon(void)
+gid_t u_getgroup(const char *name)
{
- pid_t kid;
-
- if ((kid = fork()) < 0)
- return (-1);
- if (kid)
- _exit(0);
- u_detach();
- return (0);
+ struct group *gr;
+ char *p;
+ unsigned long i = strtoul(name, &p, 0);
+
+ if (!*p)
+ gr = getgrgid(i);
+ else
+ gr = getgrnam(name);
+ if (!gr)
+ die(EXIT_FAILURE, "group `%s' not found", name);
+ return (gr->gr_gid);
+}
+
+/* --- @u_setugid@ --- *
+ *
+ * Arguments: @uid_t u@ = user to set
+ * @gid_t g@ = group to set
+ *
+ * Returns: ---
+ *
+ * Use: Sets user and group to the given values; aborts on failure.
+ */
+
+void u_setugid(uid_t u, gid_t g)
+{
+ if (g != (gid_t)-1) {
+ if (setgid(g) || (getuid() == 0 && setgroups(1, &g))) {
+ die(EXIT_FAILURE, "couldn't setgid to %u: %s",
+ (unsigned)g, strerror(errno));
+ }
+ }
+ if (u != (uid_t)-1) {
+ if (setuid(u)) {
+ die(EXIT_FAILURE, "couldn't setuid to %u: %s",
+ (unsigned)u, strerror(errno));
+ }
+ }
}
/*----- That's all, folks -------------------------------------------------*/