syslog2iptables: src/syslogconfig.cpp annotate

annotate src/syslogconfig.cpp @ 64:4b147494fc64

Added tag stable-1-0-16 for changeset 60f59936fabb

author	Carl Byington <carl@five-ten-sg.com>
date	Sat, 19 Dec 2015 10:25:11 -0800
parents	60f59936fabb
children	0e736950a117

rev	line source
36 6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	1 /*
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	2
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	3 Copyright (c) 2007 Carl Byington - 510 Software Group, released under
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	4 the GPL version 3 or any later version at your choice available at
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	5 http://www.gnu.org/licenses/gpl-3.0.txt
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	6
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	7 */
1 551433a01cab initial coding carl parents: diff changeset	8
551433a01cab initial coding carl parents: diff changeset	9 #include "includes.h"
2 6e88da080f08 initial coding carl parents: 1 diff changeset	10 #include <fcntl.h>
3 8fe310e5cd44 initial coding carl parents: 2 diff changeset	11 #include <sys/socket.h>
8fe310e5cd44 initial coding carl parents: 2 diff changeset	12 #include <netinet/in.h>
8fe310e5cd44 initial coding carl parents: 2 diff changeset	13 #include <arpa/inet.h>
8fe310e5cd44 initial coding carl parents: 2 diff changeset	14 #include <netdb.h>
4 2737ab01659a initial coding carl parents: 3 diff changeset	15 #include <limits.h>
1 551433a01cab initial coding carl parents: diff changeset	16
48 ba0259c9e411 Fixes to compile on Fedora 9 and for const correctness Carl Byington <carl@five-ten-sg.com> parents: 42 diff changeset	17 const char *token_add;
ba0259c9e411 Fixes to compile on Fedora 9 and for const correctness Carl Byington <carl@five-ten-sg.com> parents: 42 diff changeset	18 const char *token_bucket;
51 206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	19 const char *token_context;
48 ba0259c9e411 Fixes to compile on Fedora 9 and for const correctness Carl Byington <carl@five-ten-sg.com> parents: 42 diff changeset	20 const char *token_file;
ba0259c9e411 Fixes to compile on Fedora 9 and for const correctness Carl Byington <carl@five-ten-sg.com> parents: 42 diff changeset	21 const char *token_ignore;
ba0259c9e411 Fixes to compile on Fedora 9 and for const correctness Carl Byington <carl@five-ten-sg.com> parents: 42 diff changeset	22 const char *token_include;
ba0259c9e411 Fixes to compile on Fedora 9 and for const correctness Carl Byington <carl@five-ten-sg.com> parents: 42 diff changeset	23 const char *token_index;
ba0259c9e411 Fixes to compile on Fedora 9 and for const correctness Carl Byington <carl@five-ten-sg.com> parents: 42 diff changeset	24 const char *token_lbrace;
ba0259c9e411 Fixes to compile on Fedora 9 and for const correctness Carl Byington <carl@five-ten-sg.com> parents: 42 diff changeset	25 const char *token_message;
ba0259c9e411 Fixes to compile on Fedora 9 and for const correctness Carl Byington <carl@five-ten-sg.com> parents: 42 diff changeset	26 const char *token_pattern;
ba0259c9e411 Fixes to compile on Fedora 9 and for const correctness Carl Byington <carl@five-ten-sg.com> parents: 42 diff changeset	27 const char *token_rbrace;
ba0259c9e411 Fixes to compile on Fedora 9 and for const correctness Carl Byington <carl@five-ten-sg.com> parents: 42 diff changeset	28 const char *token_remove;
ba0259c9e411 Fixes to compile on Fedora 9 and for const correctness Carl Byington <carl@five-ten-sg.com> parents: 42 diff changeset	29 const char *token_semi;
ba0259c9e411 Fixes to compile on Fedora 9 and for const correctness Carl Byington <carl@five-ten-sg.com> parents: 42 diff changeset	30 const char *token_slash;
ba0259c9e411 Fixes to compile on Fedora 9 and for const correctness Carl Byington <carl@five-ten-sg.com> parents: 42 diff changeset	31 const char *token_threshold;
51 206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	32 string_set all_strings;// owns all the strings, only modified by the config loader thread
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	33 recorder_map recorders; // all the recorders are named
36 6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	34 const int maxlen = 1000; // used for snprintf buffers
58 b45dddebe8fc Add exponential increase in penalty for repeat offenders Carl Byington <carl@five-ten-sg.com> parents: 51 diff changeset	35 const int scale_max = 500000;
3 8fe310e5cd44 initial coding carl parents: 2 diff changeset	36
8fe310e5cd44 initial coding carl parents: 2 diff changeset	37
8fe310e5cd44 initial coding carl parents: 2 diff changeset	38 ////////////////////////////////////////////////
8fe310e5cd44 initial coding carl parents: 2 diff changeset	39 //
51 206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	40
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	41 IPR::IPR() {
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	42 reference_count = 0;
58 b45dddebe8fc Add exponential increase in penalty for repeat offenders Carl Byington <carl@five-ten-sg.com> parents: 51 diff changeset	43 daily_timer = 86400;
51 206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	44 }
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	45
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	46 IPR* IPR::find(const char* name) {
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	47 recorder_map::iterator m = recorders.find(name);
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	48 if (m == recorders.end()) recorders[name] = new IPR;
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	49 recorders[name]->reference(1);
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	50 return recorders[name];
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	51 }
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	52
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	53
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	54 void IPR::release(const char* name) {
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	55 recorder_map::iterator m = recorders.find(name);
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	56 IPR* i = (*m).second;
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	57 int r = i->reference(-1);
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	58 if (r == 0) {
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	59 delete i;
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	60 recorders.erase(m);
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	61 }
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	62 }
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	63
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	64
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	65 void IPR::add(int ip, int amount, CONTEXT &con, const char file_name, int pattern_index, const char message) {
36 6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	66 if (con.looking(ip)) {
63 60f59936fabb good authentication prevents ip blocking for awhile Carl Byington <carl@five-ten-sg.com> parents: 59 diff changeset	67 if (amount > 0) {
60f59936fabb good authentication prevents ip blocking for awhile Carl Byington <carl@five-ten-sg.com> parents: 59 diff changeset	68 ip_buckets::iterator j = repeat_offenders.find(ip);
60f59936fabb good authentication prevents ip blocking for awhile Carl Byington <carl@five-ten-sg.com> parents: 59 diff changeset	69 int scale = (j == repeat_offenders.end()) ? 1 : (*j).second.count;
60f59936fabb good authentication prevents ip blocking for awhile Carl Byington <carl@five-ten-sg.com> parents: 59 diff changeset	70 amount *= scale;
58 b45dddebe8fc Add exponential increase in penalty for repeat offenders Carl Byington <carl@five-ten-sg.com> parents: 51 diff changeset	71
63 60f59936fabb good authentication prevents ip blocking for awhile Carl Byington <carl@five-ten-sg.com> parents: 59 diff changeset	72 ip_buckets::iterator i = violations.find(ip);
60f59936fabb good authentication prevents ip blocking for awhile Carl Byington <carl@five-ten-sg.com> parents: 59 diff changeset	73 if (i == violations.end()) {
60f59936fabb good authentication prevents ip blocking for awhile Carl Byington <carl@five-ten-sg.com> parents: 59 diff changeset	74 bucket b;
60f59936fabb good authentication prevents ip blocking for awhile Carl Byington <carl@five-ten-sg.com> parents: 59 diff changeset	75 b.count = amount;
60f59936fabb good authentication prevents ip blocking for awhile Carl Byington <carl@five-ten-sg.com> parents: 59 diff changeset	76 b.blocked = (con.get_threshold() <= b.count);
60f59936fabb good authentication prevents ip blocking for awhile Carl Byington <carl@five-ten-sg.com> parents: 59 diff changeset	77 violations[ip] = b;
60f59936fabb good authentication prevents ip blocking for awhile Carl Byington <carl@five-ten-sg.com> parents: 59 diff changeset	78 if (b.blocked) {
60f59936fabb good authentication prevents ip blocking for awhile Carl Byington <carl@five-ten-sg.com> parents: 59 diff changeset	79 update(ip, true, scale, file_name, pattern_index, message);
60f59936fabb good authentication prevents ip blocking for awhile Carl Byington <carl@five-ten-sg.com> parents: 59 diff changeset	80 changed(con, ip, true);
60f59936fabb good authentication prevents ip blocking for awhile Carl Byington <carl@five-ten-sg.com> parents: 59 diff changeset	81 }
60f59936fabb good authentication prevents ip blocking for awhile Carl Byington <carl@five-ten-sg.com> parents: 59 diff changeset	82 }
60f59936fabb good authentication prevents ip blocking for awhile Carl Byington <carl@five-ten-sg.com> parents: 59 diff changeset	83 else {
60f59936fabb good authentication prevents ip blocking for awhile Carl Byington <carl@five-ten-sg.com> parents: 59 diff changeset	84 bucket &b = (*i).second;
60f59936fabb good authentication prevents ip blocking for awhile Carl Byington <carl@five-ten-sg.com> parents: 59 diff changeset	85 if ((b.count >= 0) && (b.count < 2600000)) {
60f59936fabb good authentication prevents ip blocking for awhile Carl Byington <carl@five-ten-sg.com> parents: 59 diff changeset	86 // good authentication (count<0) prevents blocking
60f59936fabb good authentication prevents ip blocking for awhile Carl Byington <carl@five-ten-sg.com> parents: 59 diff changeset	87 // not much point in blocking for more than a month
60f59936fabb good authentication prevents ip blocking for awhile Carl Byington <carl@five-ten-sg.com> parents: 59 diff changeset	88 b.count += amount;
60f59936fabb good authentication prevents ip blocking for awhile Carl Byington <carl@five-ten-sg.com> parents: 59 diff changeset	89 if ((!b.blocked) && (con.get_threshold() <= b.count)) {
60f59936fabb good authentication prevents ip blocking for awhile Carl Byington <carl@five-ten-sg.com> parents: 59 diff changeset	90 b.blocked = true;
60f59936fabb good authentication prevents ip blocking for awhile Carl Byington <carl@five-ten-sg.com> parents: 59 diff changeset	91 update(ip, true, scale, file_name, pattern_index, message);
60f59936fabb good authentication prevents ip blocking for awhile Carl Byington <carl@five-ten-sg.com> parents: 59 diff changeset	92 changed(con, ip, true);
60f59936fabb good authentication prevents ip blocking for awhile Carl Byington <carl@five-ten-sg.com> parents: 59 diff changeset	93 }
60f59936fabb good authentication prevents ip blocking for awhile Carl Byington <carl@five-ten-sg.com> parents: 59 diff changeset	94 }
36 6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	95 }
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	96 }
63 60f59936fabb good authentication prevents ip blocking for awhile Carl Byington <carl@five-ten-sg.com> parents: 59 diff changeset	97
60f59936fabb good authentication prevents ip blocking for awhile Carl Byington <carl@five-ten-sg.com> parents: 59 diff changeset	98 else { // amount < 0
60f59936fabb good authentication prevents ip blocking for awhile Carl Byington <carl@five-ten-sg.com> parents: 59 diff changeset	99 char buf[maxlen];
60f59936fabb good authentication prevents ip blocking for awhile Carl Byington <carl@five-ten-sg.com> parents: 59 diff changeset	100 in_addr ad;
60f59936fabb good authentication prevents ip blocking for awhile Carl Byington <carl@five-ten-sg.com> parents: 59 diff changeset	101 ad.s_addr = htonl(ip);
60f59936fabb good authentication prevents ip blocking for awhile Carl Byington <carl@five-ten-sg.com> parents: 59 diff changeset	102 snprintf(buf, maxlen, "%s for %s", message, inet_ntoa(ad));
60f59936fabb good authentication prevents ip blocking for awhile Carl Byington <carl@five-ten-sg.com> parents: 59 diff changeset	103 my_syslog(buf);
60f59936fabb good authentication prevents ip blocking for awhile Carl Byington <carl@five-ten-sg.com> parents: 59 diff changeset	104
60f59936fabb good authentication prevents ip blocking for awhile Carl Byington <carl@five-ten-sg.com> parents: 59 diff changeset	105 ip_buckets::iterator j = repeat_offenders.find(ip);
60f59936fabb good authentication prevents ip blocking for awhile Carl Byington <carl@five-ten-sg.com> parents: 59 diff changeset	106 if (j != repeat_offenders.end()) {
60f59936fabb good authentication prevents ip blocking for awhile Carl Byington <carl@five-ten-sg.com> parents: 59 diff changeset	107 repeat_offenders.erase(j++);
60f59936fabb good authentication prevents ip blocking for awhile Carl Byington <carl@five-ten-sg.com> parents: 59 diff changeset	108 snprintf(buf, maxlen, "removing %s from repeat offenders", inet_ntoa(ad));
60f59936fabb good authentication prevents ip blocking for awhile Carl Byington <carl@five-ten-sg.com> parents: 59 diff changeset	109 my_syslog(buf);
60f59936fabb good authentication prevents ip blocking for awhile Carl Byington <carl@five-ten-sg.com> parents: 59 diff changeset	110 }
60f59936fabb good authentication prevents ip blocking for awhile Carl Byington <carl@five-ten-sg.com> parents: 59 diff changeset	111 ip_buckets::iterator i = violations.find(ip);
60f59936fabb good authentication prevents ip blocking for awhile Carl Byington <carl@five-ten-sg.com> parents: 59 diff changeset	112 if (i == violations.end()) {
60f59936fabb good authentication prevents ip blocking for awhile Carl Byington <carl@five-ten-sg.com> parents: 59 diff changeset	113 bucket b;
60f59936fabb good authentication prevents ip blocking for awhile Carl Byington <carl@five-ten-sg.com> parents: 59 diff changeset	114 b.count = amount;
60f59936fabb good authentication prevents ip blocking for awhile Carl Byington <carl@five-ten-sg.com> parents: 59 diff changeset	115 b.blocked = false;
60f59936fabb good authentication prevents ip blocking for awhile Carl Byington <carl@five-ten-sg.com> parents: 59 diff changeset	116 violations[ip] = b;
60f59936fabb good authentication prevents ip blocking for awhile Carl Byington <carl@five-ten-sg.com> parents: 59 diff changeset	117 }
60f59936fabb good authentication prevents ip blocking for awhile Carl Byington <carl@five-ten-sg.com> parents: 59 diff changeset	118 else {
60f59936fabb good authentication prevents ip blocking for awhile Carl Byington <carl@five-ten-sg.com> parents: 59 diff changeset	119 bucket &b = (*i).second;
60f59936fabb good authentication prevents ip blocking for awhile Carl Byington <carl@five-ten-sg.com> parents: 59 diff changeset	120 b.count = amount;
60f59936fabb good authentication prevents ip blocking for awhile Carl Byington <carl@five-ten-sg.com> parents: 59 diff changeset	121 if (b.blocked) {
60f59936fabb good authentication prevents ip blocking for awhile Carl Byington <carl@five-ten-sg.com> parents: 59 diff changeset	122 update(ip, false, 0, NULL, 0, NULL);
60f59936fabb good authentication prevents ip blocking for awhile Carl Byington <carl@five-ten-sg.com> parents: 59 diff changeset	123 changed(con, ip, false);
36 6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	124 }
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	125 }
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	126 }
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	127 }
3 8fe310e5cd44 initial coding carl parents: 2 diff changeset	128 }
8fe310e5cd44 initial coding carl parents: 2 diff changeset	129
8fe310e5cd44 initial coding carl parents: 2 diff changeset	130
51 206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	131 void IPR::leak(int amount, CONTEXT &con) {
36 6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	132 for (ip_buckets::iterator i=violations.begin(); i!=violations.end(); ) {
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	133 int ip = (*i).first;
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	134 bucket &b = (*i).second;
63 60f59936fabb good authentication prevents ip blocking for awhile Carl Byington <carl@five-ten-sg.com> parents: 59 diff changeset	135 if (b.count < 0) {
60f59936fabb good authentication prevents ip blocking for awhile Carl Byington <carl@five-ten-sg.com> parents: 59 diff changeset	136 if (b.count >= -amount) violations.erase(i++);
60f59936fabb good authentication prevents ip blocking for awhile Carl Byington <carl@five-ten-sg.com> parents: 59 diff changeset	137 else {
60f59936fabb good authentication prevents ip blocking for awhile Carl Byington <carl@five-ten-sg.com> parents: 59 diff changeset	138 b.count += amount;
60f59936fabb good authentication prevents ip blocking for awhile Carl Byington <carl@five-ten-sg.com> parents: 59 diff changeset	139 i++;
36 6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	140 }
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	141 }
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	142 else {
63 60f59936fabb good authentication prevents ip blocking for awhile Carl Byington <carl@five-ten-sg.com> parents: 59 diff changeset	143 if (b.count <= amount) {
60f59936fabb good authentication prevents ip blocking for awhile Carl Byington <carl@five-ten-sg.com> parents: 59 diff changeset	144 if (b.blocked) {
60f59936fabb good authentication prevents ip blocking for awhile Carl Byington <carl@five-ten-sg.com> parents: 59 diff changeset	145 update(ip, false, 0, NULL, 0, NULL);
60f59936fabb good authentication prevents ip blocking for awhile Carl Byington <carl@five-ten-sg.com> parents: 59 diff changeset	146 changed(con, ip, false);
60f59936fabb good authentication prevents ip blocking for awhile Carl Byington <carl@five-ten-sg.com> parents: 59 diff changeset	147 }
60f59936fabb good authentication prevents ip blocking for awhile Carl Byington <carl@five-ten-sg.com> parents: 59 diff changeset	148 violations.erase(i++);
60f59936fabb good authentication prevents ip blocking for awhile Carl Byington <carl@five-ten-sg.com> parents: 59 diff changeset	149 }
60f59936fabb good authentication prevents ip blocking for awhile Carl Byington <carl@five-ten-sg.com> parents: 59 diff changeset	150 else {
60f59936fabb good authentication prevents ip blocking for awhile Carl Byington <carl@five-ten-sg.com> parents: 59 diff changeset	151 b.count -= amount;
60f59936fabb good authentication prevents ip blocking for awhile Carl Byington <carl@five-ten-sg.com> parents: 59 diff changeset	152 i++;
60f59936fabb good authentication prevents ip blocking for awhile Carl Byington <carl@five-ten-sg.com> parents: 59 diff changeset	153 }
36 6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	154 }
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	155 }
58 b45dddebe8fc Add exponential increase in penalty for repeat offenders Carl Byington <carl@five-ten-sg.com> parents: 51 diff changeset	156 daily_timer -= amount;
b45dddebe8fc Add exponential increase in penalty for repeat offenders Carl Byington <carl@five-ten-sg.com> parents: 51 diff changeset	157 if (daily_timer < 0) {
b45dddebe8fc Add exponential increase in penalty for repeat offenders Carl Byington <carl@five-ten-sg.com> parents: 51 diff changeset	158 daily_timer = 86400;
b45dddebe8fc Add exponential increase in penalty for repeat offenders Carl Byington <carl@five-ten-sg.com> parents: 51 diff changeset	159 for (ip_buckets::iterator j=repeat_offenders.begin(); j!=repeat_offenders.end(); ) {
b45dddebe8fc Add exponential increase in penalty for repeat offenders Carl Byington <carl@five-ten-sg.com> parents: 51 diff changeset	160 int ip = (*j).first;
b45dddebe8fc Add exponential increase in penalty for repeat offenders Carl Byington <carl@five-ten-sg.com> parents: 51 diff changeset	161 bucket &b = (*j).second;
b45dddebe8fc Add exponential increase in penalty for repeat offenders Carl Byington <carl@five-ten-sg.com> parents: 51 diff changeset	162 b.count = b.count * 2 / 3;
b45dddebe8fc Add exponential increase in penalty for repeat offenders Carl Byington <carl@five-ten-sg.com> parents: 51 diff changeset	163 if (b.count <= 2) {
b45dddebe8fc Add exponential increase in penalty for repeat offenders Carl Byington <carl@five-ten-sg.com> parents: 51 diff changeset	164 repeat_offenders.erase(j++);
b45dddebe8fc Add exponential increase in penalty for repeat offenders Carl Byington <carl@five-ten-sg.com> parents: 51 diff changeset	165 char buf[maxlen];
b45dddebe8fc Add exponential increase in penalty for repeat offenders Carl Byington <carl@five-ten-sg.com> parents: 51 diff changeset	166 in_addr ad;
b45dddebe8fc Add exponential increase in penalty for repeat offenders Carl Byington <carl@five-ten-sg.com> parents: 51 diff changeset	167 ad.s_addr = htonl(ip);
b45dddebe8fc Add exponential increase in penalty for repeat offenders Carl Byington <carl@five-ten-sg.com> parents: 51 diff changeset	168 snprintf(buf, maxlen, "removing %s from repeat offenders", inet_ntoa(ad));
b45dddebe8fc Add exponential increase in penalty for repeat offenders Carl Byington <carl@five-ten-sg.com> parents: 51 diff changeset	169 my_syslog(buf);
b45dddebe8fc Add exponential increase in penalty for repeat offenders Carl Byington <carl@five-ten-sg.com> parents: 51 diff changeset	170 }
b45dddebe8fc Add exponential increase in penalty for repeat offenders Carl Byington <carl@five-ten-sg.com> parents: 51 diff changeset	171 else {
b45dddebe8fc Add exponential increase in penalty for repeat offenders Carl Byington <carl@five-ten-sg.com> parents: 51 diff changeset	172 j++;
b45dddebe8fc Add exponential increase in penalty for repeat offenders Carl Byington <carl@five-ten-sg.com> parents: 51 diff changeset	173 }
b45dddebe8fc Add exponential increase in penalty for repeat offenders Carl Byington <carl@five-ten-sg.com> parents: 51 diff changeset	174 }
b45dddebe8fc Add exponential increase in penalty for repeat offenders Carl Byington <carl@five-ten-sg.com> parents: 51 diff changeset	175 }
36 6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	176 }
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	177
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	178
51 206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	179 void IPR::free_all(CONTEXT &con) {
37 e4eb969dfc4a shutdown removes iptables entries that we added carl parents: 36 diff changeset	180 if (debug_syslog > 2) {
e4eb969dfc4a shutdown removes iptables entries that we added carl parents: 36 diff changeset	181 my_syslog("syslog2iptables shutting down");
e4eb969dfc4a shutdown removes iptables entries that we added carl parents: 36 diff changeset	182 }
36 6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	183 for (ip_buckets::iterator i=violations.begin(); i!=violations.end(); i++) {
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	184 int ip = (*i).first;
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	185 bucket &b = (*i).second;
58 b45dddebe8fc Add exponential increase in penalty for repeat offenders Carl Byington <carl@five-ten-sg.com> parents: 51 diff changeset	186 if (b.blocked) {
b45dddebe8fc Add exponential increase in penalty for repeat offenders Carl Byington <carl@five-ten-sg.com> parents: 51 diff changeset	187 update(ip, false, 0, NULL, 0, NULL);
36 6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	188 changed(con, ip, false);
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	189 }
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	190 }
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	191 violations.clear();
20 0d65c3de34fd add better logging carl parents: 12 diff changeset	192 }
0d65c3de34fd add better logging carl parents: 12 diff changeset	193
0d65c3de34fd add better logging carl parents: 12 diff changeset	194
58 b45dddebe8fc Add exponential increase in penalty for repeat offenders Carl Byington <carl@five-ten-sg.com> parents: 51 diff changeset	195 void IPR::update(int ip, bool added, int scale, const char file_name, int pattern_index, const char message) {
36 6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	196 if (debug_syslog > 2) {
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	197 char buf[maxlen];
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	198 in_addr ad;
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	199 ad.s_addr = htonl(ip);
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	200 if (added) {
58 b45dddebe8fc Add exponential increase in penalty for repeat offenders Carl Byington <carl@five-ten-sg.com> parents: 51 diff changeset	201 if (message) snprintf(buf, maxlen, "dropping traffic from/to %s based on %s in %s, scale %d", inet_ntoa(ad), message, file_name, scale);
36 6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	202 else snprintf(buf, maxlen, "dropping traffic from/to %s based on pattern match %d in %s", inet_ntoa(ad), pattern_index, file_name);
58 b45dddebe8fc Add exponential increase in penalty for repeat offenders Carl Byington <carl@five-ten-sg.com> parents: 51 diff changeset	203 ip_buckets::iterator j = repeat_offenders.find(ip);
b45dddebe8fc Add exponential increase in penalty for repeat offenders Carl Byington <carl@five-ten-sg.com> parents: 51 diff changeset	204 if (j == repeat_offenders.end()) {
b45dddebe8fc Add exponential increase in penalty for repeat offenders Carl Byington <carl@five-ten-sg.com> parents: 51 diff changeset	205 bucket b;
b45dddebe8fc Add exponential increase in penalty for repeat offenders Carl Byington <carl@five-ten-sg.com> parents: 51 diff changeset	206 b.count = 2;
b45dddebe8fc Add exponential increase in penalty for repeat offenders Carl Byington <carl@five-ten-sg.com> parents: 51 diff changeset	207 b.blocked = true; // unused
b45dddebe8fc Add exponential increase in penalty for repeat offenders Carl Byington <carl@five-ten-sg.com> parents: 51 diff changeset	208 repeat_offenders[ip] = b;
b45dddebe8fc Add exponential increase in penalty for repeat offenders Carl Byington <carl@five-ten-sg.com> parents: 51 diff changeset	209 }
b45dddebe8fc Add exponential increase in penalty for repeat offenders Carl Byington <carl@five-ten-sg.com> parents: 51 diff changeset	210 else {
b45dddebe8fc Add exponential increase in penalty for repeat offenders Carl Byington <carl@five-ten-sg.com> parents: 51 diff changeset	211 bucket &b = (*j).second;
b45dddebe8fc Add exponential increase in penalty for repeat offenders Carl Byington <carl@five-ten-sg.com> parents: 51 diff changeset	212 if (b.count < scale_max) b.count = b.count * 3 / 2;
b45dddebe8fc Add exponential increase in penalty for repeat offenders Carl Byington <carl@five-ten-sg.com> parents: 51 diff changeset	213 }
36 6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	214 }
58 b45dddebe8fc Add exponential increase in penalty for repeat offenders Carl Byington <carl@five-ten-sg.com> parents: 51 diff changeset	215 else snprintf(buf, maxlen, "allowing traffic from/to %s", inet_ntoa(ad));
36 6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	216 my_syslog(buf);
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	217 }
3 8fe310e5cd44 initial coding carl parents: 2 diff changeset	218 }
8fe310e5cd44 initial coding carl parents: 2 diff changeset	219
8fe310e5cd44 initial coding carl parents: 2 diff changeset	220
51 206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	221 void IPR::changed(CONTEXT &con, int ip, bool added) {
36 6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	222 int t = con.get_threshold();
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	223 char buf[maxlen];
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	224 if (added) {
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	225 bucket &b = violations[ip];
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	226 if (con.looking(ip) && (b.count > t)) {
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	227 in_addr ad;
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	228 ad.s_addr = htonl(ip);
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	229 snprintf(buf, maxlen, con.add_command, inet_ntoa(ad));
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	230 system(buf);
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	231 }
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	232 }
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	233 else {
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	234 in_addr ad;
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	235 ad.s_addr = htonl(ip);
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	236 snprintf(buf, maxlen, con.remove_command, inet_ntoa(ad));
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	237 system(buf);
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	238 }
3 8fe310e5cd44 initial coding carl parents: 2 diff changeset	239 }
1 551433a01cab initial coding carl parents: diff changeset	240
551433a01cab initial coding carl parents: diff changeset	241
3 8fe310e5cd44 initial coding carl parents: 2 diff changeset	242 ////////////////////////////////////////////////
8fe310e5cd44 initial coding carl parents: 2 diff changeset	243 //
48 ba0259c9e411 Fixes to compile on Fedora 9 and for const correctness Carl Byington <carl@five-ten-sg.com> parents: 42 diff changeset	244 int ip_address(const char *have);
ba0259c9e411 Fixes to compile on Fedora 9 and for const correctness Carl Byington <carl@five-ten-sg.com> parents: 42 diff changeset	245 int ip_address(const char *have) {
36 6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	246 int ipaddr = 0;
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	247 in_addr ip;
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	248 if (inet_aton(have, &ip)) ipaddr = ip.s_addr;
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	249 else {
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	250 struct hostent *host = gethostbyname(have);
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	251 if (host && host->h_addrtype == AF_INET) memcpy(&ipaddr, host->h_addr, sizeof(ipaddr));
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	252 }
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	253 return ntohl(ipaddr);
3 8fe310e5cd44 initial coding carl parents: 2 diff changeset	254 }
8fe310e5cd44 initial coding carl parents: 2 diff changeset	255
8fe310e5cd44 initial coding carl parents: 2 diff changeset	256
8fe310e5cd44 initial coding carl parents: 2 diff changeset	257 ////////////////////////////////////////////////
8fe310e5cd44 initial coding carl parents: 2 diff changeset	258 //
48 ba0259c9e411 Fixes to compile on Fedora 9 and for const correctness Carl Byington <carl@five-ten-sg.com> parents: 42 diff changeset	259 PATTERN::PATTERN(TOKEN &tok, const char pattern_, int index_, int amount_, const char msg_) {
36 6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	260 pattern = pattern_;
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	261 index = index_;
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	262 amount = amount_;
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	263 message = msg_;
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	264 if (pattern) {
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	265 int rc = regcomp(&re, pattern, REG_ICASE \| REG_EXTENDED);
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	266 if (rc) {
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	267 char bu[maxlen];
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	268 regerror(rc, &re, bu, maxlen);
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	269 char buf[maxlen];
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	270 snprintf(buf, sizeof(buf), "pattern %s not valid - %s", pattern, bu);
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	271 tok.token_error(buf);
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	272 pattern = NULL;
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	273 }
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	274 }
3 8fe310e5cd44 initial coding carl parents: 2 diff changeset	275 }
8fe310e5cd44 initial coding carl parents: 2 diff changeset	276
8fe310e5cd44 initial coding carl parents: 2 diff changeset	277
8fe310e5cd44 initial coding carl parents: 2 diff changeset	278 PATTERN::~PATTERN() {
36 6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	279 regfree(&re);
3 8fe310e5cd44 initial coding carl parents: 2 diff changeset	280 }
8fe310e5cd44 initial coding carl parents: 2 diff changeset	281
8fe310e5cd44 initial coding carl parents: 2 diff changeset	282
51 206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	283 bool PATTERN::process(char buf, CONTEXT &con, const char file_name, int pattern_index) {
36 6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	284 if (pattern) {
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	285 const int nmatch = index+1;
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	286 regmatch_t match[nmatch];
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	287 if (0 == regexec(&re, buf, nmatch, match, 0)) {
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	288 int s = match[index].rm_so;
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	289 int e = match[index].rm_eo;
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	290 if (s != -1) {
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	291 if (debug_syslog > 3) {
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	292 my_syslog(buf); // show lines with matches
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	293 }
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	294 buf[e] = '\0';
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	295 int ip = ip_address(buf+s);
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	296 if (ip) {
51 206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	297 con.recorder->add(ip, amount, con, file_name, pattern_index, message);
36 6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	298 }
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	299 return true;
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	300 }
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	301 }
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	302 }
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	303 return false;
3 8fe310e5cd44 initial coding carl parents: 2 diff changeset	304 }
8fe310e5cd44 initial coding carl parents: 2 diff changeset	305
8fe310e5cd44 initial coding carl parents: 2 diff changeset	306
8fe310e5cd44 initial coding carl parents: 2 diff changeset	307 void PATTERN::dump(int level) {
36 6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	308 char indent[maxlen];
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	309 int i = min(maxlen-1, level*4);
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	310 memset(indent, ' ', i);
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	311 indent[i] = '\0';
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	312 printf("%s pattern \"%s\" {; \n", indent, pattern);
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	313 printf("%s index %d; \n", indent, index);
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	314 printf("%s bucket %d; \n", indent, amount);
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	315 if (message) printf("%s message \"%s\"; \n", indent, message);
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	316 printf("%s }; \n", indent);
3 8fe310e5cd44 initial coding carl parents: 2 diff changeset	317 }
8fe310e5cd44 initial coding carl parents: 2 diff changeset	318
8fe310e5cd44 initial coding carl parents: 2 diff changeset	319
8fe310e5cd44 initial coding carl parents: 2 diff changeset	320 ////////////////////////////////////////////////
8fe310e5cd44 initial coding carl parents: 2 diff changeset	321 //
51 206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	322 CONTEXT::CONTEXT(const char *nam) {
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	323 name = nam;
36 6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	324 threshold = 500;
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	325 add_command = "/sbin/iptables -I INPUT --src %s --jump DROP";
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	326 remove_command = "/sbin/iptables -D INPUT --src %s --jump DROP";
51 206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	327 recorder = IPR::find(name);
1 551433a01cab initial coding carl parents: diff changeset	328 }
551433a01cab initial coding carl parents: diff changeset	329
551433a01cab initial coding carl parents: diff changeset	330
51 206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	331 ////////////////////////////////////////////////
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	332 //
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	333 CONTEXT::~CONTEXT() {
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	334 ignore.clear();
36 6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	335 for (syslogconfig_list::iterator i=syslogconfigs.begin(); i!=syslogconfigs.end(); i++) {
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	336 SYSLOGCONFIG c = i;
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	337 delete c;
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	338 }
51 206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	339 IPR::release(name);
1 551433a01cab initial coding carl parents: diff changeset	340 }
551433a01cab initial coding carl parents: diff changeset	341
551433a01cab initial coding carl parents: diff changeset	342
51 206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	343 void CONTEXT::add_syslogconfig(SYSLOGCONFIGP con) {
36 6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	344 syslogconfigs.push_back(con);
1 551433a01cab initial coding carl parents: diff changeset	345 }
551433a01cab initial coding carl parents: diff changeset	346
551433a01cab initial coding carl parents: diff changeset	347
51 206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	348 void CONTEXT::add_pair(IPPAIR pair) {
36 6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	349 ignore.push_back(pair);
3 8fe310e5cd44 initial coding carl parents: 2 diff changeset	350 }
8fe310e5cd44 initial coding carl parents: 2 diff changeset	351
8fe310e5cd44 initial coding carl parents: 2 diff changeset	352
51 206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	353 void CONTEXT::dump() {
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	354 string indents(" ");
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	355 const char *indent = indents.c_str();
3 8fe310e5cd44 initial coding carl parents: 2 diff changeset	356
51 206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	357 printf("context %s {\n", name);
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	358 printf("%s threshold %d; \n\n", indent, threshold);
27 28fec0c67646 make add/remove commands configureable carl parents: 24 diff changeset	359
51 206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	360 printf("%s add_command \"%s\"; \n", indent, add_command);
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	361 printf("%s remove_command \"%s\"; \n\n", indent, remove_command);
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	362
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	363 printf("%s ignore { \n", indent);
36 6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	364 for (ippair_list::iterator i=ignore.begin(); i!=ignore.end(); i++) {
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	365 IPPAIR &p = *i;
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	366 in_addr ip;
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	367 ip.s_addr = htonl(p.first);
51 206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	368 printf("%s %s/%d; \n", indent, inet_ntoa(ip), p.cidr);
36 6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	369 }
51 206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	370 printf("%s }; \n\n", indent);
3 8fe310e5cd44 initial coding carl parents: 2 diff changeset	371
36 6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	372 for (syslogconfig_list::iterator i=syslogconfigs.begin(); i!=syslogconfigs.end(); i++) {
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	373 SYSLOGCONFIGP c = *i;
51 206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	374 c->dump(1);
36 6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	375 }
51 206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	376 printf("}; \n\n");
1 551433a01cab initial coding carl parents: diff changeset	377 }
551433a01cab initial coding carl parents: diff changeset	378
551433a01cab initial coding carl parents: diff changeset	379
51 206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	380 void CONTEXT::read(CONFIG &con) {
36 6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	381 while (true) {
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	382 bool have = false;
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	383 for (syslogconfig_list::iterator i=syslogconfigs.begin(); i!=syslogconfigs.end(); i++) {
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	384 SYSLOGCONFIGP c = *i;
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	385 have \|= c->read(*this);
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	386 }
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	387 if (!have) break;
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	388 }
2 6e88da080f08 initial coding carl parents: 1 diff changeset	389 }
6e88da080f08 initial coding carl parents: 1 diff changeset	390
6e88da080f08 initial coding carl parents: 1 diff changeset	391
51 206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	392 void CONTEXT::free_all() {
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	393 recorder->free_all(*this);
3 8fe310e5cd44 initial coding carl parents: 2 diff changeset	394 }
8fe310e5cd44 initial coding carl parents: 2 diff changeset	395
8fe310e5cd44 initial coding carl parents: 2 diff changeset	396
51 206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	397 void CONTEXT::leak(int delta) {
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	398 recorder->leak(delta, *this);
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	399
36 6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	400 }
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	401
51 206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	402
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	403 bool CONTEXT::looking(int ip) {
36 6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	404 for (ippair_list::iterator i=ignore.begin(); i!=ignore.end(); i++) {
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	405 IPPAIR &p = *i;
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	406 if ((p.first <= ip) && (ip <= p.last)) return false;
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	407 }
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	408 return true;
3 8fe310e5cd44 initial coding carl parents: 2 diff changeset	409 }
8fe310e5cd44 initial coding carl parents: 2 diff changeset	410
8fe310e5cd44 initial coding carl parents: 2 diff changeset	411 ////////////////////////////////////////////////
8fe310e5cd44 initial coding carl parents: 2 diff changeset	412 //
51 206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	413 CONFIG::CONFIG() {
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	414 reference_count = 0;
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	415 generation = 0;
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	416 load_time = 0;
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	417 }
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	418
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	419
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	420 CONFIG::~CONFIG() {
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	421 for (context_list::iterator i=contexts.begin(); i!=contexts.end(); i++) {
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	422 CONTEXT c = i;
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	423 delete c;
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	424 }
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	425 }
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	426
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	427
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	428 void CONFIG::dump() {
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	429 for (context_list::iterator i=contexts.begin(); i!=contexts.end(); i++) {
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	430 CONTEXTP c = *i;
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	431 c->dump();
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	432 }
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	433 }
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	434
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	435
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	436 void CONFIG::read() {
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	437 for (context_list::iterator i=contexts.begin(); i!=contexts.end(); i++) {
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	438 CONTEXT c = i;
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	439 c->read(*this);
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	440 }
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	441 }
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	442
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	443
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	444 void CONFIG::sleep(int duration, time_t &previous) {
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	445 ::sleep(duration);
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	446 time_t now = time(NULL);
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	447 for (context_list::iterator i=contexts.begin(); i!=contexts.end(); i++) {
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	448 CONTEXT c = i;
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	449 c->leak(now-previous);
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	450 }
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	451 previous = now;
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	452 }
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	453
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	454
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	455 void CONFIG::free_all() {
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	456 for (context_list::iterator i=contexts.begin(); i!=contexts.end(); i++) {
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	457 CONTEXT c = i;
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	458 c->free_all();
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	459 }
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	460 }
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	461
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	462
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	463 ////////////////////////////////////////////////
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	464 //
48 ba0259c9e411 Fixes to compile on Fedora 9 and for const correctness Carl Byington <carl@five-ten-sg.com> parents: 42 diff changeset	465 SYSLOGCONFIG::SYSLOGCONFIG(TOKEN &tok, const char *file_name_) {
36 6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	466 tokp = &tok;
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	467 file_name = file_name_;
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	468 open(true);
1 551433a01cab initial coding carl parents: diff changeset	469 }
551433a01cab initial coding carl parents: diff changeset	470
551433a01cab initial coding carl parents: diff changeset	471
551433a01cab initial coding carl parents: diff changeset	472 SYSLOGCONFIG::~SYSLOGCONFIG() {
36 6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	473 close();
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	474 for (pattern_list::iterator i=patterns.begin(); i!=patterns.end(); i++) {
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	475 PATTERN p = i;
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	476 delete p;
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	477 }
2 6e88da080f08 initial coding carl parents: 1 diff changeset	478 }
6e88da080f08 initial coding carl parents: 1 diff changeset	479
6e88da080f08 initial coding carl parents: 1 diff changeset	480
4 2737ab01659a initial coding carl parents: 3 diff changeset	481 void SYSLOGCONFIG::open(bool msg) {
36 6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	482 fd = ::open(file_name, O_RDONLY);
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	483 len = 0;
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	484 if (fd == -1) {
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	485 if (msg) {
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	486 char buf[maxlen];
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	487 snprintf(buf, sizeof(buf), "syslog file %s not readable", file_name);
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	488 tokp->token_error(buf);
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	489 }
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	490 }
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	491 else {
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	492 if (debug_syslog > 1) {
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	493 snprintf(buf, sizeof(buf), "syslog file %s opened", file_name);
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	494 my_syslog(buf);
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	495 }
48 ba0259c9e411 Fixes to compile on Fedora 9 and for const correctness Carl Byington <carl@five-ten-sg.com> parents: 42 diff changeset	496 if (msg) lseek(fd, 0, SEEK_END);
36 6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	497 if (fstat(fd, &openfdstat)) {
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	498 close();
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	499 snprintf(buf, sizeof(buf), "syslog file %s cannot stat after open", file_name);
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	500 tokp->token_error(buf);
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	501 }
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	502 // specify that this fd gets closed on exec, so that selinux
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	503 // won't complain about iptables trying to read log files.
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	504 int oldflags = fcntl(fd, F_GETFD, 0);
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	505 if (oldflags >= 0) {
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	506 fcntl(fd, F_SETFD, oldflags \| FD_CLOEXEC);
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	507 }
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	508 }
3 8fe310e5cd44 initial coding carl parents: 2 diff changeset	509 }
8fe310e5cd44 initial coding carl parents: 2 diff changeset	510
8fe310e5cd44 initial coding carl parents: 2 diff changeset	511
51 206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	512 bool SYSLOGCONFIG::read(CONTEXT &con) {
36 6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	513 if (failed()) {
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	514 open(false);
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	515 if (failed()) return false;
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	516 }
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	517 int n = ::read(fd, buf+len, buflen-len);
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	518 bool have = (n > 0);
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	519 if (have) {
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	520 len += n;
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	521 while (true) {
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	522 char p = (char)memchr(buf, '\n', len);
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	523 if (!p) break;
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	524 n = p-buf;
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	525 *p = '\0';
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	526 process(con); // process null terminated string
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	527 len -= n+1;
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	528 memmove(buf, p+1, len);
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	529 }
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	530 // no <lf> in a full buffer
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	531 if (len == buflen) len = 0;
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	532 }
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	533 else {
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	534 // check for file close
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	535 struct stat filenamest;
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	536 if (0 == stat(file_name, &filenamest)) {
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	537 if ((filenamest.st_dev != openfdstat.st_dev) \|\|
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	538 (filenamest.st_ino != openfdstat.st_ino)) {
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	539 close();
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	540 }
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	541 }
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	542 else {
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	543 // filename no longer exists
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	544 close();
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	545 }
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	546 }
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	547 return have;
2 6e88da080f08 initial coding carl parents: 1 diff changeset	548 }
6e88da080f08 initial coding carl parents: 1 diff changeset	549
6e88da080f08 initial coding carl parents: 1 diff changeset	550
4 2737ab01659a initial coding carl parents: 3 diff changeset	551 void SYSLOGCONFIG::close() {
36 6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	552 if (debug_syslog > 1) {
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	553 snprintf(buf, sizeof(buf), "syslog file %s closed", file_name);
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	554 my_syslog(buf);
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	555 }
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	556 if (fd != -1) ::close(fd);
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	557 fd = -1;
4 2737ab01659a initial coding carl parents: 3 diff changeset	558 }
2737ab01659a initial coding carl parents: 3 diff changeset	559
2737ab01659a initial coding carl parents: 3 diff changeset	560
2737ab01659a initial coding carl parents: 3 diff changeset	561 void SYSLOGCONFIG::add_pattern(PATTERNP pat) {
36 6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	562 patterns.push_back(pat);
4 2737ab01659a initial coding carl parents: 3 diff changeset	563 }
2737ab01659a initial coding carl parents: 3 diff changeset	564
2737ab01659a initial coding carl parents: 3 diff changeset	565
51 206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	566 void SYSLOGCONFIG::process(CONTEXT &con) {
36 6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	567 int pi=0;
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	568 for (pattern_list::iterator i=patterns.begin(); i!=patterns.end(); i++) {
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	569 PATTERN p = i;
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	570 if (p->process(buf, con, file_name, pi)) break;
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	571 pi++;
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	572 }
1 551433a01cab initial coding carl parents: diff changeset	573 }
551433a01cab initial coding carl parents: diff changeset	574
551433a01cab initial coding carl parents: diff changeset	575
551433a01cab initial coding carl parents: diff changeset	576 void SYSLOGCONFIG::dump(int level) {
36 6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	577 char indent[maxlen];
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	578 int i = min(maxlen-1, level*4);
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	579 memset(indent, ' ', i);
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	580 indent[i] = '\0';
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	581 printf("%s file \"%s\" {\n", indent, file_name);
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	582 for (pattern_list::iterator i=patterns.begin(); i!=patterns.end(); i++) {
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	583 PATTERN p = i;
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	584 p->dump(level+1);
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	585 }
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	586 printf("%s }; \n", indent);
1 551433a01cab initial coding carl parents: diff changeset	587 }
551433a01cab initial coding carl parents: diff changeset	588
551433a01cab initial coding carl parents: diff changeset	589
551433a01cab initial coding carl parents: diff changeset	590 ////////////////////////////////////////////////
551433a01cab initial coding carl parents: diff changeset	591 // helper to discard the strings held by a string_set
551433a01cab initial coding carl parents: diff changeset	592 //
551433a01cab initial coding carl parents: diff changeset	593 void discard(string_set &s) {
36 6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	594 for (string_set::iterator i=s.begin(); i!=s.end(); i++) {
48 ba0259c9e411 Fixes to compile on Fedora 9 and for const correctness Carl Byington <carl@five-ten-sg.com> parents: 42 diff changeset	595 free((void)i);
36 6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	596 }
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	597 s.clear();
1 551433a01cab initial coding carl parents: diff changeset	598 }
551433a01cab initial coding carl parents: diff changeset	599
551433a01cab initial coding carl parents: diff changeset	600
551433a01cab initial coding carl parents: diff changeset	601 ////////////////////////////////////////////////
551433a01cab initial coding carl parents: diff changeset	602 // helper to register a string in a string set
551433a01cab initial coding carl parents: diff changeset	603 //
48 ba0259c9e411 Fixes to compile on Fedora 9 and for const correctness Carl Byington <carl@five-ten-sg.com> parents: 42 diff changeset	604 const char* register_string(string_set &s, const char *name) {
ba0259c9e411 Fixes to compile on Fedora 9 and for const correctness Carl Byington <carl@five-ten-sg.com> parents: 42 diff changeset	605 string_set::const_iterator i = s.find(name);
36 6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	606 if (i != s.end()) return *i;
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	607 char *x = strdup(name);
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	608 s.insert(x);
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	609 return x;
1 551433a01cab initial coding carl parents: diff changeset	610 }
551433a01cab initial coding carl parents: diff changeset	611
551433a01cab initial coding carl parents: diff changeset	612
551433a01cab initial coding carl parents: diff changeset	613 ////////////////////////////////////////////////
551433a01cab initial coding carl parents: diff changeset	614 // register a global string
551433a01cab initial coding carl parents: diff changeset	615 //
48 ba0259c9e411 Fixes to compile on Fedora 9 and for const correctness Carl Byington <carl@five-ten-sg.com> parents: 42 diff changeset	616 const char* register_string(const char *name) {
36 6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	617 return register_string(all_strings, name);
1 551433a01cab initial coding carl parents: diff changeset	618 }
551433a01cab initial coding carl parents: diff changeset	619
551433a01cab initial coding carl parents: diff changeset	620
551433a01cab initial coding carl parents: diff changeset	621 ////////////////////////////////////////////////
38 26c29da3fbdf shutdown removes iptables entries that we added carl parents: 37 diff changeset	622 // clear all global strings, helper for valgrind checking
26c29da3fbdf shutdown removes iptables entries that we added carl parents: 37 diff changeset	623 //
26c29da3fbdf shutdown removes iptables entries that we added carl parents: 37 diff changeset	624 void clear_strings() {
26c29da3fbdf shutdown removes iptables entries that we added carl parents: 37 diff changeset	625 discard(all_strings);
26c29da3fbdf shutdown removes iptables entries that we added carl parents: 37 diff changeset	626 }
26c29da3fbdf shutdown removes iptables entries that we added carl parents: 37 diff changeset	627
26c29da3fbdf shutdown removes iptables entries that we added carl parents: 37 diff changeset	628
26c29da3fbdf shutdown removes iptables entries that we added carl parents: 37 diff changeset	629 ////////////////////////////////////////////////
1 551433a01cab initial coding carl parents: diff changeset	630 //
48 ba0259c9e411 Fixes to compile on Fedora 9 and for const correctness Carl Byington <carl@five-ten-sg.com> parents: 42 diff changeset	631 bool tsa(TOKEN &tok, const char *token);
ba0259c9e411 Fixes to compile on Fedora 9 and for const correctness Carl Byington <carl@five-ten-sg.com> parents: 42 diff changeset	632 bool tsa(TOKEN &tok, const char *token) {
ba0259c9e411 Fixes to compile on Fedora 9 and for const correctness Carl Byington <carl@five-ten-sg.com> parents: 42 diff changeset	633 const char *have = tok.next();
36 6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	634 if (have == token) return true;
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	635 tok.token_error(token, have);
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	636 return false;
1 551433a01cab initial coding carl parents: diff changeset	637 }
551433a01cab initial coding carl parents: diff changeset	638
551433a01cab initial coding carl parents: diff changeset	639
551433a01cab initial coding carl parents: diff changeset	640 ////////////////////////////////////////////////
551433a01cab initial coding carl parents: diff changeset	641 //
51 206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	642 bool parse_pattern(TOKEN &tok, SYSLOGCONFIG &con, CONTEXT &me);
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	643 bool parse_pattern(TOKEN &tok, SYSLOGCONFIG &con, CONTEXT &me) {
48 ba0259c9e411 Fixes to compile on Fedora 9 and for const correctness Carl Byington <carl@five-ten-sg.com> parents: 42 diff changeset	644 const char *pat = tok.next();
ba0259c9e411 Fixes to compile on Fedora 9 and for const correctness Carl Byington <carl@five-ten-sg.com> parents: 42 diff changeset	645 int ind = 0;
ba0259c9e411 Fixes to compile on Fedora 9 and for const correctness Carl Byington <carl@five-ten-sg.com> parents: 42 diff changeset	646 int buc = 0;
ba0259c9e411 Fixes to compile on Fedora 9 and for const correctness Carl Byington <carl@five-ten-sg.com> parents: 42 diff changeset	647 const char *msg = NULL;
36 6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	648 if (!tsa(tok, token_lbrace)) return false;
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	649 while (true) {
48 ba0259c9e411 Fixes to compile on Fedora 9 and for const correctness Carl Byington <carl@five-ten-sg.com> parents: 42 diff changeset	650 const char *have = tok.next();
36 6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	651 if (!have) break;
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	652 if (have == token_rbrace) break;
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	653 if (have == token_index) {
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	654 have = tok.next();
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	655 ind = atoi(have);
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	656 if (!tsa(tok, token_semi)) return false;
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	657 }
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	658 else if (have == token_bucket) {
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	659 have = tok.next();
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	660 buc = atoi(have);
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	661 if (!tsa(tok, token_semi)) return false;
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	662 }
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	663 else if (have == token_message) {
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	664 msg = tok.next();
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	665 if (!tsa(tok, token_semi)) return false;
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	666 }
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	667 else {
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	668 tok.token_error("index/bucket", have);
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	669 return false;
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	670 }
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	671 }
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	672 if (!tsa(tok, token_semi)) return false;
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	673 PATTERNP patt = new PATTERN(tok, pat, ind, buc, msg);
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	674 con.add_pattern(patt);
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	675 return true;
3 8fe310e5cd44 initial coding carl parents: 2 diff changeset	676 }
8fe310e5cd44 initial coding carl parents: 2 diff changeset	677
8fe310e5cd44 initial coding carl parents: 2 diff changeset	678
8fe310e5cd44 initial coding carl parents: 2 diff changeset	679 ////////////////////////////////////////////////
8fe310e5cd44 initial coding carl parents: 2 diff changeset	680 //
51 206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	681 bool parse_ignore(TOKEN &tok, CONFIG &dc, CONTEXT &me);
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	682 bool parse_ignore(TOKEN &tok, CONFIG &dc, CONTEXT &me) {
36 6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	683 if (!tsa(tok, token_lbrace)) return false;
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	684 while (true) {
48 ba0259c9e411 Fixes to compile on Fedora 9 and for const correctness Carl Byington <carl@five-ten-sg.com> parents: 42 diff changeset	685 const char *have = tok.next();
36 6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	686 if (!have) break;
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	687 if (have == token_rbrace) break;
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	688 int ipaddr = ip_address(have);
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	689 if (ipaddr == 0) {
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	690 tok.token_error("ip address", have);
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	691 return false;
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	692 }
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	693 if (!tsa(tok, token_slash)) return false;
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	694 have = tok.next();
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	695 int mask = atoi(have);
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	696 if ((mask < 8) \|\| (mask > 32)) {
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	697 tok.token_error("cidr 8..32 value", have);
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	698 return false;
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	699 }
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	700 if (!tsa(tok, token_semi)) return false;
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	701 IPPAIR pair;
59 f133196b8591 fix c++11 compiler warnings Carl Byington <carl@five-ten-sg.com> parents: 58 diff changeset	702 const unsigned int masks[33] = {0xffffffff, // 0
f133196b8591 fix c++11 compiler warnings Carl Byington <carl@five-ten-sg.com> parents: 58 diff changeset	703 0x7fffffff, // 1
f133196b8591 fix c++11 compiler warnings Carl Byington <carl@five-ten-sg.com> parents: 58 diff changeset	704 0x3fffffff, // 2
f133196b8591 fix c++11 compiler warnings Carl Byington <carl@five-ten-sg.com> parents: 58 diff changeset	705 0x1fffffff, // 3
f133196b8591 fix c++11 compiler warnings Carl Byington <carl@five-ten-sg.com> parents: 58 diff changeset	706 0x0fffffff, // 4
f133196b8591 fix c++11 compiler warnings Carl Byington <carl@five-ten-sg.com> parents: 58 diff changeset	707 0x07ffffff, // 5
f133196b8591 fix c++11 compiler warnings Carl Byington <carl@five-ten-sg.com> parents: 58 diff changeset	708 0x03ffffff, // 6
f133196b8591 fix c++11 compiler warnings Carl Byington <carl@five-ten-sg.com> parents: 58 diff changeset	709 0x01ffffff, // 7
f133196b8591 fix c++11 compiler warnings Carl Byington <carl@five-ten-sg.com> parents: 58 diff changeset	710 0x00ffffff, // 8
f133196b8591 fix c++11 compiler warnings Carl Byington <carl@five-ten-sg.com> parents: 58 diff changeset	711 0x007fffff, // 9
f133196b8591 fix c++11 compiler warnings Carl Byington <carl@five-ten-sg.com> parents: 58 diff changeset	712 0x003fffff, // 10
f133196b8591 fix c++11 compiler warnings Carl Byington <carl@five-ten-sg.com> parents: 58 diff changeset	713 0x001fffff, // 11
f133196b8591 fix c++11 compiler warnings Carl Byington <carl@five-ten-sg.com> parents: 58 diff changeset	714 0x000fffff, // 12
f133196b8591 fix c++11 compiler warnings Carl Byington <carl@five-ten-sg.com> parents: 58 diff changeset	715 0x0007ffff, // 13
f133196b8591 fix c++11 compiler warnings Carl Byington <carl@five-ten-sg.com> parents: 58 diff changeset	716 0x0003ffff, // 14
f133196b8591 fix c++11 compiler warnings Carl Byington <carl@five-ten-sg.com> parents: 58 diff changeset	717 0x0001ffff, // 15
f133196b8591 fix c++11 compiler warnings Carl Byington <carl@five-ten-sg.com> parents: 58 diff changeset	718 0x0000ffff, // 16
f133196b8591 fix c++11 compiler warnings Carl Byington <carl@five-ten-sg.com> parents: 58 diff changeset	719 0x00007fff, // 17
f133196b8591 fix c++11 compiler warnings Carl Byington <carl@five-ten-sg.com> parents: 58 diff changeset	720 0x00003fff, // 18
f133196b8591 fix c++11 compiler warnings Carl Byington <carl@five-ten-sg.com> parents: 58 diff changeset	721 0x00001fff, // 19
f133196b8591 fix c++11 compiler warnings Carl Byington <carl@five-ten-sg.com> parents: 58 diff changeset	722 0x00000fff, // 20
f133196b8591 fix c++11 compiler warnings Carl Byington <carl@five-ten-sg.com> parents: 58 diff changeset	723 0x000007ff, // 21
f133196b8591 fix c++11 compiler warnings Carl Byington <carl@five-ten-sg.com> parents: 58 diff changeset	724 0x000003ff, // 22
f133196b8591 fix c++11 compiler warnings Carl Byington <carl@five-ten-sg.com> parents: 58 diff changeset	725 0x000001ff, // 23
f133196b8591 fix c++11 compiler warnings Carl Byington <carl@five-ten-sg.com> parents: 58 diff changeset	726 0x000000ff, // 24
f133196b8591 fix c++11 compiler warnings Carl Byington <carl@five-ten-sg.com> parents: 58 diff changeset	727 0x0000007f, // 25
f133196b8591 fix c++11 compiler warnings Carl Byington <carl@five-ten-sg.com> parents: 58 diff changeset	728 0x0000003f, // 26
f133196b8591 fix c++11 compiler warnings Carl Byington <carl@five-ten-sg.com> parents: 58 diff changeset	729 0x0000001f, // 27
f133196b8591 fix c++11 compiler warnings Carl Byington <carl@five-ten-sg.com> parents: 58 diff changeset	730 0x0000000f, // 28
f133196b8591 fix c++11 compiler warnings Carl Byington <carl@five-ten-sg.com> parents: 58 diff changeset	731 0x00000007, // 29
f133196b8591 fix c++11 compiler warnings Carl Byington <carl@five-ten-sg.com> parents: 58 diff changeset	732 0x00000003, // 30
f133196b8591 fix c++11 compiler warnings Carl Byington <carl@five-ten-sg.com> parents: 58 diff changeset	733 0x00000001, // 31
f133196b8591 fix c++11 compiler warnings Carl Byington <carl@five-ten-sg.com> parents: 58 diff changeset	734 0x00000000}; // 32
36 6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	735 pair.first = ipaddr;
59 f133196b8591 fix c++11 compiler warnings Carl Byington <carl@five-ten-sg.com> parents: 58 diff changeset	736 pair.last = ipaddr \| (int)masks[mask];
36 6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	737 pair.cidr = mask;
51 206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	738 me.add_pair(pair);
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	739 }
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	740 if (!tsa(tok, token_semi)) return false;
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	741 return true;
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	742 }
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	743
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	744
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	745 ////////////////////////////////////////////////
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	746 //
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	747 bool parse_syslogconfig(TOKEN &tok, CONFIG &dc, CONTEXT &me);
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	748 bool parse_syslogconfig(TOKEN &tok, CONFIG &dc, CONTEXT &me) {
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	749 const char *name = tok.next();
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	750 if (!tsa(tok, token_lbrace)) return false;
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	751 SYSLOGCONFIGP con = new SYSLOGCONFIG(tok, name);
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	752 if (con->failed()) {
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	753 delete con;
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	754 return false;
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	755 }
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	756 me.add_syslogconfig(con);
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	757 while (true) {
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	758 const char *have = tok.next();
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	759 if (!have) break;
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	760 if (have == token_rbrace) break;
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	761 if (have == token_pattern) {
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	762 if (!parse_pattern(tok, *con, me)) return false;
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	763 }
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	764 else {
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	765 tok.token_error("pattern", have);
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	766 return false;
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	767 }
36 6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	768 }
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	769 if (!tsa(tok, token_semi)) return false;
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	770 return true;
3 8fe310e5cd44 initial coding carl parents: 2 diff changeset	771 }
8fe310e5cd44 initial coding carl parents: 2 diff changeset	772
8fe310e5cd44 initial coding carl parents: 2 diff changeset	773
8fe310e5cd44 initial coding carl parents: 2 diff changeset	774 ////////////////////////////////////////////////
8fe310e5cd44 initial coding carl parents: 2 diff changeset	775 //
51 206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	776 bool parse_context(TOKEN &tok, CONFIG &dc, CONTEXTP parent);
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	777 bool parse_context(TOKEN &tok, CONFIG &dc, CONTEXTP parent) {
48 ba0259c9e411 Fixes to compile on Fedora 9 and for const correctness Carl Byington <carl@five-ten-sg.com> parents: 42 diff changeset	778 const char *name = tok.next();
36 6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	779 if (!tsa(tok, token_lbrace)) return false;
51 206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	780 CONTEXTP con = new CONTEXT(name);
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	781
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	782 while (true) {
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	783 const char *have = tok.next();
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	784 if (!have) break;
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	785 if (have == token_rbrace) break; // done
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	786 if (have == token_threshold) {
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	787 have = tok.next();
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	788 con->set_threshold(atoi(have));
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	789 if (!tsa(tok, token_semi)) return false;
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	790 }
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	791 else if (have == token_ignore) {
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	792 if (!parse_ignore(tok, dc, *con)) return false;
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	793 }
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	794 else if (have == token_add) {
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	795 have = tok.next();
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	796 con->set_add(have);
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	797 if (!tsa(tok, token_semi)) return false;
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	798 }
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	799 else if (have == token_remove) {
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	800 have = tok.next();
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	801 con->set_remove(have);
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	802 if (!tsa(tok, token_semi)) return false;
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	803 }
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	804 else if (have == token_file) {
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	805 if (!parse_syslogconfig(tok, dc, *con)) return false;
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	806 }
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	807 else {
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	808 tok.token_error("threshold/ignore/add_command/remove_command/file", have);
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	809 return false;
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	810 }
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	811 }
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	812 if (!tsa(tok, token_semi)) {
36 6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	813 delete con;
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	814 return false;
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	815 }
51 206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	816 dc.add_context(con);
36 6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	817 return true;
1 551433a01cab initial coding carl parents: diff changeset	818 }
551433a01cab initial coding carl parents: diff changeset	819
551433a01cab initial coding carl parents: diff changeset	820
551433a01cab initial coding carl parents: diff changeset	821 ////////////////////////////////////////////////
551433a01cab initial coding carl parents: diff changeset	822 // parse a config file
551433a01cab initial coding carl parents: diff changeset	823 //
48 ba0259c9e411 Fixes to compile on Fedora 9 and for const correctness Carl Byington <carl@five-ten-sg.com> parents: 42 diff changeset	824 bool load_conf(CONFIG &dc, const char *fn) {
36 6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	825 int count = 0;
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	826 TOKEN tok(fn, &dc.config_files);
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	827 while (true) {
48 ba0259c9e411 Fixes to compile on Fedora 9 and for const correctness Carl Byington <carl@five-ten-sg.com> parents: 42 diff changeset	828 const char *have = tok.next();
36 6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	829 if (!have) break;
51 206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	830 if (have == token_context) {
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	831 if (!parse_context(tok, dc, NULL)) {
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	832 tok.token_error("load_conf() failed to parse context");
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	833 return false;
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	834 }
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	835 else count++;
36 6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	836 }
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	837 else {
51 206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	838 tok.token_error(token_context, have);
36 6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	839 return false;
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	840 }
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	841 }
51 206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	842 tok.token_error("load_conf() found %d contexts in %s", count, fn);
206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	843 return (!dc.contexts.empty());
1 551433a01cab initial coding carl parents: diff changeset	844 }
551433a01cab initial coding carl parents: diff changeset	845
551433a01cab initial coding carl parents: diff changeset	846
551433a01cab initial coding carl parents: diff changeset	847 ////////////////////////////////////////////////
551433a01cab initial coding carl parents: diff changeset	848 // init the tokens
551433a01cab initial coding carl parents: diff changeset	849 //
551433a01cab initial coding carl parents: diff changeset	850 void token_init() {
36 6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	851 token_add = register_string("add_command");
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	852 token_bucket = register_string("bucket");
51 206448c00b55 Allow multiple contexts with independent add/remove commands. Carl Byington <carl@five-ten-sg.com> parents: 48 diff changeset	853 token_context = register_string("context");
36 6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	854 token_file = register_string("file");
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	855 token_ignore = register_string("ignore");
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	856 token_include = register_string("include");
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	857 token_index = register_string("index");
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	858 token_lbrace = register_string("{");
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	859 token_message = register_string("message");
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	860 token_pattern = register_string("pattern");
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	861 token_rbrace = register_string("}");
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	862 token_remove = register_string("remove_command");
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	863 token_semi = register_string(";");
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	864 token_slash = register_string("/");
6a2f26976898 shutdown removes iptables entries that we added carl parents: 35 diff changeset	865 token_threshold = register_string("threshold");
1 551433a01cab initial coding carl parents: diff changeset	866 }

Mercurial > syslog2iptables

annotate src/syslogconfig.cpp @ 64:4b147494fc64