add entrophysource support to hyperdrive

anonrig · anonrig · commit 8d2b2b6959be · 2025-10-30T12:10:50.000-04:00
diff --git a/src/workerd/api/hyperdrive.c++ b/src/workerd/api/hyperdrive.c++
@@ -13,15 +13,23 @@
 #include <kj/string.h>
 
 namespace workerd::api {
-Hyperdrive::Hyperdrive(
-    uint clientIndex, kj::String database, kj::String user, kj::String password, kj::String scheme)
+Hyperdrive::Hyperdrive(uint clientIndex,
+    kj::String database,
+    kj::String user,
+    kj::String password,
+    kj::String scheme,
+    const kj::Maybe<kj::EntropySource&>& maybeEntropySource)
     : clientIndex(clientIndex),
       database(kj::mv(database)),
       user(kj::mv(user)),
       password(kj::mv(password)),
       scheme(kj::mv(scheme)) {
   kj::byte randomBytes[16]{};
-  KJ_ASSERT(RAND_bytes(randomBytes, sizeof(randomBytes)) == 1);
+  KJ_IF_SOME(entrophySource, maybeEntropySource) {
+    entrophySource.generate(kj::arrayPtr(randomBytes, sizeof(randomBytes)));
+  } else {
+    KJ_ASSERT(RAND_bytes(randomBytes, sizeof(randomBytes)) == 1);
+  }
   randomHost = kj::str(kj::encodeHex(randomBytes), ".hyperdrive.local");
 }
 
diff --git a/src/workerd/api/hyperdrive.h b/src/workerd/api/hyperdrive.h
@@ -6,6 +6,8 @@
 
 #include <workerd/jsg/jsg.h>
 
+#include <kj/compat/http.h>
+
 namespace kj {
 class AsyncIoStream;
 }
@@ -26,7 +28,8 @@ class Hyperdrive: public jsg::Object {
       kj::String database,
       kj::String user,
       kj::String password,
-      kj::String scheme);
+      kj::String scheme,
+      const kj::Maybe<kj::EntropySource&>& = kj::none);
   jsg::Ref<Socket> connect(jsg::Lock& js);
   kj::StringPtr getDatabase();
   kj::StringPtr getUser();
