]> Pileus Git - ~andy/fetchmail/blob - uid.c
Guard some buffers.
[~andy/fetchmail] / uid.c
1 /*
2  * uid.c -- UIDL handling for POP3 servers without LAST
3  *
4  * For license terms, see the file COPYING in this directory.
5  */
6
7 #include "config.h"
8
9 #include <sys/stat.h>
10 #include <errno.h>
11 #include <stdio.h>
12 #include <limits.h>
13 #if defined(STDC_HEADERS)
14 #include <stdlib.h>
15 #include <string.h>
16 #endif
17 #if defined(HAVE_UNISTD_H)
18 #include <unistd.h>
19 #endif
20
21 #include "fetchmail.h"
22 #include "i18n.h"
23
24 /*
25  * Machinery for handling UID lists live here.  This is mainly to support
26  * RFC1725/RFC1939-conformant POP3 servers without a LAST command, but may also
27  * be useful for making the IMAP4 querying logic UID-oriented, if a future
28  * revision of IMAP forces me to.
29  *
30  * These functions are also used by the rest of the code to maintain
31  * string lists.
32  *
33  * Here's the theory:
34  *
35  * At start of a query, we have a (possibly empty) list of UIDs to be
36  * considered seen in `oldsaved'.  These are messages that were left in
37  * the mailbox and *not deleted* on previous queries (we don't need to
38  * remember the UIDs of deleted messages because ... well, they're gone!)
39  * This list is initially set up by initialize_saved_list() from the
40  * .fetchids file.
41  *
42  * Early in the query, during the execution of the protocol-specific
43  * getrange code, the driver expects that the host's `newsaved' member
44  * will be filled with a list of UIDs and message numbers representing
45  * the mailbox state.  If this list is empty, the server did
46  * not respond to the request for a UID listing.
47  *
48  * Each time a message is fetched, we can check its UID against the
49  * `oldsaved' list to see if it is old.
50  *
51  * Each time a message-id is seen, we mark it with MARK_SEEN.
52  *
53  * Each time a message is deleted, we mark its id UID_DELETED in the
54  * `newsaved' member.  When we want to assert that an expunge has been
55  * done on the server, we call expunge_uid() to register that all
56  * deleted messages are gone by marking them UID_EXPUNGED.
57  *
58  * At the end of the query, the `newsaved' member becomes the
59  * `oldsaved' list.  The old `oldsaved' list is freed.
60  *
61  * At the end of the fetchmail run, seen and non-EXPUNGED members of all
62  * current `oldsaved' lists are flushed out to the .fetchids file to
63  * be picked up by the next run.  If there are no un-expunged
64  * messages, the file is deleted.
65  *
66  * Note: some comparisons (those used for DNS address lists) are caseblind!
67  */
68
69 /* UIDs associated with un-queried hosts */
70 static struct idlist *scratchlist;
71
72 #ifdef POP3_ENABLE
73 void initialize_saved_lists(struct query *hostlist, const char *idfile)
74 /* read file of saved IDs and attach to each host */
75 {
76     struct stat statbuf;
77     FILE        *tmpfp;
78     struct query *ctl;
79
80     /* make sure lists are initially empty */
81     for (ctl = hostlist; ctl; ctl = ctl->next) {
82         ctl->skipped = (struct idlist *)NULL;
83         ctl->oldsaved = (struct idlist *)NULL;
84         ctl->newsaved = (struct idlist *)NULL;
85         ctl->oldsavedend = &ctl->oldsaved;
86     }
87
88     errno = 0;
89
90     /*
91      * Croak if the uidl directory does not exist.
92      * This probably means an NFS mount failed and we can't
93      * see a uidl file that ought to be there.
94      * Question: is this a portable check? It's not clear
95      * that all implementations of lstat() will return ENOTDIR
96      * rather than plain ENOENT in this case...
97      */
98     if (lstat(idfile, &statbuf) < 0) {
99         if (errno == ENOTDIR)
100         {
101             report(stderr, GT_("lstat: %s: %s\n"), idfile, strerror(errno));
102             exit(PS_IOERR);
103         }
104     }
105
106     /* let's get stored message UIDs from previous queries */
107     if ((tmpfp = fopen(idfile, "r")) != (FILE *)NULL)
108     {
109         char buf[POPBUFSIZE+1];
110         char *host = NULL;      /* pacify -Wall */
111         char *user;
112         char *id;
113         char *atsign;   /* temp pointer used in parsing user and host */
114         char *delimp1;
115         char saveddelim1;
116         char *delimp2;
117         char saveddelim2 = '\0';        /* pacify -Wall */
118
119         while (fgets(buf, POPBUFSIZE, tmpfp) != (char *)NULL)
120         {
121             /*
122              * At this point, we assume the bug has two fields -- a user@host 
123              * part, and an ID part. Either field may contain spurious @ signs.
124              * The previous version of this code presumed one could split at 
125              * the rightmost '@'.  This is not correct, as InterMail puts an 
126              * '@' in the UIDL.
127              */
128           
129             /* first, skip leading spaces */
130             user = buf + strspn(buf, " \t");
131
132             /*
133              * First, we split the buf into a userhost part and an id
134              * part ... but id doesn't necessarily start with a '<',
135              * espescially if the POP server returns an X-UIDL header
136              * instead of a Message-ID, as GMX's (www.gmx.net) POP3
137              * StreamProxy V1.0 does.
138              */
139             if ((id = strchr(user, ' ')) != NULL )
140             {
141
142               /*
143                * this is one other trick. The userhost part 
144                * may contain ' ' in the user part, at least in
145                * the lotus notes case.
146                * So we start looking for the '@' after which the
147                * host will follow with the ' ' seperator finaly id.
148                */
149                 delimp1 = strchr(user, '@');
150                 id = strchr(delimp1,' ');
151                 for (delimp1 = id; delimp1 >= user; delimp1--)
152                     if ((*delimp1 != ' ') && (*delimp1 != '\t'))
153                         break;
154
155                 /* 
156                  * It should be safe to assume that id starts after
157                  * the " " - after all, we're writing the " "
158                  * ourselves in write_saved_lists() :-)
159                  */
160                 id = id + strspn(id, " ");
161
162                 delimp1++; /* but what if there is only white space ?!? */
163                 saveddelim1 = *delimp1; /* save char after token */
164                 *delimp1 = '\0';                /* delimit token with \0 */
165                 if (id != NULL) 
166                 {
167                     /* now remove trailing white space chars from id */
168                     if ((delimp2 = strpbrk(id, " \t\n")) != NULL ) {
169                         saveddelim2 = *delimp2;
170                         *delimp2 = '\0';
171                     }
172                     atsign = strrchr(user, '@');
173                     if (atsign) {
174                         *atsign = '\0';
175                         host = atsign + 1;
176
177                     }
178                     for (ctl = hostlist; ctl; ctl = ctl->next) {
179                         if (strcasecmp(host, ctl->server.queryname) == 0
180                             && strcasecmp(user, ctl->remotename) == 0) {
181         
182                             save_str(&ctl->oldsaved, id, UID_SEEN);
183                             break;
184                         }
185                     }
186                     /* 
187                      * If it's not in a host we're querying,
188                      * save it anyway.  Otherwise we'd lose UIDL
189                      * information any time we queried an explicit
190                      * subset of hosts.
191                      */
192                     if (ctl == (struct query *)NULL) {
193                                 /* restore string */
194                         *delimp1 = saveddelim1;
195                         *atsign = '@';
196                         if (delimp2 != NULL) {
197                             *delimp2 = saveddelim2;
198                         }
199                         save_str(&scratchlist, buf, UID_SEEN);
200                     }
201                 }
202             }
203         }
204         fclose(tmpfp);  /* not checking should be safe, mode was "r" */
205     }
206
207     if (outlevel >= O_DEBUG)
208     {
209         struct idlist   *idp;
210         int uidlcount = 0;
211
212         for (ctl = hostlist; ctl; ctl = ctl->next)
213             if (ctl->server.uidl)
214             {
215                 report_build(stdout, GT_("Old UID list from %s:"), 
216                              ctl->server.pollname);
217                 for (idp = ctl->oldsaved; idp; idp = idp->next)
218                     report_build(stdout, " %s", idp->id);
219                 if (!idp)
220                     report_build(stdout, GT_(" <empty>"));
221                 report_complete(stdout, "\n");
222                 uidlcount++;
223             }
224
225         if (uidlcount)
226         {
227             report_build(stdout, GT_("Scratch list of UIDs:"));
228             for (idp = scratchlist; idp; idp = idp->next)
229                 report_build(stdout, " %s", idp->id);
230             if (!idp)
231                 report_build(stdout, GT_(" <empty>"));
232             report_complete(stdout, "\n");
233         }
234     }
235 }
236 #endif /* POP3_ENABLE */
237
238 /* return a pointer to the last element of the list to help the quick,
239  * constant-time addition to the list, NOTE: this function does not dup
240  * the string, the caller must do that. */
241 /*@shared@*/ struct idlist **save_str_quick(/*@shared@*/ struct idlist **idl,
242                                /*@only@*/ char *str, flag status)
243 /* save a number/UID pair on the given UID list */
244 {
245     struct idlist **end;
246
247     /* do it nonrecursively so the list is in the right order */
248     for (end = idl; *end; end = &(*end)->next)
249         continue;
250
251     *end = (struct idlist *)xmalloc(sizeof(struct idlist));
252     (*end)->val.status.mark = status;
253     (*end)->id = (unsigned char *)str;
254     (*end)->next = NULL;
255
256     return end;
257 }
258
259 /* return the end list element for direct modification */
260 struct idlist *save_str(struct idlist **idl, const char *str, flag st)
261 {
262     return *save_str_quick(idl, str ? xstrdup(str) : NULL,
263                            st);
264 }
265
266 void free_str_list(struct idlist **idl)
267 /* free the given UID list */
268 {
269     if (*idl == (struct idlist *)NULL)
270         return;
271
272     free_str_list(&(*idl)->next);
273     free ((*idl)->id);
274     free(*idl);
275     *idl = (struct idlist *)NULL;
276 }
277
278 void save_str_pair(struct idlist **idl, const char *str1, const char *str2)
279 /* save an ID pair on the given list */
280 {
281     struct idlist **end;
282
283     /* do it nonrecursively so the list is in the right order */
284     for (end = idl; *end; end = &(*end)->next)
285         continue;
286
287     *end = (struct idlist *)xmalloc(sizeof(struct idlist));
288     (*end)->id = str1 ? xstrdup(str1) : (char *)NULL;
289     if (str2)
290         (*end)->val.id2 = xstrdup(str2);
291     else
292         (*end)->val.id2 = (char *)NULL;
293     (*end)->next = (struct idlist *)NULL;
294 }
295
296 #ifdef __UNUSED__
297 void free_str_pair_list(struct idlist **idl)
298 /* free the given ID pair list */
299 {
300     if (*idl == (struct idlist *)NULL)
301         return;
302
303     free_idpair_list(&(*idl)->next);
304     free ((*idl)->id);
305     free ((*idl)->val.id2);
306     free(*idl);
307     *idl = (struct idlist *)NULL;
308 }
309 #endif
310
311 int str_in_list(struct idlist **idl, const char *str, const flag caseblind)
312 /* is a given ID in the given list? (comparison may be caseblind) */
313 {
314     struct idlist *walk;
315     if (caseblind) {
316         for( walk = *idl; walk; walk = walk->next )
317             if( strcasecmp( str, (char *)walk->id) == 0 )
318                 return 1;
319     } else {
320         for( walk = *idl; walk; walk = walk->next )
321             if( strcmp( str, (char *)walk->id) == 0 )
322                 return 1;
323     }
324     return 0;
325 }
326
327 int str_nr_in_list( struct idlist **idl, const char *str )
328   /* return the position of str in idl */
329 {
330     int nr;
331     struct idlist *walk;
332     if ( !str )
333         return -1;
334     for( walk = *idl, nr = 0; walk; nr ++, walk = walk->next )
335         if( strcmp( str, walk->id) == 0 )
336             return nr;
337     return -1;
338 }
339
340 int str_nr_last_in_list( struct idlist **idl, const char *str)
341 /* return the last position of str in idl */
342 {
343     int nr, ret = -1;
344     struct idlist *walk;
345     if ( !str )
346         return -1;
347     for( walk = *idl, nr = 0; walk; nr ++, walk = walk->next )
348         if( strcmp( str, walk->id) == 0 )
349             ret = nr;
350     return ret;
351 }
352
353 void str_set_mark( struct idlist **idl, const char *str, const flag val)
354 /* update the mark on an of an id to given value */
355 {
356     int nr;
357     struct idlist *walk;
358     if (!str)
359         return;
360     for(walk = *idl, nr = 0; walk; nr ++, walk = walk->next)
361         if (strcmp(str, walk->id) == 0)
362             walk->val.status.mark = val;
363 }
364
365 int count_list( struct idlist **idl)
366 /* count the number of elements in the list */
367 {
368   if( !*idl )
369     return 0;
370   return 1 + count_list( &(*idl)->next );
371 }
372
373 /*@null@*/ char *str_from_nr_list(struct idlist **idl, long number)
374 /* return the number'th string in idl */
375 {
376     if( !*idl  || number < 0)
377         return 0;
378     if( number == 0 )
379         return (*idl)->id;
380     return str_from_nr_list(&(*idl)->next, number-1);
381 }
382
383
384 char *str_find(struct idlist **idl, long number)
385 /* return the id of the given number in the given list. */
386 {
387     if (*idl == (struct idlist *) 0)
388         return((char *) 0);
389     else if (number == (*idl)->val.status.num)
390         return((*idl)->id);
391     else
392         return(str_find(&(*idl)->next, number));
393 }
394
395 char *idpair_find(struct idlist **idl, const char *id)
396 /* return the id of the given id in the given list (caseblind comparison) */
397 {
398     if (*idl == (struct idlist *) 0)
399         return((char *) 0);
400     else if (strcasecmp(id, (*idl)->id) == 0)
401         return((*idl)->val.id2 ? (*idl)->val.id2 : (*idl)->id);
402     else
403         return(idpair_find(&(*idl)->next, id));
404 }
405
406 int delete_str(struct idlist **idl, long num)
407 /* delete given message from given list */
408 {
409     struct idlist       *idp;
410
411     for (idp = *idl; idp; idp = idp->next)
412         if (idp->val.status.num == num)
413         {
414             idp->val.status.mark = UID_DELETED;
415             return(1);
416         }
417     return(0);
418 }
419
420 struct idlist *copy_str_list(struct idlist *idl)
421 /* copy the given UID list */
422 {
423     struct idlist *newnode ;
424
425     if (idl == (struct idlist *)NULL)
426         return(NULL);
427     else
428     {
429         newnode = (struct idlist *)xmalloc(sizeof(struct idlist));
430         memcpy(newnode, idl, sizeof(struct idlist));
431         newnode->next = copy_str_list(idl->next);
432         return(newnode);
433     }
434 }
435
436 void append_str_list(struct idlist **idl, struct idlist **nidl)
437 /* append nidl to idl (does not copy *) */
438 {
439     if ((*nidl) == (struct idlist *)NULL || *nidl == *idl)
440         return;
441     else if ((*idl) == (struct idlist *)NULL)
442         *idl = *nidl;
443     else if ((*idl)->next == (struct idlist *)NULL)
444         (*idl)->next = *nidl;
445     else if ((*idl)->next != *nidl)
446         append_str_list(&(*idl)->next, nidl);
447 }
448
449 #ifdef POP3_ENABLE
450 void expunge_uids(struct query *ctl)
451 /* assert that all UIDs marked deleted have actually been expunged */
452 {
453     struct idlist *idl;
454
455     for (idl = ctl->newsaved; idl; idl = idl->next)
456         if (idl->val.status.mark == UID_DELETED)
457             idl->val.status.mark = UID_EXPUNGED;
458 }
459
460 void uid_swap_lists(struct query *ctl) 
461 /* finish a query */
462 {
463     /* debugging code */
464     if (ctl->server.uidl && outlevel >= O_DEBUG)
465     {
466         struct idlist *idp;
467
468         report_build(stdout, GT_("New UID list from %s:"), ctl->server.pollname);
469         for (idp = ctl->newsaved; idp; idp = idp->next)
470             report_build(stdout, " %s = %d", idp->id, idp->val.status.mark);
471         if (!idp)
472             report_build(stdout, GT_(" <empty>"));
473         report_complete(stdout, "\n");
474     }
475
476     /*
477      * Don't swap UID lists unless we've actually seen UIDLs.
478      * This is necessary in order to keep UIDL information
479      * from being heedlessly deleted later on.
480      *
481      * Older versions of fetchmail did
482      *
483      *     free_str_list(&scratchlist);
484      *
485      * after swap.  This was wrong; we need to preserve the UIDL information
486      * from unqueried hosts.  Unfortunately, not doing this means that
487      * under some circumstances UIDLs can end up being stored forever --
488      * specifically, if a user description is removed from .fetchmailrc
489      * with UIDLs from that account in .fetchids, there is no way for
490      * them to ever get garbage-collected.
491      */
492     if (ctl->newsaved)
493     {
494         /* old state of mailbox may now be irrelevant */
495         if (outlevel >= O_DEBUG)
496             report(stdout, GT_("swapping UID lists\n"));
497         free_str_list(&ctl->oldsaved);
498         ctl->oldsaved = ctl->newsaved;
499         ctl->newsaved = (struct idlist *) NULL;
500     }
501     else if (outlevel >= O_DEBUG)
502         report(stdout, GT_("not swapping UID lists, no UIDs seen this query\n"));
503 }
504
505 void write_saved_lists(struct query *hostlist, const char *idfile)
506 /* perform end-of-run write of seen-messages list */
507 {
508     long        idcount;
509     FILE        *tmpfp;
510     struct query *ctl;
511     struct idlist *idp;
512
513     /* if all lists are empty, nuke the file */
514     idcount = 0;
515     for (ctl = hostlist; ctl; ctl = ctl->next) {
516         for (idp = ctl->oldsaved; idp; idp = idp->next)
517             if (idp->val.status.mark == UID_SEEN
518                     || idp->val.status.mark == UID_DELETED)
519                 idcount++;
520     }
521
522     /* either nuke the file or write updated last-seen IDs */
523     if (!idcount && !scratchlist)
524     {
525         if (outlevel >= O_DEBUG)
526             report(stdout, GT_("Deleting fetchids file.\n"));
527         unlink(idfile);
528     }
529     else
530     {
531         if (outlevel >= O_DEBUG)
532             report(stdout, GT_("Writing fetchids file.\n"));
533         /* FIXME: do not overwrite the old idfile */
534         if ((tmpfp = fopen(idfile, "w")) != (FILE *)NULL) {
535             for (ctl = hostlist; ctl; ctl = ctl->next) {
536                 for (idp = ctl->oldsaved; idp; idp = idp->next)
537                     if (idp->val.status.mark == UID_SEEN
538                                 || idp->val.status.mark == UID_DELETED)
539                         fprintf(tmpfp, "%s@%s %s\n", 
540                             ctl->remotename, ctl->server.queryname, idp->id);
541             }
542             for (idp = scratchlist; idp; idp = idp->next)
543                 fputs(idp->id, tmpfp);
544             fclose(tmpfp);
545         }
546     }
547 }
548 #endif /* POP3_ENABLE */
549
550 /* uid.c ends here */