Merge "resourceloader: Clarify operator precedence"

[lhc/web/wiklou.git] / includes / shell / Command.php

diff --git a/includes/shell/Command.php b/includes/shell/Command.php
index 864e69a..fd8f6a0 100644 (file)
--- a/includes/shell/Command.php
+++ b/includes/shell/Command.php
@@ -24,6 +24,8 @@ use Exception;
 use MediaWiki\ProcOpenError;
 use MediaWiki\ShellDisabledError;
 use Profiler;
+use Psr\Log\LoggerAwareTrait;
+use Psr\Log\NullLogger;
 
 /**
  * Class used for executing shell commands
@@ -31,11 +33,22 @@ use Profiler;
  * @since 1.30
  */
 class Command {
+       use LoggerAwareTrait;
+
        /** @var string */
        private $command = '';
 
        /** @var array */
-       private $limits = [];
+       private $limits = [
+               // seconds
+               'time' => 180,
+               // seconds
+               'walltime' => 180,
+               // KB
+               'memory' => 307200,
+               // KB
+               'filesize' => 102400,
+       ];
 
        /** @var string[] */
        private $env = [];
@@ -49,13 +62,20 @@ class Command {
        /** @var bool */
        private $everExecuted = false;
 
+       /** @var string|false */
+       private $cGroup = false;
+
        /**
         * Constructor. Don't call directly, instead use Shell::command()
+        *
+        * @throws ShellDisabledError
         */
        public function __construct() {
                if ( Shell::isDisabled() ) {
                        throw new ShellDisabledError();
                }
+
+               $this->setLogger( new NullLogger() );
        }
 
        /**
@@ -112,11 +132,10 @@ class Command {
         * Sets execution limits
         *
         * @param array $limits Optional array with limits(filesize, memory, time, walltime).
-        *   This overrides the global wgMaxShell* limits.
         * @return $this
         */
        public function limits( array $limits ) {
-               $this->limits = $limits;
+               $this->limits = $limits + $this->limits;
 
                return $this;
        }
@@ -158,6 +177,18 @@ class Command {
                return $this;
        }
 
+       /**
+        * Sets cgroup for this command
+        *
+        * @param string|false $cgroup
+        * @return $this
+        */
+       public function cgroup( $cgroup ) {
+               $this->cGroup = $cgroup;
+
+               return $this;
+       }
+
        /**
         * Executes command. Afterwards, getExitCode() and getOutput() can be used to access execution
         * results.
@@ -168,8 +199,7 @@ class Command {
         * @throws ShellDisabledError
         */
        public function execute() {
-               global $IP, $wgMaxShellMemory, $wgMaxShellFileSize, $wgMaxShellTime,
-                          $wgMaxShellWallClockTime, $wgShellCgroup;
+               global $IP;
 
                $this->everExecuted = true;
 
@@ -197,18 +227,12 @@ class Command {
 
                $useLogPipe = false;
                if ( is_executable( '/bin/bash' ) ) {
-                       $time = intval( isset( $this->limits['time'] ) ? $this->limits['time'] : $wgMaxShellTime );
-                       if ( isset( $this->limits['walltime'] ) ) {
-                               $wallTime = intval( $this->limits['walltime'] );
-                       } elseif ( isset( $this->limits['time'] ) ) {
-                               $wallTime = $time;
-                       } else {
-                               $wallTime = intval( $wgMaxShellWallClockTime );
-                       }
-                       $mem = intval( isset( $this->limits['memory'] ) ? $this->limits['memory'] : $wgMaxShellMemory );
-                       $filesize = intval( isset( $this->limits['filesize'] )
-                               ? $this->limits['filesize']
-                               : $wgMaxShellFileSize );
+                       $time = intval( $this->limits['time'] );
+                       $wallTime = intval( $this->limits['walltime'] );
+                       // for b/c, wall time falls back to time
+                       $wallTime = min( $time, $wallTime );
+                       $mem = intval( $this->limits['memory'] );
+                       $filesize = intval( $this->limits['filesize'] );
 
                        if ( $time > 0 || $mem > 0 || $filesize > 0 || $wallTime > 0 ) {
                                $cmd = '/bin/bash ' . escapeshellarg( "$IP/includes/limit.sh" ) . ' ' .
@@ -216,7 +240,7 @@ class Command {
                                           escapeshellarg(
                                                   "MW_INCLUDE_STDERR=" . ( $this->useStderr ? '1' : '' ) . ';' .
                                                   "MW_CPU_LIMIT=$time; " .
-                                                  'MW_CGROUP=' . escapeshellarg( $wgShellCgroup ) . '; ' .
+                                                  'MW_CGROUP=' . escapeshellarg( $this->cGroup ) . '; ' .
                                                   "MW_MEM_LIMIT=$mem; " .
                                                   "MW_FILE_SIZE_LIMIT=$filesize; " .
                                                   "MW_WALL_CLOCK_LIMIT=$wallTime; " .
@@ -252,7 +276,7 @@ class Command {
                $scoped = Profiler::instance()->scopedProfileIn( __FUNCTION__ . '-' . $profileMethod );
                $proc = proc_open( $cmd, $desc, $pipes );
                if ( !$proc ) {
-                       wfDebugLog( 'exec', "proc_open() failed: $cmd" );
+                       $this->logger->error( "proc_open() failed: {command}", [ 'command' => $cmd ] );
                        throw new ProcOpenError();
                }
                $outBuffer = $logBuffer = '';
@@ -329,7 +353,7 @@ class Command {
                                                $lines = explode( "\n", $logBuffer );
                                                $logBuffer = array_pop( $lines );
                                                foreach ( $lines as $line ) {
-                                                       wfDebugLog( 'exec', $line );
+                                                       $this->logger->info( $line );
                                                }
                                        }
                                }
@@ -369,7 +393,7 @@ class Command {
                }
 
                if ( $logMsg !== false ) {
-                       wfDebugLog( 'exec', "$logMsg: $cmd" );
+                       $this->logger->warning( "$logMsg: {command}", [ 'command' => $cmd ] );
                }
 
                return new Result( $retval, $outBuffer );