quota.php

См. документацию.

<?php
 
IncludeModuleLangFile(__FILE__);
 
class CDiskQuota
{
    var $max_execution_time = 20; // 20 sec
    var $LAST_ERROR = false;
    private static $instance;
    protected static $recalculateDb = false;
 
    public static function getInstance()
    {
        if (!isset(self::$instance))
        {
            $c = __CLASS__;
            self::$instance = new $c;
        }
 
        return self::$instance;
    }
 
    public function __construct($params = array())
    {
        if(array_key_exists("max_execution_time", $params) && intval($params["max_execution_time"]) > 0)
            $this->max_execution_time = intval($params["max_execution_time"]);
    }
 
    public static function recalculateDb(bool $mode = true): void
    {
        static::$recalculateDb = $mode;
    }
 
    public static function SetDBSize()
    {
        global $DB;
        $DBSize = 0;
 
        if (static::$recalculateDb && COption::GetOptionInt("main", "disk_space") > 0)
        {
            $db_res = $DB->Query("
                SELECT sum(Data_length + Index_length) as TOTAL_LEN
                FROM information_schema.tables
                WHERE table_schema = '{$DB->ForSql($DB->DBName)}'
            ");
            if ($db_res && ($res = $db_res->Fetch()))
            {
                $DBSize = $res['TOTAL_LEN'];
            }
 
            COption::SetOptionString("main_size", "~db", $DBSize);
            $params = array("status" => "d", "time" => time());
            COption::SetOptionString("main_size", "~db_params", serialize($params));
 
            static::recalculateDb(false);
        }
        else
        {
            $params = array("time" => false);
        }
 
        return array("status" => "done", "size" => $DBSize, "time" => $params["time"]);
    }
 
    public function SetDirSize($path="", $name="", $recount=false, $skip_dir = false)
    {
        if (empty($name))
            $name = $path;
 
        if ((empty($name) && empty($path)) || ($path == "files"))
        {
            $name = "files";
            $path = "";
        }
 
        $abs_path = str_replace("//", "/", $_SERVER['DOCUMENT_ROOT']."/".$path."/");
 
        $result = array();
 
        if (empty($abs_path))
            return false;
 
        $record = array("size" => COption::GetOptionString("main_size", "~".$name));
 
        if ($skip_dir)
        {
            if (!is_array($skip_dir))
                $skip_dir = array($skip_dir);
            foreach ($skip_dir as $key => $path_to_skip_dir)
                $skip_dir[$key] = str_replace("//", "/", $abs_path.$path_to_skip_dir."/");
        }
 
        if (!empty($record["size"]) && !$recount)
        {
            $record = array_merge(
                unserialize(COption::GetOptionString("main_size", "~".$name."_params"), ['allowed_classes' => false]),
                $record);
 
            $record["path_to_last_file"] = str_replace("//", "/", $abs_path.$record["file"]);
            if (is_file($record["path_to_last_file"]) && $record["status"] == "c")
            {
                $res = $this->GetDirListFromLastFile($abs_path, $record["path_to_last_file"], true, $skip_dir);
                if ($res["status"] == "done" || $res["status"] == "continue")
                {
                    $properties = array(
                        "status" => mb_substr($res["status"], 0, 1),
                        "file" => str_replace($abs_path, "", str_replace("//", "/", $res["last_file"])),
                        "time" => time());
 
                    $record["size"] = doubleVal($record["size"])+doubleVal($res["size"]);
                    COption::SetOptionString("main_size", "~".$name, $record["size"]);
                    COption::SetOptionString("main_size", "~".$name."_params", serialize($properties));
                    $result = $res;
                    $result["size"] = $record["size"];
                }
            }
            elseif ($record["status"] == "d")
            {
                return array("status" => "done", "last_file" => $record["file"], "size" => $record["size"]);
            }
        }
 
        if (empty($result))
        {
 
            $res = $this->GetDirListSimple($abs_path, true, $skip_dir);
 
            if ($res["status"] == "done" || $res["status"] == "continue")
            {
                $properties = array(
                    "status" => mb_substr($res["status"], 0, 1),
                    "file" => str_replace($abs_path, "", str_replace("//", "/", $res["last_file"])),
                    "time" => time());
 
                COption::SetOptionString("main_size", "~".$name, doubleVal($res["size"]));
                COption::SetOptionString("main_size", "~".$name."_params", serialize($properties));
                $result = $res;
            }
        }
 
        if (!empty($result))
            return $result;
        return array("status" => "error");
    }
 
    public function GetDirListSimple($path, $check_time = true, $skip_dir=false)
    {
        $path = str_replace("//", "/", $path."/");
        $res = array();
        $size = 0;
        $handle = @opendir($path);
 
        if ($handle)
        {
            while($file = readdir($handle))
            {
                if($file == "." || $file == "..")
                {
                    continue;
                }
 
                if(is_dir($path.$file))
                {
                    if (is_array($skip_dir) && (in_array(str_replace("//", "/", $path.$file."/"), $skip_dir)))
                    {
                        continue;
                    }
 
                    $res_rec = $this->GetDirListSimple($path.$file, $check_time, $skip_dir);
                    $res = array_merge($res, $res_rec["tree"]);
                    $size += doubleVal($res_rec["size"]);
                    if ($res_rec["status"] == "continue")
                    {
                        $res_rec["tree"] = $res;
                        $res_rec["size"] = doubleVal($size);
                        return $res_rec;
                    }
                }
                else
                {
                    $res[] = $path.$file;
                    $size += filesize($path.$file);
                    if ($check_time && intval(microtime(true) - START_EXEC_TIME) >= $this->max_execution_time)
                    {
                        return array("tree" => $res, "status" => "continue", "last_file" => $path.$file, "size" => $size);
                    }
                }
            }
            @closedir($handle);
        }
        else
        {
            return array("status" => "error");
        }
        return array("tree" => $res, "status" => "done", "last_file" => $path.$file, "size" => $size);
    }
 
    public function GetDirListFromLastFile($path, $path_to_last_file="", $check_time = true, $skip_dir = false)
    {
        $path = str_replace("//", "/", $path."/");
        $path_to_last_file = str_replace("//", "/", $path_to_last_file);
        $path_to_lf = str_replace($path, "", $path_to_last_file);
        $size = 0;
        $res = array();
        $file = '';
        $path_tree = explode("/", $path_to_lf);
        while ($lf = array_pop($path_tree))
        {
            $path_to_dir = str_replace("//", "/", $path.implode("/", $path_tree)."/");
            $handle = @opendir($path_to_dir);
            $search = true;
            if ($handle)
            {
                while($file = readdir($handle))
                {
                    if($file == "." || $file == ".." || $search)
                    {
                        if ($file == $lf)
                            $search = false;
                        continue;
                    }
 
                    if(is_dir($path_to_dir.$file))
                    {
                        if (is_array($skip_dir) && (in_array(str_replace("//", "/", $path.$file."/"), $skip_dir)))
                        {
                            continue;
                        }
                        $res_rec = $this->GetDirListSimple($path_to_dir.$file, $check_time);
                        $res = array_merge($res, $res_rec["tree"]);
                        $size += doubleVal($res_rec["size"]);
                        if ($res_rec["status"] == "continue")
                        {
                            $res_rec["tree"] = $res;
                            $res_rec["size"] = doubleVal($size);
                            return $res_rec;
                        }
                    }
                    else
                    {
 
                        $res[] = $path_to_dir.$file;
                        $size += filesize($path_to_dir.$file);
                        if ($check_time && intval(microtime(true) - START_EXEC_TIME) >= $this->max_execution_time)
                        {
                            return array("tree" => $res, "status" => "continue", "last_file" => $path_to_dir.$file, "size" => $size);
                        }
                    }
                }
            }
            @closedir($handle);
        }
        return array("tree" => $res, "status" => "done", "last_file" => $path.$file, "size" => $size);
    }
 
    public function Recount($id, $recount=false)
    {
        if ((COption::GetOptionInt("main", "disk_space") <= 0))
            return true;
 
        if ($id != "files" && (!is_dir($_SERVER['DOCUMENT_ROOT']."/".$id)))
            return array("status" => "error");
 
        if ($id == "files")
        {
            $result = $this->SetDirSize("", "files", $recount, array("bitrix"));
        }
        else
        {
            $result = $this->SetDirSize($id, "", $recount);
        }
 
        if (empty($result["time"]))
        {
            $result["time"] = time();
        }
        return $result;
    }
 
    public function GetDiskQuota()
    {
        if (COption::GetOptionInt("main", "disk_space") <= 0)
            return true;
 
        $this->LAST_ERROR = "";
        $arMsg = Array();
 
        if (COption::GetOptionInt("main_size", "~db") <= 0)
        {
            static::recalculateDb();
        }
 
        $quota = doubleVal(COption::GetOptionInt("main", "disk_space")*1024*1024 -
            COption::GetOptionInt("main_size", "~db") -
            COption::GetOptionInt("main_size", "~files"));
 
        if ($quota > 0)
        {
            return $quota;
        }
 
        $this->LAST_ERROR = GetMessage("MAIN_QUOTA_BAD");
        $arMsg[] = array("id"=>"QUOTA_BAD", "text"=> GetMessage("MAIN_QUOTA_BAD"));
 
        $e = new CAdminException($arMsg);
        $GLOBALS["APPLICATION"]->ThrowException($e);
 
        return false;
    }
 
    public static function UpdateDiskQuota($type, $size, $action)
    {
        if (COption::GetOptionInt("main", "disk_space") <= 0)
            return true;
 
        if (empty($type) || empty($size) || empty($action))
            return false;
 
        if (is_array($size))
            $size = mb_strlen(implode("", $size));
        elseif (doubleval($size) > 0)
            $size = doubleval($size);
        else
            $size = mb_strlen($size);
 
        $size = doubleval($size);
 
        $name = mb_strtolower($type) == "db" ? "db" : "files";
 
        if (in_array(mb_strtolower($action), array("delete", "del")))
        {
            COption::SetOptionString("main_size", "~".$name,
                doubleval(COption::GetOptionInt("main_size", "~".$name) - $size));
            return true;
        }
        if (in_array(mb_strtolower($action), array("update", "insert", "add", "copy")))
        {
            COption::SetOptionString("main_size", "~".$name,
                doubleval(COption::GetOptionInt("main_size", "~".$name) + $size));
            return true;
        }
        return false;
    }
 
    public function CheckDiskQuota($params = array())
    {
        if (COption::GetOptionInt("main", "disk_space") <= 0)
            return true;
 
        if (defined("SKIP_DISK_QUOTA_CHECK") && constant("SKIP_DISK_QUOTA_CHECK") === true)
            return true;
 
        $quota = $this->GetDiskQuota();
 
        if ($quota === true || $quota === false)
            return $quota;
 
        if (!empty($params))
        {
            if (is_array($params))
            {
                if (is_set($params, "FILE_SIZE"))
                    $size = $params["FILE_SIZE"];
                elseif (is_set($params, "SIZE"))
                    $size = $params["SIZE"];
                elseif (is_set($params, "file_size"))
                    $size = $params["file_size"];
                elseif (is_set($params, "size"))
                    $size = $params["size"];
                else
                    $size = mb_strlen(serialize($params));
 
                if ($size !== false)
                    return ((double)$quota - $size) > 0;
            }
            if (!is_array($params) && doubleVal($params) > 0 && ((double)$quota - $params) > 0)
                return true;
            if (((double)$quota - mb_strlen($params)) > 0)
                return true;
        }
        return false;
    }
}