src/auth/userdb-template.c

	userdb-template.c revision 38f53f9d3eda5002b3478f7679d352ee4da60d2f
4c78d9e646c4a1158d7167806937c02d86cdfc25Aki Tuomi/* Copyright (c) 2003-2017 Dovecot authors, see the included COPYING file */
4c78d9e646c4a1158d7167806937c02d86cdfc25Aki Tuomi
4c78d9e646c4a1158d7167806937c02d86cdfc25Aki Tuomi#include "auth-common.h"
4c78d9e646c4a1158d7167806937c02d86cdfc25Aki Tuomi#include "array.h"
4c78d9e646c4a1158d7167806937c02d86cdfc25Aki Tuomi#include "str.h"
4c78d9e646c4a1158d7167806937c02d86cdfc25Aki Tuomi#include "userdb.h"
4c78d9e646c4a1158d7167806937c02d86cdfc25Aki Tuomi#include "userdb-template.h"
4c78d9e646c4a1158d7167806937c02d86cdfc25Aki Tuomi
4c78d9e646c4a1158d7167806937c02d86cdfc25Aki Tuomistruct userdb_template {
4c78d9e646c4a1158d7167806937c02d86cdfc25Aki Tuomi    ARRAY(const char *) args;
4c78d9e646c4a1158d7167806937c02d86cdfc25Aki Tuomi};
4c78d9e646c4a1158d7167806937c02d86cdfc25Aki Tuomi
4c78d9e646c4a1158d7167806937c02d86cdfc25Aki Tuomistruct userdb_template *
4c78d9e646c4a1158d7167806937c02d86cdfc25Aki Tuomiuserdb_template_build(pool_t pool, const char *userdb_name, const char *args)
4c78d9e646c4a1158d7167806937c02d86cdfc25Aki Tuomi{
4c78d9e646c4a1158d7167806937c02d86cdfc25Aki Tuomi    struct userdb_template *tmpl;
4c78d9e646c4a1158d7167806937c02d86cdfc25Aki Tuomi    const char *const *tmp, *key, *value, *nonull_value;
4c78d9e646c4a1158d7167806937c02d86cdfc25Aki Tuomi    uid_t uid;
4c78d9e646c4a1158d7167806937c02d86cdfc25Aki Tuomi    gid_t gid;
4c78d9e646c4a1158d7167806937c02d86cdfc25Aki Tuomi
4c78d9e646c4a1158d7167806937c02d86cdfc25Aki Tuomi    tmpl = p_new(pool, struct userdb_template, 1);
4c78d9e646c4a1158d7167806937c02d86cdfc25Aki Tuomi
4c78d9e646c4a1158d7167806937c02d86cdfc25Aki Tuomi    tmp = t_strsplit_spaces(args, " ");
4c78d9e646c4a1158d7167806937c02d86cdfc25Aki Tuomi    p_array_init(&tmpl->args, pool, str_array_length(tmp));
4c78d9e646c4a1158d7167806937c02d86cdfc25Aki Tuomi
4c78d9e646c4a1158d7167806937c02d86cdfc25Aki Tuomi    for (; *tmp != NULL; tmp++) {
4c78d9e646c4a1158d7167806937c02d86cdfc25Aki Tuomi        value = strchr(*tmp, '=');
4c78d9e646c4a1158d7167806937c02d86cdfc25Aki Tuomi        if (value == NULL)
4c78d9e646c4a1158d7167806937c02d86cdfc25Aki Tuomi            key = *tmp;
4c78d9e646c4a1158d7167806937c02d86cdfc25Aki Tuomi        else
4c78d9e646c4a1158d7167806937c02d86cdfc25Aki Tuomi            key = t_strdup_until(*tmp, value++);
4c78d9e646c4a1158d7167806937c02d86cdfc25Aki Tuomi
4c78d9e646c4a1158d7167806937c02d86cdfc25Aki Tuomi
4c78d9e646c4a1158d7167806937c02d86cdfc25Aki Tuomi        if (*key == '\0')
4c78d9e646c4a1158d7167806937c02d86cdfc25Aki Tuomi            i_fatal("Invalid userdb template %s - key must not be empty",
4c78d9e646c4a1158d7167806937c02d86cdfc25Aki Tuomi                args);
4c78d9e646c4a1158d7167806937c02d86cdfc25Aki Tuomi
4c78d9e646c4a1158d7167806937c02d86cdfc25Aki Tuomi        nonull_value = value == NULL ? "" : value;
4c78d9e646c4a1158d7167806937c02d86cdfc25Aki Tuomi        if (strcasecmp(key, "uid") == 0) {
4c78d9e646c4a1158d7167806937c02d86cdfc25Aki Tuomi            uid = userdb_parse_uid(NULL, nonull_value);
4c78d9e646c4a1158d7167806937c02d86cdfc25Aki Tuomi            if (uid == (uid_t)-1) {
4c78d9e646c4a1158d7167806937c02d86cdfc25Aki Tuomi                i_fatal("%s userdb: Invalid uid: %s",
4c78d9e646c4a1158d7167806937c02d86cdfc25Aki Tuomi                    userdb_name, nonull_value);
4c78d9e646c4a1158d7167806937c02d86cdfc25Aki Tuomi            }
4c78d9e646c4a1158d7167806937c02d86cdfc25Aki Tuomi            value = dec2str(uid);
4c78d9e646c4a1158d7167806937c02d86cdfc25Aki Tuomi        } else if (strcasecmp(key, "gid") == 0) {
4c78d9e646c4a1158d7167806937c02d86cdfc25Aki Tuomi            gid = userdb_parse_gid(NULL, nonull_value);
4c78d9e646c4a1158d7167806937c02d86cdfc25Aki Tuomi            if (gid == (gid_t)-1) {
4c78d9e646c4a1158d7167806937c02d86cdfc25Aki Tuomi                i_fatal("%s userdb: Invalid gid: %s",
4c78d9e646c4a1158d7167806937c02d86cdfc25Aki Tuomi                    userdb_name, nonull_value);
4c78d9e646c4a1158d7167806937c02d86cdfc25Aki Tuomi            }
4c78d9e646c4a1158d7167806937c02d86cdfc25Aki Tuomi            value = dec2str(gid);
4c78d9e646c4a1158d7167806937c02d86cdfc25Aki Tuomi        } else if (*key == '\0') {
4c78d9e646c4a1158d7167806937c02d86cdfc25Aki Tuomi            i_fatal("%s userdb: Empty key (=%s)",
4c78d9e646c4a1158d7167806937c02d86cdfc25Aki Tuomi                userdb_name, nonull_value);
4c78d9e646c4a1158d7167806937c02d86cdfc25Aki Tuomi        }
4c78d9e646c4a1158d7167806937c02d86cdfc25Aki Tuomi        key = p_strdup(pool, key);
4c78d9e646c4a1158d7167806937c02d86cdfc25Aki Tuomi        value = p_strdup(pool, value);
4c78d9e646c4a1158d7167806937c02d86cdfc25Aki Tuomi
4c78d9e646c4a1158d7167806937c02d86cdfc25Aki Tuomi        array_append(&tmpl->args, &key, 1);
4c78d9e646c4a1158d7167806937c02d86cdfc25Aki Tuomi        array_append(&tmpl->args, &value, 1);
4c78d9e646c4a1158d7167806937c02d86cdfc25Aki Tuomi    }
4c78d9e646c4a1158d7167806937c02d86cdfc25Aki Tuomi    return tmpl;
4c78d9e646c4a1158d7167806937c02d86cdfc25Aki Tuomi}
4c78d9e646c4a1158d7167806937c02d86cdfc25Aki Tuomi
4c78d9e646c4a1158d7167806937c02d86cdfc25Aki Tuomiint userdb_template_export(struct userdb_template *tmpl,
4c78d9e646c4a1158d7167806937c02d86cdfc25Aki Tuomi               struct auth_request *auth_request,
               const char **error_r)
{
        const struct var_expand_table *table;
    string_t *str;
    const char *const *args, *value;
    unsigned int i, count;

    if (userdb_template_is_empty(tmpl))
        return 0;

    str = t_str_new(256);
    table = auth_request_get_var_expand_table(auth_request, NULL);

    args = array_get(&tmpl->args, &count);
    i_assert((count % 2) == 0);
    for (i = 0; i < count; i += 2) {
        if (args[i+1] == NULL)
            value = "";
        else {
            str_truncate(str, 0);
            if (auth_request_var_expand_with_table(str, args[i+1],
                    auth_request, table, NULL, error_r) <= 0)
                return -1;
            value = str_c(str);
        }
        auth_request_set_userdb_field(auth_request, args[i], value);
    }
    return 0;
}

bool userdb_template_remove(struct userdb_template *tmpl,
                const char *key, const char **value_r)
{
    const char *const *args;
    unsigned int i, count;

    args = array_get(&tmpl->args, &count);
    i_assert((count % 2) == 0);
    for (i = 0; i < count; i += 2) {
        if (strcmp(args[i], key) == 0) {
            *value_r = args[i+1];
            array_delete(&tmpl->args, i, 2);
            return TRUE;
        }
    }
    return FALSE;
}

bool userdb_template_is_empty(struct userdb_template *tmpl)
{
    return array_count(&tmpl->args) == 0;
}

const char *const *userdb_template_get_args(struct userdb_template *tmpl, unsigned int *count_r)
{
    return array_get(&tmpl->args, count_r);
}