<?php
/* $Id$ */
/*
 snort_rulesets.php
 Copyright (C) 2006 Scott Ullrich
 Copyright (C) 2009 Robert Zelaya
 All rights reserved.

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

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

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

 THIS SOFTWARE IS PROVIDED ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES,
 INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY
 AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
 AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY,
 OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
 SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
 INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
 CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
 ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
 POSSIBILITY OF SUCH DAMAGE.
 */

/* Setup enviroment */

/* TODO: review if include files are needed */
require_once("globals.inc");
require_once("functions.inc");
require_once("service-utils.inc");
require_once("/usr/local/pkg/snort/snort.inc");

//////////////////////////////

function read_header2($ch, $string) {
	global $file_size, $fout;
	$length = strlen($string);
	$regs = "";
	ereg("(Content-Length:) (.*)", $string, $regs);
	if($regs[2] <> "") {
		$file_size = intval($regs[2]);
	}
	ob_flush();
	return $length;
}

function read_body2($ch, $string) {
	global $fout, $file_size, $downloaded, $sendto, $static_status, $static_output, $lastseen, $pkg_interface;
	$length = strlen($string);
	$downloaded += intval($length);
	$downloadProgress = round(100 * (1 - $downloaded / $file_size), 0);
	$downloadProgress = 100 - $downloadProgress;
	if($lastseen <> $downloadProgress and $downloadProgress < 101) {
		if($sendto == "status") {
			$tostatus = $static_status . $downloadProgress . "%";
			echo "$tostatus";
		} else {
			$tooutput = $static_output . $downloadProgress . "%";
			echo "$tooutput";
		}
		echo "$downloadProgress";
		$lastseen = $downloadProgress;
	}
	if($fout)
	fwrite($fout, $string);
	ob_flush();
	return $length;
}

////////////////////////////////>>>>>>>>>>

function read_body_firmware($ch, $string) {
	global $fout, $file_size, $downloaded, $counter, $version, $latest_version, $current_installed_pfsense_version;
	$length = strlen($string);
	$downloaded += intval($length);
	$downloadProgress = round(100 * (1 - $downloaded / $file_size), 0);
	$downloadProgress = 100 - $downloadProgress;
	$a = $file_size;
	$b = $downloaded;
	$c = $downloadProgress;
	$text  = "  Snort download in progress\\n";
	$text .= "----------------------------------------------------\\n";
	$text .= "  Downloaded      : {$b}\\n";
	$text .= "----------------------------------------------------\\n";
	$counter++;
	if($counter > 150) {
		update_output_window($text);
		update_progress_bar($downloadProgress);
		flush();
		$counter = 0;
	}
	fwrite($fout, $string);
	return $length;
}

function download_file_with_progress_bar2($url_file, $destination_file, $readbody = 'read_body') {
	global $ch, $fout, $file_size, $downloaded;
	$file_size  = 1;
	$downloaded = 1;
	/* open destination file */
	$fout = fopen($destination_file, "wb");

	/*
	 *      Originally by Author: Keyvan Minoukadeh
	 *      Modified by Scott Ullrich to return Content-Length size
	 */

	$ch = curl_init();
	curl_setopt($ch, CURLOPT_URL, $url_file);
	curl_setopt($ch, CURLOPT_HEADERFUNCTION, 'read_header');
	curl_setopt($ch, CURLOPT_FOLLOWLOCATION, true);
	curl_setopt($ch, CURLOPT_SSL_VERIFYPEER, false);
	curl_setopt($ch, CURLOPT_WRITEFUNCTION, $readbody);
	curl_setopt($ch, CURLOPT_NOPROGRESS, '1');
	curl_setopt($ch, CURLOPT_CONNECTTIMEOUT, '5');
	curl_setopt($ch, CURLOPT_TIMEOUT, 0);

	curl_exec($ch);
	$http_code = curl_getinfo($ch, CURLINFO_HTTP_CODE);
	if($fout)
	fclose($fout);
	curl_close($ch);
	return ($http_code == 200) ? true : $http_code;
}

//////////////////////////////

$tmpfname = "/usr/local/etc/snort/tmp/snort_rules_up";
$snortdir = "/usr/local/etc/snort";
$snortdir_wan = "/usr/local/etc/snort";
$snort_filename_md5 = "snortrules-snapshot-2860.tar.gz.md5";
$snort_filename = "snortrules-snapshot-2860.tar.gz";
$emergingthreats_filename_md5 = "emerging.rules.tar.gz.md5";
$emergingthreats_filename = "emerging.rules.tar.gz";
$pfsense_rules_filename_md5 = "pfsense_rules.tar.gz.md5";
$pfsense_rules_filename = "pfsense_rules.tar.gz";

/* define checks */
$oinkid = $config['installedpackages']['snortglobal']['oinkmastercode'];
$snortdownload = $config['installedpackages']['snortglobal']['snortdownload'];
$emergingthreats = $config['installedpackages']['snortglobal']['emergingthreats'];

