mirror of
https://github.com/danog/tgseclib.git
synced 2024-12-11 16:49:41 +01:00
342 lines
9.4 KiB
PHP
342 lines
9.4 KiB
PHP
<?php
|
|
|
|
/**
|
|
* Pure-PHP implementation of SCP.
|
|
*
|
|
* PHP version 5
|
|
*
|
|
* The API for this library is modeled after the API from PHP's {@link http://php.net/book.ftp FTP extension}.
|
|
*
|
|
* Here's a short example of how to use this library:
|
|
* <code>
|
|
* <?php
|
|
* include 'vendor/autoload.php';
|
|
*
|
|
* $ssh = new \phpseclib\Net\SSH2('www.domain.tld');
|
|
* if (!$ssh->login('username', 'password')) {
|
|
* exit('bad login');
|
|
* }
|
|
* $scp = new \phpseclib\Net\SCP($ssh);
|
|
*
|
|
* $scp->put('abcd', str_repeat('x', 1024*1024));
|
|
* ?>
|
|
* </code>
|
|
*
|
|
* @category Net
|
|
* @package SCP
|
|
* @author Jim Wigginton <terrafrost@php.net>
|
|
* @copyright 2010 Jim Wigginton
|
|
* @license http://www.opensource.org/licenses/mit-license.html MIT License
|
|
* @link http://phpseclib.sourceforge.net
|
|
*/
|
|
|
|
namespace phpseclib\Net;
|
|
|
|
use phpseclib\Exception\FileNotFoundException;
|
|
use phpseclib\Common\Functions\Strings;
|
|
use phpseclib\Common\Functions\Objects;
|
|
|
|
/**
|
|
* Pure-PHP implementations of SCP.
|
|
*
|
|
* @package SCP
|
|
* @author Jim Wigginton <terrafrost@php.net>
|
|
* @access public
|
|
*/
|
|
class SCP
|
|
{
|
|
/**#@+
|
|
* @access public
|
|
* @see \phpseclib\Net\SCP::put()
|
|
*/
|
|
/**
|
|
* Reads data from a local file.
|
|
*/
|
|
const SOURCE_LOCAL_FILE = 1;
|
|
/**
|
|
* Reads data from a string.
|
|
*/
|
|
const SOURCE_STRING = 2;
|
|
/**#@-*/
|
|
|
|
/**#@+
|
|
* @access private
|
|
* @see \phpseclib\Net\SCP::_send()
|
|
* @see \phpseclib\Net\SCP::_receive()
|
|
*/
|
|
/**
|
|
* SSH1 is being used.
|
|
*/
|
|
const MODE_SSH1 = 1;
|
|
/**
|
|
* SSH2 is being used.
|
|
*/
|
|
const MODE_SSH2 = 2;
|
|
/**#@-*/
|
|
|
|
/**
|
|
* SSH Object
|
|
*
|
|
* @var object
|
|
* @access private
|
|
*/
|
|
private $ssh;
|
|
|
|
/**
|
|
* Packet Size
|
|
*
|
|
* @var int
|
|
* @access private
|
|
*/
|
|
private $packet_size;
|
|
|
|
/**
|
|
* Mode
|
|
*
|
|
* @var int
|
|
* @access private
|
|
*/
|
|
private $mode;
|
|
|
|
/**
|
|
* Default Constructor.
|
|
*
|
|
* Connects to an SSH server
|
|
*
|
|
* @param \phpseclib\Net\SSH1|\phpseclib\Net\SSH2 $ssh
|
|
* @return \phpseclib\Net\SCP
|
|
* @access public
|
|
*/
|
|
public function __construct($ssh)
|
|
{
|
|
if ($ssh instanceof SSH2) {
|
|
$this->mode = self::MODE_SSH2;
|
|
} elseif ($ssh instanceof SSH1) {
|
|
$this->packet_size = 50000;
|
|
$this->mode = self::MODE_SSH1;
|
|
} else {
|
|
return;
|
|
}
|
|
|
|
$this->ssh = $ssh;
|
|
}
|
|
|
|
/**
|
|
* Uploads a file to the SCP server.
|
|
*
|
|
* By default, \phpseclib\Net\SCP::put() does not read from the local filesystem. $data is dumped directly into $remote_file.
|
|
* So, for example, if you set $data to 'filename.ext' and then do \phpseclib\Net\SCP::get(), you will get a file, twelve bytes
|
|
* long, containing 'filename.ext' as its contents.
|
|
*
|
|
* Setting $mode to self::SOURCE_LOCAL_FILE will change the above behavior. With self::SOURCE_LOCAL_FILE, $remote_file will
|
|
* contain as many bytes as filename.ext does on your local filesystem. If your filename.ext is 1MB then that is how
|
|
* large $remote_file will be, as well.
|
|
*
|
|
* Currently, only binary mode is supported. As such, if the line endings need to be adjusted, you will need to take
|
|
* care of that, yourself.
|
|
*
|
|
* @param string $remote_file
|
|
* @param string $data
|
|
* @param int $mode
|
|
* @param callable $callback
|
|
* @throws \phpseclib\Exception\FileNotFoundException if you're uploading via a file and the file doesn't exist
|
|
* @return bool
|
|
* @access public
|
|
*/
|
|
public function put($remote_file, $data, $mode = self::SOURCE_STRING, $callback = null)
|
|
{
|
|
if (!isset($this->ssh)) {
|
|
return false;
|
|
}
|
|
|
|
if (!$this->ssh->exec('scp -t ' . escapeshellarg($remote_file), false)) { // -t = to
|
|
return false;
|
|
}
|
|
|
|
$temp = $this->receive();
|
|
if ($temp !== chr(0)) {
|
|
return false;
|
|
}
|
|
|
|
if ($this->mode == self::MODE_SSH2) {
|
|
$this->packet_size = Objects::getVar($this->ssh, 'packet_size_client_to_server')[SSH2::CHANNEL_EXEC] - 4;
|
|
}
|
|
|
|
$remote_file = basename($remote_file);
|
|
|
|
if ($mode == self::SOURCE_STRING) {
|
|
$size = strlen($data);
|
|
} else {
|
|
if (!is_file($data)) {
|
|
throw new FileNotFoundException("$data is not a valid file");
|
|
}
|
|
|
|
$fp = @fopen($data, 'rb');
|
|
if (!$fp) {
|
|
return false;
|
|
}
|
|
$size = filesize($data);
|
|
}
|
|
|
|
$this->send('C0644 ' . $size . ' ' . $remote_file . "\n");
|
|
|
|
$temp = $this->receive();
|
|
if ($temp !== chr(0)) {
|
|
return false;
|
|
}
|
|
|
|
$sent = 0;
|
|
while ($sent < $size) {
|
|
$temp = $mode & self::SOURCE_STRING ? substr($data, $sent, $this->packet_size) : fread($fp, $this->packet_size);
|
|
$this->send($temp);
|
|
$sent+= strlen($temp);
|
|
|
|
if (is_callable($callback)) {
|
|
call_user_func($callback, $sent);
|
|
}
|
|
}
|
|
$this->close();
|
|
|
|
if ($mode != self::SOURCE_STRING) {
|
|
fclose($fp);
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
/**
|
|
* Downloads a file from the SCP server.
|
|
*
|
|
* Returns a string containing the contents of $remote_file if $local_file is left undefined or a boolean false if
|
|
* the operation was unsuccessful. If $local_file is defined, returns true or false depending on the success of the
|
|
* operation
|
|
*
|
|
* @param string $remote_file
|
|
* @param string $local_file
|
|
* @return mixed
|
|
* @access public
|
|
*/
|
|
public function get($remote_file, $local_file = false)
|
|
{
|
|
if (!isset($this->ssh)) {
|
|
return false;
|
|
}
|
|
|
|
if (!$this->ssh->exec('scp -f ' . escapeshellarg($remote_file), false)) { // -f = from
|
|
return false;
|
|
}
|
|
|
|
$this->send("\0");
|
|
|
|
if (!preg_match('#(?<perms>[^ ]+) (?<size>\d+) (?<name>.+)#', rtrim($this->receive()), $info)) {
|
|
return false;
|
|
}
|
|
|
|
$this->send("\0");
|
|
|
|
$size = 0;
|
|
|
|
if ($local_file !== false) {
|
|
$fp = @fopen($local_file, 'wb');
|
|
if (!$fp) {
|
|
return false;
|
|
}
|
|
}
|
|
|
|
$content = '';
|
|
while ($size < $info['size']) {
|
|
$data = $this->receive();
|
|
// SCP usually seems to split stuff out into 16k chunks
|
|
$size+= strlen($data);
|
|
|
|
if ($local_file === false) {
|
|
$content.= $data;
|
|
} else {
|
|
fputs($fp, $data);
|
|
}
|
|
}
|
|
|
|
$this->close();
|
|
|
|
if ($local_file !== false) {
|
|
fclose($fp);
|
|
return true;
|
|
}
|
|
|
|
return $content;
|
|
}
|
|
|
|
/**
|
|
* Sends a packet to an SSH server
|
|
*
|
|
* @param string $data
|
|
* @access private
|
|
*/
|
|
private function send($data)
|
|
{
|
|
switch ($this->mode) {
|
|
case self::MODE_SSH2:
|
|
Objects::callFunc($this->ssh, 'send_channel_packet', [SSH2::CHANNEL_EXEC, $data]);
|
|
break;
|
|
case self::MODE_SSH1:
|
|
$data = pack('CNa*', NET_SSH1_CMSG_STDIN_DATA, strlen($data), $data);
|
|
Objects::callFunc($this->ssh, 'send_binary_packet', [$data]);
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Receives a packet from an SSH server
|
|
*
|
|
* @return string
|
|
* @throws \UnexpectedValueException on receipt of an unexpected packet
|
|
* @access private
|
|
*/
|
|
private function receive()
|
|
{
|
|
switch ($this->mode) {
|
|
case self::MODE_SSH2:
|
|
return Objects::callFunc($this->ssh, 'get_channel_packet', [SSH2::CHANNEL_EXEC, true]);
|
|
case self::MODE_SSH1:
|
|
if (!Objects::getVar($this->ssh, 'bitmap')) {
|
|
return false;
|
|
}
|
|
while (true) {
|
|
$response = Objects::getFunc($this->ssh, 'get_binary_packet');
|
|
switch ($response[SSH1::RESPONSE_TYPE]) {
|
|
case NET_SSH1_SMSG_STDOUT_DATA:
|
|
if (strlen($response[SSH1::RESPONSE_DATA]) < 4) {
|
|
return false;
|
|
}
|
|
extract(unpack('Nlength', $response[SSH1::RESPONSE_DATA]));
|
|
return Strings::shift($response[SSH1::RESPONSE_DATA], $length);
|
|
case NET_SSH1_SMSG_STDERR_DATA:
|
|
break;
|
|
case NET_SSH1_SMSG_EXITSTATUS:
|
|
Objects::callFunc($this->ssh, 'send_binary_packet', [chr(NET_SSH1_CMSG_EXIT_CONFIRMATION)]);
|
|
fclose(Objects::getVar($this->ssh, 'fsock'));
|
|
Objects::setVar($this->ssh, 'bitmap', 0);
|
|
return false;
|
|
default:
|
|
throw new \UnexpectedValueException('Unknown packet received');
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Closes the connection to an SSH server
|
|
*
|
|
* @access private
|
|
*/
|
|
private function close()
|
|
{
|
|
switch ($this->mode) {
|
|
case self::MODE_SSH2:
|
|
Objects::callFunc($this->ssh, 'close_channel', [SSH2::CHANNEL_EXEC, true]);
|
|
break;
|
|
case self::MODE_SSH1:
|
|
Objects::callFunc($this->ssh, 'disconnect');
|
|
}
|
|
}
|
|
}
|