up

Author: Pnkcaht

Valkan/Internal/discovery/discovery.go

@@ -0,0 +1,156 @@
+package discovery
+
+import (
+	"context"
+	"fmt"
+	"net"
+	"time"
+
+	"golang.org/x/net/icmp"
+	"golang.org/x/net/ipv4"
+)
+
+// DiscoveryResult armazena o resultado da descoberta de um host
+type DiscoveryResult struct {
+	IP     string
+	Alive  bool
+	Method string
+}
+
+// RunDiscovery executa descoberta em uma rede no formato CIDR usando método "icmp" ou "tcp"
+// Agora aceita portaTCP como parâmetro, caso o método seja "tcp"
+func RunDiscovery(ctx context.Context, cidr string, method string, timeout time.Duration, tcpPort int) ([]DiscoveryResult, error) {
+	ips, err := ExpandCIDR(cidr)
+	if err != nil {
+		return nil, fmt.Errorf("CIDR inválido: %v", err)
+	}
+
+	results := DiscoverHostsConcurrently(ctx, ips, method, timeout, tcpPort)
+	return results, nil
+}
+
+// PingICMP envia um pacote ICMP Echo Request para um IP
+func PingICMP(ip string, timeout time.Duration) bool {
+	c, err := icmp.ListenPacket("ip4:icmp", "0.0.0.0")
+	if err != nil {
+		return false // Sem permissão para socket raw
+	}
+	defer c.Close()
+
+	msg := icmp.Message{
+		Type: ipv4.ICMPTypeEcho,
+		Code: 0,
+		Body: &icmp.Echo{
+			ID:   1234,
+			Seq:  1,
+			Data: []byte("HELLO-PING"),
+		},
+	}
+	msgBytes, err := msg.Marshal(nil)
+	if err != nil {
+		return false
+	}
+
+	dst, err := net.ResolveIPAddr("ip4", ip)
+	if err != nil {
+		return false
+	}
+
+	if _, err := c.WriteTo(msgBytes, dst); err != nil {
+		return false
+	}
+
+	c.SetReadDeadline(time.Now().Add(timeout))
+
+	buf := make([]byte, 1500)
+	n, _, err := c.ReadFrom(buf)
+	if err != nil {
+		return false
+	}
+
+	rm, err := icmp.ParseMessage(1, buf[:n])
+	if err != nil {
+		return false
+	}
+
+	return rm.Type == ipv4.ICMPTypeEchoReply
+}
+
+// PingTCP tenta conectar em uma porta TCP
+func PingTCP(ip string, port int, timeout time.Duration) bool {
+	addr := fmt.Sprintf("%s:%d", ip, port)
+	conn, err := net.DialTimeout("tcp", addr, timeout)
+	if err != nil {
+		return false
+	}
+	conn.Close()
+	return true
+}
+
+// ExpandCIDR retorna todos os IPs dentro de um CIDR
+func ExpandCIDR(cidr string) ([]string, error) {
+	ip, ipnet, err := net.ParseCIDR(cidr)
+	if err != nil {
+		return nil, err
+	}
+
+	var ips []string
+	for ip := ip.Mask(ipnet.Mask); ipnet.Contains(ip); incIP(ip) {
+		ipCopy := make(net.IP, len(ip))
+		copy(ipCopy, ip)
+		ips = append(ips, ipCopy.String())
+	}
+
+	// Remove endereço de rede e broadcast
+	if len(ips) > 2 {
+		return ips[1 : len(ips)-1], nil
+	}
+	return ips, nil
+}
+
+// incIP incrementa o IP
+func incIP(ip net.IP) {
+	for j := len(ip) - 1; j >= 0; j-- {
+		ip[j]++
+		if ip[j] > 0 {
+			break
+		}
+	}
+}
+
+// DiscoverHostsConcurrently faz a varredura concorrente
+func DiscoverHostsConcurrently(ctx context.Context, ips []string, method string, timeout time.Duration, tcpPort int) []DiscoveryResult {
+	results := make([]DiscoveryResult, 0, len(ips))
+	resultsChan := make(chan DiscoveryResult, len(ips))
+
+	for _, ip := range ips {
+		go func(ip string) {
+			alive := false
+
+			if method == "icmp" {
+				alive = PingICMP(ip, timeout)
+			} else if method == "tcp" {
+				alive = PingTCP(ip, tcpPort, timeout)
+			}
+
+			resultsChan <- DiscoveryResult{
+				IP:     ip,
+				Alive:  alive,
+				Method: method,
+			}
+		}(ip)
+	}
+
+	for i := 0; i < len(ips); i++ {
+		select {
+		case res := <-resultsChan:
+			if res.Alive {
+				results = append(results, res)
+			}
+		case <-ctx.Done():
+			return results
+		}
+	}
+
+	return results
+}

