source: trunk/src/shell32/shellole.c

/*
 *      handling of SHELL32.DLL OLE-Objects
 *
 *      Copyright 1997  Marcus Meissner
 *      Copyright 1998  Juergen Schmied  <juergen.schmied@metronet.de>
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 */

#include "config.h"

#include <stdlib.h>
#include <string.h>

#include "shellapi.h"
#include "shlobj.h"
#include "shlguid.h"
#include "winreg.h"
#include "winerror.h"

#include "undocshell.h"
#include "wine/unicode.h"
#include "shell32_main.h"

#include "wine/debug.h"

WINE_DEFAULT_DEBUG_CHANNEL(shell);

DWORD WINAPI SHCLSIDFromStringA (LPCSTR clsid, CLSID *id);
extern IShellFolder * IShellFolder_Constructor(
        IShellFolder * psf,
        LPITEMIDLIST pidl);
extern HRESULT IFSFolder_Constructor(
        IUnknown * pUnkOuter,
        REFIID riid,
        LPVOID * ppv);

/*************************************************************************
 * SHCoCreateInstance [SHELL32.102]
 *
 * NOTES
 *     exported by ordinal
 */
LRESULT WINAPI SHCoCreateInstance(
        LPCSTR aclsid,
        REFCLSID clsid,
        LPUNKNOWN unknownouter,
        REFIID refiid,
        LPVOID *ppv)
{
        DWORD   hres;
        IID     iid;
        CLSID * myclsid = (CLSID*)clsid;

        if (!clsid)
        {
          if (!aclsid) return REGDB_E_CLASSNOTREG;
          SHCLSIDFromStringA(aclsid, &iid);
          myclsid = &iid;
        }

        TRACE("(%p,\n\tCLSID:\t%s, unk:%p\n\tIID:\t%s,%p)\n",
                aclsid,debugstr_guid(myclsid),unknownouter,debugstr_guid(refiid),ppv);

        if IsEqualCLSID(myclsid, &CLSID_ShellFSFolder)
        {
          hres = IFSFolder_Constructor(unknownouter, refiid, ppv);
        }
        else
        {
          CoInitialize(NULL);
          hres = CoCreateInstance(myclsid, unknownouter, CLSCTX_INPROC_SERVER, refiid, ppv);
        }

        if(hres!=S_OK)
        {
          ERR("failed (0x%08lx) to create \n\tCLSID:\t%s\n\tIID:\t%s\n",
              hres, debugstr_guid(myclsid), debugstr_guid(refiid));
          ERR("class not found in registry\n");
        }

        TRACE("-- instance: %p\n",*ppv);
        return hres;
}

/*************************************************************************
 * DllGetClassObject   [SHELL32.128]
 */
HRESULT WINAPI SHELL32_DllGetClassObject(REFCLSID rclsid, REFIID iid,LPVOID *ppv)
{       HRESULT hres = E_OUTOFMEMORY;
        LPCLASSFACTORY lpclf;

        TRACE("\n\tCLSID:\t%s,\n\tIID:\t%s\n",debugstr_guid(rclsid),debugstr_guid(iid));

        *ppv = NULL;

        if(IsEqualCLSID(rclsid, &CLSID_ShellDesktop)||
           IsEqualCLSID(rclsid, &CLSID_ShellLink))
        {
          lpclf = IClassFactory_Constructor( rclsid );

          if(lpclf)
          {
            hres = IClassFactory_QueryInterface(lpclf,iid, ppv);
            IClassFactory_Release(lpclf);
          }
        }
        else
        {
          WARN("-- CLSID not found\n");
          hres = CLASS_E_CLASSNOTAVAILABLE;
        }
        TRACE("-- pointer to class factory: %p\n",*ppv);
        return hres;
}

/*************************************************************************
 * SHCLSIDFromString                            [SHELL32.147]
 *
 * NOTES
 *     exported by ordinal
 */
