<?php

if (!defined('IN_TRAFSTAT')) {
	die("Hacking attempt!");
}

/* Определим флаг сортировки. */
$avlSortCmp = array('port', 'allOut', 'allIn');
if (isset($_GET['sort']))
	$temp = in_array($_GET['sort'], $avlSortCmp) ? $_GET['sort'] : 'allIn';
else
	$temp = 'allIn';
define('SORT_CMP', $temp);

/* Определим запрашиваемый месяц */
define('PERIOD', get_period());
/* Определим запрашиваемый день месяца */
define('PERIOD_DAY', get_period_day());

/* Сформируем даты, между которыми будет формироваться статистика. */
define('BEGIN_DATE', get_begin_date(PERIOD, PERIOD_DAY));
define('END_DATE', get_end_date(BEGIN_DATE, PERIOD_DAY));

/* Определим пользователя по которому запрашивается статистика. */
if (isset($_GET['user'])) {
	$temp = mysql_real_escape_string($_GET['user']);
	define('USER', $temp);
	define('USER_NAME', long2ip(USER));
} else {
	message_die("Undefined variable 'user'.");
}

/* определим удаленный сервер по которому запрашивается статистика */
if (isset($_GET['server'])) {
	$temp = mysql_real_escape_string($_GET['server']);
	define('SERVER', $temp);
	define('SERVER_NAME', long2ip(SERVER));
} else {
	message_die("Undefined variable 'server'.");
}

/* количество запросов к базе */
$queryCount = 0;

function my_mysql_query($query)
{
	global $queryCount;

	$queryCount++;
	$result = mysql_query($query) or
		message_die("Query failed: " . mysql_error());
	return $result;
}

/* Получим статистику по портам сервера, к которому обращался клиент. */
$dataTraf = array();
$patternsQueryTraf = array('/%USER%/', '/%SERVER%/',
	'/%BEGIN_DATE%/', '/%END_DATE%/');

/*
 * Запрос устанавливает переменную '@proto' используемую в дальнейших запросах
 * для идентификации протокола, по которому идет запрос.
 * Внимание, после вызова команды musql_close, данная переменная
 * больше не доступна.
 */
$querySetProto = 'select @proto := id from proto where name = "%PROTO%"';

/*
 * Запрос возвращает список портов сервера к которому обращался клиент и
 * количество трафика по этому порту за заданный период.
 */
$queryTraf =
'select
	out.dstPort as port,
	out.trafOut as trafOut,
	in.trafIn as trafIn
from
	(
		select dstPort, sum(psize) as trafOut
		from traffic
		where (pId = @proto) && (srcPort = 0) &&
			(srcIp = %USER%) && (dstIp = %SERVER%) &&
			(date >= "%BEGIN_DATE%" && date < "%END_DATE%")
		group by dstPort
	) as `out` join
	(
		select srcPort, sum(psize) as trafIn
		from traffic
		where (pId = @proto) && (dstPort = 0) &&
			(dstIp = %USER%) && (srcIp = %SERVER%) &&
			(date >= "%BEGIN_DATE%" && date < "%END_DATE%")
		group by srcPort
	) as `in` on out.dstPort = in.srcPort

union all

select
	out.dstPort as port,
	out.trafOut as trafOut,
	ifnull(in.trafIn,0) as trafIn
from
	(
		select dstPort, sum(psize) as trafOut
		from traffic
		where (pId = @proto) && (srcPort = 0) &&
			(srcIp = %USER%) && (dstIp = %SERVER%) &&
			(date >= "%BEGIN_DATE%" && date < "%END_DATE%")
		group by dstPort
	) as `out` left outer join
	(
		select srcPort, sum(psize) as trafIn
		from traffic
		where (pId = @proto) && (dstPort = 0) &&
			(dstIp = %USER%) && (srcIp = %SERVER%) &&
			(date >= "%BEGIN_DATE%" && date < "%END_DATE%")
		group by srcPort
	) as `in` on out.dstPort = in.srcPort
where in.srcPort is null

union all

select
	in.srcPort as port,
	ifnull(out.trafOut,0) as trafOut,
	in.trafIn as trafIn
