#cs ----------------------------------------------------------------------------

 AutoIt Version: 3.3.4.0
 Author:         Joseph Barone 2010-2015

 Script Function:
	Modele pour l'utilisation de la fonction ssh (plink).

#ce ----------------------------------------------------------------------------

; Script Start - Add your code below here

#include <GUIConstantsEx.au3>
#include <WindowsConstants.au3>
#include <ButtonConstants.au3>
#include <EditConstants.au3>
#include <ScrollBarConstants.au3>
#include "SSH_udf.au3"

_SSHStartup()

Global $butt[6]

$Gui = GUICreate("SSH Console",550,400)
$label1 = GUICtrlCreateLabel("Adresse IP:",10,10,150,20) ; ip
$input1 = GUICtrlCreateInput("",10,25,150,20) ; ip
$label2 = GUICtrlCreateLabel("Tcp port:",170,10,150,20) ; port
$input2 = GUICtrlCreateInput("",170,25,60,20) ; port
$label6 = GUICtrlCreateLabel("Protocol:",240,10,40,20) ; port
$input6 = GUICtrlCreateInput("ssh",240,25,60,20) ; port
$butt[0] = GUICtrlCreateButton("Connect0",310,5,80,20,$BS_DEFPUSHBUTTON)
$butt[1] = GUICtrlCreateButton("Connect1",390,5,80,20)
$butt[2] = GUICtrlCreateButton("Connect2",470,5,80,20)
$butt[3] = GUICtrlCreateButton("Connect3",310,25,80,20)
$butt[4] = GUICtrlCreateButton("Connect4",390,25,80,20)
$butt[5] = GUICtrlCreateButton("Connect5",470,25,80,20)

$Checkbox = GUICtrlCreateCheckbox("RSA, Fingerprint, Alert AutoValidate", 330, 45, 205, 25)

$label4 = GUICtrlCreateLabel("Login:",10,45,150,20) ; login
$input4 = GUICtrlCreateInput("",10,60,150,20) ; login
$label5 = GUICtrlCreateLabel("Mot de passe:",170,45,150,20) ; password
$input5 = GUICtrlCreateInput("",170,60,150,20,$ES_PASSWORD) ; password

$label3 = GUICtrlCreateLabel("Commande a envoyer:",10,85,150,20) ; send command
$input3 = GUICtrlCreateInput("",10,100,310,20) ; send commande vers l'hote
$butt2 = GUICtrlCreateButton("Send",330,100,80,20,$BS_DEFPUSHBUTTON)
GUICtrlSetState($butt2,$GUI_DISABLE)

$edit = GUICtrlCreateEdit("",10,130,530,260,$WS_VSCROLL);,$WS_DISABLED))

GUISetState()


Global $ConnectedSocket[6] = [-1,-1,-1,-1,-1,-1]
Global $Auth[6] = [0,0,0,0,0,0]

Global $msg, $recv, $ret

; GUI Message Loop
;==============================================
While 1
	$nmsg = GUIGetMsg()

    If $nmsg = $GUI_EVENT_CLOSE Then ExitLoop

;;; connect part:
	$b = 0
	While 1
	If $nMsg = $butt[$b] Then
		If GUICtrlRead($butt[$b]) = "Connect"&$b Then
			_SSHParametreSet("yesall",GUICtrlRead($Checkbox))
			_SSHParametreSet("protocol",GUICtrlRead($input6))
			;_SSHParametreSet("login",GUICtrlRead($input4))
			;_SSHParametreSet("passwd",GUICtrlRead($input5))
			$ConnectedSocket[$b] = _SSHConnect(GUICtrlRead($input1),GUICtrlRead($input2))
			If $ConnectedSocket[$b] = 0 Then
				MsgBox(0,"Erreur", "Impossible de ce connecter!!")
			Else
				GUICtrlSetData($butt[$b],"Close"&$b)
				GUICtrlSetState($butt2,$GUI_ENABLE)
			EndIf
			DisplayArray($_config)
			ConsoleWrite("nb config:"&$_nbconf&@CRLF)
		Else
			_SSHCloseSocket($ConnectedSocket[$b])
			GUICtrlSetData($butt[$b],"Connect"&$b)
			GUICtrlSetState($butt2,$GUI_DISABLE)
			DisplayArray($_config)
			ConsoleWrite("nb config:"&$_nbconf&@CRLF)
			$Auth[$b] = 0
		EndIf
	EndIf

;;; send part:
	If $nMsg = $butt2 Then
		$ret = _SSHSend( $ConnectedSocket[$b] , GUICtrlRead($input3)&@crlf)
		GUICtrlSetData($input3,"")
		GUICtrlSetState($input3,$GUI_FOCUS)
	EndIf


;;; receive part:
    $recv = _SSHRecv( $ConnectedSocket[$b])
    If $recv <> "" Then
		ConsoleWrite($recv&@CRLF)
		if StringInStr($recv,"login as:") And $Auth[$b] = 0 Then
			ConsoleRead('Login received!'&@CRLF)
			$ret = _SSHSend( $ConnectedSocket[$b] , GUICtrlRead($input4)&@crlf)
			GUICtrlSetData($edit, GUICtrlRead($edit) & $recv & GUICtrlRead($input4))
			$Auth[$b] = 1
		Elseif StringInStr($recv,"password:") And $Auth[$b] = 1 Then
			ConsoleRead('Password received!'&@CRLF)
			$ret = _SSHSend( $ConnectedSocket[$b] , GUICtrlRead($input5)&@crlf)
			$Auth[$b] = 2
		Else
			GUICtrlSetData($edit, GUICtrlRead($edit) & $recv )
		EndIf
	EndIf
	GUICtrlSendMsg($edit, $EM_SCROLL, $SB_PAGEDOWN, 0)
	$b += 1
	If $b = UBound($butt) Then ExitLoop
	WEnd

WEnd

_SSHCloseSocket($ConnectedSocket)
_SSHShutdown()

Exit

Func DisplayArray($array)
	$i = 0
	$j = 0
	if UBound($array,1) = 0 Then Return
	While 1
		ConsoleWrite("(["&UBound($array,1)&"]["&UBound($array,2)&"]) ")
		While 1
			ConsoleWrite("["&$j&"]"&$array[$i][$j]&" ")
			$j += 1
			if UBound($array,2)<=$j Then ExitLoop
		WEnd
		ConsoleWrite(@CRLF)
		$i += 1
		$j = 0
		if UBound($array,1)<=$i Then ExitLoop
	WEnd
EndFunc