src/lib/var-expand.c

	var-expand.c revision 4ee00532a265bdfb38539d811fcd12d51210ac35
5f5870385cff47efd2f58e7892f251cf13761528Timo Sirainen/* Copyright (c) 2003-2012 Dovecot authors, see the included COPYING file */
4cb2599c5cdf27362a66ac475ce295409c093c92Timo Sirainen
4cb2599c5cdf27362a66ac475ce295409c093c92Timo Sirainen#include "lib.h"
3e859421cc59d28d4ba99f32830e3d0531334813Timo Sirainen#include "array.h"
1036d2db2b718bdc5b10f0773dd01d62638e9ba9Timo Sirainen#include "md5.h"
bb979386852c7689dc66c0fce03319382f66d501Timo Sirainen#include "hash.h"
1036d2db2b718bdc5b10f0773dd01d62638e9ba9Timo Sirainen#include "hex-binary.h"
927d3977d5598f12ae18d4fa3f22b9e913f7dd46Timo Sirainen#include "hostpid.h"
4cb2599c5cdf27362a66ac475ce295409c093c92Timo Sirainen#include "str.h"
3b94ff5951db4d4eddb7a80ed4e3f61207202635Timo Sirainen#include "strescape.h"
4cb2599c5cdf27362a66ac475ce295409c093c92Timo Sirainen#include "var-expand.h"
4cb2599c5cdf27362a66ac475ce295409c093c92Timo Sirainen
7cd08c67fde5371f662d8c95b30c341741950420Timo Sirainen#include <stdlib.h>
d758298b3b6f1ebcd494392c0f20b0e119a9e85eTimo Sirainen#include <unistd.h>
546d3609e0811a147269ee9979eb90649445f5acTimo Sirainen#include <ctype.h>
7cd08c67fde5371f662d8c95b30c341741950420Timo Sirainen
f77ffa31038d46ca9c6d24d93e3d76c9aa8d4d0cTimo Sirainen#define TABLE_LAST(t) \
f77ffa31038d46ca9c6d24d93e3d76c9aa8d4d0cTimo Sirainen    ((t)->key == '\0' && (t)->long_key == NULL)
f77ffa31038d46ca9c6d24d93e3d76c9aa8d4d0cTimo Sirainen
bb979386852c7689dc66c0fce03319382f66d501Timo Sirainenstruct var_expand_context {
a40649829bce4c8de6210a2cb4a4b4cf5bb40da8Timo Sirainen    int offset;
a40649829bce4c8de6210a2cb4a4b4cf5bb40da8Timo Sirainen    unsigned int width;
d35364f4d7d139b4150d290e14717e10f1ede4cdTimo Sirainen    bool zero_padding;
bb979386852c7689dc66c0fce03319382f66d501Timo Sirainen};
bb979386852c7689dc66c0fce03319382f66d501Timo Sirainen
3b94ff5951db4d4eddb7a80ed4e3f61207202635Timo Sirainenstruct var_expand_modifier {
3b94ff5951db4d4eddb7a80ed4e3f61207202635Timo Sirainen    char key;
bb979386852c7689dc66c0fce03319382f66d501Timo Sirainen    const char *(*func)(const char *, struct var_expand_context *);
3b94ff5951db4d4eddb7a80ed4e3f61207202635Timo Sirainen};
3b94ff5951db4d4eddb7a80ed4e3f61207202635Timo Sirainen
bb979386852c7689dc66c0fce03319382f66d501Timo Sirainenstatic const char *
43d32cbe60fdaef2699d99f1ca259053e9350411Timo Sirainenm_str_lcase(const char *str, struct var_expand_context *ctx ATTR_UNUSED)
bb979386852c7689dc66c0fce03319382f66d501Timo Sirainen{
bb979386852c7689dc66c0fce03319382f66d501Timo Sirainen    return t_str_lcase(str);
bb979386852c7689dc66c0fce03319382f66d501Timo Sirainen}
bb979386852c7689dc66c0fce03319382f66d501Timo Sirainen
bb979386852c7689dc66c0fce03319382f66d501Timo Sirainenstatic const char *
43d32cbe60fdaef2699d99f1ca259053e9350411Timo Sirainenm_str_ucase(const char *str, struct var_expand_context *ctx ATTR_UNUSED)
bb979386852c7689dc66c0fce03319382f66d501Timo Sirainen{
cd700cc7be045389dd1c948b2372b30ee99c5795Timo Sirainen    return t_str_ucase(str);
bb979386852c7689dc66c0fce03319382f66d501Timo Sirainen}
bb979386852c7689dc66c0fce03319382f66d501Timo Sirainen
bb979386852c7689dc66c0fce03319382f66d501Timo Sirainenstatic const char *
43d32cbe60fdaef2699d99f1ca259053e9350411Timo Sirainenm_str_escape(const char *str, struct var_expand_context *ctx ATTR_UNUSED)
bb979386852c7689dc66c0fce03319382f66d501Timo Sirainen{
bb979386852c7689dc66c0fce03319382f66d501Timo Sirainen    return str_escape(str);
bb979386852c7689dc66c0fce03319382f66d501Timo Sirainen}
bb979386852c7689dc66c0fce03319382f66d501Timo Sirainen
bb979386852c7689dc66c0fce03319382f66d501Timo Sirainenstatic const char *
43d32cbe60fdaef2699d99f1ca259053e9350411Timo Sirainenm_str_hex(const char *str, struct var_expand_context *ctx ATTR_UNUSED)
7cd08c67fde5371f662d8c95b30c341741950420Timo Sirainen{
7cd08c67fde5371f662d8c95b30c341741950420Timo Sirainen    unsigned long long l;
7cd08c67fde5371f662d8c95b30c341741950420Timo Sirainen
7cd08c67fde5371f662d8c95b30c341741950420Timo Sirainen    l = strtoull(str, NULL, 10);
7cd08c67fde5371f662d8c95b30c341741950420Timo Sirainen    return t_strdup_printf("%llx", l);
7cd08c67fde5371f662d8c95b30c341741950420Timo Sirainen}
7cd08c67fde5371f662d8c95b30c341741950420Timo Sirainen
8e0f89885f2ce60961da9ae2d96d71c7109e6032Timo Sirainenstatic const char *
43d32cbe60fdaef2699d99f1ca259053e9350411Timo Sirainenm_str_reverse(const char *str, struct var_expand_context *ctx ATTR_UNUSED)
8e0f89885f2ce60961da9ae2d96d71c7109e6032Timo Sirainen{
8e0f89885f2ce60961da9ae2d96d71c7109e6032Timo Sirainen    size_t len = strlen(str);
8e0f89885f2ce60961da9ae2d96d71c7109e6032Timo Sirainen    char *p, *rev;
8e0f89885f2ce60961da9ae2d96d71c7109e6032Timo Sirainen
8e0f89885f2ce60961da9ae2d96d71c7109e6032Timo Sirainen    rev = t_malloc(len + 1);
8e0f89885f2ce60961da9ae2d96d71c7109e6032Timo Sirainen    rev[len] = '\0';
8e0f89885f2ce60961da9ae2d96d71c7109e6032Timo Sirainen
8e0f89885f2ce60961da9ae2d96d71c7109e6032Timo Sirainen    for (p = rev + len - 1; *str != '\0'; str++)
8e0f89885f2ce60961da9ae2d96d71c7109e6032Timo Sirainen        *p-- = *str;
8e0f89885f2ce60961da9ae2d96d71c7109e6032Timo Sirainen    return rev;
8e0f89885f2ce60961da9ae2d96d71c7109e6032Timo Sirainen}
8e0f89885f2ce60961da9ae2d96d71c7109e6032Timo Sirainen
bb979386852c7689dc66c0fce03319382f66d501Timo Sirainenstatic const char *m_str_hash(const char *str, struct var_expand_context *ctx)
bb979386852c7689dc66c0fce03319382f66d501Timo Sirainen{
bb979386852c7689dc66c0fce03319382f66d501Timo Sirainen    unsigned int value = str_hash(str);
bb979386852c7689dc66c0fce03319382f66d501Timo Sirainen    string_t *hash = t_str_new(20);
bb979386852c7689dc66c0fce03319382f66d501Timo Sirainen
bb979386852c7689dc66c0fce03319382f66d501Timo Sirainen    if (ctx->width != 0) {
bb979386852c7689dc66c0fce03319382f66d501Timo Sirainen        value %= ctx->width;
bb979386852c7689dc66c0fce03319382f66d501Timo Sirainen        ctx->width = 0;
bb979386852c7689dc66c0fce03319382f66d501Timo Sirainen    }
869b0fa6ef7fc0326c9f75ae449e5a9f97a796e2Timo Sirainen
bb979386852c7689dc66c0fce03319382f66d501Timo Sirainen    str_printfa(hash, "%x", value);
a40649829bce4c8de6210a2cb4a4b4cf5bb40da8Timo Sirainen    while ((int)str_len(hash) < ctx->offset)
bb979386852c7689dc66c0fce03319382f66d501Timo Sirainen        str_insert(hash, 0, "0");
bb979386852c7689dc66c0fce03319382f66d501Timo Sirainen        ctx->offset = 0;
bb979386852c7689dc66c0fce03319382f66d501Timo Sirainen
bb979386852c7689dc66c0fce03319382f66d501Timo Sirainen    return str_c(hash);
bb979386852c7689dc66c0fce03319382f66d501Timo Sirainen}
bb979386852c7689dc66c0fce03319382f66d501Timo Sirainen
546d3609e0811a147269ee9979eb90649445f5acTimo Sirainenstatic const char *
546d3609e0811a147269ee9979eb90649445f5acTimo Sirainenm_str_md5(const char *str, struct var_expand_context *ctx ATTR_UNUSED)
1036d2db2b718bdc5b10f0773dd01d62638e9ba9Timo Sirainen{
1036d2db2b718bdc5b10f0773dd01d62638e9ba9Timo Sirainen    unsigned char digest[16];
1036d2db2b718bdc5b10f0773dd01d62638e9ba9Timo Sirainen
1036d2db2b718bdc5b10f0773dd01d62638e9ba9Timo Sirainen    md5_get_digest(str, strlen(str), digest);
1036d2db2b718bdc5b10f0773dd01d62638e9ba9Timo Sirainen
1036d2db2b718bdc5b10f0773dd01d62638e9ba9Timo Sirainen    return binary_to_hex(digest, sizeof(digest));
1036d2db2b718bdc5b10f0773dd01d62638e9ba9Timo Sirainen}
1036d2db2b718bdc5b10f0773dd01d62638e9ba9Timo Sirainen
546d3609e0811a147269ee9979eb90649445f5acTimo Sirainenstatic const char *
546d3609e0811a147269ee9979eb90649445f5acTimo Sirainenm_str_ldap_dn(const char *str, struct var_expand_context *ctx ATTR_UNUSED)
28a57cc7b6f29cc55a4a586c80902b21daf5d55bTimo Sirainen{
28a57cc7b6f29cc55a4a586c80902b21daf5d55bTimo Sirainen    string_t *ret = t_str_new(256);
28a57cc7b6f29cc55a4a586c80902b21daf5d55bTimo Sirainen
28a57cc7b6f29cc55a4a586c80902b21daf5d55bTimo Sirainen    while (*str) {
28a57cc7b6f29cc55a4a586c80902b21daf5d55bTimo Sirainen        if (*str == '.')
28a57cc7b6f29cc55a4a586c80902b21daf5d55bTimo Sirainen            str_append(ret, ",dc=");
28a57cc7b6f29cc55a4a586c80902b21daf5d55bTimo Sirainen        else
28a57cc7b6f29cc55a4a586c80902b21daf5d55bTimo Sirainen            str_append_c(ret, *str);
28a57cc7b6f29cc55a4a586c80902b21daf5d55bTimo Sirainen        str++;
28a57cc7b6f29cc55a4a586c80902b21daf5d55bTimo Sirainen    }
28a57cc7b6f29cc55a4a586c80902b21daf5d55bTimo Sirainen
28a57cc7b6f29cc55a4a586c80902b21daf5d55bTimo Sirainen    return str_free_without_data(&ret);
28a57cc7b6f29cc55a4a586c80902b21daf5d55bTimo Sirainen}
28a57cc7b6f29cc55a4a586c80902b21daf5d55bTimo Sirainen
546d3609e0811a147269ee9979eb90649445f5acTimo Sirainenstatic const char *
546d3609e0811a147269ee9979eb90649445f5acTimo Sirainenm_str_trim(const char *str, struct var_expand_context *ctx ATTR_UNUSED)
546d3609e0811a147269ee9979eb90649445f5acTimo Sirainen{
546d3609e0811a147269ee9979eb90649445f5acTimo Sirainen    unsigned int len;
546d3609e0811a147269ee9979eb90649445f5acTimo Sirainen
546d3609e0811a147269ee9979eb90649445f5acTimo Sirainen    len = strlen(str);
546d3609e0811a147269ee9979eb90649445f5acTimo Sirainen    while (len > 0 && i_isspace(str[len-1]))
546d3609e0811a147269ee9979eb90649445f5acTimo Sirainen        len--;
546d3609e0811a147269ee9979eb90649445f5acTimo Sirainen    return t_strndup(str, len);
546d3609e0811a147269ee9979eb90649445f5acTimo Sirainen}
546d3609e0811a147269ee9979eb90649445f5acTimo Sirainen
7cd08c67fde5371f662d8c95b30c341741950420Timo Sirainen#define MAX_MODIFIER_COUNT 10
3b94ff5951db4d4eddb7a80ed4e3f61207202635Timo Sirainenstatic const struct var_expand_modifier modifiers[] = {
bb979386852c7689dc66c0fce03319382f66d501Timo Sirainen    { 'L', m_str_lcase },
bb979386852c7689dc66c0fce03319382f66d501Timo Sirainen    { 'U', m_str_ucase },
bb979386852c7689dc66c0fce03319382f66d501Timo Sirainen    { 'E', m_str_escape },
bb979386852c7689dc66c0fce03319382f66d501Timo Sirainen    { 'X', m_str_hex },
8e0f89885f2ce60961da9ae2d96d71c7109e6032Timo Sirainen    { 'R', m_str_reverse },
bb979386852c7689dc66c0fce03319382f66d501Timo Sirainen    { 'H', m_str_hash },
1036d2db2b718bdc5b10f0773dd01d62638e9ba9Timo Sirainen    { 'M', m_str_md5 },
28a57cc7b6f29cc55a4a586c80902b21daf5d55bTimo Sirainen    { 'D', m_str_ldap_dn },
546d3609e0811a147269ee9979eb90649445f5acTimo Sirainen    { 'T', m_str_trim },
3b94ff5951db4d4eddb7a80ed4e3f61207202635Timo Sirainen    { '\0', NULL }
3b94ff5951db4d4eddb7a80ed4e3f61207202635Timo Sirainen};
3b94ff5951db4d4eddb7a80ed4e3f61207202635Timo Sirainen
e130dc7c712a10215b1e6be56403bbb934826251Timo Sirainenstatic const char *
e130dc7c712a10215b1e6be56403bbb934826251Timo Sirainenvar_expand_func(const struct var_expand_func_table *func_table,
e130dc7c712a10215b1e6be56403bbb934826251Timo Sirainen        const char *key, const char *data, void *context)
e130dc7c712a10215b1e6be56403bbb934826251Timo Sirainen{
e130dc7c712a10215b1e6be56403bbb934826251Timo Sirainen    if (strcmp(key, "env") == 0)
e130dc7c712a10215b1e6be56403bbb934826251Timo Sirainen        return getenv(data);
e130dc7c712a10215b1e6be56403bbb934826251Timo Sirainen    if (func_table == NULL)
e130dc7c712a10215b1e6be56403bbb934826251Timo Sirainen        return NULL;
e130dc7c712a10215b1e6be56403bbb934826251Timo Sirainen
e130dc7c712a10215b1e6be56403bbb934826251Timo Sirainen    for (; func_table->key != NULL; func_table++) {
e130dc7c712a10215b1e6be56403bbb934826251Timo Sirainen        if (strcmp(func_table->key, key) == 0)
e130dc7c712a10215b1e6be56403bbb934826251Timo Sirainen            return func_table->func(data, context);
e130dc7c712a10215b1e6be56403bbb934826251Timo Sirainen    }
e130dc7c712a10215b1e6be56403bbb934826251Timo Sirainen    return NULL;
e130dc7c712a10215b1e6be56403bbb934826251Timo Sirainen}
e130dc7c712a10215b1e6be56403bbb934826251Timo Sirainen
927d3977d5598f12ae18d4fa3f22b9e913f7dd46Timo Sirainenstatic const char *
927d3977d5598f12ae18d4fa3f22b9e913f7dd46Timo Sirainenvar_expand_long(const struct var_expand_table *table,
e130dc7c712a10215b1e6be56403bbb934826251Timo Sirainen        const struct var_expand_func_table *func_table,
e130dc7c712a10215b1e6be56403bbb934826251Timo Sirainen        const void *key_start, unsigned int key_len, void *context)
927d3977d5598f12ae18d4fa3f22b9e913f7dd46Timo Sirainen{
927d3977d5598f12ae18d4fa3f22b9e913f7dd46Timo Sirainen        const struct var_expand_table *t;
927d3977d5598f12ae18d4fa3f22b9e913f7dd46Timo Sirainen    const char *value = NULL;
927d3977d5598f12ae18d4fa3f22b9e913f7dd46Timo Sirainen
93794594bc682b12353d2d0db08d91ae3e7c56c6Timo Sirainen    if (table != NULL) {
93794594bc682b12353d2d0db08d91ae3e7c56c6Timo Sirainen        for (t = table; !TABLE_LAST(t); t++) {
93794594bc682b12353d2d0db08d91ae3e7c56c6Timo Sirainen            if (t->long_key != NULL &&
93794594bc682b12353d2d0db08d91ae3e7c56c6Timo Sirainen                strncmp(t->long_key, key_start, key_len) == 0 &&
93794594bc682b12353d2d0db08d91ae3e7c56c6Timo Sirainen                t->long_key[key_len] == '\0') {
93794594bc682b12353d2d0db08d91ae3e7c56c6Timo Sirainen                return t->value != NULL ? t->value : "";
93794594bc682b12353d2d0db08d91ae3e7c56c6Timo Sirainen            }
927d3977d5598f12ae18d4fa3f22b9e913f7dd46Timo Sirainen        }
927d3977d5598f12ae18d4fa3f22b9e913f7dd46Timo Sirainen    }
927d3977d5598f12ae18d4fa3f22b9e913f7dd46Timo Sirainen
927d3977d5598f12ae18d4fa3f22b9e913f7dd46Timo Sirainen    /* built-in variables: */
927d3977d5598f12ae18d4fa3f22b9e913f7dd46Timo Sirainen    T_BEGIN {
927d3977d5598f12ae18d4fa3f22b9e913f7dd46Timo Sirainen        const char *key = t_strndup(key_start, key_len);
927d3977d5598f12ae18d4fa3f22b9e913f7dd46Timo Sirainen
927d3977d5598f12ae18d4fa3f22b9e913f7dd46Timo Sirainen        switch (key_len) {
927d3977d5598f12ae18d4fa3f22b9e913f7dd46Timo Sirainen        case 3:
927d3977d5598f12ae18d4fa3f22b9e913f7dd46Timo Sirainen            if (strcmp(key, "pid") == 0)
927d3977d5598f12ae18d4fa3f22b9e913f7dd46Timo Sirainen                value = my_pid;
d758298b3b6f1ebcd494392c0f20b0e119a9e85eTimo Sirainen            else if (strcmp(key, "uid") == 0)
d758298b3b6f1ebcd494392c0f20b0e119a9e85eTimo Sirainen                value = dec2str(geteuid());
d758298b3b6f1ebcd494392c0f20b0e119a9e85eTimo Sirainen            else if (strcmp(key, "gid") == 0)
d758298b3b6f1ebcd494392c0f20b0e119a9e85eTimo Sirainen                value = dec2str(getegid());
927d3977d5598f12ae18d4fa3f22b9e913f7dd46Timo Sirainen            break;
927d3977d5598f12ae18d4fa3f22b9e913f7dd46Timo Sirainen        case 8:
927d3977d5598f12ae18d4fa3f22b9e913f7dd46Timo Sirainen            if (strcmp(key, "hostname") == 0)
927d3977d5598f12ae18d4fa3f22b9e913f7dd46Timo Sirainen                value = my_hostname;
927d3977d5598f12ae18d4fa3f22b9e913f7dd46Timo Sirainen            break;
927d3977d5598f12ae18d4fa3f22b9e913f7dd46Timo Sirainen        }
e130dc7c712a10215b1e6be56403bbb934826251Timo Sirainen        if (value == NULL) {
e130dc7c712a10215b1e6be56403bbb934826251Timo Sirainen            const char *data = strchr(key, ':');
e130dc7c712a10215b1e6be56403bbb934826251Timo Sirainen
e130dc7c712a10215b1e6be56403bbb934826251Timo Sirainen            if (data != NULL)
e130dc7c712a10215b1e6be56403bbb934826251Timo Sirainen                key = t_strdup_until(key, data++);
e130dc7c712a10215b1e6be56403bbb934826251Timo Sirainen            else
e130dc7c712a10215b1e6be56403bbb934826251Timo Sirainen                data = "";
e130dc7c712a10215b1e6be56403bbb934826251Timo Sirainen            value = var_expand_func(func_table, key, data, context);
e130dc7c712a10215b1e6be56403bbb934826251Timo Sirainen        }
927d3977d5598f12ae18d4fa3f22b9e913f7dd46Timo Sirainen    } T_END;
927d3977d5598f12ae18d4fa3f22b9e913f7dd46Timo Sirainen    return value;
927d3977d5598f12ae18d4fa3f22b9e913f7dd46Timo Sirainen}
927d3977d5598f12ae18d4fa3f22b9e913f7dd46Timo Sirainen
e130dc7c712a10215b1e6be56403bbb934826251Timo Sirainenvoid var_expand_with_funcs(string_t *dest, const char *str,
e130dc7c712a10215b1e6be56403bbb934826251Timo Sirainen               const struct var_expand_table *table,
e130dc7c712a10215b1e6be56403bbb934826251Timo Sirainen               const struct var_expand_func_table *func_table,
e130dc7c712a10215b1e6be56403bbb934826251Timo Sirainen               void *context)
4cb2599c5cdf27362a66ac475ce295409c093c92Timo Sirainen{
3b94ff5951db4d4eddb7a80ed4e3f61207202635Timo Sirainen        const struct var_expand_modifier *m;
3b94ff5951db4d4eddb7a80ed4e3f61207202635Timo Sirainen        const struct var_expand_table *t;
4cb2599c5cdf27362a66ac475ce295409c093c92Timo Sirainen    const char *var;
bb979386852c7689dc66c0fce03319382f66d501Timo Sirainen        struct var_expand_context ctx;
bb979386852c7689dc66c0fce03319382f66d501Timo Sirainen    const char *(*modifier[MAX_MODIFIER_COUNT])
bb979386852c7689dc66c0fce03319382f66d501Timo Sirainen        (const char *, struct var_expand_context *);
f77ffa31038d46ca9c6d24d93e3d76c9aa8d4d0cTimo Sirainen    const char *end;
f77ffa31038d46ca9c6d24d93e3d76c9aa8d4d0cTimo Sirainen    unsigned int i, len, modifier_count;
4cb2599c5cdf27362a66ac475ce295409c093c92Timo Sirainen
bb979386852c7689dc66c0fce03319382f66d501Timo Sirainen    memset(&ctx, 0, sizeof(ctx));
4cb2599c5cdf27362a66ac475ce295409c093c92Timo Sirainen    for (; *str != '\0'; str++) {
4cb2599c5cdf27362a66ac475ce295409c093c92Timo Sirainen        if (*str != '%')
4cb2599c5cdf27362a66ac475ce295409c093c92Timo Sirainen            str_append_c(dest, *str);
4cb2599c5cdf27362a66ac475ce295409c093c92Timo Sirainen        else {
d35364f4d7d139b4150d290e14717e10f1ede4cdTimo Sirainen            int sign = 1;
d35364f4d7d139b4150d290e14717e10f1ede4cdTimo Sirainen
3b94ff5951db4d4eddb7a80ed4e3f61207202635Timo Sirainen            str++;
d35364f4d7d139b4150d290e14717e10f1ede4cdTimo Sirainen            memset(&ctx, 0, sizeof(ctx));
55de1a4d11765e795ec96fddd4858b188be4b892Timo Sirainen
7cd08c67fde5371f662d8c95b30c341741950420Timo Sirainen            /* [<offset>.]<width>[<modifiers>]<variable> */
a40649829bce4c8de6210a2cb4a4b4cf5bb40da8Timo Sirainen            if (*str == '-') {
a40649829bce4c8de6210a2cb4a4b4cf5bb40da8Timo Sirainen                sign = -1;
a40649829bce4c8de6210a2cb4a4b4cf5bb40da8Timo Sirainen                str++;
a40649829bce4c8de6210a2cb4a4b4cf5bb40da8Timo Sirainen            }
7cd08c67fde5371f662d8c95b30c341741950420Timo Sirainen            if (*str == '0') {
d35364f4d7d139b4150d290e14717e10f1ede4cdTimo Sirainen                ctx.zero_padding = TRUE;
7cd08c67fde5371f662d8c95b30c341741950420Timo Sirainen                str++;
7cd08c67fde5371f662d8c95b30c341741950420Timo Sirainen            }
3b94ff5951db4d4eddb7a80ed4e3f61207202635Timo Sirainen            while (*str >= '0' && *str <= '9') {
bb979386852c7689dc66c0fce03319382f66d501Timo Sirainen                ctx.width = ctx.width*10 + (*str - '0');
4cb2599c5cdf27362a66ac475ce295409c093c92Timo Sirainen                str++;
4cb2599c5cdf27362a66ac475ce295409c093c92Timo Sirainen            }
4cb2599c5cdf27362a66ac475ce295409c093c92Timo Sirainen
d35364f4d7d139b4150d290e14717e10f1ede4cdTimo Sirainen            if (*str == '.') {
a40649829bce4c8de6210a2cb4a4b4cf5bb40da8Timo Sirainen                ctx.offset = sign * (int)ctx.width;
bb979386852c7689dc66c0fce03319382f66d501Timo Sirainen                ctx.width = 0;
55de1a4d11765e795ec96fddd4858b188be4b892Timo Sirainen                str++;
d35364f4d7d139b4150d290e14717e10f1ede4cdTimo Sirainen
d35364f4d7d139b4150d290e14717e10f1ede4cdTimo Sirainen                /* if offset was prefixed with zero (or it was
d35364f4d7d139b4150d290e14717e10f1ede4cdTimo Sirainen                   plain zero), just ignore that. zero padding
d35364f4d7d139b4150d290e14717e10f1ede4cdTimo Sirainen                   is done with the width. */
d35364f4d7d139b4150d290e14717e10f1ede4cdTimo Sirainen                ctx.zero_padding = FALSE;
d35364f4d7d139b4150d290e14717e10f1ede4cdTimo Sirainen                if (*str == '0') {
d35364f4d7d139b4150d290e14717e10f1ede4cdTimo Sirainen                    ctx.zero_padding = TRUE;
d35364f4d7d139b4150d290e14717e10f1ede4cdTimo Sirainen                    str++;
d35364f4d7d139b4150d290e14717e10f1ede4cdTimo Sirainen                }
d35364f4d7d139b4150d290e14717e10f1ede4cdTimo Sirainen
55de1a4d11765e795ec96fddd4858b188be4b892Timo Sirainen                while (*str >= '0' && *str <= '9') {
bb979386852c7689dc66c0fce03319382f66d501Timo Sirainen                    ctx.width = ctx.width*10 + (*str - '0');
55de1a4d11765e795ec96fddd4858b188be4b892Timo Sirainen                    str++;
55de1a4d11765e795ec96fddd4858b188be4b892Timo Sirainen                }
55de1a4d11765e795ec96fddd4858b188be4b892Timo Sirainen            }
55de1a4d11765e795ec96fddd4858b188be4b892Timo Sirainen
7cd08c67fde5371f662d8c95b30c341741950420Timo Sirainen                        modifier_count = 0;
7cd08c67fde5371f662d8c95b30c341741950420Timo Sirainen            while (modifier_count < MAX_MODIFIER_COUNT) {
7cd08c67fde5371f662d8c95b30c341741950420Timo Sirainen                modifier[modifier_count] = NULL;
7cd08c67fde5371f662d8c95b30c341741950420Timo Sirainen                for (m = modifiers; m->key != '\0'; m++) {
7cd08c67fde5371f662d8c95b30c341741950420Timo Sirainen                    if (m->key == *str) {
7cd08c67fde5371f662d8c95b30c341741950420Timo Sirainen                        /* @UNSAFE */
7cd08c67fde5371f662d8c95b30c341741950420Timo Sirainen                        modifier[modifier_count] =
7cd08c67fde5371f662d8c95b30c341741950420Timo Sirainen                            m->func;
7cd08c67fde5371f662d8c95b30c341741950420Timo Sirainen                        str++;
7cd08c67fde5371f662d8c95b30c341741950420Timo Sirainen                        break;
7cd08c67fde5371f662d8c95b30c341741950420Timo Sirainen                    }
3b94ff5951db4d4eddb7a80ed4e3f61207202635Timo Sirainen                }
7cd08c67fde5371f662d8c95b30c341741950420Timo Sirainen                if (modifier[modifier_count] == NULL)
7cd08c67fde5371f662d8c95b30c341741950420Timo Sirainen                    break;
7cd08c67fde5371f662d8c95b30c341741950420Timo Sirainen                modifier_count++;
3b94ff5951db4d4eddb7a80ed4e3f61207202635Timo Sirainen            }
3b94ff5951db4d4eddb7a80ed4e3f61207202635Timo Sirainen
3b94ff5951db4d4eddb7a80ed4e3f61207202635Timo Sirainen            if (*str == '\0')
4cb2599c5cdf27362a66ac475ce295409c093c92Timo Sirainen                break;
3b94ff5951db4d4eddb7a80ed4e3f61207202635Timo Sirainen
3b94ff5951db4d4eddb7a80ed4e3f61207202635Timo Sirainen            var = NULL;
f77ffa31038d46ca9c6d24d93e3d76c9aa8d4d0cTimo Sirainen            if (*str == '{' && (end = strchr(str, '}')) != NULL) {
f77ffa31038d46ca9c6d24d93e3d76c9aa8d4d0cTimo Sirainen                /* %{long_key} */
f77ffa31038d46ca9c6d24d93e3d76c9aa8d4d0cTimo Sirainen                len = end - (str + 1);
e130dc7c712a10215b1e6be56403bbb934826251Timo Sirainen                var = var_expand_long(table, func_table,
e130dc7c712a10215b1e6be56403bbb934826251Timo Sirainen                              str+1, len, context);
927d3977d5598f12ae18d4fa3f22b9e913f7dd46Timo Sirainen                if (var != NULL)
927d3977d5598f12ae18d4fa3f22b9e913f7dd46Timo Sirainen                    str = end;
93794594bc682b12353d2d0db08d91ae3e7c56c6Timo Sirainen            } else if (table != NULL) {
f77ffa31038d46ca9c6d24d93e3d76c9aa8d4d0cTimo Sirainen                for (t = table; !TABLE_LAST(t); t++) {
f77ffa31038d46ca9c6d24d93e3d76c9aa8d4d0cTimo Sirainen                    if (t->key == *str) {
f77ffa31038d46ca9c6d24d93e3d76c9aa8d4d0cTimo Sirainen                        var = t->value != NULL ?
f77ffa31038d46ca9c6d24d93e3d76c9aa8d4d0cTimo Sirainen                            t->value : "";
f77ffa31038d46ca9c6d24d93e3d76c9aa8d4d0cTimo Sirainen                        break;
f77ffa31038d46ca9c6d24d93e3d76c9aa8d4d0cTimo Sirainen                    }
3b94ff5951db4d4eddb7a80ed4e3f61207202635Timo Sirainen                }
4cb2599c5cdf27362a66ac475ce295409c093c92Timo Sirainen            }
4cb2599c5cdf27362a66ac475ce295409c093c92Timo Sirainen
3b94ff5951db4d4eddb7a80ed4e3f61207202635Timo Sirainen            if (var == NULL) {
3b94ff5951db4d4eddb7a80ed4e3f61207202635Timo Sirainen                /* not found */
3b94ff5951db4d4eddb7a80ed4e3f61207202635Timo Sirainen                if (*str == '%')
3b94ff5951db4d4eddb7a80ed4e3f61207202635Timo Sirainen                    var = "%";
3b94ff5951db4d4eddb7a80ed4e3f61207202635Timo Sirainen            }
4cb2599c5cdf27362a66ac475ce295409c093c92Timo Sirainen
4cb2599c5cdf27362a66ac475ce295409c093c92Timo Sirainen            if (var != NULL) {
7cd08c67fde5371f662d8c95b30c341741950420Timo Sirainen                for (i = 0; i < modifier_count; i++)
bb979386852c7689dc66c0fce03319382f66d501Timo Sirainen                    var = modifier[i](var, &ctx);
a40649829bce4c8de6210a2cb4a4b4cf5bb40da8Timo Sirainen
a40649829bce4c8de6210a2cb4a4b4cf5bb40da8Timo Sirainen                if (ctx.offset < 0) {
a40649829bce4c8de6210a2cb4a4b4cf5bb40da8Timo Sirainen                    /* if offset is < 0 then we want to
a40649829bce4c8de6210a2cb4a4b4cf5bb40da8Timo Sirainen                       start at the end */
a40649829bce4c8de6210a2cb4a4b4cf5bb40da8Timo Sirainen                    size_t len = strlen(var);
a40649829bce4c8de6210a2cb4a4b4cf5bb40da8Timo Sirainen
a40649829bce4c8de6210a2cb4a4b4cf5bb40da8Timo Sirainen                    if (len > (size_t)-ctx.offset)
a40649829bce4c8de6210a2cb4a4b4cf5bb40da8Timo Sirainen                        var += len + ctx.offset;
a40649829bce4c8de6210a2cb4a4b4cf5bb40da8Timo Sirainen                } else {
a40649829bce4c8de6210a2cb4a4b4cf5bb40da8Timo Sirainen                    while (*var != '\0' && ctx.offset > 0) {
a40649829bce4c8de6210a2cb4a4b4cf5bb40da8Timo Sirainen                        ctx.offset--;
a40649829bce4c8de6210a2cb4a4b4cf5bb40da8Timo Sirainen                        var++;
a40649829bce4c8de6210a2cb4a4b4cf5bb40da8Timo Sirainen                    }
bb979386852c7689dc66c0fce03319382f66d501Timo Sirainen                }
bb979386852c7689dc66c0fce03319382f66d501Timo Sirainen                if (ctx.width == 0)
4cb2599c5cdf27362a66ac475ce295409c093c92Timo Sirainen                    str_append(dest, var);
d35364f4d7d139b4150d290e14717e10f1ede4cdTimo Sirainen                else if (!ctx.zero_padding)
bb979386852c7689dc66c0fce03319382f66d501Timo Sirainen                    str_append_n(dest, var, ctx.width);
7cd08c67fde5371f662d8c95b30c341741950420Timo Sirainen                else {
d35364f4d7d139b4150d290e14717e10f1ede4cdTimo Sirainen                    /* %05d -like padding. no truncation. */
7cd08c67fde5371f662d8c95b30c341741950420Timo Sirainen                    size_t len = strlen(var);
bb979386852c7689dc66c0fce03319382f66d501Timo Sirainen                    while (len < ctx.width) {
7cd08c67fde5371f662d8c95b30c341741950420Timo Sirainen                        str_append_c(dest, '0');
bb979386852c7689dc66c0fce03319382f66d501Timo Sirainen                        ctx.width--;
7cd08c67fde5371f662d8c95b30c341741950420Timo Sirainen                    }
7cd08c67fde5371f662d8c95b30c341741950420Timo Sirainen                    str_append(dest, var);
7cd08c67fde5371f662d8c95b30c341741950420Timo Sirainen                }
4cb2599c5cdf27362a66ac475ce295409c093c92Timo Sirainen            }
4cb2599c5cdf27362a66ac475ce295409c093c92Timo Sirainen        }
4cb2599c5cdf27362a66ac475ce295409c093c92Timo Sirainen    }
4cb2599c5cdf27362a66ac475ce295409c093c92Timo Sirainen}
ee3cb11d230d549367a1213aefe4598345796256Timo Sirainen
e130dc7c712a10215b1e6be56403bbb934826251Timo Sirainenvoid var_expand(string_t *dest, const char *str,
e130dc7c712a10215b1e6be56403bbb934826251Timo Sirainen        const struct var_expand_table *table)
e130dc7c712a10215b1e6be56403bbb934826251Timo Sirainen{
e130dc7c712a10215b1e6be56403bbb934826251Timo Sirainen    var_expand_with_funcs(dest, str, table, NULL, NULL);
e130dc7c712a10215b1e6be56403bbb934826251Timo Sirainen}
e130dc7c712a10215b1e6be56403bbb934826251Timo Sirainen
ee3cb11d230d549367a1213aefe4598345796256Timo Sirainenchar var_get_key(const char *str)
b58cb4b59ccd78ee1c3e0df0bc13c300d1bec380Timo Sirainen{
b58cb4b59ccd78ee1c3e0df0bc13c300d1bec380Timo Sirainen    unsigned int idx, size;
b58cb4b59ccd78ee1c3e0df0bc13c300d1bec380Timo Sirainen
b58cb4b59ccd78ee1c3e0df0bc13c300d1bec380Timo Sirainen    var_get_key_range(str, &idx, &size);
b58cb4b59ccd78ee1c3e0df0bc13c300d1bec380Timo Sirainen    return str[idx];
b58cb4b59ccd78ee1c3e0df0bc13c300d1bec380Timo Sirainen}
b58cb4b59ccd78ee1c3e0df0bc13c300d1bec380Timo Sirainen
b58cb4b59ccd78ee1c3e0df0bc13c300d1bec380Timo Sirainenvoid var_get_key_range(const char *str, unsigned int *idx_r,
b58cb4b59ccd78ee1c3e0df0bc13c300d1bec380Timo Sirainen               unsigned int *size_r)
ee3cb11d230d549367a1213aefe4598345796256Timo Sirainen{
ee3cb11d230d549367a1213aefe4598345796256Timo Sirainen    const struct var_expand_modifier *m;
b58cb4b59ccd78ee1c3e0df0bc13c300d1bec380Timo Sirainen    unsigned int i = 0;
ee3cb11d230d549367a1213aefe4598345796256Timo Sirainen
7cd08c67fde5371f662d8c95b30c341741950420Timo Sirainen    /* [<offset>.]<width>[<modifiers>]<variable> */
b58cb4b59ccd78ee1c3e0df0bc13c300d1bec380Timo Sirainen    while ((str[i] >= '0' && str[i] <= '9') || str[i] == '-')
b58cb4b59ccd78ee1c3e0df0bc13c300d1bec380Timo Sirainen        i++;
ee3cb11d230d549367a1213aefe4598345796256Timo Sirainen
b58cb4b59ccd78ee1c3e0df0bc13c300d1bec380Timo Sirainen    if (str[i] == '.') {
b58cb4b59ccd78ee1c3e0df0bc13c300d1bec380Timo Sirainen        i++;
b58cb4b59ccd78ee1c3e0df0bc13c300d1bec380Timo Sirainen        while (str[i] >= '0' && str[i] <= '9')
b58cb4b59ccd78ee1c3e0df0bc13c300d1bec380Timo Sirainen            i++;
ee3cb11d230d549367a1213aefe4598345796256Timo Sirainen    }
ee3cb11d230d549367a1213aefe4598345796256Timo Sirainen
7cd08c67fde5371f662d8c95b30c341741950420Timo Sirainen    do {
7cd08c67fde5371f662d8c95b30c341741950420Timo Sirainen        for (m = modifiers; m->key != '\0'; m++) {
b58cb4b59ccd78ee1c3e0df0bc13c300d1bec380Timo Sirainen            if (m->key == str[i]) {
b58cb4b59ccd78ee1c3e0df0bc13c300d1bec380Timo Sirainen                i++;
7cd08c67fde5371f662d8c95b30c341741950420Timo Sirainen                break;
7cd08c67fde5371f662d8c95b30c341741950420Timo Sirainen            }
ee3cb11d230d549367a1213aefe4598345796256Timo Sirainen        }
7cd08c67fde5371f662d8c95b30c341741950420Timo Sirainen    } while (m->key != '\0');
ee3cb11d230d549367a1213aefe4598345796256Timo Sirainen
b58cb4b59ccd78ee1c3e0df0bc13c300d1bec380Timo Sirainen    if (str[i] != '{') {
b58cb4b59ccd78ee1c3e0df0bc13c300d1bec380Timo Sirainen        /* short key */
b58cb4b59ccd78ee1c3e0df0bc13c300d1bec380Timo Sirainen        *idx_r = i;
b58cb4b59ccd78ee1c3e0df0bc13c300d1bec380Timo Sirainen        *size_r = str[i] == '\0' ? 0 : 1;
b58cb4b59ccd78ee1c3e0df0bc13c300d1bec380Timo Sirainen    } else {
b58cb4b59ccd78ee1c3e0df0bc13c300d1bec380Timo Sirainen        /* long key */
b58cb4b59ccd78ee1c3e0df0bc13c300d1bec380Timo Sirainen        *idx_r = ++i;
b58cb4b59ccd78ee1c3e0df0bc13c300d1bec380Timo Sirainen        for (; str[i] != '\0'; i++) {
b58cb4b59ccd78ee1c3e0df0bc13c300d1bec380Timo Sirainen            if (str[i] == '}')
b58cb4b59ccd78ee1c3e0df0bc13c300d1bec380Timo Sirainen                break;
b58cb4b59ccd78ee1c3e0df0bc13c300d1bec380Timo Sirainen        }
b58cb4b59ccd78ee1c3e0df0bc13c300d1bec380Timo Sirainen        *size_r = i - *idx_r;
b58cb4b59ccd78ee1c3e0df0bc13c300d1bec380Timo Sirainen    }
ee3cb11d230d549367a1213aefe4598345796256Timo Sirainen}
3e859421cc59d28d4ba99f32830e3d0531334813Timo Sirainen
771178a5c5285aa596723591271c8936c4007f1bTimo Sirainenstatic bool var_has_long_key(const char **str, const char *long_key)
771178a5c5285aa596723591271c8936c4007f1bTimo Sirainen{
771178a5c5285aa596723591271c8936c4007f1bTimo Sirainen    const char *start, *end;
771178a5c5285aa596723591271c8936c4007f1bTimo Sirainen
771178a5c5285aa596723591271c8936c4007f1bTimo Sirainen    start = strchr(*str, '{');
771178a5c5285aa596723591271c8936c4007f1bTimo Sirainen    i_assert(start != NULL);
771178a5c5285aa596723591271c8936c4007f1bTimo Sirainen
771178a5c5285aa596723591271c8936c4007f1bTimo Sirainen    end = strchr(++start, '}');
771178a5c5285aa596723591271c8936c4007f1bTimo Sirainen    if (end == NULL)
771178a5c5285aa596723591271c8936c4007f1bTimo Sirainen        return FALSE;
771178a5c5285aa596723591271c8936c4007f1bTimo Sirainen
771178a5c5285aa596723591271c8936c4007f1bTimo Sirainen    if (strncmp(start, long_key, end-start) == 0 &&
771178a5c5285aa596723591271c8936c4007f1bTimo Sirainen        long_key[end-start] == '\0')
771178a5c5285aa596723591271c8936c4007f1bTimo Sirainen        return TRUE;
771178a5c5285aa596723591271c8936c4007f1bTimo Sirainen
771178a5c5285aa596723591271c8936c4007f1bTimo Sirainen    *str = end;
771178a5c5285aa596723591271c8936c4007f1bTimo Sirainen    return FALSE;
771178a5c5285aa596723591271c8936c4007f1bTimo Sirainen}
771178a5c5285aa596723591271c8936c4007f1bTimo Sirainen
f77ffa31038d46ca9c6d24d93e3d76c9aa8d4d0cTimo Sirainenbool var_has_key(const char *str, char key, const char *long_key)
5daf9fa081859b99c5082b680f3f8a70b82a76f0Timo Sirainen{
f77ffa31038d46ca9c6d24d93e3d76c9aa8d4d0cTimo Sirainen    char c;
f77ffa31038d46ca9c6d24d93e3d76c9aa8d4d0cTimo Sirainen
5daf9fa081859b99c5082b680f3f8a70b82a76f0Timo Sirainen    for (; *str != '\0'; str++) {
5daf9fa081859b99c5082b680f3f8a70b82a76f0Timo Sirainen        if (*str == '%' && str[1] != '\0') {
5daf9fa081859b99c5082b680f3f8a70b82a76f0Timo Sirainen            str++;
f77ffa31038d46ca9c6d24d93e3d76c9aa8d4d0cTimo Sirainen            c = var_get_key(str);
f77ffa31038d46ca9c6d24d93e3d76c9aa8d4d0cTimo Sirainen            if (c == key)
5daf9fa081859b99c5082b680f3f8a70b82a76f0Timo Sirainen                return TRUE;
f77ffa31038d46ca9c6d24d93e3d76c9aa8d4d0cTimo Sirainen
771178a5c5285aa596723591271c8936c4007f1bTimo Sirainen            if (c == '{' && long_key != NULL) {
771178a5c5285aa596723591271c8936c4007f1bTimo Sirainen                if (var_has_long_key(&str, long_key))
f77ffa31038d46ca9c6d24d93e3d76c9aa8d4d0cTimo Sirainen                    return TRUE;
f77ffa31038d46ca9c6d24d93e3d76c9aa8d4d0cTimo Sirainen            }
5daf9fa081859b99c5082b680f3f8a70b82a76f0Timo Sirainen        }
5daf9fa081859b99c5082b680f3f8a70b82a76f0Timo Sirainen    }
5daf9fa081859b99c5082b680f3f8a70b82a76f0Timo Sirainen    return FALSE;
5daf9fa081859b99c5082b680f3f8a70b82a76f0Timo Sirainen}
5daf9fa081859b99c5082b680f3f8a70b82a76f0Timo Sirainen
3e859421cc59d28d4ba99f32830e3d0531334813Timo Sirainenconst struct var_expand_table *
3e859421cc59d28d4ba99f32830e3d0531334813Timo Sirainenvar_expand_table_build(char key, const char *value, char key2, ...)
3e859421cc59d28d4ba99f32830e3d0531334813Timo Sirainen{
4ee00532a265bdfb38539d811fcd12d51210ac35Timo Sirainen    ARRAY(struct var_expand_table) variables;
3e859421cc59d28d4ba99f32830e3d0531334813Timo Sirainen    struct var_expand_table *var;
3e859421cc59d28d4ba99f32830e3d0531334813Timo Sirainen    va_list args;
3e859421cc59d28d4ba99f32830e3d0531334813Timo Sirainen
3e859421cc59d28d4ba99f32830e3d0531334813Timo Sirainen    i_assert(key != '\0');
3e859421cc59d28d4ba99f32830e3d0531334813Timo Sirainen
3e859421cc59d28d4ba99f32830e3d0531334813Timo Sirainen    t_array_init(&variables, 16);
3e859421cc59d28d4ba99f32830e3d0531334813Timo Sirainen    var = array_append_space(&variables);
3e859421cc59d28d4ba99f32830e3d0531334813Timo Sirainen    var->key = key;
3e859421cc59d28d4ba99f32830e3d0531334813Timo Sirainen    var->value = value;
3e859421cc59d28d4ba99f32830e3d0531334813Timo Sirainen
3e859421cc59d28d4ba99f32830e3d0531334813Timo Sirainen    va_start(args, key2);
3e859421cc59d28d4ba99f32830e3d0531334813Timo Sirainen    for (key = key2; key != '\0'; key = va_arg(args, int)) {
3e859421cc59d28d4ba99f32830e3d0531334813Timo Sirainen        var = array_append_space(&variables);
3e859421cc59d28d4ba99f32830e3d0531334813Timo Sirainen        var->key = key;
3e859421cc59d28d4ba99f32830e3d0531334813Timo Sirainen        var->value = va_arg(args, const char *);
3e859421cc59d28d4ba99f32830e3d0531334813Timo Sirainen    }
3e859421cc59d28d4ba99f32830e3d0531334813Timo Sirainen    va_end(args);
3e859421cc59d28d4ba99f32830e3d0531334813Timo Sirainen
3e859421cc59d28d4ba99f32830e3d0531334813Timo Sirainen    /* 0, NULL entry */
31a574fda352ef4f71dbff9c30e15e4744e132c0Timo Sirainen    array_append_zero(&variables);
3e859421cc59d28d4ba99f32830e3d0531334813Timo Sirainen    return array_idx(&variables, 0);
3e859421cc59d28d4ba99f32830e3d0531334813Timo Sirainen}