citadel/imap_fetch.c

   1 /*
   2  * $Id$
   3  *
   4  * Implements the FETCH command in IMAP.
   5  * This command is way too convoluted.  Marc Crispin is a fscking idiot.
   6  *
   7  */
   8
   9
  10 #include "sysdep.h"
  11 #include <stdlib.h>
  12 #include <unistd.h>
  13 #include <stdio.h>
  14 #include <fcntl.h>
  15 #include <signal.h>
  16 #include <pwd.h>
  17 #include <errno.h>
  18 #include <sys/types.h>
  19 #include <sys/time.h>
  20 #include <sys/wait.h>
  21 #include <ctype.h>
  22 #include <string.h>
  23 #include <limits.h>
  24 #include "citadel.h"
  25 #include "server.h"
  26 #include <time.h>
  27 #include "sysdep_decls.h"
  28 #include "citserver.h"
  29 #include "support.h"
  30 #include "config.h"
  31 #include "dynloader.h"
  32 #include "room_ops.h"
  33 #include "user_ops.h"
  34 #include "policy.h"
  35 #include "database.h"
  36 #include "msgbase.h"
  37 #include "tools.h"
  38 #include "internet_addressing.h"
  39 #include "mime_parser.h"
  40 #include "serv_imap.h"
  41 #include "imap_tools.h"
  42 #include "imap_fetch.h"
  43 #include "genstamp.h"
  44
  45
  46
  47 struct imap_fetch_part {
  48         char desired_section[SIZ];
  49         FILE *output_fp;
  50 };
  51
  52 /*
  53  * Individual field functions for imap_do_fetch_msg() ...
  54  */
  55
  56
  57
  58 void imap_fetch_uid(int seq) {
  59         cprintf("UID %ld", IMAP->msgids[seq-1]);
  60 }
  61
  62 void imap_fetch_flags(struct CtdlMessage *msg) {
  63         cprintf("FLAGS ()");    /* FIXME do something here */
  64 }
  65
  66 void imap_fetch_internaldate(struct CtdlMessage *msg) {
  67         char buf[SIZ];
  68         time_t msgdate;
  69
  70         if (msg->cm_fields['T'] != NULL) {
  71                 msgdate = atol(msg->cm_fields['T']);
  72         }
  73         else {
  74                 msgdate = time(NULL);
  75         }
  76
  77         datestring(buf, msgdate, DATESTRING_IMAP);
  78         cprintf("INTERNALDATE \"%s\"", buf);
  79 }
  80
  81
  82 /*
  83  * Fetch RFC822-formatted messages.
  84  *
  85  * 'whichfmt' should be set to one of:
  86  *      "RFC822"        entire message
  87  *      "RFC822.HEADER" headers only (with trailing blank line)
  88  *      "RFC822.SIZE"   size of translated message
  89  *      "RFC822.TEXT"   body only (without leading blank line)
  90  */
  91 void imap_fetch_rfc822(int msgnum, char *whichfmt, struct CtdlMessage *msg) {
  92         FILE *tmp;
  93         char buf[1024];
  94         char *ptr;
  95         long headers_size, text_size, total_size;
  96         long bytes_remaining = 0;
  97         long blocksize;
  98
  99         tmp = tmpfile();
 100         if (tmp == NULL) {
 101                 lprintf(1, "Cannot open temp file: %s\n", strerror(errno));
 102                 return;
 103         }
 104
 105         /*
 106          * Load the message into a temp file for translation and measurement
 107          */
 108         CtdlRedirectOutput(tmp, -1);
 109         CtdlOutputPreLoadedMsg(msg, msgnum, MT_RFC822, 0, 0, 1);
 110         CtdlRedirectOutput(NULL, -1);
 111         if (!is_valid_message(msg)) {
 112                 lprintf(1, "WARNING: output clobbered the message!\n");
 113         }
 114
 115         /*
 116          * Now figure out where the headers/text break is.  IMAP considers the
 117          * intervening blank line to be part of the headers, not the text.
 118          */
 119         rewind(tmp);
 120         headers_size = 0L;
 121         do {
 122                 ptr = fgets(buf, sizeof buf, tmp);
 123                 if (ptr != NULL) {
 124                         striplt(buf);
 125                         if (strlen(buf) == 0) headers_size = ftell(tmp);
 126                 }
 127         } while ( (headers_size == 0L) && (ptr != NULL) );
 128         fseek(tmp, 0L, SEEK_END);
 129         total_size = ftell(tmp);
 130         text_size = total_size - headers_size;
 131
 132         if (!strcasecmp(whichfmt, "RFC822.SIZE")) {
 133                 cprintf("RFC822.SIZE %ld", total_size);
 134                 fclose(tmp);
 135                 return;
 136         }
 137
 138         else if (!strcasecmp(whichfmt, "RFC822")) {
 139                 bytes_remaining = total_size;
 140                 rewind(tmp);
 141         }
 142
 143         else if (!strcasecmp(whichfmt, "RFC822.HEADER")) {
 144                 bytes_remaining = headers_size;
 145                 rewind(tmp);
 146         }
 147
 148         else if (!strcasecmp(whichfmt, "RFC822.TEXT")) {
 149                 bytes_remaining = text_size;
 150                 fseek(tmp, headers_size, SEEK_SET);
 151         }
 152
 153         cprintf("%s {%ld}\r\n", whichfmt, bytes_remaining);
 154         blocksize = sizeof(buf);
 155         while (bytes_remaining > 0L) {
 156                 if (blocksize > bytes_remaining) blocksize = bytes_remaining;
 157                 fread(buf, blocksize, 1, tmp);
 158                 client_write(buf, blocksize);
 159                 bytes_remaining = bytes_remaining - blocksize;
 160         }
 161
 162         fclose(tmp);
 163 }
 164
 165
 166
 167 /*
 168  * Load a specific part of a message into the temp file to be output to a
 169  * client.  FIXME we can handle parts like "2" and "2.1" and even "2.MIME"
 170  * but we still can't handle "2.HEADER" (which might not be a problem, because
 171  * we currently don't have the ability to break out nested RFC822's anyway).
 172  *
 173  * Note: mime_parser() was called with dont_decode set to 1, so we have the
 174  * luxury of simply spewing without having to re-encode.
 175  */
 176 void imap_load_part(char *name, char *filename, char *partnum, char *disp,
 177                     void *content, char *cbtype, size_t length, char *encoding,
 178                     void *cbuserdata)
 179 {
 180         struct imap_fetch_part *imfp;
 181         char mbuf2[1024];
 182
 183         imfp = (struct imap_fetch_part *)cbuserdata;
 184
 185         if (!strcasecmp(partnum, imfp->desired_section)) {
 186                 fwrite(content, length, 1, imfp->output_fp);
 187         }
 188
 189         sprintf(mbuf2, "%s.MIME", partnum);
 190
 191         if (!strcasecmp(imfp->desired_section, mbuf2)) {
 192                 fprintf(imfp->output_fp, "Content-type: %s", cbtype);
 193                 if (strlen(name) > 0)
 194                         fprintf(imfp->output_fp, "; name=\"%s\"", name);
 195                 fprintf(imfp->output_fp, "\r\n");
 196                 if (strlen(encoding) > 0)
 197                         fprintf(imfp->output_fp,
 198                                 "Content-Transfer-Encoding: %s\r\n", encoding);
 199                 if (strlen(encoding) > 0) {
 200                         fprintf(imfp->output_fp, "Content-Disposition: %s",
 201                                         disp);
 202                         if (strlen(filename) > 0) {
 203                                 fprintf(imfp->output_fp, "; filename=\"%s\"",
 204                                         filename);
 205                         }
 206                         fprintf(imfp->output_fp, "\r\n");
 207                 }
 208                 fprintf(imfp->output_fp, "Content-Length: %d\r\n", length);
 209                 fprintf(imfp->output_fp, "\r\n");
 210         }
 211
 212
 213 }
 214
 215
 216 /*
 217  * Called by imap_fetch_envelope() to output the "From" field.
 218  * This is in its own function because its logic is kind of complex.  We
 219  * really need to make this suck less.
 220  */
 221 void imap_output_envelope_from(struct CtdlMessage *msg) {
 222         char user[1024], node[1024], name[1024];
 223
 224         cprintf("((");                          /* open double-parens */
 225         imap_strout(msg->cm_fields['A']);       /* personal name */
 226         cprintf(" NIL ");                       /* source route (not used) */
 227
 228         if (msg->cm_fields['F'] != NULL) {
 229                 process_rfc822_addr(msg->cm_fields['F'], user, node, name);
 230                 imap_strout(user);              /* mailbox name (user id) */
 231                 cprintf(" ");
 232                 if (!strcasecmp(node, config.c_nodename)) {
 233                         imap_strout(config.c_fqdn);
 234                 }
 235                 else {
 236                         imap_strout(node);              /* host name */
 237                 }
 238         }
 239         else {
 240                 imap_strout(msg->cm_fields['A']); /* mailbox name (user id) */
 241                 cprintf(" ");
 242                 imap_strout(msg->cm_fields['N']);       /* host name */
 243         }
 244
 245         cprintf(")) ");                         /* close double-parens */
 246 }
 247
 248
 249 /*
 250  * Implements the ENVELOPE fetch item
 251  *
 252  * FIXME ... we only output some of the fields right now.  Definitely need
 253  *           to do all of them.  Accurately, too.
 254  *
 255  * Note that the imap_strout() function can cleverly output NULL fields as NIL,
 256  * so we don't have to check for that condition like we do elsewhere.
 257  */
 258 void imap_fetch_envelope(long msgnum, struct CtdlMessage *msg) {
 259         char datestringbuf[SIZ];
 260         time_t msgdate;
 261         char *fieldptr = NULL;
 262
 263         /* Parse the message date into an IMAP-format date string */
 264         if (msg->cm_fields['T'] != NULL) {
 265                 msgdate = atol(msg->cm_fields['T']);
 266         }
 267         else {
 268                 msgdate = time(NULL);
 269         }
 270         datestring(datestringbuf, msgdate, DATESTRING_IMAP);
 271
 272         /* Now start spewing data fields.  The order is important, as it is
 273          * defined by the protocol specification.  Nonexistent fields must
 274          * be output as NIL, existent fields must be quoted or literalled.
 275          * The imap_strout() function conveniently does all this for us.
 276          */
 277         cprintf("ENVELOPE (");
 278
 279         /* Date */
 280         imap_strout(datestringbuf);
 281         cprintf(" ");
 282
 283         /* Subject */
 284         imap_strout(msg->cm_fields['U']);
 285         cprintf(" ");
 286
 287         /* From */
 288         imap_output_envelope_from(msg);
 289
 290         /* Sender */
 291         if (0) {
 292                 /* FIXME ... check for a *real* Sender: field */
 293         }
 294         else {
 295                 imap_output_envelope_from(msg);
 296         }
 297
 298         /* Reply-to */
 299         if (0) {
 300                 /* FIXME ... check for a *real* Reply-to: field */
 301         }
 302         else {
 303                 imap_output_envelope_from(msg);
 304         }
 305
 306         cprintf("NIL ");        /* to */
 307
 308         cprintf("NIL ");        /* cc */
 309
 310         cprintf("NIL ");        /* bcc */
 311
 312         /* In-reply-to */
 313         fieldptr = rfc822_fetch_field(msg->cm_fields['M'], "In-reply-to");
 314         imap_strout(fieldptr);
 315         cprintf(" ");
 316         if (fieldptr != NULL) phree(fieldptr);
 317
 318         /* message ID */
 319         imap_strout(msg->cm_fields['I']);
 320
 321         cprintf(") ");
 322 }
 323
 324
 325 /*
 326  * Strip any non header information out of a chunk of RFC822 data on disk
 327  */
 328 void imap_strip_headers(FILE *fp) {
 329         char buf[1024];
 330
 331         rewind(fp);
 332         while (fgets(buf, sizeof buf, fp) != NULL) {
 333                 striplt(buf);
 334                 if (strlen(buf) == 0) {
 335                         fflush(fp);
 336                         ftruncate(fileno(fp), ftell(fp));
 337                 }
 338         }
 339         fflush(fp);
 340         fprintf(fp, "\r\n");    /* add the trailing newline */
 341         rewind(fp);
 342 }
 343
 344
 345 /*
 346  * Implements the BODY and BODY.PEEK fetch items
 347  */
 348 void imap_fetch_body(long msgnum, char *item, int is_peek,
 349                 struct CtdlMessage *msg) {
 350         char section[1024];
 351         char partial[1024];
 352         int is_partial = 0;
 353         char buf[1024];
 354         int i;
 355         FILE *tmp;
 356         long bytes_remaining = 0;
 357         long blocksize;
 358         long pstart, pbytes;
 359         struct imap_fetch_part imfp;
 360
 361         /* extract section */
 362         strcpy(section, item);
 363         for (i=0; i<strlen(section); ++i) {
 364                 if (section[i]=='[') strcpy(section, &section[i+1]);
 365         }
 366         for (i=0; i<strlen(section); ++i) {
 367                 if (section[i]==']') section[i] = 0;
 368         }
 369         lprintf(9, "Section is %s\n", section);
 370
 371         /* extract partial */
 372         strcpy(partial, item);
 373         for (i=0; i<strlen(partial); ++i) {
 374                 if (partial[i]=='<') {
 375                         strcpy(partial, &partial[i+1]);
 376                         is_partial = 1;
 377                 }
 378         }
 379         for (i=0; i<strlen(partial); ++i) {
 380                 if (partial[i]=='>') partial[i] = 0;
 381         }
 382         if (is_partial == 0) strcpy(partial, "");
 383         if (strlen(partial) > 0) lprintf(9, "Partial is %s\n", partial);
 384
 385         tmp = tmpfile();
 386         if (tmp == NULL) {
 387                 lprintf(1, "Cannot open temp file: %s\n", strerror(errno));
 388                 return;
 389         }
 390
 391         /* Now figure out what the client wants, and get it */
 392
 393         if (!strcmp(section, "")) {             /* the whole thing */
 394                 CtdlRedirectOutput(tmp, -1);
 395                 CtdlOutputPreLoadedMsg(msg, msgnum, MT_RFC822, 0, 0, 1);
 396                 CtdlRedirectOutput(NULL, -1);
 397         }
 398
 399         /*
 400          * Be obnoxious and send the entire header, even if the client only
 401          * asks for certain fields.  FIXME this shortcut later.
 402          */
 403         else if (!strncasecmp(section, "HEADER", 6)) {
 404                 CtdlRedirectOutput(tmp, -1);
 405                 CtdlOutputPreLoadedMsg(msg, msgnum, MT_RFC822, 1, 0, 1);
 406                 CtdlRedirectOutput(NULL, -1);
 407                 imap_strip_headers(tmp);
 408         }
 409
 410         /*
 411          * Anything else must be a part specifier.
 412          * (Note value of 1 passed as 'dont_decode' so client gets it encoded)
 413          */
 414         else {
 415                 safestrncpy(imfp.desired_section, section,
 416                                 sizeof(imfp.desired_section));
 417                 imfp.output_fp = tmp;
 418
 419                 mime_parser(msg->cm_fields['M'], NULL,
 420                                 *imap_load_part, NULL, NULL,
 421                                 (void *)&imfp,
 422                                 1);
 423         }
 424
 425
 426         fseek(tmp, 0L, SEEK_END);
 427         bytes_remaining = ftell(tmp);
 428
 429         if (is_partial == 0) {
 430                 rewind(tmp);
 431                 cprintf("BODY[%s] {%ld}\r\n", section, bytes_remaining);
 432         }
 433         else {
 434                 sscanf(partial, "%ld.%ld", &pstart, &pbytes);
 435                 if ((bytes_remaining - pstart) < pbytes) {
 436                         pbytes = bytes_remaining - pstart;
 437                 }
 438                 fseek(tmp, pstart, SEEK_SET);
 439                 bytes_remaining = pbytes;
 440                 cprintf("BODY[%s] {%ld}<%ld>\r\n",
 441                         section, bytes_remaining, pstart);
 442         }
 443
 444         blocksize = sizeof(buf);
 445         while (bytes_remaining > 0L) {
 446                 if (blocksize > bytes_remaining) blocksize = bytes_remaining;
 447                 fread(buf, blocksize, 1, tmp);
 448                 client_write(buf, blocksize);
 449                 bytes_remaining = bytes_remaining - blocksize;
 450         }
 451
 452         fclose(tmp);
 453
 454         if (is_peek) {
 455                 /* FIXME set the last read pointer or something */
 456         }
 457 }
 458
 459 /*
 460  * Called immediately before outputting a multipart bodystructure
 461  */
 462 void imap_fetch_bodystructure_pre(
 463                 char *name, char *filename, char *partnum, char *disp,
 464                 void *content, char *cbtype, size_t length, char *encoding,
 465                 void *cbuserdata
 466                 ) {
 467
 468         cprintf("(");
 469 }
 470
 471
 472
 473 /*
 474  * Called immediately after outputting a multipart bodystructure
 475  */
 476 void imap_fetch_bodystructure_post(
 477                 char *name, char *filename, char *partnum, char *disp,
 478                 void *content, char *cbtype, size_t length, char *encoding,
 479                 void *cbuserdata
 480                 ) {
 481
 482         char subtype[SIZ];
 483
 484         extract_token(subtype, cbtype, 1, '/');
 485         imap_strout(subtype);
 486         cprintf(")");
 487 }
 488
 489
 490
 491 /*
 492  * Output the info for a MIME part in the format required by BODYSTRUCTURE.
 493  *
 494  * FIXME .... this needs oodles of work to get completed.
 495  */
 496 void imap_fetch_bodystructure_part(
 497                 char *name, char *filename, char *partnum, char *disp,
 498                 void *content, char *cbtype, size_t length, char *encoding,
 499                 void *cbuserdata
 500                 ) {
 501
 502         char buf[SIZ];
 503         int have_cbtype = 0;
 504         int have_encoding = 0;
 505
 506         cprintf("(");
 507
 508         if (cbtype != NULL) if (strlen(cbtype)>0) have_cbtype = 1;
 509
 510         if (have_cbtype) {
 511                 extract_token(buf, cbtype, 0, '/');
 512                 imap_strout(buf);
 513                 cprintf(" ");
 514                 extract_token(buf, cbtype, 1, '/');
 515                 imap_strout(buf);
 516                 cprintf(" ");
 517         }
 518         else {
 519                 cprintf("\"TEXT\" \"PLAIN\" ");
 520         }
 521
 522         cprintf("(\"CHARSET\" \"US-ASCII\"");
 523
 524         if (name != NULL) if (strlen(name)>0) {
 525                 cprintf(" \"NAME\" ");
 526                 imap_strout(name);
 527         }
 528
 529         if (filename != NULL) if (strlen(filename)>0) {
 530                 cprintf(" \"FILENAME\" ");
 531                 imap_strout(name);
 532         }
 533
 534         cprintf(") ");
 535
 536         cprintf("NIL NIL ");
 537
 538         if (encoding != NULL) if (strlen(encoding) > 0)  have_encoding = 1;
 539
 540         if (have_encoding) {
 541                 imap_strout(encoding);
 542         }
 543         else {
 544                 imap_strout("7BIT");
 545         }
 546         cprintf(" ");
 547
 548         cprintf("%ld ", length);        /* bytes */
 549         cprintf("NIL ");                /* lines */
 550         cprintf("NIL NIL NIL)");        /* I don't know what this is */
 551
 552 }
 553
 554
 555
 556 /*
 557  * Spew the BODYSTRUCTURE data for a message.  (Do you need a silencer if
 558  * you're going to shoot a MIME?  Do you need a reason to shoot Mark Crispin?
 559  * No, and no.)
 560  *
 561  * FIXME finish the implementation
 562  */
 563 void imap_fetch_bodystructure (long msgnum, char *item,
 564                 struct CtdlMessage *msg) {
 565         FILE *tmp;
 566         char buf[1024];
 567         long lines = 0L;
 568         long bytes = 0L;
 569
 570
 571         /* For non-RFC822 (ordinary Citadel) messages, this is short and
 572          * sweet...
 573          */
 574         if (msg->cm_format_type != FMT_RFC822) {
 575
 576                 /* *sigh* We have to RFC822-format the message just to be able
 577                  * to measure it.
 578                  */
 579                 tmp = tmpfile();
 580                 if (tmp == NULL) return;
 581                 CtdlRedirectOutput(tmp, -1);
 582                 CtdlOutputPreLoadedMsg(msg, msgnum, MT_RFC822, 0, 0, 1);
 583                 CtdlRedirectOutput(NULL, -1);
 584
 585                 rewind(tmp);
 586                 while (fgets(buf, sizeof buf, tmp) != NULL) ++lines;
 587                 bytes = ftell(tmp);
 588                 fclose(tmp);
 589
 590                 cprintf("BODYSTRUCTURE (\"TEXT\" \"PLAIN\" "
 591                         "(\"CHARSET\" \"US-ASCII\") NIL NIL "
 592                         "\"7BIT\" %ld %ld)", bytes, lines);
 593
 594                 return;
 595         }
 596
 597         /* For messages already stored in RFC822 format, we have to parse. */
 598         cprintf("BODYSTRUCTURE ");
 599         mime_parser(msg->cm_fields['M'],
 600                         NULL,
 601                         *imap_fetch_bodystructure_part, /* part */
 602                         *imap_fetch_bodystructure_pre,  /* pre-multi */
 603                         *imap_fetch_bodystructure_post, /* post-multi */
 604                         NULL,
 605                         0);
 606 }
 607
 608
 609
 610
 611
 612
 613 /*
 614  * imap_do_fetch() calls imap_do_fetch_msg() to output the deta of an
 615  * individual message, once it has been successfully loaded from disk.
 616  */
 617 void imap_do_fetch_msg(int seq, struct CtdlMessage *msg,
 618                         int num_items, char **itemlist) {
 619         int i;
 620
 621         cprintf("* %d FETCH (", seq);
 622
 623         for (i=0; i<num_items; ++i) {
 624
 625                 if (!strncasecmp(itemlist[i], "BODY[", 5)) {
 626                         imap_fetch_body(IMAP->msgids[seq-1], itemlist[i],
 627                                         0, msg);
 628                 }
 629                 else if (!strncasecmp(itemlist[i], "BODY.PEEK[", 10)) {
 630                         imap_fetch_body(IMAP->msgids[seq-1], itemlist[i],
 631                                         1, msg);
 632                 }
 633                 else if (!strcasecmp(itemlist[i], "BODYSTRUCTURE")) {
 634                         imap_fetch_bodystructure(IMAP->msgids[seq-1],
 635                                         itemlist[i], msg);
 636                 }
 637                 else if (!strcasecmp(itemlist[i], "ENVELOPE")) {
 638                         imap_fetch_envelope(IMAP->msgids[seq-1], msg);
 639                 }
 640                 else if (!strcasecmp(itemlist[i], "FLAGS")) {
 641                         imap_fetch_flags(msg);
 642                 }
 643                 else if (!strcasecmp(itemlist[i], "INTERNALDATE")) {
 644                         imap_fetch_internaldate(msg);
 645                 }
 646                 else if (!strcasecmp(itemlist[i], "RFC822")) {
 647                         imap_fetch_rfc822(IMAP->msgids[seq-1], itemlist[i], msg);
 648                 }
 649                 else if (!strcasecmp(itemlist[i], "RFC822.HEADER")) {
 650                         imap_fetch_rfc822(IMAP->msgids[seq-1], itemlist[i], msg);
 651                 }
 652                 else if (!strcasecmp(itemlist[i], "RFC822.SIZE")) {
 653                         imap_fetch_rfc822(IMAP->msgids[seq-1], itemlist[i], msg);
 654                 }
 655                 else if (!strcasecmp(itemlist[i], "RFC822.TEXT")) {
 656                         imap_fetch_rfc822(IMAP->msgids[seq-1], itemlist[i], msg);
 657                 }
 658                 else if (!strcasecmp(itemlist[i], "UID")) {
 659                         imap_fetch_uid(seq);
 660                 }
 661
 662                 if (i != num_items-1) cprintf(" ");
 663         }
 664
 665         cprintf(")\r\n");
 666 }
 667
 668
 669
 670 /*
 671  * imap_fetch() calls imap_do_fetch() to do its actual work, once it's
 672  * validated and boiled down the request a bit.
 673  */
 674 void imap_do_fetch(int num_items, char **itemlist) {
 675         int i;
 676         struct CtdlMessage *msg;
 677
 678         if (IMAP->num_msgs > 0)
 679          for (i = 0; i < IMAP->num_msgs; ++i)
 680           if (IMAP->flags[i] && IMAP_FETCHED) {
 681                 msg = CtdlFetchMessage(IMAP->msgids[i]);
 682                 if (msg != NULL) {
 683                         imap_do_fetch_msg(i+1, msg, num_items, itemlist);
 684                         CtdlFreeMessage(msg);
 685                 }
 686                 else {
 687                         cprintf("* %d FETCH <internal error>\r\n", i+1);
 688                 }
 689         }
 690 }
 691
 692
 693
 694 /*
 695  * Back end for imap_handle_macros()
 696  * Note that this function *only* looks at the beginning of the string.  It
 697  * is not a generic search-and-replace function.
 698  */
 699 void imap_macro_replace(char *str, char *find, char *replace) {
 700         char holdbuf[1024];
 701
 702         if (!strncasecmp(str, find, strlen(find))) {
 703                 if (str[strlen(find)]==' ') {
 704                         strcpy(holdbuf, &str[strlen(find)+1]);
 705                         strcpy(str, replace);
 706                         strcat(str, " ");
 707                         strcat(str, holdbuf);
 708                 }
 709                 if (str[strlen(find)]==0) {
 710                         strcpy(holdbuf, &str[strlen(find)+1]);
 711                         strcpy(str, replace);
 712                 }
 713         }
 714 }
 715
 716
 717
 718 /*
 719  * Handle macros embedded in FETCH data items.
 720  * (What the heck are macros doing in a wire protocol?  Are we trying to save
 721  * the computer at the other end the trouble of typing a lot of characters?)
 722  */
 723 void imap_handle_macros(char *str) {
 724         int i;
 725         int nest = 0;
 726
 727         for (i=0; i<strlen(str); ++i) {
 728                 if (str[i]=='(') ++nest;
 729                 if (str[i]=='[') ++nest;
 730                 if (str[i]=='<') ++nest;
 731                 if (str[i]=='{') ++nest;
 732                 if (str[i]==')') --nest;
 733                 if (str[i]==']') --nest;
 734                 if (str[i]=='>') --nest;
 735                 if (str[i]=='}') --nest;
 736
 737                 if (nest <= 0) {
 738                         imap_macro_replace(&str[i],
 739                                 "ALL",
 740                                 "FLAGS INTERNALDATE RFC822.SIZE ENVELOPE"
 741                         );
 742                         imap_macro_replace(&str[i],
 743                                 "BODY",
 744                                 "BODYSTRUCTURE"
 745                         );
 746                         imap_macro_replace(&str[i],
 747                                 "FAST",
 748                                 "FLAGS INTERNALDATE RFC822.SIZE"
 749                         );
 750                         imap_macro_replace(&str[i],
 751                                 "FULL",
 752                                 "FLAGS INTERNALDATE RFC822.SIZE ENVELOPE BODY"
 753                         );
 754                 }
 755         }
 756 }
 757
 758
 759 /*
 760  * Break out the data items requested, possibly a parenthesized list.
 761  * Returns the number of data items, or -1 if the list is invalid.
 762  * NOTE: this function alters the string it is fed, and uses it as a buffer
 763  * to hold the data for the pointers it returns.
 764  */
 765 int imap_extract_data_items(char **argv, char *items) {
 766         int num_items = 0;
 767         int nest = 0;
 768         int i, initial_len;
 769         char *start;
 770
 771         /* Convert all whitespace to ordinary space characters. */
 772         for (i=0; i<strlen(items); ++i) {
 773                 if (isspace(items[i])) items[i]=' ';
 774         }
 775
 776         /* Strip leading and trailing whitespace, then strip leading and
 777          * trailing parentheses if it's a list
 778          */
 779         striplt(items);
 780         if ( (items[0]=='(') && (items[strlen(items)-1]==')') ) {
 781                 items[strlen(items)-1] = 0;
 782                 strcpy(items, &items[1]);
 783                 striplt(items);
 784         }
 785
 786         /* Parse any macro data items */
 787         imap_handle_macros(items);
 788
 789         /*
 790          * Now break out the data items.  We throw in one trailing space in
 791          * order to avoid having to break out the last one manually.
 792          */
 793         strcat(items, " ");
 794         start = items;
 795         initial_len = strlen(items);
 796         for (i=0; i<initial_len; ++i) {
 797                 if (items[i]=='(') ++nest;
 798                 if (items[i]=='[') ++nest;
 799                 if (items[i]=='<') ++nest;
 800                 if (items[i]=='{') ++nest;
 801                 if (items[i]==')') --nest;
 802                 if (items[i]==']') --nest;
 803                 if (items[i]=='>') --nest;
 804                 if (items[i]=='}') --nest;
 805
 806                 if (nest <= 0) if (items[i]==' ') {
 807                         items[i] = 0;
 808                         argv[num_items++] = start;
 809                         start = &items[i+1];
 810                 }
 811         }
 812
 813         return(num_items);
 814
 815 }
 816
 817
 818 /*
 819  * One particularly hideous aspect of IMAP is that we have to allow the client
 820  * to specify arbitrary ranges and/or sets of messages to fetch.  Citadel IMAP
 821  * handles this by setting the IMAP_FETCHED flag for each message specified in
 822  * the ranges/sets, then looping through the message array, outputting messages
 823  * with the flag set.  We don't bother returning an error if an out-of-range
 824  * number is specified (we just return quietly) because any client braindead
 825  * enough to request a bogus message number isn't going to notice the
 826  * difference anyway.
 827  *
 828  * This function clears out the IMAP_FETCHED bits, then sets that bit for each
 829  * message included in the specified range.
 830  *
 831  * Set is_uid to 1 to fetch by UID instead of sequence number.
 832  */
 833 void imap_pick_range(char *range, int is_uid) {
 834         int i;
 835         int num_sets;
 836         int s;
 837         char setstr[SIZ], lostr[SIZ], histr[SIZ];       /* was 1024 */
 838         int lo, hi;
 839         char *actual_range;
 840
 841         actual_range = range;
 842         if (!strcasecmp(range, "ALL")) actual_range = "1:*";
 843
 844         /*
 845          * Clear out the IMAP_FETCHED flags for all messages.
 846          */
 847         for (i = 1; i <= IMAP->num_msgs; ++i) {
 848                 IMAP->flags[i-1] = IMAP->flags[i-1] & ~IMAP_FETCHED;
 849         }
 850
 851         /*
 852          * Now set it for all specified messages.
 853          */
 854         num_sets = num_tokens(actual_range, ',');
 855         for (s=0; s<num_sets; ++s) {
 856                 extract_token(setstr, actual_range, s, ',');
 857
 858                 extract_token(lostr, setstr, 0, ':');
 859                 if (num_tokens(setstr, ':') >= 2) {
 860                         extract_token(histr, setstr, 1, ':');
 861                         if (!strcmp(histr, "*")) sprintf(histr, "%d", INT_MAX);
 862                 }
 863                 else {
 864                         strcpy(histr, lostr);
 865                 }
 866                 lo = atoi(lostr);
 867                 hi = atoi(histr);
 868
 869                 /* Loop through the array, flipping bits where appropriate */
 870                 for (i = 1; i <= IMAP->num_msgs; ++i) {
 871                         if (is_uid) {   /* fetch by sequence number */
 872                                 if ( (IMAP->msgids[i-1]>=lo)
 873                                    && (IMAP->msgids[i-1]<=hi)) {
 874                                         IMAP->flags[i-1] =
 875                                                 IMAP->flags[i-1] | IMAP_FETCHED;
 876                                 }
 877                         }
 878                         else {          /* fetch by uid */
 879                                 if ( (i>=lo) && (i<=hi)) {
 880                                         IMAP->flags[i-1] =
 881                                                 IMAP->flags[i-1] | IMAP_FETCHED;
 882                                 }
 883                         }
 884                 }
 885         }
 886 }
 887
 888
 889
 890 /*
 891  * This function is called by the main command loop.
 892  */
 893 void imap_fetch(int num_parms, char *parms[]) {
 894         char items[SIZ];        /* was 1024 */
 895         char *itemlist[SIZ];
 896         int num_items;
 897         int i;
 898
 899         if (num_parms < 4) {
 900                 cprintf("%s BAD invalid parameters\r\n", parms[0]);
 901                 return;
 902         }
 903
 904         imap_pick_range(parms[2], 0);
 905
 906         strcpy(items, "");
 907         for (i=3; i<num_parms; ++i) {
 908                 strcat(items, parms[i]);
 909                 if (i < (num_parms-1)) strcat(items, " ");
 910         }
 911
 912         num_items = imap_extract_data_items(itemlist, items);
 913         if (num_items < 1) {
 914                 cprintf("%s BAD invalid data item list\r\n", parms[0]);
 915                 return;
 916         }
 917
 918         imap_do_fetch(num_items, itemlist);
 919         cprintf("%s OK FETCH completed\r\n", parms[0]);
 920 }
 921
 922 /*
 923  * This function is called by the main command loop.
 924  */
 925 void imap_uidfetch(int num_parms, char *parms[]) {
 926         char items[SIZ];        /* was 1024 */
 927         char *itemlist[SIZ];
 928         int num_items;
 929         int i;
 930         int have_uid_item = 0;
 931
 932         if (num_parms < 5) {
 933                 cprintf("%s BAD invalid parameters\r\n", parms[0]);
 934                 return;
 935         }
 936
 937         imap_pick_range(parms[3], 1);
 938
 939         strcpy(items, "");
 940         for (i=4; i<num_parms; ++i) {
 941                 strcat(items, parms[i]);
 942                 if (i < (num_parms-1)) strcat(items, " ");
 943         }
 944
 945         num_items = imap_extract_data_items(itemlist, items);
 946         if (num_items < 1) {
 947                 cprintf("%s BAD invalid data item list\r\n", parms[0]);
 948                 return;
 949         }
 950
 951         /* If the "UID" item was not included, we include it implicitly
 952          * because this is a UID FETCH command
 953          */
 954         for (i=0; i<num_items; ++i) {
 955                 if (!strcasecmp(itemlist[i], "UID")) ++have_uid_item;
 956         }
 957         if (have_uid_item == 0) itemlist[num_items++] = "UID";
 958
 959         imap_do_fetch(num_items, itemlist);
 960         cprintf("%s OK UID FETCH completed\r\n", parms[0]);
 961 }
 962
 963