slow.h #1

/***********************************************************
* Tune the slowmilter program
*
* $Id: slow.h,v 1.5 2003/10/08 14:34:19 bcx Exp bcx $
*
* Copyright Bryan Costales 2003
* Placed into the pubic domain 2003
***********************************************************/
# include <arpa/inet.h>
# include <ctype.h>
# include <db.h>
# include <errno.h>
# include <fcntl.h>
# include <libgen.h>
# include <memory.h>
# include <netdb.h>
# include <netinet/in.h>
# include <arpa/nameser.h>
# include <sys/param.h>
# include <stdio.h>
# include <resolv.h>
# include <stdlib.h>
# include <string.h>
# include <sys/socket.h>
# include <sys/stat.h>
# include <sys/types.h>
# include <sys/types.h> 
# include <sysexits.h>
# include <syslog.h>
# include <time.h>
# include <unistd.h>

# include <libmilter/mfapi.h>

# ifndef TRUE
#       define TRUE (1) 
# endif
# ifndef FALSE 
#       define FALSE (0)
# endif 
# define bool int

/********************************************************
** Definitions used for MX lookups.
*/
# define MAXMXHOSTS 100
# ifndef HFIXEDSZ
#	define HFIXEDSZ 12	/* sizeof(HEADER) */
# endif
typedef union
{
	HEADER  qb1;
	u_char  qb2[PACKETSZ];
} QBUF;

# ifndef INT16SZ
#	define INT16SZ  2
# endif
# ifndef INT32SZ
#	define INT32SZ 4
# endif
# ifndef MAXHOSTNAMELEN
#	define MAXHOSTNAMELEN 128
# endif
# if defined(__RES) && (__RES >= 19940415)
#	define RES_UNC_T      char *
# else
#	define RES_UNC_T      u_char *
# endif

char **MxHosts;
char *LocalHostName;

/********************************************************
** Definition of the Database keys and data.
*/

/* 
 * This defines what is stored in the database. 
 * Be sure to bump the version when you add new items
 * to the SECONDARY_DATUM below.
 */
# define DBVERSION 1
EXTERN time_t DBVersion;

typedef unsigned long BITMAP;
# define bitset(bit, word) (((word) & (bit)) != 0)
# define setbit(bit, word) ((word) |= (bit))

/*
 * The primary database has binary IP numbers as keys.
 * IP==0 is special because its data is the version of the database.
 * The data is an array of dates in unix time_t format.
 */
typedef uint32_t     PRIMARY_KEY;	/* IPv4 IP number in numeric binary */
typedef time_t	     PRIMARY_DATUM;	/* A unit of date/time in binary */
typedef PRIMARY_DATUM **PRIMARY_DATA;	/* FIFO list of date/times */

/*
 * The secondary database has a combined key made up of an IP number
 * and a time_t date. That combined key documents a single event.
 * The data is a SECONDARY_DATUM structure (see below).
 */
typedef struct {
	PRIMARY_KEY	ip;
	PRIMARY_DATUM	date;
} SECONDARY_KEY;

typedef struct {
	BITMAP		eventmap;	/* See the BIT_ definitions below */
	unsigned int	envrcpts;	/* excess envelope recipients */
	unsigned int	headrcpts;	/* excess header recipients */
	unsigned int	honeyrcpts;	/* total honey pot recipients */
	unsigned int	badrcpts;	/* total bad recipients */
} SECONDARY_DATUM;


/********************************************************
** Should we log? And if so, how?
*/
# ifndef USE_SYSLOG
#	define USE_SYSLOG	1
# endif
# ifndef LOG_FACILITY
#	define LOG_FACILITY	LOG_MAIL
# endif

EXTERN unsigned int Debugbits;
# define isdebug(tok) bitset(tok, Debugbits)
# define BUG_MX          (0x00000001) /* debug bitmap */
# define BUG_DB          (0x00000002) /* debug bitmap */
# define BUG_PUSHN       (0x00000004) /* debug bitmap */ 
# define BUG_HONEY       (0x00000008) /* debug bitmap */ 
# define BUG_SOURCE	 (0x00000010) /* debug bitmap */
# define BUG_HEADERS	 (0x00000020) /* debug bitmap */
# define BUG_CONNECT	 (0x00000040) /* debug bitmap */
# define BUG_HELO	 (0x00000080) /* debug bitmap */
# define BUG_MAILFROM	 (0x00000100) /* debug bitmap */
# define BUG_RCPTTO	 (0x00000200) /* debug bitmap */
# define BUG_HEADER	 (0x00000400) /* debug bitmap */
# define BUG_EOM	 (0x00000800) /* debug bitmap */
# define BUG_EOH	 (0x00001000) /* debug bitmap */
# define BUG_CLOSE	 (0x00002000) /* debug bitmap */
# define BUG_ABORT	 (0x00004000) /* debug bitmap */


/********************************************************
** The various modes of this program.
*/
# define MODE_RUN_AS_MILTER	(0)
# define MODE_DUMP_DATABASE	(1)
# define MODE_ADD_BY_HAND	(2)
# define MODE_GARBAGE_COLLECT	(3)
# define MODE_BOOTSTRAP		(4)

/********************************************************
** Defaults that can be reset with -D when compiling.
*/
EXTERN char *ListenPort;
# ifndef LISTENPORT
#	define LISTENPORT "inet:99@localhost"    /* how to listen */
# endif

