X-Git-Url: https://git.openstreetmap.org./nominatim.git/blobdiff_plain/1108bf7d8679b85b7aa9f829019acf4dc8210dd3..9431e80eb4aa6ebcfa63e5e0d8c0cbbe7569c7c2:/lib/cmd.php?ds=sidebyside diff --git a/lib/cmd.php b/lib/cmd.php index 2fd3c49b..77878c15 100644 --- a/lib/cmd.php +++ b/lib/cmd.php @@ -120,15 +120,6 @@ function showUsage($aSpec, $bExit = false, $sError = false) exit; } -function chksql($oSql, $sMsg = false) -{ - if (PEAR::isError($oSql)) { - fail($sMsg || $oSql->getMessage(), $oSql->userinfo); - } - - return $oSql; -} - function info($sMsg) { echo date('Y-m-d H:i:s == ').$sMsg."\n"; @@ -155,18 +146,20 @@ function repeatWarnings() function runSQLScript($sScript, $bfatal = true, $bVerbose = false, $bIgnoreErrors = false) { // Convert database DSN to psql parameters - $aDSNInfo = DB::parseDSN(CONST_Database_DSN); + $aDSNInfo = \Nominatim\DB::parseDSN(CONST_Database_DSN); if (!isset($aDSNInfo['port']) || !$aDSNInfo['port']) $aDSNInfo['port'] = 5432; - $sCMD = 'psql -p '.$aDSNInfo['port'].' -d '.$aDSNInfo['database']; + $sCMD = 'psql' + .' -p '.escapeshellarg($aDSNInfo['port']) + .' -d '.escapeshellarg($aDSNInfo['database']); if (isset($aDSNInfo['hostspec']) && $aDSNInfo['hostspec']) { - $sCMD .= ' -h ' . $aDSNInfo['hostspec']; + $sCMD .= ' -h ' . escapeshellarg($aDSNInfo['hostspec']); } if (isset($aDSNInfo['username']) && $aDSNInfo['username']) { - $sCMD .= ' -U ' . $aDSNInfo['username']; + $sCMD .= ' -U ' . escapeshellarg($aDSNInfo['username']); } - $procenv = null; + $aProcEnv = null; if (isset($aDSNInfo['password']) && $aDSNInfo['password']) { - $procenv = array_merge(array('PGPASSWORD' => $aDSNInfo['password']), $_ENV); + $aProcEnv = array_merge(array('PGPASSWORD' => $aDSNInfo['password']), $_ENV); } if (!$bVerbose) { $sCMD .= ' -q'; @@ -180,15 +173,19 @@ function runSQLScript($sScript, $bfatal = true, $bVerbose = false, $bIgnoreError 2 => STDERR ); $ahPipes = null; - $hProcess = @proc_open($sCMD, $aDescriptors, $ahPipes, null, $procenv); + $hProcess = @proc_open($sCMD, $aDescriptors, $ahPipes, null, $aProcEnv); if (!is_resource($hProcess)) { fail('unable to start pgsql'); } + if (!$bVerbose) { + fwrite($ahPipes[0], 'set client_min_messages to WARNING;'); + } + while (strlen($sScript)) { - $written = fwrite($ahPipes[0], $sScript); - if ($written <= 0) break; - $sScript = substr($sScript, $written); + $iWritten = fwrite($ahPipes[0], $sScript); + if ($iWritten <= 0) break; + $sScript = substr($sScript, $iWritten); } fclose($ahPipes[0]); $iReturn = proc_close($hProcess); @@ -198,19 +195,21 @@ function runSQLScript($sScript, $bfatal = true, $bVerbose = false, $bIgnoreError } -function runWithEnv($cmd, $env) +function runWithEnv($sCmd, $aEnv) { - $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); + $aFDs = array( + 0 => array('pipe', 'r'), + 1 => STDOUT, + 2 => STDERR + ); + $aPipes = null; + $hProc = @proc_open($sCmd, $aFDs, $aPipes, null, $aEnv); + if (!is_resource($hProc)) { + fail('unable to run command:' . $sCmd); } - fclose($pipes[0]); // no stdin + fclose($aPipes[0]); // no stdin - $stat = proc_close($proc); - return $stat; + $iStat = proc_close($hProc); + return $iStat; }