<?php
require_once("config_defaults.inc");
require_once('config.inc');
require_once('plugins/mobile-detect/Mobile_Detect.php');

//Class Autoloader
spl_autoload_register(function ($className) {
    $className = strtolower($className);
    $path = __DIR__."/lib/class/{$className}.php";
    if (file_exists($path)) {
        require_once($path);
    } else {
        die("The file {$className}.php could not be found.");
    }
});

/* make sure we have enough memory, as some pages can be quite memory intensive */
ini_set("memory_limit", "256M");
/* note: you might want to put the data from asinfo.txt into an SQL
   database to avoid having to read the whole file all the time */
function getASInfo($asnum) {
	global $asinfodb;

	if (!isset($asinfodb))
		$asinfodb = readasinfodb();

	if (@$asinfodb[$asnum])
		return $asinfodb[$asnum];
	else
		return array('name' => "AS$asnum", 'descr' => "AS $asnum");
}

function readasinfodb() {
	global $asinfofile;

	if (!file_exists($asinfofile))
		return array();

	$fd = fopen($asinfofile, "r");
	$asinfodb = array();
	while (!feof($fd)) {
		$line = trim(fgets($fd));
		if (preg_match("/(^\\s*#)|(^\\s*$)/", $line))
			continue;	/* empty line or comment */

		$asnarr = explode("\t", $line);
		$asn = $asnarr[0];
		$asname = $asnarr[1];
		$descr = $asnarr[2];
		if (isset($asnarr[3])) $country = $asnarr[3];

		$asinfodb[$asn] = array(
			'name' => $asname,
			'descr' => $descr,
			'country' => $country
		);
	}
	fclose($fd);

	return $asinfodb;
}

function getknownlinks() {
	global $knownlinksfile;
	$fd = fopen($knownlinksfile, "r");
	$knownlinks = array();
	while (!feof($fd)) {
		$line = trim(fgets($fd));
		if (preg_match("/(^\\s*#)|(^\\s*$)/", $line))
			continue;	/* empty line or comment */

		list($routerip,$ifindex,$tag,$descr,$color) = preg_split("/\\t+/", $line);
		$known = false;
		foreach ($knownlinks as $link) {
		    if (in_array($tag,$link)) {$known=true;}
		}
		if (!$known) {
		    $knownlinks[] = array(
			'routerip' => $routerip,
			'ifindex' => $ifindex,
			'tag' => $tag,
			'descr' => $descr,
			'color' => $color
		    );
		}
	}
	fclose($fd);

	return $knownlinks;
}

function getasstats_top($ntop, $statfile, $selected_links, $list_asn = NULL, $v = NULL) {
	try{
		$db = new SQLite3($statfile);
	}catch(Exception $e){
		return array();
	}
	if(sizeof($selected_links) == 0){
		$selected_links = array();
		foreach(getknownlinks() as $link)
			$selected_links[] = $link['tag'];
	}
	$nlinks = 0;
	$query_total = '0';
	$query_links = '';
	foreach($selected_links as $tag){
		if ($v == 4 || $v == NULL) {
			$query_links .= "${tag}_in, ${tag}_out, ";
		} else {
	        	$query_links .=	"${tag}_v6_in, ${tag}_v6_out, ";
		}
		$nlinks += 4;
		if ($v == 4 || $v == NULL) {
			$query_total .= " + ${tag}_in + ${tag}_out";
		} else {
			$query_total .= " + ${tag}_v6_in + ${tag}_v6_out";
		}
	}
  if ( $list_asn ) {
    $where = implode(",", $list_asn);
    $query = "SELECT asn, $query_links $query_total as total FROM stats WHERE asn IN ( $where ) ORDER BY total desc limit $ntop";
  } else {
	  $query = "select asn, $query_links $query_total as total from stats order by total desc limit $ntop";
  }
	$asn = $db->query($query);
	$asstats = array();
	while($row = $asn->fetchArray()){
		$tot_in = 0;
		$tot_out = 0;
		$tot_v6_in = 0;
		$tot_v6_out = 0;
		foreach($row as $key => $value){
			if (strpos($key, '_in') !== false) {
				if (strpos($key, '_v6_') !== false)
					$tot_v6_in += $value;
				else
					$tot_in += $value;
			} else if (strpos($key, '_out') !== false) {
				if (strpos($key, '_v6_') !== false)
					$tot_v6_out += $value;
				else
					$tot_out += $value;
			}
		}
		$asstats[$row['asn']] = array($tot_in, $tot_out, $tot_v6_in, $tot_v6_out);
	}
	return $asstats;
}

function format_bytes($bytes) {
	if ($bytes >= 1099511627776)
		return sprintf("%.2f TB", $bytes / 1099511627776);
	else if ($bytes >= 1073741824)
		return sprintf("%.2f GB", $bytes / 1073741824);
	else if ($bytes >= 1048576)
		return sprintf("%.2f MB", $bytes / 1048576);
	else if ($bytes >= 1024)
		return sprintf("%d KB", $bytes / 1024);
	else
		return "$bytes bytes";
}