if ($oinkid == '' && $snortdownload == 'on')
{
	echo "You must obtain an oinkid from snort.org and set its value in the Snort settings tab.\n";
	exec("/usr/bin/logger -p daemon.info -i -t SnortStartup 'You must obtain an oinkid from snort.org and set its value in the Snort settings tab.'");
	exit;
}

if ($snortdownload != "on" && $emergingthreats != "on")
{
	echo 'Snort Global Settings: download snort.org rules = off and download emergingthreat rules = off.\n';
	exec("/usr/bin/logger -p daemon.info -i -t SnortStartup 'No rules have been selected to download.'");
	exit;
}

/*  Time stamps define */
$last_md5_download = $config['installedpackages']['snortglobal']['last_md5_download'];
$last_rules_install = $config['installedpackages']['snortglobal']['last_rules_install'];

$up_date_time = date('l jS \of F Y h:i:s A');
echo "\n";
echo "#########################\n";
echo "$up_date_time\n";
echo "#########################\n";
echo "\n\n";

exec("/usr/bin/logger -p daemon.info -i -t SnortStartup 'Checking for needed updates...'");

/* Begin main code */
conf_mount_rw();

if (!file_exists('/usr/local/etc/snort/tmp')) {
	exec('/bin/mkdir -p /usr/local/etc/snort/tmp');
}

/* Set user agent to Mozilla */
ini_set('user_agent','Mozilla/4.0 (compatible; MSIE 6.0)');
ini_set("memory_limit","150M");

/* mark the time update started */
$config['installedpackages']['snortglobal']['last_md5_download'] = date("Y-M-jS-h:i-A");

/* send current buffer */
ob_flush();
conf_mount_rw();

/* send current buffer */
ob_flush();
conf_mount_rw();

/*  remove old $tmpfname files */
if (file_exists("{$tmpfname}")) {
	echo 'Removing old tmp files...' . "\n";
	exec("/bin/rm -r {$tmpfname}");
	apc_clear_cache();
}

/*  Make shure snortdir exits */
exec("/bin/mkdir -p {$snortdir}");
exec("/bin/mkdir -p {$snortdir}/rules");
exec("/bin/mkdir -p {$snortdir}/signatures");
exec("/bin/mkdir -p /usr/local/lib/snort/dynamicrules/");

/* send current buffer */
ob_flush();
conf_mount_rw();

/* If tmp dir does not exist create it */
if (file_exists($tmpfname)) {
	echo 'The directory tmp exists...' . "\n";
} else {
	mkdir("{$tmpfname}", 700);
}

/*  download md5 sig from snort.org */
if ($snortdownload == 'on')
{
	if (file_exists("{$tmpfname}/{$snort_filename_md5}") &&
	filesize("{$tmpfname}/{$snort_filename_md5}") > 0) {
		echo 'snort.org md5 temp file exists...' . "\n";
	} else {
		echo 'Downloading snort.org md5 file...' . "\n";
		ini_set('user_agent','Mozilla/4.0 (compatible; MSIE 6.0)');
		$image = @file_get_contents("http://www.snort.org/pub-bin/oinkmaster.cgi/{$oinkid}/{$snort_filename_md5}");
		$f = fopen("{$tmpfname}/{$snort_filename_md5}", 'w');
		fwrite($f, $image);
		fclose($f);
		echo 'Done downloading snort.org md5' . "\n";
	}
}

/*  download md5 sig from emergingthreats.net */
if ($emergingthreats == "on")
{
	echo 'Downloading emergingthreats md5 file...' . "\n";
	ini_set('user_agent','Mozilla/4.0 (compatible; MSIE 6.0)');
	$image = @file_get_contents("http://rules.emergingthreats.net/open/snort-2.8.6/emerging.rules.tar.gz.md5");
	//    $image = @file_get_contents("http://www.mtest.local/pub-bin/oinkmaster.cgi/{$oinkid}/version.txt");
	$f = fopen("{$tmpfname}/{$emergingthreats_filename_md5}", 'w');
	fwrite($f, $image);
	fclose($f);
	echo 'Done downloading emergingthreats md5' . "\n";
}

/*  download md5 sig from pfsense.org */
if (file_exists("{$tmpfname}/{$pfsense_rules_filename_md5}")) {
	echo 'Pfsense md5 temp file exists...' . "\n";
} else {
	echo 'Downloading pfsense md5 file...' . "\n";
	ini_set('user_agent','Mozilla/4.0 (compatible; MSIE 6.0)');
	$image = @file_get_contents("http://www.pfsense.com/packages/config/snort/pfsense_rules/pfsense_rules.tar.gz.md5");
	//    $image = @file_get_contents("http://www.mtest.local/pub-bin/oinkmaster.cgi/{$oinkid}/pfsense_rules.tar.gz.md5");
	$f = fopen("{$tmpfname}/pfsense_rules.tar.gz.md5", 'w');
	fwrite($f, $image);
	fclose($f);
	echo 'Done downloading pfsense md5.' . "\n";
}