DWORD WINAPI SHCLSIDFromStringA (LPCSTR clsid, CLSID *id)
{
    WCHAR buffer[40];
    TRACE("(%p(%s) %p)\n", clsid, clsid, id);
    if (!MultiByteToWideChar( CP_ACP, 0, clsid, -1, buffer, sizeof(buffer)/sizeof(WCHAR) ))
        return CO_E_CLASSSTRING;
    return CLSIDFromString( buffer, id );
}
DWORD WINAPI SHCLSIDFromStringW (LPWSTR clsid, CLSID *id)
{
        TRACE("(%p(%s) %p)\n", clsid, debugstr_w(clsid), id);
        return CLSIDFromString(clsid, id);
}
DWORD WINAPI SHCLSIDFromStringAW (LPVOID clsid, CLSID *id)
{
        if (SHELL_OsIsUnicode())
          return SHCLSIDFromStringW (clsid, id);
        return SHCLSIDFromStringA (clsid, id);
}

/*************************************************************************
 *                       SHGetMalloc                    [SHELL32.@]
 * returns the interface to shell malloc.
 *
 * [SDK header win95/shlobj.h:
 * equivalent to:  #define SHGetMalloc(ppmem)   CoGetMalloc(MEMCTX_TASK, ppmem)
 * ]
 * What we are currently doing is not very wrong, since we always use the same
 * heap (ProcessHeap).
 */
DWORD WINAPI SHGetMalloc(LPMALLOC *lpmal)
{
        TRACE("(%p)\n", lpmal);
        return CoGetMalloc(MEMCTX_TASK, lpmal);
}

/*************************************************************************
 * SHGetDesktopFolder                   [SHELL32.@]
 */
LPSHELLFOLDER pdesktopfolder=NULL;

DWORD WINAPI SHGetDesktopFolder(IShellFolder **psf)
{
        HRESULT hres = S_OK;
        LPCLASSFACTORY lpclf;
        TRACE("%p->(%p)\n",psf,*psf);

        *psf=NULL;

        if (!pdesktopfolder)
        {
          lpclf = IClassFactory_Constructor(&CLSID_ShellDesktop);
          if(lpclf)
          {
            hres = IClassFactory_CreateInstance(lpclf,NULL,(REFIID)&IID_IShellFolder, (void*)&pdesktopfolder);
            IClassFactory_Release(lpclf);
          }
        }

        if (pdesktopfolder)
        {
          /* even if we create the folder, add a ref so the application canŽt destroy the folder*/
          IShellFolder_AddRef(pdesktopfolder);
          *psf = pdesktopfolder;
        }

        TRACE("-- %p->(%p)\n",psf, *psf);
        return hres;
}

/**************************************************************************
*  IClassFactory Implementation
*/

typedef struct
{
    /* IUnknown fields */
    ICOM_VFIELD(IClassFactory);
    DWORD                       ref;
    CLSID                       *rclsid;
} IClassFactoryImpl;

static ICOM_VTABLE(IClassFactory) clfvt;

/**************************************************************************
 *  IClassFactory_Constructor
 */

LPCLASSFACTORY IClassFactory_Constructor(REFCLSID rclsid)
{
        IClassFactoryImpl* lpclf;

        lpclf= (IClassFactoryImpl*)HeapAlloc(GetProcessHeap(),0,sizeof(IClassFactoryImpl));
        lpclf->ref = 1;
        ICOM_VTBL(lpclf) = &clfvt;
        lpclf->rclsid = (CLSID*)rclsid;

        TRACE("(%p)->()\n",lpclf);
        InterlockedIncrement(&shell32_ObjCount);
        return (LPCLASSFACTORY)lpclf;
}
/**************************************************************************
 *  IClassFactory_QueryInterface
 */
