Denis Vlasenko | 239d06b | 2008-11-06 23:42:42 +0000 | [diff] [blame] | 1 | /* 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 | |
| 15 | static 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 | |
| 31 | static void pop3_check(const char *fmt, const char *param) |
| 32 | { |
| 33 | pop3_checkr(fmt, param, NULL); |
| 34 | } |
| 35 | |
| 36 | int popmaildir_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE; |
| 37 | int 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 | } |