Function main

main.go:45–154

()

}

func main() {
	// 显示大字标题
	showTitle()

	// 获取本机IP
	localIP, err := getLocalIP()
	if err != nil {
		printError(fmt.Sprintf("获取本机IP失败: %v", err))
		localIP = "127.0.0.1" // 默认值
	}

	// 询问是否使用本机IP
	useLocalIP := askYesNo(fmt.Sprintf("本机IP为：%s，是否使用该IP？", localIP), true)
	var targetIP string
	if useLocalIP {
		targetIP = localIP
	} else {
		fmt.Print("请输入要使用的IP地址: ")
		targetIP = getStringInput()
		if net.ParseIP(targetIP) == nil {
			printError("无效的IP地址格式，使用默认IP")
			targetIP = localIP
		}
	}

	// 询问是否使用/24段
	use24Subnet := askYesNo("是否使用/24段？", true)
	var scanTarget string
	if use24Subnet {
		scanTarget = targetIP + "/24"
	} else {
		fmt.Print("请输入子网掩码位数 (如: /20, /16): ")
		maskInput := getStringInput()
		if maskInput == "" {
			scanTarget = targetIP + "/24"
			printInfo("使用默认/24段")
		} else {
			// 处理用户输入，确保以/开头
			if !strings.HasPrefix(maskInput, "/") {
				maskInput = "/" + maskInput
			}
			
			// 验证掩码位数是否有效
			if isValidMask(maskInput) {
				// 计算网络地址
				networkAddr, err := calculateNetworkAddress(targetIP, maskInput)
				if err != nil {
					printError("计算网络地址失败，使用默认/24段")
					scanTarget = targetIP + "/24"
				} else {
					scanTarget = networkAddr + maskInput
					printInfo(fmt.Sprintf("计算得到网段: %s", scanTarget))
				}
			} else {
				printError("无效的子网掩码位数，使用默认/24段")
				scanTarget = targetIP + "/24"
			}
		}