IMAP auth test function;

src/tests/testimap.c

@@ -31,11 +31,174 @@
 #include <string.h>
 #include <openssl/ssl.h>
 #include <openssl/err.h>
+#include <stdarg.h>
+#include <getopt.h>
 
-#include <eport.h>
+#include <ejabbermsg.h>
 #include <tlsport.h>
+#include <imapf.h>
+
+#define PRG_NAME  "ejabberauth imap function test"
+
+/*
+ * The idea is stupid simple: take hostname, port, username and password
+ * move this data to the jabber message manually, connect to the host,
+ * run the function, check result and disconnect.
+ * Extra option -e (--expected-result) ok/fail to check the output from
+ * imap according to the expectations
+ */
+
+static void print_help(const char *cname)
+{
+  fprintf(stdout, "\n%s - %s\n\n", cname, PRG_NAME);
+  fprintf(stdout, "USAGE:\n");
+  fprintf(stdout, "\t%s -H | --host <IMAP hostname> -p | --port <port> -u | --user <username>"
+          " -P | --password <password> [ -e | --expected-result <ok | fail>] [ -v | --verbose ]",
+          cname);
+  fprintf(stdout, "\n\n\t%s -h | --help\n", cname);
+
+  return;
+}
+
+static int ejda_username_split(char *buf, char **user, char **domain)
+{
+  char *u, *d;
+
+  if(!buf) return 1;
+
+  u = strchr(buf, '@');
+  if(!u) return 1;
+  else {
+    *u = '\0';
+    *u++;
+    d = u; u = buf;
+  }
+
+  *user = u; *domain = d;
+
+  return 0;
+}
 
 int main(int argc, char **argv)
 {
+  char *host, *port, *user, *password;
+  char *utbuf = NULL;
+  int verbose = 0, opt, expected = 0;
+  struct ejabber_msg m;
+  struct tlsport p;
+
+  host = port = user = password = NULL;
+
+  /* parse options */
+  while(1) {
+    int option_index = 0;
+    static struct option long_options[] = {
+      {"help", no_argument, NULL, 'h'},
+      {"host", required_argument, NULL, 'H'},
+      {"port", required_argument, NULL, 'p'},
+      {"user", required_argument, NULL, 'u'},
+      {"password", required_argument, NULL, 'P'},
+      {"expected-result", required_argument, NULL, 'e'},
+      {"verbose", no_argument, NULL, 'v'},
+      {NULL, 0, NULL, 0},
+    };
+
+    if((opt = getopt_long(argc, argv, "hH:p:u:P:e:v", long_options,
+                          &option_index)) == -1) break;
+    switch(opt) {
+    case 'h':
+      /* print help information and exit */
+      print_help(argv[0]);
       return 0;
+      break;
+    case 'H': host = optarg; break;
+    case 'p': port = optarg; break;
+    case 'u': user = optarg; break;
+    case 'P': password = optarg; break;
+    case 'e':
+      if(!strcmp(optarg, "fail")) expected = 1;
+      else expected = 0;
+      break;
+    case 'v': verbose = 1; break;
+    default: abort(); break; /* something that's shouldn't ever happen */
     }
+  }
+
+  if(!host) {
+    fprintf(stderr, "Error: no hostname given.\n");
+  main_einval:
+    print_help(argv[0]);
+    return EINVAL;
+  }
+  if(!port) {
+    fprintf(stderr, "Error: no port given.\n");
+    goto main_einval;
+  }
+  if(!password) {
+    fprintf(stderr, "Error: no password provided.\n");
+    goto main_einval;
+  }
+
+  /* split user name */
+  opt = strlen(user) + sizeof(char);
+  if((utbuf = malloc(opt)) == NULL) {
+    fprintf(stderr, "Error: not enough memory\n");
+    return ENOMEM;
+  } else {
+    memset(utbuf, '\0', opt);
+    memcpy(utbuf, user, opt - sizeof(char));
+    if(ejda_username_split(utbuf, &m.user, &m.domain)) {
+      fprintf(stderr, "Error: username is invalid.\n");
+      free(utbuf);
+      return EINVAL;
+    }
+  }
+  /* prepare other data in the message */
+  m.password = password;
+
+  /* TLS stuff */
+  ssllib_init();
+
+  if(verbose) {
+    fprintf(stdout, "Connecting to TLS port %s:%s ... ", host, port);
+    fflush(stdout);
+  }
+
+  if(tls_connect(host, port, &p)) {
+    if(!verbose)
+      fprintf(stderr, "Error: failed to connect TLS port.\n");
+    else fprintf(stdout, " failed\n");
+    opt = -1;
+    goto mainmoi;
+  }
+  if(verbose) fprintf(stdout, " ok\n");
+
+  /* IMAP auth */
+  if(verbose) {
+    fprintf(stdout, "IMAP auth u=%s:d=%s:p=%s ... ", m.user, m.domain, m.password);
+    fflush(stdout);
+  }
+
+  opt = imap_auth(&p, &m);
+  /* invert result */
+  if(expected && opt) opt = 0;
+  if(expected && !opt) opt = 1;
+
+  if(opt) {
+    if(!verbose) fprintf(stderr, "IMAP auth returns not expected result.\n");
+    else fprintf(stdout, " failed\n");
+    opt = -1;
+  } else {
+    opt = 0;
+    if(verbose) fprintf(stdout, " ok\n");
+  }
+
+  tls_close(&p);
+
+ mainmoi:
+  ssllib_free();
+  free(utbuf);
+
+  return opt;
+}
+