fixup! dgram: make UDPWrap more reusable

nodejs · Oct 11, 2019 · a915058 · a915058
1 parent f652f65
commit a915058
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/udp_wrap.cc b/src/udp_wrap.cc
@@ -496,6 +496,8 @@ void UDPWrap::DoSend(const FunctionCallbackInfo<Value>& args, int family) {
 ssize_t UDPWrap::Send(uv_buf_t* bufs_ptr,
                       size_t count,
                       const sockaddr* addr) {
+  if (IsHandleClosing()) return UV_EBADF;
+
   size_t msg_size = 0;
   for (size_t i = 0; i < count; i++)
     msg_size += bufs_ptr[i].len;
@@ -589,6 +591,7 @@ void UDPWrap::RecvStart(const FunctionCallbackInfo<Value>& args) {
 }
 
 int UDPWrap::RecvStart() {
+  if (IsHandleClosing()) return UV_EBADF;
   int err = uv_udp_recv_start(&handle_, OnAlloc, OnRecv);
   // UV_EALREADY means that the socket is already bound but that's okay
   if (err == UV_EALREADY)
@@ -606,6 +609,7 @@ void UDPWrap::RecvStop(const FunctionCallbackInfo<Value>& args) {
 }
 
 int UDPWrap::RecvStop() {
+  if (IsHandleClosing()) return UV_EBADF;
   return uv_udp_recv_stop(&handle_);
 }