From 533fd44ecc7048fa6480b1f6f4488088b3399abc Mon Sep 17 00:00:00 2001 From: Neil Smith Date: Sat, 14 May 2016 08:26:16 +0100 Subject: [PATCH] Built enigma machine --- enigma.ipynb | 1400 ++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 1400 insertions(+) create mode 100644 enigma.ipynb diff --git a/enigma.ipynb b/enigma.ipynb new file mode 100644 index 0000000..bfd3302 --- /dev/null +++ b/enigma.ipynb @@ -0,0 +1,1400 @@ +{ + "cells": [ + { + "cell_type": "markdown", + "metadata": {}, + "source": [ + "# Enigma machine\n", + "Specification from [Codes and Ciphers](http://www.codesandciphers.org.uk/enigma/rotorspec.htm) page.\n", + "\n", + "Example Enigma machines from [Louise Dale](http://enigma.louisedade.co.uk/enigma.html) (full simulation) and [EnigmaCo](http://enigmaco.de/enigma/enigma.html) (good animation of the wheels, but no ring settings).\n", + "\n", + "There's also the nice Enigma simulator for Android by [Franklin Heath](https://franklinheath.co.uk/2012/02/04/our-first-app-published-enigma-simulator/), available on the [Google Play store](https://play.google.com/store/apps/details?id=uk.co.franklinheath.enigmasim&hl=en_GB)." + ] + }, + { + "cell_type": "code", + "execution_count": 333, + "metadata": { + "collapsed": true + }, + "outputs": [], + "source": [ + "import string\n", + "\n", + "cat = ''.join" + ] + }, + { + "cell_type": "code", + "execution_count": 334, + "metadata": { + "collapsed": true + }, + "outputs": [], + "source": [ + "wheel_i_spec = 'ekmflgdqvzntowyhxuspaibrcj'\n", + "wheel_ii_spec = 'ajdksiruxblhwtmcqgznpyfvoe'\n", + "wheel_iii_spec = 'bdfhjlcprtxvznyeiwgakmusqo'\n", + "wheel_iv_spec = 'esovpzjayquirhxlnftgkdcmwb'\n", + "wheel_v_spec = 'vzbrgityupsdnhlxawmjqofeck'\n", + "wheel_vi_spec = 'jpgvoumfyqbenhzrdkasxlictw'\n", + "wheel_vii_spec = 'nzjhgrcxmyswboufaivlpekqdt'\n", + "wheel_viii_spec = 'fkqhtlxocbjspdzramewniuygv'\n", + "beta_wheel_spec = 'leyjvcnixwpbqmdrtakzgfuhos'\n", + "gamma_wheel_spec = 'fsokanuerhmbtiycwlqpzxvgjd'\n", + "\n", + "wheel_i_pegs = ['q']\n", + "wheel_ii_pegs = ['e']\n", + "wheel_iii_pegs = ['v']\n", + "wheel_iv_pegs = ['j']\n", + "wheel_v_pegs = ['z']\n", + "wheel_vi_pegs = ['z', 'm']\n", + "wheel_vii_pegs = ['z', 'm']\n", + "wheel_viii_pegs = ['z', 'm']\n", + "\n", + "reflector_b_spec = '(ay) (br) (cu) (dh) (eq) (fs) (gl) (ip) (jx) (kn) (mo) (tz) (vw)'\n", + "reflector_c_spec = '(af) (bv) (cp) (dj) (ei) (go) (hy) (kr) (lz) (mx) (nw) (tq) (su)'" + ] + }, + { + "cell_type": "code", + "execution_count": 335, + "metadata": { + "collapsed": false + }, + "outputs": [], + "source": [ + "class LetterTransformer(object):\n", + " def __init__(self, specification, raw_transform=False):\n", + " if raw_transform:\n", + " transform = specification\n", + " else:\n", + " transform = self.parse_specification(specification)\n", + " self.validate_transform(transform)\n", + " self.make_transform_map(transform)\n", + " \n", + " def parse_specification(self, specification):\n", + " return specification\n", + " \n", + " def validate_transform(self, transform):\n", + " \"\"\"A set of pairs, of from-to\"\"\"\n", + " if len(transform) != 26:\n", + " raise ValueError(\"Transform specification has {} pairs, requires 26\".\n", + " format(len(transform)))\n", + " for p in transform:\n", + " if len(p) != 2:\n", + " raise ValueError(\"Not all mappings in transform \"\n", + " \"have two elements\")\n", + " if len(set([p[0] for p in transform])) != 26:\n", + " raise ValueError(\"Transform specification must list 26 origin letters\") \n", + " if len(set([p[1] for p in transform])) != 26:\n", + " raise ValueError(\"Transform specification must list 26 destination letters\") \n", + "\n", + " def make_empty_transform(self):\n", + " self.forward_map = [0] * 26\n", + " self.backward_map = [0] * 26\n", + " \n", + " def make_transform_map(self, transform):\n", + " self.make_empty_transform()\n", + " for p in transform:\n", + " self.forward_map[ord(p[0]) - ord('a')] = ord(p[1]) - ord('a')\n", + " self.backward_map[ord(p[1]) - ord('a')] = ord(p[0]) - ord('a')\n", + " return self.forward_map, self.backward_map\n", + " \n", + " def forward(self, letter):\n", + " if letter in string.ascii_lowercase:\n", + " return chr(\n", + " (self.forward_map[(ord(letter) - ord('a')) % 26] + ord('a')))\n", + " else:\n", + " return ''\n", + " \n", + " def backward(self, letter):\n", + " if letter in string.ascii_lowercase:\n", + " return chr(\n", + " (self.backward_map[(ord(letter) - ord('a')) % 26] + ord('a')))\n", + " else:\n", + " return ''" + ] + }, + { + "cell_type": "code", + "execution_count": 336, + "metadata": { + "collapsed": false + }, + "outputs": [ + { + "data": { + "text/plain": [ + "[('z', 'a'),\n", + " ('a', 'b'),\n", + " ('b', 'c'),\n", + " ('c', 'd'),\n", + " ('d', 'e'),\n", + " ('e', 'f'),\n", + " ('f', 'g'),\n", + " ('g', 'h'),\n", + " ('h', 'i'),\n", + " ('i', 'j'),\n", + " ('j', 'k'),\n", + " ('k', 'l'),\n", + " ('l', 'm'),\n", + " ('m', 'n'),\n", + " ('n', 'o'),\n", + " ('o', 'p'),\n", + " ('p', 'q'),\n", + " ('q', 'r'),\n", + " ('r', 's'),\n", + " ('s', 't'),\n", + " ('t', 'u'),\n", + " ('u', 'v'),\n", + " ('v', 'w'),\n", + " ('w', 'x'),\n", + " ('x', 'y'),\n", + " ('y', 'z')]" + ] + }, + "execution_count": 336, + "metadata": {}, + "output_type": "execute_result" + } + ], + "source": [ + "tmap = [('z', 'a')] + [(l, string.ascii_lowercase[i+1]) for i, l in enumerate(string.ascii_lowercase[:-1])]\n", + "tmap" + ] + }, + { + "cell_type": "code", + "execution_count": 337, + "metadata": { + "collapsed": false, + "scrolled": true + }, + "outputs": [ + { + "data": { + "text/plain": [ + "([1,\n", + " 2,\n", + " 3,\n", + " 4,\n", + " 5,\n", + " 6,\n", + " 7,\n", + " 8,\n", + " 9,\n", + " 10,\n", + " 11,\n", + " 12,\n", + " 13,\n", + " 14,\n", + " 15,\n", + " 16,\n", + " 17,\n", + " 18,\n", + " 19,\n", + " 20,\n", + " 21,\n", + " 22,\n", + " 23,\n", + " 24,\n", + " 25,\n", + " 0],\n", + " [25,\n", + " 0,\n", + " 1,\n", + " 2,\n", + " 3,\n", + " 4,\n", + " 5,\n", + " 6,\n", + " 7,\n", + " 8,\n", + " 9,\n", + " 10,\n", + " 11,\n", + " 12,\n", + " 13,\n", + " 14,\n", + " 15,\n", + " 16,\n", + " 17,\n", + " 18,\n", + " 19,\n", + " 20,\n", + " 21,\n", + " 22,\n", + " 23,\n", + " 24])" + ] + }, + "execution_count": 337, + "metadata": {}, + "output_type": "execute_result" + } + ], + "source": [ + "lt = LetterTransformer(tmap)\n", + "lt.forward_map, lt.backward_map" + ] + }, + { + "cell_type": "code", + "execution_count": 338, + "metadata": { + "collapsed": false + }, + "outputs": [ + { + "data": { + "text/plain": [ + "'bcdefghijklmnopqrstuvwxyza'" + ] + }, + "execution_count": 338, + "metadata": {}, + "output_type": "execute_result" + } + ], + "source": [ + "cat(lt.forward(l) for l in string.ascii_lowercase)" + ] + }, + { + "cell_type": "code", + "execution_count": 339, + "metadata": { + "collapsed": false + }, + "outputs": [ + { + "data": { + "text/plain": [ + "'zabcdefghijklmnopqrstuvwxy'" + ] + }, + "execution_count": 339, + "metadata": {}, + "output_type": "execute_result" + } + ], + "source": [ + "cat(lt.backward(l) for l in string.ascii_lowercase)" + ] + }, + { + "cell_type": "code", + "execution_count": 340, + "metadata": { + "collapsed": true + }, + "outputs": [], + "source": [ + "class Plugboard(LetterTransformer):\n", + " def parse_specification(self, specification):\n", + " return [tuple(p) for p in specification.split()]\n", + " \n", + " def validate_transform(self, transform):\n", + " \"\"\"A set of pairs, of from-to\"\"\"\n", + " for p in transform:\n", + " if len(p) != 2:\n", + " raise ValueError(\"Not all mappings in transform\"\n", + " \"have two elements\")\n", + " \n", + " def make_empty_transform(self):\n", + " self.forward_map = list(range(26))\n", + " self.backward_map = list(range(26))\n", + " \n", + " def make_transform_map(self, transform):\n", + " expanded_transform = transform + [tuple(reversed(p)) for p in transform]\n", + " return super(Plugboard, self).make_transform_map(expanded_transform)" + ] + }, + { + "cell_type": "code", + "execution_count": 341, + "metadata": { + "collapsed": false + }, + "outputs": [], + "source": [ + "pb = Plugboard([('a', 'z'), ('b', 'y')], raw_transform=True)" + ] + }, + { + "cell_type": "code", + "execution_count": 342, + "metadata": { + "collapsed": false + }, + "outputs": [ + { + "data": { + "text/plain": [ + "'zycdefghijklmnopqrstuvwxba'" + ] + }, + "execution_count": 342, + "metadata": {}, + "output_type": "execute_result" + } + ], + "source": [ + "cat(pb.forward(l) for l in string.ascii_lowercase)" + ] + }, + { + "cell_type": "code", + "execution_count": 343, + "metadata": { + "collapsed": false + }, + "outputs": [ + { + "data": { + "text/plain": [ + "'zycdefghijklmnopqrstuvwxba'" + ] + }, + "execution_count": 343, + "metadata": {}, + "output_type": "execute_result" + } + ], + "source": [ + "cat(pb.backward(l) for l in string.ascii_lowercase)" + ] + }, + { + "cell_type": "code", + "execution_count": 344, + "metadata": { + "collapsed": false + }, + "outputs": [], + "source": [ + "pb = Plugboard('az by')" + ] + }, + { + "cell_type": "code", + "execution_count": 345, + "metadata": { + "collapsed": false + }, + "outputs": [ + { + "data": { + "text/plain": [ + "('zycdefghijklmnopqrstuvwxba', 'zycdefghijklmnopqrstuvwxba')" + ] + }, + "execution_count": 345, + "metadata": {}, + "output_type": "execute_result" + } + ], + "source": [ + "cat(pb.forward(l) for l in string.ascii_lowercase), cat(pb.backward(l) for l in string.ascii_lowercase)" + ] + }, + { + "cell_type": "code", + "execution_count": 346, + "metadata": { + "collapsed": true + }, + "outputs": [], + "source": [ + "class Reflector(Plugboard):\n", + " def parse_specification(self, specification):\n", + " return [tuple(p) for p in specification[1:-1].split(') (')]\n", + " \n", + " def validate_transform(self, transform):\n", + " if len(transform) != 13:\n", + " raise ValueError(\"Reflector specification has {} pairs, requires 13\".\n", + " format(len(transform)))\n", + " if len(set([p[0] for p in transform] + \n", + " [p[1] for p in transform])) != 26:\n", + " raise ValueError(\"Reflector specification does not contain 26 letters\")\n", + " try:\n", + " super(Reflector, self).validate_transform(transform)\n", + " except ValueError as v:\n", + " raise ValueError(\"Not all mappings in reflector have two elements\")" + ] + }, + { + "cell_type": "code", + "execution_count": 347, + "metadata": { + "collapsed": false + }, + "outputs": [ + { + "data": { + "text/plain": [ + "[('a', 'y'),\n", + " ('b', 'r'),\n", + " ('c', 'u'),\n", + " ('d', 'h'),\n", + " ('e', 'q'),\n", + " ('f', 's'),\n", + " ('g', 'l'),\n", + " ('i', 'p'),\n", + " ('j', 'x'),\n", + " ('k', 'n'),\n", + " ('m', 'o'),\n", + " ('t', 'z'),\n", + " ('v', 'w')]" + ] + }, + "execution_count": 347, + "metadata": {}, + "output_type": "execute_result" + } + ], + "source": [ + "# reflector_b_text = '(AY) (BR) (CU) (DH) (EQ) (FS) (GL) (IP) (JX) (KN) (MO) (TZ) (VW)'\n", + "reflector_b_l = [tuple(p) for p in reflector_b_spec.lower()[1:-1].split(') (')]\n", + "reflector_b_l" + ] + }, + { + "cell_type": "code", + "execution_count": 348, + "metadata": { + "collapsed": false + }, + "outputs": [], + "source": [ + "reflector_b = Reflector(reflector_b_spec, raw_transform=False)" + ] + }, + { + "cell_type": "code", + "execution_count": 349, + "metadata": { + "collapsed": false + }, + "outputs": [ + { + "data": { + "text/plain": [ + "'yruhqsldpxngokmiebfzcwvjat'" + ] + }, + "execution_count": 349, + "metadata": {}, + "output_type": "execute_result" + } + ], + "source": [ + "cat(reflector_b.forward(l) for l in string.ascii_lowercase)" + ] + }, + { + "cell_type": "code", + "execution_count": 350, + "metadata": { + "collapsed": true + }, + "outputs": [], + "source": [ + "reflector_c = Reflector(reflector_c_spec)" + ] + }, + { + "cell_type": "code", + "execution_count": 351, + "metadata": { + "collapsed": false + }, + "outputs": [ + { + "data": { + "text/plain": [ + "'fvpjiaoyedrzxwgctkuqsbnmhl'" + ] + }, + "execution_count": 351, + "metadata": {}, + "output_type": "execute_result" + } + ], + "source": [ + "cat(reflector_c.forward(l) for l in string.ascii_lowercase)" + ] + }, + { + "cell_type": "code", + "execution_count": 352, + "metadata": { + "collapsed": true + }, + "outputs": [], + "source": [ + "class SimpleWheel(LetterTransformer):\n", + " def __init__(self, transform, position='a', raw_transform=False):\n", + " super(SimpleWheel, self).__init__(transform, raw_transform)\n", + " self.set_position(position)\n", + " \n", + " def parse_specification(self, specification):\n", + " return list(zip(string.ascii_lowercase, specification))\n", + " \n", + " def set_position(self, position):\n", + " self.position = ord(position) - ord('a')\n", + " self.position_l = position\n", + " \n", + " def forward(self, letter):\n", + " if letter in string.ascii_lowercase:\n", + " return chr(\n", + " (self.forward_map[(ord(letter) - ord('a') + self.position) % 26] - \n", + " self.position) % 26 + \n", + " ord('a'))\n", + " else:\n", + " return ''\n", + " \n", + " def backward(self, letter):\n", + " if letter in string.ascii_lowercase:\n", + " return chr(\n", + " (self.backward_map[(ord(letter) - ord('a') + self.position) % 26] - \n", + " self.position) % 26 + \n", + " ord('a'))\n", + " else:\n", + " return ''\n", + " \n", + " def advance(self):\n", + " self.position = (self.position + 1) % 26\n", + " self.position_l = chr(self.position + ord('a'))\n", + " return self.position" + ] + }, + { + "cell_type": "code", + "execution_count": 353, + "metadata": { + "collapsed": false, + "scrolled": true + }, + "outputs": [ + { + "data": { + "text/plain": [ + "[('a', 'e'),\n", + " ('b', 'k'),\n", + " ('c', 'm'),\n", + " ('d', 'f'),\n", + " ('e', 'l'),\n", + " ('f', 'g'),\n", + " ('g', 'd'),\n", + " ('h', 'q'),\n", + " ('i', 'v'),\n", + " ('j', 'z'),\n", + " ('k', 'n'),\n", + " ('l', 't'),\n", + " ('m', 'o'),\n", + " ('n', 'w'),\n", + " ('o', 'y'),\n", + " ('p', 'h'),\n", + " ('q', 'x'),\n", + " ('r', 'u'),\n", + " ('s', 's'),\n", + " ('t', 'p'),\n", + " ('u', 'a'),\n", + " ('v', 'i'),\n", + " ('w', 'b'),\n", + " ('x', 'r'),\n", + " ('y', 'c'),\n", + " ('z', 'j')]" + ] + }, + "execution_count": 353, + "metadata": {}, + "output_type": "execute_result" + } + ], + "source": [ + "rotor_1_transform = list(zip(string.ascii_lowercase, 'EKMFLGDQVZNTOWYHXUSPAIBRCJ'.lower()))\n", + "rotor_1_transform" + ] + }, + { + "cell_type": "code", + "execution_count": 354, + "metadata": { + "collapsed": false + }, + "outputs": [], + "source": [ + "wheel_1 = SimpleWheel(rotor_1_transform, raw_transform=True)" + ] + }, + { + "cell_type": "code", + "execution_count": 355, + "metadata": { + "collapsed": false + }, + "outputs": [ + { + "data": { + "text/plain": [ + "('ekmflgdqvzntowyhxuspaibrcj', 'uwygadfpvzbeckmthxslrinqoj')" + ] + }, + "execution_count": 355, + "metadata": {}, + "output_type": "execute_result" + } + ], + "source": [ + "cat(wheel_1.forward(l) for l in string.ascii_lowercase), cat(wheel_1.backward(l) for l in string.ascii_lowercase)" + ] + }, + { + "cell_type": "code", + "execution_count": 356, + "metadata": { + "collapsed": false + }, + "outputs": [], + "source": [ + "wheel_2 = SimpleWheel(wheel_ii_spec)" + ] + }, + { + "cell_type": "code", + "execution_count": 357, + "metadata": { + "collapsed": false + }, + "outputs": [ + { + "data": { + "text/plain": [ + "('ajdksiruxblhwtmcqgznpyfvoe', 'ajpczwrlfbdkotyuqgenhxmivs')" + ] + }, + "execution_count": 357, + "metadata": {}, + "output_type": "execute_result" + } + ], + "source": [ + "cat(wheel_2.forward(l) for l in string.ascii_lowercase), cat(wheel_2.backward(l) for l in string.ascii_lowercase)" + ] + }, + { + "cell_type": "code", + "execution_count": 358, + "metadata": { + "collapsed": false + }, + "outputs": [ + { + "data": { + "text/plain": [ + "'cegikboqswuymxdhvfzjltrpna'" + ] + }, + "execution_count": 358, + "metadata": {}, + "output_type": "execute_result" + } + ], + "source": [ + "wheel_3 = SimpleWheel(wheel_iii_spec)\n", + "wheel_3.set_position('a')\n", + "wheel_3.advance()\n", + "cat(wheel_3.forward(l) for l in string.ascii_lowercase)" + ] + }, + { + "cell_type": "code", + "execution_count": 524, + "metadata": { + "collapsed": false + }, + "outputs": [], + "source": [ + "class Wheel(SimpleWheel):\n", + " def __init__(self, transform, ring_peg_letters, ring_setting=1, position='a', raw_transform=False):\n", + " self.ring_peg_letters = ring_peg_letters\n", + " self.ring_setting = ring_setting\n", + " super(Wheel, self).__init__(transform, position=position, raw_transform=raw_transform)\n", + " self.set_position(position)\n", + " \n", + " def set_position(self, position):\n", + " super(Wheel, self).set_position(position)\n", + " self.peg_positions = [(ord(p) - ord(position)) % 26 for p in self.ring_peg_letters]\n", + " self.position = (self.position - self.ring_setting + 1) % 26\n", + " \n", + " def advance(self):\n", + " super(Wheel, self).advance()\n", + " self.peg_positions = [(p - 1) % 26 for p in self.peg_positions]\n", + " return self.position" + ] + }, + { + "cell_type": "code", + "execution_count": 525, + "metadata": { + "collapsed": false + }, + "outputs": [], + "source": [ + "wheel_3 = Wheel(wheel_iii_spec, wheel_iii_pegs, position='b', ring_setting=1)" + ] + }, + { + "cell_type": "code", + "execution_count": 526, + "metadata": { + "collapsed": false + }, + "outputs": [ + { + "data": { + "text/plain": [ + "(1, [20])" + ] + }, + "execution_count": 526, + "metadata": {}, + "output_type": "execute_result" + } + ], + "source": [ + "wheel_3.position, wheel_3.peg_positions" + ] + }, + { + "cell_type": "code", + "execution_count": 527, + "metadata": { + "collapsed": false + }, + "outputs": [ + { + "data": { + "text/plain": [ + "(25, [24, 11])" + ] + }, + "execution_count": 527, + "metadata": {}, + "output_type": "execute_result" + } + ], + "source": [ + "wheel_6 = Wheel(wheel_vi_spec, wheel_vi_pegs, position='b', ring_setting=3)\n", + "wheel_6.position, wheel_6.peg_positions" + ] + }, + { + "cell_type": "code", + "execution_count": 528, + "metadata": { + "collapsed": false, + "scrolled": true + }, + "outputs": [ + { + "name": "stdout", + "output_type": "stream", + "text": [ + "0 [23, 10]\n", + "1 [22, 9]\n", + "2 [21, 8]\n", + "3 [20, 7]\n", + "4 [19, 6]\n", + "5 [18, 5]\n", + "6 [17, 4]\n", + "7 [16, 3]\n", + "8 [15, 2]\n", + "9 [14, 1]\n", + "10 [13, 0]\n", + "11 [12, 25]\n", + "12 [11, 24]\n", + "13 [10, 23]\n", + "14 [9, 22]\n", + "15 [8, 21]\n", + "16 [7, 20]\n", + "17 [6, 19]\n", + "18 [5, 18]\n", + "19 [4, 17]\n", + "20 [3, 16]\n", + "21 [2, 15]\n", + "22 [1, 14]\n", + "23 [0, 13]\n", + "24 [25, 12]\n", + "25 [24, 11]\n", + "0 [23, 10]\n" + ] + } + ], + "source": [ + "for _ in range(27):\n", + " wheel_6.advance()\n", + " print(wheel_6.position, wheel_6.peg_positions)" + ] + }, + { + "cell_type": "code", + "execution_count": 529, + "metadata": { + "collapsed": true + }, + "outputs": [], + "source": [ + "class Enigma(object):\n", + " def __init__(self, reflector_spec,\n", + " left_wheel_spec, left_wheel_pegs,\n", + " middle_wheel_spec, middle_wheel_pegs,\n", + " right_wheel_spec, right_wheel_pegs,\n", + " left_ring_setting, middle_ring_setting, right_ring_setting,\n", + " plugboard_setting):\n", + " self.reflector = Reflector(reflector_spec)\n", + " self.left_wheel = Wheel(left_wheel_spec, left_wheel_pegs, ring_setting=left_ring_setting)\n", + " self.middle_wheel = Wheel(middle_wheel_spec, middle_wheel_pegs, ring_setting=middle_ring_setting)\n", + " self.right_wheel = Wheel(right_wheel_spec, right_wheel_pegs, ring_setting=right_ring_setting)\n", + " self.plugboard = Plugboard(plugboard_setting)\n", + " \n", + " def set_wheels(self, left_wheel_position, middle_wheel_position, right_wheel_position):\n", + " self.left_wheel.set_position(left_wheel_position)\n", + " self.middle_wheel.set_position(middle_wheel_position)\n", + " self.right_wheel.set_position(right_wheel_position)\n", + " \n", + " def lookup(self, letter):\n", + " a = self.plugboard.forward(letter)\n", + " b = self.right_wheel.forward(a)\n", + " c = self.middle_wheel.forward(b)\n", + " d = self.left_wheel.forward(c)\n", + " e = self.reflector.forward(d)\n", + " f = self.left_wheel.backward(e)\n", + " g = self.middle_wheel.backward(f)\n", + " h = self.right_wheel.backward(g)\n", + " i = self.plugboard.backward(h)\n", + " return i\n", + " \n", + " def advance(self):\n", + " advance_middle = False\n", + " advance_left = False\n", + " if 0 in self.right_wheel.peg_positions:\n", + " advance_middle = True\n", + " if 0 in self.middle_wheel.peg_positions:\n", + " advance_left = True\n", + " if 0 in self.middle_wheel.peg_positions and 25 in self.right_wheel.peg_positions:\n", + " advance_middle = True\n", + " advance_left = True\n", + " self.right_wheel.advance()\n", + " if advance_middle: self.middle_wheel.advance()\n", + " if advance_left: self.left_wheel.advance()\n", + " \n", + " def encipher_letter(self, letter):\n", + " self.advance()\n", + " return self.lookup(letter)\n", + " \n", + " def encipher(self, message):\n", + " enciphered = ''\n", + " for letter in message:\n", + " enciphered += self.encipher_letter(letter)\n", + " return enciphered" + ] + }, + { + "cell_type": "code", + "execution_count": 548, + "metadata": { + "collapsed": false + }, + "outputs": [], + "source": [ + "enigma = Enigma(reflector_b_spec, \n", + " wheel_i_spec, wheel_i_pegs,\n", + " wheel_ii_spec, wheel_ii_pegs,\n", + " wheel_iii_spec, wheel_iii_pegs,\n", + " 1, 1, 1,\n", + " '')" + ] + }, + { + "cell_type": "code", + "execution_count": 549, + "metadata": { + "collapsed": false + }, + "outputs": [ + { + "data": { + "text/plain": [ + "'u'" + ] + }, + "execution_count": 549, + "metadata": {}, + "output_type": "execute_result" + } + ], + "source": [ + "enigma.lookup('a')" + ] + }, + { + "cell_type": "code", + "execution_count": 550, + "metadata": { + "collapsed": false + }, + "outputs": [ + { + "data": { + "text/plain": [ + "'a'" + ] + }, + "execution_count": 550, + "metadata": {}, + "output_type": "execute_result" + } + ], + "source": [ + "enigma.lookup('u')" + ] + }, + { + "cell_type": "code", + "execution_count": 551, + "metadata": { + "collapsed": false, + "scrolled": true + }, + "outputs": [ + { + "name": "stdout", + "output_type": "stream", + "text": [ + "a a a ; [16] [4] [21]\n", + "a a b ; [16] [4] [20]\n", + "a a c ; [16] [4] [19]\n", + "a a d ; [16] [4] [18]\n", + "a a e ; [16] [4] [17]\n", + "a a f ; [16] [4] [16]\n", + "a a g ; [16] [4] [15]\n", + "a a h ; [16] [4] [14]\n", + "a a i ; [16] [4] [13]\n", + "a a j ; [16] [4] [12]\n", + "a a k ; [16] [4] [11]\n", + "a a l ; [16] [4] [10]\n", + "a a m ; [16] [4] [9]\n", + "a a n ; [16] [4] [8]\n", + "a a o ; [16] [4] [7]\n", + "a a p ; [16] [4] [6]\n", + "a a q ; [16] [4] [5]\n", + "a a r ; [16] [4] [4]\n", + "a a s ; [16] [4] [3]\n", + "a a t ; [16] [4] [2]\n", + "a a u ; [16] [4] [1]\n", + "a a v ; [16] [4] [0]\n", + "a b w ; [16] [3] [25]\n", + "a b x ; [16] [3] [24]\n", + "a b y ; [16] [3] [23]\n", + "a b z ; [16] [3] [22]\n" + ] + } + ], + "source": [ + "enigma.set_wheels('a', 'a', 'a')\n", + "for _ in range(26):\n", + " print(enigma.left_wheel.position_l, enigma.middle_wheel.position_l, enigma.right_wheel.position_l, ';',\n", + " enigma.left_wheel.peg_positions, enigma.middle_wheel.peg_positions, enigma.right_wheel.peg_positions)\n", + " enigma.advance()" + ] + }, + { + "cell_type": "code", + "execution_count": 552, + "metadata": { + "collapsed": false, + "scrolled": true + }, + "outputs": [ + { + "name": "stdout", + "output_type": "stream", + "text": [ + "a d t ; [16] [1] [2]\n", + "a d u ; [16] [1] [1]\n", + "a d v ; [16] [1] [0]\n", + "a e w ; [16] [0] [25]\n", + "b f x ; [15] [25] [24]\n", + "b f y ; [15] [25] [23]\n", + "b f z ; [15] [25] [22]\n", + "b f a ; [15] [25] [21]\n", + "b f b ; [15] [25] [20]\n", + "b f c ; [15] [25] [19]\n", + "b f d ; [15] [25] [18]\n", + "b f e ; [15] [25] [17]\n", + "b f f ; [15] [25] [16]\n", + "b f g ; [15] [25] [15]\n", + "b f h ; [15] [25] [14]\n", + "b f i ; [15] [25] [13]\n", + "b f j ; [15] [25] [12]\n", + "b f k ; [15] [25] [11]\n", + "b f l ; [15] [25] [10]\n", + "b f m ; [15] [25] [9]\n", + "b f n ; [15] [25] [8]\n", + "b f o ; [15] [25] [7]\n", + "b f p ; [15] [25] [6]\n", + "b f q ; [15] [25] [5]\n", + "b f r ; [15] [25] [4]\n", + "b f s ; [15] [25] [3]\n" + ] + } + ], + "source": [ + "enigma.set_wheels('a', 'd', 't')\n", + "for _ in range(26):\n", + " print(enigma.left_wheel.position_l, enigma.middle_wheel.position_l, enigma.right_wheel.position_l, ';',\n", + " enigma.left_wheel.peg_positions, enigma.middle_wheel.peg_positions, enigma.right_wheel.peg_positions)\n", + " enigma.advance()" + ] + }, + { + "cell_type": "code", + "execution_count": 553, + "metadata": { + "collapsed": false + }, + "outputs": [ + { + "data": { + "text/plain": [ + "'i'" + ] + }, + "execution_count": 553, + "metadata": {}, + "output_type": "execute_result" + } + ], + "source": [ + "enigma.set_wheels('a', 'a', 'a')\n", + "enigma.advance()\n", + "enigma.lookup('h')" + ] + }, + { + "cell_type": "code", + "execution_count": 560, + "metadata": { + "collapsed": false + }, + "outputs": [ + { + "data": { + "text/plain": [ + "'ilbdartydc'" + ] + }, + "execution_count": 560, + "metadata": {}, + "output_type": "execute_result" + } + ], + "source": [ + "enigma.set_wheels('a', 'a', 'a')\n", + "ct = enigma.encipher('hellothere')\n", + "assert(ct == 'ilbdartydc')\n", + "ct" + ] + }, + { + "cell_type": "code", + "execution_count": 555, + "metadata": { + "collapsed": false + }, + "outputs": [ + { + "data": { + "text/plain": [ + "'hellothere'" + ] + }, + "execution_count": 555, + "metadata": {}, + "output_type": "execute_result" + } + ], + "source": [ + "enigma.set_wheels('a', 'a', 'a')\n", + "pt = enigma.encipher(ct)\n", + "pt" + ] + }, + { + "cell_type": "code", + "execution_count": 556, + "metadata": { + "collapsed": false + }, + "outputs": [ + { + "data": { + "text/plain": [ + "'i'" + ] + }, + "execution_count": 556, + "metadata": {}, + "output_type": "execute_result" + } + ], + "source": [ + "enigma.set_wheels('a', 'a', 'a')\n", + "enigma.advance()\n", + "enigma.lookup('h')" + ] + }, + { + "cell_type": "code", + "execution_count": 557, + "metadata": { + "collapsed": false + }, + "outputs": [ + { + "data": { + "text/plain": [ + "('h', 'q', 'q', 'x', 'j', 'z', 's', 'i', 'i')" + ] + }, + "execution_count": 557, + "metadata": {}, + "output_type": "execute_result" + } + ], + "source": [ + "enigma.set_wheels('a', 'a', 'a')\n", + "enigma.advance()\n", + "a = enigma.plugboard.forward('h')\n", + "b = enigma.right_wheel.forward(a)\n", + "c = enigma.middle_wheel.forward(b)\n", + "d = enigma.left_wheel.forward(c)\n", + "e = enigma.reflector.forward(d)\n", + "f = enigma.left_wheel.backward(e)\n", + "g = enigma.middle_wheel.backward(f)\n", + "h = enigma.right_wheel.backward(g)\n", + "i = enigma.plugboard.backward(h)\n", + "a, b, c, d, e, f, g, h, i" + ] + }, + { + "cell_type": "code", + "execution_count": 558, + "metadata": { + "collapsed": false + }, + "outputs": [ + { + "data": { + "text/plain": [ + "'h'" + ] + }, + "execution_count": 558, + "metadata": {}, + "output_type": "execute_result" + } + ], + "source": [ + "pb = Plugboard('')\n", + "pb.forward('h')" + ] + }, + { + "cell_type": "code", + "execution_count": 561, + "metadata": { + "collapsed": false + }, + "outputs": [ + { + "data": { + "text/plain": [ + "'bahxvfrpdc'" + ] + }, + "execution_count": 561, + "metadata": {}, + "output_type": "execute_result" + } + ], + "source": [ + "enigma.set_wheels('a', 'd', 't')\n", + "ct = enigma.encipher('hellothere')\n", + "assert(ct == 'bahxvfrpdc')\n", + "ct" + ] + }, + { + "cell_type": "code", + "execution_count": 562, + "metadata": { + "collapsed": false + }, + "outputs": [ + { + "data": { + "text/plain": [ + "'kvmmwrlqlqsqpeugjrcxzwpfyiyybwloewrouvkpoztceuwtfjzqwpbqldttsr'" + ] + }, + "execution_count": 562, + "metadata": {}, + "output_type": "execute_result" + } + ], + "source": [ + "enigma.set_wheels('b', 'd', 'q')\n", + "ct = enigma.encipher('aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa')\n", + "assert(ct == 'kvmmwrlqlqsqpeugjrcxzwpfyiyybwloewrouvkpoztceuwtfjzqwpbqldttsr')\n", + "assert(enigma.left_wheel.position_l == 'c')\n", + "assert(enigma.middle_wheel.position_l == 'h')\n", + "assert(enigma.right_wheel.position_l == 'a')\n", + "ct" + ] + }, + { + "cell_type": "code", + "execution_count": 563, + "metadata": { + "collapsed": false + }, + "outputs": [ + { + "data": { + "text/plain": [ + "'c'" + ] + }, + "execution_count": 563, + "metadata": {}, + "output_type": "execute_result" + } + ], + "source": [ + "enigma.left_wheel.position_l" + ] + }, + { + "cell_type": "code", + "execution_count": 564, + "metadata": { + "collapsed": true + }, + "outputs": [], + "source": [ + "# Setting sheet line 31 from http://www.codesandciphers.org.uk/enigma/enigma3.htm\n", + "# Enigma simulation settings are \n", + "# http://enigma.louisedade.co.uk/enigma.html?m3;b;b153;AFTX;AJFE;AU-BG-EY-FP-HL-IN-JZ-OS-QR-TX\n", + "w_enigma = Enigma(reflector_b_spec, \n", + " wheel_i_spec, wheel_i_pegs,\n", + " wheel_v_spec, wheel_v_pegs,\n", + " wheel_iii_spec, wheel_iii_pegs,\n", + " 6, 20, 24,\n", + " 'ua pf rq so ni ey bg hl tx zj')" + ] + }, + { + "cell_type": "code", + "execution_count": 566, + "metadata": { + "collapsed": false + }, + "outputs": [ + { + "name": "stdout", + "output_type": "stream", + "text": [ + "wzmfiwuntn\n" + ] + }, + { + "data": { + "text/plain": [ + "'wzmfiwuntn'" + ] + }, + "execution_count": 566, + "metadata": {}, + "output_type": "execute_result" + } + ], + "source": [ + "w_enigma.set_wheels('j', 'e', 'u')\n", + "ct = w_enigma.encipher('hellothere')\n", + "print(ct)\n", + "assert(ct == 'wzmfiwuntn')\n", + "ct" + ] + }, + { + "cell_type": "code", + "execution_count": 568, + "metadata": { + "collapsed": false + }, + "outputs": [ + { + "data": { + "text/plain": [ + "'ayraqvsfkhflwsmkqicvfwawswmiwvvlteb'" + ] + }, + "execution_count": 568, + "metadata": {}, + "output_type": "execute_result" + } + ], + "source": [ + "w_enigma.set_wheels('n', 'y', 'q')\n", + "ct = w_enigma.encipher('hellotomweshouldgetbeerorcoffeesoon')\n", + "ct" + ] + }, + { + "cell_type": "code", + "execution_count": null, + "metadata": { + "collapsed": true + }, + "outputs": [], + "source": [] + } + ], + "metadata": { + "kernelspec": { + "display_name": "Python 3", + "language": "python", + "name": "python3" + }, + "language_info": { + "codemirror_mode": { + "name": "ipython", + "version": 3 + }, + "file_extension": ".py", + "mimetype": "text/x-python", + "name": "python", + "nbconvert_exporter": "python", + "pygments_lexer": "ipython3", + "version": "3.5.1+" + } + }, + "nbformat": 4, + "nbformat_minor": 0 +} -- 2.34.1