/*

Copyright (c) 2007 Carl Byington - 510 Software Group, released under
the GPL version 3 or any later version at your choice available at
http://www.gnu.org/licenses/gpl-3.0.txt

Based on spamass-milter by Georg C. F. Greve <greve@gnu.org>

*/

#ifndef _SPAMASS_MILTER_H
#define _SPAMASS_MILTER_H

extern "C" {
	#include <libmilter/mfapi.h>
}

#ifdef HAVE_CDEFS_H
#include <sys/cdefs.h>
#endif

#include <list>

extern char *spamc;
extern char *spamc_empty;
using namespace std;

class mlfiPriv;

class SpamAssassin {
public:
	SpamAssassin(mlfiPriv *priv_, int ip, char *helo_, char *from, char *qid);
	~SpamAssassin();

	void mlfi_envrcpt(SMFICTX *ctx, char *envrcpt);
	void mlfi_header(char* headerf, char* headerv);
	void mlfi_eoh();
	void mlfi_body(u_char *bodyp, size_t bodylen);
	int  mlfi_eom();

private:
	void Connect();
	void output(const char*buffer, size_t size);
	void output(const char*buffer);
	void output(string buffer);
	void close_output();
	void input();
	int  read_pipe();
	void empty_and_close_pipe();
	void closeall(int fd);
	char *getorwarnmacro(SMFICTX *ctx, char *macro, char *def, char *scope);
	void warnmacro(char *macro, char *scope);

public:
	bool error; 		// spamc died or cannot work
	bool running;		// running implies (connected and pid)
	bool first_recipient;	// have we seen any recipients?

	// connection back to main dnsbl priv structure for logging
	mlfiPriv *priv;
	int 	 ip4;		// ip4 address of smtp client

	// strings owned by main dnsbl
	char *helo; 		// client helo value
	char *envfrom;		// envelope from value for this message
	char *queueid;		// the sendmail queue id for this message

	// Process handling variables
	pid_t pid;
	int pipe_io[2][2];
	string spamc_output;
	string spamc_input;
};

#endif