check_ip

check_ip.py

# -*- coding: utf-8 -*-
#14-6-6
# create by: snower

import struct
from collections import defaultdict
import socket


default_networds = (
    "10.0.0.0/8",
    "172.16.0.0/12",
    "192.168.0.0/16"
)

networks = defaultdict(dict)
masks = []

def load_networks():
    global networks, masks
    networks, masks = defaultdict(dict), []
    try:
        try:
            with open("china_ip_list.txt") as fp:
                for line in fp:
                    if "#" in line or not line.strip():
                        continue
                    info = line.strip().split("/")
                    mask = int(info[1]) if len(info) >= 2 else 32
                    try:
                        network = struct.unpack(">I", socket.inet_aton(info[0]))[0] >> (32 - mask)
                    except:
                        continue
                    networks[mask][network] = True
        except:
            pass

        for line in default_networds:
            info = line.strip().split("/")
            mask = int(info[1]) if len(info) >= 2 else 32
            network = struct.unpack(">I", socket.inet_aton(info[0]))[0] >> (32 - mask)
            networks[mask][network] = True

        masks = sorted(networks.keys())
        for i in range(len(masks)):
            for j in range(i + 1, len(masks)):
                for network in networks[masks[j]]:
                    network = network >> (masks[j] - masks[i])
                    if network in networks[masks[i]]:
                        continue
                    networks[masks[i]][network] = False
    except Exception as e:
        return

def check_ip(ip):
    try:
        ip = struct.unpack(">I", socket.inet_aton(ip))[0]
        for mask in masks:
            network = ip >> (32 - mask)
            if network not in networks[mask]:
                return False
            if networks[mask][network]:
                return True
    except Exception as e:
        return False
    return False