VirusTotal API URL扫描结果获取:正确处理ID的关键

VirusTotal API URL扫描结果获取:正确处理ID的关键

本教程详细介绍了如何使用Python通过VirusTotal API进行URL扫描。文章重点阐述了在提交URL进行扫描后获取的分析ID与查询扫描结果所需的URL ID之间的区别与正确处理方法,旨在帮助用户避免常见的“Wrong URL id”错误,确保能准确检索到URL扫描报告。

1. VirusTotal API 简介与URL扫描流程

virustotal是一个免费的在线服务,用于分析可疑文件和url,以检测病毒、蠕虫、木马和其他类型的恶意软件。其api允许开发者集成virustotal的功能到自己的应用程序中,实现自动化扫描和报告检索。

使用VirusTotal API扫描URL通常包括两个主要步骤:

提交URL进行扫描: 向API发送一个POST请求,将待扫描的URL提交给VirusTotal。此操作会返回一个“分析ID”或“提交ID”。获取扫描结果: 使用第一步中获得的特定ID,向API发送一个GET请求,以检索该URL的详细扫描报告。

在实际操作中,许多开发者在第二步获取扫描结果时会遇到Wrong URL id的错误,这通常是因为对ID的处理方式存在误解。

2. 提交URL进行扫描

首先,我们需要使用POST请求将URL提交给VirusTotal。以下是使用Python requests 库实现此操作的示例代码:

import requestsimport jsondef submit_url_for_scan(scan_url, api_key):    """    提交URL到VirusTotal进行扫描。    :param scan_url: 待扫描的URL字符串。    :param api_key: 您的VirusTotal API密钥。    :return: 扫描任务的分析ID (analysis_id) 或 None。    """    url = "https://www.virustotal.com/api/v3/urls"    payload = {"url": scan_url}    headers = {        "accept": "application/json",        "x-apikey": api_key,    }    try:        response = requests.post(url, data=payload, headers=headers)        response.raise_for_status()  # 检查HTTP响应状态码,如果不是2xx则抛出异常        response_json = response.json()        # 从响应中提取分析ID        analysis_id = response_json.get('data', {}).get('id')        if analysis_id:            print(f"成功提交URL: {scan_url},分析ID: {analysis_id}")            return analysis_id        else:            print(f"提交URL失败或未获取到分析ID: {response_json}")            return None    except requests.exceptions.RequestException as e:        print(f"提交URL时发生网络或HTTP错误: {e}")        return None    except json.JSONDecodeError:        print(f"提交URL时响应内容不是有效的JSON: {response.text}")        return None# 示例调用 (请替换为您的实际API密钥)# api_key = "YOUR_VIRUSTOTAL_API_KEY"# analysis_id_example = submit_url_for_scan("https://www.youtube.com/", api_key)# print(f"获得的分析ID: {analysis_id_example}")

执行上述代码后,如果成功,您将获得一个类似 u-dbae2d0204aa489e234eb2f903a0127b17c712386428cab12b86c5f68aa75867-1701503514 的字符串。这个字符串是VirusTotal为您的提交任务生成的“分析ID”或“提交ID”。

3. 正确获取URL扫描结果:ID处理的关键

许多用户会直接将上述获得的 analysis_id 用于查询URL扫描结果的GET请求,从而导致 Wrong URL id 错误。这是因为VirusTotal的URL分析结果查询API(/api/v3/urls/{id})所需的{id}并不是完整的分析ID,而是分析ID中的一个特定部分——通常是去除前缀和时间戳的哈希值部分。

问题根源:提交URL后返回的分析ID格式通常是 u-{哈希值}-{时间戳}。例如:u-dbae2d0204aa489e234eb2f903a0127b17c712386428cab12b86c5f68aa75867-1701503514。然而,用于查询具体URL报告的GET请求,其URL路径中的ID部分(即{id})需要的是中间的哈希值部分,而不是完整的分析ID。

