vfdshext.cpp

Go to the documentation of this file.

/*
    vfdshext.cpp

    Virtual Floppy Drive for Windows
    Driver control library
    shell extension COM shell extension class

    Copyright (c) 2003-2005 Ken Kato
*/

#define WIN32_LEAN_AND_MEAN
#include <windows.h>
#include <shellapi.h>
#include <shlobj.h>

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

//  class header
#include "vfdshext.h"

//
//  Constructor
//
CVfdShExt::CVfdShExt()
{
    VFDTRACE(0, ("CVfdShExt::CVfdShExt()\n"));

    m_cRefCnt       = 0L;
    m_pDataObj      = NULL;
    m_nDevice       = (ULONG)-1;
    m_sTarget[0]    = '\0';
    m_bDragDrop     = FALSE;

    g_cDllRefCnt++;
}

//
//  Destructor
//
CVfdShExt::~CVfdShExt()
{
    VFDTRACE(0, ("CVfdShExt::~CVfdShExt()\n"));

    if (m_pDataObj) {
        m_pDataObj->Release();
    }

    g_cDllRefCnt--;
}

//  IUnknown members

STDMETHODIMP CVfdShExt::QueryInterface(
    REFIID          riid,
    LPVOID          *ppv)
{
    *ppv = NULL;

    if (IsEqualIID(riid, IID_IShellExtInit) ||
        IsEqualIID(riid, IID_IUnknown)) {
        VFDTRACE(0,
            ("CVfdShExt::QueryInterface()==>IID_IShellExtInit\n"));

        *ppv = (LPSHELLEXTINIT)this;
    }
    else if (IsEqualIID(riid, IID_IContextMenu)) {
        VFDTRACE(0,
            ("CVfdShExt::QueryInterface()==>IID_IContextMenu\n"));

        *ppv = (LPCONTEXTMENU)this;
    }
    else if (IsEqualIID(riid, IID_IShellPropSheetExt)) {
        VFDTRACE(0,
            ("CVfdShExt::QueryInterface()==>IID_IShellPropSheetExt\n"));

        *ppv = (LPSHELLPROPSHEETEXT)this;
    }

    if (*ppv) {
        AddRef();

        return NOERROR;
    }

    VFDTRACE(0,
        ("CVfdShExt::QueryInterface()==>Unknown Interface!\n"));

    return E_NOINTERFACE;
}

STDMETHODIMP_(ULONG) CVfdShExt::AddRef()
{
    VFDTRACE(0, ("CVfdShExt::AddRef()\n"));

    return ++m_cRefCnt;
}

STDMETHODIMP_(ULONG) CVfdShExt::Release()
{
    VFDTRACE(0, ("CVfdShExt::Release()\n"));

    if (--m_cRefCnt) {
        return m_cRefCnt;
    }

#ifndef __REACTOS__
    delete this;
#endif

    return 0L;
}

//  IShellExtInit members

//
//  Initialize
//  Called by the shell to initialize the shell extension object
//
STDMETHODIMP CVfdShExt::Initialize(
    LPCITEMIDLIST   pIDFolder,
    LPDATAOBJECT    pDataObj,
    HKEY            hRegKey)
{
    CHAR drive = '\0';

    VFDTRACE(0, ("CVfdShExt::Initialize()\n"));

    UNREFERENCED_PARAMETER(hRegKey);

    // Initialize can be called more than once

    if (m_pDataObj) {
        m_pDataObj->Release();
        m_pDataObj = NULL;
    }

    m_nDevice = (ULONG)-1;
    m_sTarget[0] = '\0';

    //  Get the folder name
    if (SHGetPathFromIDList(pIDFolder, m_sTarget)) {

        //  act as a Drag-and-Drop Handler

        VFDTRACE(0, ("Drag-Drop: %s\n", m_sTarget));

        if (GetDriveType(m_sTarget) != DRIVE_REMOVABLE) {
            VFDTRACE(0, ("Not a VFD drive\n"));
            return NOERROR;
        }

        drive = m_sTarget[0];
        m_bDragDrop = TRUE;
    }
    else {

        //  act as a context menu handler

        VFDTRACE(0, ("Context menu:\n"));
        m_bDragDrop = FALSE;
    }

    // Extract the target object name

    if (pDataObj) {

        STGMEDIUM medium;
        FORMATETC fmt = {
            CF_HDROP, NULL, DVASPECT_CONTENT, -1, TYMED_HGLOBAL
        };

        if (SUCCEEDED(pDataObj->GetData(&fmt, &medium))) {
            if (DragQueryFile((HDROP)medium.hGlobal, (UINT)-1, NULL, 0)) {

                DragQueryFile((HDROP)medium.hGlobal,
                    0, m_sTarget, sizeof(m_sTarget));
            }

            ReleaseStgMedium(&medium);
        }
    }

    VFDTRACE(0, ("Target %s\n", m_sTarget));

    if (!drive) {
        //  Contect menu handler
        //  -- Data object is the target drive
        drive = m_sTarget[0];
    }

    HANDLE hDevice = VfdOpenDevice(drive);

    if (hDevice == INVALID_HANDLE_VALUE) {
        VFDTRACE(0, ("Not a VFD drive\n"));
        return NOERROR;
    }

    ULONG ret = VfdGetDeviceNumber(hDevice, &m_nDevice);

    CloseHandle(hDevice);

    if (ret != ERROR_SUCCESS) {
        m_nDevice = (ULONG)-1;
        return NOERROR;
    }

    VFDTRACE(0, ("VFD device %d\n", m_nDevice));
    //  Store the data object

    m_pDataObj = pDataObj;
    m_pDataObj->AddRef();

    return NOERROR;
}

/*
STDMETHODIMP CVfdShExt::GetInfoFlags(
    DWORD *pdwFlags)
{
    VFDTRACE(0, ("CVfdShExt::GetInfoFlags\n"));
    *pdwFlags = 0;
    return NOERROR;
}

STDMETHODIMP CVfdShExt::GetInfoTip(
    DWORD dwFlags,
    LPWSTR *ppwszTip)
{
    VFDTRACE(0, ("CVfdShExt::GetInfoTip\n"));
    *ppwszTip = NULL;
    return NOERROR;
}
*/