From: Ken Hornstein <kenh@pobox.com>
Date: Thu, 2 Nov 2017 03:46:17 +0000 (-0400)
Subject: Start of making APPEND functional; not quite there yet.
X-Git-Url: https://diplodocus.org/git/nmh/commitdiff_plain/4d6f3230807e28cbf9a8739aba1fa83825a78aa5?ds=inline;hp=bd4d13521ec0229e8255d0ef8663a8541daca3d7

Start of making APPEND functional; not quite there yet.
---

diff --git a/uip/imaptest.c b/uip/imaptest.c
index 3bf3b2d2..8767685a 100644
--- a/uip/imaptest.c
+++ b/uip/imaptest.c
@@ -29,6 +29,7 @@
     X("queue", 0, QUEUESW) \
     X("noqueue", 0, NOQUEUESW) \
     X("append filename", 0, APPENDSW) \
+    X("afolder foldername", 0, AFOLDERSW) \
     X("timestamp", 0, TIMESTAMPSW) \
     X("notimestamp", 0, NOTIMESTAMPSW) \
     X("version", 0, VERSIONSW) \
@@ -46,7 +47,10 @@ struct imap_msg;
 
 struct imap_msg {
     char *command;		/* Command to send */
+    const char *folder;		/* Folder (for append) */
     bool queue;			/* If true, queue for later delivery */
+    bool append;		/* If true, append "command" to mbox */
+    size_t msgsize;		/* RFC822 size of message */
     struct imap_msg *next;	/* Next pointer */
 };
 
@@ -85,7 +89,11 @@ static void ts_report(struct timeval *tv, const char *fmt, ...)
 		      CHECK_PRINTF(2, 3);
 static void imap_negotiate_tls(netsec_context *);
 
-static void add_msg(bool queue, const char *fmt, ...) CHECK_PRINTF(2, 3);
+static void add_msg(bool queue, struct imap_msg **, const char *fmt, ...)
+		    CHECK_PRINTF(3, 4);
+static void add_append(const char *filename, const char *folder);
+
+static size_t rfc822size(const char *filename);
 
 static bool timestamp = false;
 
@@ -97,6 +105,7 @@ main (int argc, char **argv)
     int fd;
     char *saslmech = NULL, *host = NULL, *port = "143", *user = NULL;
     char *cp, **argp, buf[BUFSIZ], *oauth_svc = NULL, *errstr, **arguments, *p;
+    char *afolder = NULL;
     netsec_context *nsc = NULL;
     struct imap_msg *imsg;
     size_t len;
@@ -138,6 +147,18 @@ main (int argc, char **argv)
 		    die("missing argument to %s", argp[-2]);
 		continue;
 
+	    case AFOLDERSW:
+		if (!(afolder = *argp++) || *afolder == '-')
+		    die("missing argument to %s", argp[-2]);
+		continue;
+	    case APPENDSW:
+		if (!*argp || (**argp == '-'))
+		    die("missing argument to %s", argp[-2]);
+		if (! afolder)
+		    die("Append folder must be set with -afolder first");
+		add_append(*argp++, afolder);
+		continue;
+
 	    case SNOOPSW:
 		snoop = true;
 		continue;
@@ -186,9 +207,9 @@ main (int argc, char **argv)
 	} else if (*cp == '+') {
 	    if (*(cp + 1) == '\0')
 		die("Invalid null folder name");
-	    add_msg(0, "SELECT \"%s\"", cp + 1);
+	    add_msg(0, NULL, "SELECT \"%s\"", cp + 1);
 	} else {
-	    add_msg(queue, "%s", cp);
+	    add_msg(queue, NULL, "%s", cp);
 	}
     }
 
@@ -779,7 +800,7 @@ getline:
  */
 
 static void
-add_msg(bool queue, const char *fmt, ...)
+add_msg(bool queue, struct imap_msg **ret_imsg, const char *fmt, ...)
 {
     struct imap_msg *imsg;
     va_list ap;
@@ -798,6 +819,8 @@ add_msg(bool queue, const char *fmt, ...)
     imsg = mh_xmalloc(sizeof(*imsg));
 
     imsg->command = msg;
+    imsg->folder = NULL;
+    imsg->append = false;
     imsg->queue = queue;
     imsg->next = NULL;
 
@@ -808,6 +831,26 @@ add_msg(bool queue, const char *fmt, ...)
 	msgqueue_tail->next = imsg;
 	msgqueue_tail = imsg;
     }
+
+    if (ret_imsg)
+	*ret_imsg = imsg;
+}
+
+/*
+ * Add an APPEND command to the queue
+ */
+
+static void
+add_append(const char *filename, const char *folder)
+{
+    size_t filesize = rfc822size(filename);
+    struct imap_msg *imsg;
+
+    add_msg(false, &imsg, "%s", filename);
+
+    imsg->folder = folder;
+    imsg->append = true;
+    imsg->msgsize = filesize;
 }
 
 /*
@@ -852,3 +895,36 @@ ts_report(struct timeval *tv, const char *fmt, ...)
 
     printf(": %f sec\n", delta);
 }
+
+/*
+ * Calculate the RFC 822 size of file.
+ */
+
+static size_t
+rfc822size(const char *filename)
+{
+    FILE *f;
+    size_t total = 0, linecap = 0;
+    ssize_t rc;
+    char *line = NULL;
+
+    if (! (f = fopen(filename, "r")))
+	die("Unable to open %s: %s", filename, strerror(errno));
+
+    while ((rc = getline(&line, &linecap, f)) > 0) {
+	total += rc;
+	if (line[rc - 1] == '\n' && (rc == 1 || line[rc - 2] != '\r'))
+	    total++;
+	if (line[rc - 1] != '\n')
+	    total += 2;
+    }
+
+    free(line);
+
+    if (! feof(f))
+	die("Error while reading %s: %s", filename, strerror(errno));
+
+    fclose(f);
+
+    return total;
+}