解决方案:我们需要从完整的分析ID中提取出中间的哈希值部分。一种简单有效的方法是使用字符串的 split(‘-‘) 方法,并取其第二个元素(索引为1)。

# 假设我们获得的分析ID是full_analysis_id = 'u-dbae2d0204aa489e234eb2f903a0127b17c712386428cab12b86c5f68aa75867-1701503514'# 正确提取用于查询的URL IDurl_id_for_query = full_analysis_id.split('-')[1] # 此时 url_id_for_query 将是 'dbae2d0204aa489e234eb2f903a0127b17c712386428cab12b86c5f68aa75867'

4. 完整的URL扫描与结果获取示例代码

结合上述提交和ID处理逻辑,以下是一个完整的Python函数,用于提交URL并获取其VirusTotal扫描报告:

import requestsimport jsonimport timedef scan_and_get_url_report(scan_url, api_key, wait_time=10, max_retries=5):    """    提交URL到VirusTotal进行扫描,并获取其报告。    :param scan_url: 待扫描的URL字符串。    :param api_key: 您的VirusTotal API密钥。    :param wait_time: 每次重试之间的等待时间(秒)。    :param max_retries: 最大重试次数。    :return: URL扫描报告的JSON数据,如果失败则为None。    """    # 1. 提交URL进行扫描    submit_url = "https://www.virustotal.com/api/v3/urls"    payload = {"url": scan_url}    headers = {        "accept": "application/json",        "x-apikey": api_key,    }    print(f"正在提交URL '{scan_url}' 进行扫描...")    try:        post_response = requests.post(submit_url, data=payload, headers=headers)        post_response.raise_for_status()        post_response_json = post_response.json()        full_analysis_id = post_response_json.get('data', {}).get('id')        if not full_analysis_id:            print(f"提交URL失败或未获取到分析ID: {post_response_json}")            return None        print(f"成功获取分析ID: {full_analysis_id}")        # 2. 从分析ID中提取用于查询的URL ID        # 格式通常为 u-{哈希值}-{时间戳}        # 我们需要的是中间的哈希值部分        try:            url_id_for_query = full_analysis_id.split('-')[1]            print(f"提取出用于查询的URL ID: {url_id_for_query}")        except IndexError:            print(f"无法从分析ID '{full_analysis_id}' 中正确提取URL ID。ID格式可能不符合预期。")            return None    except requests.exceptions.RequestException as e:        print(f"提交URL时发生网络或HTTP错误: {e}")        return None    except json.JSONDecodeError:        print(f"提交URL时响应内容不是有效的JSON: {post_response.text}")        return None    # 3. 循环查询扫描结果    get_report_url = f"https://www.virustotal.com/api/v3/urls/{url_id_for_query}"    for attempt in range(max_retries):        print(f"尝试获取报告 (第 {attempt + 1}/{max_retries} 次)...")        try:            get_response = requests.get(get_report_url, headers=headers)            get_response.raise_for_status()            report_json = get_response.json()            # 检查报告中是否存在错误,特别是BadRequestError            if 'error' in report_json:                error_message = report_json['error'].get('message', '未知错误')                error_code = report_json['error'].get('code', '未知代码')                print(f"获取报告时API返回错误: {error_code} - {error_message}")                # 如果是 Wrong URL id 错误,通常是ID处理问题,但我们已经处理了                # 可能是报告尚未生成,或URL本身的问题                if error_code == "BadRequestError" and "Wrong URL id" in error_message:                    print("尽管已尝试正确处理ID,但仍收到 'Wrong URL id' 错误。请检查原始分析ID是否有效或VirusTotal是否已处理此URL。")                # 对于某些错误,可能需要重试,但对于Wrong URL id,重试意义不大                # 对于报告未就绪的情况,可以继续等待                if "Not found" in error_message or "still processing" in error_message.lower():                    print(f"报告尚未就绪或正在处理中,等待 {wait_time} 秒后重试...")                    time.sleep(wait_time)                    continue # 继续下一次循环尝试                else:                    return None # 其他错误直接退出            # 如果报告成功返回且没有错误,则返回报告            print("成功获取URL扫描报告。")            return report_json        except requests.exceptions.RequestException as e:            print(f"获取报告时发生网络或HTTP错误: {e}")        except json.JSONDecodeError:            print(f"获取报告时响应内容不是有效的JSON: {get_response.text}")        print(f"等待 {wait_time} 秒后重试...")        time.sleep(wait_time)    print("达到最大重试次数,未能获取URL扫描报告。")    return None# --- 示例调用 ---# 请替换为您的实际API密钥VIRUSTOTAL_API_KEY = "YOUR_VIRUSTOTAL_API_KEY" TARGET_URL = "https://www.youtube.com/" # 替换为您想要扫描的URLif VIRUSTOTAL_API_KEY == "YOUR_VIRUSTOTAL_API_KEY":    print("请将 'YOUR_VIRUSTOTAL_API_KEY' 替换为您的实际VirusTotal API密钥。")else:    report = scan_and_get_url_report(TARGET_URL, VIRUSTOTAL_API_KEY)    if report:        # 打印部分报告内容,例如检测引擎数量        print("n--- 扫描报告摘要 ---")        attributes = report.get('data', {}).get('attributes', {})        last_analysis_stats = attributes.get('last_analysis_stats', {})        print(f"URL: {TARGET_URL}")        print(f"恶意检测数: {last_analysis_stats.get('malicious', 0)}")        print(f"可疑检测数: {last_analysis_stats.get('suspicious', 0)}")        print(f"无害检测数: {last_analysis_stats.get('harmless', 0)}")        print(f"未检测数: {last_analysis_stats.get('undetected', 0)}")        # print(json.dumps(report, indent=2)) # 打印完整报告    else:        print(f"未能获取 {TARGET_URL} 的扫描报告。")

