Utils.php

<?php
 
/*
 *
 *  ____            _        _   __  __ _                  __  __ ____
 * |  _ \ ___   ___| | _____| |_|  \/  (_)_ __   ___      |  \/  |  _ \
 * | |_) / _ \ / __| |/ / _ \ __| |\/| | | '_ \ / _ \_____| |\/| | |_) |
 * |  __/ (_) | (__|   <  __/ |_| |  | | | | | |  __/_____| |  | |  __/
 * |_|   \___/ \___|_|\_\___|\__|_|  |_|_|_| |_|\___|     |_|  |_|_|
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * @author PocketMine Team
 * @link http://www.pocketmine.net/
 *
 *
 */
 
declare(strict_types=1);
 
namespace pocketmine\utils;
 
use DaveRandom\CallbackValidator\CallbackType;
use pocketmine\entity\Location;
use pocketmine\errorhandler\ErrorTypeToStringMap;
use pocketmine\math\Vector3;
use pocketmine\thread\ThreadCrashInfoFrame;
use Ramsey\Uuid\Uuid;
use Ramsey\Uuid\UuidInterface;
use function array_combine;
use function array_map;
use function array_reverse;
use function array_values;
use function bin2hex;
use function chunk_split;
use function class_exists;
use function count;
use function debug_zval_dump;
use function dechex;
use function exec;
use function explode;
use function file;
use function file_exists;
use function file_get_contents;
use function function_exists;
use function get_class;
use function get_current_user;
use function get_loaded_extensions;
use function getenv;
use function gettype;
use function implode;
use function interface_exists;
use function is_a;
use function is_array;
use function is_bool;
use function is_float;
use function is_infinite;
use function is_int;
use function is_nan;
use function is_object;
use function is_string;
use function mb_check_encoding;
use function mt_getrandmax;
use function mt_rand;
use function ob_end_clean;
use function ob_get_contents;
use function ob_start;
use function opcache_get_status;
use function ord;
use function php_uname;
use function phpversion;
use function preg_grep;
use function preg_match;
use function preg_match_all;
use function preg_replace;
use function shell_exec;
use function spl_object_id;
use function str_contains;
use function str_pad;
use function str_split;
use function str_starts_with;
use function stripos;
use function strlen;
use function substr;
use function sys_get_temp_dir;
use function trim;
use function xdebug_get_function_stack;
use const PHP_EOL;
use const PHP_INT_MAX;
use const PHP_INT_SIZE;
use const PHP_MAXPATHLEN;
use const STR_PAD_LEFT;
use const STR_PAD_RIGHT;
 
final class Utils{
    public const OS_WINDOWS = "win";
    public const OS_IOS = "ios";
    public const OS_MACOS = "mac";
    public const OS_ANDROID = "android";
    public const OS_LINUX = "linux";
    public const OS_BSD = "bsd";
    public const OS_UNKNOWN = "other";
 
    private static ?string $os = null;
    private static ?UuidInterface $serverUniqueId = null;
    private static ?int $cpuCores = null;
 
    public static function getNiceClosureName(\Closure $closure) : string{
        $func = new \ReflectionFunction($closure);
        if(!str_contains($func->getName(), '{closure')){
            //closure wraps a named function, can be done with reflection or fromCallable()
            //isClosure() is useless here because it just tells us if $func is reflecting a Closure object
 
            $scope = $func->getClosureScopeClass();
            if($scope !== null){ //class method
                return
                    $scope->getName() .
                    ($func->getClosureThis() !== null ? "->" : "::") .
                    $func->getName(); //name doesn't include class in this case
            }
 
            //non-class function
            return $func->getName();
        }
        $filename = $func->getFileName();
 
        return "closure@" . ($filename !== false ?
                Filesystem::cleanPath($filename) . "#L" . $func->getStartLine() :
                "internal"
            );
    }
 
