3rdparty/lua/src/modules/wow64.c


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47

///////////////////////////////////////////////////////////////////////////////
///
/// Written 2012, Oliver Schneider (assarbad.net) - PUBLIC DOMAIN/CC0
///
/// Original filename: wow64.c
/// Project          : WinDirStat
/// Author(s)        : Oliver Schneider
///
/// Purpose          : Lua function, implemented in C, to check whether the
///                    current process is running as WOW64 process.
///
///////////////////////////////////////////////////////////////////////////////

#include <Windows.h>
#include <tchar.h>
#include <lua.h>

static int luaC_iswow64_(lua_State* L)
{
    static int *pcachedResult = NULL;
    if(!pcachedResult)
    {
        typedef BOOL (WINAPI *TFNIsWow64Process) (HANDLE, PBOOL);
        static int cachedResult = 0;
        TFNIsWow64Process pfnIsWow64Process = (TFNIsWow64Process)GetProcAddress(GetModuleHandle(TEXT("kernel32.dll")), "IsWow64Process");
        if(pfnIsWow64Process)
        {
            BOOL bIsWow64 = FALSE;
            // Ignore result, instead presume it's not WOW64
            (void)pfnIsWow64Process(GetCurrentProcess(), &bIsWow64);
            cachedResult = (bIsWow64) ? 1 : 0;
        }
        pcachedResult = &cachedResult;
    }
    if(!pcachedResult)
    {
        lua_pushnil(L);
        return 1;
    }
    lua_pushboolean(L, *pcachedResult);
    return 1;
}

LUA_API const luaL_Reg wow64_funcs[] = {
    {"iswow64", luaC_iswow64_},
    {NULL, NULL}
};