Diffstat (limited to 'libetpan/src/driver/implementation/mbox/mboxdriver_cached.c') (more/less context) (ignore whitespace changes)
-rw-r--r-- | libetpan/src/driver/implementation/mbox/mboxdriver_cached.c | 1337 |
1 files changed, 1337 insertions, 0 deletions
diff --git a/libetpan/src/driver/implementation/mbox/mboxdriver_cached.c b/libetpan/src/driver/implementation/mbox/mboxdriver_cached.c new file mode 100644 index 0000000..eaa0e48 --- a/dev/null +++ b/libetpan/src/driver/implementation/mbox/mboxdriver_cached.c | |||
@@ -0,0 +1,1337 @@ | |||
1 | /* | ||
2 | * libEtPan! -- a mail stuff library | ||
3 | * | ||
4 | * Copyright (C) 2001, 2005 - DINH Viet Hoa | ||
5 | * All rights reserved. | ||
6 | * | ||
7 | * Redistribution and use in source and binary forms, with or without | ||
8 | * modification, are permitted provided that the following conditions | ||
9 | * are met: | ||
10 | * 1. Redistributions of source code must retain the above copyright | ||
11 | * notice, this list of conditions and the following disclaimer. | ||
12 | * 2. Redistributions in binary form must reproduce the above copyright | ||
13 | * notice, this list of conditions and the following disclaimer in the | ||
14 | * documentation and/or other materials provided with the distribution. | ||
15 | * 3. Neither the name of the libEtPan! project nor the names of its | ||
16 | * contributors may be used to endorse or promote products derived | ||
17 | * from this software without specific prior written permission. | ||
18 | * | ||
19 | * THIS SOFTWARE IS PROVIDED BY THE AUTHORS AND CONTRIBUTORS ``AS IS'' AND | ||
20 | * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE | ||
21 | * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE | ||
22 | * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHORS OR CONTRIBUTORS BE LIABLE | ||
23 | * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL | ||
24 | * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS | ||
25 | * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) | ||
26 | * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT | ||
27 | * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY | ||
28 | * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF | ||
29 | * SUCH DAMAGE. | ||
30 | */ | ||
31 | |||
32 | /* | ||
33 | * $Id$ | ||
34 | */ | ||
35 | |||
36 | #include "mboxdriver_cached.h" | ||
37 | |||
38 | #include <stdio.h> | ||
39 | #include <string.h> | ||
40 | #include <dirent.h> | ||
41 | #include <unistd.h> | ||
42 | #include <ctype.h> | ||
43 | #include <sys/types.h> | ||
44 | #include <sys/stat.h> | ||
45 | #include <fcntl.h> | ||
46 | #include <stdlib.h> | ||
47 | |||
48 | #include "mail.h" | ||
49 | #include "mail_cache_db.h" | ||
50 | #include "mboxdriver.h" | ||
51 | #include "mboxdriver_tools.h" | ||
52 | #include "maildriver_tools.h" | ||
53 | #include "mailmbox.h" | ||
54 | #include "maildriver.h" | ||
55 | #include "carray.h" | ||
56 | #include "generic_cache.h" | ||
57 | #include "imfcache.h" | ||
58 | #include "mboxdriver_cached_message.h" | ||
59 | #include "libetpan-config.h" | ||
60 | |||
61 | static int mboxdriver_cached_initialize(mailsession * session); | ||
62 | |||
63 | static void mboxdriver_cached_uninitialize(mailsession * session); | ||
64 | |||
65 | static int mboxdriver_cached_parameters(mailsession * session, | ||
66 | int id, void * value); | ||
67 | |||
68 | static int mboxdriver_cached_connect_path(mailsession * session, char * path); | ||
69 | |||
70 | static int mboxdriver_cached_logout(mailsession * session); | ||
71 | |||
72 | static int mboxdriver_cached_check_folder(mailsession * session); | ||
73 | |||
74 | static int mboxdriver_cached_expunge_folder(mailsession * session); | ||
75 | |||
76 | static int mboxdriver_cached_status_folder(mailsession * session, char * mb, | ||
77 | uint32_t * result_messages, uint32_t * result_recent, | ||
78 | uint32_t * result_unseen); | ||
79 | static int mboxdriver_cached_messages_number(mailsession * session, char * mb, | ||
80 | uint32_t * result); | ||
81 | static int mboxdriver_cached_recent_number(mailsession * session, char * mb, | ||
82 | uint32_t * result); | ||
83 | static int mboxdriver_cached_unseen_number(mailsession * session, char * mb, | ||
84 | uint32_t * result); | ||
85 | |||
86 | static int mboxdriver_cached_append_message(mailsession * session, | ||
87 | char * message, size_t size); | ||
88 | |||
89 | static int mboxdriver_cached_append_message_flags(mailsession * session, | ||
90 | char * message, size_t size, struct mail_flags * flags); | ||
91 | |||
92 | static int | ||
93 | mboxdriver_cached_get_messages_list(mailsession * session, | ||
94 | struct mailmessage_list ** result); | ||
95 | |||
96 | static int | ||
97 | mboxdriver_cached_get_envelopes_list(mailsession * session, | ||
98 | struct mailmessage_list * env_list); | ||
99 | |||
100 | static int mboxdriver_cached_remove_message(mailsession * session, | ||
101 | uint32_t num); | ||
102 | |||
103 | static int mboxdriver_cached_get_message(mailsession * session, | ||
104 | uint32_t num, mailmessage ** result); | ||
105 | |||
106 | static int mboxdriver_cached_get_message_by_uid(mailsession * session, | ||
107 | const char * uid, | ||
108 | mailmessage ** result); | ||
109 | |||
110 | static mailsession_driver local_mbox_cached_session_driver = { | ||
111 | .sess_name = "mbox-cached", | ||
112 | |||
113 | .sess_initialize = mboxdriver_cached_initialize, | ||
114 | .sess_uninitialize = mboxdriver_cached_uninitialize, | ||
115 | |||
116 | .sess_parameters = mboxdriver_cached_parameters, | ||
117 | |||
118 | .sess_connect_path = mboxdriver_cached_connect_path, | ||
119 | .sess_connect_stream = NULL, | ||
120 | .sess_starttls = NULL, | ||
121 | .sess_login = NULL, | ||
122 | .sess_logout = mboxdriver_cached_logout, | ||
123 | .sess_noop = NULL, | ||
124 | |||
125 | .sess_build_folder_name = NULL, | ||
126 | .sess_create_folder = NULL, | ||
127 | .sess_delete_folder = NULL, | ||
128 | .sess_rename_folder = NULL, | ||
129 | .sess_check_folder = mboxdriver_cached_check_folder, | ||
130 | .sess_examine_folder = NULL, | ||
131 | .sess_select_folder = NULL, | ||
132 | .sess_expunge_folder = mboxdriver_cached_expunge_folder, | ||
133 | .sess_status_folder = mboxdriver_cached_status_folder, | ||
134 | .sess_messages_number = mboxdriver_cached_messages_number, | ||
135 | .sess_recent_number = mboxdriver_cached_recent_number, | ||
136 | .sess_unseen_number = mboxdriver_cached_unseen_number, | ||
137 | .sess_list_folders = NULL, | ||
138 | .sess_lsub_folders = NULL, | ||
139 | .sess_subscribe_folder = NULL, | ||
140 | .sess_unsubscribe_folder = NULL, | ||
141 | |||
142 | .sess_append_message = mboxdriver_cached_append_message, | ||
143 | .sess_append_message_flags = mboxdriver_cached_append_message_flags, | ||
144 | |||
145 | .sess_copy_message = NULL, | ||
146 | .sess_move_message = NULL, | ||
147 | |||
148 | .sess_get_messages_list = mboxdriver_cached_get_messages_list, | ||
149 | .sess_get_envelopes_list = mboxdriver_cached_get_envelopes_list, | ||
150 | .sess_remove_message = mboxdriver_cached_remove_message, | ||
151 | #if 0 | ||
152 | .sess_search_messages = maildriver_generic_search_messages, | ||
153 | #endif | ||
154 | |||
155 | .sess_get_message = mboxdriver_cached_get_message, | ||
156 | .sess_get_message_by_uid = mboxdriver_cached_get_message_by_uid, | ||
157 | }; | ||
158 | |||
159 | mailsession_driver * mbox_cached_session_driver = | ||
160 | &local_mbox_cached_session_driver; | ||
161 | |||
162 | |||
163 | #define ENV_NAME "env.db" | ||
164 | #define FLAGS_NAME "flags.db" | ||
165 | |||
166 | |||
167 | |||
168 | static int mbox_error_to_mail_error(int error) | ||
169 | { | ||
170 | switch (error) { | ||
171 | case MAILMBOX_NO_ERROR: | ||
172 | return MAIL_NO_ERROR; | ||
173 | |||
174 | case MAILMBOX_ERROR_PARSE: | ||
175 | return MAIL_ERROR_PARSE; | ||
176 | |||
177 | case MAILMBOX_ERROR_INVAL: | ||
178 | return MAIL_ERROR_INVAL; | ||
179 | |||
180 | case MAILMBOX_ERROR_FILE_NOT_FOUND: | ||
181 | return MAIL_ERROR_PARSE; | ||
182 | |||
183 | case MAILMBOX_ERROR_MEMORY: | ||
184 | return MAIL_ERROR_MEMORY; | ||
185 | |||
186 | case MAILMBOX_ERROR_TEMPORARY_FILE: | ||
187 | return MAIL_ERROR_PARSE; | ||
188 | |||
189 | case MAILMBOX_ERROR_FILE: | ||
190 | return MAIL_ERROR_FILE; | ||
191 | |||
192 | case MAILMBOX_ERROR_MSG_NOT_FOUND: | ||
193 | return MAIL_ERROR_MSG_NOT_FOUND; | ||
194 | |||
195 | case MAILMBOX_ERROR_READONLY: | ||
196 | return MAIL_ERROR_READONLY; | ||
197 | |||
198 | default: | ||
199 | return MAIL_ERROR_INVAL; | ||
200 | } | ||
201 | } | ||
202 | |||
203 | |||
204 | |||
205 | |||
206 | static inline struct mbox_cached_session_state_data * | ||
207 | get_cached_data(mailsession * session) | ||
208 | { | ||
209 | return session->sess_data; | ||
210 | } | ||
211 | |||
212 | static inline mailsession * get_ancestor(mailsession * session) | ||
213 | { | ||
214 | return get_cached_data(session)->mbox_ancestor; | ||
215 | } | ||
216 | |||
217 | static inline struct mbox_session_state_data * | ||
218 | get_ancestor_data(mailsession * session) | ||
219 | { | ||
220 | return get_ancestor(session)->sess_data; | ||
221 | } | ||
222 | |||
223 | static inline struct mailmbox_folder * | ||
224 | get_mbox_session(mailsession * session) | ||
225 | { | ||
226 | return get_ancestor_data(session)->mbox_folder; | ||
227 | } | ||
228 | |||
229 | static int mboxdriver_cached_initialize(mailsession * session) | ||
230 | { | ||
231 | struct mbox_cached_session_state_data * cached_data; | ||
232 | struct mbox_session_state_data * mbox_data; | ||
233 | |||
234 | cached_data = malloc(sizeof(* cached_data)); | ||
235 | if (cached_data == NULL) | ||
236 | goto err; | ||
237 | |||
238 | cached_data->mbox_flags_store = mail_flags_store_new(); | ||
239 | if (cached_data->mbox_flags_store == NULL) | ||
240 | goto free; | ||
241 | |||
242 | cached_data->mbox_ancestor = mailsession_new(mbox_session_driver); | ||
243 | if (cached_data->mbox_ancestor == NULL) | ||
244 | goto free_store; | ||
245 | |||
246 | cached_data->mbox_quoted_mb = NULL; | ||
247 | /* | ||
248 | UID must be enabled to take advantage of the cache | ||
249 | */ | ||
250 | mbox_data = cached_data->mbox_ancestor->sess_data; | ||
251 | mbox_data->mbox_force_no_uid = FALSE; | ||
252 | |||
253 | session->sess_data = cached_data; | ||
254 | |||
255 | return MAIL_NO_ERROR; | ||
256 | |||
257 | free_store: | ||
258 | mail_flags_store_free(cached_data->mbox_flags_store); | ||
259 | free: | ||
260 | free(cached_data); | ||
261 | err: | ||
262 | return MAIL_ERROR_MEMORY; | ||
263 | } | ||
264 | |||
265 | static void free_state(struct mbox_cached_session_state_data * mbox_data) | ||
266 | { | ||
267 | if (mbox_data->mbox_quoted_mb) { | ||
268 | free(mbox_data->mbox_quoted_mb); | ||
269 | mbox_data->mbox_quoted_mb = NULL; | ||
270 | } | ||
271 | } | ||
272 | |||
273 | static int mbox_flags_store_process(char * flags_directory, char * quoted_mb, | ||
274 | struct mail_flags_store * flags_store) | ||
275 | { | ||
276 | char filename_flags[PATH_MAX]; | ||
277 | struct mail_cache_db * cache_db_flags; | ||
278 | MMAPString * mmapstr; | ||
279 | unsigned int i; | ||
280 | int r; | ||
281 | int res; | ||
282 | |||
283 | if (carray_count(flags_store->fls_tab) == 0) | ||
284 | return MAIL_NO_ERROR; | ||
285 | |||
286 | if (quoted_mb == NULL) | ||
287 | return MAIL_NO_ERROR; | ||
288 | |||
289 | snprintf(filename_flags, PATH_MAX, "%s%c%s%c%s", | ||
290 | flags_directory, MAIL_DIR_SEPARATOR, quoted_mb, | ||
291 | MAIL_DIR_SEPARATOR, FLAGS_NAME); | ||
292 | |||
293 | r = mail_cache_db_open_lock(filename_flags, &cache_db_flags); | ||
294 | if (r < 0) { | ||
295 | res = MAIL_ERROR_FILE; | ||
296 | goto err; | ||
297 | } | ||
298 | |||
299 | mmapstr = mmap_string_new(""); | ||
300 | if (mmapstr == NULL) { | ||
301 | res = MAIL_ERROR_MEMORY; | ||
302 | goto close_db_flags; | ||
303 | } | ||
304 | |||
305 | for(i = 0 ; i < carray_count(flags_store->fls_tab) ; i ++) { | ||
306 | mailmessage * msg; | ||
307 | |||
308 | msg = carray_get(flags_store->fls_tab, i); | ||
309 | |||
310 | r = mboxdriver_write_cached_flags(cache_db_flags, mmapstr, | ||
311 | msg->msg_uid, msg->msg_flags); | ||
312 | if (r != MAIL_NO_ERROR) { | ||
313 | /* ignore errors */ | ||
314 | } | ||
315 | } | ||
316 | |||
317 | mmap_string_free(mmapstr); | ||
318 | mail_cache_db_close_unlock(filename_flags, cache_db_flags); | ||
319 | |||
320 | mail_flags_store_clear(flags_store); | ||
321 | |||
322 | return MAIL_NO_ERROR; | ||
323 | |||
324 | close_db_flags: | ||
325 | mail_cache_db_close_unlock(filename_flags, cache_db_flags); | ||
326 | err: | ||
327 | return res; | ||
328 | } | ||
329 | |||
330 | static void mboxdriver_cached_uninitialize(mailsession * session) | ||
331 | { | ||
332 | struct mbox_cached_session_state_data * data; | ||
333 | |||
334 | data = get_cached_data(session); | ||
335 | |||
336 | mbox_flags_store_process(data->mbox_flags_directory, | ||
337 | data->mbox_quoted_mb, | ||
338 | data->mbox_flags_store); | ||
339 | |||
340 | mail_flags_store_free(data->mbox_flags_store); | ||
341 | |||
342 | free_state(data); | ||
343 | mailsession_free(data->mbox_ancestor); | ||
344 | free(data); | ||
345 | |||
346 | session->sess_data = NULL; | ||
347 | } | ||
348 | |||
349 | static int mboxdriver_cached_parameters(mailsession * session, | ||
350 | int id, void * value) | ||
351 | { | ||
352 | struct mbox_cached_session_state_data * data; | ||
353 | int r; | ||
354 | |||
355 | data = get_cached_data(session); | ||
356 | |||
357 | switch (id) { | ||
358 | case MBOXDRIVER_CACHED_SET_CACHE_DIRECTORY: | ||
359 | strncpy(data->mbox_cache_directory, value, PATH_MAX); | ||
360 | data->mbox_cache_directory[PATH_MAX - 1] = '\0'; | ||
361 | |||
362 | r = generic_cache_create_dir(data->mbox_cache_directory); | ||
363 | if (r != MAIL_NO_ERROR) | ||
364 | return r; | ||
365 | |||
366 | return MAIL_NO_ERROR; | ||
367 | |||
368 | case MBOXDRIVER_CACHED_SET_FLAGS_DIRECTORY: | ||
369 | strncpy(data->mbox_flags_directory, value, PATH_MAX); | ||
370 | data->mbox_flags_directory[PATH_MAX - 1] = '\0'; | ||
371 | |||
372 | r = generic_cache_create_dir(data->mbox_flags_directory); | ||
373 | if (r != MAIL_NO_ERROR) | ||
374 | return r; | ||
375 | |||
376 | return MAIL_NO_ERROR; | ||
377 | |||
378 | case MBOXDRIVER_SET_NO_UID: | ||
379 | return MAIL_ERROR_INVAL; | ||
380 | |||
381 | default: | ||
382 | return mailsession_parameters(data->mbox_ancestor, id, value); | ||
383 | } | ||
384 | } | ||
385 | |||
386 | |||
387 | static int get_cache_directory(mailsession * session, | ||
388 | char * path, char ** result) | ||
389 | { | ||
390 | char * quoted_mb; | ||
391 | char dirname[PATH_MAX]; | ||
392 | int res; | ||
393 | int r; | ||
394 | struct mbox_cached_session_state_data * cached_data; | ||
395 | |||
396 | cached_data = get_cached_data(session); | ||
397 | |||
398 | quoted_mb = maildriver_quote_mailbox(path); | ||
399 | if (quoted_mb == NULL) { | ||
400 | res = MAIL_ERROR_MEMORY; | ||
401 | goto err; | ||
402 | } | ||
403 | |||
404 | snprintf(dirname, PATH_MAX, "%s%c%s", | ||
405 | cached_data->mbox_cache_directory, MAIL_DIR_SEPARATOR, quoted_mb); | ||
406 | |||
407 | r = generic_cache_create_dir(dirname); | ||
408 | if (r != MAIL_NO_ERROR) { | ||
409 | res = r; | ||
410 | goto free; | ||
411 | } | ||
412 | |||
413 | snprintf(dirname, PATH_MAX, "%s%c%s", | ||
414 | cached_data->mbox_flags_directory, MAIL_DIR_SEPARATOR, quoted_mb); | ||
415 | |||
416 | r = generic_cache_create_dir(dirname); | ||
417 | if (r != MAIL_NO_ERROR) { | ||
418 | res = r; | ||
419 | goto free; | ||
420 | } | ||
421 | |||
422 | * result = quoted_mb; | ||
423 | |||
424 | return MAIL_NO_ERROR; | ||
425 | |||
426 | free: | ||
427 | free(quoted_mb); | ||
428 | err: | ||
429 | return res; | ||
430 | } | ||
431 | |||
432 | |||
433 | |||
434 | |||
435 | #define FILENAME_MAX_UID "max-uid" | ||
436 | |||
437 | /* write max uid current value */ | ||
438 | |||
439 | static int write_max_uid_value(mailsession * session) | ||
440 | { | ||
441 | int r; | ||
442 | char filename[PATH_MAX]; | ||
443 | FILE * f; | ||
444 | int res; | ||
445 | |||
446 | #if 0 | ||
447 | struct mbox_session_state_data * mbox_data; | ||
448 | #endif | ||
449 | struct mbox_cached_session_state_data * cached_data; | ||
450 | int fd; | ||
451 | |||
452 | MMAPString * mmapstr; | ||
453 | size_t cur_token; | ||
454 | struct mailmbox_folder * folder; | ||
455 | |||
456 | /* expunge the mailbox */ | ||
457 | |||
458 | #if 0 | ||
459 | mbox_data = get_ancestor(session)->data; | ||
460 | #endif | ||
461 | folder = get_mbox_session(session); | ||
462 | |||
463 | r = mailmbox_validate_write_lock(folder); | ||
464 | if (r != MAILMBOX_NO_ERROR) { | ||
465 | res = mbox_error_to_mail_error(r); | ||
466 | goto err; | ||
467 | } | ||
468 | |||
469 | r = mailmbox_expunge_no_lock(folder); | ||
470 | if (r != MAILMBOX_NO_ERROR) { | ||
471 | res = r; | ||
472 | goto unlock; | ||
473 | } | ||
474 | |||
475 | cached_data = get_cached_data(session); | ||
476 | |||
477 | snprintf(filename, PATH_MAX, "%s%c%s%c%s", | ||
478 | cached_data->mbox_flags_directory, MAIL_DIR_SEPARATOR, | ||
479 | cached_data->mbox_quoted_mb, MAIL_DIR_SEPARATOR, FILENAME_MAX_UID); | ||
480 | |||
481 | fd = creat(filename, S_IRUSR | S_IWUSR); | ||
482 | if (fd < 0) { | ||
483 | res = MAIL_ERROR_FILE; | ||
484 | goto err; | ||
485 | } | ||
486 | |||
487 | f = fdopen(fd, "w"); | ||
488 | if (f == NULL) { | ||
489 | close(fd); | ||
490 | res = MAIL_ERROR_FILE; | ||
491 | goto unlock; | ||
492 | } | ||
493 | |||
494 | mmapstr = mmap_string_new(""); | ||
495 | if (mmapstr == NULL) { | ||
496 | res = MAIL_ERROR_MEMORY; | ||
497 | goto close; | ||
498 | } | ||
499 | |||
500 | r = mail_serialize_clear(mmapstr, &cur_token); | ||
501 | if (r != MAIL_NO_ERROR) { | ||
502 | res = r; | ||
503 | goto free_mmapstr; | ||
504 | } | ||
505 | |||
506 | r = mailimf_cache_int_write(mmapstr, &cur_token, | ||
507 | folder->mb_written_uid); | ||
508 | if (r != MAIL_NO_ERROR) { | ||
509 | res = r; | ||
510 | goto free_mmapstr; | ||
511 | } | ||
512 | |||
513 | fwrite(mmapstr->str, 1, mmapstr->len, f); | ||
514 | |||
515 | mmap_string_free(mmapstr); | ||
516 | fclose(f); | ||
517 | mailmbox_write_unlock(folder); | ||
518 | |||
519 | return MAIL_NO_ERROR; | ||
520 | |||
521 | free_mmapstr: | ||
522 | mmap_string_free(mmapstr); | ||
523 | close: | ||
524 | fclose(f); | ||
525 | unlock: | ||
526 | mailmbox_read_unlock(folder); | ||
527 | err: | ||
528 | return res; | ||
529 | } | ||
530 | |||
531 | static int read_max_uid_value(mailsession * session, uint32_t * result) | ||
532 | { | ||
533 | int r; | ||
534 | char filename[PATH_MAX]; | ||
535 | FILE * f; | ||
536 | uint32_t written_uid; | ||
537 | int res; | ||
538 | |||
539 | struct mbox_cached_session_state_data * cached_data; | ||
540 | |||
541 | MMAPString * mmapstr; | ||
542 | size_t cur_token; | ||
543 | char buf[sizeof(uint32_t)]; | ||
544 | size_t read_size; | ||
545 | |||
546 | cached_data = get_cached_data(session); | ||
547 | |||
548 | snprintf(filename, PATH_MAX, "%s%c%s%c%s", | ||
549 | cached_data->mbox_flags_directory, MAIL_DIR_SEPARATOR, | ||
550 | cached_data->mbox_quoted_mb, MAIL_DIR_SEPARATOR, FILENAME_MAX_UID); | ||
551 | |||
552 | f = fopen(filename, "r"); | ||
553 | if (f == NULL) { | ||
554 | res = MAIL_ERROR_FILE; | ||
555 | goto err; | ||
556 | } | ||
557 | |||
558 | read_size = fread(buf, 1, sizeof(uint32_t), f); | ||
559 | |||
560 | mmapstr = mmap_string_new_len(buf, read_size); | ||
561 | if (mmapstr == NULL) { | ||
562 | res = MAIL_ERROR_MEMORY; | ||
563 | goto close; | ||
564 | } | ||
565 | |||
566 | cur_token = 0; | ||
567 | |||
568 | r = mailimf_cache_int_read(mmapstr, &cur_token, &written_uid); | ||
569 | if (r != MAIL_NO_ERROR) { | ||
570 | fclose(f); | ||
571 | res = r; | ||
572 | goto free_mmapstr; | ||
573 | } | ||
574 | |||
575 | mmap_string_free(mmapstr); | ||
576 | fclose(f); | ||
577 | |||
578 | * result = written_uid; | ||
579 | |||
580 | return MAIL_NO_ERROR; | ||
581 | |||
582 | free_mmapstr: | ||
583 | mmap_string_free(mmapstr); | ||
584 | close: | ||
585 | fclose(f); | ||
586 | err: | ||
587 | return res; | ||
588 | } | ||
589 | |||
590 | static int mboxdriver_cached_connect_path(mailsession * session, char * path) | ||
591 | { | ||
592 | int r; | ||
593 | int res; | ||
594 | char * quoted_mb; | ||
595 | struct mbox_cached_session_state_data * cached_data; | ||
596 | struct mbox_session_state_data * ancestor_data; | ||
597 | struct mailmbox_folder * folder; | ||
598 | uint32_t written_uid; | ||
599 | |||
600 | folder = get_mbox_session(session); | ||
601 | if (folder != NULL) { | ||
602 | res = MAIL_ERROR_BAD_STATE; | ||
603 | goto err; | ||
604 | } | ||
605 | |||
606 | r = get_cache_directory(session, path, "ed_mb); | ||
607 | if (r != MAIL_NO_ERROR) { | ||
608 | res = r; | ||
609 | goto err; | ||
610 | } | ||
611 | |||
612 | cached_data = get_cached_data(session); | ||
613 | free_state(cached_data); | ||
614 | |||
615 | cached_data->mbox_quoted_mb = quoted_mb; | ||
616 | |||
617 | written_uid = 0; | ||
618 | r = read_max_uid_value(session, &written_uid); | ||
619 | /* ignore errors */ | ||
620 | |||
621 | ancestor_data = get_ancestor_data(session); | ||
622 | |||
623 | r = mailmbox_init(path, | ||
624 | ancestor_data->mbox_force_read_only, | ||
625 | ancestor_data->mbox_force_no_uid, | ||
626 | written_uid, | ||
627 | &folder); | ||
628 | |||
629 | if (r != MAILMBOX_NO_ERROR) { | ||
630 | cached_data->mbox_quoted_mb = NULL; | ||
631 | |||
632 | res = mboxdriver_mbox_error_to_mail_error(r); | ||
633 | goto free; | ||
634 | } | ||
635 | |||
636 | ancestor_data->mbox_folder = folder; | ||
637 | |||
638 | return MAIL_NO_ERROR; | ||
639 | |||
640 | free: | ||
641 | free(quoted_mb); | ||
642 | err: | ||
643 | return res; | ||
644 | } | ||
645 | |||
646 | |||
647 | static int mboxdriver_cached_logout(mailsession * session) | ||
648 | { | ||
649 | struct mbox_cached_session_state_data * cached_data; | ||
650 | int r; | ||
651 | |||
652 | r = write_max_uid_value(session); | ||
653 | |||
654 | cached_data = get_cached_data(session); | ||
655 | |||
656 | mbox_flags_store_process(cached_data->mbox_flags_directory, | ||
657 | cached_data->mbox_quoted_mb, | ||
658 | cached_data->mbox_flags_store); | ||
659 | |||
660 | r = mailsession_logout(get_ancestor(session)); | ||
661 | if (r != MAIL_NO_ERROR) | ||
662 | return r; | ||
663 | |||
664 | free_state(cached_data); | ||
665 | |||
666 | return MAIL_NO_ERROR; | ||
667 | } | ||
668 | |||
669 | static int mboxdriver_cached_check_folder(mailsession * session) | ||
670 | { | ||
671 | struct mbox_cached_session_state_data * cached_data; | ||
672 | |||
673 | cached_data = get_cached_data(session); | ||
674 | |||
675 | mbox_flags_store_process(cached_data->mbox_flags_directory, | ||
676 | cached_data->mbox_quoted_mb, | ||
677 | cached_data->mbox_flags_store); | ||
678 | |||
679 | return MAIL_NO_ERROR; | ||
680 | } | ||
681 | |||
682 | static int mboxdriver_cached_expunge_folder(mailsession * session) | ||
683 | { | ||
684 | struct mailmbox_folder * folder; | ||
685 | int res; | ||
686 | char filename_flags[PATH_MAX]; | ||
687 | struct mail_cache_db * cache_db_flags; | ||
688 | MMAPString * mmapstr; | ||
689 | struct mbox_cached_session_state_data * data; | ||
690 | int r; | ||
691 | unsigned int i; | ||
692 | |||
693 | folder = get_mbox_session(session); | ||
694 | if (folder == NULL) { | ||
695 | res = MAIL_ERROR_BAD_STATE; | ||
696 | goto err; | ||
697 | } | ||
698 | |||
699 | data = get_cached_data(session); | ||
700 | if (data->mbox_quoted_mb == NULL) { | ||
701 | res = MAIL_ERROR_BAD_STATE; | ||
702 | goto err; | ||
703 | } | ||
704 | |||
705 | mbox_flags_store_process(data->mbox_flags_directory, | ||
706 | data->mbox_quoted_mb, | ||
707 | data->mbox_flags_store); | ||
708 | |||
709 | snprintf(filename_flags, PATH_MAX, "%s%c%s%c%s", | ||
710 | data->mbox_flags_directory, MAIL_DIR_SEPARATOR, data->mbox_quoted_mb, | ||
711 | MAIL_DIR_SEPARATOR, FLAGS_NAME); | ||
712 | |||
713 | r = mail_cache_db_open_lock(filename_flags, &cache_db_flags); | ||
714 | if (r < 0) { | ||
715 | res = MAIL_ERROR_FILE; | ||
716 | goto err; | ||
717 | } | ||
718 | |||
719 | mmapstr = mmap_string_new(""); | ||
720 | if (mmapstr == NULL) { | ||
721 | res = MAIL_ERROR_MEMORY; | ||
722 | goto close_db_flags; | ||
723 | } | ||
724 | |||
725 | for(i = 0 ; i < carray_count(folder->mb_tab) ; i ++) { | ||
726 | struct mailmbox_msg_info * msg_info; | ||
727 | struct mail_flags * flags; | ||
728 | |||
729 | msg_info = carray_get(folder->mb_tab, i); | ||
730 | if (msg_info == NULL) | ||
731 | continue; | ||
732 | |||
733 | if (msg_info->msg_deleted) | ||
734 | continue; | ||
735 | |||
736 | r = mboxdriver_get_cached_flags(cache_db_flags, mmapstr, | ||
737 | session, msg_info->msg_uid, &flags); | ||
738 | if (r != MAIL_NO_ERROR) | ||
739 | continue; | ||
740 | |||
741 | if (flags->fl_flags & MAIL_FLAG_DELETED) { | ||
742 | r = mailmbox_delete_msg(folder, msg_info->msg_uid); | ||
743 | } | ||
744 | |||
745 | mail_flags_free(flags); | ||
746 | } | ||
747 | |||
748 | mmap_string_free(mmapstr); | ||
749 | mail_cache_db_close_unlock(filename_flags, cache_db_flags); | ||
750 | |||
751 | r = mailmbox_expunge(folder); | ||
752 | |||
753 | return MAIL_NO_ERROR; | ||
754 | |||
755 | close_db_flags: | ||
756 | mail_cache_db_close_unlock(filename_flags, cache_db_flags); | ||
757 | err: | ||
758 | return res; | ||
759 | } | ||
760 | |||
761 | static int mboxdriver_cached_status_folder(mailsession * session, char * mb, | ||
762 | uint32_t * result_messages, uint32_t * result_recent, | ||
763 | uint32_t * result_unseen) | ||
764 | { | ||
765 | struct mailmbox_folder * folder; | ||
766 | int res; | ||
767 | char filename_flags[PATH_MAX]; | ||
768 | struct mail_cache_db * cache_db_flags; | ||
769 | MMAPString * mmapstr; | ||
770 | struct mbox_cached_session_state_data * data; | ||
771 | int r; | ||
772 | unsigned int i; | ||
773 | uint32_t recent; | ||
774 | uint32_t unseen; | ||
775 | uint32_t num; | ||
776 | |||
777 | num = 0; | ||
778 | recent = 0; | ||
779 | unseen = 0; | ||
780 | |||
781 | folder = get_mbox_session(session); | ||
782 | if (folder == NULL) { | ||
783 | res = MAIL_ERROR_BAD_STATE; | ||
784 | goto err; | ||
785 | } | ||
786 | |||
787 | data = get_cached_data(session); | ||
788 | if (data->mbox_quoted_mb == NULL) { | ||
789 | res = MAIL_ERROR_BAD_STATE; | ||
790 | goto err; | ||
791 | } | ||
792 | |||
793 | r = mailmbox_validate_read_lock(folder); | ||
794 | if (r != MAIL_NO_ERROR) { | ||
795 | res = MAIL_ERROR_BAD_STATE; | ||
796 | goto err; | ||
797 | } | ||
798 | |||
799 | mailmbox_read_unlock(folder); | ||
800 | |||
801 | mbox_flags_store_process(data->mbox_flags_directory, data->mbox_quoted_mb, | ||
802 | data->mbox_flags_store); | ||
803 | |||
804 | snprintf(filename_flags, PATH_MAX, "%s%c%s%c%s", | ||
805 | data->mbox_flags_directory, MAIL_DIR_SEPARATOR, data->mbox_quoted_mb, | ||
806 | MAIL_DIR_SEPARATOR, FLAGS_NAME); | ||
807 | |||
808 | r = mail_cache_db_open_lock(filename_flags, &cache_db_flags); | ||
809 | if (r < 0) { | ||
810 | res = MAIL_ERROR_FILE; | ||
811 | goto err; | ||
812 | } | ||
813 | |||
814 | mmapstr = mmap_string_new(""); | ||
815 | if (mmapstr == NULL) { | ||
816 | res = MAIL_ERROR_MEMORY; | ||
817 | goto close_db_flags; | ||
818 | } | ||
819 | |||
820 | for(i = 0 ; i < carray_count(folder->mb_tab) ; i ++) { | ||
821 | struct mailmbox_msg_info * msg_info; | ||
822 | struct mail_flags * flags; | ||
823 | |||
824 | msg_info = carray_get(folder->mb_tab, i); | ||
825 | if (msg_info == NULL) | ||
826 | continue; | ||
827 | |||
828 | if (msg_info->msg_deleted) | ||
829 | continue; | ||
830 | |||
831 | r = mboxdriver_get_cached_flags(cache_db_flags, mmapstr, | ||
832 | session, msg_info->msg_uid, &flags); | ||
833 | if (r != MAIL_NO_ERROR) { | ||
834 | recent ++; | ||
835 | unseen ++; | ||
836 | num ++; | ||
837 | continue; | ||
838 | } | ||
839 | |||
840 | if ((flags->fl_flags & MAIL_FLAG_NEW) != 0) { | ||
841 | recent ++; | ||
842 | } | ||
843 | if ((flags->fl_flags & MAIL_FLAG_SEEN) == 0) { | ||
844 | unseen ++; | ||
845 | } | ||
846 | |||
847 | num ++; | ||
848 | |||
849 | mail_flags_free(flags); | ||
850 | } | ||
851 | |||
852 | mmap_string_free(mmapstr); | ||
853 | mail_cache_db_close_unlock(filename_flags, cache_db_flags); | ||
854 | |||
855 | * result_messages = num; | ||
856 | * result_recent = recent; | ||
857 | * result_unseen = unseen; | ||
858 | |||
859 | return MAIL_NO_ERROR; | ||
860 | |||
861 | close_db_flags: | ||
862 | mail_cache_db_close_unlock(filename_flags, cache_db_flags); | ||
863 | err: | ||
864 | return res; | ||
865 | } | ||
866 | |||
867 | static int mboxdriver_cached_messages_number(mailsession * session, char * mb, | ||
868 | uint32_t * result) | ||
869 | { | ||
870 | return mailsession_messages_number(get_ancestor(session), mb, result); | ||
871 | } | ||
872 | |||
873 | |||
874 | static int mboxdriver_cached_recent_number(mailsession * session, char * mb, | ||
875 | uint32_t * result) | ||
876 | { | ||
877 | uint32_t messages; | ||
878 | uint32_t recent; | ||
879 | uint32_t unseen; | ||
880 | int r; | ||
881 | |||
882 | r = mboxdriver_cached_status_folder(session, mb, &messages, &recent, &unseen); | ||
883 | if (r != MAIL_NO_ERROR) | ||
884 | return r; | ||
885 | |||
886 | * result = recent; | ||
887 | |||
888 | return MAIL_NO_ERROR; | ||
889 | } | ||
890 | |||
891 | static int mboxdriver_cached_unseen_number(mailsession * session, char * mb, | ||
892 | uint32_t * result) | ||
893 | { | ||
894 | uint32_t messages; | ||
895 | uint32_t recent; | ||
896 | uint32_t unseen; | ||
897 | int r; | ||
898 | |||
899 | r = mboxdriver_cached_status_folder(session, mb, | ||
900 | &messages, &recent, &unseen); | ||
901 | if (r != MAIL_NO_ERROR) | ||
902 | return r; | ||
903 | |||
904 | * result = unseen; | ||
905 | |||
906 | return MAIL_NO_ERROR; | ||
907 | } | ||
908 | |||
909 | /* messages operations */ | ||
910 | |||
911 | static int mboxdriver_cached_append_message(mailsession * session, | ||
912 | char * message, size_t size) | ||
913 | { | ||
914 | return mboxdriver_cached_append_message_flags(session, | ||
915 | message, size, NULL); | ||
916 | } | ||
917 | |||
918 | static int mboxdriver_cached_append_message_flags(mailsession * session, | ||
919 | char * message, size_t size, struct mail_flags * flags) | ||
920 | { | ||
921 | int r; | ||
922 | struct mailmbox_folder * folder; | ||
923 | struct mbox_cached_session_state_data * data; | ||
924 | unsigned int uid; | ||
925 | struct mailmbox_msg_info * msg_info; | ||
926 | chashdatum key; | ||
927 | chashdatum value; | ||
928 | struct mail_cache_db * cache_db_flags; | ||
929 | char filename_flags[PATH_MAX]; | ||
930 | MMAPString * mmapstr; | ||
931 | char keyname[PATH_MAX]; | ||
932 | |||
933 | folder = get_mbox_session(session); | ||
934 | if (folder == NULL) | ||
935 | return MAIL_ERROR_APPEND; | ||
936 | |||
937 | r = mailmbox_append_message_uid(folder, message, size, &uid); | ||
938 | |||
939 | switch (r) { | ||
940 | case MAILMBOX_ERROR_FILE: | ||
941 | return MAIL_ERROR_DISKSPACE; | ||
942 | case MAILMBOX_NO_ERROR: | ||
943 | break; | ||
944 | default: | ||
945 | return mboxdriver_mbox_error_to_mail_error(r); | ||
946 | } | ||
947 | |||
948 | /* could store in flags store instead */ | ||
949 | |||
950 | if (flags == NULL) | ||
951 | goto exit; | ||
952 | |||
953 | key.data = &uid; | ||
954 | key.len = sizeof(uid); | ||
955 | r = chash_get(folder->mb_hash, &key, &value); | ||
956 | if (r < 0) | ||
957 | goto exit; | ||
958 | |||
959 | msg_info = value.data; | ||
960 | |||
961 | data = get_cached_data(session); | ||
962 | |||
963 | snprintf(filename_flags, PATH_MAX, "%s%c%s%c%s", | ||
964 | data->mbox_flags_directory, MAIL_DIR_SEPARATOR, data->mbox_quoted_mb, | ||
965 | MAIL_DIR_SEPARATOR, FLAGS_NAME); | ||
966 | |||
967 | r = mail_cache_db_open_lock(filename_flags, &cache_db_flags); | ||
968 | if (r < 0) | ||
969 | goto exit; | ||
970 | |||
971 | mmapstr = mmap_string_new(""); | ||
972 | if (mmapstr == NULL) | ||
973 | goto close_db_flags; | ||
974 | |||
975 | snprintf(keyname, PATH_MAX, "%u-%lu", uid, | ||
976 | (unsigned long) msg_info->msg_body_len); | ||
977 | |||
978 | r = mboxdriver_write_cached_flags(cache_db_flags, mmapstr, keyname, flags); | ||
979 | |||
980 | mmap_string_free(mmapstr); | ||
981 | mail_cache_db_close_unlock(filename_flags, cache_db_flags); | ||
982 | |||
983 | if (r != MAIL_NO_ERROR) | ||
984 | goto exit; | ||
985 | |||
986 | return MAIL_NO_ERROR; | ||
987 | |||
988 | close_db_flags: | ||
989 | mail_cache_db_close_unlock(filename_flags, cache_db_flags); | ||
990 | exit: | ||
991 | return MAIL_NO_ERROR; | ||
992 | } | ||
993 | |||
994 | static int | ||
995 | mboxdriver_cached_get_messages_list(mailsession * session, | ||
996 | struct mailmessage_list ** result) | ||
997 | { | ||
998 | struct mailmbox_folder * folder; | ||
999 | int res; | ||
1000 | |||
1001 | folder = get_mbox_session(session); | ||
1002 | if (folder == NULL) { | ||
1003 | res = MAIL_ERROR_BAD_STATE; | ||
1004 | goto err; | ||
1005 | } | ||
1006 | |||
1007 | return mbox_get_uid_messages_list(folder, | ||
1008 | session, mbox_cached_message_driver, result); | ||
1009 | |||
1010 | err: | ||
1011 | return res; | ||
1012 | } | ||
1013 | |||
1014 | static int | ||
1015 | get_cached_envelope(struct mail_cache_db * cache_db, MMAPString * mmapstr, | ||
1016 | mailsession * session, uint32_t num, | ||
1017 | struct mailimf_fields ** result) | ||
1018 | { | ||
1019 | int r; | ||
1020 | char keyname[PATH_MAX]; | ||
1021 | struct mailimf_fields * fields; | ||
1022 | int res; | ||
1023 | struct mailmbox_msg_info * info; | ||
1024 | struct mailmbox_folder * folder; | ||
1025 | chashdatum key; | ||
1026 | chashdatum data; | ||
1027 | |||
1028 | folder = get_mbox_session(session); | ||
1029 | if (folder == NULL) { | ||
1030 | res = MAIL_ERROR_BAD_STATE; | ||
1031 | goto err; | ||
1032 | } | ||
1033 | |||
1034 | key.data = # | ||
1035 | key.len = sizeof(num); | ||
1036 | |||
1037 | r = chash_get(folder->mb_hash, &key, &data); | ||
1038 | if (r < 0) { | ||
1039 | res = MAIL_ERROR_MSG_NOT_FOUND; | ||
1040 | goto err; | ||
1041 | } | ||
1042 | |||
1043 | info = data.data; | ||
1044 | |||
1045 | snprintf(keyname, PATH_MAX, "%u-%lu-envelope", num, | ||
1046 | (unsigned long) info->msg_body_len); | ||
1047 | |||
1048 | r = generic_cache_fields_read(cache_db, mmapstr, keyname, &fields); | ||
1049 | if (r != MAIL_NO_ERROR) { | ||
1050 | res = r; | ||
1051 | goto err; | ||
1052 | } | ||
1053 | |||
1054 | * result = fields; | ||
1055 | |||
1056 | return MAIL_NO_ERROR; | ||
1057 | |||
1058 | err: | ||
1059 | return res; | ||
1060 | } | ||
1061 | |||
1062 | static int | ||
1063 | write_cached_envelope(struct mail_cache_db * cache_db, MMAPString * mmapstr, | ||
1064 | mailsession * session, uint32_t num, | ||
1065 | struct mailimf_fields * fields) | ||
1066 | { | ||
1067 | int r; | ||
1068 | char keyname[PATH_MAX]; | ||
1069 | int res; | ||
1070 | struct mailmbox_msg_info * info; | ||
1071 | struct mailmbox_folder * folder; | ||
1072 | chashdatum key; | ||
1073 | chashdatum data; | ||
1074 | |||
1075 | folder = get_mbox_session(session); | ||
1076 | if (folder == NULL) { | ||
1077 | res = MAIL_ERROR_BAD_STATE; | ||
1078 | goto err; | ||
1079 | } | ||
1080 | |||
1081 | key.data = # | ||
1082 | key.len = sizeof(num); | ||
1083 | |||
1084 | r = chash_get(folder->mb_hash, &key, &data); | ||
1085 | if (r < 0) { | ||
1086 | res = MAIL_ERROR_MSG_NOT_FOUND; | ||
1087 | goto err; | ||
1088 | } | ||
1089 | |||
1090 | info = data.data; | ||
1091 | |||
1092 | snprintf(keyname, PATH_MAX, "%u-%lu-envelope", num, | ||
1093 | (unsigned long) info->msg_body_len); | ||
1094 | |||
1095 | r = generic_cache_fields_write(cache_db, mmapstr, keyname, fields); | ||
1096 | if (r != MAIL_NO_ERROR) { | ||
1097 | res = r; | ||
1098 | goto err; | ||
1099 | } | ||
1100 | |||
1101 | return MAIL_NO_ERROR; | ||
1102 | |||
1103 | err: | ||
1104 | return res; | ||
1105 | } | ||
1106 | |||
1107 | static int | ||
1108 | mboxdriver_cached_get_envelopes_list(mailsession * session, | ||
1109 | struct mailmessage_list * env_list) | ||
1110 | { | ||
1111 | int r; | ||
1112 | unsigned int i; | ||
1113 | struct mbox_cached_session_state_data * cached_data; | ||
1114 | char filename_env[PATH_MAX]; | ||
1115 | char filename_flags[PATH_MAX]; | ||
1116 | struct mail_cache_db * cache_db_env; | ||
1117 | struct mail_cache_db * cache_db_flags; | ||
1118 | MMAPString * mmapstr; | ||
1119 | int res; | ||
1120 | struct mailmbox_folder * folder; | ||
1121 | |||
1122 | folder = get_mbox_session(session); | ||
1123 | if (folder == NULL) { | ||
1124 | res = MAIL_ERROR_BAD_STATE; | ||
1125 | goto err; | ||
1126 | } | ||
1127 | |||
1128 | cached_data = get_cached_data(session); | ||
1129 | if (cached_data->mbox_quoted_mb == NULL) { | ||
1130 | res = MAIL_ERROR_BAD_STATE; | ||
1131 | goto err; | ||
1132 | } | ||
1133 | |||
1134 | mbox_flags_store_process(cached_data->mbox_flags_directory, | ||
1135 | cached_data->mbox_quoted_mb, | ||
1136 | cached_data->mbox_flags_store); | ||
1137 | |||
1138 | mmapstr = mmap_string_new(""); | ||
1139 | if (mmapstr == NULL) { | ||
1140 | res = MAIL_ERROR_MEMORY; | ||
1141 | goto err; | ||
1142 | } | ||
1143 | |||
1144 | snprintf(filename_env, PATH_MAX, "%s%c%s%c%s", | ||
1145 | cached_data->mbox_cache_directory, MAIL_DIR_SEPARATOR, | ||
1146 | cached_data->mbox_quoted_mb, | ||
1147 | MAIL_DIR_SEPARATOR, ENV_NAME); | ||
1148 | |||
1149 | r = mail_cache_db_open_lock(filename_env, &cache_db_env); | ||
1150 | if (r < 0) { | ||
1151 | res = MAIL_ERROR_MEMORY; | ||
1152 | goto free_mmapstr; | ||
1153 | } | ||
1154 | |||
1155 | snprintf(filename_flags, PATH_MAX, "%s%c%s%c%s", | ||
1156 | cached_data->mbox_flags_directory, MAIL_DIR_SEPARATOR, | ||
1157 | cached_data->mbox_quoted_mb, | ||
1158 | MAIL_DIR_SEPARATOR, FLAGS_NAME); | ||
1159 | |||
1160 | r = mail_cache_db_open_lock(filename_flags, &cache_db_flags); | ||
1161 | if (r < 0) { | ||
1162 | res = MAIL_ERROR_FILE; | ||
1163 | goto close_db_env; | ||
1164 | } | ||
1165 | |||
1166 | /* fill with cached */ | ||
1167 | |||
1168 | for(i = 0 ; i < carray_count(env_list->msg_tab) ; i ++) { | ||
1169 | mailmessage * msg; | ||
1170 | struct mailimf_fields * fields; | ||
1171 | struct mail_flags * flags; | ||
1172 | |||
1173 | msg = carray_get(env_list->msg_tab, i); | ||
1174 | |||
1175 | if (msg->msg_fields == NULL) { | ||
1176 | r = get_cached_envelope(cache_db_env, mmapstr, session, | ||
1177 | msg->msg_index, &fields); | ||
1178 | if (r == MAIL_NO_ERROR) { | ||
1179 | msg->msg_cached = TRUE; | ||
1180 | msg->msg_fields = fields; | ||
1181 | } | ||
1182 | } | ||
1183 | |||
1184 | if (msg->msg_flags == NULL) { | ||
1185 | r = mboxdriver_get_cached_flags(cache_db_flags, mmapstr, | ||
1186 | session, msg->msg_index, | ||
1187 | &flags); | ||
1188 | if (r == MAIL_NO_ERROR) { | ||
1189 | msg->msg_flags = flags; | ||
1190 | } | ||
1191 | } | ||
1192 | } | ||
1193 | |||
1194 | mail_cache_db_close_unlock(filename_flags, cache_db_flags); | ||
1195 | mail_cache_db_close_unlock(filename_env, cache_db_env); | ||
1196 | |||
1197 | r = mailsession_get_envelopes_list(get_ancestor(session), env_list); | ||
1198 | |||
1199 | if (r != MAIL_NO_ERROR) { | ||
1200 | res = r; | ||
1201 | goto free_mmapstr; | ||
1202 | } | ||
1203 | |||
1204 | /* add flags */ | ||
1205 | |||
1206 | for(i = 0 ; i < carray_count(env_list->msg_tab) ; i ++) { | ||
1207 | mailmessage * msg; | ||
1208 | |||
1209 | msg = carray_get(env_list->msg_tab, i); | ||
1210 | |||
1211 | if (msg->msg_flags == NULL) | ||
1212 | msg->msg_flags = mail_flags_new_empty(); | ||
1213 | } | ||
1214 | |||
1215 | r = mail_cache_db_open_lock(filename_env, &cache_db_env); | ||
1216 | if (r < 0) { | ||
1217 | res = MAIL_ERROR_MEMORY; | ||
1218 | goto free_mmapstr; | ||
1219 | } | ||
1220 | |||
1221 | r = mail_cache_db_open_lock(filename_flags, &cache_db_flags); | ||
1222 | if (r < 0) { | ||
1223 | res = MAIL_ERROR_FILE; | ||
1224 | goto close_db_env; | ||
1225 | } | ||
1226 | |||
1227 | /* must write cache */ | ||
1228 | |||
1229 | for(i = 0 ; i < carray_count(env_list->msg_tab) ; i ++) { | ||
1230 | mailmessage * msg; | ||
1231 | |||
1232 | msg = carray_get(env_list->msg_tab, i); | ||
1233 | |||
1234 | if (msg->msg_fields != NULL) { | ||
1235 | if (!msg->msg_cached) { | ||
1236 | /* msg->msg_index is the numerical UID of the message */ | ||
1237 | r = write_cached_envelope(cache_db_env, mmapstr, | ||
1238 | session, msg->msg_index, msg->msg_fields); | ||
1239 | } | ||
1240 | } | ||
1241 | |||
1242 | if (msg->msg_flags != NULL) { | ||
1243 | r = mboxdriver_write_cached_flags(cache_db_flags, mmapstr, | ||
1244 | msg->msg_uid, msg->msg_flags); | ||
1245 | } | ||
1246 | } | ||
1247 | |||
1248 | /* flush cache */ | ||
1249 | |||
1250 | maildriver_cache_clean_up(cache_db_env, cache_db_flags, env_list); | ||
1251 | |||
1252 | mail_cache_db_close_unlock(filename_flags, cache_db_flags); | ||
1253 | mail_cache_db_close_unlock(filename_env, cache_db_env); | ||
1254 | |||
1255 | mmap_string_free(mmapstr); | ||
1256 | |||
1257 | return MAIL_NO_ERROR; | ||
1258 | |||
1259 | close_db_env: | ||
1260 | mail_cache_db_close_unlock(filename_env, cache_db_env); | ||
1261 | free_mmapstr: | ||
1262 | mmap_string_free(mmapstr); | ||
1263 | err: | ||
1264 | return res; | ||
1265 | } | ||
1266 | |||
1267 | |||
1268 | static int | ||
1269 | mboxdriver_cached_remove_message(mailsession * session, uint32_t num) | ||
1270 | { | ||
1271 | return mailsession_remove_message(get_ancestor(session), num); | ||
1272 | } | ||
1273 | |||
1274 | static int mboxdriver_cached_get_message(mailsession * session, | ||
1275 | uint32_t num, mailmessage ** result) | ||
1276 | { | ||
1277 | mailmessage * msg_info; | ||
1278 | int r; | ||
1279 | |||
1280 | msg_info = mailmessage_new(); | ||
1281 | if (msg_info == NULL) | ||
1282 | return MAIL_ERROR_MEMORY; | ||
1283 | |||
1284 | r = mailmessage_init(msg_info, session, mbox_cached_message_driver, num, 0); | ||
1285 | if (r != MAIL_NO_ERROR) { | ||
1286 | mailmessage_free(msg_info); | ||
1287 | return r; | ||
1288 | } | ||
1289 | |||
1290 | * result = msg_info; | ||
1291 | |||
1292 | return MAIL_NO_ERROR; | ||
1293 | } | ||
1294 | |||
1295 | static int mboxdriver_cached_get_message_by_uid(mailsession * session, | ||
1296 | const char * uid, | ||
1297 | mailmessage ** result) | ||
1298 | { | ||
1299 | uint32_t num; | ||
1300 | char * p; | ||
1301 | chashdatum key; | ||
1302 | chashdatum data; | ||
1303 | struct mailmbox_msg_info * info; | ||
1304 | struct mailmbox_folder * folder; | ||
1305 | int r; | ||
1306 | |||
1307 | if (uid == NULL) | ||
1308 | return MAIL_ERROR_INVAL; | ||
1309 | |||
1310 | num = strtoul(uid, &p, 10); | ||
1311 | if (p == uid || * p != '-') | ||
1312 | return MAIL_ERROR_INVAL; | ||
1313 | |||
1314 | folder = get_mbox_session(session); | ||
1315 | if (folder == NULL) | ||
1316 | return MAIL_ERROR_BAD_STATE; | ||
1317 | |||
1318 | key.data = # | ||
1319 | key.len = sizeof(num); | ||
1320 | |||
1321 | r = chash_get(folder->mb_hash, &key, &data); | ||
1322 | if (r == 0) { | ||
1323 | char * body_len_p = p + 1; | ||
1324 | size_t body_len; | ||
1325 | |||
1326 | info = data.data; | ||
1327 | /* Check if the cached message has the same UID */ | ||
1328 | body_len = strtoul(body_len_p, &p, 10); | ||
1329 | if (p == body_len_p || * p != '\0') | ||
1330 | return MAIL_ERROR_INVAL; | ||
1331 | |||
1332 | if (body_len == info->msg_body_len) | ||
1333 | return mboxdriver_cached_get_message(session, num, result); | ||
1334 | } | ||
1335 | |||
1336 | return MAIL_ERROR_MSG_NOT_FOUND; | ||
1337 | } | ||