]> andersk Git - openssh.git/blobdiff - logintest.c
- djm@cvs.openbsd.org 2010/01/30 02:54:53
[openssh.git] / logintest.c
index c7a3be625fc1543c488c1b028f2ffdb57fe5c65e..7e9fbbfbbdc1e5a6cac726af7de233410e9438d8 100644 (file)
@@ -9,11 +9,6 @@
  * 2. Redistributions in binary form must reproduce the above copyright
  *    notice, this list of conditions and the following disclaimer in the
  *    documentation and/or other materials provided with the distribution.
- * 3. All advertising materials mentioning features or use of this software
- *    must display the following acknowledgement:
- *      This product includes software developed by Markus Friedl.
- * 4. The name of the author may not be used to endorse or promote products
- *    derived from this software without specific prior written permission.
  *
  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  */
 
-/** 
+/**
  ** logintest.c:  simple test driver for platform-independent login recording
  **               and lastlog retrieval
  **/
 
-#include "config.h"
+#include "includes.h"
 
 #include <sys/types.h>
 #include <sys/wait.h>
+#include <sys/socket.h>
+
+#include <netinet/in.h>
+
 #include <unistd.h>
 #include <stdlib.h>
 #include <stdio.h>
@@ -48,8 +47,9 @@
 
 #include "loginrec.h"
 
-RCSID("$Id$");
+extern char *__progname;
 
+#define PAUSE_BEFORE_LOGOUT 3
 
 int nologtest = 0;
 int compile_opts_only = 0;
@@ -57,228 +57,252 @@ int be_verbose = 0;
 
 
 /* Dump a logininfo to stdout. Assumes a tab size of 8 chars. */
-void dump_logininfo(struct logininfo *li, char *descname)
+void
+dump_logininfo(struct logininfo *li, char *descname)
 {
-  /* yes I know how nasty this is */
-  printf("struct logininfo %s = {\n\t"
-        "progname\t'%s'\n\ttype\t\t%d\n\t"
-        "pid\t\t%d\n\tuid\t\t%d\n\t"
-        "line\t\t'%s'\n\tusername\t'%s'\n\t"
-        "hostname\t'%s'\n\texit\t\t%d\n\ttermination\t%d\n\t"
-        "tv_sec\t%d\n\ttv_usec\t%d\n\t"
-        "struct login_netinfo hostaddr {\n\t\t"
-        "struct sockaddr sa {\n"
-        "\t\t\tfamily\t%d\n\t\t}\n"
-        "\t\t** !!! IP6 stuff not supported yet **\n"
-        "\t}\n"
-        "}\n",
-        descname, li->progname, li->type, 
-        li->pid, li->uid, li->line,
-        li->username, li->hostname, li->exit, 
-        li->termination, li->tv_sec, li->tv_usec, 
-        li->hostaddr.sa.sa_family);
+       /* yes I know how nasty this is */
+       printf("struct logininfo %s = {\n\t"
+              "progname\t'%s'\n\ttype\t\t%d\n\t"
+              "pid\t\t%d\n\tuid\t\t%d\n\t"
+              "line\t\t'%s'\n\tusername\t'%s'\n\t"
+              "hostname\t'%s'\n\texit\t\t%d\n\ttermination\t%d\n\t"
+              "tv_sec\t%d\n\ttv_usec\t%d\n\t"
+              "struct login_netinfo hostaddr {\n\t\t"
+              "struct sockaddr sa {\n"
+              "\t\t\tfamily\t%d\n\t\t}\n"
+              "\t}\n"
+              "}\n",
+              descname, li->progname, li->type,
+              li->pid, li->uid, li->line,
+              li->username, li->hostname, li->exit,
+              li->termination, li->tv_sec, li->tv_usec,
+              li->hostaddr.sa.sa_family);
 }
 
 