from
	(
		select dstPort, sum(psize) as trafOut
		from traffic
		where (pId = @proto) && (srcPort = 0) &&
			(srcIp = %USER%) && (dstIp = %SERVER%) &&
			(date >= "%BEGIN_DATE%" && date < "%END_DATE%")
		group by dstPort
	) as `out` right outer join
	(
		select srcPort, sum(psize) as trafIn
		from traffic
		where (pId = @proto) && (dstPort = 0) &&
			(dstIp = %USER%) && (srcIp = %SERVER%) &&
			(date >= "%BEGIN_DATE%" && date < "%END_DATE%")
		group by srcPort
	) as `in` on out.dstPort = in.srcPort
where out.dstPort is null';

/* установим в MySQL переменную @proto для протокола UDP */
$queryTemp = preg_replace('/%PROTO%/', 'udp', $querySetProto);
$result = my_mysql_query($queryTemp);
mysql_free_result($result);

/* Формируем статистику запрошеных портов по протоколу UDP. */
$replacements = array(USER, SERVER, BEGIN_DATE, END_DATE);
$queryTemp = preg_replace($patternsQueryTraf, $replacements, $queryTraf);
$result = my_mysql_query($queryTemp);

while ($line = mysql_fetch_assoc($result)) {
	$index = $line['port'] . ':udp';
	$dataTraf[$index]['allOut'] = $line['trafOut'];
	$dataTraf[$index]['allIn'] = $line['trafIn'];
}

mysql_free_result($result);

/* установим в MySQL переменную @proto для протокола TCP */
$queryTemp = preg_replace('/%PROTO%/', 'tcp', $querySetProto);
$result = my_mysql_query($queryTemp);
mysql_free_result($result);

/* Формируем статистику запрошенных портов по протоколу TCP. */
$replacements = array(USER, SERVER, BEGIN_DATE, END_DATE);
$queryTemp = preg_replace($patternsQueryTraf, $replacements, $queryTraf);
$result = my_mysql_query($queryTemp);

while ($line = mysql_fetch_assoc($result)) {
	$index = $line['port'] . ':tcp';
	$dataTraf[$index]['allOut'] = $line['trafOut'];
	$dataTraf[$index]['allIn'] = $line['trafIn'];
}

mysql_free_result($result);

/* Получим статистику за период сумарного трафика */
$patternsQueryPeriod = array("/%USER%/", "/%SERVER%/",
	"/%BEGIN_DATE%/", "/%END_DATE%/");

/* сформируем массив данных для построения графика */
$dateBeginArray = array();
$dateEndArray = array();
list($year, $month, $day) = split('-', BEGIN_DATE);
$initDate = mktime(0, 0, 0, $month, $day, $year);
list($year, $month, $day) = split('-', END_DATE);
$endDate = mktime(0, 0, 0, $month, $day, $year);

while ($initDate < $endDate) {
	$tempDate = date("Y-m-d H:i:s", $initDate);
	$dateBeginArray[] = $tempDate;

	list($date, $time) = split(' ', $tempDate);
	list($year, $month, $day) = split('-', $date);
	list($h, $m, $s) = split(':', $time);
	$initDate = PERIOD_DAY ?
		mktime($h + 1, 0, 0, $month, $day, $year) :
		mktime(0, 0, 0, $month, $day + 1, $year);
	$dateEndArray[] = date("Y-m-d H:i:s", $initDate);
}

/* Запрос возвращает количество прошедшего трафика за заданный период. */
$queryPeriod = '
select
	ifnull(out.psizeSum,0) as outTraf,
	ifnull(in.psizeSum,0) as inTraf
from
	(
		select sum(psize) as psizeSum
		from traffic
		where (srcPort = 0) && (srcIp = %USER%) && (dstIp = %SERVER%) &&
			(date >= "%BEGIN_DATE%" && date < "%END_DATE%")
	) as `out` join
	(
		select sum(psize) as psizeSum
		from traffic
		where (dstPort = 0) && (dstIp = %USER%) && (srcIp = %SERVER%) &&
			(date >= "%BEGIN_DATE%" && date < "%END_DATE%")
	) as `in`';

