summaryrefslogtreecommitdiffstats
path: root/library/Pdfexport/ShellCommand.php
diff options
context:
space:
mode:
Diffstat (limited to '')
-rw-r--r--library/Pdfexport/ShellCommand.php148
1 files changed, 148 insertions, 0 deletions
diff --git a/library/Pdfexport/ShellCommand.php b/library/Pdfexport/ShellCommand.php
new file mode 100644
index 0000000..c669773
--- /dev/null
+++ b/library/Pdfexport/ShellCommand.php
@@ -0,0 +1,148 @@
+<?php
+
+/* Icinga PDF Export | (c) 2018 Icinga GmbH | GPLv2 */
+
+namespace Icinga\Module\Pdfexport;
+
+class ShellCommand
+{
+ /** @var string Command to execute */
+ protected $command;
+
+ /** @var int Exit code of the command */
+ protected $exitCode;
+
+ /** @var resource Process resource */
+ protected $resource;
+
+ /**
+ * Create a new command
+ *
+ * @param string $command The command to execute
+ * @param bool $escape Whether to escape the command
+ */
+ public function __construct($command, $escape = true)
+ {
+ $command = (string) $command;
+
+ $this->command = $escape ? escapeshellcmd($command) : $command;
+ }
+
+ /**
+ * Get the exit code of the command
+ *
+ * @return int
+ */
+ public function getExitCode()
+ {
+ return $this->exitCode;
+ }
+
+ /**
+ * Get the status of the command
+ *
+ * @return object
+ */
+ public function getStatus()
+ {
+ $status = (object) proc_get_status($this->resource);
+ if ($status->running === false && $this->exitCode === null) {
+ // The exit code is only valid the first time proc_get_status is
+ // called in terms of running false, hence we capture it
+ $this->exitCode = $status->exitcode;
+ }
+
+ return $status;
+ }
+
+ /**
+ * Execute the command
+ *
+ * @return object
+ *
+ * @throws \Exception
+ */
+ public function execute()
+ {
+ if ($this->resource !== null) {
+ throw new \Exception('Command already started');
+ }
+
+ $descriptors = [
+ ['pipe', 'r'], // stdin
+ ['pipe', 'w'], // stdout
+ ['pipe', 'w'] // stderr
+ ];
+
+ $this->resource = proc_open(
+ $this->command,
+ $descriptors,
+ $pipes
+ );
+
+ if (! is_resource($this->resource)) {
+ throw new \Exception(sprintf(
+ "Can't fork '%s'",
+ $this->command
+ ));
+ }
+
+ $namedpipes = (object) [
+ 'stdin' => &$pipes[0],
+ 'stdout' => &$pipes[1],
+ 'stderr' => &$pipes[2]
+ ];
+
+ fclose($namedpipes->stdin);
+
+ $read = [$namedpipes->stderr, $namedpipes->stdout];
+ $origRead = $read;
+ $write = null; // stdin not handled
+ $except = null;
+ $stdout = '';
+ $stderr = '';
+
+ stream_set_blocking($namedpipes->stdout, 0); // non-blocking
+ stream_set_blocking($namedpipes->stderr, 0);
+
+ while (stream_select($read, $write, $except, 0, 20000) !== false) {
+ foreach ($read as $pipe) {
+ if ($pipe === $namedpipes->stdout) {
+ $stdout .= stream_get_contents($pipe);
+ }
+
+ if ($pipe === $namedpipes->stderr) {
+ $stderr .= stream_get_contents($pipe);
+ }
+ }
+
+ foreach ($origRead as $i => $str) {
+ if (feof($str)) {
+ unset($origRead[$i]);
+ }
+ }
+
+ if (empty($origRead)) {
+ break;
+ }
+
+ // Reset pipes
+ $read = $origRead;
+ }
+
+ fclose($namedpipes->stderr);
+ fclose($namedpipes->stdout);
+
+ $exitCode = proc_close($this->resource);
+ if ($this->exitCode === null) {
+ $this->exitCode = $exitCode;
+ }
+
+ $this->resource = null;
+
+ return (object) [
+ 'stdout' => $stdout,
+ 'stderr' => $stderr
+ ];
+ }
+}