{
 "cells": [
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "# XOR decryption (Euler Problem 59)"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {
    "collapsed": true
   },
   "source": [
    "[https://projecteuler.net/problem=59](https://projecteuler.net/problem=59)\n",
    "\n",
    "Each character on a computer is assigned a unique code and the preferred standard is ASCII (American Standard Code for Information Interchange). For example, uppercase A = 65, asterisk (*) = 42, and lowercase k = 107.\n",
    "\n",
    "A modern encryption method is to take a text file, convert the bytes to ASCII, then XOR each byte with a given value, taken from a secret key. The advantage with the XOR function is that using the same encryption key on the cipher text, restores the plain text; for example, 65 XOR 42 = 107, then 107 XOR 42 = 65.\n",
    "\n",
    "For unbreakable encryption, the key is the same length as the plain text message, and the key is made up of random bytes. The user would keep the encrypted message and the encryption key in different locations, and without both \"halves\", it is impossible to decrypt the message.\n",
    "\n",
    "Unfortunately, this method is impractical for most users, so the modified method is to use a password as a key. If the password is shorter than the message, which is likely, the key is repeated cyclically throughout the message. The balance for this method is using a sufficiently long password key for security, but short enough to be memorable.\n",
    "\n",
    "Your task has been made easy, as the encryption key consists of three lower case characters. Using cipher.txt (right click and 'Save Link/Target As...'), a file containing the encrypted ASCII codes, and the knowledge that the plain text must contain common English words, decrypt the message and find the sum of the ASCII values in the original text."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [],
   "source": [
    "def get_encrypted_msg():\n",
    "    with open(\"EulerProblem059.txt\", \"r\") as f:\n",
    "        msg = [int(d) for d in f.read().split(',')]\n",
    "        return msg\n",
    "    \n",
    "def get_words():\n",
    "    with open(\"EulerProblem042.txt\", \"r\") as f:\n",
    "        words = [w.strip('\"') for w in f.read().split(\",\")]\n",
    "        return set(words)\n",
    "\n",
    "assert(65 ^ 42 == 107)\n",
    "msg = get_encrypted_msg()\n",
    "word_list = get_words()"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "We only have a three leter code which makes this problem easy to solve. Normally, "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {},
   "outputs": [],
   "source": [
    "def generate_keys():\n",
    "    alphabet = list(map(chr, range(97, 123)))\n",
    "    return [(ord(a), ord(b), ord(c)) for a in alphabet for b in alphabet for c in alphabet]\n",
    "\n",
    "keys = generate_keys()\n",
    "assert(len(keys) == 26 * 26 * 26)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "metadata": {},
   "outputs": [],
   "source": [
    "def to_string(msg):\n",
    "    return \"\".join(map(chr, msg))\n",
    "\n",
    "assert(to_string([65, 42, 107]) == \"A*k\")\n",
    "\n",
    "def to_ascii(msg):\n",
    "    return list(map(ord, msg))\n",
    "\n",
    "assert(to_ascii(\"A*k\") == [65, 42, 107])"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "metadata": {},
   "outputs": [],
   "source": [
    "def decrypt_msg(key, msg):\n",
    "    return [d ^ key[i % len(key)] for i, d in enumerate(msg)]\n",
    "\n",
    "assert(decrypt_msg([1, 2, 3], [1, 2, 3, 4, 5, 7]) == [0, 0, 0, 5, 7, 4])\n",
    "\n",
    "def clean_word(word):\n",
    "    return word.strip('').strip(\",\").strip(\".\").strip(\"!\").strip(\"?\").upper()\n",
    "\n",
    "def test_key(key, msg):\n",
    "    msg = to_string(decrypt_msg(key, msg))\n",
    "    word_count = 0\n",
    "    words = msg.split(\" \")\n",
    "    for word in words:\n",
    "        if clean_word(word) in word_list:\n",
    "            word_count += 1   \n",
    "    return word_count / len(words)\n",
    "    "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "metadata": {},
   "outputs": [],
   "source": [
    "ss = [(test_key(key, msg), key) for key in keys]\n",
    "s = sorted(ss, reverse=True)[0]\n",
    "key = s[1]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 6,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "(The Gospel of John, chapter 1) 1 In the beginning the Word already existed. He was with God, and he was God. 2 He was in the beginning with God. 3 He created everything there is. Nothing exists that he didn't make. 4 Life itself was in him, and this life gives light to everyone. 5 The light shines through the darkness, and the darkness can never extinguish it. 6 God sent John the Baptist 7 to tell everyone about the light so that everyone might believe because of his testimony. 8 John himself was not the light; he was only a witness to the light. 9 The one who is the true light, who gives light to everyone, was going to come into the world. 10 But although the world was made through him, the world didn't recognize him when he came. 11 Even in his own land and among his own people, he was not accepted. 12 But to all who believed him and accepted him, he gave the right to become children of God. 13 They are reborn! This is not a physical birth resulting from human passion or plan, this rebirth comes from God.14 So the Word became human and lived here on earth among us. He was full of unfailing love and faithfulness. And we have seen his glory, the glory of the only Son of the Father.\n"
     ]
    }
   ],
   "source": [
    "print(to_string(decrypt_msg(key, msg)))"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 7,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "107359\n"
     ]
    }
   ],
   "source": [
    "s = sum(decrypt_msg(key, msg))\n",
    "print(s)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {
    "collapsed": true
   },
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "completion_date": "Sat, 5 Jan 2019, 00:10",
  "kernelspec": {
   "display_name": "Python 3",
   "language": "python3.6",
   "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.6.5"
  },
  "tags": [
   "xor",
   "cipher",
   "encryption"
  ]
 },
 "nbformat": 4,
 "nbformat_minor": 2
}
