main.php | searchcode

/v1.9/main.php

https://bitbucket.org/rev22/timekoin
PHP | 292 lines | 202 code | 41 blank | 49 comment | 54 complexity | c4cbe5830ac772c5b05cc9b9f7ab66e4 MD5 | raw file

<?PHP
include 'configuration.php';
include 'function.php';

if($_GET["action"]=="begin_main")
{
	if(mysql_connect(MYSQL_IP,MYSQL_USERNAME,MYSQL_PASSWORD) == FALSE)
	{
		// Database connect error
		$datbase_error = TRUE;
	}

	if(mysql_select_db(MYSQL_DATABASE) == FALSE)
	{
		// Database select error
		$datbase_error = TRUE;
	}

	// Check last heartbeat and make sure it was more than X seconds ago
	$main_heartbeat_active = mysql_result(mysql_query("SELECT * FROM `main_loop_status` WHERE `field_name` = 'main_heartbeat_active' LIMIT 1"),0,"field_data");

	if($main_heartbeat_active == FALSE && $datbase_error == FALSE)
	{
		// Database Initialization
		initialization_database();

		mysql_query("UPDATE `main_loop_status` SET `field_data` = '" . time() . "' WHERE `main_loop_status`.`field_name` = 'main_last_heartbeat' LIMIT 1");

		// Set loop at active now
		mysql_query("UPDATE `main_loop_status` SET `field_data` = '1' WHERE `main_loop_status`.`field_name` = 'main_heartbeat_active' LIMIT 1");

		source_bg('main.php');

		activate(TIMEKOINSYSTEM, 1); // In case this was disabled from a emergency stop call in the server GUI
		header("Location: index.php?menu=system&code=1");
		exit;
	}
	else
	{
		header("Location: index.php?menu=system&code=99");
		exit;
	}
}

$mysql_link = mysql_connect(MYSQL_IP,MYSQL_USERNAME,MYSQL_PASSWORD);
mysql_select_db(MYSQL_DATABASE);
$context = stream_context_create(array('http' => array('header'=>'Connection: close'))); // Force close socket after complete
ini_set('user_agent', 'Timekoin Server (Main) v' . TIMEKOIN_VERSION);
ini_set('default_socket_timeout', 3); // Timeout for request in seconds

