Hans Leidekker : wpcap: Handle different layout of the native packet header structure on 32-bit.

[Alexandre Julliard]

Module: wine
Branch: master
Commit: 727472ab7d60e285a28fb67fd5e3e21907dd060a
URL:    https://gitlab.winehq.org/wine/wine/-/commit/727472ab7d60e285a28fb67fd5e3e21...

Author: Hans Leidekker <hans(a)codeweavers.com>
Date:   Thu Jun 27 09:56:45 2024 +0200

wpcap: Handle different layout of the native packet header structure on 32-bit.

Wine-Bug: https://bugs.winehq.org/show_bug.cgi?id=56871

---

 dlls/wpcap/unixlib.c | 52 +++++++++++++++++++++++++++++++++++++++-------------
 1 file changed, 39 insertions(+), 13 deletions(-)

diff --git a/dlls/wpcap/unixlib.c b/dlls/wpcap/unixlib.c
index 4f5c2337b10..6db4800d16b 100644
--- a/dlls/wpcap/unixlib.c
+++ b/dlls/wpcap/unixlib.c
@@ -145,13 +145,25 @@ static NTSTATUS wrap_datalink_val_to_name( void *args )
 static NTSTATUS wrap_dump( void *args )
 {
     const struct dump_params *params = args;
-    struct pcap_pkthdr hdr_unix;
 
-    hdr_unix.ts.tv_sec  = params->hdr->ts.tv_sec;
-    hdr_unix.ts.tv_usec = params->hdr->ts.tv_usec;
-    hdr_unix.caplen     = params->hdr->caplen;
-    hdr_unix.len        = params->hdr->len;
-    pcap_dump( params->user, &hdr_unix, params->packet );
+    if (sizeof(void *) == 4)
+    {
+        struct pcap_pkthdr_win32 hdr32;
+        hdr32.ts.tv_sec  = params->hdr->ts.tv_sec;
+        hdr32.ts.tv_usec = params->hdr->ts.tv_usec;
+        hdr32.caplen     = params->hdr->caplen;
+        hdr32.len        = params->hdr->len;
+        pcap_dump( params->user, (const struct pcap_pkthdr *)&hdr32, params->packet );
+    }
+    else
+    {
+        struct pcap_pkthdr hdr64;
+        hdr64.ts.tv_sec  = params->hdr->ts.tv_sec;
+        hdr64.ts.tv_usec = params->hdr->ts.tv_usec;
+        hdr64.caplen     = params->hdr->caplen;
+        hdr64.len        = params->hdr->len;
+        pcap_dump( params->user, &hdr64, params->packet );
+    }
     return STATUS_SUCCESS;
 }
 
@@ -300,16 +312,30 @@ static NTSTATUS wrap_minor_version( void *args )
 static NTSTATUS wrap_next_ex( void *args )
 {
     struct next_ex_params *params = args;
-    struct pcap_pkthdr *hdr_unix;
     int ret;
 
-    if ((ret = pcap_next_ex( (pcap_t *)(ULONG_PTR)params->handle, &hdr_unix, params->data )) == 1)
+    if (sizeof(void *) == 4)
     {
-        if (hdr_unix->ts.tv_sec > INT_MAX || hdr_unix->ts.tv_usec > INT_MAX) WARN( "truncating timeval values(s)\n" );
-        params->hdr->ts.tv_sec  = hdr_unix->ts.tv_sec;
-        params->hdr->ts.tv_usec = hdr_unix->ts.tv_usec;
-        params->hdr->caplen     = hdr_unix->caplen;
-        params->hdr->len        = hdr_unix->len;
+        struct pcap_pkthdr_win32 *hdr32;
+        if ((ret = pcap_next_ex( (pcap_t *)(ULONG_PTR)params->handle, (struct pcap_pkthdr **)&hdr32, params->data )) == 1)
+        {
+            params->hdr->ts.tv_sec  = hdr32->ts.tv_sec;
+            params->hdr->ts.tv_usec = hdr32->ts.tv_usec;
+            params->hdr->caplen     = hdr32->caplen;
+            params->hdr->len        = hdr32->len;
+        }
+    }
+    else
+    {
+        struct pcap_pkthdr *hdr64;
+        if ((ret = pcap_next_ex( (pcap_t *)(ULONG_PTR)params->handle, &hdr64, params->data )) == 1)
+        {
+            if (hdr64->ts.tv_sec > INT_MAX || hdr64->ts.tv_usec > INT_MAX) WARN( "truncating timeval values(s)\n" );
+            params->hdr->ts.tv_sec  = hdr64->ts.tv_sec;
+            params->hdr->ts.tv_usec = hdr64->ts.tv_usec;
+            params->hdr->caplen     = hdr64->caplen;
+            params->hdr->len        = hdr64->len;
+        }
     }
     return ret;
 }