    public static function getNiceClassName(object $obj) : string{
        $reflect = new \ReflectionClass($obj);
        if($reflect->isAnonymous()){
            $filename = $reflect->getFileName();
 
            return "anonymous@" . ($filename !== false ?
                    Filesystem::cleanPath($filename) . "#L" . $reflect->getStartLine() :
                    "internal"
                );
        }
 
        return $reflect->getName();
    }
 
    public static function cloneCallback() : \Closure{
        return static function(object $o){
            return clone $o;
        };
    }
 
    public static function cloneObjectArray(array $array) : array{
        $callback = self::cloneCallback();
        return array_map($callback, $array);
    }
 
    public static function getMachineUniqueId(string $extra = "") : UuidInterface{
        if(self::$serverUniqueId !== null && $extra === ""){
            return self::$serverUniqueId;
        }
 
        $machine = php_uname("a");
        $cpuinfo = @file("/proc/cpuinfo");
        if($cpuinfo !== false){
            $cpuinfoLines = preg_grep("/(model name|Processor|Serial)/", $cpuinfo);
            if($cpuinfoLines === false){
                throw new AssumptionFailedError("Pattern is valid, so this shouldn't fail ...");
            }
            $machine .= implode("", $cpuinfoLines);
        }
        $machine .= sys_get_temp_dir();
        $machine .= $extra;
        $os = Utils::getOS();
        if($os === Utils::OS_WINDOWS){
            @exec("ipconfig /ALL", $mac);
            $mac = implode("\n", $mac);
            if(preg_match_all("#Physical Address[. ]{1,}: ([0-9A-F\\-]{17})#", $mac, $matches) > 0){
                foreach($matches[1] as $i => $v){
                    if($v == "00-00-00-00-00-00"){
                        unset($matches[1][$i]);
                    }
                }
                $machine .= implode(" ", $matches[1]); //Mac Addresses
            }
        }elseif($os === Utils::OS_LINUX){
            if(file_exists("/etc/machine-id")){
                $machine .= file_get_contents("/etc/machine-id");
            }else{
                @exec("ifconfig 2>/dev/null", $mac);
                $mac = implode("\n", $mac);
                if(preg_match_all("#HWaddr[ \t]{1,}([0-9a-f:]{17})#", $mac, $matches) > 0){
                    foreach($matches[1] as $i => $v){
                        if($v == "00:00:00:00:00:00"){
                            unset($matches[1][$i]);
                        }
                    }
                    $machine .= implode(" ", $matches[1]); //Mac Addresses
                }
            }
        }elseif($os === Utils::OS_ANDROID){
            $machine .= @file_get_contents("/system/build.prop");
        }elseif($os === Utils::OS_MACOS){
            $machine .= shell_exec("system_profiler SPHardwareDataType | grep UUID");
        }
        $data = $machine . PHP_MAXPATHLEN;
        $data .= PHP_INT_MAX;
        $data .= PHP_INT_SIZE;
        $data .= get_current_user();
        foreach(get_loaded_extensions() as $ext){
            $data .= $ext . ":" . phpversion($ext);
        }
 
        //TODO: use of NIL as namespace is a hack; it works for now, but we should have a proper namespace UUID
        $uuid = Uuid::uuid3(Uuid::NIL, $data);
 
        if($extra === ""){
            self::$serverUniqueId = $uuid;
        }
 
        return $uuid;
    }
 
    public static function getOS(bool $recalculate = false) : string{
        if(self::$os === null || $recalculate){
            $uname = php_uname("s");
            if(stripos($uname, "Darwin") !== false){
                if(str_starts_with(php_uname("m"), "iP")){
                    self::$os = self::OS_IOS;
                }else{
                    self::$os = self::OS_MACOS;
                }
            }elseif(stripos($uname, "Win") !== false || $uname === "Msys"){
                self::$os = self::OS_WINDOWS;
            }elseif(stripos($uname, "Linux") !== false){
                if(@file_exists("/system/build.prop")){
                    self::$os = self::OS_ANDROID;
                }else{
                    self::$os = self::OS_LINUX;
                }
            }elseif(stripos($uname, "BSD") !== false || $uname === "DragonFly"){
                self::$os = self::OS_BSD;
            }else{
                self::$os = self::OS_UNKNOWN;
            }
        }
 
        return self::$os;
    }
 
