X-Git-Url: http://git.tomasm.cz/krypto.git/blobdiff_plain/167d6e0ae3d38f47d812cbb089cf26c5ff18ea3e..bd598aa43bf66c3afcb7a597a7790c8acfb49f7b:/robot.py?ds=sidebyside diff --git a/robot.py b/robot.py index 6ae0358..f227eb5 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) @@ -33,19 +39,46 @@ predmet, _, opsny = mail['Subject'].strip().upper().partition(' ') odesilatel = mail['From'] if predmet != settings.subject: sys.exit() -if (opsny == 'H'): +if opsny == 'H': posli_mail(odesilatel, "Napoveda", "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") + "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 " + "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): - posli_mail(odesilatel, "Omluva", "Omlouvam se, ale Vase zprava uz bohuzel neni na serveru ulozena.\n") + 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 = [] @@ -54,33 +87,35 @@ analyza = [] analyza.append("Puvodni zprava:") analyza.append(telo) -# 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)) - -# Jakobsen -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' +# 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): + mod_m = True + analyza.append("\nAutomaticky zvolena analyza s mezerami.") else: - return char + 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)))) -posunuty = zprava -for i in range(26): - posunuty = map(posun, posunuty) - analyza.append(''.join(posunuty)) +# 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:") @@ -95,47 +130,65 @@ 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: + analyza.append("\nIndexy koincidence (bez mezer):") + analyza.append(koincidence.tabulka_indexu(zprava)) + +# 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 -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()