src/lib-storage/mail-user.c

	mail-user.c revision a64adf62fa33f2463a86f990217b0c9078531a40
/* Copyright (c) 2008-2009 Dovecot authors, see the included COPYING file */

#include "lib.h"
#include "array.h"
#include "hostpid.h"
#include "network.h"
#include "var-expand.h"
#include "settings-parser.h"
#include "auth-master.h"
#include "mail-storage-settings.h"
#include "mail-namespace.h"
#include "mail-user.h"

#include <stdlib.h>

struct mail_user_module_register mail_user_module_register = { 0 };
void (*hook_mail_user_created)(struct mail_user *user) = NULL;

static struct auth_master_connection *auth_master_conn;

static void mail_user_deinit_base(struct mail_user *user)
{
    mail_namespaces_deinit(&user->namespaces);
    pool_unref(&user->pool);
}

struct mail_user *mail_user_alloc(const char *username,
                  const struct mail_user_settings *set)
{
    struct mail_user *user;
    pool_t pool;

    i_assert(username != NULL);
    i_assert(*username != '\0');

    pool = pool_alloconly_create("mail user", 2048);
    user = p_new(pool, struct mail_user, 1);
    user->pool = pool;
    user->refcount = 1;
    user->username = p_strdup(pool, username);
    user->unexpanded_set = set;
    user->set = settings_dup(&mail_user_setting_parser_info, set, pool);
    user->v.deinit = mail_user_deinit_base;
    p_array_init(&user->module_contexts, user->pool, 5);
    return user;
}

int mail_user_init(struct mail_user *user, const char **error_r)
{
    const char *home, *key, *value;

    if (user->_home == NULL &&
        settings_vars_have_key(&mail_user_setting_parser_info, user->set,
                   'h', "home", &key, &value) &&
        mail_user_get_home(user, &home) <= 0) {
        *error_r = t_strdup_printf(
            "userdb didn't return a home directory, "
            "but %s used it (%%h): %s", key, value);
        return -1;
    }

    settings_var_expand(&mail_user_setting_parser_info, user->set,
                user->pool, mail_user_var_expand_table(user));

    user->initialized = TRUE;
    if (hook_mail_user_created != NULL)
        hook_mail_user_created(user);
    return 0;
}

void mail_user_ref(struct mail_user *user)
{
    i_assert(user->refcount > 0);

    user->refcount++;
}

void mail_user_unref(struct mail_user **_user)
{
    struct mail_user *user = *_user;

    i_assert(user->refcount > 0);

    *_user = NULL;
    if (--user->refcount == 0)
        user->v.deinit(user);
}

struct mail_user *mail_user_find(struct mail_user *user, const char *name)
{
    struct mail_namespace *ns;

    for (ns = user->namespaces; ns != NULL; ns = ns->next) {
        if (ns->owner != NULL && strcmp(ns->owner->username, name) == 0)
            return ns->owner;
    }
    return NULL;
}

void mail_user_set_vars(struct mail_user *user, uid_t uid, const char *service,
            const struct ip_addr *local_ip,
            const struct ip_addr *remote_ip)
{
    user->uid = uid;
    user->service = p_strdup(user->pool, service);
    if (local_ip != NULL) {
        user->local_ip = p_new(user->pool, struct ip_addr, 1);
        *user->local_ip = *local_ip;
    }
    if (remote_ip != NULL) {
        user->remote_ip = p_new(user->pool, struct ip_addr, 1);
        *user->remote_ip = *remote_ip;
    }
}

