AWVS API接口 食用简单方式

在网络安全领域，自动化扫描工具如 AWVS（Acunetix Web Vulnerability Scanner）极大地提升了漏洞检测的效率。通过 AWVS 提供的 API，我们可以方便地管理扫描任务、获取目标等信息。本文将详细介绍如何使用 AWVS API，包括获取所有目标和添加扫描任务的具体代码实现。

环境准备

在开始之前，请确保你已经安装了 requests 和 json 模块。你可以使用以下命令安装 requests：

pip install requests

API 配置

首先，我们需要配置 AWVS API 的相关参数，包括 API 地址和 API 密钥。以下是我们需要用到的基础代码：

import requests
import json
import time

# AWVS API配置
api_url = "https://192.168.126.8:3443/api/v1"
api_key = "你的_api_key"
headers = {
    "X-Auth": api_key,
    "Content-Type": "application/json"
}

请将 "你的_api_key" 替换为你的实际 API 密钥。

获取所有目标

为了获取所有的扫描目标，我们需要编写一个函数 get_all_targets，该函数将分页获取目标数据。代码如下：

# 获取所有目标
def get_all_targets():
    all_targets = []
    page = 1
    limit = 100  # 每页的目标数量

    while True:
        print(f"正在获取第 {page} 页...")
        response = requests.get(f"{api_url}/targets?l={limit}&c={page}", headers=headers, verify=False)
        response.raise_for_status()
        data = response.json()
        
        targets = data.get("targets", [])
        all_targets.extend(targets)
        
        print(f"从第 {page} 页获取了 {len(targets)} 个目标")
        
        pagination = data.get("pagination", {})
        if pagination.get("count", 0) <= page * limit:
            break
        
        page += 1

    total_count = len(all_targets)
    print(f"总共获取到 {total_count} 个目标")
    return all_targets, total_count

函数说明

1. 分页获取目标：通过 while 循环逐页请求目标数据，直到没有更多数据。
2. 异常处理：使用 response.raise_for_status() 检查响应状态，如果不是 2xx 则抛出异常。
3. 返回数据：最后返回所有目标和总目标数量。

添加新的扫描任务

获取目标后，我们可以为这些目标添加扫描任务。下面是 add_scan 函数的实现：

# 添加新的扫描任务
def add_scan(target_id):
    data = {
        "target_id": target_id,
        "profile_id": "11111111-1111-1111-1111-111111111111",  # Full Scan profile
        "schedule": {
            "disable": False,
            "start_date": None,
            "time_sensitive": False
        }
    }
    
    while True:
        try:
            response = requests.post(f"{api_url}/scans", headers=headers, data=json.dumps(data), verify=False)
            response.raise_for_status()  # 如果响应状态码不是 2xx，将引发异常
            return True
        except requests.exceptions.RequestException as e:
            print(f"添加扫描任务时发生错误: {e}")
            print("等待 5 秒后重试...")
            time.sleep(5)

函数说明

1. 任务数据：构造扫描任务的数据，包括目标 ID 和扫描配置。
2. 异常处理：在请求过程中，如果发生错误，会等待 5 秒后重试。

主函数

最后，我们将上述函数整合到主函数中，执行目标获取和扫描任务添加的操作：

# 主函数
def main():
    targets, total_count = get_all_targets()
    print(f"总目标数量: {total_count}")
    
    start_adding = False
    for target in targets:
        target_id = target["target_id"]
        address = target["address"]
        
        if address == "http://appserver.google.com":
            start_adding = True
        
        if start_adding:
            print(f"正在为目标 {address} 添加扫描任务...")
            if add_scan(target_id):
                print(f"成功为目标 {address} 添加扫描任务")
            else:
                print(f"为目标 {address} 添加扫描任务失败")
        else:
            print(f"跳过目标 {address}")

if __name__ == "__main__":
    # 禁用不安全的HTTPS警告
    requests.packages.urllib3.disable_warnings(requests.packages.urllib3.exceptions.InsecureRequestWarning)
    main()

代码说明

• 目标筛选：代码中设置了一个标志 start_adding，在满足特定条件后开始为后续目标添加扫描任务。
• 执行流程：通过调用 get_all_targets 和 add_scan 函数完成整个流程。

总结

通过 AWVS API，我们可以方便地获取目标和添加扫描任务，大大提升了安全扫描的效率。希望这篇文章能够帮助你更好地理解和使用 AWVS API。如果你有任何问题或建议，欢迎在私信留言讨论！