import PySimpleGUI as sg from padding_oracle_demo import po_main import time # 引入time模块 import AES_CBC as CBC import AES_CFB as CFB import AES_CTR as CTR import AES_ECB as ECB import AES_OFB as OFB import time # 密钥设置，请设置为16.24.32 ENCKEY = '' user_get_enc = '' # 定义窗口 def PO_attack(): layout = [ [sg.T('密文填塞攻击')], [sg.T('正常用户:')], [sg.Text('请输入需要加密的明文 :'), sg.InputText('', size=(32, 1), key='-user_mw-')], [sg.Text('请输入使用的密钥 :'), sg.InputText('', size=(32, 1), key='-user_key-')], [sg.Text('请输入使用的IV :'), sg.InputText('', size=(32, 1), key='-user_iv-')], [sg.Button('开始加密', key='-user_start-')], [sg.Text('加密密文:')], [sg.Output(size=(80, 2), key='-user_enc-', echo_stdout_stderr=True)], # [sg.Output(size=(100, 5))], [sg.Text('')], [sg.T('攻击者:')], [sg.Text('请输入需要攻击的密文 :')], [sg.Multiline('', size=(80, 2), key='-attacker_enc-')], [sg.Text('请输入需要攻击的IV :'), sg.InputText('', size=(32, 1), key='-attacker_iv-')], [sg.Text('请输入需要加密的明文 :'), sg.InputText('', size=(32, 1), key='-attacker_mw-')], [sg.Button('开始攻击', key='-attacker_start-')], [sg.Output(size=(80, 3), key='-attacker_output-', echo_stdout_stderr=True)], [sg.Text('')] ] return sg.Window('密文填塞攻击攻击', layout) class GUI: # 主窗口 def __init__(self): sg.theme('DefaultNoMoreNagging') list1 = ['ECB模式', 'CBC模式', 'CFB模式', 'OFB模式', 'CTR模式'] list2 = ['加密', '解密'] layout = [ [sg.Text('请选择加密/解密'), sg.Combo(list2, key='-crypt-', size=5, enable_events=True)], [sg.Text('请输入密钥'), sg.InputText('', key='-key_text-')], [sg.FileBrowse('选择文件', target='-FilePath-', file_types=(('ALL Files', '*.txt'),)), sg.InputText('', size=(100, 1), key='-FilePath-')], [sg.Button('开始执行加密/解密', size=(50, 2), key='-start-')], [sg.Text('采用ECB模式'), sg.Text('', size=5, key='-ECBkey-', border_width=1, relief='ridge'), sg.Text('的结果是：'), sg.InputText('', key='-ECB-'), sg.T('用时：'), sg.InputText('', key='ECBtime', )], [sg.Text('采用CBC模式'), sg.Text('', size=5, border_width=1, relief='ridge', key='-CBCkey-'), sg.Text('的结果是：'), sg.InputText('', key='-CBC-'), sg.T('用时：'), sg.InputText('', key='CBCtime')], [sg.Text('采用CFB模式'), sg.Text('', size=5, border_width=1, relief='ridge', key='-CFBkey-'), sg.Text('的结果是：'), sg.InputText('', key='-CFB-'), sg.T('用时：'), sg.InputText('', key='CFBtime')], [sg.Text('采用OFB模式'), sg.Text('', size=5, border_width=1, relief='ridge', key='-OFBkey-'), sg.Text('的结果是：'), sg.InputText('', key='-OFB-'), sg.T('用时：'), sg.InputText('', key='OFBtime')], [sg.Text('采用CTR模式'), sg.Text('', size=5, border_width=1, relief='ridge', key='-CTRkey-'), sg.Text('的结果是：'), sg.InputText('', key='-CTR-'), sg.T('用时：'), sg.InputText('', key='CTRtime')], [sg.Button('Padding_Oracle_Attack', size=(50, 1), key='-PO_Attack-')] ] self.window = sg.Window('AES', layout, size=(1200, 500), element_padding=(5, 10), font='宋体', grab_anywhere=True, resizable=True, element_justification='left') # 事件 def run(self): while True: event, values = self.window.read() print(event) if event == None: break elif event == '-crypt-': print('加密模式') self.window['-ECBkey-'].update(values['-crypt-']) self.window['-CBCkey-'].update(values['-crypt-']) self.window['-CFBkey-'].update(values['-crypt-']) self.window['-OFBkey-'].update(values['-crypt-']) self.window['-CTRkey-'].update(values['-crypt-']) print('111') elif event == '-start-': print("开始执行") with open(values['-FilePath-'], 'r', encoding='utf-8') as fp: M = fp.read() print(M) # 加密 if values['-crypt-'] == '加密': # CBC模式 cbc = CBC.aestest(values['-key_text-']) T1_CBC = time.time() for i in range(1000): C_CBC = cbc.encrypt(M) T2_CBC = time.time() T_CBC = T2_CBC - T1_CBC # CBC加密时间 print(T_CBC) # print(C_CBC) self.window['-CBC-'].update(C_CBC) self.window['CBCtime'].update(T_CBC) # CFB模式 cfb = CFB.aestest(values['-key_text-']) T1_CFB = time.time() for i in range(1000): C_CFB = cfb.encrypt(M) T2_CFB = time.time() T_CFB = T2_CFB - T1_CFB # CFB加密时间 print(T_CFB) self.window['-CFB-'].update(C_CFB) self.window['CFBtime'].update(T_CFB) # CTR模式 ctr = CTR.aestest(values['-key_text-']) T1_CTR = time.time() for i in range(1000): C_CTR = ctr.encrypt(M) T2_CTR = time.time() T_CTR = T2_CTR - T1_CTR # CTR加密时间 print(T_CTR) self.window['-CTR-'].update(C_CTR) self.window['CTRtime'].update(T_CTR) # ECB模式 ecb = ECB.aestest(values['-key_text-']) T1_ECB = time.time() for i in range(1000): C_ECB = ecb.encrypt(M) T2_ECB = time.time() T_ECB = T2_ECB - T1_ECB # ECB加密时间 print(T_ECB) self.window['-ECB-'].update(C_ECB) self.window['ECBtime'].update(T_ECB) # OFB模式 ofb = OFB.aestest(values['-key_text-']) T1_OFB = time.time() for i in range(1000): C_OFB = ofb.encrypt(M) T2_OFB = time.time() T_OFB = T2_OFB - T1_OFB # OFB加密时间 print(T_OFB) self.window['-OFB-'].update(C_OFB) self.window['OFBtime'].update(T_OFB) elif values['-crypt-'] == '解密': # CBC模式 cbc = CBC.aestest(values['-key_text-']) T1_CBC = time.time() for i in range(1000): C_CBC = cbc.decrypt(M) T2_CBC = time.time() T_CBC = T2_CBC - T1_CBC # CBC解密时间 print(T_CBC) self.window['-CBC-'].update(C_CBC) self.window['CBCtime'].update(T_CBC) # CFB模式 cfb = CFB.aestest(values['-key_text-']) T1_CFB = time.time() for i in range(1000): C_CFB = cfb.decrypt(M) T2_CFB = time.time() T_CFB = T2_CFB - T1_CFB # CFB解密时间 print(T_CFB) self.window['-CFB-'].update(C_CFB) self.window['CFBtime'].update(T_CFB) # CTR模式 ctr = CTR.aestest(values['-key_text-']) T1_CTR = time.time() for i in range(1000): C_CTR = ctr.decrypt(M) T2_CTR = time.time() T_CTR = T2_CTR - T1_CTR # CTR解密时间 print(T_CTR) self.window['-CTR-'].update(C_CTR) self.window['CTRtime'].update(T_CTR) # ECB模式 ecb = ECB.aestest(values['-key_text-']) T1_ECB = time.time() for i in range(1000): C_ECB = ecb.decrypt(M) T2_ECB = time.time() T_ECB = T2_ECB - T1_ECB # ECB解密时间 print(T_ECB) self.window['-ECB-'].update(C_ECB) self.window['ECBtime'].update(T_ECB) # OFB模式 ofb = OFB.aestest(values['-key_text-']) T1_OFB = time.time() for i in range(1000): C_OFB = ofb.decrypt(M) T2_OFB = time.time() T_OFB = T2_OFB - T1_OFB # OFB解密时间 print(T_OFB) self.window['-OFB-'].update(C_OFB) self.window['OFBtime'].update(T_OFB) elif event == '-key_text-': print(values['-key_text-']) # 打印出密钥 elif event == '-PO_Attack-': print('Padding模式') win1 = PO_attack() while True: event1, value1 = win1.read() if event1 == '-user_start-': print(value1['-user_mw-']) user_mw = value1['-user_mw-'] user_iv = value1['-user_iv-'] ENCKEY = value1['-user_key-'].encode('utf-8') user_enc = po_main(user_mw, user_iv, ENCKEY) print(user_enc) user_get_enc = user_enc # 给攻击者使用，因为输出的是字符串包裹的bytes难以转换 win1['-user_enc-'].update(f'正常用户密文： {user_enc}') win1['-attacker_output-'].update('') if event1 == '-attacker_start-': try: attacker_enc = value1['-attacker_enc-'] print(attacker_enc) print(type(attacker_enc), '\n') except: print('change failed!\n') attacker_enc = user_get_enc attacker_iv = value1['-attacker_iv-'] attacker_mw = value1['-attacker_mw-'] attacker_user_mw, attacker_enc_mw = po_main(None, attacker_iv, ENCKEY, attacker_enc, attacker_mw) print(attacker_user_mw, attacker_enc_mw) win1['-attacker_output-'].update(f'正常用户明文： {attacker_user_mw}\n攻击者生成的密文： {attacker_enc_mw}') elif event1 is None: break self.window.close() if __name__ == '__main__': BaseW = GUI() BaseW.run()