X-Git-Url: http://git.tomasm.cz/krypto.git/blobdiff_plain/df0003e93591f53b015980aaaec0a68f8e565ead..3fa7f2c2cc9ab864e921709ccc0deeff35ab6831:/robot.py diff --git a/robot.py b/robot.py index e2b3820..a49c337 100755 --- a/robot.py +++ b/robot.py @@ -9,8 +9,10 @@ from ocesavac import ocesat import jakobsen from jakobsen import reference import koincidence +import posuny import pickle import os +import re ABECEDA = 'ABCDEFGHIJKLMNOPQRSTUVWXYZ' @@ -20,12 +22,16 @@ class settings: settings = pickle.load(open('./robot_settings')) os.chdir(settings.path) -def posli_mail(komu, co, text): +def posli_mail(komu, co, text, loguj=None): msg = MIMEText(text) msg['Subject'] = co msg['From'] = settings.name + " <" + settings.mail + ">" msg['To'] = komu smtplib.SMTP('localhost').sendmail(settings.mail, komu, msg.as_string()) + if loguj: + f = open(loguj, 'w') + f.write(msg.as_string()) + f.close() # Cteni mailu mail = email.message_from_file(sys.stdin) @@ -40,13 +46,32 @@ if opsny == 'H': "\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 " + "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 '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() @@ -61,18 +86,22 @@ 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 -elif 'M' in opsny: - mod_m = False -elif ' ' in ocesat(zprava, True): +if 'A' in opsny: + mod_x = True mod_m = True -else: - mod_m = False -if mod_m: - analyza.append("\nAnalyza s mezerami.") -else: - analyza.append("\nAnalyza bez mezer.") +if not mod_m and not mod_x: + if ' ' in ocesat(zprava, True): + 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:") @@ -86,33 +115,6 @@ if mod_m: slova = ocesat(zprava, True).split() analyza.append('{0:.3}'.format(float(sum([len(word) for word in slova]))/len(slova))) -# Index koincidence -# TODO verze s mezerami -analyza.append("\nIndexy koincidence (vzdy bez mezer):") -analyza.append(koincidence.tabulka_indexu(zprava)) - -# Jakobsen -# TODO verze bez mezer -if mod_m: - analyza.append("\nJakobsen:") - analyza.append(jakobsen.desifruj(zprava, './ref')) - -# 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() @@ -126,47 +128,62 @@ 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 +# TODO verze s mezerami +analyza.append("\nIndexy koincidence (vzdy bez mezer):") +analyza.append(koincidence.tabulka_indexu(zprava)) + +# Jakobsen +if mod_m: + analyza.append("\nJakobsen (s mezerami):") + analyza.append(jakobsen.desifruj(zprava, './ref')) +if mod_x: + analyza.append("\nJakobsen (bez mezer):") + analyza.append(jakobsen.desifruj(zprava, './bref')) + +# 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 -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)) +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"])) -# 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()