1 files changed, 12 insertions, 27 deletions

diff --git a/sshd.c b/sshd.c
index b7066df5c..7cc7044be 100644
--- a/sshd.c
+++ b/sshd.c
@@ -1,4 +1,4 @@
-/* $OpenBSD: sshd.c,v 1.390 2012/04/12 02:42:32 djm Exp $ */
+/* $OpenBSD: sshd.c,v 1.391 2012/05/13 01:42:32 dtucker Exp $ */
 /*
  * Author: Tatu Ylonen <ylo@cs.hut.fi>
  * Copyright (c) 1995 Tatu Ylonen <ylo@cs.hut.fi>, Espoo, Finland
@@ -1325,14 +1325,14 @@ main(int ac, char **av)
         int opt, i, j, on = 1;
         int sock_in = -1, sock_out = -1, newsock = -1;
         const char *remote_ip;
-        char *test_user = NULL, *test_host = NULL, *test_addr = NULL;
         int remote_port;
-        char *line, *p, *cp;
+        char *line;
         int config_s[2] = { -1 , -1 };
         u_int64_t ibytes, obytes;
         mode_t new_umask;
         Key *key;
         Authctxt *authctxt;
+        struct connection_info *connection_info = get_connection_info(0, 0);
 
 #ifdef HAVE_SECUREWARE
         (void)set_auth_parameters(ac, av);
@@ -1454,20 +1454,9 @@ main(int ac, char **av)
                         test_flag = 2;
                         break;
                 case 'C':
-                        cp = optarg;
-                        while ((p = strsep(&cp, ",")) && *p != '\0') {
-                                if (strncmp(p, "addr=", 5) == 0)
-                                        test_addr = xstrdup(p + 5);
-                                else if (strncmp(p, "host=", 5) == 0)
-                                        test_host = xstrdup(p + 5);
-                                else if (strncmp(p, "user=", 5) == 0)
-                                        test_user = xstrdup(p + 5);
-                                else {
-                                        fprintf(stderr, "Invalid test "
-                                            "mode specification %s\n", p);
-                                        exit(1);
-                                }
-                        }
+                        if (parse_server_match_testspec(connection_info,
+                            optarg) == -1)
+                                exit(1);
                         break;
                 case 'u':
                         utmp_len = (u_int)strtonum(optarg, 0, MAXHOSTNAMELEN+1, NULL);
@@ -1479,7 +1468,7 @@ main(int ac, char **av)
                 case 'o':
                         line = xstrdup(optarg);
                         if (process_server_config_line(&options, line,
-                            "command-line", 0, NULL, NULL, NULL, NULL) != 0)
+                            "command-line", 0, NULL, NULL) != 0)
                                 exit(1);
                         xfree(line);
                         break;
@@ -1535,13 +1524,10 @@ main(int ac, char **av)
          * the parameters we need.  If we're not doing an extended test,
          * do not silently ignore connection test params.
          */
-        if (test_flag >= 2 &&
-           (test_user != NULL || test_host != NULL || test_addr != NULL)
-            && (test_user == NULL || test_host == NULL || test_addr == NULL))
+        if (test_flag >= 2 && server_match_spec_complete(connection_info) == 0)
                 fatal("user, host and addr are all required when testing "
                    "Match configs");
-        if (test_flag < 2 && (test_user != NULL || test_host != NULL ||
-            test_addr != NULL))
+        if (test_flag < 2 && server_match_spec_complete(connection_info) >= 0)
                 fatal("Config test connection parameter (-C) provided without "
                    "test mode (-T)");
 
@@ -1553,7 +1539,7 @@ main(int ac, char **av)
                 load_server_config(config_file_name, &cfg);
 
         parse_server_config(&options, rexeced_flag ? "rexec" : config_file_name,
-            &cfg, NULL, NULL, NULL);
+            &cfg, NULL);
 
         seed_rng();
 
@@ -1715,9 +1701,8 @@ main(int ac, char **av)
         }
 
         if (test_flag > 1) {
-                if (test_user != NULL && test_addr != NULL && test_host != NULL)
-                        parse_server_match_config(&options, test_user,
-                            test_host, test_addr);
+                if (server_match_spec_complete(connection_info) == 1)
+                        parse_server_match_config(&options, connection_info);
                 dump_config(&options);
         }