Diffstat (limited to 'libetpan/src/driver/implementation/mh/mhdriver_cached_message.c') (more/less context) (ignore whitespace changes)
-rw-r--r-- | libetpan/src/driver/implementation/mh/mhdriver_cached_message.c | 338 |
1 files changed, 338 insertions, 0 deletions
diff --git a/libetpan/src/driver/implementation/mh/mhdriver_cached_message.c b/libetpan/src/driver/implementation/mh/mhdriver_cached_message.c new file mode 100644 index 0000000..a59beea --- a/dev/null +++ b/libetpan/src/driver/implementation/mh/mhdriver_cached_message.c | |||
@@ -0,0 +1,338 @@ | |||
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 "mhdriver_message.h" | ||
37 | |||
38 | #include "mailmessage_tools.h" | ||
39 | #include "mhdriver_tools.h" | ||
40 | #include "mhdriver_cached.h" | ||
41 | #include "mailmh.h" | ||
42 | #include "generic_cache.h" | ||
43 | |||
44 | #include "mail_cache_db.h" | ||
45 | |||
46 | #include <unistd.h> | ||
47 | #include <sys/mman.h> | ||
48 | #include <sys/types.h> | ||
49 | #include <sys/stat.h> | ||
50 | #include <fcntl.h> | ||
51 | #include <string.h> | ||
52 | #include <stdlib.h> | ||
53 | |||
54 | static int mh_prefetch(mailmessage * msg_info); | ||
55 | |||
56 | static void mh_prefetch_free(struct generic_message_t * msg); | ||
57 | |||
58 | static int mh_initialize(mailmessage * msg_info); | ||
59 | |||
60 | static int mh_fetch_size(mailmessage * msg_info, | ||
61 | size_t * result); | ||
62 | |||
63 | static int mh_get_flags(mailmessage * msg_info, | ||
64 | struct mail_flags ** result); | ||
65 | |||
66 | static void mh_uninitialize(mailmessage * msg_info); | ||
67 | |||
68 | static void mh_flush(mailmessage * msg_info); | ||
69 | |||
70 | static void mh_check(mailmessage * msg_info); | ||
71 | |||
72 | static int mh_fetch_header(mailmessage * msg_info, | ||
73 | char ** result, | ||
74 | size_t * result_len); | ||
75 | |||
76 | static mailmessage_driver local_mh_cached_message_driver = { | ||
77 | .msg_name = "mh-cached", | ||
78 | |||
79 | .msg_initialize = mh_initialize, | ||
80 | .msg_uninitialize = mh_uninitialize, | ||
81 | |||
82 | .msg_flush = mh_flush, | ||
83 | .msg_check = mh_check, | ||
84 | |||
85 | .msg_fetch_result_free = mailmessage_generic_fetch_result_free, | ||
86 | |||
87 | .msg_fetch = mailmessage_generic_fetch, | ||
88 | .msg_fetch_header = mh_fetch_header, | ||
89 | .msg_fetch_body = mailmessage_generic_fetch_body, | ||
90 | .msg_fetch_size = mh_fetch_size, | ||
91 | .msg_get_bodystructure = mailmessage_generic_get_bodystructure, | ||
92 | .msg_fetch_section = mailmessage_generic_fetch_section, | ||
93 | .msg_fetch_section_header = mailmessage_generic_fetch_section_header, | ||
94 | .msg_fetch_section_mime = mailmessage_generic_fetch_section_mime, | ||
95 | .msg_fetch_section_body = mailmessage_generic_fetch_section_body, | ||
96 | .msg_fetch_envelope = mailmessage_generic_fetch_envelope, | ||
97 | |||
98 | .msg_get_flags = mh_get_flags, | ||
99 | }; | ||
100 | |||
101 | mailmessage_driver * mh_cached_message_driver = | ||
102 | &local_mh_cached_message_driver; | ||
103 | |||
104 | static inline struct mh_cached_session_state_data * | ||
105 | get_cached_session_data(mailmessage * msg) | ||
106 | { | ||
107 | return msg->msg_session->sess_data; | ||
108 | } | ||
109 | |||
110 | static inline mailsession * get_ancestor_session(mailmessage * msg) | ||
111 | { | ||
112 | return get_cached_session_data(msg)->mh_ancestor; | ||
113 | } | ||
114 | |||
115 | static inline struct mh_session_state_data * | ||
116 | get_ancestor_session_data(mailmessage * msg) | ||
117 | { | ||
118 | return get_ancestor_session(msg)->sess_data; | ||
119 | } | ||
120 | |||
121 | static inline struct mailmh * | ||
122 | get_mh_session(mailmessage * msg) | ||
123 | { | ||
124 | return get_ancestor_session_data(msg)->mh_session; | ||
125 | } | ||
126 | |||
127 | static inline struct mailmh_folder * | ||
128 | get_mh_cur_folder(mailmessage * msg) | ||
129 | { | ||
130 | return get_ancestor_session_data(msg)->mh_cur_folder; | ||
131 | } | ||
132 | |||
133 | static int mh_prefetch(mailmessage * msg_info) | ||
134 | { | ||
135 | struct generic_message_t * msg; | ||
136 | int r; | ||
137 | char * msg_content; | ||
138 | size_t msg_length; | ||
139 | |||
140 | r = mhdriver_fetch_message(get_ancestor_session(msg_info), | ||
141 | msg_info->msg_index, &msg_content, &msg_length); | ||
142 | if (r != MAIL_NO_ERROR) | ||
143 | return r; | ||
144 | |||
145 | msg = msg_info->msg_data; | ||
146 | |||
147 | msg->msg_message = msg_content; | ||
148 | msg->msg_length = msg_length; | ||
149 | |||
150 | return MAIL_NO_ERROR; | ||
151 | } | ||
152 | |||
153 | static void mh_prefetch_free(struct generic_message_t * msg) | ||
154 | { | ||
155 | if (msg->msg_message != NULL) { | ||
156 | mmap_string_unref(msg->msg_message); | ||
157 | msg->msg_message = NULL; | ||
158 | } | ||
159 | } | ||
160 | |||
161 | static int mh_initialize(mailmessage * msg_info) | ||
162 | { | ||
163 | struct generic_message_t * msg; | ||
164 | int r; | ||
165 | char * uid; | ||
166 | char static_uid[PATH_MAX]; | ||
167 | struct mailmh_msg_info * mh_msg_info; | ||
168 | chashdatum key; | ||
169 | chashdatum data; | ||
170 | struct mailmh_folder * folder; | ||
171 | |||
172 | folder = get_mh_cur_folder(msg_info); | ||
173 | |||
174 | key.data = &msg_info->msg_index; | ||
175 | key.len = sizeof(msg_info->msg_index); | ||
176 | r = chash_get(folder->fl_msgs_hash, &key, &data); | ||
177 | if (r < 0) | ||
178 | return MAIL_ERROR_INVAL; | ||
179 | |||
180 | mh_msg_info = data.data; | ||
181 | |||
182 | snprintf(static_uid, PATH_MAX, "%u-%lu-%lu", msg_info->msg_index, | ||
183 | mh_msg_info->msg_mtime, (unsigned long) mh_msg_info->msg_size); | ||
184 | uid = strdup(static_uid); | ||
185 | if (uid == NULL) | ||
186 | return MAIL_ERROR_MEMORY; | ||
187 | |||
188 | r = mailmessage_generic_initialize(msg_info); | ||
189 | if (r != MAIL_NO_ERROR) { | ||
190 | free(uid); | ||
191 | return r; | ||
192 | } | ||
193 | |||
194 | msg = msg_info->msg_data; | ||
195 | msg->msg_prefetch = mh_prefetch; | ||
196 | msg->msg_prefetch_free = mh_prefetch_free; | ||
197 | msg_info->msg_uid = uid; | ||
198 | |||
199 | return MAIL_NO_ERROR; | ||
200 | } | ||
201 | |||
202 | static void mh_uninitialize(mailmessage * msg_info) | ||
203 | { | ||
204 | mailmessage_generic_uninitialize(msg_info); | ||
205 | } | ||
206 | |||
207 | |||
208 | #define FLAGS_NAME "flags.db" | ||
209 | |||
210 | static void mh_flush(mailmessage * msg_info) | ||
211 | { | ||
212 | mailmessage_generic_flush(msg_info); | ||
213 | } | ||
214 | |||
215 | static void mh_check(mailmessage * msg_info) | ||
216 | { | ||
217 | int r; | ||
218 | |||
219 | if (msg_info->msg_flags != NULL) { | ||
220 | r = mail_flags_store_set(get_cached_session_data(msg_info)->mh_flags_store, | ||
221 | msg_info); | ||
222 | /* ignore errors */ | ||
223 | } | ||
224 | } | ||
225 | |||
226 | static int mh_fetch_size(mailmessage * msg_info, | ||
227 | size_t * result) | ||
228 | { | ||
229 | int r; | ||
230 | size_t size; | ||
231 | |||
232 | r = mhdriver_fetch_size(get_ancestor_session(msg_info), | ||
233 | msg_info->msg_index, &size); | ||
234 | if (r != MAIL_NO_ERROR) | ||
235 | return r; | ||
236 | |||
237 | * result = size; | ||
238 | |||
239 | return MAIL_NO_ERROR; | ||
240 | } | ||
241 | |||
242 | static int mh_get_flags(mailmessage * msg_info, | ||
243 | struct mail_flags ** result) | ||
244 | { | ||
245 | int r; | ||
246 | struct mail_flags * flags; | ||
247 | struct mail_cache_db * cache_db_flags; | ||
248 | char filename_flags[PATH_MAX]; | ||
249 | int res; | ||
250 | struct mh_cached_session_state_data * cached_data; | ||
251 | MMAPString * mmapstr; | ||
252 | |||
253 | if (msg_info->msg_flags != NULL) { | ||
254 | * result = msg_info->msg_flags; | ||
255 | |||
256 | return MAIL_NO_ERROR; | ||
257 | } | ||
258 | |||
259 | cached_data = get_cached_session_data(msg_info); | ||
260 | |||
261 | flags = mail_flags_store_get(cached_data->mh_flags_store, | ||
262 | msg_info->msg_index); | ||
263 | |||
264 | if (flags == NULL) { | ||
265 | if (cached_data->mh_quoted_mb == NULL) { | ||
266 | res = MAIL_ERROR_BAD_STATE; | ||
267 | goto err; | ||
268 | } | ||
269 | |||
270 | snprintf(filename_flags, PATH_MAX, "%s/%s/%s", | ||
271 | cached_data->mh_flags_directory, | ||
272 | cached_data->mh_quoted_mb, FLAGS_NAME); | ||
273 | |||
274 | r = mail_cache_db_open_lock(filename_flags, &cache_db_flags); | ||
275 | if (r < 0) { | ||
276 | res = MAIL_ERROR_MEMORY; | ||
277 | goto err; | ||
278 | } | ||
279 | |||
280 | mmapstr = mmap_string_new(""); | ||
281 | if (mmapstr == NULL) { | ||
282 | res = MAIL_ERROR_MEMORY; | ||
283 | goto close_db_flags; | ||
284 | } | ||
285 | |||
286 | r = mhdriver_get_cached_flags(cache_db_flags, mmapstr, | ||
287 | msg_info->msg_session, msg_info->msg_index, &flags); | ||
288 | if (r != MAIL_NO_ERROR) { | ||
289 | flags = mail_flags_new_empty(); | ||
290 | if (flags == NULL) { | ||
291 | res = MAIL_ERROR_MEMORY; | ||
292 | goto free_mmapstr; | ||
293 | } | ||
294 | } | ||
295 | |||
296 | mmap_string_free(mmapstr); | ||
297 | mail_cache_db_close_unlock(filename_flags, cache_db_flags); | ||
298 | } | ||
299 | |||
300 | msg_info->msg_flags = flags; | ||
301 | |||
302 | * result = flags; | ||
303 | |||
304 | return MAIL_NO_ERROR; | ||
305 | |||
306 | free_mmapstr: | ||
307 | mmap_string_free(mmapstr); | ||
308 | close_db_flags: | ||
309 | mail_cache_db_close_unlock(filename_flags, cache_db_flags); | ||
310 | err: | ||
311 | return res; | ||
312 | } | ||
313 | |||
314 | static int mh_fetch_header(mailmessage * msg_info, | ||
315 | char ** result, | ||
316 | size_t * result_len) | ||
317 | { | ||
318 | struct generic_message_t * msg; | ||
319 | int r; | ||
320 | char * msg_content; | ||
321 | size_t msg_length; | ||
322 | |||
323 | msg = msg_info->msg_data; | ||
324 | if (msg->msg_message != NULL) { | ||
325 | return mailmessage_generic_fetch_header(msg_info, result, result_len); | ||
326 | } | ||
327 | else { | ||
328 | r = mhdriver_fetch_header(get_ancestor_session(msg_info), | ||
329 | msg_info->msg_index, &msg_content, &msg_length); | ||
330 | if (r != MAIL_NO_ERROR) | ||
331 | return r; | ||
332 | |||
333 | * result = msg_content; | ||
334 | * result_len = msg_length; | ||
335 | |||
336 | return MAIL_NO_ERROR; | ||
337 | } | ||
338 | } | ||