-int testAPI()
+int
+testAPI()
 {
-  struct logininfo *li1;
-  struct passwd *pw;
-  struct hostent *he;
-  struct sockaddr_in sa_in4;
-  char cmdstring[256], stripline[8];
-  char username[32];
+       struct logininfo *li1;
+       struct passwd *pw;
+       struct hostent *he;
+       struct sockaddr_in sa_in4;
+       char cmdstring[256], stripline[8];
+       char username[32];
+#ifdef HAVE_TIME_H
+       time_t t0, t1, t2, logintime, logouttime;
+       char s_t0[64],s_t1[64],s_t2[64];
+       char s_logintime[64], s_logouttime[64]; /* ctime() strings */
+#endif
+
+       printf("**\n** Testing the API...\n**\n");
+
+       pw = getpwuid(getuid());
+       strlcpy(username, pw->pw_name, sizeof(username));
+
+       /* gethostname(hostname, sizeof(hostname)); */
+
+       printf("login_alloc_entry test (no host info):\n");
+
+       /* FIXME fake tty more effectively - this could upset some platforms */
+       li1 = login_alloc_entry((int)getpid(), username, NULL, ttyname(0));
+       strlcpy(li1->progname, "OpenSSH-logintest", sizeof(li1->progname));
+
+       if (be_verbose)
+               dump_logininfo(li1, "li1");
+
+       printf("Setting host address info for 'localhost' (may call out):\n");
+       if (! (he = gethostbyname("localhost"))) {
+               printf("Couldn't set hostname(lookup failed)\n");
+       } else {
+               /* NOTE: this is messy, but typically a program wouldn't have to set
+                *  any of this, a sockaddr_in* would be already prepared */
+               memcpy((void *)&(sa_in4.sin_addr), (void *)&(he->h_addr_list[0][0]),
+                      sizeof(struct in_addr));
+               login_set_addr(li1, (struct sockaddr *) &sa_in4, sizeof(sa_in4));
+               strlcpy(li1->hostname, "localhost", sizeof(li1->hostname));
+       }
+       if (be_verbose)
+               dump_logininfo(li1, "li1");
+
+       if ((int)geteuid() != 0) {
+               printf("NOT RUNNING LOGIN TESTS - you are not root!\n");
+               return 1;
+       }
+
+       if (nologtest)
+               return 1;
+
+       line_stripname(stripline, li1->line, sizeof(stripline));
+
+       printf("Performing an invalid login attempt (no type field)\n--\n");
+       login_write(li1);
+       printf("--\n(Should have written errors to stderr)\n");
+
+#ifdef HAVE_TIME_H
+       (void)time(&t0);
+       strlcpy(s_t0, ctime(&t0), sizeof(s_t0));
+       t1 = login_get_lastlog_time(getuid());
+       strlcpy(s_t1, ctime(&t1), sizeof(s_t1));
+       printf("Before logging in:\n\tcurrent time is %d - %s\t"
+              "lastlog time is %d - %s\n",
+              (int)t0, s_t0, (int)t1, s_t1);
+#endif
+
+       printf("Performing a login on line %s ", stripline);
 #ifdef HAVE_TIME_H
-  time_t t0, t1, t2;
-  char s_t0[64],s_t1[64],s_t2[64]; /* ctime() strings */
+       (void)time(&logintime);
+       strlcpy(s_logintime, ctime(&logintime), sizeof(s_logintime));
+       printf("at %d - %s", (int)logintime, s_logintime);
 #endif
+       printf("--\n");
+       login_login(li1);
 
-  printf("**\n** Testing the API...\n**\n");
-
-  pw = getpwuid(getuid());
-  strlcpy(username, pw->pw_name, sizeof(username));
-
-  /* gethostname(hostname, sizeof(hostname)); */
-
-  printf("login_alloc_entry test (no host info):\n");
-  /* !!! fake tty more effectively */
-  li1 = login_alloc_entry((int)getpid(), username, NULL, ttyname(0));
-  strlcpy(li1->progname, "OpenSSH-logintest", sizeof(li1->progname));
-
-  if (be_verbose)
-    dump_logininfo(li1, "li1");
-
-  printf("Setting host address info for 'localhost' (may call out):\n");
-  if (! (he = gethostbyname("localhost"))) {
-    printf("Couldn't set hostname(lookup failed)\n");
-  } else {
-    /* NOTE: this is messy, but typically a program wouldn't have to set
-     *  any of this, a sockaddr_in* would be already prepared */
-    memcpy((void *)&(sa_in4.sin_addr), (void *)&(he->h_addr_list[0][0]),
-          sizeof(struct in_addr));
-    login_set_addr(li1, (struct sockaddr *) &sa_in4, sizeof(sa_in4));
-    strlcpy(li1->hostname, "localhost", sizeof(li1->hostname));
-  }
-  if (be_verbose)
-    dump_logininfo(li1, "li1");
-
-  if ((int)geteuid() != 0) {
-    printf("NOT RUNNING LOGIN TESTS - you are not root!\n");
-    return 1; /* this isn't necessarily an error */
-  }
-
-  if (nologtest)
-    return 1;
-  
-  line_stripname(stripline, li1->line, sizeof(stripline));
-
-  printf("Performing an invalid login attempt (no type field)\n--\n");
-  login_write(li1);
-  printf("--\n(Should have written an error to stderr)\n");
+       snprintf(cmdstring, sizeof(cmdstring), "who | grep '%s '",
+                stripline);
+       system(cmdstring);
 
+       printf("--\nPausing for %d second(s)...\n", PAUSE_BEFORE_LOGOUT);
+       sleep(PAUSE_BEFORE_LOGOUT);
+
+       printf("Performing a logout ");
 #ifdef HAVE_TIME_H
-  (void)time(&t0);
-  strlcpy(s_t0, ctime(&t0), sizeof(s_t0));
-  t1 = login_get_lastlog_time(getuid());
-  strlcpy(s_t1, ctime(&t1), sizeof(s_t1));
-  printf("Before logging in:\n\tcurrent time is %d - %s\t"
-        "lastlog time is %d - %s\n",
-        (int)t0, s_t0, (int)t1, s_t1);
+       (void)time(&logouttime);
+       strlcpy(s_logouttime, ctime(&logouttime), sizeof(s_logouttime));
+       printf("at %d - %s", (int)logouttime, s_logouttime);
 #endif
+       printf("\nThe root login shown above should be gone.\n"
+              "If the root login hasn't gone, but another user on the same\n"
+              "pty has, this is OK - we're hacking it here, and there\n"
+              "shouldn't be two users on one pty in reality...\n"
+              "-- ('who' output follows)\n");
+       login_logout(li1);
 
-  printf("Performing a login on line %s...\n--\n", stripline);
-  login_login(li1);
-  
-  snprintf(cmdstring, sizeof(cmdstring), "who | grep '%s '",
-          stripline);
-  system(cmdstring);
-  
-  printf("--\nWaiting for a few seconds...\n");
-  sleep(2);
-
-  printf("Performing a logout (the root login "
-        "shown above should be gone)\n"
-        "If the root login hasn't gone, but another user on the same\n"
-        "pty has, this is OK - we're hacking it here, and there\n"
-        "shouldn't be two users on one pty in reality...\n"
-        "-- ('who' output follows)\n");
-  login_logout(li1);
-
-  system(cmdstring);
-  printf("-- ('who' output ends)\n");
+       system(cmdstring);
+       printf("-- ('who' output ends)\n");
 
 #ifdef HAVE_TIME_H
-  t2 = login_get_lastlog_time(getuid());
-  strlcpy(s_t2, ctime(&t2), sizeof(s_t2));
-  printf("After logging in, lastlog time is %d - %s\n", (int)t2, s_t2);
-  if (t1 == t2)
-    printf("The lastlog times before and after logging in are the "
-          "same.\nThis indicates that lastlog is ** NOT WORKING "
-          "CORRECTLY **\n");
-  else if (t0 != t2)
-    printf("** The login time and the lastlog time differ.\n"
-          "** This indicates that lastlog is either recording the "
-          "wrong time,\n** or retrieving the wrong entry.\n");
-  else
-    printf("lastlog agrees with the login time. This is a good thing.\n");
+       t2 = login_get_lastlog_time(getuid());
+       strlcpy(s_t2, ctime(&t2), sizeof(s_t2));
+       printf("After logging in, lastlog time is %d - %s\n", (int)t2, s_t2);
+       if (t1 == t2)
+               printf("The lastlog times before and after logging in are the "
+                      "same.\nThis indicates that lastlog is ** NOT WORKING "
+                      "CORRECTLY **\n");
+       else if (t0 != t2)
+               /* We can be off by a second or so, even when recording works fine.
+                * I'm not 100% sure why, but it's true. */
+               printf("** The login time and the lastlog time differ.\n"
+                      "** This indicates that lastlog is either recording the "
+                      "wrong time,\n** or retrieving the wrong entry.\n"
+                      "If it's off by less than %d second(s) "
+                      "run the test again.\n", PAUSE_BEFORE_LOGOUT);
+       else
+               printf("lastlog agrees with the login time. This is a good thing.\n");
 
 #endif
 
-  printf("--\nThe output of 'last' shown next should have "
-        "an entry for root \n  on %s for the time shown above:\n--\n", 
-        stripline);
-  snprintf(cmdstring, sizeof(cmdstring), "last | grep '%s ' | head -3",
-          stripline);
-  system(cmdstring);
-            
-  printf("--\nEnd of login test.\n");
+       printf("--\nThe output of 'last' shown next should have "
+              "an entry for root \n  on %s for the time shown above:\n--\n",
+              stripline);
+       snprintf(cmdstring, sizeof(cmdstring), "last | grep '%s ' | head -3",
+                stripline);
+       system(cmdstring);
+
+       printf("--\nEnd of login test.\n");
 
-  login_free_entry(li1);
+       login_free_entry(li1);
 
-  return 1;
+       return 1;
 } /* testAPI() */
 
 
