postscreen_state.c [plain text]
#include <sys_defs.h>
#include <msg.h>
#include <mymalloc.h>
#include <name_mask.h>
#include <htable.h>
#include <mail_proto.h>
#include <mail_server.h>
#include <postscreen.h>
PSC_STATE *psc_new_session_state(VSTREAM *stream,
const char *client_addr,
const char *client_port,
const char *server_addr,
const char *server_port)
{
PSC_STATE *state;
state = (PSC_STATE *) mymalloc(sizeof(*state));
if ((state->smtp_client_stream = stream) != 0)
psc_check_queue_length++;
state->smtp_server_fd = (-1);
state->smtp_client_addr = mystrdup(client_addr);
state->smtp_client_port = mystrdup(client_port);
state->smtp_server_addr = mystrdup(server_addr);
state->smtp_server_port = mystrdup(server_port);
state->send_buf = vstring_alloc(100);
state->test_name = "TEST NAME HERE";
state->dnsbl_reply = 0;
state->final_reply = "421 4.3.2 Service currently unavailable\r\n";
state->rcpt_reply = "450 4.3.2 Service currently unavailable\r\n";
state->command_count = 0;
state->protocol = MAIL_PROTO_SMTP;
state->helo_name = 0;
state->sender = 0;
state->cmd_buffer = 0;
state->read_state = 0;
state->ehlo_discard_mask = 0;
state->expand_buf = 0;
state->where = PSC_SMTPD_CMD_CONNECT;
if (psc_stress == 0
&& psc_check_queue_length >= psc_hiwat_check_queue_length) {
psc_stress = 1;
msg_info("entering STRESS mode with %d connections",
psc_check_queue_length);
}
if ((state->client_info = (PSC_CLIENT_INFO *)
htable_find(psc_client_concurrency, client_addr)) == 0) {
state->client_info = (PSC_CLIENT_INFO *)
mymalloc(sizeof(state->client_info[0]));
(void) htable_enter(psc_client_concurrency, client_addr,
(void *) state->client_info);
PSC_INIT_TESTS(state);
state->client_info->concurrency = 1;
state->client_info->pass_new_count = 0;
} else {
PSC_INIT_TEST_FLAGS_ONLY(state);
state->client_info->concurrency += 1;
}
return (state);
}
void psc_free_session_state(PSC_STATE *state)
{
const char *myname = "psc_free_session_state";
HTABLE_INFO *ht;
if ((ht = htable_locate(psc_client_concurrency,
state->smtp_client_addr)) == 0)
msg_panic("%s: unknown client address: %s",
myname, state->smtp_client_addr);
if (--(state->client_info->concurrency) == 0)
htable_delete(psc_client_concurrency, state->smtp_client_addr, myfree);
if (state->smtp_client_stream != 0) {
event_server_disconnect(state->smtp_client_stream);
psc_check_queue_length--;
}
if (state->smtp_server_fd >= 0) {
close(state->smtp_server_fd);
psc_post_queue_length--;
}
if (state->send_buf != 0)
state->send_buf = vstring_free(state->send_buf);
myfree(state->smtp_client_addr);
myfree(state->smtp_client_port);
myfree(state->smtp_server_addr);
myfree(state->smtp_server_port);
if (state->dnsbl_reply)
vstring_free(state->dnsbl_reply);
if (state->helo_name)
myfree(state->helo_name);
if (state->sender)
myfree(state->sender);
if (state->cmd_buffer)
vstring_free(state->cmd_buffer);
if (state->expand_buf)
vstring_free(state->expand_buf);
myfree((void *) state);
if (psc_check_queue_length < 0 || psc_post_queue_length < 0)
msg_panic("bad queue length: check_queue=%d, post_queue=%d",
psc_check_queue_length, psc_post_queue_length);
if (psc_stress != 0
&& psc_check_queue_length <= psc_lowat_check_queue_length) {
psc_stress = 0;
msg_info("leaving STRESS mode with %d connections",
psc_check_queue_length);
}
}
const char *psc_print_state_flags(int flags, const char *context)
{
static const NAME_MASK flags_mask[] = {
"NOFORWARD", PSC_STATE_FLAG_NOFORWARD,
"USING_TLS", PSC_STATE_FLAG_USING_TLS,
"NEW", PSC_STATE_FLAG_NEW,
"BLIST_FAIL", PSC_STATE_FLAG_BLIST_FAIL,
"HANGUP", PSC_STATE_FLAG_HANGUP,
"WLIST_FAIL", PSC_STATE_FLAG_WLIST_FAIL,
"PREGR_FAIL", PSC_STATE_FLAG_PREGR_FAIL,
"PREGR_PASS", PSC_STATE_FLAG_PREGR_PASS,
"PREGR_TODO", PSC_STATE_FLAG_PREGR_TODO,
"PREGR_DONE", PSC_STATE_FLAG_PREGR_DONE,
"DNSBL_FAIL", PSC_STATE_FLAG_DNSBL_FAIL,
"DNSBL_PASS", PSC_STATE_FLAG_DNSBL_PASS,
"DNSBL_TODO", PSC_STATE_FLAG_DNSBL_TODO,
"DNSBL_DONE", PSC_STATE_FLAG_DNSBL_DONE,
"PIPEL_FAIL", PSC_STATE_FLAG_PIPEL_FAIL,
"PIPEL_PASS", PSC_STATE_FLAG_PIPEL_PASS,
"PIPEL_TODO", PSC_STATE_FLAG_PIPEL_TODO,
"PIPEL_SKIP", PSC_STATE_FLAG_PIPEL_SKIP,
"NSMTP_FAIL", PSC_STATE_FLAG_NSMTP_FAIL,
"NSMTP_PASS", PSC_STATE_FLAG_NSMTP_PASS,
"NSMTP_TODO", PSC_STATE_FLAG_NSMTP_TODO,
"NSMTP_SKIP", PSC_STATE_FLAG_NSMTP_SKIP,
"BARLF_FAIL", PSC_STATE_FLAG_BARLF_FAIL,
"BARLF_PASS", PSC_STATE_FLAG_BARLF_PASS,
"BARLF_TODO", PSC_STATE_FLAG_BARLF_TODO,
"BARLF_SKIP", PSC_STATE_FLAG_BARLF_SKIP,
0,
};
return (str_name_mask_opt((VSTRING *) 0, context, flags_mask, flags,
NAME_MASK_PIPE | NAME_MASK_NUMBER));
}