Pileus Git - ~andy/fetchmail/blob - driver.c

   1 /*
   2  * driver.c -- generic driver for mail fetch method protocols
   3  *
   4  * Copyright 1997 by Eric S. Raymond
   5  * For license terms, see the file COPYING in this directory.
   6  */
   7
   8 #include  "config.h"
   9 #include  <stdio.h>
  10 #include  <setjmp.h>
  11 #include  <errno.h>
  12 #include  <string.h>
  13 #ifdef HAVE_MEMORY_H
  14 #include  <memory.h>
  15 #endif /* HAVE_MEMORY_H */
  16 #if defined(STDC_HEADERS)
  17 #include  <stdlib.h>
  18 #include  <limits.h>
  19 #endif
  20 #if defined(HAVE_UNISTD_H)
  21 #include <unistd.h>
  22 #endif
  23 #if defined(HAVE_SYS_ITIMER_H)
  24 #include <sys/itimer.h>
  25 #endif
  26 #include  <signal.h>
  27 #ifdef HAVE_SYS_WAIT_H
  28 #include <sys/wait.h>
  29 #endif
  30
  31 #ifdef HAVE_SYS_SOCKET_H
  32 #include <sys/socket.h>
  33 #endif
  34 #ifdef HAVE_NET_SOCKET_H
  35 #include <net/socket.h>
  36 #endif
  37 #include <netdb.h>
  38 #ifdef HAVE_PKG_hesiod
  39 #include <hesiod.h>
  40 #endif
  41
  42 #include <langinfo.h>
  43
  44 #include "kerberos.h"
  45 #ifdef KERBEROS_V4
  46 #include <netinet/in.h>
  47 #endif /* KERBEROS_V4 */
  48
  49 #include "i18n.h"
  50 #include "socket.h"
  51
  52 #include "fetchmail.h"
  53 #include "getaddrinfo.h"
  54 #include "tunable.h"
  55
  56 /* throw types for runtime errors */
  57 #define THROW_TIMEOUT   1               /* server timed out */
  58
  59 /* magic values for the message length array */
  60 #define MSGLEN_UNKNOWN  0               /* length unknown (0 is impossible) */
  61 #define MSGLEN_INVALID  -1              /* length passed back is invalid */
  62 #define MSGLEN_TOOLARGE -2              /* message is too large */
  63 #define MSGLEN_OLD      -3              /* message is old */
  64
  65 int pass;               /* how many times have we re-polled? */
  66 int stage;              /* where are we? */
  67 int phase;              /* where are we, for error-logging purposes? */
  68 int batchcount;         /* count of messages sent in current batch */
  69 flag peek_capable;      /* can we peek for better error recovery? */
  70 int mailserver_socket_temp = -1;        /* socket to free if connect timeout */
  71
  72 struct addrinfo *ai0, *ai1;     /* clean these up after signal */
  73
  74 static volatile int timeoutcount = 0;   /* count consecutive timeouts */
  75 static volatile int idletimeout = 0;    /* timeout occured in idle stage? */
  76
  77 static sigjmp_buf       restart;
  78
  79 int is_idletimeout(void)
  80 /* last timeout occured in idle stage? */
  81 {
  82     return idletimeout;
  83 }
  84
  85 void resetidletimeout(void)
  86 {
  87     idletimeout = 0;
  88 }
  89
  90 void set_timeout(int timeleft)
  91 /* reset the nonresponse-timeout */
  92 {
  93 #if !defined(__EMX__) && !defined(__BEOS__)
  94     struct itimerval ntimeout;
  95
  96     if (timeleft == 0)
  97         timeoutcount = 0;
  98
  99     ntimeout.it_interval.tv_sec = ntimeout.it_interval.tv_usec = 0;
 100     ntimeout.it_value.tv_sec  = timeleft;
 101     ntimeout.it_value.tv_usec = 0;
 102     setitimer(ITIMER_REAL, &ntimeout, (struct itimerval *)NULL);
 103 #endif
 104 }
 105
 106 static RETSIGTYPE timeout_handler (int signal)
 107 /* handle SIGALRM signal indicating a server timeout */
 108 {
 109     (void)signal;
 110     if(stage != STAGE_IDLE) {
 111         timeoutcount++;
 112         /* XXX FIXME: this siglongjmp must die - it's not safe to be
 113          * called from a function handler and breaks, for instance,
 114          * getaddrinfo() */
 115         siglongjmp(restart, THROW_TIMEOUT);
 116     } else
 117         idletimeout = 1;
 118 }
 119
 120 #define CLEANUP_TIMEOUT 60 /* maximum timeout during cleanup */
 121
 122 static int cleanupSockClose (int fd)
 123 /* close sockets in maximum CLEANUP_TIMEOUT seconds during cleanup */
 124 {
 125     int scerror;
 126     SIGHANDLERTYPE alrmsave;
 127     alrmsave = set_signal_handler(SIGALRM, null_signal_handler);
 128     set_timeout(CLEANUP_TIMEOUT);
 129     scerror = SockClose(fd);
 130     set_timeout(0);
 131     set_signal_handler(SIGALRM, alrmsave);
 132     return (scerror);
 133 }
 134
 135 #ifdef KERBEROS_V4
 136 static int kerberos_auth(socket, canonical, principal)
 137 /* authenticate to the server host using Kerberos V4 */
 138 int socket;             /* socket to server host */
 139 char *canonical;        /* server name */
 140 char *principal;
 141 {
 142     KTEXT ticket;
 143     MSG_DAT msg_data;
 144     CREDENTIALS cred;
 145     Key_schedule schedule;
 146     int rem;
 147     char * prin_copy = (char *) NULL;
 148     char * prin = (char *) NULL;
 149     char * inst = (char *) NULL;
 150     char * realm = (char *) NULL;
 151
 152     if (principal != (char *)NULL && *principal)
 153     {
 154         char *cp;
 155         prin = prin_copy = xstrdup(principal);
 156         for (cp = prin_copy; *cp && *cp != '.'; ++cp)
 157             ;
 158         if (*cp)
 159         {
 160             *cp++ = '\0';
 161             inst = cp;
 162             while (*cp && *cp != '@')
 163                 ++cp;
 164             if (*cp)
 165             {
 166                 *cp++ = '\0';
 167                 realm = cp;
 168             }
 169         }
 170     }
 171
 172     ticket = xmalloc(sizeof (KTEXT_ST));
 173     rem = (krb_sendauth (0L, socket, ticket,
 174                          prin ? prin : "pop",
 175                          inst ? inst : canonical,
 176                          realm ? realm : ((char *) (krb_realmofhost (canonical))),
 177                          ((unsigned long) 0),
 178                          (&msg_data),
 179                          (&cred),
 180                          (schedule),
 181                          ((struct sockaddr_in *) 0),
 182                          ((struct sockaddr_in *) 0),
 183                          "KPOPV0.1"));
 184     free(ticket);
 185     if (prin_copy)
 186     {
 187         free(prin_copy);
 188     }
 189     if (rem != KSUCCESS)
 190     {
 191         report(stderr, GT_("kerberos error %s\n"), (krb_get_err_text (rem)));
 192         return (PS_AUTHFAIL);
 193     }
 194     return (0);
 195 }
 196 #endif /* KERBEROS_V4 */
 197
 198 #ifdef KERBEROS_V5
 199 static int kerberos5_auth(socket, canonical)
 200 /* authenticate to the server host using Kerberos V5 */
 201 int socket;             /* socket to server host */
 202 const char *canonical;  /* server name */
 203 {
 204     krb5_error_code retval;
 205     krb5_context context;
 206     krb5_ccache ccdef;
 207     krb5_principal client = NULL, server = NULL;
 208     krb5_error *err_ret = NULL;
 209
 210     krb5_auth_context auth_context = NULL;
 211
 212     krb5_init_context(&context);
 213     krb5_auth_con_init(context, &auth_context);
 214
 215     if ((retval = krb5_cc_default(context, &ccdef))) {
 216         report(stderr, "krb5_cc_default: %s\n", error_message(retval));
 217         return(PS_ERROR);
 218     }
 219
 220     if ((retval = krb5_cc_get_principal(context, ccdef, &client))) {
 221         report(stderr, "krb5_cc_get_principal: %s\n", error_message(retval));
 222         return(PS_ERROR);
 223     }
 224
 225     if ((retval = krb5_sname_to_principal(context, canonical, "pop",
 226            KRB5_NT_UNKNOWN,
 227            &server))) {
 228         report(stderr, "krb5_sname_to_principal: %s\n", error_message(retval));
 229         return(PS_ERROR);
 230     }
 231
 232     retval = krb5_sendauth(context, &auth_context, (krb5_pointer) &socket,
 233          "KPOPV1.0", client, server,
 234          AP_OPTS_MUTUAL_REQUIRED,
 235          NULL,  /* no data to checksum */
 236          0,   /* no creds, use ccache instead */
 237          ccdef,
 238          &err_ret, 0,
 239
 240          NULL); /* don't need reply */
 241
 242     krb5_free_principal(context, server);
 243     krb5_free_principal(context, client);
 244     krb5_auth_con_free(context, auth_context);
 245
 246     if (retval) {
 247 #ifdef HEIMDAL
 248       if (err_ret && err_ret->e_text) {
 249           report(stderr, GT_("krb5_sendauth: %s [server says '%*s'] \n"),
 250                  error_message(retval),
 251                  err_ret->e_text);
 252 #else
 253       if (err_ret && err_ret->text.length) {
 254           report(stderr, GT_("krb5_sendauth: %s [server says '%*s'] \n"),
 255                  error_message(retval),
 256                  err_ret->text.length,
 257                  err_ret->text.data);
 258 #endif
 259           krb5_free_error(context, err_ret);
 260       } else
 261           report(stderr, "krb5_sendauth: %s\n", error_message(retval));
 262       return(PS_ERROR);
 263     }
 264
 265     return 0;
 266 }
 267 #endif /* KERBEROS_V5 */
 268
 269 static void clean_skipped_list(struct idlist **skipped_list)
 270 /* struct "idlist" contains no "prev" ptr; we must remove unused items first */
 271 {
 272     struct idlist *current=NULL, *prev=NULL, *tmp=NULL, *head=NULL;
 273     prev = current = head = *skipped_list;
 274
 275     if (!head)
 276         return;
 277     do
 278     {
 279         /* if item has no reference, remove it */
 280         if (current && current->val.status.mark == 0)
 281         {
 282             if (current == head) /* remove first item (head) */
 283             {
 284                 head = current->next;
 285                 if (current->id) free(current->id);
 286                 free(current);
 287                 prev = current = head;
 288             }
 289             else /* remove middle/last item */
 290             {
 291                 tmp = current->next;
 292                 prev->next = tmp;
 293                 if (current->id) free(current->id);
 294                 free(current);
 295                 current = tmp;
 296             }
 297         }
 298         else /* skip this item */
 299         {
 300             prev = current;
 301             current = current->next;
 302         }
 303     } while(current);
 304
 305     *skipped_list = head;
 306 }
 307
 308 static void send_size_warnings(struct query *ctl)
 309 /* send warning mail with skipped msg; reset msg count when user notified */
 310 {
 311     int size, nbr;
 312     int msg_to_send = FALSE;
 313     struct idlist *head=NULL, *current=NULL;
 314     int max_warning_poll_count;
 315
 316     head = ctl->skipped;
 317     if (!head)
 318         return;
 319
 320     /* don't start a notification message unless we need to */
 321     for (current = head; current; current = current->next)
 322         if (current->val.status.num == 0 && current->val.status.mark)
 323             msg_to_send = TRUE;
 324     if (!msg_to_send)
 325         return;
 326
 327     /*
 328      * There's no good way to recover if we can't send notification mail,
 329      * but it's not a disaster, either, since the skipped mail will not
 330      * be deleted.
 331      */
 332     if (open_warning_by_mail(ctl))
 333         return;
 334     stuff_warning(iana_charset, ctl,
 335            GT_("Subject: Fetchmail oversized-messages warning"));
 336     stuff_warning(NULL, ctl, "%s", "");
 337     if (ctl->limitflush)
 338         stuff_warning(NULL, ctl,
 339                 GT_("The following oversized messages were deleted on server %s account %s:"),
 340                 ctl->server.pollname, ctl->remotename);
 341     else
 342         stuff_warning(NULL, ctl,
 343                 GT_("The following oversized messages remain on server %s account %s:"),
 344                 ctl->server.pollname, ctl->remotename);
 345
 346     stuff_warning(NULL, ctl, "%s", "");
 347
 348     if (run.poll_interval == 0)
 349         max_warning_poll_count = 0;
 350     else
 351         max_warning_poll_count = ctl->warnings/run.poll_interval;
 352
 353     /* parse list of skipped msg, adding items to the mail */
 354     for (current = head; current; current = current->next)
 355     {
 356         if (current->val.status.num == 0 && current->val.status.mark)
 357         {
 358             nbr = current->val.status.mark;
 359             size = atoi(current->id);
 360             if (ctl->limitflush)
 361                 stuff_warning(NULL, ctl,
 362                         GT_("  %d msg %d octets long deleted by fetchmail."),
 363                         nbr, size);
 364             else
 365                 stuff_warning(NULL, ctl,
 366                         GT_("  %d msg %d octets long skipped by fetchmail."),
 367                         nbr, size);
 368         }
 369         current->val.status.num++;
 370         current->val.status.mark = 0;
 371
 372         if (current->val.status.num >= max_warning_poll_count)
 373             current->val.status.num = 0;
 374     }
 375
 376     stuff_warning(NULL, ctl, "%s", "");
 377
 378     close_warning_by_mail(ctl, (struct msgblk *)NULL);
 379 }
 380
 381 static void mark_oversized(struct query *ctl, int size)
 382 /* mark a message oversized */
 383 {
 384     struct idlist *current=NULL, *tmp=NULL;
 385     char sizestr[32];
 386     int cnt;
 387
 388     /* convert size to string */
 389     snprintf(sizestr, sizeof(sizestr), "%d", size);
 390
 391     /* build a list of skipped messages
 392      * val.id = size of msg (string cnvt)
 393      * val.status.num = warning_poll_count
 394      * val.status.mask = nbr of msg this size
 395      */
 396
 397     current = ctl->skipped;
 398
 399     /* initialise warning_poll_count to the
 400      * current value so that all new msg will
 401      * be included in the next mail
 402      */
 403     cnt = current ? current->val.status.num : 0;
 404
 405     /* if entry exists, increment the count */
 406     if (current && (tmp = str_in_list(&current, sizestr, FALSE)))
 407     {
 408         tmp->val.status.mark++;
 409     }
 410     /* otherwise, create a new entry */
 411     /* initialise with current poll count */
 412     else
 413     {
 414         tmp = save_str(&ctl->skipped, sizestr, 1);
 415         tmp->val.status.num = cnt;
 416     }
 417 }
 418
 419 static int fetch_messages(int mailserver_socket, struct query *ctl,
 420                           int count, int **msgsizes, int maxfetch,
 421                           int *fetches, int *dispatches, int *deletions)
 422 /* fetch messages in lockstep mode */
 423 {
 424     flag force_retrieval;
 425     int num, firstnum = 1, lastnum = 0, err, len;
 426     int fetchsizelimit = ctl->fetchsizelimit;
 427     int msgsize;
 428     int initialfetches = *fetches;
 429
 430     if (ctl->server.base_protocol->getpartialsizes && NUM_NONZERO(fetchsizelimit))
 431     {
 432         /* for POP3, we can get the size of one mail only! Unfortunately, this
 433          * protocol specific test cannot be done elsewhere as the protocol
 434          * could be "auto". */
 435         switch (ctl->server.protocol)
 436         {
 437             case P_POP3: case P_APOP: case P_RPOP:
 438             fetchsizelimit = 1;
 439         }
 440
 441         /* Time to allocate memory to store the sizes */
 442         xfree(*msgsizes);
 443         *msgsizes = (int *)xmalloc(sizeof(int) * fetchsizelimit);
 444     }
 445
 446     /*
 447      * What forces this code is that in POP2 and
 448      * IMAP2bis you can't fetch a message without
 449      * having it marked `seen'.  In POP3 and IMAP4, on the
 450      * other hand, you can (peek_capable is set by
 451      * each driver module to convey this; it's not a
 452      * method constant because of the difference between
 453      * IMAP2bis and IMAP4, and because POP3 doesn't  peek
 454      * if fetchall is on).
 455      *
 456      * The result of being unable to peek is that if there's
 457      * any kind of transient error (DNS lookup failure, or
 458      * sendmail refusing delivery due to process-table limits)
 459      * the message will be marked "seen" on the server without
 460      * having been delivered.  This is not a big problem if
 461      * fetchmail is running in foreground, because the user
 462      * will see a "skipped" message when it next runs and get
 463      * clued in.
 464      *
 465      * But in daemon mode this leads to the message
 466      * being silently ignored forever.  This is not
 467      * acceptable.
 468      *
 469      * We compensate for this by checking the error
 470      * count from the previous pass and forcing all
 471      * messages to be considered new if it's nonzero.
 472      */
 473     force_retrieval = !peek_capable && (ctl->errcount > 0);
 474
 475     for (num = 1; num <= count; num++)
 476     {
 477         flag suppress_delete = FALSE;
 478         flag suppress_forward = FALSE;
 479         flag suppress_readbody = FALSE;
 480         flag retained = FALSE;
 481         int msgcode = MSGLEN_UNKNOWN;
 482
 483         /* check if the message is old
 484          * Note: the size of the message may not be known here */
 485         if (ctl->fetchall || force_retrieval)
 486             ;
 487         else if (ctl->server.base_protocol->is_old && (ctl->server.base_protocol->is_old)(mailserver_socket,ctl,num))
 488             msgcode = MSGLEN_OLD;
 489         if (msgcode == MSGLEN_OLD)
 490         {
 491                 /* To avoid flooding the syslog when using --keep,
 492                  * report "Skipped message" only when:
 493                  *  1) --verbose is on, or
 494                  *  2) fetchmail does not use syslog
 495                  */
 496             if (   (outlevel >= O_VERBOSE) ||
 497                    (outlevel > O_SILENT && !run.use_syslog)
 498                )
 499             {
 500                 report_build(stdout,
 501                              GT_("skipping message %s@%s:%d"),
 502                              ctl->remotename, ctl->server.truename, num);
 503             }
 504
 505             goto flagthemail;
 506         }
 507
 508         if (ctl->server.base_protocol->getpartialsizes && NUM_NONZERO(fetchsizelimit) &&
 509             lastnum < num)
 510         {
 511             /* Instead of getting the sizes of all mails at the start, we get
 512              * the sizes in blocks of fetchsizelimit. This leads to better
 513              * performance when there are too many mails (say, 10000) in
 514              * the mailbox and either we are not getting all the mails at
 515              * one go (--fetchlimit 100) or there is a frequent socket
 516              * error while just getting the sizes of all mails! */
 517
 518             int i;
 519             int oldstage = stage;
 520             firstnum = num;
 521             lastnum = num + fetchsizelimit - 1;
 522             if (lastnum > count)
 523                 lastnum = count;
 524             for (i = 0; i < fetchsizelimit; i++)
 525                 (*msgsizes)[i] = 0;
 526
 527             stage = STAGE_GETSIZES;
 528             err = (ctl->server.base_protocol->getpartialsizes)(mailserver_socket, num, lastnum, *msgsizes);
 529             if (err != 0) {
 530                 return err;
 531             }
 532             stage = oldstage;
 533         }
 534
 535         msgsize = *msgsizes ? (*msgsizes)[num-firstnum] : 0;
 536
 537         /* check if the message is oversized */
 538         if (NUM_NONZERO(ctl->limit) && (msgsize > ctl->limit))
 539             msgcode = MSGLEN_TOOLARGE;
 540 /*      else if (msgsize == 512)
 541             msgcode = MSGLEN_OLD;  (hmh) sample code to skip message */
 542
 543         if (msgcode < 0)
 544         {
 545             if (msgcode == MSGLEN_TOOLARGE)
 546             {
 547                 mark_oversized(ctl, msgsize);
 548                 if (!ctl->limitflush)
 549                     suppress_delete = TRUE;
 550             }
 551             if (outlevel > O_SILENT)
 552             {
 553                 /* old messages are already handled above */
 554                 report_build(stdout,
 555                              GT_("skipping message %s@%s:%d (%d octets)"),
 556                              ctl->remotename, ctl->server.truename, num,
 557                              msgsize);
 558                 switch (msgcode)
 559                 {
 560                 case MSGLEN_INVALID:
 561                     /*
 562                      * Invalid lengths are produced by Post Office/NT's
 563                      * annoying habit of randomly prepending bogus
 564                      * LIST items of length -1.  Patrick Audley
 565                      * <paudley@pobox.com> tells us: LIST shows a
 566                      * size of -1, RETR and TOP return "-ERR
 567                      * System error - couldn't open message", and
 568                      * DELE succeeds but doesn't actually delete
 569                      * the message.
 570                      */
 571                     report_build(stdout, GT_(" (length -1)"));
 572                     break;
 573                 case MSGLEN_TOOLARGE:
 574                     report_build(stdout, GT_(" (oversized)"));
 575                     break;
 576                 }
 577             }
 578         }
 579         else
 580         {
 581             flag wholesize = !ctl->server.base_protocol->fetch_body;
 582             flag separatefetchbody = (ctl->server.base_protocol->fetch_body) ? TRUE : FALSE;
 583
 584             /* request a message */
 585             err = (ctl->server.base_protocol->fetch_headers)(mailserver_socket,ctl,num, &len);
 586             if (err == PS_TRANSIENT)    /* server is probably Exchange */
 587             {
 588                 report(stdout,
 589                              GT_("couldn't fetch headers, message %s@%s:%d (%d octets)\n"),
 590                              ctl->remotename, ctl->server.truename, num,
 591                              msgsize);
 592                 continue;
 593             }
 594             else if (err != 0)
 595                 return(err);
 596
 597             /* -1 means we didn't see a size in the response */
 598             if (len == -1)
 599             {
 600                 len = msgsize;
 601                 wholesize = TRUE;
 602             }
 603
 604             if (outlevel > O_SILENT)
 605             {
 606                 report_build(stdout, GT_("reading message %s@%s:%d of %d"),
 607                              ctl->remotename, ctl->server.truename,
 608                              num, count);
 609
 610                 if (len > 0)
 611                     report_build(stdout, wholesize ? GT_(" (%d octets)")
 612                                  : GT_(" (%d header octets)"), len);
 613                 if (outlevel >= O_VERBOSE)
 614                     report_complete(stdout, "\n");
 615             }
 616
 617             /*
 618              * Read the message headers and ship them to the
 619              * output sink.
 620              */
 621             err = readheaders(mailserver_socket, len, msgsize,
 622                              ctl, num,
 623                              /* pass the suppress_readbody flag only if the underlying
 624                               * protocol does not fetch the body separately */
 625                              separatefetchbody ? 0 : &suppress_readbody);
 626             if (err == PS_RETAINED)
 627                 suppress_forward = suppress_delete = retained = TRUE;
 628             else if (err == PS_TRANSIENT)
 629                 suppress_delete = suppress_forward = TRUE;
 630             else if (err == PS_REFUSED)
 631                 suppress_forward = TRUE;
 632             else if (err == PS_TRUNCATED)
 633                 suppress_readbody = TRUE;
 634             else if (err)
 635                 return(err);
 636
 637             /* tell server we got it OK and resynchronize */
 638             if (separatefetchbody && ctl->server.base_protocol->trail)
 639             {
 640                 if (outlevel >= O_VERBOSE && !is_a_file(1) && !run.use_syslog)
 641                 {
 642                     fputc('\n', stdout);
 643                     fflush(stdout);
 644                 }
 645
 646                 if ((err = (ctl->server.base_protocol->trail)(mailserver_socket, ctl, tag)))
 647                     return(err);
 648             }
 649
 650             /* do not read the body which is not being forwarded only if
 651              * the underlying protocol allows the body to be fetched
 652              * separately */
 653             if (separatefetchbody && suppress_forward)
 654                 suppress_readbody = TRUE;
 655
 656             /*
 657              * If we're using IMAP4 or something else that
 658              * can fetch headers separately from bodies,
 659              * it's time to request the body now.  This
 660              * fetch may be skipped if we got an anti-spam
 661              * or other PS_REFUSED error response during
 662              * readheaders.
 663              */
 664             if (!suppress_readbody)
 665             {
 666                 if (separatefetchbody)
 667                 {
 668                     len = -1;
 669                     if ((err=(ctl->server.base_protocol->fetch_body)(mailserver_socket,ctl,num,&len)))
 670                         return(err);
 671                     /*
 672                      * Work around a bug in Novell's
 673                      * broken GroupWise IMAP server;
 674                      * its body FETCH response is missing
 675                      * the required length for the data
 676                      * string.  This violates RFC2060.
 677                      */
 678                     if (len == -1)
 679                         len = msgsize - msgblk.msglen;
 680                     if (outlevel > O_SILENT && !wholesize)
 681                         report_build(stdout,
 682                                         GT_(" (%d body octets)"), len);
 683                 }
 684
 685                 /* process the body now */
 686                 err = readbody(mailserver_socket,
 687                               ctl,
 688                               !suppress_forward,
 689                               len);
 690                 if (err == PS_TRANSIENT)
 691                     suppress_delete = suppress_forward = TRUE;
 692                 else if (err)
 693                     return(err);
 694
 695                 /* tell server we got it OK and resynchronize */
 696                 if (ctl->server.base_protocol->trail)
 697                 {
 698                     if (outlevel >= O_VERBOSE && !is_a_file(1) && !run.use_syslog)
 699                     {
 700                         fputc('\n', stdout);
 701                         fflush(stdout);
 702                     }
 703
 704                     err = (ctl->server.base_protocol->trail)(mailserver_socket, ctl, tag);
 705                     if (err != 0)
 706                         return(err);
 707                 }
 708             }
 709
 710             /* count # messages forwarded on this pass */
 711             if (!suppress_forward)
 712                 (*dispatches)++;
 713
 714             /*
 715              * Check to see if the numbers matched?
 716              *
 717              * Yes, some servers foo this up horribly.
 718              * All IMAP servers seem to get it right, and
 719              * so does Eudora QPOP at least in 2.xx
 720              * versions.
 721              *
 722              * Microsoft Exchange gets it completely
 723              * wrong, reporting compressed rather than
 724              * actual sizes (so the actual length of
 725              * message is longer than the reported size).
 726              * Another fine example of Microsoft brain death!
 727              *
 728              * Some older POP servers, like the old UCB
 729              * POP server and the pre-QPOP QUALCOMM
 730              * versions, report a longer size in the LIST
 731              * response than actually gets shipped up.
 732              * It's unclear what is going on here, as the
 733              * QUALCOMM server (at least) seems to be
 734              * reporting the on-disk size correctly.
 735              *
 736              * qmail-pop3d also goofs up message sizes and does not
 737              * count the line end characters properly.
 738              */
 739             if (msgblk.msglen != msgsize)
 740             {
 741                 if (outlevel >= O_DEBUG)
 742                     report(stdout,
 743                            GT_("message %s@%s:%d was not the expected length (%d actual != %d expected)\n"),
 744                            ctl->remotename, ctl->server.truename, num,
 745                            msgblk.msglen, msgsize);
 746             }
 747
 748             /* end-of-message processing starts here */
 749             if (!close_sink(ctl, &msgblk, !suppress_forward))
 750             {
 751                 ctl->errcount++;
 752                 suppress_delete = TRUE;
 753             }
 754             if (!retained)
 755                 (*fetches)++;
 756         }
 757
 758 flagthemail:
 759         /*
 760          * At this point in flow of control, either
 761          * we've bombed on a protocol error or had
 762          * delivery refused by the SMTP server
 763          * (unlikely -- I've never seen it) or we've
 764          * seen `accepted for delivery' and the
 765          * message is shipped.  It's safe to mark the
 766          * message seen and delete it on the server
 767          * now.
 768          */
 769
 770         /* maybe we delete this message now? */
 771         if (retained)
 772         {
 773             if (outlevel > O_SILENT)
 774                 report_complete(stdout, GT_(" retained\n"));
 775         }
 776         else if (ctl->server.base_protocol->delete_msg
 777                  && !suppress_delete
 778                  && ((msgcode >= 0 && !ctl->keep)
 779                      || (msgcode == MSGLEN_OLD && ctl->flush)
 780                      || (msgcode == MSGLEN_TOOLARGE && ctl->limitflush)))
 781         {
 782             (*deletions)++;
 783             if (outlevel > O_SILENT)
 784                 report_complete(stdout, GT_(" flushed\n"));
 785             err = (ctl->server.base_protocol->delete_msg)(mailserver_socket, ctl, num);
 786             if (err != 0)
 787                 return(err);
 788         }
 789         else
 790         {
 791             if (   (outlevel >= O_VERBOSE) ||
 792                         /* To avoid flooding the syslog when using --keep,
 793                          * report "Skipped message" only when:
 794                          *  1) --verbose is on, or
 795                          *  2) fetchmail does not use syslog, or
 796                          *  3) the message was skipped for some other
 797                          *     reason than just being old.
 798                          */
 799                    (outlevel > O_SILENT && (!run.use_syslog || msgcode != MSGLEN_OLD))
 800                )
 801             report_complete(stdout, GT_(" not flushed\n"));
 802
 803             /* maybe we mark this message as seen now? */
 804             if (ctl->server.base_protocol->mark_seen
 805                 && !suppress_delete
 806                 && (msgcode >= 0 && ctl->keep))
 807             {
 808                 err = (ctl->server.base_protocol->mark_seen)(mailserver_socket, ctl, num);
 809                 if (err != 0)
 810                     return(err);
 811             }
 812         }
 813
 814         /* perhaps this as many as we're ready to handle */
 815         if (maxfetch && maxfetch <= *fetches && num < count)
 816         {
 817             int remcount = count - (*fetches - initialfetches);
 818             report(stdout,
 819                    ngettext("fetchlimit %d reached; %d message left on server %s account %s\n",
 820                             "fetchlimit %d reached; %d messages left on server %s account %s\n", remcount),
 821                    maxfetch, remcount, ctl->server.truename, ctl->remotename);
 822             return(PS_MAXFETCH);
 823         }
 824     } /* for (num = 1; num <= count; num++) */
 825
 826     return(PS_SUCCESS);
 827 }
 828
 829 /* retrieve messages from server using given protocol method table */
 830 static int do_session(
 831         /* parsed options with merged-in defaults */
 832         struct query *ctl,
 833         /* protocol method table */
 834         const struct method *proto,
 835         /* maximum number of messages to fetch */
 836         const int maxfetch)
 837 {
 838     static int *msgsizes;
 839     volatile int err, mailserver_socket = -1;   /* pacifies -Wall */
 840     int tmperr;
 841     int deletions = 0, js;
 842     const char *msg;
 843     SIGHANDLERTYPE alrmsave;
 844
 845     ctl->server.base_protocol = proto;
 846
 847     msgsizes = NULL;
 848     pass = 0;
 849     err = 0;
 850     init_transact(proto);
 851
 852     /* set up the server-nonresponse timeout */
 853     alrmsave = set_signal_handler(SIGALRM, timeout_handler);
 854     mytimeout = ctl->server.timeout;
 855
 856     if ((js = sigsetjmp(restart,1)))
 857     {
 858         /* exception caught */
 859         sigset_t        allsigs;
 860
 861         sigfillset(&allsigs);
 862         sigprocmask(SIG_UNBLOCK, &allsigs, NULL);
 863
 864         if (ai0) {
 865             fm_freeaddrinfo(ai0); ai0 = NULL;
 866         }
 867
 868         if (ai1) {
 869             fm_freeaddrinfo(ai1); ai1 = NULL;
 870         }
 871
 872         if (js == THROW_TIMEOUT)
 873         {
 874             if (phase == OPEN_WAIT)
 875                 report(stdout,
 876                        GT_("timeout after %d seconds waiting to connect to server %s.\n"),
 877                        ctl->server.timeout, ctl->server.pollname);
 878             else if (phase == SERVER_WAIT)
 879                 report(stdout,
 880                        GT_("timeout after %d seconds waiting for server %s.\n"),
 881                        ctl->server.timeout, ctl->server.pollname);
 882             else if (phase == FORWARDING_WAIT)
 883                 report(stdout,
 884                        GT_("timeout after %d seconds waiting for %s.\n"),
 885                        ctl->server.timeout,
 886                        ctl->mda ? "MDA" : "SMTP");
 887             else if (phase == LISTENER_WAIT)
 888                 report(stdout,
 889                        GT_("timeout after %d seconds waiting for listener to respond.\n"), ctl->server.timeout);
 890             else
 891                 report(stdout,
 892                        GT_("timeout after %d seconds.\n"), ctl->server.timeout);
 893
 894             /*
 895              * If we've exceeded our threshold for consecutive timeouts,
 896              * try to notify the user, then mark the connection wedged.
 897              * Don't do this if the connection can idle, though; idle
 898              * timeouts just mean the frequency of mail is low.
 899              */
 900             if (timeoutcount > MAX_TIMEOUTS
 901                 && !open_warning_by_mail(ctl))
 902             {
 903                 stuff_warning(iana_charset, ctl,
 904                               GT_("Subject: fetchmail sees repeated timeouts"));
 905                 stuff_warning(NULL, ctl, "%s", "");
 906                 stuff_warning(NULL, ctl,
 907                               GT_("Fetchmail saw more than %d timeouts while attempting to get mail from %s@%s.\n"),
 908                               MAX_TIMEOUTS,
 909                               ctl->remotename, ctl->server.truename);
 910                 stuff_warning(NULL, ctl,
 911     GT_("This could mean that your mailserver is stuck, or that your SMTP\n" \
 912     "server is wedged, or that your mailbox file on the server has been\n" \
 913     "corrupted by a server error.  You can run `fetchmail -v -v' to\n" \
 914     "diagnose the problem.\n\n" \
 915     "Fetchmail won't poll this mailbox again until you restart it.\n"));
 916                 close_warning_by_mail(ctl, (struct msgblk *)NULL);
 917                 ctl->wedged = TRUE;
 918             }
 919         }
 920
 921         err = PS_SOCKET;
 922         goto cleanUp;
 923     }
 924     else
 925     {
 926         /* sigsetjmp returned zero -> normal operation */
 927         char buf[MSGBUFSIZE+1], *realhost;
 928         int count, newm, bytes;
 929         int fetches, dispatches, oldphase;
 930         struct idlist *idp;
 931
 932         /* execute pre-initialization command, if any */
 933         if (ctl->preconnect && (err = system(ctl->preconnect)))
 934         {
 935             report(stderr,
 936                    GT_("pre-connection command failed with status %d\n"), err);
 937             err = PS_SYNTAX;
 938             goto closeUp;
 939         }
 940
 941         /* open a socket to the mail server */
 942         oldphase = phase;
 943         phase = OPEN_WAIT;
 944         set_timeout(mytimeout);
 945
 946 #ifdef HAVE_PKG_hesiod
 947         /* If either the pollname or vianame are "hesiod" we want to
 948            lookup the user's hesiod pobox host */
 949         if (!strcasecmp(ctl->server.queryname, "hesiod")) {
 950             struct hes_postoffice *hes_p;
 951             hes_p = hes_getmailhost(ctl->remotename);
 952             if (hes_p != NULL && strcmp(hes_p->po_type, "POP") == 0) {
 953                  free(ctl->server.queryname);
 954                  ctl->server.queryname = xstrdup(hes_p->po_host);
 955                  if (ctl->server.via)
 956                      free(ctl->server.via);
 957                  ctl->server.via = xstrdup(hes_p->po_host);
 958             } else {
 959                  report(stderr,
 960                         GT_("couldn't find HESIOD pobox for %s\n"),
 961                         ctl->remotename);
 962             }
 963         }
 964 #endif /* HESIOD */
 965
 966         /*
 967          * Canonicalize the server truename for later use.  This also
 968          * functions as a probe for whether the mailserver is accessible.
 969          * We try it on each poll cycle until we get a result.  This way,
 970          * fetchmail won't fail if started up when the network is inaccessible.
 971          */
 972         if (ctl->server.dns && !ctl->server.trueaddr)
 973         {
 974             if (ctl->server.lead_server)
 975             {
 976                 char    *leadname = ctl->server.lead_server->truename;
 977
 978                 /* prevent core dump from ill-formed or duplicate entry */
 979                 if (!leadname)
 980                 {
 981                     report(stderr, GT_("Lead server has no name.\n"));
 982                     err = PS_DNS;
 983                     set_timeout(0);
 984                     phase = oldphase;
 985                     goto closeUp;
 986                 }
 987
 988                 xfree(ctl->server.truename);
 989                 ctl->server.truename = xstrdup(leadname);
 990             }
 991             else
 992             {
 993                 struct addrinfo hints, *res;
 994                 int error;
 995
 996                 memset(&hints, 0, sizeof(hints));
 997                 hints.ai_socktype = SOCK_STREAM;
 998                 hints.ai_family = AF_UNSPEC;
 999                 hints.ai_flags = AI_CANONNAME;
1000
1001                 error = fm_getaddrinfo(ctl->server.queryname, NULL, &hints, &res);
1002                 if (error)
1003                 {
1004                     report(stderr,
1005                            GT_("couldn't find canonical DNS name of %s (%s)\n"),
1006                            ctl->server.pollname, ctl->server.queryname);
1007                     err = PS_DNS;
1008                     set_timeout(0);
1009                     phase = oldphase;
1010                     goto closeUp;
1011                 }
1012                 else
1013                 {
1014                     xfree(ctl->server.truename);
1015                     /* Older FreeBSD versions return NULL in ai_canonname
1016                      * if they cannot canonicalize, rather than copying
1017                      * the queryname here, as IEEE Std 1003.1-2001
1018                      * requires. Work around NULL. */
1019                     if (res->ai_canonname != NULL) {
1020                         ctl->server.truename = xstrdup(res->ai_canonname);
1021                     } else {
1022                         ctl->server.truename = xstrdup(ctl->server.queryname);
1023                     }
1024                     ctl->server.trueaddr = (struct sockaddr *)xmalloc(res->ai_addrlen);
1025                     ctl->server.trueaddr_len = res->ai_addrlen;
1026                     memcpy(ctl->server.trueaddr, res->ai_addr, res->ai_addrlen);
1027                     fm_freeaddrinfo(res);
1028                 }
1029             }
1030         }
1031
1032         realhost = ctl->server.via ? ctl->server.via : ctl->server.pollname;
1033
1034         /* allow time for the port to be set up if we have a plugin */
1035         if (ctl->server.plugin)
1036             (void)sleep(1);
1037         if ((mailserver_socket = SockOpen(realhost,
1038                              ctl->server.service ? ctl->server.service : ( ctl->use_ssl ? ctl->server.base_protocol->sslservice : ctl->server.base_protocol->service ),
1039                              ctl->server.plugin, &ai0)) == -1)
1040         {
1041             char        errbuf[BUFSIZ];
1042             int err_no = errno;
1043             /*
1044              * Avoid generating a bogus error every poll cycle when we're
1045              * in daemon mode but the connection to the outside world
1046              * is down.
1047              */
1048             if (!((err_no == EHOSTUNREACH || err_no == ENETUNREACH)
1049                   && run.poll_interval))
1050             {
1051                 report_build(stderr, GT_("%s connection to %s failed"),
1052                              ctl->server.base_protocol->name, ctl->server.pollname);
1053                     strlcpy(errbuf, strerror(err_no), sizeof(errbuf));
1054                 report_complete(stderr, ": %s\n", errbuf);
1055
1056 #ifdef __UNUSED__
1057                 /*
1058                  * Don't use this.  It was an attempt to address Debian bug
1059                  * #47143 (Notify user by mail when pop server nonexistent).
1060                  * Trouble is, that doesn't work; you trip over the case
1061                  * where your SLIP or PPP link is down...
1062                  */
1063                 /* warn the system administrator */
1064                 if (open_warning_by_mail(ctl) == 0)
1065                 {
1066                     stuff_warning(iana_charset, ctl,
1067                          GT_("Subject: Fetchmail unreachable-server warning."));
1068                     stuff_warning(NULL, ctl, "");
1069                     stuff_warning(NULL, ctl, GT_("Fetchmail could not reach the mail server %s:"),
1070                                   ctl->server.pollname);
1071                     stuff_warning(NULL, ctl, errbuf, ctl->server.pollname);
1072                     close_warning_by_mail(ctl, (struct msgblk *)NULL);
1073                 }
1074 #endif
1075             }
1076             err = PS_SOCKET;
1077             set_timeout(0);
1078             phase = oldphase;
1079             goto closeUp;
1080         }
1081
1082 #ifdef SSL_ENABLE
1083         /* Save the socket opened. Useful if Fetchmail hangs on SSLOpen
1084          * because the socket can be closed.
1085          */
1086         mailserver_socket_temp = mailserver_socket;
1087         set_timeout(mytimeout);
1088
1089         /* perform initial SSL handshake on open connection */
1090         /* Note:  We pass the realhost name over for certificate
1091                 verification.  We may want to make this configurable */
1092         if (ctl->use_ssl && SSLOpen(mailserver_socket,ctl->sslcert,ctl->sslkey,ctl->sslproto,ctl->sslcertck,
1093             ctl->sslcertpath,ctl->sslfingerprint,realhost,ctl->server.pollname) == -1)
1094         {
1095             report(stderr, GT_("SSL connection failed.\n"));
1096             err = PS_SOCKET;
1097             goto cleanUp;
1098         }
1099
1100         /* Fetchmail didn't hang on SSLOpen,
1101          * then no need to set mailserver_socket_temp
1102          */
1103         mailserver_socket_temp = -1;
1104 #endif
1105
1106         /* A timeout is still defined before SSLOpen,
1107          * then Fetchmail hanging on SSLOpen is handled.
1108          */
1109         set_timeout(0);
1110         phase = oldphase;
1111 #ifdef KERBEROS_V4
1112         if (ctl->server.authenticate == A_KERBEROS_V4 && (strcasecmp(proto->name,"IMAP") != 0))
1113         {
1114             set_timeout(mytimeout);
1115             err = kerberos_auth(mailserver_socket, ctl->server.truename,
1116                                ctl->server.principal);
1117             set_timeout(0);
1118             if (err != 0)
1119                 goto cleanUp;
1120         }
1121 #endif /* KERBEROS_V4 */
1122
1123 #ifdef KERBEROS_V5
1124         if (ctl->server.authenticate == A_KERBEROS_V5)
1125         {
1126             set_timeout(mytimeout);
1127             err = kerberos5_auth(mailserver_socket, ctl->server.truename);
1128             set_timeout(0);
1129             if (err != 0)
1130                 goto cleanUp;
1131         }
1132 #endif /* KERBEROS_V5 */
1133
1134         /* accept greeting message from mail server */
1135         err = (ctl->server.base_protocol->parse_response)(mailserver_socket, buf);
1136         if (err != 0)
1137             goto cleanUp;
1138
1139         /* try to get authorized to fetch mail */
1140         stage = STAGE_GETAUTH;
1141         if (ctl->server.base_protocol->getauth)
1142         {
1143             err = (ctl->server.base_protocol->getauth)(mailserver_socket, ctl, buf);
1144
1145             if (err != 0)
1146             {
1147                 if (err == PS_LOCKBUSY)
1148                     report(stderr, GT_("Lock-busy error on %s@%s\n"),
1149                           ctl->remotename,
1150                           ctl->server.truename);
1151                 else if (err == PS_SERVBUSY)
1152                     report(stderr, GT_("Server busy error on %s@%s\n"),
1153                           ctl->remotename,
1154                           ctl->server.truename);
1155                 else if (err == PS_AUTHFAIL)
1156                 {
1157                     report(stderr, GT_("Authorization failure on %s@%s%s\n"),
1158                            ctl->remotename,
1159                            ctl->server.truename,
1160                            (ctl->wehaveauthed ? GT_(" (previously authorized)") : "")
1161                         );
1162
1163                     /*
1164                      * If we're running in background, try to mail the
1165                      * calling user a heads-up about the authentication
1166                      * failure once it looks like this isn't a fluke
1167                      * due to the server being temporarily inaccessible.
1168                      * When we get third succesive failure, we notify the user
1169                      * but only if we haven't already managed to get
1170                      * authorization.  After that, once we get authorization
1171                      * we let the user know service is restored.
1172                      */
1173                     if (run.poll_interval
1174                         && !ctl->wehavesentauthnote
1175                         && ((ctl->wehaveauthed && ++ctl->authfailcount >= 10)
1176                             || (!ctl->wehaveauthed && ++ctl->authfailcount >= 3))
1177                         && !open_warning_by_mail(ctl))
1178                     {
1179                         ctl->wehavesentauthnote = 1;
1180                         stuff_warning(iana_charset, ctl,
1181                                       GT_("Subject: fetchmail authentication failed on %s@%s"),
1182                             ctl->remotename, ctl->server.truename);
1183                         stuff_warning(NULL, ctl, "%s", "");
1184                         stuff_warning(NULL, ctl,
1185                                       GT_("Fetchmail could not get mail from %s@%s.\n"),
1186                                       ctl->remotename,
1187                                       ctl->server.truename);
1188                         if (ctl->wehaveauthed) {
1189                             stuff_warning(NULL, ctl, GT_("\
1190 The attempt to get authorization failed.\n\
1191 Since we have already succeeded in getting authorization for this\n\
1192 connection, this is probably another failure mode (such as busy server)\n\
1193 that fetchmail cannot distinguish because the server didn't send a useful\n\
1194 error message."));
1195                             stuff_warning(NULL, ctl, GT_("\
1196 \n\
1197 However, if you HAVE changed your account details since starting the\n\
1198 fetchmail daemon, you need to stop the daemon, change your configuration\n\
1199 of fetchmail, and then restart the daemon.\n\
1200 \n\
1201 The fetchmail daemon will continue running and attempt to connect\n\
1202 at each cycle.  No future notifications will be sent until service\n\
1203 is restored."));
1204                         } else {
1205                             stuff_warning(NULL, ctl, GT_("\
1206 The attempt to get authorization failed.\n\
1207 This probably means your password is invalid, but some servers have\n\
1208 other failure modes that fetchmail cannot distinguish from this\n\
1209 because they don't send useful error messages on login failure.\n\
1210 \n\
1211 The fetchmail daemon will continue running and attempt to connect\n\
1212 at each cycle.  No future notifications will be sent until service\n\
1213 is restored."));
1214                         }
1215                         close_warning_by_mail(ctl, (struct msgblk *)NULL);
1216                     }
1217                 }
1218                 else if (err == PS_REPOLL)
1219                 {
1220                   if (outlevel >= O_VERBOSE)
1221                     report(stderr, GT_("Repoll immediately on %s@%s\n"),
1222                            ctl->remotename,
1223                            ctl->server.truename);
1224                 }
1225                 else
1226                     report(stderr, GT_("Unknown login or authentication error on %s@%s\n"),
1227                            ctl->remotename,
1228                            ctl->server.truename);
1229
1230                 goto cleanUp;
1231             }
1232             else
1233             {
1234                 /*
1235                  * This connection has given us authorization at least once.
1236                  *
1237                  * There are dodgy server (clubinternet.fr for example) that
1238                  * give spurious authorization failures on patently good
1239                  * account/password details, then 5 minutes later let you in!
1240                  *
1241                  * This is meant to build in some tolerance of such nasty bits
1242                  * of work.
1243                  */
1244                 ctl->wehaveauthed = 1;
1245                 /*if (ctl->authfailcount >= 3)*/
1246                 if (ctl->wehavesentauthnote)
1247                 {
1248                     ctl->wehavesentauthnote = 0;
1249                     report(stderr,
1250                            GT_("Authorization OK on %s@%s\n"),
1251                            ctl->remotename,
1252                            ctl->server.truename);
1253                     if (!open_warning_by_mail(ctl))
1254                     {
1255                         stuff_warning(iana_charset, ctl,
1256                               GT_("Subject: fetchmail authentication OK on %s@%s"),
1257                                       ctl->remotename, ctl->server.truename);
1258                         stuff_warning(NULL, ctl, "%s", "");
1259                         stuff_warning(NULL, ctl,
1260                               GT_("Fetchmail was able to log into %s@%s.\n"),
1261                                       ctl->remotename,
1262                                       ctl->server.truename);
1263                         stuff_warning(NULL, ctl,
1264                                       GT_("Service has been restored.\n"));
1265                         close_warning_by_mail(ctl, (struct msgblk *)NULL);
1266
1267                     }
1268                 }
1269                 /*
1270                  * Reporting only after the first three
1271                  * consecutive failures, or ten consecutive
1272                  * failures after we have managed to get
1273                  * authorization.
1274                  */
1275                 ctl->authfailcount = 0;
1276             }
1277         }
1278
1279         ctl->errcount = fetches = 0;
1280
1281         /* now iterate over each folder selected */
1282         for (idp = ctl->mailboxes; idp; idp = idp->next)
1283         {
1284             ctl->folder = idp->id;
1285             pass = 0;
1286             do {
1287                 dispatches = 0;
1288                 ++pass;
1289
1290                 /* reset timeout, in case we did an IDLE */
1291                 mytimeout = ctl->server.timeout;
1292
1293                 if (outlevel >= O_DEBUG)
1294                 {
1295                     if (idp->id)
1296                         report(stdout, GT_("selecting or re-polling folder %s\n"), idp->id);
1297                     else
1298                         report(stdout, GT_("selecting or re-polling default folder\n"));
1299                 }
1300
1301                 /* compute # of messages and number of new messages waiting */
1302                 stage = STAGE_GETRANGE;
1303                 err = (ctl->server.base_protocol->getrange)(mailserver_socket, ctl, idp->id, &count, &newm, &bytes);
1304                 if (err != 0)
1305                     goto cleanUp;
1306
1307                 /* show user how many messages we downloaded */
1308                 if (idp->id)
1309                     (void) snprintf(buf, sizeof(buf),
1310                                    GT_("%s at %s (folder %s)"),
1311                                    ctl->remotename, ctl->server.pollname, idp->id);
1312                 else
1313                     (void) snprintf(buf, sizeof(buf), GT_("%s at %s"),
1314                                    ctl->remotename, ctl->server.pollname);
1315                 if (outlevel > O_SILENT)
1316                 {
1317                     if (count == -1)            /* only used for ETRN */
1318                         report(stdout, GT_("Polling %s\n"), ctl->server.truename);
1319                     else if (count != 0)
1320                     {
1321                         if (newm != -1 && (count - newm) > 0)
1322                             report_build(stdout, ngettext("%d message (%d %s) for %s", "%d messages (%d %s) for %s", (unsigned long)count),
1323                                   count,
1324                                   count - newm,
1325                                   ngettext("seen", "seen", (unsigned long)count-newm),
1326                                   buf);
1327                         else
1328                             report_build(stdout, ngettext("%d message for %s",
1329                                                           "%d messages for %s",
1330                                                           count),
1331                                   count, buf);
1332                         if (bytes == -1)
1333                             report_complete(stdout, ".\n");
1334                         else
1335                             report_complete(stdout, GT_(" (%d octets).\n"), bytes);
1336                     }
1337                     else
1338                     {
1339                         /* these are pointless in normal daemon mode */
1340                         if (pass == 1 && (run.poll_interval == 0 || outlevel >= O_VERBOSE))
1341                             report(stdout, GT_("No mail for %s\n"), buf);
1342                     }
1343                 }
1344
1345                 /* very important, this is where we leave the do loop */
1346                 if (count == 0)
1347                     break;
1348
1349                 if (check_only)
1350                 {
1351                     if (newm == -1 || ctl->fetchall)
1352                         newm = count;
1353                     fetches = newm;     /* set error status correctly */
1354                     /*
1355                      * There used to be a `goto noerror' here, but this
1356                      * prevented checking of multiple folders.  This
1357                      * comment is a reminder in case I introduced some
1358                      * subtle bug by removing it...
1359                      */
1360                 }
1361                 else if (count > 0)
1362                 {
1363                     int         i;
1364
1365                     /*
1366                      * Don't trust the message count passed by the server.
1367                      * Without this check, it might be possible to do a
1368                      * DNS-spoofing attack that would pass back a ridiculous
1369                      * count, and allocate a malloc area that would overlap
1370                      * a portion of the stack.
1371                      */
1372                     if ((unsigned)count > INT_MAX/sizeof(int))
1373                     {
1374                         report(stderr, GT_("bogus message count!"));
1375                         err = PS_PROTOCOL;
1376                         goto cleanUp;
1377                     }
1378
1379                     /*
1380                      * We need the size of each message before it's
1381                      * loaded in order to pass it to the ESMTP SIZE
1382                      * option.  If the protocol has a getsizes method,
1383                      * we presume this means it doesn't get reliable
1384                      * sizes from message fetch responses.
1385                      *
1386                      * If the protocol supports getting sizes of subset of
1387                      * messages, we skip this step now.
1388                      */
1389                     if (proto->getsizes &&
1390                         !(proto->getpartialsizes && NUM_NONZERO(ctl->fetchsizelimit)))
1391                     {
1392                         xfree(msgsizes);
1393                         msgsizes = (int *)xmalloc(sizeof(int) * count);
1394                         for (i = 0; i < count; i++)
1395                             msgsizes[i] = 0;
1396
1397                         stage = STAGE_GETSIZES;
1398                         err = (proto->getsizes)(mailserver_socket, count, msgsizes);
1399                         if (err != 0)
1400                             goto cleanUp;
1401
1402                         if (bytes == -1)
1403                         {
1404                             bytes = 0;
1405                             for (i = 0; i < count; i++)
1406                                 bytes += msgsizes[i];
1407                         }
1408                     }
1409
1410                     /* read, forward, and delete messages */
1411                     stage = STAGE_FETCH;
1412
1413                     /* fetch in lockstep mode */
1414                     err = fetch_messages(mailserver_socket, ctl,
1415                                          count, &msgsizes,
1416                                          maxfetch,
1417                                          &fetches, &dispatches, &deletions);
1418                     if (err != PS_SUCCESS && err != PS_MAXFETCH)
1419                         goto cleanUp;
1420
1421                     if (!check_only && ctl->skipped
1422                         && run.poll_interval > 0 && !nodetach)
1423                     {
1424                         clean_skipped_list(&ctl->skipped);
1425                         send_size_warnings(ctl);
1426                     }
1427                 }
1428
1429                 /* end-of-mailbox processing before we repoll or switch to another one */
1430                 if (ctl->server.base_protocol->end_mailbox_poll)
1431                 {
1432                     err = (ctl->server.base_protocol->end_mailbox_poll)(mailserver_socket, ctl);
1433                     if (err)
1434                         goto cleanUp;
1435                 }
1436                 /* Return now if we have reached the fetchlimit */
1437                 if (maxfetch && maxfetch <= fetches)
1438                     goto no_error;
1439             } while
1440                   /*
1441                    * Only repoll if we either had some actual forwards
1442                    * or are idling for new mails and had no errors.
1443                    * Otherwise it is far too easy to get into infinite loops.
1444                    */
1445                   (ctl->server.base_protocol->retry && (dispatches || ctl->idle) && !ctl->errcount);
1446         }
1447
1448         /* XXX: From this point onwards, preserve err unless a new error has occurred */
1449
1450     no_error:
1451         /* PS_SUCCESS, PS_MAXFETCH: ordinary termination with no errors -- officially log out */
1452         stage = STAGE_LOGOUT;
1453         tmperr = (ctl->server.base_protocol->logout_cmd)(mailserver_socket, ctl);
1454         if (tmperr != PS_SUCCESS)
1455             err = tmperr;
1456         /*
1457          * Hmmmm...arguably this would be incorrect if we had fetches but
1458          * no dispatches (due to oversized messages, etc.)
1459          */
1460         else if (err == PS_SUCCESS && fetches == 0)
1461             err = PS_NOMAIL;
1462         /*
1463          * Close all SMTP delivery sockets.  For optimum performance
1464          * we'd like to hold them open til end of run, but (1) this
1465          * loses if our poll interval is longer than the MTA's
1466          * inactivity timeout, and (2) some MTAs (like smail) don't
1467          * deliver after each message, but rather queue up mail and
1468          * wait to actually deliver it until the input socket is
1469          * closed.
1470          *
1471          * don't send QUIT for ODMR case because we're acting as a
1472          * proxy between the SMTP server and client.
1473          */
1474         smtp_close(ctl, ctl->server.protocol != P_ODMR);
1475         cleanupSockClose(mailserver_socket);
1476         goto closeUp;
1477
1478     cleanUp:
1479         /* we only get here on error */
1480         if (err != 0 && err != PS_SOCKET && err != PS_REPOLL)
1481         {
1482             stage = STAGE_LOGOUT;
1483             (ctl->server.base_protocol->logout_cmd)(mailserver_socket, ctl);
1484         }
1485
1486         /* try to clean up all streams */
1487         release_sink(ctl);
1488         /*
1489          * Sending SMTP QUIT on signal is theoretically nice, but led
1490          * to a subtle bug.  If fetchmail was terminated by signal
1491          * while it was shipping message text, it would hang forever
1492          * waiting for a command acknowledge.  In theory we could
1493          * enable the QUIT only outside of the message send.  In
1494          * practice, we don't care.  All mailservers hang up on a
1495          * dropped TCP/IP connection anyway.
1496          */
1497         smtp_close(ctl, 0);
1498         if (mailserver_socket != -1) {
1499             cleanupSockClose(mailserver_socket);
1500             mailserver_socket = -1;
1501         }
1502         /* If there was a connect timeout, the socket should be closed.
1503          * mailserver_socket_temp contains the socket to close.
1504          */
1505         if (mailserver_socket_temp != -1) {
1506             cleanupSockClose(mailserver_socket_temp);
1507             mailserver_socket_temp = -1;
1508         }
1509     }
1510
1511     /* no report on PS_AUTHFAIL */
1512     msg = NULL;
1513     switch (err)
1514     {
1515     case PS_SOCKET:
1516         msg = GT_("socket");
1517         break;
1518     case PS_SYNTAX:
1519         msg = GT_("missing or bad RFC822 header");
1520         break;
1521     case PS_IOERR:
1522         msg = GT_("MDA");
1523         break;
1524     case PS_ERROR:
1525         msg = GT_("client/server synchronization");
1526         break;
1527     case PS_PROTOCOL:
1528         msg = GT_("client/server protocol");
1529         break;
1530     case PS_LOCKBUSY:
1531         msg = GT_("lock busy on server");
1532         break;
1533     case PS_SMTP:
1534         msg = GT_("SMTP transaction");
1535         break;
1536     case PS_DNS:
1537         msg = GT_("DNS lookup");
1538         break;
1539     case PS_UNDEFINED:
1540         msg = GT_("undefined");
1541         break;
1542     }
1543     if (msg) {
1544         if (phase == FORWARDING_WAIT || phase == LISTENER_WAIT
1545                 || err == PS_SMTP)
1546             report(stderr, GT_("%s error while fetching from %s@%s and delivering to SMTP host %s\n"),
1547                     msg, ctl->remotename, ctl->server.pollname,
1548                     ctl->smtphost ? ctl->smtphost : GT_("unknown"));
1549         else
1550             report(stderr, GT_("%s error while fetching from %s@%s\n"),
1551                     msg, ctl->remotename, ctl->server.pollname);
1552     }
1553
1554 closeUp:
1555     xfree(msgsizes);
1556     ctl->folder = NULL;
1557
1558     /* execute wrapup command, if any */
1559     if (ctl->postconnect && (tmperr = system(ctl->postconnect)))
1560     {
1561         report(stderr, GT_("post-connection command failed with status %d\n"), tmperr);
1562         if (err == PS_SUCCESS)
1563             err = PS_SYNTAX;
1564     }
1565
1566     set_timeout(0); /* cancel any pending alarm */
1567     set_signal_handler(SIGALRM, alrmsave);
1568     return(err);
1569 }
1570
1571 /** retrieve messages from server using given protocol method table */
1572 int do_protocol(struct query *ctl /** parsed options with merged-in defaults */,
1573                 const struct method *proto /** protocol method table */)
1574 {
1575     int err;
1576
1577 #ifndef KERBEROS_V4
1578     if (ctl->server.authenticate == A_KERBEROS_V4)
1579     {
1580         report(stderr, GT_("Kerberos V4 support not linked.\n"));
1581         return(PS_ERROR);
1582     }
1583 #endif /* KERBEROS_V4 */
1584
1585 #ifndef KERBEROS_V5
1586     if (ctl->server.authenticate == A_KERBEROS_V5)
1587     {
1588         report(stderr, GT_("Kerberos V5 support not linked.\n"));
1589         return(PS_ERROR);
1590     }
1591 #endif /* KERBEROS_V5 */
1592
1593     /* lacking methods, there are some options that may fail */
1594     if (!proto->is_old)
1595     {
1596         /* check for unsupported options */
1597         if (ctl->flush) {
1598             report(stderr,
1599                     GT_("Option --flush is not supported with %s\n"),
1600                     proto->name);
1601             return(PS_SYNTAX);
1602         }
1603         else if (ctl->fetchall) {
1604             report(stderr,
1605                     GT_("Option --all is not supported with %s\n"),
1606                     proto->name);
1607             return(PS_SYNTAX);
1608         }
1609     }
1610     if (!(proto->getsizes || proto->getpartialsizes)
1611             && NUM_SPECIFIED(ctl->limit))
1612     {
1613         report(stderr,
1614                 GT_("Option --limit is not supported with %s\n"),
1615                 proto->name);
1616         return(PS_SYNTAX);
1617     }
1618
1619     /*
1620      * If no expunge limit or we do expunges within the driver,
1621      * then just do one session, passing in any fetchlimit.
1622      */
1623     if ((ctl->keep && !ctl->flush) ||
1624         proto->retry || !NUM_SPECIFIED(ctl->expunge))
1625         return(do_session(ctl, proto, NUM_VALUE_OUT(ctl->fetchlimit)));
1626     /*
1627      * There's an expunge limit, and it isn't handled in the driver itself.
1628      * OK; do multiple sessions, each fetching a limited # of messages.
1629      * Stop if the total count of retrieved messages exceeds ctl->fetchlimit
1630      * (if it was nonzero).
1631      */
1632     else
1633     {
1634         int totalcount = 0;
1635         int lockouts   = 0;
1636         int expunge    = NUM_VALUE_OUT(ctl->expunge);
1637         int fetchlimit = NUM_VALUE_OUT(ctl->fetchlimit);
1638
1639         do {
1640             if (fetchlimit > 0 && (expunge == 0 || expunge > fetchlimit - totalcount))
1641                 expunge = fetchlimit - totalcount;
1642             err = do_session(ctl, proto, expunge);
1643             totalcount += expunge;
1644             if (NUM_SPECIFIED(ctl->fetchlimit) && totalcount >= fetchlimit)
1645                 break;
1646             if (err != PS_LOCKBUSY)
1647                 lockouts = 0;
1648             else if (lockouts >= MAX_LOCKOUTS)
1649                 break;
1650             else /* err == PS_LOCKBUSY */
1651             {
1652                 /*
1653                  * Allow time for the server lock to release.  if we
1654                  * don't do this, we'll often hit a locked-mailbox
1655                  * condition and fail.
1656                  */
1657                 lockouts++;
1658                 sleep(3);
1659             }
1660         } while
1661             (err == PS_MAXFETCH || err == PS_LOCKBUSY);
1662
1663         return(err);
1664     }
1665 }
1666
1667
1668 /* driver.c ends here */