1 /* -*- mode: c; c-basic-offset: 4; tab-width: 4; indent-tabs-mode: t -*-
3 * Conky, a system monitor, based on torsmo
5 * Any original torsmo code is licensed under the BSD license
7 * All code written since the fork of torsmo is licensed under the GPL
9 * Please see COPYING for details
11 * Copyright (c) 2004, Hannu Saransaari and Lauri Hakkarainen
12 * Copyright (c) 2005-2009 Brenden Matthews, Philip Kovacs, et. al.
14 * All rights reserved.
16 * This program is free software: you can redistribute it and/or modify
17 * it under the terms of the GNU General Public License as published by
18 * the Free Software Foundation, either version 3 of the License, or
19 * (at your option) any later version.
21 * This program is distributed in the hope that it will be useful,
22 * but WITHOUT ANY WARRANTY; without even the implied warranty of
23 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
24 * GNU General Public License for more details.
25 * You should have received a copy of the GNU General Public License
26 * along with this program. If not, see <http://www.gnu.org/licenses/>.
28 * vim: ts=4 sw=4 noet ai cindent syntax=c
43 #include <netinet/in.h>
45 #include <sys/socket.h>
48 #include <sys/param.h>
54 /* MAX() is defined by a header included from conky.h
55 * maybe once this is not true anymore, so have an alternative
58 * #define MAX(a, b) ((a > b) ? a : b)
61 char *current_mail_spool;
63 void update_mail_count(struct local_mail_s *mail)
71 /* TODO: use that fine file modification notify on Linux 2.4 */
73 /* don't check mail so often (9.5s is minimum interval) */
74 if (current_update_time - mail->last_update < 9.5) {
77 mail->last_update = current_update_time;
80 if (stat(mail->mbox, &st)) {
84 NORM_ERR("can't stat %s: %s", mail->mbox, strerror(errno));
91 if (S_ISDIR(st.st_mode)) {
94 struct dirent *dirent;
97 mail->mail_count = mail->new_mail_count = 0;
98 mail->seen_mail_count = mail->unseen_mail_count = 0;
99 mail->flagged_mail_count = mail->unflagged_mail_count = 0;
100 mail->forwarded_mail_count = mail->unforwarded_mail_count = 0;
101 mail->replied_mail_count = mail->unreplied_mail_count = 0;
102 mail->draft_mail_count = mail->trashed_mail_count = 0;
103 dirname = (char *) malloc(sizeof(char) * (strlen(mail->mbox) + 5));
108 strcpy(dirname, mail->mbox);
109 strcat(dirname, "/");
110 /* checking the cur subdirectory */
111 strcat(dirname, "cur");
113 dir = opendir(dirname);
115 NORM_ERR("cannot open directory");
119 dirent = readdir(dir);
121 /* . and .. are skipped */
122 if (dirent->d_name[0] != '.') {
124 mailflags = (char *) malloc(sizeof(char) * strlen(strrchr(dirent->d_name, ',')));
130 strcpy(mailflags, strrchr(dirent->d_name, ','));
131 if (!strchr(mailflags, 'T')) { /* The message is not in the trash */
132 if (strchr(mailflags, 'S')) { /*The message has been seen */
133 mail->seen_mail_count++;
135 mail->unseen_mail_count++;
137 if (strchr(mailflags, 'F')) { /*The message was flagged */
138 mail->flagged_mail_count++;
140 mail->unflagged_mail_count++;
142 if (strchr(mailflags, 'P')) { /*The message was forwarded */
143 mail->forwarded_mail_count++;
145 mail->unforwarded_mail_count++;
147 if (strchr(mailflags, 'R')) { /*The message was replied */
148 mail->replied_mail_count++;
150 mail->unreplied_mail_count++;
152 if (strchr(mailflags, 'D')) { /*The message is a draft */
153 mail->draft_mail_count++;
156 mail->trashed_mail_count++;
160 dirent = readdir(dir);
164 dirname[strlen(dirname) - 3] = '\0';
165 strcat(dirname, "new");
167 dir = opendir(dirname);
169 NORM_ERR("cannot open directory");
173 dirent = readdir(dir);
175 /* . and .. are skipped */
176 if (dirent->d_name[0] != '.') {
177 mail->new_mail_count++;
179 mail->unseen_mail_count++; /* new messages cannot have been seen */
181 dirent = readdir(dir);
190 if (st.st_mtime != mail->last_mtime) {
191 /* yippee, modification time has changed, let's read mail count! */
194 int reading_status = 0;
196 /* could lock here but I don't think it's really worth it because
197 * this isn't going to write mail spool */
199 mail->new_mail_count = mail->mail_count = 0;
201 /* these flags are not supported for mbox */
202 mail->seen_mail_count = mail->unseen_mail_count = -1;
203 mail->flagged_mail_count = mail->unflagged_mail_count = -1;
204 mail->forwarded_mail_count = mail->unforwarded_mail_count = -1;
205 mail->replied_mail_count = mail->unreplied_mail_count = -1;
206 mail->draft_mail_count = mail->trashed_mail_count = -1;
208 fp = open_file(mail->mbox, &rep);
213 /* NOTE: adds mail as new if there isn't Status-field at all */
219 if (fgets(buf, 128, fp) == NULL) {
223 if (strncmp(buf, "From ", 5) == 0) {
224 /* ignore MAILER-DAEMON */
225 if (strncmp(buf + 5, "MAILER-DAEMON ", 14) != 0) {
229 if (reading_status == 1) {
230 mail->new_mail_count++;
236 if (reading_status == 1
237 && strncmp(buf, "X-Mozilla-Status:", 17) == 0) {
238 int xms = strtol(buf + 17, NULL, 16);
239 /* check that mail isn't marked for deletion */
241 mail->trashed_mail_count++;
243 /* Don't check whether the trashed email is unread */
246 /* check that mail isn't already read */
247 if (!(xms & 0x0001)) {
248 mail->new_mail_count++;
252 /* check for an additional X-Status header */
256 if (reading_status == 1 && strncmp(buf, "Status:", 7) == 0) {
257 /* check that mail isn't already read */
258 if (strchr(buf + 7, 'R') == NULL) {
259 mail->new_mail_count++;
266 if (reading_status >= 1 && strncmp(buf, "X-Status:", 9) == 0) {
267 /* check that mail isn't marked for deletion */
268 if (strchr(buf + 9, 'D') != NULL) {
269 mail->trashed_mail_count++;
270 /* If the mail was previously detected as new,
271 subtract it from the new mail count */
273 mail->new_mail_count--;
282 while (strchr(buf, '\n') == NULL && !feof(fp)) {
289 if (reading_status) {
290 mail->new_mail_count++;
293 mail->last_mtime = st.st_mtime;
297 #define MAXDATASIZE 1000
299 struct mail_s *parse_mail_args(char type, const char *arg)
304 mail = malloc(sizeof(struct mail_s));
305 memset(mail, 0, sizeof(struct mail_s));
307 if (sscanf(arg, "%128s %128s %128s", mail->host, mail->user, mail->pass)
309 if (type == POP3_TYPE) {
310 NORM_ERR("Scanning POP3 args failed");
311 } else if (type == IMAP_TYPE) {
312 NORM_ERR("Scanning IMAP args failed");
316 // see if password needs prompting
317 if (mail->pass[0] == '*' && mail->pass[1] == '\0') {
318 int fp = fileno(stdin);
321 tcgetattr(fp, &term);
322 term.c_lflag &= ~ECHO;
323 tcsetattr(fp, TCSANOW, &term);
324 printf("Enter mailbox password (%s@%s): ", mail->user, mail->host);
325 scanf("%128s", mail->pass);
327 term.c_lflag |= ECHO;
328 tcsetattr(fp, TCSANOW, &term);
330 // now we check for optional args
331 tmp = strstr(arg, "-r ");
334 sscanf(tmp, "%u", &mail->retries);
336 mail->retries = 5; // 5 retries after failure
338 tmp = strstr(arg, "-i ");
341 sscanf(tmp, "%f", &mail->interval);
343 mail->interval = 300; // 5 minutes
345 tmp = strstr(arg, "-p ");
348 sscanf(tmp, "%lu", &mail->port);
350 if (type == POP3_TYPE) {
351 mail->port = 110; // default pop3 port
352 } else if (type == IMAP_TYPE) {
353 mail->port = 143; // default imap port
356 if (type == IMAP_TYPE) {
357 tmp = strstr(arg, "-f ");
360 sscanf(tmp, "%s", mail->folder);
362 strncpy(mail->folder, "INBOX", 128); // default imap inbox
365 tmp = strstr(arg, "-e ");
370 if (tmp[0] == '\'') {
371 len = strstr(tmp + 1, "'") - tmp - 1;
376 strncpy(mail->command, tmp + 1, len);
378 mail->command[0] = '\0';
380 mail->p_timed_thread = NULL;
384 int imap_command(int sockfd, const char *command, char *response, const char *verify)
386 struct timeval timeout;
388 int res, numbytes = 0;
389 if (send(sockfd, command, strlen(command), 0) == -1) {
393 timeout.tv_sec = 60; // 60 second timeout i guess
396 FD_SET(sockfd, &fdset);
397 res = select(sockfd + 1, &fdset, NULL, NULL, &timeout);
399 if ((numbytes = recv(sockfd, response, MAXDATASIZE - 1, 0)) == -1) {
404 DBGP2("imap_command() command: %s", command);
405 DBGP2("imap_command() received: %s", response);
406 response[numbytes] = '\0';
407 if (strstr(response, verify) == NULL) {
413 int imap_check_status(char *recvbuf, struct mail_s *mail)
416 reply = strstr(recvbuf, " (MESSAGES ");
417 if (!reply || strlen(reply) < 2) {
421 *strchr(reply, ')') = '\0';
423 NORM_ERR("Error parsing IMAP response: %s", recvbuf);
426 timed_thread_lock(mail->p_timed_thread);
427 sscanf(reply, "MESSAGES %lu UNSEEN %lu", &mail->messages,
429 timed_thread_unlock(mail->p_timed_thread);
434 void imap_unseen_command(struct mail_s *mail, unsigned long old_unseen, unsigned long old_messages)
436 if (strlen(mail->command) > 1 && (mail->unseen > old_unseen
437 || (mail->messages > old_messages && mail->unseen > 0))) {
439 if (system(mail->command) == -1) {
445 void *imap_thread(void *arg)
447 int sockfd, numbytes;
448 char recvbuf[MAXDATASIZE];
449 char sendbuf[MAXDATASIZE];
450 unsigned int fail = 0;
451 unsigned long old_unseen = ULONG_MAX;
452 unsigned long old_messages = ULONG_MAX;
453 struct stat stat_buf;
454 struct hostent he, *he_res = 0;
457 struct sockaddr_in their_addr; // connector's address information
458 struct mail_s *mail = (struct mail_s *)arg;
460 int threadfd = timed_thread_readfd(mail->p_timed_thread);
462 #ifdef HAVE_GETHOSTBYNAME_R
463 if (gethostbyname_r(mail->host, &he, hostbuff, sizeof(hostbuff), &he_res, &he_errno)) { // get the host info
464 NORM_ERR("IMAP gethostbyname_r: %s", hstrerror(h_errno));
467 #else /* HAVE_GETHOSTBYNAME_R */
468 if ((he_res = gethostbyname(mail->host)) == NULL) { // get the host info
469 herror("gethostbyname");
472 #endif /* HAVE_GETHOSTBYNAME_R */
473 while (fail < mail->retries) {
474 struct timeval timeout;
479 NORM_ERR("Trying IMAP connection again for %s@%s (try %u/%u)",
480 mail->user, mail->host, fail + 1, mail->retries);
483 if ((sockfd = socket(PF_INET, SOCK_STREAM, 0)) == -1) {
490 their_addr.sin_family = AF_INET;
491 // short, network byte order
492 their_addr.sin_port = htons(mail->port);
493 their_addr.sin_addr = *((struct in_addr *) he_res->h_addr);
494 // zero the rest of the struct
495 memset(&(their_addr.sin_zero), '\0', 8);
497 if (connect(sockfd, (struct sockaddr *) &their_addr,
498 sizeof(struct sockaddr)) == -1) {
504 timeout.tv_sec = 60; // 60 second timeout i guess
507 FD_SET(sockfd, &fdset);
508 res = select(sockfd + 1, &fdset, NULL, NULL, &timeout);
510 if ((numbytes = recv(sockfd, recvbuf, MAXDATASIZE - 1, 0)) == -1) {
516 NORM_ERR("IMAP connection failed: timeout");
520 recvbuf[numbytes] = '\0';
521 DBGP2("imap_thread() received: %s", recvbuf);
522 if (strstr(recvbuf, "* OK") != recvbuf) {
523 NORM_ERR("IMAP connection failed, probably not an IMAP server");
527 strncpy(sendbuf, "a1 login ", MAXDATASIZE);
528 strncat(sendbuf, mail->user, MAXDATASIZE - strlen(sendbuf) - 1);
529 strncat(sendbuf, " ", MAXDATASIZE - strlen(sendbuf) - 1);
530 strncat(sendbuf, mail->pass, MAXDATASIZE - strlen(sendbuf) - 1);
531 strncat(sendbuf, "\r\n", MAXDATASIZE - strlen(sendbuf) - 1);
532 if (imap_command(sockfd, sendbuf, recvbuf, "a1 OK")) {
536 if (strstr(recvbuf, " IDLE ") != NULL) {
540 strncpy(sendbuf, "a2 STATUS ", MAXDATASIZE);
541 strncat(sendbuf, mail->folder, MAXDATASIZE - strlen(sendbuf) - 1);
542 strncat(sendbuf, " (MESSAGES UNSEEN)\r\n",
543 MAXDATASIZE - strlen(sendbuf) - 1);
544 if (imap_command(sockfd, sendbuf, recvbuf, "a2 OK")) {
549 if (imap_check_status(recvbuf, mail)) {
553 imap_unseen_command(mail, old_unseen, old_messages);
555 old_unseen = mail->unseen;
556 old_messages = mail->messages;
559 strncpy(sendbuf, "a4 SELECT ", MAXDATASIZE);
560 strncat(sendbuf, mail->folder, MAXDATASIZE - strlen(sendbuf) - 1);
561 strncat(sendbuf, "\r\n", MAXDATASIZE - strlen(sendbuf) - 1);
562 if (imap_command(sockfd, sendbuf, recvbuf, "a4 OK")) {
567 strncpy(sendbuf, "a5 IDLE\r\n", MAXDATASIZE);
568 if (imap_command(sockfd, sendbuf, recvbuf, "+ idling")) {
576 * RFC 2177 says we have to re-idle every 29 minutes.
577 * We'll do it every 20 minutes to be safe.
579 timeout.tv_sec = 1200;
583 FD_SET(sockfd, &fdset);
584 FD_SET(threadfd, &fdset);
585 res = select(MAX(sockfd + 1, threadfd + 1), &fdset, NULL, NULL, NULL);
586 if (timed_thread_test(mail->p_timed_thread, 1) || (res == -1 && errno == EINTR) || FD_ISSET(threadfd, &fdset)) {
587 if ((fstat(sockfd, &stat_buf) == 0) && S_ISSOCK(stat_buf.st_mode)) {
588 /* if a valid socket, close it */
591 timed_thread_exit(mail->p_timed_thread);
592 } else if (res > 0) {
593 if ((numbytes = recv(sockfd, recvbuf, MAXDATASIZE - 1, 0)) == -1) {
594 perror("recv idling");
602 recvbuf[numbytes] = '\0';
603 DBGP2("imap_thread() received: %s", recvbuf);
604 if (strlen(recvbuf) > 2) {
605 unsigned long messages, recent;
607 char force_check = 0;
608 buf = strstr(buf, "EXISTS");
609 while (buf && strlen(buf) > 1 && strstr(buf + 1, "EXISTS")) {
610 buf = strstr(buf + 1, "EXISTS");
613 // back up until we reach '*'
614 while (buf >= recvbuf && buf[0] != '*') {
617 if (sscanf(buf, "* %lu EXISTS\r\n", &messages) == 1) {
618 timed_thread_lock(mail->p_timed_thread);
619 if (mail->messages != messages) {
621 mail->messages = messages;
623 timed_thread_unlock(mail->p_timed_thread);
627 buf = strstr(buf, "RECENT");
628 while (buf && strlen(buf) > 1 && strstr(buf + 1, "RECENT")) {
629 buf = strstr(buf + 1, "RECENT");
632 // back up until we reach '*'
633 while (buf >= recvbuf && buf[0] != '*') {
636 if (sscanf(buf, "* %lu RECENT\r\n", &recent) != 1) {
641 * check if we got a FETCH from server, recent was
642 * something other than 0, or we had a timeout
645 if (recent > 0 || (buf && strstr(buf, " FETCH ")) || timeout.tv_sec == 0 || force_check) {
646 // re-check messages and unseen
647 if (imap_command(sockfd, "DONE\r\n", recvbuf, "a5 OK")) {
651 strncpy(sendbuf, "a2 STATUS ", MAXDATASIZE);
652 strncat(sendbuf, mail->folder, MAXDATASIZE - strlen(sendbuf) - 1);
653 strncat(sendbuf, " (MESSAGES UNSEEN)\r\n",
654 MAXDATASIZE - strlen(sendbuf) - 1);
655 if (imap_command(sockfd, sendbuf, recvbuf, "a2 OK")) {
659 if (imap_check_status(recvbuf, mail)) {
663 strncpy(sendbuf, "a5 IDLE\r\n", MAXDATASIZE);
664 if (imap_command(sockfd, sendbuf, recvbuf, "+ idling")) {
670 * check if we got a BYE from server
673 if (buf && strstr(buf, "* BYE")) {
674 // need to re-connect
681 imap_unseen_command(mail, old_unseen, old_messages);
683 old_unseen = mail->unseen;
684 old_messages = mail->messages;
688 strncpy(sendbuf, "a3 logout\r\n", MAXDATASIZE);
689 if (send(sockfd, sendbuf, strlen(sendbuf), 0) == -1) {
694 timeout.tv_sec = 60; // 60 second timeout i guess
697 FD_SET(sockfd, &fdset);
698 res = select(sockfd + 1, &fdset, NULL, NULL, &timeout);
700 if ((numbytes = recv(sockfd, recvbuf, MAXDATASIZE - 1, 0)) == -1) {
706 recvbuf[numbytes] = '\0';
707 DBGP2("imap_thread() received: %s", recvbuf);
708 if (strstr(recvbuf, "a3 OK") == NULL) {
709 NORM_ERR("IMAP logout failed: %s", recvbuf);
715 if ((fstat(sockfd, &stat_buf) == 0) && S_ISSOCK(stat_buf.st_mode)) {
716 /* if a valid socket, close it */
719 if (timed_thread_test(mail->p_timed_thread, 0)) {
720 timed_thread_exit(mail->p_timed_thread);
728 int pop3_command(int sockfd, const char *command, char *response, const char *verify)
730 struct timeval timeout;
732 int res, numbytes = 0;
733 if (send(sockfd, command, strlen(command), 0) == -1) {
737 timeout.tv_sec = 60; // 60 second timeout i guess
740 FD_SET(sockfd, &fdset);
741 res = select(sockfd + 1, &fdset, NULL, NULL, &timeout);
743 if ((numbytes = recv(sockfd, response, MAXDATASIZE - 1, 0)) == -1) {
748 DBGP2("pop3_command() received: %s", response);
749 response[numbytes] = '\0';
750 if (strstr(response, verify) == NULL) {
756 void *pop3_thread(void *arg)
758 int sockfd, numbytes;
759 char recvbuf[MAXDATASIZE];
760 char sendbuf[MAXDATASIZE];
762 unsigned int fail = 0;
763 unsigned long old_unseen = ULONG_MAX;
764 struct stat stat_buf;
765 struct hostent he, *he_res = 0;
768 struct sockaddr_in their_addr; // connector's address information
769 struct mail_s *mail = (struct mail_s *)arg;
771 #ifdef HAVE_GETHOSTBYNAME_R
772 if (gethostbyname_r(mail->host, &he, hostbuff, sizeof(hostbuff), &he_res, &he_errno)) { // get the host info
773 NORM_ERR("POP3 gethostbyname_r: %s", hstrerror(h_errno));
776 #else /* HAVE_GETHOSTBYNAME_R */
777 if ((he_res = gethostbyname(mail->host)) == NULL) { // get the host info
778 herror("gethostbyname");
781 #endif /* HAVE_GETHOSTBYNAME_R */
782 while (fail < mail->retries) {
783 struct timeval timeout;
788 NORM_ERR("Trying POP3 connection again for %s@%s (try %u/%u)",
789 mail->user, mail->host, fail + 1, mail->retries);
792 if ((sockfd = socket(PF_INET, SOCK_STREAM, 0)) == -1) {
799 their_addr.sin_family = AF_INET;
800 // short, network byte order
801 their_addr.sin_port = htons(mail->port);
802 their_addr.sin_addr = *((struct in_addr *) he_res->h_addr);
803 // zero the rest of the struct
804 memset(&(their_addr.sin_zero), '\0', 8);
806 if (connect(sockfd, (struct sockaddr *) &their_addr,
807 sizeof(struct sockaddr)) == -1) {
813 timeout.tv_sec = 60; // 60 second timeout i guess
816 FD_SET(sockfd, &fdset);
817 res = select(sockfd + 1, &fdset, NULL, NULL, &timeout);
819 if ((numbytes = recv(sockfd, recvbuf, MAXDATASIZE - 1, 0)) == -1) {
825 NORM_ERR("POP3 connection failed: timeout\n");
829 DBGP2("pop3_thread received: %s", recvbuf);
830 recvbuf[numbytes] = '\0';
831 if (strstr(recvbuf, "+OK ") != recvbuf) {
832 NORM_ERR("POP3 connection failed, probably not a POP3 server");
836 strncpy(sendbuf, "USER ", MAXDATASIZE);
837 strncat(sendbuf, mail->user, MAXDATASIZE - strlen(sendbuf) - 1);
838 strncat(sendbuf, "\r\n", MAXDATASIZE - strlen(sendbuf) - 1);
839 if (pop3_command(sockfd, sendbuf, recvbuf, "+OK ")) {
844 strncpy(sendbuf, "PASS ", MAXDATASIZE);
845 strncat(sendbuf, mail->pass, MAXDATASIZE - strlen(sendbuf) - 1);
846 strncat(sendbuf, "\r\n", MAXDATASIZE - strlen(sendbuf) - 1);
847 if (pop3_command(sockfd, sendbuf, recvbuf, "+OK ")) {
848 NORM_ERR("POP3 server login failed: %s", recvbuf);
853 strncpy(sendbuf, "STAT\r\n", MAXDATASIZE);
854 if (pop3_command(sockfd, sendbuf, recvbuf, "+OK ")) {
860 // now we get the data
863 NORM_ERR("Error parsing POP3 response: %s", recvbuf);
867 timed_thread_lock(mail->p_timed_thread);
868 sscanf(reply, "%lu %lu", &mail->unseen, &mail->used);
869 timed_thread_unlock(mail->p_timed_thread);
872 strncpy(sendbuf, "QUIT\r\n", MAXDATASIZE);
873 if (pop3_command(sockfd, sendbuf, recvbuf, "+OK")) {
874 NORM_ERR("POP3 logout failed: %s", recvbuf);
879 if (strlen(mail->command) > 1 && mail->unseen > old_unseen) {
881 if (system(mail->command) == -1) {
886 old_unseen = mail->unseen;
888 if ((fstat(sockfd, &stat_buf) == 0) && S_ISSOCK(stat_buf.st_mode)) {
889 /* if a valid socket, close it */
892 if (timed_thread_test(mail->p_timed_thread, 0)) {
893 timed_thread_exit(mail->p_timed_thread);