use Config->Route->DatapathType for data allocation

microsoft · Sep 7, 2023 · 02b39de · 02b39de
1 parent 9720513
commit 02b39de
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/platform/datapath_win.c b/src/platform/datapath_win.c
@@ -444,9 +444,9 @@ CxPlatSendDataAlloc(
 {
     CXPLAT_SEND_DATA* SendData = NULL;
     // TODO: fallback?
-    if (Socket->UseTcp || (Socket->Type == CXPLAT_SOCKET_UDP &&
-        Socket->RawSocketAvailable &&
-        !IS_LOOPBACK(Config->Route->RemoteAddress))) {
+    if (Socket->UseTcp || Config->Route->DatapathType == CXPLAT_DATAPATH_TYPE_XDP ||
+        (Config->Route->DatapathType == CXPLAT_DATAPATH_TYPE_UNKNOWN &&
+        Socket->RawSocketAvailable && !IS_LOOPBACK(Config->Route->RemoteAddress))) {
         SendData = RawSendDataAlloc(SocketToRaw(Socket), Config);
         if (SendData) {
             DatapathType(SendData) = CXPLAT_DATAPATH_TYPE_XDP;