/*
 *  Copyright (C) 2007-2008 Sourcefire, Inc.
 *
 *  Authors: Nigel Horne
 *
 *  This program is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License version 2 as
 *  published by the Free Software Foundation.
 *
 *  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., 51 Franklin Street, Fifth Floor, Boston,
 *  MA 02110-1301, USA.
 */

#ifndef __BLOB_H
#define __BLOB_H

/*
 * Resizable chunk of memory
 */
typedef struct blob {
	char	*name;	/* filename */
	unsigned	char	*data;	/* the stuff itself */
	off_t	len;	/* number of bytes of data so far */
	off_t	size;	/* number of bytes allocated to data so far */
	int	isClosed;
#ifdef	CL_DEBUG
	object_type	magic;	/* verify that this is a blob */
#endif
} blob;

blob	*blobCreate(void);
void	blobDestroy(blob *b);
void	blobArrayDestroy(blob *b[], int n);
void	*blobToMem(blob *b);
void	blobSetFilename(blob *b, const char *dir, const char *filename);
int	blobAddData(blob *b, const unsigned char *data, size_t len);
unsigned char *blobGetData(const blob *b);
size_t	blobGetDataSize(const blob *b);
void	blobClose(blob *b);
int	blobcmp(const blob *b1, const blob *b2);
int	blobGrow(blob *b, size_t len);

/*
 * Like a blob, but associated with a file stored in the temporary directory
 */
typedef	struct fileblob {
	FILE	*fp;
	int	fd;
	blob	b;	/*
			 * b.name is the name of the attachment as stored in the
			 * email, not the full path name of the temporary file
			 */
	char	*fullname;	/* full pathname of the file */
	cli_ctx	*ctx;	/* When set we can scan the blob, otherwise NULL */
	unsigned	long	bytes_scanned;
	unsigned	int	isNotEmpty : 1;
	unsigned	int	isInfected : 1;
} fileblob;

fileblob	*fileblobCreate(void);
int	fileblobScanAndDestroy(fileblob *fb);
void	fileblobDestructiveDestroy(fileblob *fb);
void	fileblobDestroy(fileblob *fb);
void	fileblobSetFilename(fileblob *fb, const char *dir, const char *filename);
void    fileblobPartialSet(fileblob *fb, const char *fullname, const char *arg);
const	char	*fileblobGetFilename(const fileblob *fb);
void	fileblobSetCTX(fileblob *fb, cli_ctx *ctx);
int	fileblobAddData(fileblob *fb, const unsigned char *data, size_t len);
int	fileblobScan(const fileblob *fb);
int	fileblobInfected(const fileblob *fb);
void	sanitiseName(char *name);

#endif /*_BLOB_H*/
