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