rc.update_urltables 1.54 KB
Newer Older
1
#!/usr/local/bin/php
Ad Schellevis's avatar
Ad Schellevis committed
2
<?php
3

Ad Schellevis's avatar
Ad Schellevis committed
4 5
require_once("config.inc");
require_once("util.inc");
6
require_once("pfsense-utils.inc");
Ad Schellevis's avatar
Ad Schellevis committed
7

8
if (!isset($config['aliases']['alias'])) {
Ad Schellevis's avatar
Ad Schellevis committed
9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56
	// No aliases
	exit;
}

// Gather list of urltable aliases
$todo = array();
foreach ($config['aliases']['alias'] as $alias) {
	if (preg_match('/urltable/i', $alias['type'])) {
		$tmp = array();
		$tmp['type'] = $alias['type'];
		$tmp['name'] = $alias['name'];
		$tmp['url']  = $alias['url'];
		$tmp['freq'] = $alias['updatefreq'];
		$todo[] = $tmp;
	}
}

if (count($todo) > 0) {
	log_error("{$argv[0]}: Starting up.");

	if ($argv[1] != "now") {
		// Wait a little before updating.
		$wait = mt_rand(5, 60);
		log_error("{$argv[0]}: Sleeping for {$wait} seconds.");
		sleep($wait);
	}

	log_error("{$argv[0]}: Starting URL table alias updates");

	$filter_reload = false;
	foreach ($todo as $t) {
		$r = process_alias_urltable($t['name'], $t['url'], $t['freq']);
		if ($r == 1) {
			$result = "";
			// TODO: Change it when pf supports tables with ports
			if ($t['type'] == "urltable")
				exec("/sbin/pfctl -t " . escapeshellarg($t['name']) . " -T replace -f /var/db/aliastables/" . escapeshellarg($t['name']) . ".txt 2>&1", $result);
			else
				$filter_reload = true;
			log_error("{$argv[0]}: Updated {$t['name']} content from {$t['url']}: {$result[0]}");
		} elseif ($r == -1) {
			log_error("{$argv[0]}: {$t['name']} does not need updating.");
		} else {
			log_error("{$argv[0]}: ERROR: could not update {$t['name']} content from {$t['url']}");
		}
	}

	if ($filter_reload)
57
		configd_run("filter reload");
Ad Schellevis's avatar
Ad Schellevis committed
58
}