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