Add optional 'binary' parameter to crypto.md5

PlutoLang · Feb 8, 2025 · dda08fd · dda08fd
1 parent 08da954
commit dda08fd
Showing 1 changed file with 12 additions and 5 deletions.
diff --git a/src/lcryptolib.cpp b/src/lcryptolib.cpp
@@ -218,13 +218,20 @@ static int sdbm(lua_State *L)
 static int md5(lua_State *L)
 {
   size_t len;
-  unsigned char buffer[16];
   const auto str = luaL_checklstring(L, 1, &len);
+  const bool binary = lua_istrue(L, 2);
+
+  unsigned char buffer[16];
   md5_fn((unsigned char*)str, (int)len, buffer);
-
-  char hexbuff[32];
-  soup::string::bin2hexAt(hexbuff, (const char*)buffer, sizeof(buffer), soup::string::charset_hex_lower);
-  lua_pushlstring(L, hexbuff, sizeof(hexbuff));
+
+  if (binary) {
+    lua_pushlstring(L, (const char*)buffer, sizeof(buffer));
+  }
+  else {
+    char hexbuff[32];
+    soup::string::bin2hexAt(hexbuff, (const char*)buffer, sizeof(buffer), soup::string::charset_hex_lower);
+    lua_pushlstring(L, hexbuff, sizeof(hexbuff));
+  }
   return 1;
 }