function getRRDFileForAS($as, $peer = 0) {
	global $rrdpath;
	$prefix = ($peer == 1) ? "$rrdpath/peeras" : "$rrdpath";
	return "$prefix/" . sprintf("%02x", $as % 256) . "/$as.rrd";
}

function getASSET($asset) {
	global $whois, $assetpath, $asset_cache_life;

	/* sanity check */
	if (!preg_match("/^[a-zA-Z0-9:_-]+$/", $asset)) return null;

	$assetfile = $assetpath."/".$asset.".txt";
	# check if file exist and cache
	$filemtime = @filemtime($assetfile);
	if (!$filemtime or (time() - $filemtime >= $asset_cache_life)) {
  	$cmd = $whois ." -h whois.radb.net '!i".$asset."'";
    $return_aslist = explode("\n",shell_exec($cmd));

		/* find the line that contains the AS-SET members */
		$aslist = array();
		foreach ($return_aslist as $asline) {
			if (preg_match("/^AS/", $asline)) {
				$aslist = explode(" ", $asline);
				break;
			}
		}

    $f = fopen($assetfile,"w");
    foreach ($aslist as $as) {
    	fputs($f,$as."\n");
    }
    fclose($f);
	# else read cache file
	} else {
		$f = fopen($assetfile, "r");
    	$aslist = array();
		while (!feof($f)) {
			$line = trim(fgets($f));
			if (!empty($line))
				$aslist[] = $line;
		}
	}
	return $aslist;
}

function clearCacheFileASSET($asset) {
	global $assetpath;

	/* sanity check */
	if (!preg_match("/^[a-zA-Z0-9:_-]+$/", $asset))
		return;
	if ( $asset == "all" ) {
		$files = glob($assetpath."/*.txt");
		foreach($files as $file) {
			unlink($file);
		}
	} else {
		$file = $assetpath."/".$asset.".txt";
		unlink($file);
	}
}

# return the html used in top.php : <a href=blabla><img src=blabla/></url>
function getHTMLUrl($as, $ipversion, $desc, $start, $end, $peerusage, $selected_links = array()){
	$img = getHTMLImg($as, $ipversion, $desc, $start, $end, $peerusage, '', '', false, $selected_links);
	$result = "<a href='history.php?as=$as&peerusage=$peerusage&v=$ipversion' target='_blank'>$img</a>";
	return($result);
}

# return the html used in history.php (for example) : <img src=blabla/>
function getHTMLImg($as, $ipversion, $desc, $start, $end, $peerusage, $alt, $class, $history = false, $selected_links=array()){
	global $top_graph_width;
	global $top_graph_height;
	$dname = rawurlencode("AS$as - $desc - IPV$ipversion");
	$result = "<img class='img-responsive' alt='$alt' class='$class' src='gengraph.php?v=$ipversion&as=$as&peerusage=$peerusage&dname=$dname&start=$start&end=$end";
	if(!$history)
		$result .= "&width=$top_graph_width&height=$top_graph_height&nolegend=1";
	if(sizeof($selected_links) != 0){
		$result .= "&selected_links=";
		foreach($selected_links as $link)
			$result .= "$link,";
		$result = rtrim($result, ',');
	}
	$result .= "'";
	if(!$history)
		$result .= " width='$top_graph_width' height='$top_graph_height' border='0'";
	$result .= "/>";
	return($result);
}

function statsFileForHours($hours) {
	global $top_intervals, $daystatsfile;
	foreach ($top_intervals as $interval) {
		if ($interval['hours'] == $hours && @$interval['statsfile']) {
			return $interval['statsfile'];
		}
	}
	return $daystatsfile;
}

function statsLabelForHours($hours) {
	global $top_intervals;
	foreach ($top_intervals as $interval) {
		if ($interval['hours'] == $hours && @$interval['label']) {
			return $interval['label'];
		}
	}
	return (int)$hours . " hours";
}

function debug($txt) {
  print "<pre>";
  print_r($txt);
  print "</pre>";
}

