kill.go

package commands

import (
	"errors"
	"fmt"
	"io"

	"github.com/NHAS/reverse_ssh/internal/server/users"
	"github.com/NHAS/reverse_ssh/internal/terminal"
	"github.com/NHAS/reverse_ssh/internal/terminal/autocomplete"
	"github.com/NHAS/reverse_ssh/pkg/logger"
)

type kill struct {
	log logger.Logger
}

func (k *kill) ValidArgs() map[string]string {
	return map[string]string{"y": "Do not prompt for confirmation before killing clients"}
}

func (k *kill) Run(user *users.User, tty io.ReadWriter, line terminal.ParsedLine) error {

	if len(line.Arguments) != 1 {
		return errors.New(k.Help(false))
	}

	connections, err := user.SearchClients(line.Arguments[0].Value())
	if err != nil {
		return err
	}

	if len(connections) == 0 {
		return fmt.Errorf("No clients matched '%s'", line.Arguments[0].Value())
	}

	if !line.IsSet("y") {

		fmt.Fprintf(tty, "Kill %d clients? [N/y] ", len(connections))

		if term, ok := tty.(*terminal.Terminal); ok {
			term.EnableRaw()
		}

		b := make([]byte, 1)
		_, err := tty.Read(b)
		if err != nil {
			if term, ok := tty.(*terminal.Terminal); ok {
				term.DisableRaw()
			}
			return err
		}
		if term, ok := tty.(*terminal.Terminal); ok {
			term.DisableRaw()
		}

		if !(b[0] == 'y' || b[0] == 'Y') {
			return fmt.Errorf("\nUser did not enter y/Y, aborting")
		}

		fmt.Fprint(tty, "\n")
	}

	killedClients := 0
	for id, serverConn := range connections {
		serverConn.SendRequest("kill", false, nil)

		if len(connections) == 1 {
			return fmt.Errorf("%s killed", id)
		}
		killedClients++
	}

	return fmt.Errorf("%d connections killed", killedClients)
}

func (k *kill) Expect(line terminal.ParsedLine) []string {
	if len(line.Arguments) <= 1 {
		return []string{autocomplete.RemoteId}
	}
	return nil
}

func (k *kill) Help(explain bool) string {
	if explain {
		return "Stop the execute of the rssh client."
	}

	return terminal.MakeHelpText(k.ValidArgs(),
		"kill <remote_id>",
		"kill <glob pattern>",
		"Stop the execute of the rssh client.",
	)
}

func Kill(log logger.Logger) *kill {
	return &kill{
		log: log,
	}
}