Valkan/Internal/ui/terminal.go

@@ -11,6 +11,7 @@ import (
 	"time"
 
 	"github.com/Vyzer9/Valkan/Valkan/Internal/detection"
+	"github.com/Vyzer9/Valkan/Valkan/Internal/discovery" // importe o discovery aqui
 	"github.com/Vyzer9/Valkan/Valkan/Internal/plugins"
 	"github.com/Vyzer9/Valkan/Valkan/Internal/scanner"
 )
@@ -92,8 +93,9 @@ func ShowMenu() {
 		fmt.Println()
 		fmt.Println(Yellow + Bold + "Menu:" + Reset)
 		fmt.Println(Yellow + "1) Scanner" + Reset)
-		fmt.Println(Yellow + "2) Help" + Reset)
-		fmt.Println(Yellow + "3) Sair" + Reset)
+		fmt.Println(Yellow + "2) Discovery" + Reset)
+		fmt.Println(Yellow + "3) Help" + Reset)
+		fmt.Println(Yellow + "4) Sair" + Reset)
 		fmt.Print(Reset + "Escolha uma opção: " + Reset)
 
 		input, _ := reader.ReadString('\n')
@@ -196,9 +198,63 @@ func ShowMenu() {
 			}
 
 		case "2":
-			showHelp()
+			fmt.Println(Yellow + "Iniciando Discovery..." + Reset)
+
+			reader := bufio.NewReader(os.Stdin)
+
+			// Pedir CIDR para o usuário
+			fmt.Print("Digite a rede no formato CIDR (ex: 192.168.1.0/24): ")
+			cidr, _ := reader.ReadString('\n')
+			cidr = strings.TrimSpace(cidr)
+			if cidr == "" {
+				fmt.Println(Red + "CIDR inválido." + Reset)
+				continue
+			}
+
+			// Pedir método
+			fmt.Print("Digite o método (icmp/tcp): ")
+			method, _ := reader.ReadString('\n')
+			method = strings.TrimSpace(method)
+			if method != "icmp" && method != "tcp" {
+				fmt.Println(Red + "Método inválido." + Reset)
+				continue
+			}
+
+			// Se método for TCP, pedir porta
+			tcpPort := 80 // valor padrão
+			if method == "tcp" {
+				fmt.Print("Digite a porta TCP para testar (ex: 80): ")
+				portInput, _ := reader.ReadString('\n')
+				portInput = strings.TrimSpace(portInput)
+				_, err := fmt.Sscanf(portInput, "%d", &tcpPort)
+				if err != nil || tcpPort <= 0 || tcpPort > 65535 {
+					fmt.Println(Red + "Porta inválida." + Reset)
+					continue
+				}
+			}
+
+			ctx := context.Background()
+			timeout := 2 * time.Second
+
+			// Agora com tcpPort como argumento
+			results, err := discovery.RunDiscovery(ctx, cidr, method, timeout, tcpPort)
+			if err != nil {
+				fmt.Println(Red + "Erro ao executar discovery: " + err.Error() + Reset)
+			} else {
+				if len(results) == 0 {
+					fmt.Println(Yellow + "Nenhum host ativo encontrado." + Reset)
+				} else {
+					fmt.Println(Green + "Resultados do Discovery:" + Reset)
+					for _, res := range results {
+						fmt.Printf("IP: %s, Vivo: %t, Método: %s\n", res.IP, res.Alive, res.Method)
+					}
+				}
+			}
 
 		case "3":
+			showHelp()
+
+		case "4":
 			fmt.Println("Saindo...")
 			return
 
@@ -241,10 +297,13 @@ func showHelp() {
    - Busca portas abertas e tenta identificar serviços (apenas TCP).
    - Salva resultados em arquivo "resultados_scan.txt".
 
-2) Help
+2) Discovery
+   - Executa uma varredura de descoberta na rede (ex: ping sweep, hosts ativos).
+
+3) Help
    - Mostra este menu de ajuda.
 
-3) Sair
+4) Sair
    - Fecha o programa.
 
 Dicas:

go.mod

@@ -14,7 +14,8 @@ require (
 	github.com/rivo/uniseg v0.4.7 // indirect
 	github.com/spf13/cobra v1.10.1 // indirect
 	github.com/spf13/pflag v1.0.9 // indirect
-	golang.org/x/sys v0.29.0 // indirect
-	golang.org/x/term v0.28.0 // indirect
-	golang.org/x/text v0.21.0 // indirect
+	golang.org/x/net v0.44.0 // indirect
+	golang.org/x/sys v0.36.0 // indirect
+	golang.org/x/term v0.35.0 // indirect
+	golang.org/x/text v0.29.0 // indirect
 )

go.sum

