pfSense

#!/usr/local/bin/php -f

<?php

echo "Starting the pfSense console firmware update system";

echo ".";

require("globals.inc");

$g['booting'] = true;

require("functions.inc");

echo ".";

require("config.inc");

echo ".";

$g['booting'] = false;

$fp = fopen('php://stdin', 'r');

echo ".\n\n";

$shell_active = true;

echo "1) Update from a URL\n";

echo "2) Update from a local file\n";

echo "Q) Quit\n";

echo "\nPlease select an option to continue: ";

$command = strtoupper(chop(fgets($fp)));

switch ($command) {

	case "q":

	case "quit":

		echo "\n";

		fclose($fp);

		die;

	break;

	case "1":

		echo "\nEnter the URL to the .tgz update file:\n> ";

		$url = chop(fgets($fp));

		if(!$url) { 

			fclose($fp);

			die;

		}

		$status = does_url_exist($url);

		if($status) {

			conf_mount_rw();

			echo "\nFetching file size...\n";

			$file_size = exec("fetch -s \"$url\"");

			$file_size = trim($file_size, "\r");

			echo "\nFile size: $file_size\n";

			echo "\nFetching file...\n";

			exec("fetch -1 -w15 -a -v -o /root/firmware.tgz \"$url\"");

			if($file_size <> filesize("/root/firmware.tgz")) {

				echo "\nFile size mismatch.  Upgrade cancelled.\n\n";

				fclose($fp);

				die;

			}			

			if(!file_exists("/root/firmware.tgz")) {

				echo "Something went wrong during file transfer.  Exiting.\n\n";

				fclose($fp);

				die;

			}

			$status = does_url_exist("$url.md5");

			if($status) { 

				echo "\nFetching MD5...\n";

				exec("fetch -1 -w15 -a -v -o /root/firmware.tgz.md5 \"$url.md5\"");

			} else {

				echo "\n\nWARNING.\n";

				echo "\nCould not locate a MD5 file.  We cannot verify the download once its done.\n\n";

				sleep(15);

			}

			if(file_exists("/root/firmware.tgz.md5")) {

				$source_md5 = trim(`cat /root/firmware.tgz.md5 | awk '{ print \$4 }'`,"\r");

				$file_md5 = trim(`md5 /root/firmware.tgz | awk '{ print \$4 }'`,"\r");

				echo "URL MD5: $source_md5\n";

				echo "Downloaded file MD5: $file_md5\n";

				if($source_md5 <> $file_md5) {

					echo "\n\nMD5 checksum does not match.  Cancelling upgrade.\n\n";

					fclose($fp);

					exec("rm -f /root/*.md5");

					die -1;

				}

				echo "\nMD5 checksum matches.\n";

				exec("rm -f /root/*.md5");

			}

			if(file_exists("/root/firmware.tgz"))

				do_upgrade("/root/firmware.tgz");

		} else {

			echo "\nCould not download update.\n\n";

			fclose($fp);

			die -1;

		}

	case "2":

		echo "\nEnter the complete path to the .tgz update file: ";

		$path = chop(fgets($fp));

		if(!$path) {

			fclose($fp);

			die;

		}

		if(file_exists($path)) {

			do_upgrade($path);

		} else {

			echo "\nCould not find file.\n\n";

			fclose($fp);

			die -1;

		}

}

function check_for_kernel_file() {

	global $fp;

	if(!file_exists("/boot/kernel/pfsense_kernel.txt")) {

		echo "\nPlease select which kernel you would like to use:\n";

		echo "\n1. Non SMP kernel";

		echo "\n2. SMP kernel";

		echo "\n3. Embedded kernel (no video or keyboard)";

		echo "\n4. Developers kernel (slower performing, more debugging)\n";

		echo "\nPlease enter a number [1-4]: ";

		$selection = strtoupper(chop(fgets($fp)));

		switch ($selection) {

			case "1":

				touch("/boot/kernel/pfsense_kernel.txt");

			break;

			case "2":

				exec("echo SMP > /boot/kernel/pfsense_kernel.txt");

			break;

			case "3":

				exec("echo wrap > /boot/kernel/pfsense_kernel.txt");

			break;

			case "4":

				exec("echo Developers > /boot/kernel/pfsense_kernel.txt");

			break;		

		}

		echo "\n";

	}

}

function do_upgrade($path) {

	global $fp;

	check_for_kernel_file();

	echo "\nOne moment please... Invoking firmware upgrade...\n";

	exec("/etc/rc.firmware pfSenseupgrade $path");

	unlink_if_exists($path);

	die;

}

exec("rm -f /root/*.md5");

fclose($fp);

?>