X-Git-Url: https://git.openstreetmap.org./nominatim.git/blobdiff_plain/6c1977b448e8b195bf96b6144674ffe0527e79de..1108bf7d8679b85b7aa9f829019acf4dc8210dd3:/lib/cmd.php diff --git a/lib/cmd.php b/lib/cmd.php index a1084938..2fd3c49b 100644 --- a/lib/cmd.php +++ b/lib/cmd.php @@ -16,7 +16,7 @@ function getCmdOpt($aArg, $aSpec, &$aResult, $bExitOnError = false, $bExitOnUnkn $aResult = array(); $bUnknown = false; - $iSize = sizeof($aArg); + $iSize = count($aArg); for ($i = 1; $i < $iSize; $i++) { if (isset($aQuick[$aArg[$i]])) { $aLine = $aQuick[$aArg[$i]]; @@ -134,7 +134,7 @@ function info($sMsg) echo date('Y-m-d H:i:s == ').$sMsg."\n"; } -$aWarnings = []; +$aWarnings = array(); function warn($sMsg) @@ -158,6 +158,16 @@ function runSQLScript($sScript, $bfatal = true, $bVerbose = false, $bIgnoreError $aDSNInfo = DB::parseDSN(CONST_Database_DSN); if (!isset($aDSNInfo['port']) || !$aDSNInfo['port']) $aDSNInfo['port'] = 5432; $sCMD = 'psql -p '.$aDSNInfo['port'].' -d '.$aDSNInfo['database']; + if (isset($aDSNInfo['hostspec']) && $aDSNInfo['hostspec']) { + $sCMD .= ' -h ' . $aDSNInfo['hostspec']; + } + if (isset($aDSNInfo['username']) && $aDSNInfo['username']) { + $sCMD .= ' -U ' . $aDSNInfo['username']; + } + $procenv = null; + if (isset($aDSNInfo['password']) && $aDSNInfo['password']) { + $procenv = array_merge(array('PGPASSWORD' => $aDSNInfo['password']), $_ENV); + } if (!$bVerbose) { $sCMD .= ' -q'; } @@ -170,7 +180,7 @@ function runSQLScript($sScript, $bfatal = true, $bVerbose = false, $bIgnoreError 2 => STDERR ); $ahPipes = null; - $hProcess = @proc_open($sCMD, $aDescriptors, $ahPipes); + $hProcess = @proc_open($sCMD, $aDescriptors, $ahPipes, null, $procenv); if (!is_resource($hProcess)) { fail('unable to start pgsql'); } @@ -186,3 +196,21 @@ function runSQLScript($sScript, $bfatal = true, $bVerbose = false, $bIgnoreError fail("pgsql returned with error code ($iReturn)"); } } + + +function runWithEnv($cmd, $env) +{ + $fds = array(0 => array('pipe', 'r'), + 1 => STDOUT, + 2 => STDERR); + $pipes = null; + $proc = @proc_open($cmd, $fds, $pipes, null, $env); + if (!is_resource($proc)) { + fail('unable to run command:' . $cmd); + } + + fclose($pipes[0]); // no stdin + + $stat = proc_close($proc); + return $stat; +}