    public static function getCoreCount(bool $recalculate = false) : int{
        if(self::$cpuCores !== null && !$recalculate){
            return self::$cpuCores;
        }
 
        $processors = 0;
        switch(Utils::getOS()){
            case Utils::OS_LINUX:
            case Utils::OS_ANDROID:
                if(($cpuinfo = @file('/proc/cpuinfo')) !== false){
                    foreach($cpuinfo as $l){
                        if(preg_match('/^processor[ \t]*:[ \t]*[0-9]+$/m', $l) > 0){
                            ++$processors;
                        }
                    }
                }elseif(($cpuPresent = @file_get_contents("/sys/devices/system/cpu/present")) !== false){
                    if(preg_match("/^([0-9]+)\\-([0-9]+)$/", trim($cpuPresent), $matches) > 0){
                        $processors = ((int) $matches[2]) - ((int) $matches[1]);
                    }
                }
                break;
            case Utils::OS_BSD:
            case Utils::OS_MACOS:
                $processors = (int) shell_exec("sysctl -n hw.ncpu");
                break;
            case Utils::OS_WINDOWS:
                $processors = (int) getenv("NUMBER_OF_PROCESSORS");
                break;
        }
        return self::$cpuCores = $processors;
    }
 
    public static function hexdump(string $bin) : string{
        $output = "";
        $bin = str_split($bin, 16);
        foreach($bin as $counter => $line){
            $hex = chunk_split(chunk_split(str_pad(bin2hex($line), 32, " ", STR_PAD_RIGHT), 2, " "), 24, " ");
            $ascii = preg_replace('#([^\x20-\x7E])#', ".", $line);
            $output .= str_pad(dechex($counter << 4), 4, "0", STR_PAD_LEFT) . "  " . $hex . " " . $ascii . PHP_EOL;
        }
 
        return $output;
    }
 
    public static function printable(mixed $str) : string{
        if(!is_string($str)){
            return gettype($str);
        }
 
        return preg_replace('#([^\x20-\x7E])#', '.', $str);
    }
 
    public static function javaStringHash(string $string) : int{
        $hash = 0;
        for($i = 0, $len = strlen($string); $i < $len; $i++){
            $ord = ord($string[$i]);
            if(($ord & 0x80) !== 0){
                $ord -= 0x100;
            }
            $hash = 31 * $hash + $ord;
            $hash &= 0xFFFFFFFF;
        }
        return $hash;
    }
 
    public static function getReferenceCount(object $value, bool $includeCurrent = true) : int{
        ob_start();
        debug_zval_dump($value);
        $contents = ob_get_contents();
        if($contents === false) throw new AssumptionFailedError("ob_get_contents() should never return false here");
        $ret = explode("\n", $contents);
        ob_end_clean();
 
        if(preg_match('/^.* refcount\\(([0-9]+)\\)\\{$/', trim($ret[0]), $m) > 0){
            return ((int) $m[1]) - ($includeCurrent ? 3 : 4); //$value + zval call + extra call
        }
        return -1;
    }
 
    private static function printableExceptionMessage(\Throwable $e) : string{
        $errstr = preg_replace('/\s+/', ' ', trim($e->getMessage()));
 
        $errno = $e->getCode();
        if(is_int($errno)){
            try{
                $errno = ErrorTypeToStringMap::get($errno);
            }catch(\InvalidArgumentException $ex){
                //pass
            }
        }
 
        $errfile = Filesystem::cleanPath($e->getFile());
        $errline = $e->getLine();
 
        return get_class($e) . ": \"$errstr\" ($errno) in \"$errfile\" at line $errline";
    }
 
