You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
492 lines
10 KiB
492 lines
10 KiB
/*
|
|
* libEtPan! -- a mail stuff library
|
|
*
|
|
* Copyright (C) 2001, 2005 - DINH Viet Hoa
|
|
* All rights reserved.
|
|
*
|
|
* Redistribution and use in source and binary forms, with or without
|
|
* modification, are permitted provided that the following conditions
|
|
* are met:
|
|
* 1. Redistributions of source code must retain the above copyright
|
|
* notice, this list of conditions and the following disclaimer.
|
|
* 2. Redistributions in binary form must reproduce the above copyright
|
|
* notice, this list of conditions and the following disclaimer in the
|
|
* documentation and/or other materials provided with the distribution.
|
|
* 3. Neither the name of the libEtPan! project nor the names of its
|
|
* contributors may be used to endorse or promote products derived
|
|
* from this software without specific prior written permission.
|
|
*
|
|
* THIS SOFTWARE IS PROVIDED BY THE AUTHORS AND CONTRIBUTORS ``AS IS'' AND
|
|
* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
|
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
|
* ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHORS OR CONTRIBUTORS BE LIABLE
|
|
* FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
|
|
* DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
|
|
* OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
|
|
* HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
|
|
* LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
|
|
* OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
|
|
* SUCH DAMAGE.
|
|
*/
|
|
|
|
/*
|
|
* $Id: mailstream.c,v 1.25 2011/03/11 21:49:36 hoa Exp $
|
|
*/
|
|
|
|
#ifdef HAVE_CONFIG_H
|
|
# include <config.h>
|
|
#endif
|
|
|
|
#ifdef WIN32
|
|
# include "win_etpan.h"
|
|
#endif
|
|
|
|
#include "mailstream.h"
|
|
#include "maillock.h"
|
|
#include "mailstream_cfstream.h"
|
|
#include "mailstream_cancel.h"
|
|
#include <string.h>
|
|
#include <stdlib.h>
|
|
#include <sys/types.h>
|
|
#include <sys/stat.h>
|
|
|
|
#define DEFAULT_NETWORK_TIMEOUT 300
|
|
|
|
struct timeval mailstream_network_delay =
|
|
{ DEFAULT_NETWORK_TIMEOUT, 0 };
|
|
|
|
mailstream * mailstream_new(mailstream_low * low, size_t buffer_size)
|
|
{
|
|
mailstream * s;
|
|
|
|
s = malloc(sizeof(* s));
|
|
if (s == NULL)
|
|
goto err;
|
|
|
|
s->read_buffer = malloc(buffer_size);
|
|
if (s->read_buffer == NULL)
|
|
goto free_s;
|
|
s->read_buffer_len = 0;
|
|
|
|
s->write_buffer = malloc(buffer_size);
|
|
if (s->write_buffer == NULL)
|
|
goto free_read_buffer;
|
|
s->write_buffer_len = 0;
|
|
|
|
s->buffer_max_size = buffer_size;
|
|
s->low = NULL;
|
|
|
|
s->idle = NULL;
|
|
s->idling = 0;
|
|
|
|
s->logger = NULL;
|
|
s->logger_context = NULL;
|
|
|
|
mailstream_set_low(s, low);
|
|
|
|
return s;
|
|
|
|
free_read_buffer:
|
|
free(s->read_buffer);
|
|
free_s:
|
|
free(s);
|
|
err:
|
|
return NULL;
|
|
}
|
|
|
|
static size_t write_to_internal_buffer(mailstream * s,
|
|
const void * buf, size_t count)
|
|
{
|
|
memcpy(s->write_buffer + s->write_buffer_len, buf, count);
|
|
s->write_buffer_len += count;
|
|
|
|
return count;
|
|
}
|
|
|
|
static ssize_t write_direct(mailstream * s, const void * buf, size_t count)
|
|
{
|
|
size_t left;
|
|
const char * cur_buf;
|
|
ssize_t written;
|
|
|
|
cur_buf = buf;
|
|
left = count;
|
|
while (left > 0) {
|
|
written = mailstream_low_write(s->low, cur_buf, left);
|
|
|
|
if (written < 0) {
|
|
if (count == left)
|
|
return -1;
|
|
else
|
|
return count - left;
|
|
}
|
|
|
|
cur_buf += written;
|
|
left -= written;
|
|
}
|
|
|
|
return count;
|
|
}
|
|
|
|
ssize_t mailstream_write(mailstream * s, const void * buf, size_t count)
|
|
{
|
|
int r;
|
|
|
|
if (s == NULL)
|
|
return -1;
|
|
|
|
if (count + s->write_buffer_len > s->buffer_max_size) {
|
|
r = mailstream_flush(s);
|
|
if (r == -1)
|
|
return -1;
|
|
|
|
if (count > s->buffer_max_size)
|
|
return write_direct(s, buf, count);
|
|
}
|
|
|
|
return write_to_internal_buffer(s, buf, count);
|
|
}
|
|
|
|
int mailstream_flush(mailstream * s)
|
|
{
|
|
char * cur_buf;
|
|
size_t left;
|
|
ssize_t written;
|
|
|
|
if (s == NULL)
|
|
return -1;
|
|
|
|
cur_buf = s->write_buffer;
|
|
left = s->write_buffer_len;
|
|
while (left > 0) {
|
|
written = mailstream_low_write(s->low, cur_buf, left);
|
|
|
|
if (written < 0)
|
|
goto move_buffer;
|
|
cur_buf += written;
|
|
left -= written;
|
|
}
|
|
|
|
s->write_buffer_len = 0;
|
|
|
|
return 0;
|
|
|
|
move_buffer:
|
|
memmove(s->write_buffer, cur_buf, left);
|
|
s->write_buffer_len = left;
|
|
return -1;
|
|
}
|
|
|
|
static ssize_t read_from_internal_buffer(mailstream * s,
|
|
void * buf, size_t count)
|
|
{
|
|
if (count >= s->read_buffer_len)
|
|
count = s->read_buffer_len;
|
|
if (count != 0)
|
|
memcpy(buf, s->read_buffer, count);
|
|
|
|
s->read_buffer_len -= count;
|
|
if (s->read_buffer_len != 0)
|
|
memmove(s->read_buffer, s->read_buffer + count,
|
|
s->read_buffer_len);
|
|
|
|
return count;
|
|
}
|
|
|
|
static ssize_t read_through_buffer(mailstream * s, void * buf, size_t count)
|
|
{
|
|
size_t left;
|
|
char * cur_buf;
|
|
ssize_t bytes_read;
|
|
|
|
cur_buf = buf;
|
|
left = count;
|
|
|
|
while (left > 0) {
|
|
bytes_read = mailstream_low_read(s->low, cur_buf, left);
|
|
|
|
if (bytes_read < 0) {
|
|
if (count == left)
|
|
return -1;
|
|
else
|
|
return count - left;
|
|
}
|
|
else if (bytes_read == 0)
|
|
return count - left;
|
|
|
|
cur_buf += bytes_read;
|
|
left -= bytes_read;
|
|
}
|
|
|
|
return count;
|
|
}
|
|
|
|
ssize_t mailstream_read(mailstream * s, void * buf, size_t count)
|
|
{
|
|
ssize_t read_bytes;
|
|
char * cur_buf;
|
|
size_t left;
|
|
|
|
if (s == NULL)
|
|
return -1;
|
|
|
|
left = count;
|
|
cur_buf = buf;
|
|
read_bytes = read_from_internal_buffer(s, cur_buf, left);
|
|
cur_buf += read_bytes;
|
|
left -= read_bytes;
|
|
|
|
if (left == 0) {
|
|
return read_bytes;
|
|
}
|
|
|
|
if (left > s->buffer_max_size) {
|
|
read_bytes = read_through_buffer(s, cur_buf, left);
|
|
if (read_bytes == -1) {
|
|
if (count == left)
|
|
return -1;
|
|
else {
|
|
return count - left;
|
|
}
|
|
}
|
|
|
|
cur_buf += read_bytes;
|
|
left -= read_bytes;
|
|
|
|
return count - left;
|
|
}
|
|
|
|
read_bytes = mailstream_low_read(s->low, s->read_buffer, s->buffer_max_size);
|
|
if (read_bytes < 0) {
|
|
if (left == count)
|
|
return -1;
|
|
else {
|
|
return count - left;
|
|
}
|
|
}
|
|
else
|
|
s->read_buffer_len += read_bytes;
|
|
|
|
read_bytes = read_from_internal_buffer(s, cur_buf, left);
|
|
cur_buf += read_bytes;
|
|
left -= read_bytes;
|
|
|
|
return count - left;
|
|
}
|
|
|
|
mailstream_low * mailstream_get_low(mailstream * s)
|
|
{
|
|
return s->low;
|
|
}
|
|
|
|
static void low_logger(mailstream_low * s, int log_type,
|
|
const char * str, size_t size, void * context)
|
|
{
|
|
mailstream * stream = context;
|
|
if (stream->logger != NULL) {
|
|
stream->logger(context, log_type, str, size, stream->logger_context);
|
|
}
|
|
}
|
|
|
|
void mailstream_set_low(mailstream * s, mailstream_low * low)
|
|
{
|
|
s->low = low;
|
|
mailstream_low_set_logger(low, low_logger, s);
|
|
}
|
|
|
|
int mailstream_close(mailstream * s)
|
|
{
|
|
if (s->idle != NULL) {
|
|
mailstream_cancel_free(s->idle);
|
|
}
|
|
|
|
mailstream_low_close(s->low);
|
|
mailstream_low_free(s->low);
|
|
|
|
free(s->read_buffer);
|
|
free(s->write_buffer);
|
|
|
|
free(s);
|
|
|
|
return 0;
|
|
}
|
|
|
|
ssize_t mailstream_feed_read_buffer(mailstream * s)
|
|
{
|
|
ssize_t read_bytes;
|
|
|
|
if (s == NULL)
|
|
return -1;
|
|
|
|
if (s->read_buffer_len == 0) {
|
|
read_bytes = mailstream_low_read(s->low, s->read_buffer,
|
|
s->buffer_max_size);
|
|
if (read_bytes < 0)
|
|
return -1;
|
|
s->read_buffer_len += read_bytes;
|
|
}
|
|
|
|
return s->read_buffer_len;
|
|
}
|
|
|
|
void mailstream_cancel(mailstream * s)
|
|
{
|
|
if (s == NULL)
|
|
return;
|
|
|
|
mailstream_low_cancel(s->low);
|
|
}
|
|
|
|
void mailstream_log_error(mailstream * s, char * buf, size_t count)
|
|
{
|
|
mailstream_low_log_error(s->low, buf, count);
|
|
}
|
|
|
|
void mailstream_set_privacy(mailstream * s, int can_be_public)
|
|
{
|
|
mailstream_low_set_privacy(s->low, can_be_public);
|
|
}
|
|
|
|
|
|
int mailstream_wait_idle(mailstream * s, int max_idle_delay)
|
|
{
|
|
int fd;
|
|
int idle_fd;
|
|
int cancel_fd;
|
|
int maxfd;
|
|
fd_set readfds;
|
|
struct timeval delay;
|
|
int r;
|
|
|
|
if (s->low->driver == mailstream_cfstream_driver) {
|
|
return mailstream_cfstream_wait_idle(s, max_idle_delay);
|
|
}
|
|
|
|
if (s->idle == NULL) {
|
|
return MAILSTREAM_IDLE_ERROR;
|
|
}
|
|
if (mailstream_low_get_cancel(mailstream_get_low(s)) == NULL) {
|
|
return MAILSTREAM_IDLE_ERROR;
|
|
}
|
|
fd = mailstream_low_get_fd(mailstream_get_low(s));
|
|
idle_fd = mailstream_cancel_get_fd(s->idle);
|
|
cancel_fd = mailstream_cancel_get_fd(mailstream_low_get_cancel(mailstream_get_low(s)));
|
|
|
|
FD_ZERO(&readfds);
|
|
FD_SET(fd, &readfds);
|
|
FD_SET(idle_fd, &readfds);
|
|
FD_SET(cancel_fd, &readfds);
|
|
maxfd = fd;
|
|
if (idle_fd > maxfd) {
|
|
maxfd = idle_fd;
|
|
}
|
|
if (cancel_fd > maxfd) {
|
|
maxfd = cancel_fd;
|
|
}
|
|
delay.tv_sec = max_idle_delay;
|
|
delay.tv_usec = 0;
|
|
|
|
r = select(maxfd + 1, &readfds, NULL, NULL, &delay);
|
|
if (r == 0) {
|
|
// timeout
|
|
return MAILSTREAM_IDLE_TIMEOUT;
|
|
}
|
|
else if (r == -1) {
|
|
// do nothing
|
|
return MAILSTREAM_IDLE_ERROR;
|
|
}
|
|
else {
|
|
if (FD_ISSET(fd, &readfds)) {
|
|
// has something on socket
|
|
return MAILSTREAM_IDLE_HASDATA;
|
|
}
|
|
if (FD_ISSET(idle_fd, &readfds)) {
|
|
// idle interrupted
|
|
mailstream_cancel_ack(s->idle);
|
|
return MAILSTREAM_IDLE_INTERRUPTED;
|
|
}
|
|
if (FD_ISSET(cancel_fd, &readfds)) {
|
|
// idle cancelled
|
|
mailstream_cancel_ack(mailstream_low_get_cancel(mailstream_get_low(s)));
|
|
return MAILSTREAM_IDLE_CANCELLED;
|
|
}
|
|
return MAILSTREAM_IDLE_ERROR;
|
|
}
|
|
}
|
|
|
|
int mailstream_setup_idle(mailstream * s)
|
|
{
|
|
if (s->idling) {
|
|
return -1;
|
|
}
|
|
|
|
if (s->low->driver == mailstream_cfstream_driver) {
|
|
mailstream_cfstream_setup_idle(s);
|
|
}
|
|
else {
|
|
s->idle = mailstream_cancel_new();
|
|
if (s->idle == NULL)
|
|
return -1;
|
|
}
|
|
|
|
s->idling = 1;
|
|
|
|
return 0;
|
|
}
|
|
|
|
void mailstream_interrupt_idle(mailstream * s)
|
|
{
|
|
if (!s->idling) {
|
|
return;
|
|
}
|
|
|
|
if (s->low->driver == mailstream_cfstream_driver) {
|
|
mailstream_cfstream_interrupt_idle(s);
|
|
}
|
|
else {
|
|
mailstream_cancel_notify(s->idle);
|
|
}
|
|
}
|
|
|
|
void mailstream_unsetup_idle(mailstream * s)
|
|
{
|
|
if (!s->idling) {
|
|
return;
|
|
}
|
|
|
|
if (s->low->driver == mailstream_cfstream_driver) {
|
|
mailstream_cfstream_unsetup_idle(s);
|
|
}
|
|
else {
|
|
mailstream_cancel_free(s->idle);
|
|
s->idle = NULL;
|
|
}
|
|
|
|
s->idling = 0;
|
|
}
|
|
|
|
void mailstream_set_logger(mailstream * s, void (* logger)(mailstream * s, int log_type,
|
|
const char * str, size_t size, void * context), void * logger_context)
|
|
{
|
|
s->logger = logger;
|
|
s->logger_context = logger_context;
|
|
}
|
|
|
|
carray * mailstream_get_certificate_chain(mailstream * s)
|
|
{
|
|
return mailstream_low_get_certificate_chain(s->low);
|
|
}
|
|
|
|
void mailstream_certificate_chain_free(carray * certificate_chain)
|
|
{
|
|
unsigned int i;
|
|
|
|
if (certificate_chain == NULL)
|
|
return;
|
|
|
|
for(i = 0 ; i < carray_count(certificate_chain) ; i ++) {
|
|
free(carray_get(certificate_chain, i));
|
|
}
|
|
carray_free(certificate_chain);
|
|
}
|
|
|