function menu($selected_links) {
	global $top_intervals;

	$dpagename = basename($_SERVER['PHP_SELF'], ".php");
	$active_top = $dpagename == "index" ? "active": "";
	$active_searchas = $dpagename == "history" ? 'class="active"' : "";
	$active_searchasset = $dpagename == "asset" ? 'class="active"' : "";
  $active_ix = $dpagename == "ix" ? 'class="active"' : "";
  $active_linkusage = $dpagename == "linkusage" ? 'class="active"' : "";

  $return = '<header class="main-header">';
  $return .= '<nav class="navbar navbar-static-top">';
  $return .= '<div class="container pull-left">';

  $return .='<div class="navbar-header">';
  $return .='<a href="index.php" class="navbar-brand"><b>AS-Stats</b></a>';
  $return .='<button type="button" class="navbar-toggle collapsed" data-toggle="collapse" data-target="#navbar-collapse">';
  $return .='<i class="fa fa-bars"></i>';
  $return .='</button>';
  $return .='</div>';

  $return .='<div class="collapse navbar-collapse pull-left" id="navbar-collapse">';
  $return .='<ul class="nav navbar-nav">';

  if ( count($top_intervals) < 2 ) {
		$return .= '<li class="'.$active_top.'"><a href="index.php">Top AS</a></li>';
	} else {
		$return .= '<li class="dropdown '.$active_top.'">';
		$return .= '<a href="#" class="dropdown-toggle" data-toggle="dropdown" aria-expanded="false">Top AS <span class="caret"></span></a>';
		$return .= '<ul class="dropdown-menu" role="menu">';
		foreach ($top_intervals as $interval) {
			$return .= '<li><a href="index.php?numhours=' . $interval['hours'] . '">Top AS - ' . $interval['label'] . '</a></li>';
		}
		$return .= '</ul>';
		$return .= '</li>';
	}

	$return .='<li '.$active_searchas.'><a href="history.php">View AS</a></li>';
	$return .='<li '.$active_searchasset.'><a href="asset.php">View AS-SET</a></li>';
  $return .='<li '.$active_ix.'><a href="ix.php">View IX Stats</a></li>';
  //$return .='<li '.$active_linkusage.'><a href="linkusage.php">Link Usage</a></li>';

  if ( count($top_intervals) < 2 ) {
		$return .= '<li '.$active_linkusage.'><a href="linkusage.php">Link Usage</a></li>';
	} else {
		$return .= '<li class="dropdown '.$active_linkusage.'">';
		$return .= '<a href="#" class="dropdown-toggle" data-toggle="dropdown" aria-expanded="false">Link Usage <span class="caret"></span></a>';
		$return .= '<ul class="dropdown-menu" role="menu">';
		foreach ($top_intervals as $interval) {
			$return .= '<li><a href="linkusage.php?numhours=' . $interval['hours'] . '">Link Usage - ' . $interval['label'] . '</a></li>';
		}
		$return .= '</ul>';
		$return .= '</li>';
	}

  $return .='</ul>';

	$val_ntop = isset($_GET['n']) ? htmlentities($_GET['n']) : "";

	if ($dpagename == "index" || $dpagename == "ix") {
		$return .='<form class="navbar-form navbar-left" role="search">';
		foreach($selected_links as $tag){
			$tag = "link_".$tag;
			$return .='<input type="hidden" name="'.$tag.'" value="on">';
		}

    if ( $dpagename == "ix" && isset($_GET['ix']) ) {
      $return .='<input type="hidden" name="ix" value="'.htmlspecialchars($_GET['ix']).'">';
    }

	  $return .='<div class="input-group">';
	  $return .='<input type="text" class="form-control menu-input" name="n" placeholder="Top AS" value="'.$val_ntop.'">';
		$return .='<span class="input-group-btn">';
	  $return .='<button type="submit" class="btn btn-flat button-input"><i class="fa fa-search"></i></button>';
	  $return .='</span>';
	  $return .='</div>';
	  $return .='</form>';
	}

  $return .='</div>';
  $return .='</div>';
  $return .='</nav>';
  $return .='</header>';

  return $return;
}

function footer() {
  $footer = '<footer class="main-footer font-default">';
  $footer .= '<div class="pull-right hidden-xs">';
  $footer .= '<b>GUI Version</b> 0.2';
  $footer .= '</div>';
  $footer .= '<strong>AS-Stats v1.6</strong> written by Manuel Kasper for Monzoon Networks AG.';
	$footer .= '&nbsp;GUI written by Nicolas Debrigode for Hexanet SAS.';

  $footer .= '</footer>';

  return $footer;
}

function content_header($titre, $small) {
	global $outispositive;

  $header = '<section class="content-header">';
  $header .= '<h1>';
  $header .= $titre;
  $header .= '<small><i>'. $small .'</i></small>';
  $header .= '</h1>';

	if ($outispositive) {
		$txt_traffic = '<i class="fa fa-arrow-up"></i> Outbound traffic <i class="fa fa-arrow-down"></i> Inbound traffic';
	} else {
		$txt_traffic = '<i class="fa fa-arrow-up"></i> Inbound traffic <i class="fa fa-arrow-down"></i> Outbound traffic';
	}

	$header .= '<ol class="breadcrumb">';
	$header .= '<li> '. $txt_traffic .' </li>';
	$header .= '</ol>';

  $header .= '</section>';

	$header .= '<hr>';

  return $header;
}

function box_linkusage ($title, $img) {
	$return = '<div class="box box-primary">';
	$return .= '<div class="box-header with-border">';
	$return .= '<i class="fa fa-bar-chart-o"></i>';
	$return .= '<h3 class="box-title">'.$title.'</h3>';
	$return .= '</div>';
	$return .= '<div class="box-body">';
	$return .= '<center>'.$img.'</center>';
	$return .= '</div>';
	$return .= '</div>';
	return $return;
}
?>