- Rewrote bsd-login to use proper utmp API if available. Major cleanup

   of fallback DIY code.

diff --git a/ChangeLog b/ChangeLog
index a87f32b74817a9f57188521d0d69e7fb3a156ade..fa1f65be7fdf442016757eadfcf7353b1ddd0764 100644 (file)
--- a/ChangeLog
+++ b/ChangeLog
@@ -2,6 +2,8 @@
  - Cleanup of auth.c, login.c and fake-*
  - Cleanup of auth-pam.c, save and print "account expired" error messages
  - Fix EGD read bug by IWAMURO Motonori <iwa@mmp.fujitsu.co.jp>
+ - Rewrote bsd-login to use proper utmp API if available. Major cleanup
+   of fallback DIY code.
 
 20000530
  - Define atexit for old Solaris

diff --git a/bsd-login.c b/bsd-login.c
index ecd5e05af1b86ed5667f01a394104d88db0f5a0e..a6f4acca345719067d9ef4578189c09eb221773f 100644 (file)
--- a/bsd-login.c
+++ b/bsd-login.c
@@ -1,5 +1,5 @@
 /*
- * This file has been modified from the original OpenBSD version 
+ * This file has been heavily modified from the original OpenBSD version 
  */
 
 /*     $OpenBSD: login.c,v 1.5 1998/07/13 02:11:12 millert Exp $       */
@@ -74,7 +74,7 @@ struct utmp * utp;
        int t = 0;
        struct utmp * u;
 
-#if defined(HAVE_TYPE_IN_UTMP) || defined(HAVE_TYPE_IN_UTMPX)
+# if defined(HAVE_TYPE_IN_UTMP) || defined(HAVE_TYPE_IN_UTMPX)
        setutent();
 
        while((u = getutent()) != NULL) {
@@ -93,35 +93,47 @@ struct utmp * utp;
        }
 
        endutent();
-#endif
+# endif /* defined(HAVE_TYPE_IN_UTMP) || defined(HAVE_TYPE_IN_UTMPX) */
        return(-1);
 }
-#else
-int find_tty_slot( utp )
-struct utmp * utp;
+#else /* USER_PROCESS */
+int find_tty_slot(struct utmp *utp)
 {
        return(ttyslot());
 }
-#endif
+#endif /* USER_PROCESS */
 
 #if defined(HAVE_UTMPX_H) && defined(USE_UTMPX)
-void
-login(utp,utx)
-       struct utmp *utp;
-       struct utmpx *utx;
+void login(struct utmpx *utx)
 #else /* defined(HAVE_UTMPX_H) && defined(USE_UTMPX) */
