/*++/* NAME/* smtp-sink 8/* SUMMARY/* multi-threaded SMTP/LMTP test server/* SYNOPSIS/* smtp-sink [-cLpv] [-w delay] [host]:port backlog/* DESCRIPTION/* \\\\fIsmtp-sink\\\\fR listens on the named host (or address) and port./* It takes SMTP messages from the network and throws them away./* The purpose is to measure SMTP client performance, not protocol/* compliance./* This program is the complement of the \\\\fIsmtp-source\\\\fR program./* .IP -c/* Display a running counter that is updated whenever an SMTP/* QUIT command is executed./* .IP -L/* Speak LMTP rather than SMTP./* .IP -p/* Disable ESMTP command pipelining./* .IP -v/* Show the SMTP conversations./* .IP \\\"-w delay\\\"/* Wait \\\\fIdelay\\\\fR seconds before responding to a DATA command./* SEE ALSO/* smtp-source, SMTP/LMTP test message generator/* LICENSE/* .ad/* .fi/* The Secure Mailer license must be distributed with this software./* AUTHOR(S)/* Wietse Venema/* IBM T.J. Watson Research/* P.O. Box 704/* Yorktown Heights, NY 10598, USA/*--*//* System library. */#include #include #include #include #include #include #include #ifdef STRCASECMP_IN_STRINGS_H#include #endif/* Utility library. */#include #include #include #include #include #include #include #include #include #include /* Global library. */#include /* Application-specific. */typedef struct SINK_STATE { VSTREAM *stream; int data_state; int (*read) (struct SINK_STATE *); int rcpts;} SINK_STATE;#define ST_ANY 0#define ST_CR 1#define ST_CR_LF 2#define ST_CR_LF_DOT 3#define ST_CR_LF_DOT_CR 4#define ST_CR_LF_DOT_CR_LF 5static int var_tmout;static int var_max_line_length;static char *var_myhostname;static VSTRING *buffer;static int command_read(SINK_STATE *);static int data_read(SINK_STATE *);static void disconnect(SINK_STATE *);static int count;static int counter;static int disable_pipelining;static int fixed_delay;static int enable_lmtp;/* ehlo_response - respond to EHLO command */static void ehlo_response(SINK_STATE *state){ smtp_printf(state->stream, \\\"250-%s\\\", var_myhostname); if (!disable_pipelining) smtp_printf(state->stream, \\\"250-PIPELINING\\\"); smtp_printf(state->stream, \\\"250 8BITMIME\\\");}/* ok_response - send 250 OK */static void ok_response(SINK_STATE *state){ smtp_printf(state->stream, \\\"250 Ok\\\");}/* mail_response - reset recipient count, send 250 OK */static void mail_response(SINK_STATE *state){ state->rcpts = 0; ok_response(state);}/* rcpt_response - bump recipient count, send 250 OK */static void rcpt_response(SINK_STATE *state){ state->rcpts++; ok_response(state);}/* data_response - respond to DATA command */static void data_response(SINK_STATE *state){ state->data_state = ST_CR_LF; smtp_printf(state->stream, \\\"354 End data with .\\\"); state->read = data_read;}/* data_event - delayed response to DATA command */static void data_event(int unused_event, char *context){ SINK_STATE *state = (SINK_STATE *) context; data_response(state);}/* dot_response - response to . command */static void dot_response(SINK_STATE *state){ if (enable_lmtp) { while (state->rcpts-- > 0) /* XXX this could block */ ok_response(state); } else { ok_response(state); }}/* quit_response - respond to QUIT command */static void quit_response(SINK_STATE *state){ smtp_printf(state->stream, \\\"221 Bye\\\"); if (count) { counter++; vstream_printf(\\\"%d\\\\r\\\", counter); vstream_fflush(VSTREAM_OUT); }}/* data_read - read data from socket */static int data_read(SINK_STATE *state){ int ch; struct data_trans { int state; int want; int next_state; }; static struct data_trans data_trans[] = { ST_ANY, \\\'\\\\r\\\', ST_CR, ST_CR, \\\'\\\\n\\\', ST_CR_LF, ST_CR_LF, \\\'.\\\', ST_CR_LF_DOT, ST_CR_LF_DOT, \\\'\\\\r\\\', ST_CR_LF_DOT_CR, ST_CR_LF_DOT_CR, \\\'\\\\n\\\', ST_CR_LF_DOT_CR_LF, }; struct data_trans *dp; /* * We must avoid blocking I/O, so get out of here as soon as both the * VSTREAM and kernel read buffers dry up. */ while (vstream_peek(state->stream) > 0 || peekfd(vstream_fileno(state->stream)) > 0) { if ((ch = VSTREAM_GETC(state->stream)) == VSTREAM_EOF) return (-1); for (dp = data_trans; dp->state != state->data_state; dp++) /* void */ ; /* * Try to match the current character desired by the state machine. * If that fails, try to restart the machine with a match for its * first state. This covers the case of a CR/LF/CR/LF sequence * (empty line) right before the end of the message data. */ if (ch == dp->want) state->data_state = dp->next_state; else if (ch == data_trans[0].want) state->data_state = data_trans[0].next_state; else state->data_state = ST_ANY; if (state->data_state == ST_CR_LF_DOT_CR_LF) { if (msg_verbose) msg_info(\\\".\\\"); dot_response(state); state->read = command_read; break; } } return (0);} /* * The table of all SMTP commands that we can handle. */typedef struct SINK_COMMAND { char *name; void (*response) (SINK_STATE *);} SINK_COMMAND;static SINK_COMMAND command_table[] = { \\\"helo\\\", ok_response, \\\"ehlo\\\", ehlo_response, \\\"lhlo\\\", ehlo_response, \\\"mail\\\", mail_response, \\\"rcpt\\\", rcpt_response, \\\"data\\\", data_response, \\\"rset\\\", ok_response, \\\"noop\\\", ok_response, \\\"vrfy\\\", ok_response, \\\"quit\\\", quit_response, 0,};/* command_read - talk the SMTP protocol, server side */static int command_read(SINK_STATE *state){ char *command; SINK_COMMAND *cmdp; smtp_get(buffer, state->stream, var_max_line_length); if ((command = strtok(vstring_str(buffer), \\\" \\\\t\\\")) == 0) { smtp_printf(state->stream, \\\"500 Error: unknown command\\\"); return (0); } if (msg_verbose) msg_info(\\\"%s\\\", command); for (cmdp = command_table; cmdp->name != 0; cmdp++) if (strcasecmp(command, cmdp->name) == 0) break; if (cmdp->name == 0) { smtp_printf(state->stream, \\\"500 Error: unknown command\\\"); return (0); } if (cmdp->response == data_response && fixed_delay > 0) { event_request_timer(data_event, (char *) state, fixed_delay); } else { cmdp->response(state); if (cmdp->response == quit_response) return (-1); } return (0);}/* read_event - handle command or data read events */static void read_event(int unused_event, char *context){ SINK_STATE *state = (SINK_STATE *) context; do { switch (vstream_setjmp(state->stream)) { default: msg_panic(\\\"unknown error reading input\\\"); case SMTP_ERR_TIME: msg_panic(\\\"attempt to read non-readable socket\\\"); /* NOTREACHED */ case SMTP_ERR_EOF: msg_warn(\\\"lost connection\\\"); disconnect(state); return; case 0: if (state->read(state) < 0) { if (msg_verbose) msg_info(\\\"disconnect\\\"); disconnect(state); return; } } } while (vstream_peek(state->stream) > 0);}/* disconnect - handle disconnection events */static void disconnect(SINK_STATE *state){ event_disable_readwrite(vstream_fileno(state->stream)); vstream_fclose(state->stream); myfree((char *) state);}/* connect_event - handle connection events */static void connect_event(int unused_event, char *context){ int sock = (int) context; SINK_STATE *state; int fd; if ((fd = accept(sock, (struct sockaddr *) 0, (SOCKADDR_SIZE *) 0)) >= 0) { if (msg_verbose) msg_info(\\\"connect\\\"); non_blocking(fd, NON_BLOCKING); state = (SINK_STATE *) mymalloc(sizeof(*state)); state->stream = vstream_fdopen(fd, O_RDWR); state->read = command_read; state->data_state = 0; smtp_timeout_setup(state->stream, var_tmout); smtp_printf(state->stream, \\\"220 %s ESMTP\\\", var_myhostname); event_enable_read(fd, read_event, (char *) state); }}/* usage - explain */static void usage(char *myname){ msg_fatal(\\\"usage: %s [-cLpv] [host]:port backlog\\\", myname);}int main(int argc, char **argv){ int sock; int backlog; int ch; /* * Initialize diagnostics. */ msg_vstream_init(argv[0], VSTREAM_ERR); /* * Parse JCL. */ while ((ch = GETOPT(argc, argv, \\\"cLpvw:\\\")) > 0) { switch (ch) { case \\\'c\\\': count++; break; case \\\'L\\\': enable_lmtp = 1; break; case \\\'p\\\': disable_pipelining = 1; break; case \\\'v\\\': msg_verbose++; break; case \\\'w\\\': if ((fixed_delay = atoi(optarg)) <= 0) usage(argv[0]); break; default: usage(argv[0]); } } if (argc - optind != 2) usage(argv[0]); if ((backlog = atoi(argv[optind + 1])) <= 0) usage(argv[0]); /* * Initialize. */ buffer = vstring_alloc(1024); var_myhostname = \\\"smtp-sink\\\"; sock = inet_listen(argv[optind], backlog, BLOCKING); /* * Start the event handler. */ event_enable_read(sock, connect_event, (char *) sock); for (;;) event_loop(-1);} |
