<?php
|
|
/*
|
* Author Gody - Orm 2016
|
* You need to configure daily log rotation for pureftp (/etc/logorate.d/pure-ftpd-comon)
|
* TODO: replace logrotate to ISPConfig log rotation
|
*/
|
|
class cronjob_ftplogfiles extends cronjob {
|
|
// job schedule
|
protected $_schedule = '0 0 * * *';
|
|
/* this function is optional if it contains no custom code */
|
public function onPrepare() {
|
global $app;
|
|
parent::onPrepare();
|
}
|
|
/* this function is optional if it contains no custom code */
|
public function onBeforeRun() {
|
global $app;
|
|
return parent::onBeforeRun();
|
}
|
|
public function onRunJob() {
|
global $app, $conf;
|
|
//######################################################################################################
|
// Make the ftp logfiles directories world readable to enable ftp access
|
//######################################################################################################
|
|
if(is_dir('/var/log/pure-ftpd/')) exec('chmod +r /var/log/pure-ftpd/*');
|
|
//######################################################################################################
|
// Manage and compress ftp logfiles and create traffic statistics
|
//######################################################################################################
|
$sql = "SELECT domain_id, domain, type, document_root, web_folder, parent_domain_id FROM web_domain WHERE (type = 'vhost' or type = 'vhostsubdomain' or type = 'vhostalias') AND server_id = ?";
|
$records = $app->db->queryAllRecords($sql, $conf['server_id']);
|
|
function parse_ftp_log($line){
|
if(preg_match('/^\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3} - (.+) \[(\d+\/\w+\/\d+):.+\] "(PUT|GET) .+" \d+ (\d+)$/', $line, $matches) == false) return false;
|
|
$date = date('Y-m-d', strtotime(str_replace('/', '-', $matches[2]))); // Correction date
|
|
switch($matches[3])
|
{
|
case 'PUT':
|
$direction = 'in';
|
break;
|
case 'GET':
|
$direction = 'out';
|
break;
|
}
|
// Returned array
|
return array('username' => $matches[1], 'date' => $date,'direction' => $direction, 'size' => $matches[4]);
|
}
|
|
function add_ftp_traffic(&$traffic_array, $parsed_line)
|
{
|
if(is_array($traffic_array[$parsed_line['date']]) && array_key_exists($parsed_line['domain'], $traffic_array[$parsed_line['date']]))
|
{
|
$traffic_array[$parsed_line['date']][$parsed_line['domain']][$parsed_line['direction']] += $parsed_line['size'];
|
}
|
else
|
{
|
$traffic_array[$parsed_line['date']][$parsed_line['domain']] = array('in' => 0, 'out' => 0 );
|
$traffic_array[$parsed_line['date']][$parsed_line['domain']][$parsed_line['direction']] = $parsed_line['size'];
|
}
|
}
|
|
$fp = fopen('/var/log/pure-ftpd/transfer.log.1', 'r');
|
$ftp_traffic = array();
|
|
// cumule des stats journalière dans un tableau
|
while($line = fgets($fp))
|
{
|
$parsed_line = parse_ftp_log($line);
|
|
$sql = "SELECT wd.domain FROM ftp_user AS fu INNER JOIN web_domain AS wd ON fu.parent_domain_id = wd.domain_id WHERE fu.username = ? ";
|
$temp = $app->db->queryOneRecord($sql, $parsed_line['username'] );
|
|
$parsed_line['domain'] = $temp['domain'];
|
|
add_ftp_traffic($ftp_traffic, $parsed_line);
|
}
|
|
fclose($fp);
|
|
// Save du tableau en BD
|
foreach($ftp_traffic as $traffic_date => $all_traffic)
|
{
|
foreach ( $all_traffic as $hostname =>$traffic)
|
{
|
$sql1 = "SELECT hostname FROM ftp_traffic WHERE hostname = ? AND traffic_date = ?";
|
$tmp = $app->dbmaster->queryOneRecord($sql1, $hostname , $traffic_date);
|
|
if(is_array($tmp) && count($tmp) > 0) {
|
$sql = "UPDATE ftp_traffic SET in_bytes=in_bytes+ ?, out_bytes=out_bytes+ ? WHERE hostname = ? AND traffic_date = ? ";
|
}
|
else
|
{
|
$sql = "INSERT INTO ftp_traffic (in_bytes, out_bytes, hostname, traffic_date ) VALUES ( ?, ?, ?, ? )";
|
}
|
|
$resultat = $app->dbmaster->query($sql, $traffic['in'], $traffic['out'], $hostname, $traffic_date );
|
|
//if($resultat == 1){
|
//echo 'finished.'.PHP_EOL; // maybe you have a better solution ? }
|
}
|
}
|
|
parent::onRunJob();
|
}
|
|
/* this function is optional if it contains no custom code */
|
public function onAfterRun() {
|
global $app;
|
|
parent::onAfterRun();
|
}
|
}
|
|
?>
|
