Files
2021-08-06 10:35:01 +02:00

21 lines
957 B
PHP

<?php
// Copyright (c) 2021 Battelle Energy Alliance, LLC. All rights reserved.
error_reporting(E_ALL | E_STRICT);
require('UploadHandler.php');
class CustomUploadHandler extends UploadHandler {
protected function trim_file_name($file_path, $name, $size, $type, $error, $index, $content_range) {
$enabled_carve_modes = array("interesting", "mapped", "known", "all");
if (isset($_POST["tags"]) && !empty($_POST["tags"])) {
$name = $_POST["tags"] . "," . $name;
}
if (isset($_POST["auto-carve"]) && in_array($_POST["auto-carve"], $enabled_carve_modes)) {
$name = "AUTOCARVE".$_POST["auto-carve"]."," . $name;
}
if (isset($_POST["auto-zeek"]) && $_POST["auto-zeek"] == "enabled") {
$name = "AUTOZEEK," . $name;
}
return parent::trim_file_name($file_path, preg_replace("/[^a-zA-Z0-9\s_\(\)\.,-]/", "", $name), $size, $type, $error, $index, $content_range);
}
}
$upload_handler = new CustomUploadHandler();