<?php
/***
* Copyright 2016 Igor Dyshlenko
*
* Permission is hereby granted, free of charge, to any person obtaining a copy
* of this software and associated documentation files (the "Software"), to deal
* in the Software without restriction, including without limitation the rights
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
* copies of the Software, and to permit persons to whom the Software is
* furnished to do so, subject to the following conditions:
*
* The above copyright notice and this permission notice shall be included in
* all copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
* SOFTWARE.
*/
/**
* Ssh2Connector
* The class provides basic functions for connecting and exchanging data with
* the ssh2 server.
*
* @author Igor Dyshlenko
* @category Console
* @license https://opensource.org/licenses/MIT MIT
*/
class Ssh2Connector implements ShellConnector {
protected $host, $port, $userName;
protected $callbacks;
protected
$connection = null,
$connected = false,
$connErrCode = null,
$connErrMsg = '',
$connErrLang = null;
protected $shell = null;
protected $logger = null;
protected $writeBuffer = null;
/**
* Constructor
* @param string $host
* @param int $port - default 22.
* @param mixed $logger - PEAR Log class object for logging all events, any
* other value is ignored.
* @throws BadFunctionCallException if ssh2_connect() function doesn't exist.
* @throws LogicException if connect to ssh server is fail.
*/
public function __construct($host, $port=22, $logger=null) {
$this->logger = new LogWrapper($logger);
if (!function_exists('ssh2_connect')) {
$msg = 'Function ssh2_connect doesn\'t exist.';
$this->logger->err(__METHOD__ . ': ' . $msg);
throw new BadFunctionCallException($msg);
}
$this->host = $host;
$this->port = $port;
$this->callbacks = array('disconnect' => array($this, 'ssh_disconnect'));
if (!($this->connection = ssh2_connect($host, $port, null, $this->callbacks))) {
$msg = 'Fail: unable to establish connection to ' . $host . ':' . $port .
"\n" . $this->connErrCode . ': ' . $this->connErrMsg;
$this->logger->err(__METHOD__ . ': ' . $msg);
throw new LogicException($msg);
}
$this->logger->debug(__METHOD__ . ': Connection to ' . $host . ':' . $port . ' established.');
$this->connected = true;
}
public function __destruct() {
$this->logout();
$this->disconnect();
}
/**
* Login function
* @param string $userName - user name (login).
* @param string $pass - password.
* @return bool true if success.
* @throws LogicException if authentication error.
*/
public function login($userName, $pass='') {
if (!ssh2_auth_password($this->connection, $userName, $pass)) {
$msg = 'Fail: unable to authenticate user ' . $userName;
$this->logger->err(__METHOD__ . ': ' . $msg);
throw new LogicException($msg);
}
$this->userName = $userName;
if (!($this->shell = ssh2_shell($this->connection, 'vt102', null, 80, 40, SSH2_TERM_UNIT_CHARS))) {
$msg = 'Fail: unable to establish shell.';
$this->logger->err(__METHOD__ . ': ' . $msg);
throw new LogicException($msg);
}
$this->loggedIn = true;
$this->logger->debug(__METHOD__ . ': user ' . $userName . ' logged in.');
return true;
}
/**
* Logout function
* @return bool true if success, false if fail.
*/
public function logout() {
if ($this->isLoggedIn()) {
fclose($this->shell);
$this->logger->debug(__METHOD__ . ': user ' . $this->userName . ' logged out.');
return true;
}
return false;
}
public function disconnect() {}
/**
* Get "is ssh2 connected" state
* @return bool
*/
public function isConnected() {
return $this->connected;
}
/**
* Get "is logged in" state
* @return bool
*/
public function isLoggedIn() {
return is_resource($this->shell);
}
/**
* Callback "disconnect" function for ssh2_connect().
* @param int $reason - Error Number
* @param string $message - Error message
* @param mixed $language
*/
public function ssh_disconnect($reason, $message, $language) {
$this->connected = false;
$this->connErrCode = $reason;
$this->connErrMsg = $message;
$this->connErrLang = $language;
$this->logger->debug(__METHOD__ . ': SSH2 connection ' . $this->userName . '@' . $this->host .
':' . $this->port . ' closed. ' . $reason . ': ' . $message);
}
/**
* Get Error Message returned ssh2_connect() function.
* @return string error message if error, empty string ('') otherwise
*/
public function getError() {
return $this->connErrMsg;
}
/**
* Get Error Number returned ssh2_connect() function.
* @return mixed - int error code if error, NULL otherwise
*/
public function getErrno() {
return $this->connErrCode;
}
/**
* Read character from stream.
* @return mixed - string (readed character) or FALSE if EOF or error.
*/
public function read() {
if (!($this->isLoggedIn() && $this->isConnected()) || feof($this->shell)){
$this->logger->debug(__METHOD__ . ': SSH2 connection ' . $this->userName . '@' . $this->host .
':' . $this->port . ' closed or logged out or feof().');
return false;
}
$char = fread($this->shell, 1);
if ($char === '' && !feof($this->shell)) {
usleep(100000);
$char = fread($this->shell, 1);
}
if ($char === false) {
$this->logger->debug(__METHOD__ . ': Error read from stream.');
}
return $char;
}
/**
* Write data to ssh connection.
* @param string $data - data for write
* @return int - number of written chars or FALSE if error
*/
public function write($data) {
$total = strlen($data);
$written = $n = 0;
$buf = $data;
while ($written < $total){
$buf = substr($buf,$n);
if (($n = fwrite($this->shell,$buf,4096)) === false) {
if (feof($this->shell)) {
$this->logger->debug(__METHOD__ . ': ' . $this->userName . '@' . $this->host . ' disconnected.');
break;
} else {
$msg = 'Error writing to socket.';
$this->logger->err(__METHOD__ . ': ' . $msg);
throw new RuntimeException($msg);
}
}
$written += $n;
}
fflush($this->shell);
return !$this->connected || feof($this->shell) ? false : $written;
}
}
|