X-Git-Url: https://diplodocus.org/git/nmh/blobdiff_plain/56224eac0c7aa1912b4e0de4fe38b664888f0f2b..04ee0c3f6ad7e4177cc43a0c57f046013a4fbac9:/uip/imaptest.c

diff --git a/uip/imaptest.c b/uip/imaptest.c
index cd11aab5..be276fb1 100644
--- a/uip/imaptest.c
+++ b/uip/imaptest.c
@@ -5,9 +5,12 @@
  * complete copyright information.
  */
 
-#include <h/mh.h>
-#include <h/utils.h>
-#include <h/netsec.h>
+#include "h/mh.h"
+#include "sbr/print_version.h"
+#include "sbr/print_help.h"
+#include "sbr/error.h"
+#include "h/utils.h"
+#include "h/netsec.h"
 #include <stdarg.h>
 #include <sys/time.h>
 #include "h/done.h"
@@ -33,6 +36,7 @@
     X("batch filename", 0, BATCHSW) \
     X("timestamp", 0, TIMESTAMPSW) \
     X("notimestamp", 0, NOTIMESTAMPSW) \
+    X("timeout", 0, TIMEOUTSW) \
     X("version", 0, VERSIONSW) \
     X("help", 0, HELPSW) \
 
@@ -106,7 +110,7 @@ main (int argc, char **argv)
 {
     bool sasl = false, tls = false, initialtls = false;
     bool snoop = false, queue = false;
-    int fd;
+    int fd, timeout = 0;
     char *saslmech = NULL, *host = NULL, *port = "143", *user = NULL;
     char *cp, **argp, buf[BUFSIZ], *oauth_svc = NULL, *errstr, **arguments, *p;
     char *afolder = NULL;
@@ -169,6 +173,13 @@ main (int argc, char **argv)
 		batchfile(*argp++, afolder, queue);
 		continue;
 
+	    case TIMEOUTSW:
+		if (! *argp || (**argp == '-'))
+		    die("missing argument to %s", argp[-1]);
+		if (! (timeout = atoi(*argp++)))
+		    die("Invalid timeout: %s", argp[-1]);
+		continue;
+
 	    case SNOOPSW:
 		snoop = true;
 		continue;
@@ -233,6 +244,9 @@ main (int argc, char **argv)
 
     netsec_set_hostname(nsc, host);
 
+    if (timeout)
+	netsec_set_timeout(nsc, timeout);
+
     if (snoop)
 	netsec_set_snoop(nsc, 1);