appl/login/utmpx_login.c

   1 /************************************************************************
   2 * Copyright 1995 by Wietse Venema.  All rights reserved.  Some individual
   3 * files may be covered by other copyrights.
   4 *
   5 * This material was originally written and compiled by Wietse Venema at
   6 * Eindhoven University of Technology, The Netherlands, in 1990, 1991,
   7 * 1992, 1993, 1994 and 1995.
   8 *
   9 * Redistribution and use in source and binary forms, with or without
  10 * modification, are permitted provided that this entire copyright notice
  11 * is duplicated in all such copies.
  12 *
  13 * This software is provided "as is" and without any expressed or implied
  14 * warranties, including, without limitation, the implied warranties of
  15 * merchantibility and fitness for any particular purpose.
  16 ************************************************************************/
  17 /* Author: Wietse Venema <wietse@wzv.win.tue.nl> */
  18
  19 #include "login_locl.h"
  20
  21 RCSID("$Id$");
  22
  23 /* utmpx_login - update utmp and wtmp after login */
  24
  25 #ifndef HAVE_UTMPX_H
  26 int utmpx_login(char *line, const char *user, const char *host) { return 0; }
  27 #else
  28
  29 static void
  30 utmpx_update(struct utmpx *ut, char *line, const char *user, const char *host)
  31 {
  32     struct timeval tmp;
  33     char *clean_tty = clean_ttyname(line);
  34
  35     strncpy(ut->ut_line, clean_tty, sizeof(ut->ut_line));
  36 #ifdef HAVE_STRUCT_UTMPX_UT_ID
  37     strncpy(ut->ut_id, make_id(clean_tty), sizeof(ut->ut_id));
  38 #endif
  39     strncpy(ut->ut_user, user, sizeof(ut->ut_user));
  40     shrink_hostname (host, ut->ut_host, sizeof(ut->ut_host));
  41 #ifdef HAVE_STRUCT_UTMPX_UT_SYSLEN
  42     ut->ut_syslen = strlen(host) + 1;
  43     if (ut->ut_syslen > sizeof(ut->ut_host))
  44         ut->ut_syslen = sizeof(ut->ut_host);
  45 #endif
  46     ut->ut_type = USER_PROCESS;
  47     gettimeofday (&tmp, 0);
  48     ut->ut_tv.tv_sec = tmp.tv_sec;
  49     ut->ut_tv.tv_usec = tmp.tv_usec;
  50     pututxline(ut);
  51 #ifdef WTMPX_FILE
  52     updwtmpx(WTMPX_FILE, ut);
  53 #elif defined(WTMP_FILE)
  54     { /* XXX should be removed, just drop wtmp support */
  55         struct utmp utmp;
  56         int fd;
  57
  58         prepare_utmp (&utmp, line, user, host);
  59         if ((fd = open(_PATH_WTMP, O_WRONLY|O_APPEND, 0)) >= 0) {
  60             write(fd, &utmp, sizeof(struct utmp));
  61             close(fd);
  62         }
  63     }
  64 #endif
  65 }
  66
  67 int
  68 utmpx_login(char *line, const char *user, const char *host)
  69 {
  70     struct utmpx *ut, save_ut;
  71     pid_t   mypid = getpid();
  72     int     ret = (-1);
  73
  74     /*
  75      * SYSV4 ttymon and login use tty port names with the "/dev/" prefix
  76      * stripped off. Rlogind and telnetd, on the other hand, make utmpx
  77      * entries with device names like /dev/pts/nnn. We therefore cannot use
  78      * getutxline(). Return nonzero if no utmp entry was found with our own
  79      * process ID for a login or user process.
  80      */
  81
  82     while ((ut = getutxent())) {
  83         /* Try to find a reusable entry */
  84         if (ut->ut_pid == mypid
  85             && (   ut->ut_type == INIT_PROCESS
  86                 || ut->ut_type == LOGIN_PROCESS
  87                 || ut->ut_type == USER_PROCESS)) {
  88             save_ut = *ut;
  89             utmpx_update(&save_ut, line, user, host);
  90             ret = 0;
  91             break;
  92         }
  93     }
  94     if (ret == -1) {
  95         /* Grow utmpx file by one record. */
  96         struct utmpx newut;
  97         memset(&newut, 0, sizeof(newut));
  98         newut.ut_pid = mypid;
  99         utmpx_update(&newut, line, user, host);
 100         ret = 0;
 101     }
 102     endutxent();
 103     return (ret);
 104 }
 105 #endif /* HAVE_UTMPX_H */