155682Smarkm/* 278527Sassar * Copyright (c) 1995 - 2001 Kungliga Tekniska H�gskolan 355682Smarkm * (Royal Institute of Technology, Stockholm, Sweden). 455682Smarkm * All rights reserved. 555682Smarkm * 655682Smarkm * Redistribution and use in source and binary forms, with or without 755682Smarkm * modification, are permitted provided that the following conditions 855682Smarkm * are met: 955682Smarkm * 1055682Smarkm * 1. Redistributions of source code must retain the above copyright 1155682Smarkm * notice, this list of conditions and the following disclaimer. 1255682Smarkm * 1355682Smarkm * 2. Redistributions in binary form must reproduce the above copyright 1455682Smarkm * notice, this list of conditions and the following disclaimer in the 1555682Smarkm * documentation and/or other materials provided with the distribution. 1655682Smarkm * 1755682Smarkm * 3. Neither the name of the Institute nor the names of its contributors 1855682Smarkm * may be used to endorse or promote products derived from this software 1955682Smarkm * without specific prior written permission. 2055682Smarkm * 2155682Smarkm * THIS SOFTWARE IS PROVIDED BY THE INSTITUTE AND CONTRIBUTORS ``AS IS'' AND 2255682Smarkm * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 2355682Smarkm * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 2455682Smarkm * ARE DISCLAIMED. IN NO EVENT SHALL THE INSTITUTE OR CONTRIBUTORS BE LIABLE 2555682Smarkm * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 2655682Smarkm * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 2755682Smarkm * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 2855682Smarkm * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 2955682Smarkm * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 3055682Smarkm * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 3155682Smarkm * SUCH DAMAGE. 3255682Smarkm */ 3355682Smarkm 3455682Smarkm#include "login_locl.h" 3555682Smarkm 36178825SdfrRCSID("$Id: utmp_login.c 9661 2001-02-08 16:08:47Z assar $"); 3755682Smarkm 3878527Sassar/* try to put something useful from hostname into dst, dst_sz: 3978527Sassar * full name, first component or address */ 4078527Sassar 4155682Smarkmvoid 4278527Sassarshrink_hostname (const char *hostname, 4378527Sassar char *dst, size_t dst_sz) 4478527Sassar{ 4578527Sassar char local_hostname[MaxHostNameLen]; 4678527Sassar char *ld, *hd; 4778527Sassar int ret; 4878527Sassar struct addrinfo *ai; 4978527Sassar 5078527Sassar if (strlen(hostname) < dst_sz) { 5178527Sassar strlcpy (dst, hostname, dst_sz); 5278527Sassar return; 5378527Sassar } 5478527Sassar gethostname (local_hostname, sizeof(local_hostname)); 5578527Sassar hd = strchr (hostname, '.'); 5678527Sassar ld = strchr (local_hostname, '.'); 5778527Sassar if (hd != NULL && ld != NULL && strcmp(hd, ld) == 0 5878527Sassar && hd - hostname < dst_sz) { 5978527Sassar strlcpy (dst, hostname, dst_sz); 6078527Sassar dst[hd - hostname] = '\0'; 6178527Sassar return; 6278527Sassar } 6378527Sassar 6478527Sassar ret = getaddrinfo (hostname, NULL, NULL, &ai); 6578527Sassar if (ret) { 6678527Sassar strncpy (dst, hostname, dst_sz); 6778527Sassar return; 6878527Sassar } 6978527Sassar ret = getnameinfo (ai->ai_addr, ai->ai_addrlen, 7078527Sassar dst, dst_sz, 7178527Sassar NULL, 0, 7278527Sassar NI_NUMERICHOST); 7378527Sassar freeaddrinfo (ai); 7478527Sassar if (ret) { 7578527Sassar strncpy (dst, hostname, dst_sz); 7678527Sassar return; 7778527Sassar } 7878527Sassar} 7978527Sassar 8078527Sassarvoid 8155682Smarkmprepare_utmp (struct utmp *utmp, char *tty, 8255682Smarkm const char *username, const char *hostname) 8355682Smarkm{ 8455682Smarkm char *ttyx = clean_ttyname (tty); 8555682Smarkm 8655682Smarkm memset(utmp, 0, sizeof(*utmp)); 8755682Smarkm utmp->ut_time = time(NULL); 8855682Smarkm strncpy(utmp->ut_line, ttyx, sizeof(utmp->ut_line)); 8955682Smarkm strncpy(utmp->ut_name, username, sizeof(utmp->ut_name)); 9055682Smarkm 9155682Smarkm# ifdef HAVE_STRUCT_UTMP_UT_USER 9255682Smarkm strncpy(utmp->ut_user, username, sizeof(utmp->ut_user)); 9355682Smarkm# endif 9455682Smarkm 9555682Smarkm# ifdef HAVE_STRUCT_UTMP_UT_ADDR 9655682Smarkm if (hostname[0]) { 9755682Smarkm struct hostent *he; 9855682Smarkm if ((he = gethostbyname(hostname))) 9955682Smarkm memcpy(&utmp->ut_addr, he->h_addr_list[0], 10055682Smarkm sizeof(utmp->ut_addr)); 10155682Smarkm } 10255682Smarkm# endif 10355682Smarkm 10455682Smarkm# ifdef HAVE_STRUCT_UTMP_UT_HOST 10578527Sassar shrink_hostname (hostname, utmp->ut_host, sizeof(utmp->ut_host)); 10655682Smarkm# endif 10755682Smarkm 10855682Smarkm# ifdef HAVE_STRUCT_UTMP_UT_TYPE 10955682Smarkm utmp->ut_type = USER_PROCESS; 11055682Smarkm# endif 11155682Smarkm 11255682Smarkm# ifdef HAVE_STRUCT_UTMP_UT_PID 11355682Smarkm utmp->ut_pid = getpid(); 11455682Smarkm# endif 11555682Smarkm 11655682Smarkm# ifdef HAVE_STRUCT_UTMP_UT_ID 11755682Smarkm strncpy(utmp->ut_id, make_id(ttyx), sizeof(utmp->ut_id)); 11855682Smarkm# endif 11955682Smarkm} 12055682Smarkm 12155682Smarkm#ifdef HAVE_UTMPX_H 12255682Smarkmvoid utmp_login(char *tty, const char *username, const char *hostname) 12355682Smarkm{ 12455682Smarkm return; 12555682Smarkm} 12655682Smarkm#else 12755682Smarkm 12855682Smarkm/* update utmp and wtmp - the BSD way */ 12955682Smarkm 13055682Smarkmvoid utmp_login(char *tty, const char *username, const char *hostname) 13155682Smarkm{ 13255682Smarkm struct utmp utmp; 13355682Smarkm int fd; 13455682Smarkm 13555682Smarkm prepare_utmp (&utmp, tty, username, hostname); 13655682Smarkm 13755682Smarkm#ifdef HAVE_SETUTENT 13855682Smarkm utmpname(_PATH_UTMP); 13955682Smarkm setutent(); 14055682Smarkm pututline(&utmp); 14155682Smarkm endutent(); 14255682Smarkm#else 14355682Smarkm 14455682Smarkm#ifdef HAVE_TTYSLOT 14555682Smarkm { 14655682Smarkm int ttyno; 14755682Smarkm ttyno = ttyslot(); 14855682Smarkm if (ttyno > 0 && (fd = open(_PATH_UTMP, O_WRONLY, 0)) >= 0) { 14955682Smarkm lseek(fd, (long)(ttyno * sizeof(struct utmp)), SEEK_SET); 15055682Smarkm write(fd, &utmp, sizeof(struct utmp)); 15155682Smarkm close(fd); 15255682Smarkm } 15355682Smarkm } 15455682Smarkm#endif /* HAVE_TTYSLOT */ 15555682Smarkm#endif /* HAVE_SETUTENT */ 15655682Smarkm 15755682Smarkm if ((fd = open(_PATH_WTMP, O_WRONLY|O_APPEND, 0)) >= 0) { 15855682Smarkm write(fd, &utmp, sizeof(struct utmp)); 15955682Smarkm close(fd); 16055682Smarkm } 16155682Smarkm} 16255682Smarkm#endif /* !HAVE_UTMPX_H */ 163