utmp_init.c

来自「压缩包中包含LINUX下多个命令的源码」· C语言 代码 · 共 104 行

/************************************************************************
* Copyright 1995 by Wietse Venema.  All rights reserved. Individual files
* may be covered by other copyrights (as noted in the file itself.)
*
* This material was originally written and compiled by Wietse Venema at
* Eindhoven University of Technology, The Netherlands, in 1990, 1991,
* 1992, 1993, 1994 and 1995.
*
* Redistribution and use in source and binary forms are permitted
* provided that this entire copyright notice is duplicated in all such
* copies.
*
* This software is provided "as is" and without any expressed or implied
* warranties, including, without limitation, the implied warranties of
* merchantibility and fitness for any particular purpose.
************************************************************************/
/* Author: Wietse Venema <[e-mail]> */
/* light changes where done to accomodate libinetutils: Alain Magloire */

#ifdef HAVE_CONFIG_H
# include <config.h>
#endif

#include <sys/types.h>

#ifdef HAVE_SYS_TIME_H
# include <sys/time.h>
#else
# include <time.h>
#endif
#if defined (UTMPX) && defined(HAVE_UTMPX_H)
# define __USE_GNU
# include <utmpx.h>
#else
# include <utmp.h>
#endif
#ifdef HAVE_STRING_H
# include <string.h>
#else
# include <strings.h>
#endif
#include <unistd.h>

/* utmp_init - update utmp and wtmp before login */

void
utmp_init(char   *line, char   *user, char   *id)
{
#ifdef UTMPX
    struct utmpx utx;
#else
    struct utmp utx;
#endif

    memset((char *) &utx, 0, sizeof(utx));
#if defined(HAVE_STRUCT_UTMP_UT_ID)
    strncpy(utx.ut_id, id, sizeof(utx.ut_id));
#endif
#if defined(HAVE_STRUCT_UTMP_UT_USER)
    strncpy(utx.ut_user, user, sizeof(utx.ut_user));
#else
    strncpy(utx.ut_name, user, sizeof(utx.ut_name));
#endif
    strncpy(utx.ut_line, line, sizeof(utx.ut_line));
#if defined(HAVE_STRUCT_UTMP_UT_PID)
    utx.ut_pid = getpid();
#endif
#if defined(HAVE_STRUCT_UTMP_UT_TYPE)
    utx.ut_type = LOGIN_PROCESS;
#endif
#if defined(HAVE_STRUCT_UTMPX_UT_TV)
    gettimeofday(&(utx.ut_tv), 0);
#else
    time(&(utx.ut_time));
#endif
#ifdef UTMPX
    pututxline(&utx);
# ifdef HAVE_UPDWTMPX
    updwtmpx(PATH_WTMPX, &utx);
# endif
    endutxent();
#else
    pututline(&utx);
# ifdef HAVE_UPDWTMP
    updwtmp(PATH_WTMP, &utx);
# else
    logwtmp(line, user, id);
# endif
    endutent();
#endif
}

/* utmp_ptsid - generate utmp id for pseudo terminal */

char   *
utmp_ptsid(char   *line, char   *tag)
{
    static char buf[5];

    strncpy(buf, tag, 2);
    strncpy(buf + 2, line + strlen(line) - 2, 2);
    return (buf);
}