Advertisement
tobast

sha256

Jun 20th, 2014
254
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Lua 6.29 KB | None | 0 0
  1. ------------------------------------------------------------
  2. ---- SHA-256 library, credits goes to GravityScore
  3. ---- from the ComputerCraft forum
  4. ------------------------------------------------------------
  5.  
  6. --  
  7. --  Adaptation of the Secure Hashing Algorithm (SHA-244/256)
  8. --  Found Here: http://lua-users.org/wiki/SecureHashAlgorithm
  9. --  
  10. --  Using an adapted version of the bit library
  11. --  Found Here: https://bitbucket.org/Boolsheet/bslf/src/1ee664885805/bit.lua
  12. --  
  13.  
  14. local MOD = 2^32
  15. local MODM = MOD-1
  16.  
  17. local function memoize(f)
  18.         local mt = {}
  19.         local t = setmetatable({}, mt)
  20.         function mt:__index(k)
  21.                 local v = f(k)
  22.                 t[k] = v
  23.                 return v
  24.         end
  25.         return t
  26. end
  27.  
  28. local function make_bitop_uncached(t, m)
  29.         local function bitop(a, b)
  30.                 local res,p = 0,1
  31.                 while a ~= 0 and b ~= 0 do
  32.                         local am, bm = a % m, b % m
  33.                         res = res + t[am][bm] * p
  34.                         a = (a - am) / m
  35.                         b = (b - bm) / m
  36.                         p = p*m
  37.                 end
  38.                 res = res + (a + b) * p
  39.                 return res
  40.         end
  41.         return bitop
  42. end
  43.  
  44. local function make_bitop(t)
  45.         local op1 = make_bitop_uncached(t,2^1)
  46.         local op2 = memoize(function(a) return memoize(function(b) return op1(a, b) end) end)
  47.         return make_bitop_uncached(op2, 2 ^ (t.n or 1))
  48. end
  49.  
  50. local bxor1 = make_bitop({[0] = {[0] = 0,[1] = 1}, [1] = {[0] = 1, [1] = 0}, n = 4})
  51.  
  52. local function bxor(a, b, c, ...)
  53.         local z = nil
  54.         if b then
  55.                 a = a % MOD
  56.                 b = b % MOD
  57.                 z = bxor1(a, b)
  58.                 if c then z = bxor(z, c, ...) end
  59.                 return z
  60.         elseif a then return a % MOD
  61.         else return 0 end
  62. end
  63.  
  64. local function band(a, b, c, ...)
  65.         local z
  66.         if b then
  67.                 a = a % MOD
  68.                 b = b % MOD
  69.                 z = ((a + b) - bxor1(a,b)) / 2
  70.                 if c then z = bit32_band(z, c, ...) end
  71.                 return z
  72.         elseif a then return a % MOD
  73.         else return MODM end
  74. end
  75.  
  76. local function bnot(x) return (-1 - x) % MOD end
  77.  
  78. local function rshift1(a, disp)
  79.         if disp < 0 then return lshift(a,-disp) end
  80.         return math.floor(a % 2 ^ 32 / 2 ^ disp)
  81. end
  82.  
  83. local function rshift(x, disp)
  84.         if disp > 31 or disp < -31 then return 0 end
  85.         return rshift1(x % MOD, disp)
  86. end
  87.  
  88. local function lshift(a, disp)
  89.         if disp < 0 then return rshift(a,-disp) end
  90.         return (a * 2 ^ disp) % 2 ^ 32
  91. end
  92.  
  93. local function rrotate(x, disp)
  94.     x = x % MOD
  95.     disp = disp % 32
  96.     local low = band(x, 2 ^ disp - 1)
  97.     return rshift(x, disp) + lshift(low, 32 - disp)
  98. end
  99.  
  100. local k = {
  101.         0x428a2f98, 0x71374491, 0xb5c0fbcf, 0xe9b5dba5,
  102.         0x3956c25b, 0x59f111f1, 0x923f82a4, 0xab1c5ed5,
  103.         0xd807aa98, 0x12835b01, 0x243185be, 0x550c7dc3,
  104.         0x72be5d74, 0x80deb1fe, 0x9bdc06a7, 0xc19bf174,
  105.         0xe49b69c1, 0xefbe4786, 0x0fc19dc6, 0x240ca1cc,
  106.         0x2de92c6f, 0x4a7484aa, 0x5cb0a9dc, 0x76f988da,
  107.         0x983e5152, 0xa831c66d, 0xb00327c8, 0xbf597fc7,
  108.         0xc6e00bf3, 0xd5a79147, 0x06ca6351, 0x14292967,
  109.         0x27b70a85, 0x2e1b2138, 0x4d2c6dfc, 0x53380d13,
  110.         0x650a7354, 0x766a0abb, 0x81c2c92e, 0x92722c85,
  111.         0xa2bfe8a1, 0xa81a664b, 0xc24b8b70, 0xc76c51a3,
  112.         0xd192e819, 0xd6990624, 0xf40e3585, 0x106aa070,
  113.         0x19a4c116, 0x1e376c08, 0x2748774c, 0x34b0bcb5,
  114.         0x391c0cb3, 0x4ed8aa4a, 0x5b9cca4f, 0x682e6ff3,
  115.         0x748f82ee, 0x78a5636f, 0x84c87814, 0x8cc70208,
  116.         0x90befffa, 0xa4506ceb, 0xbef9a3f7, 0xc67178f2,
  117. }
  118.  
  119. local function str2hexa(s)
  120.         return (string.gsub(s, ".", function(c) return string.format("%02x", string.byte(c)) end))
  121. end
  122.  
  123. local function num2s(l, n)
  124.         local s = ""
  125.         for i = 1, n do
  126.                 local rem = l % 256
  127.                 s = string.char(rem) .. s
  128.                 l = (l - rem) / 256
  129.         end
  130.         return s
  131. end
  132.  
  133. local function s232num(s, i)
  134.         local n = 0
  135.         for i = i, i + 3 do n = n*256 + string.byte(s, i) end
  136.         return n
  137. end
  138.  
  139. local function preproc(msg, len)
  140.         local extra = 64 - ((len + 9) % 64)
  141.         len = num2s(8 * len, 8)
  142.         msg = msg .. "\128" .. string.rep("\0", extra) .. len
  143.         assert(#msg % 64 == 0)
  144.         return msg
  145. end
  146.  
  147. local function initH256(H)
  148.         H[1] = 0x6a09e667
  149.         H[2] = 0xbb67ae85
  150.         H[3] = 0x3c6ef372
  151.         H[4] = 0xa54ff53a
  152.         H[5] = 0x510e527f
  153.         H[6] = 0x9b05688c
  154.         H[7] = 0x1f83d9ab
  155.         H[8] = 0x5be0cd19
  156.         return H
  157. end
  158.  
  159. local function digestblock(msg, i, H)
  160.         local w = {}
  161.         for j = 1, 16 do w[j] = s232num(msg, i + (j - 1)*4) end
  162.         for j = 17, 64 do
  163.                 local v = w[j - 15]
  164.                 local s0 = bxor(rrotate(v, 7), rrotate(v, 18), rshift(v, 3))
  165.                 v = w[j - 2]
  166.                 w[j] = w[j - 16] + s0 + w[j - 7] + bxor(rrotate(v, 17), rrotate(v, 19), rshift(v, 10))
  167.         end
  168.  
  169.         local a, b, c, d, e, f, g, h = H[1], H[2], H[3], H[4], H[5], H[6], H[7], H[8]
  170.         for i = 1, 64 do
  171.                 local s0 = bxor(rrotate(a, 2), rrotate(a, 13), rrotate(a, 22))
  172.                 local maj = bxor(band(a, b), band(a, c), band(b, c))
  173.                 local t2 = s0 + maj
  174.                 local s1 = bxor(rrotate(e, 6), rrotate(e, 11), rrotate(e, 25))
  175.                 local ch = bxor (band(e, f), band(bnot(e), g))
  176.                 local t1 = h + s1 + ch + k[i] + w[i]
  177.                 h, g, f, e, d, c, b, a = g, f, e, d + t1, c, b, a, t1 + t2
  178.         end
  179.  
  180.         H[1] = band(H[1] + a)
  181.         H[2] = band(H[2] + b)
  182.         H[3] = band(H[3] + c)
  183.         H[4] = band(H[4] + d)
  184.         H[5] = band(H[5] + e)
  185.         H[6] = band(H[6] + f)
  186.         H[7] = band(H[7] + g)
  187.         H[8] = band(H[8] + h)
  188. end
  189.  
  190. function sha256(msg)
  191.         msg = preproc(msg, #msg)
  192.         local H = initH256({})
  193.         for i = 1, #msg, 64 do digestblock(msg, i, H) end
  194.         return str2hexa(num2s(H[1], 4) .. num2s(H[2], 4) .. num2s(H[3], 4) .. num2s(H[4], 4) ..
  195.                 num2s(H[5], 4) .. num2s(H[6], 4) .. num2s(H[7], 4) .. num2s(H[8], 4))
  196. end
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement