From a3a152d34fd8f0fc52069a53c9c7854e1aeb47d1 Mon Sep 17 00:00:00 2001
From: theanarkh <theratliter@gmail.com>
Date: Thu, 18 Aug 2022 00:08:58 +0800
Subject: [PATCH] dgram: add dgram send queue info

PR-URL: https://github.com/nodejs/node/pull/44149
Reviewed-By: Matteo Collina <matteo.collina@gmail.com>
Reviewed-By: Anna Henningsen <anna@addaleax.net>
---
 doc/api/dgram.md                            | 17 +++++++++++++
 lib/dgram.js                                |  7 ++++++
 src/udp_wrap.cc                             | 20 +++++++++++++++
 src/udp_wrap.h                              |  3 +++
 test/parallel/test-dgram-send-queue-info.js | 27 +++++++++++++++++++++
 5 files changed, 74 insertions(+)
 create mode 100644 test/parallel/test-dgram-send-queue-info.js

diff --git a/doc/api/dgram.md b/doc/api/dgram.md
index ea26785b570..78930ff45dd 100644
--- a/doc/api/dgram.md
+++ b/doc/api/dgram.md
@@ -454,6 +454,23 @@ added: v8.7.0
 
 This method throws [`ERR_SOCKET_BUFFER_SIZE`][] if called on an unbound socket.
 
+### `socket.getSendQueueSize()`
+
+<!-- YAML
+added: REPLACEME
+-->
+
+* Returns: {number} Number of bytes queued for sending.
+
+### `socket.getSendQueueCount()`
+
+<!-- YAML
+added: REPLACEME
+-->
+
+* Returns: {number} Number of send requests currently in the queue awaiting
+  to be processed.
+
 ### `socket.ref()`
 
 <!-- YAML
diff --git a/lib/dgram.js b/lib/dgram.js
index 246649ad65c..3dcd99c7c45 100644
--- a/lib/dgram.js
+++ b/lib/dgram.js
@@ -976,6 +976,13 @@ Socket.prototype.getSendBufferSize = function() {
   return bufferSize(this, 0, SEND_BUFFER);
 };
 
+Socket.prototype.getSendQueueSize = function() {
+  return this[kStateSymbol].handle.getSendQueueSize();
+};
+
+Socket.prototype.getSendQueueCount = function() {
+  return this[kStateSymbol].handle.getSendQueueCount();
+};
 
 // Deprecated private APIs.
 ObjectDefineProperty(Socket.prototype, '_handle', {
diff --git a/src/udp_wrap.cc b/src/udp_wrap.cc
index 23b4afedb52..5ea38f6732d 100644
--- a/src/udp_wrap.cc
+++ b/src/udp_wrap.cc
@@ -182,6 +182,9 @@ void UDPWrap::Initialize(Local<Object> target,
   SetProtoMethod(isolate, t, "setBroadcast", SetBroadcast);
   SetProtoMethod(isolate, t, "setTTL", SetTTL);
   SetProtoMethod(isolate, t, "bufferSize", BufferSize);
+  SetProtoMethodNoSideEffect(isolate, t, "getSendQueueSize", GetSendQueueSize);
+  SetProtoMethodNoSideEffect(
+      isolate, t, "getSendQueueCount", GetSendQueueCount);
 
   t->Inherit(HandleWrap::GetConstructorTemplate(env));
 
@@ -783,6 +786,23 @@ MaybeLocal<Object> UDPWrap::Instantiate(Environment* env,
   return env->udp_constructor_function()->NewInstance(env->context());
 }
 
+void UDPWrap::GetSendQueueSize(const FunctionCallbackInfo<Value>& args) {
+  UDPWrap* wrap;
+  ASSIGN_OR_RETURN_UNWRAP(
+      &wrap, args.Holder(), args.GetReturnValue().Set(UV_EBADF));
+
+  size_t size = uv_udp_get_send_queue_size(&wrap->handle_);
+  args.GetReturnValue().Set(static_cast<double>(size));
+}
+
+void UDPWrap::GetSendQueueCount(const FunctionCallbackInfo<Value>& args) {
+  UDPWrap* wrap;
+  ASSIGN_OR_RETURN_UNWRAP(
+      &wrap, args.Holder(), args.GetReturnValue().Set(UV_EBADF));
+
+  size_t count = uv_udp_get_send_queue_count(&wrap->handle_);
+  args.GetReturnValue().Set(static_cast<double>(count));
+}
 
 }  // namespace node
 
diff --git a/src/udp_wrap.h b/src/udp_wrap.h
index 75a123d8fa7..83aeb1ec084 100644
--- a/src/udp_wrap.h
+++ b/src/udp_wrap.h
@@ -150,6 +150,9 @@ class UDPWrap final : public HandleWrap,
   static void SetBroadcast(const v8::FunctionCallbackInfo<v8::Value>& args);
   static void SetTTL(const v8::FunctionCallbackInfo<v8::Value>& args);
   static void BufferSize(const v8::FunctionCallbackInfo<v8::Value>& args);
+  static void GetSendQueueSize(const v8::FunctionCallbackInfo<v8::Value>& args);
+  static void GetSendQueueCount(
+      const v8::FunctionCallbackInfo<v8::Value>& args);
 
   // UDPListener implementation
   uv_buf_t OnAlloc(size_t suggested_size) override;
diff --git a/test/parallel/test-dgram-send-queue-info.js b/test/parallel/test-dgram-send-queue-info.js
new file mode 100644
index 00000000000..07b282b29fc
--- /dev/null
+++ b/test/parallel/test-dgram-send-queue-info.js
@@ -0,0 +1,27 @@
+// Flags: --test-udp-no-try-send
+'use strict';
+const common = require('../common');
+const assert = require('assert');
+const dgram = require('dgram');
+
+const socket = dgram.createSocket('udp4');
+assert.strictEqual(socket.getSendQueueSize(), 0);
+assert.strictEqual(socket.getSendQueueCount(), 0);
+socket.close();
+
+const server = dgram.createSocket('udp4');
+const client = dgram.createSocket('udp4');
+
+server.bind(0, common.mustCall(() => {
+  client.connect(server.address().port, common.mustCall(() => {
+    const data = 'hello';
+    client.send(data);
+    client.send(data);
+    // See uv__send in win/udp.c
+    assert.strictEqual(client.getSendQueueSize(),
+                       common.isWindows ? 0 : data.length * 2);
+    assert.strictEqual(client.getSendQueueCount(), 2);
+    client.close();
+    server.close();
+  }));
+}));