X-Git-Url: http://git.tomasm.cz/krypto.git/blobdiff_plain/811d25c5cc569eec8ae7cfff984e150ce024e218..ee02d49796a9fcc1ff8be4a69149c0a371121d8e:/robot.py?ds=sidebyside diff --git a/robot.py b/robot.py index 1ea93e2..16eaaf0 100755 --- a/robot.py +++ b/robot.py @@ -20,30 +20,44 @@ class settings: settings = pickle.load(open('./robot_settings')) os.chdir(settings.path) +def posli_mail(komu, co, text): + 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()) + # 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': + posli_mail(odesilatel, "Napoveda", "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 " + "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()) + 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: @@ -55,21 +69,65 @@ analyza = [] 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): + 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("\nDelka zpravy (bez mezer a zvlastnich znaku):") -analyza.append(str(len(ocesat(zprava, False)))) +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 -analyza.append("\nIndexy koincidence:") +# TODO verze s mezerami +analyza.append("\nIndexy koincidence (vzdy bez mezer):") analyza.append(koincidence.tabulka_indexu(zprava)) # Jakobsen -analyza.append("\nJakobsen:") -analyza.append(jakobsen.desifruj(zprava, './ref')) +# TODO verze bez mezer +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 -analyza.append("\nPosuny v abecede:") - +# TODO najit nejvhodnejsi posuny a vypsat jenom ty def posun(char): if 65 <= ord(char) < 90: return chr(ord(char) + 1) @@ -78,39 +136,29 @@ def posun(char): 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)) +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)) # 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"