userdb-template.c revision 65a1c01472ebf88b9f5bbd6636827902936d5497
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen/* Copyright (c) 2003-2011 Dovecot authors, see the included COPYING file */
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen#include "auth-common.h"
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen#include "array.h"
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen#include "str.h"
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen#include "var-expand.h"
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen#include "userdb.h"
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen#include "userdb-template.h"
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainenstruct userdb_template {
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen ARRAY_DEFINE(args, const char *);
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen};
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainenstruct userdb_template *
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainenuserdb_template_build(pool_t pool, const char *userdb_name, const char *args)
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen{
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen struct userdb_template *tmpl;
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen const char *const *tmp, *key, *value, *nonull_value;
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen uid_t uid;
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen gid_t gid;
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen tmpl = p_new(pool, struct userdb_template, 1);
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen tmp = t_strsplit_spaces(args, " ");
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen p_array_init(&tmpl->args, pool, str_array_length(tmp));
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen for (; *tmp != NULL; tmp++) {
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen value = strchr(*tmp, '=');
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen if (value == NULL)
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen key = *tmp;
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen else
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen key = t_strdup_until(*tmp, value++);
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen nonull_value = value == NULL ? "" : value;
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen if (strcasecmp(key, "uid") == 0) {
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen uid = userdb_parse_uid(NULL, nonull_value);
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen if (uid == (uid_t)-1) {
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen i_fatal("%s userdb: Invalid uid: %s",
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen userdb_name, nonull_value);
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen }
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen value = dec2str(uid);
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen } else if (strcasecmp(key, "gid") == 0) {
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen gid = userdb_parse_gid(NULL, nonull_value);
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen if (gid == (gid_t)-1) {
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen i_fatal("%s userdb: Invalid gid: %s",
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen userdb_name, nonull_value);
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen }
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen value = dec2str(gid);
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen } else if (*key == '\0') {
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen i_fatal("%s userdb: Empty key (=%s)",
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen userdb_name, nonull_value);
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen }
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen key = p_strdup(pool, key);
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen value = p_strdup(pool, value);
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen array_append(&tmpl->args, &key, 1);
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen array_append(&tmpl->args, &value, 1);
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen }
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen return tmpl;
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen}
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainenvoid userdb_template_export(struct userdb_template *tmpl,
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen struct auth_request *auth_request)
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen{
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen const struct var_expand_table *table;
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen string_t *str;
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen const char *const *args, *value;
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen unsigned int i, count;
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen str = t_str_new(256);
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen table = auth_request_get_var_expand_table(auth_request, NULL);
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen args = array_get(&tmpl->args, &count);
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen i_assert((count % 2) == 0);
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen for (i = 0; i < count; i += 2) {
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen if (args[i+1] == NULL)
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen value = NULL;
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen else {
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen str_truncate(str, 0);
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen var_expand(str, args[i+1], table);
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen value = str_c(str);
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen }
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen auth_request_set_userdb_field(auth_request, args[i], value);
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen }
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen}
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainenbool userdb_template_remove(struct userdb_template *tmpl,
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen const char *key, const char **value_r)
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen{
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen const char *const *args;
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen unsigned int i, count;
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen args = array_get(&tmpl->args, &count);
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen i_assert((count % 2) == 0);
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen for (i = 0; i < count; i += 2) {
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen if (strcmp(args[i], key) == 0) {
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen *value_r = args[i+1];
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen array_delete(&tmpl->args, i, 2);
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen return TRUE;
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen }
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen }
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen return FALSE;
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen}
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainenbool userdb_template_is_empty(struct userdb_template *tmpl)
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen{
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen return array_count(&tmpl->args) == 0;
65a1c01472ebf88b9f5bbd6636827902936d5497Timo Sirainen}