Project

General

Profile

Download (10.6 KB) Statistics
| Branch: | Tag: | Revision:
1
<?php
2
/*
3
	easyrule.inc.php
4

    
5
	Copyright (C) 2009-2010 Jim Pingle (jpingle@gmail.com)
6
	Originally Sponsored By Anathematic @ pfSense Forums
7
	All rights reserved.
8

    
9
	Redistribution and use in source and binary forms, with or without
10
	modification, are permitted provided that the following conditions are met:
11

    
12
	1. Redistributions of source code must retain the above copyright notice,
13
	this list of conditions and the following disclaimer.
14

    
15
	2. Redistributions in binary form must reproduce the above copyright
16
	notice, this list of conditions and the following disclaimer in the
17
	documentation and/or other materials provided with the distribution.
18

    
19
	THIS SOFTWARE IS PROVIDED ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES,
20
	INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY
21
	AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
22
	AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY,
23
	OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
24
	SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
25
	INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
26
	CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
27
	ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
28
	POSSIBILITY OF SUCH DAMAGE.
29
*/
30
/*
31
	pfSense_BUILDER_BINARIES:
32
	pfSense_MODULE:	filter
33
*/
34

    
35
$blockaliasname = 'EasyRuleBlockHosts';
36
$protocols_with_ports = array('tcp', 'udp');
37
require_once("functions.inc");
38
require_once("util.inc");
39
require_once("config.inc");
40

    
41
function easyrule_find_rule_interface($int) {
42
	global $config;
43
	/* Borrowed from firewall_rules.php */
44
	$iflist = get_configured_interface_with_descr(false, true);
45

    
46
	if ($config['pptpd']['mode'] == "server")
47
		$iflist['pptp'] = "PPTP VPN";
48

    
49
	if ($config['pppoe']['mode'] == "server")
50
		$iflist['pppoe'] = "PPPoE VPN";
51

    
52
	if ($config['l2tp']['mode'] == "server")
53
                $iflist['l2tp'] = "L2TP VPN";
54

    
55
	/* add ipsec interfaces */
56
	if (isset($config['ipsec']['enable']) || isset($config['ipsec']['client']['enable'])){
57
		$iflist["enc0"] = "IPSEC";
58
	}
59

    
60
	if (isset($iflist[$int]))
61
		return $int;
62

    
63
	foreach ($iflist as $if => $ifd) {
64
		if (strtolower($int) == strtolower($ifd))
65
			return $if;
66
	}
67

    
68
	if (substr($int, 0, 4) == "ovpn")
69
		return "openvpn";
70

    
71
	return false;
72
}
73

    
74
function easyrule_block_rule_exists($int = 'wan') {
75
	global $blockaliasname, $config;
76
	/* No rules, we we know it doesn't exist */
77
	if (!is_array($config['filter']['rule'])) {
78
		return false;
79
	}
80

    
81
	/* Search through the rules for one referencing our alias */
82
	foreach ($config['filter']['rule'] as $rule) {
83
		if (!is_array($rule) || !is_array($rule['source']))
84
			continue;
85
		if ($rule['source']['address'] == $blockaliasname . strtoupper($int) && ($rule['interface'] == $int))
86
			return true;
87
	}
88
	return false;
89
}
90

    
91
function easyrule_block_rule_create($int = 'wan') {
92
	global $blockaliasname, $config;
93
	/* If the alias doesn't exist, exit.
94
	 * Can't create an empty alias, and we don't know a host */
95
	if (easyrule_block_alias_getid($int) === false)
96
		return false;
97

    
98
	/* If the rule already exists, no need to do it again */
99
	if (easyrule_block_rule_exists($int))
100
		return true;
101

    
102
	/* No rules, start a new array */
103
	if (!is_array($config['filter']['rule'])) {
104
		$config['filter']['rule'] = array();
105
	}
106

    
107
	filter_rules_sort();
108
	$a_filter = &$config['filter']['rule'];
109

    
110
	/* Make up a new rule */
111
	$filterent = array();
112
	$filterent['type'] = 'block';
113
	$filterent['interface'] = $int;
114
	$filterent['source']['address'] = $blockaliasname . strtoupper($int);
115
	$filterent['destination']['any'] = '';
116
	$filterent['descr'] = gettext("Easy Rule: Blocked from Firewall Log View");
117

    
118
	array_splice($a_filter, 0, 0, array($filterent));
119

    
120
	return true;
121
}
122

    
123
function easyrule_block_alias_getid($int = 'wan') {
124
	global $blockaliasname, $config;
125
	if (!is_array($config['aliases']))
126
		return false;
127

    
128
	/* Hunt down an alias with the name we want, return its id */
129
	foreach ($config['aliases']['alias'] as $aliasid => $alias)
130
		if ($alias['name'] == $blockaliasname . strtoupper($int))
131
			return $aliasid;
132

    
133
	return false;
134
}
135

    
136
function easyrule_block_alias_add($host, $int = 'wan') {
137
	global $blockaliasname, $config;
138
	/* If the host isn't a valid IP address, bail */
139
	if (!is_ipaddr($host) && !is_subnet($host))
140
		return false;
141

    
142
	/* If there are no aliases, start an array */
143
	if (!is_array($config['aliases']['alias']))
144
		$config['aliases']['alias'] = array();
145

    
146
	$a_aliases = &$config['aliases']['alias'];
147

    
148
	/* Try to get the ID if the alias already exists */
149
	$id = easyrule_block_alias_getid($int);
150
	if ($id === false)
151
	  unset($id);
152

    
153
	$alias = array();
154

    
155
	if (is_subnet($host)) {
156
		list($host, $mask) = explode("/", $host);
157
	} elseif (is_specialnet($host)) {
158
		$mask = 0;
159
	} else {
160
		$mask = 32;
161
	}
162

    
163
	if (isset($id) && $a_aliases[$id]) {
164
		/* Make sure this IP isn't already in the list. */
165
		if (in_array($host.'/'.$mask, explode(" ", $a_aliases[$id]['address'])))
166
			return true;
167
		/* Since the alias already exists, just add to it. */
168
		$alias['name']    = $a_aliases[$id]['name'];
169
		$alias['type']    = $a_aliases[$id]['type'];
170
		$alias['descr']   = $a_aliases[$id]['descr'];
171

    
172
		$alias['address'] = $a_aliases[$id]['address'] . ' ' . $host . '/' . $mask;
173
		$alias['detail']  = $a_aliases[$id]['detail'] . gettext('Entry added') . ' ' . date('r') . '||';
174
	} else {
175
		/* Create a new alias with all the proper information */
176
	 	$alias['name']    = $blockaliasname . strtoupper($int);
177
	 	$alias['type']    = 'network';
178
		$alias['descr']   = gettext("Hosts blocked from Firewall Log view");
179

    
180
		$alias['address'] = $host . '/' . $mask;
181
		$alias['detail']  = gettext('Entry added') . ' ' . date('r') . '||';
182
	}
183

    
184
	/* Replace the old alias if needed, otherwise tack it on the end */
185
	if (isset($id) && $a_aliases[$id])
186
		$a_aliases[$id] = $alias;
187
	else
188
		$a_aliases[] = $alias;
189

    
190
	// Sort list
191
	$a_aliases = msort($a_aliases, "name");
192

    
193
	return true;
194
}
195

    
196
function easyrule_block_host_add($host, $int = 'wan') {
197
	global $retval;
198
	/* Bail if the supplied host is not a valid IP address */
199
	if (!is_ipaddr($host) && !is_subnet($host))
200
		return false;
201

    
202
	/* Flag whether or not we need to reload the filter */
203
	$dirty = false;
204

    
205
	/* Attempt to add this host to the alias */
206
	if (easyrule_block_alias_add($host, $int)) {
207
		$dirty = true;
208
	} else {
209
		/* Couldn't add the alias, or adding the host failed. */
210
		return false;
211
	}
212

    
213
	/* Attempt to add the firewall rule if it doesn't exist.
214
	 * Failing to add the rule isn't necessarily an error, it may
215
	 * have been modified by the user in some way. Adding to the
216
	 * Alias is what's important.
217
	 */
218
	if (!easyrule_block_rule_exists($int)) {
219
		if (easyrule_block_rule_create($int)) {
220
			$dirty = true;
221
		} else {
222
			return false;
223
		}
224
	}
225

    
226
	/* If needed, write the config and reload the filter */
227
	if ($dirty) {
228
		write_config();
229
		$retval = filter_configure();
230
		if (!empty($_SERVER['DOCUMENT_ROOT'])) {
231
			header("Location: firewall_aliases.php");
232
			exit;
233
		} else {
234
			return true;
235
		}
236
	} else {
237
		return false;
238
	}
239
}
240

    
241
function easyrule_pass_rule_add($int, $proto, $srchost, $dsthost, $dstport, $ipproto) {
242
	global $config;
243

    
244
	/* No rules, start a new array */
245
	if (!is_array($config['filter']['rule'])) {
246
		$config['filter']['rule'] = array();
247
	}
248

    
249
	filter_rules_sort();
250
	$a_filter = &$config['filter']['rule'];
251

    
252
	/* Make up a new rule */
253
	$filterent = array();
254
	$filterent['type'] = 'pass';
255
	$filterent['interface'] = $int;
256
	$filterent['ipprotocol'] = $ipproto;
257
	$filterent['descr'] = gettext("Easy Rule: Passed from Firewall Log View");
258

    
259
	if ($proto != "any")
260
		$filterent['protocol'] = $proto;
261
	else
262
		unset($filterent['protocol']);
263

    
264
	/* Default to only allow echo requests, since that's what most people want and
265
	 *  it should be a safe choice. */
266
	if ($proto == "icmp")
267
		$filterent['icmptype'] = 'echoreq';
268

    
269
	if (is_subnet($srchost)) {
270
		list($srchost, $srcmask) = explode("/", $srchost);
271
	} elseif (is_specialnet($srchost)) {
272
		$srcmask = 0;
273
	} else {
274
		$srcmask = 32;
275
	}
276

    
277
	if (is_subnet($dsthost)) {
278
		list($dsthost, $dstmask) = explode("/", $dsthost);
279
	} elseif (is_specialnet($dsthost)) {
280
		$dstmask = 0;
281
	} else {
282
		$dstmask = 32;
283
	}
284

    
285
	pconfig_to_address($filterent['source'], $srchost, $srcmask);
286
	pconfig_to_address($filterent['destination'], $dsthost, $dstmask, '', $dstport, $dstport);
287

    
288
	$a_filter[] = $filterent;
289

    
290
	write_config($filterent['descr']);
291
	$retval = filter_configure();
292
	if (!empty($_SERVER['DOCUMENT_ROOT'])) {
293
		header("Location: firewall_rules.php?if={$int}");
294
		exit;
295
	} else {
296
		return true;
297
	}
298
}
299

    
300
function easyrule_parse_block($int, $src) {
301
	if (!empty($src) && !empty($int)) {
302
		if (!is_ipaddr($src) && !is_subnet($src)) {
303
			return gettext("Tried to block invalid IP:") . ' ' . htmlspecialchars($src);
304
		}
305
		$int = easyrule_find_rule_interface($int);
306
		if ($int === false) {
307
			return gettext("Invalid interface for block rule:") . ' ' . htmlspecialchars($int);
308
		}
309
		if (easyrule_block_host_add($src, $int)) {
310
			return gettext("Host added successfully");
311
		} else {
312
			return gettext("Failed to create block rule, alias, or add host.");
313
		}
314
	} else {
315
		return gettext("Tried to block but had no host IP or interface");
316
	}
317
	return gettext("Unknown block error.");
318
}
319
function easyrule_parse_pass($int, $proto, $src, $dst, $dstport = 0, $ipproto = inet) {
320
	/* Check for valid int, srchost, dsthost, dstport, and proto */
321
	global $protocols_with_ports;
322

    
323
	if (!empty($int) && !empty($proto) && !empty($src) && !empty($dst)) {
324
		$int = easyrule_find_rule_interface($int);
325
		if ($int === false) {
326
			return gettext("Invalid interface for pass rule:") . ' ' . htmlspecialchars($int);
327
		}
328
		if (getprotobyname($proto) == -1) {
329
			return gettext("Invalid protocol for pass rule:") . ' ' . htmlspecialchars($proto);
330
		}
331
		if (!is_ipaddr($src) && !is_subnet($src) && !is_ipaddroralias($src) && !is_specialnet($src)) {
332
			return gettext("Tried to pass invalid source IP:") . ' ' . htmlspecialchars($src);
333
		}
334
		if (!is_ipaddr($dst) && !is_subnet($dst) && !is_ipaddroralias($dst) && !is_specialnet($dst)) {
335
			return gettext("Tried to pass invalid destination IP:") . ' ' . htmlspecialchars($dst);
336
		}
337
		if (in_array($proto, $protocols_with_ports)) {
338
			if (empty($dstport)) {
339
				return gettext("Missing destination port:") . ' ' . htmlspecialchars($dstport);
340
			}
341
			if (!is_port($dstport) && ($dstport != "any")) {
342
				return gettext("Tried to pass invalid destination port:") . ' ' . htmlspecialchars($dstport);
343
			}
344
		} else {
345
			$dstport = 0;
346
		}
347
		/* Should have valid input... */
348
		if (easyrule_pass_rule_add($int, $proto, $src, $dst, $dstport, $ipproto)) {
349
			return gettext("Successfully added pass rule!");
350
		} else {
351
			return gettext("Failed to add pass rule.");
352
		}
353
	} else {
354
		return gettext("Missing parameters for pass rule.");
355
	}
356
	return gettext("Unknown pass error.");
357
}
358

    
359
?>
(19-19/66)