mirror of
https://bitbucket.org/jsuto/piler.git
synced 2024-12-26 08:30:12 +01:00
272 lines
6.7 KiB
C
272 lines
6.7 KiB
C
/*
|
|
* retr.c, SJ
|
|
*/
|
|
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <strings.h>
|
|
#include <sys/types.h>
|
|
#include <sys/socket.h>
|
|
#include <sys/stat.h>
|
|
#include <fcntl.h>
|
|
#include <unistd.h>
|
|
#include <signal.h>
|
|
#include <syslog.h>
|
|
#include <time.h>
|
|
#include <openssl/ssl.h>
|
|
#include <openssl/err.h>
|
|
#include <piler.h>
|
|
|
|
|
|
int stat_file(struct session_data *sdata, char *f, char **buf, int buflen, struct __config *cfg){
|
|
struct stat st;
|
|
|
|
snprintf(*buf, buflen, "%s/%02x/%c%c%c/%c%c/%c%c/%s", cfg->queuedir, cfg->server_id, f[8], f[9], f[10], f[RND_STR_LEN-4], f[RND_STR_LEN-3], f[RND_STR_LEN-2], f[RND_STR_LEN-1], f);
|
|
if(!stat(*buf, &st)) return st.st_size;
|
|
|
|
snprintf(*buf, TINYBUFSIZE-1, "%s/%02x/%c%c/%c%c/%c%c/%s", cfg->queuedir, cfg->server_id, f[RND_STR_LEN-6], f[RND_STR_LEN-5], f[RND_STR_LEN-4], f[RND_STR_LEN-3], f[RND_STR_LEN-2], f[RND_STR_LEN-1], f);
|
|
if(!stat(*buf, &st)) return st.st_size;
|
|
|
|
return 0;
|
|
}
|
|
|
|
|
|
int stat_message(struct session_data *sdata, struct __data *data, char **buf, int buflen, struct __config *cfg){
|
|
int i, attachments, len=0;
|
|
struct ptr_array ptr_arr[MAX_ATTACHMENTS];
|
|
char puf[TINYBUFSIZE];
|
|
|
|
if(strlen(sdata->ttmpfile) != RND_STR_LEN){
|
|
return ERR;
|
|
}
|
|
|
|
snprintf(*buf, buflen-2, "%s.m", sdata->ttmpfile);
|
|
len = strlen(*buf);
|
|
|
|
attachments = query_attachments(sdata, data, &ptr_arr[0], cfg);
|
|
|
|
if(attachments == -1){
|
|
return ERR;
|
|
}
|
|
|
|
for(i=1; i<=attachments; i++){
|
|
if(len < buflen){
|
|
snprintf(puf, sizeof(puf)-1, " %s.a%d", ptr_arr[i].piler_id, ptr_arr[i].attachment_id);
|
|
strncat(*buf, puf, buflen);
|
|
len += strlen(puf);
|
|
}
|
|
}
|
|
|
|
strncat(*buf, "\r\n", buflen-2);
|
|
|
|
return OK;
|
|
}
|
|
|
|
|
|
int do_ssl_handshake(struct session_data *sdata, struct __data *data, int new_sd, struct __config *cfg){
|
|
int ssl_ok=0, rc;
|
|
char ssl_error[SMALLBUFSIZE];
|
|
|
|
if(data->ctx){
|
|
data->ssl = SSL_new(data->ctx);
|
|
|
|
if(data->ssl){
|
|
if(SSL_set_fd(data->ssl, new_sd) == 1){
|
|
ssl_ok = 1;
|
|
}
|
|
else syslog(LOG_PRIORITY, "SSL_set_fd() failed");
|
|
}
|
|
else syslog(LOG_PRIORITY, "SSL_new() failed");
|
|
}
|
|
else syslog(LOG_PRIORITY, "SSL ctx is null!");
|
|
|
|
|
|
if(ssl_ok == 0){
|
|
send(new_sd, SMTP_RESP_421_ERR_TMP, strlen(SMTP_RESP_421_ERR_TMP), 0);
|
|
return ERR;
|
|
}
|
|
|
|
|
|
rc = SSL_accept(data->ssl);
|
|
|
|
if(cfg->verbosity >= _LOG_DEBUG) syslog(LOG_PRIORITY, "SSL_accept() finished");
|
|
|
|
if(rc == 1){
|
|
sdata->tls = 1;
|
|
return OK;
|
|
}
|
|
else {
|
|
ERR_error_string_n(ERR_get_error(), ssl_error, SMALLBUFSIZE);
|
|
syslog(LOG_PRIORITY, "SSL_accept() failed, rc=%d, errorcode: %d, error text: %s\n", rc, SSL_get_error(data->ssl, rc), ssl_error);
|
|
return ERR;
|
|
}
|
|
|
|
}
|
|
|
|
|
|
int handle_pilerget_request(int new_sd, struct __data *data, struct __config *cfg){
|
|
int len, n, auth_ok=0, n_files=0;
|
|
char *q, buf[MAXBUFSIZE], puf[MAXBUFSIZE], muf[TINYBUFSIZE], resp[MAXBUFSIZE];
|
|
struct session_data sdata;
|
|
int db_conn=0;
|
|
struct __counters counters;
|
|
|
|
struct timezone tz;
|
|
struct timeval tv1, tv2;
|
|
|
|
#ifdef HAVE_LIBWRAP
|
|
struct request_info req;
|
|
|
|
request_init(&req, RQ_DAEMON, PILERGETD_PROGNAME, RQ_FILE, new_sd, 0);
|
|
fromhost(&req);
|
|
if(!hosts_access(&req)){
|
|
send(new_sd, SMTP_RESP_550_ERR_YOU_ARE_BANNED_BY_LOCAL_POLICY, strlen(SMTP_RESP_550_ERR_YOU_ARE_BANNED_BY_LOCAL_POLICY), 0);
|
|
syslog(LOG_PRIORITY, "denied connection from %s by tcp_wrappers", eval_client(&req));
|
|
return 0;
|
|
}
|
|
#endif
|
|
|
|
|
|
init_session_data(&sdata, cfg);
|
|
sdata.tls = 0;
|
|
|
|
bzero(&counters, sizeof(counters));
|
|
|
|
|
|
/* open database connection */
|
|
|
|
db_conn = 0;
|
|
|
|
#ifdef NEED_MYSQL
|
|
if(open_database(&sdata, cfg) == OK){
|
|
db_conn = 1;
|
|
}
|
|
else
|
|
syslog(LOG_PRIORITY, "%s", ERR_MYSQL_CONNECT);
|
|
#endif
|
|
|
|
if(db_conn == 0){
|
|
send(new_sd, SMTP_RESP_421_ERR_TMP, strlen(SMTP_RESP_421_ERR_TMP), 0);
|
|
syslog(LOG_PRIORITY, "cannot make prepared statement");
|
|
return 0;
|
|
}
|
|
|
|
|
|
gettimeofday(&tv1, &tz);
|
|
|
|
|
|
if(cfg->tls_enable > 0){
|
|
if(do_ssl_handshake(&sdata, data, new_sd, cfg) == ERR) goto QUITTING;
|
|
}
|
|
else {
|
|
auth_ok = 1;
|
|
}
|
|
|
|
|
|
snprintf(buf, MAXBUFSIZE-1, SMTP_RESP_220_BANNER, cfg->hostid);
|
|
|
|
write1(new_sd, buf, strlen(buf), sdata.tls, data->ssl);
|
|
|
|
if(cfg->verbosity >= _LOG_DEBUG) syslog(LOG_PRIORITY, "sent: %s", buf);
|
|
|
|
while((n = recvtimeoutssl(new_sd, puf, MAXBUFSIZE, TIMEOUT, sdata.tls, data->ssl)) > 0){
|
|
|
|
if(strncasecmp(puf, "MESSAGE ", strlen("MESSAGE ")) == 0 && auth_ok == 1){
|
|
trimBuffer(puf);
|
|
q = &resp[0];
|
|
|
|
memset(resp, 0, sizeof(resp));
|
|
|
|
snprintf(sdata.ttmpfile, sizeof(sdata.ttmpfile)-1, "%s", &puf[8]);
|
|
stat_message(&sdata, data, &q, sizeof(resp)-2, cfg);
|
|
|
|
write1(new_sd, resp, strlen(resp), sdata.tls, data->ssl);
|
|
|
|
continue;
|
|
}
|
|
|
|
|
|
if(strncasecmp(puf, "AUTH ", strlen("AUTH ")) == 0){
|
|
trimBuffer(puf);
|
|
q = &muf[0];
|
|
|
|
if(strcmp(&puf[5], cfg->pilergetd_password)){
|
|
snprintf(resp, sizeof(resp)-1, "550 AUTH FAILED\r\n");
|
|
}
|
|
else {
|
|
snprintf(resp, sizeof(resp)-1, "250 AUTH\r\n");
|
|
auth_ok = 1;
|
|
}
|
|
|
|
write1(new_sd, resp, strlen(resp), sdata.tls, data->ssl);
|
|
|
|
continue;
|
|
}
|
|
|
|
|
|
if(strncasecmp(puf, "STAT ", strlen("STAT ")) == 0 && auth_ok == 1){
|
|
trimBuffer(puf);
|
|
q = &muf[0];
|
|
|
|
len = stat_file(&sdata, &puf[5], &q, sizeof(muf)-2, cfg);
|
|
|
|
snprintf(resp, sizeof(resp)-1, "SIZE %s %d\r\n", &puf[5], len);
|
|
|
|
write1(new_sd, resp, strlen(resp), sdata.tls, data->ssl);
|
|
|
|
continue;
|
|
}
|
|
|
|
|
|
if(strncasecmp(puf, "RETR ", strlen("RETR ")) == 0 && auth_ok == 1){
|
|
|
|
trimBuffer(puf);
|
|
q = &muf[0];
|
|
|
|
if(strlen(&puf[5]) >= RND_STR_LEN){
|
|
len = stat_file(&sdata, &puf[5], &q, sizeof(muf)-2, cfg);
|
|
file_from_archive_to_network(muf, new_sd, sdata.tls, data, cfg);
|
|
n_files++;
|
|
}
|
|
else {
|
|
snprintf(resp, sizeof(resp)-1, "ERR\r\n");
|
|
write1(new_sd, resp, strlen(resp), sdata.tls, data->ssl);
|
|
}
|
|
|
|
continue;
|
|
}
|
|
|
|
|
|
if(strncasecmp(puf, SMTP_CMD_QUIT, strlen(SMTP_CMD_QUIT)) == 0){
|
|
snprintf(resp, sizeof(resp)-1, SMTP_RESP_221_GOODBYE, cfg->hostid);
|
|
write1(new_sd, resp, strlen(resp), sdata.tls, data->ssl);
|
|
|
|
gettimeofday(&tv2, &tz);
|
|
|
|
break;
|
|
}
|
|
|
|
snprintf(resp, sizeof(resp)-1, "550 ERR INVALID COMMAND\r\n");
|
|
write1(new_sd, resp, strlen(resp), sdata.tls, data->ssl);
|
|
|
|
}
|
|
|
|
|
|
|
|
QUITTING:
|
|
|
|
#ifdef NEED_MYSQL
|
|
close_database(&sdata);
|
|
#endif
|
|
|
|
SSL_shutdown(data->ssl);
|
|
SSL_free(data->ssl);
|
|
|
|
if(cfg->verbosity >= _LOG_DEBUG) syslog(LOG_PRIORITY, "served %d files", n_files);
|
|
|
|
return 1;
|
|
}
|
|
|
|
|