lib-smtp: server: Add smtp_server_command_get_reply_count().

Returns the number of replies expected to the command.

diff --git a/src/lib-smtp/smtp-server-command.c b/src/lib-smtp/smtp-server-command.c
index aeb097d0cc2e203cf4dd6067b76654f92ecb1d57..1c0e526e463458c363ba7885bc2d9c44b602df05 100644 (file)
--- a/src/lib-smtp/smtp-server-command.c
+++ b/src/lib-smtp/smtp-server-command.c
@@ -387,6 +387,13 @@ void smtp_server_command_set_reply_count(struct smtp_server_command *cmd,
        cmd->replies_expected = count;
 }
 
+unsigned int
+smtp_server_command_get_reply_count(struct smtp_server_command *cmd)
+{
+       i_assert(cmd->replies_expected > 0);
+       return cmd->replies_expected;
+}
+
 void smtp_server_command_ready_to_reply(struct smtp_server_command *cmd)
 {
        cmd->state = SMTP_SERVER_COMMAND_STATE_READY_TO_REPLY;

diff --git a/src/lib-smtp/smtp-server.h b/src/lib-smtp/smtp-server.h
index e632ab40f391001ba03951136126bb8e6ac36241..957d41fd5f0bbd190a768b3dae798e5eb88119b8 100644 (file)
--- a/src/lib-smtp/smtp-server.h
+++ b/src/lib-smtp/smtp-server.h
@@ -502,6 +502,8 @@ void smtp_server_command_unregister(struct smtp_server *server,
 
 void smtp_server_command_set_reply_count(struct smtp_server_command *cmd,
                                         unsigned int count);
+unsigned int
+smtp_server_command_get_reply_count(struct smtp_server_command *cmd);
 
 void smtp_server_command_fail(struct smtp_server_command *cmd,
                              unsigned int status, const char *enh_code,