psapi.c File Reference

#include <apitest.h>
#include <psapi.h>
#include <tlhelp32.h>
#include <stdio.h>

Include dependency graph for psapi.c:

Go to the source code of this file.

Classes
struct	TEST_MODULE_INFO

Functions
static LPVOID	IntGetImageBase (LPCSTR Image)
static BOOLEAN	IntGetModuleInformation (LPCSTR Module, BOOLEAN IsDriver, BOOLEAN IsProcMod, BOOLEAN BaseName, TEST_MODULE_INFO *Info)
	START_TEST (GetDeviceDriverFileName)
	START_TEST (GetDeviceDriverBaseName)

Function Documentation

IntGetImageBase()

static LPVOID IntGetImageBase ( LPCSTR  Image )

static

Definition at line 20 of file psapi.c.

{
    HANDLE Snap;
    MODULEENTRY32 Module;

    Snap = CreateToolhelp32Snapshot(TH32CS_SNAPMODULE, 0);
    if (Snap == INVALID_HANDLE_VALUE)
    {
        return NULL;
    }

    Module.dwSize = sizeof(MODULEENTRY32);
    if(!Module32First(Snap, &Module))
    {
        CloseHandle(Snap);
        return NULL;
    }

    do
    {
        if (lstrcmpiA(Module.szExePath, Image) == 0)
        {
            CloseHandle(Snap);
            return (LPVOID)Module.modBaseAddr;
        }
    } while(Module32Next(Snap, &Module));

    CloseHandle(Snap);
    return NULL;
}

Referenced by IntGetModuleInformation().

IntGetModuleInformation()

static BOOLEAN IntGetModuleInformation ( LPCSTR  Module, BOOLEAN  IsDriver, BOOLEAN  IsProcMod, BOOLEAN  BaseName, TEST_MODULE_INFO *  Info  )

static

Definition at line 51 of file psapi.c.

{
    CHAR System[255];
    UINT Len;

    memset(Info, 0, sizeof(TEST_MODULE_INFO));

    /* Get system path */
    Len = GetSystemWindowsDirectory(System, 255);
    if (Len > 255 || Len == 0)
    {
        printf("GetSystemWindowsDirectory failed\n");
        return FALSE;
    }

    /* Make path to module */
    strcat(System, "\\system32\\");
    if (IsDriver) strcat(System, "drivers\\");
    strcat(System, Module);

    /* Get base address */
    if (IsProcMod)
    {
        Info->ImageBase = IntGetImageBase(System);
        if (!Info->ImageBase)
        {
            printf("IntGetImageBase failed\n");
            return FALSE;
        }
    }
    else
    {
        /* FIXME */
        printf("Not supported yet!\n");
        return FALSE;
    }

    if (BaseName)
    {
        strcpy(Info->Path, Module);
        Info->Len = lstrlenA(Info->Path);
    }
    else
    {
        /* Skip disk */
        strcpy(Info->Path, System + 2);
        Info->Len = lstrlenA(Info->Path);
    }

    return TRUE;
}

Referenced by START_TEST().

START_TEST() [1/2]

START_TEST

(

GetDeviceDriverFileName

)

Definition at line 103 of file psapi.c.

