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
|
#include <string.h>
#include "lauxlib.h"
#include "lua.h"
#include "nettle/sha2.h"
//#include "../qcommon/q3_lauxlib.h" FIXME? This doesn't seem to be hooked correctly into tremded.exe
#define SHA256_CTX "sha256_ctx*"
static int lsha256(lua_State *L)
{
struct sha256_ctx *ctx;
ctx = (struct sha256_ctx *)lua_newuserdata(L, sizeof(struct sha256_ctx));
luaL_setmetatable(L, SHA256_CTX);
sha256_init(ctx);
return 1;
}
static int lsha256_digest(lua_State *L)
{
struct sha256_ctx *ctx;
char digest[SHA256_DIGEST_SIZE];
ctx = luaL_checkudata(L, 1, SHA256_CTX);
sha256_digest(ctx, sizeof(digest), digest);
lua_pushlstring(L, digest, sizeof(digest));
return 1;
}
static int lsha256_update(lua_State *L)
{
struct sha256_ctx *ctx;
const char *data;
size_t len;
ctx = luaL_checkudata(L, 1, SHA256_CTX);
if (lua_isnil(L, 2)) {
return 0;
}
data = luaL_tolstring(L, 2, &len);
nettle_sha256_update(ctx, len, data);
return 0;
}
static int lsha256_tostring(lua_State *L)
{
struct sha256_ctx *ctx, ctx2;
char digest[SHA256_DIGEST_SIZE];
char hex[SHA256_DIGEST_SIZE*2+1];
int i;
ctx = luaL_checkudata(L, 1, SHA256_CTX);
memcpy(&ctx2, ctx, sizeof(struct sha256_ctx));
sha256_digest(&ctx2, sizeof(digest), digest);
for (i = 0; i < sizeof(digest); i++) {
sprintf(hex + 2 * i, "%02hhx", digest[i]);
}
lua_pushstring(L, hex);
return 1;
}
/* functions for 'nettle' library */
static const luaL_Reg nettlelib[] = {
{"sha256", lsha256},
{NULL, NULL}
};
/* functions for sha256 objects */
static const luaL_Reg lsha256_methods[] = {
{"digest", lsha256_digest},
{"update", lsha256_update},
{"__tostring", lsha256_tostring},
{NULL, NULL}
};
static void createmeta (lua_State *L)
{
luaL_newmetatable(L, SHA256_CTX); /* create metatable for file handles */
lua_pushvalue(L, -1); /* push metatable */
lua_setfield(L, -2, "__index"); /* metatable.__index = metatable */
luaL_setfuncs(L, lsha256_methods, 0); /* add file methods to new metatable */
lua_pop(L, 1); /* pop new metatable */
}
LUAMOD_API int luaopen_nettle (lua_State *L)
{
luaL_newlib(L, nettlelib);
createmeta(L);
return 1;
}
|
