Fix null

The function should return null in case the data is null, for consistency with existing hashing functions like md5, digest, etc
cardano-community · Jun 17, 2023 · 7af522e · 7af522e
1 parent 80f8b00
commit 7af522e
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/pg_blake2b.c b/pg_blake2b.c
@@ -13,13 +13,21 @@ pg_blake2b(PG_FUNCTION_ARGS)
 
     bytea   *digest;
     bytea   *key = NULL;
-    bytea   *data = PG_GETARG_BYTEA_PP(0);
+    bytea   *data;
 
     size_t  keylen = 0;
-    size_t  datalen = VARSIZE(data) - VARHDRSZ;
+    size_t  datalen;
 
     int16   digest_size = BLAKE2B_OUTBYTES;
 
+    if (PG_ARGISNULL(0))
+    {
+        PG_RETURN_NULL();
+    }
+
+    data = PG_GETARG_BYTEA_PP(0);
+    datalen = VARSIZE(data) - VARHDRSZ;
+
     if (!PG_ARGISNULL(1))
     {
         digest_size = PG_GETARG_INT16(1);