-void testLineName(char *line)
+void
+testLineName(char *line)
 {
-  /* have to null-terminate - these functions are designed for
-   * structures with fixed-length char arrays, and don't null-term.*/
-  char full[17], strip[9], abbrev[5];
+       /* have to null-terminate - these functions are designed for
+        * structures with fixed-length char arrays, and don't null-term.*/
+       char full[17], strip[9], abbrev[5];
 
-  memset(full, '\0', sizeof(full));
-  memset(strip, '\0', sizeof(strip));
-  memset(abbrev, '\0', sizeof(abbrev));
+       memset(full, '\0', sizeof(full));
+       memset(strip, '\0', sizeof(strip));
+       memset(abbrev, '\0', sizeof(abbrev));
 
-  line_fullname(full, line, sizeof(full)-1);
-  line_stripname(strip, full, sizeof(strip)-1);
-  line_abbrevname(abbrev, full, sizeof(abbrev)-1);
-  printf("%s: %s, %s, %s\n", line, full, strip, abbrev);
+       line_fullname(full, line, sizeof(full)-1);
+       line_stripname(strip, full, sizeof(strip)-1);
+       line_abbrevname(abbrev, full, sizeof(abbrev)-1);
+       printf("%s: %s, %s, %s\n", line, full, strip, abbrev);
 
 } /* testLineName() */
 
 