    public static function printableExceptionInfo(\Throwable $e, $trace = null) : array{
        if($trace === null){
            $trace = $e->getTrace();
        }
 
        $lines = [self::printableExceptionMessage($e)];
        $lines[] = "--- Stack trace ---";
        foreach(Utils::printableTrace($trace) as $line){
            $lines[] = "  " . $line;
        }
        for($prev = $e->getPrevious(); $prev !== null; $prev = $prev->getPrevious()){
            $lines[] = "--- Previous ---";
            $lines[] = self::printableExceptionMessage($prev);
            foreach(Utils::printableTrace($prev->getTrace()) as $line){
                $lines[] = "  " . $line;
            }
        }
        $lines[] = "--- End of exception information ---";
        return $lines;
    }
 
    private static function stringifyValueForTrace(mixed $value, int $maxStringLength) : string{
        return match(true){
            is_object($value) => "object " . self::getNiceClassName($value) . "#" . spl_object_id($value),
            is_array($value) => "array[" . count($value) . "]",
            is_string($value) => "string[" . strlen($value) . "] " . substr(Utils::printable($value), 0, $maxStringLength),
            is_bool($value) => $value ? "true" : "false",
            is_int($value) => "int " . $value,
            is_float($value) => "float " . $value,
            $value === null => "null",
            default => gettype($value) . " " . Utils::printable((string) $value)
        };
    }
 
    public static function printableTrace(array $trace, int $maxStringLength = 80) : array{
        $messages = [];
        for($i = 0; isset($trace[$i]); ++$i){
            $params = "";
            if(isset($trace[$i]["args"]) || isset($trace[$i]["params"])){
                if(isset($trace[$i]["args"])){
                    $args = $trace[$i]["args"];
                }else{
                    $args = $trace[$i]["params"];
                }
                $paramsList = [];
                $offset = 0;
                foreach($args as $argId => $value){
                    $paramsList[] = ($argId === $offset ? "" : "$argId: ") . self::stringifyValueForTrace($value, $maxStringLength);
                    $offset++;
                }
                $params = implode(", ", $paramsList);
            }
            $messages[] = "#$i " . (isset($trace[$i]["file"]) ? Filesystem::cleanPath($trace[$i]["file"]) : "") . "(" . (isset($trace[$i]["line"]) ? $trace[$i]["line"] : "") . "): " . (isset($trace[$i]["class"]) ? $trace[$i]["class"] . (($trace[$i]["type"] === "dynamic" || $trace[$i]["type"] === "->") ? "->" : "::") : "") . $trace[$i]["function"] . "(" . Utils::printable($params) . ")";
        }
        return $messages;
    }
 
    public static function printableTraceWithMetadata(array $rawTrace, int $maxStringLength = 80) : array{
        $printableTrace = self::printableTrace($rawTrace, $maxStringLength);
        $safeTrace = [];
        foreach($printableTrace as $frameId => $printableFrame){
            $rawFrame = $rawTrace[$frameId];
            $safeTrace[$frameId] = new ThreadCrashInfoFrame(
                $printableFrame,
                $rawFrame["file"] ?? null,
                $rawFrame["line"] ?? 0
            );
        }
 
        return $safeTrace;
    }
 
    public static function currentTrace(int $skipFrames = 0) : array{
        ++$skipFrames; //omit this frame from trace, in addition to other skipped frames
        if(function_exists("xdebug_get_function_stack") && count($trace = @xdebug_get_function_stack()) !== 0){
            $trace = array_reverse($trace);
        }else{
            $e = new \Exception();
            $trace = $e->getTrace();
        }
        for($i = 0; $i < $skipFrames; ++$i){
            unset($trace[$i]);
        }
        return array_values($trace);
    }
 
    public static function printableCurrentTrace(int $skipFrames = 0) : array{
        return self::printableTrace(self::currentTrace(++$skipFrames));
    }
 
    public static function parseDocComment(string $docComment) : array{
        $rawDocComment = substr($docComment, 3, -2); //remove the opening and closing markers
        preg_match_all('/(*ANYCRLF)^[\t ]*(?:\* )?@([a-zA-Z\-]+)(?:[\t ]+(.+?))?[\t ]*$/m', $rawDocComment, $matches);
 
        return array_combine($matches[1], $matches[2]);
    }
 