static HRESULT WINAPI IClassFactory_fnQueryInterface(
  LPCLASSFACTORY iface, REFIID riid, LPVOID *ppvObj)
{
        ICOM_THIS(IClassFactoryImpl,iface);
        TRACE("(%p)->(\n\tIID:\t%s)\n",This,debugstr_guid(riid));

        *ppvObj = NULL;

        if(IsEqualIID(riid, &IID_IUnknown))          /*IUnknown*/
        { *ppvObj = This;
        }
        else if(IsEqualIID(riid, &IID_IClassFactory))  /*IClassFactory*/
        { *ppvObj = (IClassFactory*)This;
        }

        if(*ppvObj)
        { IUnknown_AddRef((LPUNKNOWN)*ppvObj);
          TRACE("-- Interface: (%p)->(%p)\n",ppvObj,*ppvObj);
          return S_OK;
        }
        TRACE("-- Interface: %s E_NOINTERFACE\n", debugstr_guid(riid));
        return E_NOINTERFACE;
}
/******************************************************************************
 * IClassFactory_AddRef
 */
static ULONG WINAPI IClassFactory_fnAddRef(LPCLASSFACTORY iface)
{
        ICOM_THIS(IClassFactoryImpl,iface);
        TRACE("(%p)->(count=%lu)\n",This,This->ref);

        InterlockedIncrement(&shell32_ObjCount);
        return InterlockedIncrement(&This->ref);
}
/******************************************************************************
 * IClassFactory_Release
 */
static ULONG WINAPI IClassFactory_fnRelease(LPCLASSFACTORY iface)
{
        ICOM_THIS(IClassFactoryImpl,iface);
        TRACE("(%p)->(count=%lu)\n",This,This->ref);

        InterlockedDecrement(&shell32_ObjCount);
        if (!InterlockedDecrement(&This->ref))
        {
          TRACE("-- destroying IClassFactory(%p)\n",This);
          HeapFree(GetProcessHeap(),0,This);
          return 0;
        }
        return This->ref;
}
/******************************************************************************
 * IClassFactory_CreateInstance
 */
static HRESULT WINAPI IClassFactory_fnCreateInstance(
  LPCLASSFACTORY iface, LPUNKNOWN pUnknown, REFIID riid, LPVOID *ppObject)
{
        ICOM_THIS(IClassFactoryImpl,iface);
        IUnknown *pObj = NULL;
        HRESULT hres;

        TRACE("%p->(%p,\n\tIID:\t%s,%p)\n",This,pUnknown,debugstr_guid(riid),ppObject);

        *ppObject = NULL;

        if(pUnknown)
        {
          return(CLASS_E_NOAGGREGATION);
        }

        if (IsEqualCLSID(This->rclsid, &CLSID_ShellDesktop))
        {
          pObj = (IUnknown *)ISF_Desktop_Constructor();
        }
        else if (IsEqualCLSID(This->rclsid, &CLSID_ShellLink))
        {
          pObj = (IUnknown *)IShellLink_Constructor(FALSE);
        }
        else
        {
          ERR("unknown IID requested\n\tIID:\t%s\n",debugstr_guid(riid));
          return(E_NOINTERFACE);
        }

        if (!pObj)
        {
          return(E_OUTOFMEMORY);
        }

        hres = IUnknown_QueryInterface(pObj,riid, ppObject);
        IUnknown_Release(pObj);

        TRACE("-- Object created: (%p)->%p\n",This,*ppObject);

        return hres;
}
/******************************************************************************
 * IClassFactory_LockServer
 */
static HRESULT WINAPI IClassFactory_fnLockServer(LPCLASSFACTORY iface, BOOL fLock)
{
        ICOM_THIS(IClassFactoryImpl,iface);
        TRACE("%p->(0x%x), not implemented\n",This, fLock);
        return E_NOTIMPL;
}

static ICOM_VTABLE(IClassFactory) clfvt =
{
    ICOM_MSVTABLE_COMPAT_DummyRTTIVALUE
    IClassFactory_fnQueryInterface,
    IClassFactory_fnAddRef,
  IClassFactory_fnRelease,
  IClassFactory_fnCreateInstance,
  IClassFactory_fnLockServer
};

/**************************************************************************
 * Default ClassFactory Implementation
 *
 * SHCreateDefClassObject
 *
 * NOTES
 *  helper function for dll's without a own classfactory
 *  a generic classfactory is returned
 *  when the CreateInstance of the cf is called the callback is executed
 */
