postgres_driver limit max num hashes to 100 (#2976)

waku-org · Aug 19, 2024 · b8550c5 · b8550c5
1 parent d24b56b
commit b8550c5
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/waku/waku_archive/driver/postgres_driver/postgres_driver.nim b/waku/waku_archive/driver/postgres_driver/postgres_driver.nim
@@ -815,6 +815,10 @@ method getMessages*(
 ): Future[ArchiveDriverResult[seq[ArchiveRow]]] {.async.} =
   debug "beginning of getMessages"
 
+  const MAX_ALLOWED_HASHES = 100
+  if hashes.len > MAX_ALLOWED_HASHES:
+    return err(fmt"can not attend queries with more than {MAX_ALLOWED_HASHES} hashes")
+
   let hexHashes = hashes.mapIt(toHex(it))
 
   if cursor.isNone() and pubsubTopic.isNone() and contentTopics.len == 0 and