/* If md5 file is empty wait 15min exit */
if ($snortdownload == 'on')
{
	if (0 == filesize("{$tmpfname}/{$snort_filename_md5}"))
	{
		echo 'Please wait... You may only check for New Rules every 15 minutes...' . "\n";
		echo 'Rules are released every month from snort.org. You may download the Rules at any time.' . "\n";
		conf_mount_ro();
		exit(0);
	}
}

/* If emergingthreats md5 file is empty wait 15min exit not needed */

/* If pfsense md5 file is empty wait 15min exit */
if (0 == filesize("{$tmpfname}/$pfsense_rules_filename_md5")){
	echo 'Please wait... You may only check for New Pfsense Rules every 15 minutes...' . "\n";
	echo 'Rules are released to support Pfsense packages.' . "\n";
	conf_mount_ro();
	exit(0);
}

/* Check if were up to date snort.org */
if ($snortdownload == 'on')
{
	if (file_exists("{$snortdir}/{$snort_filename_md5}"))
	{
		$md5_check_new_parse = file_get_contents("{$tmpfname}/{$snort_filename_md5}");
		$md5_check_new = `/bin/echo "{$md5_check_new_parse}" | /usr/bin/awk '{ print $1 }'`;
		$md5_check_old_parse = file_get_contents("{$snortdir}/{$snort_filename_md5}");
		$md5_check_old = `/bin/echo "{$md5_check_old_parse}" | /usr/bin/awk '{ print $1 }'`;
		/* Write out time of last sucsessful md5 to cache */
		write_config(); // Will cause switch back to read-only on nanobsd
		conf_mount_rw();
		if ($md5_check_new == $md5_check_old)
		{
			echo 'Your rules are up to date...' . "\n";
			echo 'You may start Snort now, check update.' . "\n";
			$snort_md5_check_ok = on;
		}
	}
}

/* Check if were up to date emergingthreats.net */
if ($emergingthreats == "on")
{
	if (file_exists("{$snortdir}/{$emergingthreats_filename_md5}"))
	{
		$emerg_md5_check_new_parse = file_get_contents("{$tmpfname}/{$emergingthreats_filename_md5}");
		$emerg_md5_check_new = `/bin/echo "{$emerg_md5_check_new_parse}" | /usr/bin/awk '{ print $1 }'`;
		$emerg_md5_check_old_parse = file_get_contents("{$snortdir}/{$emergingthreats_filename_md5}");
		$emerg_md5_check_old = `/bin/echo "{$emerg_md5_check_old_parse}" | /usr/bin/awk '{ print $1 }'`;
		/* Write out time of last sucsessful md5 to cache */
		// Will cause switch back to read-only on nanobsd
		write_config();
		conf_mount_rw();
		if ($emerg_md5_check_new == $emerg_md5_check_old)
		{
			$emerg_md5_check_ok = on;
		}
	}
}

/* Check if were up to date pfsense.org */
if (file_exists("{$snortdir}/pfsense_rules.tar.gz.md5"))
{
	$pfsense_check_new_parse = file_get_contents("{$tmpfname}/pfsense_rules.tar.gz.md5");
	$pfsense_md5_check_new = `/bin/echo "{$pfsense_md5_check_new_parse}" | /usr/bin/awk '{ print $1 }'`;
	$pfsense_md5_check_old_parse = file_get_contents("{$snortdir}/pfsense_rules.tar.gz.md5");
	$pfsense_md5_check_old = `/bin/echo "{$pfsense_md5_check_old_parse}" | /usr/bin/awk '{ print $1 }'`;
	/* Write out time of last sucsessful md5 to cache */
	// Will cause switch back to read-only on nanobsd
	write_config();
	conf_mount_rw();
	if ($pfsense_md5_check_new == $pfsense_md5_check_old)
	{
		$pfsense_md5_check_ok = on;
	}
}

/* Check if were up to date is so, exit */
/* WARNING This code needs constant checks */
if ($snortdownload == 'on' && $emergingthreats == 'on')
{
	if ($snort_md5_check_ok == 'on' && $emerg_md5_check_ok == 'on')
	{
		echo 'All your rules are up to date...' . "\n";
		echo 'You may start Snort now...' . "\n";
		conf_mount_ro();
		exit(0);
	}
}

if ($snortdownload == 'on' && $emergingthreats == 'off')
{
	if ($snort_md5_check_ok == 'on')
	{
		echo 'Your snort.org rules are up to date...' . "\n";
		echo 'You may start Snort now...' . "\n";
		conf_mount_ro();
		exit(0);
	}
}

if ($snortdownload == 'off' && $emergingthreats == 'on')
{
	if ($emerg_md5_check_ok == 'on')
	{
		echo 'Your Emergingthreats rules are up to date...' . "\n";
		echo 'You may start Snort now...' . "\n";
		conf_mount_ro();
		exit(0);
	}
}

