Initial Commit
This commit is contained in:
David Stone
179
dependencies/amphp/process/lib/Internal/Windows/Runner.php
vendored
Normal file
179
dependencies/amphp/process/lib/Internal/Windows/Runner.php
vendored
Normal file
@ -0,0 +1,179 @@
|
||||
<?php
|
||||
|
||||
namespace WP_Ultimo\Dependencies\Amp\Process\Internal\Windows;
|
||||
|
||||
use WP_Ultimo\Dependencies\Amp\Deferred;
|
||||
use WP_Ultimo\Dependencies\Amp\Loop;
|
||||
use WP_Ultimo\Dependencies\Amp\Process\Internal\ProcessHandle;
|
||||
use WP_Ultimo\Dependencies\Amp\Process\Internal\ProcessRunner;
|
||||
use WP_Ultimo\Dependencies\Amp\Process\Internal\ProcessStatus;
|
||||
use WP_Ultimo\Dependencies\Amp\Process\ProcessException;
|
||||
use WP_Ultimo\Dependencies\Amp\Process\ProcessInputStream;
|
||||
use WP_Ultimo\Dependencies\Amp\Process\ProcessOutputStream;
|
||||
use WP_Ultimo\Dependencies\Amp\Promise;
|
||||
use const WP_Ultimo\Dependencies\Amp\Process\BIN_DIR;
|
||||
/**
|
||||
* @internal
|
||||
* @codeCoverageIgnore Windows only.
|
||||
*/
|
||||
final class Runner implements ProcessRunner
|
||||
{
|
||||
const FD_SPEC = [
|
||||
["pipe", "r"],
|
||||
// stdin
|
||||
["pipe", "w"],
|
||||
// stdout
|
||||
["pipe", "w"],
|
||||
// stderr
|
||||
["pipe", "w"],
|
||||
];
|
||||
const WRAPPER_EXE_PATH = \PHP_INT_SIZE === 8 ? BIN_DIR . '\\windows\\ProcessWrapper64.exe' : BIN_DIR . '\\windows\\ProcessWrapper.exe';
|
||||
private static $pharWrapperPath;
|
||||
private $socketConnector;
|
||||
private function makeCommand(string $workingDirectory) : string
|
||||
{
|
||||
$wrapperPath = self::WRAPPER_EXE_PATH;
|
||||
// We can't execute the exe from within the PHAR, so copy it out...
|
||||
if (\strncmp($wrapperPath, "phar://", 7) === 0) {
|
||||
if (self::$pharWrapperPath === null) {
|
||||
$fileHash = \hash_file('sha1', self::WRAPPER_EXE_PATH);
|
||||
self::$pharWrapperPath = \sys_get_temp_dir() . "/amphp-process-wrapper-" . $fileHash;
|
||||
if (!\file_exists(self::$pharWrapperPath) || \hash_file('sha1', self::$pharWrapperPath) !== $fileHash) {
|
||||
\copy(self::WRAPPER_EXE_PATH, self::$pharWrapperPath);
|
||||
}
|
||||
}
|
||||
$wrapperPath = self::$pharWrapperPath;
|
||||
}
|
||||
$result = \sprintf('%s --address=%s --port=%d --token-size=%d', \escapeshellarg($wrapperPath), $this->socketConnector->address, $this->socketConnector->port, SocketConnector::SECURITY_TOKEN_SIZE);
|
||||
if ($workingDirectory !== '') {
|
||||
$result .= ' ' . \escapeshellarg('--cwd=' . \rtrim($workingDirectory, '\\'));
|
||||
}
|
||||
return $result;
|
||||
}
|
||||
public function __construct()
|
||||
{
|
||||
$this->socketConnector = new SocketConnector();
|
||||
}
|
||||
/** @inheritdoc */
|
||||
public function start(string $command, string $cwd = null, array $env = [], array $options = []) : ProcessHandle
|
||||
{
|
||||
if (\strpos($command, "\x00") !== \false) {
|
||||
throw new ProcessException("Can't execute commands that contain null bytes.");
|
||||
}
|
||||
$options['bypass_shell'] = \true;
|
||||
$handle = new Handle();
|
||||
$handle->proc = @\proc_open($this->makeCommand($cwd ?? ''), self::FD_SPEC, $pipes, $cwd ?: null, $env ?: null, $options);
|
||||
if (!\is_resource($handle->proc)) {
|
||||
$message = "Could not start process";
|
||||
if ($error = \error_get_last()) {
|
||||
$message .= \sprintf(" Errno: %d; %s", $error["type"], $error["message"]);
|
||||
}
|
||||
throw new ProcessException($message);
|
||||
}
|
||||
$status = \proc_get_status($handle->proc);
|
||||
if (!$status) {
|
||||
\proc_close($handle->proc);
|
||||
throw new ProcessException("Could not get process status");
|
||||
}
|
||||
$securityTokens = \random_bytes(SocketConnector::SECURITY_TOKEN_SIZE * 6);
|
||||
$written = \fwrite($pipes[0], $securityTokens . "\x00" . $command . "\x00");
|
||||
\fclose($pipes[0]);
|
||||
\fclose($pipes[1]);
|
||||
if ($written !== SocketConnector::SECURITY_TOKEN_SIZE * 6 + \strlen($command) + 2) {
|
||||
\fclose($pipes[2]);
|
||||
\proc_close($handle->proc);
|
||||
throw new ProcessException("Could not send security tokens / command to process wrapper");
|
||||
}
|
||||
$handle->securityTokens = \str_split($securityTokens, SocketConnector::SECURITY_TOKEN_SIZE);
|
||||
$handle->wrapperPid = $status['pid'];
|
||||
$handle->wrapperStderrPipe = $pipes[2];
|
||||
$stdinDeferred = new Deferred();
|
||||
$handle->stdioDeferreds[] = $stdinDeferred;
|
||||
$handle->stdin = new ProcessOutputStream($stdinDeferred->promise());
|
||||
$stdoutDeferred = new Deferred();
|
||||
$handle->stdioDeferreds[] = $stdoutDeferred;
|
||||
$handle->stdout = new ProcessInputStream($stdoutDeferred->promise());
|
||||
$stderrDeferred = new Deferred();
|
||||
$handle->stdioDeferreds[] = $stderrDeferred;
|
||||
$handle->stderr = new ProcessInputStream($stderrDeferred->promise());
|
||||
$this->socketConnector->registerPendingProcess($handle);
|
||||
return $handle;
|
||||
}
|
||||
/** @inheritdoc */
|
||||
public function join(ProcessHandle $handle) : Promise
|
||||
{
|
||||
/** @var Handle $handle */
|
||||
$handle->exitCodeRequested = \true;
|
||||
if ($handle->exitCodeWatcher !== null) {
|
||||
Loop::reference($handle->exitCodeWatcher);
|
||||
}
|
||||
return $handle->joinDeferred->promise();
|
||||
}
|
||||
/** @inheritdoc */
|
||||
public function kill(ProcessHandle $handle)
|
||||
{
|
||||
/** @var Handle $handle */
|
||||
\exec('taskkill /F /T /PID ' . $handle->wrapperPid . ' 2>&1', $output, $exitCode);
|
||||
$failStart = \false;
|
||||
if ($handle->childPidWatcher !== null) {
|
||||
Loop::cancel($handle->childPidWatcher);
|
||||
$handle->childPidWatcher = null;
|
||||
$handle->pidDeferred->fail(new ProcessException("The process was killed"));
|
||||
$failStart = \true;
|
||||
}
|
||||
if ($handle->exitCodeWatcher !== null) {
|
||||
Loop::cancel($handle->exitCodeWatcher);
|
||||
$handle->exitCodeWatcher = null;
|
||||
$handle->joinDeferred->fail(new ProcessException("The process was killed"));
|
||||
}
|
||||
$handle->status = ProcessStatus::ENDED;
|
||||
if ($failStart || $handle->stdioDeferreds) {
|
||||
$this->socketConnector->failHandleStart($handle, "The process was killed");
|
||||
}
|
||||
$this->free($handle);
|
||||
}
|
||||
/** @inheritdoc */
|
||||
public function signal(ProcessHandle $handle, int $signo)
|
||||
{
|
||||
throw new ProcessException('Signals are not supported on Windows');
|
||||
}
|
||||
/** @inheritdoc */
|
||||
public function destroy(ProcessHandle $handle)
|
||||
{
|
||||
/** @var Handle $handle */
|
||||
if ($handle->status < ProcessStatus::ENDED && \is_resource($handle->proc)) {
|
||||
try {
|
||||
$this->kill($handle);
|
||||
return;
|
||||
} catch (ProcessException $e) {
|
||||
// ignore
|
||||
}
|
||||
}
|
||||
$this->free($handle);
|
||||
}
|
||||
private function free(Handle $handle)
|
||||
{
|
||||
if ($handle->childPidWatcher !== null) {
|
||||
Loop::cancel($handle->childPidWatcher);
|
||||
$handle->childPidWatcher = null;
|
||||
}
|
||||
if ($handle->exitCodeWatcher !== null) {
|
||||
Loop::cancel($handle->exitCodeWatcher);
|
||||
$handle->exitCodeWatcher = null;
|
||||
}
|
||||
$handle->stdin->close();
|
||||
$handle->stdout->close();
|
||||
$handle->stderr->close();
|
||||
foreach ($handle->sockets as $socket) {
|
||||
if (\is_resource($socket)) {
|
||||
@\fclose($socket);
|
||||
}
|
||||
}
|
||||
if (\is_resource($handle->wrapperStderrPipe)) {
|
||||
@\fclose($handle->wrapperStderrPipe);
|
||||
}
|
||||
if (\is_resource($handle->proc)) {
|
||||
\proc_close($handle->proc);
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user