form_question_rule_type_file_virus_check.inc

<?php
require_once dirname(__FILE__).'/../../form_question_rule/form_question_rule.inc';
require_once SQ_FUDGE_PATH.'/antivirus/antivirus.inc';

class Form_Question_Rule_Type_File_Virus_Check extends Form_Question_Rule
{


    function __construct($assetid=0)
    {
        parent::__construct($assetid);

    }//end constructor



    function evaluate($answer, $rule_data, &$q_asset)
    {
        //get all data we need from either current submission or incomplete submission
        if (!empty($_FILES['q'.$q_asset->id]['tmp_name'])) {
            $temp_file = $_FILES['q'.$q_asset->id]['tmp_name'];
            $answer_name = $_FILES['q'.$q_asset->id]['name'];           
        }
        else {
            $extra_data = $q_asset->getExtraData();
            $temp_file = array_get_index($extra_data, 'temp_filesystem_path','');
            $answer_name = basename($temp_file);
        }
        
        // if no file uploaded, no need to scan virus
        if ((empty($answer_name) || $answer_name == 'No file uploaded') && !isset($temp_file)) return TRUE;
        
        // if disabled, no need to proceed
        if(isset($rule_data['check_virus']) && $rule_data['check_virus'] == 2) return TRUE;
        
        $report;    
        $status = Antivirus::scan_file($temp_file,$report);
        // If infected, remove and warn the user
        if (!$status) {
            unlink($temp_file);
            return FALSE;
        }

        return TRUE;
    }//end evaluate()


    function generateJSCode(&$q_asset, $rule_data)
    {
        // Generate a customisable indicator DIV
        
        // if disabled
        if(isset($rule_data['check_virus']) && $rule_data['check_virus'] == 2) return '';
        
        $content = isset($rule_data['indicator_content']) && !empty ($rule_data['indicator_content']) ? $rule_data['indicator_content'] : translate('core_form_rule_file_virus_check_indicator_content_default');
        $class = isset($rule_data['indicator_class']) && !empty ($rule_data['indicator_class']) ? $rule_data['indicator_class'] : translate('core_form_rule_file_virus_check_indicator_class_default');
        ob_start();
        ?>
        var newdiv = document.createElement('div');
        var divIdName = 'file_upload_virus_check_indicator';
        if(document.getElementById(divIdName) == null ){
            var indicator_content = '<?php echo $content; ?>';
            newdiv.setAttribute('id',divIdName);
            newdiv.setAttribute('class','<?php echo $class; ?>');
            newdiv.innerHTML = indicator_content;
            form.appendChild(newdiv);
        }
        <?php
        $contents = ob_get_contents();
        ob_end_clean();
        return $contents;

    }//end generateJSCode()


    function defaultError(&$q_asset, $rule_data)
    {
        return translate('core_form_rule_file_virus_check_error_true', $q_asset->getValue(), $q_asset->attr('name'));

    }//end defaultError()


}//end class
?>