np_enum.c

Go to the documentation of this file.

#include <windows.h>
#include <stdio.h>
 
DWORD debug_shift = 0;
 
#define INC_SHIFT ++debug_shift;
#define DEC_SHIFT --debug_shift;
#define PRT_SHIFT do { DWORD cur = 0; for (; cur < debug_shift; ++cur) printf("\t"); } while (0);
 
void np_enum(NETRESOURCEW * resource)
{
    DWORD ret;
    HANDLE handle;
    DWORD size = 0x1000;
    NETRESOURCEW * out;
    BOOL check = FALSE;
 
    if (resource && resource->lpRemoteName)
        check = TRUE;
 
    ret = WNetOpenEnum(RESOURCE_GLOBALNET, RESOURCETYPE_DISK, 0, resource, &handle);
    if (ret != WN_SUCCESS)
        return;
 
    out = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, size);
    if (!out)
    {
        WNetCloseEnum(handle);
        return;
    }
 
    INC_SHIFT
 
    if (check)
    {
        printf("Called with lpRemoteName not null, current value: %S\n", resource->lpRemoteName);
    }
 
    do
    {
        DWORD count = -1;
 
        ret = WNetEnumResource(handle, &count, out, &size);
        if (ret == WN_SUCCESS || ret == WN_MORE_DATA)
        {
            NETRESOURCEW * current;
 
            current = out;
            for (; count; count--)
            {
                PRT_SHIFT;
                printf("lpRemoteName: %S\n", current->lpRemoteName);
 
                if ((current->dwUsage & RESOURCEUSAGE_CONTAINER) == RESOURCEUSAGE_CONTAINER)
                {
                    PRT_SHIFT;
                    printf("Found provider: %S\n", current->lpProvider);
                    np_enum(current);
                }
 
                current++;
            }
        }
    } while (ret != WN_NO_MORE_ENTRIES);
    DEC_SHIFT;
 
    HeapFree(GetProcessHeap(), 0, out);
    WNetCloseEnum(handle);
}
 
int wmain(int argc, const WCHAR *argv[])
{
    np_enum(NULL);
 
    return 0;
}