while(1) // Begin Infinite Loop
{
	// Are we to remain active?
	$loop_active = mysql_result(mysql_query("SELECT * FROM `main_loop_status` WHERE `field_name` = 'main_heartbeat_active' LIMIT 1"),0,"field_data");

	if($loop_active === FALSE) // Databaes Error
	{
		// Database Error, try to re-establish a connection after 5 seconds
		mysql_close($mysql_link);
		sleep(5);
		$mysql_link = mysql_connect(MYSQL_IP,MYSQL_USERNAME,MYSQL_PASSWORD);
		mysql_select_db(MYSQL_DATABASE);

		// Keep track of errors in case this can't be recovered from
		$datbase_error = TRUE;
		$database_error_counter++;
	}
	else
	{
		$datbase_error = 0;
		$database_error_counter = 0;
	}

	if($loop_active == 1)
	{
		// Main loop work goes below
		// Set the working status of 2
		mysql_query("UPDATE `main_loop_status` SET `field_data` = '2' WHERE `main_loop_status`.`field_name` = 'main_heartbeat_active' LIMIT 1");
	timekoin_debug_msg("main", "main loop start");
	//*****************************************************************************************************
	//*****************************************************************************************************	
	// Do a random time sync check and report any errors to the user
	if(rand(1,25) == 10)
	{
	  source_bg('timecheck.php');
	}
	//*****************************************************************************************************
	//*****************************************************************************************************
	// Check for spamming IPs
	        timekoin_debug_msg("main", "check for spamming IPs");
		$sql = "SELECT * FROM `ip_activity` GROUP BY `ip`";
		$sql_result = mysql_query($sql);
		$sql_num_results = mysql_num_rows($sql_result);

		$request_max = mysql_result(mysql_query("SELECT * FROM `options` WHERE `field_name` = 'server_request_max' LIMIT 1"),0,"field_data");

		for ($i = 0; $i < $sql_num_results; $i++)
		{
			$sql_row = mysql_fetch_array($sql_result);
			$select_IP = $sql_row["ip"];

			$sql = "SELECT * FROM `ip_activity` WHERE `ip` = '$select_IP'";
			$sql_num_results2 = mysql_num_rows(mysql_query($sql));

			if($sql_num_results2 > $request_max)
			{
				// More than X request per cycle means something is wrong
				// so this IP needs to be banned for a while
				mysql_query("INSERT INTO `ip_banlist` (`when` ,`ip`)VALUES (" . time() . ", '$select_IP')");
				write_log("IP Address $select_IP was added to the ban list due to excessive traffic. Default max query is $request_max per cycle, IP was doing $sql_num_results2 query per cycle instead.", "MA");
			}
		}

		// Clear IP Activity for next cycle
		mysql_query("TRUNCATE TABLE `ip_activity`");

		// Clear out ban list of IPs older than 1 day
		if(rand(1,60) == 30) // Randomize a little to save DB usage
		{
			mysql_query("DELETE FROM `ip_banlist` WHERE `ip_banlist`.`when` < " . (time() - 86400));
		}
	//*****************************************************************************************************
	//*****************************************************************************************************
		// Check to make sure we are not behind a firewall with no Inbound ports
	        timekoin_debug_msg("main", "check to make sure we are not behind a firewall with no Inbound port");
		if($sql_num_results == 0) // Randomize a little
		{
			if(rand(1,3) == 2)
			{
				// No activity from any peer, keep track of this
				$no_peer_activity = mysql_result(mysql_query("SELECT * FROM `options` WHERE `field_name` = 'no_peer_activity' LIMIT 1"),0,"field_data");
				$no_peer_activity++;
				
				if($no_peer_activity < 9)
				{
					mysql_query("UPDATE `options` SET `field_data` = '$no_peer_activity' WHERE `options`.`field_name` = 'no_peer_activity' LIMIT 1");
				}
			}
		}
		else
		{
			if(rand(1,3) == 2)
			{
				$no_peer_activity = mysql_result(mysql_query("SELECT * FROM `options` WHERE `field_name` = 'no_peer_activity' LIMIT 1"),0,"field_data");
				
				if($no_peer_activity > 1)
				{
					// Disable Firewalled Mode, Inbound is working again
					mysql_query("UPDATE `options` SET `field_data` = '0' WHERE `options`.`field_name` = 'firewall_blocked_peer' LIMIT 1");
					mysql_query("UPDATE `options` SET `field_data` = '0' WHERE `options`.`field_name` = 'no_peer_activity' LIMIT 1");
					$no_peer_activity = FALSE;
				}
			}
		}

		if($no_peer_activity > 5 && $no_peer_activity < 7)
		{
			// No Inbound connection working, the only way to submit transactions is out remotely.
			// Switch to firewalled mode.
			mysql_query("UPDATE `options` SET `field_data` = '1' WHERE `options`.`field_name` = 'firewall_blocked_peer' LIMIT 1");
		}
	//*****************************************************************************************************
	//*****************************************************************************************************
	        timekoin_debug_msg("main", "checking transaction clerk");
		$script_loop_active = mysql_result(mysql_query("SELECT * FROM `main_loop_status` WHERE `field_name` = 'transclerk_heartbeat_active' LIMIT 1"),0,"field_data");
		// Check if script is already running
		if($script_loop_active == 0)
		{
		  source_bg('transclerk.php');
		}

		sleep(1); // 1 second for sanity reasons

	        timekoin_debug_msg("main", "checking foundation manager");
		$script_loop_active = mysql_result(mysql_query("SELECT * FROM `main_loop_status` WHERE `field_name` = 'foundation_heartbeat_active' LIMIT 1"),0,"field_data");
		// Check if script is already running
		if($script_loop_active == 0)
		{
		  source_bg('foundation.php', TRUE);
		}

		sleep(1); // 1 second for sanity reasons

	        timekoin_debug_msg("main", "checking generation manager");
		$script_loop_active = mysql_result(mysql_query("SELECT * FROM `main_loop_status` WHERE `field_name` = 'generation_heartbeat_active' LIMIT 1"),0,"field_data");
		// Check if script is already running
		if($script_loop_active == 0)
		{
		  source_bg('generation.php');
		}

		sleep(1); // 1 second for sanity reasons

	        timekoin_debug_msg("main", "checking treasurer processor");
		$script_loop_active = mysql_result(mysql_query("SELECT * FROM `main_loop_status` WHERE `field_name` = 'treasurer_heartbeat_active' LIMIT 1"),0,"field_data");
		// Check if script is already running
		if($script_loop_active == 0)
		{
		  source_bg('treasurer.php');
		}

		sleep(2); // 1 second for sanity reasons

	        timekoin_debug_msg("main", "checking peer list processor");
		$script_loop_active = mysql_result(mysql_query("SELECT * FROM `main_loop_status` WHERE `field_name` = 'peerlist_heartbeat_active' LIMIT 1"),0,"field_data");
		// Check if script is already running
		if($script_loop_active == 0)
		{
		  source_bg('peerlist.php');
		}

		sleep(1); // 1 second for sanity reasons

	        timekoin_debug_msg("main", "checking queue clerk");
		$script_loop_active = mysql_result(mysql_query("SELECT * FROM `main_loop_status` WHERE `field_name` = 'queueclerk_heartbeat_active' LIMIT 1"),0,"field_data");
		// Check if script is already running
		if($script_loop_active == 0 || $script_loop_active == 1)
		{
		  source_bg('queueclerk.php');	
		}

		sleep(2); // 1 second for sanity reasons

	        timekoin_debug_msg("main", "checking genpeer manager");
		$script_loop_active = mysql_result(mysql_query("SELECT * FROM `main_loop_status` WHERE `field_name` = 'genpeer_heartbeat_active' LIMIT 1"),0,"field_data");
		// Check if script is already running
		if($script_loop_active == 0)
		{
		  source_bg('genpeer.php');	
		}

		sleep(1); // 1 second for sanity reasons

	        timekoin_debug_msg("main", "checking watchdog");
		if(rand(1,3) == 2) // Randomize checking to keep database load down
		{
			// Check watchdog script to make sure it is still running
			$script_loop_active = mysql_result(mysql_query("SELECT * FROM `main_loop_status` WHERE `field_name` = 'watchdog_heartbeat_active' LIMIT 1"),0,"field_data");
			$watchdog_last_heartbeat = mysql_result(mysql_query("SELECT * FROM `main_loop_status` WHERE `field_name` = 'watchdog_last_heartbeat' LIMIT 1"),0,"field_data");

			if($script_loop_active > 0)
			{
				// Watchdog should still be active
				if((time() - $watchdog_last_heartbeat) > 60) // Greater than double the loop time, something is wrong
				{
					// Watchdog stop was unexpected
					write_log("Watchdog is Stalled...", "MA");
				}
			}
		}
	//*****************************************************************************************************
	//*****************************************************************************************************	
		// (Very Last Thing to do in Script)
	        timekoin_debug_msg("main", "loop end");
		sleep(1);

		// Time to wake up and start again
		mysql_query("UPDATE `main_loop_status` SET `field_data` = '" . time() . "' WHERE `main_loop_status`.`field_name` = 'main_last_heartbeat' LIMIT 1");

		// Check loop status...
		$loop_active = mysql_result(mysql_query("SELECT * FROM `main_loop_status` WHERE `field_name` = 'main_heartbeat_active' LIMIT 1"),0,"field_data");

		if($loop_active == 3) // Do a final check to make sure we shouldn't stop running instead
		{
			// Stop the loop and reset status back to 0
			mysql_query("UPDATE `main_loop_status` SET `field_data` = '0' WHERE `main_loop_status`.`field_name` = 'main_heartbeat_active' LIMIT 1");
			exit;
		}
		else
		{
			mysql_query("UPDATE `main_loop_status` SET `field_data` = '1' WHERE `main_loop_status`.`field_name` = 'main_heartbeat_active' LIMIT 1");
		}
	} // Check if Active
	else
	{
		// Something is not working right, delay to avoid fast infinite loop
		if($datbase_error == TRUE && $database_error_counter < 6)
		{
		        timekoin_debug_msg("main", "database connection error");
			// Wait 2 seconds for database to come back online
			sleep(2);
		}
		else
		{
			// Script was called improperly from somewhere or while it was already running, exit to avoid loop stacking
			exit;
		}
	}
}
// End Infinite Loop

?>