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