route.py

# coding:utf-8
import subprocess
import re
import ip2asn
import logging


class TracerouteLib:
    # TODO:这里的LIST是一股脑存进去 主函数里面再次去重 相当抽象 需要重构

    GOOD_ROUTE_LIST = []

    def __init__(self, ip2asn_db_path):
        self.i2a = ip2asn.IP2ASN(ip2asn_db_path)
        self.good_route_map = {
            4538: "CERNET",
            4837: "联通4837",
            9929: "联通9929",
            4809: "电信CN2",
            10099: "联通CUG",
            23764: "电信CTG",
            58807: "移动CMIN2"
        }

    def parse_traceroute_output(self, output):
        ip_pattern = re.compile(r'\b(?:[0-9]{1,3}\.){3}[0-9]{1,3}\b')
        return ip_pattern.findall(output)

    def execute_traceroute(self, target_host):
        command = ['tracert', '-d', '-w', '1', target_host]
        try:
            result = subprocess.run(command, capture_output=True, text=True, check=True)
            return result.stdout
        except subprocess.CalledProcessError as e:
            raise Exception(f"Traceroute failed for {target_host}: {e}")

    def get_traceroute_ips(self, target_host):
        output = self.execute_traceroute(target_host)
        return self.parse_traceroute_output(output) if output else []

    def traceroute_with_good_route_check(self, target_host, logger):
        logger.info(f"Starting traceroute for {target_host}")
        ips = self.get_traceroute_ips(target_host)
        if ips:
            good_routes_found = set()
            for ip in ips:
                try:
                    asn_data = self.i2a.lookup_address(ip)
                    asn = int(asn_data['ASN'])
                    owner = asn_data['owner']
                    route_name = self.good_route_map.get(asn, None)  # 使用None代替'Unknown ASN'作为默认值

                    # 记录日志信息
                    if route_name:
                        # 如果是优质ASN，记录优质路由名称
                        logger.info(f"IP: {ip}, ASN: {asn}, Owner: {owner}, Route: {route_name}")
                        good_routes_found.add(route_name)
                        self.GOOD_ROUTE_LIST.append({'ip': target_host, 'route': route_name})
                    else:
                        # 如果不是优质ASN，记录ASN编号
                        logger.info(f"IP: {ip}, ASN: {asn}, Owner: {owner}")
                except Exception as e:
                    logger.error(f"Error looking up IP {ip}: {e}")

            # 记录所有找到的优质路由名称
            if good_routes_found:
                good_routes_message = ", ".join(good_routes_found)
                logger.info(f"====Good Routes Found: {good_routes_message}====")
            else:
                logger.info("====No Good Route Found====")
        else:
            logger.error(f"No IPs found for {target_host}")


def setup_logger(logger_name, log_file):
    logger = logging.getLogger(logger_name)
    logger.setLevel(logging.INFO)

    handler = logging.FileHandler(log_file, encoding='utf-8')
    formatter = logging.Formatter('%(asctime)s - %(levelname)s - %(message)s')
    handler.setFormatter(formatter)
    logger.addHandler(handler)
    return logger


from concurrent.futures import ThreadPoolExecutor


def thread_traceroute(lib, target_host, logger):
    try:
        lib.traceroute_with_good_route_check(target_host, logger)
    except Exception as e:
        logger.error(f"Error for {target_host}: {e}")


def main():
    targets = []  # 这里填入你的目标主机列表
    file = open('cnout', 'r')
    for line in file:
        targets.append(line.strip('\n'))
    file.close()
    ip2asn_db_path = 'ip2asn-v4-u32.tsv'  # 指定IP2ASN数据库文件路径
    log_directory = "logs"  # 日志文件存放目录

    # 确保日志目录存在
    import os
    os.makedirs(log_directory, exist_ok=True)

    # 创建TracerouteLib实例
    lib = TracerouteLib(ip2asn_db_path)

    # 使用线程池执行traceroute操作
    with ThreadPoolExecutor(max_workers=len(targets)) as executor:
        for target in targets:
            logger_name = f"traceroute_{target}"
            log_file = os.path.join(log_directory, f"{target}.log")
            logger = setup_logger(logger_name, log_file)
            executor.submit(thread_traceroute, lib, target, logger)

    unique_dicts = set(frozenset(d.items()) for d in TracerouteLib.GOOD_ROUTE_LIST)

    # 将 frozenset 转换回字典列表
    unique_list_of_dicts = [dict(d) for d in unique_dicts]
    print(unique_list_of_dicts)
    for item in unique_list_of_dicts:
        file = open(item["route"] + '_list.txt', "a+")
        file.write(f"{item['ip']}\n")
        file.close()
        # 同样，因为重复问题 暂时用try-except兜住了
        try:
            targets.remove(item['ip'])
        except ValueError:
            continue

    file = open('normal_list.txt', 'a+')
    for item in targets:
        file.write(f"{item}\n")
    file.close()


if __name__ == "__main__":
    main()