    public static function testValidInstance(string $className, string $baseName) : void{
        $baseInterface = false;
        if(!class_exists($baseName)){
            if(!interface_exists($baseName)){
                throw new \InvalidArgumentException("Base class $baseName does not exist");
            }
            $baseInterface = true;
        }
        if(!class_exists($className)){
            throw new \InvalidArgumentException("Class $className does not exist or is not a class");
        }
        if(!is_a($className, $baseName, true)){
            throw new \InvalidArgumentException("Class $className does not " . ($baseInterface ? "implement" : "extend") . " $baseName");
        }
        $class = new \ReflectionClass($className);
        if(!$class->isInstantiable()){
            throw new \InvalidArgumentException("Class $className cannot be constructed");
        }
    }
 
    public static function validateCallableSignature(callable|CallbackType $signature, callable $subject) : void{
        if(!($signature instanceof CallbackType)){
            $signature = CallbackType::createFromCallable($signature);
        }
        if(!$signature->isSatisfiedBy($subject)){
            throw new \TypeError("Declaration of callable `" . CallbackType::createFromCallable($subject) . "` must be compatible with `" . $signature . "`");
        }
    }
 
    public static function validateArrayValueType(array $array, \Closure $validator) : void{
        foreach($array as $k => $v){
            try{
                $validator($v);
            }catch(\TypeError $e){
                throw new \TypeError("Incorrect type of element at \"$k\": " . $e->getMessage(), 0, $e);
            }
        }
    }
 
    public static function stringifyKeys(array $array) : \Generator{
        foreach($array as $key => $value){ // @phpstan-ignore-line - this is where we fix the stupid bullshit with array keys :)
            yield (string) $key => $value;
        }
    }
 
    public static function promoteKeys(array $array) : array{
        return $array;
    }
 
    public static function checkUTF8(string $string) : void{
        if(!mb_check_encoding($string, 'UTF-8')){
            throw new \InvalidArgumentException("Text must be valid UTF-8");
        }
    }
 
    public static function assumeNotFalse(mixed $value, \Closure|string $context = "This should never be false") : mixed{
        if($value === false){
            throw new AssumptionFailedError("Assumption failure: " . (is_string($context) ? $context : $context()) . " (THIS IS A BUG)");
        }
        return $value;
    }
 
    public static function checkFloatNotInfOrNaN(string $name, float $float) : void{
        if(is_nan($float)){
            throw new \InvalidArgumentException("$name cannot be NaN");
        }
        if(is_infinite($float)){
            throw new \InvalidArgumentException("$name cannot be infinite");
        }
    }
 
    public static function checkVector3NotInfOrNaN(Vector3 $vector3) : void{
        if($vector3 instanceof Location){ //location could be masquerading as vector3
            self::checkFloatNotInfOrNaN("yaw", $vector3->yaw);
            self::checkFloatNotInfOrNaN("pitch", $vector3->pitch);
        }
        self::checkFloatNotInfOrNaN("x", $vector3->x);
        self::checkFloatNotInfOrNaN("y", $vector3->y);
        self::checkFloatNotInfOrNaN("z", $vector3->z);
    }
 
    public static function checkLocationNotInfOrNaN(Location $location) : void{
        self::checkVector3NotInfOrNaN($location);
    }
 
    public static function getOpcacheJitMode() : ?int{
        if(
            function_exists('opcache_get_status') &&
            ($opcacheStatus = opcache_get_status(false)) !== false &&
            isset($opcacheStatus["jit"]["on"])
        ){
            $jit = $opcacheStatus["jit"];
            if($jit["on"] === true){
                return (($jit["opt_flags"] >> 2) * 1000) +
                    (($jit["opt_flags"] & 0x03) * 100) +
                    ($jit["kind"] * 10) +
                    $jit["opt_level"];
            }
 
            //jit available, but disabled
            return 0;
        }
 
        //jit not available
        return null;
    }
 
    public static function getRandomFloat() : float{
        return mt_rand() / mt_getrandmax();
    }
}