1 files changed, 31 insertions, 6 deletions

diff --git a/src/hash/sdbm.c b/src/hash/sdbm.c
index b32a627..bc5f0d9 100644
--- a/src/hash/sdbm.c
+++ b/src/hash/sdbm.c
@@ -2,18 +2,43 @@
 #include <stdio.h>
 #include <stdint.h>
 
+struct sdbm_hash sdbm_init(){
+  return (struct sdbm_hash){.hash = 0};
+}
+
+void sdbm_update(uint8_t* in, size_t len, struct sdbm_hash* hash){
+  for(int i = 0; i != len; i++){
+    hash->hash = (uint64_t)*in + (hash->hash << 6) + (hash->hash << 16) - hash->hash;
+    in++;
+  }
+}
+
+uint64_t sdbm_final(struct sdbm_hash* hash){
+  return hash->hash;
+}
+
 uint64_t sdbm(uint8_t* in, size_t len){
-    uint64_t hash = 0;
+  struct sdbm_hash a = sdbm_init();
+  sdbm_update(in, len, &a);
+  return sdbm_final(&a);
+}
+
+common_hash_init_update(sdbm);
 
-    for(int i = 0; i != len; i++){
-        hash = (uint64_t)*in + (hash << 6) + (hash << 16) - hash;
-        in++;
-    }
+int l_sdbm_final(lua_State* L){
+  lua_pushstring(L, "ud");
+  lua_gettable(L, 1);
 
-    return hash;
+  struct sdbm_hash* a = (struct sdbm_hash*)lua_touserdata(L, -1);
+  uint64_t u = sdbm_final(a);
+  char digest[64];
+  sprintf(digest,"%016llx",u);
+  lua_pushstring(L, digest);
+  return 1;
 }
 
 int l_sdbm(lua_State* L){ 
+  if(lua_gettop(L) == 0) return l_sdbm_init(L);
   size_t len = 0;
   uint8_t* a = (uint8_t*)luaL_checklstring(L, 1, &len);