题解 | #验证IP地址#

package main

import (
	"strconv"
	"strings"
)

func validIPAddress(queryIP string) string {
    // 以 "." 为分隔符，尝试将输入的字符串分割成四部分
	if sp := strings.Split(queryIP, "."); len(sp) == 4 {
        // 对每一部分进行验证
		for _, s := range sp {
            // 如果部分长度大于1且以0开头，则返回 "Neither"
			if len(s) > 1 && s[0] == '0' {
				return "Neither"
			}
            // 将部分转换为整数，如果转换失败或者值大于255，则返回 "Neither"
			if v, err := strconv.Atoi(s); err != nil || v > 255 {
				return "Neither"
			}
		}
        // 如果通过以上验证，则返回 "IPv4"
		return "IPv4"
	}
    // 以 ":" 为分隔符，尝试将输入的字符串分割成八部分
	if sp := strings.Split(queryIP, ":"); len(sp) == 8 {
        // 对每一部分进行验证
		for _, s := range sp {
            // 如果部分长度大于4，则返回 "Neither"
			if len(s) > 4 {
				return "Neither"
			}
            // 将部分解析为 16 进制的无符号整数，如果解析失败，则返回 "Neither"
			if _, err := strconv.ParseUint(s, 16, 64); err != nil {
				return "Neither"
			}
		}
        // 如果通过以上验证，则返回 "IPv6"
		return "IPv6"
	}
    // 如果输入既不满足 IPv4 的条件，也不满足 IPv6 的条件，则返回 "Neither"
	return "Neither"
}