loginrec.c

   1 /*
   2  * Copyright (c) 2000 Andre Lucas.  All rights reserved.
   3  * Portions copyright (c) 1998 Todd C. Miller
   4  * Portions copyright (c) 1996 Jason Downs
   5  * Portions copyright (c) 1996 Theo de Raadt
   6  *
   7  * Redistribution and use in source and binary forms, with or without
   8  * modification, are permitted provided that the following conditions
   9  * are met:
  10  * 1. Redistributions of source code must retain the above copyright
  11  *    notice, this list of conditions and the following disclaimer.
  12  * 2. Redistributions in binary form must reproduce the above copyright
  13  *    notice, this list of conditions and the following disclaimer in the
  14  *    documentation and/or other materials provided with the distribution.
  15  *
  16  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
  17  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
  18  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
  19  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
  20  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
  21  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  22  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  23  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  24  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
  25  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  26  */
  27
  28 /*
  29  * The btmp logging code is derived from login.c from util-linux and is under
  30  * the the following license:
  31  *
  32  * Copyright (c) 1980, 1987, 1988 The Regents of the University of California.
  33  * All rights reserved.
  34  *
  35  * Redistribution and use in source and binary forms are permitted
  36  * provided that the above copyright notice and this paragraph are
  37  * duplicated in all such forms and that any documentation,
  38  * advertising materials, and other materials related to such
  39  * distribution and use acknowledge that the software was developed
  40  * by the University of California, Berkeley.  The name of the
  41  * University may not be used to endorse or promote products derived
  42  * from this software without specific prior written permission.
  43  * THIS SOFTWARE IS PROVIDED ``AS IS'' AND WITHOUT ANY EXPRESS OR
  44  * IMPLIED WARRANTIES, INCLUDING, WITHOUT LIMITATION, THE IMPLIED
  45  * WARRANTIES OF MERCHANTIBILITY AND FITNESS FOR A PARTICULAR PURPOSE.
  46  */
  47
  48
  49 /**
  50  ** loginrec.c:  platform-independent login recording and lastlog retrieval
  51  **/
  52
  53 /*
  54  *  The new login code explained
  55  *  ============================
  56  *
  57  *  This code attempts to provide a common interface to login recording
  58  *  (utmp and friends) and last login time retrieval.
  59  *
  60  *  Its primary means of achieving this is to use 'struct logininfo', a
  61  *  union of all the useful fields in the various different types of
  62  *  system login record structures one finds on UNIX variants.
  63  *
  64  *  We depend on autoconf to define which recording methods are to be
  65  *  used, and which fields are contained in the relevant data structures
  66  *  on the local system. Many C preprocessor symbols affect which code
  67  *  gets compiled here.
  68  *
  69  *  The code is designed to make it easy to modify a particular
  70  *  recording method, without affecting other methods nor requiring so
  71  *  many nested conditional compilation blocks as were commonplace in
  72  *  the old code.
  73  *
  74  *  For login recording, we try to use the local system's libraries as
  75  *  these are clearly most likely to work correctly. For utmp systems
  76  *  this usually means login() and logout() or setutent() etc., probably
  77  *  in libutil, along with logwtmp() etc. On these systems, we fall back
  78  *  to writing the files directly if we have to, though this method
  79  *  requires very thorough testing so we do not corrupt local auditing
  80  *  information. These files and their access methods are very system
  81  *  specific indeed.
  82  *
  83  *  For utmpx systems, the corresponding library functions are
  84  *  setutxent() etc. To the author's knowledge, all utmpx systems have
  85  *  these library functions and so no direct write is attempted. If such
  86  *  a system exists and needs support, direct analogues of the [uw]tmp
  87  *  code should suffice.
  88  *
  89  *  Retrieving the time of last login ('lastlog') is in some ways even
  90  *  more problemmatic than login recording. Some systems provide a
  91  *  simple table of all users which we seek based on uid and retrieve a
  92  *  relatively standard structure. Others record the same information in
  93  *  a directory with a separate file, and others don't record the
  94  *  information separately at all. For systems in the latter category,
  95  *  we look backwards in the wtmp or wtmpx file for the last login entry
  96  *  for our user. Naturally this is slower and on busy systems could
  97  *  incur a significant performance penalty.
  98  *
  99  *  Calling the new code
 100  *  --------------------
 101  *
 102  *  In OpenSSH all login recording and retrieval is performed in
 103  *  login.c. Here you'll find working examples. Also, in the logintest.c
 104  *  program there are more examples.
 105  *
 106  *  Internal handler calling method
 107  *  -------------------------------
 108  *
 109  *  When a call is made to login_login() or login_logout(), both
 110  *  routines set a struct logininfo flag defining which action (log in,
 111  *  or log out) is to be taken. They both then call login_write(), which
 112  *  calls whichever of the many structure-specific handlers autoconf
 113  *  selects for the local system.
 114  *
 115  *  The handlers themselves handle system data structure specifics. Both
 116  *  struct utmp and struct utmpx have utility functions (see
 117  *  construct_utmp*()) to try to make it simpler to add extra systems
 118  *  that introduce new features to either structure.
 119  *
 120  *  While it may seem terribly wasteful to replicate so much similar
 121  *  code for each method, experience has shown that maintaining code to
 122  *  write both struct utmp and utmpx in one function, whilst maintaining
 123  *  support for all systems whether they have library support or not, is
 124  *  a difficult and time-consuming task.
 125  *
 126  *  Lastlog support proceeds similarly. Functions login_get_lastlog()
 127  *  (and its OpenSSH-tuned friend login_get_lastlog_time()) call
 128  *  getlast_entry(), which tries one of three methods to find the last
 129  *  login time. It uses local system lastlog support if it can,
 130  *  otherwise it tries wtmp or wtmpx before giving up and returning 0,
 131  *  meaning "tilt".
 132  *
 133  *  Maintenance
 134  *  -----------
 135  *
 136  *  In many cases it's possible to tweak autoconf to select the correct
 137  *  methods for a particular platform, either by improving the detection
 138  *  code (best), or by presetting DISABLE_<method> or CONF_<method>_FILE
 139  *  symbols for the platform.
 140  *
 141  *  Use logintest to check which symbols are defined before modifying
 142  *  configure.ac and loginrec.c. (You have to build logintest yourself
 143  *  with 'make logintest' as it's not built by default.)
 144  *
 145  *  Otherwise, patches to the specific method(s) are very helpful!
 146  */
 147
 148 #include "includes.h"
 149
 150 #include "ssh.h"
 151 #include "xmalloc.h"
 152 #include "loginrec.h"
 153 #include "log.h"
 154 #include "atomicio.h"
 155 #include "packet.h"
 156 #include "canohost.h"
 157
 158 #ifdef HAVE_UTIL_H
 159 # include <util.h>
 160 #endif
 161
 162 #ifdef HAVE_LIBUTIL_H
 163 # include <libutil.h>
 164 #endif
 165
 166 RCSID("$Id$");
 167
 168 /**
 169  ** prototypes for helper functions in this file
 170  **/
 171
 172 #if HAVE_UTMP_H
 173 void set_utmp_time(struct logininfo *li, struct utmp *ut);
 174 void construct_utmp(struct logininfo *li, struct utmp *ut);
 175 #endif
 176
 177 #ifdef HAVE_UTMPX_H
 178 void set_utmpx_time(struct logininfo *li, struct utmpx *ut);
 179 void construct_utmpx(struct logininfo *li, struct utmpx *ut);
 180 #endif
 181
 182 int utmp_write_entry(struct logininfo *li);
 183 int utmpx_write_entry(struct logininfo *li);
 184 int wtmp_write_entry(struct logininfo *li);
 185 int wtmpx_write_entry(struct logininfo *li);
 186 int lastlog_write_entry(struct logininfo *li);
 187 int syslogin_write_entry(struct logininfo *li);
 188
 189 int getlast_entry(struct logininfo *li);
 190 int lastlog_get_entry(struct logininfo *li);
 191 int wtmp_get_entry(struct logininfo *li);
 192 int wtmpx_get_entry(struct logininfo *li);
 193
 194 /* pick the shortest string */
 195 #define MIN_SIZEOF(s1,s2) (sizeof(s1) < sizeof(s2) ? sizeof(s1) : sizeof(s2))
 196
 197 /**
 198  ** platform-independent login functions
 199  **/
 200
 201 /*
 202  * login_login(struct logininfo *) - Record a login
 203  *
 204  * Call with a pointer to a struct logininfo initialised with
 205  * login_init_entry() or login_alloc_entry()
 206  *
 207  * Returns:
 208  *  >0 if successful
 209  *  0  on failure (will use OpenSSH's logging facilities for diagnostics)
 210  */
 211 int
 212 login_login(struct logininfo *li)
 213 {
 214         li->type = LTYPE_LOGIN;
 215         return (login_write(li));
 216 }
 217
 218
 219 /*
 220  * login_logout(struct logininfo *) - Record a logout
 221  *
 222  * Call as with login_login()
 223  *
 224  * Returns:
 225  *  >0 if successful
 226  *  0  on failure (will use OpenSSH's logging facilities for diagnostics)
 227  */
 228 int
 229 login_logout(struct logininfo *li)
 230 {
 231         li->type = LTYPE_LOGOUT;
 232         return (login_write(li));
 233 }
 234
 235 /*
 236  * login_get_lastlog_time(int) - Retrieve the last login time
 237  *
 238  * Retrieve the last login time for the given uid. Will try to use the
 239  * system lastlog facilities if they are available, but will fall back
 240  * to looking in wtmp/wtmpx if necessary
 241  *
 242  * Returns:
 243  *   0 on failure, or if user has never logged in
 244  *   Time in seconds from the epoch if successful
 245  *
 246  * Useful preprocessor symbols:
 247  *   DISABLE_LASTLOG: If set, *never* even try to retrieve lastlog
 248  *                    info
 249  *   USE_LASTLOG: If set, indicates the presence of system lastlog
 250  *                facilities. If this and DISABLE_LASTLOG are not set,
 251  *                try to retrieve lastlog information from wtmp/wtmpx.
 252  */
 253 unsigned int
 254 login_get_lastlog_time(const int uid)
 255 {
 256         struct logininfo li;
 257
 258         if (login_get_lastlog(&li, uid))
 259                 return (li.tv_sec);
 260         else
 261                 return (0);
 262 }
 263
 264 /*
 265  * login_get_lastlog(struct logininfo *, int)   - Retrieve a lastlog entry
 266  *
 267  * Retrieve a logininfo structure populated (only partially) with
 268  * information from the system lastlog data, or from wtmp/wtmpx if no
 269  * system lastlog information exists.
 270  *
 271  * Note this routine must be given a pre-allocated logininfo.
 272  *
 273  * Returns:
 274  *  >0: A pointer to your struct logininfo if successful
 275  *  0  on failure (will use OpenSSH's logging facilities for diagnostics)
 276  */
 277 struct logininfo *
 278 login_get_lastlog(struct logininfo *li, const int uid)
 279 {
 280         struct passwd *pw;
 281
 282         memset(li, '\0', sizeof(*li));
 283         li->uid = uid;
 284
 285         /*
 286          * If we don't have a 'real' lastlog, we need the username to
 287          * reliably search wtmp(x) for the last login (see
 288          * wtmp_get_entry().)
 289          */
 290         pw = getpwuid(uid);
 291         if (pw == NULL)
 292                 fatal("%s: Cannot find account for uid %i", __func__, uid);
 293
 294         /* No MIN_SIZEOF here - we absolutely *must not* truncate the
 295          * username (XXX - so check for trunc!) */
 296         strlcpy(li->username, pw->pw_name, sizeof(li->username));
 297
 298         if (getlast_entry(li))
 299                 return (li);
 300         else
 301                 return (NULL);
 302 }
 303
 304
 305 /*
 306  * login_alloc_entry(int, char*, char*, char*)    - Allocate and initialise
 307  *                                                  a logininfo structure
 308  *
 309  * This function creates a new struct logininfo, a data structure
 310  * meant to carry the information required to portably record login info.
 311  *
 312  * Returns a pointer to a newly created struct logininfo. If memory
 313  * allocation fails, the program halts.
 314  */
 315 struct
 316 logininfo *login_alloc_entry(int pid, const char *username,
 317     const char *hostname, const char *line)
 318 {
 319         struct logininfo *newli;
 320
 321         newli = xmalloc(sizeof(*newli));
 322         login_init_entry(newli, pid, username, hostname, line);
 323         return (newli);
 324 }
 325
 326
 327 /* login_free_entry(struct logininfo *)    - free struct memory */
 328 void
 329 login_free_entry(struct logininfo *li)
 330 {
 331         xfree(li);
 332 }
 333
 334
 335 /* login_init_entry(struct logininfo *, int, char*, char*, char*)
 336  *                                        - initialise a struct logininfo
 337  *
 338  * Populates a new struct logininfo, a data structure meant to carry
 339  * the information required to portably record login info.
 340  *
 341  * Returns: 1
 342  */
 343 int
 344 login_init_entry(struct logininfo *li, int pid, const char *username,
 345     const char *hostname, const char *line)
 346 {
 347         struct passwd *pw;
 348
 349         memset(li, 0, sizeof(*li));
 350
 351         li->pid = pid;
 352
 353         /* set the line information */
 354         if (line)
 355                 line_fullname(li->line, line, sizeof(li->line));
 356
 357         if (username) {
 358                 strlcpy(li->username, username, sizeof(li->username));
 359                 pw = getpwnam(li->username);
 360                 if (pw == NULL) {
 361                         fatal("%s: Cannot find user \"%s\"", __func__,
 362                             li->username);
 363                 }
 364                 li->uid = pw->pw_uid;
 365         }
 366
 367         if (hostname)
 368                 strlcpy(li->hostname, hostname, sizeof(li->hostname));
 369
 370         return (1);
 371 }
 372
 373 /*
 374  * login_set_current_time(struct logininfo *)    - set the current time
 375  *
 376  * Set the current time in a logininfo structure. This function is
 377  * meant to eliminate the need to deal with system dependencies for
 378  * time handling.
 379  */
 380 void
 381 login_set_current_time(struct logininfo *li)
 382 {
 383         struct timeval tv;
 384
 385         gettimeofday(&tv, NULL);
 386
 387         li->tv_sec = tv.tv_sec;
 388         li->tv_usec = tv.tv_usec;
 389 }
 390
 391 /* copy a sockaddr_* into our logininfo */
 392 void
 393 login_set_addr(struct logininfo *li, const struct sockaddr *sa,
 394     const unsigned int sa_size)
 395 {
 396         unsigned int bufsize = sa_size;
 397
 398         /* make sure we don't overrun our union */
 399         if (sizeof(li->hostaddr) < sa_size)
 400                 bufsize = sizeof(li->hostaddr);
 401
 402         memcpy(&li->hostaddr.sa, sa, bufsize);
 403 }
 404
 405
 406 /**
 407  ** login_write: Call low-level recording functions based on autoconf
 408  ** results
 409  **/
 410 int
 411 login_write(struct logininfo *li)
 412 {
 413 #ifndef HAVE_CYGWIN
 414         if (geteuid() != 0) {
 415                 logit("Attempt to write login records by non-root user (aborting)");
 416                 return (1);
 417         }
 418 #endif
 419
 420         /* set the timestamp */
 421         login_set_current_time(li);
 422 #ifdef USE_LOGIN
 423         syslogin_write_entry(li);
 424 #endif
 425 #ifdef USE_LASTLOG
 426         if (li->type == LTYPE_LOGIN)
 427                 lastlog_write_entry(li);
 428 #endif
 429 #ifdef USE_UTMP
 430         utmp_write_entry(li);
 431 #endif
 432 #ifdef USE_WTMP
 433         wtmp_write_entry(li);
 434 #endif
 435 #ifdef USE_UTMPX
 436         utmpx_write_entry(li);
 437 #endif
 438 #ifdef USE_WTMPX
 439         wtmpx_write_entry(li);
 440 #endif
 441 #ifdef CUSTOM_SYS_AUTH_RECORD_LOGIN
 442         if (li->type == LTYPE_LOGIN &&
 443            !sys_auth_record_login(li->username,li->hostname,li->line))
 444                 logit("Writing login record failed for %s", li->username);
 445 #endif
 446         return (0);
 447 }
 448
 449 #ifdef LOGIN_NEEDS_UTMPX
 450 int
 451 login_utmp_only(struct logininfo *li)
 452 {
 453         li->type = LTYPE_LOGIN;
 454         login_set_current_time(li);
 455 # ifdef USE_UTMP
 456         utmp_write_entry(li);
 457 # endif
 458 # ifdef USE_WTMP
 459         wtmp_write_entry(li);
 460 # endif
 461 # ifdef USE_UTMPX
 462         utmpx_write_entry(li);
 463 # endif
 464 # ifdef USE_WTMPX
 465         wtmpx_write_entry(li);
 466 # endif
 467         return (0);
 468 }
 469 #endif
 470
 471 /**
 472  ** getlast_entry: Call low-level functions to retrieve the last login
 473  **                time.
 474  **/
 475
 476 /* take the uid in li and return the last login time */
 477 int
 478 getlast_entry(struct logininfo *li)
 479 {
 480 #ifdef USE_LASTLOG
 481         return(lastlog_get_entry(li));
 482 #else /* !USE_LASTLOG */
 483
 484 #if defined(DISABLE_LASTLOG)
 485         /* On some systems we shouldn't even try to obtain last login
 486          * time, e.g. AIX */
 487         return (0);
 488 # elif defined(USE_WTMP) && \
 489     (defined(HAVE_TIME_IN_UTMP) || defined(HAVE_TV_IN_UTMP))
 490         /* retrieve last login time from utmp */
 491         return (wtmp_get_entry(li));
 492 # elif defined(USE_WTMPX) && \
 493     (defined(HAVE_TIME_IN_UTMPX) || defined(HAVE_TV_IN_UTMPX))
 494         /* If wtmp isn't available, try wtmpx */
 495         return (wtmpx_get_entry(li));
 496 # else
 497         /* Give up: No means of retrieving last login time */
 498         return (0);
 499 # endif /* DISABLE_LASTLOG */
 500 #endif /* USE_LASTLOG */
 501 }
 502
 503
 504
 505 /*
 506  * 'line' string utility functions
 507  *
 508  * These functions process the 'line' string into one of three forms:
 509  *
 510  * 1. The full filename (including '/dev')
 511  * 2. The stripped name (excluding '/dev')
 512  * 3. The abbreviated name (e.g. /dev/ttyp00 -> yp00
 513  *                               /dev/pts/1  -> ts/1 )
 514  *
 515  * Form 3 is used on some systems to identify a .tmp.? entry when
 516  * attempting to remove it. Typically both addition and removal is
 517  * performed by one application - say, sshd - so as long as the choice
 518  * uniquely identifies a terminal it's ok.
 519  */
 520
 521
 522 /*
 523  * line_fullname(): add the leading '/dev/' if it doesn't exist make
 524  * sure dst has enough space, if not just copy src (ugh)
 525  */
 526 char *
 527 line_fullname(char *dst, const char *src, int dstsize)
 528 {
 529         memset(dst, '\0', dstsize);
 530         if ((strncmp(src, "/dev/", 5) == 0) || (dstsize < (strlen(src) + 5)))
 531                 strlcpy(dst, src, dstsize);
 532         else {
 533                 strlcpy(dst, "/dev/", dstsize);
 534                 strlcat(dst, src, dstsize);
 535         }
 536         return (dst);
 537 }
 538
 539 /* line_stripname(): strip the leading '/dev' if it exists, return dst */
 540 char *
 541 line_stripname(char *dst, const char *src, int dstsize)
 542 {
 543         memset(dst, '\0', dstsize);
 544         if (strncmp(src, "/dev/", 5) == 0)
 545                 strlcpy(dst, src + 5, dstsize);
 546         else
 547                 strlcpy(dst, src, dstsize);
 548         return (dst);
 549 }
 550
 551 /*
 552  * line_abbrevname(): Return the abbreviated (usually four-character)
 553  * form of the line (Just use the last <dstsize> characters of the
 554  * full name.)
 555  *
 556  * NOTE: use strncpy because we do NOT necessarily want zero
 557  * termination
 558  */
 559 char *
 560 line_abbrevname(char *dst, const char *src, int dstsize)
 561 {
 562         size_t len;
 563
 564         memset(dst, '\0', dstsize);
 565
 566         /* Always skip prefix if present */
 567         if (strncmp(src, "/dev/", 5) == 0)
 568                 src += 5;
 569
 570 #ifdef WITH_ABBREV_NO_TTY
 571         if (strncmp(src, "tty", 3) == 0)
 572                 src += 3;
 573 #endif
 574
 575         len = strlen(src);
 576
 577         if (len > 0) {
 578                 if (((int)len - dstsize) > 0)
 579                         src +=  ((int)len - dstsize);
 580
 581                 /* note: _don't_ change this to strlcpy */
 582                 strncpy(dst, src, (size_t)dstsize);
 583         }
 584
 585         return (dst);
 586 }
 587
 588 /**
 589  ** utmp utility functions
 590  **
 591  ** These functions manipulate struct utmp, taking system differences
 592  ** into account.
 593  **/
 594
 595 #if defined(USE_UTMP) || defined (USE_WTMP) || defined (USE_LOGIN)
 596
 597 /* build the utmp structure */
 598 void
 599 set_utmp_time(struct logininfo *li, struct utmp *ut)
 600 {
 601 # if defined(HAVE_TV_IN_UTMP)
 602         ut->ut_tv.tv_sec = li->tv_sec;
 603         ut->ut_tv.tv_usec = li->tv_usec;
 604 # elif defined(HAVE_TIME_IN_UTMP)
 605         ut->ut_time = li->tv_sec;
 606 # endif
 607 }
 608
 609 void
 610 construct_utmp(struct logininfo *li,
 611                     struct utmp *ut)
 612 {
 613 # ifdef HAVE_ADDR_V6_IN_UTMP
 614         struct sockaddr_in6 *sa6;
 615 # endif
 616
 617         memset(ut, '\0', sizeof(*ut));
 618
 619         /* First fill out fields used for both logins and logouts */
 620
 621 # ifdef HAVE_ID_IN_UTMP
 622         line_abbrevname(ut->ut_id, li->line, sizeof(ut->ut_id));
 623 # endif
 624
 625 # ifdef HAVE_TYPE_IN_UTMP
 626         /* This is done here to keep utmp constants out of struct logininfo */
 627         switch (li->type) {
 628         case LTYPE_LOGIN:
 629                 ut->ut_type = USER_PROCESS;
 630 #ifdef _UNICOS
 631                 cray_set_tmpdir(ut);
 632 #endif
 633                 break;
 634         case LTYPE_LOGOUT:
 635                 ut->ut_type = DEAD_PROCESS;
 636 #ifdef _UNICOS
 637                 cray_retain_utmp(ut, li->pid);
 638 #endif
 639                 break;
 640         }
 641 # endif
 642         set_utmp_time(li, ut);
 643
 644         line_stripname(ut->ut_line, li->line, sizeof(ut->ut_line));
 645
 646 # ifdef HAVE_PID_IN_UTMP
 647         ut->ut_pid = li->pid;
 648 # endif
 649
 650         /* If we're logging out, leave all other fields blank */
 651         if (li->type == LTYPE_LOGOUT)
 652                 return;
 653
 654         /*
 655          * These fields are only used when logging in, and are blank
 656          * for logouts.
 657          */
 658
 659         /* Use strncpy because we don't necessarily want null termination */
 660         strncpy(ut->ut_name, li->username,
 661             MIN_SIZEOF(ut->ut_name, li->username));
 662 # ifdef HAVE_HOST_IN_UTMP
 663         strncpy(ut->ut_host, li->hostname,
 664             MIN_SIZEOF(ut->ut_host, li->hostname));
 665 # endif
 666 # ifdef HAVE_ADDR_IN_UTMP
 667         /* this is just a 32-bit IP address */
 668         if (li->hostaddr.sa.sa_family == AF_INET)
 669                 ut->ut_addr = li->hostaddr.sa_in.sin_addr.s_addr;
 670 # endif
 671 # ifdef HAVE_ADDR_V6_IN_UTMP
 672         /* this is just a 128-bit IPv6 address */
 673         if (li->hostaddr.sa.sa_family == AF_INET6) {
 674                 sa6 = ((struct sockaddr_in6 *)&li->hostaddr.sa);
 675                 memcpy(ut->ut_addr_v6, sa6->sin6_addr.s6_addr, 16);
 676                 if (IN6_IS_ADDR_V4MAPPED(&sa6->sin6_addr)) {
 677                         ut->ut_addr_v6[0] = ut->ut_addr_v6[3];
 678                         ut->ut_addr_v6[1] = 0;
 679                         ut->ut_addr_v6[2] = 0;
 680                         ut->ut_addr_v6[3] = 0;
 681                 }
 682         }
 683 # endif
 684 }
 685 #endif /* USE_UTMP || USE_WTMP || USE_LOGIN */
 686
 687 /**
 688  ** utmpx utility functions
 689  **
 690  ** These functions manipulate struct utmpx, accounting for system
 691  ** variations.
 692  **/
 693
 694 #if defined(USE_UTMPX) || defined (USE_WTMPX)
 695 /* build the utmpx structure */
 696 void
 697 set_utmpx_time(struct logininfo *li, struct utmpx *utx)
 698 {
 699 # if defined(HAVE_TV_IN_UTMPX)
 700         utx->ut_tv.tv_sec = li->tv_sec;
 701         utx->ut_tv.tv_usec = li->tv_usec;
 702 # elif defined(HAVE_TIME_IN_UTMPX)
 703         utx->ut_time = li->tv_sec;
 704 # endif
 705 }
 706
 707 void
 708 construct_utmpx(struct logininfo *li, struct utmpx *utx)
 709 {
 710 # ifdef HAVE_ADDR_V6_IN_UTMP
 711         struct sockaddr_in6 *sa6;
 712 #  endif
 713         memset(utx, '\0', sizeof(*utx));
 714
 715 # ifdef HAVE_ID_IN_UTMPX
 716         line_abbrevname(utx->ut_id, li->line, sizeof(utx->ut_id));
 717 # endif
 718
 719         /* this is done here to keep utmp constants out of loginrec.h */
 720         switch (li->type) {
 721         case LTYPE_LOGIN:
 722                 utx->ut_type = USER_PROCESS;
 723                 break;
 724         case LTYPE_LOGOUT:
 725                 utx->ut_type = DEAD_PROCESS;
 726                 break;
 727         }
 728         line_stripname(utx->ut_line, li->line, sizeof(utx->ut_line));
 729         set_utmpx_time(li, utx);
 730         utx->ut_pid = li->pid;
 731
 732         /* strncpy(): Don't necessarily want null termination */
 733         strncpy(utx->ut_name, li->username,
 734             MIN_SIZEOF(utx->ut_name, li->username));
 735
 736         if (li->type == LTYPE_LOGOUT)
 737                 return;
 738
 739         /*
 740          * These fields are only used when logging in, and are blank
 741          * for logouts.
 742          */
 743
 744 # ifdef HAVE_HOST_IN_UTMPX
 745         strncpy(utx->ut_host, li->hostname,
 746             MIN_SIZEOF(utx->ut_host, li->hostname));
 747 # endif
 748 # ifdef HAVE_ADDR_IN_UTMPX
 749         /* this is just a 32-bit IP address */
 750         if (li->hostaddr.sa.sa_family == AF_INET)
 751                 utx->ut_addr = li->hostaddr.sa_in.sin_addr.s_addr;
 752 # endif
 753 # ifdef HAVE_ADDR_V6_IN_UTMP
 754         /* this is just a 128-bit IPv6 address */
 755         if (li->hostaddr.sa.sa_family == AF_INET6) {
 756                 sa6 = ((struct sockaddr_in6 *)&li->hostaddr.sa);
 757                 memcpy(ut->ut_addr_v6, sa6->sin6_addr.s6_addr, 16);
 758                 if (IN6_IS_ADDR_V4MAPPED(&sa6->sin6_addr)) {
 759                         ut->ut_addr_v6[0] = ut->ut_addr_v6[3];
 760                         ut->ut_addr_v6[1] = 0;
 761                         ut->ut_addr_v6[2] = 0;
 762                         ut->ut_addr_v6[3] = 0;
 763                 }
 764         }
 765 # endif
 766 # ifdef HAVE_SYSLEN_IN_UTMPX
 767         /* ut_syslen is the length of the utx_host string */
 768         utx->ut_syslen = MIN(strlen(li->hostname), sizeof(utx->ut_host));
 769 # endif
 770 }
 771 #endif /* USE_UTMPX || USE_WTMPX */
 772
 773 /**
 774  ** Low-level utmp functions
 775  **/
 776
 777 /* FIXME: (ATL) utmp_write_direct needs testing */
 778 #ifdef USE_UTMP
 779
 780 /* if we can, use pututline() etc. */
 781 # if !defined(DISABLE_PUTUTLINE) && defined(HAVE_SETUTENT) && \
 782         defined(HAVE_PUTUTLINE)
 783 #  define UTMP_USE_LIBRARY
 784 # endif
 785
 786
 787 /* write a utmp entry with the system's help (pututline() and pals) */
 788 # ifdef UTMP_USE_LIBRARY
 789 static int
 790 utmp_write_library(struct logininfo *li, struct utmp *ut)
 791 {
 792         setutent();
 793         pututline(ut);
 794 #  ifdef HAVE_ENDUTENT
 795         endutent();
 796 #  endif
 797         return (1);
 798 }
 799 # else /* UTMP_USE_LIBRARY */
 800
 801 /*
 802  * Write a utmp entry direct to the file
 803  * This is a slightly modification of code in OpenBSD's login.c
 804  */
 805 static int
 806 utmp_write_direct(struct logininfo *li, struct utmp *ut)
 807 {
 808         struct utmp old_ut;
 809         register int fd;
 810         int tty;
 811
 812         /* FIXME: (ATL) ttyslot() needs local implementation */
 813
 814 #if defined(HAVE_GETTTYENT)
 815         struct ttyent *ty;
 816
 817         tty=0;
 818         setttyent();
 819         while (NULL != (ty = getttyent())) {
 820                 tty++;
 821                 if (!strncmp(ty->ty_name, ut->ut_line, sizeof(ut->ut_line)))
 822                         break;
 823         }
 824         endttyent();
 825
 826         if (NULL == ty) {
 827                 logit("%s: tty not found", __func__);
 828                 return (0);
 829         }
 830 #else /* FIXME */
 831
 832         tty = ttyslot(); /* seems only to work for /dev/ttyp? style names */
 833
 834 #endif /* HAVE_GETTTYENT */
 835
 836         if (tty > 0 && (fd = open(UTMP_FILE, O_RDWR|O_CREAT, 0644)) >= 0) {
 837                 off_t pos, ret;
 838
 839                 pos = (off_t)tty * sizeof(struct utmp);
 840                 if ((ret = lseek(fd, pos, SEEK_SET)) == -1) {
 841                         logit("%s: lseek: %s", __func__, strerror(errno));
 842                         return (0);
 843                 }
 844                 if (ret != pos) {
 845                         logit("%s: Couldn't seek to tty %d slot in %s",
 846                             __func__, tty, UTMP_FILE);
 847                         return (0);
 848                 }
 849                 /*
 850                  * Prevent luser from zero'ing out ut_host.
 851                  * If the new ut_line is empty but the old one is not
 852                  * and ut_line and ut_name match, preserve the old ut_line.
 853                  */
 854                 if (atomicio(read, fd, &old_ut, sizeof(old_ut)) == sizeof(old_ut) &&
 855                     (ut->ut_host[0] == '\0') && (old_ut.ut_host[0] != '\0') &&
 856                     (strncmp(old_ut.ut_line, ut->ut_line, sizeof(ut->ut_line)) == 0) &&
 857                     (strncmp(old_ut.ut_name, ut->ut_name, sizeof(ut->ut_name)) == 0))
 858                         memcpy(ut->ut_host, old_ut.ut_host, sizeof(ut->ut_host));
 859
 860                 if ((ret = lseek(fd, pos, SEEK_SET)) == -1) {
 861                         logit("%s: lseek: %s", __func__, strerror(errno));
 862                         return (0);
 863                 }
 864                 if (ret != pos) {
 865                         logit("%s: Couldn't seek to tty %d slot in %s",
 866                             __func__, tty, UTMP_FILE);
 867                         return (0);
 868                 }
 869                 if (atomicio(vwrite, fd, ut, sizeof(*ut)) != sizeof(*ut)) {
 870                         logit("%s: error writing %s: %s", __func__,
 871                             UTMP_FILE, strerror(errno));
 872                 }
 873
 874                 close(fd);
 875                 return (1);
 876         } else {
 877                 return (0);
 878         }
 879 }
 880 # endif /* UTMP_USE_LIBRARY */
 881
 882 static int
 883 utmp_perform_login(struct logininfo *li)
 884 {
 885         struct utmp ut;
 886
 887         construct_utmp(li, &ut);
 888 # ifdef UTMP_USE_LIBRARY
 889         if (!utmp_write_library(li, &ut)) {
 890                 logit("%s: utmp_write_library() failed", __func__);
 891                 return (0);
 892         }
 893 # else
 894         if (!utmp_write_direct(li, &ut)) {
 895                 logit("%s: utmp_write_direct() failed", __func__);
 896                 return (0);
 897         }
 898 # endif
 899         return (1);
 900 }
 901
 902
 903 static int
 904 utmp_perform_logout(struct logininfo *li)
 905 {
 906         struct utmp ut;
 907
 908         construct_utmp(li, &ut);
 909 # ifdef UTMP_USE_LIBRARY
 910         if (!utmp_write_library(li, &ut)) {
 911                 logit("%s: utmp_write_library() failed", __func__);
 912                 return (0);
 913         }
 914 # else
 915         if (!utmp_write_direct(li, &ut)) {
 916                 logit("%s: utmp_write_direct() failed", __func__);
 917                 return (0);
 918         }
 919 # endif
 920         return (1);
 921 }
 922
 923
 924 int
 925 utmp_write_entry(struct logininfo *li)
 926 {
 927         switch(li->type) {
 928         case LTYPE_LOGIN:
 929                 return (utmp_perform_login(li));
 930
 931         case LTYPE_LOGOUT:
 932                 return (utmp_perform_logout(li));
 933
 934         default:
 935                 logit("%s: invalid type field", __func__);
 936                 return (0);
 937         }
 938 }
 939 #endif /* USE_UTMP */
 940
 941
 942 /**
 943  ** Low-level utmpx functions
 944  **/
 945
 946 /* not much point if we don't want utmpx entries */
 947 #ifdef USE_UTMPX
 948
 949 /* if we have the wherewithall, use pututxline etc. */
 950 # if !defined(DISABLE_PUTUTXLINE) && defined(HAVE_SETUTXENT) && \
 951         defined(HAVE_PUTUTXLINE)
 952 #  define UTMPX_USE_LIBRARY
 953 # endif
 954
 955
 956 /* write a utmpx entry with the system's help (pututxline() and pals) */
 957 # ifdef UTMPX_USE_LIBRARY
 958 static int
 959 utmpx_write_library(struct logininfo *li, struct utmpx *utx)
 960 {
 961         setutxent();
 962         pututxline(utx);
 963
 964 #  ifdef HAVE_ENDUTXENT
 965         endutxent();
 966 #  endif
 967         return (1);
 968 }
 969
 970 # else /* UTMPX_USE_LIBRARY */
 971
 972 /* write a utmp entry direct to the file */
 973 static int
 974 utmpx_write_direct(struct logininfo *li, struct utmpx *utx)
 975 {
 976         logit("%s: not implemented!", __func__);
 977         return (0);
 978 }
 979 # endif /* UTMPX_USE_LIBRARY */
 980
 981 static int
 982 utmpx_perform_login(struct logininfo *li)
 983 {
 984         struct utmpx utx;
 985
 986         construct_utmpx(li, &utx);
 987 # ifdef UTMPX_USE_LIBRARY
 988         if (!utmpx_write_library(li, &utx)) {
 989                 logit("%s: utmp_write_library() failed", __func__);
 990                 return (0);
 991         }
 992 # else
 993         if (!utmpx_write_direct(li, &ut)) {
 994                 logit("%s: utmp_write_direct() failed", __func__);
 995                 return (0);
 996         }
 997 # endif
 998         return (1);
 999 }
