vfdlib.c

Go to the documentation of this file.

/*
    vfdlib.c

    Virtual Floppy Drive for Windows
    Driver control library
    Miscellaneous functions

    Copyright (C) 2003-2008 Ken Kato
*/

#ifdef __cplusplus
#pragma message(__FILE__": Compiled as C++ for testing purpose.")
#endif  // __cplusplus

#define WIN32_LEAN_AND_MEAN
#include <windows.h>
#include <stdio.h>

#ifdef _DEBUG
#include <psapi.h>
#endif  // _DEBUG

#include "vfdtypes.h"
#include "vfdapi.h"
#include "vfdlib.h"

//
// DLL Global variables
//

#ifndef __REACTOS__
extern HINSTANCE    g_hDllModule = NULL;    // Handle to this DLL itself
extern UINT         g_cDllRefCnt = 0;       // Reference count of this DLL
extern UINT         g_nNotifyMsg = 0;       // VFD notification message
#else
HINSTANCE   g_hDllModule = NULL;    // Handle to this DLL itself
UINT            g_cDllRefCnt = 0;       // Reference count of this DLL
UINT            g_nNotifyMsg = 0;       // VFD notification message
#endif

//
// DllMain
//
BOOL WINAPI DllMain(
    HINSTANCE       hInstance,
    DWORD           dwReason,
    LPVOID          lpReserved)
{
#ifdef _DEBUG
    char            name[MAX_PATH];
    HMODULE         hMod;
    DWORD           size;

    if (EnumProcessModules(GetCurrentProcess(), &hMod, sizeof(hMod), &size)) {
        GetModuleBaseName(GetCurrentProcess(), hMod, name, sizeof(name));
    }
    else {
        strcpy(name, "unknown");
    }
#endif  // _DEBUG

    UNREFERENCED_PARAMETER(lpReserved);

    if (dwReason == DLL_PROCESS_ATTACH) {
        VFDTRACE(0, ("DLL_PROCESS_ATTACH - %s\n", name));

        //  this DLL doesn't need DLL_THREAD_ATTACH and DLL_THREAD_DETACH
        DisableThreadLibraryCalls(hInstance);

        //  store the DLL instance handle
        g_hDllModule = hInstance;

        //  register the VFD notification message
        g_nNotifyMsg = RegisterWindowMessage(VFD_NOTIFY_MESSAGE);

    }
    else if (dwReason == DLL_PROCESS_DETACH) {
        VFDTRACE(0, ("DLL_PROCESS_DETACH - %s\n", name));
    }

    return TRUE;
}

//
//  Check running platform
//
BOOL WINAPI VfdIsValidPlatform()
{
    BOOL (WINAPI *pfnIsWow64Process)(HANDLE, PBOOL);
    BOOL wow64;

    if (GetVersion() & 0x80000000) {
        return FALSE;       // doesn't work on Win9x
    }

    pfnIsWow64Process = (BOOL (WINAPI *)(HANDLE, PBOOL))
        GetProcAddress(GetModuleHandle("kernel32.dll"), "IsWow64Process");

    if (pfnIsWow64Process == NULL) {
        return TRUE;        // NT4 or 2000 -- assured to be 32 bit
    }

    wow64 = FALSE;

    if (!pfnIsWow64Process(GetCurrentProcess(), &wow64)) {
        return FALSE;
    }

    return !wow64;
}

//
//  Get VFD notification message value
//
UINT WINAPI VfdGetNotifyMessage()
{
    return g_nNotifyMsg;
}


//
//  Get message text from this DLL module
//
PSTR ModuleMessage(
    DWORD           nFormat, ...)
{
    PSTR    p;
    va_list args;

    va_start(args, nFormat);

    if (!FormatMessage(
        FORMAT_MESSAGE_FROM_HMODULE |
        FORMAT_MESSAGE_ALLOCATE_BUFFER,
        g_hDllModule, nFormat, 0, (LPTSTR)&p, 0, &args)) {
        p = NULL;
    }

    va_end(args);

    return p;
}

//
//  Get system error message string
//
PCSTR SystemMessage(
    DWORD           nError)
{
    static CHAR     msg[256];

    if (!FormatMessage(
        FORMAT_MESSAGE_FROM_SYSTEM |
        FORMAT_MESSAGE_IGNORE_INSERTS,
        NULL, nError, 0, msg, sizeof(msg), NULL)) {

        _snprintf(msg, sizeof(msg),
            "Unknown system error %lu (0x%08x)\n", nError, nError);
    }

    return msg;
}

#ifdef _DEBUG
//
//  Format and output debug string
//
void DebugTrace(
    PCSTR           sFormat, ...)
{
    CHAR            buf[512];
    int             len;
    va_list         args;

    len = _snprintf(buf, sizeof(buf),
        "%s(%lu) : ", TraceFile, TraceLine);

    va_start(args, sFormat);

    _vsnprintf(buf + len, sizeof(buf) - len, sFormat, args);

    OutputDebugString(buf);
}
#endif  // _DEBUG