Initial revision

1 files changed, 278 insertions, 0 deletions

diff --git a/kmicromail/libetpan/tests/smtpsend.c b/kmicromail/libetpan/tests/smtpsend.c
new file mode 100644
index 0000000..d5229d9
--- a/dev/null
+++ b/kmicromail/libetpan/tests/smtpsend.c
@@ -0,0 +1,278 @@
+/*
+ * Simple Mail Submission Agent using SMTP with libEtPan!
+ * TODO: Full sendmail like interface
+ */
+
+#include <libetpan/libetpan.h>
+#include <netdb.h>
+#include <netinet/in.h>
+#include <sys/socket.h>
+#include <stdlib.h>
+#include <string.h>
+#include <sys/types.h>
+#include <sys/mman.h>
+#include <sys/stat.h>
+#include <unistd.h>
+#include <sys/ioctl.h>
+#include <fcntl.h>
+#include <pwd.h>
+
+#define _GNU_SOURCE
+#include <getopt.h>
+
+/* globals */
+char *smtp_server;
+uint smtp_port = 25;
+char *smtp_user;
+char *smtp_password;
+char *smtp_from;
+int smtp_tls = 0;
+int smtp_esmtp = 1;
+
+struct mem_message {
+  char *data;
+  size_t len;
+  MMAPString *mstring;
+};
+
+#define BLOCKSIZE 4096
+
+int collect(struct mem_message *message) {
+  struct stat sb;
+  int len;
+
+  memset(message, 0, sizeof(struct mem_message));
+
+  /* if stdin is a file whose size is known, try to mmap it */
+  if (!fstat(0, &sb) && S_ISREG(sb.st_mode) && sb.st_size >= 0) {
+    message->len = sb.st_size;
+    if ((message->data = mmap(NULL, message->len, PROT_READ, MAP_SHARED,
+                              STDIN_FILENO, 0)) != MAP_FAILED)
+      return 0;
+  }
+
+  /* read the buffer from stdin by blocks, until EOF or error.
+     save the message in a mmap_string */
+  if ((message->mstring = mmap_string_sized_new(BLOCKSIZE)) == NULL) {
+    perror("mmap_string_new");
+    goto error;
+  }
+  message->len = 0;
+
+  while ((len = read(STDIN_FILENO,
+                     message->mstring->str + message->len, BLOCKSIZE)) > 0) {
+    message->len += len;
+    /* reserve room for next block */
+    if ((mmap_string_set_size(message->mstring,
+                              message->len + BLOCKSIZE)) == NULL) {
+      perror("mmap_string_set_size");
+      goto error;
+    }
+  }
+
+  if (len == 0) {
+    message->data = message->mstring->str;
+    return 0; /* OK */
+  }
+
+  perror("read");
+
+ error:
+  if (message->mstring != NULL)
+    mmap_string_free(message->mstring);
+  return -1;
+}
+
+char *guessfrom() {
+  uid_t uid;
+  struct passwd *pw;
+  char hostname[256];
+  int len;
+  char *gfrom;
+
+  if (gethostname(hostname, sizeof(hostname))) {
+    perror("gethostname");
+    return NULL;
+  }
+  hostname[sizeof(hostname) - 1] = '\0';
+
+  uid = getuid();
+  pw = getpwuid(uid);
+  
+  len = ((pw != NULL) ? strlen(pw->pw_name) : 12)
+    + strlen(hostname) + 2;
+
+  if ((gfrom = malloc(len)) == NULL) {
+    perror("malloc");
+    return NULL;
+  }
+  if (pw != NULL && pw->pw_name != NULL) 
+    snprintf(gfrom, len, "%s@%s", pw->pw_name, hostname);
+  else
+    snprintf(gfrom, len, "#%u@%s", uid, hostname);
+  return gfrom;
+}
+
+void release(struct mem_message *message) {
+  if (message->mstring != NULL)
+    mmap_string_free(message->mstring);
+  else if (message->data != NULL)
+    munmap(message->data, message->len);
+}
+
+int send_message(char *data, size_t len, char**rcpts) {
+  int s = -1;
+  int ret;
+  char **r;
+  int esmtp = 0;
+  mailsmtp *smtp = NULL;
+
+  if ((smtp = mailsmtp_new(0, NULL)) == NULL) {
+    perror("mailsmtp_new");
+    goto error;
+  }
+
+  /* first open the stream */
+  if ((ret = mailsmtp_socket_connect(smtp, 
+                                     (smtp_server != NULL ? smtp_server : "localhost"),
+                                     smtp_port)) != MAILSMTP_NO_ERROR) {
+    fprintf(stderr, "mailsmtp_socket_connect: %s\n", mailsmtp_strerror(ret));
+    goto error;
+  }
+  
+  /* then introduce ourselves */
+  if (smtp_esmtp && (ret = mailesmtp_ehlo(smtp)) == MAILSMTP_NO_ERROR)
+    esmtp = 1;
+  else if (!smtp_esmtp || ret == MAILSMTP_ERROR_NOT_IMPLEMENTED)
+    ret = mailsmtp_helo(smtp);
+  if (ret != MAILSMTP_NO_ERROR) {
+    fprintf(stderr, "mailsmtp_helo: %s\n", mailsmtp_strerror(ret));
+    goto error;
+  }
+
+  if (esmtp && smtp_tls &&
+      (ret = mailsmtp_socket_starttls(smtp)) != MAILSMTP_NO_ERROR) {
+    fprintf(stderr, "mailsmtp_starttls: %s\n", mailsmtp_strerror(ret));
+    goto error;
+  }
+  
+  if (esmtp && smtp_user != NULL &&
+      (ret = mailsmtp_auth(smtp, smtp_user,
+                           (smtp_password != NULL) ? smtp_password : ""))
+      != MAILSMTP_NO_ERROR) {
+    fprintf(stderr, "mailsmtp_auth: %s: %s\n", smtp_user, mailsmtp_strerror(ret));
+    goto error;
+  }
+
+  /* source */
+  if ((ret = (esmtp ?
+              mailesmtp_mail(smtp, smtp_from, 1, "etPanSMTPTest") :
+              mailsmtp_mail(smtp, smtp_from))) != MAILSMTP_NO_ERROR) {
+    fprintf(stderr, "mailsmtp_mail: %s, %s\n", smtp_from, mailsmtp_strerror(ret));
+    goto error;
+  }
+  
+  /* recipients */
+  for (r = rcpts; *r != NULL; r++) {
+    if ((ret = (esmtp ?
+                mailesmtp_rcpt(smtp, *r,
+                               MAILSMTP_DSN_NOTIFY_FAILURE|MAILSMTP_DSN_NOTIFY_DELAY,
+                               NULL) :
+                mailsmtp_rcpt(smtp, *r))) != MAILSMTP_NO_ERROR) {
+      fprintf(stderr, "mailsmtp_rcpt: %s: %s\n", *r, mailsmtp_strerror(ret));
+      goto error;
+    }
+  }
+  
+  /* message */
+  if ((ret = mailsmtp_data(smtp)) != MAILSMTP_NO_ERROR) {
+    fprintf(stderr, "mailsmtp_data: %s\n", mailsmtp_strerror(ret));
+    goto error;
+  }
+  if ((ret = mailsmtp_data_message(smtp, data, len)) != MAILSMTP_NO_ERROR) {
+    fprintf(stderr, "mailsmtp_data_message: %s\n", mailsmtp_strerror(ret));
+    goto error;
+  }
+  mailsmtp_free(smtp);
+  return 0;
+
+ error:
+  if (smtp != NULL)
+    mailsmtp_free(smtp);
+  if (s >= 0)
+    close(s);
+  return -1;
+}
+
+int main(int argc, char **argv) {
+  struct mem_message message;
+  int index, r;
+
+  static struct option long_options[] = {
+    {"server",   1, 0, 's'},
+    {"port",     1, 0, 'p'},
+    {"user",     1, 0, 'u'},
+    {"password", 1, 0, 'v'},
+    {"from",     1, 0, 'f'},
+    {"tls",      0, 0, 'S'},
+    {"no-esmtp", 0, 0, 'E'},
+  };
+
+  while(1) {
+    if ((r = getopt_long(argc, argv, "s:p:u:v:f:SE", long_options, &index)) < 0)
+      break;
+    switch (r) {
+    case 's':
+      if (smtp_server != NULL)
+        free(smtp_server);
+      smtp_server = strdup(optarg);
+      break;
+    case 'p':
+      smtp_port = strtoul(optarg, NULL, 10);
+      break;
+    case 'u':
+      if (smtp_user != NULL)
+        free(smtp_user);
+      smtp_user = strdup(optarg);
+      break;
+    case 'v':
+      if (smtp_password != NULL)
+        free(smtp_password);
+      smtp_password = strdup(optarg);
+      break;
+    case 'f':
+      if (smtp_from != NULL)
+        free(smtp_from);
+      smtp_from = strdup(optarg);
+      break;
+    case 'S':
+      smtp_tls = 1;
+      break;
+    case 'E':
+      smtp_esmtp = 0;
+      break;
+    }
+  }
+
+  argc -= optind;
+  argv += optind;
+
+  if (argc < 1) {
+    fprintf(stderr, "usage: smtpsend [-f from] [-u user] [-v password] [-s server] [-p port] [-S] <rcpts>...\n");
+    return EXIT_FAILURE;
+  }
+
+  if (smtp_from == NULL && (smtp_from = guessfrom()) == NULL) {
+    fprintf(stderr, "can't guess a valid from, please use -f option.\n");
+    return EXIT_FAILURE;
+  }
+  
+  /* reads message from stdin */
+  if (collect(&message))
+    return EXIT_FAILURE;
+  
+  send_message(message.data, message.len, argv);
+
+  release(&message);
+  return EXIT_SUCCESS;
+}