#ifdef __WIN32OS2__
typedef HRESULT (* CALLBACK LPFNCREATEINSTANCE)(IUnknown* pUnkOuter, REFIID riid, LPVOID* ppvObject);
#else
typedef HRESULT (CALLBACK *LPFNCREATEINSTANCE)(IUnknown* pUnkOuter, REFIID riid, LPVOID* ppvObject);
#endif

typedef struct
{
    ICOM_VFIELD(IClassFactory);
    DWORD                       ref;
    CLSID                       *rclsid;
    LPFNCREATEINSTANCE          lpfnCI;
    const IID *                 riidInst;
    ULONG *                     pcRefDll; /* pointer to refcounter in external dll (ugrrr...) */
} IDefClFImpl;

static ICOM_VTABLE(IClassFactory) dclfvt;

/**************************************************************************
 *  IDefClF_fnConstructor
 */

IClassFactory * IDefClF_fnConstructor(LPFNCREATEINSTANCE lpfnCI, PLONG pcRefDll, REFIID riidInst)
{
        IDefClFImpl* lpclf;

        lpclf = (IDefClFImpl*)HeapAlloc(GetProcessHeap(),0,sizeof(IDefClFImpl));
        lpclf->ref = 1;
        ICOM_VTBL(lpclf) = &dclfvt;
        lpclf->lpfnCI = lpfnCI;
        lpclf->pcRefDll = pcRefDll;

        if (pcRefDll) InterlockedIncrement(pcRefDll);
        lpclf->riidInst = riidInst;

        TRACE("(%p)\n\tIID:\t%s\n",lpclf, debugstr_guid(riidInst));
        InterlockedIncrement(&shell32_ObjCount);
        return (LPCLASSFACTORY)lpclf;
}
/**************************************************************************
 *  IDefClF_fnQueryInterface
 */
static HRESULT WINAPI IDefClF_fnQueryInterface(
  LPCLASSFACTORY iface, REFIID riid, LPVOID *ppvObj)
{
        ICOM_THIS(IDefClFImpl,iface);

        TRACE("(%p)->(\n\tIID:\t%s)\n",This,debugstr_guid(riid));

        *ppvObj = NULL;

        if(IsEqualIID(riid, &IID_IUnknown))          /*IUnknown*/
        { *ppvObj = This;
        }
        else if(IsEqualIID(riid, &IID_IClassFactory))  /*IClassFactory*/
        { *ppvObj = (IClassFactory*)This;
        }

        if(*ppvObj)
        { IUnknown_AddRef((LPUNKNOWN)*ppvObj);
          TRACE("-- Interface: (%p)->(%p)\n",ppvObj,*ppvObj);
          return S_OK;
        }
        TRACE("-- Interface: %s E_NOINTERFACE\n", debugstr_guid(riid));
        return E_NOINTERFACE;
}
/******************************************************************************
 * IDefClF_fnAddRef
 */
static ULONG WINAPI IDefClF_fnAddRef(LPCLASSFACTORY iface)
{
        ICOM_THIS(IDefClFImpl,iface);
        TRACE("(%p)->(count=%lu)\n",This,This->ref);

        InterlockedIncrement(&shell32_ObjCount);
        return InterlockedIncrement(&This->ref);
}
/******************************************************************************
 * IDefClF_fnRelease
 */
static ULONG WINAPI IDefClF_fnRelease(LPCLASSFACTORY iface)
{
        ICOM_THIS(IDefClFImpl,iface);
        TRACE("(%p)->(count=%lu)\n",This,This->ref);

        InterlockedDecrement(&shell32_ObjCount);

        if (!InterlockedDecrement(&This->ref))
        {
          if (This->pcRefDll) InterlockedDecrement(This->pcRefDll);

          TRACE("-- destroying IClassFactory(%p)\n",This);
          HeapFree(GetProcessHeap(),0,This);
          return 0;
        }
        return This->ref;
}
/******************************************************************************
 * IDefClF_fnCreateInstance
 */
