static DWORD _enumerateConnectedW(PWNetEnumerator enumerator, DWORD* user_count,
void* user_buffer, DWORD* user_size)
{
- DWORD ret, index, count, size, i, left;
+ DWORD ret, index, count, total_count, size, i, left;
void* end;
NETRESOURCEW* curr, * buffer;
HANDLE* handles;
curr = user_buffer;
end = (char *)user_buffer + size;
count = *user_count;
+ total_count = 0;
ret = WN_NO_MORE_ENTRIES;
for (index = 0; index < providerTable->numProviders; index++)
ret = providerTable->table[index].enumResource(handles[index],
&count, buffer,
&size);
+ total_count += count;
if (ret == WN_MORE_DATA)
break;
++curr;
}
- count = *user_count - count;
+ if (*user_count != -1)
+ count = *user_count - total_count;
+ else
+ count = *user_count;
size = left;
}
- if (ret != WN_SUCCESS || count == 0)
+ if (ret != WN_SUCCESS || total_count == 0)
break;
}
}
- if (count == 0)
+ if (total_count == 0)
ret = WN_NO_MORE_ENTRIES;
- *user_count = *user_count - count;
+ *user_count = total_count;
if (ret != WN_MORE_DATA && ret != WN_NO_MORE_ENTRIES)
ret = WN_SUCCESS;