EXTERN char *PrimaryDatabaseLoc;
EXTERN DB *Dp1;		/* The primary database */
# ifndef PRIMARYDBFILE
#	define PRIMARYDBFILE "/etc/mail/milter/slowip.db"
# endif
EXTERN char *SecondaryDatabaseLoc;
EXTERN DB *Dp2;		/* The secondary database */
# ifndef SECONDARYDBFILE
#	define SECONDARYDBFILE "/etc/mail/milter/slowevents.db"
# endif

# ifndef CFFILE
#	define CFFILE "/etc/mail/milter/slow.conf" /* configuration file */
# endif

# ifndef HONEYFILE
#	define HONEYFILE "/etc/mail/milter/honeyfile" /* configuration file */
# endif

/********************************************************
** Defaults that can only be overridden in the
** configuration file.
*/

EXTERN char *XTrackHeader;
#define X_TRACK_HEADER "X-SlowMilter"

EXTERN bool LookBackOne;
EXTERN bool AllowWhiteListIPs;

EXTERN int MaxHeaderRcpts;
# define MAX_HEADER_RCPTS			(4)
# define BIT_TOO_MANY_HDR_RCPTS	(0x00000001) /* database bitmap */

EXTERN int MaxRcptsPerEnvelope;
# define MAX_RCPTS_PER_ENV			(200)
# define BIT_TOO_MANY_ENV_RCPTS	(0x00000002) /* database bitmap */

EXTERN char **HoneyHosts;
EXTERN char *HoneyFile;
# define BIT_GOT_HONEYPOT  	(0x00000004) /* database bitmap */

EXTERN bool CheckMessageId;
# define BIT_BAD_MSG_ID		(0x00000008) /* database bitmap */

EXTERN bool CheckFromHeader;
# define BIT_BAD_FROM_HEAD	(0x00000010) /* database bitmap */

EXTERN bool CheckSenderHost;
# define BIT_BAD_HOST		(0x00000020) /* database bitmap */

EXTERN bool LookupIPatRBL;
# define BIT_RBL_BAD		(0x00000040) /* database bitmap */

EXTERN bool MilterAbort;
# define BIT_MILTER_ABORTED	(0x00000080) /* database bitmap */

EXTERN bool AdvanceWrite;
# define BIT_ADVANCEWRITE	(0x00000100) /* database bitmap */

EXTERN bool CheckForged;
# define BIT_FORGED		(0x00000200) /* database bitmap */

# define BIT_FROMMX		(0x00000400) /* database bitmap */
# define BIT_NOTINET		(0x00000800) /* database bitmap */

/********************************************************
** Global variables set by main() and read-only by
** all the threads.
*/
EXTERN char *Prog;	/* Basename of the milter invocation */

/********************************************************
** Pattern for the context specific data
*/
typedef struct priv {
	PRIMARY_DATUM	 ip;		/* IPv4 number in binary */
	char   		*iptxt;		/* IPv4 number in human readable text */
	BITMAP		 bits;		/* Bit map of binary events */
	int		 envrcpts;	/* Number good recipients */
	int		 badrcpts; 	/* Number bad recipients */
	int		 honeyrcpts; 	/* Number honeypot recipients */
	int		 nhdrrcpts;	/* Number of header recipients */
	char	       **hdrrcpts;	/* List of header recipients */
	bool		 gotmid;	/* Found a Message-ID header */
} PRIV;

/*********************************************
** Misc. sizes and limits.
*/
# define MAX_FAILS_M (5) /* how many times should malloc() fail? */

/*********************************************
** Function prototypes
*/

/* in main.c */
sfsistat  push_to_db(SMFICTX *, PRIV *);


/* in readconf.c */
void setdefaults(void);		/* in readconf.c */
void readconf(char *, char *);	/* in readconf.c */

/* in util.c */
void milterr(char *, char *, int, int, int, char *, char *);
char **push_list(char *, char **);
char **free_list(char **);
char **mkargv(char *, int);
void print_list(char *, char **, FILE *);
bool is_in_list(char *, char **);
void read_honey(void);
void set_debug(char *);
char *cleanup(char *string);
void *try_alloc(char *, int, size_t);
void *try_strdup(char *, int, char *);
void *try_realloc(char *, int, void *, size_t);


/* in database.c */
DB *open_database(char *, int);
void close_database(DB *);
int adddate_db(DB *, char *, PRIMARY_KEY, PRIMARY_DATUM);
SECONDARY_DATUM *ipgetdateinfo_db(DB *, char *, SECONDARY_KEY *, bool);
int ipputdateinfo_db(DB *, char *, SECONDARY_KEY *, SECONDARY_DATUM *);
int dump_database(DB *, DB *, char *, char *, FILE *);

/* in mx.c */
char **getmx(char *, char **);

/* in header.c */
char **header_recipient(char **, char *);
char *deRFC822(char *);
PRIMARY_KEY header_received(char *);
bool header_messageid(char *);

/* regression testing in test.c */
void test_connect(char *);
void test_helo(char *);
void test_mailfrom(char *);
void test_rcptto(char *);
void test_header(char *);
void test_eoh(char *);
void test_eom(char *);
void test_close(char *);
void test_abort(char *);
void test_db(char *);