$imgDataOut = array();
$imgDataIn = array();
for ($i = 0; isset($dateBeginArray[$i]); $i++) {
	$replacements = array(USER, SERVER, $dateBeginArray[$i], $dateEndArray[$i]);
	$queryTemp = preg_replace($patternsQueryPeriod,
		$replacements, $queryPeriod);
	$result = my_mysql_query($queryTemp);
	$line = mysql_fetch_assoc($result);
	$imgDataOut[] = $line['outTraf'];
	$imgDataIn[] = $line['inTraf'];
	mysql_free_result($result);
}

/* сформируем линк для графика суммарного трафика */
$imgServer = './graph.php?graph=statgraph&amp;leg=' .
	urlencode('Sent') . '|' . urlencode('Received') .
	'&amp;xt=' . urlencode(PERIOD_DAY ? 'Hours' : 'Days') .
	'&amp;data=' . implode(',', $imgDataOut) . '|' . implode(',', $imgDataIn);

/* Функция сортировки данных */
$dataTrafCopy = $dataTraf;
function sort_cmp($a, $b)
{
	if (strcmp(SORT_CMP, 'port')) {
		global $dataTrafCopy;

		$v1 = $dataTrafCopy[$a][SORT_CMP];
		$v2 = $dataTrafCopy[$b][SORT_CMP];

		if ($v1 == $v2)
			return 0;
		return ($v1 > $v2) ? -1 : 1;
	} else {
		return strcmp($a, $b);
	}
}

uksort($dataTraf, "sort_cmp");

/* формирует ссылку для сортировки данных */
function make_sort_link($val, $sort)
{
	if ($sort != SORT_CMP) {
		$link = '<a href="./index.php?content=' . CONTENT . '&amp;user=' .
			USER . '&amp;server=' . SERVER . '&amp;period=' . PERIOD .
			'&amp;period_day=' . PERIOD_DAY . '&amp;sort=' . $sort .
			'">' . $val . '</a>';
	} else {
		$link = $val;
	}

	return $link;
}

