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