const struct var_expand_table *
mail_user_var_expand_table(struct mail_user *user)
{
    static struct var_expand_table static_tab[] = {
        { 'u', NULL, "user" },
        { 'n', NULL, "username" },
        { 'd', NULL, "domain" },
        { 's', NULL, "service" },
        { 'h', NULL, "home" },
        { 'l', NULL, "lip" },
        { 'r', NULL, "rip" },
        { 'p', NULL, "pid" },
        { 'i', NULL, "uid" },
        { '\0', NULL, NULL }
    };
    struct var_expand_table *tab;

    if (user->var_expand_table != NULL)
        return user->var_expand_table;

    tab = p_malloc(user->pool, sizeof(static_tab));
    memcpy(tab, static_tab, sizeof(static_tab));

    tab[0].value = user->username;
    tab[1].value = p_strdup(user->pool, t_strcut(user->username, '@'));
    tab[2].value = strchr(user->username, '@');
    if (tab[2].value != NULL) tab[2].value++;
    tab[3].value = user->service;
    tab[4].value = user->_home; /* don't look it up unless we need it */
    tab[5].value = user->local_ip == NULL ? NULL :
        p_strdup(user->pool, net_ip2addr(user->local_ip));
    tab[6].value = user->remote_ip == NULL ? NULL :
        p_strdup(user->pool, net_ip2addr(user->remote_ip));
    tab[7].value = my_pid;
    tab[8].value = p_strdup(user->pool, dec2str(user->uid));

    user->var_expand_table = tab;
    return user->var_expand_table;
}

void mail_user_set_home(struct mail_user *user, const char *home)
{
    user->_home = p_strdup(user->pool, home);
    user->home_looked_up = TRUE;
}

void mail_user_add_namespace(struct mail_user *user,
                 struct mail_namespace **namespaces)
{
    struct mail_namespace **tmp, *next, *ns = *namespaces;

    for (; ns != NULL; ns = next) {
        next = ns->next;

        tmp = &user->namespaces;
        for (; *tmp != NULL; tmp = &(*tmp)->next) {
            if (strlen(ns->prefix) < strlen((*tmp)->prefix))
                break;
        }
        ns->next = *tmp;
        *tmp = ns;
    }
    *namespaces = user->namespaces;
}

void mail_user_drop_useless_namespaces(struct mail_user *user)
{
    struct mail_namespace *ns, *next;

    for (ns = user->namespaces; ns != NULL; ns = next) {
        next = ns->next;

        if ((ns->flags & NAMESPACE_FLAG_USABLE) == 0 &&
            (ns->flags & NAMESPACE_FLAG_AUTOCREATED) != 0)
            mail_namespace_destroy(ns);
    }
}

const char *mail_user_home_expand(struct mail_user *user, const char *path)
{
    (void)mail_user_try_home_expand(user, &path);
    return path;
}

int mail_user_get_home(struct mail_user *user, const char **home_r)
{
    struct auth_user_reply reply;
    pool_t userdb_pool;
    int ret;

    if (user->home_looked_up) {
        *home_r = user->_home;
        return user->_home != NULL ? 1 : 0;
    }

    userdb_pool = pool_alloconly_create("userdb lookup", 512);
    ret = auth_master_user_lookup(auth_master_conn, user->username,
                      AUTH_SERVICE_INTERNAL,
                      userdb_pool, &reply);
    if (ret < 0)
        *home_r = NULL;
    else {
        user->_home = ret == 0 ? NULL :
            p_strdup(user->pool, reply.home);
        user->home_looked_up = TRUE;
        ret = user->_home != NULL ? 1 : 0;
        *home_r = user->_home;
    }
    pool_unref(&userdb_pool);
    return ret;
}

int mail_user_try_home_expand(struct mail_user *user, const char **pathp)
{
    const char *home, *path = *pathp;

    if (mail_user_get_home(user, &home) < 0)
        return -1;

    if (path[0] == '~' && (path[1] == '/' || path[1] == '\0')) {
        if (home == NULL)
            return -1;

        *pathp = t_strconcat(home, path + 1, NULL);
    }
    return 0;
}

void mail_users_init(const char *auth_socket_path, bool debug)
{
    const char *base_dir;

    if (auth_socket_path == NULL) {
        base_dir = getenv("BASE_DIR");
        if (base_dir == NULL)
            base_dir = PKG_RUNDIR;
        auth_socket_path = t_strconcat(base_dir, "/auth-master", NULL);
    }
    auth_master_conn = auth_master_init(auth_socket_path, debug);
}

void mail_users_deinit(void)
{
    auth_master_deinit(&auth_master_conn);
}