98 lines
		
	
	
		
			2.5 KiB
		
	
	
	
		
			PHP
		
	
	
	
	
	
			
		
		
	
	
			98 lines
		
	
	
		
			2.5 KiB
		
	
	
	
		
			PHP
		
	
	
	
	
	
<?php
 | 
						|
 | 
						|
/*
 | 
						|
 * This file is part of the Symfony package.
 | 
						|
 *
 | 
						|
 * (c) Fabien Potencier <fabien@symfony.com>
 | 
						|
 *
 | 
						|
 * For the full copyright and license information, please view the LICENSE
 | 
						|
 * file that was distributed with this source code.
 | 
						|
 */
 | 
						|
 | 
						|
namespace Symfony\Component\VarDumper\Server;
 | 
						|
 | 
						|
use Symfony\Component\VarDumper\Cloner\Data;
 | 
						|
use Symfony\Component\VarDumper\Dumper\ContextProvider\ContextProviderInterface;
 | 
						|
 | 
						|
/**
 | 
						|
 * Forwards serialized Data clones to a server.
 | 
						|
 *
 | 
						|
 * @author Maxime Steinhausser <maxime.steinhausser@gmail.com>
 | 
						|
 */
 | 
						|
class Connection
 | 
						|
{
 | 
						|
    private string $host;
 | 
						|
 | 
						|
    /**
 | 
						|
     * @var resource|null
 | 
						|
     */
 | 
						|
    private $socket;
 | 
						|
 | 
						|
    /**
 | 
						|
     * @param string                     $host             The server host
 | 
						|
     * @param ContextProviderInterface[] $contextProviders Context providers indexed by context name
 | 
						|
     */
 | 
						|
    public function __construct(
 | 
						|
        string $host,
 | 
						|
        private array $contextProviders = [],
 | 
						|
    ) {
 | 
						|
        if (!str_contains($host, '://')) {
 | 
						|
            $host = 'tcp://'.$host;
 | 
						|
        }
 | 
						|
 | 
						|
        $this->host = $host;
 | 
						|
    }
 | 
						|
 | 
						|
    public function getContextProviders(): array
 | 
						|
    {
 | 
						|
        return $this->contextProviders;
 | 
						|
    }
 | 
						|
 | 
						|
    public function write(Data $data): bool
 | 
						|
    {
 | 
						|
        $socketIsFresh = !$this->socket;
 | 
						|
        if (!$this->socket = $this->socket ?: $this->createSocket()) {
 | 
						|
            return false;
 | 
						|
        }
 | 
						|
 | 
						|
        $context = ['timestamp' => microtime(true)];
 | 
						|
        foreach ($this->contextProviders as $name => $provider) {
 | 
						|
            $context[$name] = $provider->getContext();
 | 
						|
        }
 | 
						|
        $context = array_filter($context);
 | 
						|
        $encodedPayload = base64_encode(serialize([$data, $context]))."\n";
 | 
						|
 | 
						|
        set_error_handler(static fn () => null);
 | 
						|
        try {
 | 
						|
            if (-1 !== stream_socket_sendto($this->socket, $encodedPayload)) {
 | 
						|
                return true;
 | 
						|
            }
 | 
						|
            if (!$socketIsFresh) {
 | 
						|
                stream_socket_shutdown($this->socket, \STREAM_SHUT_RDWR);
 | 
						|
                fclose($this->socket);
 | 
						|
                $this->socket = $this->createSocket();
 | 
						|
            }
 | 
						|
            if (-1 !== stream_socket_sendto($this->socket, $encodedPayload)) {
 | 
						|
                return true;
 | 
						|
            }
 | 
						|
        } finally {
 | 
						|
            restore_error_handler();
 | 
						|
        }
 | 
						|
 | 
						|
        return false;
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * @return resource|null
 | 
						|
     */
 | 
						|
    private function createSocket()
 | 
						|
    {
 | 
						|
        set_error_handler(static fn () => null);
 | 
						|
        try {
 | 
						|
            return stream_socket_client($this->host, $errno, $errstr, 3) ?: null;
 | 
						|
        } finally {
 | 
						|
            restore_error_handler();
 | 
						|
        }
 | 
						|
    }
 | 
						|
}
 |