5. 注意事项与最佳实践

API 密钥安全: 您的API密钥是敏感信息,请勿将其硬编码到公开的代码库中,或以不安全的方式共享。建议使用环境变量或配置文件来管理API密钥。错误处理: 在实际应用中,应更健壮地处理API请求可能出现的各种错误,例如网络问题、HTTP状态码非2xx、JSON解析失败、API限速等。示例代码中已包含基本的错误处理。API 限速: VirusTotal API对请求频率有限制。免费公共API通常有较低的限速(例如每分钟4个请求)。如果您的应用需要进行大量扫描,可能需要购买私有API计划。在设计代码时,应考虑加入适当的延迟或重试机制来应对限速。报告延迟: 提交URL后,VirusTotal需要时间来完成扫描。立即查询报告可能会得到“正在处理中”或“未找到”的响应。因此,在获取报告时,通常需要一个循环等待和重试机制,直到报告可用。示例代码中的 wait_time 和 max_retries 参数就是为此目的。官方文档: 始终参考VirusTotal官方API文档(https://developers.virustotal.com/reference/overview)获取最新、最准确的信息。API的行为和ID格式可能会更新。

总结

通过本教程,我们深入探讨了使用Python与VirusTotal API进行URL扫描的关键步骤,特别是解决了在获取扫描结果时常见的Wrong URL id错误。核心在于理解VirusTotal返回的“分析ID”与查询报告所需的“URL ID”之间的差异,并学会通过字符串处理(split(‘-‘)[1])正确提取后者。遵循这些指导和最佳实践,开发者可以更有效地利用VirusTotal API,构建稳定可靠的恶意URL检测系统。

以上就是VirusTotal API URL扫描结果获取:正确处理ID的关键的详细内容,更多请关注创想鸟其它相关文章!

版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容, 请发送邮件至 chuangxiangniao@163.com 举报,一经查实,本站将立刻删除。
发布者:程序猿,转转请注明出处:https://www.chuangxiangniao.com/p/1371306.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2025年12月14日 11:19:50
下一篇 2025年12月14日 11:20:01

相关推荐

发表回复

登录后才能评论
关注微信