static int client_export_iter_session(struct client *client)
{
struct client_export_cmd *cmd = client->cmd_export;
struct mail_session *session = client->mail_session_iter;
i_assert(cmd->level == MAIL_EXPORT_LEVEL_SESSION);
mail_session_unref(&client->mail_session_iter);
if (!cmd->header_sent) {
o_stream_nsend_str(client->output,
"session\tuser\tip\tservice\tpid\tconnected"
"\tlast_update\tnum_cmds"
MAIL_STATS_HEADER);
cmd->header_sent = TRUE;
}
for (; session != NULL; session = session->stable_next) {
if (client_is_busy(client))
break;
if (!mail_export_filter_match_session(&cmd->filter, session))
continue;
str_truncate(cmd->str, 0);
T_BEGIN {
str_append(cmd->str, guid_128_to_string(session->guid));
str_append_c(cmd->str, '\t');
str_append_tabescaped(cmd->str, session->user->name);
str_append_c(cmd->str, '\t');
if (session->ip != NULL) {
str_append(cmd->str,
net_ip2addr(&session->ip->ip));
}
str_append_c(cmd->str, '\t');
str_append_tabescaped(cmd->str, session->service);
} T_END;
str_printfa(cmd->str, "\t%ld", (long)session->pid);
str_printfa(cmd->str, "\t%d", !session->disconnected);
client_export_timeval(cmd->str, &session->last_update);
str_printfa(cmd->str, "\t%u", session->num_cmds);
client_export_mail_stats(cmd->str, &session->stats);
str_append_c(cmd->str, '\n');
o_stream_nsend(client->output, str_data(cmd->str),
str_len(cmd->str));
}
if (session != NULL) {
client->mail_session_iter = session;
mail_session_ref(session);
return 0;
}
return 1;
}
static struct mail_command *
mail_command_add(struct mail_session *session, const char *name,
const char *args)
{
struct mail_command *cmd;
cmd = i_malloc(MALLOC_ADD(sizeof(struct mail_command), stats_alloc_size()));
cmd->stats = (void *)(cmd + 1);
cmd->refcount = 1; /* unrefed at "done" */
cmd->session = session;
cmd->name = i_strdup(name);
cmd->args = i_strdup(args);
cmd->last_update = ioloop_timeval;
DLLIST2_APPEND_FULL(&stable_mail_commands_head,
&stable_mail_commands_tail, cmd,
stable_prev, stable_next);
DLLIST_PREPEND_FULL(&session->commands, cmd,
session_prev, session_next);
mail_session_ref(cmd->session);
global_memory_alloc(mail_command_memsize(cmd));
return cmd;
}
static bool client_export_iter_init(struct client *client)
{
struct client_export_cmd *cmd = client->cmd_export;
if (cmd->filter.user != NULL && strchr(cmd->filter.user, '*') == NULL &&
(cmd->level == MAIL_EXPORT_LEVEL_USER ||
cmd->level == MAIL_EXPORT_LEVEL_SESSION)) {
/* exact user */
struct mail_user *user = mail_user_lookup(cmd->filter.user);
if (user == NULL)
return FALSE;
if (cmd->level == MAIL_EXPORT_LEVEL_SESSION) {
client->mail_session_iter = user->sessions;
if (client->mail_session_iter == NULL)
return FALSE;
mail_session_ref(client->mail_session_iter);
cmd->export_iter = client_export_iter_session;
} else {
client->mail_user_iter = user;
mail_user_ref(user);
cmd->export_iter = client_export_iter_user;
}
return TRUE;
}
if (cmd->filter.ip_bits == IPADDR_BITS(&cmd->filter.ip) &&
(cmd->level == MAIL_EXPORT_LEVEL_IP ||
cmd->level == MAIL_EXPORT_LEVEL_SESSION)) {
/* exact IP address */
struct mail_ip *ip = mail_ip_lookup(&cmd->filter.ip);
if (ip == NULL)
return FALSE;
if (cmd->level == MAIL_EXPORT_LEVEL_SESSION) {
client->mail_session_iter = ip->sessions;
if (client->mail_session_iter == NULL)
return FALSE;
mail_session_ref(client->mail_session_iter);
cmd->export_iter = client_export_iter_session;
} else {
client->mail_ip_iter = ip;
mail_ip_ref(ip);
cmd->export_iter = client_export_iter_ip;
}
return TRUE;
}
if (cmd->filter.domain != NULL &&
strchr(cmd->filter.domain, '*') == NULL &&
(cmd->level == MAIL_EXPORT_LEVEL_DOMAIN ||
cmd->level == MAIL_EXPORT_LEVEL_USER)) {
/* exact domain */
struct mail_domain *domain =
mail_domain_lookup(cmd->filter.domain);
if (domain == NULL)
return FALSE;
if (cmd->level == MAIL_EXPORT_LEVEL_USER) {
client->mail_user_iter = domain->users;
mail_user_ref(client->mail_user_iter);
cmd->export_iter = client_export_iter_user;
} else {
client->mail_domain_iter = domain;
mail_domain_ref(domain);
cmd->export_iter = client_export_iter_domain;
}
return TRUE;
}
switch (cmd->level) {
case MAIL_EXPORT_LEVEL_COMMAND:
client->mail_cmd_iter = stable_mail_commands_head;
if (client->mail_cmd_iter == NULL)
return FALSE;
mail_command_ref(client->mail_cmd_iter);
cmd->export_iter = client_export_iter_command;
break;
case MAIL_EXPORT_LEVEL_SESSION:
client->mail_session_iter = stable_mail_sessions;
if (client->mail_session_iter == NULL)
return FALSE;
mail_session_ref(client->mail_session_iter);
cmd->export_iter = client_export_iter_session;
break;
case MAIL_EXPORT_LEVEL_USER:
client->mail_user_iter = stable_mail_users;
if (client->mail_user_iter == NULL)
return FALSE;
mail_user_ref(client->mail_user_iter);
cmd->export_iter = client_export_iter_user;
break;
case MAIL_EXPORT_LEVEL_DOMAIN:
client->mail_domain_iter = stable_mail_domains;
if (client->mail_domain_iter == NULL)
return FALSE;
mail_domain_ref(client->mail_domain_iter);
cmd->export_iter = client_export_iter_domain;
break;
case MAIL_EXPORT_LEVEL_IP:
client->mail_ip_iter = stable_mail_ips;
if (client->mail_ip_iter == NULL)
return FALSE;
mail_ip_ref(client->mail_ip_iter);
cmd->export_iter = client_export_iter_ip;
break;
}
i_assert(cmd->export_iter != NULL);
return TRUE;
}
