Use correct length increment for msg_controllen (#3065)

microsoft · Sep 14, 2022 · 1dfe7c9 · 1dfe7c9
1 parent e8dfa0b
commit 1dfe7c9
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/platform/datapath_epoll.c b/src/platform/datapath_epoll.c
@@ -2571,7 +2571,9 @@ CxPlatSocketSendInternal(
         *(int *)CMSG_DATA(CMsg) = SendData->ECN;
 
         if (!Socket->Connected) {
-            Mhdr->msg_controllen += CMSG_SPACE(sizeof(struct in6_pktinfo));
+            Mhdr->msg_controllen += RemoteAddress->Ip.sa_family == QUIC_ADDRESS_FAMILY_INET
+                ? CMSG_SPACE(sizeof(struct in_pktinfo))
+                : CMSG_SPACE(sizeof(struct in6_pktinfo));
             CMsg = CMSG_NXTHDR(Mhdr, CMsg);
             CXPLAT_DBG_ASSERT(LocalAddress != NULL);
             CXPLAT_DBG_ASSERT(CMsg != NULL);