PHP Classes

File: Ssh2Connector.php

Recommend this page to a friend!
  Classes of Igor Dyshlenko   Apc Switch   Ssh2Connector.php   Download  
File: Ssh2Connector.php
Role: Class source
Content type: text/plain
Description: Ssh2Connector class
Class: Apc Switch
Manage an APC Rack PDU over ssh
Author: By
Last change:
Date: 6 years ago
Size: 6,713 bytes
 

Contents

Class file image Download
<?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;
    }
}