-void
-login(utp)
-       struct utmp *utp;
+void login(struct utmp *utp)
 #endif /* defined(HAVE_UTMPX_H) && defined(USE_UTMPX) */
 {
-#if defined(HAVE_HOST_IN_UTMP)
-       struct utmp old_ut;
-#endif
-#if defined(HAVE_UTMPX_H) && defined(USE_UTMPX)
-       struct utmpx *old_utx;
-#endif /* defined(HAVE_UTMPX_H) && defined(USE_UTMPX) */
-       register int fd;
+       /* Use proper API if we have it */
+#if defined(USE_UTMPX)
+# if defined(HAVE_PUTUTXLINE)
+       setutxent();
+       pututxline(utx);
+       endutxent();    
+# endif /* defined(HAVE_PUTUTXLINE) */
+# if defined(HAVE_UPDWTMPX)
+       updwtmpx(_PATH_WTMPX, utx);
+# endif /* defined(HAVE_UPDWTMPX) */
+#else /* defined(USE_UTMPX) */
+# if defined(HAVE_PUTUTLINE)
+       setutent();
+       pututline(utp);
+       endutent();     
+# endif /* defined(HAVE_PUTUTLINE) */
+# if defined(HAVE_UPDWTMPX)
+       updwtmp(_PATH_WTMP, utp);
+# endif /* defined(HAVE_UPDWTMP) */
+#endif /* defined(USE_UTMPX) */
+
+       /* Otherwise DIY */
+#if (defined(USE_UTMPX) && !defined(HAVE_PUTUTXLINE)) || \
+       (!defined(USE_UTMPX) && !defined(HAVE_PUTUTLINE)) 
+       int fd;
        int tty;
 
        /* can't use ttyslot here, as that will not work for logout
@@ -132,72 +144,52 @@ login(utp)
        tty = find_tty_slot(utp);
 
 #ifdef USE_UTMPX
+       /* If no tty was found, append it to utmpx */
+       if (tty == -1) {
+               if ((fd = open(_PATH_UTMPX, O_WRONLY|O_APPEND, 0)) >= 0) {
+                       (void)write(fd, utp, sizeof(struct utmp));
+                       (void)close(fd);
+                       return;
+               }
+       }
+       /* Otherwise, tty was found - update at its location */
        fd = open(_PATH_UTMPX, O_RDWR|O_CREAT, 0644);
        if (fd == -1) {
                log("Couldn't open %s: %s", _PATH_UTMPX, strerror(errno));
+               return;
+       }
+       lseek(fd, (off_t)(tty * sizeof(struct utmpx)), SEEK_SET);
+       write(fd, utx, sizeof(struct utmpx));
+       close(fd);
+       if ((fd = open(_PATH_WTMPX, O_WRONLY|O_APPEND, 0)) >= 0) {
+               (void)write(fd, utx, sizeof(struct utmpx));
+               (void)close(fd);
+       }
 #else /* USE_UTMPX */
-       fd = open(_PATH_UTMP, O_RDWR|O_CREAT, 0644);
-       if (fd == -1) {
-               log("Couldn't open %s: %s", _PATH_UTMP, strerror(errno));
-#endif /* USE_UTMPX */
-       } else {
-               /* If no tty was found... */
-               if (tty == -1) {
-                       /* ... append it to utmp on login */
-#if defined(HAVE_TYPE_IN_UTMP) || defined(HAVE_TYPE_IN_UTMPX)
-                       if (utp->ut_type == USER_PROCESS) {
-#ifdef USE_UTMPX
-                               if ((fd = open(_PATH_UTMPX, O_WRONLY|O_APPEND, 0)) >= 0) {
-#else /* USE_UTMPX */
-                               if ((fd = open(_PATH_UTMP, O_WRONLY|O_APPEND, 0)) >= 0) {
-#endif /* USE_UTMPX */
-                                       (void)write(fd, utp, sizeof(struct utmp));
-                                       (void)close(fd);
-                               }
-                       } else {
-                               /* Shouldn't get to here unless somthing happened to utmp */
-                               /* Between login and logout */
-                               log("No tty slot found at logout");
-                       }
-#endif
-               } else {
-                       /* Otherwise, tty was found - update at its location */
-#if defined(HAVE_HOST_IN_UTMP)
-# ifndef UT_LINESIZE
-#  define UT_LINESIZE (sizeof(old_ut.ut_line))
-#  define UT_NAMESIZE (sizeof(old_ut.ut_name))
-#  define UT_HOSTSIZE (sizeof(old_ut.ut_host))
-# endif
-                       (void)lseek(fd, (off_t)(tty * sizeof(struct utmp)), SEEK_SET);
-                       /*
-                        * Prevent luser from zero'ing out ut_host.
-                        * If the new ut_line is empty but the old one is not
-                        * and ut_line and ut_name match, preserve the old ut_line.
-                        */
-                       if (read(fd, &old_ut, sizeof(struct utmp)) ==
-                        sizeof(struct utmp) && utp->ut_host[0] == '\0' &&
-                        old_ut.ut_host[0] != '\0' &&
-                        strncmp(old_ut.ut_line, utp->ut_line, UT_LINESIZE) == 0 &&
-                        strncmp(old_ut.ut_name, utp->ut_name, UT_NAMESIZE) == 0)
-                               (void)memcpy(utp->ut_host, old_ut.ut_host, UT_HOSTSIZE);
-#endif /* defined(HAVE_HOST_IN_UTMP) */
-                       (void)lseek(fd, (off_t)(tty * sizeof(struct utmp)), SEEK_SET);
+       /* If no tty was found, append it to utmp */
+       if (tty == -1) {
+               if ((fd = open(_PATH_UTMP, O_WRONLY|O_APPEND, 0)) >= 0) {
                        (void)write(fd, utp, sizeof(struct utmp));
                        (void)close(fd);
+                       return;
                }
        }
-
+       /* Otherwise, tty was found - update at its location */
+       fd = open(_PATH_UTMP, O_RDWR|O_CREAT, 0644);
+       if (fd == -1) {
+               log("Couldn't open %s: %s", _PATH_UTMP, strerror(errno));
+               return;
+       }
+       lseek(fd, (off_t)(tty * sizeof(struct utmp)), SEEK_SET);
+       write(fd, utp, sizeof(struct utmp));
+       close(fd);
        if ((fd = open(_PATH_WTMP, O_WRONLY|O_APPEND, 0)) >= 0) {
                (void)write(fd, utp, sizeof(struct utmp));
                (void)close(fd);
        }
-#if defined(HAVE_UTMPX_H) && defined(USE_UTMPX)
-       old_utx = pututxline(utx);
-# ifdef HAVE_UPDWTMPX
-       updwtmpx(_PATH_WTMPX, utx);
-# endif /* HAVE_UPDWTMPX */
-       endutxent();
-#endif /* defined(HAVE_UTMPX_H) && defined(USE_UTMPX) */
+#endif /* USE_UTMPX */
+#endif /* (defined(USE_UTMPX) && !defined(HAVE_PUTUTXLINE)) || \
+                       (!defined(USE_UTMPX) && !defined(HAVE_PUTUTLINE)) */
 }
 
 #endif /* HAVE_LOGIN */

diff --git a/configure.in b/configure.in
index 3c6694dbaed2cb4ba97340b03a78074a490e91bd..86284aa220858f92388bd342cbcc2555b5e6d38e 100644 (file)
--- a/configure.in
+++ b/configure.in
@@ -135,7 +135,7 @@ fi
 AC_CHECK_HEADERS(bstring.h endian.h lastlog.h login.h maillock.h netdb.h netgroup.h netinet/in_systm.h paths.h poll.h pty.h shadow.h security/pam_appl.h sys/bitypes.h sys/bsdtty.h sys/cdefs.h sys/poll.h sys/select.h sys/stropts.h sys/sysmacros.h sys/time.h sys/ttcompat.h stddef.h util.h utmp.h utmpx.h)
 
 # Checks for library functions.
-AC_CHECK_FUNCS(arc4random atexit b64_ntop bcopy bindresvport_af clock freeaddrinfo gai_strerror getaddrinfo getnameinfo getrusage innetgr md5_crypt memmove mkdtemp on_exit openpty rresvport_af setenv seteuid setlogin setproctitle setreuid snprintf strlcat strlcpy updwtmpx vsnprintf vhangup _getpty __b64_ntop)
+AC_CHECK_FUNCS(arc4random atexit b64_ntop bcopy bindresvport_af clock freeaddrinfo gai_strerror getaddrinfo getnameinfo getrusage innetgr md5_crypt memmove mkdtemp on_exit openpty pututline pututxline rresvport_af setenv seteuid setlogin setproctitle setreuid snprintf strlcat strlcpy updwtmp updwtmpx vsnprintf vhangup _getpty __b64_ntop)
 
 AC_CHECK_FUNC(login, 
        [AC_DEFINE(HAVE_LOGIN)],

diff --git a/login.c b/login.c
index 04e36affd8ed5a137d4289ae9de6ba9fe2d838c4..a1a336140c96249e26bf01b1e25b993e4d04d341 100644 (file)
--- a/login.c
+++ b/login.c
@@ -253,7 +253,7 @@ record_login(pid_t pid, const char *ttyname, const char *user, uid_t uid,
 #endif /* defined(HAVE_UTMPX_H) && defined(USE_UTMPX) */
 
 #if defined(HAVE_UTMPX_H) && defined(USE_UTMPX)
-       login(&u, &utx);
+       login(&utx);
 #else /* defined(HAVE_UTMPX_H) && defined(USE_UTMPX) */
        login(&u);
 #endif /* defined(HAVE_UTMPX_H) && defined(USE_UTMPX) */