You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
70 lines
2.4 KiB
70 lines
2.4 KiB
# -*- coding: utf-8 -*-
|
|
|
|
from Connexion import Connexion
|
|
import tkinter as tk
|
|
|
|
|
|
class Interface(tk.Tk):
|
|
"""interface graphique pour envoyer/recevoir/chiffrer/dechiffrer"""
|
|
def __init__(self, sens):
|
|
tk.Tk.__init__(self)
|
|
self.connexion = Connexion(sens)
|
|
self.title(sens)
|
|
self.entree = tk.StringVar()
|
|
tk.Entry(self, textvariable=self.entree, width=30).pack()
|
|
tk.Label(text="Message reçu : ").pack()
|
|
self.message = tk.StringVar()
|
|
tk.Label(textvariable=self.message).pack()
|
|
self.sendButton = tk.Button(self, text="Send", command=self.send)
|
|
self.sendButton.pack()
|
|
self.recvButton = tk.Button(self, text="Receive", command=self.receive)
|
|
self.recvButton.pack()
|
|
#
|
|
self._cleSymetrique = None
|
|
|
|
def send(self):
|
|
"""Envoie le message chiffré."""
|
|
print("self.entree.get() dans Interface.py : ", self.entree.get())
|
|
message = self.entree.get().encode("utf8")
|
|
# à compléter pour chiffrer le message
|
|
if self._cleSymetrique != None:
|
|
message = chiffre_xor(message, self._cleSymetrique)
|
|
self.connexion.send(message)
|
|
|
|
def receive(self):
|
|
"""Réceptionne un message depuis le serveur et l'affiche.
|
|
Attention : bloque tant que le serveur n'envoie rien."""
|
|
message = self.connexion.recv()
|
|
# à compléter pour déchiffrer le message
|
|
if self._cleSymetrique != None:
|
|
message = chiffre_xor(message, self._cleSymetrique)
|
|
self.message.set(message.decode("utf8"))
|
|
|
|
def set_cleSymetrique(self, val):
|
|
"""Définit/met à jour la clé symétrique"""
|
|
self._cleSymetrique = val
|
|
|
|
def destroy(self):
|
|
self.connexion.close()
|
|
tk.Tk.destroy(self)
|
|
|
|
|
|
|
|
def chiffre_xor(message, cle):
|
|
"""chiffre/déchiffre un message à l'aide d'une clé"""
|
|
res = b""
|
|
for i in range(len(message)):
|
|
lettre = bytes([message[i] ^ cle[i%len(cle)]])
|
|
res += lettre
|
|
return res
|
|
|
|
def puissance_mod_rapide(a, b, n):
|
|
"""renvoie a**b mod n"""
|
|
if b == 1:
|
|
return a
|
|
elif b == 2:
|
|
return (a**2)%n
|
|
elif b % 2 == 0:
|
|
return puissance_mod_rapide((a**2)%n, b // 2, n)
|
|
else:
|
|
return (a * puissance_mod_rapide((a**2)%n, b // 2, n)) % n
|