src/auth/passdb-template.c

	passdb-template.c revision d9fe7936864be4fa9e1784dc7dbf451a81f4ab9c
/* Copyright (c) 2003-2011 Dovecot authors, see the included COPYING file */

#include "auth-common.h"
#include "array.h"
#include "str.h"
#include "var-expand.h"
#include "passdb.h"
#include "passdb-template.h"

struct passdb_template {
    ARRAY_DEFINE(args, const char *);
};

struct passdb_template *passdb_template_build(pool_t pool, const char *args)
{
    struct passdb_template *tmpl;
    const char *const *tmp, *key, *value;

    tmpl = p_new(pool, struct passdb_template, 1);

    tmp = t_strsplit_spaces(args, " ");
    p_array_init(&tmpl->args, pool, str_array_length(tmp));

    for (; *tmp != NULL; tmp++) {
        value = strchr(*tmp, '=');
        if (value == NULL)
            key = *tmp;
        else
            key = t_strdup_until(*tmp, value++);

        key = p_strdup(pool, key);
        value = p_strdup(pool, value);
        array_append(&tmpl->args, &key, 1);
        array_append(&tmpl->args, &value, 1);
    }
    return tmpl;
}

void passdb_template_export(struct passdb_template *tmpl,
                struct auth_request *auth_request)
{
        const struct var_expand_table *table;
    string_t *str;
    const char *const *args, *value;
    unsigned int i, count;

    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);
            var_expand(str, args[i+1], table);
            value = str_c(str);
        }
        auth_request_set_field(auth_request, args[i], value,
                       STATIC_PASS_SCHEME);
    }
}

bool passdb_template_remove(struct passdb_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;
}