From ad8520b74108000715115b732c6e80ace28481e8 Mon Sep 17 00:00:00 2001
From: Oswald Buddenhagen <ossi@users.sf.net>
Date: Sun, 19 Jun 2022 13:13:53 +0200
Subject: [PATCH] revamp sending of IMAP commands with literals

include the literal's length into the command itself, so error messages
become clearer. the commands are initially built for LITERAL+, and are
"edited down" right before sending if necessary.
---
 src/drv_imap.c | 37 ++++++++++++++++++-------------------
 1 file changed, 18 insertions(+), 19 deletions(-)

diff --git a/src/drv_imap.c b/src/drv_imap.c
index 8bae526..59fbd4c 100644
--- a/src/drv_imap.c
+++ b/src/drv_imap.c
@@ -302,24 +302,23 @@ done_imap_cmd( imap_store_t *ctx, imap_cmd_t *cmd, int response )
 static void
 send_imap_cmd( imap_store_t *ctx, imap_cmd_t *cmd )
 {
-	int litplus, iovcnt = 3;
-	uint tbufl, lbufl;
+	int litplus = 0, iovcnt = 3;
 	conn_iovec_t iov[5];
 	char tagbuf[16];
-	char lenbuf[16];
 
 	cmd->tag = ++ctx->nexttag;
-	tbufl = nfsnprintf( tagbuf, sizeof(tagbuf), "%d ", cmd->tag );
-	if (!cmd->param.data) {
-		memcpy( lenbuf, "\r\n", 3 );
-		lbufl = 2;
-		litplus = 0;
-	} else if ((cmd->param.to_trash && ctx->trashnc == TrashUnknown) || !CAP(LITERALPLUS) || cmd->param.data_len >= 100*1024) {
-		lbufl = nfsnprintf( lenbuf, sizeof(lenbuf), "{%u}\r\n", cmd->param.data_len );
-		litplus = 0;
-	} else {
-		lbufl = nfsnprintf( lenbuf, sizeof(lenbuf), "{%u+}\r\n", cmd->param.data_len );
-		litplus = 1;
+	uint tbufl = nfsnprintf( tagbuf, sizeof(tagbuf), "%d ", cmd->tag );
+	uint cmdl = strlen( cmd->cmd );
+	if (cmd->param.data) {
+		assert( cmdl > 2 && !memcmp( cmd->cmd + cmdl - 2, "+}", 2 ) );
+		if ((!cmd->param.to_trash || ctx->trashnc != TrashUnknown) &&
+		    CAP(LITERALPLUS) && cmd->param.data_len <= 100*1024) {
+			litplus = 1;
+		} else {
+			cmd->cmd[cmdl - 2] = '}';
+			cmd->cmd[cmdl - 1] = 0;
+			cmdl--;
+		}
 	}
 	if (DFlags & DEBUG_NET) {
 		if (ctx->num_in_progress)
@@ -329,17 +328,17 @@ send_imap_cmd( imap_store_t *ctx, imap_cmd_t *cmd )
 		else if (starts_with( cmd->cmd, -1, "AUTHENTICATE PLAIN", 18 ))
 			printf( "%s>>> %sAUTHENTICATE PLAIN <authdata>\r\n", ctx->label, tagbuf );
 		else
-			printf( "%s>>> %s%s%s", ctx->label, tagbuf, cmd->cmd, lenbuf );
+			printf( "%s>>> %s%s\r\n", ctx->label, tagbuf, cmd->cmd );
 		fflush( stdout );
 	}
 	iov[0].buf = tagbuf;
 	iov[0].len = tbufl;
 	iov[0].takeOwn = KeepOwn;
 	iov[1].buf = cmd->cmd;
-	iov[1].len = strlen( cmd->cmd );
+	iov[1].len = cmdl;
 	iov[1].takeOwn = KeepOwn;
-	iov[2].buf = lenbuf;
-	iov[2].len = lbufl;
+	iov[2].buf = "\r\n";
+	iov[2].len = 2;
 	iov[2].takeOwn = KeepOwn;
 	if (litplus) {
 		if (DFlags & DEBUG_NET_ALL) {
@@ -3264,7 +3263,7 @@ DIAG_POP
 		}
 	}
 	imap_exec( ctx, &cmd->gen, imap_store_msg_p2,
-	           "APPEND \"%\\s\" %s%s", buf, flagstr, datestr );
+	           "APPEND \"%\\s\" %s%s{%u+}", buf, flagstr, datestr, data->len );
 	free( buf );
 }