Project

General

Profile

Download (11 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) && ($rule['ipprotocol'] == $ipproto))
86
			return true;
87
	}
88
	return false;
89
}
90

    
91
function easyrule_block_rule_create($int = 'wan', $ipproto = "inet") {
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, $ipproto))
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['ipprotocol'] = $ipproto;
115
	$filterent['source']['address'] = $blockaliasname . strtoupper($int);
116
	$filterent['destination']['any'] = '';
117
	$filterent['descr'] = gettext("Easy Rule: Blocked from Firewall Log View");
118

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

    
121
	return true;
122
}
123

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

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

    
134
	return false;
135
}
136

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

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

    
148
	$a_aliases = &$config['aliases']['alias'];
149

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

    
155
	$alias = array();
156

    
157
	if (is_subnet($host)) {
158
		list($host, $mask) = explode("/", $host);
159
	} elseif (is_specialnet($host)) {
160
		$mask = 0;
161
	} elseif (is_ipaddrv6($host)) {
162
		$mask = 128;
163
	} else {
164
		$mask = 32;
165
	}
166

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

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

    
184
		$alias['address'] = $host . '/' . $mask;
185
		$alias['detail']  = gettext('Entry added') . ' ' . date('r') . '||';
186
	}
187

    
188
	/* Replace the old alias if needed, otherwise tack it on the end */
189
	if (isset($id) && $a_aliases[$id])
190
		$a_aliases[$id] = $alias;
191
	else
192
		$a_aliases[] = $alias;
193

    
194
	// Sort list
195
	$a_aliases = msort($a_aliases, "name");
196

    
197
	return true;
198
}
199

    
200
function easyrule_block_host_add($host, $int = 'wan', $ipproto = "inet") {
201
	global $retval;
202
	/* Bail if the supplied host is not a valid IP address */
203
	$host = trim($host, "[]");
204
	if (!is_ipaddr($host) && !is_subnet($host))
205
		return false;
206

    
207
	/* Flag whether or not we need to reload the filter */
208
	$dirty = false;
209

    
210
	/* Attempt to add this host to the alias */
211
	if (easyrule_block_alias_add($host, $int)) {
212
		$dirty = true;
213
	} else {
214
		/* Couldn't add the alias, or adding the host failed. */
215
		return false;
216
	}
217

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

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

    
246
function easyrule_pass_rule_add($int, $proto, $srchost, $dsthost, $dstport, $ipproto) {
247
	global $config;
248

    
249
	/* No rules, start a new array */
250
	if (!is_array($config['filter']['rule'])) {
251
		$config['filter']['rule'] = array();
252
	}
253

    
254
	filter_rules_sort();
255
	$a_filter = &$config['filter']['rule'];
256

    
257
	/* Make up a new rule */
258
	$filterent = array();
259
	$filterent['type'] = 'pass';
260
	$filterent['interface'] = $int;
261
	$filterent['ipprotocol'] = $ipproto;
262
	$filterent['descr'] = gettext("Easy Rule: Passed from Firewall Log View");
263

    
264
	if ($proto != "any")
265
		$filterent['protocol'] = $proto;
266
	else
267
		unset($filterent['protocol']);
268

    
269
	/* Default to only allow echo requests, since that's what most people want and
270
	 *  it should be a safe choice. */
271
	if ($proto == "icmp")
272
		$filterent['icmptype'] = 'echoreq';
273

    
274
	if (is_subnet($srchost)) {
275
		list($srchost, $srcmask) = explode("/", $srchost);
276
	} elseif (is_specialnet($srchost)) {
277
		$srcmask = 0;
278
	} else {
279
		$srcmask = 32;
280
	}
281

    
282
	if (is_subnet($dsthost)) {
283
		list($dsthost, $dstmask) = explode("/", $dsthost);
284
	} elseif (is_specialnet($dsthost)) {
285
		$dstmask = 0;
286
	} else {
287
		$dstmask = 32;
288
	}
289

    
290
	pconfig_to_address($filterent['source'], $srchost, $srcmask);
291
	pconfig_to_address($filterent['destination'], $dsthost, $dstmask, '', $dstport, $dstport);
292

    
293
	$a_filter[] = $filterent;
294

    
295
	write_config($filterent['descr']);
296
	$retval = filter_configure();
297
	if (!empty($_SERVER['DOCUMENT_ROOT'])) {
298
		header("Location: firewall_rules.php?if={$int}");
299
		exit;
300
	} else {
301
		return true;
302
	}
303
}
304

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

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

    
367
?>
(18-18/66)