package main

import (
 "fmt"
 "log"
 "os"
 "os/exec"
 "strings"
)

func main() {
 const CAP_PATH = "/tmp/tg_cap.pcap" // Caminho temporário para o arquivo de captura pcap
 const CAP_TEXT = "/tmp/tg_text.txt" // Caminho temporário para o arquivo de texto com informações
 const CAP_DURATION = "5"            // Duração da captura em segundos

 // Obter o endereço IP externo do dispositivo
 ipCmd := exec.Command("curl", "-s", "icanhazip.com")
 ipOutput, err := ipCmd.Output()
 if err != nil {
  log.Fatal("Falha ao obter o endereço IP:", err)
 }
 MY_IP := strings.TrimSpace(string(ipOutput))

 // Verificar se o Wireshark está instalado
 _, err = exec.LookPath("tshark")
 if err != nil {
  log.Println("[-] Wireshark não foi encontrado. Tente instalar o Wireshark primeiro.")
  log.Println("[+] Debian baseado: sudo apt-get install -y tshark")
  log.Println("[+] Baseado em RedHat: sudo yum install -y tshark")
  os.Exit(1)
 }

 fmt.Println("[+] Descobrindo o Endereço IP do Usuário no Telegram usando Golang")
 fmt.Println("[+] Iniciando a captura de tráfego. Aguarde", CAP_DURATION, "segundos...")

 // Iniciar a captura de tráfego com o Wireshark
 captureCmd := exec.Command("tshark", "-w", CAP_PATH, "-a", "duration:"+CAP_DURATION)
 captureOutput, err := captureCmd.CombinedOutput()
 if err != nil {
  log.Fatal("Erro na captura de tráfego:", err)
 }

 fmt.Println("[+] Tráfego capturado.")

 // Converter o arquivo pcap para um arquivo de texto legível
 convertCmd := exec.Command("tshark", "-r", CAP_PATH)
 convertOutput, err := convertCmd.Output()
 if err != nil {
  log.Fatal("Erro ao converter o arquivo pcap para texto:", err)
 }

 err = os.WriteFile(CAP_TEXT, convertOutput, 0644)
 if err != nil {
  log.Fatal("Erro ao escrever o arquivo de texto:", err)
 }

 fmt.Println("[+] Arquivo pcap convertido com sucesso.")

 // Verificar se o tráfego do Telegram está presente no arquivo de texto
 if strings.Contains(string(convertOutput), "STUN 106") {
  fmt.Println("[+] Tráfego do Telegram foi encontrado.")

  // Extrair o endereço IP do texto
  extractCmd := exec.Command("cat", CAP_TEXT, "|", "grep", "STUN 106", "|", "sed", "'s/^.*XOR-MAPPED-ADDRESS: //'", "|", "awk", "'{match($0,/[0-9]+\\.[0-9]+\\.[0-9]+\\.[0-9]+/); ip = substr($0,RSTART,RLENGTH); print ip}' | awk '!seen[$0]++'")
  extractOutput, err := extractCmd.Output()
  if err != nil {
   log.Fatal("Erro ao extrair o endereço IP:", err)
  }

  TG_OUT := strings.TrimSpace(string(extractOutput))
  IP_1 := strings.Fields(TG_OUT)[0]
  IP_2 := strings.Fields(TG_OUT)[1]

  var IP string

  // Verificar se o endereço IP é o nosso ou do destinatário
  if MY_IP == IP_1 {
   IP = IP_2
  } else if MY_IP == IP_2 {
   IP = IP_1
  } else {
   IP = "[-] Desculpe. Endereço IP não encontrado."
   os.Exit(1)
  }

  // Obter informações de host para o endereço IP
  hostCmd := exec.Command("host", IP)
  hostOutput, err := hostCmd.Output()
  if err != nil {
   log.Fatal("Erro ao obter informações de host:", err)
  }

  fmt.Println("[+]")
  fmt.Println("[+] Endereço IP:", IP)
  fmt.Println("[+] Host:", strings.TrimSpace(string(hostOutput)))
  fmt.Println("[+]")

  // Limpar os arquivos temporários
  err = os.Remove(CAP_PATH)
  if err != nil {
   log.Fatal("Erro na limpeza:", err)
  }

  err = os.Remove(CAP_TEXT)
  if err != nil {
   log.Fatal("Erro na limpeza:", err)
  }

  fmt.Println("[+] Limpeza concluída.")
 } else {
  fmt.Println("[-] Tráfego do Telegram não foi encontrado.")
  fmt.Println("[!]")
  fmt.Println("[!] Execute este script somente >>>APÓS<<< a resposta.")
  fmt.Println("[!]")
  os.Exit(1)
 }

 fmt.Println("[?]")
 fmt.Print("[?] Executar whois", IP, "? (S/N): ")

 // Verificar se o usuário deseja executar o comando whois
 var answer string
 fmt.Scanln(&answer)

 if strings.ToUpper(answer) == "S" {
  whoisCmd := exec.Command("whois", IP)
  whoisOutput, err := whoisCmd.Output()
  if err != nil {
   log.Fatal("Erro ao executar o comando whois:", err)
  }

  fmt.Println(string(whoisOutput))
 } else {
  fmt.Println("[+] Tchau tchau!")
  os.Exit(0)
 }
}