class.registry.php

Go to the documentation of this file.

<?php
/*
 * Copyright (c) 2021-2024 Bearsampp
 * License:  GNU General Public License version 3 or later; see LICENSE.txt
 * Author: Bear
 * Website: https://bearsampp.com
 * Github: https://github.com/Bearsampp
 */

class Registry
{
    const END_PROCESS_STR = 'FINISHED!';
 
    const HKEY_CLASSES_ROOT = 'HKCR';
    const HKEY_CURRENT_USER = 'HKCU';
    const HKEY_LOCAL_MACHINE = 'HKLM';
    const HKEY_USERS = 'HKEY_USERS';
 
    const REG_SZ = 'REG_SZ';
    const REG_EXPAND_SZ = 'REG_EXPAND_SZ';
    const REG_BINARY = 'REG_BINARY';
    const REG_DWORD = 'REG_DWORD';
    const REG_MULTI_SZ = 'REG_MULTI_SZ';
 
    const REG_ERROR_ENTRY = 'REG_ERROR_ENTRY';
    const REG_ERROR_SET = 'REG_ERROR_SET';
    const REG_NO_ERROR = 'REG_NO_ERROR';
 
    const ENV_KEY = 'SYSTEM\CurrentControlSet\Control\Session Manager\Environment';
 
    // App bins entry
    const APP_BINS_REG_ENTRY = 'BEARSAMPP_BINS';
 
    // App path entry
    const APP_PATH_REG_ENTRY = 'BEARSAMPP_PATH';
 
    // System path entry
    const SYSPATH_REG_ENTRY = 'Path';
 
    // Processor architecture
    const PROCESSOR_REG_SUBKEY = 'HARDWARE\DESCRIPTION\System\CentralProcessor\0';
    const PROCESSOR_REG_ENTRY = 'Identifier';
 
    private $latestError;

    public function __construct()
    {
        Log::initClass($this);
        $this->latestError = null;
    }

    private function writeLog($log)
    {
        global $bearsamppRoot;
        Log::debug($log, $bearsamppRoot->getRegistryLogFilePath());
    }

    public function exists($key, $subkey, $entry = null)
    {
        $this->writeLog('Exists ' . $key . '\\' . $subkey . '\\' . $entry);
 
        // Use Win32Native COM implementation
        $result = Win32Native::registryExists($key, $subkey, $entry);
 
        $this->writeLog('-> result: ' . ($result ? '1' : '0'));
 
        return $result;
    }

    public function getValue($key, $subkey, $entry = null)
    {
        global $bearsamppLang;
 
        $this->latestError = null;
 
        $this->writeLog('GetValue ' . $key . '\\' . $subkey . '\\' . $entry);
 
        // Use Win32Native COM implementation
        $result = Win32Native::registryGetValue($key, $subkey, $entry);
 
        $this->writeLog('-> result: ' . $result);
 
        if ($result === null) {
            $this->latestError = $bearsamppLang->getValue(Lang::ERROR) . ' Registry value not found';
            return false;
        }
 
        return $result;
    }

    public function setStringValue($key, $subkey, $entry, $value)
    {
        return $this->setValue($key, $subkey, $entry, $value, 'SetStringValue');
    }

    public function setExpandStringValue($key, $subkey, $entry, $value)
    {
        return $this->setValue($key, $subkey, $entry, $value, 'SetExpandedStringValue');
    }

    public function deleteValue($key, $subkey, $entry)
    {
        $this->writeLog('delete');
        return $this->setValue($key, $subkey, $entry, null, 'DeleteValue');
    }

    private function setValue($key, $subkey, $entry, $value, $type)
    {
        global $bearsamppLang;
 
        $this->latestError = null;
 
        $this->writeLog('SetValue ' . $key . '\\' . $subkey . '\\' . $entry);
        $this->writeLog('-> value: ' . $value);
        $this->writeLog('-> type: ' . $type);
 
        // Map the VBS type to Win32Native registry type
        $regType = self::REG_SZ; // Default
        if ($type == 'SetExpandedStringValue') {
            $regType = self::REG_EXPAND_SZ;
        } elseif ($type == 'SetStringValue') {
            $regType = self::REG_SZ;
        }
 
        // Handle delete operations
        if ($type == 'DeleteValue' && !empty($entry)) {
            // Delete a value
            $result = Win32Native::registryDeleteValue($key, $subkey, $entry);
        } elseif ($type == 'DeleteValue' && empty($entry)) {
            // Delete a key
            $result = Win32Native::registryDeleteKey($key, $subkey);
        } else {
            // Set a value
            $result = Win32Native::registrySetValue($key, $subkey, $entry, $value, $regType);
        }
 
        if ($subkey == self::ENV_KEY) {
            Batch::refreshEnvVars();
        }
 
        $this->writeLog('-> result: ' . ($result ? 'success' : 'failed'));
 
        if (!$result) {
            $this->latestError = $bearsamppLang->getValue(Lang::ERROR) . ' Registry operation failed';
            return false;
        }
 
        // Verify the value was set correctly (for set operations)
        if ($type != 'DeleteValue' && !empty($value)) {
            $verifyValue = Win32Native::registryGetValue($key, $subkey, $entry);
            if ($verifyValue != $value) {
                $this->latestError = sprintf($bearsamppLang->getValue(Lang::REGISTRY_SET_ERROR_TEXT), $value);
                return false;
            }
        }
 
        return true;
    }

    public function getLatestError()
    {
        return $this->latestError;
    }
}