X-Git-Url: http://git.tomasm.cz/krypto.git/blobdiff_plain/ee02d49796a9fcc1ff8be4a69149c0a371121d8e..7cf21962d42a0e4136f55a245b83474be4a42e90:/robot.py diff --git a/robot.py b/robot.py index 16eaaf0..16897a6 100755 --- a/robot.py +++ b/robot.py @@ -1,5 +1,7 @@ #!/usr/bin/env python +"""Mailove rozhrani pro sadu kryptoanalytickych nastroju.""" + import sys import smtplib import email @@ -9,10 +11,12 @@ from ocesavac import ocesat import jakobsen from jakobsen import reference import koincidence +import posuny import pickle import os - -ABECEDA = 'ABCDEFGHIJKLMNOPQRSTUVWXYZ' +import re +from spolecne import ABECEDA +import vigenere class settings: pass @@ -20,12 +24,16 @@ class settings: settings = pickle.load(open('./robot_settings')) os.chdir(settings.path) -def posli_mail(komu, co, text): +def posli_mail(komu, predmet, text, loguj=None): msg = MIMEText(text) - msg['Subject'] = co + msg['Subject'] = predmet msg['From'] = settings.name + " <" + settings.mail + ">" msg['To'] = komu smtplib.SMTP('localhost').sendmail(settings.mail, komu, msg.as_string()) + if loguj: + soubor = open(loguj, 'w') + soubor.write(msg.as_string()) + soubor.close() # Cteni mailu mail = email.message_from_file(sys.stdin) @@ -34,33 +42,47 @@ odesilatel = mail['From'] if predmet != settings.subject: sys.exit() if opsny == 'H': - posli_mail(odesilatel, "Napoveda", "Napoveda:\n" - "Prikazy robotovi vkladejte do predmetu zpravy zaslane na adresu " + - settings.mail + ".\nNa velikosti pismen nezalezi." + 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" - "\"" + settings.subject + " P\" Vypise vsechny posunuti textu.\n" - "\"" + settings.subject + " A\" Vypise uplne vsechno co dava " + "\"", 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" + "\"", settings.subject, " P\" Vypise vsechny posunuti textu.\n" + "\"", settings.subject, " A\" Vypise uplne vsechno co dava " "alespon trochu smysl (nedoporucuji).\n" - "\"" + settings.subject + " H\" Tato napoveda.\n" + "\"", settings.subject, " H\" Tato napoveda.\n" "\nVolby (mimo \"H\") je mozno libovolne kombinovat.\n" - "Napriklad \"" + settings.subject + " LXB\" analyzuje posledni " + "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')) - if (mail['From'] != odesilatel): + m1 = re.match(r".*\<(.*)\>.*", mail['From']) + m2 = re.match(r".*\<(.*)\>.*", odesilatel) + if m1: + f1 = m1.group(1) + else: + f1 = mail['From'] + if m2: + f2 = m2.group(1) + else: + f2 = odesilatel + if (f1 != f2): posli_mail(odesilatel, "Omluva", "Omlouvam se," " ale Vase zprava uz bohuzel neni na serveru ulozena.\n") sys.exit() telo = mail.get_payload().strip() -if len(telo) < 2: +if len(ocesat(telo, False)) < 2: + posli_mail(odesilatel, "Omluva", "Omlouvam se," + " ale Vase zprava je prilis kratka.\n") sys.exit() zprava = telo.upper() analyza = [] @@ -80,7 +102,7 @@ if 'A' in opsny: mod_x = True mod_m = True if not mod_m and not mod_x: - if ' ' in ocesat(zprava, True): + if ' ' in ocesat(zprava, True)[1:-1]: mod_m = True analyza.append("\nAutomaticky zvolena analyza s mezerami.") else: @@ -90,14 +112,17 @@ if not mod_m and not mod_x: # 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)))) + 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))) + analyza.append('{0:.3}'.format( + float(sum([len(word) for word in slova]))/len(slova))) # Frekvence znaku analyza.append("\nFrekvencni analyza:") @@ -113,38 +138,42 @@ for char in sorted(freq, key=freq.get, reverse=True): float(freq[char])/celkem)) # Index koincidence -# TODO verze s mezerami -analyza.append("\nIndexy koincidence (vzdy bez mezer):") -analyza.append(koincidence.tabulka_indexu(zprava)) +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 -# TODO verze bez mezer if mod_m: analyza.append("\nJakobsen (s mezerami):") - analyza.append(jakobsen.desifruj(zprava, './ref')) + analyza.append(jakobsen.desifruj(zprava)) if mod_x: analyza.append("\nJakobsen (bez mezer):") - analyza.append(jakobsen.desifruj(zprava, './bref')) + analyza.append(jakobsen.desifruj(zprava, False)) # Posuny v abecede -# TODO najit nejvhodnejsi posuny a vypsat jenom ty -def posun(char): - if 65 <= ord(char) < 90: - return chr(ord(char) + 1) - if ord(char) == 90: - return 'A' - else: - return char - +(posun, posunuty_text) = posuny.nejlepsi(zprava) if 'P' in opsny or 'A' in opsny: - analyza.append("\nPosuny v abecede:") - posunuty = zprava - for i in range(26): - posunuty = map(posun, posunuty) - analyza.append(''.join(posunuty)) + 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): +if 'B' in opsny or 'A' in opsny: analyza.append("\nBigramy podle cetnosti:") bigramy = dict() for i in ABECEDA: @@ -165,26 +194,21 @@ analyza.append(''.join(["\nS pozdravem\nVas Robot\n" "\n--\nPro napovedu zaslete mail s predmetem \"", settings.subject," H\".\n"])) -# Kompozice odpovedi -msg = MIMEText("\n".join(analyza)) -msg['Subject'] = "Analyza Vasi zpravy" -msg['From'] = settings.name + " <" + settings.mail + ">" -msg['To'] = odesilatel - # Odeslani odpovedi -s = smtplib.SMTP('localhost') -s.sendmail(settings.mail, odesilatel, msg.as_string()) -s.quit() - -# Zalogovani posledni zpravy -f = open('./robot_last.tmp', 'w') -f.write(msg.as_string()) -f.close() +posli_mail(odesilatel, + "Analyza Vasi zpravy", + "\n".join(analyza), + './robot_last.tmp') f = open('./msg_last.tmp', 'w') pickle.dump(mail, f) f.close() f = open('./robot.log', 'a') -f.write(asctime() + " " + odesilatel + "\n") +m = re.match(r".*\<(.*)\>.*", odesilatel) +if m: + odesilatel2 = m.group(1) +else: + odesilatel2 = odesilatel +f.write(asctime() + " " + odesilatel2 + " " + opsny + "\n") f.close()