Skip to content
This repository has been archived by the owner on Mar 14, 2020. It is now read-only.

Commit

Permalink
Modify SHA256 to inherit from MDConstruction
Browse files Browse the repository at this point in the history
  • Loading branch information
@Richard-W
Richard-W committed Aug 20, 2015
1 parent 8124e8d commit 4eed3f6
Showing 1 changed file with 63 additions and 116 deletions.
179 changes: 63 additions & 116 deletions src/main/scala/hash/SHA256.scala
View file
Original file line number Diff line number Diff line change
Expand Up @@ -21,131 +21,78 @@ import java.nio.{ ByteBuffer, ByteOrder }
import scala.annotation.tailrec

/** Secure Hash Algorithm 1 */
object SHA256 extends Hash {
object SHA256 extends MDConstruction[(Int, Int, Int, Int, Int, Int, Int, Int)] {

def apply: Iteratee[Seq[Byte], Seq[Byte]] = {
def bytes2word(bytes: Seq[Byte]): Int =
ByteBuffer.allocate(4).order(ByteOrder.BIG_ENDIAN).put(bytes.toArray).getInt(0)
private def bytes2word(bytes: Seq[Byte]): Int =
ByteBuffer.allocate(4).order(ByteOrder.BIG_ENDIAN).put(bytes.toArray).getInt(0)

def word2bytes(word: Int): Seq[Byte] =
ByteBuffer.allocate(4).order(ByteOrder.BIG_ENDIAN).putInt(word).array
private def word2bytes(word: Int): Seq[Byte] =
ByteBuffer.allocate(4).order(ByteOrder.BIG_ENDIAN).putInt(word).array

def rotRight(word: Int, r: Int) = {
(word >>> r) | (word << (32 - r))
}

val k = Seq(
0x428a2f98, 0x71374491, 0xb5c0fbcf, 0xe9b5dba5, 0x3956c25b, 0x59f111f1, 0x923f82a4, 0xab1c5ed5,
0xd807aa98, 0x12835b01, 0x243185be, 0x550c7dc3, 0x72be5d74, 0x80deb1fe, 0x9bdc06a7, 0xc19bf174,
0xe49b69c1, 0xefbe4786, 0x0fc19dc6, 0x240ca1cc, 0x2de92c6f, 0x4a7484aa, 0x5cb0a9dc, 0x76f988da,
0x983e5152, 0xa831c66d, 0xb00327c8, 0xbf597fc7, 0xc6e00bf3, 0xd5a79147, 0x06ca6351, 0x14292967,
0x27b70a85, 0x2e1b2138, 0x4d2c6dfc, 0x53380d13, 0x650a7354, 0x766a0abb, 0x81c2c92e, 0x92722c85,
0xa2bfe8a1, 0xa81a664b, 0xc24b8b70, 0xc76c51a3, 0xd192e819, 0xd6990624, 0xf40e3585, 0x106aa070,
0x19a4c116, 0x1e376c08, 0x2748774c, 0x34b0bcb5, 0x391c0cb3, 0x4ed8aa4a, 0x5b9cca4f, 0x682e6ff3,
0x748f82ee, 0x78a5636f, 0x84c87814, 0x8cc70208, 0x90befffa, 0xa4506ceb, 0xbef9a3f7, 0xc67178f2
)

def compressBlock(state: (Int, Int, Int, Int, Int, Int, Int, Int), block: Seq[Byte]): (Int, Int, Int, Int, Int, Int, Int, Int) = {
val w = ((block.grouped(4).toSeq map { bytes2word(_) }) ++ Seq.fill[Int](64) { 0 }).toArray
for (i <- (16 until 80)) {
val s0 = rotRight(w(i - 15), 7) ^ rotRight(w(i - 15), 18) ^ (w(i - 15) >>> 3)
val s1 = rotRight(w(i - 2), 17) ^ rotRight(w(i - 2), 19) ^ (w(i - 2) >>> 10)
w(i) = w(i - 16) + s0 + w(i - 7) + s1
}

var a: Int = state._1
var b: Int = state._2
var c: Int = state._3
var d: Int = state._4
var e: Int = state._5
var f: Int = state._6
var g: Int = state._7
var h: Int = state._8
var temp: Int = 0

for (i <- (0 until 64)) {
val s1 = rotRight(e, 6) ^ rotRight(e, 11) ^ rotRight(e, 25)
val ch = (e & f) ^ ((~e) & g)
val temp1 = h + s1 + ch + k(i) + w(i)
val s0 = rotRight(a, 2) ^ rotRight(a, 13) ^ rotRight(a, 22)
val maj = (a & b) ^ (a & c) ^ (b & c)
val temp2 = s0 + maj
private def rotRight(word: Int, r: Int) = {
(word >>> r) | (word << (32 - r))
}

h = g
g = f
f = e
e = d + temp1
d = c
c = b
b = a
a = temp1 + temp2
}
(state._1 + a, state._2 + b, state._3 + c, state._4 + d, state._5 + e, state._6 + f, state._7 + g, state._8 + h)
private val k = Seq(
0x428a2f98, 0x71374491, 0xb5c0fbcf, 0xe9b5dba5, 0x3956c25b, 0x59f111f1, 0x923f82a4, 0xab1c5ed5,
0xd807aa98, 0x12835b01, 0x243185be, 0x550c7dc3, 0x72be5d74, 0x80deb1fe, 0x9bdc06a7, 0xc19bf174,
0xe49b69c1, 0xefbe4786, 0x0fc19dc6, 0x240ca1cc, 0x2de92c6f, 0x4a7484aa, 0x5cb0a9dc, 0x76f988da,
0x983e5152, 0xa831c66d, 0xb00327c8, 0xbf597fc7, 0xc6e00bf3, 0xd5a79147, 0x06ca6351, 0x14292967,
0x27b70a85, 0x2e1b2138, 0x4d2c6dfc, 0x53380d13, 0x650a7354, 0x766a0abb, 0x81c2c92e, 0x92722c85,
0xa2bfe8a1, 0xa81a664b, 0xc24b8b70, 0xc76c51a3, 0xd192e819, 0xd6990624, 0xf40e3585, 0x106aa070,
0x19a4c116, 0x1e376c08, 0x2748774c, 0x34b0bcb5, 0x391c0cb3, 0x4ed8aa4a, 0x5b9cca4f, 0x682e6ff3,
0x748f82ee, 0x78a5636f, 0x84c87814, 0x8cc70208, 0x90befffa, 0xa4506ceb, 0xbef9a3f7, 0xc67178f2
)

def compressionFunction(state: (Int, Int, Int, Int, Int, Int, Int, Int), block: Seq[Byte]): (Int, Int, Int, Int, Int, Int, Int, Int) = {
val w = ((block.grouped(4).toSeq map { bytes2word(_) }) ++ Seq.fill[Int](64) { 0 }).toArray
for (i <- (16 until 80)) {
val s0 = rotRight(w(i - 15), 7) ^ rotRight(w(i - 15), 18) ^ (w(i - 15) >>> 3)
val s1 = rotRight(w(i - 2), 17) ^ rotRight(w(i - 2), 19) ^ (w(i - 2) >>> 10)
w(i) = w(i - 16) + s0 + w(i - 7) + s1
}

Iteratee.fold[Seq[Byte], (Int, Int, Int, Int, Int, Int, Int, Int, Long, Seq[Byte])](0x6a09e667, 0xbb67ae85, 0x3c6ef372, 0xa54ff53a, 0x510e527f, 0x9b05688c, 0x1f83d9ab, 0x5be0cd19, 0, Seq()) {
case ((h0f, h1f, h2f, h3f, h4f, h5f, h6f, h7f, length, buf), input) ⇒
val newLength = length + input.length
val buffer = buf ++ input
val numBlocks = buffer.length / 64
val newBuf = buffer.slice(numBlocks * 64, buffer.length)
val blocks = buffer.slice(0, numBlocks * 64).grouped(64).toSeq

var h0 = h0f
var h1 = h1f
var h2 = h2f
var h3 = h3f
var h4 = h4f
var h5 = h5f
var h6 = h6f
var h7 = h7f

for (block <- blocks) {
val t = compressBlock((h0, h1, h2, h3, h4, h5, h6, h7), block)
h0 = t._1
h1 = t._2
h2 = t._3
h3 = t._4
h4 = t._5
h5 = t._6
h6 = t._7
h7 = t._8
}

Success((h0, h1, h2, h3, h4, h5, h6, h7, newLength, newBuf))
} map {
case (h0f, h1f, h2f, h3f, h4f, h5f, h6f, h7f, length, buf) ⇒
val messageLength = length
val bitLength = messageLength * 8
val bitLengthEncoded = ByteBuffer.allocate(8).order(ByteOrder.BIG_ENDIAN).putLong(bitLength).array.toSeq
val nullPaddingLength = 64 - ((buf.length + 1 + 8) % 64)
val lastBlocks = (buf :+ 128.toByte) ++ Seq.fill[Byte](nullPaddingLength) { 0.toByte } ++ bitLengthEncoded

val blocks = lastBlocks.grouped(64).toSeq
var a: Int = state._1
var b: Int = state._2
var c: Int = state._3
var d: Int = state._4
var e: Int = state._5
var f: Int = state._6
var g: Int = state._7
var h: Int = state._8
var temp: Int = 0

for (i <- (0 until 64)) {
val s1 = rotRight(e, 6) ^ rotRight(e, 11) ^ rotRight(e, 25)
val ch = (e & f) ^ ((~e) & g)
val temp1 = h + s1 + ch + k(i) + w(i)
val s0 = rotRight(a, 2) ^ rotRight(a, 13) ^ rotRight(a, 22)
val maj = (a & b) ^ (a & c) ^ (b & c)
val temp2 = s0 + maj

h = g
g = f
f = e
e = d + temp1
d = c
c = b
b = a
a = temp1 + temp2
}
(state._1 + a, state._2 + b, state._3 + c, state._4 + d, state._5 + e, state._6 + f, state._7 + g, state._8 + h)
}

var h0 = h0f
var h1 = h1f
var h2 = h2f
var h3 = h3f
var h4 = h4f
var h5 = h5f
var h6 = h6f
var h7 = h7f
val initialState = (0x6a09e667, 0xbb67ae85, 0x3c6ef372, 0xa54ff53a, 0x510e527f, 0x9b05688c, 0x1f83d9ab, 0x5be0cd19)

for (block <- blocks) {
val t = compressBlock((h0, h1, h2, h3, h4, h5, h6, h7), block)
h0 = t._1
h1 = t._2
h2 = t._3
h3 = t._4
h4 = t._5
h5 = t._6
h6 = t._7
h7 = t._8
}
def addPadding(state: (Int, Int, Int, Int, Int, Int, Int, Int), buf: Seq[Byte], messageLength: Long): Seq[Byte] = {
val bitLength = messageLength * 8
val bitLengthEncoded = ByteBuffer.allocate(8).order(ByteOrder.BIG_ENDIAN).putLong(bitLength).array.toSeq
val nullPaddingLength = 64 - ((buf.length + 1 + 8) % 64)
(buf :+ 128.toByte) ++ Seq.fill[Byte](nullPaddingLength) { 0.toByte } ++ bitLengthEncoded
}

word2bytes(h0) ++ word2bytes(h1) ++ word2bytes(h2) ++ word2bytes(h3) ++ word2bytes(h4) ++ word2bytes(h5) ++ word2bytes(h6) ++ word2bytes(h7)
}
def finalizeState(state: (Int, Int, Int, Int, Int, Int, Int, Int), messageLength: Long): Seq[Byte] = {
word2bytes(state._1) ++ word2bytes(state._2) ++ word2bytes(state._3) ++ word2bytes(state._4) ++ word2bytes(state._5) ++ word2bytes(state._6) ++ word2bytes(state._7) ++ word2bytes(state._8)
}

val length = 32
Expand Down

0 comments on commit 4eed3f6

Please sign in to comment.