static HRESULT WINAPI IDefClF_fnCreateInstance(
  LPCLASSFACTORY iface, LPUNKNOWN pUnkOuter, REFIID riid, LPVOID *ppvObject)
{
        ICOM_THIS(IDefClFImpl,iface);

        TRACE("%p->(%p,\n\tIID:\t%s,%p)\n",This,pUnkOuter,debugstr_guid(riid),ppvObject);

        *ppvObject = NULL;

        if(pUnkOuter)
          return(CLASS_E_NOAGGREGATION);

        if ( This->riidInst==NULL ||
             IsEqualCLSID(riid, This->riidInst) ||
             IsEqualCLSID(riid, &IID_IUnknown) )
        {
          return This->lpfnCI(pUnkOuter, riid, ppvObject);
        }

        ERR("unknown IID requested\n\tIID:\t%s\n",debugstr_guid(riid));
        return E_NOINTERFACE;
}
/******************************************************************************
 * IDefClF_fnLockServer
 */
static HRESULT WINAPI IDefClF_fnLockServer(LPCLASSFACTORY iface, BOOL fLock)
{
        ICOM_THIS(IDefClFImpl,iface);
        TRACE("%p->(0x%x), not implemented\n",This, fLock);
        return E_NOTIMPL;
}

static ICOM_VTABLE(IClassFactory) dclfvt =
{
    ICOM_MSVTABLE_COMPAT_DummyRTTIVALUE
    IDefClF_fnQueryInterface,
    IDefClF_fnAddRef,
  IDefClF_fnRelease,
  IDefClF_fnCreateInstance,
  IDefClF_fnLockServer
};

/******************************************************************************
 * SHCreateDefClassObject                       [SHELL32.70]
 */
HRESULT WINAPI SHCreateDefClassObject(
        REFIID  riid,
        LPVOID* ppv,
        LPFNCREATEINSTANCE lpfnCI,      /* [in] create instance callback entry */
        LPDWORD pcRefDll,               /* [in/out] ref count of the dll */
        REFIID  riidInst)               /* [in] optional interface to the instance */
{
        TRACE("\n\tIID:\t%s %p %p %p \n\tIIDIns:\t%s\n",
              debugstr_guid(riid), ppv, lpfnCI, pcRefDll, debugstr_guid(riidInst));

        if ( IsEqualCLSID(riid, &IID_IClassFactory) )
        {
          IClassFactory * pcf = IDefClF_fnConstructor(lpfnCI, pcRefDll, riidInst);
          if (pcf)
          {
            *ppv = pcf;
            return NOERROR;
          }
          return E_OUTOFMEMORY;
        }
        return E_NOINTERFACE;
}

/*************************************************************************
 *  DragAcceptFiles             [SHELL32.54]
 */
void WINAPI DragAcceptFiles(HWND hWnd, BOOL b)
{
        LONG exstyle;

        if( !IsWindow(hWnd) ) return;
        exstyle = GetWindowLongA(hWnd,GWL_EXSTYLE);
        if (b)
          exstyle |= WS_EX_ACCEPTFILES;
        else
          exstyle &= ~WS_EX_ACCEPTFILES;
        SetWindowLongA(hWnd,GWL_EXSTYLE,exstyle);
}

/*************************************************************************
 * DragFinish           [SHELL32.80]
 */
void WINAPI DragFinish(HDROP h)
{
        TRACE("\n");
        GlobalFree((HGLOBAL)h);
}

/*************************************************************************
 * DragQueryPoint               [SHELL32.135]
 */
BOOL WINAPI DragQueryPoint(HDROP hDrop, POINT *p)
{
        DROPFILES *lpDropFileStruct;
        BOOL bRet;

        TRACE("\n");

        lpDropFileStruct = (DROPFILES *) GlobalLock(hDrop);

        *p = lpDropFileStruct->pt;
        bRet = lpDropFileStruct->fNC;

        GlobalUnlock(hDrop);
        return bRet;
}

/*************************************************************************
 *  DragQueryFile               [SHELL32.81]
 *  DragQueryFileA              [SHELL32.82]
 */
