#!/usr/bin/env python
'''
Copyright (C) 2020, WAFW00F Developers.
See the LICENSE file for copying permission.
'''

NAME = 'DOSarrest (DOSarrest Internet Security)'


def is_waf(self):
    # 定义用于检测是否存在特定 Web 应用防火墙（WAF）的方案列表。
    schemes = [
        # 检查响应头中是否存在'X-DIS-Request-ID'字段且其值不为空。
        self.matchHeader(('X-DIS-Request-ID', '.+')),
        # 发现一些 DOSArrest 的样本返回'Server: DoSArrest/3.5'，检查响应头中的'Server'字段是否包含'DOSarrest'。
        self.matchHeader(('Server', r'DOSarrest(.*)?'))
    ]
    # 如果方案列表中的任何一个方案为真，则认为检测到了 WAF。
    if any(i for i in schemes):
        return True
    else:
        return False