判別IP為IPV4或者IPV6 （Go語言）

 

編寫一個函數來驗證輸入的字符串是否是有效的 IPv4 或 IPv6 地址。

如果是有效的 IPv4 地址，返回 "IPv4" ；
如果是有效的 IPv6 地址，返回 "IPv6" ；
如果不是上述類型的 IP 地址，返回 "Neither" 。
IPv4 地址由十進制數和點來表示，每個地址包含 4 個十進制數，其范圍為 0 - 255， 用(".")分割。比如，172.16.254.1；
同時，IPv4 地址內的數不會以 0 開頭。比如，地址 172.16.254.01 是不合法的。
IPv6 地址由 8 組 16 進制的數字來表示，每組表示 16 比特。這些組數字通過 (":")分割。比如,  2001:0db8:85a3:0000:0000:8a2e:0370:7334 是一個有效的地址。而且，我們可以加入一些以 0 開頭的數字，字母可以使用大寫，也可以是小寫。所以， 2001:db8:85a3:0:0:8A2E:0370:7334 也是一個有效的 IPv6 address地址 (即，忽略 0 開頭，忽略大小寫)。然而，我們不能因為某個組的值為 0，而使用一個空的組，以至於出現 (::) 的情況。 比如， 2001:0db8:85a3::8A2E:0370:7334 是無效的 IPv6 地址。同時，在 IPv6 地址中，多余的 0 也是不被允許的。比如， 02001:0db8:85a3:0000:0000:8a2e:0370:7334 是無效的。

示例 1：
輸入：IP = "172.16.254.1"
輸出："IPv4"
解釋：有效的 IPv4 地址，返回 "IPv4"
示例 2：
輸入：IP = "2001:0db8:85a3:0:0:8A2E:0370:7334"
輸出："IPv6"
解釋：有效的 IPv6 地址，返回 "IPv6"
示例 3：
輸入：IP = "256.256.256.256"
輸出："Neither"
解釋：既不是 IPv4 地址，又不是 IPv6 地址
示例 4：
輸入：IP = "2001:0db8:85a3:0:0:8A2E:0370:7334:"
輸出："Neither"
示例 5：
輸入：IP = "1e1.4.5.6"
輸出："Neither"
提示：
IP 僅由英文字母，數字，字符 '.' 和 ':' 組成。

package main

import (
    "fmt"
    "strings"
    "strconv"
)
func main() {
    //a := 0
    //fmt.Scan(&a)
    //fmt.Printf("%d\n", a)
    s :=  "1e1.4.5.6"
    case1 := "2001:0db8:85a3:0:0:8A2E:0370:7334"
    case2 := "2001:0db8:85a3:0:0:8A2E:0370:7334:"
    res := IsIpv4OrIpv6(s)
    fmt.Println(res)
    res1 := IsIpv4OrIpv6(case1)
    fmt.Println(res1)
    res2 := IsIpv4OrIpv6(case2)
    fmt.Println(res2)
    fmt.Printf("Hello World!\n");
}

func IsIpv4OrIpv6(ip string) string{
    if len(ip)<7 {
        return "Neither"
    }
    arrIpv4 := strings.Split(ip,".")
    if len(arrIpv4) == 4{
        //. 判斷IPv4
        for _,val := range arrIpv4{
            if !Judgev4(val){
                return "Neither"
            }
        }
        return "IPv4"
    }
    arrIpv6 := strings.Split(ip,":")
    if len(arrIpv6) == 8{
        // 判斷Ipv6
        for _,val := range arrIpv6{
            if !Judgev6(val){
                return "Neither"
            }
        }
        return "IPv6"
    }
    return "Neither"
    
}
func Judgev4(ips string) bool{
    if len(ips) >3{
        return false
    }
    nums ,err := strconv.Atoi(ips)
    if err!= nil {
        return false
    }
    if nums <0 || nums>255{
        return false
    }
    if len(ips)>1 && ips[0] == '0'{
        return false
    }
    return true
}
func Judgev6(ips string) bool{
    if ips == ""{
        return true
    }
    if len(ips) >4{
        return  false
    }
    for _,val := range ips{
        if !((val>='0'&&val<='9')||(val >='a' &&val <= 'f')||(val >= 'A'&&val<='F')){
            return false
        }
    }
    return true
}