/* Tests for UTMP functions.

   Copyright (C) 1998-2018 Free Software Foundation, Inc.

   This file is part of the GNU C Library.

   Contributed by Mark Kettenis <kettenis@phys.uva.nl>, 1998.

   The GNU C Library is free software; you can redistribute it and/or

   modify it under the terms of the GNU Lesser General Public

   License as published by the Free Software Foundation; either

   version 2.1 of the License, or (at your option) any later version.

   The GNU C Library is distributed in the hope that it will be useful,

   but WITHOUT ANY WARRANTY; without even the implied warranty of

   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU

   Lesser General Public License for more details.

   You should have received a copy of the GNU Lesser General Public

   License along with the GNU C Library; if not, see

   <http://www.gnu.org/licenses/>.  */

#include <errno.h>

#include <error.h>

#include <stdlib.h>

#include <string.h>

#include <sys/types.h>

#include <time.h>

#ifdef UTMPX

# include <utmpx.h>

# define utmp utmpx

# define utmpname utmpxname

# define setutent setutxent

# define getutent getutxent

# define endutent endutxent

# define getutline getutxline

# define getutid getutxid

# define pututline pututxline

#else

# include <utmp.h>

#endif

/* Prototype for our test function.  */

static int do_test (int argc, char *argv[]);

/* We have a preparation function.  */

static void do_prepare (int argc, char *argv[]);

#define PREPARE do_prepare

/* This defines the `main' function and some more.  */

#include <test-skeleton.c>

/* These are for the temporary file we generate.  */

char *name;

int fd;

static void

do_prepare (int argc, char *argv[])

{

  size_t name_len;

  name_len = strlen (test_dir);

  name = xmalloc (name_len + sizeof ("/utmpXXXXXX"));

  mempcpy (mempcpy (name, test_dir, name_len),

	   "/utmpXXXXXX", sizeof ("/utmpXXXXXX"));

  /* Open our test file.  */

  fd = mkstemp (name);

  if (fd == -1)

    error (EXIT_FAILURE, errno, "cannot open test file `%s'", name);

  add_temp_file (name);

}

struct utmp entry[] =

{

#define UT(a)  .ut_tv = { .tv_sec = (a)}

  { .ut_type = BOOT_TIME, .ut_pid = 1, UT(1000) },

  { .ut_type = RUN_LVL, .ut_pid = 1, UT(2000) },

  { .ut_type = INIT_PROCESS, .ut_pid = 5, .ut_id = "si", UT(3000) },

  { .ut_type = LOGIN_PROCESS, .ut_pid = 23, .ut_line = "tty1", .ut_id = "1",

    .ut_user = "LOGIN", UT(4000) },

  { .ut_type = USER_PROCESS, .ut_pid = 24, .ut_line = "tty2", .ut_id = "2",

    .ut_user = "albert", UT(8000) },

  { .ut_type = USER_PROCESS, .ut_pid = 196, .ut_line = "ttyp0", .ut_id = "p0",

    .ut_user = "niels", UT(10000) },

  { .ut_type = DEAD_PROCESS, .ut_line = "ttyp1", .ut_id = "p1", UT(16000) },

  { .ut_type = EMPTY },

  { .ut_type = EMPTY }

};

int num_entries = sizeof entry / sizeof (struct utmp);

time_t entry_time = 20000;

pid_t entry_pid = 234;

static int

do_init (void)

{

  int n;

  setutent ();

  for (n = 0; n < num_entries; n++)

    {

      if (pututline (&entry[n]) == NULL)

	{

	  error (0, errno, "cannot write UTMP entry");

	  return 1;

	}

    }

  endutent ();

  return 0;

}

static int

do_check (void)

{

  struct utmp *ut;

  int n;

  setutent ();

  n = 0;

  while ((ut = getutent ()))

    {

      if (n < num_entries &&

	  memcmp (ut, &entry[n], sizeof (struct utmp)))

	{

	  error (0, 0, "UTMP entry does not match");

	  return 1;

	}

      n++;

    }

  if (n != num_entries)

    {

      error (0, 0, "number of UTMP entries is incorrect");

      return 1;

    }

  endutent ();

  return 0;

}

static int

simulate_login (const char *line, const char *user)

