/*
 *  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	_MESSAGE_H
#define	_MESSAGE_H

/* The contents could change, ONLY access in message.c */
typedef struct message {
	encoding_type	*encodingTypes;
	mime_type	mimeType;
	int	numberOfEncTypes;	/* size of encodingTypes */
	char	*mimeSubtype;
	char	**mimeArguments;
	char	*mimeDispositionType;	/* probably attachment */
	text	*body_first, *body_last;
	cli_ctx	*ctx;	/* When set we can scan the message, otherwise NULL */
	int	numberOfArguments;	/* count of mimeArguments */
	int	base64chars;

	/*
	 * Markers for the start of various non MIME messages that could
	 * be included within this message
	 */
	text	*bounce;	/* start of a bounced message */
	text	*binhex;	/* start of a binhex message */
	text	*yenc;		/* start of a yEnc message */
	text	*encoding;	/* is the non MIME message encoded? */
	const text	*dedupedThisFar;

	char	base64_1, base64_2, base64_3;
	unsigned	int	isInfected : 1;
	unsigned        int     isTruncated  : 1;

} message;

message	*messageCreate(void);
void	messageDestroy(message *m);
void	messageReset(message *m);
int	messageSetMimeType(message *m, const char *type);
mime_type	messageGetMimeType(const message *m);
void	messageSetMimeSubtype(message *m, const char *subtype);
const	char	*messageGetMimeSubtype(const message *m);
void	messageSetDispositionType(message *m, const char *disptype);
const	char	*messageGetDispositionType(const message *m);
void	messageAddArgument(message *m, const char *arg);
void	messageAddArguments(message *m, const char *arg);
char	*messageFindArgument(const message *m, const char *variable);
char	*messageGetFilename(const message *m);
int	messageHasFilename(const message *m);
void	messageSetEncoding(message *m, const char *enctype);
encoding_type	messageGetEncoding(const message *m);
int	messageAddLine(message *m, line_t *line);
int	messageAddStr(message *m, const char *data);
int	messageAddStrAtTop(message *m, const char *data);
int	messageMoveText(message *m, text *t, message *old_message);
text	*messageGetBody(message *m);
unsigned	char	*base64Flush(message *m, unsigned char *buf);
fileblob	*messageToFileblob(message *m, const char *dir, int destroy);
blob	*messageToBlob(message *m, int destroy);
text	*messageToText(message *m);
text	*binhexBegin(message *m);
text	*yEncBegin(message *m);
text	*bounceBegin(message *m);
text	*encodingLine(message *m);
unsigned char	*decodeLine(message *m, encoding_type enctype, const char *line, unsigned char *buf, size_t buflen);
int	isuuencodebegin(const char *line);
void	messageSetCTX(message *m, cli_ctx *ctx);
int	messageContainsVirus(const message *m);
int messageSavePartial(message *m, const char *dir, const char *id, unsigned part);

#endif	/*_MESSAGE_H*/