@@ -44,6 +44,8 @@ golang.org/x/net v0.10.0/go.mod h1:0qNGK6F8kojg2nk9dLZ2mShWaEBan6FAoqfSigmmuDg=
 golang.org/x/net v0.15.0/go.mod h1:idbUs1IY1+zTqbi8yxTbhexhEEk5ur9LInksu6HrEpk=
 golang.org/x/net v0.21.0/go.mod h1:bIjVDfnllIU7BJ2DNgfnXvpSvtn8VRwhlsaeUTyUS44=
 golang.org/x/net v0.25.0/go.mod h1:JkAGAh7GEvH74S6FOH42FLoXpXbE/aqXSrIQjXgsiwM=
+golang.org/x/net v0.44.0 h1:evd8IRDyfNBMBTTY5XRF1vaZlD+EmWx6x8PkhR04H/I=
+golang.org/x/net v0.44.0/go.mod h1:ECOoLqd5U3Lhyeyo/QDCEVQ4sNgYsqvCZ722XogGieY=
 golang.org/x/sync v0.0.0-20190423024810-112230192c58/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
 golang.org/x/sync v0.0.0-20220722155255-886fb9371eb4/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
 golang.org/x/sync v0.1.0/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
@@ -63,6 +65,8 @@ golang.org/x/sys v0.17.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA=
 golang.org/x/sys v0.20.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA=
 golang.org/x/sys v0.29.0 h1:TPYlXGxvx1MGTn2GiZDhnjPA9wZzZeGKHHmKhHYvgaU=
 golang.org/x/sys v0.29.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA=
+golang.org/x/sys v0.36.0 h1:KVRy2GtZBrk1cBYA7MKu5bEZFxQk4NIDV6RLVcC8o0k=
+golang.org/x/sys v0.36.0/go.mod h1:OgkHotnGiDImocRcuBABYBEXf8A9a87e/uXjp9XT3ks=
 golang.org/x/telemetry v0.0.0-20240228155512-f48c80bd79b2/go.mod h1:TeRTkGYfJXctD9OcfyVLyj2J3IxLnKwHJR8f4D8a3YE=
 golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo=
 golang.org/x/term v0.0.0-20210927222741-03fcf44c2211/go.mod h1:jbD1KX2456YbFQfuXm/mYQcufACuNUgVhRMnK/tPxf8=
@@ -73,6 +77,8 @@ golang.org/x/term v0.17.0/go.mod h1:lLRBjIVuehSbZlaOtGMbcMncT+aqLLLmKrsjNrUguwk=
 golang.org/x/term v0.20.0/go.mod h1:8UkIAJTvZgivsXaD6/pH6U9ecQzZ45awqEOzuCvwpFY=
 golang.org/x/term v0.28.0 h1:/Ts8HFuMR2E6IP/jlo7QVLZHggjKQbhu/7H0LJFr3Gg=
 golang.org/x/term v0.28.0/go.mod h1:Sw/lC2IAUZ92udQNf3WodGtn4k/XoLyZoh8v/8uiwek=
+golang.org/x/term v0.35.0 h1:bZBVKBudEyhRcajGcNc3jIfWPqV4y/Kt2XcoigOWtDQ=
+golang.org/x/term v0.35.0/go.mod h1:TPGtkTLesOwf2DE8CgVYiZinHAOuy5AYUYT1lENIZnA=
 golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=
 golang.org/x/text v0.3.3/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ=
 golang.org/x/text v0.3.7/go.mod h1:u+2+/6zg+i71rQMx5EYifcz6MCKuco9NR6JIITiCfzQ=
@@ -83,6 +89,8 @@ golang.org/x/text v0.14.0/go.mod h1:18ZOQIKpY8NJVqYksKHtTdi31H5itFRjB5/qKTNYzSU=
 golang.org/x/text v0.15.0/go.mod h1:18ZOQIKpY8NJVqYksKHtTdi31H5itFRjB5/qKTNYzSU=
 golang.org/x/text v0.21.0 h1:zyQAAkrwaneQ066sspRyJaG9VNi/YJ1NfzcGB3hZ/qo=
 golang.org/x/text v0.21.0/go.mod h1:4IBbMaMmOPCJ8SecivzSH54+73PCFmPWxNTLm+vZkEQ=
+golang.org/x/text v0.29.0 h1:1neNs90w9YzJ9BocxfsQNHKuAT4pkghyXc4nhZ6sJvk=
+golang.org/x/text v0.29.0/go.mod h1:7MhJOA9CD2qZyOKYazxdYMF85OwPdEr9jTtBpO7ydH4=
 golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=
 golang.org/x/tools v0.0.0-20191119224855-298f0cb1881e/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=
 golang.org/x/tools v0.1.12/go.mod h1:hNGJHUnrk76NpqgfD5Aqm5Crs+Hm0VOH/i9J2+nxYbc=