/* You are Not Up to date, always stop snort when updating rules for low end machines */;
echo 'You are NOT up to date...' . "\n";
echo 'Stopping Snort service...' . "\n";
$chk_if_snort_up = exec("pgrep -x snort");
if ($chk_if_snort_up != "") {
	exec("/usr/bin/touch /tmp/snort_download_halt.pid");
	exec("/bin/sh /usr/local/etc/rc.d/snort.sh stop");
	sleep(2);
}

/* download snortrules file */
if ($snortdownload == 'on')
{
	if ($snort_md5_check_ok != on) {
		if (file_exists("{$tmpfname}/{$snort_filename}")) {
			echo 'Snortrule tar file exists...' . "\n";
		} else {
			echo 'There is a new set of Snort.org rules posted. Downloading...' . "\n";
			echo 'May take 4 to 10 min...' . "\n";
			conf_mount_rw();
			download_file_with_progress_bar2("http://www.snort.org/pub-bin/oinkmaster.cgi/{$oinkid}/{$snort_filename}", $tmpfname . "/{$snort_filename}", "read_body_firmware");
			echo 'Done downloading rules file.' . "\n";
			if (150000 > filesize("{$tmpfname}/$snort_filename")){
				echo 'Error with the snort rules download...' . "\n";
				echo 'Snort rules file downloaded failed...' . "\n";
				conf_mount_ro();
				exit(0);
			}
		}
	}
}

/* download emergingthreats rules file */
if ($emergingthreats == "on")
{
	if ($emerg_md5_check_ok != on)
	{
		if (file_exists("{$tmpfname}/{$emergingthreats_filename}"))
		{
			echo 'Emergingthreats tar file exists...' . "\n";
		}else{
			echo 'There is a new set of Emergingthreats rules posted. Downloading...' . "\n";
			echo 'May take 4 to 10 min...' . "\n";
			download_file_with_progress_bar2("http://www.emergingthreats.net/rules/emerging.rules.tar.gz", $tmpfname . "/{$emergingthreats_filename}", "read_body_firmware");
			echo 'Done downloading Emergingthreats rules file.' . "\n";
		}
	}
}

/* download pfsense rules file */
if ($pfsense_md5_check_ok != on) {
	if (file_exists("{$tmpfname}/{$pfsense_rules_filename}")) {
		echo 'Snortrule tar file exists...' . "\n";
	} else {
		echo 'There is a new set of Pfsense rules posted. Downloading...' . "\n";
		echo 'May take 4 to 10 min...' . "\n";
		download_file_with_progress_bar2("http://www.pfsense.com/packages/config/snort/pfsense_rules/pfsense_rules.tar.gz", $tmpfname . "/{$pfsense_rules_filename}", "read_body_firmware");
		echo 'Done downloading rules file.' . "\n";
	}
}

/* Compair md5 sig to file sig */

//$premium_url_chk = $config['installedpackages']['snort']['config'][0]['subscriber'];
//if ($premium_url_chk == on) {
//$md5 = file_get_contents("{$tmpfname}/{$snort_filename_md5}");
//$file_md5_ondisk = `/sbin/md5 {$tmpfname}/{$snort_filename} | /usr/bin/awk '{ print $4 }'`;
// if ($md5 == $file_md5_ondisk) {
//    update_status(gettext("Valid md5 checksum pass..."));
//} else {
//    update_status(gettext("The downloaded file does not match the md5 file...P is ON"));
//    update_output_window(gettext("Error md5 Mismatch..."));
//    exit(0);
//  }
//}

//$premium_url_chk = $config['installedpackages']['snort']['config'][0]['subscriber'];
//if ($premium_url_chk != on) {
//$md55 = `/bin/cat {$tmpfname}/{$snort_filename_md5} | /usr/bin/awk '{ print $4 }'`;
//$file_md5_ondisk2 = `/sbin/md5 {$tmpfname}/{$snort_filename} | /usr/bin/awk '{ print $4 }'`;
// if ($md55 == $file_md5_ondisk2) {
//    update_status(gettext("Valid md5 checksum pass..."));
//} else {
//    update_status(gettext("The downloaded file does not match the md5 file...Not P"));
//    update_output_window(gettext("Error md5 Mismatch..."));
//    exit(0);
//    }
//}

