[+] #13 Adding a current session prompt setting showing what session …

…the user is interacting with.

lib/cli/dispatcher/command_dispatcher.go

@@ -12,6 +12,8 @@ import (
 
 type Dispatcher struct{}
 
+var ReadLineInstance *readline.Instance
+
 func parseInput(input string) (string, []string) {
 	methods := reflection.GetAllMethods(Dispatcher{})
 	args := strings.Split(strings.TrimSpace(input), " ")
@@ -58,7 +60,8 @@ func Run() {
 	completer.SetChildren(children)
 
 	// Construct the IO
-	l, err := readline.NewEx(&readline.Config{
+	var err error
+	ReadLineInstance, err = readline.NewEx(&readline.Config{
 		Prompt:              context.Ctx.CommandPrompt,
 		HistoryFile:         "/tmp/platypus.history",
 		AutoComplete:        completer,
@@ -71,13 +74,13 @@ func Run() {
 	if err != nil {
 		panic(err)
 	}
-	defer l.Close()
+	// defer l.Close()
 
-	log.Logger.SetOutput(l.Stderr())
+	log.Logger.SetOutput(ReadLineInstance.Stderr())
 
 	// Command loop
 	for {
-		line, err := l.Readline()
+		line, err := ReadLineInstance.Readline()
 		if err == readline.ErrInterrupt {
 			if len(line) == 0 {
 				break

lib/cli/dispatcher/jump.go

@@ -6,6 +6,7 @@ import (
 
 	"github.com/WangYihang/Platypus/lib/context"
 	"github.com/WangYihang/Platypus/lib/util/log"
+	"github.com/fatih/color"
 )
 
 func (dispatcher Dispatcher) Jump(args []string) {
@@ -19,6 +20,25 @@ func (dispatcher Dispatcher) Jump(args []string) {
 			if strings.HasPrefix(client.Hash, strings.ToLower(args[0])) {
 				context.Ctx.Current = client
 				log.Success("The current interactive shell is set to: %s", client.FullDesc())
+
+				// Update prompt
+				// BUG:
+				// The prompt will set only at the `Jump` command once.
+				// If we jump to a client before the os & user is detected
+				// So the prompt will be:
+				// (Unknown) 127.0.0.1:43802 [unknown] »
+				var user string
+				if client.User == "" {
+					user = "unknown"
+				} else {
+					user = client.User
+				}
+				ReadLineInstance.SetPrompt(color.CyanString(
+					"(%s) %s [%s] » ",
+					client.OS.String(),
+					client.Conn.RemoteAddr().String(),
+					user,
+				))
 				return
 			}
 		}

lib/context/server.go

@@ -185,13 +185,10 @@ func (s *TCPServer) AsTable() {
 			(*s).Port,
 			len((*s).Clients),
 		))
-		t.AppendHeader(table.Row{"ID", "Hash", "Network", "OS", "User", "Time"})
+		t.AppendHeader(table.Row{"Hash", "Network", "OS", "User", "Time"})
 
-		i := 0
 		for chash, client := range s.Clients {
-			i++
 			t.AppendRow([]interface{}{
-				i,
 				chash,
 				client.Conn.RemoteAddr().String(),
 				client.OS.String(),