Mercurial > public > m209
changeset 15:2e2692fb7de6
Created an encrypt procedure and a first test for it.
author | Brian Neal <bgneal@gmail.com> |
---|---|
date | Fri, 07 Jun 2013 22:30:33 -0500 |
parents | da830ef4ba52 |
children | 1448d698f9e4 |
files | m209/converter.py m209/key_wheel.py m209/keylist/__init__.py m209/procedure.py m209/tests/test_converter.py m209/tests/test_procedure.py |
diffstat | 6 files changed, 238 insertions(+), 4 deletions(-) [+] |
line wrap: on
line diff
--- a/m209/converter.py Thu Jun 06 20:32:37 2013 -0500 +++ b/m209/converter.py Fri Jun 07 22:30:33 2013 -0500 @@ -11,10 +11,11 @@ from . import M209Error from .data import KEY_WHEEL_DATA -from .key_wheel import KeyWheel +from .key_wheel import KeyWheel, KeyWheelError from .drum import Drum -M209_ALPHABET = set(string.ascii_uppercase) +M209_ALPHABET_LIST = string.ascii_uppercase +M209_ALPHABET_SET = set(string.ascii_uppercase) CIPHER_TABLE = list(reversed(string.ascii_uppercase)) @@ -110,6 +111,18 @@ drum = Drum(lug_list) self.drum = drum + def set_key_wheel(self, n, c): + """Set key wheel n to the letter c, where n is 0-5, inclusive. + + Key wheel 0 is the leftmost key wheel, and 5 is the rightmost. + + May raise KeyWheelError if c is not valid for key wheel n. + + """ + if not (0 <= n < len(self.key_wheels)): + raise M209Error("set_key_wheel(): invalid key wheel index {}".format(n)) + self.key_wheels[n].set_pos(c) + def set_key_wheels(self, s): """Set the key wheels from left to right to the six letter string s.""" @@ -117,7 +130,18 @@ raise M209Error("Invalid key wheels setting length") for n in range(6): - self.key_wheels[n].set_pos(s[n]) + try: + self.key_wheels[n].set_pos(s[n]) + except KeyWheelError as ex: + raise KeyWheelError('wheel #{}: {}'.format(n, ex)) + + def set_random_key_wheels(self): + """Sets the 6 key wheels to random letters and returns the letters as + a string. + + """ + letters = [kw.set_random() for kw in self.key_wheels] + return ''.join(letters) def get_settings(self): """Returns the current settings as a M209Settings named tuple.""" @@ -188,7 +212,7 @@ the internal substitution table. """ - if c not in M209_ALPHABET: + if c not in M209_ALPHABET_SET: raise M209Error("Illegal char: {}".format(c)) pins = [kw.is_effective() for kw in self.key_wheels]
--- a/m209/key_wheel.py Thu Jun 06 20:32:37 2013 -0500 +++ b/m209/key_wheel.py Fri Jun 07 22:30:33 2013 -0500 @@ -6,8 +6,11 @@ simulation. """ +import random + from . import M209Error + class KeyWheelError(M209Error): """Exception class for all key wheel errors""" pass @@ -120,3 +123,13 @@ self.pos = self.letter_offsets[c] except KeyError: raise KeyWheelError("Invalid position {}".format(c)) + + def set_random(self): + """Sets the position of the key wheel to a random letter. + + The random letter is returned as a string. + + """ + c = random.choice(self.letters) + self.set_pos(c) + return c
--- a/m209/keylist/__init__.py Thu Jun 06 20:32:37 2013 -0500 +++ b/m209/keylist/__init__.py Fri Jun 07 22:30:33 2013 -0500 @@ -0,0 +1,5 @@ +# Copyright (C) 2013 by Brian Neal. +# This file is part of m209, the M-209 simulation. +# m209 is released under the MIT License (see LICENSE.txt). + +from .key_list import KeyList
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/m209/procedure.py Fri Jun 07 22:30:33 2013 -0500 @@ -0,0 +1,136 @@ +# Copyright (C) 2013 by Brian Neal. +# This file is part of m209, the M-209 simulation. +# m209 is released under the MIT License (see LICENSE.txt). + +"""This module contains the encrypt & decrypt procedures as described in "War +Department, Official Training Film, T.F. 11 - 1400, Army, Service Forces." +A YouTube playlist of this film can be found here: + +http://www.youtube.com/playlist?list=PLCPgncK_sTnEny2-uoTV-1_GC72zo-vKq + +This procedure is also described on Mark J. Blair's pages: + +http://www.nf6x.net/2013/04/practical-use-of-the-m-209-cipher-machine-chapter-4/ +http://www.nf6x.net/2013/04/practical-use-of-the-m-209-cipher-machine-chapter-5/ + +If other procedures are discovered, this module can be expanded to a package and +the new procedures can be added as different modules. For now, this is the only +procedure known to the author. + +""" +import random + +from . import M209Error +from .converter import M209_ALPHABET_SET, M209_ALPHABET_LIST +from .key_wheel import KeyWheelError + + +class ProcedureError(M209Error): + pass + + +def encrypt(m_209, plaintext, group=True, spaces=True, key_list=None, + key_list_ind=None, ext_msg_ind=None, sys_ind=None): + """Encrypts a plaintext message using standard procedure. The encrypted text + with the required message indicators are returned as one string. + + The encrypt function accepts these parameters: + + m_209 - A M209 converter instance + plaintext - Input string of text to be encrypted + group - If True, the resulting encrypted text will be grouped into 5-letter + groups with a space between each group. If False, no spaces will be + present in the output. + spaces - If True, space characters in the input plaintext will automatically + be replaced with 'Z' characters before encrypting. + key_list - If not None, this must be a KeyList instance and it will be + used to setup the m_209 machine. If None, it is assumed the M209 is + already setup in the desired configuration. + key_list_ind - If key_list is None, then this parameter must be supplied, + and it is the key list indicator that was used to setup the M209. + ext_msg_ind - This is the external message indicator, which, if supplied, + must be a valid 6 letter string of key wheel settings. If not supplied, + one will be generated randomly. + sys_ind - This is the system indicator, which must be a string of length + 1 in the range 'A'-'Z', inclusive. If None, one is chosen at random. + + It is an error to supply both key_list and key_list_ind options. Supply one + or the other. A ProcedureError will be raised if both are supplied or if + neither are supplied. + + """ + # Ensure we have a key list indicator and there is no ambiguity: + + if (key_list and key_list_ind) or (not key_list and not key_list_ind): + raise ProcedureError("encrypt requires either key_list or key_list_ind but not both") + + # Setup M209 machine if necessary + + if key_list: + m_209.set_drum_lugs(key_list.lugs) + m_209.set_all_pins(key_list.pin_list) + key_list_ind = key_list.indicator + + if len(key_list_ind) != 2: + raise ProcedureError("key list indicator must be two letters") + + m_209.letter_counter = 0 + + # Set key wheels to external message indicator + if ext_msg_ind: + try: + m_209.set_key_wheels(ext_msg_ind) + except M209Error as ex: + raise M209Error("invalid external message indicator {} - {}".format( + ext_msg_ind, ex)) + else: + ext_msg_ind = m_209.set_random_key_wheels() + + # Ensure we have a valid system indicator + if sys_ind: + if sys_ind not in M209_ALPHABET_SET: + raise ProcedureError("invalid system indicator {}".format(sys_ind)) + else: + sys_ind = random.choice(M209_ALPHABET_LIST) + + # Generate internal message indicator + + int_msg_ind = m_209.encrypt(sys_ind * 12, group=False, spaces=False) + + # Set wheels to internal message indicator from left to right. We must skip + # letters that aren't valid for a given key wheel. + it = iter(int_msg_ind) + n = 0 + while n != 6: + try: + m_209.set_key_wheel(n, next(it)) + except KeyWheelError: + pass + except StopIteration: + assert False, "Ran out of letters building internal message indicator" + else: + n += 1 + + # Now encipher the message on the M209 + ciphertext = m_209.encrypt(plaintext, group=group, spaces=spaces) + + # If we are grouping, and the final group in the ciphertext has less than + # 5 letters, pad with X's to make a complete group: + if group: + total_len = len(ciphertext) + num_groups = total_len // 5 + num_spaces = num_groups - 1 if num_groups >= 2 else 0 + x_count = 5 - (total_len - num_spaces) % 5 + if 0 < x_count < 5: + ciphertext = ciphertext + 'X' * x_count + + # Add the message indicators to pad each end of the message + + pad1 = sys_ind * 2 + ext_msg_ind[:3] + pad2 = ext_msg_ind[3:] + key_list_ind + + msg_parts = [pad1, pad2, ciphertext, pad1, pad2] + + # Assemble the final message; group if requested + sep = ' ' if group else '' + return sep.join(msg_parts)
--- a/m209/tests/test_converter.py Thu Jun 06 20:32:37 2013 -0500 +++ b/m209/tests/test_converter.py Fri Jun 07 22:30:33 2013 -0500 @@ -87,6 +87,23 @@ check = 'OZGPK AFVAJ JYRZW LRJEG MOVLU M' self.letter_check(lugs, pin_list, check) + def test_fm_letter_check(self): + """See if we can pass a letter check using Mark J. Blair's FM key list.""" + + lugs = '1-0 2-0*8 0-3*7 0-4*5 0-5*2 1-5 1-6 3-4 4-5' + + pin_list = [ + 'BCEJOPSTUVXY', + 'ACDHJLMNOQRUYZ', + 'AEHJLOQRUV', + 'DFGILMNPQS', + 'CEHIJLNPS', + 'ACDFHIMN' + ] + + check = 'TNMYS CRMKK UHLKW LDQHM RQOLW R' + self.letter_check(lugs, pin_list, check) + def test_no_group(self): m = M209(AA_LUGS, AA_PIN_LIST)
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/m209/tests/test_procedure.py Fri Jun 07 22:30:33 2013 -0500 @@ -0,0 +1,39 @@ +# Copyright (C) 2013 by Brian Neal. +# This file is part of m209, the M-209 simulation. +# m209 is released under the MIT License (see LICENSE.txt). + +"""Unit tests for the M209 encrypt & decrypt procedures.""" + +import unittest + +from ..keylist import KeyList +from ..procedure import encrypt +from ..converter import M209 + + +PLAINTEXT = 'ATTACK AT DAWN' +CIPHERTEXT = 'GGABC DEFFM NQHNL CAARZ OLTVX GGABC DEFFM' + + +class ProcedureTestCase(unittest.TestCase): + + def setUp(self): + self.fm = KeyList(indicator="FM", + lugs = '1-0 2-0*8 0-3*7 0-4*5 0-5*2 1-5 1-6 3-4 4-5', + pin_list = [ + 'BCEJOPSTUVXY', + 'ACDHJLMNOQRUYZ', + 'AEHJLOQRUV', + 'DFGILMNPQS', + 'CEHIJLNPS', + 'ACDFHIMN' + ], + letter_check = 'TNMYS CRMKK UHLKW LDQHM RQOLW R') + self.m_209 = M209() + + def test_encrypt(self): + + result = encrypt(self.m_209, PLAINTEXT, key_list=self.fm, + ext_msg_ind='ABCDEF', sys_ind='G') + + self.assertEqual(result, CIPHERTEXT)