UINT WINAPI DragQueryFileA(
        HDROP hDrop,
        UINT lFile,
        LPSTR lpszFile,
        UINT lLength)
{
        LPSTR lpDrop;
        UINT i = 0;
        DROPFILES *lpDropFileStruct = (DROPFILES *) GlobalLock(hDrop);

        TRACE("(%08x, %x, %p, %u)\n",   hDrop,lFile,lpszFile,lLength);

        if(!lpDropFileStruct) goto end;

        lpDrop = (LPSTR) lpDropFileStruct + lpDropFileStruct->pFiles;

        if(lpDropFileStruct->fWide == TRUE) {
            LPWSTR lpszFileW = NULL;

            if(lpszFile) {
                lpszFileW = (LPWSTR) HeapAlloc(GetProcessHeap(), 0, lLength*sizeof(WCHAR));
                if(lpszFileW == NULL) {
                    goto end;
                }
            }
            i = DragQueryFileW(hDrop, lFile, lpszFileW, lLength);

            if(lpszFileW) {
                WideCharToMultiByte(CP_ACP, 0, lpszFileW, -1, lpszFile, lLength, 0, NULL);
                HeapFree(GetProcessHeap(), 0, lpszFileW);
            }
            goto end;
        }

        while (i++ < lFile)
        {
          while (*lpDrop++); /* skip filename */
          if (!*lpDrop)
          {
            i = (lFile == 0xFFFFFFFF) ? i : 0;
            goto end;
          }
        }

        i = strlen(lpDrop);
        i++;
        if (!lpszFile ) goto end;   /* needed buffer size */
        i = (lLength > i) ? i : lLength;
        lstrcpynA (lpszFile,  lpDrop,  i);
end:
        GlobalUnlock(hDrop);
        return i;
}

/*************************************************************************
 *  DragQueryFileW              [SHELL32.133]
 */
UINT WINAPI DragQueryFileW(
        HDROP hDrop,
        UINT lFile,
        LPWSTR lpszwFile,
        UINT lLength)
{
        LPWSTR lpwDrop;
        UINT i = 0;
        DROPFILES *lpDropFileStruct = (DROPFILES *) GlobalLock(hDrop);

        TRACE("(%08x, %x, %p, %u)\n", hDrop,lFile,lpszwFile,lLength);

        if(!lpDropFileStruct) goto end;

        lpwDrop = (LPWSTR) ((LPSTR)lpDropFileStruct + lpDropFileStruct->pFiles);

        if(lpDropFileStruct->fWide == FALSE) {
            LPSTR lpszFileA = NULL;
 
            if(lpszwFile) {
#ifdef __WIN32OS2__
                lpszFileA = (LPSTR) HeapAlloc(GetProcessHeap(), 0, lLength * sizeof( WCHAR ));
#else
                lpszFileA = (LPSTR) HeapAlloc(GetProcessHeap(), 0, lLength);
#endif
                if(lpszFileA == NULL) {
                    goto end;
                }
            }
#ifdef __WIN32OS2__
            i = DragQueryFileA(hDrop, lFile, lpszFileA, lLength * sizeof( WCHAR ));
#else
            i = DragQueryFileA(hDrop, lFile, lpszFileA, lLength);
#endif

            if(lpszFileA) {
                MultiByteToWideChar(CP_ACP, 0, lpszFileA, -1, lpszwFile, lLength);
                HeapFree(GetProcessHeap(), 0, lpszFileA);
            }
            goto end;
        }

        i = 0;
        while (i++ < lFile)
        {
          while (*lpwDrop++); /* skip filename */
          if (!*lpwDrop)
          {
            i = (lFile == 0xFFFFFFFF) ? i : 0;
            goto end;
          }
        }

        i = strlenW(lpwDrop);
        i++;
        if ( !lpszwFile) goto end;   /* needed buffer size */

        i = (lLength > i) ? i : lLength;
        lstrcpynW (lpszwFile, lpwDrop, i);
end:
        GlobalUnlock(hDrop);
        return i;
}