Go语言实现的一个简单的域名扫描工具

kengengping

[Golang] 纯文本查看 复制代码

package main

import (
	"fmt"
	"net"
	"sync"
)

// 定义一个全局变量用于存储开放的端口
var openPorts []int

// 定义一个互斥锁用于保护共享资源
var mutex sync.Mutex

// scanPort 函数用于扫描指定IP和端口是否开放
func scanPort(ip string, port int) {
	conn, err := net.DialTimeout("tcp", ip+":"+fmt.Sprint(port), time.Second*5)
	if err == nil {
		mutex.Lock()
		openPorts = append(openPorts, port)
		mutex.Unlock()
		conn.Close()
	}
}

// main 函数用于启动多线程扫描
func main() {
	// 目标域名
	domain := "example.com"
	// 获取所有子域名
	subdomains := getSubdomains(domain)
	// 创建多个goroutine进行并发扫描
	var wg sync.WaitGroup
	for _, subdomain := range subdomains {
		wg.Add(1)
		go func(subdomain string) {
			defer wg.Done()
			ip := getIPByDomain(subdomain)
			if ip != "" {
				for port := 80; port <= 65535; port++ {
					scanPort(ip, port)
				}
			}
		}(subdomain)
	}
	wg.Wait()
	// 输出开放的端口
	fmt.Println("Open ports:", openPorts)
}

// getSubdomains 函数用于获取目标域名的所有子域名
func getSubdomains(domain string) []string {
	// 这里仅作示例，实际应用中可以使用第三方库或API获取子域名列表
	return []string{"www", "mail", "ftp"}
}

// getIPByDomain 函数用于根据域名获取对应的IP地址
func getIPByDomain(domain string) string {
	// 这里仅作示例，实际应用中可以使用第三方库或API获取IP地址
	return "192.168.1.1"
}