scroll list when moving rows up or down
[claws.git] / src / imap.c
1 /*
2  * Sylpheed -- a GTK+ based, lightweight, and fast e-mail client
3  * Copyright (C) 1999-2001 Hiroyuki Yamamoto
4  *
5  * This program is free software; you can redistribute it and/or modify
6  * it under the terms of the GNU General Public License as published by
7  * the Free Software Foundation; either version 2 of the License, or
8  * (at your option) any later version.
9  *
10  * This program is distributed in the hope that it will be useful,
11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13  * GNU General Public License for more details.
14  *
15  * You should have received a copy of the GNU General Public License
16  * along with this program; if not, write to the Free Software
17  * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
18  */
19
20 #ifdef HAVE_CONFIG_H
21 #  include "config.h"
22 #endif
23
24 #include <glib.h>
25 #include <stdio.h>
26 #include <string.h>
27 #include <stdlib.h>
28 #include <dirent.h>
29 #include <unistd.h>
30 #include <ctype.h>
31
32 #include "intl.h"
33 #include "imap.h"
34 #include "socket.h"
35 #include "recv.h"
36 #include "procmsg.h"
37 #include "procheader.h"
38 #include "folder.h"
39 #include "statusbar.h"
40 #include "prefs_account.h"
41 #include "codeconv.h"
42 #include "utils.h"
43 #include "inputdialog.h"
44
45 #define IMAP4_PORT      143
46
47 static GList *session_list = NULL;
48
49 static gint imap_cmd_count = 0;
50
51 static IMAPSession *imap_session_get    (Folder         *folder);
52 static gchar *imap_query_password       (const gchar    *server,
53                                          const gchar    *user);
54
55 static void imap_scan_tree_recursive    (IMAPSession    *session,
56                                          FolderItem     *item,
57                                          IMAPNameSpace  *namespace);
58 static GSList *imap_parse_list          (IMAPSession    *session,
59                                          const gchar    *path);
60 static gint imap_create_trash           (Folder         *folder);
61
62 static gint imap_do_copy                (Folder         *folder,
63                                          FolderItem     *dest,
64                                          MsgInfo        *msginfo,
65                                          gboolean        remove_source);
66 static gint imap_do_copy_msgs_with_dest (Folder         *folder,
67                                          FolderItem     *dest, 
68                                          GSList         *msglist,
69                                          gboolean        remove_source);
70
71 static GSList *imap_get_uncached_messages       (IMAPSession    *session,
72                                                  FolderItem     *item,
73                                                  guint32         first_uid,
74                                                  guint32         last_uid);
75 static GSList *imap_delete_cached_messages      (GSList         *mlist,
76                                                  FolderItem     *item,
77                                                  guint32         first_uid,
78                                                  guint32         last_uid);
79 static void imap_delete_all_cached_messages     (FolderItem     *item);
80
81 static SockInfo *imap_open              (const gchar    *server,
82                                          gushort         port,
83                                          gchar          *buf);
84
85 static gint imap_set_message_flags      (IMAPSession    *session,
86                                          guint32         first_uid,
87                                          guint32         last_uid,
88                                          IMAPFlags       flag,
89                                          gboolean        is_set);
90 static gint imap_select                 (IMAPSession    *session,
91                                          IMAPFolder     *folder,
92                                          const gchar    *path,
93                                          gint           *exists,
94                                          gint           *recent,
95                                          gint           *unseen,
96                                          guint32        *uid_validity);
97 static gint imap_get_uid                (IMAPSession    *session,
98                                          gint            msgnum,
99                                          guint32        *uid);
100 static gint imap_status                 (IMAPSession    *session,
101                                          IMAPFolder     *folder,
102                                          const gchar    *path,
103                                          gint           *messages,
104                                          gint           *recent,
105                                          gint           *unseen,
106                                          guint32        *uid_validity);
107
108 static void imap_parse_namespace                (IMAPSession    *session,
109                                                  IMAPFolder     *folder);
110 static IMAPNameSpace *imap_find_namespace       (IMAPFolder     *folder,
111                                                  const gchar    *path);
112 static gchar *imap_get_real_path                (IMAPFolder     *folder,
113                                                  const gchar    *path);
114
115 static gchar *imap_parse_atom           (SockInfo       *sock,
116                                          gchar          *src,
117                                          gchar          *dest,
118                                          gint            dest_len,
119                                          GString        *str);
120 static gchar *imap_parse_one_address    (SockInfo       *sock,
121                                          gchar          *start,
122                                          gchar          *out_from_str,
123                                          gchar          *out_fromname_str,
124                                          GString        *str);
125 static gchar *imap_parse_address        (SockInfo       *sock,
126                                          gchar          *start,
127                                          gchar         **out_from_str,
128                                          gchar         **out_fromname_str,
129                                          GString        *str);
130 static MsgFlags imap_parse_flags        (const gchar    *flag_str);
131 static MsgInfo *imap_parse_envelope     (SockInfo       *sock,
132                                          GString        *line_str);
133
134 /* low-level IMAP4rev1 commands */
135 static gint imap_cmd_login      (SockInfo       *sock,
136                                  const gchar    *user,
137                                  const gchar    *pass);
138 static gint imap_cmd_logout     (SockInfo       *sock);
139 static gint imap_cmd_noop       (SockInfo       *sock);
140 static gint imap_cmd_namespace  (SockInfo       *sock,
141                                  gchar         **ns_str);
142 static gint imap_cmd_list       (SockInfo       *sock,
143                                  const gchar    *ref,
144                                  const gchar    *mailbox,
145                                  GPtrArray      *argbuf);
146 static gint imap_cmd_do_select  (SockInfo       *sock,
147                                  const gchar    *folder,
148                                  gboolean        examine,
149                                  gint           *exists,
150                                  gint           *recent,
151                                  gint           *unseen,
152                                  guint32        *uid_validity);
153 static gint imap_cmd_select     (SockInfo       *sock,
154                                  const gchar    *folder,
155                                  gint           *exists,
156                                  gint           *recent,
157                                  gint           *unseen,
158                                  guint32        *uid_validity);
159 static gint imap_cmd_examine    (SockInfo       *sock,
160                                  const gchar    *folder,
161                                  gint           *exists,
162                                  gint           *recent,
163                                  gint           *unseen,
164                                  guint32        *uid_validity);
165 static gint imap_cmd_create     (SockInfo       *sock,
166                                  const gchar    *folder);
167 static gint imap_cmd_delete     (SockInfo       *sock,
168                                  const gchar    *folder);
169 static gint imap_cmd_envelope   (SockInfo       *sock,
170                                  guint32         first_uid,
171                                  guint32         last_uid);
172 #if 0
173 static gint imap_cmd_search     (SockInfo       *sock,
174                                  GSList         *numlist);
175 #endif
176 static gint imap_cmd_fetch      (SockInfo       *sock,
177                                  guint32         uid,
178                                  const gchar    *filename);
179 static gint imap_cmd_append     (SockInfo       *sock,
180                                  const gchar    *destfolder,
181                                  const gchar    *file);
182 static gint imap_cmd_copy       (SockInfo       *sock,
183                                  guint32         uid,
184                                  const gchar    *destfolder);
185 static gint imap_cmd_store      (SockInfo       *sock,
186                                  guint32         first_uid,
187                                  guint32         last_uid,
188                                  gchar          *sub_cmd);
189 static gint imap_cmd_expunge    (SockInfo       *sock);
190
191 static gint imap_cmd_ok         (SockInfo       *sock,
192                                  GPtrArray      *argbuf);
193 static void imap_cmd_gen_send   (SockInfo       *sock,
194                                  const gchar    *format, ...);
195 static gint imap_cmd_gen_recv   (SockInfo       *sock,
196                                  gchar          *buf,
197                                  gint            size);
198
199 /* misc utility functions */
200 static gchar *strchr_cpy                        (const gchar    *src,
201                                                  gchar           ch,
202                                                  gchar          *dest,
203                                                  gint            len);
204 static gchar *search_array_contain_str          (GPtrArray      *array,
205                                                  gchar          *str);
206 static void imap_path_separator_subst           (gchar          *str,
207                                                  gchar           separator);
208
209 static IMAPSession *imap_session_get(Folder *folder)
210 {
211         RemoteFolder *rfolder = REMOTE_FOLDER(folder);
212         gushort port;
213
214         g_return_val_if_fail(folder != NULL, NULL);
215         g_return_val_if_fail(folder->type == F_IMAP, NULL);
216         g_return_val_if_fail(folder->account != NULL, NULL);
217
218         port = folder->account->set_imapport ? folder->account->imapport
219                 : IMAP4_PORT;
220
221         if (!rfolder->session) {
222                 rfolder->session =
223                         imap_session_new(folder->account->recv_server, port,
224                                          folder->account->userid,
225                                          folder->account->passwd);
226                 if (rfolder->session)
227                         imap_parse_namespace(IMAP_SESSION(rfolder->session),
228                                              IMAP_FOLDER(folder));
229                 statusbar_pop_all();
230                 return IMAP_SESSION(rfolder->session);
231         }
232
233         if (imap_cmd_noop(rfolder->session->sock) != IMAP_SUCCESS) {
234                 log_warning(_("IMAP4 connection to %s:%d has been"
235                               " disconnected. Reconnecting...\n"),
236                             folder->account->recv_server, port);
237                 session_destroy(rfolder->session);
238                 rfolder->session =
239                         imap_session_new(folder->account->recv_server, port,
240                                          folder->account->userid,
241                                          folder->account->passwd);
242                 if (rfolder->session)
243                         imap_parse_namespace(IMAP_SESSION(rfolder->session),
244                                              IMAP_FOLDER(folder));
245         }
246
247         statusbar_pop_all();
248         return IMAP_SESSION(rfolder->session);
249 }
250
251 static gchar *imap_query_password(const gchar *server, const gchar *user)
252 {
253         gchar *message;
254         gchar *pass;
255
256         message = g_strdup_printf(_("Input password for %s on %s:"),
257                                   user, server);
258         pass = input_dialog_with_invisible(_("Input password"), message, NULL);
259         g_free(message);
260
261         return pass;
262 }
263
264 Session *imap_session_new(const gchar *server, gushort port,
265                           const gchar *user, const gchar *pass)
266 {
267         gchar buf[IMAPBUFSIZE];
268         IMAPSession *session;
269         SockInfo *imap_sock;
270
271         g_return_val_if_fail(server != NULL, NULL);
272         g_return_val_if_fail(user != NULL, NULL);
273
274         if (!pass) {
275                 gchar *tmp_pass;
276                 tmp_pass = imap_query_password(server, user);
277                 if (!tmp_pass)
278                         return NULL;
279                 Xstrdup_a(pass, tmp_pass, {g_free(tmp_pass); return NULL;});
280                 g_free(tmp_pass);
281         }
282
283         log_message(_("creating IMAP4 connection to %s:%d ...\n"),
284                     server, port);
285
286         if ((imap_sock = imap_open(server, port, buf)) == NULL)
287                 return NULL;
288         if (imap_cmd_login(imap_sock, user, pass) != IMAP_SUCCESS) {
289                 imap_cmd_logout(imap_sock);
290                 sock_close(imap_sock);
291                 return NULL;
292         }
293
294         session = g_new(IMAPSession, 1);
295         SESSION(session)->type      = SESSION_IMAP;
296         SESSION(session)->server    = g_strdup(server);
297         SESSION(session)->sock      = imap_sock;
298         SESSION(session)->connected = TRUE;
299         SESSION(session)->phase     = SESSION_READY;
300         SESSION(session)->data      = NULL;
301         session->mbox = NULL;
302
303         session_list = g_list_append(session_list, session);
304
305         return SESSION(session);
306 }
307
308 void imap_session_destroy(IMAPSession *session)
309 {
310         sock_close(SESSION(session)->sock);
311         SESSION(session)->sock = NULL;
312
313         g_free(session->mbox);
314
315         session_list = g_list_remove(session_list, session);
316 }
317
318 void imap_session_destroy_all(void)
319 {
320         while (session_list != NULL) {
321                 IMAPSession *session = (IMAPSession *)session_list->data;
322
323                 imap_cmd_logout(SESSION(session)->sock);
324                 imap_session_destroy(session);
325         }
326 }
327
328 #define THROW goto catch
329
330 GSList *imap_get_msg_list(Folder *folder, FolderItem *item, gboolean use_cache)
331 {
332         GSList *mlist = NULL;
333         IMAPSession *session;
334         gint ok, exists = 0, recent = 0, unseen = 0;
335         guint32 uid_validity = 0;
336         guint32 first_uid = 0, last_uid = 0, begin;
337
338         g_return_val_if_fail(folder != NULL, NULL);
339         g_return_val_if_fail(item != NULL, NULL);
340         g_return_val_if_fail(folder->type == F_IMAP, NULL);
341         g_return_val_if_fail(folder->account != NULL, NULL);
342
343         session = imap_session_get(folder);
344
345         if (!session) {
346                 mlist = procmsg_read_cache(item, FALSE);
347                 item->last_num = procmsg_get_last_num_in_cache(mlist);
348                 procmsg_set_flags(mlist, item);
349                 statusbar_pop_all();
350                 return mlist;
351         }
352
353         ok = imap_select(session, IMAP_FOLDER(folder), item->path,
354                          &exists, &recent, &unseen, &uid_validity);
355         if (ok != IMAP_SUCCESS) THROW;
356         if (exists > 0) {
357                 ok = imap_get_uid(session, 1, &first_uid);
358                 if (ok != IMAP_SUCCESS) THROW;
359                 if (1 != exists) {
360                         ok = imap_get_uid(session, exists, &last_uid);
361                         if (ok != IMAP_SUCCESS) THROW;
362                 } else
363                         last_uid = first_uid;
364         } else {
365                 imap_delete_all_cached_messages(item);
366                 statusbar_pop_all();
367                 return NULL;
368         }
369
370         if (use_cache) {
371                 guint32 cache_last;
372
373                 mlist = procmsg_read_cache(item, FALSE);
374                 procmsg_set_flags(mlist, item);
375                 cache_last = procmsg_get_last_num_in_cache(mlist);
376
377                 /* calculating the range of envelope to get */
378                 if (item->mtime != uid_validity) {
379                         /* mailbox is changed (get all) */
380                         begin = first_uid;
381                 } else if (last_uid < cache_last) {
382                         /* mailbox is changed (get all) */
383                         begin = first_uid;
384                 } else if (last_uid == cache_last) {
385                         /* mailbox unchanged (get none)*/
386                         begin = 0;
387                 } else {
388                         begin = cache_last + 1;
389                 }
390
391                 item->mtime = uid_validity;
392
393                 if (first_uid > 0 && last_uid > 0) {
394                         mlist = imap_delete_cached_messages(mlist, item,
395                                                             0, first_uid - 1);
396                         mlist = imap_delete_cached_messages(mlist, item,
397                                                             last_uid + 1,
398                                                             UINT_MAX);
399                 }
400                 if (begin > 0)
401                         mlist = imap_delete_cached_messages(mlist, item,
402                                                             begin, UINT_MAX);
403         } else {
404                 imap_delete_all_cached_messages(item);
405                 begin = first_uid;
406         }
407
408         if (begin > 0 && begin <= last_uid) {
409                 GSList *newlist;
410                 newlist = imap_get_uncached_messages(session, item,
411                                                      begin, last_uid);
412                 mlist = g_slist_concat(mlist, newlist);
413         }
414
415         item->last_num = last_uid;
416
417 catch:
418         statusbar_pop_all();
419         return mlist;
420 }
421
422 #undef THROW
423
424 gchar *imap_fetch_msg(Folder *folder, FolderItem *item, gint uid)
425 {
426         gchar *path, *filename;
427         IMAPSession *session;
428         gint ok;
429
430         g_return_val_if_fail(folder != NULL, NULL);
431         g_return_val_if_fail(item != NULL, NULL);
432
433         path = folder_item_get_path(item);
434         if (!is_dir_exist(path))
435                 make_dir_hier(path);
436         filename = g_strconcat(path, G_DIR_SEPARATOR_S, itos(uid), NULL);
437         g_free(path);
438  
439         if (is_file_exist(filename)) {
440                 debug_print(_("message %d has been already cached.\n"), uid);
441                 return filename;
442         }
443
444         session = imap_session_get(folder);
445         if (!session) {
446                 g_free(filename);
447                 return NULL;
448         }
449
450         debug_print(_("getting message %d...\n"), uid);
451         ok = imap_cmd_fetch(SESSION(session)->sock, (guint32)uid, filename);
452
453         statusbar_pop_all();
454
455         if (ok != IMAP_SUCCESS) {
456                 g_warning(_("can't fetch message %d\n"), uid);
457                 g_free(filename);
458                 return NULL;
459         }
460
461         return filename;
462 }
463
464 gint imap_add_msg(Folder *folder, FolderItem *dest, const gchar *file,
465                   gboolean remove_source)
466 {
467         IMAPSession *session;
468         gint ok;
469
470         g_return_val_if_fail(folder != NULL, -1);
471         g_return_val_if_fail(dest != NULL, -1);
472         g_return_val_if_fail(file != NULL, -1);
473
474         session = imap_session_get(folder);
475         if (!session)
476                 return -1;
477
478         ok = imap_cmd_append(SESSION(session)->sock, dest->path, file);
479         if (ok != IMAP_SUCCESS) {
480                 g_warning(_("can't append message %s\n"), file);
481                 return -1;
482         }
483
484         if (remove_source) {
485                 if (unlink(file) < 0)
486                         FILE_OP_ERROR(file, "unlink");
487         }
488
489         return dest->last_num;
490 }
491
492 static gint imap_do_copy(Folder *folder, FolderItem *dest, MsgInfo *msginfo,
493                          gboolean remove_source)
494 {
495         gchar *destdir;
496         IMAPSession *session;
497         gint ok;
498
499         g_return_val_if_fail(folder != NULL, -1);
500         g_return_val_if_fail(folder->type == F_IMAP, -1);
501         g_return_val_if_fail(dest != NULL, -1);
502         g_return_val_if_fail(msginfo != NULL, -1);
503
504         session = imap_session_get(folder);
505         if (!session) return -1;
506
507         if (msginfo->folder == dest) {
508                 g_warning(_("the src folder is identical to the dest.\n"));
509                 return -1;
510         }
511
512         destdir = imap_get_real_path(IMAP_FOLDER(folder), dest->path);
513
514         if (remove_source)
515                 debug_print(_("Moving message %s%c%d to %s ...\n"),
516                             msginfo->folder->path, G_DIR_SEPARATOR,
517                             msginfo->msgnum, destdir);
518         else
519                 debug_print(_("Copying message %s%c%d to %s ...\n"),
520                             msginfo->folder->path, G_DIR_SEPARATOR,
521                             msginfo->msgnum, destdir);
522
523         ok = imap_cmd_copy(SESSION(session)->sock, msginfo->msgnum, destdir);
524
525         if (ok == IMAP_SUCCESS && remove_source) {
526                 imap_set_message_flags(session, msginfo->msgnum, msginfo->msgnum,
527                                        IMAP_FLAG_DELETED, TRUE);
528                 ok = imap_cmd_expunge(SESSION(session)->sock);
529         }
530
531         g_free(destdir);
532         statusbar_pop_all();
533
534         return ok;
535 }
536
537 static gint imap_do_copy_msgs_with_dest(Folder *folder, FolderItem *dest, 
538                                         GSList *msglist,
539                                         gboolean remove_source)
540 {
541         gchar *destdir;
542         GSList *cur;
543         MsgInfo *msginfo;
544         IMAPSession *session;
545         gint ok;
546
547         g_return_val_if_fail(folder != NULL, -1);
548         g_return_val_if_fail(dest != NULL, -1);
549         g_return_val_if_fail(msglist != NULL, -1);
550
551         session = imap_session_get(folder);
552         if (!session) return -1;
553
554         destdir = imap_get_real_path(IMAP_FOLDER(folder), dest->path);
555
556         for (cur = msglist; cur != NULL; cur = cur->next) {
557                 msginfo = (MsgInfo *)cur->data;
558
559                 if (msginfo->folder == dest) {
560                         g_warning(_("the src folder is identical to the dest.\n"));
561                         continue;
562                 }
563
564                 if (remove_source)
565                         debug_print(_("Moving message %s%c%d to %s ...\n"),
566                                     msginfo->folder->path, G_DIR_SEPARATOR,
567                                     msginfo->msgnum, destdir);
568                 else
569                         debug_print(_("Copying message %s%c%d to %s ...\n"),
570                                     msginfo->folder->path, G_DIR_SEPARATOR,
571                                     msginfo->msgnum, destdir);
572
573                 ok = imap_cmd_copy(SESSION(session)->sock, msginfo->msgnum,
574                                    destdir);
575
576                 if (ok == IMAP_SUCCESS && remove_source) {
577                         imap_set_message_flags
578                                 (session, msginfo->msgnum, msginfo->msgnum,
579                                  IMAP_FLAG_DELETED, TRUE);
580                 }
581         }
582
583         if (remove_source)
584                 ok = imap_cmd_expunge(SESSION(session)->sock);
585
586         g_free(destdir);
587         statusbar_pop_all();
588
589         return IMAP_SUCCESS;
590 }
591
592 gint imap_move_msg(Folder *folder, FolderItem *dest, MsgInfo *msginfo)
593 {
594         return imap_do_copy(folder, dest, msginfo, TRUE);
595 }
596
597 gint imap_move_msgs_with_dest(Folder *folder, FolderItem *dest, 
598                               GSList *msglist)
599 {
600         return imap_do_copy_msgs_with_dest(folder, dest, msglist, TRUE);
601 }
602
603 gint imap_copy_msg(Folder *folder, FolderItem *dest, MsgInfo *msginfo)
604 {
605         return imap_do_copy(folder, dest, msginfo, FALSE);
606 }
607
608 gint imap_copy_msgs_with_dest(Folder *folder, FolderItem *dest, 
609                               GSList *msglist)
610 {
611         return imap_do_copy_msgs_with_dest(folder, dest, msglist, FALSE);
612 }
613
614 gint imap_remove_msg(Folder *folder, FolderItem *item, gint uid)
615 {
616         gint exists, recent, unseen;
617         guint32 uid_validity;
618         gint ok;
619         IMAPSession *session;
620
621         g_return_val_if_fail(folder != NULL, -1);
622         g_return_val_if_fail(folder->type == F_IMAP, -1);
623         g_return_val_if_fail(item != NULL, -1);
624
625         session = imap_session_get(folder);
626         if (!session) return -1;
627
628         ok = imap_select(session, IMAP_FOLDER(folder), item->path,
629                          &exists, &recent, &unseen, &uid_validity);
630         statusbar_pop_all();
631         if (ok != IMAP_SUCCESS)
632                 return ok;
633
634         ok = imap_set_message_flags
635                 (IMAP_SESSION(REMOTE_FOLDER(folder)->session),
636                  (guint32)uid, (guint32)uid, IMAP_FLAG_DELETED, TRUE);
637         statusbar_pop_all();
638         if (ok != IMAP_SUCCESS) {
639                 log_warning(_("can't set deleted flags: %d\n"), uid);
640                 return ok;
641         }
642
643         ok = imap_cmd_expunge(SESSION(session)->sock);
644         statusbar_pop_all();
645         if (ok != IMAP_SUCCESS) {
646                 log_warning(_("can't expunge\n"));
647                 return ok;
648         }
649
650         return IMAP_SUCCESS;
651 }
652
653 gint imap_remove_all_msg(Folder *folder, FolderItem *item)
654 {
655         gint exists, recent, unseen;
656         guint32 uid_validity;
657         gint ok;
658         IMAPSession *session;
659
660         g_return_val_if_fail(folder != NULL, -1);
661         g_return_val_if_fail(item != NULL, -1);
662
663         session = imap_session_get(folder);
664         if (!session) return -1;
665
666         ok = imap_select(session, IMAP_FOLDER(folder), item->path,
667                          &exists, &recent, &unseen, &uid_validity);
668         statusbar_pop_all();
669         if (ok != IMAP_SUCCESS)
670                 return ok;
671         if (exists == 0)
672                 return IMAP_SUCCESS;
673
674         imap_cmd_gen_send(SESSION(session)->sock,
675                           "STORE 1:%d +FLAGS (\\Deleted)", exists);
676         ok = imap_cmd_ok(SESSION(session)->sock, NULL);
677         statusbar_pop_all();
678         if (ok != IMAP_SUCCESS) {
679                 log_warning(_("can't set deleted flags: 1:%d\n"), exists);
680                 return ok;
681         }
682
683         ok = imap_cmd_expunge(SESSION(session)->sock);
684         statusbar_pop_all();
685         if (ok != IMAP_SUCCESS) {
686                 log_warning(_("can't expunge\n"));
687                 return ok;
688         }
689
690         return IMAP_SUCCESS;
691 }
692
693 void imap_scan_folder(Folder *folder, FolderItem *item)
694 {
695         IMAPSession *session;
696         gint messages, recent, unseen;
697         guint32 uid_validity;
698         gint ok;
699
700         g_return_if_fail(folder != NULL);
701         g_return_if_fail(item != NULL);
702
703         session = imap_session_get(folder);
704         if (!session) return;
705
706         ok = imap_status(session, IMAP_FOLDER(folder), item->path,
707                          &messages, &recent, &unseen, &uid_validity);
708         statusbar_pop_all();
709         if (ok != IMAP_SUCCESS) return;
710
711         item->new = recent;
712         item->unread = unseen;
713         item->total = messages;
714         /* item->mtime = uid_validity; */
715 }
716
717 void imap_scan_tree(Folder *folder)
718 {
719         IMAPFolder *imapfolder = IMAP_FOLDER(folder);
720         FolderItem *item, *inbox;
721         IMAPSession *session;
722         IMAPNameSpace *namespace = NULL;
723         gchar *root_folder = NULL;
724
725         g_return_if_fail(folder != NULL);
726         g_return_if_fail(folder->account != NULL);
727
728         session = imap_session_get(folder);
729         if (!session) return;
730
731         if (imapfolder->namespace && imapfolder->namespace->data)
732                 namespace = (IMAPNameSpace *)imapfolder->namespace->data;
733
734         if (folder->account->imap_dir && *folder->account->imap_dir) {
735                 gchar *imap_dir;
736                 Xstrdup_a(imap_dir, folder->account->imap_dir, return);
737                 strtailchomp(imap_dir, '/');
738                 root_folder = g_strconcat
739                         (namespace && namespace->name ? namespace->name : "",
740                          imap_dir, NULL);
741                 if (namespace && namespace->separator)
742                         subst_char(root_folder, namespace->separator, '/');
743         }
744
745         if (root_folder)
746                 debug_print("IMAP root directory: %s\n", root_folder);
747
748         folder_tree_destroy(folder);
749         item = folder_item_new(folder->name, root_folder);
750         item->folder = folder;
751         folder->node = g_node_new(item);
752         g_free(root_folder);
753
754         imap_scan_tree_recursive(session, item, namespace);
755
756         if (!folder->inbox) {
757                 inbox = folder_item_new("INBOX", "INBOX");
758                 inbox->stype = F_INBOX;
759                 folder_item_append(item, inbox);
760                 folder->inbox = inbox;
761         }
762         if (!folder->trash)
763                 imap_create_trash(folder);
764 }
765
766 static void imap_scan_tree_recursive(IMAPSession *session,
767                                      FolderItem *item,
768                                      IMAPNameSpace *namespace)
769 {
770         IMAPFolder *imapfolder;
771         FolderItem *new_item;
772         GSList *item_list, *cur;
773         gchar *real_path;
774
775         g_return_if_fail(item != NULL);
776         g_return_if_fail(item->folder != NULL);
777         g_return_if_fail(item->no_sub == FALSE);
778
779         imapfolder = IMAP_FOLDER(item->folder);
780
781         if (item->folder->ui_func)
782                 item->folder->ui_func(item->folder, item,
783                                       item->folder->ui_func_data);
784
785         if (item->path) {
786                 real_path = imap_get_real_path(imapfolder, item->path);
787                 imap_cmd_gen_send(SESSION(session)->sock, "LIST \"\" %s%c%%",
788                                   real_path,
789                                   namespace && namespace->separator
790                                   ? namespace->separator : '/');
791         } else {
792                 real_path = g_strdup(namespace && namespace->name
793                                      ? namespace->name : "");
794                 imap_cmd_gen_send(SESSION(session)->sock, "LIST \"\" %s%%",
795                                   real_path);
796         }
797
798         strtailchomp(real_path, namespace && namespace->separator
799                      ? namespace->separator : '/');
800
801         item_list = imap_parse_list(session, real_path);
802         for (cur = item_list; cur != NULL; cur = cur->next) {
803                 new_item = cur->data;
804                 if (!strcmp(new_item->path, "INBOX")) {
805                         if (!item->folder->inbox) {
806                                 new_item->stype = F_INBOX;
807                                 item->folder->inbox = new_item;
808                         } else {
809                                 folder_item_destroy(new_item);
810                                 continue;
811                         }
812                 } else if (!item->parent && !item->folder->trash) {
813                         if (!strcasecmp(g_basename(new_item->path), "Trash")) {
814                                 new_item->stype = F_TRASH;
815                                 item->folder->trash = new_item;
816                         }
817                 }
818                 folder_item_append(item, new_item);
819                 if (new_item->no_select == FALSE)
820                         imap_scan_folder(new_item->folder, new_item);
821                 if (new_item->no_sub == FALSE)
822                         imap_scan_tree_recursive(session, new_item, namespace);
823         }
824 }
825
826 static GSList *imap_parse_list(IMAPSession *session, const gchar *path)
827 {
828         gchar buf[IMAPBUFSIZE];
829         gchar flags[256];
830         gchar separator[16];
831         gchar *p;
832         gchar *name;
833         GSList *item_list = NULL;
834         GString *str;
835         FolderItem *new_item;
836
837         debug_print("getting list of %s ...\n", *path ? path : "\"\"");
838
839         str = g_string_new(NULL);
840
841         for (;;) {
842                 if (sock_gets(SESSION(session)->sock, buf, sizeof(buf)) <= 0) {
843                         log_warning(_("error occured while getting LIST.\n"));
844                         break;
845                 }
846                 strretchomp(buf);
847                 if (buf[0] != '*' || buf[1] != ' ') {
848                         log_print("IMAP4< %s\n", buf);
849                         break;
850                 }
851                 debug_print("IMAP4< %s\n", buf);
852
853                 g_string_assign(str, buf);
854                 p = str->str + 2;
855                 if (strncmp(p, "LIST ", 5) != 0) continue;
856                 p += 5;
857
858                 if (*p != '(') continue;
859                 p++;
860                 p = strchr_cpy(p, ')', flags, sizeof(flags));
861                 if (!p) continue;
862                 while (*p == ' ') p++;
863
864                 p = strchr_cpy(p, ' ', separator, sizeof(separator));
865                 if (!p) continue;
866                 extract_quote(separator, '"');
867                 if (!strcmp(separator, "NIL"))
868                         separator[0] = '\0';
869
870                 buf[0] = '\0';
871                 while (*p == ' ') p++;
872                 if (*p == '{' || *p == '"')
873                         p = imap_parse_atom(SESSION(session)->sock, p,
874                                             buf, sizeof(buf), str);
875                 else
876                         strncpy2(buf, p, sizeof(buf));
877                 strtailchomp(buf, separator[0]);
878                 if (buf[0] == '\0') continue;
879                 if (!strcmp(buf, path)) continue;
880
881                 if (separator[0] != '\0')
882                         subst_char(buf, separator[0], '/');
883                 name = g_basename(buf);
884                 if (name[0] == '.') continue;
885
886                 new_item = folder_item_new(name, buf);
887                 if (strcasestr(flags, "\\Noinferiors") != NULL)
888                         new_item->no_sub = TRUE;
889                 if (strcasestr(flags, "\\Noselect") != NULL)
890                         new_item->no_select = TRUE;
891
892                 item_list = g_slist_append(item_list, new_item);
893
894                 debug_print("folder %s has been added.\n", buf);
895         }
896
897         g_string_free(str, TRUE);
898         statusbar_pop_all();
899
900         return item_list;
901 }
902
903 gint imap_create_tree(Folder *folder)
904 {
905         FolderItem *item;
906
907         g_return_val_if_fail(folder != NULL, -1);
908         g_return_val_if_fail(folder->node != NULL, -1);
909         g_return_val_if_fail(folder->node->data != NULL, -1);
910         g_return_val_if_fail(folder->account != NULL, -1);
911
912         imap_scan_tree(folder);
913
914         item = FOLDER_ITEM(folder->node->data);
915
916         if (!folder->inbox) {
917                 FolderItem *inbox;
918
919                 inbox = folder_item_new("INBOX", "INBOX");
920                 inbox->stype = F_INBOX;
921                 folder_item_append(item, inbox);
922                 folder->inbox = inbox;
923         }
924         if (!folder->trash)
925                 imap_create_trash(folder);
926
927         return 0;
928 }
929
930 static gint imap_create_trash(Folder *folder)
931 {
932         IMAPFolder *imapfolder = IMAP_FOLDER(folder);
933         FolderItem *item;
934         FolderItem *new_item;
935         gchar *trash_path;
936         gchar *imap_dir = "";
937
938         g_return_val_if_fail(folder != NULL, -1);
939         g_return_val_if_fail(folder->node != NULL, -1);
940         g_return_val_if_fail(folder->node->data != NULL, -1);
941         g_return_val_if_fail(folder->account != NULL, -1);
942
943         if (folder->account->imap_dir && *folder->account->imap_dir) {
944                 gchar *tmpdir;
945
946                 Xstrdup_a(tmpdir, folder->account->imap_dir, return -1);
947                 strtailchomp(tmpdir, '/');
948                 Xalloca(imap_dir, strlen(tmpdir) + 2, return -1);
949                 g_snprintf(imap_dir, strlen(tmpdir) + 2, "%s%c", tmpdir, '/');
950         }
951
952         if (imapfolder->namespace && imapfolder->namespace->data) {
953                 IMAPNameSpace *namespace =
954                         (IMAPNameSpace *)imapfolder->namespace->data;
955
956                 if (*namespace->name != '\0') {
957                         gchar *name;
958
959                         Xstrdup_a(name, namespace->name, return -1);
960                         subst_char(name, namespace->separator, '/');
961                         trash_path = g_strconcat(name, imap_dir, "Trash", NULL);
962                 } else
963                         trash_path = g_strconcat(imap_dir, "Trash", NULL);
964         } else
965                 trash_path = g_strconcat(imap_dir, "Trash", NULL);
966
967         item = FOLDER_ITEM(folder->node->data);
968         new_item = imap_create_folder(folder, item, trash_path);
969
970         if (!new_item) {
971                 gchar *path;
972
973                 new_item = folder_item_new("Trash", trash_path);
974                 folder_item_append(item, new_item);
975
976                 path = folder_item_get_path(new_item);
977                 if (!is_dir_exist(path))
978                         make_dir_hier(path);
979                 g_free(path);
980         } else {
981                 g_free(new_item->name);
982                 new_item->name = g_strdup("Trash");
983         }
984         new_item->stype = F_TRASH;
985         folder->trash = new_item;
986
987         g_free(trash_path);
988
989         return 0;
990 }
991
992 FolderItem *imap_create_folder(Folder *folder, FolderItem *parent,
993                                const gchar *name)
994 {
995         gchar *dirpath, *imappath;
996         IMAPSession *session;
997         IMAPNameSpace *namespace;
998         FolderItem *new_item;
999         gchar *new_name;
1000         const gchar *p;
1001         gint ok;
1002
1003         g_return_val_if_fail(folder != NULL, NULL);
1004         g_return_val_if_fail(folder->account != NULL, NULL);
1005         g_return_val_if_fail(parent != NULL, NULL);
1006         g_return_val_if_fail(name != NULL, NULL);
1007
1008         session = imap_session_get(folder);
1009         if (!session) return NULL;
1010
1011         if (parent->path)
1012                 dirpath = g_strconcat(parent->path, "/", name, NULL);
1013         else if ((p = strchr(name, '/')) != NULL && *(p + 1) != '\0')
1014                 dirpath = g_strdup(name);
1015         else if (folder->account->imap_dir && *folder->account->imap_dir) {
1016                 gchar *imap_dir;
1017
1018                 Xstrdup_a(imap_dir, folder->account->imap_dir, return NULL);
1019                 strtailchomp(imap_dir, '/');
1020                 dirpath = g_strconcat(imap_dir, "/", name, NULL);
1021         } else
1022                 dirpath = g_strdup(name);
1023
1024         Xstrdup_a(imappath, dirpath, {g_free(dirpath); return NULL;});
1025         Xstrdup_a(new_name, name, {g_free(dirpath); return NULL;});
1026         namespace = imap_find_namespace(IMAP_FOLDER(folder), imappath);
1027         if (namespace && namespace->separator) {
1028                 imap_path_separator_subst(imappath, namespace->separator);
1029                 imap_path_separator_subst(new_name, namespace->separator);
1030                 strtailchomp(new_name, namespace->separator);
1031         }
1032         strtailchomp(dirpath, '/');
1033
1034         if (strcmp(name, "INBOX") != 0) {
1035                 GPtrArray *argbuf;
1036                 gint i;
1037                 gboolean exist = FALSE;
1038
1039                 argbuf = g_ptr_array_new();
1040                 ok = imap_cmd_list(SESSION(session)->sock, NULL, imappath,
1041                                    argbuf);
1042                 statusbar_pop_all();
1043                 if (ok != IMAP_SUCCESS) {
1044                         log_warning(_("can't create mailbox: LIST failed\n"));
1045                         g_free(dirpath);
1046                         g_ptr_array_free(argbuf, TRUE);
1047                         return NULL;
1048                 }
1049
1050                 for (i = 0; i < argbuf->len; i++) {
1051                         gchar *str;
1052                         str = g_ptr_array_index(argbuf, i);
1053                         if (!strncmp(str, "LIST ", 5)) {
1054                                 exist = TRUE;
1055                                 break;
1056                         }
1057                 }
1058                 g_ptr_array_free(argbuf, TRUE);
1059
1060                 if (!exist) {
1061                         ok = imap_cmd_create(SESSION(session)->sock, imappath);
1062                         statusbar_pop_all();
1063                         if (ok != IMAP_SUCCESS) {
1064                                 log_warning(_("can't create mailbox\n"));
1065                                 g_free(dirpath);
1066                                 return NULL;
1067                         }
1068                 }
1069         }
1070
1071         new_item = folder_item_new(new_name, dirpath);
1072         folder_item_append(parent, new_item);
1073         g_free(dirpath);
1074
1075         dirpath = folder_item_get_path(new_item);
1076         if (!is_dir_exist(dirpath))
1077                 make_dir_hier(dirpath);
1078         g_free(dirpath);
1079
1080         return new_item;
1081 }
1082
1083 gint imap_remove_folder(Folder *folder, FolderItem *item)
1084 {
1085         gint ok;
1086         IMAPSession *session;
1087         gchar *path;
1088         gint exists, recent, unseen;
1089         guint32 uid_validity;
1090
1091         g_return_val_if_fail(folder != NULL, -1);
1092         g_return_val_if_fail(item != NULL, -1);
1093         g_return_val_if_fail(item->path != NULL, -1);
1094
1095         session = imap_session_get(folder);
1096         if (!session) return -1;
1097
1098         path = imap_get_real_path(IMAP_FOLDER(folder), item->path);
1099
1100         ok = imap_cmd_examine(SESSION(session)->sock, "INBOX",
1101                               &exists, &recent, &unseen, &uid_validity);
1102         statusbar_pop_all();
1103         if (ok != IMAP_SUCCESS) {
1104                 g_free(path);
1105                 return -1;
1106         }
1107
1108         ok = imap_cmd_delete(SESSION(session)->sock, path);
1109         statusbar_pop_all();
1110         if (ok != IMAP_SUCCESS) {
1111                 log_warning(_("can't delete mailbox\n"));
1112                 g_free(path);
1113                 return -1;
1114         }
1115
1116         g_free(path);
1117         folder_item_remove(item);
1118
1119         return 0;
1120 }
1121
1122 static GSList *imap_get_uncached_messages(IMAPSession *session,
1123                                           FolderItem *item,
1124                                           guint32 first_uid, guint32 last_uid)
1125 {
1126         gchar *tmp;
1127         GSList *newlist = NULL;
1128         GSList *llast = NULL;
1129         GString *str;
1130         MsgInfo *msginfo;
1131
1132         g_return_val_if_fail(session != NULL, NULL);
1133         g_return_val_if_fail(item != NULL, NULL);
1134         g_return_val_if_fail(item->folder != NULL, NULL);
1135         g_return_val_if_fail(item->folder->type == F_IMAP, NULL);
1136         g_return_val_if_fail(first_uid <= last_uid, NULL);
1137
1138         if (imap_cmd_envelope(SESSION(session)->sock, first_uid, last_uid)
1139             != IMAP_SUCCESS) {
1140                 log_warning(_("can't get envelope\n"));
1141                 return NULL;
1142         }
1143
1144         str = g_string_new(NULL);
1145
1146         for (;;) {
1147                 if ((tmp = sock_getline(SESSION(session)->sock)) == NULL) {
1148                         log_warning(_("error occurred while getting envelope.\n"));
1149                         g_string_free(str, TRUE);
1150                         return newlist;
1151                 }
1152                 strretchomp(tmp);
1153                 log_print("IMAP4< %s\n", tmp);
1154                 if (tmp[0] != '*' || tmp[1] != ' ') {
1155                         g_free(tmp);
1156                         break;
1157                 }
1158                 g_string_assign(str, tmp);
1159                 g_free(tmp);
1160
1161                 msginfo = imap_parse_envelope(SESSION(session)->sock, str);
1162                 if (!msginfo) {
1163                         log_warning(_("can't parse envelope: %s\n"), str->str);
1164                         continue;
1165                 }
1166
1167                 msginfo->folder = item;
1168
1169                 if (!newlist)
1170                         llast = newlist = g_slist_append(newlist, msginfo);
1171                 else {
1172                         llast = g_slist_append(llast, msginfo);
1173                         llast = llast->next;
1174                 }
1175         }
1176
1177         g_string_free(str, TRUE);
1178
1179         return newlist;
1180 }
1181
1182 static GSList *imap_delete_cached_messages(GSList *mlist, FolderItem *item,
1183                                            guint32 first_uid, guint32 last_uid)
1184 {
1185         GSList *cur, *next;
1186         MsgInfo *msginfo;
1187         gchar *dir;
1188
1189         g_return_val_if_fail(item != NULL, mlist);
1190         g_return_val_if_fail(item->folder != NULL, mlist);
1191         g_return_val_if_fail(item->folder->type == F_IMAP, mlist);
1192
1193         debug_print(_("Deleting cached messages %d - %d ... "),
1194                     first_uid, last_uid);
1195
1196         dir = folder_item_get_path(item);
1197         remove_numbered_files(dir, first_uid, last_uid);
1198         g_free(dir);
1199
1200         for (cur = mlist; cur != NULL; ) {
1201                 next = cur->next;
1202
1203                 msginfo = (MsgInfo *)cur->data;
1204                 if (msginfo != NULL && first_uid <= msginfo->msgnum &&
1205                     msginfo->msgnum <= last_uid) {
1206                         procmsg_msginfo_free(msginfo);
1207                         mlist = g_slist_remove(mlist, msginfo);
1208                 }
1209
1210                 cur = next;
1211         }
1212
1213         debug_print(_("done.\n"));
1214
1215         return mlist;
1216 }
1217
1218 static void imap_delete_all_cached_messages(FolderItem *item)
1219 {
1220         gchar *dir;
1221
1222         g_return_if_fail(item != NULL);
1223         g_return_if_fail(item->folder != NULL);
1224         g_return_if_fail(item->folder->type == F_IMAP);
1225
1226         debug_print(_("Deleting all cached messages... "));
1227
1228         dir = folder_item_get_path(item);
1229         remove_all_numbered_files(dir);
1230         g_free(dir);
1231
1232         debug_print(_("done.\n"));
1233 }
1234
1235 static SockInfo *imap_open(const gchar *server, gushort port, gchar *buf)
1236 {
1237         SockInfo *sock;
1238
1239         if ((sock = sock_connect(server, port)) == NULL) {
1240                 log_warning(_("Can't connect to IMAP4 server: %s:%d\n"),
1241                             server, port);
1242                 return NULL;
1243         }
1244
1245         imap_cmd_count = 0;
1246
1247         if (imap_cmd_noop(sock) != IMAP_SUCCESS) {
1248                 sock_close(sock);
1249                 return NULL;
1250         }
1251
1252         return sock;
1253 }
1254
1255 #define THROW goto catch
1256
1257 static void imap_parse_namespace(IMAPSession *session, IMAPFolder *folder)
1258 {
1259         gchar *ns_str;
1260         gchar *name;
1261         gchar *separator;
1262         gchar *p;
1263         IMAPNameSpace *namespace;
1264         GList *ns_list = NULL;
1265
1266         g_return_if_fail(session != NULL);
1267         g_return_if_fail(folder != NULL);
1268
1269         if (folder->namespace != NULL) return;
1270
1271         if (imap_cmd_namespace(SESSION(session)->sock, &ns_str)
1272             != IMAP_SUCCESS) {
1273                 log_warning(_("can't get namespace\n"));
1274                 return;
1275         }
1276
1277         /* get the first element */
1278         extract_one_parenthesis_with_skip_quote(ns_str, '"', '(', ')');
1279         g_strstrip(ns_str);
1280         p = ns_str;
1281
1282         while (*p != '\0') {
1283                 /* parse ("#foo" "/") */
1284
1285                 while (*p && *p != '(') p++;
1286                 if (*p == '\0') THROW;
1287                 p++;
1288
1289                 while (*p && *p != '"') p++;
1290                 if (*p == '\0') THROW;
1291                 p++;
1292                 name = p;
1293
1294                 while (*p && *p != '"') p++;
1295                 if (*p == '\0') THROW;
1296                 *p = '\0';
1297                 p++;
1298
1299                 while (*p && isspace(*p)) p++;
1300                 if (*p == '\0') THROW;
1301                 if (strncmp(p, "NIL", 3) == 0)
1302                         separator = NULL;
1303                 else if (*p == '"') {
1304                         p++;
1305                         separator = p;
1306                         while (*p && *p != '"') p++;
1307                         if (*p == '\0') THROW;
1308                         *p = '\0';
1309                         p++;
1310                 } else THROW;
1311
1312                 while (*p && *p != ')') p++;
1313                 if (*p == '\0') THROW;
1314                 p++;
1315
1316                 namespace = g_new(IMAPNameSpace, 1);
1317                 namespace->name = g_strdup(name);
1318                 namespace->separator = separator ? separator[0] : '\0';
1319                 ns_list = g_list_append(ns_list, namespace);
1320                 IMAP_FOLDER(folder)->namespace = ns_list;
1321         }
1322
1323 catch:
1324         g_free(ns_str);
1325         return;
1326 }
1327
1328 #undef THROW
1329
1330 static IMAPNameSpace *imap_find_namespace(IMAPFolder *folder,
1331                                           const gchar *path)
1332 {
1333         IMAPNameSpace *namespace = NULL;
1334         GList *ns_list;
1335         gchar *name;
1336
1337         g_return_val_if_fail(folder != NULL, NULL);
1338         g_return_val_if_fail(path != NULL, NULL);
1339
1340         ns_list = folder->namespace;
1341
1342         for (; ns_list != NULL; ns_list = ns_list->next) {
1343                 IMAPNameSpace *tmp_ns = ns_list->data;
1344
1345                 Xstrdup_a(name, tmp_ns->name, return namespace);
1346                 if (tmp_ns->separator && tmp_ns->separator != '/')
1347                         subst_char(name, tmp_ns->separator, '/');
1348                 if (strncmp(path, name, strlen(name)) == 0)
1349                         namespace = tmp_ns;
1350         }
1351
1352         return namespace;
1353 }
1354
1355 static gchar *imap_get_real_path(IMAPFolder *folder, const gchar *path)
1356 {
1357         gchar *real_path;
1358         IMAPNameSpace *namespace;
1359
1360         g_return_val_if_fail(folder != NULL, NULL);
1361         g_return_val_if_fail(path != NULL, NULL);
1362
1363         real_path = g_strdup(path);
1364         namespace = imap_find_namespace(folder, path);
1365         if (namespace && namespace->separator)
1366                 imap_path_separator_subst(real_path, namespace->separator);
1367
1368         return real_path;
1369 }
1370
1371 static gchar *imap_parse_atom(SockInfo *sock, gchar *src,
1372                               gchar *dest, gint dest_len, GString *str)
1373 {
1374         gchar *cur_pos = src;
1375
1376         g_return_val_if_fail(str != NULL, cur_pos);
1377
1378         while (*cur_pos == ' ') cur_pos++;
1379
1380         if (!strncmp(cur_pos, "NIL", 3)) {
1381                 *dest = '\0';
1382                 cur_pos += 3;
1383         } else if (*cur_pos == '\"') {
1384                 gchar *p;
1385
1386                 p = strchr_cpy(cur_pos + 1, '\"', dest, dest_len);
1387                 cur_pos = p ? p : cur_pos + 2;
1388         } else if (*cur_pos == '{') {
1389                 gchar buf[32];
1390                 gint len;
1391                 gchar *nextline;
1392
1393                 cur_pos = strchr_cpy(cur_pos + 1, '}', buf, sizeof(buf));
1394                 len = atoi(buf);
1395
1396                 if ((nextline = sock_getline(sock)) == NULL)
1397                         return cur_pos;
1398                 strretchomp(nextline);
1399                 log_print("IMAP4< %s\n", nextline);
1400                 g_string_assign(str, nextline);
1401
1402                 len = MIN(len, strlen(nextline));
1403                 memcpy(dest, nextline, MIN(len, dest_len - 1));
1404                 dest[MIN(len, dest_len - 1)] = '\0';
1405                 cur_pos = str->str + len;
1406         }
1407
1408         return cur_pos;
1409 }
1410
1411 static gchar *imap_parse_one_address(SockInfo *sock, gchar *start,
1412                                      gchar *out_from_str,
1413                                      gchar *out_fromname_str,
1414                                      GString *str)
1415 {
1416         gchar buf[IMAPBUFSIZE];
1417         gchar *userid;
1418         gchar *domain;
1419         gchar *cur_pos = start;
1420
1421         cur_pos = imap_parse_atom(sock, cur_pos, buf, sizeof(buf), str);
1422         conv_unmime_header(out_fromname_str, IMAPBUFSIZE, buf, NULL);
1423
1424         cur_pos = imap_parse_atom(sock, cur_pos, buf, sizeof(buf), str);
1425
1426         cur_pos = imap_parse_atom(sock, cur_pos, buf, sizeof(buf), str);
1427         Xstrdup_a(userid, buf, return cur_pos + 1);
1428
1429         cur_pos = imap_parse_atom(sock, cur_pos, buf, sizeof(buf), str);
1430         Xstrdup_a(domain, buf, return cur_pos + 1);
1431
1432         if (out_fromname_str[0] != '\0') {
1433                 g_snprintf(out_from_str, IMAPBUFSIZE, "\"%s\" <%s@%s>",
1434                            out_fromname_str, userid, domain);
1435         } else {
1436                 g_snprintf(out_from_str, IMAPBUFSIZE, "%s@%s",
1437                            userid, domain);
1438                 strcpy(out_fromname_str, out_from_str);
1439         }
1440
1441         while (*cur_pos == ' ') cur_pos++;
1442         g_return_val_if_fail(*cur_pos == ')', NULL);
1443
1444         return cur_pos + 1;
1445 }
1446
1447 static gchar *imap_parse_address(SockInfo *sock, gchar *start,
1448                                  gchar **out_from_str,
1449                                  gchar **out_fromname_str,
1450                                  GString *str)
1451 {
1452         gchar buf[IMAPBUFSIZE];
1453         gchar name_buf[IMAPBUFSIZE];
1454         gchar *cur_pos = start;
1455         GString *addr_str;
1456
1457         if (out_from_str)     *out_from_str     = NULL;
1458         if (out_fromname_str) *out_fromname_str = NULL;
1459         buf[0] = name_buf[0] = '\0';
1460
1461         if (!strncmp(cur_pos, "NIL", 3)) {
1462                 if (out_from_str)     *out_from_str     = g_strdup("");
1463                 if (out_fromname_str) *out_fromname_str = g_strdup("");
1464                 return cur_pos + 3;
1465         }
1466
1467         g_return_val_if_fail(*cur_pos == '(', NULL);
1468         cur_pos++;
1469
1470         addr_str = g_string_new(NULL);
1471
1472         for (;;) {
1473                 gchar ch = *cur_pos++;
1474                 if (ch == ')') break;
1475                 if (ch == '(') {
1476                         cur_pos = imap_parse_one_address
1477                                 (sock, cur_pos, buf, name_buf, str);
1478                         if (!cur_pos) {
1479                                 g_string_free(addr_str, TRUE);
1480                                 return NULL;
1481                         }
1482                         if (addr_str->str[0] != '\0')
1483                                 g_string_append(addr_str, ", ");
1484                         g_string_append(addr_str, buf);
1485                 }
1486         }
1487
1488         if (out_from_str)     *out_from_str     = g_strdup(addr_str->str);
1489         if (out_fromname_str) *out_fromname_str = g_strdup(name_buf);
1490
1491         g_string_free(addr_str, TRUE);
1492
1493         return cur_pos;
1494 }
1495
1496 static MsgFlags imap_parse_flags(const gchar *flag_str)  
1497 {
1498         const gchar *p = flag_str;
1499         MsgFlags flags;
1500
1501         flags.perm_flags = MSG_UNREAD;
1502         flags.tmp_flags  = MSG_IMAP;
1503
1504         while ((p = strchr(p, '\\')) != NULL) {
1505                 p++;
1506
1507                 if (g_strncasecmp(p, "Recent", 6) == 0) {
1508                         MSG_SET_PERM_FLAGS(flags, MSG_NEW|MSG_UNREAD);
1509                 } else if (g_strncasecmp(p, "Seen", 4) == 0) {
1510                         MSG_UNSET_PERM_FLAGS(flags, MSG_NEW|MSG_UNREAD);
1511                 } else if (g_strncasecmp(p, "Deleted", 7) == 0) {
1512                         MSG_SET_PERM_FLAGS(flags, MSG_DELETED);
1513                 } else if (g_strncasecmp(p, "Flagged", 7) == 0) {
1514                         MSG_SET_PERM_FLAGS(flags, MSG_MARKED);
1515                 }
1516         }
1517
1518         return flags;
1519 }
1520
1521 static MsgInfo *imap_parse_envelope(SockInfo *sock, GString *line_str)
1522 {
1523         MsgInfo *msginfo;
1524         gchar buf[IMAPBUFSIZE];
1525         gchar tmp[IMAPBUFSIZE];
1526         gchar *cur_pos;
1527         gint msgnum;
1528         guint32 uid = 0;
1529         size_t size = 0;
1530         gchar *date = NULL;
1531         time_t date_t = 0;
1532         gchar *subject = NULL;
1533         gchar *tmp_from;
1534         gchar *tmp_fromname;
1535         gchar *from = NULL;
1536         gchar *fromname = NULL;
1537         gchar *tmp_to;
1538         gchar *to = NULL;
1539         gchar *inreplyto = NULL;
1540         gchar *msgid = NULL;
1541         MsgFlags flags = {0, 0};
1542
1543         g_return_val_if_fail(line_str != NULL, NULL);
1544         g_return_val_if_fail(line_str->str[0] == '*' &&
1545                              line_str->str[1] == ' ', NULL);
1546
1547         cur_pos = line_str->str + 2;
1548
1549 #define PARSE_ONE_ELEMENT(ch) \
1550 { \
1551         cur_pos = strchr_cpy(cur_pos, ch, buf, sizeof(buf)); \
1552         g_return_val_if_fail(cur_pos != NULL, NULL); \
1553 }
1554
1555         PARSE_ONE_ELEMENT(' ');
1556         msgnum = atoi(buf);
1557
1558         PARSE_ONE_ELEMENT(' ');
1559         g_return_val_if_fail(!strcmp(buf, "FETCH"), NULL);
1560
1561         g_return_val_if_fail(*cur_pos == '(', NULL);
1562         cur_pos++;
1563
1564         while (*cur_pos != '\0' && *cur_pos != ')') {
1565                 while (*cur_pos == ' ') cur_pos++;
1566
1567                 if (!strncmp(cur_pos, "UID ", 4)) {
1568                         cur_pos += 4;
1569                         uid = strtoul(cur_pos, &cur_pos, 10);
1570                 } else if (!strncmp(cur_pos, "FLAGS ", 6)) {
1571                         cur_pos += 6;
1572                         g_return_val_if_fail(*cur_pos == '(', NULL);
1573                         cur_pos++;
1574                         PARSE_ONE_ELEMENT(')');
1575                         flags = imap_parse_flags(buf);
1576                 } else if (!strncmp(cur_pos, "RFC822.SIZE ", 12)) {
1577                         cur_pos += 12;
1578                         size = strtol(cur_pos, &cur_pos, 10);
1579                 } else if (!strncmp(cur_pos, "ENVELOPE ", 9)) {
1580                         cur_pos += 9;
1581                         g_return_val_if_fail(*cur_pos == '(', NULL);
1582                         cur_pos = imap_parse_atom
1583                                 (sock, cur_pos + 1, buf, sizeof(buf), line_str);
1584                         Xstrdup_a(date, buf, return NULL);
1585                         date_t = procheader_date_parse(NULL, date, 0);
1586
1587                         cur_pos = imap_parse_atom
1588                                 (sock, cur_pos, buf, sizeof(buf), line_str);
1589                         if (buf[0] != '\0') {
1590                                 conv_unmime_header(tmp, sizeof(tmp), buf, NULL);
1591                                 Xstrdup_a(subject, tmp, return NULL);
1592                         }
1593
1594                         g_return_val_if_fail(*cur_pos == ' ', NULL);
1595                         cur_pos = imap_parse_address(sock, cur_pos + 1,
1596                                                      &tmp_from, &tmp_fromname,
1597                                                      line_str);
1598                         Xstrdup_a(from, tmp_from,
1599                                   {g_free(tmp_from); g_free(tmp_fromname);
1600                                    return NULL;});
1601                         Xstrdup_a(fromname, tmp_fromname,
1602                                   {g_free(tmp_from); g_free(tmp_fromname);
1603                                    return NULL;});
1604                         g_free(tmp_from);
1605                         g_free(tmp_fromname);
1606
1607 #define SKIP_ONE_ELEMENT() \
1608 { \
1609         g_return_val_if_fail(*cur_pos == ' ', NULL); \
1610         cur_pos = imap_parse_address(sock, cur_pos + 1, NULL, NULL, \
1611                                      line_str); \
1612 }
1613
1614                         /* skip sender and reply-to */
1615                         SKIP_ONE_ELEMENT();
1616                         SKIP_ONE_ELEMENT();
1617
1618                         g_return_val_if_fail(*cur_pos == ' ', NULL);
1619                         cur_pos = imap_parse_address(sock, cur_pos + 1,
1620                                                      &tmp_to, NULL, line_str);
1621                         Xstrdup_a(to, tmp_to, {g_free(tmp_to); return NULL;});
1622                         g_free(tmp_to);
1623
1624                         /* skip Cc and Bcc */
1625                         SKIP_ONE_ELEMENT();
1626                         SKIP_ONE_ELEMENT();
1627
1628 #undef SKIP_ONE_ELEMENT
1629
1630                         g_return_val_if_fail(*cur_pos == ' ', NULL);
1631                         cur_pos = imap_parse_atom
1632                                 (sock, cur_pos, buf, sizeof(buf), line_str);
1633                         if (buf[0] != '\0') {
1634                                 eliminate_parenthesis(buf, '(', ')');
1635                                 extract_parenthesis(buf, '<', '>');
1636                                 remove_space(buf);
1637                                 Xstrdup_a(inreplyto, buf, return NULL);
1638                         }
1639
1640                         g_return_val_if_fail(*cur_pos == ' ', NULL);
1641                         cur_pos = imap_parse_atom
1642                                 (sock, cur_pos, buf, sizeof(buf), line_str);
1643                         if (buf[0] != '\0') {
1644                                 extract_parenthesis(buf, '<', '>');
1645                                 remove_space(buf);
1646                                 Xstrdup_a(msgid, buf, return NULL);
1647                         }
1648
1649                         g_return_val_if_fail(*cur_pos == ')', NULL);
1650                         cur_pos++;
1651                 } else {
1652                         g_warning("invalid FETCH response: %s\n", cur_pos);
1653                         break;
1654                 }
1655         }
1656
1657         msginfo = g_new0(MsgInfo, 1);
1658         msginfo->msgnum = uid;
1659         msginfo->size = size;
1660         msginfo->date = g_strdup(date);
1661         msginfo->date_t = date_t;
1662         msginfo->subject = g_strdup(subject);
1663         msginfo->from = g_strdup(from);
1664         msginfo->fromname = g_strdup(fromname);
1665         msginfo->to = g_strdup(to);
1666         msginfo->inreplyto = g_strdup(inreplyto);
1667         msginfo->msgid = g_strdup(msgid);
1668         msginfo->flags = flags;
1669
1670         return msginfo;
1671 }
1672
1673 static gint imap_set_message_flags(IMAPSession *session,
1674                                    guint32 first_uid,
1675                                    guint32 last_uid,
1676                                    IMAPFlags flags,
1677                                    gboolean is_set)
1678 {
1679         GString *buf;
1680         gint ok;
1681
1682         buf = g_string_new(is_set ? "+FLAGS (" : "-FLAGS (");
1683
1684         if (IMAP_IS_SEEN(flags))        g_string_append(buf, "\\Seen ");
1685         if (IMAP_IS_ANSWERED(flags))    g_string_append(buf, "\\Answered ");
1686         if (IMAP_IS_FLAGGED(flags))     g_string_append(buf, "\\Flagged ");
1687         if (IMAP_IS_DELETED(flags))     g_string_append(buf, "\\Deleted ");
1688         if (IMAP_IS_DRAFT(flags))       g_string_append(buf, "\\Draft");
1689
1690         if (buf->str[buf->len - 1] == ' ')
1691                 g_string_truncate(buf, buf->len - 1);
1692
1693         g_string_append_c(buf, ')');
1694
1695         ok = imap_cmd_store(SESSION(session)->sock, first_uid, last_uid,
1696                             buf->str);
1697         g_string_free(buf, TRUE);
1698
1699         return ok;
1700 }
1701
1702 static gint imap_select(IMAPSession *session, IMAPFolder *folder,
1703                         const gchar *path,
1704                         gint *exists, gint *recent, gint *unseen,
1705                         guint32 *uid_validity)
1706 {
1707         gchar *real_path;
1708         gint ok;
1709
1710         real_path = imap_get_real_path(folder, path);
1711         ok = imap_cmd_select(SESSION(session)->sock, real_path,
1712                              exists, recent, unseen, uid_validity);
1713         if (ok != IMAP_SUCCESS)
1714                 log_warning(_("can't select folder: %s\n"), real_path);
1715         g_free(real_path);
1716
1717         return ok;
1718 }
1719
1720 #define THROW(err) { ok = err; goto catch; }
1721
1722 static gint imap_get_uid(IMAPSession *session, gint msgnum, guint32 *uid)
1723 {
1724         gint ok;
1725         GPtrArray *argbuf;
1726         gchar *str;
1727         gint num;
1728
1729         *uid = 0;
1730         argbuf = g_ptr_array_new();
1731
1732         imap_cmd_gen_send(SESSION(session)->sock, "FETCH %d (UID)", msgnum);
1733         if ((ok = imap_cmd_ok(SESSION(session)->sock, argbuf)) != IMAP_SUCCESS)
1734                 THROW(ok);
1735
1736         str = search_array_contain_str(argbuf, "FETCH");
1737         if (!str) THROW(IMAP_ERROR);
1738
1739         if (sscanf(str, "%d FETCH (UID %d)", &num, uid) != 2 ||
1740             num != msgnum) {
1741                 g_warning("imap_get_uid(): invalid FETCH line.\n");
1742                 THROW(IMAP_ERROR);
1743         }
1744
1745 catch:
1746         ptr_array_free_strings(argbuf);
1747         g_ptr_array_free(argbuf, TRUE);
1748
1749         return ok;
1750 }
1751
1752 static gint imap_status(IMAPSession *session, IMAPFolder *folder,
1753                         const gchar *path,
1754                         gint *messages, gint *recent, gint *unseen,
1755                         guint32 *uid_validity)
1756 {
1757         gchar *real_path;
1758         gint ok;
1759         GPtrArray *argbuf;
1760         gchar *str;
1761
1762         *messages = *recent = *unseen = *uid_validity = 0;
1763
1764         argbuf = g_ptr_array_new();
1765
1766         real_path = imap_get_real_path(folder, path);
1767         if (strchr(real_path, ' ') != NULL)
1768                 imap_cmd_gen_send(SESSION(session)->sock, "STATUS \"%s\" "
1769                                         "(MESSAGES RECENT UNSEEN UIDVALIDITY)",
1770                                   real_path);
1771         else
1772                 imap_cmd_gen_send(SESSION(session)->sock, "STATUS %s "
1773                                         "(MESSAGES RECENT UNSEEN UIDVALIDITY)",
1774                                   real_path);
1775
1776         ok = imap_cmd_ok(SESSION(session)->sock, argbuf);
1777         if (ok != IMAP_SUCCESS) THROW(ok);
1778
1779         str = search_array_contain_str(argbuf, "STATUS");
1780         if (!str) THROW(IMAP_ERROR);
1781
1782         str = strchr(str, '(');
1783         if (!str) THROW(IMAP_ERROR);
1784         str++;
1785         while (*str != '\0' && *str != ')') {
1786                 while (*str == ' ') str++;
1787
1788                 if (!strncmp(str, "MESSAGES ", 9)) {
1789                         str += 9;
1790                         *messages = strtol(str, &str, 10);
1791                 } else if (!strncmp(str, "RECENT ", 7)) {
1792                         str += 7;
1793                         *recent = strtol(str, &str, 10);
1794                 } else if (!strncmp(str, "UNSEEN ", 7)) {
1795                         str += 7;
1796                         *unseen = strtol(str, &str, 10);
1797                 } else if (!strncmp(str, "UIDVALIDITY ", 12)) {
1798                         str += 12;
1799                         *uid_validity = strtoul(str, &str, 10);
1800                 } else {
1801                         g_warning("invalid STATUS response: %s\n", str);
1802                         break;
1803                 }
1804         }
1805
1806 catch:
1807         g_free(real_path);
1808         ptr_array_free_strings(argbuf);
1809         g_ptr_array_free(argbuf, TRUE);
1810
1811         return ok;
1812 }
1813
1814 #undef THROW
1815
1816
1817 /* low-level IMAP4rev1 commands */
1818
1819 static gint imap_cmd_login(SockInfo *sock,
1820                            const gchar *user, const gchar *pass)
1821 {
1822         gint ok;
1823
1824         if (strchr(user, ' ') != NULL)
1825                 imap_cmd_gen_send(sock, "LOGIN \"%s\" %s", user, pass);
1826         else
1827                 imap_cmd_gen_send(sock, "LOGIN %s %s", user, pass);
1828
1829         ok = imap_cmd_ok(sock, NULL);
1830         if (ok != IMAP_SUCCESS)
1831                 log_warning(_("IMAP4 login failed.\n"));
1832
1833         return ok;
1834 }
1835
1836 static gint imap_cmd_logout(SockInfo *sock)
1837 {
1838         imap_cmd_gen_send(sock, "LOGOUT");
1839         return imap_cmd_ok(sock, NULL);
1840 }
1841
1842 static gint imap_cmd_noop(SockInfo *sock)
1843 {
1844         imap_cmd_gen_send(sock, "NOOP");
1845         return imap_cmd_ok(sock, NULL);
1846 }
1847
1848 #define THROW(err) { ok = err; goto catch; }
1849
1850 static gint imap_cmd_namespace(SockInfo *sock, gchar **ns_str)
1851 {
1852         gint ok;
1853         GPtrArray *argbuf;
1854         gchar *str;
1855
1856         argbuf = g_ptr_array_new();
1857
1858         imap_cmd_gen_send(sock, "NAMESPACE");
1859         if ((ok = imap_cmd_ok(sock, argbuf)) != IMAP_SUCCESS) THROW(ok);
1860
1861         str = search_array_contain_str(argbuf, "NAMESPACE");
1862         if (!str) THROW(IMAP_ERROR);
1863
1864         *ns_str = g_strdup(str);
1865
1866 catch:
1867         ptr_array_free_strings(argbuf);
1868         g_ptr_array_free(argbuf, TRUE);
1869
1870         return ok;
1871 }
1872
1873 #undef THROW
1874
1875 static gint imap_cmd_list(SockInfo *sock, const gchar *ref,
1876                           const gchar *mailbox, GPtrArray *argbuf)
1877 {
1878         gchar *new_ref;
1879         gchar *new_mailbox;
1880
1881         if (!ref) ref = "\"\"";
1882         if (!mailbox) mailbox = "\"\"";
1883
1884         if (*ref != '"' && strchr(ref, ' ') != NULL)
1885                 new_ref = g_strdup_printf("\"%s\"", ref);
1886         else
1887                 new_ref = g_strdup(ref);
1888         if (*mailbox != '"' && strchr(mailbox, ' ') != NULL)
1889                 new_mailbox = g_strdup_printf("\"%s\"", mailbox);
1890         else
1891                 new_mailbox = g_strdup(mailbox);
1892
1893         imap_cmd_gen_send(sock, "LIST %s %s", new_ref, new_mailbox);
1894
1895         g_free(new_ref);
1896         g_free(new_mailbox);
1897
1898         return imap_cmd_ok(sock, argbuf);
1899 }
1900
1901 #define THROW goto catch
1902
1903 static gint imap_cmd_do_select(SockInfo *sock, const gchar *folder,
1904                                gboolean examine,
1905                                gint *exists, gint *recent, gint *unseen,
1906                                guint32 *uid_validity)
1907 {
1908         gint ok;
1909         gchar *resp_str;
1910         GPtrArray *argbuf;
1911         gchar *select_cmd;
1912
1913         *exists = *recent = *unseen = *uid_validity = 0;
1914         argbuf = g_ptr_array_new();
1915
1916         if (examine)
1917                 select_cmd = "EXAMINE";
1918         else
1919                 select_cmd = "SELECT";
1920
1921         if (strchr(folder, ' ') != NULL)
1922                 imap_cmd_gen_send(sock, "%s \"%s\"", select_cmd, folder);
1923         else
1924                 imap_cmd_gen_send(sock, "%s %s", select_cmd, folder);
1925
1926         if ((ok = imap_cmd_ok(sock, argbuf)) != IMAP_SUCCESS) THROW;
1927
1928         resp_str = search_array_contain_str(argbuf, "EXISTS");
1929         if (resp_str) {
1930                 if (sscanf(resp_str,"%d EXISTS", exists) != 1) {
1931                         g_warning("imap_cmd_select(): invalid EXISTS line.\n");
1932                         THROW;
1933                 }
1934         }
1935
1936         resp_str = search_array_contain_str(argbuf, "RECENT");
1937         if (resp_str) {
1938                 if (sscanf(resp_str, "%d RECENT", recent) != 1) {
1939                         g_warning("imap_cmd_select(): invalid RECENT line.\n");
1940                         THROW;
1941                 }
1942         }
1943
1944         resp_str = search_array_contain_str(argbuf, "UIDVALIDITY");
1945         if (resp_str) {
1946                 if (sscanf(resp_str, "OK [UIDVALIDITY %u] ", uid_validity)
1947                     != 1) {
1948                         g_warning("imap_cmd_select(): invalid UIDVALIDITY line.\n");
1949                         THROW;
1950                 }
1951         }
1952
1953         resp_str = search_array_contain_str(argbuf, "UNSEEN");
1954         if (resp_str) {
1955                 if (sscanf(resp_str, "OK [UNSEEN %d] ", unseen) != 1) {
1956                         g_warning("imap_cmd_select(): invalid UNSEEN line.\n");
1957                         THROW;
1958                 }
1959         }
1960
1961 catch:
1962         ptr_array_free_strings(argbuf);
1963         g_ptr_array_free(argbuf, TRUE);
1964
1965         return ok;
1966 }
1967
1968 static gint imap_cmd_select(SockInfo *sock, const gchar *folder,
1969                             gint *exists, gint *recent, gint *unseen,
1970                             guint32 *uid_validity)
1971 {
1972         return imap_cmd_do_select(sock, folder, FALSE,
1973                                   exists, recent, unseen, uid_validity);
1974 }
1975
1976 static gint imap_cmd_examine(SockInfo *sock, const gchar *folder,
1977                              gint *exists, gint *recent, gint *unseen,
1978                              guint32 *uid_validity)
1979 {
1980         return imap_cmd_do_select(sock, folder, TRUE,
1981                                   exists, recent, unseen, uid_validity);
1982 }
1983
1984 #undef THROW
1985
1986 static gint imap_cmd_create(SockInfo *sock, const gchar *folder)
1987 {
1988         if (strchr(folder, ' ') != NULL)
1989                 imap_cmd_gen_send(sock, "CREATE \"%s\"", folder);
1990         else
1991                 imap_cmd_gen_send(sock, "CREATE %s", folder);
1992
1993         return imap_cmd_ok(sock, NULL);
1994 }
1995
1996 static gint imap_cmd_delete(SockInfo *sock, const gchar *folder)
1997 {
1998         if (strchr(folder, ' ') != NULL)
1999                 imap_cmd_gen_send(sock, "DELETE \"%s\"", folder);
2000         else
2001                 imap_cmd_gen_send(sock, "DELETE %s", folder);
2002
2003         return imap_cmd_ok(sock, NULL);
2004 }
2005
2006 static gint imap_cmd_fetch(SockInfo *sock, guint32 uid, const gchar *filename)
2007 {
2008         gint ok;
2009         gchar buf[IMAPBUFSIZE];
2010         gchar *cur_pos;
2011         gchar size_str[32];
2012         glong size_num;
2013
2014         g_return_val_if_fail(filename != NULL, IMAP_ERROR);
2015
2016         imap_cmd_gen_send(sock, "UID FETCH %d BODY[]", uid);
2017
2018         if (sock_gets(sock, buf, sizeof(buf)) < 0)
2019                 return IMAP_ERROR;
2020         strretchomp(buf);
2021         if (buf[0] != '*' || buf[1] != ' ')
2022                 return IMAP_ERROR;
2023         log_print("IMAP4< %s\n", buf);
2024
2025         cur_pos = strchr(buf, '{');
2026         g_return_val_if_fail(cur_pos != NULL, IMAP_ERROR);
2027         cur_pos = strchr_cpy(cur_pos + 1, '}', size_str, sizeof(size_str));
2028         g_return_val_if_fail(cur_pos != NULL, IMAP_ERROR);
2029         size_num = atol(size_str);
2030
2031         if (*cur_pos != '\0') return IMAP_ERROR;
2032
2033         if (recv_bytes_write_to_file(sock, size_num, filename) != 0)
2034                 return IMAP_ERROR;
2035
2036         if (imap_cmd_gen_recv(sock, buf, sizeof(buf)) != IMAP_SUCCESS)
2037                 return IMAP_ERROR;
2038
2039         if (buf[0] == '\0' || buf[strlen(buf) - 1] != ')')
2040                 return IMAP_ERROR;
2041
2042         ok = imap_cmd_ok(sock, NULL);
2043
2044         return ok;
2045 }
2046
2047 static gint imap_cmd_append(SockInfo *sock, const gchar *destfolder,
2048                             const gchar *file)
2049 {
2050         gint ok;
2051         gint size;
2052
2053         g_return_val_if_fail(file != NULL, IMAP_ERROR);
2054
2055         size = get_file_size(file);
2056         imap_cmd_gen_send(sock, "APPEND %s {%d}", destfolder, size);
2057         ok = imap_cmd_ok(sock, NULL);
2058         if (ok != IMAP_SUCCESS) {
2059                 log_warning(_("can't append %s to %s\n"), file, destfolder);
2060                 return -1;
2061         }
2062
2063         return ok;
2064 }
2065
2066 static gint imap_cmd_copy(SockInfo *sock, guint32 uid, const gchar *destfolder)
2067 {
2068         gint ok;
2069
2070         g_return_val_if_fail(destfolder != NULL, IMAP_ERROR);
2071
2072         if (strchr(destfolder, ' ') != NULL)
2073                 imap_cmd_gen_send(sock, "UID COPY %d \"%s\"", uid, destfolder);
2074         else
2075                 imap_cmd_gen_send(sock, "UID COPY %d %s", uid, destfolder);
2076
2077         ok = imap_cmd_ok(sock, NULL);
2078         if (ok != IMAP_SUCCESS) {
2079                 log_warning(_("can't copy %d to %s\n"), uid, destfolder);
2080                 return -1;
2081         }
2082
2083         return ok;
2084 }
2085
2086 gint imap_cmd_envelope(SockInfo *sock, guint32 first_uid, guint32 last_uid)
2087 {
2088         imap_cmd_gen_send
2089                 (sock, "UID FETCH %d:%d (UID FLAGS RFC822.SIZE ENVELOPE)",
2090                  first_uid, last_uid);
2091
2092         return IMAP_SUCCESS;
2093 }
2094
2095 static gint imap_cmd_store(SockInfo *sock, guint32 first_uid, guint32 last_uid,
2096                            gchar *sub_cmd)
2097 {
2098         gint ok;
2099
2100         imap_cmd_gen_send(sock, "UID STORE %d:%d %s",
2101                           first_uid, last_uid, sub_cmd);
2102
2103         if ((ok = imap_cmd_ok(sock, NULL)) != IMAP_SUCCESS) {
2104                 log_warning(_("error while imap command: STORE %d:%d %s\n"),
2105                             first_uid, last_uid, sub_cmd);
2106                 return ok;
2107         }
2108
2109         return IMAP_SUCCESS;
2110 }
2111
2112 static gint imap_cmd_expunge(SockInfo *sock)
2113 {
2114         gint ok;
2115
2116         imap_cmd_gen_send(sock, "EXPUNGE");
2117         if ((ok = imap_cmd_ok(sock, NULL)) != IMAP_SUCCESS) {
2118                 log_warning(_("error while imap command: EXPUNGE\n"));
2119                 return ok;
2120         }
2121
2122         return IMAP_SUCCESS;
2123 }
2124
2125 static gint imap_cmd_ok(SockInfo *sock, GPtrArray *argbuf)
2126 {
2127         gint ok;
2128         gchar buf[IMAPBUFSIZE];
2129         gint cmd_num;
2130         gchar cmd_status[IMAPBUFSIZE];
2131
2132         while ((ok = imap_cmd_gen_recv(sock, buf, sizeof(buf)))
2133                == IMAP_SUCCESS) {
2134                 if (buf[0] == '*' && buf[1] == ' ') {
2135                         if (argbuf)
2136                                 g_ptr_array_add(argbuf, g_strdup(buf + 2));
2137                         continue;
2138                 }
2139
2140                 if (sscanf(buf, "%d %s", &cmd_num, cmd_status) < 2)
2141                         return IMAP_ERROR;
2142                 else if (cmd_num == imap_cmd_count &&
2143                          !strcmp(cmd_status, "OK")) {
2144                         if (argbuf)
2145                                 g_ptr_array_add(argbuf, g_strdup(buf));
2146                         return IMAP_SUCCESS;
2147                 } else
2148                         return IMAP_ERROR;
2149         }
2150
2151         return ok;
2152 }
2153
2154 static void imap_cmd_gen_send(SockInfo *sock, const gchar *format, ...)
2155 {
2156         gchar buf[IMAPBUFSIZE];
2157         gchar tmp[IMAPBUFSIZE];
2158         gchar *p;
2159         va_list args;
2160
2161         va_start(args, format);
2162         g_vsnprintf(tmp, sizeof(tmp), format, args);
2163         va_end(args);
2164
2165         imap_cmd_count++;
2166
2167         g_snprintf(buf, sizeof(buf), "%d %s\r\n", imap_cmd_count, tmp);
2168         if (!strncasecmp(tmp, "LOGIN ", 6) && (p = strchr(tmp + 6, ' '))) {
2169                 *p = '\0';
2170                 log_print("IMAP4> %d %s ********\n", imap_cmd_count, tmp);
2171         } else
2172                 log_print("IMAP4> %d %s\n", imap_cmd_count, tmp);
2173
2174         sock_write(sock, buf, strlen(buf));
2175 }
2176
2177 static gint imap_cmd_gen_recv(SockInfo *sock, gchar *buf, gint size)
2178 {
2179         if (sock_gets(sock, buf, size) == -1)
2180                 return IMAP_SOCKET;
2181
2182         strretchomp(buf);
2183
2184         log_print("IMAP4< %s\n", buf);
2185
2186         return IMAP_SUCCESS;
2187 }
2188
2189
2190 /* misc utility functions */
2191
2192 static gchar *strchr_cpy(const gchar *src, gchar ch, gchar *dest, gint len)
2193 {
2194         gchar *tmp;
2195
2196         dest[0] = '\0';
2197         tmp = strchr(src, ch);
2198         if (!tmp)
2199                 return NULL;
2200
2201         memcpy(dest, src, MIN(tmp - src, len - 1));
2202         dest[MIN(tmp - src, len - 1)] = '\0';
2203
2204         return tmp + 1;
2205 }
2206
2207 static gchar *search_array_contain_str(GPtrArray *array, gchar *str)
2208 {
2209         gint i;
2210
2211         for (i = 0; i < array->len; i++) {
2212                 gchar *tmp;
2213
2214                 tmp = g_ptr_array_index(array, i);
2215                 if (strstr(tmp, str) != NULL)
2216                         return tmp;
2217         }
2218
2219         return NULL;
2220 }
2221
2222 static void imap_path_separator_subst(gchar *str, gchar separator)
2223 {
2224         if (separator && separator != '/')
2225                 subst_char(str, '/', separator);
2226 }