mirror of
https://bitbucket.org/jsuto/piler.git
synced 2024-11-08 20:41:59 +01:00
291 lines
6.7 KiB
C
291 lines
6.7 KiB
C
/*
|
|
* imap.c, SJ
|
|
*/
|
|
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <sys/types.h>
|
|
#include <sys/stat.h>
|
|
#include <sys/socket.h>
|
|
#include <netinet/in.h>
|
|
#include <arpa/inet.h>
|
|
#include <netdb.h>
|
|
#include <sys/time.h>
|
|
#include <openssl/ssl.h>
|
|
#include <fcntl.h>
|
|
#include <ctype.h>
|
|
#include <syslog.h>
|
|
#include <unistd.h>
|
|
#include <limits.h>
|
|
#include <piler.h>
|
|
|
|
|
|
|
|
unsigned long resolve_host(char *host){
|
|
struct hostent *h;
|
|
struct in_addr addr;
|
|
|
|
if(!host) return 0;
|
|
|
|
if((addr.s_addr = inet_addr(host)) == -1){
|
|
if((h = gethostbyname(host)) == NULL){
|
|
return 0;
|
|
}
|
|
else return *(unsigned long*)h->h_addr;
|
|
}
|
|
else return addr.s_addr;
|
|
}
|
|
|
|
|
|
int get_message_length_from_imap_answer(char *s, int *_1st_line_bytes){
|
|
char *p, *q;
|
|
int len=0;
|
|
|
|
|
|
p = strstr(s, "\r\n");
|
|
if(!p){
|
|
printf("invalid reply: %s", s);
|
|
return len;
|
|
}
|
|
|
|
*p = '\0';
|
|
|
|
*_1st_line_bytes = strlen(s)+2;
|
|
|
|
if(*(p-1) == '}') *(p-1) = '\0';
|
|
|
|
|
|
q = strchr(s, '{');
|
|
if(q){
|
|
q++;
|
|
len = atoi(q);
|
|
}
|
|
|
|
*(p-1) = '}';
|
|
*p = '\r';
|
|
|
|
return len;
|
|
}
|
|
|
|
|
|
int is_last_complete_packet(char *s, int len, char *tagok, char *tagbad){
|
|
|
|
if(*(s+len-2) == '\r' && *(s+len-1) == '\n'){
|
|
if(strstr(s, tagok)) return 1;
|
|
if(strstr(s, tagbad)) return 1;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
|
|
int process_imap_folder(int sd, int *seq, char *folder, struct session_data *sdata, struct __data *data, struct __config *cfg){
|
|
int rc=ERR, i, n, pos, messages=0, len, readlen, fd, lastpos, nreads, processed_messages=0;
|
|
char *p, tag[SMALLBUFSIZE], tagok[SMALLBUFSIZE], tagbad[SMALLBUFSIZE], buf[MAXBUFSIZE], filename[SMALLBUFSIZE];
|
|
char aggrbuf[3*MAXBUFSIZE];
|
|
|
|
snprintf(tag, sizeof(tag)-1, "A%d", *seq); snprintf(tagok, sizeof(tagok)-1, "\r\nA%d OK", (*seq)++);
|
|
snprintf(buf, sizeof(buf)-1, "%s SELECT \"%s\"\r\n", tag, folder);
|
|
send(sd, buf, strlen(buf), 0);
|
|
|
|
n = recvtimeout(sd, buf, MAXBUFSIZE, 10);
|
|
|
|
if(!strstr(buf, tagok)){
|
|
trimBuffer(buf);
|
|
printf("error: %s\n", buf);
|
|
return rc;
|
|
}
|
|
|
|
p = strstr(buf, " EXISTS");
|
|
if(p){
|
|
*p = '\0';
|
|
p = strrchr(buf, '\n');
|
|
if(p){
|
|
while(!isdigit(*p)){ p++; }
|
|
messages = atoi(p);
|
|
}
|
|
}
|
|
|
|
printf("found %d messages\n", messages);
|
|
|
|
if(messages <= 0) return rc;
|
|
|
|
for(i=1; i<=messages; i++){
|
|
printf("processed: %7d\r", processed_messages); fflush(stdout);
|
|
processed_messages++;
|
|
|
|
snprintf(tag, sizeof(tag)-1, "A%d", *seq);
|
|
snprintf(tagok, sizeof(tagok)-1, "\r\nA%d OK", (*seq)++);
|
|
snprintf(tagbad, sizeof(tagbad)-1, "\r\n%s BAD", tag);
|
|
|
|
snprintf(buf, sizeof(buf)-1, "%s FETCH %d (BODY.PEEK[])\r\n", tag, i);
|
|
|
|
snprintf(filename, sizeof(filename)-1, "%s-%d.txt", folder, i);
|
|
unlink(filename);
|
|
|
|
fd = open(filename, O_CREAT|O_EXCL|O_RDWR|O_TRUNC, S_IRUSR|S_IWUSR);
|
|
if(fd == -1){
|
|
printf("cannot open: %s\n", filename);
|
|
return rc;
|
|
}
|
|
|
|
|
|
send(sd, buf, strlen(buf), 0);
|
|
|
|
readlen = 0;
|
|
pos = 0;
|
|
len = 0;
|
|
nreads = 0;
|
|
|
|
memset(aggrbuf, 0, sizeof(aggrbuf));
|
|
lastpos = 0;
|
|
|
|
|
|
while((n = recvtimeout(sd, buf, sizeof(buf), 15)) > 0){
|
|
nreads++;
|
|
readlen += n;
|
|
|
|
if(nreads == 1){
|
|
len = get_message_length_from_imap_answer(buf, &pos);
|
|
|
|
if(len < 10){
|
|
printf("%d: too short message! %s\n", i, buf);
|
|
break;
|
|
}
|
|
}
|
|
|
|
if(lastpos + n + sizeof(buf) > sizeof(aggrbuf)){
|
|
write(fd, aggrbuf, lastpos);
|
|
memset(aggrbuf, 0, sizeof(aggrbuf));
|
|
lastpos = 0;
|
|
}
|
|
|
|
memcpy(aggrbuf+lastpos, buf, n);
|
|
lastpos += n;
|
|
|
|
if(is_last_complete_packet(aggrbuf, lastpos, tagok, tagbad) == 1){
|
|
write(fd, aggrbuf, lastpos);
|
|
break;
|
|
}
|
|
else {
|
|
write(fd, aggrbuf, lastpos-n);
|
|
memmove(aggrbuf, aggrbuf+lastpos-n, n);
|
|
lastpos = n;
|
|
memset(aggrbuf+lastpos, 0, sizeof(aggrbuf)-lastpos);
|
|
}
|
|
}
|
|
|
|
|
|
close(fd);
|
|
|
|
rc = import_message(filename, sdata, data, cfg);
|
|
|
|
unlink(filename);
|
|
}
|
|
|
|
printf("\n");
|
|
|
|
return OK;
|
|
}
|
|
|
|
|
|
int connect_to_imap_server(int sd, int *seq, char *imapserver, char *username, char *password){
|
|
int n, pos=0;
|
|
char tag[SMALLBUFSIZE], tagok[SMALLBUFSIZE], buf[MAXBUFSIZE];
|
|
char auth[2*SMALLBUFSIZE];
|
|
unsigned char tmp[SMALLBUFSIZE];
|
|
unsigned long host=0;
|
|
struct sockaddr_in remote_addr;
|
|
|
|
|
|
host = resolve_host(imapserver);
|
|
|
|
remote_addr.sin_family = AF_INET;
|
|
remote_addr.sin_port = htons(143);
|
|
remote_addr.sin_addr.s_addr = host;
|
|
bzero(&(remote_addr.sin_zero),8);
|
|
|
|
|
|
if(connect(sd, (struct sockaddr *)&remote_addr, sizeof(struct sockaddr)) == -1){
|
|
printf("connect()\n");
|
|
return ERR;
|
|
}
|
|
|
|
n = recvtimeout(sd, buf, MAXBUFSIZE, 10);
|
|
|
|
|
|
/*
|
|
* create auth buffer: username + NUL character + username + NUL character + password
|
|
*/
|
|
|
|
memset(tmp, 0, sizeof(tmp));
|
|
pos = 0;
|
|
|
|
memcpy(tmp+pos, username, strlen(username));
|
|
pos = strlen(username) + 1;
|
|
memcpy(tmp+pos, username, strlen(username));
|
|
pos += strlen(username) + 1;
|
|
memcpy(tmp+pos, password, strlen(password));
|
|
pos += strlen(password);
|
|
|
|
|
|
base64_encode(&tmp[0], pos, &auth[0], sizeof(auth));
|
|
|
|
snprintf(tag, sizeof(tag)-1, "A%d", *seq); snprintf(tagok, sizeof(tagok)-1, "A%d OK", (*seq)++);
|
|
snprintf(buf, sizeof(buf)-1, "%s AUTHENTICATE PLAIN %s\r\n", tag, auth);
|
|
send(sd, buf, strlen(buf), 0);
|
|
n = recvtimeout(sd, buf, MAXBUFSIZE, 10);
|
|
if(strncmp(buf, tagok, strlen(tagok))){
|
|
printf("login failed, server reponse: %s\n", buf);
|
|
return ERR;
|
|
}
|
|
|
|
return OK;
|
|
}
|
|
|
|
|
|
int list_folders(int sd, int *seq, char *folders, int foldersize){
|
|
int n;
|
|
char *p, *q, tag[SMALLBUFSIZE], tagok[SMALLBUFSIZE], buf[MAXBUFSIZE], puf[MAXBUFSIZE];
|
|
|
|
snprintf(tag, sizeof(tag)-1, "A%d", *seq); snprintf(tagok, sizeof(tagok)-1, "A%d OK", (*seq)++);
|
|
//snprintf(buf, sizeof(buf)-1, "%s LIST \"\" %%\r\n", tag);
|
|
snprintf(buf, sizeof(buf)-1, "%s LIST \"\" \"*\"\r\n", tag);
|
|
|
|
send(sd, buf, strlen(buf), 0);
|
|
|
|
n = recvtimeout(sd, buf, MAXBUFSIZE, 10);
|
|
|
|
p = &buf[0];
|
|
do {
|
|
memset(puf, 0, sizeof(puf));
|
|
p = split(p, '\n', puf, sizeof(puf)-1);
|
|
trimBuffer(puf);
|
|
|
|
if(strncmp(puf, "* LIST ", 7) == 0){
|
|
q = strstr(puf, "\".\"");
|
|
if(q){
|
|
q += 3;
|
|
|
|
if(*q == ' ') q++;
|
|
if(*q == '"') q++;
|
|
|
|
if(q[strlen(q)-1] == '"') q[strlen(q)-1] = '\0';
|
|
|
|
strncat(folders, "\n", foldersize-1);
|
|
strncat(folders, q, foldersize-1);
|
|
|
|
}
|
|
}
|
|
else {
|
|
if(strncmp(puf, tagok, strlen(tagok)) == 0) {}
|
|
}
|
|
|
|
} while(p);
|
|
|
|
return 0;
|
|
}
|
|
|
|
|