Minor update

This commit is contained in:
Miroslav Stampar 2019-01-05 21:29:59 +01:00
parent 45d0010323
commit 12427ff6f8

View File

@ -16,8 +16,7 @@ def detect(get_page):
for vector in WAF_ATTACK_VECTORS: for vector in WAF_ATTACK_VECTORS:
page, _, _ = get_page(get=vector) page, _, _ = get_page(get=vector)
retval = re.search(r"This response was generated by Wordfence", page or "", re.I) is not None retval = any(_ in (page or "") for _ in ("A potentially unsafe operation has been detected in your request to this site", "Generated by Wordfence", "Your access to this site has been limited", "This response was generated by Wordfence"))
retval |= re.search(r"Your access to this site has been limited", page or "", re.I) is not None
if retval: if retval:
break break