0

场景:我需要一个函数来获取STDOUT通过 SSH 异步运行的命令。这有多种用途,包括(最重要的是)通过 SSH 读取文件。这个函数的一个重要特性是它是异步的,因此我可以显示从服务器返回的输出(或估计文件下载进度)。ssh_move()这与用于下载文件的常用方法不同。

function ssh_exec($dsn, $cmd, $return=true, $size_est=null){
    $BUFFER_SIZE = $return ? (1024 * 1024) : 1;
    debug('ssh2_exec '.$cmd);
    $stream = ssh2_exec(ssh_open($dsn), $cmd);
    debug('stream_set_blocking');
    stream_set_blocking($stream, true);
    debug('ssh2_fetch_stream');
    $stream_out = ssh2_fetch_stream($stream, SSH2_STREAM_STDIO);
    stream_set_blocking($stream_out, true);
    debug('stream_get_contents');
    $data = ''; $stime = $oldtime = microtime(true); $data_len = 0;
    if(!$return){
        write_message("\033[0m".'  Execution Output:'.PHP_EOL.'    ');
    }
    while(!feof($stream_out)){
        $buff = fread($stream_out, $BUFFER_SIZE);
        if($buff===false)throw new Exception('Unexpected result from fread().');
        if($buff===''){
            debug('Empty result from fread()...breaking.');
            break;
        }
        $data .= $buff;
        if($return){
            $buff_len = strlen($buff);
            $data_len += $buff_len;
            $newtime = microtime(true);
            debugo('stream_get_contents '.bytes_to_human($data_len)
                .' @ '.bytes_to_human($buff_len / ($newtime - $oldtime)).'/s'
                .' ['.($size_est ? number_format($data_len / $size_est * 100, 2) : '???').'%]');
            $oldtime = $newtime;
        }else{
            echo str_replace(PHP_EOL, PHP_EOL.'    ', $buff);
        }
    }
    if($return){
        debugo('stream_get_contents Transferred '.bytes_to_human(strlen($data)).' in '.number_format(microtime(true) - $stime, 2).'s');
        return $data;
    }
}

用法:该函数的使用如下:

$dsn = 'ssh2.exec://root:pass@host/';
$size = ssh_size($dsn, 'bigfile.zip');
$zip = ssh_exec($dsn, 'cat bigfile.zip', true, $size);

注1:一些非标准函数的解释:

  • debug($message)- 将调试消息写入控制台。
  • ssh_open($dsn)- 接受一个 SSH URI 并返回一个 SSH 连接句柄。
  • bytes_to_human($bytes)- 将字节数转换为人类可读的格式(例如:6gb)
  • debugo($message)- 相同debug()但覆盖最后一行。

注2:该参数$size_est用于进度指示器;通常你会先获取文件大小,然后尝试下载它(如我的示例所示)。它是可选的,因此当您只想运行 SSH 命令时可以忽略它。

问题:通过 运行相同的下载操作scp root@host:/bigfile.zip ./,我的速度高达 1mb/s,而这个脚本似乎限制为 70kb/s。我想知道为什么以及如何改进这一点。

编辑:另外,我想知道/如果$BUFFER_SIZE有什么不同。

4

1 回答 1

0

您应该能够使用phpseclib(一个纯 PHP SSH 实现)来执行此操作。例如。

$ssh->exec('cat bigfile.zip', false);

while (true) {
    $temp = $this->_get_channel_packet(NET_SSH2_CHANNEL_EXEC);
    if (is_bool($temp)) {
        break;
    }
    echo $temp;
}

实际上也可能更快。

于 2013-05-02T15:25:08.010 回答