X-Git-Url: http://git.tomasm.cz/krypto.git/blobdiff_plain/d4b7ddb6876e9c5f7d96329c95a3c327972c27d7..239c6137da450c54759d5d7d910b5e8f16cf37ef:/robot.py diff --git a/robot.py b/robot.py index 290798c..baa866c 100755 --- a/robot.py +++ b/robot.py @@ -2,21 +2,19 @@ """Mailove rozhrani pro sadu kryptoanalytickych nastroju.""" +import os import sys +import re import smtplib import email from email.mime.text import MIMEText from time import asctime +import pickle + from ocesavac import ocesat -import jakobsen from jakobsen import reference -import koincidence -import posuny -import pickle -import os -import re from spolecne import ABECEDA -import vigenere +import krypto class settings: pass @@ -41,15 +39,16 @@ predmet, _, opsny = mail['Subject'].strip().upper().partition(' ') odesilatel = mail['From'] if predmet != settings.subject: sys.exit() + if opsny == 'H': - posli_mail(odesilatel, "Napoveda", ''.join("Napoveda:\n" + #TODO dokumentace k Vigenerovi + #TODO dokumentace k L + posli_mail(odesilatel, "Napoveda", ''.join(["Napoveda:\n" "Prikazy robotovi vkladejte do predmetu zpravy " "zaslane na adresu ", settings.mail, ".\nNa velikosti pismen nezalezi." "\n\nSeznam prikazu:\n" "\"", settings.subject, "\" Analyza zpravy z tela mailu.\n" - "\"", settings.subject, " L\" Opakovana analyza " - "posledni zpravy.\n", "\"", settings.subject, " X\" Vynuti analyzu bez mezer.\n" "\"", settings.subject, " M\" Vynuti analyzu s mezerami.\n" "\"", settings.subject, " B\" Vypise cetnost vsech bigramu.\n" @@ -61,8 +60,9 @@ if opsny == 'H': "Napriklad \"", settings.subject, " LXB\" analyzuje posledni " "prijatou zpravu s vynucenim analyzy bez mezer a navic vypise " "cetnosti bigramu.\n" - "\nS pozdravem\nVas Robot\n")) + "\nS pozdravem\nVas Robot\n"])) sys.exit() + if 'L' in opsny: mail = pickle.load(open('./msg_last.tmp')) m1 = re.match(r".*\<(.*)\>.*", mail['From']) @@ -79,8 +79,9 @@ if 'L' in opsny: posli_mail(odesilatel, "Omluva", "Omlouvam se," " ale Vase zprava uz bohuzel neni na serveru ulozena.\n") sys.exit() + if mail.is_multipart(): - telo = mail.get_payload()[0].strip() + telo = mail.get_payload(0).get_payload().strip() else: telo = mail.get_payload().strip() @@ -89,119 +90,13 @@ if len(ocesat(telo, False)) < 2: " ale Vase zprava je prilis kratka.\n") sys.exit() zprava = telo.upper() -analyza = [] - -# Puvodni zprava -analyza.append("Puvodni zprava:") -analyza.append(telo) - -# Prepinani spacemodu -mod_m = False -mod_x = False -if 'X' in opsny: - mod_x = True -if 'M' in opsny: - mod_m = True -if 'A' in opsny: - mod_x = True - mod_m = True -if not mod_m and not mod_x: - if ' ' in ocesat(zprava, True)[1:-1]: - mod_m = True - analyza.append("\nAutomaticky zvolena analyza s mezerami.") - else: - mod_x = True - analyza.append("\nAutomaticky zvolena analyza bez mezer.") - -# Zakladni udaje -analyza.append("\nZakladni udaje:") -if mod_m: - analyza.append("Delka zpravy (vcetne mezer a zvlastnich znaku): " + - str(len(zprava))) -analyza.append("Delka zpravy (bez mezer a zvlastnich znaku): " + - str(len(ocesat(zprava, False)))) - -# Prumerna delka slova -if mod_m: - analyza.append("\nPrumerna delka slova:") - slova = ocesat(zprava, True).split() - analyza.append('{0:.3}'.format( - float(sum([len(word) for word in slova]))/len(slova))) - -# Frekvence znaku -analyza.append("\nFrekvencni analyza:") -freq = dict() -for char in ABECEDA: - freq[char] = 0 -for char in zprava: - if char in freq: - freq[char] += 1 -celkem = sum(freq.values()) -for char in sorted(freq, key=freq.get, reverse=True): - analyza.append('{0} {1:>3} {2:>6.2%}'.format(char, freq[char], - float(freq[char])/celkem)) - -# Index koincidence -if mod_m: - analyza.append("\nIndex koincidence (s mezerami):") - analyza.append('{0:.2%}'.format( - koincidence.index_koincidence(zprava, True))) -if mod_x and len(zprava) >= 10: - #TODO tady by se spravne mela porovnavat delka ocesane zpravy - analyza.append("\nIndexy koincidence (bez mezer):") - analyza.append(koincidence.tabulka_indexu(zprava)) - -# Vigenere -if 'V' in opsny: - analyza.append("\nHeslo k Vigenerovi (" + opsny[-1] + "):") - vig_heslo, vig_text = vigere.vsechno(zprava, int(opsny[-1])) - analyza.append(vig_heslo) - analyza.append(vig_text) - -# Jakobsen -if mod_m: - analyza.append("\nJakobsen (s mezerami):") - analyza.append(jakobsen.desifruj(zprava)) -if mod_x: - analyza.append("\nJakobsen (bez mezer):") - analyza.append(jakobsen.desifruj(zprava, False)) - -# Posuny v abecede -(posun, posunuty_text) = posuny.nejlepsi(zprava) -if 'P' in opsny or 'A' in opsny: - analyza.append("\nPosuny v abecede (nejlepsi posun {0}):".format(posun)) - analyza.append("\n".join("{0:>2} {1}".format( - i + 1, posuny.vsechny(zprava)[i]) for i in range(26))) -else: - analyza.append("\nNejlepsi posun ({0}):".format(posun)) - analyza.append(posunuty_text) - -# Bigramy -if 'B' in opsny or 'A' in opsny: - analyza.append("\nBigramy podle cetnosti:") - bigramy = dict() - for i in ABECEDA: - for j in ABECEDA: - bigramy[i+j] = 0 - for i in range(0, len(zprava)-1): - if zprava[i:i+2] in bigramy: - bigramy[zprava[i:i+2]] += 1 - celkem = sum(bigramy.values()) - for char in sorted(bigramy, key=bigramy.get, reverse=True): - if bigramy[char] == 0: - break - analyza.append('{0} {1:>3} {2:>6.2%}'.format(char, bigramy[char], - float(bigramy[char])/celkem)) - -# Podpis a pozdrav -analyza.append(''.join(["\nS pozdravem\nVas Robot\n" - "\n--\nPro napovedu zaslete mail s predmetem \"", - settings.subject," H\".\n"])) - -# Odeslani odpovedi + +analyza = krypto.analyza(zprava, opsny) + + posli_mail(odesilatel, "Analyza Vasi zpravy", - "\n".join(analyza), + analyza + "\n\nS pozdravem\nVas Robot", './robot_last.tmp') f = open('./msg_last.tmp', 'w')