md5.lua

--[=[

	MD5 hash and digest.

	md5(s[, #s]) -> s       compute the MD5 hash of a string or buffer.
	md5_digest() -> digest  get a closure that can consume data chunks.
	digest(s[, size])       digest a string
	digest(cdata, size)     digest a cdata buffer
	digest() -> s           return the hash

	The functions return the binary representation of the hash.
	To get the hex representation, use tohex().

]=]

if not ... then
	require'md5_test'
	require'md5_hmac_test'
	return
end

local ffi = require'ffi'
local C = ffi.load'md5'

ffi.cdef[[
typedef struct {
	uint32_t lo, hi;
	uint32_t a, b, c, d;
	uint8_t buffer[64];
	uint32_t block[16];
} MD5_CTX;

void MD5_Init(MD5_CTX *ctx);
void MD5_Update(MD5_CTX *ctx, const uint8_t *data, uint32_t size);
void MD5_Final(const uint8_t *result, MD5_CTX *ctx);
]]

function md5_digest()
	local ctx = ffi.new'MD5_CTX'
	local result = ffi.new'uint8_t[16]'
	C.MD5_Init(ctx)
	return function(data, size)
		if data then
			C.MD5_Update(ctx, data, size or #data)
		else
			C.MD5_Final(result, ctx)
			return ffi.string(result, 16)
		end
	end
end

function md5(data, size)
	local d = md5_digest(); d(data, size); return d()
end