Function scanAddress

main.go:228–331

实际的扫描函数

(addr string)

// 实际的扫描函数
func scanAddress(addr string) error {
	printInfo("正在初始化扫描...")

	// 初始化地理位置查询
	geoPaths := []string{
		"Country.mmdb",
		"GeoLite2-Country.mmdb",
		"/usr/share/GeoIP/GeoLite2-Country.mmdb",
		"/var/lib/GeoIP/GeoLite2-Country.mmdb",
		config.Output + ".geo.mmdb",
	}

	var geo *Geo
	var geoErr error
	for _, path := range geoPaths {
		if geo, geoErr = NewGeo(path); geoErr == nil {
			printInfo(fmt.Sprintf("地理位置数据库加载成功: %s", path))
			break
		}
	}

	// 如果没有找到地理位置数据库，尝试自动下载
	if geo == nil {
		printInfo("未找到地理位置数据库，正在尝试自动下载...")

		// 尝试下载到程序目录
		downloadPath := "GeoLite2-Country.mmdb"
		if TryDownloadGeoLite2DB(downloadPath) {
			// 下载成功，尝试加载
			if geo, geoErr = NewGeo(downloadPath); geoErr == nil {
				printInfo(fmt.Sprintf("地理位置数据库下载并加载成功: %s", downloadPath))
			} else {
				printError(fmt.Sprintf("下载的数据库文件加载失败: %v", geoErr))
				printInfo("将跳过地理位置查询")
			}
		} else {
			printInfo("自动下载失败，将跳过地理位置查询")
			printInfo("提示: 可手动下载 GeoLite2-Country.mmdb 文件到程序目录以启用地理位置功能")
		}
	}
	defer func() {
		if geo != nil {
			geo.Close()
		}
	}()

	// 解析主机
	host, err := ParseHost(addr)
	if err != nil {
		return fmt.Errorf("解析地址失败: %v", err)
	}

	var hostChan <-chan Host
	var totalTargets int

	// 根据主机类型创建迭代器和计算总数
	if host.Type == HostTypeIP {
		// 单个IP的无限扫描模式