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