Fix jsg::Ref<WebSocketRequestResponsePair> dtor segfault in actor-state.

src/workerd/api/actor-state.c++

@@ -880,7 +880,7 @@ void DurableObjectState::setWebSocketAutoResponse(
     // If there's no request/response pair, we unset any current set auto response configuration.
     KJ_IF_SOME(manager, a.getHibernationManager()) {
       // If there's no hibernation manager created yet, there's nothing to do here.
-      manager.unsetWebSocketAutoResponse();
+      manager.setWebSocketAutoResponse(kj::none, kj::none);
     }
     return;
   }
@@ -902,15 +902,16 @@ void DurableObjectState::setWebSocketAutoResponse(
     // If there's no hibernation manager created yet, we should create one and
     // set its auto response.
   }
-  KJ_REQUIRE_NONNULL(a.getHibernationManager()).setWebSocketAutoResponse(kj::mv(reqResp));
+  KJ_REQUIRE_NONNULL(a.getHibernationManager()).setWebSocketAutoResponse(
+      reqResp->getRequest(), reqResp->getResponse());
 }
 
 kj::Maybe<jsg::Ref<api::WebSocketRequestResponsePair>> DurableObjectState::getWebSocketAutoResponse() {
   auto& a = KJ_REQUIRE_NONNULL(IoContext::current().getActor());
   KJ_IF_SOME(manager, a.getHibernationManager()) {
     // If there's no hibernation manager created yet, there's nothing to do here.
     auto r = manager.getWebSocketAutoResponse();
-    return r;
+    return kj::mv(r);
   }
   return kj::none;
 }

src/workerd/io/hibernation-manager.c++

@@ -99,17 +99,22 @@ kj::Vector<jsg::Ref<api::WebSocket>> HibernationManagerImpl::getWebSockets(
 }
 
 void HibernationManagerImpl::setWebSocketAutoResponse(
-    jsg::Ref<api::WebSocketRequestResponsePair> reqResp) {
-  autoResponsePair = kj::mv(reqResp);
-}
-
-void HibernationManagerImpl::unsetWebSocketAutoResponse() {
+    kj::Maybe<kj::StringPtr> request, kj::Maybe<kj::StringPtr> response) {
+  KJ_IF_SOME(req, request) {
+    KJ_IF_SOME(resp, response){
+      auto autoRR = kj::heap<AutoRequestResponsePair>();
+      autoRR->response = kj::mv(req);
+      autoRR->request = kj::mv(resp);
+      autoResponsePair = kj::mv(autoRR);
+      return;
+    }
+  }
   autoResponsePair = kj::none;
 }
 
 kj::Maybe<jsg::Ref<api::WebSocketRequestResponsePair>> HibernationManagerImpl::getWebSocketAutoResponse() {
   KJ_IF_SOME(ar, autoResponsePair) {
-    return ar.addRef();
+    return api::WebSocketRequestResponsePair::constructor(kj::str(ar.request), kj::str(ar.response));
   } else {
     return kj::none;
   }
@@ -191,7 +196,7 @@ kj::Promise<void> HibernationManagerImpl::readLoop(HibernatableWebSocket& hib) {
     KJ_IF_SOME (reqResp, autoResponsePair) {
       KJ_SWITCH_ONEOF(message) {
         KJ_CASE_ONEOF(text, kj::String) {
-          if (text == (reqResp)->getRequest()) {
+          if (text == reqResp.request) {
             // If the received message matches the one set for auto-response, we must
             // short-circuit readLoop, store the current timestamp and and automatically respond
             // with the expected response.
@@ -207,15 +212,15 @@ kj::Promise<void> HibernationManagerImpl::readLoop(HibernatableWebSocket& hib) {
                 // If the actor is not hibernated/If the WebSocket is active, we need to update
                 // autoResponseTimestamp on the active websocket.
                 apiWs->setAutoResponseStatus(hib.autoResponseTimestamp, kj::READY_NOW);
-                co_await apiWs->sendAutoResponse(kj::str(reqResp->getResponse().asArray()), ws);
+                co_await apiWs->sendAutoResponse(kj::str(reqResp.response.asArray()), ws);
               }
               KJ_CASE_ONEOF(package, api::WebSocket::HibernationPackage) {
                 if (!package.closedOutgoingConnection) {
                   // We need to store the autoResponsePromise because we may instantiate an api::websocket
                   // If we do that, we have to provide it with the promise to avoid races. This can
                   // happen if we have a websocket hibernating, that unhibernates and sends a
                   // message while ws.send() for auto-response is also sending.
-                  auto p = ws.send(reqResp->getResponse().asArray()).fork();
+                  auto p = ws.send(reqResp.response.asArray()).fork();
                   hib.autoResponsePromise = p.addBranch();
                   co_await p;
                   hib.autoResponsePromise = kj::READY_NOW;

src/workerd/io/hibernation-manager.h

@@ -40,8 +40,8 @@ class HibernationManagerImpl final : public Worker::Actor::HibernationManager {
   // This converts our activeOrPackage from an api::WebSocket to a HibernationPackage.
   void hibernateWebSockets(Worker::Lock& lock) override;
 
-  void setWebSocketAutoResponse(jsg::Ref<api::WebSocketRequestResponsePair> reqResp) override;
-  void unsetWebSocketAutoResponse() override;
+  void setWebSocketAutoResponse(kj::Maybe<kj::StringPtr> request,
+      kj::Maybe<kj::StringPtr> response) override;
   kj::Maybe<jsg::Ref<api::WebSocketRequestResponsePair>> getWebSocketAutoResponse() override;
   void setTimerChannel(TimerChannel& timerChannel) override;
 
@@ -187,6 +187,11 @@ class HibernationManagerImpl final : public Worker::Actor::HibernationManager {
     TagCollection(TagCollection&& other) = default;
   };
 
+  struct AutoRequestResponsePair {
+    kj::StringPtr request;
+    kj::StringPtr response;
+  };
+
   // A hashmap of tags to HibernatableWebSockets associated with the tag.
   // We use a kj::List so we can quickly remove websockets that have disconnected.
   // Also note that we box the keys and values such that in the event of a hashmap resizing we don't
@@ -222,7 +227,7 @@ class HibernationManagerImpl final : public Worker::Actor::HibernationManager {
   };
   DisconnectHandler onDisconnect;
   kj::TaskSet readLoopTasks;
-  kj::Maybe<jsg::Ref<api::WebSocketRequestResponsePair>> autoResponsePair;
+  kj::Maybe<AutoRequestResponsePair&> autoResponsePair;
   kj::Maybe<TimerChannel&> timer;
 };
 }; // namespace workerd

src/workerd/io/worker.h

@@ -697,8 +697,8 @@ class Worker::Actor final: public kj::Refcounted {
         jsg::Lock& js,
         kj::Maybe<kj::StringPtr> tag) = 0;
     virtual void hibernateWebSockets(Worker::Lock& lock) = 0;
-    virtual void setWebSocketAutoResponse(jsg::Ref<api::WebSocketRequestResponsePair> reqResp) = 0;
-    virtual void unsetWebSocketAutoResponse() = 0;
+    virtual void setWebSocketAutoResponse(kj::Maybe<kj::StringPtr> request,
+        kj::Maybe<kj::StringPtr> response) = 0;
     virtual kj::Maybe<jsg::Ref<api::WebSocketRequestResponsePair>> getWebSocketAutoResponse() = 0;
     virtual void setTimerChannel(TimerChannel& timerChannel) = 0;
   };