Update.
[kopensolaris-gnu/glibc.git] / login / login.c
index f74ce41..cf8632d 100644 (file)
-/* Copyright (C) 1996 Free Software Foundation, Inc.
-This file is part of the GNU C Library.
-Contributed by Ulrich Drepper <drepper@cygnus.com>, 1996.
+/* Copyright (C) 1996, 1997 Free Software Foundation, Inc.
+   This file is part of the GNU C Library.
+   Contributed by Ulrich Drepper <drepper@cygnus.com>, 1996.
 
-The GNU C Library is free software; you can redistribute it and/or
-modify it under the terms of the GNU Library General Public License as
-published by the Free Software Foundation; either version 2 of the
-License, or (at your option) any later version.
+   The GNU C Library is free software; you can redistribute it and/or
+   modify it under the terms of the GNU Library General Public License as
+   published by the Free Software Foundation; either version 2 of the
+   License, or (at your option) any later version.
 
-The GNU C Library 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
-Library General Public License for more details.
+   The GNU C Library 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
+   Library General Public License for more details.
 
-You should have received a copy of the GNU Library General Public
-License along with the GNU C Library; see the file COPYING.LIB.  If
-not, write to the Free Software Foundation, Inc., 59 Temple Place - Suite 330,
-Boston, MA 02111-1307, USA.  */
+   You should have received a copy of the GNU Library General Public
+   License along with the GNU C Library; see the file COPYING.LIB.  If not,
+   write to the Free Software Foundation, Inc., 59 Temple Place - Suite 330,
+   Boston, MA 02111-1307, USA.  */
 
 #include <errno.h>
 #include <limits.h>
 #include <string.h>
 #include <unistd.h>
+#include <stdlib.h>
 #include <utmp.h>
 
 
+/* Return the result of ttyname in the buffer pointed to by TTY, which should
+   be of length BUF_LEN.  If it is too long to fit in this buffer, a
+   sufficiently long buffer is allocated using malloc, and returned in TTY.
+   0 is returned upon success, -1 otherwise.  */
+static int
+tty_name (int fd, char **tty, size_t buf_len)
+{
+  int rv;                      /* Return value.  0 = success.  */
+  char *buf = *tty;            /* Buffer for ttyname, initially the user's. */
+
+  for (;;)
+    {
+      char *new_buf;
+
+      if (buf_len)
+       {
+         rv = ttyname_r (fd, buf, buf_len);
+
+         if (rv < 0 || memchr (buf, '\0', buf_len))
+           /* We either got an error, or we succeeded and the
+              returned name fit in the buffer.  */
+           break;
+
+         /* Try again with a longer buffer.  */
+         buf_len += buf_len;   /* Double it */
+       }
+      else
+       /* No initial buffer; start out by mallocing one.  */
+       buf_len = 128;          /* First time guess.  */
+
+      if (buf != *tty)
+       /* We've already malloced another buffer at least once.  */
+       new_buf = realloc (buf, buf_len);
+      else
+       new_buf = malloc (buf_len);
+      if (! new_buf)
+       {
+         rv = -1;
+         __set_errno (ENOMEM);
+         break;
+       }
+      buf = new_buf;
+    }
+
+  if (rv == 0)
+    *tty = buf;                /* Return buffer to the user.  */
+  else if (buf != *tty)
+    free (buf);                /* Free what we malloced when returning an error.  */
+
+  return rv;
+}
+\f
 void
 login (const struct utmp *ut)
 {
-  char tty[PATH_MAX + UT_LINESIZE];
+#ifdef PATH_MAX
+  char _tty[PATH_MAX + UT_LINESIZE];
+#else
+  char _tty[512 + UT_LINESIZE];
+#endif
+  char *tty = _tty;
   int found_tty;
   const char *ttyp;
-  struct utmp_data data;
+  struct utmp copy = *ut;
+  struct utmp utbuf;
+
+  /* Fill in those fields we supply.  */
+#if _HAVE_UT_TYPE - 0
+  copy.ut_type = USER_PROCESS;
+#endif
+#if _HAVE_UT_PID - 0
+  copy.ut_pid = getpid ();
+#endif
 
   /* Seek tty.  */
-  found_tty = ttyname_r (STDIN_FILENO, tty, sizeof tty);
+  found_tty = tty_name (STDIN_FILENO, &tty, sizeof (_tty));
   if (found_tty < 0)
-    found_tty = ttyname_r (STDOUT_FILENO, tty, sizeof tty);
+    found_tty = tty_name (STDOUT_FILENO, &tty, sizeof (_tty));
   if (found_tty < 0)
-    found_tty = ttyname_r (STDERR_FILENO, tty, sizeof tty);
+    found_tty = tty_name (STDERR_FILENO, &tty, sizeof (_tty));
 
   if (found_tty >= 0)
     {
+      /* We only want to insert the name of the tty without path.  */
+      ttyp = basename (tty);
+
+      /* Position to record for this tty.  */
+      strncpy (copy.ut_line, ttyp, UT_LINESIZE);
+
       /* Tell that we want to use the UTMP file.  */
       if (utmpname (_PATH_UTMP) != 0)
        {
-         struct utmp tmp;
          struct utmp *old;
 
          /* Open UTMP file.  */
-         setutent_r (&data);
-
-         /* We only want to insert the name of the tty without path.  */
-         ttyp = basename (tty);
-
-         /* Position to record for this tty.  */
-#if _HAVE_UT_TYPE - 0
-         tmp.ut_type = USER_PROCESS;
-#endif
-         strncpy (tmp.ut_line, ttyp, UT_LINESIZE);
+         setutent ();
 
          /* Read the record.  */
-         if (getutline_r (&tmp, &old, &data) >= 0 || errno == ESRCH)
-           {
-#if _HAVE_UT_TYPE - 0
-             /* We have to fake the old entry because this `login'
-                function does not fit well into the UTMP file
-                handling scheme.  */
-             old->ut_type = ut->ut_type;
-#endif
-             pututline_r (ut, &data);
-           }
+         getutline_r (&copy, &utbuf, &old);
+
+         /* Write the entry.  */
+         pututline (&copy);
 
          /* Close UTMP file.  */
-         endutent_r (&data);
+         endutent ();
        }
+
+      if (tty != _tty)
+       free (tty);             /* Free buffer malloced by tty_name.  */
     }
 
   /* Update the WTMP file.  Here we have to add a new entry.  */
-  if (utmpname (_PATH_WTMP) != 0)
-    {
-      /* Open the WTMP file.  */
-      setutent_r (&data);
-
-      /* Position at end of file.  */
-      data.loc_utmp = lseek (data.ut_fd, 0, SEEK_END);
-      if (data.loc_utmp != -1)
-       {
-#if _HAVE_UT_TYPE - 0
-         /* We have to fake the old entry because this `login'
-            function does not fit well into the UTMP file handling
-            scheme.  */
-         data.ubuf.ut_type = ut->ut_type;
-#endif
-         pututline_r (ut, &data);
-       }
-
-      /* Close WTMP file.  */
-      endutent_r (&data);
-    }
+  updwtmp (_PATH_WTMP, &copy);
 }