<?php
$color_str;
$color_tmp;
$from = 'yesterday';
$to = 'today';
$src = 'all';
if(isset($_GET['src'])) $src = $_GET['src'];
if(isset($_GET['from'])) $from = $_GET['from'];
if(isset($_GET['to'])) $to = $_GET['to'];
$conn_string = "host=192.168.8.10 port=5432 dbname=CAMDPAMS user=olegk password=4Q2073gl";
$db_conn = pg_connect($conn_string);

function hex2rgb( $colour ) {
	if ( $colour[0] == '#' ) {
	    $colour = substr( $colour, 1 );
	}
	if ( strlen( $colour ) == 6 ) {
	    list( $r, $g, $b ) = array( $colour[0] . $colour[1], $colour[2] . $colour[3], $colour[4] . $colour[5] );
	}
	elseif ( strlen( $colour ) == 3 ) {
	    list( $r, $g, $b ) = array( $colour[0] . $colour[0], $colour[1] . $colour[1], $colour[2] . $colour[2] );
	}
	else {return false;}
	$r = hexdec( $r );
	$g = hexdec( $g );
	$b = hexdec( $b );
	return array( 'red' => $r, 'green' => $g, 'blue' => $b );
 }
 
function get_color(){
global $color_str;
global $color_tmp;
$colors =count($color_str)-1;
$t= mt_rand(0,$colors-1);
    if (!array_key_exists($color_str[$t], $color_tmp)){
	array_push($color_tmp,$color_str[$t]);
        return $color_str[$t];
    }
    else{ get_color();}
}
 
 
function start_map($from,$to,$src){ 
global $color_str;
global $color_tmp;
global $db_conn;
//header ("Content-type: image/png");
$imgname = "world_map.png";
$im = @imagecreatefrompng($imgname);
//$ink = imagecolorallocate($im, 255 , 0, 0);
$color_str = array("FFFFCC","FFFF99","FFFF66","FFFF33","FFFF00","CCCC00","FFCC66","FFCC00","FFCC33","CC9900","CC9933","996600",
			"FF9900","FF9933","CC9966","CC6600","996633","663300","FFCC99","FF9966","FF6600","CC6633","993300","660000",
			"FF6633","CC3300","FF3300","FF0000","CC0000","990000","FFCCCC","FF9999","FF6666","FF3333","FF0033","CC0033",
			"CC9999","CC6666","CC3333","993333","990033","330000","FF6699","FF3366","FF0066","CC3366","996666","663333",
			"FF99CC","FF3399","FF0099","CC0066","993366","660033","FF66CC","FF00CC","FF33CC","CC6699","CC0099","990066",
			"FFCCFF","FF99FF","FF66FF","FF33FF","FF00FF","CC3399","CC99CC","CC66CC","CC00CC","CC33CC","990099","993399",
			"CC66FF","CC33FF","CC00FF","9900CC","996699","660066","CC99FF","9933CC","9933FF","9900FF","660099","663366",
			"9966CC","9966FF","6600CC","6633CC","663399","330033","CCCCFF","9999FF","6633FF","6600FF","330099","330066",
			"9999CC","6666FF","6666CC","666699","333399","333366","3333FF","3300FF","3300CC","3333CC","000099","000066",
			"6699FF","3366FF","0000FF","0000CC","0033CC","000033","0066FF","0066CC","3366CC","0033FF","003399","003366",
			"99CCFF","3399FF","0099FF","6699CC","336699","006699","66CCFF","33CCFF","00CCFF","3399CC","0099CC","003333",
			"99CCCC","66CCCC","339999","669999","006666","336666","CCFFFF","99FFFF","66FFFF","33FFFF","00FFFF","00CCCC",
			"99FFCC","66FFCC","33FFCC","00FFCC","33CCCC","009999","66CC99","33CC99","00CC99","339966","009966","006633",
			"66FF99","33FF99","00FF99","33CC66","00CC66","009933","99FF99","66FF66","33FF66","00FF66","339933","006600",
			"CCFFCC","99CC99","66CC66","669966","336633","003300","33FF33","00FF33","00FF00","00CC00","33CC33","00CC33",
			"66FF00","66FF33","33FF00","33CC00","339900","009900","CCFF99","99FF66","66CC00","66CC33","669933","336600",
			"99FF00","99FF33","99CC66","99CC00","99CC33","669900","CCFF66","CCFF00","CCFF33","CCCC99","666633","333300",
			"CCCC66","CCCC33","999966","999933","999900","666600","CCCCCC","999999","666666","333333","000000");


$size = getimagesize("world_map.png");
$W = $size[0];
$H = $size[1];
$html = "";
$color_tmp= array();
$countr_legend = array();
$sql = "select \"CountryName\",sum(\"Total\") as total                                                                                                                           
    from \"mis\".\"report_fls_infection_country\" where \"Ts\" between '$from' and '$to' ";
if($src !='all') $sql .=" and \"CallerTypeId\" = $src ";
$sql .= "group by \"CountryName\" order by total desc";                                                                                                                                   
//and  \"CountryName\" = 'United States'    
$c_sql = "select \"CountryId\",\"CountryName\",\"Coordinates\" from mis.country_poligon join mis.country_name on (\"CountryId\" = \"Id\")                                      
          where \"CountryName\" = $1"; 
$coord_res = pg_prepare($db_conn, "country_coord", $c_sql);
$res = pg_query($db_conn,$sql);

while($cnt=pg_fetch_assoc($res)){
    $coord_res=pg_execute($db_conn, "country_coord", array($cnt['CountryName']));
//    $color = hex2rgb($color_str[$t]);
    $color = hex2rgb(get_color());
    $countr_legend[$cnt['CountryName']]=$cnt['total'];
    $ink = imagecolorallocate($im, $color['red'] , $color['green'], $color['blue']);
    while($row=pg_fetch_assoc($coord_res)){
	$points = explode(",",$row["Coordinates"]);
	$html .= "<area href=\"javascript:void(0)\" onclick=\"goCountry('".$cnt['CountryName']."','$from','$to');\" shape=\"poly\" coords=\"";
	$html .= $row["Coordinates"]."\" class=\"{stroke:false,fillColor:'ff0000',fillOpacity:0.0}\" ";
	$html .= "title=\"".$cnt['CountryName']."\" />";
	for ($i=0;$i<count($points);$i++) $points[$i]=$points[$i]-2;
	imagefilledpolygon($im, $points,count($points)/2 , $ink);
	}
	
}	
imagepng($im,"test.png");
imagedestroy($im);
print "
<!DOCTYPE html PUBLIC \"-//W3C//DTD HTML 4.01//EN\" \"http://www.w3.org/TR/html4/strict.dtd\"> 
<html>
<head>
<title>Infection world map</title>
    <script type=\"text/javascript\" src=\"jquery-1.4.2.min.js\"></script> 
    <script type=\"text/javascript\" src=\"jquery.metadata.min.js\"></script> 
    <script type=\"text/javascript\" src=\"jquery.maphilight.min.js\"></script> 
<script>
	$(function() {
    	    $('.map').maphilight();
    	    $('#squidheadlink').mouseover(function(e) {
    		$('#squidhead').mouseover();
    	    }).mouseout(function(e) {
                $('#squidhead').mouseout();
                }).click(function(e) { e.preventDefault(); });
        });
      function goCountry(cntr,from,to){
	var url = 'get_coord.php?location='+cntr+'&from='+from+'&to='+to;
	window.location = url;
      }
</script>    
</head>
<body>
<div style='width:100%'>
<div id = 'map' style='width:100%'>
<img src=\"test.png\" width=\"799\" height=\"350\" class=\"map\" usemap=\"#simple\" border='0' />
<map name=\"simple\"> ";
print $html;	
print "</map>
</div>
<hr>
<div id='legend' style='width:100%;'><table>";
$k = 6;
for($i =0;$i<count($countr_legend)-$k+1;$i+=$k ){
	  print "<tr>";
	 for($ii =0;$ii<$k;$ii++){
         print "<td><div style='width:20px;background-color:#".current($color_tmp).";'>&nbsp;</div></td><td>";
	 print "<a href='get_coord.php?src=$src&location=".key($countr_legend)."&from=$from&to=$to'>".key($countr_legend)."</a></td><td>".current($countr_legend)."</td>";
	 next($countr_legend);
	 next($color_tmp);
	  }
	  print "</tr>";
}
print "</table></div></div></body></html>";
}
function country_detail($country,$from,$to,$src){
global $db_conn;
if($src != 'all'){ $sql = "select \"MalwareName\",\"Total\" from \"mis\".\"report_fls_infection_country\" where (\"Ts\" between '$from' and '$to')
	and \"CountryName\"= '$country' and \"CallerTypeId\" = $src";
}
else{$sql = "select \"MalwareName\",sum(\"Total\")as \"Total\" from \"mis\".\"report_fls_infection_country\" 
	      where (\"Ts\" between '$from' and '$to') and \"CountryName\"= '$country' 
	      group by \"MalwareName\" order by \"Total\" desc";

}
$res = pg_query($db_conn,$sql);
flush();
print "
<!DOCTYPE html PUBLIC \"-//W3C//DTD HTML 4.01//EN\" \"http://www.w3.org/TR/html4/strict.dtd\"> 
<html>
<head>
<title>Infection detail for $country </title>
</head>
<body>
<a href='get_coord.php?from=$from&to=$to'>return to map</a>
<h1>Infection detail for $country</h1>
<table>
<tr><th>#</th>
<th>Malware Name</th>
<th>Count</th></tr>";
$i =1;
while($row=pg_fetch_assoc($res)){
    print "<tr><td>$i</td><td>".$row['MalwareName']."</td><td>".$row['Total']."</td></tr>";
    $i++;
}
print "</table></body></html>";

}

if (!isset($_GET['location'])){start_map($from,$to,$src);}
else{
country_detail($_GET['location'],$from,$to,$src);
}

?>