{

  int n;

  for (n = 0; n < num_entries; n++)

    {

      if (strcmp (line, entry[n].ut_line) == 0 ||

	  entry[n].ut_type == DEAD_PROCESS)

	{

	  if (entry[n].ut_pid == DEAD_PROCESS)

	    entry[n].ut_pid = (entry_pid += 27);

	  entry[n].ut_type = USER_PROCESS;

	  strncpy (entry[n].ut_user, user, sizeof (entry[n].ut_user));

	  entry[n].ut_tv.tv_sec = (entry_time += 1000);

	  setutent ();

	  if (pututline (&entry[n]) == NULL)

	    {

	      error (0, errno, "cannot write UTMP entry");

	      return 1;

	    }

	  endutent ();

	  return 0;

	}

    }

  error (0, 0, "no entries available");

  return 1;

}

static int

simulate_logout (const char *line)

{

  int n;

  for (n = 0; n < num_entries; n++)

    {

      if (strcmp (line, entry[n].ut_line) == 0)

	{

	  entry[n].ut_type = DEAD_PROCESS;

	  strncpy (entry[n].ut_user, "", sizeof (entry[n].ut_user));

          entry[n].ut_tv.tv_sec = (entry_time += 1000);

	  setutent ();

	  if (pututline (&entry[n]) == NULL)

	    {

	      error (0, errno, "cannot write UTMP entry");

	      return 1;

	    }

	  endutent ();

	  return 0;

	}

    }

  error (0, 0, "no entry found for `%s'", line);

  return 1;

}

static int

check_login (const char *line)

{

  struct utmp *up;

  struct utmp ut;

  int n;

  setutent ();

  strcpy (ut.ut_line, line);

  up = getutline (&ut);

  if (up == NULL)

    {

      error (0, errno, "cannot get entry for line `%s'", line);

      return 1;

    }

  endutent ();

  for (n = 0; n < num_entries; n++)

    {

      if (strcmp (line, entry[n].ut_line) == 0)

	{

	  if (memcmp (up, &entry[n], sizeof (struct utmp)))

	    {

	      error (0, 0, "UTMP entry does not match");

	      return 1;

	    }

	  return 0;

	}

    }

  error (0, 0, "bogus entry for line `%s'", line);

  return 1;

}

static int

check_logout (const char *line)

{

  struct utmp ut;

  setutent ();

  strcpy (ut.ut_line, line);

  if (getutline (&ut) != NULL)

    {

      error (0, 0, "bogus login entry for `%s'", line);

      return 1;

    }

  endutent ();

  return 0;

}

static int

check_id (const char *id)

{

  struct utmp *up;

  struct utmp ut;

  int n;

  setutent ();

  ut.ut_type = USER_PROCESS;

  strcpy (ut.ut_id, id);

  up = getutid (&ut);

  if (up == NULL)

    {

      error (0, errno, "cannot get entry for ID `%s'", id);

      return 1;

    }

  endutent ();

  for (n = 0; n < num_entries; n++)

    {

      if (strcmp (id, entry[n].ut_id) == 0)

	{

	  if (memcmp (up, &entry[n], sizeof (struct utmp)))

	    {

	      error (0, 0, "UTMP entry does not match");

	      return 1;

	    }

	  return 0;

	}

    }

  error (0, 0, "bogus entry for ID `%s'", id);

  return 1;

}

static int

check_type (int type)

{

  struct utmp *up;

  struct utmp ut;

  int n;

  setutent ();

  ut.ut_type = type;

  up = getutid (&ut);

  if (up == NULL)

    {

      error (0, errno, "cannot get entry for type `%d'", type);

      return 1;

    }

  endutent ();

  for (n = 0; n < num_entries; n++)

    {

      if (type == entry[n].ut_type)

	{

	  if (memcmp (up, &entry[n], sizeof (struct utmp)))

	    {

	      error (0, 0, "UTMP entry does not match");

	      return 1;

	    }

	  return 0;

	}

    }

  error (0, 0, "bogus entry for type `%d'", type);

  return 1;

}

static int

do_test (int argc, char *argv[])

{

  int result = 0;

  utmpname (name);

  result |= do_init ();

  result |= do_check ();

  result |= simulate_login ("tty1", "erwin");

  result |= do_check ();

  result |= simulate_login ("ttyp1", "paul");

  result |= do_check ();

  result |= simulate_logout ("tty2");

  result |= do_check ();

  result |= simulate_logout ("ttyp0");

  result |= do_check ();

  result |= simulate_login ("ttyp2", "richard");

  result |= do_check ();

  result |= check_login ("tty1");

  result |= check_logout ("ttyp0");

  result |= check_id ("p1");

  result |= check_id ("2");

  result |= check_id ("si");

  result |= check_type (BOOT_TIME);

  result |= check_type (RUN_LVL);

  return result;

}