/* Untar snort rules file individually to help people with low system specs */
if ($snortdownload == 'on')
{
	if ($snort_md5_check_ok != on) {
		if (file_exists("{$tmpfname}/{$snort_filename}")) {

			if ($pfsense_stable == 'yes')
			{
				$freebsd_version_so = 'FreeBSD-7-2';
			}else{
				$freebsd_version_so = 'FreeBSD-8-0';
			}

			echo 'Extracting Snort.org rules...' . "\n";
			echo 'May take a while...' . "\n";
			/* extract snort.org rules and  add prefix to all snort.org files*/
			exec("/bin/rm -r {$snortdir}/rules");
			sleep(2);
			exec("/usr/bin/tar xzf {$tmpfname}/{$snort_filename} -C {$snortdir} rules/");
			chdir ("/usr/local/etc/snort/rules");
			sleep(2);
			exec('/usr/local/bin/perl /usr/local/bin/snort_rename.pl s/^/snort_/ *.rules');
			/* extract so rules */
			exec("/usr/bin/tar xzf {$tmpfname}/{$snort_filename} -C {$snortdir} so_rules/precompiled/$freebsd_version_so/i386/2.8.6.0/");
			exec('/bin/mkdir -p /usr/local/lib/snort/dynamicrules/');
			exec("/bin/mv -f {$snortdir}/so_rules/precompiled/$freebsd_version_so/i386/2.8.6.0/* /usr/local/lib/snort/dynamicrules/");
			/* extract so rules none bin and rename */
			exec("/usr/bin/tar xzf {$tmpfname}/{$snort_filename} -C {$snortdir} so_rules/bad-traffic.rules/" .
			" so_rules/chat.rules/" .
			" so_rules/dos.rules/" .
			" so_rules/exploit.rules/" .
			" so_rules/icmp.rules/" .
			" so_rules/imap.rules/" .
			" so_rules/misc.rules/" .
			" so_rules/multimedia.rules/" .
			" so_rules/netbios.rules/" .
			" so_rules/nntp.rules/" .
			" so_rules/p2p.rules/" .
			" so_rules/smtp.rules/" .
			" so_rules/sql.rules/" .
			" so_rules/web-activex.rules/" .
			" so_rules/web-client.rules/" .
			" so_rules/web-iis.rules/" .
			" so_rules/web-misc.rules/");

			exec("/bin/mv -f {$snortdir}/so_rules/bad-traffic.rules {$snortdir}/rules/snort_bad-traffic.so.rules");
			exec("/bin/mv -f {$snortdir}/so_rules/chat.rules {$snortdir}/rules/snort_chat.so.rules");
			exec("/bin/mv -f {$snortdir}/so_rules/dos.rules {$snortdir}/rules/snort_dos.so.rules");
			exec("/bin/mv -f {$snortdir}/so_rules/exploit.rules {$snortdir}/rules/snort_exploit.so.rules");
			exec("/bin/mv -f {$snortdir}/so_rules/icmp.rules {$snortdir}/rules/snort_icmp.so.rules");
			exec("/bin/mv -f {$snortdir}/so_rules/imap.rules {$snortdir}/rules/snort_imap.so.rules");
			exec("/bin/mv -f {$snortdir}/so_rules/misc.rules {$snortdir}/rules/snort_misc.so.rules");
			exec("/bin/mv -f {$snortdir}/so_rules/multimedia.rules {$snortdir}/rules/snort_multimedia.so.rules");
			exec("/bin/mv -f {$snortdir}/so_rules/netbios.rules {$snortdir}/rules/snort_netbios.so.rules");
			exec("/bin/mv -f {$snortdir}/so_rules/nntp.rules {$snortdir}/rules/snort_nntp.so.rules");
			exec("/bin/mv -f {$snortdir}/so_rules/p2p.rules {$snortdir}/rules/snort_p2p.so.rules");
			exec("/bin/mv -f {$snortdir}/so_rules/smtp.rules {$snortdir}/rules/snort_smtp.so.rules");
			exec("/bin/mv -f {$snortdir}/so_rules/sql.rules {$snortdir}/rules/snort_sql.so.rules");
			exec("/bin/mv -f {$snortdir}/so_rules/web-activex.rules {$snortdir}/rules/snort_web-activex.so.rules");
			exec("/bin/mv -f {$snortdir}/so_rules/web-client.rules {$snortdir}/rules/snort_web-client.so.rules");
			exec("/bin/mv -f {$snortdir}/so_rules/web-iis.rules {$snortdir}/rules/snort_web-iis.so.rules");
			exec("/bin/mv -f {$snortdir}/so_rules/web-misc.rules {$snortdir}/rules/snort_web-misc.so.rules");
			exec("/bin/rm -r {$snortdir}/so_rules");

			/* extract base etc files */
			exec("/usr/bin/tar xzf {$tmpfname}/{$snort_filename} -C {$snortdir} etc/");
			exec("/bin/mv -f {$snortdir}/etc/* {$snortdir}");
			exec("/bin/rm -r {$snortdir}/etc");

			echo 'Done extracting Snort.org Rules.' . "\n";
		}else{
			echo 'Error extracting Snort.org Rules...' . "\n";
			echo 'Error Line 755' . "\n";
			conf_mount_ro();
			exit(0);
		}
	}
}

