/*
  *  Copyright (C) 2002 Nigel Horne <njh@bandsman.co.uk>
  *
  *  This program is free software; you can redistribute it and/or modify
  *  it under the terms of the GNU General Public License as published by
  *  the Free Software Foundation; either version 2 of the License, or
  *  (at your option) any later version.
  *
  *  This program is distributed in the hope that it will be useful,
  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  *  GNU General Public License for more details.
  *
  *  You should have received a copy of the GNU General Public License
  *  along with this program; if not, write to the Free Software
  *  Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  */

 static	char	const	rcsid[] = "$Id: blob.c,v 1.40 2005/04/04 13:52:46 nigelhorne Exp $";

 
 #if HAVE_CONFIG_H
 #include "clamav-config.h"
 #endif

 #include <stdio.h>

 #include <stdlib.h>
 #include <string.h>

 #include <errno.h>

 #include <fcntl.h>

 #include <sys/param.h>	/* for NAME_MAX */

 #ifdef	C_DARWIN

 #include <sys/types.h>
 #endif

 #include "mbox.h"
 #include "blob.h"
 #include "others.h"
 
 #ifndef	CL_DEBUG
 #define	NDEBUG	/* map CLAMAV debug onto standard */
 #endif
 

 #ifndef	O_BINARY
 #define	O_BINARY	0
 #endif
 

 #include <assert.h>
 

 #ifdef	C_MINGW
 #include <windows.h>
 #endif
 

 blob *
 blobCreate(void)
 {
 #ifdef	CL_DEBUG
 	blob *b = (blob *)cli_calloc(1, sizeof(blob));

 	if(b)

 		b->magic = BLOBCLASS;

 	cli_dbgmsg("blobCreate\n");

 	return b;
 #else
 	return (blob *)cli_calloc(1, sizeof(blob));
 #endif
 }
 
 void
 blobDestroy(blob *b)
 {

 #ifdef	CL_DEBUG
 	cli_dbgmsg("blobDestroy %d\n", b->magic);
 #else
 	cli_dbgmsg("blobDestroy\n");
 #endif
 

 	assert(b != NULL);

 	assert(b->magic == BLOBCLASS);

 
 	if(b->name)
 		free(b->name);
 	if(b->data)
 		free(b->data);
 #ifdef	CL_DEBUG

 	b->magic = INVALIDCLASS;

 #endif
 	free(b);
 }
 
 void
 blobArrayDestroy(blob *blobList[], int n)
 {

 	assert(blobList != NULL);
 

 	while(--n >= 0) {
 		cli_dbgmsg("blobArrayDestroy: %d\n", n);
 		if(blobList[n]) {
 			blobDestroy(blobList[n]);
 			blobList[n] = NULL;
 		}
 	}

 }
 

 /*ARGSUSED*/

 void

 blobSetFilename(blob *b, const char *dir, const char *filename)

 {
 	assert(b != NULL);

 	assert(b->magic == BLOBCLASS);

 	assert(filename != NULL);
 

 	cli_dbgmsg("blobSetFilename: %s\n", filename);
 

 	if(b->name)
 		free(b->name);
 

 	b->name = strdup(filename);

 	if(b->name)
 		sanitiseName(b->name);

 }
 
 const char *
 blobGetFilename(const blob *b)
 {
 	assert(b != NULL);

 	assert(b->magic == BLOBCLASS);

 	return b->name;

 }
 

 int

 blobAddData(blob *b, const unsigned char *data, size_t len)
 {

 #ifdef	HAVE_GETPAGESIZE

 	static int pagesize;
 	int growth;

 #endif
 

 	assert(b != NULL);

 	assert(b->magic == BLOBCLASS);

 	assert(data != NULL);
 
 	if(len == 0)

 		return 0;

 	if(b->isClosed) {
 		/*
 		 * Should be cli_dbgmsg, but I want to see them for now,
 		 * and cli_dbgmsg doesn't support debug levels
 		 */
 		cli_warnmsg("Reopening closed blob\n");
 		b->isClosed = 0;
 	}

 	/*
 	 * The payoff here is between reducing the number of calls to
 	 * malloc/realloc and not overallocating memory. A lot of machines
 	 * are more tight with memory than one may imagine which is why
 	 * we don't just allocate a *huge* amount and be done with it. Closing
 	 * the blob helps because that reclaims memory. If you know the maximum
 	 * size of a blob before you start adding data, use blobGrow() that's
 	 * the most optimum
 	 */
 #ifdef	HAVE_GETPAGESIZE

 	if(pagesize == 0) {
 		pagesize = getpagesize();
 		if(pagesize == 0)
 			pagesize = 4096;
 	}
 	growth = pagesize;

 	if(len >= (size_t)pagesize)

 		growth = ((len / pagesize) + 1) * pagesize;
 
 	/*printf("len %u, growth = %u\n", len, growth);*/
 

 	if(b->data == NULL) {
 		assert(b->len == 0);
 		assert(b->size == 0);
 

 		b->size = growth;
 		b->data = cli_malloc(growth);

 	} else if(b->size < b->len + len) {

 		unsigned char *p = cli_realloc(b->data, b->size + growth);

 
 		if(p == NULL)

 			return -1;

 		b->size += growth;

 		b->data = p;
 	}
 #else

 	if(b->data == NULL) {
 		assert(b->len == 0);

 		assert(b->size == 0);
 

 		b->size = len * 4;
 		b->data = cli_malloc(b->size);
 	} else if(b->size < b->len + len) {

 		unsigned char *p = cli_realloc(b->data, b->size + (len * 4));
 
 		if(p == NULL)

 			return -1;

 		b->size += len * 4;

 		b->data = p;

 	}

 #endif

 	if(b->data) {
 		memcpy(&b->data[b->len], data, len);
 		b->len += len;
 	}

 	return 0;

 }
 

 unsigned char *

 blobGetData(const blob *b)
 {
 	assert(b != NULL);

 	assert(b->magic == BLOBCLASS);

 	if(b->len == 0)
 		return NULL;

 	return(b->data);
 }
 
 unsigned long
 blobGetDataSize(const blob *b)
 {
 	assert(b != NULL);

 	assert(b->magic == BLOBCLASS);

 
 	return(b->len);
 }

 
 void
 blobClose(blob *b)
 {

 	assert(b != NULL);

 	assert(b->magic == BLOBCLASS);

 
 	if(b->isClosed) {
 		cli_dbgmsg("Attempt to close a previously closed blob\n");
 		return;
 	}

 	/*
 	 * Nothing more is going to be added to this blob. If it'll save more
 	 * than a trivial amount (say 64 bytes) of memory, shrink the allocation
 	 */
 	if((b->size - b->len) >= 64) {

 		if(b->len == 0) {	/* Not likely */
 			free(b->data);
 			b->data = NULL;
 			cli_dbgmsg("blobClose: recovered all %u bytes\n",
 				b->size);
 			b->size = 0;
 		} else {
 			unsigned char *ptr = cli_realloc(b->data, b->len);
 
 			if(ptr == NULL)
 				return;
 
 			cli_dbgmsg("blobClose: recovered %u bytes from %u\n",
 				b->size - b->len, b->size);
 			b->size = b->len;
 			b->data = ptr;
 		}

 	}

 	b->isClosed = 1;

 }
 
 /*
  * Returns 0 if the blobs are the same
  */
 int
 blobcmp(const blob *b1, const blob *b2)
 {
 	unsigned long s1, s2;
 
 	assert(b1 != NULL);
 	assert(b2 != NULL);
 
 	if(b1 == b2)
 		return 0;
 
 	s1 = blobGetDataSize(b1);
 	s2 = blobGetDataSize(b2);
 
 	if(s1 != s2)
 		return 1;
 

 	if((s1 == 0) && (s2 == 0))
 		return 0;
 

 	return memcmp(blobGetData(b1), blobGetData(b2), s1);
 }

 /*
  * Return clamav return code
  */
 int

 blobGrow(blob *b, size_t len)
 {
 	assert(b != NULL);

 	assert(b->magic == BLOBCLASS);

 
 	if(len == 0)

 		return 0;

 
 	if(b->isClosed) {
 		/*
 		 * Should be cli_dbgmsg, but I want to see them for now,
 		 * and cli_dbgmsg doesn't support debug levels
 		 */
 		cli_warnmsg("Growing closed blob\n");
 		b->isClosed = 0;
 	}
 	if(b->data == NULL) {
 		assert(b->len == 0);
 		assert(b->size == 0);
 
 		b->data = cli_malloc(len);

 		if(b->data)
 			b->size = len;

 	} else {

 		unsigned char *ptr = cli_realloc(b->data, b->size + len);
 
 		if(ptr) {
 			b->size += len;
 			b->data = ptr;
 		}

 	}

 
 	return (b->data) ? 0 : CL_EMEM;

 }

 
 fileblob *
 fileblobCreate(void)
 {
 #ifdef	CL_DEBUG
 	fileblob *fb = (fileblob *)cli_calloc(1, sizeof(fileblob));
 	if(fb)

 		fb->b.magic = BLOBCLASS;

 	cli_dbgmsg("blobCreate\n");
 	return fb;
 #else
 	return (fileblob *)cli_calloc(1, sizeof(fileblob));
 #endif
 }
 
 void
 fileblobDestroy(fileblob *fb)
 {
 	assert(fb != NULL);

 	assert(fb->b.magic == BLOBCLASS);

 	if(fb->b.name && fb->fp) {

 		fclose(fb->fp);
 		cli_dbgmsg("fileblobDestroy: %s\n", fb->b.name);
 		if(!fb->isNotEmpty) {

 			cli_dbgmsg("fileblobDestroy: not saving empty file\n");
 			unlink(fb->b.name);
 		}

 		free(fb->b.name);
 
 		assert(fb->b.data == NULL);
 	} else if(fb->b.data) {
 		cli_errmsg("fileblobDestroy: file not saved: report to bugs@clamav.net\n");
 		free(fb->b.data);

 		if(fb->b.name)
 			free(fb->b.name);

 	}

 #ifdef	CL_DEBUG
 	fb->b.magic = INVALIDCLASS;
 #endif

 	free(fb);
 }
 
 void
 fileblobSetFilename(fileblob *fb, const char *dir, const char *filename)
 {
 	int fd;
 	char fullname[NAME_MAX + 1];
 
 	if(fb->b.name)
 		return;
 

 	assert(filename != NULL);
 	assert(dir != NULL);
 

 	blobSetFilename(&fb->b, dir, filename);

 
 	/*
 	 * Reload the filename, it may be different from the one we've
 	 * asked for, e.g. '/'s taken out
 	 */
 	filename = blobGetFilename(&fb->b);
 

 	assert(filename != NULL);
 

 #ifdef	C_QNX6
 	/*
 	 * QNX6 support from mikep@kaluga.org to fix bug where mkstemp
 	 * can return ETOOLONG even when the file name isn't too long
 	 */
 	snprintf(fullname, sizeof(fullname), "%s/clamavtmpXXXXXXXXXXXXX", dir);
 #else

 	snprintf(fullname, sizeof(fullname) - 1, "%s/%.*sXXXXXX", dir,
 		(int)(sizeof(fullname) - 9 - strlen(dir)), filename);

 #endif
 

 #if    defined(C_LINUX) || defined(C_BSD) || defined(HAVE_MKSTEMP) || defined(C_SOLARIS) || defined(C_CYGWIN) || defined(C_QNX6) || defined(C_KFREEBSD_GNU)

 	cli_dbgmsg("fileblobSetFilename: mkstemp(%s)\n", fullname);

 	fd = mkstemp(fullname);
 #else

 	cli_dbgmsg("fileblobSetFilename: mktemp(%s)\n", fullname);

 	(void)mktemp(fullname);
 	fd = open(fullname, O_WRONLY|O_CREAT|O_EXCL|O_TRUNC|O_BINARY, 0600);
 #endif
 
 	if(fd < 0) {
 		cli_errmsg("Can't create temporary file %s: %s\n", fullname, strerror(errno));

 		cli_dbgmsg("%d %d\n", sizeof(fullname), strlen(fullname));

 		return;
 	}
 
 	cli_dbgmsg("Saving attachment as %s\n", fullname);
 
 	fb->fp = fdopen(fd, "wb");
 
 	if(fb->fp == NULL) {
 		cli_errmsg("Can't create file %s: %s\n", fullname, strerror(errno));

 		cli_dbgmsg("%d %d\n", sizeof(fullname), strlen(fullname));

 		close(fd);
 
 		return;
 	}
 	if(fb->b.data) {
 		if(fwrite(fb->b.data, fb->b.len, 1, fb->fp) != 1)

 			cli_errmsg("fileblobSetFilename: Can't write to temporary file %s: %s\n", fullname, strerror(errno));

 		else
 			fb->isNotEmpty = 1;

 		free(fb->b.data);
 		fb->b.data = NULL;
 		fb->b.len = fb->b.size = 0;
 	}
 }
 

 int

 fileblobAddData(fileblob *fb, const unsigned char *data, size_t len)
 {
 	if(len == 0)

 		return 0;

 	assert(data != NULL);
 

 	if(fb->fp) {

 		if(fwrite(data, len, 1, fb->fp) != 1) {

 			cli_errmsg("fileblobAddData: Can't write %u bytes to temporary file %s: %s\n", len, fb->b.name, strerror(errno));

 			return -1;
 		}
 		fb->isNotEmpty = 1;
 		return 0;
 	}
 	return blobAddData(&(fb->b), data, len);

 }
 
 const char *
 fileblobGetFilename(const fileblob *fb)
 {
 	return blobGetFilename(&(fb->b));
 }

 
 /*
  * Different operating systems allow different characters in their filenames

  * FIXME: What does QNX want? There is no #ifdef C_QNX, but if there were
  *	it may be best to treat it like MSDOS

  */
 void
 sanitiseName(char *name)
 {
 	while(*name) {
 #ifdef	C_DARWIN
 		*name &= '\177';

 #endif

 		/* Also check for tab - "Heinz Martin" <Martin@hemag.ch> */

 #if	defined(MSDOS) || defined(C_CYGWIN) || defined(WIN32) || defined(C_OS2)

 		if(strchr("/*?<>|\\\"+=,;:\t ", *name))

 #else
 		if(*name == '/')
 #endif
 			*name = '_';
 		name++;
 	}
 }