Diffstat (limited to 'kmicromail/libetpan/generic/nntpdriver_cached_message.c') (more/less context) (ignore whitespace changes)
-rw-r--r-- | kmicromail/libetpan/generic/nntpdriver_cached_message.c | 365 |
1 files changed, 365 insertions, 0 deletions
diff --git a/kmicromail/libetpan/generic/nntpdriver_cached_message.c b/kmicromail/libetpan/generic/nntpdriver_cached_message.c new file mode 100644 index 0000000..a2bb6f7 --- a/dev/null +++ b/kmicromail/libetpan/generic/nntpdriver_cached_message.c | |||
@@ -0,0 +1,365 @@ | |||
1 | /* | ||
2 | * libEtPan! -- a mail stuff library | ||
3 | * | ||
4 | * Copyright (C) 2001, 2002 - 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 REGENTS 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 REGENTS 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 "nntpdriver_cached_message.h" | ||
37 | |||
38 | #include <string.h> | ||
39 | #include <stdlib.h> | ||
40 | |||
41 | #include "mail_cache_db.h" | ||
42 | |||
43 | #include "mailmessage.h" | ||
44 | #include "mailmessage_tools.h" | ||
45 | #include "nntpdriver.h" | ||
46 | #include "nntpdriver_tools.h" | ||
47 | #include "nntpdriver_cached.h" | ||
48 | #include "nntpdriver_message.h" | ||
49 | #include "generic_cache.h" | ||
50 | |||
51 | static int nntp_prefetch(mailmessage * msg_info); | ||
52 | |||
53 | static void nntp_prefetch_free(struct generic_message_t * msg); | ||
54 | |||
55 | static int nntp_initialize(mailmessage * msg_info); | ||
56 | |||
57 | static int nntp_fetch_header(mailmessage * msg_info, | ||
58 | char ** result, | ||
59 | size_t * result_len); | ||
60 | |||
61 | static int nntp_fetch_size(mailmessage * msg_info, | ||
62 | size_t * result); | ||
63 | |||
64 | static void nntp_uninitialize(mailmessage * msg_info); | ||
65 | |||
66 | static void nntp_flush(mailmessage * msg_info); | ||
67 | |||
68 | static void nntp_check(mailmessage * msg_info); | ||
69 | |||
70 | static int nntp_get_flags(mailmessage * msg_info, | ||
71 | struct mail_flags ** result); | ||
72 | |||
73 | static mailmessage_driver local_nntp_cached_message_driver = { | ||
74 | .msg_name = "nntp-cached", | ||
75 | |||
76 | .msg_initialize = nntp_initialize, | ||
77 | .msg_uninitialize = nntp_uninitialize, | ||
78 | |||
79 | .msg_flush = nntp_flush, | ||
80 | .msg_check = nntp_check, | ||
81 | |||
82 | .msg_fetch_result_free = mailmessage_generic_fetch_result_free, | ||
83 | |||
84 | .msg_fetch = mailmessage_generic_fetch, | ||
85 | .msg_fetch_header = nntp_fetch_header, | ||
86 | .msg_fetch_body = mailmessage_generic_fetch_body, | ||
87 | .msg_fetch_size = nntp_fetch_size, | ||
88 | .msg_get_bodystructure = mailmessage_generic_get_bodystructure, | ||
89 | .msg_fetch_section = mailmessage_generic_fetch_section, | ||
90 | .msg_fetch_section_header = mailmessage_generic_fetch_section_header, | ||
91 | .msg_fetch_section_mime = mailmessage_generic_fetch_section_mime, | ||
92 | .msg_fetch_section_body = mailmessage_generic_fetch_section_body, | ||
93 | .msg_fetch_envelope = mailmessage_generic_fetch_envelope, | ||
94 | |||
95 | .msg_get_flags = nntp_get_flags, | ||
96 | }; | ||
97 | |||
98 | mailmessage_driver * nntp_cached_message_driver = | ||
99 | &local_nntp_cached_message_driver; | ||
100 | |||
101 | static inline struct nntp_cached_session_state_data * | ||
102 | get_cached_session_data(mailmessage * msg) | ||
103 | { | ||
104 | return msg->msg_session->sess_data; | ||
105 | } | ||
106 | |||
107 | static inline mailsession * get_ancestor_session(mailmessage * msg) | ||
108 | { | ||
109 | return get_cached_session_data(msg)->nntp_ancestor; | ||
110 | } | ||
111 | |||
112 | static inline struct nntp_session_state_data * | ||
113 | get_ancestor_session_data(mailmessage * msg) | ||
114 | { | ||
115 | return get_ancestor_session(msg)->sess_data; | ||
116 | } | ||
117 | |||
118 | static inline newsnntp * | ||
119 | get_nntp_session(mailmessage * msg) | ||
120 | { | ||
121 | return get_ancestor_session_data(msg)->nntp_session; | ||
122 | } | ||
123 | |||
124 | static int nntp_prefetch(mailmessage * msg_info) | ||
125 | { | ||
126 | char * msg_content; | ||
127 | size_t msg_length; | ||
128 | struct generic_message_t * msg; | ||
129 | int r; | ||
130 | struct nntp_cached_session_state_data * cached_data; | ||
131 | struct nntp_session_state_data * ancestor_data; | ||
132 | char filename[PATH_MAX]; | ||
133 | |||
134 | /* we try the cached message */ | ||
135 | |||
136 | cached_data = get_cached_session_data(msg_info); | ||
137 | |||
138 | ancestor_data = get_ancestor_session_data(msg_info); | ||
139 | |||
140 | snprintf(filename, PATH_MAX, "%s/%s/%i", cached_data->nntp_cache_directory, | ||
141 | ancestor_data->nntp_group_name, msg_info->msg_index); | ||
142 | |||
143 | r = generic_cache_read(filename, &msg_content, &msg_length); | ||
144 | if (r == MAIL_NO_ERROR) { | ||
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 | /* we get the message through the network */ | ||
154 | |||
155 | r = nntpdriver_article(get_ancestor_session(msg_info), | ||
156 | msg_info->msg_index, &msg_content, | ||
157 | &msg_length); | ||
158 | |||
159 | if (r != MAIL_NO_ERROR) | ||
160 | return r; | ||
161 | |||
162 | /* we write the message cache */ | ||
163 | |||
164 | generic_cache_store(filename, msg_content, msg_length); | ||
165 | |||
166 | msg = msg_info->msg_data; | ||
167 | |||
168 | msg->msg_message = msg_content; | ||
169 | msg->msg_length = msg_length; | ||
170 | |||
171 | return MAIL_NO_ERROR; | ||
172 | } | ||
173 | |||
174 | static void nntp_prefetch_free(struct generic_message_t * msg) | ||
175 | { | ||
176 | if (msg->msg_message != NULL) { | ||
177 | mmap_string_unref(msg->msg_message); | ||
178 | msg->msg_message = NULL; | ||
179 | } | ||
180 | } | ||
181 | |||
182 | static int nntp_initialize(mailmessage * msg_info) | ||
183 | { | ||
184 | struct generic_message_t * msg; | ||
185 | int r; | ||
186 | char * uid; | ||
187 | char static_uid[20]; | ||
188 | |||
189 | snprintf(static_uid, 20, "%u", msg_info->msg_index); | ||
190 | uid = strdup(static_uid); | ||
191 | if (uid == NULL) | ||
192 | return MAIL_ERROR_MEMORY; | ||
193 | |||
194 | r = mailmessage_generic_initialize(msg_info); | ||
195 | if (r != MAIL_NO_ERROR) { | ||
196 | free(uid); | ||
197 | return r; | ||
198 | } | ||
199 | |||
200 | msg = msg_info->msg_data; | ||
201 | msg->msg_prefetch = nntp_prefetch; | ||
202 | msg->msg_prefetch_free = nntp_prefetch_free; | ||
203 | msg_info->msg_uid = uid; | ||
204 | |||
205 | return MAIL_NO_ERROR; | ||
206 | } | ||
207 | |||
208 | |||
209 | static void nntp_uninitialize(mailmessage * msg_info) | ||
210 | { | ||
211 | mailmessage_generic_uninitialize(msg_info); | ||
212 | } | ||
213 | |||
214 | #define FLAGS_NAME "flags.db" | ||
215 | |||
216 | static void nntp_flush(mailmessage * msg_info) | ||
217 | { | ||
218 | mailmessage_generic_flush(msg_info); | ||
219 | } | ||
220 | |||
221 | |||
222 | static void nntp_check(mailmessage * msg_info) | ||
223 | { | ||
224 | int r; | ||
225 | |||
226 | if (msg_info->msg_flags != NULL) { | ||
227 | r = mail_flags_store_set(get_cached_session_data(msg_info)->nntp_flags_store, | ||
228 | msg_info); | ||
229 | /* ignore errors */ | ||
230 | } | ||
231 | } | ||
232 | |||
233 | static int nntp_fetch_header(mailmessage * msg_info, | ||
234 | char ** result, | ||
235 | size_t * result_len) | ||
236 | { | ||
237 | struct generic_message_t * msg; | ||
238 | char * headers; | ||
239 | size_t headers_length; | ||
240 | struct nntp_cached_session_state_data * cached_data; | ||
241 | struct nntp_session_state_data * ancestor_data; | ||
242 | int r; | ||
243 | char filename[PATH_MAX]; | ||
244 | |||
245 | msg = msg_info->msg_data; | ||
246 | |||
247 | if (msg->msg_message != NULL) | ||
248 | return mailmessage_generic_fetch_header(msg_info, | ||
249 | result, result_len); | ||
250 | |||
251 | /* we try the cached message */ | ||
252 | |||
253 | cached_data = get_cached_session_data(msg_info); | ||
254 | |||
255 | ancestor_data = get_ancestor_session_data(msg_info); | ||
256 | |||
257 | snprintf(filename, PATH_MAX, "%s/%s/%i-header", | ||
258 | cached_data->nntp_cache_directory, | ||
259 | ancestor_data->nntp_group_name, msg_info->msg_index); | ||
260 | |||
261 | r = generic_cache_read(filename, &headers, &headers_length); | ||
262 | if (r == MAIL_NO_ERROR) { | ||
263 | * result = headers; | ||
264 | * result_len = headers_length; | ||
265 | |||
266 | return MAIL_NO_ERROR; | ||
267 | } | ||
268 | |||
269 | /* we get the message through the network */ | ||
270 | |||
271 | r = nntpdriver_head(get_ancestor_session(msg_info), msg_info->msg_index, | ||
272 | &headers, &headers_length); | ||
273 | if (r != MAIL_NO_ERROR) | ||
274 | return r; | ||
275 | |||
276 | /* we write the message cache */ | ||
277 | |||
278 | generic_cache_store(filename, headers, headers_length); | ||
279 | |||
280 | * result = headers; | ||
281 | * result_len = headers_length; | ||
282 | |||
283 | return MAIL_NO_ERROR; | ||
284 | } | ||
285 | |||
286 | static int nntp_fetch_size(mailmessage * msg_info, | ||
287 | size_t * result) | ||
288 | { | ||
289 | return nntpdriver_size(get_ancestor_session(msg_info), | ||
290 | msg_info->msg_index, result); | ||
291 | } | ||
292 | |||
293 | static int nntp_get_flags(mailmessage * msg_info, | ||
294 | struct mail_flags ** result) | ||
295 | { | ||
296 | int r; | ||
297 | struct mail_flags * flags; | ||
298 | struct mail_cache_db * cache_db_flags; | ||
299 | char filename_flags[PATH_MAX]; | ||
300 | int res; | ||
301 | MMAPString * mmapstr; | ||
302 | |||
303 | if (msg_info->msg_flags != NULL) { | ||
304 | * result = msg_info->msg_flags; | ||
305 | |||
306 | return MAIL_NO_ERROR; | ||
307 | } | ||
308 | |||
309 | flags = mail_flags_store_get(get_cached_session_data(msg_info)->nntp_flags_store, msg_info->msg_index); | ||
310 | |||
311 | if (flags == NULL) { | ||
312 | struct nntp_cached_session_state_data * cached_data; | ||
313 | struct nntp_session_state_data * ancestor_data; | ||
314 | |||
315 | cached_data = get_cached_session_data(msg_info); | ||
316 | |||
317 | ancestor_data = get_ancestor_session_data(msg_info); | ||
318 | if (ancestor_data->nntp_group_name == NULL) { | ||
319 | res = MAIL_ERROR_BAD_STATE; | ||
320 | goto err; | ||
321 | } | ||
322 | |||
323 | snprintf(filename_flags, PATH_MAX, "%s/%s/%s", | ||
324 | cached_data->nntp_flags_directory, | ||
325 | ancestor_data->nntp_group_name, FLAGS_NAME); | ||
326 | |||
327 | r = mail_cache_db_open_lock(filename_flags, &cache_db_flags); | ||
328 | if (r < 0) { | ||
329 | res = MAIL_ERROR_MEMORY; | ||
330 | goto err; | ||
331 | } | ||
332 | |||
333 | mmapstr = mmap_string_new(""); | ||
334 | if (mmapstr == NULL) { | ||
335 | res = MAIL_ERROR_MEMORY; | ||
336 | goto close_db_flags; | ||
337 | } | ||
338 | |||
339 | r = nntpdriver_get_cached_flags(cache_db_flags, mmapstr, | ||
340 | msg_info->msg_index, &flags); | ||
341 | if (r != MAIL_NO_ERROR) { | ||
342 | flags = mail_flags_new_empty(); | ||
343 | if (flags == NULL) { | ||
344 | res = MAIL_ERROR_MEMORY; | ||
345 | goto free_mmapstr; | ||
346 | } | ||
347 | } | ||
348 | |||
349 | mmap_string_free(mmapstr); | ||
350 | mail_cache_db_close_unlock(filename_flags, cache_db_flags); | ||
351 | } | ||
352 | |||
353 | msg_info->msg_flags = flags; | ||
354 | |||
355 | * result = flags; | ||
356 | |||
357 | return MAIL_NO_ERROR; | ||
358 | |||
359 | free_mmapstr: | ||
360 | mmap_string_free(mmapstr); | ||
361 | close_db_flags: | ||
362 | mail_cache_db_close_unlock(filename_flags, cache_db_flags); | ||
363 | err: | ||
364 | return res; | ||
365 | } | ||