/* Untar emergingthreats rules to tmp */
if ($emergingthreats == "on")
{
	if ($emerg_md5_check_ok != on)
	{
		if (file_exists("{$tmpfname}/{$emergingthreats_filename}"))
		{
			echo 'Extracting rules...' . "\n";
			echo 'May take a while...' . "\n";
			exec("/usr/bin/tar xzf {$tmpfname}/{$emergingthreats_filename} -C {$snortdir} rules/");
		}
	}
}

/* Untar Pfsense rules to tmp */
if ($pfsense_md5_check_ok != on) {
	if (file_exists("{$tmpfname}/{$pfsense_rules_filename}")) {
		echo 'Extracting Pfsense rules...' . "\n";
		echo 'May take a while...' . "\n";
		exec("/usr/bin/tar xzf {$tmpfname}/{$pfsense_rules_filename} -C {$snortdir} rules/");
	}
}

/* Untar snort signatures */
if ($snort_md5_check_ok != on) {
	if (file_exists("{$tmpfname}/{$snort_filename}")) {
		$signature_info_chk = $config['installedpackages']['snortglobal']['signatureinfo'];
		if ($premium_url_chk == on) {
			echo 'Extracting Signatures...' . "\n";
			echo 'May take a while...' . "\n";
			exec("/usr/bin/tar xzf {$tmpfname}/{$snort_filename} -C {$snortdir} doc/signatures/");
			echo 'Done extracting Signatures.' . "\n";
		}
	}
}

/*  Copy md5 sig to snort dir */
if ($snortdownload == 'on')
{
	if ($snort_md5_check_ok != on) {
		if (file_exists("{$tmpfname}/$snort_filename_md5")) {
			echo 'Copying md5 sig to snort directory...' . "\n";
			exec("/bin/cp {$tmpfname}/$snort_filename_md5 {$snortdir}/$snort_filename_md5");
		}else{
			echo 'The md5 file does not exist...' . "\n";
			echo 'Error copying config...' . "\n";
			conf_mount_ro();
			exit(0);
		}
	}
}

/*  Copy emergingthreats md5 sig to snort dir */
if ($emergingthreats == "on")
{
	if ($emerg_md5_check_ok != on)
	{
		if (file_exists("{$tmpfname}/$emergingthreats_filename_md5"))
		{
			echo 'Copying md5 sig to snort directory...' . "\n";
			exec("/bin/cp {$tmpfname}/$emergingthreats_filename_md5 {$snortdir}/$emergingthreats_filename_md5");
		}else{
			echo 'The emergingthreats md5 file does not exist...' . "\n";
			echo 'Error copying config...' . "\n";
			conf_mount_ro();
			exit(0);
		}
	}
}

/*  Copy Pfsense md5 sig to snort dir */
if ($pfsense_md5_check_ok != on) {
	if (file_exists("{$tmpfname}/$pfsense_rules_filename_md5")) {
		echo 'Copying Pfsense md5 sig to snort directory...' . "\n";
		exec("/bin/cp {$tmpfname}/$pfsense_rules_filename_md5 {$snortdir}/$pfsense_rules_filename_md5");
	} else {
		update_status(gettext(""));
		echo 'The Pfsense md5 file does not exist...' . "\n";
		echo 'Error copying config...' . "\n";
		conf_mount_ro();
		exit(0);
	}
}

/*  Copy signatures dir to snort dir */
if ($snortdownload == 'on')
{
	if ($snort_md5_check_ok != on)
	{
		$signature_info_chk = $config['installedpackages']['snortglobal']['signatureinfo'];
		if ($premium_url_chk == on)
		{
			if (file_exists("{$snortdir}/doc/signatures")) {
				echo 'Copying signatures...' . "\n";
				echo 'May take a while...' . "\n";
				exec("/bin/mv -f {$snortdir}/doc/signatures {$snortdir}/signatures");
				exec("/bin/rm -r {$snortdir}/doc/signatures");
				echo 'Done copying signatures.' . "\n";
			}else{
				echo 'Directory signatures exist...' . "\n";
				echo 'Error copying signature...' . "\n";
				conf_mount_ro();
				exit(0);
			}
		}
	}
}

/* double make shure cleanup emerg rules that dont belong */
if (file_exists("/usr/local/etc/snort/rules/emerging-botcc-BLOCK.rules")) {
	apc_clear_cache();
	exec("/bin/rm /usr/local/etc/snort/rules/emerging-botcc-BLOCK.rules");
	exec("/bin/rm /usr/local/etc/snort/rules/emerging-botcc.rules");
	exec("/bin/rm /usr/local/etc/snort/rules/emerging-compromised-BLOCK.rules");
	exec("/bin/rm /usr/local/etc/snort/rules/emerging-drop-BLOCK.rules");
	exec("/bin/rm /usr/local/etc/snort/rules/emerging-dshield-BLOCK.rules");
	exec("/bin/rm /usr/local/etc/snort/rules/emerging-rbn-BLOCK.rules");
	exec("/bin/rm /usr/local/etc/snort/rules/emerging-tor-BLOCK.rules");
}

