pattern.c

   1 /*
   2  * Copyright (C) 1996-2000,2006-7 Michael R. Elkins <me@mutt.org>, and others
   3  *
   4  *     This program is free software; you can redistribute it and/or modify
   5  *     it under the terms of the GNU General Public License as published by
   6  *     the Free Software Foundation; either version 2 of the License, or
   7  *     (at your option) any later version.
   8  *
   9  *     This program is distributed in the hope that it will be useful,
  10  *     but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  *     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12  *     GNU General Public License for more details.
  13  *
  14  *     You should have received a copy of the GNU General Public License
  15  *     along with this program; if not, write to the Free Software
  16  *     Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
  17  */
  18
  19 #if HAVE_CONFIG_H
  20 # include "config.h"
  21 #endif
  22
  23 #include "mutt.h"
  24 #include "mapping.h"
  25 #include "keymap.h"
  26 #include "mailbox.h"
  27 #include "copy.h"
  28
  29 #include <string.h>
  30 #include <stdlib.h>
  31 #include <ctype.h>
  32 #include <sys/stat.h>
  33 #include <unistd.h>
  34 #include <stdarg.h>
  35
  36 #include "mutt_crypt.h"
  37 #include "mutt_curses.h"
  38 #include "group.h"
  39
  40 #ifdef USE_IMAP
  41 #include "mx.h"
  42 #include "imap/imap.h"
  43 #endif
  44
  45 static int eat_regexp (pattern_t *pat, BUFFER *, BUFFER *);
  46 static int eat_date (pattern_t *pat, BUFFER *, BUFFER *);
  47 static int eat_range (pattern_t *pat, BUFFER *, BUFFER *);
  48 static int patmatch (const pattern_t *pat, const char *buf);
  49
  50 static struct pattern_flags
  51 {
  52   int tag;      /* character used to represent this op */
  53   int op;       /* operation to perform */
  54   int class;
  55   int (*eat_arg) (pattern_t *, BUFFER *, BUFFER *);
  56 }
  57 Flags[] =
  58 {
  59   { 'A', M_ALL,                 0,              NULL },
  60   { 'b', M_BODY,                M_FULL_MSG,     eat_regexp },
  61   { 'B', M_WHOLE_MSG,           M_FULL_MSG,     eat_regexp },
  62   { 'c', M_CC,                  0,              eat_regexp },
  63   { 'C', M_RECIPIENT,           0,              eat_regexp },
  64   { 'd', M_DATE,                0,              eat_date },
  65   { 'D', M_DELETED,             0,              NULL },
  66   { 'e', M_SENDER,              0,              eat_regexp },
  67   { 'E', M_EXPIRED,             0,              NULL },
  68   { 'f', M_FROM,                0,              eat_regexp },
  69   { 'F', M_FLAG,                0,              NULL },
  70   { 'g', M_CRYPT_SIGN,          0,              NULL },
  71   { 'G', M_CRYPT_ENCRYPT,       0,              NULL },
  72   { 'h', M_HEADER,              M_FULL_MSG,     eat_regexp },
  73   { 'H', M_HORMEL,              0,              eat_regexp },
  74   { 'i', M_ID,                  0,              eat_regexp },
  75   { 'k', M_PGP_KEY,             0,              NULL },
  76   { 'l', M_LIST,                0,              NULL },
  77   { 'L', M_ADDRESS,             0,              eat_regexp },
  78   { 'm', M_MESSAGE,             0,              eat_range },
  79   { 'n', M_SCORE,               0,              eat_range },
  80   { 'N', M_NEW,                 0,              NULL },
  81   { 'O', M_OLD,                 0,              NULL },
  82   { 'p', M_PERSONAL_RECIP,      0,              NULL },
  83   { 'P', M_PERSONAL_FROM,       0,              NULL },
  84   { 'Q', M_REPLIED,             0,              NULL },
  85   { 'r', M_DATE_RECEIVED,       0,              eat_date },
  86   { 'R', M_READ,                0,              NULL },
  87   { 's', M_SUBJECT,             0,              eat_regexp },
  88   { 'S', M_SUPERSEDED,          0,              NULL },
  89   { 't', M_TO,                  0,              eat_regexp },
  90   { 'T', M_TAG,                 0,              NULL },
  91   { 'u', M_SUBSCRIBED_LIST,     0,              NULL },
  92   { 'U', M_UNREAD,              0,              NULL },
  93   { 'v', M_COLLAPSED,           0,              NULL },
  94   { 'V', M_CRYPT_VERIFIED,      0,              NULL },
  95   { 'x', M_REFERENCE,           0,              eat_regexp },
  96   { 'X', M_MIMEATTACH,          0,              eat_range },
  97   { 'y', M_XLABEL,              0,              eat_regexp },
  98   { 'z', M_SIZE,                0,              eat_range },
  99   { '=', M_DUPLICATED,          0,              NULL },
 100   { '$', M_UNREFERENCED,        0,              NULL },
 101   { 0,   0,                     0,              NULL }
 102 };
 103
 104 static pattern_t *SearchPattern = NULL; /* current search pattern */
 105 static char LastSearch[STRING] = { 0 }; /* last pattern searched for */
 106 static char LastSearchExpn[LONG_STRING] = { 0 }; /* expanded version of
 107                                                     LastSearch */
 108
 109 #define M_MAXRANGE -1
 110
 111 /* constants for parse_date_range() */
 112 #define M_PDR_NONE      0x0000
 113 #define M_PDR_MINUS     0x0001
 114 #define M_PDR_PLUS      0x0002
 115 #define M_PDR_WINDOW    0x0004
 116 #define M_PDR_ABSOLUTE  0x0008
 117 #define M_PDR_DONE      0x0010
 118 #define M_PDR_ERROR     0x0100
 119 #define M_PDR_ERRORDONE (M_PDR_ERROR | M_PDR_DONE)
 120
 121
 122 /* if no uppercase letters are given, do a case-insensitive search */
 123 int mutt_which_case (const char *s)
 124 {
 125   wchar_t w;
 126   mbstate_t mb;
 127   size_t l;
 128
 129   memset (&mb, 0, sizeof (mb));
 130
 131   for (; (l = mbrtowc (&w, s, MB_CUR_MAX, &mb)) != 0; s += l)
 132   {
 133     if (l == (size_t) -2)
 134       continue; /* shift sequences */
 135     if (l == (size_t) -1)
 136       return 0; /* error; assume case-sensitive */
 137     if (iswalpha ((wint_t) w) && iswupper ((wint_t) w))
 138       return 0; /* case-sensitive */
 139   }
 140
 141   return REG_ICASE; /* case-insensitive */
 142 }
 143
 144 static int
 145 msg_search (CONTEXT *ctx, pattern_t* pat, int msgno)
 146 {
 147   char tempfile[_POSIX_PATH_MAX];
 148   MESSAGE *msg = NULL;
 149   STATE s;
 150   struct stat st;
 151   FILE *fp = NULL;
 152   long lng = 0;
 153   int match = 0;
 154   HEADER *h = ctx->hdrs[msgno];
 155   char *buf;
 156   size_t blen;
 157
 158   if ((msg = mx_open_message (ctx, msgno)) != NULL)
 159   {
 160     if (option (OPTTHOROUGHSRC))
 161     {
 162       /* decode the header / body */
 163       memset (&s, 0, sizeof (s));
 164       s.fpin = msg->fp;
 165       s.flags = M_CHARCONV;
 166       mutt_mktemp (tempfile, sizeof (tempfile));
 167       if ((s.fpout = safe_fopen (tempfile, "w+")) == NULL)
 168       {
 169         mutt_perror (tempfile);
 170         return (0);
 171       }
 172
 173       if (pat->op != M_BODY)
 174         mutt_copy_header (msg->fp, h, s.fpout, CH_FROM | CH_DECODE, NULL);
 175
 176       if (pat->op != M_HEADER)
 177       {
 178         mutt_parse_mime_message (ctx, h);
 179
 180         if (WithCrypto && (h->security & ENCRYPT)
 181             && !crypt_valid_passphrase(h->security))
 182         {
 183           mx_close_message (&msg);
 184           if (s.fpout)
 185           {
 186             safe_fclose (&s.fpout);
 187             unlink (tempfile);
 188           }
 189           return (0);
 190         }
 191
 192         fseeko (msg->fp, h->offset, 0);
 193         mutt_body_handler (h->content, &s);
 194       }
 195
 196       fp = s.fpout;
 197       fflush (fp);
 198       fseek (fp, 0, 0);
 199       fstat (fileno (fp), &st);
 200       lng = (long) st.st_size;
 201     }
 202     else
 203     {
 204       /* raw header / body */
 205       fp = msg->fp;
 206       if (pat->op != M_BODY)
 207       {
 208         fseeko (fp, h->offset, 0);
 209         lng = h->content->offset - h->offset;
 210       }
 211       if (pat->op != M_HEADER)
 212       {
 213         if (pat->op == M_BODY)
 214           fseeko (fp, h->content->offset, 0);
 215         lng += h->content->length;
 216       }
 217     }
 218
 219     blen = STRING;
 220     buf = safe_malloc (blen);
 221
 222     /* search the file "fp" */
 223     while (lng > 0)
 224     {
 225       if (pat->op == M_HEADER)
 226       {
 227         if (*(buf = mutt_read_rfc822_line (fp, buf, &blen)) == '\0')
 228           break;
 229       }
 230       else if (fgets (buf, blen - 1, fp) == NULL)
 231         break; /* don't loop forever */
 232       if (patmatch (pat, buf) == 0)
 233       {
 234         match = 1;
 235         break;
 236       }
 237       lng -= mutt_strlen (buf);
 238     }
 239
 240     FREE (&buf);
 241
 242     mx_close_message (&msg);
 243
 244     if (option (OPTTHOROUGHSRC))
 245     {
 246       safe_fclose (&fp);
 247       unlink (tempfile);
 248     }
 249   }
 250
 251   return match;
 252 }
 253
 254 static int eat_regexp (pattern_t *pat, BUFFER *s, BUFFER *err)
 255 {
 256   BUFFER buf;
 257   char errmsg[STRING];
 258   int r;
 259
 260   memset (&buf, 0, sizeof (buf));
 261   if (mutt_extract_token (&buf, s, M_TOKEN_PATTERN | M_TOKEN_COMMENT) != 0 ||
 262       !buf.data)
 263   {
 264     snprintf (err->data, err->dsize, _("Error in expression: %s"), s->dptr);
 265     return (-1);
 266   }
 267   if (!*buf.data)
 268   {
 269     snprintf (err->data, err->dsize, _("Empty expression"));
 270     return (-1);
 271   }
 272
 273 #if 0
 274   /* If there are no RE metacharacters, use simple search anyway */
 275   if (!pat->stringmatch && !strpbrk (buf.data, "|[{.*+?^$"))
 276     pat->stringmatch = 1;
 277 #endif
 278
 279   if (pat->stringmatch)
 280   {
 281     pat->p.str = safe_strdup (buf.data);
 282     pat->ign_case = mutt_which_case (buf.data) == REG_ICASE;
 283     FREE (&buf.data);
 284   }
 285   else if (pat->groupmatch)
 286   {
 287     pat->p.g = mutt_pattern_group (buf.data);
 288     FREE (&buf.data);
 289   }
 290   else
 291   {
 292     pat->p.rx = safe_malloc (sizeof (regex_t));
 293     r = REGCOMP (pat->p.rx, buf.data, REG_NEWLINE | REG_NOSUB | mutt_which_case (buf.data));
 294     if (r)
 295     {
 296       regerror (r, pat->p.rx, errmsg, sizeof (errmsg));
 297       mutt_buffer_printf (err, "'%s': %s", buf.data, errmsg);
 298       FREE (&buf.data);
 299       FREE (&pat->p.rx);
 300       return (-1);
 301     }
 302     FREE (&buf.data);
 303   }
 304
 305   return 0;
 306 }
 307
 308 int eat_range (pattern_t *pat, BUFFER *s, BUFFER *err)
 309 {
 310   char *tmp;
 311   int do_exclusive = 0;
 312   int skip_quote = 0;
 313
 314   /*
 315    * If simple_search is set to "~m %s", the range will have double quotes
 316    * around it...
 317    */
 318   if (*s->dptr == '"')
 319   {
 320     s->dptr++;
 321     skip_quote = 1;
 322   }
 323   if (*s->dptr == '<')
 324     do_exclusive = 1;
 325   if ((*s->dptr != '-') && (*s->dptr != '<'))
 326   {
 327     /* range minimum */
 328     if (*s->dptr == '>')
 329     {
 330       pat->max = M_MAXRANGE;
 331       pat->min = strtol (s->dptr + 1, &tmp, 0) + 1; /* exclusive range */
 332     }
 333     else
 334       pat->min = strtol (s->dptr, &tmp, 0);
 335     if (toupper ((unsigned char) *tmp) == 'K') /* is there a prefix? */
 336     {
 337       pat->min *= 1024;
 338       tmp++;
 339     }
 340     else if (toupper ((unsigned char) *tmp) == 'M')
 341     {
 342       pat->min *= 1048576;
 343       tmp++;
 344     }
 345     if (*s->dptr == '>')
 346     {
 347       s->dptr = tmp;
 348       return 0;
 349     }
 350     if (*tmp != '-')
 351     {
 352       /* exact value */
 353       pat->max = pat->min;
 354       s->dptr = tmp;
 355       return 0;
 356     }
 357     tmp++;
 358   }
 359   else
 360   {
 361     s->dptr++;
 362     tmp = s->dptr;
 363   }
 364
 365   if (isdigit ((unsigned char) *tmp))
 366   {
 367     /* range maximum */
 368     pat->max = strtol (tmp, &tmp, 0);
 369     if (toupper ((unsigned char) *tmp) == 'K')
 370     {
 371       pat->max *= 1024;
 372       tmp++;
 373     }
 374     else if (toupper ((unsigned char) *tmp) == 'M')
 375     {
 376       pat->max *= 1048576;
 377       tmp++;
 378     }
 379     if (do_exclusive)
 380       (pat->max)--;
 381   }
 382   else
 383     pat->max = M_MAXRANGE;
 384
 385   if (skip_quote && *tmp == '"')
 386     tmp++;
 387
 388   SKIPWS (tmp);
 389   s->dptr = tmp;
 390   return 0;
 391 }
 392
 393 static const char *getDate (const char *s, struct tm *t, BUFFER *err)
 394 {
 395   char *p;
 396   time_t now = time (NULL);
 397   struct tm *tm = localtime (&now);
 398
 399   t->tm_mday = strtol (s, &p, 10);
 400   if (t->tm_mday < 1 || t->tm_mday > 31)
 401   {
 402     snprintf (err->data, err->dsize, _("Invalid day of month: %s"), s);
 403     return NULL;
 404   }
 405   if (*p != '/')
 406   {
 407     /* fill in today's month and year */
 408     t->tm_mon = tm->tm_mon;
 409     t->tm_year = tm->tm_year;
 410     return p;
 411   }
 412   p++;
 413   t->tm_mon = strtol (p, &p, 10) - 1;
 414   if (t->tm_mon < 0 || t->tm_mon > 11)
 415   {
 416     snprintf (err->data, err->dsize, _("Invalid month: %s"), p);
 417     return NULL;
 418   }
 419   if (*p != '/')
 420   {
 421     t->tm_year = tm->tm_year;
 422     return p;
 423   }
 424   p++;
 425   t->tm_year = strtol (p, &p, 10);
 426   if (t->tm_year < 70) /* year 2000+ */
 427     t->tm_year += 100;
 428   else if (t->tm_year > 1900)
 429     t->tm_year -= 1900;
 430   return p;
 431 }
 432
 433 /* Ny   years
 434    Nm   months
 435    Nw   weeks
 436    Nd   days */
 437 static const char *get_offset (struct tm *tm, const char *s, int sign)
 438 {
 439   char *ps;
 440   int offset = strtol (s, &ps, 0);
 441   if ((sign < 0 && offset > 0) || (sign > 0 && offset < 0))
 442     offset = -offset;
 443
 444   switch (*ps)
 445   {
 446     case 'y':
 447       tm->tm_year += offset;
 448       break;
 449     case 'm':
 450       tm->tm_mon += offset;
 451       break;
 452     case 'w':
 453       tm->tm_mday += 7 * offset;
 454       break;
 455     case 'd':
 456       tm->tm_mday += offset;
 457       break;
 458     default:
 459       return s;
 460   }
 461   mutt_normalize_time (tm);
 462   return (ps + 1);
 463 }
 464
 465 static void adjust_date_range (struct tm *min, struct tm *max)
 466 {
 467   if (min->tm_year > max->tm_year
 468       || (min->tm_year == max->tm_year && min->tm_mon > max->tm_mon)
 469       || (min->tm_year == max->tm_year && min->tm_mon == max->tm_mon
 470         && min->tm_mday > max->tm_mday))
 471   {
 472     int tmp;
 473
 474     tmp = min->tm_year;
 475     min->tm_year = max->tm_year;
 476     max->tm_year = tmp;
 477
 478     tmp = min->tm_mon;
 479     min->tm_mon = max->tm_mon;
 480     max->tm_mon = tmp;
 481
 482     tmp = min->tm_mday;
 483     min->tm_mday = max->tm_mday;
 484     max->tm_mday = tmp;
 485
 486     min->tm_hour = min->tm_min = min->tm_sec = 0;
 487     max->tm_hour = 23;
 488     max->tm_min = max->tm_sec = 59;
 489   }
 490 }
 491
 492 static const char * parse_date_range (const char* pc, struct tm *min,
 493     struct tm *max, int haveMin, struct tm *baseMin, BUFFER *err)
 494 {
 495   int flag = M_PDR_NONE;
 496   while (*pc && ((flag & M_PDR_DONE) == 0))
 497   {
 498     const char *pt;
 499     char ch = *pc++;
 500     SKIPWS (pc);
 501     switch (ch)
 502     {
 503       case '-':
 504       {
 505         /* try a range of absolute date minus offset of Ndwmy */
 506         pt = get_offset (min, pc, -1);
 507         if (pc == pt)
 508         {
 509           if (flag == M_PDR_NONE)
 510           { /* nothing yet and no offset parsed => absolute date? */
 511             if (!getDate (pc, max, err))
 512               flag |= (M_PDR_ABSOLUTE | M_PDR_ERRORDONE);  /* done bad */
 513             else
 514             {
 515               /* reestablish initial base minimum if not specified */
 516               if (!haveMin)
 517                 memcpy (min, baseMin, sizeof(struct tm));
 518               flag |= (M_PDR_ABSOLUTE | M_PDR_DONE);  /* done good */
 519             }
 520           }
 521           else
 522             flag |= M_PDR_ERRORDONE;
 523         }
 524         else
 525         {
 526           pc = pt;
 527           if (flag == M_PDR_NONE && !haveMin)
 528           { /* the very first "-3d" without a previous absolute date */
 529             max->tm_year = min->tm_year;
 530             max->tm_mon = min->tm_mon;
 531             max->tm_mday = min->tm_mday;
 532           }
 533           flag |= M_PDR_MINUS;
 534         }
 535       }
 536       break;
 537       case '+':
 538       { /* enlarge plusRange */
 539         pt = get_offset (max, pc, 1);
 540         if (pc == pt)
 541           flag |= M_PDR_ERRORDONE;
 542         else
 543         {
 544           pc = pt;
 545           flag |= M_PDR_PLUS;
 546         }
 547       }
 548       break;
 549       case '*':
 550       { /* enlarge window in both directions */
 551         pt = get_offset (min, pc, -1);
 552         if (pc == pt)
 553           flag |= M_PDR_ERRORDONE;
 554         else
 555         {
 556           pc = get_offset (max, pc, 1);
 557           flag |= M_PDR_WINDOW;
 558         }
 559       }
 560       break;
 561       default:
 562         flag |= M_PDR_ERRORDONE;
 563     }
 564     SKIPWS (pc);
 565   }
 566   if ((flag & M_PDR_ERROR) && !(flag & M_PDR_ABSOLUTE))
 567   { /* getDate has its own error message, don't overwrite it here */
 568     snprintf (err->data, err->dsize, _("Invalid relative date: %s"), pc-1);
 569   }
 570   return ((flag & M_PDR_ERROR) ? NULL : pc);
 571 }
 572
 573 static int eat_date (pattern_t *pat, BUFFER *s, BUFFER *err)
 574 {
 575   BUFFER buffer;
 576   struct tm min, max;
 577
 578   memset (&buffer, 0, sizeof (buffer));
 579   if (mutt_extract_token (&buffer, s, M_TOKEN_COMMENT | M_TOKEN_PATTERN) != 0
 580       || !buffer.data)
 581   {
 582     strfcpy (err->data, _("error in expression"), err->dsize);
 583     return (-1);
 584   }
 585
 586   memset (&min, 0, sizeof (min));
 587   /* the `0' time is Jan 1, 1970 UTC, so in order to prevent a negative time
 588      when doing timezone conversion, we use Jan 2, 1970 UTC as the base
 589      here */
 590   min.tm_mday = 2;
 591   min.tm_year = 70;
 592
 593   memset (&max, 0, sizeof (max));
 594
 595   /* Arbitrary year in the future.  Don't set this too high
 596      or mutt_mktime() returns something larger than will
 597      fit in a time_t on some systems */
 598   max.tm_year = 130;
 599   max.tm_mon = 11;
 600   max.tm_mday = 31;
 601   max.tm_hour = 23;
 602   max.tm_min = 59;
 603   max.tm_sec = 59;
 604
 605   if (strchr ("<>=", buffer.data[0]))
 606   {
 607     /* offset from current time
 608        <3d      less than three days ago
 609        >3d      more than three days ago
 610        =3d      exactly three days ago */
 611     time_t now = time (NULL);
 612     struct tm *tm = localtime (&now);
 613     int exact = 0;
 614
 615     if (buffer.data[0] == '<')
 616     {
 617       memcpy (&min, tm, sizeof (min));
 618       tm = &min;
 619     }
 620     else
 621     {
 622       memcpy (&max, tm, sizeof (max));
 623       tm = &max;
 624
 625       if (buffer.data[0] == '=')
 626         exact++;
 627     }
 628     tm->tm_hour = 23;
 629     tm->tm_min = tm->tm_sec = 59;
 630
 631     /* force negative offset */
 632     get_offset (tm, buffer.data + 1, -1);
 633
 634     if (exact)
 635     {
 636       /* start at the beginning of the day in question */
 637       memcpy (&min, &max, sizeof (max));
 638       min.tm_hour = min.tm_sec = min.tm_min = 0;
 639     }
 640   }
 641   else
 642   {
 643     const char *pc = buffer.data;
 644
 645     int haveMin = FALSE;
 646     int untilNow = FALSE;
 647     if (isdigit ((unsigned char)*pc))
 648     {
 649       /* mininum date specified */
 650       if ((pc = getDate (pc, &min, err)) == NULL)
 651       {
 652         FREE (&buffer.data);
 653         return (-1);
 654       }
 655       haveMin = TRUE;
 656       SKIPWS (pc);
 657       if (*pc == '-')
 658       {
 659         const char *pt = pc + 1;
 660         SKIPWS (pt);
 661         untilNow = (*pt == '\0');
 662       }
 663     }
 664
 665     if (!untilNow)
 666     { /* max date or relative range/window */
 667
 668       struct tm baseMin;
 669
 670       if (!haveMin)
 671       { /* save base minimum and set current date, e.g. for "-3d+1d" */
 672         time_t now = time (NULL);
 673         struct tm *tm = localtime (&now);
 674         memcpy (&baseMin, &min, sizeof(baseMin));
 675         memcpy (&min, tm, sizeof (min));
 676         min.tm_hour = min.tm_sec = min.tm_min = 0;
 677       }
 678
 679       /* preset max date for relative offsets,
 680          if nothing follows we search for messages on a specific day */
 681       max.tm_year = min.tm_year;
 682       max.tm_mon = min.tm_mon;
 683       max.tm_mday = min.tm_mday;
 684
 685       if (!parse_date_range (pc, &min, &max, haveMin, &baseMin, err))
 686       { /* bail out on any parsing error */
 687         FREE (&buffer.data);
 688         return (-1);
 689       }
 690     }
 691   }
 692
 693   /* Since we allow two dates to be specified we'll have to adjust that. */
 694   adjust_date_range (&min, &max);
 695
 696   pat->min = mutt_mktime (&min, 1);
 697   pat->max = mutt_mktime (&max, 1);
 698
 699   FREE (&buffer.data);
 700
 701   return 0;
 702 }
 703
 704 static int patmatch (const pattern_t* pat, const char* buf)
 705 {
 706   if (pat->stringmatch)
 707     return pat->ign_case ? !strcasestr (buf, pat->p.str) :
 708                            !strstr (buf, pat->p.str);
 709   else if (pat->groupmatch)
 710     return !mutt_group_match (pat->p.g, buf);
 711   else
 712     return regexec (pat->p.rx, buf, 0, NULL, 0);
 713 }
 714
 715 static struct pattern_flags *lookup_tag (char tag)
 716 {
 717   int i;
 718
 719   for (i = 0; Flags[i].tag; i++)
 720     if (Flags[i].tag == tag)
 721       return (&Flags[i]);
 722   return NULL;
 723 }
 724
 725 static /* const */ char *find_matching_paren (/* const */ char *s)
 726 {
 727   int level = 1;
 728
 729   for (; *s; s++)
 730   {
 731     if (*s == '(')
 732       level++;
 733     else if (*s == ')')
 734     {
 735       level--;
 736       if (!level)
 737         break;
 738     }
 739   }
 740   return s;
 741 }
 742
 743 void mutt_pattern_free (pattern_t **pat)
 744 {
 745   pattern_t *tmp;
 746
 747   while (*pat)
 748   {
 749     tmp = *pat;
 750     *pat = (*pat)->next;
 751
 752     if (tmp->stringmatch)
 753       FREE (&tmp->p.str);
 754     else if (tmp->groupmatch)
 755       tmp->p.g = NULL;
 756     else if (tmp->p.rx)
 757     {
 758       regfree (tmp->p.rx);
 759       FREE (&tmp->p.rx);
 760     }
 761
 762     if (tmp->child)
 763       mutt_pattern_free (&tmp->child);
 764     FREE (&tmp);
 765   }
 766 }
 767
 768 pattern_t *mutt_pattern_comp (/* const */ char *s, int flags, BUFFER *err)
 769 {
 770   pattern_t *curlist = NULL;
 771   pattern_t *tmp, *tmp2;
 772   pattern_t *last = NULL;
 773   int not = 0;
 774   int alladdr = 0;
 775   int or = 0;
 776   int implicit = 1;     /* used to detect logical AND operator */
 777   struct pattern_flags *entry;
 778   char *p;
 779   char *buf;
 780   BUFFER ps;
 781
 782   memset (&ps, 0, sizeof (ps));
 783   ps.dptr = s;
 784   ps.dsize = mutt_strlen (s);
 785
 786   while (*ps.dptr)
 787   {
 788     SKIPWS (ps.dptr);
 789     switch (*ps.dptr)
 790     {
 791       case '^':
 792         ps.dptr++;
 793         alladdr = !alladdr;
 794         break;
 795       case '!':
 796         ps.dptr++;
 797         not = !not;
 798         break;
 799       case '|':
 800         if (!or)
 801         {
 802           if (!curlist)
 803           {
 804             snprintf (err->data, err->dsize, _("error in pattern at: %s"), ps.dptr);
 805             return NULL;
 806           }
 807           if (curlist->next)
 808           {
 809             /* A & B | C == (A & B) | C */
 810             tmp = new_pattern ();
 811             tmp->op = M_AND;
 812             tmp->child = curlist;
 813
 814             curlist = tmp;
 815             last = curlist;
 816           }
 817
 818           or = 1;
 819         }
 820         ps.dptr++;
 821         implicit = 0;
 822         not = 0;
 823         alladdr = 0;
 824         break;
 825       case '%':
 826       case '=':
 827       case '~':
 828         if (!*(ps.dptr + 1))
 829         {
 830           snprintf (err->data, err->dsize, _("missing pattern: %s"), ps.dptr);
 831           mutt_pattern_free (&curlist);
 832           return NULL;
 833         }
 834         if (*(ps.dptr + 1) == '(')
 835         {
 836           ps.dptr ++; /* skip ~ */
 837           p = find_matching_paren (ps.dptr + 1);
 838           if (*p != ')')
 839           {
 840             snprintf (err->data, err->dsize, _("mismatched brackets: %s"), ps.dptr);
 841             mutt_pattern_free (&curlist);
 842             return NULL;
 843           }
 844           tmp = new_pattern ();
 845           tmp->op = M_THREAD;
 846           if (last)
 847             last->next = tmp;
 848           else
 849             curlist = tmp;
 850           last = tmp;
 851           tmp->not ^= not;
 852           tmp->alladdr |= alladdr;
 853           not = 0;
 854           alladdr = 0;
 855           /* compile the sub-expression */
 856           buf = mutt_substrdup (ps.dptr + 1, p);
 857           if ((tmp2 = mutt_pattern_comp (buf, flags, err)) == NULL)
 858           {
 859             FREE (&buf);
 860             mutt_pattern_free (&curlist);
 861             return NULL;
 862           }
 863           FREE (&buf);
 864           tmp->child = tmp2;
 865           ps.dptr = p + 1; /* restore location */
 866           break;
 867         }
 868         if (implicit && or)
 869         {
 870           /* A | B & C == (A | B) & C */
 871           tmp = new_pattern ();
 872           tmp->op = M_OR;
 873           tmp->child = curlist;
 874           curlist = tmp;
 875           last = tmp;
 876           or = 0;
 877         }
 878
 879         tmp = new_pattern ();
 880         tmp->not = not;
 881         tmp->alladdr = alladdr;
 882         tmp->stringmatch = (*ps.dptr == '=') ? 1 : 0;
 883         tmp->groupmatch  = (*ps.dptr == '%') ? 1 : 0;
 884         not = 0;
 885         alladdr = 0;
 886
 887         if (last)
 888           last->next = tmp;
 889         else
 890           curlist = tmp;
 891         last = tmp;
 892
 893         ps.dptr++; /* move past the ~ */
 894         if ((entry = lookup_tag (*ps.dptr)) == NULL)
 895         {
 896           snprintf (err->data, err->dsize, _("%c: invalid pattern modifier"), *ps.dptr);
 897           mutt_pattern_free (&curlist);
 898           return NULL;
 899         }
 900         if (entry->class && (flags & entry->class) == 0)
 901         {
 902           snprintf (err->data, err->dsize, _("%c: not supported in this mode"), *ps.dptr);
 903           mutt_pattern_free (&curlist);
 904           return NULL;
 905         }
 906         tmp->op = entry->op;
 907
 908         ps.dptr++; /* eat the operator and any optional whitespace */
 909         SKIPWS (ps.dptr);
 910
 911         if (entry->eat_arg)
 912         {
 913           if (!*ps.dptr)
 914           {
 915             snprintf (err->data, err->dsize, _("missing parameter"));
 916             mutt_pattern_free (&curlist);
 917             return NULL;
 918           }
 919           if (entry->eat_arg (tmp, &ps, err) == -1)
 920           {
 921             mutt_pattern_free (&curlist);
 922             return NULL;
 923           }
 924         }
 925         implicit = 1;
 926         break;
 927       case '(':
 928         p = find_matching_paren (ps.dptr + 1);
 929         if (*p != ')')
 930         {
 931           snprintf (err->data, err->dsize, _("mismatched parenthesis: %s"), ps.dptr);
 932           mutt_pattern_free (&curlist);
 933           return NULL;
 934         }
 935         /* compile the sub-expression */
 936         buf = mutt_substrdup (ps.dptr + 1, p);
 937         if ((tmp = mutt_pattern_comp (buf, flags, err)) == NULL)
 938         {
 939           FREE (&buf);
 940           mutt_pattern_free (&curlist);
 941           return NULL;
 942         }
 943         FREE (&buf);
 944         if (last)
 945           last->next = tmp;
 946         else
 947           curlist = tmp;
 948         last = tmp;
 949         tmp->not ^= not;
 950         tmp->alladdr |= alladdr;
 951         not = 0;
 952         alladdr = 0;
 953         ps.dptr = p + 1; /* restore location */
 954         break;
 955       default:
 956         snprintf (err->data, err->dsize, _("error in pattern at: %s"), ps.dptr);
 957         mutt_pattern_free (&curlist);
 958         return NULL;
 959     }
 960   }
 961   if (!curlist)
 962   {
 963     strfcpy (err->data, _("empty pattern"), err->dsize);
 964     return NULL;
 965   }
 966   if (curlist->next)
 967   {
 968     tmp = new_pattern ();
 969     tmp->op = or ? M_OR : M_AND;
 970     tmp->child = curlist;
 971     curlist = tmp;
 972   }
 973   return (curlist);
 974 }
 975
 976 static int
 977 perform_and (pattern_t *pat, pattern_exec_flag flags, CONTEXT *ctx, HEADER *hdr)
 978 {
 979   for (; pat; pat = pat->next)
 980     if (mutt_pattern_exec (pat, flags, ctx, hdr) <= 0)
 981       return 0;
 982   return 1;
 983 }
 984
 985 static int
 986 perform_or (struct pattern_t *pat, pattern_exec_flag flags, CONTEXT *ctx, HEADER *hdr)
 987 {
 988   for (; pat; pat = pat->next)
 989     if (mutt_pattern_exec (pat, flags, ctx, hdr) > 0)
 990       return 1;
 991   return 0;
 992 }
 993
 994 static int match_adrlist (pattern_t *pat, int match_personal, int n, ...)
 995 {
 996   va_list ap;
 997   ADDRESS *a;
 998
 999   va_start (ap, n);
1000   for ( ; n ; n --)
1001   {
1002     for (a = va_arg (ap, ADDRESS *) ; a ; a = a->next)
1003     {
1004       if (pat->alladdr ^ ((a->mailbox && patmatch (pat, a->mailbox) == 0) ||
1005            (match_personal && a->personal && patmatch (pat, a->personal) == 0)))
1006       {
1007         va_end (ap);
1008         return (! pat->alladdr); /* Found match, or non-match if alladdr */
1009       }
1010     }
1011   }
1012   va_end (ap);
1013   return pat->alladdr; /* No matches, or all matches if alladdr */
1014 }
1015
1016 static int match_reference (pattern_t *pat, LIST *refs)
1017 {
1018   for (; refs; refs = refs->next)
1019     if (patmatch (pat, refs->data) == 0)
1020       return 1;
1021   return 0;
1022 }
1023
1024 /*
1025  * Matches subscribed mailing lists
1026  */
1027 int mutt_is_list_recipient (int alladdr, ADDRESS *a1, ADDRESS *a2)
1028 {
1029   for (; a1 ; a1 = a1->next)
1030     if (alladdr ^ mutt_is_subscribed_list (a1))
1031       return (! alladdr);
1032   for (; a2 ; a2 = a2->next)
1033     if (alladdr ^ mutt_is_subscribed_list (a2))
1034       return (! alladdr);
1035   return alladdr;
1036 }
1037
1038 /*
1039  * Matches known mailing lists
1040  * The function name may seem a little bit misleading: It checks all
1041  * recipients in To and Cc for known mailing lists, subscribed or not.
1042  */
1043 int mutt_is_list_cc (int alladdr, ADDRESS *a1, ADDRESS *a2)
1044 {
1045   for (; a1 ; a1 = a1->next)
1046     if (alladdr ^ mutt_is_mail_list (a1))
1047       return (! alladdr);
1048   for (; a2 ; a2 = a2->next)
1049     if (alladdr ^ mutt_is_mail_list (a2))
1050       return (! alladdr);
1051   return alladdr;
1052 }
1053
1054 static int match_user (int alladdr, ADDRESS *a1, ADDRESS *a2)
1055 {
1056   for (; a1 ; a1 = a1->next)
1057     if (alladdr ^ mutt_addr_is_user (a1))
1058       return (! alladdr);
1059   for (; a2 ; a2 = a2->next)
1060     if (alladdr ^ mutt_addr_is_user (a2))
1061       return (! alladdr);
1062   return alladdr;
1063 }
1064
1065 static int match_threadcomplete(struct pattern_t *pat, pattern_exec_flag flags, CONTEXT *ctx, THREAD *t,int left,int up,int right,int down)
1066 {
1067   int a;
1068   HEADER *h;
1069
1070   if(!t)
1071     return 0;
1072   h = t->message;
1073   if(h)
1074     if(mutt_pattern_exec(pat, flags, ctx, h))
1075       return 1;
1076
1077   if(up && (a=match_threadcomplete(pat, flags, ctx, t->parent,1,1,1,0)))
1078     return a;
1079   if(right && t->parent && (a=match_threadcomplete(pat, flags, ctx, t->next,0,0,1,1)))
1080     return a;
1081   if(left && t->parent && (a=match_threadcomplete(pat, flags, ctx, t->prev,1,0,0,1)))
1082     return a;
1083   if(down && (a=match_threadcomplete(pat, flags, ctx, t->child,1,0,1,1)))
1084     return a;
1085   return 0;
1086 }
1087
1088 /* flags
1089         M_MATCH_FULL_ADDRESS    match both personal and machine address */
1090 int
1091 mutt_pattern_exec (struct pattern_t *pat, pattern_exec_flag flags, CONTEXT *ctx, HEADER *h)
1092 {
1093   switch (pat->op)
1094   {
1095     case M_AND:
1096       return (pat->not ^ (perform_and (pat->child, flags, ctx, h) > 0));
1097     case M_OR:
1098       return (pat->not ^ (perform_or (pat->child, flags, ctx, h) > 0));
1099     case M_THREAD:
1100       return (pat->not ^ match_threadcomplete(pat->child, flags, ctx, h->thread, 1, 1, 1, 1));
1101     case M_ALL:
1102       return (!pat->not);
1103     case M_EXPIRED:
1104       return (pat->not ^ h->expired);
1105     case M_SUPERSEDED:
1106       return (pat->not ^ h->superseded);
1107     case M_FLAG:
1108       return (pat->not ^ h->flagged);
1109     case M_TAG:
1110       return (pat->not ^ h->tagged);
1111     case M_NEW:
1112       return (pat->not ? h->old || h->read : !(h->old || h->read));
1113     case M_UNREAD:
1114       return (pat->not ? h->read : !h->read);
1115     case M_REPLIED:
1116       return (pat->not ^ h->replied);
1117     case M_OLD:
1118       return (pat->not ? (!h->old || h->read) : (h->old && !h->read));
1119     case M_READ:
1120       return (pat->not ^ h->read);
1121     case M_DELETED:
1122       return (pat->not ^ h->deleted);
1123     case M_MESSAGE:
1124       return (pat->not ^ (h->msgno >= pat->min - 1 && (pat->max == M_MAXRANGE ||
1125                                                    h->msgno <= pat->max - 1)));
1126     case M_DATE:
1127       return (pat->not ^ (h->date_sent >= pat->min && h->date_sent <= pat->max));
1128     case M_DATE_RECEIVED:
1129       return (pat->not ^ (h->received >= pat->min && h->received <= pat->max));
1130     case M_BODY:
1131     case M_HEADER:
1132     case M_WHOLE_MSG:
1133       /*
1134        * ctx can be NULL in certain cases, such as when replying to a message from the attachment menu and
1135        * the user has a reply-hook using "~h" (bug #2190).
1136        */
1137       if (!ctx)
1138               return 0;
1139 #ifdef USE_IMAP
1140       /* IMAP search sets h->matched at search compile time */
1141       if (ctx->magic == M_IMAP && pat->stringmatch)
1142         return (h->matched);
1143 #endif
1144       return (pat->not ^ msg_search (ctx, pat, h->msgno));
1145     case M_SENDER:
1146       return (pat->not ^ match_adrlist (pat, flags & M_MATCH_FULL_ADDRESS, 1,
1147                                         h->env->sender));
1148     case M_FROM:
1149       return (pat->not ^ match_adrlist (pat, flags & M_MATCH_FULL_ADDRESS, 1,
1150                                         h->env->from));
1151     case M_TO:
1152       return (pat->not ^ match_adrlist (pat, flags & M_MATCH_FULL_ADDRESS, 1,
1153                                         h->env->to));
1154     case M_CC:
1155       return (pat->not ^ match_adrlist (pat, flags & M_MATCH_FULL_ADDRESS, 1,
1156                                         h->env->cc));
1157     case M_SUBJECT:
1158       return (pat->not ^ (h->env->subject && patmatch (pat, h->env->subject) == 0));
1159     case M_ID:
1160       return (pat->not ^ (h->env->message_id && patmatch (pat, h->env->message_id) == 0));
1161     case M_SCORE:
1162       return (pat->not ^ (h->score >= pat->min && (pat->max == M_MAXRANGE ||
1163                                                    h->score <= pat->max)));
1164     case M_SIZE:
1165       return (pat->not ^ (h->content->length >= pat->min && (pat->max == M_MAXRANGE || h->content->length <= pat->max)));
1166     case M_REFERENCE:
1167       return (pat->not ^ (match_reference (pat, h->env->references) ||
1168                           match_reference (pat, h->env->in_reply_to)));
1169     case M_ADDRESS:
1170       return (pat->not ^ match_adrlist (pat, flags & M_MATCH_FULL_ADDRESS, 4,
1171                                         h->env->from, h->env->sender,
1172                                         h->env->to, h->env->cc));
1173     case M_RECIPIENT:
1174            return (pat->not ^ match_adrlist (pat, flags & M_MATCH_FULL_ADDRESS,
1175                                              2, h->env->to, h->env->cc));
1176     case M_LIST:        /* known list, subscribed or not */
1177       return (pat->not ^ mutt_is_list_cc (pat->alladdr, h->env->to, h->env->cc));
1178     case M_SUBSCRIBED_LIST:
1179       return (pat->not ^ mutt_is_list_recipient (pat->alladdr, h->env->to, h->env->cc));
1180     case M_PERSONAL_RECIP:
1181       return (pat->not ^ match_user (pat->alladdr, h->env->to, h->env->cc));
1182     case M_PERSONAL_FROM:
1183       return (pat->not ^ match_user (pat->alladdr, h->env->from, NULL));
1184     case M_COLLAPSED:
1185       return (pat->not ^ (h->collapsed && h->num_hidden > 1));
1186    case M_CRYPT_SIGN:
1187      if (!WithCrypto)
1188        break;
1189      return (pat->not ^ ((h->security & SIGN) ? 1 : 0));
1190    case M_CRYPT_VERIFIED:
1191      if (!WithCrypto)
1192        break;
1193      return (pat->not ^ ((h->security & GOODSIGN) ? 1 : 0));
1194    case M_CRYPT_ENCRYPT:
1195      if (!WithCrypto)
1196        break;
1197      return (pat->not ^ ((h->security & ENCRYPT) ? 1 : 0));
1198    case M_PGP_KEY:
1199      if (!(WithCrypto & APPLICATION_PGP))
1200        break;
1201      return (pat->not ^ ((h->security & APPLICATION_PGP) && (h->security & PGPKEY)));
1202     case M_XLABEL:
1203       return (pat->not ^ (h->env->x_label && patmatch (pat, h->env->x_label) == 0));
1204     case M_HORMEL:
1205       return (pat->not ^ (h->env->spam && h->env->spam->data && patmatch (pat, h->env->spam->data) == 0));
1206     case M_DUPLICATED:
1207       return (pat->not ^ (h->thread && h->thread->duplicate_thread));
1208     case M_MIMEATTACH:
1209       {
1210       int count = mutt_count_body_parts (ctx, h);
1211       return (pat->not ^ (count >= pat->min && (pat->max == M_MAXRANGE ||
1212                                                 count <= pat->max)));
1213       }
1214     case M_UNREFERENCED:
1215       return (pat->not ^ (h->thread && !h->thread->child));
1216   }
1217   mutt_error (_("error: unknown op %d (report this error)."), pat->op);
1218   return (-1);
1219 }
1220
1221 static void quote_simple(char *tmp, size_t len, const char *p)
1222 {
1223   int i = 0;
1224
1225   tmp[i++] = '"';
1226   while (*p && i < len - 3)
1227   {
1228     if (*p == '\\' || *p == '"')
1229       tmp[i++] = '\\';
1230     tmp[i++] = *p++;
1231   }
1232   tmp[i++] = '"';
1233   tmp[i] = 0;
1234 }
1235
1236 /* convert a simple search into a real request */
1237 void mutt_check_simple (char *s, size_t len, const char *simple)
1238 {
1239   char tmp[LONG_STRING];
1240   int do_simple = 1;
1241   char *p;
1242
1243   for (p = s; p && *p; p++)
1244   {
1245     if (*p == '\\' && *(p + 1))
1246       p++;
1247     else if (*p == '~' || *p == '=' || *p == '%')
1248     {
1249       do_simple = 0;
1250       break;
1251     }
1252   }
1253
1254   /* XXX - is ascii_strcasecmp() right here, or should we use locale's
1255    * equivalences?
1256    */
1257
1258   if (do_simple) /* yup, so spoof a real request */
1259   {
1260     /* convert old tokens into the new format */
1261     if (ascii_strcasecmp ("all", s) == 0 ||
1262         !mutt_strcmp ("^", s) || !mutt_strcmp (".", s)) /* ~A is more efficient */
1263       strfcpy (s, "~A", len);
1264     else if (ascii_strcasecmp ("del", s) == 0)
1265       strfcpy (s, "~D", len);
1266     else if (ascii_strcasecmp ("flag", s) == 0)
1267       strfcpy (s, "~F", len);
1268     else if (ascii_strcasecmp ("new", s) == 0)
1269       strfcpy (s, "~N", len);
1270     else if (ascii_strcasecmp ("old", s) == 0)
1271       strfcpy (s, "~O", len);
1272     else if (ascii_strcasecmp ("repl", s) == 0)
1273       strfcpy (s, "~Q", len);
1274     else if (ascii_strcasecmp ("read", s) == 0)
1275       strfcpy (s, "~R", len);
1276     else if (ascii_strcasecmp ("tag", s) == 0)
1277       strfcpy (s, "~T", len);
1278     else if (ascii_strcasecmp ("unread", s) == 0)
1279       strfcpy (s, "~U", len);
1280     else
1281     {
1282       quote_simple (tmp, sizeof(tmp), s);
1283       mutt_expand_fmt (s, len, simple, tmp);
1284     }
1285   }
1286 }
1287
1288 int mutt_pattern_func (int op, char *prompt)
1289 {
1290   pattern_t *pat;
1291   char buf[LONG_STRING] = "", *simple, error[STRING];
1292   BUFFER err;
1293   int i;
1294   progress_t progress;
1295
1296   strfcpy (buf, NONULL (Context->pattern), sizeof (buf));
1297   if (mutt_get_field (prompt, buf, sizeof (buf), M_PATTERN | M_CLEAR) != 0 || !buf[0])
1298     return (-1);
1299
1300   mutt_message _("Compiling search pattern...");
1301
1302   simple = safe_strdup (buf);
1303   mutt_check_simple (buf, sizeof (buf), NONULL (SimpleSearch));
1304
1305   memset (&err, 0, sizeof(err));
1306   err.data = error;
1307   err.dsize = sizeof (error);
1308   if ((pat = mutt_pattern_comp (buf, M_FULL_MSG, &err)) == NULL)
1309   {
1310     FREE (&simple);
1311     mutt_error ("%s", err.data);
1312     return (-1);
1313   }
1314
1315 #ifdef USE_IMAP
1316   if (Context->magic == M_IMAP && imap_search (Context, pat) < 0)
1317     return -1;
1318 #endif
1319
1320   mutt_progress_init (&progress, _("Executing command on matching messages..."),
1321                       M_PROGRESS_MSG, ReadInc,
1322                       (op == M_LIMIT) ? Context->msgcount : Context->vcount);
1323
1324 #define THIS_BODY Context->hdrs[i]->content
1325
1326   if (op == M_LIMIT)
1327   {
1328     Context->vcount    = 0;
1329     Context->vsize     = 0;
1330     Context->collapsed = 0;
1331
1332     for (i = 0; i < Context->msgcount; i++)
1333     {
1334       mutt_progress_update (&progress, i, -1);
1335       /* new limit pattern implicitly uncollapses all threads */
1336       Context->hdrs[i]->virtual = -1;
1337       Context->hdrs[i]->limited = 0;
1338       Context->hdrs[i]->collapsed = 0;
1339       Context->hdrs[i]->num_hidden = 0;
1340       if (mutt_pattern_exec (pat, M_MATCH_FULL_ADDRESS, Context, Context->hdrs[i]))
1341       {
1342         Context->hdrs[i]->virtual = Context->vcount;
1343         Context->hdrs[i]->limited = 1;
1344         Context->v2r[Context->vcount] = i;
1345         Context->vcount++;
1346         Context->vsize+=THIS_BODY->length + THIS_BODY->offset -
1347           THIS_BODY->hdr_offset;
1348       }
1349     }
1350   }
1351   else
1352   {
1353     for (i = 0; i < Context->vcount; i++)
1354     {
1355       mutt_progress_update (&progress, i, -1);
1356       if (mutt_pattern_exec (pat, M_MATCH_FULL_ADDRESS, Context, Context->hdrs[Context->v2r[i]]))
1357       {
1358         switch (op)
1359         {
1360           case M_DELETE:
1361           case M_UNDELETE:
1362             mutt_set_flag (Context, Context->hdrs[Context->v2r[i]], M_DELETE,
1363                           (op == M_DELETE));
1364             break;
1365           case M_TAG:
1366           case M_UNTAG:
1367             mutt_set_flag (Context, Context->hdrs[Context->v2r[i]], M_TAG,
1368                            (op == M_TAG));
1369             break;
1370         }
1371       }
1372     }
1373   }
1374
1375 #undef THIS_BODY
1376
1377   mutt_clear_error ();
1378
1379   if (op == M_LIMIT)
1380   {
1381     /* drop previous limit pattern */
1382     FREE (&Context->pattern);
1383     if (Context->limit_pattern)
1384       mutt_pattern_free (&Context->limit_pattern);
1385
1386     if (Context->msgcount && !Context->vcount)
1387       mutt_error _("No messages matched criteria.");
1388
1389     /* record new limit pattern, unless match all */
1390     if (mutt_strcmp (buf, "~A") != 0)
1391     {
1392       Context->pattern = simple;
1393       simple = NULL; /* don't clobber it */
1394       Context->limit_pattern = mutt_pattern_comp (buf, M_FULL_MSG, &err);
1395     }
1396   }
1397   FREE (&simple);
1398   mutt_pattern_free (&pat);
1399   return 0;
1400 }
1401
1402 int mutt_search_command (int cur, int op)
1403 {
1404   int i, j;
1405   char buf[STRING];
1406   char temp[LONG_STRING];
1407   char error[STRING];
1408   int incr;
1409   HEADER *h;
1410   progress_t progress;
1411   const char* msg = NULL;
1412
1413   if (!*LastSearch || (op != OP_SEARCH_NEXT && op != OP_SEARCH_OPPOSITE))
1414   {
1415     strfcpy (buf, *LastSearch ? LastSearch : "", sizeof (buf));
1416     if (mutt_get_field ((op == OP_SEARCH || op == OP_SEARCH_NEXT) ?
1417                         _("Search for: ") : _("Reverse search for: "),
1418                         buf, sizeof (buf),
1419                       M_CLEAR | M_PATTERN) != 0 || !buf[0])
1420       return (-1);
1421
1422     if (op == OP_SEARCH || op == OP_SEARCH_NEXT)
1423       unset_option (OPTSEARCHREVERSE);
1424     else
1425       set_option (OPTSEARCHREVERSE);
1426
1427     /* compare the *expanded* version of the search pattern in case
1428        $simple_search has changed while we were searching */
1429     strfcpy (temp, buf, sizeof (temp));
1430     mutt_check_simple (temp, sizeof (temp), NONULL (SimpleSearch));
1431
1432     if (!SearchPattern || mutt_strcmp (temp, LastSearchExpn))
1433     {
1434       BUFFER err;
1435       memset(&err, 0, sizeof(err));
1436       set_option (OPTSEARCHINVALID);
1437       strfcpy (LastSearch, buf, sizeof (LastSearch));
1438       mutt_message _("Compiling search pattern...");
1439       mutt_pattern_free (&SearchPattern);
1440       err.data = error;
1441       err.dsize = sizeof (error);
1442       if ((SearchPattern = mutt_pattern_comp (temp, M_FULL_MSG, &err)) == NULL)
1443       {
1444         mutt_error ("%s", error);
1445         LastSearch[0] = '\0';
1446         return (-1);
1447       }
1448       mutt_clear_error ();
1449     }
1450   }
1451
1452   if (option (OPTSEARCHINVALID))
1453   {
1454     for (i = 0; i < Context->msgcount; i++)
1455       Context->hdrs[i]->searched = 0;
1456 #ifdef USE_IMAP
1457     if (Context->magic == M_IMAP && imap_search (Context, SearchPattern) < 0)
1458       return -1;
1459 #endif
1460     unset_option (OPTSEARCHINVALID);
1461   }
1462
1463   incr = (option (OPTSEARCHREVERSE)) ? -1 : 1;
1464   if (op == OP_SEARCH_OPPOSITE)
1465     incr = -incr;
1466
1467   mutt_progress_init (&progress, _("Searching..."), M_PROGRESS_MSG,
1468                       ReadInc, Context->vcount);
1469
1470   for (i = cur + incr, j = 0 ; j != Context->vcount; j++)
1471   {
1472     mutt_progress_update (&progress, j, -1);
1473     if (i > Context->vcount - 1)
1474     {
1475       i = 0;
1476       if (option (OPTWRAPSEARCH))
1477         msg = _("Search wrapped to top.");
1478       else
1479       {
1480         mutt_message _("Search hit bottom without finding match");
1481         return (-1);
1482       }
1483     }
1484     else if (i < 0)
1485     {
1486       i = Context->vcount - 1;
1487       if (option (OPTWRAPSEARCH))
1488         msg = _("Search wrapped to bottom.");
1489       else
1490       {
1491         mutt_message _("Search hit top without finding match");
1492         return (-1);
1493       }
1494     }
1495
1496     h = Context->hdrs[Context->v2r[i]];
1497     if (h->searched)
1498     {
1499       /* if we've already evaulated this message, use the cached value */
1500       if (h->matched)
1501       {
1502         mutt_clear_error();
1503         if (msg && *msg)
1504           mutt_message (msg);
1505         return i;
1506       }
1507     }
1508     else
1509     {
1510       /* remember that we've already searched this message */
1511       h->searched = 1;
1512       if ((h->matched = (mutt_pattern_exec (SearchPattern, M_MATCH_FULL_ADDRESS, Context, h) > 0)))
1513       {
1514         mutt_clear_error();
1515         if (msg && *msg)
1516           mutt_message (msg);
1517         return i;
1518       }
1519     }
1520
1521     if (SigInt)
1522     {
1523       mutt_error _("Search interrupted.");
1524       SigInt = 0;
1525       return (-1);
1526     }
1527
1528     i += incr;
1529   }
1530
1531   mutt_error _("Not found.");
1532   return (-1);
1533 }