feat(ext/net): Add abort signal to net tcp connect

ext/net/01_net.js

@@ -579,23 +579,36 @@ function createListenDatagram(udpOpFn, unixOpFn) {
 async function connect(args) {
   switch (args.transport ?? "tcp") {
     case "tcp": {
+      let cancelRid;
+      let abortHandler;
+      if (args?.signal) {
+        args.signal.throwIfAborted();
+        cancelRid = createCancelHandle();
+        abortHandler = () => core.tryClose(cancelRid);
+        args.signal[abortSignal.add](abortHandler);
+      }
       const port = validatePort(args.port);
-      const { 0: rid, 1: localAddr, 2: remoteAddr } = await op_net_connect_tcp(
+
+      try {
+        const { 0: rid, 1: localAddr, 2: remoteAddr } =
+          await op_net_connect_tcp(
             {
               hostname: args.hostname ?? "127.0.0.1",
               port,
             },
             args.timeout,
+            cancelRid,
           );
         localAddr.transport = "tcp";
         remoteAddr.transport = "tcp";
 
-      const tcpConn = new TcpConn(rid, remoteAddr, localAddr);
+        return new TcpConn(rid, remoteAddr, localAddr);
-      if (args.signal) {
+      } finally {
-        args.signal.addEventListener("abort", () => tcpConn.close());
+        if (args?.signal) {
+          args.signal[abortSignal.remove](abortHandler);
+          args.signal.throwIfAborted();
+        }
       }
-
-      return tcpConn;
     }
     case "unix": {
       const { 0: rid, 1: localAddr, 2: remoteAddr } = await op_net_connect_unix(

ext/net/ops.rs

@@ -373,11 +373,12 @@ pub async fn op_net_connect_tcp<NP>(
   state: Rc<RefCell<OpState>>,
   #[serde] addr: IpAddr,
   #[serde] timeout: Option<u64>,
+  #[smi] resource_abort_id: Option<ResourceId>,
 ) -> Result<(ResourceId, IpAddr, IpAddr), NetError>
 where
   NP: NetPermissions + 'static,
 {
-  op_net_connect_tcp_inner::<NP>(state, addr, timeout).await
+  op_net_connect_tcp_inner::<NP>(state, addr, timeout, resource_abort_id).await
 }
 
 #[inline]
@@ -385,6 +386,7 @@ pub async fn op_net_connect_tcp_inner<NP>(
   state: Rc<RefCell<OpState>>,
   addr: IpAddr,
   timeout: Option<u64>,
+  resource_abort_id: Option<ResourceId>,
 ) -> Result<(ResourceId, IpAddr, IpAddr), NetError>
 where
   NP: NetPermissions + 'static,
@@ -400,10 +402,28 @@ where
     .await?
     .next()
     .ok_or_else(|| NetError::NoResolvedAddress)?;
-  let tcp_stream = match TcpStream::connect_timeout(&addr, timeout).await {
+
+  let cancel_handle = resource_abort_id.and_then(|rid| {
+    state
+      .borrow_mut()
+      .resource_table
+      .get::<CancelHandle>(rid)
+      .ok()
+  });
+
+  let tcp_stream_result = if let Some(cancel_handle) = &cancel_handle {
+    TcpStream::connect_timeout(&addr, timeout)
+      .or_cancel(cancel_handle)
+      .await?
+  } else {
+    TcpStream::connect_timeout(&addr, timeout).await
+  };
+
+  let tcp_stream = match tcp_stream_result {
     Ok(tcp_stream) => tcp_stream,
     Err(e) => return Err(NetError::Io(e)),
   };
+
   let local_addr = tcp_stream.local_addr()?;
   let remote_addr = tcp_stream.peer_addr()?;
 
@@ -1155,8 +1175,9 @@ mod tests {
       port: server_addr[1].parse().unwrap(),
     };
 
-    let mut connect_fut =
+    let mut connect_fut = op_net_connect_tcp_inner::<TestPermission>(
-      op_net_connect_tcp_inner::<TestPermission>(conn_state, ip_addr, None)
+      conn_state, ip_addr, None, None,
+    )
     .boxed_local();
     let mut rid = None;
 

tests/unit/net_test.ts

@@ -1295,16 +1295,21 @@ Deno.test(
 
 Deno.test(
   { permissions: { net: true } },
-  async function netTcpAbortSignal() {
+  async function netTcpWithAbortSignal() {
     const controller = new AbortController();
+    setTimeout(() => controller.abort(), 1000);
+    const error = await assertRejects(
+      async () => {
         await Deno.connect({
           hostname: "deno.com",
-      port: 80,
+          port: 50000,
           transport: "tcp",
           signal: controller.signal,
         });
-    controller.abort();
+      },
-    assertEquals(controller.signal.aborted, true);
+    );
+    assert(error instanceof DOMException);
+    assertEquals(error.name, "AbortError");
   },
 );