static int client_export_iter_domain(struct client *client)
{
struct client_export_cmd *cmd = client->cmd_export;
struct mail_domain *domain = client->mail_domain_iter;
i_assert(cmd->level == MAIL_EXPORT_LEVEL_DOMAIN);
mail_domain_unref(&client->mail_domain_iter);
if (!cmd->header_sent) {
o_stream_nsend_str(client->output,
"domain\treset_timestamp\tlast_update"
"\tnum_logins\tnum_cmds\tnum_connected_sessions"MAIL_STATS_HEADER);
cmd->header_sent = TRUE;
}
for (; domain != NULL; domain = domain->stable_next) {
if (client_is_busy(client))
break;
if (!mail_export_filter_match_domain(&cmd->filter, domain))
continue;
str_truncate(cmd->str, 0);
str_append_tabescaped(cmd->str, domain->name);
str_printfa(cmd->str, "\t%ld", (long)domain->reset_timestamp);
client_export_timeval(cmd->str, &domain->last_update);
str_printfa(cmd->str, "\t%u\t%u\t%u",
domain->num_logins, domain->num_cmds,
domain->num_connected_sessions);
client_export_mail_stats(cmd->str, &domain->stats);
str_append_c(cmd->str, '\n');
o_stream_nsend(client->output, str_data(cmd->str),
str_len(cmd->str));
}
if (domain != NULL) {
client->mail_domain_iter = domain;
mail_domain_ref(domain);
return 0;
}
return 1;
}
struct mail_user *mail_user_login(const char *username)
{
struct mail_user *user;
const char *domain;
user = hash_table_lookup(mail_users_hash, username);
if (user != NULL) {
user->num_logins++;
mail_user_refresh(user, NULL);
mail_domain_login(user->domain);
return user;
}
domain = strchr(username, '@');
if (domain != NULL)
domain++;
else
domain = "";
user = i_malloc(sizeof(struct mail_user) + stats_alloc_size());
user->stats = (void *)(user + 1);
user->name = i_strdup(username);
user->reset_timestamp = ioloop_time;
user->domain = mail_domain_login_create(domain);
hash_table_insert(mail_users_hash, user->name, user);
DLLIST_PREPEND_FULL(&stable_mail_users, user,
stable_prev, stable_next);
DLLIST2_APPEND_FULL(&mail_users_head, &mail_users_tail, user,
sorted_prev, sorted_next);
DLLIST_PREPEND_FULL(&user->domain->users, user,
domain_prev, domain_next);
mail_domain_ref(user->domain);
user->num_logins++;
user->last_update = ioloop_timeval;
global_memory_alloc(mail_user_memsize(user));
return user;
}
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;
}