{
    DWORD Len;
    CHAR FileName[255];
    TEST_MODULE_INFO ModInfo;

    SetLastError(0xDEADBEEF);
    Len = GetDeviceDriverFileNameA(0, FileName, 255);
    ok(Len == 0, "Len: %lu\n", Len);
    ok(GetLastError() == ERROR_INVALID_HANDLE, "Error: %lx\n", GetLastError());

    if (IntGetModuleInformation("ntdll.dll", FALSE, TRUE, FALSE, &ModInfo))
    {
        SetLastError(0xDEADBEEF);
        Len = GetDeviceDriverFileNameA(ModInfo.ImageBase, FileName, 255);
        ok(Len == ModInfo.Len, "Len: %lu\n", Len);
        ok(GetLastError() == 0xDEADBEEF, "Error: %lx\n", GetLastError());
        ok(lstrcmpiA(ModInfo.Path, FileName) == 0, "File name: %s\n", FileName);

        /* Test with too small buffer */
        SetLastError(0xDEADBEEF);
        ModInfo.Len--;
        ModInfo.Path[ModInfo.Len] = 0;
        FileName[ModInfo.Len] = 0;
        Len = GetDeviceDriverFileNameA(ModInfo.ImageBase, FileName, ModInfo.Len);
        ok(Len == ModInfo.Len, "Len: %lu\n", Len);
        ok(GetLastError() == 0xDEADBEEF, "Error: %lx\n", GetLastError());
        ok(lstrcmpiA(ModInfo.Path, FileName) == 0, "File name: %s\n", FileName);
    }
    else
    {
        skip("Couldn't find info about ntdll.dll\n");
    }

    if (IntGetModuleInformation("msvcrt.dll", FALSE, TRUE, FALSE, &ModInfo))
    {
        SetLastError(0xDEADBEEF);
        Len = GetDeviceDriverFileNameA(ModInfo.ImageBase, FileName, 255);
        ok(Len == 0, "Len: %lu\n", Len);
        ok(GetLastError() == ERROR_INVALID_HANDLE, "Error: %lx\n", GetLastError());
    }
    else
    {
        skip("Couldn't find info about msvcrt.dll\n");
    }

    if (IntGetModuleInformation("psapi.dll", FALSE, TRUE, FALSE, &ModInfo))
    {
        SetLastError(0xDEADBEEF);
        Len = GetDeviceDriverFileNameA(ModInfo.ImageBase, FileName, 255);
        ok(Len == 0, "Len: %lu\n", Len);
        ok(GetLastError() == ERROR_INVALID_HANDLE, "Error: %lx\n", GetLastError());
    }
    else
    {
        skip("Couldn't find info about psapi.dll\n");
    }
}

START_TEST() [2/2]

START_TEST

(

GetDeviceDriverBaseName

)

Definition at line 162 of file psapi.c.

{
    DWORD Len;
    CHAR FileName[255];
    TEST_MODULE_INFO ModInfo;

    SetLastError(0xDEADBEEF);
    Len = GetDeviceDriverBaseNameA(0, FileName, 255);
    ok(Len == 0, "Len: %lu\n", Len);
    ok(GetLastError() == ERROR_INVALID_HANDLE, "Error: %lx\n", GetLastError());

    if (IntGetModuleInformation("ntdll.dll", FALSE, TRUE, TRUE, &ModInfo))
    {
        SetLastError(0xDEADBEEF);
        Len = GetDeviceDriverBaseNameA(ModInfo.ImageBase, FileName, 255);
        ok(Len == ModInfo.Len, "Len: %lu\n", Len);
        ok(GetLastError() == 0xDEADBEEF, "Error: %lx\n", GetLastError());
        ok(lstrcmpiA(ModInfo.Path, FileName) == 0, "File name: %s\n", FileName);

        /* Test with too small buffer */
        SetLastError(0xDEADBEEF);
        ModInfo.Len--;
        ModInfo.Path[ModInfo.Len] = 0;
        FileName[ModInfo.Len] = 0;
        Len = GetDeviceDriverBaseNameA(ModInfo.ImageBase, FileName, ModInfo.Len);
        ok(Len == ModInfo.Len, "Len: %lu\n", Len);
        ok(GetLastError() == 0xDEADBEEF, "Error: %lx\n", GetLastError());
        ok(lstrcmpiA(ModInfo.Path, FileName) == 0, "File name: %s\n", FileName);
    }
    else
    {
        skip("Couldn't find info about ntdll.dll\n");
    }

    if (IntGetModuleInformation("msvcrt.dll", FALSE, TRUE, TRUE, &ModInfo))
    {
        SetLastError(0xDEADBEEF);
        Len = GetDeviceDriverBaseNameA(ModInfo.ImageBase, FileName, 255);
        ok(Len == 0, "Len: %lu\n", Len);
        ok(GetLastError() == ERROR_INVALID_HANDLE, "Error: %lx\n", GetLastError());
    }
    else
    {
        skip("Couldn't find info about msvcrt.dll\n");
    }

    if (IntGetModuleInformation("psapi.dll", FALSE, TRUE, TRUE, &ModInfo))
    {
        SetLastError(0xDEADBEEF);
        Len = GetDeviceDriverBaseNameA(ModInfo.ImageBase, FileName, 255);
        ok(Len == 0, "Len: %lu\n", Len);
        ok(GetLastError() == ERROR_INVALID_HANDLE, "Error: %lx\n", GetLastError());
    }
    else
    {
        skip("Couldn't find info about psapi.dll\n");
    }
}