nuclei/pkg/js/libs/telnet/telnet.go

46 lines
1.1 KiB
Go

package telnet
import (
"context"
"net"
"strconv"
"time"
"github.com/praetorian-inc/fingerprintx/pkg/plugins"
"github.com/praetorian-inc/fingerprintx/pkg/plugins/services/telnet"
"github.com/projectdiscovery/nuclei/v3/pkg/protocols/common/protocolstate"
)
// TelnetClient is a minimal Telnet client for nuclei scripts.
type TelnetClient struct{}
// IsTelnetResponse is the response from the IsTelnet function.
type IsTelnetResponse struct {
IsTelnet bool
Banner string
}
// IsTelnet checks if a host is running a Telnet server.
func (c *TelnetClient) IsTelnet(host string, port int) (IsTelnetResponse, error) {
resp := IsTelnetResponse{}
timeout := 5 * time.Second
conn, err := protocolstate.Dialer.Dial(context.TODO(), "tcp", net.JoinHostPort(host, strconv.Itoa(port)))
if err != nil {
return resp, err
}
defer conn.Close()
telnetPlugin := telnet.TELNETPlugin{}
service, err := telnetPlugin.Run(conn, timeout, plugins.Target{Host: host})
if err != nil {
return resp, err
}
if service == nil {
return resp, nil
}
resp.Banner = service.Metadata().(plugins.ServiceTelnet).ServerData
resp.IsTelnet = true
return resp, nil
}