Upd: Hide sensitive information in output logs

Author: Driver-C

pkg/control/control.go

@@ -3,6 +3,7 @@ package control
 import (
 	"context"
 	"github.com/Driver-C/tryssh/pkg/launcher"
+	"github.com/cheggaaa/pb/v3"
 	"sync"
 	"time"
 )
@@ -17,8 +18,10 @@ const (
 )
 
 func ConcurrencyTryToConnect(concurrency int, connectors []launcher.Connector) []launcher.Connector {
-	var hitConnectors []launcher.Connector
-	var mutex sync.Mutex
+	hitConnectors := make([]launcher.Connector, 0)
+	mutex := new(sync.Mutex)
+	bar := pb.StartNew(len(connectors))
+	bar.Set("prefix", "Attempting:")
 	// If the number of connectors is less than the set concurrency, change the concurrency to the number of connectors
 	if concurrency > len(connectors) {
 		concurrency = len(connectors)
@@ -56,13 +59,16 @@ func ConcurrencyTryToConnect(concurrency int, connectors []launcher.Connector) [
 						mutex.Lock()
 						hitConnectors = append(hitConnectors, connector)
 						mutex.Unlock()
+						bar.Finish()
 						cancelFunc()
 					}
+					bar.Increment()
 				}
 			}
-		}(ctx, cancelFunc, connectorsChan, &wg, &mutex)
+		}(ctx, cancelFunc, connectorsChan, &wg, mutex)
 	}
 	wg.Wait()
+	bar.Finish()
 	cancelFunc()
 	return hitConnectors
 }

pkg/control/scp.go

@@ -84,7 +84,7 @@ func (cc *ScpController) tryCopyWithoutCache(user string) {
 		connectors[i] = l
 	}
 	hitLaunchers := ConcurrencyTryToConnect(cc.concurrency, connectors)
-	if hitLaunchers != nil {
+	if len(hitLaunchers) > 0 {
 		utils.Logger.Infoln("Login succeeded. The cache will be added.\n")
 		hitLauncher := hitLaunchers[0].(*launcher.ScpLauncher)
 		// The new server cache information

pkg/control/ssh.go

@@ -56,7 +56,7 @@ func (sc *SshController) tryLoginWithoutCache(user string) {
 		connectors[i] = l
 	}
 	hitLaunchers := ConcurrencyTryToConnect(sc.concurrency, connectors)
-	if hitLaunchers != nil {
+	if len(hitLaunchers) > 0 {
 		utils.Logger.Infoln("Login succeeded. The cache will be added.\n")
 		hitLauncher := hitLaunchers[0].(*launcher.SshLauncher)
 		// The new server cache information

pkg/launcher/base.go

@@ -81,11 +81,7 @@ func (sc *SshConnector) CreateConnection() (sshClient *ssh.Client, err error) {
 	if err != nil {
 		if strings.Contains(err.Error(), SSHKeyKeyword) {
 			// If it's a public key verification issue, just exit
-			utils.Logger.Fatalf("Unable to connect: %s@%s, Password:%s Cause: %s\n",
-				sc.User, addr, sc.Password, err.Error())
-		} else {
-			utils.Logger.Warnf("Unable to connect: %s@%s, Password:%s Cause: %s\n",
-				sc.User, addr, sc.Password, err.Error())
+			utils.Logger.Fatalf("Unable to connect: %s Cause: %s\n", addr, err.Error())
 		}
 	}
 	return

pkg/launcher/ssh.go

@@ -37,10 +37,7 @@ func (h *SshLauncher) dialServer() (res bool) {
 	if err == nil {
 		utils.Logger.Infoln("[ LOGIN SUCCESSFUL ]\n")
 		utils.Logger.Infoln("User:", sshClient.User())
-		utils.Logger.Infoln("Password:", h.Password)
 		utils.Logger.Infoln("Port:", h.Port)
-		utils.Logger.Infoln("Ssh Server Version:", string(sshClient.ServerVersion()))
-		utils.Logger.Infof("Ssh Client Version: %s\n\n", string(sshClient.ClientVersion()))
 		res = true
 		h.createTerminal(sshClient)
 	} else {