system/wine
system/wine
1
0
Fork 0
mirror of git://source.winehq.org/git/wine.git synced 2024-07-21 02:06:39 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

iphlpapi: Reimplement get_pid_map() using the list_processes request.

Browse source 
Signed-off-by: Zebediah Figura <zfigura@codeweavers.com>
Signed-off-by: Alexandre Julliard <julliard@winehq.org>
This commit is contained in:
Zebediah Figura 2020-07-06 21:06:01 -05:00 committed by Alexandre Julliard
parent 8622eb326f
commit 747573f1e4
1 changed files with 53 additions and 40 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

93
dlls/iphlpapi/ipstats.c
View file

@ -1887,56 +1887,69 @@ struct pid_map
static struct pid_map *get_pid_map( unsigned int *num_entries ) static struct pid_map *get_pid_map( unsigned int *num_entries )
{ {
HANDLE snapshot = NULL;
struct pid_map *map; struct pid_map *map;
unsigned int i = 0, count = 16, size = count * sizeof(*map); unsigned int i = 0, map_count = 16, buffer_len = 4096, process_count, pos = 0;
NTSTATUS ret; NTSTATUS ret;
char *buffer = NULL, *new_buffer;
if (!(map = HeapAlloc( GetProcessHeap(), 0, size ))) return NULL; if (!(buffer = HeapAlloc( GetProcessHeap(), 0, buffer_len ))) return NULL;
SERVER_START_REQ( create_snapshot ) for (;;)
{ {
req->flags = SNAP_PROCESS; SERVER_START_REQ( list_processes )
req->attributes = 0;
if (!(ret = wine_server_call( req )))
snapshot = wine_server_ptr_handle( reply->handle );
}
SERVER_END_REQ;
*num_entries = 0;
while (ret == STATUS_SUCCESS)
{
SERVER_START_REQ( next_process )
{ {
req->handle = wine_server_obj_handle( snapshot ); wine_server_set_reply( req, buffer, buffer_len );
req->reset = (i == 0); ret = wine_server_call( req );
if (!(ret = wine_server_call( req ))) buffer_len = reply->info_size;
{ process_count = reply->process_count;
if (i >= count)
{
struct pid_map *new_map;
count *= 2;
size = count * sizeof(*new_map);
if (!(new_map = HeapReAlloc( GetProcessHeap(), 0, map, size )))
{
HeapFree( GetProcessHeap(), 0, map );
map = NULL;
goto done;
}
map = new_map;
}
map[i].pid = reply->pid;
map[i].unix_pid = reply->unix_pid;
(*num_entries)++;
i++;
}
} }
SERVER_END_REQ; SERVER_END_REQ;
if (ret != STATUS_INFO_LENGTH_MISMATCH) break;
if (!(new_buffer = HeapReAlloc( GetProcessHeap(), 0, buffer, buffer_len )))
{
HeapFree( GetProcessHeap(), 0, buffer );
return NULL;
}
} }
done: if (!(map = HeapAlloc( GetProcessHeap(), 0, map_count * sizeof(*map) )))
NtClose( snapshot ); {
HeapFree( GetProcessHeap(), 0, buffer );
return NULL;
}
for (i = 0; i < process_count; ++i)
{
const struct process_info *process;
pos = (pos + 7) & ~7;
process = (const struct process_info *)(buffer + pos);
if (i >= map_count)
{
struct pid_map *new_map;
map_count *= 2;
if (!(new_map = HeapReAlloc( GetProcessHeap(), 0, map, map_count * sizeof(*map))))
{
HeapFree( GetProcessHeap(), 0, map );
HeapFree( GetProcessHeap(), 0, buffer );
return NULL;
}
map = new_map;
}
map[i].pid = process->pid;
map[i].unix_pid = process->unix_pid;
pos += sizeof(struct process_info) + process->name_len;
pos = (pos + 7) & ~7;
pos += process->thread_count * sizeof(struct thread_info);
}
HeapFree( GetProcessHeap(), 0, buffer );
*num_entries = process_count;
return map; return map;
} }