| /* |
| * IEnumIDList |
| * |
| * Copyright 1998 Juergen Schmied <juergen.schmied@metronet.de> |
| */ |
| |
| #include <stdlib.h> |
| #include <string.h> |
| #include "debug.h" |
| #include "wine/obj_base.h" |
| #include "wine/obj_enumidlist.h" |
| #include "winerror.h" |
| |
| #include "pidl.h" |
| #include "shlguid.h" |
| #include "shell32_main.h" |
| |
| DEFAULT_DEBUG_CHANNEL(shell) |
| |
| typedef struct tagENUMLIST |
| { |
| struct tagENUMLIST *pNext; |
| LPITEMIDLIST pidl; |
| |
| } ENUMLIST, *LPENUMLIST; |
| |
| typedef struct |
| { |
| ICOM_VTABLE(IEnumIDList)* lpvtbl; |
| DWORD ref; |
| LPENUMLIST mpFirst; |
| LPENUMLIST mpLast; |
| LPENUMLIST mpCurrent; |
| |
| } IEnumIDListImpl; |
| |
| static struct ICOM_VTABLE(IEnumIDList) eidlvt; |
| |
| /************************************************************************** |
| * IEnumIDList_fnConstructor |
| */ |
| |
| IEnumIDList * IEnumIDList_Constructor( |
| LPCSTR lpszPath, |
| DWORD dwFlags) |
| { IEnumIDListImpl* lpeidl; |
| |
| lpeidl = (IEnumIDListImpl*)HeapAlloc(GetProcessHeap(),0,sizeof(IEnumIDListImpl)); |
| if (! lpeidl) |
| return NULL; |
| |
| lpeidl->ref = 1; |
| lpeidl->lpvtbl = &eidlvt; |
| lpeidl->mpFirst=NULL; |
| lpeidl->mpLast=NULL; |
| lpeidl->mpCurrent=NULL; |
| |
| TRACE(shell,"(%p)->(%s flags=0x%08lx)\n",lpeidl,debugstr_a(lpszPath),dwFlags); |
| |
| if(!IEnumIDList_CreateEnumList((IEnumIDList*)lpeidl, lpszPath, dwFlags)) |
| { if (lpeidl) |
| { HeapFree(GetProcessHeap(),0,lpeidl); |
| } |
| return NULL; |
| } |
| |
| TRACE(shell,"-- (%p)->()\n",lpeidl); |
| shell32_ObjCount++; |
| return (IEnumIDList*)lpeidl; |
| } |
| |
| /************************************************************************** |
| * EnumIDList_QueryInterface |
| */ |
| static HRESULT WINAPI IEnumIDList_fnQueryInterface( |
| IEnumIDList * iface, |
| REFIID riid, |
| LPVOID *ppvObj) |
| { |
| ICOM_THIS(IEnumIDListImpl,iface); |
| |
| char xriid[50]; |
| WINE_StringFromCLSID((LPCLSID)riid,xriid); |
| TRACE(shell,"(%p)->(\n\tIID:\t%s,%p)\n",This,xriid,ppvObj); |
| |
| *ppvObj = NULL; |
| |
| if(IsEqualIID(riid, &IID_IUnknown)) /*IUnknown*/ |
| { *ppvObj = This; |
| } |
| else if(IsEqualIID(riid, &IID_IEnumIDList)) /*IEnumIDList*/ |
| { *ppvObj = (IEnumIDList*)This; |
| } |
| |
| if(*ppvObj) |
| { IEnumIDList_AddRef((IEnumIDList*)*ppvObj); |
| TRACE(shell,"-- Interface: (%p)->(%p)\n",ppvObj,*ppvObj); |
| return S_OK; |
| } |
| |
| TRACE(shell,"-- Interface: E_NOINTERFACE\n"); |
| return E_NOINTERFACE; |
| } |
| |
| /****************************************************************************** |
| * IEnumIDList_fnAddRef |
| */ |
| static ULONG WINAPI IEnumIDList_fnAddRef( |
| IEnumIDList * iface) |
| { |
| ICOM_THIS(IEnumIDListImpl,iface); |
| |
| TRACE(shell,"(%p)->(%lu)\n",This,This->ref); |
| |
| shell32_ObjCount++; |
| return ++(This->ref); |
| } |
| /****************************************************************************** |
| * IEnumIDList_fnRelease |
| */ |
| static ULONG WINAPI IEnumIDList_fnRelease( |
| IEnumIDList * iface) |
| { |
| ICOM_THIS(IEnumIDListImpl,iface); |
| |
| TRACE(shell,"(%p)->(%lu)\n",This,This->ref); |
| |
| shell32_ObjCount--; |
| |
| if (!--(This->ref)) |
| { TRACE(shell," destroying IEnumIDList(%p)\n",This); |
| IEnumIDList_DeleteList((IEnumIDList*)This); |
| HeapFree(GetProcessHeap(),0,This); |
| return 0; |
| } |
| return This->ref; |
| } |
| |
| /************************************************************************** |
| * IEnumIDList_fnNext |
| */ |
| |
| static HRESULT WINAPI IEnumIDList_fnNext( |
| IEnumIDList * iface, |
| ULONG celt, |
| LPITEMIDLIST * rgelt, |
| ULONG *pceltFetched) |
| { |
| ICOM_THIS(IEnumIDListImpl,iface); |
| |
| ULONG i; |
| HRESULT hr = S_OK; |
| LPITEMIDLIST temp; |
| |
| TRACE(shell,"(%p)->(%ld,%p, %p)\n",This,celt,rgelt,pceltFetched); |
| |
| /* It is valid to leave pceltFetched NULL when celt is 1. Some of explorer's |
| * subsystems actually use it (and so may a third party browser) |
| */ |
| if(pceltFetched) |
| *pceltFetched = 0; |
| |
| *rgelt=0; |
| |
| if(celt > 1 && !pceltFetched) |
| { return E_INVALIDARG; |
| } |
| |
| for(i = 0; i < celt; i++) |
| { if(!(This->mpCurrent)) |
| { hr = S_FALSE; |
| break; |
| } |
| temp = ILClone(This->mpCurrent->pidl); |
| rgelt[i] = temp; |
| This->mpCurrent = This->mpCurrent->pNext; |
| } |
| if(pceltFetched) |
| { *pceltFetched = i; |
| } |
| |
| return hr; |
| } |
| |
| /************************************************************************** |
| * IEnumIDList_fnSkip |
| */ |
| static HRESULT WINAPI IEnumIDList_fnSkip( |
| IEnumIDList * iface,ULONG celt) |
| { |
| ICOM_THIS(IEnumIDListImpl,iface); |
| |
| DWORD dwIndex; |
| HRESULT hr = S_OK; |
| |
| TRACE(shell,"(%p)->(%lu)\n",This,celt); |
| |
| for(dwIndex = 0; dwIndex < celt; dwIndex++) |
| { if(!This->mpCurrent) |
| { hr = S_FALSE; |
| break; |
| } |
| This->mpCurrent = This->mpCurrent->pNext; |
| } |
| return hr; |
| } |
| /************************************************************************** |
| * IEnumIDList_fnReset |
| */ |
| static HRESULT WINAPI IEnumIDList_fnReset( |
| IEnumIDList * iface) |
| { |
| ICOM_THIS(IEnumIDListImpl,iface); |
| |
| TRACE(shell,"(%p)\n",This); |
| This->mpCurrent = This->mpFirst; |
| return S_OK; |
| } |
| /************************************************************************** |
| * IEnumIDList_fnClone |
| */ |
| static HRESULT WINAPI IEnumIDList_fnClone( |
| IEnumIDList * iface,LPENUMIDLIST * ppenum) |
| { |
| ICOM_THIS(IEnumIDListImpl,iface); |
| |
| TRACE(shell,"(%p)->() to (%p)->() E_NOTIMPL\n",This,ppenum); |
| return E_NOTIMPL; |
| } |
| /************************************************************************** |
| * EnumIDList_CreateEnumList() |
| * fixme: devices not handled |
| * fixme: add wildcards to path |
| */ |
| static BOOL WINAPI IEnumIDList_fnCreateEnumList( |
| IEnumIDList * iface, |
| LPCSTR lpszPath, |
| DWORD dwFlags) |
| { |
| ICOM_THIS(IEnumIDListImpl,iface); |
| |
| LPITEMIDLIST pidl=NULL; |
| LPPIDLDATA pData=NULL; |
| WIN32_FIND_DATAA stffile; |
| HANDLE hFile; |
| DWORD dwDrivemap; |
| CHAR szDriveName[4]; |
| CHAR szPath[MAX_PATH]; |
| |
| TRACE(shell,"(%p)->(path=%s flags=0x%08lx) \n",This,debugstr_a(lpszPath),dwFlags); |
| |
| if (lpszPath && lpszPath[0]!='\0') |
| { strcpy(szPath, lpszPath); |
| PathAddBackslashA(szPath); |
| strcat(szPath,"*.*"); |
| } |
| |
| /*enumerate the folders*/ |
| if(dwFlags & SHCONTF_FOLDERS) |
| { /* special case - we can't enumerate the Desktop level Objects (MyComputer,Nethood... |
| so we need to fake an enumeration of those.*/ |
| if(!lpszPath) |
| { TRACE (shell,"-- (%p)-> enumerate SHCONTF_FOLDERS (special) items\n",This); |
| /*create the pidl for This item */ |
| pidl = _ILCreateMyComputer(); |
| if(pidl) |
| { if(!IEnumIDList_AddToEnumList((IEnumIDList*)This, pidl)) |
| return FALSE; |
| } |
| } |
| else if (lpszPath[0]=='\0') /* enumerate the drives*/ |
| { TRACE (shell,"-- (%p)-> enumerate SHCONTF_FOLDERS (drives)\n",This); |
| dwDrivemap = GetLogicalDrives(); |
| strcpy (szDriveName,"A:\\"); |
| while (szDriveName[0]<='Z') |
| { if(dwDrivemap & 0x00000001L) |
| { pidl = _ILCreateDrive(szDriveName); |
| if(pidl) |
| { if(!IEnumIDList_AddToEnumList((IEnumIDList*)This, pidl)) |
| return FALSE; |
| } |
| } |
| szDriveName[0]++; |
| dwDrivemap = dwDrivemap >> 1; |
| } |
| } |
| else |
| { TRACE (shell,"-- (%p)-> enumerate SHCONTF_FOLDERS of %s\n",This,debugstr_a(szPath)); |
| hFile = FindFirstFileA(szPath,&stffile); |
| if ( hFile != INVALID_HANDLE_VALUE ) |
| { do |
| { if ( (stffile.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) && strcmp (stffile.cFileName, ".") && strcmp (stffile.cFileName, "..")) |
| { pidl = _ILCreateFolder( stffile.cAlternateFileName, stffile.cFileName); |
| if(pidl) |
| { pData = _ILGetDataPointer(pidl); |
| FileTimeToDosDateTime(&stffile.ftLastWriteTime,&pData->u.folder.uFileDate,&pData->u.folder.uFileTime); |
| pData->u.folder.dwFileSize = stffile.nFileSizeLow; |
| pData->u.folder.uFileAttribs=stffile.dwFileAttributes; |
| if(!IEnumIDList_AddToEnumList((IEnumIDList*)This, pidl)) |
| { return FALSE; |
| } |
| } |
| else |
| { return FALSE; |
| } |
| } |
| } while( FindNextFileA(hFile,&stffile)); |
| FindClose (hFile); |
| } |
| } |
| } |
| /*enumerate the non-folder items (values) */ |
| if(dwFlags & SHCONTF_NONFOLDERS) |
| { if(lpszPath) |
| { TRACE (shell,"-- (%p)-> enumerate SHCONTF_NONFOLDERS of %s\n",This,debugstr_a(szPath)); |
| hFile = FindFirstFileA(szPath,&stffile); |
| if ( hFile != INVALID_HANDLE_VALUE ) |
| { do |
| { if (! (stffile.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) ) |
| { pidl = _ILCreateValue( stffile.cAlternateFileName, stffile.cFileName); |
| if(pidl) |
| { pData = _ILGetDataPointer(pidl); |
| FileTimeToDosDateTime(&stffile.ftLastWriteTime,&pData->u.file.uFileDate,&pData->u.file.uFileTime); |
| pData->u.file.dwFileSize = stffile.nFileSizeLow; |
| pData->u.file.uFileAttribs=stffile.dwFileAttributes; |
| if(!IEnumIDList_AddToEnumList((IEnumIDList*)This, pidl)) |
| { return FALSE; |
| } |
| } |
| else |
| { return FALSE; |
| } |
| } |
| } while( FindNextFileA(hFile,&stffile)); |
| FindClose (hFile); |
| } |
| } |
| } |
| return TRUE; |
| } |
| |
| /************************************************************************** |
| * EnumIDList_AddToEnumList() |
| */ |
| static BOOL WINAPI IEnumIDList_fnAddToEnumList( |
| IEnumIDList * iface, |
| LPITEMIDLIST pidl) |
| { |
| ICOM_THIS(IEnumIDListImpl,iface); |
| |
| LPENUMLIST pNew; |
| |
| TRACE(shell,"(%p)->(pidl=%p)\n",This,pidl); |
| pNew = (LPENUMLIST)SHAlloc(sizeof(ENUMLIST)); |
| if(pNew) |
| { /*set the next pointer */ |
| pNew->pNext = NULL; |
| pNew->pidl = pidl; |
| |
| /*is This the first item in the list? */ |
| if(!This->mpFirst) |
| { This->mpFirst = pNew; |
| This->mpCurrent = pNew; |
| } |
| |
| if(This->mpLast) |
| { /*add the new item to the end of the list */ |
| This->mpLast->pNext = pNew; |
| } |
| |
| /*update the last item pointer */ |
| This->mpLast = pNew; |
| TRACE(shell,"-- (%p)->(first=%p, last=%p)\n",This,This->mpFirst,This->mpLast); |
| return TRUE; |
| } |
| return FALSE; |
| } |
| /************************************************************************** |
| * EnumIDList_DeleteList() |
| */ |
| static BOOL WINAPI IEnumIDList_fnDeleteList( |
| IEnumIDList * iface) |
| { |
| ICOM_THIS(IEnumIDListImpl,iface); |
| |
| LPENUMLIST pDelete; |
| |
| TRACE(shell,"(%p)->()\n",This); |
| |
| while(This->mpFirst) |
| { pDelete = This->mpFirst; |
| This->mpFirst = pDelete->pNext; |
| SHFree(pDelete->pidl); |
| SHFree(pDelete); |
| } |
| This->mpFirst = This->mpLast = This->mpCurrent = NULL; |
| return TRUE; |
| } |
| |
| /************************************************************************** |
| * IEnumIDList_fnVTable |
| */ |
| static ICOM_VTABLE (IEnumIDList) eidlvt = |
| { IEnumIDList_fnQueryInterface, |
| IEnumIDList_fnAddRef, |
| IEnumIDList_fnRelease, |
| IEnumIDList_fnNext, |
| IEnumIDList_fnSkip, |
| IEnumIDList_fnReset, |
| IEnumIDList_fnClone, |
| IEnumIDList_fnCreateEnumList, |
| IEnumIDList_fnAddToEnumList, |
| IEnumIDList_fnDeleteList |
| }; |