mconfig.c

一百个病毒的源代码 包括熊猫烧香等 极其具有研究价值

/*
** Modular Logfile Analyzer
** Copyright 2000 Jan Kneschke <[e-mail]>
**
** Homepage: http://www.kneschke.de/projekte/modlogan
**

    This program is free software; you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation; either version 2 of the License, or
    (at your option) any later version, and provided that the above
    copyright and permission notice is included with all distributed
    copies of this or derived software.

    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., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA

**
** $Id: mconfig.c,v 1.28 2001/01/11 22:21:58 jk Exp $
*/

#include <libintl.h>
#include <locale.h>
#include <stdlib.h> 
#include <stdio.h> 
#include <string.h>
#include <errno.h>
#include <ctype.h>
#include <sys/stat.h>

#ifdef HAVE_LIBADNS
#include <adns.h>
#endif

#include "mlocale.h"
#include "mconfig.h"
#include "mdatatypes.h"
#include "misc.h"
#include "mplugins.h"


static int linenumber = 0;
static char *filename = NULL;

#define M_CONFIG_GROUP_UNKNOWN	-1
#define M_CONFIG_GROUP_IGNORE	0
#define M_CONFIG_GROUP_GLOBAL	1
#define M_CONFIG_GROUP_INPUT	2
#define M_CONFIG_GROUP_OUTPUT	3
#define M_CONFIG_GROUP_PROCESSOR	4

