blob: f37db03d53080b951fe7804f8cb3de22978ea24e [file] [log] [blame]
Denis Vlasenko239d06b2008-11-06 23:42:42 +00001/* vi: set sw=4 ts=4: */
2/*
3 * popmaildir: a simple yet powerful POP3 client
4 * Delivers contents of remote mailboxes to local Maildir
5 *
6 * Inspired by original utility by Nikola Vladov
7 *
8 * Copyright (C) 2008 by Vladimir Dronnikov <dronnikov@gmail.com>
9 *
Denys Vlasenko0ef64bd2010-08-16 20:14:46 +020010 * Licensed under GPLv2, see file LICENSE in this source tree.
Denis Vlasenko239d06b2008-11-06 23:42:42 +000011 */
12#include "libbb.h"
13#include "mail.h"
14
15static void pop3_checkr(const char *fmt, const char *param, char **ret)
16{
17 const char *msg = command(fmt, param);
18 char *answer = xmalloc_fgetline(stdin);
Denis Vlasenko4abaec52009-03-12 15:35:26 +000019 if (answer && '+' == answer[0]) {
Denis Vlasenko239d06b2008-11-06 23:42:42 +000020 if (timeout)
21 alarm(0);
Denis Vlasenko4abaec52009-03-12 15:35:26 +000022 if (ret) {
23 // skip "+OK "
24 memmove(answer, answer + 4, strlen(answer) - 4);
25 *ret = answer;
26 } else
Denis Vlasenko239d06b2008-11-06 23:42:42 +000027 free(answer);
28 return;
29 }
30 bb_error_msg_and_die("%s failed: %s", msg, answer);
31}
32
33static void pop3_check(const char *fmt, const char *param)
34{
35 pop3_checkr(fmt, param, NULL);
36}
37
38int popmaildir_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
39int popmaildir_main(int argc UNUSED_PARAM, char **argv)
40{
41 char *buf;
42 unsigned nmsg;
43 char *hostname;
44 pid_t pid;
45 const char *retr;
46#if ENABLE_FEATURE_POPMAILDIR_DELIVERY
47 const char *delivery;
48#endif
49 unsigned opt_nlines = 0;
50
51 enum {
52 OPT_b = 1 << 0, // -b binary mode. Ignored
53 OPT_d = 1 << 1, // -d,-dd,-ddd debug. Ignored
54 OPT_m = 1 << 2, // -m show used memory. Ignored
55 OPT_V = 1 << 3, // -V version. Ignored
56 OPT_c = 1 << 4, // -c use tcpclient. Ignored
57 OPT_a = 1 << 5, // -a use APOP protocol
58 OPT_s = 1 << 6, // -s skip authorization
59 OPT_T = 1 << 7, // -T get messages with TOP instead with RETR
60 OPT_k = 1 << 8, // -k keep retrieved messages on the server
61 OPT_t = 1 << 9, // -t90 set timeout to 90 sec
62 OPT_R = 1 << 10, // -R20000 remove old messages on the server >= 20000 bytes (requires -k). Ignored
63 OPT_Z = 1 << 11, // -Z11-23 remove messages from 11 to 23 (dangerous). Ignored
64 OPT_L = 1 << 12, // -L50000 not retrieve new messages >= 50000 bytes. Ignored
65 OPT_H = 1 << 13, // -H30 type first 30 lines of a message; (-L12000 -H30). Ignored
66 OPT_M = 1 << 14, // -M\"program arg1 arg2 ...\"; deliver by program. Treated like -F
67 OPT_F = 1 << 15, // -F\"program arg1 arg2 ...\"; filter by program. Treated like -M
68 };
69
70 // init global variables
71 INIT_G();
72
73 // parse options
74 opt_complementary = "-1:dd:t+:R+:L+:H+";
75 opts = getopt32(argv,
Denis Vlasenko5e34ff22009-04-21 11:09:40 +000076 "bdmVcasTkt:" "R:Z:L:H:" IF_FEATURE_POPMAILDIR_DELIVERY("M:F:"),
Denis Vlasenko239d06b2008-11-06 23:42:42 +000077 &timeout, NULL, NULL, NULL, &opt_nlines
Denis Vlasenko5e34ff22009-04-21 11:09:40 +000078 IF_FEATURE_POPMAILDIR_DELIVERY(, &delivery, &delivery) // we treat -M and -F the same
Denis Vlasenko239d06b2008-11-06 23:42:42 +000079 );
80 //argc -= optind;
81 argv += optind;
82
83 // get auth info
84 if (!(opts & OPT_s))
85 get_cred_or_die(STDIN_FILENO);
86
87 // goto maildir
88 xchdir(*argv++);
89
90 // launch connect helper, if any
91 if (*argv)
92 launch_helper((const char **)argv);
93
94 // get server greeting
95 pop3_checkr(NULL, NULL, &buf);
96
97 // authenticate (if no -s given)
98 if (!(opts & OPT_s)) {
Denis Vlasenko4abaec52009-03-12 15:35:26 +000099 // server supports APOP and we want it?
100 if ('<' == buf[0] && (opts & OPT_a)) {
101 union { // save a bit of stack
102 md5_ctx_t ctx;
103 char hex[16 * 2 + 1];
104 } md5;
105 uint32_t res[16 / 4];
106
Denis Vlasenko239d06b2008-11-06 23:42:42 +0000107 char *s = strchr(buf, '>');
108 if (s)
Denis Vlasenko4abaec52009-03-12 15:35:26 +0000109 s[1] = '\0';
110 // get md5 sum of "<stamp>password" string
111 md5_begin(&md5.ctx);
112 md5_hash(buf, strlen(buf), &md5.ctx);
113 md5_hash(G.pass, strlen(G.pass), &md5.ctx);
114 md5_end(res, &md5.ctx);
115 *bin2hex(md5.hex, (char*)res, 16) = '\0';
Denis Vlasenko239d06b2008-11-06 23:42:42 +0000116 // APOP
Denis Vlasenko4abaec52009-03-12 15:35:26 +0000117 s = xasprintf("%s %s", G.user, md5.hex);
Denis Vlasenko239d06b2008-11-06 23:42:42 +0000118 pop3_check("APOP %s", s);
Denis Vlasenko4abaec52009-03-12 15:35:26 +0000119 free(s);
120 free(buf);
Denis Vlasenko239d06b2008-11-06 23:42:42 +0000121 // server ignores APOP -> use simple text authentication
122 } else {
123 // USER
124 pop3_check("USER %s", G.user);
125 // PASS
126 pop3_check("PASS %s", G.pass);
127 }
128 }
129
130 // get mailbox statistics
131 pop3_checkr("STAT", NULL, &buf);
132
133 // prepare message filename suffix
134 hostname = safe_gethostname();
135 pid = getpid();
136
137 // get messages counter
138 // NOTE: we don't use xatou(buf) since buf is "nmsg nbytes"
139 // we only need nmsg and atoi is just exactly what we need
140 // if atoi fails to convert buf into number it returns 0
141 // in this case the following loop simply will not be executed
142 nmsg = atoi(buf);
Denis Vlasenko4abaec52009-03-12 15:35:26 +0000143 free(buf);
Denis Vlasenko239d06b2008-11-06 23:42:42 +0000144
145 // loop through messages
146 retr = (opts & OPT_T) ? xasprintf("TOP %%u %u", opt_nlines) : "RETR %u";
147 for (; nmsg; nmsg--) {
148
149 char *filename;
150 char *target;
151 char *answer;
152 FILE *fp;
153#if ENABLE_FEATURE_POPMAILDIR_DELIVERY
154 int rc;
155#endif
156 // generate unique filename
157 filename = xasprintf("tmp/%llu.%u.%s",
158 monotonic_us(), (unsigned)pid, hostname);
159
160 // retrieve message in ./tmp/ unless filter is specified
161 pop3_check(retr, (const char *)(ptrdiff_t)nmsg);
162
163#if ENABLE_FEATURE_POPMAILDIR_DELIVERY
164 // delivery helper ordered? -> setup pipe
165 if (opts & (OPT_F|OPT_M)) {
166 // helper will have $FILENAME set to filename
167 xsetenv("FILENAME", filename);
168 fp = popen(delivery, "w");
169 unsetenv("FILENAME");
170 if (!fp) {
171 bb_perror_msg("delivery helper");
172 break;
173 }
174 } else
175#endif
176 // create and open file filename
177 fp = xfopen_for_write(filename);
178
179 // copy stdin to fp (either filename or delivery helper)
180 while ((answer = xmalloc_fgets_str(stdin, "\r\n")) != NULL) {
181 char *s = answer;
182 if ('.' == answer[0]) {
183 if ('.' == answer[1])
184 s++;
185 else if ('\r' == answer[1] && '\n' == answer[2] && '\0' == answer[3])
186 break;
187 }
188 //*strchrnul(s, '\r') = '\n';
189 fputs(s, fp);
190 free(answer);
191 }
192
193#if ENABLE_FEATURE_POPMAILDIR_DELIVERY
194 // analyse delivery status
195 if (opts & (OPT_F|OPT_M)) {
196 rc = pclose(fp);
197 if (99 == rc) // 99 means bail out
198 break;
199// if (rc) // !0 means skip to the next message
200 goto skip;
201// // 0 means continue
202 } else {
203 // close filename
204 fclose(fp);
205 }
206#endif
207
208 // delete message from server
209 if (!(opts & OPT_k))
210 pop3_check("DELE %u", (const char*)(ptrdiff_t)nmsg);
211
212 // atomically move message to ./new/
213 target = xstrdup(filename);
214 strncpy(target, "new", 3);
215 // ... or just stop receiving on failure
216 if (rename_or_warn(filename, target))
217 break;
218 free(target);
219
220#if ENABLE_FEATURE_POPMAILDIR_DELIVERY
221 skip:
222#endif
223 free(filename);
224 }
225
226 // Bye
227 pop3_check("QUIT", NULL);
228
229 if (ENABLE_FEATURE_CLEAN_UP) {
230 free(G.user);
231 free(G.pass);
232 }
233
234 return EXIT_SUCCESS;
235}