-int testOutput() {
-  printf("**\n** Testing linename functions\n**\n");
-  testLineName("/dev/pts/1");
-  testLineName("pts/1");
-  testLineName("pts/999");
-  testLineName("/dev/ttyp00");
-  testLineName("ttyp00");
-
-  return 1;
+int
+testOutput()
+{
+       printf("**\n** Testing linename functions\n**\n");
+       testLineName("/dev/pts/1");
+       testLineName("pts/1");
+       testLineName("pts/999");
+       testLineName("/dev/ttyp00");
+       testLineName("ttyp00");
+
+       return 1;
 } /* testOutput() */
 
 
 /* show which options got compiled in */
-void showOptions(void)
-{  
-  printf("**\n** Compile-time options\n**\n");
-  
-  printf("login recording methods selected:\n");
+void
+showOptions(void)
+{
+       printf("**\n** Compile-time options\n**\n");
+
+       printf("login recording methods selected:\n");
 #ifdef USE_LOGIN
-  printf("\tUSE_LOGIN\n");
+       printf("\tUSE_LOGIN\n");
 #endif
 #ifdef USE_UTMP
-  printf("\tUSE_UTMP (UTMP_FILE=%s)\n", UTMP_FILE);
+       printf("\tUSE_UTMP (UTMP_FILE=%s)\n", UTMP_FILE);
 #endif
 #ifdef USE_UTMPX
-  printf("\tUSE_UTMPX (UTMPX_FILE=%s)\n", UTMPX_FILE);
+       printf("\tUSE_UTMPX (UTMPX_FILE=%s)\n", UTMPX_FILE);
 #endif
 #ifdef USE_WTMP
-  printf("\tUSE_WTMP (WTMP_FILE=%s)\n", WTMP_FILE);
+       printf("\tUSE_WTMP (WTMP_FILE=%s)\n", WTMP_FILE);
 #endif
 #ifdef USE_WTMPX
-  printf("\tUSE_WTMPX (WTMPX_FILE=%s)\n", WTMPX_FILE);
+       printf("\tUSE_WTMPX (WTMPX_FILE=%s)\n", WTMPX_FILE);
 #endif
 #ifdef USE_LASTLOG
-  printf("\tUSE_LASTLOG (LASTLOG_FILE=%s)\n", LASTLOG_FILE);
+       printf("\tUSE_LASTLOG (LASTLOG_FILE=%s)\n", LASTLOG_FILE);
 #endif
-  printf("\n");
+       printf("\n");
 
 } /* showOptions() */
 
 
-int main(int argc, char *argv[])
+int
+main(int argc, char *argv[])
 {
-  printf("Platform-independent login recording test driver\n");
-
-  if (argc == 2) {
-    if (strncmp(argv[1], "-i", 3) == 0)
-      compile_opts_only = 1;
-    else if (strncmp(argv[1], "-v", 3) == 0)
-      be_verbose=1;
-  }
-      
-  if (!compile_opts_only) {
-    if (be_verbose && !testOutput())
-      return 1;
-    
-    if (!testAPI())
-      return 1;
-  }
-
-  showOptions();
-  
-  return 0;
+       printf("Platform-independent login recording test driver\n");
+
+       __progname = ssh_get_progname(argv[0]);
+       if (argc == 2) {
+               if (strncmp(argv[1], "-i", 3) == 0)
+                       compile_opts_only = 1;
+               else if (strncmp(argv[1], "-v", 3) == 0)
+                       be_verbose=1;
+       }
+
+       if (!compile_opts_only) {
+               if (be_verbose && !testOutput())
+                       return 1;
+
+               if (!testAPI())
+                       return 1;
+       }
+
+       showOptions();
+
+       return 0;
 } /* main() */
 
This page took 0.76683 seconds and 4 git commands to generate.