improve parsing of arguement to connect

[moira.git] / clients / mrtest / mrtest.c

diff --git a/clients/mrtest/mrtest.c b/clients/mrtest/mrtest.c
index 1e2c188f09084417d8f7061543d553836d3e617c..f4df4e417d36690fa92b687638f104e9755685c1 100644 (file)
--- a/clients/mrtest/mrtest.c
+++ b/clients/mrtest/mrtest.c
@@ -57,6 +57,7 @@ sms()
                com_err(whoami, status, 0);
                exit(1);
        }
+       exit(0);
 }
 
 test_noop()
@@ -79,11 +80,16 @@ test_connect(argc, argv)
 int argc;
 char *argv[];
 {
-       char *server = "";
+       char *server = "", serverbuf[256], *index();
        int status;
 
-       if (argc > 1)
-         server = argv[1];
+       if (argc > 1) {
+           server = argv[1];
+           if (index(server, ':') == NULL) {
+               server = serverbuf;
+               sprintf(serverbuf, "%s:%s", argv[1], "sms_db");
+           }
+       }
        status = sms_connect(server);
        if (status) ss_perror(ss, status, 0);
 }