X-Git-Url: http://git.tomasm.cz/krypto.git/blobdiff_plain/811d25c5cc569eec8ae7cfff984e150ce024e218..239c6137da450c54759d5d7d910b5e8f16cf37ef:/robot.py?ds=sidebyside diff --git a/robot.py b/robot.py index 1ea93e2..baa866c 100755 --- a/robot.py +++ b/robot.py @@ -1,18 +1,20 @@ #!/usr/bin/env python +"""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 -from ocesavac import ocesat -import jakobsen -from jakobsen import reference -import koincidence import pickle -import os -ABECEDA = 'ABCDEFGHIJKLMNOPQRSTUVWXYZ' +from ocesavac import ocesat +from jakobsen import reference +from spolecne import ABECEDA +import krypto class settings: pass @@ -20,123 +22,92 @@ class settings: settings = pickle.load(open('./robot_settings')) os.chdir(settings.path) +def posli_mail(komu, predmet, text, loguj=None): + msg = MIMEText(text) + 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) predmet, _, opsny = mail['Subject'].strip().upper().partition(' ') odesilatel = mail['From'] if predmet != settings.subject: sys.exit() -if (opsny == 'H'): - msg = MIMEText("Napoveda:\n" - "Pro opakovanou analyzu posledni zpravy zaslete mail s predmetem \"" + - settings.subject + " L\". Na velikosti pismen v predmetu nezalezi." - "\n\nS pozdravem\nVas Robot\n") - msg['Subject'] = "Napoveda" - msg['From'] = settings.name + " <" + settings.mail + ">" - msg['To'] = odesilatel - smtplib.SMTP('localhost').sendmail(settings.mail, odesilatel, msg.as_string()) + +if opsny == 'H': + #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, " 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" + "\nVolby (mimo \"H\") je mozno libovolne kombinovat.\n" + "Napriklad \"", settings.subject, " LXB\" analyzuje posledni " + "prijatou zpravu s vynucenim analyzy bez mezer a navic vypise " + "cetnosti bigramu.\n" + "\nS pozdravem\nVas Robot\n"])) sys.exit() -if (opsny == 'L'): + +if 'L' in opsny: mail = pickle.load(open('./msg_last.tmp')) - if (mail['From'] != odesilatel): - msg = MIMEText("Omlouvam se, ale Vase zprava uz bohuzel neni na serveru ulozena.") - msg['Subject'] = "Omluva" - msg['From'] = settings.name + " <" + settings.mail + ">" - msg['To'] = odesilatel - smtplib.SMTP('localhost').sendmail(settings.mail, odesilatel, msg.as_string()) + 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: - sys.exit() -zprava = telo.upper() -analyza = [] -# Puvodni zprava -analyza.append("Puvodni zprava:") -analyza.append(telo) +if mail.is_multipart(): + telo = mail.get_payload(0).get_payload().strip() +else: + telo = mail.get_payload().strip() -# Zakladni udaje -analyza.append("\nDelka zpravy (bez mezer a zvlastnich znaku):") -analyza.append(str(len(ocesat(zprava, False)))) - -# Index koincidence -analyza.append("\nIndexy koincidence:") -analyza.append(koincidence.tabulka_indexu(zprava)) +if len(ocesat(telo, False)) < 2: + posli_mail(odesilatel, "Omluva", "Omlouvam se," + " ale Vase zprava je prilis kratka.\n") + sys.exit() +zprava = telo.upper() -# Jakobsen -analyza.append("\nJakobsen:") -analyza.append(jakobsen.desifruj(zprava, './ref')) +analyza = krypto.analyza(zprava, opsny) -# Posuny v abecede -analyza.append("\nPosuny v abecede:") -def posun(char): - if 65 <= ord(char) < 90: - return chr(ord(char) + 1) - if ord(char) == 90: - return 'A' - else: - return char - -posunuty = zprava -for i in range(26): - posunuty = map(posun, posunuty) - analyza.append(''.join(posunuty)) - -# 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)) - -# Bigramy -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"])) - -# 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", + analyza + "\n\nS pozdravem\nVas Robot", + './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()