if (file_exists("/usr/local/lib/snort/dynamicrules/lib_sfdynamic_example_rule.so")) {
	exec("/bin/rm /usr/local/lib/snort/dynamicrules/lib_sfdynamic_example_rule.so");
	exec("/bin/rm /usr/local/lib/snort/dynamicrules/lib_sfdynamic_example\*");
}

/* make shure default rules are in the right format */
exec("/usr/local/bin/perl -pi -e 's/#alert/# alert/g' /usr/local/etc/snort/rules/*.rules");
exec("/usr/local/bin/perl -pi -e 's/##alert/# alert/g' /usr/local/etc/snort/rules/*.rules");
exec("/usr/local/bin/perl -pi -e 's/## alert/# alert/g' /usr/local/etc/snort/rules/*.rules");

/* create a msg-map for snort  */
echo 'Updating Alert Messages...' . "\n";
echo 'Please Wait...' . "\n";
exec("/usr/local/bin/perl /usr/local/bin/create-sidmap.pl /usr/local/etc/snort/rules > /usr/local/etc/snort/sid-msg.map");

//////////////////

/* open oinkmaster_conf for writing" function */
function oinkmaster_conf($id, $if_real, $iface_uuid)
{

	global $config, $g, $id, $if_real, $snortdir_wan, $snortdir, $snort_md5_check_ok, $emerg_md5_check_ok, $pfsense_md5_check_ok;
	conf_mount_rw();

	/*  enable disable setting will carry over with updates */
	/*  TODO carry signature changes with the updates */
	if ($snort_md5_check_ok != on || $emerg_md5_check_ok != on || $pfsense_md5_check_ok != on) {

		if (!empty($config['installedpackages']['snortglobal']['rule'][$id]['rule_sid_on'])) {
			$enabled_sid_on = $config['installedpackages']['snortglobal']['rule'][$id]['rule_sid_on'];
			$enabled_sid_on_array = split('\|\|', $enabled_sid_on);
			foreach($enabled_sid_on_array as $enabled_item_on)
			$selected_sid_on_sections .= "$enabled_item_on\n";
		}

		if (!empty($config['installedpackages']['snortglobal']['rule'][$id]['rule_sid_off'])) {
			$enabled_sid_off = $config['installedpackages']['snortglobal']['rule'][$id]['rule_sid_off'];
			$enabled_sid_off_array = split('\|\|', $enabled_sid_off);
			foreach($enabled_sid_off_array as $enabled_item_off)
			$selected_sid_off_sections .= "$enabled_item_off\n";
		}

		$snort_sid_text = <<<EOD

###########################################
#                                         #
# this is auto generated on snort updates #
#                                         #
###########################################

path = /bin:/usr/bin:/usr/local/bin

update_files = \.rules$|\.config$|\.conf$|\.txt$|\.map$

url = dir:///usr/local/etc/snort/rules

		$selected_sid_on_sections

		$selected_sid_off_sections

EOD;

		/* open snort's oinkmaster.conf for writing */
		$oinkmasterlist = fopen("/usr/local/etc/snort/snort_{$iface_uuid}_{$if_real}/oinkmaster_{$iface_uuid}_{$if_real}.conf", "w");

		fwrite($oinkmasterlist, "$snort_sid_text");

		/* close snort's oinkmaster.conf file */
		fclose($oinkmasterlist);

	}
}

