[c5c522c] | 1 | /* $Id: agent_shared.h,v 1.35 2011/07/11 21:27:15 sbajic Exp $ */ |
---|
| 2 | |
---|
| 3 | /* |
---|
| 4 | DSPAM |
---|
| 5 | COPYRIGHT (C) 2002-2012 DSPAM PROJECT |
---|
| 6 | |
---|
| 7 | This program is free software: you can redistribute it and/or modify |
---|
| 8 | it under the terms of the GNU Affero General Public License as |
---|
| 9 | published by the Free Software Foundation, either version 3 of the |
---|
| 10 | License, or (at your option) any later version. |
---|
| 11 | |
---|
| 12 | This program is distributed in the hope that it will be useful, |
---|
| 13 | but WITHOUT ANY WARRANTY; without even the implied warranty of |
---|
| 14 | MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
---|
| 15 | GNU Affero General Public License for more details. |
---|
| 16 | |
---|
| 17 | You should have received a copy of the GNU Affero General Public License |
---|
| 18 | along with this program. If not, see <http://www.gnu.org/licenses/>. |
---|
| 19 | |
---|
| 20 | */ |
---|
| 21 | |
---|
| 22 | #include <sys/types.h> |
---|
| 23 | #ifndef _WIN32 |
---|
| 24 | #include <pwd.h> |
---|
| 25 | #endif |
---|
| 26 | #include "buffer.h" |
---|
| 27 | |
---|
| 28 | #ifdef HAVE_CONFIG_H |
---|
| 29 | #include <auto-config.h> |
---|
| 30 | #endif |
---|
| 31 | #include "libdspam.h" |
---|
| 32 | |
---|
| 33 | #ifndef _AGENT_SHARED_H |
---|
| 34 | # define _AGENT_SHARED_H |
---|
| 35 | |
---|
| 36 | extern char *__pw_name; |
---|
| 37 | extern uid_t __pw_uid; |
---|
| 38 | |
---|
| 39 | #define STATUS( ... ) snprintf(ATX->status, sizeof(ATX->status), __VA_ARGS__); |
---|
| 40 | |
---|
| 41 | #define SYNTAX "Syntax: dspam [--help|--version|[--client|--daemon [--nofork]] [--debug] --mode=[toe|tum|teft|notrain|unlearn] --user [user1 user2 ... userN] [--feature=[no,wh,tb=N]] [--class=[spam|innocent]] [--source=[error|corpus|inoculation]] [--profile=[PROFILE]] [--deliver=[spam,[innocent|nonspam],summary,stdout]] [--process|--classify] [--stdout] [--mail-from=sender-address] [--rcpt-to recipient-address(es)] [--signature=DSPAM-Signature] [passthru-arguments]]" |
---|
| 42 | |
---|
| 43 | #define SIGNATURE_BEGIN "!DSPAM:" |
---|
| 44 | #define SIGNATURE_END "!" |
---|
| 45 | #define LOOSE_SIGNATURE_BEGIN "X-DSPAM-Signature:" |
---|
| 46 | #define SIGNATURE_DELIMITER ": " |
---|
| 47 | |
---|
| 48 | /* AGENT_CTX: Agent context. Defines the behavior of the agent */ |
---|
| 49 | |
---|
| 50 | typedef struct { |
---|
| 51 | int operating_mode; /* Processing Mode IN DSM_ */ |
---|
| 52 | int fork; /* Fork daemon: IN 1:0 */ |
---|
| 53 | int client_mode; /* Client Mode: IN 1:0 */ |
---|
| 54 | int training_mode; /* Training Mode IN DST_ */ |
---|
| 55 | int classification; /* Classification IN DSR_ */ |
---|
| 56 | int source; /* Classification Source IN DSS_ */ |
---|
| 57 | int spam_action; /* Action on Spam IN DSA_ */ |
---|
| 58 | #ifdef TRUSTED_USER_SECURITY |
---|
| 59 | int trusted; /* Trusted User? IN */ |
---|
| 60 | #endif |
---|
| 61 | int feature; /* Feature Overridden? IN */ |
---|
| 62 | int train_pristine; /* Train Pristine? IN */ |
---|
| 63 | int tokenizer; /* Tokenizer IN */ |
---|
| 64 | void *dbh; /* Database Handle IN */ |
---|
| 65 | u_int64_t flags; /* Flags DAF_ IN */ |
---|
| 66 | int training_buffer; /* Sedation Level 0-10 IN */ |
---|
| 67 | char *recipient; /* Current Recipient */ |
---|
| 68 | char mailer_args[256]; /* Delivery Args IN */ |
---|
| 69 | char spam_args[256]; /* Quarantine Args IN */ |
---|
| 70 | char managed_group[256]; /* Managed Groupname IN */ |
---|
| 71 | char profile[32]; /* Storage Profile IN */ |
---|
| 72 | char signature[128]; /* Signature Serial IN/OUT */ |
---|
| 73 | char mailfrom[256]; /* For LMTP or SMTP */ |
---|
| 74 | struct nt *users; /* Destination Users IN */ |
---|
| 75 | struct nt *inoc_users; /* Inoculate list OUT */ |
---|
| 76 | struct nt *classify_users; /* Classify list OUT */ |
---|
| 77 | struct nt *recipients; /* Recipients IN */ |
---|
| 78 | struct nt *results; /* Process Results OUT */ |
---|
| 79 | struct _ds_spam_signature SIG; /* Signature object OUT */ |
---|
| 80 | int learned; /* Message learned? OUT */ |
---|
| 81 | FILE *sockfd; /* Socket FD if not STDOUT */ |
---|
| 82 | int sockfd_output; /* Output sent to sockfd? */ |
---|
| 83 | char client_args[1024]; /* Args for client connection */ |
---|
| 84 | double timestart; |
---|
| 85 | agent_pref_t PTX; |
---|
| 86 | |
---|
| 87 | char status[256]; |
---|
| 88 | #ifdef DEBUG |
---|
| 89 | char debug_args[1024]; |
---|
| 90 | #endif |
---|
| 91 | #ifndef _WIN32 |
---|
| 92 | #ifdef TRUSTED_USER_SECURITY |
---|
| 93 | struct passwd *p; |
---|
| 94 | #if defined(_REENTRANT) && defined(HAVE_GETPWUID_R) |
---|
| 95 | struct passwd pwbuf; |
---|
| 96 | #endif |
---|
| 97 | #endif |
---|
| 98 | #endif |
---|
| 99 | } AGENT_CTX; |
---|
| 100 | |
---|
| 101 | int process_features (AGENT_CTX *ATX, const char *features); |
---|
| 102 | int process_mode (AGENT_CTX *ATX, const char *mode); |
---|
| 103 | int check_configuration (AGENT_CTX *ATX); |
---|
| 104 | int apply_defaults (AGENT_CTX *ATX); |
---|
| 105 | int process_arguments (AGENT_CTX *ATX, int argc, char **argv); |
---|
| 106 | int initialize_atx (AGENT_CTX *ATX); |
---|
| 107 | int process_parseto (AGENT_CTX *ATX, const char *buf); |
---|
| 108 | buffer *read_stdin (AGENT_CTX *ATX); |
---|
| 109 | |
---|
| 110 | int init_pwent_cache(void); |
---|
| 111 | |
---|
| 112 | #ifndef MIN |
---|
| 113 | # define MAX(a,b) ((a)>(b)?(a):(b)) |
---|
| 114 | # define MIN(a,b) ((a)<(b)?(a):(b)) |
---|
| 115 | #endif /* !MIN */ |
---|
| 116 | |
---|
| 117 | /* |
---|
| 118 | * Agent context flag (DAF) |
---|
| 119 | * Do not confuse with libdspam's classification context flags (DSF) |
---|
| 120 | * |
---|
| 121 | */ |
---|
| 122 | |
---|
| 123 | #define DAF_STDOUT 0x01 |
---|
| 124 | #define DAF_DELIVER_SPAM 0x02 |
---|
| 125 | #define DAF_DELIVER_INNOCENT 0x04 |
---|
| 126 | #define DAF_WHITELIST 0x08 |
---|
| 127 | #define DAF_GLOBAL 0x10 |
---|
| 128 | #define DAF_INOCULATE 0x20 |
---|
| 129 | #define DAF_NOISE 0x40 |
---|
| 130 | #define DAF_MERGED 0x80 |
---|
| 131 | #define DAF_SUMMARY 0x100 |
---|
| 132 | #define DAF_UNLEARN 0x200 |
---|
| 133 | #define DAF_FIXED_TR_MODE 0x400 |
---|
| 134 | |
---|
| 135 | #define DAZ_WORD 0x01 |
---|
| 136 | #define DAZ_CHAIN 0x02 |
---|
| 137 | #define DAZ_SBPH 0x03 |
---|
| 138 | #define DAZ_OSB 0x04 |
---|
| 139 | |
---|
| 140 | #endif /* _AGENT_SHARED_H */ |
---|