Cross Reference: /dovecot/src/lib-storage/index/maildir/maildir-keywords.c
maildir-keywords.c revision 3bc2b84acd8c5731a0bc9d69fbcd3103b68e9b52
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
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen/* Copyright (C) 2005 Timo Sirainen */
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen/* note that everything here depends on uidlist file being locked the whole
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen time. that's why we don't have any locking of our own, or that we do things
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen that would be racy otherwise. */
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen#include "lib.h"
16f816d3f3c32ae3351834253f52ddd0212bcbf3Timo Sirainen#include "array.h"
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen#include "ioloop.h"
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen#include "hash.h"
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen#include "str.h"
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen#include "istream.h"
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen#include "file-dotlock.h"
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen#include "write-full.h"
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen#include "maildir-storage.h"
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen#include "maildir-uidlist.h"
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen#include "maildir-keywords.h"
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen#include <stdlib.h>
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen#include <sys/stat.h>
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen#include <utime.h>
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainenstruct maildir_keywords {
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen struct maildir_mailbox *mbox;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen char *path;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen pool_t pool;
8d80659e504ffb34bb0c6a633184fece35751b18Timo Sirainen ARRAY_TYPE(keywords) list;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen struct hash_table *hash; /* name -> idx+1 */
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen struct dotlock_settings dotlock_settings;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen time_t synced_mtime;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen unsigned int synced:1;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen unsigned int changed:1;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen};
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainenstruct maildir_keywords_sync_ctx {
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen struct maildir_keywords *mk;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen struct mail_index *index;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
8d80659e504ffb34bb0c6a633184fece35751b18Timo Sirainen const ARRAY_TYPE(keywords) *keywords;
8d80659e504ffb34bb0c6a633184fece35751b18Timo Sirainen ARRAY_DEFINE(idx_to_chr, char);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen unsigned int chridx_to_idx[MAILDIR_MAX_KEYWORDS];
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen};
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainenstruct maildir_keywords *maildir_keywords_init(struct maildir_mailbox *mbox)
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen{
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen struct maildir_keywords *mk;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen mk = i_new(struct maildir_keywords, 1);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen mk->mbox = mbox;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen mk->path = i_strconcat(mbox->control_dir,
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen "/" MAILDIR_KEYWORDS_NAME, NULL);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen mk->pool = pool_alloconly_create("maildir keywords", 256);
91dca97b367c54a139c268b56a0c67f564bd9197Timo Sirainen i_array_init(&mk->list, MAILDIR_MAX_KEYWORDS);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen mk->hash = hash_create(default_pool, mk->pool, 0,
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen strcase_hash, (hash_cmp_callback_t *)strcasecmp);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen return mk;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen}
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainenvoid maildir_keywords_deinit(struct maildir_keywords *mk)
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen{
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen hash_destroy(mk->hash);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen array_free(&mk->list);
79fcd3f95a6266cc62ceaa753e56dd4456ab7c4bTimo Sirainen pool_unref(mk->pool);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen i_free(mk->path);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen i_free(mk);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen}
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainenstatic void maildir_keywords_clear(struct maildir_keywords *mk)
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen{
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen array_clear(&mk->list);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen hash_clear(mk->hash, FALSE);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen p_clear(mk->pool);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen}
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainenstatic int maildir_keywords_sync(struct maildir_keywords *mk)
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen{
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen struct istream *input;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen struct stat st;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen char *line, *p, *new_name;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen const char **strp;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen int fd, idx;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
1e47cfede3a0b62654105daab00e97b5d660bc6bTimo Sirainen /* Remember that we rely on uidlist file locking in here. That's why
1e47cfede3a0b62654105daab00e97b5d660bc6bTimo Sirainen we rely on stat()'s timestamp and don't bother handling ESTALE
1e47cfede3a0b62654105daab00e97b5d660bc6bTimo Sirainen errors. */
1e47cfede3a0b62654105daab00e97b5d660bc6bTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen if (stat(mk->path, &st) < 0) {
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen if (errno == ENOENT) {
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen maildir_keywords_clear(mk);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen mk->synced = TRUE;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen return 0;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen }
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen mail_storage_set_critical(STORAGE(mk->mbox->storage),
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen "stat(%s) failed: %m", mk->path);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen return -1;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen }
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen if (st.st_mtime == mk->synced_mtime) {
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen /* hasn't changed */
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen mk->synced = TRUE;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen return 0;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen }
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen mk->synced_mtime = st.st_mtime;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen fd = open(mk->path, O_RDONLY);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen if (fd == -1) {
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen if (errno == ENOENT) {
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen maildir_keywords_clear(mk);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen mk->synced = TRUE;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen return 0;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen }
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen mail_storage_set_critical(STORAGE(mk->mbox->storage),
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen "open(%s) failed: %m", mk->path);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen return -1;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen }
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen maildir_keywords_clear(mk);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen input = i_stream_create_file(fd, default_pool, 1024, FALSE);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen while ((line = i_stream_read_next_line(input)) != NULL) {
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen p = strchr(line, ' ');
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen if (p == NULL) {
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen /* note that when converting .customflags file this
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen case happens in the first line. */
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen continue;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen }
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen *p++ = '\0';
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen idx = atoi(line);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen if (idx < 0 || idx >= MAILDIR_MAX_KEYWORDS) {
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen /* shouldn't happen */
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen continue;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen }
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen /* save it */
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen new_name = p_strdup(mk->pool, p);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen hash_insert(mk->hash, new_name, POINTER_CAST(idx + 1));
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
8d80659e504ffb34bb0c6a633184fece35751b18Timo Sirainen strp = array_idx_modifiable(&mk->list, idx);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen *strp = new_name;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen }
cd56a23e21f1df3f79648cf07e2f4385e2fadebbTimo Sirainen i_stream_destroy(&input);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen if (close(fd) < 0) {
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen mail_storage_set_critical(STORAGE(mk->mbox->storage),
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen "close(%s) failed: %m", mk->path);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen return -1;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen }
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen mk->synced = TRUE;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen return 0;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen}
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainenstatic int
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainenmaildir_keywords_lookup(struct maildir_keywords *mk, const char *name)
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen{
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen void *p;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen i_assert(maildir_uidlist_is_locked(mk->mbox->uidlist));
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen p = hash_lookup(mk->hash, name);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen if (p == NULL) {
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen if (mk->synced)
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen return -1;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen if (maildir_keywords_sync(mk) < 0)
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen return -1;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen p = hash_lookup(mk->hash, name);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen if (p == NULL)
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen return -1;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen }
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen return POINTER_CAST_TO(p, int)-1;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen}
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainenstatic void
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainenmaildir_keywords_create(struct maildir_keywords *mk, const char *name,
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen unsigned int chridx)
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen{
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen const char **strp;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen char *new_name;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen i_assert(chridx < MAILDIR_MAX_KEYWORDS);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen new_name = p_strdup(mk->pool, name);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen hash_insert(mk->hash, new_name, POINTER_CAST(chridx + 1));
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
8d80659e504ffb34bb0c6a633184fece35751b18Timo Sirainen strp = array_idx_modifiable(&mk->list, chridx);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen *strp = new_name;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen mk->changed = TRUE;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen}
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainenstatic int
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainenmaildir_keywords_lookup_or_create(struct maildir_keywords *mk, const char *name)
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen{
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen const char *const *keywords;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen unsigned int i, count;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen int idx;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen idx = maildir_keywords_lookup(mk, name);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen if (idx >= 0)
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen return idx;
3bc2b84acd8c5731a0bc9d69fbcd3103b68e9b52Timo Sirainen if (!mk->synced) {
3bc2b84acd8c5731a0bc9d69fbcd3103b68e9b52Timo Sirainen /* we couldn't open the dovecot-keywords file. */
3bc2b84acd8c5731a0bc9d69fbcd3103b68e9b52Timo Sirainen return -1;
3bc2b84acd8c5731a0bc9d69fbcd3103b68e9b52Timo Sirainen }
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen /* see if we are full */
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen keywords = array_get(&mk->list, &count);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen for (i = 0; i < count; i++) {
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen if (keywords[i] == NULL)
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen break;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen }
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen if (i == count && count >= MAILDIR_MAX_KEYWORDS)
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen return -1;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen maildir_keywords_create(mk, name, i);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen return i;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen}
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainenstatic const char *
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainenmaildir_keywords_idx(struct maildir_keywords *mk, unsigned int idx)
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen{
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen const char *const *keywords;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen unsigned int count;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen i_assert(maildir_uidlist_is_locked(mk->mbox->uidlist));
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen keywords = array_get(&mk->list, &count);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen if (idx >= count) {
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen if (mk->synced)
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen return NULL;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen if (maildir_keywords_sync(mk) < 0)
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen return NULL;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen keywords = array_get(&mk->list, &count);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen }
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen return idx >= count ? NULL : keywords[idx];
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen}
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainenstatic int maildir_keywords_commit(struct maildir_keywords *mk)
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen{
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen struct dotlock *dotlock;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen const char *lock_path, *const *keywords;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen unsigned int i, count;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen struct utimbuf ut;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen string_t *str;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen int fd;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen mk->synced = FALSE;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen if (!mk->changed)
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen return 0;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen /* we could just create the temp file directly, but doing it this
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen ways avoids potential problems with overwriting contents in
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen malicious symlinks */
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen t_push();
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen lock_path = t_strconcat(mk->path, ".lock", NULL);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen (void)unlink(lock_path);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen fd = file_dotlock_open(&mk->dotlock_settings, mk->path,
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen DOTLOCK_CREATE_FLAG_NONBLOCK, &dotlock);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen if (fd == -1) {
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen mail_storage_set_critical(STORAGE(mk->mbox->storage),
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen "file_dotlock_open(%s) failed: %m", mk->path);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen t_pop();
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen return -1;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen }
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen str = t_str_new(256);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen keywords = array_get(&mk->list, &count);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen for (i = 0; i < count; i++) {
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen if (keywords[i] != NULL)
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen str_printfa(str, "%u %s\n", i, keywords[i]);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen }
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen if (write_full(fd, str_data(str), str_len(str)) < 0) {
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen mail_storage_set_critical(STORAGE(mk->mbox->storage),
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen "write_full(%s) failed: %m", mk->path);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen (void)file_dotlock_delete(&dotlock);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen t_pop();
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen return -1;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen }
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen /* mtime must grow every time */
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen mk->synced_mtime = ioloop_time <= mk->synced_mtime ?
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen mk->synced_mtime + 1 : ioloop_time;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen ut.actime = ioloop_time;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen ut.modtime = mk->synced_mtime;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen if (utime(lock_path, &ut) < 0) {
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen mail_storage_set_critical(STORAGE(mk->mbox->storage),
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen "utime(%s) failed: %m", lock_path);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen return -1;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen }
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen if (file_dotlock_replace(&dotlock, 0) < 0) {
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen mail_storage_set_critical(STORAGE(mk->mbox->storage),
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen "file_dotlock_replace(%s) failed: %m", mk->path);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen t_pop();
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen return -1;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen }
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen mk->changed = FALSE;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen t_pop();
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen return 0;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen}
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainenstruct maildir_keywords_sync_ctx *
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainenmaildir_keywords_sync_init(struct maildir_keywords *mk,
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen struct mail_index *index)
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen{
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen struct maildir_keywords_sync_ctx *ctx;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen ctx = i_new(struct maildir_keywords_sync_ctx, 1);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen ctx->mk = mk;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen ctx->index = index;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen ctx->keywords = mail_index_get_keywords(index);
91dca97b367c54a139c268b56a0c67f564bd9197Timo Sirainen i_array_init(&ctx->idx_to_chr, MAILDIR_MAX_KEYWORDS);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen return ctx;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen}
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainenvoid maildir_keywords_sync_deinit(struct maildir_keywords_sync_ctx *ctx)
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen{
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen maildir_keywords_commit(ctx->mk);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen array_free(&ctx->idx_to_chr);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen i_free(ctx);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen}
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainenunsigned int maildir_keywords_char_idx(struct maildir_keywords_sync_ctx *ctx,
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen char keyword)
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen{
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen const char *name;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen unsigned int chridx, idx;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen i_assert(keyword >= MAILDIR_KEYWORD_FIRST &&
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen keyword <= MAILDIR_KEYWORD_LAST);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen chridx = keyword - MAILDIR_KEYWORD_FIRST;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen if (ctx->chridx_to_idx[chridx] != 0)
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen return ctx->chridx_to_idx[chridx];
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen /* lookup / create */
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen name = maildir_keywords_idx(ctx->mk, chridx);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen if (name == NULL) {
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen /* name is lost. just generate one ourself. */
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen name = t_strdup_printf("unknown-%u", chridx);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen while (maildir_keywords_lookup(ctx->mk, name) >= 0) {
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen /* don't create a duplicate name.
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen keep changing the name until it doesn't exist */
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen name = t_strconcat(name, "?", NULL);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen }
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen maildir_keywords_create(ctx->mk, name, chridx);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen }
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen if (!mail_index_keyword_lookup(ctx->index, name, TRUE, &idx))
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen i_unreached();
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen ctx->chridx_to_idx[chridx] = idx;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen return idx;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen}
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainenchar maildir_keywords_idx_char(struct maildir_keywords_sync_ctx *ctx,
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen unsigned int idx)
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen{
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen const char *const *name_p;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen char *chr_p;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen int chridx;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
8d80659e504ffb34bb0c6a633184fece35751b18Timo Sirainen chr_p = array_idx_modifiable(&ctx->idx_to_chr, idx);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen if (*chr_p != '\0')
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen return *chr_p;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen name_p = array_idx(ctx->keywords, idx);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen chridx = maildir_keywords_lookup_or_create(ctx->mk, *name_p);
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen if (chridx < 0)
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen return '\0';
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen *chr_p = chridx + MAILDIR_KEYWORD_FIRST;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen return *chr_p;
cce169a321c9c629e4f2db1a69dae3b75bbcb27aTimo Sirainen}