summaryrefslogtreecommitdiff
path: root/src
diff options
context:
space:
mode:
authorStef Walter <stef@memberwebs.com>2004-08-30 04:28:25 +0000
committerStef Walter <stef@memberwebs.com>2004-08-30 04:28:25 +0000
commit3fbc5b33686d7f63341e1750799c6986a4e362f6 (patch)
tree5ead7f32d5df4429aaddee15d1d0b562b4458b41 /src
parent80bf46c1397a1e6599697688abd6c7ed1c27961d (diff)
- IO fixes
Diffstat (limited to 'src')
-rw-r--r--src/clamsmtpd.c7
-rw-r--r--src/clio.c69
2 files changed, 59 insertions, 17 deletions
diff --git a/src/clamsmtpd.c b/src/clamsmtpd.c
index e94b9b3..e68bdf1 100644
--- a/src/clamsmtpd.c
+++ b/src/clamsmtpd.c
@@ -958,7 +958,7 @@ static int avcheck_data(clamsmtp_context_t* ctx, char* logline)
strlcat(buf, "/clamsmtpd.XXXXXX", MAXPATHLEN);
/* transfer_to_file deletes the temp file on failure */
- if((r = transfer_to_file(ctx, buf)) != -1)
+ if((r = transfer_to_file(ctx, buf)) > 0)
{
havefile = 1;
r = clam_scan_file(ctx, buf, logline);
@@ -1275,6 +1275,7 @@ static int transfer_to_file(clamsmtp_context_t* ctx, char* tempname)
/* We check errors on this later */
fwrite(ctx->line, 1, ctx->linelen, tfile);
+ count += ctx->linelen;
/* Check if this line ended with a CRLF */
ended_crlf = (strcmp(CRLF, ctx->line + (ctx->linelen - KL(CRLF))) == 0);
@@ -1300,7 +1301,7 @@ cleanup:
if(tfile == NULL)
close(tfd);
- if(ret == -1)
+ if(ret <= 0)
{
messagex(ctx, LOG_DEBUG, "discarding temporary file");
unlink(tempname);
@@ -1323,7 +1324,7 @@ static int transfer_from_file(clamsmtp_context_t* ctx, const char* filename)
RETURN(-1);
}
- messagex(ctx, LOG_DEBUG, "opened temporary file: %s", filename);
+ messagex(ctx, LOG_DEBUG, "sending from temporary file: %s", filename);
while(fgets(ctx->line, LINE_LENGTH, file) != NULL)
{
diff --git a/src/clio.c b/src/clio.c
index cfd4fa7..1794118 100644
--- a/src/clio.c
+++ b/src/clio.c
@@ -1,3 +1,45 @@
+/*
+ * Copyright (c) 2004, Nate Nielsen
+ * All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions
+ * are met:
+ *
+ * * Redistributions of source code must retain the above
+ * copyright notice, this list of conditions and the
+ * following disclaimer.
+ * * 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.
+ * * The names of contributors to this software may not be
+ * used to endorse or promote products derived from this
+ * software without specific prior written permission.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS 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
+ * COPYRIGHT OWNER 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.
+ *
+ *
+ * CONTRIBUTORS
+ * Nate Nielsen <nielsen@memberwebs.com>
+ */
+
+/*
+ * select() and stdio are basically mutually exclusive.
+ * Hence all of this code to try to get some buffering
+ * along with select IO multiplexing.
+ */
#include <sys/time.h>
#include <sys/types.h>
@@ -18,7 +60,7 @@
#include "util.h"
#define MAX_LOG_LINE 79
-#define IO_UNKNOWN "??? "
+#define GET_IO_NAME(io) ((io)->name ? (io)->name : "??? ")
static void close_raw(int* fd)
{
@@ -48,8 +90,7 @@ static void log_io_data(clamsmtp_context_t* ctx, clio_t* io, const char* data, i
memcpy(buf, data, len);
buf[len] = 0;
- messagex(ctx, LOG_DEBUG, "%s%s%s",
- io->name ? io->name : IO_UNKNOWN,
+ messagex(ctx, LOG_DEBUG, "%s%s%s", GET_IO_NAME(io),
read ? " < " : " > ", buf);
data += pos;
@@ -93,8 +134,7 @@ cleanup:
}
ASSERT(io->fd != -1);
- messagex(ctx, LOG_DEBUG, "%s connected to: %s",
- io->name ? io->name : IO_UNKNOWN, addrname);
+ messagex(ctx, LOG_DEBUG, "%s connected to: %s", GET_IO_NAME(io), addrname);
return 0;
}
@@ -105,8 +145,7 @@ void clio_disconnect(clamsmtp_context_t* ctx, clio_t* io)
if(clio_valid(io))
{
close_raw(&(io->fd));
- messagex(ctx, LOG_NOTICE, "%s connection closed",
- io->name ? io->name : IO_UNKNOWN);
+ messagex(ctx, LOG_DEBUG, "%s connection closed", GET_IO_NAME(io));
}
}
@@ -212,6 +251,13 @@ int clio_read_line(clamsmtp_context_t* ctx, clio_t* io, int opts)
continue;
}
+ if(errno == ECONNRESET) /* Not usually a big deal so supresse the error */
+ messagex(ctx, LOG_DEBUG, "connection disconnected by peer: %s", GET_IO_NAME(io));
+ else if(errno == EAGAIN)
+ messagex(ctx, LOG_WARNING, "network read operation timed out: %s", GET_IO_NAME(io));
+ else
+ message(ctx, LOG_ERR, "couldn't read data from socket: %s", GET_IO_NAME(io));
+
/*
* The basic logic here is that if we've had a fatal error
* reading from the socket once then we shut it down as it's
@@ -219,11 +265,6 @@ int clio_read_line(clamsmtp_context_t* ctx, clio_t* io, int opts)
*/
close_raw(&(io->fd));
- if(errno == EAGAIN)
- messagex(ctx, LOG_WARNING, "network read operation timed out");
- else
- message(ctx, LOG_ERR, "couldn't read data from socket");
-
return -1;
}
}
@@ -354,9 +395,9 @@ int clio_write_data_raw(clamsmtp_context_t* ctx, clio_t* io, unsigned char* buf,
close_raw(&(io->fd));
if(errno == EAGAIN)
- messagex(ctx, LOG_WARNING, "network write operation timed out");
+ messagex(ctx, LOG_WARNING, "network write operation timed out: %s", GET_IO_NAME(io));
else
- message(ctx, LOG_ERR, "couldn't write data to socket");
+ message(ctx, LOG_ERR, "couldn't write data to socket: %s", GET_IO_NAME(io));
return -1;
}