external/bsd/tmux/dist/utmp.c

   1 /*      $NetBSD: utmp.c,v 1.1 2011/09/17 01:50:08 christos Exp $        */
   2
   3 /*-
   4  * Copyright (c) 2011 The NetBSD Foundation, Inc.
   5  * All rights reserved.
   6  *
   7  * This code is derived from software contributed to The NetBSD Foundation
   8  * by Christos Zoulas.
   9  *
  10  * Redistribution and use in source and binary forms, with or without
  11  * modification, are permitted provided that the following conditions
  12  * are met:
  13  * 1. Redistributions of source code must retain the above copyright
  14  *    notice, this list of conditions and the following disclaimer.
  15  * 2. Redistributions in binary form must reproduce the above copyright
  16  *    notice, this list of conditions and the following disclaimer in the
  17  *    documentation and/or other materials provided with the distribution.
  18  * 3. All advertising materials mentioning features or use of this software
  19  *    must display the following acknowledgement:
  20  *        This product includes software developed by the NetBSD
  21  *        Foundation, Inc. and its contributors.
  22  * 4. Neither the name of The NetBSD Foundation nor the names of its
  23  *    contributors may be used to endorse or promote products derived
  24  *    from this software without specific prior written permission.
  25  *
  26  * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
  27  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
  28  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
  29  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
  30  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
  31  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
  32  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
  33  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
  34  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  35  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
  36  * POSSIBILITY OF SUCH DAMAGE.
  37  */
  38 #include <sys/cdefs.h>
  39 __RCSID("$NetBSD: utmp.c,v 1.1 2011/09/17 01:50:08 christos Exp $");
  40
  41 #include <sys/types.h>
  42 #include <sys/param.h>
  43 #include <syslog.h>
  44 #include <string.h>
  45 #include <time.h>
  46 #include <stdlib.h>
  47 #include <unistd.h>
  48 #include <util.h>
  49 #include <paths.h>
  50
  51 #ifdef SUPPORT_UTMP
  52 #include <utmp.h>
  53 #endif
  54 #ifdef SUPPORT_UTMPX
  55 #include <utmpx.h>
  56 #endif
  57
  58 #include "tmux.h"
  59
  60 struct window_utmp {
  61 #ifdef SUPPORT_UTMP
  62         struct utmp ut;
  63 #endif
  64 #ifdef SUPPORT_UTMPX
  65         struct utmpx utx;
  66 #endif
  67 };
  68
  69 #ifdef SUPPORT_UTMPX
  70 static void
  71 login_utmpx(struct utmpx *utmpx, const char *username, const char *hostname,
  72         const char *tty, const struct timeval *now)
  73 {
  74         const char *t;
  75
  76         (void)memset(utmpx, 0, sizeof(*utmpx));
  77         utmpx->ut_tv = *now;
  78         (void)strncpy(utmpx->ut_name, username, sizeof(utmpx->ut_name));
  79         if (hostname)
  80                 (void)strncpy(utmpx->ut_host, hostname, sizeof(utmpx->ut_host));
  81         (void)strncpy(utmpx->ut_line, tty, sizeof(utmpx->ut_line));
  82         utmpx->ut_type = USER_PROCESS;
  83         utmpx->ut_pid = getpid();
  84         t = tty + strlen(tty);
  85         if ((size_t)(t - tty) >= sizeof(utmpx->ut_id)) {
  86             (void)strncpy(utmpx->ut_id, t - sizeof(utmpx->ut_id),
  87                 sizeof(utmpx->ut_id));
  88         } else {
  89             (void)strncpy(utmpx->ut_id, tty, sizeof(utmpx->ut_id));
  90         }
  91         (void)pututxline(utmpx);
  92         endutxent();
  93 }
  94
  95 static void
  96 logout_utmpx(struct utmpx *utmpx, const struct timeval *now)
  97 {
  98         utmpx->ut_type = DEAD_PROCESS;
  99         utmpx->ut_tv = *now;
 100         utmpx->ut_pid = 0;
 101         (void)pututxline(utmpx);
 102         endutxent();
 103 }
 104 #endif
 105
 106 #ifdef SUPPORT_UTMP
 107 static void
 108 login_utmp(struct utmp *utmp, const char *username, const char *hostname,
 109     const char *tty, const struct timeval *now)
 110 {
 111         (void)memset(utmp, 0, sizeof(*utmp));
 112         utmp->ut_time = now->tv_sec;
 113         (void)strncpy(utmp->ut_name, username, sizeof(utmp->ut_name));
 114         if (hostname)
 115                 (void)strncpy(utmp->ut_host, hostname, sizeof(utmp->ut_host));
 116         (void)strncpy(utmp->ut_line, tty, sizeof(utmp->ut_line));
 117         login(utmp);
 118 }
 119
 120 static void
 121 logout_utmp(struct utmp *utmp, const struct timeval *now)
 122 {
 123         logout(utmp->ut_line);
 124 }
 125 #endif
 126
 127 struct window_utmp *
 128 utmp_create(const char *tty)
 129 {
 130         struct window_utmp *wu;
 131         struct timeval tv;
 132         char username[LOGIN_NAME_MAX];
 133
 134         if (getlogin_r(username, sizeof(username)) == -1)
 135                 return NULL;
 136
 137         if ((wu = malloc(sizeof(*wu))) == NULL)
 138                 return NULL;
 139
 140         tty += sizeof(_PATH_DEV) - 1;
 141
 142         (void)gettimeofday(&tv, NULL);
 143 #ifdef SUPPORT_UTMPX
 144         login_utmpx(&wu->utx, username, "tmux", tty, &tv);
 145 #endif
 146 #ifdef SUPPORT_UTMP
 147         login_utmp(&wu->ut, username, "tmux", tty, &tv);
 148 #endif
 149         return wu;
 150 }
 151
 152 void
 153 utmp_destroy(struct window_utmp *wu)
 154 {
 155         struct timeval tv;
 156
 157         if (wu == NULL)
 158                 return;
 159
 160         (void)gettimeofday(&tv, NULL);
 161 #ifdef SUPPORT_UTMPX
 162         logout_utmpx(&wu->utx, &tv);
 163 #endif
 164 #ifdef SUPPORT_UTMP
 165         logout_utmp(&wu->ut, &tv);
 166 #endif
 167 }