/*

  *  Copyright (C) 2013-2019 Cisco Systems, Inc. and/or its affiliates. All rights reserved.
  *  Copyright (C) 2007-2013 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

 #include "json_api.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;

 #if HAVE_JSON

     json_object *jobj;

 #endif

 } 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);

 #if HAVE_JSON
 json_object *messageGetJObj(message *m);
 #endif

 #endif /*_MESSAGE_H*/