summaryrefslogtreecommitdiff
path: root/src/utils/CryptoUtils.scala
blob: 267889505db31a00792470e4d36886835e054258 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
package ixee.cryptopals.utils

import ixee.cryptopals.utils.crypto._
import ixee.cryptopals.utils.TupleUtils._
import ixee.cryptopals.utils.StreamUtils._
import ixee.cryptopals.utils.FunctionUtils._
import ixee.cryptopals.utils.ConversionUtils._
import javax.crypto.Cipher
import javax.crypto.spec.SecretKeySpec

object CryptoUtils {

  def pkcs7pad(s: Seq[Byte], blockSize: Int): Seq[Byte] = {
    val padLength = blockSize - (s.length % blockSize)
    s ++ Stream.continually(padLength.toByte).take(padLength)
  }

  def stripPkcs7Pad(s: Seq[Byte]): Seq[Byte] =
    s.dropRight(s.last)

  def cbcEncrypt(builder: CbcBuilder)(data: Seq[Byte]) =
    builder.encrypt.end(data)

  def cbcDecrypt(builder: CbcBuilder)(data: Seq[Byte]) =
    stripPkcs7Pad(builder.decrypt.end(data))

  def ecbEncrypt(builder: EcbBuilder)(data: Seq[Byte]) =
    builder.encrypt.end(data)

  def ecbDecrypt(builder: EcbBuilder)(data: Seq[Byte]) =
    stripPkcs7Pad(builder.decrypt.end(data))

  def detectMode(xs: Seq[Byte]): String = {
    def dupBlocks(xs: Seq[Seq[Byte]]) =
      pairsOf(xs).map(tup(_ == _)).count(_ == true)

    def countDupBlocks(xs: Seq[Byte]): Int =
      dupBlocks(xs.grouped(16).toSeq.init.toStream)

                              //.... well very probably.
    if (countDupBlocks(xs) > 0) "ECB"
    else "CBC"
  }

  def isEcb(xs: Seq[Byte]): Boolean = detectMode(xs) == "ECB"

  def detectEcbBlockSize(encryptor: Seq[Byte] => Seq[Byte]): Int = {
    val pad = (2 to 512 by 2).map("@" * _).zipWithIndex.map {
      _.mapAll(_1 = _.asBytes, _2 = _ + 1)
    }

    val minPadSize = pad
      .map( { encryptor(_) } <-: _ )
      .find( { x => isEcb(x._1) })
      .map(_._2)

    minPadSize.get // if this was a None we have serious possibility of this not being ECB
  }

  def extractUnknownViaEcbOracle(encrypt: Seq[Byte] => Seq[Byte]) = {
    val blockSize = detectEcbBlockSize(encrypt)
    val baseCiphertext = encrypt(Seq())
    def blocksIn(xs: Seq[Byte]) = xs.length / blockSize
    val baseBlockCount = blocksIn(baseCiphertext)

    def rainbow(prefix: Seq[Byte]): Map[Seq[Byte], Byte] = {
      (0 to 255)
        .map(_.toByte)
        .map(prefix :+ _)
        .map(encrypt)
        .map(_.take(16).toSeq)
        .zipWithIndex
        .map(_ :-> { _.toByte } )
        .toMap
    }

    def suffixRainbow(prefix: Seq[Byte]): Map[Seq[Byte], Byte] = {
      (0 to 255)
        .map(_.toByte)
        .map(_ +: prefix)
        .map(encrypt)
        .map(_.take(16).toSeq)
        .zipWithIndex
        .map(_ :-> { _.toByte } )
        .toMap
    }

    def probeFirstBlockAndPaddings: (Seq[Byte], Map[Int, Seq[Byte]]) = {
      def prefix(known: Seq[Byte]) = ("@" * (blockSize - 1 - known.length)).asBytes
      def genRainbow(known: Seq[Byte]) = rainbow(prefix(known) ++ known)
      def firstCryptedBlock(known: Seq[Byte]) = encrypt(prefix(known)).take(blockSize).toSeq
      def nextByte(known: Seq[Byte]) = genRainbow(known)(firstCryptedBlock(known))
      (0 until 16).foldLeft((Seq[Byte](), Map[Int, Seq[Byte]]())) { (ac, idx) =>
        (ac._1 :+ nextByte(ac._1), ac._2 + (idx -> encrypt(prefix(ac._1))))
      }
    }

    def probeLastBlockSize: Int = {
      val firstLargerCiphertext = (0 until blockSize)
        .map(" " * _).map(_.asBytes).map(encrypt)
        .zipWithIndex
        .find(x => blocksIn(x._1) != baseBlockCount)

        // this will always be Some(_) because
        // somewhere between 0..blockSize WILL grow the text.
      blockSize - firstLargerCiphertext.get._2
    }

    val (firstBlock, ciphertexts) = probeFirstBlockAndPaddings
    /*
     * zip together cipher blocks so that they look like
     * rot0b0, rot1b0, rot2b0, rot3b0, rot4b0, rot5b0, rot6b0, rot7b0
     * rot0b1, rot1b1, rot2b1, ...
     *
     */
    val cipherBlocks = ciphertexts.toSeq.sortBy(_._1).map(_ :-> { x =>
      x.grouped(16).toSeq
    })

    val (middleBlocks, lastBlocks) =
      cipherBlocks.foldLeft((Seq[Seq[Seq[Byte]]](), Seq[Option[Seq[Byte]]]()))( (ac, curr) => {
        val maybeLastBlock =
          if (curr._2.length > baseBlockCount) Some(curr._2.last)
          else None

        val middleBlocks = curr._2.take(baseBlockCount).tail

        (ac._1 :+ middleBlocks, maybeLastBlock +: ac._2)
      }) :-> { _.flatten :+ baseCiphertext.takeRight(16)}

    def breakLastBlock(blocks: Seq[Seq[Byte]]) = {
      /*
       * plaintext looks something like...
       * X 15 15 15 15 15 ... 15
       * Y X  14 14 14 14 ... 14
       * ... pkcs7
       */
      // drop the first block because it will be 16 16 16 16 16 16 ... 16
      blocks.tail.foldLeft(Seq[Byte]()) { (bytes, block) => {
        val postfix = pkcs7pad("?".asBytes ++ bytes, blockSize).tail
        suffixRainbow(postfix)(block) +: bytes
      }}
    }

    def breakBlock(plaintext: Seq[Byte], padded: Seq[Seq[Byte]]): Seq[Byte] =
      padded.foldLeft(Seq[Byte]()) { (blockText: Seq[Byte], block: Seq[Byte]) => {
        val prefix = (plaintext ++ blockText).takeRight(blockSize - 1)
        val b = rainbow(prefix)(block)
        blockText :+ b
      }}

    val middleBytes =
      middleBlocks.transpose.init.foldLeft(firstBlock) { (plaintext, padded: Seq[Seq[Byte]]) =>
        plaintext ++ breakBlock(plaintext, padded)
      }

    middleBytes ++ breakLastBlock(lastBlocks)
  }

}