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.
This commit is contained in:
Oswald Buddenhagen 2022-06-19 13:13:53 +02:00
parent abd31aad61
commit ad8520b741

View File

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