Mark UDP socket as ready for write in send msg.

This unblock the pending select.
project-chip · Jun 25, 2021 · 902383f · 902383f
1 parent fb54fc4
commit 902383f
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/inet/UDPEndPoint.cpp b/src/inet/UDPEndPoint.cpp
@@ -649,6 +649,10 @@ INET_ERROR UDPEndPoint::SendMsg(const IPPacketInfo * pktInfo, System::PacketBuff
     SuccessOrExit(res);
 
     res = IPEndPointBasis::SendMsg(pktInfo, std::move(msg), sendFlags);
+
+    // Wait for ability to write on this endpoint.
+    mSocket.SetCallback(HandlePendingIO, reinterpret_cast<intptr_t>(this));
+    mSocket.OnRequestCallbackOnPendingWrite();
 #endif // CHIP_SYSTEM_CONFIG_USE_SOCKETS
 
 #if CHIP_SYSTEM_CONFIG_USE_NETWORK_FRAMEWORK