blob: da9ea50eca0a1a6edc295bdb8586b3180d15e20a [file] [log] [blame]
andre2ff7b5d2000-06-03 14:57:40 +00001/*
2 * Copyright (c) 2000 Andre Lucas. All rights reserved.
3 *
4 * Redistribution and use in source and binary forms, with or without
5 * modification, are permitted provided that the following conditions
6 * are met:
7 * 1. Redistributions of source code must retain the above copyright
8 * notice, this list of conditions and the following disclaimer.
9 * 2. Redistributions in binary form must reproduce the above copyright
10 * notice, this list of conditions and the following disclaimer in the
11 * documentation and/or other materials provided with the distribution.
12 * 3. All advertising materials mentioning features or use of this software
13 * must display the following acknowledgement:
14 * This product includes software developed by Markus Friedl.
15 * 4. The name of the author may not be used to endorse or promote products
16 * derived from this software without specific prior written permission.
17 *
18 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
19 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
20 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
21 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
22 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
23 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
24 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
25 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
26 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
27 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
28 */
29
Kevin Stevesef4eea92001-02-05 12:42:17 +000030/**
andre2ff7b5d2000-06-03 14:57:40 +000031 ** logintest.c: simple test driver for platform-independent login recording
32 ** and lastlog retrieval
33 **/
34
andreb4db42f2000-06-19 08:25:36 +000035#include "includes.h"
andre2ff7b5d2000-06-03 14:57:40 +000036
37#include <sys/types.h>
38#include <sys/wait.h>
39#include <unistd.h>
40#include <stdlib.h>
41#include <stdio.h>
42#include <string.h>
43#include <pwd.h>
44#include <netdb.h>
45#ifdef HAVE_TIME_H
46#include <time.h>
47#endif
48
49#include "loginrec.h"
50
Kevin Stevesff8b4952001-04-05 23:05:22 +000051RCSID("$Id: logintest.c,v 1.8 2001/04/05 23:05:22 stevesk Exp $");
andre2ff7b5d2000-06-03 14:57:40 +000052
Kevin Stevesff8b4952001-04-05 23:05:22 +000053#ifdef HAVE___PROGNAME
54extern char *__progname;
55#else
56char *__progname;
57#endif
andre2ff7b5d2000-06-03 14:57:40 +000058
andreb4db42f2000-06-19 08:25:36 +000059#define PAUSE_BEFORE_LOGOUT 3
60
andre2ff7b5d2000-06-03 14:57:40 +000061int nologtest = 0;
62int compile_opts_only = 0;
63int be_verbose = 0;
64
65
andre2ff7b5d2000-06-03 14:57:40 +000066/* Dump a logininfo to stdout. Assumes a tab size of 8 chars. */
andre6e5d3472000-06-13 00:43:47 +000067void
68dump_logininfo(struct logininfo *li, char *descname)
andre61e67252000-06-04 17:07:49 +000069{
andre6e5d3472000-06-13 00:43:47 +000070 /* yes I know how nasty this is */
71 printf("struct logininfo %s = {\n\t"
72 "progname\t'%s'\n\ttype\t\t%d\n\t"
73 "pid\t\t%d\n\tuid\t\t%d\n\t"
74 "line\t\t'%s'\n\tusername\t'%s'\n\t"
75 "hostname\t'%s'\n\texit\t\t%d\n\ttermination\t%d\n\t"
76 "tv_sec\t%d\n\ttv_usec\t%d\n\t"
77 "struct login_netinfo hostaddr {\n\t\t"
78 "struct sockaddr sa {\n"
79 "\t\t\tfamily\t%d\n\t\t}\n"
andre6e5d3472000-06-13 00:43:47 +000080 "\t}\n"
81 "}\n",
Kevin Stevesef4eea92001-02-05 12:42:17 +000082 descname, li->progname, li->type,
andre6e5d3472000-06-13 00:43:47 +000083 li->pid, li->uid, li->line,
Kevin Stevesef4eea92001-02-05 12:42:17 +000084 li->username, li->hostname, li->exit,
85 li->termination, li->tv_sec, li->tv_usec,
andre6e5d3472000-06-13 00:43:47 +000086 li->hostaddr.sa.sa_family);
andre2ff7b5d2000-06-03 14:57:40 +000087}
88
89
andre6e5d3472000-06-13 00:43:47 +000090int
91testAPI()
andre61e67252000-06-04 17:07:49 +000092{
andre6e5d3472000-06-13 00:43:47 +000093 struct logininfo *li1;
94 struct passwd *pw;
95 struct hostent *he;
96 struct sockaddr_in sa_in4;
97 char cmdstring[256], stripline[8];
98 char username[32];
andre2ff7b5d2000-06-03 14:57:40 +000099#ifdef HAVE_TIME_H
andreb4db42f2000-06-19 08:25:36 +0000100 time_t t0, t1, t2, logintime, logouttime;
101 char s_t0[64],s_t1[64],s_t2[64];
102 char s_logintime[64], s_logouttime[64]; /* ctime() strings */
andre2ff7b5d2000-06-03 14:57:40 +0000103#endif
104
andre6e5d3472000-06-13 00:43:47 +0000105 printf("**\n** Testing the API...\n**\n");
andre2ff7b5d2000-06-03 14:57:40 +0000106
andre6e5d3472000-06-13 00:43:47 +0000107 pw = getpwuid(getuid());
108 strlcpy(username, pw->pw_name, sizeof(username));
andre2ff7b5d2000-06-03 14:57:40 +0000109
andre6e5d3472000-06-13 00:43:47 +0000110 /* gethostname(hostname, sizeof(hostname)); */
andre2ff7b5d2000-06-03 14:57:40 +0000111
andre6e5d3472000-06-13 00:43:47 +0000112 printf("login_alloc_entry test (no host info):\n");
andreb4db42f2000-06-19 08:25:36 +0000113
114 /* FIXME fake tty more effectively - this could upset some platforms */
andre6e5d3472000-06-13 00:43:47 +0000115 li1 = login_alloc_entry((int)getpid(), username, NULL, ttyname(0));
116 strlcpy(li1->progname, "OpenSSH-logintest", sizeof(li1->progname));
andre2ff7b5d2000-06-03 14:57:40 +0000117
andre6e5d3472000-06-13 00:43:47 +0000118 if (be_verbose)
119 dump_logininfo(li1, "li1");
andre2ff7b5d2000-06-03 14:57:40 +0000120
andre6e5d3472000-06-13 00:43:47 +0000121 printf("Setting host address info for 'localhost' (may call out):\n");
122 if (! (he = gethostbyname("localhost"))) {
123 printf("Couldn't set hostname(lookup failed)\n");
124 } else {
125 /* NOTE: this is messy, but typically a program wouldn't have to set
126 * any of this, a sockaddr_in* would be already prepared */
127 memcpy((void *)&(sa_in4.sin_addr), (void *)&(he->h_addr_list[0][0]),
128 sizeof(struct in_addr));
129 login_set_addr(li1, (struct sockaddr *) &sa_in4, sizeof(sa_in4));
130 strlcpy(li1->hostname, "localhost", sizeof(li1->hostname));
131 }
132 if (be_verbose)
133 dump_logininfo(li1, "li1");
andre2ff7b5d2000-06-03 14:57:40 +0000134
andre6e5d3472000-06-13 00:43:47 +0000135 if ((int)geteuid() != 0) {
136 printf("NOT RUNNING LOGIN TESTS - you are not root!\n");
andreb4db42f2000-06-19 08:25:36 +0000137 return 1;
andre6e5d3472000-06-13 00:43:47 +0000138 }
andre2ff7b5d2000-06-03 14:57:40 +0000139
andre6e5d3472000-06-13 00:43:47 +0000140 if (nologtest)
141 return 1;
Kevin Stevesef4eea92001-02-05 12:42:17 +0000142
andre6e5d3472000-06-13 00:43:47 +0000143 line_stripname(stripline, li1->line, sizeof(stripline));
andre2ff7b5d2000-06-03 14:57:40 +0000144
andre6e5d3472000-06-13 00:43:47 +0000145 printf("Performing an invalid login attempt (no type field)\n--\n");
146 login_write(li1);
andreb4db42f2000-06-19 08:25:36 +0000147 printf("--\n(Should have written errors to stderr)\n");
andre2ff7b5d2000-06-03 14:57:40 +0000148
149#ifdef HAVE_TIME_H
andre6e5d3472000-06-13 00:43:47 +0000150 (void)time(&t0);
151 strlcpy(s_t0, ctime(&t0), sizeof(s_t0));
152 t1 = login_get_lastlog_time(getuid());
153 strlcpy(s_t1, ctime(&t1), sizeof(s_t1));
154 printf("Before logging in:\n\tcurrent time is %d - %s\t"
155 "lastlog time is %d - %s\n",
156 (int)t0, s_t0, (int)t1, s_t1);
andre2ff7b5d2000-06-03 14:57:40 +0000157#endif
158
andreb4db42f2000-06-19 08:25:36 +0000159 printf("Performing a login on line %s ", stripline);
160#ifdef HAVE_TIME_H
161 (void)time(&logintime);
162 strlcpy(s_logintime, ctime(&logintime), sizeof(s_logintime));
163 printf("at %d - %s", (int)logintime, s_logintime);
164#endif
165 printf("--\n");
andre6e5d3472000-06-13 00:43:47 +0000166 login_login(li1);
Kevin Stevesef4eea92001-02-05 12:42:17 +0000167
andre6e5d3472000-06-13 00:43:47 +0000168 snprintf(cmdstring, sizeof(cmdstring), "who | grep '%s '",
169 stripline);
170 system(cmdstring);
Kevin Stevesef4eea92001-02-05 12:42:17 +0000171
andreb4db42f2000-06-19 08:25:36 +0000172 printf("--\nPausing for %d second(s)...\n", PAUSE_BEFORE_LOGOUT);
173 sleep(PAUSE_BEFORE_LOGOUT);
andre2ff7b5d2000-06-03 14:57:40 +0000174
andre6e5d3472000-06-13 00:43:47 +0000175 printf("Performing a logout ");
176#ifdef HAVE_TIME_H
177 (void)time(&logouttime);
178 strlcpy(s_logouttime, ctime(&logouttime), sizeof(s_logouttime));
179 printf("at %d - %s", (int)logouttime, s_logouttime);
180#endif
andreb4db42f2000-06-19 08:25:36 +0000181 printf("\nThe root login shown above should be gone.\n"
andre6e5d3472000-06-13 00:43:47 +0000182 "If the root login hasn't gone, but another user on the same\n"
183 "pty has, this is OK - we're hacking it here, and there\n"
184 "shouldn't be two users on one pty in reality...\n"
185 "-- ('who' output follows)\n");
186 login_logout(li1);
andre2ff7b5d2000-06-03 14:57:40 +0000187
andre6e5d3472000-06-13 00:43:47 +0000188 system(cmdstring);
189 printf("-- ('who' output ends)\n");
andre2ff7b5d2000-06-03 14:57:40 +0000190
191#ifdef HAVE_TIME_H
andre6e5d3472000-06-13 00:43:47 +0000192 t2 = login_get_lastlog_time(getuid());
193 strlcpy(s_t2, ctime(&t2), sizeof(s_t2));
194 printf("After logging in, lastlog time is %d - %s\n", (int)t2, s_t2);
195 if (t1 == t2)
196 printf("The lastlog times before and after logging in are the "
197 "same.\nThis indicates that lastlog is ** NOT WORKING "
198 "CORRECTLY **\n");
199 else if (t0 != t2)
andreb4db42f2000-06-19 08:25:36 +0000200 /* We can be off by a second or so, even when recording works fine.
201 * I'm not 100% sure why, but it's true. */
andre6e5d3472000-06-13 00:43:47 +0000202 printf("** The login time and the lastlog time differ.\n"
203 "** This indicates that lastlog is either recording the "
andreb4db42f2000-06-19 08:25:36 +0000204 "wrong time,\n** or retrieving the wrong entry.\n"
205 "If it's off by less than %d second(s) "
206 "run the test again.\n", PAUSE_BEFORE_LOGOUT);
andre6e5d3472000-06-13 00:43:47 +0000207 else
208 printf("lastlog agrees with the login time. This is a good thing.\n");
andre2ff7b5d2000-06-03 14:57:40 +0000209
210#endif
211
andre6e5d3472000-06-13 00:43:47 +0000212 printf("--\nThe output of 'last' shown next should have "
Kevin Stevesef4eea92001-02-05 12:42:17 +0000213 "an entry for root \n on %s for the time shown above:\n--\n",
andre6e5d3472000-06-13 00:43:47 +0000214 stripline);
215 snprintf(cmdstring, sizeof(cmdstring), "last | grep '%s ' | head -3",
216 stripline);
217 system(cmdstring);
Kevin Stevesef4eea92001-02-05 12:42:17 +0000218
andre6e5d3472000-06-13 00:43:47 +0000219 printf("--\nEnd of login test.\n");
andre2ff7b5d2000-06-03 14:57:40 +0000220
andre6e5d3472000-06-13 00:43:47 +0000221 login_free_entry(li1);
andre2ff7b5d2000-06-03 14:57:40 +0000222
andre6e5d3472000-06-13 00:43:47 +0000223 return 1;
andre2ff7b5d2000-06-03 14:57:40 +0000224} /* testAPI() */
225
226
andre6e5d3472000-06-13 00:43:47 +0000227void
228testLineName(char *line)
andre61e67252000-06-04 17:07:49 +0000229{
andre6e5d3472000-06-13 00:43:47 +0000230 /* have to null-terminate - these functions are designed for
231 * structures with fixed-length char arrays, and don't null-term.*/
232 char full[17], strip[9], abbrev[5];
andre2ff7b5d2000-06-03 14:57:40 +0000233
andre6e5d3472000-06-13 00:43:47 +0000234 memset(full, '\0', sizeof(full));
235 memset(strip, '\0', sizeof(strip));
236 memset(abbrev, '\0', sizeof(abbrev));
andre2ff7b5d2000-06-03 14:57:40 +0000237
andre6e5d3472000-06-13 00:43:47 +0000238 line_fullname(full, line, sizeof(full)-1);
239 line_stripname(strip, full, sizeof(strip)-1);
240 line_abbrevname(abbrev, full, sizeof(abbrev)-1);
241 printf("%s: %s, %s, %s\n", line, full, strip, abbrev);
andre2ff7b5d2000-06-03 14:57:40 +0000242
243} /* testLineName() */
244
245
andre6e5d3472000-06-13 00:43:47 +0000246int
247testOutput()
248{
249 printf("**\n** Testing linename functions\n**\n");
250 testLineName("/dev/pts/1");
251 testLineName("pts/1");
252 testLineName("pts/999");
253 testLineName("/dev/ttyp00");
254 testLineName("ttyp00");
andre2ff7b5d2000-06-03 14:57:40 +0000255
andre6e5d3472000-06-13 00:43:47 +0000256 return 1;
andre2ff7b5d2000-06-03 14:57:40 +0000257} /* testOutput() */
258
259
260/* show which options got compiled in */
andre6e5d3472000-06-13 00:43:47 +0000261void
262showOptions(void)
Kevin Stevesef4eea92001-02-05 12:42:17 +0000263{
andre6e5d3472000-06-13 00:43:47 +0000264 printf("**\n** Compile-time options\n**\n");
Kevin Stevesef4eea92001-02-05 12:42:17 +0000265
andre6e5d3472000-06-13 00:43:47 +0000266 printf("login recording methods selected:\n");
andre2ff7b5d2000-06-03 14:57:40 +0000267#ifdef USE_LOGIN
andre6e5d3472000-06-13 00:43:47 +0000268 printf("\tUSE_LOGIN\n");
andre2ff7b5d2000-06-03 14:57:40 +0000269#endif
270#ifdef USE_UTMP
andre6e5d3472000-06-13 00:43:47 +0000271 printf("\tUSE_UTMP (UTMP_FILE=%s)\n", UTMP_FILE);
andre2ff7b5d2000-06-03 14:57:40 +0000272#endif
273#ifdef USE_UTMPX
andre6e5d3472000-06-13 00:43:47 +0000274 printf("\tUSE_UTMPX (UTMPX_FILE=%s)\n", UTMPX_FILE);
andre2ff7b5d2000-06-03 14:57:40 +0000275#endif
276#ifdef USE_WTMP
andre6e5d3472000-06-13 00:43:47 +0000277 printf("\tUSE_WTMP (WTMP_FILE=%s)\n", WTMP_FILE);
andre2ff7b5d2000-06-03 14:57:40 +0000278#endif
279#ifdef USE_WTMPX
andre6e5d3472000-06-13 00:43:47 +0000280 printf("\tUSE_WTMPX (WTMPX_FILE=%s)\n", WTMPX_FILE);
andre2ff7b5d2000-06-03 14:57:40 +0000281#endif
282#ifdef USE_LASTLOG
andre6e5d3472000-06-13 00:43:47 +0000283 printf("\tUSE_LASTLOG (LASTLOG_FILE=%s)\n", LASTLOG_FILE);
andre2ff7b5d2000-06-03 14:57:40 +0000284#endif
andre6e5d3472000-06-13 00:43:47 +0000285 printf("\n");
andre2ff7b5d2000-06-03 14:57:40 +0000286
andre2ff7b5d2000-06-03 14:57:40 +0000287} /* showOptions() */
288
289
andre6e5d3472000-06-13 00:43:47 +0000290int
291main(int argc, char *argv[])
andre61e67252000-06-04 17:07:49 +0000292{
andre6e5d3472000-06-13 00:43:47 +0000293 printf("Platform-independent login recording test driver\n");
andre2ff7b5d2000-06-03 14:57:40 +0000294
Kevin Stevesff8b4952001-04-05 23:05:22 +0000295 __progname = get_progname(argv[0]);
andre6e5d3472000-06-13 00:43:47 +0000296 if (argc == 2) {
297 if (strncmp(argv[1], "-i", 3) == 0)
298 compile_opts_only = 1;
299 else if (strncmp(argv[1], "-v", 3) == 0)
300 be_verbose=1;
301 }
Kevin Stevesef4eea92001-02-05 12:42:17 +0000302
andre6e5d3472000-06-13 00:43:47 +0000303 if (!compile_opts_only) {
304 if (be_verbose && !testOutput())
305 return 1;
Kevin Stevesef4eea92001-02-05 12:42:17 +0000306
andre6e5d3472000-06-13 00:43:47 +0000307 if (!testAPI())
308 return 1;
309 }
andre2ff7b5d2000-06-03 14:57:40 +0000310
andre6e5d3472000-06-13 00:43:47 +0000311 showOptions();
Kevin Stevesef4eea92001-02-05 12:42:17 +0000312
andre6e5d3472000-06-13 00:43:47 +0000313 return 0;
andre2ff7b5d2000-06-03 14:57:40 +0000314} /* main() */
315