/* 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