// Read the data files
void
InputEvents::LoadDefaults(InputConfig &input_config)
{
// Get defaults
input_config.SetDefaults();
if (IsAltair()) {
#include "InputEvents_altair.cpp"
apply_defaults(input_config,
default_modes,
default_events,
ARRAY_SIZE(default_events),
default_gesture2event,
default_key2event, default_gc2event, default_n2event,
default_labels);
} else {
#include "InputEvents_default.cpp"
apply_defaults(input_config,
default_modes,
default_events,
ARRAY_SIZE(default_events),
default_gesture2event,
default_key2event, default_gc2event, default_n2event,
default_labels);
}
}
ConfigSettings::ConfigSettings(){
num_numids = 0;
//defaults
strcpy(_d_card, "hw:0");
_d_scaling = Element::LINEAR;
_d_auto_mute = true;
_d_vertical = false;
_d_window_x=-1;
_d_window_y=-1;
_d_window_width=270;
_d_window_height=256;
_d_slider_width=102;
_d_slider_height=20;
_d_slider_margin = 2;
_d_seg_thickness = 2;
_d_seg_spacing = 1;
_d_background_color[0]=0.0;
_d_background_color[1]=0.0;
_d_background_color[2]=0.0;
_d_border_color[0]=0.0;
_d_border_color[1]=0.0;
_d_border_color[2]=0.0;
_d_unlit_color[0]=0.6;
_d_unlit_color[1]=0.2;
_d_unlit_color[2]=0.0;
_d_lit_color[0]=1.0;
_d_lit_color[1]=0.8;
_d_lit_color[2]=0.0;
_d_enable_tray_icon = true;
_d_enable_tray_menu = true;
_d_enable_tray_icon_background_color = false;
_d_tray_icon_background_color[0]=0.8;
_d_tray_icon_background_color[1]=0.8;
_d_tray_icon_background_color[2]=0.8;
_d_tray_slider_vertical = true;
_d_tray_slider_width=20;
_d_tray_slider_height=102;
_d_tray_slider_offset=-1;
strcpy(icon_file_names[0], VOL_MUTED_IMAGE);
strcpy(icon_file_names[1], VOL_NONE_IMAGE);
strcpy(icon_file_names[2], VOL_LOW_IMAGE);
strcpy(icon_file_names[3], VOL_MEDIUM_IMAGE);
strcpy(icon_file_names[4], VOL_HIGH_IMAGE);
tray_icon = NULL;
tray_icon_image = NULL;
tray_icon_menu = NULL;
tray_slider = NULL;
tray_control = NULL;
apply_defaults();
}
bool
conftab::run (const str &file, u_int opts, int fd, status_t *sp)
{
bool errors = false;
bool unknown = false;
if (opts & (CONFTAB_APPLY_DEFAULTS|CONFTAB_VERBOSE)) {
reset ();
}
if (file) {
parseargs pa (file, fd);
vec<str> av;
int line;
if (opts & CONFTAB_VERBOSE) {
warn << "Parsing configuration file: " << file << "\n";
}
while (pa.getline (&av, &line)) {
if (!match (av, file, line, &errors)) {
warn << file << ":" << line << ": unknown config parameter\n";
unknown = true;
}
}
}
if (opts & CONFTAB_APPLY_DEFAULTS)
apply_defaults ();
if (opts & CONFTAB_VERBOSE)
report ();
if (sp) {
if (errors) *sp = ERROR;
else if (unknown) *sp = UNKNOWN;
else *sp = OK;
}
return !(errors || unknown);
}
int main(int argc, char *argv[])
{
mmatic *mm = mmatic_create();
mmatic *mmtmp = mmatic_create();
struct mg *mg;
int i;
/*
* initialize and parse config
*/
mg = mmatic_zalloc(mm, sizeof(struct mg));
mg->mm = mm;
mg->mmtmp = mmtmp;
apply_defaults(mg);
/* get my id number from hostname */
fetch_myid(mg);
/* parse command line options */
if (parse_argv(mg, argc, argv))
return 1;
/* parse configuration file options */
if (mg->options.conf_file) {
if (parse_config(mg))
return 4;
}
/*
* config syntax looks OK, see if it is feasible
*/
/* initialize random number generator */
srand48(mg->options.myid);
/* init libevent */
mg->evb = event_init();
event_set_log_callback(libevent_log);
/* init stats structures so mgstats_aggregator_add() used somewhere below works */
mgstats_init(mg);
/* attach to raw interfaces */
if (mgi_init(mg, handle_packet) <= 0) {
dbg(0, "no available interfaces found\n");
return 2;
}
/* parse traffic file */
if (parse_traffic(mg))
return 3;
/*
* all OK, prepare to start
*/
/* synchronize time reference point on all nodes */
mgc_sync(mg);
/* schedule stats writing */
mgstats_start(mg);
/* attach global stats */
_stats_init(mg);
/* schedule heartbeat and disk sync signals */
heartbeat_init(mg);
sync_init(mg);
/* schedule the real work of this node: line generators */
for (i = 1; i < TRAFFIC_LINE_MAX; i++) {
if (!(mg->lines[i] && mg->lines[i]->my))
continue;
/* this will schedule first execution */
mgs_sleep(mg->lines[i], NULL);
mg->running++;
}
/* suppose last frame was received now */
gettimeofday(&mg->last, NULL);
/*
* start!
*/
dbg(0, "Starting\n");
event_base_dispatch(mg->evb);
/*******************************/
/*
* cleanup after end of libevent loop
*/
event_base_free(mg->evb);
mmatic_free(mg->mm);
mmatic_free(mg->mmtmp);
fflush(NULL);
sync();
return 0;
}
//parse the config file
void ConfigSettings::parse_config(char *config_file){
//first set the defaults
apply_defaults();
//store the filename
strcpy(_config_file, config_file);
FILE *cfile = fopen(_config_file, "r");
if (!cfile){
fprintf(stdout, _("Cannot read file: %s\nUsing defaults...\n"), _config_file);
return;
}
char buffer[256];
char *tmpptr;
while (fgets(buffer, 256, cfile)){
//use the # as a comment, and ignore newlines
if (buffer[0] == '#' || buffer[0] == '\n'){
continue;
}
tmpptr=strtok(buffer, "=\n");
if (strcmp(tmpptr, "card")==0){
tmpptr=strtok(NULL, "=\"\n");
strcpy(card, tmpptr);
} else if (strcmp(tmpptr, "scaling")==0){
tmpptr=strtok(NULL, "=\n");
scaling=atoi(tmpptr);
if (scaling < 0 || scaling > NUM_SCALE_T - 1){ scaling = _d_scaling; } //if out of bounds, use default
} else if (strcmp(tmpptr, "auto_mute")==0){
tmpptr=strtok(NULL, "=\n");
auto_mute=(bool)atoi(tmpptr);
} else if (strcmp(tmpptr, "vertical")==0){
tmpptr=strtok(NULL, "=\n");
vertical=(bool)atoi(tmpptr);
} else if (strcmp(tmpptr, "window_x")==0){
tmpptr=strtok(NULL, "=\n");
window_x=atoi(tmpptr);
} else if (strcmp(tmpptr, "window_y")==0){
tmpptr=strtok(NULL, "=\n");
window_y=atoi(tmpptr);
} else if (strcmp(tmpptr, "window_width")==0){
tmpptr=strtok(NULL, "=\n");
window_width=atoi(tmpptr);
} else if (strcmp(tmpptr, "window_height")==0){
tmpptr=strtok(NULL, "=\n");
window_height=atoi(tmpptr);
} else if (strcmp(tmpptr, "slider_width")==0){
tmpptr=strtok(NULL, "=\n");
slider_width=atoi(tmpptr);
} else if (strcmp(tmpptr, "slider_height")==0){
tmpptr=strtok(NULL, "=\n");
slider_height=atoi(tmpptr);
} else if (strcmp(tmpptr, "slider_margin")==0){
tmpptr=strtok(NULL, "=\n");
slider_margin=atoi(tmpptr);
} else if (strcmp(tmpptr, "seg_thickness")==0){
tmpptr=strtok(NULL, "=\n");
seg_thickness=atoi(tmpptr);
} else if (strcmp(tmpptr, "seg_spacing")==0){
tmpptr=strtok(NULL, "=\n");
seg_spacing=atoi(tmpptr);
} else if (strcmp(tmpptr, "background_color")==0){
tmpptr=strtok(NULL, "=\n");
htonf(background_color, tmpptr);
} else if (strcmp(tmpptr, "border_color")==0){
tmpptr=strtok(NULL, "=\n");
htonf(border_color, tmpptr);
} else if (strcmp(tmpptr, "unlit_color")==0){
tmpptr=strtok(NULL, "=\n");
htonf(unlit_color, tmpptr);
} else if (strcmp(tmpptr, "lit_color")==0){
tmpptr=strtok(NULL, "=\n");
htonf(lit_color, tmpptr);
} else if (strcmp(tmpptr, "enable_tray_icon")==0){
tmpptr=strtok(NULL, "=\n");
enable_tray_icon=(bool)atoi(tmpptr);
} else if (strcmp(tmpptr, "enable_tray_menu")==0){
tmpptr=strtok(NULL, "=\n");
enable_tray_menu=(bool)atoi(tmpptr);
} else if (strcmp(tmpptr, "tray_icon_background_color")==0){
tmpptr=strtok(NULL, "=\n");
htonf(tray_icon_background_color, tmpptr);
enable_tray_icon_background_color=true;
} else if (strcmp(tmpptr, "tray_slider_vertical")==0){
tmpptr=strtok(NULL, "=\n");
tray_slider_vertical=(bool)atoi(tmpptr);
} else if (strcmp(tmpptr, "tray_slider_width")==0){
tmpptr=strtok(NULL, "=\n");
tray_slider_width=atoi(tmpptr);
} else if (strcmp(tmpptr, "tray_slider_height")==0){
tmpptr=strtok(NULL, "=\n");
tray_slider_height=atoi(tmpptr);
} else if (strcmp(tmpptr, "tray_slider_offset")==0){
tmpptr=strtok(NULL, "=\n");
tray_slider_offset=atoi(tmpptr);
} else if (strcmp(tmpptr, "tray_control")==0){
tmpptr=strtok(NULL, "=\n");
tray_control_numid=atoi(tmpptr);
} else if (strcmp(tmpptr, "sliders:")==0){
int n;
for (n=0; fgets(buffer, 256, cfile); n++){
char *buff = buffer;
//skip over any whitespace
while (buff[0] && (buff[0] == ' ' || buff[0] == '\t')){ buff++; }
//skip any commented lines or newlines
if (buff[0] == '#' || buff[0] == '\n'){
n--;
continue;
}
//save the numid and initialize the name to an empty string
numid_list[n] = atoi(buff);
name_list[n][0] = '\0';
}
num_numids=n;
}
}
fclose(cfile);
}
void *process_connection(void *ptr) {
char *server_ident = _ds_read_attribute(agent_config, "ServerIdent");
THREAD_CTX *TTX = (THREAD_CTX *) ptr;
AGENT_CTX *ATX = NULL;
char *input, *cmdline = NULL, *token, *ptrptr;
buffer *message = NULL;
char *parms=NULL, *p=NULL;
int i, locked = -1, invalid = 0;
int server_mode = SSM_DSPAM;
char *argv[64];
char buf[1024];
int tries = 0;
int argc = 0;
FILE *fd = 0;
if (_ds_read_attribute(agent_config, "ServerMode") &&
!strcasecmp(_ds_read_attribute(agent_config, "ServerMode"), "standard"))
{
server_mode = SSM_STANDARD;
}
if (_ds_read_attribute(agent_config, "ServerMode") &&
!strcasecmp(_ds_read_attribute(agent_config, "ServerMode"), "auto"))
{
server_mode = SSM_AUTO;
}
/* Initialize a file descriptor hook for dspam to use as stdout */
fd = fdopen(TTX->sockfd, "w");
if (!fd) {
close(TTX->sockfd);
goto CLOSE;
}
setbuf(fd, NULL);
TTX->packet_buffer = buffer_create(NULL);
if (TTX->packet_buffer == NULL)
goto CLOSE;
/*
* Send greeting banner
* in auto mode, we want to look like a regular LMTP server so we don't
* cause any compatibility problems. in dspam mode, we can change this.
*/
snprintf(buf, sizeof(buf), "%d DSPAM %sLMTP %s %s",
LMTP_GREETING,
(server_mode == SSM_DSPAM) ? "D" : "",
VERSION,
(server_mode == SSM_DSPAM) ? "Authentication Required" : "Ready");
if (send_socket(TTX, buf)<=0)
goto CLOSE;
TTX->authenticated = 0;
/* LHLO */
input = daemon_expect(TTX, "LHLO");
if (input == NULL)
goto CLOSE;
if (server_mode == SSM_AUTO && input[4]) {
char buff[128];
/*
* Auto-detect the server mode based on whether or not the ident is
* assigned a password in dspam.conf
*/
snprintf(buff, sizeof(buff), "ServerPass.%s", input + 5);
chomp(buff);
if (_ds_read_attribute(agent_config, buff))
server_mode = SSM_DSPAM;
else
server_mode = SSM_STANDARD;
}
free(input);
/* Advertise extensions */
if (daemon_extension(TTX, (server_ident) ? server_ident :
"localhost.localdomain")<=0)
goto CLOSE;
if (daemon_extension(TTX, "PIPELINING")<=0)
goto CLOSE;
if (daemon_extension(TTX, "ENHANCEDSTATUSCODES")<=0)
goto CLOSE;
if (server_mode == SSM_DSPAM)
if (daemon_extension(TTX, "DSPAMPROCESSMODE")<=0)
goto CLOSE;
if (daemon_extension(TTX, "8BITMIME")<=0)
goto CLOSE;
if (daemon_reply(TTX, LMTP_OK, "", "SIZE")<=0)
goto CLOSE;
/* Main protocol loop */
while(1) {
char processmode[256];
parms = NULL;
/* Configure a new agent context for each pass */
ATX = calloc(1, sizeof(AGENT_CTX));
if (ATX == NULL) {
LOG(LOG_CRIT, ERR_MEM_ALLOC);
daemon_reply(TTX, LMTP_TEMP_FAIL, "4.3.0", ERR_MEM_ALLOC);
goto CLOSE;
}
if (initialize_atx(ATX)) {
LOG(LOG_ERR, ERR_AGENT_INIT_ATX);
daemon_reply(TTX, LMTP_BAD_CMD, "5.3.0", ERR_AGENT_INIT_ATX);
goto CLOSE;
}
/* MAIL FROM (and authentication, if SSM_DSPAM) */
processmode[0] = 0;
while(!TTX->authenticated) {
input = daemon_expect(TTX, "MAIL FROM");
if (RSET(input))
goto RSET;
if (input == NULL)
goto CLOSE;
else {
char *pass, *ident;
chomp(input);
if (server_mode == SSM_STANDARD) {
TTX->authenticated = 1;
ATX->mailfrom[0] = 0;
_ds_extract_address(ATX->mailfrom, input, sizeof(ATX->mailfrom));
if (daemon_reply(TTX, LMTP_OK, "2.1.0", "OK")<=0) {
free(input);
goto CLOSE;
}
} else {
char id[256];
pass = ident = NULL;
id[0] = 0;
if (!_ds_extract_address(id, input, sizeof(id))) {
pass = strtok_r(id, "@", &ptrptr);
ident = strtok_r(NULL, "@", &ptrptr);
}
if (pass && ident) {
char *serverpass;
char *ptr, *ptr2, *ptr3;
snprintf(buf, sizeof(buf), "ServerPass.%s", ident);
serverpass = _ds_read_attribute(agent_config, buf);
snprintf(buf, sizeof(buf), "ServerPass.%s", ident);
if (serverpass && !strcmp(pass, serverpass)) {
TTX->authenticated = 1;
/* Parse PROCESSMODE service tag */
ptr = strstr(input, "DSPAMPROCESSMODE=\"");
if (ptr) {
char *mode;
int i;
ptr2 = strchr(ptr, '"')+1;
mode = ptr2;
while((ptr3 = strstr(ptr2, "\\\"")))
ptr2 = ptr3+2;
ptr3 = strchr(ptr2+2, '"');
if (ptr3)
ptr3[0] = 0;
strlcpy(processmode, mode, sizeof(processmode));
ptr = processmode;
for(i=0; ptr[i]; i++) {
if (ptr[i] == '\\' && ptr[i+1] == '"') {
strcpy(ptr+i, ptr+i+1);
}
}
LOGDEBUG("process mode: '%s'", processmode);
}
if (daemon_reply(TTX, LMTP_OK, "2.1.0", "OK")<=0) {
free(input);
goto CLOSE;
}
}
}
}
free(input);
if (!TTX->authenticated) {
LOGDEBUG("fd %d authentication failure.", TTX->sockfd);
if (daemon_reply(TTX, LMTP_AUTH_ERROR, "5.1.0",
"Authentication Required")<=0)
{
free(input);
goto CLOSE;
}
tries++;
if (tries>=3) {
struct timeval tv;
tv.tv_sec = 5;
tv.tv_usec = 0;
select(0, NULL, NULL, NULL, &tv);
goto CLOSE;
}
}
}
}
/* MAIL FROM response */
snprintf(buf, sizeof(buf), "%d OK", LMTP_OK);
argc = 1;
argv[0] = "dspam";
argv[1] = 0;
/* Load open-LMTP configuration parameters */
if (server_mode == SSM_STANDARD) {
parms = _ds_read_attribute(agent_config, "ServerParameters");
if (parms) {
p = strdup(parms);
if (p) {
token = strtok_r(p, " ", &ptrptr);
while(token != NULL && argc<63) {
argv[argc] = token;
argc++;
argv[argc] = 0;
token = strtok_r(NULL, " ", &ptrptr);
}
}
}
}
/* RCPT TO */
while(ATX->users->items == 0 || invalid) {
free(cmdline);
cmdline = daemon_getline(TTX, 300);
while(cmdline &&
(!strncasecmp(cmdline, "RCPT TO:", 8) ||
!strncasecmp(cmdline, "RSET", 4)))
{
char username[256];
char *at = NULL;
if (!strncasecmp(cmdline, "RSET", 4)) {
snprintf(buf, sizeof(buf), "%d OK", LMTP_OK);
if (send_socket(TTX, buf)<=0)
goto CLOSE;
goto RSET;
}
if (_ds_extract_address(username, cmdline, sizeof(username)) ||
username[0] == 0 || username[0] == '-' || username[0] == '@')
{
if ((server_mode == SSM_DSPAM) || (server_mode == SSM_STANDARD && _ds_validate_address(username) == 0)) {
daemon_reply(TTX, LMTP_BAD_CMD, "5.1.2", ERR_LMTP_BAD_RCPT);
goto GETCMD;
}
}
if (_ds_match_attribute(agent_config, "Broken", "case"))
lc(username, username);
/* Chop of @.* from the recipient */
if (_ds_match_attribute(agent_config, "StripRcptDomain", "on")) {
at = strchr(username, '@');
if (at != NULL)
*at = '\0';
}
if (server_mode == SSM_DSPAM) {
nt_add(ATX->users, username);
}
else {
if (!parms || !strstr(parms, "--user "))
nt_add(ATX->users, username);
if (!ATX->recipients) {
ATX->recipients = nt_create(NT_CHAR);
if (ATX->recipients == NULL) {
LOG(LOG_CRIT, ERR_MEM_ALLOC);
goto CLOSE;
}
}
if (at != NULL)
*at = '@'; /* always add complete address (user@domain) to recipient list */
nt_add(ATX->recipients, username);
}
if (daemon_reply(TTX, LMTP_OK, "2.1.5", "OK")<=0)
goto CLOSE;
GETCMD:
free(cmdline);
cmdline = daemon_getline(TTX, 300);
}
if (cmdline == NULL)
goto CLOSE;
if (!strncasecmp(cmdline, "RSET", 4)) {
snprintf(buf, sizeof(buf), "%d OK", LMTP_OK);
if (send_socket(TTX, buf)<=0)
goto CLOSE;
goto RSET;
}
if (!strncasecmp(cmdline, "quit", 4)) {
daemon_reply(TTX, LMTP_OK, "2.0.0", "OK");
goto CLOSE;
}
/* Parse DSPAMPROCESSMODE input and set up process arguments */
if (server_mode == SSM_DSPAM && processmode[0] != 0) {
token = strtok_r(processmode, " ", &ptrptr);
while(token != NULL && argc<63) {
argv[argc] = token;
argc++;
argv[argc] = 0;
token = strtok_r(NULL, " ", &ptrptr);
}
}
invalid = 0;
if (process_arguments(ATX, argc, argv) || apply_defaults(ATX))
{
LOG(LOG_ERR, ERR_AGENT_INIT_ATX);
daemon_reply(TTX, LMTP_NO_RCPT, "5.1.0", ERR_AGENT_INIT_ATX);
invalid = 1;
} else if (ATX->users->items == 0) {
daemon_reply(TTX, LMTP_NO_RCPT, "5.1.1", ERR_AGENT_USER_UNDEFINED);
}
}
ATX->sockfd = fd;
ATX->sockfd_output = 0;
/* Something's terribly misconfigured */
if (check_configuration(ATX)) {
LOG(LOG_ERR, ERR_AGENT_MISCONFIGURED);
daemon_reply(TTX, LMTP_BAD_CMD, "5.3.5", ERR_AGENT_MISCONFIGURED);
goto CLOSE;
}
/* DATA */
if (cmdline != NULL) {
if (strncasecmp(cmdline, "DATA", 4)) {
if (daemon_reply(TTX, LMTP_BAD_CMD, "5.0.0", "Need DATA Here")<0)
goto CLOSE;
input = daemon_expect(TTX, "DATA");
if (input == NULL)
goto CLOSE;
if (RSET(input))
goto RSET;
}
}
if (daemon_reply(TTX, LMTP_DATA, "", INFO_LMTP_DATA)<=0)
goto CLOSE;
/*
* Read in the message from a DATA. I personally like to just hang up on
* a client stupid enough to pass in a NULL message for DATA, but you're
* welcome to do whatever you want.
*/
message = read_sock(TTX, ATX);
if (message == NULL || message->data == NULL || message->used == 0) {
daemon_reply(TTX, LMTP_FAILURE, "5.2.0", ERR_LMTP_MSG_NULL);
goto CLOSE;
}
/*
* Lock a database handle. We currently use the modulus of the socket
* id against the number of database connections in the cache. This
* seems to work rather well, as we would need to lock up the entire
* cache to wrap back around. And if we do wrap back around, that means
* we're busy enough to justify spinning on the current lock (vs. seeking
* out a free handle, which there likely are none).
*/
i = (TTX->sockfd % TTX->DTX->connection_cache);
LOGDEBUG("using database handle id %d", i);
if (TTX->DTX->flags & DRF_RWLOCK) {
if (ATX->operating_mode == DSM_CLASSIFY ||
ATX->training_mode == DST_NOTRAIN ||
(ATX->training_mode == DST_TOE && ATX->classification == DSR_NONE))
{
pthread_rwlock_rdlock(&TTX->DTX->connections[i]->rwlock);
} else {
pthread_rwlock_wrlock(&TTX->DTX->connections[i]->rwlock);
}
} else {
pthread_mutex_lock(&TTX->DTX->connections[i]->lock);
}
LOGDEBUG("handle locked");
ATX->dbh = TTX->DTX->connections[i]->dbh;
locked = i;
/* Process the message by tying back into the agent functions */
ATX->results = nt_create(NT_PTR);
if (ATX->results == NULL) {
LOG(LOG_CRIT, ERR_MEM_ALLOC);
goto CLOSE;
}
process_users(ATX, message);
/*
* Unlock the database handle as soon as we're done. We also need to
* refresh our handle index with a new handle if for some reason we
* had to re-establish a dewedged connection.
*/
if (TTX->DTX->connections[locked]->dbh != ATX->dbh)
TTX->DTX->connections[locked]->dbh = ATX->dbh;
if (TTX->DTX->flags & DRF_RWLOCK) {
pthread_rwlock_unlock(&TTX->DTX->connections[locked]->rwlock);
} else {
pthread_mutex_unlock(&TTX->DTX->connections[locked]->lock);
}
locked = -1;
/* Send a terminating '.' if --stdout in 'dspam' mode */
if (ATX->sockfd_output) {
if (send_socket(TTX, ".")<=0)
goto CLOSE;
/* Otherwise, produce standard delivery results */
} else {
struct nt_node *node_nt, *node_res = NULL;
struct nt_c c_nt;
if (ATX->recipients)
node_nt = c_nt_first(ATX->recipients, &c_nt);
else
node_nt = c_nt_first(ATX->users, &c_nt);
if (ATX->results)
node_res = ATX->results->first;
while(node_res && node_nt != NULL) {
agent_result_t result = (agent_result_t) node_res->ptr;
if (result != NULL && result->exitcode == ERC_SUCCESS)
{
if (server_mode == SSM_DSPAM) {
snprintf(buf, sizeof(buf),
"%d 2.6.0 <%s> Message accepted for delivery: %s",
LMTP_OK, (char *) node_nt->ptr,
(result->classification == DSR_ISSPAM) ? "SPAM" : "INNOCENT");
} else {
snprintf(buf, sizeof(buf),
"%d 2.6.0 <%s> Message accepted for delivery",
LMTP_OK, (char *) node_nt->ptr);
}
}
else
{
if (result != NULL && result->exitcode == ERC_PERMANENT_DELIVERY) {
snprintf(buf, sizeof(buf), "%d 5.3.0 <%s> %s",
LMTP_FAILURE, (char *) node_nt->ptr,
(result->text[0]) ? result->text : "Permanent error occured");
} else {
if (result != NULL && result->text[0]) {
snprintf(buf, sizeof(buf),
"%d 4.3.0 <%s> %s",
LMTP_TEMP_FAIL, (char *) node_nt->ptr, result->text);
} else {
snprintf(buf, sizeof(buf),
"%d 4.3.0 <%s> Error occured during %s",
LMTP_TEMP_FAIL, (char *) node_nt->ptr,
(result != NULL && result->exitcode == ERC_DELIVERY) ? "delivery" : "processing");
}
}
}
if (send_socket(TTX, buf)<=0)
goto CLOSE;
if (ATX->recipients)
node_nt = c_nt_next(ATX->recipients, &c_nt);
else
node_nt = c_nt_next(ATX->users, &c_nt);
if (node_res)
node_res = node_res->next;
}
}
/* Cleanup and get ready for another message */
RSET:
fflush(fd);
buffer_destroy(message);
message = NULL;
if (ATX != NULL) {
nt_destroy(ATX->users);
nt_destroy(ATX->recipients);
nt_destroy(ATX->results);
free(ATX);
ATX = NULL;
free(cmdline);
cmdline = NULL;
TTX->authenticated = 0;
/* argc = 0; */
}
free(p);
p = NULL;
} /* while(1) */
/* Close connection and return */
CLOSE:
if (locked>=0)
pthread_mutex_unlock(&TTX->DTX->connections[locked]->lock);
if (fd)
fclose(fd);
buffer_destroy(TTX->packet_buffer);
if (message)
buffer_destroy(message);
if (ATX != NULL) {
nt_destroy(ATX->users);
nt_destroy(ATX->recipients);
nt_destroy(ATX->results);
}
free(ATX);
free(cmdline);
free(TTX);
decrement_thread_count();
pthread_exit(0);
return 0;
}
int
main (int argc, char *argv[])
{
AGENT_CTX ATX;
int exitcode = EXIT_SUCCESS;
buffer *message = NULL; /* input Message */
int agent_init = 0; /* agent is initialized */
setbuf (stdout, NULL); /* unbuffered output */
#ifdef DEBUG
DO_DEBUG = 0;
#endif
srand ((long) time << (long) getpid());
umask (006);
#ifndef DAEMON
LOG(LOG_ERR, ERR_DAEMON_NO_SUPPORT);
exit(EXIT_FAILURE);
#endif
/* Read dspam.conf into global config structure (ds_config_t) */
agent_config = read_config(NULL);
if (!agent_config) {
LOG(LOG_ERR, ERR_AGENT_READ_CONFIG);
exitcode = EXIT_FAILURE;
goto BAIL;
}
if (!_ds_read_attribute(agent_config, "Home")) {
LOG(LOG_ERR, ERR_AGENT_DSPAM_HOME);
exitcode = EXIT_FAILURE;
goto BAIL;
}
/* Set up agent context to define behavior of processor */
if (initialize_atx(&ATX)) {
LOG(LOG_ERR, ERR_AGENT_INIT_ATX);
exitcode = EXIT_FAILURE;
goto BAIL;
} else {
agent_init = 1;
}
if (process_arguments(&ATX, argc, argv)) {
LOG(LOG_ERR, ERR_AGENT_INIT_ATX);
exitcode = EXIT_FAILURE;
goto BAIL;
}
if (apply_defaults(&ATX)) {
LOG(LOG_ERR, ERR_AGENT_INIT_ATX);
exitcode = EXIT_FAILURE;
goto BAIL;
}
if (check_configuration(&ATX)) {
LOG(LOG_ERR, ERR_AGENT_MISCONFIGURED);
exitcode = EXIT_FAILURE;
goto BAIL;
}
/* Read the message in and apply ParseTo services */
message = read_stdin(&ATX);
if (message == NULL) {
exitcode = EXIT_FAILURE;
goto BAIL;
}
if (ATX.users->items == 0)
{
LOG (LOG_ERR, ERR_AGENT_USER_UNDEFINED);
fprintf (stderr, "%s\n", SYNTAX);
exitcode = EXIT_FAILURE;
goto BAIL;
}
/* Perform client-based processing */
#ifdef DAEMON
if (_ds_read_attribute(agent_config, "ClientIdent") &&
(_ds_read_attribute(agent_config, "ClientHost") ||
_ds_read_attribute(agent_config, "ServerDomainSocketPath")))
{
exitcode = client_process(&ATX, message);
} else {
LOG(LOG_ERR, ERR_CLIENT_INVALID_CONFIG);
exitcode = EINVAL;
}
#endif
BAIL:
if (message)
buffer_destroy(message);
if (agent_init)
nt_destroy(ATX.users);
if (agent_config)
_ds_destroy_config(agent_config);
exit (exitcode);
}