int mconfig_insert_value(void *dest, int type, char *value, int value_def) {
	pcre *match = NULL;
	const char *errptr;
	int erroffset = 0;
#define N 20 + 1
	int ovector[3 * N], n;
	const char **list;
	
	switch(type) {
		case M_CONFIG_TYPE_STRING: {
			char *str = *(char **)(dest);
			
			switch(value_def) {
				case M_CONFIG_VALUE_IGNORE: {
					if (str) break;

					str = malloc(strlen(value)+1);
					strcpy(str, value);
					
					*(char **)(dest) = str;
					break;
				}
				case M_CONFIG_VALUE_APPEND:
					fprintf(stderr,"%s.%d: append not supported for string type\n", __FILE__, __LINE__);
					break;
				case M_CONFIG_VALUE_OVERWRITE:
					if (str) free(str);
					str = malloc(strlen(value)+1);
					strcpy(str, value);
			
					*(char **)(dest) = str;
					break;
				default:
					fprintf(stderr,"%s.%d: unknown type of value definetion\n", __FILE__, __LINE__);
					break;
			}
			
			
			break;
		}
		case M_CONFIG_TYPE_STRING_LIST: {
			switch(value_def) {
				case M_CONFIG_VALUE_IGNORE:
					fprintf(stderr,"%s.%d: ignore not supported for grouping type\n", __FILE__, __LINE__);
					break;
				case M_CONFIG_VALUE_APPEND: {
					data_StrInt *data;
					mlist *l = *(mlist **)(dest);
						
					data = createStrInt(value, 1);
					mlist_insert(l, data);
					break;
				}
				case M_CONFIG_VALUE_OVERWRITE:
					fprintf(stderr,"%s.%d: overwrite not supported for grouping type\n", __FILE__, __LINE__);
					break;
				default:
					fprintf(stderr,"%s.%d: unknown type of value definetion\n", __FILE__, __LINE__);
					break;
			}
			
			break;
		}
		case M_CONFIG_TYPE_COLTRIPPL: {
			char *str = *(char **)(dest);
			
			if (!is_htmltripple(value)) {
				fprintf(stderr, "%s (%s:%i): %s\n", _("Invalid colortripple"), filename, linenumber, value);
				break;
			}
			
			switch(value_def) {
				case M_CONFIG_VALUE_IGNORE:
					if (str) break;
					str = malloc(strlen(value)+1);
					strcpy(str, value);
					
					*(char **)(dest) = str;
					break;
				case M_CONFIG_VALUE_APPEND:
					fprintf(stderr,"%s.%d: append not supported for colortripple type\n", __FILE__, __LINE__);
					break;
				case M_CONFIG_VALUE_OVERWRITE:
					if (str) free(str);
					str = malloc(strlen(value)+1);
					strcpy(str, value);
					
					*(char **)(dest) = str;
					break;
				default:
					fprintf(stderr,"%s.%d: unknown type of value definetion\n", __FILE__, __LINE__);
					break;
			}
					
				
			break;
		}
		case M_CONFIG_TYPE_SUBSTITUTE: {
			if ((match = pcre_compile(
				"^\"(.+)\",(.+)$", 
				0, &errptr, &erroffset, NULL)) == NULL) {
		
				fprintf(stderr, "%s.%d: rexexp compilation error at %s\n", __FILE__, __LINE__, errptr);
		
				return -1;
			} 
	
			if ((n = pcre_exec(match, NULL, value, strlen(value), 0, 0, ovector, 3 * N)) < 0) {
				if (n == PCRE_ERROR_NOMATCH) {
					fprintf(stderr, "%s.%d: string doesn't match: %s\n", __FILE__, __LINE__, value);
					fprintf(stderr, "%s (%s:%i): %s\n", "grouping", filename, linenumber, value);
				} else {
					fprintf(stderr, "%s.%d: execution error while matching: %d\n", __FILE__, __LINE__, n);
				}
				return -1;
			}
	
			pcre_get_substring_list(value, ovector, n, &list);
			
			switch(value_def) {
				case M_CONFIG_VALUE_IGNORE:
					fprintf(stderr,"%s.%d: ignore not supported for grouping type\n", __FILE__, __LINE__);
					break;
				case M_CONFIG_VALUE_APPEND: {
					data_Match *data;
					mlist *l = *(mlist **)(dest);
						
					data = createMatch((char *)list[1], (char *)list[2]);
					mlist_append(l, data);
					break;
				}
				case M_CONFIG_VALUE_OVERWRITE:
					fprintf(stderr,"%s.%d: overwrite not supported for grouping type\n", __FILE__, __LINE__);
					break;
				default:
					fprintf(stderr,"%s.%d: unknown type of value definetion\n", __FILE__, __LINE__);
					break;
			}
			
			free(list);
			free(match);
			break;
		}
		case M_CONFIG_TYPE_MATCH: {
			if ((match = pcre_compile(
				"^\"(.+)\"$", 
				0, &errptr, &erroffset, NULL)) == NULL) {
		
				fprintf(stderr, "%s.%d: rexexp compilation error at %s\n", __FILE__, __LINE__, errptr);
		
				return -1;
			} 
	
			if ((n = pcre_exec(match, NULL, value, strlen(value), 0, 0, ovector, 3 * N)) < 0) {
				if (n == PCRE_ERROR_NOMATCH) {
					fprintf(stderr, "%s.%d: string doesn't match: %s\n", __FILE__, __LINE__, value);
					fprintf(stderr, "%s (%s:%i): %s\n", "hiding", filename, linenumber, value);
				} else {
					fprintf(stderr, "%s.%d: execution error while matching: %d\n", __FILE__, __LINE__, n);
				}
				return -1;
			}
	
			pcre_get_substring_list(value, ovector, n, &list);
			
			switch(value_def) {
				case M_CONFIG_VALUE_IGNORE:
					fprintf(stderr,"%s.%d: ignore not supported for hiding type\n", __FILE__, __LINE__);
					break;
				case M_CONFIG_VALUE_APPEND: {
					data_Match *data;
					mlist *l = *(mlist **)(dest);
					
					data = createMatch((char *)list[1], "");
					
					mlist_append(l, data);
					break;
				}
				case M_CONFIG_VALUE_OVERWRITE:
					fprintf(stderr,"%s.%d: overwrite not supported for hiding type\n", __FILE__, __LINE__);
					break;
				default:
					fprintf(stderr,"%s.%d: unknown type of value definetion\n", __FILE__, __LINE__);
					break;
			}
			
			free(list);
			free(match);
			break;
		}
		case M_CONFIG_TYPE_INT: {
			int str = *(int *)(dest);
			char *endp;
			
			switch(value_def) {
				case M_CONFIG_VALUE_IGNORE:
					fprintf(stderr,"%s.%d: ignore not supported for int type\n", __FILE__, __LINE__);
					break;
				case M_CONFIG_VALUE_APPEND:
					fprintf(stderr,"%s.%d: append not supported for int type\n", __FILE__, __LINE__);
					break;
				case M_CONFIG_VALUE_OVERWRITE: {
					str = strtol(value, &endp, 10);
					
					if (*endp != '\0') {
						fprintf(stderr, "%s (%s:%i): %s\n", _("Value isn't an integer in line"), filename, linenumber, value);
					} else {
						*(int *)(dest) = str;
					}
					break;
				}
				default:
					fprintf(stderr,"%s.%d: unknown type of value definetion\n", __FILE__, __LINE__);
					break;
			}
			
			break;
		}
		case M_CONFIG_TYPE_CHAR: {
			char str = *(char *)(dest);
			
			switch(value_def) {
				case M_CONFIG_VALUE_IGNORE:
					fprintf(stderr,"%s.%d: ignore not supported for int type\n", __FILE__, __LINE__);
					break;
				case M_CONFIG_VALUE_APPEND:
					fprintf(stderr,"%s.%d: append not supported for int type\n", __FILE__, __LINE__);
					break;
				case M_CONFIG_VALUE_OVERWRITE: {
					str = *value;
					
					*(char *)(dest) = str;
					break;
				}
				default:
					fprintf(stderr,"%s.%d: unknown type of value definetion\n", __FILE__, __LINE__);
					break;
			}
			
			break;
		}
		default:
			fprintf(stderr,"%s.%d: unknown type of key-value definetion\n", __FILE__, __LINE__);
			break;
	}
#undef N	
	return 0;
}

int mconfig_set_defaults(mconfig *conf) {
	struct stat st;
	if (conf == NULL) return -1;
	
	if (conf->outputdir == NULL) {
		fprintf(stderr, "no output directory was set\n");
		return -1;
	}
	
	if (stat(conf->outputdir, &st) != 0) {
		fprintf(stderr, "Can't check if the output directory is ok (%s): %s\n", conf->outputdir, strerror(errno));
		return -1;
	} else {
		if (S_ISDIR(st.st_mode) && 
			(((st.st_mode & S_IWUSR) && (st.st_mode & S_IRUSR)) ||
			((st.st_mode & S_IWGRP) && (st.st_mode & S_IRGRP)) ||
			((st.st_mode & S_IWOTH) && (st.st_mode & S_IROTH)))
			) {
			/* every thing seems to be ok. */
		} else {
			fprintf(stderr, "the output directory doesn't have the right permissions: %s\n", conf->outputdir);
			return -1;
		}
	}

#ifdef HAVE_LIBADNS
	conf->adns = (adns_state *)malloc(sizeof(adns_state));
	
	if (adns_init(conf->adns, 0, 0)) {
		fprintf(stderr, "Can't init the resolver\n");
		return -1;
	}
#endif

	return 0;
}