test.lua

#!/usr/bin/env lua

local bignum = require("bignum")
local hash = require("hash")
local srp = require("wow/srp")
local srphost = require("wow/srp/host")
local srpuser = require("wow/srp/user")

local ok = "\27[32mOK\27[0m "
local nok = "\27[31mnok\27[0m"

local tests = {
  function()
    local desc = "check sha1 hash"
    local chk = nil

    -- echo -n "USERNAME:PASSWORD" | sha1sum | awk '{ print $1 }' 
    local digest = "5b039d152722e351c8bdebcf06fd8cd4e5244d78"
    local sha1 = hash.sha1_init()
    sha1:update("USERNAME:PASSWORD")
    sha1:final()
    local result = sha1:__tostring()
    if digest == result then chk = ok end

    return chk, desc
  end,

  function()
    local desc = "check password verifier"
    local chk = nil

    -- username:password
    local s = srp.hex2bn("C87C2F705F3A3DE385F4F0E49386D6688061AF13DB4653AD434C82015ECA2969")
    local v = srp.hex2bn("4BEC2A9A0BE2296F67058E1C1AD6FA1EF1E73432BB6872617FA2E3DB7610BB90")

    local p = srp.p("username", "password")
    local x, salt = srp.x(p, s)
    local verifier = srp.v(x, srp.dec2bn(srp.g), srp.hex2bn(srp.N))
    if v:__tostring() == verifier:__tostring() and s:__tostring() == salt:__tostring() then chk = ok end

    return chk, desc
  end,

  function()
    local desc = "check session key computation"
    local chk = nil

    local username = "username"
    local password = "password"
    local s = "C87C2F705F3A3DE385F4F0E49386D6688061AF13DB4653AD434C82015ECA2969"
    local v = "4BEC2A9A0BE2296F67058E1C1AD6FA1EF1E73432BB6872617FA2E3DB7610BB90"

    local host = srphost.auth_challenge(username, v, s)
    local user = srpuser.auth_challenge(username, password, host.s:bn2hex(), host.B:bn2hex(), srp.g, srp.N)

    if host:logon_proof(user.A:bn2hex(), user.M1:bn2hex()) then
      if user:logon_proof(host.M2) then chk = ok end
    end

    return chk, desc
  end
}

local ok, desc
local rc = 0
for _, test in pairs(tests) do
  ok, desc = test()
  print(("[%s] %s"):format(ok or nok, desc))
  if not ok then rc = 1 end
end

os.exit(rc)