/*  Run oinkmaster to snort_wan and cp configs */
/*  If oinkmaster is not needed cp rules normally */
/*  TODO add per interface settings here */
function oinkmaster_run($id, $if_real, $iface_uuid)
{

	global $config, $g, $id, $if_real, $snortdir_wan, $snortdir, $snort_md5_check_ok, $emerg_md5_check_ok, $pfsense_md5_check_ok;
	conf_mount_rw();

	if ($snort_md5_check_ok != on || $emerg_md5_check_ok != on || $pfsense_md5_check_ok != on)
	{

		if ($config['installedpackages']['snortglobal']['rule'][$id]['rule_sid_on'] == '' && $config['installedpackages']['snortglobal']['rule'][$id]['rule_sid_off'] == '')
		{
			echo 'Your first set of rules are being copied...' . "\n";
			echo 'May take a while...' . "\n";
			exec("/bin/cp {$snortdir}/rules/* {$snortdir_wan}/snort_{$iface_uuid}_{$if_real}/rules/");
			exec("/bin/cp {$snortdir}/classification.config {$snortdir_wan}/snort_{$iface_uuid}_{$if_real}");
			exec("/bin/cp {$snortdir}/gen-msg.map {$snortdir_wan}/snort_{$iface_uuid}_{$if_real}");
			exec("/bin/cp {$snortdir}/generators {$snortdir_wan}/snort_{$iface_uuid}_{$if_real}");
			exec("/bin/cp {$snortdir}/reference.config {$snortdir_wan}/snort_{$iface_uuid}_{$if_real}");
			exec("/bin/cp {$snortdir}/sid {$snortdir_wan}/snort_{$iface_uuid}_{$if_real}");
			exec("/bin/cp {$snortdir}/sid-msg.map {$snortdir_wan}/snort_{$iface_uuid}_{$if_real}");
			exec("/bin/cp {$snortdir}/unicode.map {$snortdir_wan}/snort_{$iface_uuid}_{$if_real}");
		}else{
			echo 'Your enable and disable changes are being applied to your fresh set of rules...' . "\n";
			echo 'May take a while...' . "\n";
			exec("/bin/cp {$snortdir}/rules/* {$snortdir_wan}/snort_{$iface_uuid}_{$if_real}/rules/");
			exec("/bin/cp {$snortdir}/classification.config {$snortdir_wan}/snort_{$iface_uuid}_{$if_real}");
			exec("/bin/cp {$snortdir}/gen-msg.map {$snortdir_wan}/snort_{$iface_uuid}_{$if_real}");
			exec("/bin/cp {$snortdir}/generators {$snortdir_wan}/snort_{$iface_uuid}_{$if_real}");
			exec("/bin/cp {$snortdir}/reference.config {$snortdir_wan}/snort_{$iface_uuid}_{$if_real}");
			exec("/bin/cp {$snortdir}/sid {$snortdir_wan}/snort_{$iface_uuid}_{$if_real}");
			exec("/bin/cp {$snortdir}/sid-msg.map {$snortdir_wan}/snort_{$iface_uuid}_{$if_real}");
			exec("/bin/cp {$snortdir}/unicode.map {$snortdir_wan}/snort_{$iface_uuid}_{$if_real}");

			/*  might have to add a sleep for 3sec for flash drives or old drives */
			exec("/usr/local/bin/perl /usr/local/bin/oinkmaster.pl -C /usr/local/etc/snort/snort_{$iface_uuid}_{$if_real}/oinkmaster_{$iface_uuid}_{$if_real}.conf -o /usr/local/etc/snort/snort_{$iface_uuid}_{$if_real}/rules > /usr/local/etc/snort/oinkmaster_{$iface_uuid}_{$if_real}.log");
		}
	}
}

/* Start the proccess for every interface rule */
/* TODO: try to make the code smother */

if (!empty($config['installedpackages']['snortglobal']['rule']))
{

	$rule_array = $config['installedpackages']['snortglobal']['rule'];
	$id = -1;
	foreach ($rule_array as $value) {

		$id += 1;

		$result_lan = $config['installedpackages']['snortglobal']['rule'][$id]['interface'];
		$if_real = convert_friendly_interface_to_real_interface_name2($result_lan);
		$iface_uuid = $config['installedpackages']['snortglobal']['rule'][$id]['uuid'];

		/* make oinkmaster.conf for each interface rule */
		oinkmaster_conf($id, $if_real, $iface_uuid);

		/* run oinkmaster for each interface rule */
		oinkmaster_run($id, $if_real, $iface_uuid);

	}
}

//////////////

/* mark the time update finnished */
$config['installedpackages']['snortglobal']['last_rules_install'] = date("Y-M-jS-h:i-A");

/*  remove old $tmpfname files */
if (file_exists('/usr/local/etc/snort/tmp'))
{
	echo 'Cleaning up...' . "\n";
	exec("/bin/rm -r /usr/local/etc/snort/tmp/snort_rules_up");
	sleep(2);
	exec("/bin/rm -r /usr/local/etc/snort/tmp/rules_bk");
	apc_clear_cache();
}

/* php code to flush out cache some people are reportting missing files this might help  */
sleep(2);
apc_clear_cache();
exec("/bin/sync ;/bin/sync ;/bin/sync ;/bin/sync ;/bin/sync ;/bin/sync ;/bin/sync ;/bin/sync");

/* make all dirs snorts */
exec("/usr/sbin/chown -R snort:snort /var/log/snort");
exec("/usr/sbin/chown -R snort:snort /usr/local/etc/snort");
exec("/usr/sbin/chown -R snort:snort /usr/local/lib/snort");
exec("/bin/chmod -R 755  /var/log/snort");
exec("/bin/chmod -R 755  /usr/local/etc/snort");
exec("/bin/chmod -R 755  /usr/local/lib/snort");


/* if snort is running hardrestart, if snort is not running do nothing */
if (file_exists("/tmp/snort_download_halt.pid")) {
	exec("/bin/sh /usr/local/etc/rc.d/snort.sh start");
	echo 'The Rules update finished...' . "\n";
	echo 'Snort has restarted with your new set of rules...' . "\n";
	exec("/bin/rm /tmp/snort_download_halt.pid");
} else {
	echo 'The Rules update finished...' . "\n";
	echo 'You may start snort now...' . "\n\n";
}

/* script finnished make system read only again */
conf_mount_ro();

?>