1000
1001
1002 static int
1003 utmpx_perform_logout(struct logininfo *li)
1004 {
1005         struct utmpx utx;
1006
1007         construct_utmpx(li, &utx);
1008 # ifdef HAVE_ID_IN_UTMPX
1009         line_abbrevname(utx.ut_id, li->line, sizeof(utx.ut_id));
1010 # endif
1011 # ifdef HAVE_TYPE_IN_UTMPX
1012         utx.ut_type = DEAD_PROCESS;
1013 # endif
1014
1015 # ifdef UTMPX_USE_LIBRARY
1016         utmpx_write_library(li, &utx);
1017 # else
1018         utmpx_write_direct(li, &utx);
1019 # endif
1020         return (1);
1021 }
1022
1023 int
1024 utmpx_write_entry(struct logininfo *li)
1025 {
1026         switch(li->type) {
1027         case LTYPE_LOGIN:
1028                 return (utmpx_perform_login(li));
1029         case LTYPE_LOGOUT:
1030                 return (utmpx_perform_logout(li));
1031         default:
1032                 logit("%s: invalid type field", __func__);
1033                 return (0);
1034         }
1035 }
1036 #endif /* USE_UTMPX */
1037
1038
1039 /**
1040  ** Low-level wtmp functions
1041  **/
1042
1043 #ifdef USE_WTMP
1044
1045 /*
1046  * Write a wtmp entry direct to the end of the file
1047  * This is a slight modification of code in OpenBSD's logwtmp.c
1048  */
1049 static int
1050 wtmp_write(struct logininfo *li, struct utmp *ut)
1051 {
1052         struct stat buf;
1053         int fd, ret = 1;
1054
1055         if ((fd = open(WTMP_FILE, O_WRONLY|O_APPEND, 0)) < 0) {
1056                 logit("%s: problem writing %s: %s", __func__,
1057                     WTMP_FILE, strerror(errno));
1058                 return (0);
1059         }
1060         if (fstat(fd, &buf) == 0)
1061                 if (atomicio(vwrite, fd, ut, sizeof(*ut)) != sizeof(*ut)) {
1062                         ftruncate(fd, buf.st_size);
1063                         logit("%s: problem writing %s: %s", __func__,
1064                             WTMP_FILE, strerror(errno));
1065                         ret = 0;
1066                 }
1067         close(fd);
1068         return (ret);
1069 }
1070
1071 static int
1072 wtmp_perform_login(struct logininfo *li)
1073 {
1074         struct utmp ut;
1075
1076         construct_utmp(li, &ut);
1077         return (wtmp_write(li, &ut));
1078 }
1079
1080
1081 static int
1082 wtmp_perform_logout(struct logininfo *li)
1083 {
1084         struct utmp ut;
1085
1086         construct_utmp(li, &ut);
1087         return (wtmp_write(li, &ut));
1088 }
1089
1090
1091 int
1092 wtmp_write_entry(struct logininfo *li)
1093 {
1094         switch(li->type) {
1095         case LTYPE_LOGIN:
1096                 return (wtmp_perform_login(li));
1097         case LTYPE_LOGOUT:
1098                 return (wtmp_perform_logout(li));
1099         default:
1100                 logit("%s: invalid type field", __func__);
1101                 return (0);
1102         }
1103 }
1104
1105
1106 /*
1107  * Notes on fetching login data from wtmp/wtmpx
1108  *
1109  * Logouts are usually recorded with (amongst other things) a blank
1110  * username on a given tty line.  However, some systems (HP-UX is one)
1111  * leave all fields set, but change the ut_type field to DEAD_PROCESS.
1112  *
1113  * Since we're only looking for logins here, we know that the username
1114  * must be set correctly. On systems that leave it in, we check for
1115  * ut_type==USER_PROCESS (indicating a login.)
1116  *
1117  * Portability: Some systems may set something other than USER_PROCESS
1118  * to indicate a login process. I don't know of any as I write. Also,
1119  * it's possible that some systems may both leave the username in
1120  * place and not have ut_type.
1121  */
1122
1123 /* return true if this wtmp entry indicates a login */
1124 static int
1125 wtmp_islogin(struct logininfo *li, struct utmp *ut)
1126 {
1127         if (strncmp(li->username, ut->ut_name,
1128             MIN_SIZEOF(li->username, ut->ut_name)) == 0) {
1129 # ifdef HAVE_TYPE_IN_UTMP
1130                 if (ut->ut_type & USER_PROCESS)
1131                         return (1);
1132 # else
1133                 return (1);
1134 # endif
1135         }
1136         return (0);
1137 }
1138
1139 int
1140 wtmp_get_entry(struct logininfo *li)
1141 {
1142         struct stat st;
1143         struct utmp ut;
1144         int fd, found = 0;
1145
1146         /* Clear the time entries in our logininfo */
1147         li->tv_sec = li->tv_usec = 0;
1148
1149         if ((fd = open(WTMP_FILE, O_RDONLY)) < 0) {
1150                 logit("%s: problem opening %s: %s", __func__,
1151                     WTMP_FILE, strerror(errno));
1152                 return (0);
1153         }
1154         if (fstat(fd, &st) != 0) {
1155                 logit("%s: couldn't stat %s: %s", __func__,
1156                     WTMP_FILE, strerror(errno));
1157                 close(fd);
1158                 return (0);
1159         }
1160
1161         /* Seek to the start of the last struct utmp */
1162         if (lseek(fd, -(off_t)sizeof(struct utmp), SEEK_END) == -1) {
1163                 /* Looks like we've got a fresh wtmp file */
1164                 close(fd);
1165                 return (0);
1166         }
1167
1168         while (!found) {
1169                 if (atomicio(read, fd, &ut, sizeof(ut)) != sizeof(ut)) {
1170                         logit("%s: read of %s failed: %s", __func__,
1171                             WTMP_FILE, strerror(errno));
1172                         close (fd);
1173                         return (0);
1174                 }
1175                 if ( wtmp_islogin(li, &ut) ) {
1176                         found = 1;
1177                         /*
1178                          * We've already checked for a time in struct
1179                          * utmp, in login_getlast()
1180                          */
1181 # ifdef HAVE_TIME_IN_UTMP
1182                         li->tv_sec = ut.ut_time;
1183 # else
1184 #  if HAVE_TV_IN_UTMP
1185                         li->tv_sec = ut.ut_tv.tv_sec;
1186 #  endif
1187 # endif
1188                         line_fullname(li->line, ut.ut_line,
1189                             MIN_SIZEOF(li->line, ut.ut_line));
1190 # ifdef HAVE_HOST_IN_UTMP
1191                         strlcpy(li->hostname, ut.ut_host,
1192                             MIN_SIZEOF(li->hostname, ut.ut_host));
1193 # endif
1194                         continue;
1195                 }
1196                 /* Seek back 2 x struct utmp */
1197                 if (lseek(fd, -(off_t)(2 * sizeof(struct utmp)), SEEK_CUR) == -1) {
1198                         /* We've found the start of the file, so quit */
1199                         close(fd);
1200                         return (0);
1201                 }
1202         }
1203
1204         /* We found an entry. Tidy up and return */
1205         close(fd);
1206         return (1);
1207 }
1208 # endif /* USE_WTMP */
1209
1210
1211 /**
1212  ** Low-level wtmpx functions
1213  **/
1214
1215 #ifdef USE_WTMPX
1216 /*
1217  * Write a wtmpx entry direct to the end of the file
1218  * This is a slight modification of code in OpenBSD's logwtmp.c
1219  */
1220 static int
1221 wtmpx_write(struct logininfo *li, struct utmpx *utx)
1222 {
1223 #ifndef HAVE_UPDWTMPX
1224         struct stat buf;
1225         int fd, ret = 1;
1226
1227         if ((fd = open(WTMPX_FILE, O_WRONLY|O_APPEND, 0)) < 0) {
1228                 logit("%s: problem opening %s: %s", __func__,
1229                     WTMPX_FILE, strerror(errno));
1230                 return (0);
1231         }
1232
1233         if (fstat(fd, &buf) == 0)
1234                 if (atomicio(vwrite, fd, utx, sizeof(*utx)) != sizeof(*utx)) {
1235                         ftruncate(fd, buf.st_size);
1236                         logit("%s: problem writing %s: %s", __func__,
1237                             WTMPX_FILE, strerror(errno));
1238                         ret = 0;
1239                 }
1240         close(fd);
1241
1242         return (ret);
1243 #else
1244         updwtmpx(WTMPX_FILE, utx);
1245         return (1);
1246 #endif
1247 }
1248
1249
1250 static int
1251 wtmpx_perform_login(struct logininfo *li)
1252 {
1253         struct utmpx utx;
1254
1255         construct_utmpx(li, &utx);
1256         return (wtmpx_write(li, &utx));
1257 }
1258
1259
1260 static int
1261 wtmpx_perform_logout(struct logininfo *li)
1262 {
1263         struct utmpx utx;
1264
1265         construct_utmpx(li, &utx);
1266         return (wtmpx_write(li, &utx));
1267 }
1268
1269
1270 int
1271 wtmpx_write_entry(struct logininfo *li)
1272 {
1273         switch(li->type) {
1274         case LTYPE_LOGIN:
1275                 return (wtmpx_perform_login(li));
1276         case LTYPE_LOGOUT:
1277                 return (wtmpx_perform_logout(li));
1278         default:
1279                 logit("%s: invalid type field", __func__);
1280                 return (0);
1281         }
1282 }
1283
1284 /* Please see the notes above wtmp_islogin() for information about the
1285    next two functions */
1286
1287 /* Return true if this wtmpx entry indicates a login */
1288 static int
1289 wtmpx_islogin(struct logininfo *li, struct utmpx *utx)
1290 {
1291         if (strncmp(li->username, utx->ut_name,
1292             MIN_SIZEOF(li->username, utx->ut_name)) == 0 ) {
1293 # ifdef HAVE_TYPE_IN_UTMPX
1294                 if (utx->ut_type == USER_PROCESS)
1295                         return (1);
1296 # else
1297                 return (1);
1298 # endif
1299         }
1300         return (0);
1301 }
1302
1303
1304 int
1305 wtmpx_get_entry(struct logininfo *li)
1306 {
1307         struct stat st;
1308         struct utmpx utx;
1309         int fd, found=0;
1310
1311         /* Clear the time entries */
1312         li->tv_sec = li->tv_usec = 0;
1313
1314         if ((fd = open(WTMPX_FILE, O_RDONLY)) < 0) {
1315                 logit("%s: problem opening %s: %s", __func__,
1316                     WTMPX_FILE, strerror(errno));
1317                 return (0);
1318         }
1319         if (fstat(fd, &st) != 0) {
1320                 logit("%s: couldn't stat %s: %s", __func__,
1321                     WTMPX_FILE, strerror(errno));
1322                 close(fd);
1323                 return (0);
1324         }
1325
1326         /* Seek to the start of the last struct utmpx */
1327         if (lseek(fd, -(off_t)sizeof(struct utmpx), SEEK_END) == -1 ) {
1328                 /* probably a newly rotated wtmpx file */
1329                 close(fd);
1330                 return (0);
1331         }
1332
1333         while (!found) {
1334                 if (atomicio(read, fd, &utx, sizeof(utx)) != sizeof(utx)) {
1335                         logit("%s: read of %s failed: %s", __func__,
1336                             WTMPX_FILE, strerror(errno));
1337                         close (fd);
1338                         return (0);
1339                 }
1340                 /*
1341                  * Logouts are recorded as a blank username on a particular
1342                  * line. So, we just need to find the username in struct utmpx
1343                  */
1344                 if (wtmpx_islogin(li, &utx)) {
1345                         found = 1;
1346 # if defined(HAVE_TV_IN_UTMPX)
1347                         li->tv_sec = utx.ut_tv.tv_sec;
1348 # elif defined(HAVE_TIME_IN_UTMPX)
1349                         li->tv_sec = utx.ut_time;
1350 # endif
1351                         line_fullname(li->line, utx.ut_line, sizeof(li->line));
1352 # if defined(HAVE_HOST_IN_UTMPX)
1353                         strlcpy(li->hostname, utx.ut_host,
1354                             MIN_SIZEOF(li->hostname, utx.ut_host));
1355 # endif
1356                         continue;
1357                 }
1358                 if (lseek(fd, -(off_t)(2 * sizeof(struct utmpx)), SEEK_CUR) == -1) {
1359                         close(fd);
1360                         return (0);
1361                 }
1362         }
1363
1364         close(fd);
1365         return (1);
1366 }
1367 #endif /* USE_WTMPX */
1368
1369 /**
1370  ** Low-level libutil login() functions
1371  **/
1372
1373 #ifdef USE_LOGIN
1374 static int
1375 syslogin_perform_login(struct logininfo *li)
1376 {
1377         struct utmp *ut;
1378
1379         ut = xmalloc(sizeof(*ut));
1380         construct_utmp(li, ut);
1381         login(ut);
1382         free(ut);
1383
1384         return (1);
1385 }
1386
1387 static int
1388 syslogin_perform_logout(struct logininfo *li)
1389 {
1390 # ifdef HAVE_LOGOUT
1391         char line[UT_LINESIZE];
1392
1393         (void)line_stripname(line, li->line, sizeof(line));
1394
1395         if (!logout(line))
1396                 logit("%s: logout() returned an error", __func__);
1397 #  ifdef HAVE_LOGWTMP
1398         else
1399                 logwtmp(line, "", "");
1400 #  endif
1401         /* FIXME: (ATL - if the need arises) What to do if we have
1402          * login, but no logout?  what if logout but no logwtmp? All
1403          * routines are in libutil so they should all be there,
1404          * but... */
1405 # endif
1406         return (1);
1407 }
1408
1409 int
1410 syslogin_write_entry(struct logininfo *li)
1411 {
1412         switch (li->type) {
1413         case LTYPE_LOGIN:
1414                 return (syslogin_perform_login(li));
1415         case LTYPE_LOGOUT:
1416                 return (syslogin_perform_logout(li));
1417         default:
1418                 logit("%s: Invalid type field", __func__);
1419                 return (0);
1420         }
1421 }
1422 #endif /* USE_LOGIN */
1423
1424 /* end of file log-syslogin.c */
1425
1426 /**
1427  ** Low-level lastlog functions
1428  **/
1429
1430 #ifdef USE_LASTLOG
1431 #define LL_FILE 1
1432 #define LL_DIR 2
1433 #define LL_OTHER 3
1434
1435 static void
1436 lastlog_construct(struct logininfo *li, struct lastlog *last)
1437 {
1438         /* clear the structure */
1439         memset(last, '\0', sizeof(*last));
1440
1441         line_stripname(last->ll_line, li->line, sizeof(last->ll_line));
1442         strlcpy(last->ll_host, li->hostname,
1443                 MIN_SIZEOF(last->ll_host, li->hostname));
1444         last->ll_time = li->tv_sec;
1445 }
1446
1447 static int
1448 lastlog_filetype(char *filename)
1449 {
1450         struct stat st;
1451
1452         if (stat(LASTLOG_FILE, &st) != 0) {
1453                 logit("%s: Couldn't stat %s: %s", __func__,
1454                     LASTLOG_FILE, strerror(errno));
1455                 return (0);
1456         }
1457         if (S_ISDIR(st.st_mode))
1458                 return (LL_DIR);
1459         else if (S_ISREG(st.st_mode))
1460                 return (LL_FILE);
1461         else
1462                 return (LL_OTHER);
1463 }
1464
1465
1466 /* open the file (using filemode) and seek to the login entry */
1467 static int
1468 lastlog_openseek(struct logininfo *li, int *fd, int filemode)
1469 {
1470         off_t offset;
1471         int type;
1472         char lastlog_file[1024];
1473
1474         type = lastlog_filetype(LASTLOG_FILE);
1475         switch (type) {
1476         case LL_FILE:
1477                 strlcpy(lastlog_file, LASTLOG_FILE,
1478                     sizeof(lastlog_file));
1479                 break;
1480         case LL_DIR:
1481                 snprintf(lastlog_file, sizeof(lastlog_file), "%s/%s",
1482                     LASTLOG_FILE, li->username);
1483                 break;
1484         default:
1485                 logit("%s: %.100s is not a file or directory!", __func__,
1486                     LASTLOG_FILE);
1487                 return (0);
1488         }
1489
1490         *fd = open(lastlog_file, filemode, 0600);
1491         if (*fd < 0) {
1492                 debug("%s: Couldn't open %s: %s", __func__,
1493                     lastlog_file, strerror(errno));
1494                 return (0);
1495         }
1496
1497         if (type == LL_FILE) {
1498                 /* find this uid's offset in the lastlog file */
1499                 offset = (off_t) ((long)li->uid * sizeof(struct lastlog));
1500
1501                 if (lseek(*fd, offset, SEEK_SET) != offset) {
1502                         logit("%s: %s->lseek(): %s", __func__,
1503                             lastlog_file, strerror(errno));
1504                         return (0);
1505                 }
1506         }
1507
1508         return (1);
1509 }
1510
1511 static int
1512 lastlog_perform_login(struct logininfo *li)
1513 {
1514         struct lastlog last;
1515         int fd;
1516
1517         /* create our struct lastlog */
1518         lastlog_construct(li, &last);
1519
1520         if (!lastlog_openseek(li, &fd, O_RDWR|O_CREAT))
1521                 return (0);
1522
1523         /* write the entry */
1524         if (atomicio(vwrite, fd, &last, sizeof(last)) != sizeof(last)) {
1525                 close(fd);
1526                 logit("%s: Error writing to %s: %s", __func__,
1527                     LASTLOG_FILE, strerror(errno));
1528                 return (0);
1529         }
1530
1531         close(fd);
1532         return (1);
1533 }
1534
1535 int
1536 lastlog_write_entry(struct logininfo *li)
1537 {
1538         switch(li->type) {
1539         case LTYPE_LOGIN:
1540                 return (lastlog_perform_login(li));
1541         default:
1542                 logit("%s: Invalid type field", __func__);
1543                 return (0);
1544         }
1545 }
1546
1547 static void
1548 lastlog_populate_entry(struct logininfo *li, struct lastlog *last)
1549 {
1550         line_fullname(li->line, last->ll_line, sizeof(li->line));
1551         strlcpy(li->hostname, last->ll_host,
1552             MIN_SIZEOF(li->hostname, last->ll_host));
1553         li->tv_sec = last->ll_time;
1554 }
1555
1556 int
1557 lastlog_get_entry(struct logininfo *li)
1558 {
1559         struct lastlog last;
1560         int fd, ret;
1561
1562         if (!lastlog_openseek(li, &fd, O_RDONLY))
1563                 return (0);
1564
1565         ret = atomicio(read, fd, &last, sizeof(last));
1566         close(fd);
1567
1568         switch (ret) {
1569         case 0:
1570                 memset(&last, '\0', sizeof(last));
1571                 /* FALLTHRU */
1572         case sizeof(last):
1573                 lastlog_populate_entry(li, &last);
1574                 return (1);
1575         case -1:
1576                 error("%s: Error reading from %s: %s", __func__,
1577                     LASTLOG_FILE, strerror(errno));
1578                 return (0);
1579         default:
1580                 error("%s: Error reading from %s: Expecting %d, got %d",
1581                     __func__, LASTLOG_FILE, sizeof(last), ret);
1582                 return (0);
1583         }
1584
1585         /* NOTREACHED */
1586         return (0);
1587 }
1588 #endif /* USE_LASTLOG */
1589
1590 #ifdef USE_BTMP
1591   /*
1592    * Logs failed login attempts in _PATH_BTMP if that exists.
1593    * The most common login failure is to give password instead of username.
1594    * So the _PATH_BTMP file checked for the correct permission, so that
1595    * only root can read it.
1596    */
1597
1598 void
1599 record_failed_login(const char *username, const char *hostname,
1600     const char *ttyn)
1601 {
1602         int fd;
1603         struct utmp ut;
1604         struct sockaddr_storage from;
1605         size_t fromlen = sizeof(from);
1606         struct sockaddr_in *a4;
1607         struct sockaddr_in6 *a6;
1608         time_t t;
1609         struct stat fst;
1610
1611         if (geteuid() != 0)
1612                 return;
1613         if ((fd = open(_PATH_BTMP, O_WRONLY | O_APPEND)) < 0) {
1614                 debug("Unable to open the btmp file %s: %s", _PATH_BTMP,
1615                     strerror(errno));
1616                 return;
1617         }
1618         if (fstat(fd, &fst) < 0) {
1619                 logit("%s: fstat of %s failed: %s", __func__, _PATH_BTMP,
1620                     strerror(errno));
1621                 goto out;
1622         }
1623         if((fst.st_mode & (S_IRWXG | S_IRWXO)) || (fst.st_uid != 0)){
1624                 logit("Excess permission or bad ownership on file %s",
1625                     _PATH_BTMP);
1626                 goto out;
1627         }
1628
1629         memset(&ut, 0, sizeof(ut));
1630         /* strncpy because we don't necessarily want nul termination */
1631         strncpy(ut.ut_user, username, sizeof(ut.ut_user));
1632         strlcpy(ut.ut_line, "ssh:notty", sizeof(ut.ut_line));
1633
1634         time(&t);
1635         ut.ut_time = t;     /* ut_time is not always a time_t */
1636         ut.ut_type = LOGIN_PROCESS;
1637         ut.ut_pid = getpid();
1638
1639         /* strncpy because we don't necessarily want nul termination */
1640         strncpy(ut.ut_host, hostname, sizeof(ut.ut_host));
1641
1642         if (packet_connection_is_on_socket() &&
1643             getpeername(packet_get_connection_in(),
1644             (struct sockaddr *)&from, &fromlen) == 0) {
1645                 ipv64_normalise_mapped(&from, &fromlen);
1646                 if (from.ss_family == AF_INET) {
1647                         a4 = (struct sockaddr_in *)&from;
1648                         memcpy(&ut.ut_addr, &(a4->sin_addr),
1649                             MIN_SIZEOF(ut.ut_addr, a4->sin_addr));
1650                 }
1651 #ifdef HAVE_ADDR_V6_IN_UTMP
1652                 if (from.ss_family == AF_INET6) {
1653                         a6 = (struct sockaddr_in6 *)&from;
1654                         memcpy(&ut.ut_addr_v6, &(a6->sin6_addr),
1655                             MIN_SIZEOF(ut.ut_addr_v6, a6->sin6_addr));
1656                 }
1657 #endif
1658         }
1659
1660         if (atomicio(vwrite, fd, &ut, sizeof(ut)) != sizeof(ut))
1661                 error("Failed to write to %s: %s", _PATH_BTMP,
1662                     strerror(errno));
1663
1664 out:
1665         close(fd);
1666 }
1667 #endif  /* USE_BTMP */