function make_link($type, $server, $portStr)
{
	$str = <<<EOF
<script type="text/javascript">
//<![CDATA[
$(document).ready(function() {
	var sl = $("#ajax-$type a");
	var ip = sl.attr('href').replace(/$type:\/\//, '');
	$.post('./ajax/statistic_reverse_ip.php', {'ip': ip}, function(data) {
		var href = '$type://' + data;
		sl.attr('href', href);
	});
});
//]]>
</script><span id="ajax-$type"><a href="$type://$server" target="_blank">$portStr</a></span>
EOF;

	return $str;
}

/**
 * Отобразим страницу.
 */
list($year, $month) = split('-', BEGIN_DATE);
$day = PERIOD_DAY;

$title = 'Статистика за ' . make_month($year, $month, $day) .
	': Хост ' . USER_NAME . ', Сервер ' . SERVER_NAME;
print_html_head($title, 'reverse_ip');

/* отобразим название продукта */
print_head();

/* отобразим график содержащай статистику */
echo '<center><table>' . "\n";
echo '<tr><td class="pointer">';
print_login();
echo "</td></tr>\n";
echo '<tr><th class="title"><h2>Статистика за ' .
	make_month($year, $month, $day) . "</h2></th></tr>\n";
echo '<tr><th class="title"><h2>Хост ' . USER_NAME . ', Сервер ' .
	make_span(SERVER_NAME, 'ajax-reverse_ip') . "</h2></th></tr>\n";
echo "<tr><td></td></tr>\n";
echo '<tr><td class="title">';
echo '<img src="' . $imgServer . '" />';
echo '</td></tr>' . "\n";
echo '</table></center>' . "\n";

print_full_line();

/* отобразим таблицу содержащую статистику */
echo '<center><table>' . "\n";
echo '<tr><td class="pointer" colspan="4">| ' .
	'<a href="./index.php?content=timeserver&amp;user=' . USER .
	'&amp;server=' . SERVER . '&amp;period=' . PERIOD . '&amp;period_day=' .
	PERIOD_DAY . "\"><b>Сводные данные</b></a> |</td></tr>\n";

if (SESS_TYPE === 'admin') {
	echo '<tr><td class="navigation" colspan="4">' .
		'<a href="./index.php?content=period">' .
		'<b>Статистика по месяцам</b></a> > ' .
		'<a href="./index.php?content=statistic&amp;period=' . PERIOD .
		"&amp;period_day=$day\"><b>" . make_month($year, $month, $day) .
		'</b></a> > ' . '<a href="./index.php?content=user&amp;user=' . USER .
		'&amp;period=' . PERIOD . "&amp;period_day=$day\"><b>Хост " . USER_NAME .
		'</b></a> > <b>Сервер ' . make_span(SERVER_NAME, 'ajax-reverse_ip') .
		"</b></td></tr>\n";
} else if (SESS_TYPE === 'guest') {
	echo '<tr><td class="navigation" colspan="4">' .
		'<a href="./index.php?content=user&amp;user=' . USER . '&amp;period=' .
		PERIOD . "&amp;period_day=$day\"><b>Хост " . USER_NAME . '</b></a> > ' .
		'<b>Сервер ' . make_span(SERVER_NAME, 'ajax-reverse_ip') .
		"</b></td></tr>\n";
}

echo '<tr><td class="page" colspan="4">';
$endDate = date('t', PERIOD);
for ($i = 1; $i <= $endDate; $i++) {
	if ($i != PERIOD_DAY) {
		echo '<a href="./index.php?content=' . CONTENT . '&amp;user=' . USER .
			'&amp;server=' . SERVER . '&amp;period=' . PERIOD .
			"&amp;period_day=${i}&amp;sort=" . SORT_CMP . "\">$i</a> ";
	} else {
		echo "[<b>$i</b>] ";
	}
}
if (PERIOD_DAY) {
	echo '<a href="./index.php?content=' . CONTENT . '&amp;user=' . USER .
		'&amp;server=' . SERVER . '&amp;period=' . PERIOD .
		'&amp;period_day=0&amp;sort=' . SORT_CMP . "\">все</a></td></tr>\n";
} else {
	echo "[<b>все</b>]</td></tr>\n";
}

echo '<tr>';
print_data("Номер", "head0", "th");
print_data(make_sort_link("Порт", "port"), "head1", "th");
print_data(make_sort_link("Отправлено", "allOut"), "head1", "th");
print_data(make_sort_link("Получено", "allIn"), "head1", "th");
echo '</tr>' . "\n";

$count = 0;
$sumOut = $sumIn = 0;

foreach ($dataTraf as $port => $traf) {
	$count++;
	$sumOut += $traf['allOut'];
	$sumIn += $traf['allIn'];

	list($portNumb, $servName) = split(':', $port);
	$portName = getservbyport($portNumb, $servName);
	if ($portName) {
		$portStr = "$servName:$portName ($portNumb)";
	} else {
		$portStr = "$servName:$portNumb";
	}

	if ($portNumb == 80 && !strcmp($servName, 'tcp')) {
		$portStr = make_link('http', SERVER_NAME, $portStr);
	} else if ($portNumb == 21 && !strcmp($servName, 'tcp')) {
		$portStr = make_link('ftp', SERVER_NAME, $portStr);
	} else if ($portNumb == 443 && !strcmp($servName, 'tcp')) {
		$portStr = make_link('https', SERVER_NAME, $portStr);
	}

	echo '<tr>';
	print_data($count, "data0");
	print_data($portStr, "data1");
	print_data(val_trun($traf['allOut']), "data2");
	print_data(val_trun($traf['allIn']), "data2");
	echo '</tr>' . "\n";
}

echo '<tr>';
echo '<td class="page"></td>';
print_data("Итого:", "head4", "th");
print_data(val_trun($sumOut), "head3");
print_data(val_trun($sumIn), "head3");
echo '</tr>' . "\n";
echo '</table></center>' . "\n";

print_full_line();

/* отобразим информацию о программе */
print_info($queryCount);
print_html_tail();

?>