1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
|
/* Copyright (c) 2003-2018 Dovecot authors, see the included COPYING file */
#include "lib.h"
#include "sort.h"
#include "mail-cache.h"
#include "mail-storage-private.h"
static struct mailbox_header_lookup_ctx *
mailbox_header_lookup_init_real(struct mailbox *box,
const char *const headers[])
{
struct mail_cache_field *fields, header_field = {
.type = MAIL_CACHE_FIELD_HEADER,
.decision = MAIL_CACHE_DECISION_NO,
};
struct mailbox_header_lookup_ctx *ctx;
const char *const *name;
const char **sorted_headers, **dest_name;
pool_t pool;
unsigned int i, j, count;
i_assert(*headers != NULL);
for (count = 0, name = headers; *name != NULL; name++)
count++;
/* @UNSAFE: headers need to be sorted for filter stream. */
sorted_headers = t_new(const char *, count);
memcpy(sorted_headers, headers, count * sizeof(*sorted_headers));
i_qsort(sorted_headers, count, sizeof(*sorted_headers), i_strcasecmp_p);
headers = sorted_headers;
/* @UNSAFE */
fields = t_new(struct mail_cache_field, count);
for (i = j = 0; i < count; i++) {
if (i > 0 && strcasecmp(headers[i-1], headers[i]) == 0)
continue;
header_field.name = t_strconcat("hdr.", headers[i], NULL);
fields[j++] = header_field;
}
count = j;
mail_cache_register_fields(box->cache, fields, count);
pool = pool_alloconly_create("mailbox_header_lookup_ctx", 1024);
ctx = p_new(pool, struct mailbox_header_lookup_ctx, 1);
ctx->box = box;
ctx->refcount = 1;
ctx->pool = pool;
ctx->count = count;
ctx->idx = p_new(pool, unsigned int, count);
/* @UNSAFE */
dest_name = p_new(pool, const char *, count + 1);
for (i = 0; i < count; i++) {
ctx->idx[i] = fields[i].idx;
dest_name[i] = p_strdup(pool, fields[i].name + strlen("hdr."));
}
ctx->name = dest_name;
return ctx;
}
struct mailbox_header_lookup_ctx *
mailbox_header_lookup_init(struct mailbox *box, const char *const headers[])
{
struct mailbox_header_lookup_ctx *ctx;
T_BEGIN {
ctx = mailbox_header_lookup_init_real(box, headers);
} T_END;
return ctx;
}
void mailbox_header_lookup_ref(struct mailbox_header_lookup_ctx *ctx)
{
i_assert(ctx->refcount > 0);
ctx->refcount++;
}
void mailbox_header_lookup_unref(struct mailbox_header_lookup_ctx **_ctx)
{
struct mailbox_header_lookup_ctx *ctx = *_ctx;
if (ctx == NULL)
return;
*_ctx = NULL;
i_assert(ctx->refcount > 0);
if (--ctx->refcount > 0)
return;
pool_unref(&ctx->pool);
}
struct mailbox_header_lookup_ctx *
mailbox_header_lookup_merge(const struct mailbox_header_lookup_ctx *hdr1,
const struct mailbox_header_lookup_ctx *hdr2)
{
ARRAY_TYPE(const_string) names;
unsigned int i;
i_assert(hdr1->box == hdr2->box);
t_array_init(&names, 32);
for (i = 0; i < hdr1->count; i++)
array_push_back(&names, &hdr1->name[i]);
for (i = 0; i < hdr2->count; i++)
array_push_back(&names, &hdr2->name[i]);
array_append_zero(&names);
return mailbox_header_lookup_init(hdr1->box, array_front(&names));
}
|
