基于Python实现子域名收集工具

子域名收集是网络安全评估和信息收集的一部分。在Python中，你可以使用多种方法来实现子域名收集工具。以下是一个基于socket和re模块的简单子域名收集工具的示例：

import socket
import re
from urllib.request import urlopen

def resolve_host(name, timeout=5):
    try:
        host = socket.gethostbyname(name)
        return host
    except socket.gaierror:
        return None

def discover_subdomains(domain, timeout=5):
    subdomains = set()
    domain = domain + '.'  # 确保域名以点结尾
    for char in 'abcdefghijklmnopqrstuvwxyz':
        second_level = domain + char
        third_level = second_level + 'www.'
        try:
            # 尝试解析第二级和第三级域名
            resolve_host(second_level, timeout)
            resolve_host(third_level, timeout)
            # 如果能够解析，添加到子域名列表中
            subdomains.add(second_level)
            subdomains.add(third_level)
            # 尝试获取A记录
            a_records = socket.gethostbyname_ex(third_level)[2]
            for a_record in a_records:
                subdomains.add(third_level + '.' + a_record[4])
        except socket.gaierror:
            continue
    return subdomains

def fetch_subdomains(domain, timeout=10):
    subdomains = set()
    try:
        # 尝试从证书透明度日志中获取子域名
        url = f'https://crt.sh/?q=%s&output=json' % domain
        response = urlopen(url, timeout=timeout)
        data = response.read().decode('utf-8')
        matches = re.findall(r'"name":"([^"]+)"', data)
        for match in matches:
            subdomains.add(match + '.' + domain)
    except Exception as e:
        print(f"Error fetching subdomains from CRT: {e}")
    return subdomains

def main(domain):
    # 收集子域名
    subdomains = discover_subdomains(domain)
    subdomains |= fetch_subdomains(domain)
    
    # 去重并排序
    subdomains = sorted(subdomains)
    
    # 输出子域名
    for subdomain in subdomains:
        print(subdomain)

if __name__ == '__main__':
    target_domain = 'example.com'  # 替换为你想要收集子域名的目标域名
    main(target_domain)

这个工具使用了两种方法来收集子域名：

1. 基于DNS解析的子域名发现：通过尝试解析目标域名的所有可能的二级和三级组合，来发现子域名。这种方法简单但可能不够全面，因为它依赖于DNS解析的结果。

2. 从证书透明度日志中获取子域名：通过查询CRT（Certificate Transparency）日志，可以获取到目标域名的所有证书记录，从而发现子域名。这种方法可以发现一些通过DNS解析方法可能发现不了的子域名。

请注意，这个工具仅用于教育和合法的网络安全评估目的。在使用子域名收集工具时，应始终遵守相关法律法规和道德标准，确保你有权进行此类信息收集。未经授权的信息收集可能会违反法律或道德规范。