DDNS/ddns/ip.py

#!/usr/bin/env python
# -*- coding:utf-8 -*-
from re import compile
from os import name as os_name
from socket import socket, getaddrinfo, gethostname, AF_INET, AF_INET6, SOCK_DGRAM
from logging import debug, error

from .util.http import request
from .util.try_run import try_run

# 模块级别的SSL验证配置，默认使用auto模式
ssl_verify = "auto"

# IPV4正则
IPV4_REG = r"((25[0-5]|(2[0-4]|1{0,1}[0-9]){0,1}[0-9])\.){3,3}(25[0-5]|(2[0-4]|1{0,1}[0-9]){0,1}[0-9])"
# IPV6正则
# https://community.helpsystems.com/forums/intermapper/miscellaneous-topics/5acc4fcf-fa83-e511-80cf-0050568460e4
IPV6_REG = r"((([0-9A-Fa-f]{1,4}:){7}([0-9A-Fa-f]{1,4}|:))|(([0-9A-Fa-f]{1,4}:){6}(:[0-9A-Fa-f]{1,4}|((25[0-5]|2[0-4]\d|1\d\d|[1-9]?\d)(\.(25[0-5]|2[0-4]\d|1\d\d|[1-9]?\d)){3})|:))|(([0-9A-Fa-f]{1,4}:){5}(((:[0-9A-Fa-f]{1,4}){1,2})|:((25[0-5]|2[0-4]\d|1\d\d|[1-9]?\d)(\.(25[0-5]|2[0-4]\d|1\d\d|[1-9]?\d)){3})|:))|(([0-9A-Fa-f]{1,4}:){4}(((:[0-9A-Fa-f]{1,4}){1,3})|((:[0-9A-Fa-f]{1,4})?:((25[0-5]|2[0-4]\d|1\d\d|[1-9]?\d)(\.(25[0-5]|2[0-4]\d|1\d\d|[1-9]?\d)){3}))|:))|(([0-9A-Fa-f]{1,4}:){3}(((:[0-9A-Fa-f]{1,4}){1,4})|((:[0-9A-Fa-f]{1,4}){0,2}:((25[0-5]|2[0-4]\d|1\d\d|[1-9]?\d)(\.(25[0-5]|2[0-4]\d|1\d\d|[1-9]?\d)){3}))|:))|(([0-9A-Fa-f]{1,4}:){2}(((:[0-9A-Fa-f]{1,4}){1,5})|((:[0-9A-Fa-f]{1,4}){0,3}:((25[0-5]|2[0-4]\d|1\d\d|[1-9]?\d)(\.(25[0-5]|2[0-4]\d|1\d\d|[1-9]?\d)){3}))|:))|(([0-9A-Fa-f]{1,4}:){1}(((:[0-9A-Fa-f]{1,4}){1,6})|((:[0-9A-Fa-f]{1,4}){0,4}:((25[0-5]|2[0-4]\d|1\d\d|[1-9]?\d)(\.(25[0-5]|2[0-4]\d|1\d\d|[1-9]?\d)){3}))|:))|(:(((:[0-9A-Fa-f]{1,4}){1,7})|((:[0-9A-Fa-f]{1,4}){0,5}:((25[0-5]|2[0-4]\d|1\d\d|[1-9]?\d)(\.(25[0-5]|2[0-4]\d|1\d\d|[1-9]?\d)){3}))|:)))"  # noqa: E501

# 公网IPv4 API列表，按优先级排序
PUBLIC_IPV4_APIS = [
    "https://api.ipify.org",
    "https://ipv4.ddnsip.cn",
    "https://ipinfo.io/ip",
    "https://api-ipv4.ip.sb/ip",
    "http://checkip.amazonaws.com",
]

# 公网IPv6 API列表，按优先级排序
PUBLIC_IPV6_APIS = [
    "https://api6.ipify.org/",
    "https://ipv6.ddnsip.cn",
    "https://api-ipv6.ip.sb/ip",
    "http://ipv6.icanhazip.com",
]


def default_v4():  # 默认连接外网的ipv4
    s = socket(AF_INET, SOCK_DGRAM)
    s.connect(("1.1.1.1", 53))
    ip = s.getsockname()[0]
    s.close()
    return ip


def default_v6():  # 默认连接外网的ipv6
    s = socket(AF_INET6, SOCK_DGRAM)
    s.connect(("1:1:1:1:1:1:1:1", 8))
    ip = s.getsockname()[0]
    s.close()
    return ip


def local_v6(i=0):  # 本地ipv6地址
    info = getaddrinfo(gethostname(), 0, AF_INET6)
    debug(info)
    return info[int(i)][4][0]


def local_v4(i=0):  # 本地ipv4地址
    info = getaddrinfo(gethostname(), 0, AF_INET)
    debug(info)
    return info[int(i)][-1][0]


def _open(url, reg):
    try:
        debug("open: %s", url)
        # IP 模块重试3次
        response = request("GET", url, verify=ssl_verify, retries=2)
        res = response.body
        debug("response: %s", res)
        match = compile(reg).search(res)
        if match:
            return match.group()
        error("No match found in response: %s", res)
    except Exception as e:
        error(e)


def _try_multiple_apis(api_list, reg, ip_type):
    """
    Try multiple API endpoints until one succeeds
    """
    for url in api_list:
        try:
            debug("Trying %s API: %s", ip_type, url)
            result = _open(url, reg)
            if result:
                debug("Successfully got %s from %s: %s", ip_type, url, result)
                return result
            else:
                debug("No valid IP found from %s", url)
        except Exception as e:
            debug("Failed to get %s from %s: %s", ip_type, url, e)
    return None


def public_v4(url=None, reg=IPV4_REG):  # 公网IPV4地址
    if url:
        # 使用指定URL
        return _open(url, reg)
    else:
        # 使用多个API自动重试
        return _try_multiple_apis(PUBLIC_IPV4_APIS, reg, "IPv4")


def public_v6(url=None, reg=IPV6_REG):  # 公网IPV6地址
    if url:
        # 使用指定URL
        return _open(url, reg)
    else:
        # 使用多个API自动重试
        return _try_multiple_apis(PUBLIC_IPV6_APIS, reg, "IPv6")


def _read_network_config():
    # type: () -> str | None
    if os_name == "nt":
        return try_run(["ipconfig"])

    output = try_run(["ip", "address"])
    if output is None:
        output = try_run(["ifconfig"])
    return output


def _ip_regex_match(parrent_regex, match_regex):
    ip_pattern = compile(parrent_regex)
    matcher = compile(match_regex)

    output = _read_network_config()
    for s in (output or "").splitlines(True):
        addr = ip_pattern.search(s)
        if addr and matcher.match(addr.group(1)):
            return addr.group(1)


def regex_v4(reg):  # ipv4 正则提取
    if os_name == "nt":  # Windows: IPv4 xxx: 192.168.1.2
        regex_str = r"IPv4 .*: ((?:\d{1,3}\.){3}\d{1,3})\W"
    else:
        regex_str = r"inet (?:addr\:)?((?:\d{1,3}\.){3}\d{1,3})[\s/]"
    return _ip_regex_match(regex_str, reg)


def regex_v6(reg):  # ipv6 正则提取
    if os_name == "nt":  # Windows: IPv4 xxx: ::1
        regex_str = r"IPv6 .*: ([\:\dabcdef]*)?\W"
    else:
        regex_str = r"inet6 (?:addr\:\s*)?([\:\dabcdef]*)?[\s/%]"
    return _ip_regex_match(regex_str, reg)