123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404 |
- /* Tests for UTMP functions.
- Copyright (C) 1998, 2001-2003 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, write to the Free
- Software Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA
- 02111-1307 USA. */
- #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
- #if _HAVE_UT_TYPE || defined UTMPX
- /* 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 = malloc (name_len + sizeof ("/utmpXXXXXX"));
- mempcpy (mempcpy (name, test_dir, name_len),
- "/utmpXXXXXX", sizeof ("/utmpXXXXXX"));
- add_temp_file (name);
- /* Open our test file. */
- fd = mkstemp (name);
- if (fd == -1)
- error (EXIT_FAILURE, errno, "cannot open test file `%s'", name);
- }
- struct utmp entry[] =
- {
- #if _HAVE_UT_TV || defined UTMPX
- #define UT(a) .ut_tv = { .tv_sec = (a)}
- #else
- #define UT(a) .ut_time = (a)
- #endif
- { .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));
- #if _HAVE_UT_TV - 0 || defined UTMPX
- entry[n].ut_tv.tv_sec = (entry_time += 1000);
- #else
- entry[n].ut_time = (entry_time += 1000);
- #endif
- 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));
- #if _HAVE_UT_TV - 0 || defined UTMPX
- entry[n].ut_tv.tv_sec = (entry_time += 1000);
- #else
- entry[n].ut_time = (entry_time += 1000);
- #endif
- 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;
- }
- #else
- /* No field 'ut_type' in struct utmp. */
- int
- main ()
- {
- return 0;
- }
- #endif
|