[PATCH 1/1] nsiproxy.sys: Only get owning pid when needed in udp_endpoint_enumerate_all().

[Paul Gofman]

From: Paul Gofman

---
 dlls/nsiproxy.sys/udp.c | 37 +++++++++++++++++++++++--------------
 1 file changed, 23 insertions(+), 14 deletions(-)

diff --git a/dlls/nsiproxy.sys/udp.c b/dlls/nsiproxy.sys/udp.c
index 3cb215c1c58..1db7b3df4eb 100644
--- a/dlls/nsiproxy.sys/udp.c
+++ b/dlls/nsiproxy.sys/udp.c
@@ -240,7 +240,7 @@ static NTSTATUS udp_endpoint_enumerate_all( void *key_data, UINT key_size, void

memset( &key, 0, sizeof(key) );
         memset( &stat, 0, sizeof(stat) );
-        pid_map = get_pid_map( &pid_map_size );
+        if (stat_out) pid_map = get_pid_map( &pid_map_size );

/* skip header line */
         ptr = fgets( buf, sizeof(buf), fp );
@@ -254,10 +254,13 @@ static NTSTATUS udp_endpoint_enumerate_all( void *key_data, UINT key_size, void
             key.local.Ipv4.sin_addr.WS_s_addr = addr;
             key.local.Ipv4.sin_port = htons( key.local.Ipv4.sin_port );

-            stat.pid = find_owning_pid( pid_map, pid_map_size, inode );
-            stat.create_time = 0; /* FIXME */
-            stat.flags = 0; /* FIXME */
-            stat.mod_info = 0; /* FIXME */
+            if (stat_out)
+            {
+                stat.pid = find_owning_pid( pid_map, pid_map_size, inode );
+                stat.create_time = 0; /* FIXME */
+                stat.flags = 0; /* FIXME */
+                stat.mod_info = 0; /* FIXME */
+            }

if (num < *count)
             {
@@ -290,10 +293,13 @@ static NTSTATUS udp_endpoint_enumerate_all( void *key_data, UINT key_size, void
                 key.local.Ipv6.sin6_scope_id = find_ipv6_addr_scope( &key.local.Ipv6.sin6_addr, addr_scopes,
                                                                      addr_scopes_size );

-                stat.pid = find_owning_pid( pid_map, pid_map_size, inode );
-                stat.create_time = 0; /* FIXME */
-                stat.flags = 0; /* FIXME */
-                stat.mod_info = 0; /* FIXME */
+                if (stat_out)
+                {
+                    stat.pid = find_owning_pid( pid_map, pid_map_size, inode );
+                    stat.create_time = 0; /* FIXME */
+                    stat.flags = 0; /* FIXME */
+                    stat.mod_info = 0; /* FIXME */
+                }

if (num < *count)
                 {
@@ -344,7 +350,7 @@ static NTSTATUS udp_endpoint_enumerate_all( void *key_data, UINT key_size, void
         if (len <= sizeof(struct xinpgen)) goto err;

addr_scopes = get_ipv6_addr_scope_table( &addr_scopes_size );
-        pid_map = get_pid_map( &pid_map_size );
+        if (stat_out) pid_map = get_pid_map( &pid_map_size );

orig_xig = (struct xinpgen *)buf;
         xig = orig_xig;
@@ -393,10 +399,13 @@ static NTSTATUS udp_endpoint_enumerate_all( void *key_data, UINT key_size, void
                                                                      addr_scopes_size );
             }

-            stat.pid = find_owning_pid( pid_map, pid_map_size, (UINT_PTR)sock->so_pcb );
-            stat.create_time = 0; /* FIXME */
-            stat.flags = !(in->inp_flags & INP_ANONPORT);
-            stat.mod_info = 0; /* FIXME */
+            if (stat_out)
+            {
+                stat.pid = find_owning_pid( pid_map, pid_map_size, (UINT_PTR)sock->so_pcb );
+                stat.create_time = 0; /* FIXME */
+                stat.flags = !(in->inp_flags & INP_ANONPORT);
+                stat.mod_info = 0; /* FIXME */
+            }

if (num < *count)
             {