Source code of Windows XP (NT5)
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 

4873 lines
153 KiB

#include "shellprv.h"
#include "caggunk.h"
#include "views.h"
#include "ids.h"
#include "shitemid.h"
#include "fstreex.h"
#include "clsobj.h"
#include "datautil.h"
#include "winnetp.h" // RESOURCE_SHAREABLE
#include "prop.h"
#include "infotip.h"
#include "basefvcb.h"
#include "netview.h"
#include "printer.h"
#include "fsdata.h"
#include "idldrop.h"
#include "enumidlist.h"
#include "util.h"
#include <webvw.h>
#define WNNC_NET_LARGEST WNNC_NET_SYMFONET
HRESULT CNetRootDropTarget_CreateInstance(HWND hwnd, LPCITEMIDLIST pidl, IDropTarget **ppdropt);
class CNetData : public CFSIDLData
{
public:
CNetData(LPCITEMIDLIST pidlFolder, UINT cidl, LPCITEMIDLIST apidl[]): CFSIDLData(pidlFolder, cidl, apidl, NULL) { };
// IDataObject methods overwrite
STDMETHODIMP GetData(FORMATETC *pFmtEtc, STGMEDIUM *pstm);
STDMETHODIMP QueryGetData(FORMATETC *pFmtEtc);
protected:
STDMETHODIMP GetHDrop(FORMATETC *pformatetcIn, STGMEDIUM *pmedium);
};
// {22BEB58B-0794-11d2-A4AA-00C04F8EEB3E}
const GUID CLSID_CNetFldr = { 0x22beb58b, 0x794, 0x11d2, 0xa4, 0xaa, 0x0, 0xc0, 0x4f, 0x8e, 0xeb, 0x3e };
// idlist.c
STDAPI_(void) StrRetFormat(STRRET *pStrRet, LPCITEMIDLIST pidlRel, LPCTSTR pszTemplate, LPCTSTR pszAppend);
// in stdenum.cpp
STDAPI_(void *) CStandardEnum_CreateInstance(REFIID riid, BOOL bInterfaces, int cElement, int cbElement, void *rgElements,
void (WINAPI * pfnCopyElement)(void *, const void *, DWORD));
// is a \\server\printer object
BOOL _IsPrintShare(LPCIDNETRESOURCE pidn)
{
return NET_GetDisplayType(pidn) == RESOURCEDISPLAYTYPE_SHARE &&
NET_GetType(pidn) == RESOURCETYPE_PRINT;
}
// column information
enum
{
ICOL_NAME = 0,
ICOL_COMMENT,
ICOL_COMPUTERNAME,
ICOL_NETWORKLOCATION
};
const COLUMN_INFO s_net_cols[] =
{
DEFINE_COL_STR_ENTRY(SCID_NAME, 30, IDS_NAME_COL),
DEFINE_COL_STR_ENTRY(SCID_Comment, 30, IDS_EXCOL_COMMENT),
DEFINE_COL_STR_ENTRY(SCID_COMPUTERNAME, 30, IDS_EXCOL_COMPUTER),
DEFINE_COL_STR_ENTRY(SCID_NETWORKLOCATION, 30, IDS_NETWORKLOCATION),
};
#define MAX_ICOL_NETFOLDER (ICOL_COMMENT+1)
#define MAX_ICOL_NETROOT (ICOL_NETWORKLOCATION+1)
STDAPI CNetwork_DFMCallBackBG(IShellFolder *psf, HWND hwnd, IDataObject *pdtobj, UINT uMsg, WPARAM wParam, LPARAM lParam);
class CNetFolderViewCB;
class CNetFolderEnum;
class CNetFolder : public CAggregatedUnknown,
public IShellFolder2,
public IPersistFolder3,
public IShellIconOverlay
{
friend CNetFolderViewCB;
friend CNetFolderEnum;
public:
// IUnknown
STDMETHODIMP QueryInterface(REFIID riid, void ** ppvObj)
{ return CAggregatedUnknown::QueryInterface(riid, ppvObj); };
STDMETHODIMP_(ULONG) AddRef(void)
{ return CAggregatedUnknown::AddRef(); };
STDMETHODIMP_(ULONG) Release(void)
{ return CAggregatedUnknown::Release(); };
// IShellFolder
STDMETHODIMP ParseDisplayName(HWND hwnd, LPBC pbc, LPOLESTR lpszDisplayName,
ULONG* pchEaten, LPITEMIDLIST* ppidl, ULONG* pdwAttributes);
STDMETHODIMP EnumObjects(HWND hwnd, DWORD grfFlags, IEnumIDList ** ppenumIDList);
STDMETHODIMP BindToObject(LPCITEMIDLIST pidl, LPBC pbc, REFIID riid, void **ppv);
STDMETHODIMP BindToStorage(LPCITEMIDLIST pidl, LPBC pbc, REFIID riid, void **ppvObj);
STDMETHODIMP CompareIDs(LPARAM lParam, LPCITEMIDLIST pidl1, LPCITEMIDLIST pidl2);
STDMETHODIMP CreateViewObject(HWND hwndOwner, REFIID riid, void **ppv);
STDMETHODIMP GetAttributesOf(UINT cidl, LPCITEMIDLIST* apidl, ULONG* rgfInOut);
STDMETHODIMP GetUIObjectOf(HWND hwndOwner, UINT cidl, LPCITEMIDLIST* apidl,
REFIID riid, UINT* prgfInOut, void **ppv);
STDMETHODIMP GetDisplayNameOf(LPCITEMIDLIST pidl, DWORD uFlags, LPSTRRET lpName);
STDMETHODIMP SetNameOf(HWND hwnd, LPCITEMIDLIST pidl, LPCOLESTR lpszName, DWORD uFlags,
LPITEMIDLIST* ppidlOut);
// IShellFolder2
STDMETHODIMP GetDefaultSearchGUID(GUID *pGuid);
STDMETHODIMP EnumSearches(IEnumExtraSearch **ppenum);
STDMETHODIMP GetDefaultColumn(DWORD dwRes, ULONG* pSort, ULONG* pDisplay);
STDMETHODIMP GetDefaultColumnState(UINT iColumn, DWORD* pbState)
{ return _GetDefaultColumnState(MAX_ICOL_NETFOLDER, iColumn, pbState); }
STDMETHODIMP GetDetailsEx(LPCITEMIDLIST pidl, const SHCOLUMNID* pscid, VARIANT* pv);
STDMETHODIMP GetDetailsOf(LPCITEMIDLIST pidl, UINT iColumn, SHELLDETAILS* pDetails)
{ return _GetDetailsOf(MAX_ICOL_NETFOLDER, pidl, iColumn, pDetails); }
STDMETHODIMP MapColumnToSCID(UINT iColumn, SHCOLUMNID* pscid)
{ return _MapColumnToSCID(MAX_ICOL_NETFOLDER, iColumn, pscid); }
// IPersist
STDMETHODIMP GetClassID(CLSID* pClassID);
// IPersistFolder
STDMETHODIMP Initialize(LPCITEMIDLIST pidl);
// IPersistFolder2
STDMETHODIMP GetCurFolder(LPITEMIDLIST* ppidl);
// IPersistFolder3
STDMETHOD(InitializeEx)(IBindCtx *pbc, LPCITEMIDLIST pidlRoot, const PERSIST_FOLDER_TARGET_INFO *ppfai);
STDMETHOD(GetFolderTargetInfo)(PERSIST_FOLDER_TARGET_INFO *ppfai);
// *** IShellIconOverlay methods***
STDMETHOD(GetOverlayIndex)(LPCITEMIDLIST pidl, int * pIndex);
STDMETHOD(GetOverlayIconIndex)(LPCITEMIDLIST pidl, int * pIconIndex);
protected:
CNetFolder(IUnknown* punkOuter);
~CNetFolder();
virtual HRESULT v_GetFileFolder(IShellFolder2 **ppsfFiles)
{ *ppsfFiles = NULL; return E_NOTIMPL; };
// used by the CAggregatedUnknown stuff
HRESULT v_InternalQueryInterface(REFIID riid, void **ppvObj);
HRESULT _OpenKeys(LPCIDNETRESOURCE pidn, HKEY ahkeys[]);
LPCTSTR _GetProvider(LPCIDNETRESOURCE pidn, IBindCtx *pbc, LPTSTR pszProvider, UINT cchProvider);
DWORD _OpenEnum(HWND hwnd, DWORD grfFlags, LPNETRESOURCE pnr, HANDLE *phEnum);
static HRESULT _CreateNetIDList(LPIDNETRESOURCE pidnIn,
LPCTSTR pszName, LPCTSTR pszProvider, LPCTSTR pszComment,
LPITEMIDLIST *ppidl);
static HRESULT _NetResToIDList(NETRESOURCE *pnr,
BOOL fKeepNullRemoteName,
BOOL fKeepProviderName,
BOOL fKeepComment,
LPITEMIDLIST *ppidl);
static HRESULT _CreateEntireNetwork(LPITEMIDLIST *ppidl);
static HRESULT _CreateEntireNetworkFullIDList(LPITEMIDLIST *ppidl);
LPTSTR _GetNameForParsing(LPCWSTR pwszName, LPTSTR pszBuffer, INT cchBuffer, LPTSTR *ppszRegItem);
HRESULT _ParseRest(LPBC pbc, LPCWSTR pszRest, LPITEMIDLIST* ppidl, DWORD* pdwAttributes);
HRESULT _AddUnknownIDList(DWORD dwDisplayType, LPITEMIDLIST *ppidl);
HRESULT _ParseSimple(LPBC pbc, LPWSTR pszName, LPITEMIDLIST* ppidl, DWORD* pdwAttributes);
HRESULT _NetResToIDLists(NETRESOURCE *pnr, DWORD dwbuf, LPITEMIDLIST *ppidl);
HRESULT _ParseNetName(HWND hwnd, LPBC pbc, LPCWSTR pwszName, ULONG* pchEaten,
LPITEMIDLIST* ppidl, DWORD* pdwAttributes);
LONG _GetFilePIDLType(LPCITEMIDLIST pidl);
LPITEMIDLIST _AddProviderToPidl(LPITEMIDLIST pidl, LPCTSTR lpProvider);
BOOL _MakeStripToLikeKinds(UINT *pcidl, LPCITEMIDLIST **papidl, BOOL fNetObjects);
HRESULT _GetDefaultColumnState(UINT cColumns, UINT iColumn, DWORD* pdwState);
HRESULT _GetDetailsOf(UINT cColumns, LPCITEMIDLIST pidl, UINT iColumn, SHELLDETAILS *pDetails);
HRESULT _MapColumnToSCID(UINT cColumns, UINT iColumn, SHCOLUMNID* pscid);
LPFNDFMCALLBACK _GetCallbackType(LPCIDNETRESOURCE pidn)
{ return _IsPrintShare(pidn) ? &PrinterDFMCallBack : &DFMCallBack; };
static HRESULT CALLBACK _AttributesCallbackRoot(IShellFolder2* psf, LPCITEMIDLIST pidl, ULONG* prgfInOut);
LPITEMIDLIST _pidl;
LPITEMIDLIST _pidlTarget; // pidl of where the folder is in the namespace
LPCIDNETRESOURCE _pidnForProvider; // optional provider for this container...
LPTSTR _pszResName; // optional resource name of this container
UINT _uDisplayType; // display type of the folder
IShellFolder2* _psfFiles;
IUnknown* _punkReg;
private:
HRESULT _CreateInstance(LPCITEMIDLIST pidlAbs, LPCITEMIDLIST pidlTarget,
UINT uDisplayType,
LPCIDNETRESOURCE pidnForProvider, LPCTSTR pszResName,
REFIID riid, void **ppv);
friend HRESULT CNetwork_DFMCallBackBG(IShellFolder *psf, HWND hwnd,
IDataObject *pdtobj, UINT uMsg,
WPARAM wParam, LPARAM lParam);
static DWORD CALLBACK _PropertiesThreadProc(void *pv);
static HRESULT DFMCallBack(IShellFolder* psf, HWND hwnd,
IDataObject* pdtobj, UINT uMsg,
WPARAM wParam, LPARAM lParam);
static HRESULT PrinterDFMCallBack(IShellFolder* psf, HWND hwnd,
IDataObject* pdtobj, UINT uMsg,
WPARAM wParam, LPARAM lParam);
static HRESULT CALLBACK _AttributesCallback(IShellFolder2* psf, LPCITEMIDLIST pidl, ULONG* prgfInOut);
BOOL _GetPathForShare(LPCIDNETRESOURCE pidn, LPTSTR pszPath);
HRESULT _GetPathForItem(LPCIDNETRESOURCE pidn, LPTSTR pszPath);
HRESULT _GetPathForItemW(LPCIDNETRESOURCE pidn, LPWSTR pszPath);
HRESULT _CreateFolderForItem(LPBC pbc, LPCITEMIDLIST pidl, LPCITEMIDLIST pidlTarget, LPCIDNETRESOURCE pidnForProvider, REFIID riid, void **ppv);
HRESULT _GetFormatName(LPCIDNETRESOURCE pidn, STRRET* pStrRet);
HRESULT _GetIconOverlayInfo(LPCIDNETRESOURCE pidn, int *pIndex, DWORD dwFlags);
HKEY _OpenProviderTypeKey(LPCIDNETRESOURCE pidn);
HKEY _OpenProviderKey(LPCIDNETRESOURCE pidn);
static void WINAPI _CopyEnumElement(void* pDest, const void* pSource, DWORD dwSize);
HRESULT _GetNetResource(LPCIDNETRESOURCE pidn, NETRESOURCEW* pnr, int cb);
};
class CNetRootFolder : public CNetFolder
{
public:
// IUnknown
STDMETHODIMP QueryInterface(REFIID riid, void ** ppvObj)
{ return CNetFolder::QueryInterface(riid, ppvObj); };
STDMETHODIMP_(ULONG) AddRef(void)
{ return CNetFolder::AddRef(); };
STDMETHODIMP_(ULONG) Release(void)
{ return CNetFolder::Release(); };
// IShellFolder
STDMETHODIMP ParseDisplayName(HWND hwnd, LPBC pbc, LPOLESTR lpszDisplayName,
ULONG* pchEaten, LPITEMIDLIST* ppidl, ULONG* pdwAttributes);
STDMETHODIMP EnumObjects(HWND hwnd, DWORD grfFlags, IEnumIDList ** ppenumIDList);
STDMETHODIMP BindToObject(LPCITEMIDLIST pidl, LPBC pbc, REFIID riid, void **ppv);
STDMETHODIMP BindToStorage(LPCITEMIDLIST pidl, LPBC pbc, REFIID riid, void **ppvObj)
{ return CNetFolder::BindToStorage(pidl, pbc, riid, ppvObj); };
STDMETHODIMP CompareIDs(LPARAM lParam, LPCITEMIDLIST pidl1, LPCITEMIDLIST pidl2);
STDMETHODIMP CreateViewObject(HWND hwndOwner, REFIID riid, void **ppv);
STDMETHODIMP GetAttributesOf(UINT cidl, LPCITEMIDLIST* apidl, ULONG* rgfInOut);
STDMETHODIMP GetUIObjectOf(HWND hwndOwner, UINT cidl, LPCITEMIDLIST* apidl,
REFIID riid, UINT* prgfInOut, void **ppv);
STDMETHODIMP GetDisplayNameOf(LPCITEMIDLIST pidl, DWORD uFlags, LPSTRRET lpName);
STDMETHODIMP SetNameOf(HWND hwnd, LPCITEMIDLIST pidl, LPCOLESTR lpszName, DWORD uFlags,
LPITEMIDLIST* ppidlOut);
// IShellFolder2
STDMETHODIMP GetDefaultSearchGUID(GUID *pGuid)
{ return CNetFolder::GetDefaultSearchGUID(pGuid); };
STDMETHODIMP EnumSearches(IEnumExtraSearch **ppenum)
{ return CNetFolder::EnumSearches(ppenum); };
STDMETHODIMP GetDefaultColumn(DWORD dwRes, ULONG* pSort, ULONG* pDisplay)
{ return CNetFolder::GetDefaultColumn(dwRes, pSort, pDisplay); };
STDMETHODIMP GetDefaultColumnState(UINT iColumn, DWORD* pbState)
{ return _GetDefaultColumnState(MAX_ICOL_NETROOT, iColumn, pbState); } // +1 for <= check
STDMETHODIMP GetDetailsEx(LPCITEMIDLIST pidl, const SHCOLUMNID* pscid, VARIANT* pv)
{ return CNetFolder::GetDetailsEx(pidl, pscid, pv); };
STDMETHODIMP GetDetailsOf(LPCITEMIDLIST pidl, UINT iColumn, SHELLDETAILS* pDetails)
{ return _GetDetailsOf(MAX_ICOL_NETROOT, pidl, iColumn, pDetails); };
STDMETHODIMP MapColumnToSCID(UINT iColumn, SHCOLUMNID* pscid)
{ return _MapColumnToSCID(MAX_ICOL_NETROOT, iColumn, pscid); }
// IPersist
STDMETHODIMP GetClassID(CLSID* pClassID);
// IPersistFolder
STDMETHODIMP Initialize(LPCITEMIDLIST pidl);
// IPersistFolder2
STDMETHODIMP GetCurFolder(LPITEMIDLIST* ppidl) { return CNetFolder::GetCurFolder(ppidl); };
// IPersistFolder3
STDMETHOD(InitializeEx)(IBindCtx *pbc, LPCITEMIDLIST pidlRoot, const PERSIST_FOLDER_TARGET_INFO *ppfai)
{ return CNetFolder::InitializeEx(pbc, pidlRoot, ppfai); };
STDMETHOD(GetFolderTargetInfo)(PERSIST_FOLDER_TARGET_INFO *ppfai)
{ return CNetFolder::GetFolderTargetInfo(ppfai); };
protected:
CNetRootFolder(IUnknown* punkOuter) : CNetFolder(punkOuter) { };
~CNetRootFolder() { ASSERT(NULL != _spThis); _spThis = NULL; };
BOOL v_HandleDelete(PLONG pcRef);
HRESULT v_GetFileFolder(IShellFolder2 **ppsfFiles);
private:
HRESULT _TryParseEntireNet(HWND hwnd, LPBC pbc, WCHAR *pwszName, LPITEMIDLIST *ppidl, DWORD *pdwAttributes);
friend HRESULT CNetwork_CreateInstance(IUnknown* punkOuter, REFIID riid, void **ppv);
static CNetRootFolder* _spThis;
};
class CNetFolderViewCB : public CBaseShellFolderViewCB
{
public:
CNetFolderViewCB(CNetFolder *pFolder);
// IShellFolderViewCB
STDMETHODIMP RealMessage(UINT uMsg, WPARAM wParam, LPARAM lParam);
private:
~CNetFolderViewCB();
HRESULT OnINVOKECOMMAND(DWORD pv, UINT wP);
HRESULT OnGETHELPTEXT(DWORD pv, UINT wPl, UINT wPh, LPTSTR lP);
HRESULT OnREFRESH(DWORD pv, BOOL fPreRefresh);
HRESULT OnDELAYWINDOWCREATE(DWORD pv, HWND hwnd);
HRESULT OnGETCOLSAVESTREAM(DWORD pv, WPARAM wP, IStream **pps);
HRESULT OnDEFITEMCOUNT(DWORD pv, UINT *pnItems);
HRESULT OnGetZone(DWORD pv, DWORD * pdwZone);
HRESULT OnEnumeratedItems(DWORD pv, UINT celt, LPCITEMIDLIST *rgpidl);
HRESULT OnDefViewMode(DWORD pv, FOLDERVIEWMODE* pvm);
HRESULT OnGetDeferredViewSettings(DWORD pv, SFVM_DEFERRED_VIEW_SETTINGS* pSettings);
BOOL _GetProviderKeyName(LPTSTR pszName, UINT uNameLen);
BOOL _EntireNetworkAvailable();
CNetFolder *_pFolder;
UINT _cItems;
// Web View implementation
HRESULT OnGetWebViewLayout(DWORD pv, UINT uViewMode, SFVM_WEBVIEW_LAYOUT_DATA* pData);
HRESULT OnGetWebViewContent(DWORD pv, SFVM_WEBVIEW_CONTENT_DATA* pData);
HRESULT OnGetWebViewTasks(DWORD pv, SFVM_WEBVIEW_TASKSECTION_DATA* pTasks);
public:
static HRESULT _CanShowHNW(IUnknown* pv, IShellItemArray *psiItemArray, BOOL fOkToBeSlow, UISTATE* puisState);
static HRESULT _CanViewComputersNearMe(IUnknown* pv, IShellItemArray *psiItemArray, BOOL fOkToBeSlow, UISTATE* puisState);
static HRESULT _CanSearchActiveDirectory(IUnknown* pv, IShellItemArray *psiItemArray, BOOL fOkToBeSlow, UISTATE* puisState);
static HRESULT _DoRunDll32(LPTSTR pszParameters); // helper to do a ShellExecute of RunDll32.
static HRESULT _OnViewNetConnections(IUnknown* pv, IShellItemArray *psiItemArray, IBindCtx *pbc);
static HRESULT _OnAddNetworkPlace(IUnknown* pv, IShellItemArray *psiItemArray, IBindCtx *pbc)
{ return _DoRunDll32(TEXT("netplwiz.dll,AddNetPlaceRunDll")); }
static HRESULT _OnHomeNetworkWizard(IUnknown* pv, IShellItemArray *psiItemArray, IBindCtx *pbc)
{ return _DoRunDll32(TEXT("hnetwiz.dll,HomeNetWizardRunDll")); }
static HRESULT _OnViewComputersNearMe(IUnknown* pv, IShellItemArray *psiItemArray, IBindCtx *pbc);
static HRESULT _OnSearchActiveDirectory(IUnknown* pv, IShellItemArray *psiItemArray, IBindCtx *pbc)
{ return _DoRunDll32(TEXT("dsquery.dll,OpenQueryWindow")); }
};
#define NETFLDR_EVENTS \
SHCNE_RENAMEITEM | SHCNE_RENAMEFOLDER | \
SHCNE_CREATE | SHCNE_DELETE | SHCNE_UPDATEDIR | SHCNE_UPDATEITEM | \
SHCNE_MKDIR | SHCNE_RMDIR
CNetFolderViewCB::CNetFolderViewCB(CNetFolder *pFolder) :
CBaseShellFolderViewCB(pFolder->_pidl, NETFLDR_EVENTS), _pFolder(pFolder)
{
_pFolder->AddRef();
}
CNetFolderViewCB::~CNetFolderViewCB()
{
_pFolder->Release();
}
HRESULT CNetFolderViewCB::OnINVOKECOMMAND(DWORD pv, UINT wP)
{
return CNetwork_DFMCallBackBG(_pFolder, _hwndMain, NULL, DFM_INVOKECOMMAND, wP, 0);
}
HRESULT CNetFolderViewCB::OnGETHELPTEXT(DWORD pv, UINT wPl, UINT wPh, LPTSTR lP)
{
return CNetwork_DFMCallBackBG(_pFolder, _hwndMain, NULL, DFM_GETHELPTEXTW, MAKEWPARAM(wPl, wPh), (LPARAM)lP);
}
HRESULT CNetFolderViewCB::OnREFRESH(DWORD pv, BOOL fPreRefresh)
{
if (fPreRefresh)
{
RefreshNetCrawler();
}
return S_OK;
}
HRESULT CNetFolderViewCB::OnDELAYWINDOWCREATE(DWORD pv, HWND hwnd)
{
// only do delay window processing in the net root.
if (RESOURCEDISPLAYTYPE_GENERIC == _pFolder->_uDisplayType) // MyNetPlaces
{
RefreshNetCrawler();
}
return S_OK;
}
HRESULT CNetFolderViewCB::OnGETCOLSAVESTREAM(DWORD pv, WPARAM wP, IStream **pps)
{
LPCTSTR pszValName;
switch (_pFolder->_uDisplayType)
{
case RESOURCEDISPLAYTYPE_DOMAIN:
pszValName = TEXT("NetDomainColsX");
break;
case RESOURCEDISPLAYTYPE_SERVER:
pszValName = TEXT("NetServerColsX");
break;
default:
return E_FAIL;
}
*pps = OpenRegStream(HKEY_CURRENT_USER, REGSTR_PATH_EXPLORER, pszValName, (DWORD) wP);
return *pps ? S_OK : E_FAIL;
}
// HRESULT CNetFolderViewCB::OnGetZone(DWORD pv, DWORD * pdwZone);
HRESULT CNetFolderViewCB::OnEnumeratedItems(DWORD pv, UINT celt, LPCITEMIDLIST *rgpidl)
{
_cItems = celt;
return S_OK;
}
HRESULT CNetFolderViewCB::OnDefViewMode(DWORD pv, FOLDERVIEWMODE* pvm)
{
if (_cItems < DEFVIEW_FVM_MANY_CUTOFF)
*pvm = FVM_TILE;
else
*pvm = FVM_ICON; // used to pick icon only for My Net Places ((_pFolder->_uDisplayType == RESOURCEDISPLAYTYPE_GENERIC))
return S_OK;
}
HRESULT CNetFolderViewCB::OnGetDeferredViewSettings(DWORD pv, SFVM_DEFERRED_VIEW_SETTINGS* pSettings)
{
OnDefViewMode(pv, &pSettings->fvm);
// if this is the root folder then lets sort accordingly
if (_pFolder->_uDisplayType == RESOURCEDISPLAYTYPE_GENERIC)
{
pSettings->fGroupView = TRUE;
pSettings->uSortCol = ICOL_NETWORKLOCATION;
pSettings->iSortDirection = 1;
}
return S_OK;
}
HRESULT CNetFolderViewCB::OnGetZone(DWORD pv, DWORD * pdwZone)
{
if (pdwZone)
*pdwZone = URLZONE_INTRANET; // default is "Local Intranet"
return S_OK;
}
HRESULT CNetFolderViewCB::OnGetWebViewLayout(DWORD pv, UINT uViewMode, SFVM_WEBVIEW_LAYOUT_DATA* pData)
{
ZeroMemory(pData, sizeof(*pData));
pData->dwLayout = SFVMWVL_NORMAL;
return S_OK;
}
// HNW is shown on X86 pro or personal workgroup only
HRESULT CNetFolderViewCB::_CanShowHNW(IUnknown* pv, IShellItemArray *psiItemArray, BOOL fOkToBeSlow, UISTATE* puisState)
{
#ifdef _WIN64
*puisState = UIS_DISABLED;
return S_OK;
#else
if (IsOS(OS_ANYSERVER))
*puisState = UIS_DISABLED; // Server-type OS
else
*puisState = !IsOS(OS_DOMAINMEMBER) ? UIS_ENABLED : UIS_DISABLED;
return S_OK;
#endif
}
HRESULT CNetFolderViewCB::_CanViewComputersNearMe(IUnknown* pv, IShellItemArray *psiItemArray, BOOL fOkToBeSlow, UISTATE* puisState)
{
if (!SHRestricted(REST_NOCOMPUTERSNEARME))
*puisState = !IsOS(OS_DOMAINMEMBER) ? UIS_ENABLED : UIS_DISABLED;
else
*puisState = UIS_DISABLED;
return S_OK;
}
HRESULT CNetFolderViewCB::_CanSearchActiveDirectory(IUnknown* pv, IShellItemArray *psiItemArray, BOOL fOkToBeSlow, UISTATE* puisState)
{
if (IsOS(OS_DOMAINMEMBER) && (GetEnvironmentVariable(TEXT("USERDNSDOMAIN"), NULL, 0) > 0))
*puisState = UIS_ENABLED;
else
*puisState = UIS_DISABLED;
return S_OK;
}
HRESULT CNetFolderViewCB::_OnViewNetConnections(IUnknown* pv, IShellItemArray *psiItemArray, IBindCtx *pbc)
{
LPITEMIDLIST pidl;
HRESULT hr = SHGetFolderLocation(NULL, CSIDL_CONNECTIONS, NULL, 0, &pidl);
if (SUCCEEDED(hr))
{
hr = ((CNetFolderViewCB*)(void*)pv)->_BrowseObject(pidl);
ILFree(pidl);
}
return hr;
}
HRESULT CNetFolderViewCB::_DoRunDll32(LPTSTR pszParameters)
{
SHELLEXECUTEINFO sei = {0};
sei.cbSize = sizeof(sei);
sei.lpFile = TEXT("rundll32.exe");
sei.lpParameters = pszParameters;
sei.nShow = SW_SHOWNORMAL;
return ShellExecuteEx(&sei) ? S_OK : E_FAIL;
}
HRESULT CNetFolderViewCB::_OnViewComputersNearMe(IUnknown* pv, IShellItemArray *psiItemArray, IBindCtx *pbc)
{
LPITEMIDLIST pidl;
HRESULT hr = SHGetFolderLocation(NULL, CSIDL_COMPUTERSNEARME, NULL, 0, &pidl);
if (SUCCEEDED(hr))
{
hr = ((CNetFolderViewCB*)(void*)pv)->_BrowseObject(pidl);
ILFree(pidl);
}
return hr;
}
const WVTASKITEM c_MyNetPlacesTaskHeader = WVTI_HEADER(L"shell32.dll", IDS_HEADER_MYNETPLACES, IDS_HEADER_MYNETPLACES_TT);
const WVTASKITEM c_MyNetPlacesTaskList[] =
{
WVTI_ENTRY_ALL(CLSID_NULL, L"shell32.dll", IDS_TASK_ADDNETWORKPLACE, IDS_TASK_ADDNETWORKPLACE_TT, IDI_TASK_ADDNETWORKPLACE, NULL, CNetFolderViewCB::_OnAddNetworkPlace),
WVTI_ENTRY_ALL(CLSID_NULL, L"shell32.dll", IDS_TASK_VIEWNETCONNECTIONS, IDS_TASK_VIEWNETCONNECTIONS_TT, IDI_TASK_VIEWNETCONNECTIONS, NULL, CNetFolderViewCB::_OnViewNetConnections),
WVTI_ENTRY_ALL(CLSID_NULL, L"shell32.dll", IDS_TASK_HOMENETWORKWIZARD, IDS_TASK_HOMENETWORKWIZARD_TT, IDI_TASK_HOMENETWORKWIZARD, CNetFolderViewCB::_CanShowHNW, CNetFolderViewCB::_OnHomeNetworkWizard),
WVTI_ENTRY_ALL(CLSID_NULL, L"shell32.dll", IDS_TASK_COMPUTERSNEARME, IDS_TASK_COMPUTERSNEARME_TT, IDI_GROUP, CNetFolderViewCB::_CanViewComputersNearMe, CNetFolderViewCB::_OnViewComputersNearMe),
WVTI_ENTRY_ALL(CLSID_NULL, L"shell32.dll", IDS_TASK_SEARCHDS, IDS_TASK_SEARCHDS_TT, IDI_TASK_SEARCHDS, CNetFolderViewCB::_CanSearchActiveDirectory, CNetFolderViewCB::_OnSearchActiveDirectory),
};
BOOL CNetFolderViewCB::_EntireNetworkAvailable()
{
BOOL fRet = FALSE;
// Only enable if we're in a Domain
if (IsOS(OS_DOMAINMEMBER) && !SHRestricted(REST_NOENTIRENETWORK))
{
LPITEMIDLIST pidl;
if (SUCCEEDED(CNetFolder::_CreateEntireNetworkFullIDList(&pidl)))
{
// ... and we're not already in the "Entire Network" folder.
if (!ILIsEqual(_pidl, pidl))
{
fRet = TRUE;
}
ILFree(pidl);
}
}
return fRet;
}
HRESULT CNetFolderViewCB::OnGetWebViewContent(DWORD pv, SFVM_WEBVIEW_CONTENT_DATA* pData)
{
ZeroMemory(pData, sizeof(*pData));
Create_IUIElement(&c_MyNetPlacesTaskHeader, &(pData->pFolderTaskHeader));
LPCTSTR rgCsidls[] = { MAKEINTRESOURCE(CSIDL_DRIVES), MAKEINTRESOURCE(CSIDL_PERSONAL), MAKEINTRESOURCE(CSIDL_COMMON_DOCUMENTS), MAKEINTRESOURCE(CSIDL_PRINTERS) };
if (_EntireNetworkAvailable())
{
LPITEMIDLIST pidlEntireNetwork = NULL;
CNetFolder::_CreateEntireNetworkFullIDList(&pidlEntireNetwork);
CreateIEnumIDListOnCSIDLs2(_pidl, pidlEntireNetwork, rgCsidls, ARRAYSIZE(rgCsidls), &(pData->penumOtherPlaces));
ILFree(pidlEntireNetwork);
}
else
{
CreateIEnumIDListOnCSIDLs(_pidl, rgCsidls, ARRAYSIZE(rgCsidls), &(pData->penumOtherPlaces));
}
return S_OK;
}
HRESULT CNetFolderViewCB::OnGetWebViewTasks(DWORD pv, SFVM_WEBVIEW_TASKSECTION_DATA* pTasks)
{
ZeroMemory(pTasks, sizeof(*pTasks));
Create_IEnumUICommand((IUnknown*)(void*)this, c_MyNetPlacesTaskList, ARRAYSIZE(c_MyNetPlacesTaskList), &pTasks->penumFolderTasks);
return S_OK;
}
STDMETHODIMP CNetFolderViewCB::RealMessage(UINT uMsg, WPARAM wParam, LPARAM lParam)
{
switch (uMsg)
{
HANDLE_MSG(0, SFVM_INVOKECOMMAND, OnINVOKECOMMAND);
HANDLE_MSG(0, SFVM_GETHELPTEXT, OnGETHELPTEXT);
HANDLE_MSG(0, SFVM_DELAYWINDOWCREATE, OnDELAYWINDOWCREATE);
HANDLE_MSG(0, SFVM_GETCOLSAVESTREAM, OnGETCOLSAVESTREAM);
HANDLE_MSG(0, SFVM_GETZONE, OnGetZone);
HANDLE_MSG(0, SFVM_ENUMERATEDITEMS, OnEnumeratedItems);
HANDLE_MSG(0, SFVM_DEFVIEWMODE, OnDefViewMode);
HANDLE_MSG(0, SFVM_GETDEFERREDVIEWSETTINGS, OnGetDeferredViewSettings);
HANDLE_MSG(0, SFVM_REFRESH, OnREFRESH);
HANDLE_MSG(0, SFVM_GETWEBVIEWLAYOUT, OnGetWebViewLayout);
HANDLE_MSG(0, SFVM_GETWEBVIEWCONTENT, OnGetWebViewContent);
HANDLE_MSG(0, SFVM_GETWEBVIEWTASKS, OnGetWebViewTasks);
default:
return E_FAIL;
}
return S_OK;
}
// Replace all the space characters in the provider name with '_'.
void ReplaceSpacesWithUnderscore(LPTSTR psz)
{
while (psz = StrChr(psz, TEXT(' ')))
{
*psz = TEXT('_');
psz++; // DBCS safe
}
}
BOOL CNetFolderViewCB::_GetProviderKeyName(LPTSTR pszName, UINT uNameLen)
{
if (_pFolder->_GetProvider(NULL, NULL, pszName, uNameLen))
{
ReplaceSpacesWithUnderscore(pszName);
}
return (BOOL)*pszName;
}
// Define a collate order for the hood object types
#define _HOOD_COL_RON 0
#define _HOOD_COL_REMOTE 1
#define _HOOD_COL_FILE 2
#define _HOOD_COL_NET 3
const static ICONMAP c_aicmpNet[] = {
{ SHID_NET_NETWORK , II_NETWORK },
{ SHID_NET_DOMAIN , II_GROUP },
{ SHID_NET_SERVER , II_SERVER },
{ SHID_NET_SHARE , (UINT)EIRESID(IDI_SERVERSHARE) },
{ SHID_NET_DIRECTORY , II_FOLDER },
{ SHID_NET_PRINTER , II_PRINTER },
{ SHID_NET_RESTOFNET , II_WORLD },
{ SHID_NET_SHAREADMIN , II_DRIVEFIXED },
{ SHID_NET_TREE , II_TREE },
{ SHID_NET_NDSCONTAINER, (UINT)EIRESID(IDI_NDSCONTAINER) },
};
enum
{
NKID_PROVIDERTYPE = 0,
NKID_PROVIDER,
NKID_NETCLASS,
NKID_NETWORK,
NKID_DIRECTORY,
NKID_FOLDER
};
#define NKID_COUNT 6
// This is one-entry cache for remote junctions resolution
TCHAR g_szLastAttemptedJunctionName[MAX_PATH] = {0};
TCHAR g_szLastResolvedJunctionName[MAX_PATH] = {0};
REGITEMSINFO g_riiNetRoot =
{
REGSTR_PATH_EXPLORER TEXT("\\NetworkNeighborhood\\NameSpace"),
NULL,
TEXT(':'),
SHID_NET_REGITEM,
1,
SFGAO_CANLINK,
0,
NULL,
RIISA_ORIGINAL,
NULL,
0,
0,
};
CNetRootFolder* CNetRootFolder::_spThis = NULL;
HRESULT CNetFolder::_CreateInstance(LPCITEMIDLIST pidlAbs, LPCITEMIDLIST pidlTarget, UINT uDisplayType,
LPCIDNETRESOURCE pidnForProvider, LPCTSTR pszResName,
REFIID riid, void **ppv)
{
HRESULT hr = E_OUTOFMEMORY;
*ppv = NULL;
if (!ILIsEmpty(pidlAbs))
{
CNetFolder* pNetF = new CNetFolder(NULL);
if (NULL != pNetF)
{
pNetF->_uDisplayType = uDisplayType;
if (pidnForProvider)
{
//Make sure that the pidnProvider has provider information.
ASSERT(NET_FHasProvider(pidnForProvider))
//We are interested only in the provider informarion which is contained in the first entry.
//Its enough if we clone only the first item in the pidl.
pNetF->_pidnForProvider = (LPCIDNETRESOURCE)ILCloneFirst((LPCITEMIDLIST)pidnForProvider);
}
if (pszResName && *pszResName)
pNetF->_pszResName = StrDup(pszResName);
pNetF->_pidl = ILClone(pidlAbs);
pNetF->_pidlTarget = ILClone(pidlTarget);
if (pNetF->_pidl && (!pidlTarget || (pidlTarget && pNetF->_pidlTarget)))
{
if (uDisplayType == RESOURCEDISPLAYTYPE_SERVER)
{
// This is a remote computer. See if there are any remote
// computer registry items. If so, aggregate with the registry
// class.
REGITEMSINFO riiComputer =
{
REGSTR_PATH_EXPLORER TEXT("\\RemoteComputer\\NameSpace"),
NULL,
TEXT(':'),
SHID_NET_REMOTEREGITEM,
-1,
SFGAO_FOLDER | SFGAO_CANLINK,
0, // no required reg items
NULL,
RIISA_ORIGINAL,
pszResName,
0,
0,
};
CRegFolder_CreateInstance(&riiComputer,
(IUnknown*) (IShellFolder*) pNetF,
IID_PPV_ARG(IUnknown, &pNetF->_punkReg));
}
else if (uDisplayType == RESOURCEDISPLAYTYPE_ROOT)
{
//
// this is the entire net icon, so lets create an instance of the regitem folder
// so we can merge in the items from there.
//
REGITEMSINFO riiEntireNet =
{
REGSTR_PATH_EXPLORER TEXT("\\NetworkNeighborhood\\EntireNetwork\\NameSpace"),
NULL,
TEXT(':'),
SHID_NET_REGITEM,
-1,
SFGAO_CANLINK,
0, // no required reg items
NULL,
RIISA_ORIGINAL,
NULL,
0,
0,
};
CRegFolder_CreateInstance(&riiEntireNet,
(IUnknown*) (IShellFolder*) pNetF,
IID_PPV_ARG(IUnknown, &pNetF->_punkReg));
}
else
{
ASSERT(hr == E_OUTOFMEMORY);
}
hr = pNetF->QueryInterface(riid, ppv);
}
pNetF->Release();
}
else
{
ASSERT(hr == E_OUTOFMEMORY);
}
}
else
{
ASSERT(0);
hr = E_INVALIDARG;
}
return hr;
}
HRESULT CNetwork_CreateInstance(IUnknown* punkOuter, REFIID riid, void **ppv)
{
HRESULT hr = S_OK;
*ppv = NULL;
// Must enter critical section to avoid racing against v_HandleDelete
ENTERCRITICAL;
if (NULL != CNetRootFolder::_spThis)
{
hr = CNetRootFolder::_spThis->QueryInterface(riid, ppv);
}
else
{
CNetRootFolder* pNetRootF = new CNetRootFolder(punkOuter);
if (pNetRootF)
{
// Initialize it ourselves to ensure that the cached value
// is the correct one.
hr = pNetRootF->Initialize((LPCITEMIDLIST)&c_idlNet);
if (SUCCEEDED(hr))
{
pNetRootF->_uDisplayType = RESOURCEDISPLAYTYPE_GENERIC;
ASSERT(NULL == pNetRootF->_punkReg);
if (SHRestricted(REST_NOSETFOLDERS))
g_riiNetRoot.iReqItems = 0;
// create the regitems object, he has the NetRoot object as his outer guy.
hr = CRegFolder_CreateInstance(&g_riiNetRoot,
SAFECAST(pNetRootF, IShellFolder2*),
IID_PPV_ARG(IUnknown, &pNetRootF->_punkReg));
// NOTE: not using SHInterlockedCompareExchange() because we have the critsec
CNetRootFolder::_spThis = pNetRootF;
hr = pNetRootF->QueryInterface(riid, ppv);
}
// Release the self-reference, but keep the the _spThis pointer intact
// (it will be reset to NULL in the destructor)
pNetRootF->Release();
}
else
{
hr = E_OUTOFMEMORY;
}
}
LEAVECRITICAL;
return hr;
}
CNetFolder::CNetFolder(IUnknown* punkOuter) :
CAggregatedUnknown (punkOuter)
{
// Assert that we're still using a zero-init flag inside the new operator
ASSERT(NULL == _pidl);
ASSERT(NULL == _pidlTarget);
ASSERT(NULL == _pidnForProvider);
ASSERT(NULL == _pszResName);
ASSERT(0 == _uDisplayType);
ASSERT(NULL == _psfFiles);
ASSERT(NULL == _punkReg);
DllAddRef();
}
CNetFolder::~CNetFolder()
{
ILFree(_pidl);
ILFree(_pidlTarget);
ILFree((LPITEMIDLIST)_pidnForProvider);
if (NULL != _pszResName)
{
LocalFree(_pszResName);
}
if (_psfFiles)
{
_psfFiles->Release();
}
SHReleaseInnerInterface(SAFECAST(this, IShellFolder *), &_punkReg);
DllRelease();
}
CNetFolder *FolderToNetFolder(IUnknown *punk)
{
CNetFolder * pThis = NULL;
return punk && SUCCEEDED(punk->QueryInterface(CLSID_CNetFldr, (void **)&pThis)) ? pThis : NULL;
}
HRESULT CNetFolder::v_InternalQueryInterface(REFIID riid, void **ppv)
{
static const QITAB qit[] = {
QITABENT(CNetFolder, IShellFolder2), // IID_IShellFolder2
QITABENTMULTI(CNetFolder, IShellFolder, IShellFolder2), // IID_IShellFolder
QITABENT(CNetFolder, IPersistFolder3), // IID_IPersistFolder3
QITABENT(CNetFolder, IShellIconOverlay), // IID_IShellIconOverlay
QITABENTMULTI(CNetFolder, IPersistFolder2, IPersistFolder3), // IID_IPersistFolder2
QITABENTMULTI(CNetFolder, IPersistFolder, IPersistFolder3), // IID_IPersistFolder
QITABENTMULTI(CNetFolder, IPersist, IPersistFolder3), // IID_IPersist
QITABENTMULTI2(CNetFolder, IID_IPersistFreeThreadedObject, IPersist), // IID_IPersistFreeThreadedObject
{ 0 },
};
if (IsEqualIID(riid, CLSID_CNetFldr))
{
*ppv = this; // get class pointer (unrefed!)
return S_OK;
}
HRESULT hr;
if (_punkReg && RegGetsFirstShot(riid))
{
hr = _punkReg->QueryInterface(riid, ppv);
}
else
{
hr = QISearch(this, qit, riid, ppv);
if ((E_NOINTERFACE == hr) && _punkReg)
{
hr = _punkReg->QueryInterface(riid, ppv);
}
}
return hr;
}
BOOL CNetRootFolder::v_HandleDelete(PLONG pcRef)
{
ASSERT(NULL != pcRef);
ENTERCRITICAL;
// Once inside the critical section things are slightly more stable.
// CNetwork_CreateInstance won't be able to rescue the cached reference
// (and bump the refcount from 0 to 1). And we don't have to worry
// about somebody Release()ing us down to zero a second time, since
// no new references can show up.
//
// HOWEVER! All those scary things could've happened WHILE WE WERE
// WAITING TO ENTER THE CRITICAL SECTION.
//
// While we were waiting, somebody could've called CNetwork_CreateInstance,
// which bumps the reference count back up. So don't destroy ourselves
// if our object got "rescued".
//
// What's more, while we were waiting, that somebody could've then
// Release()d us back down to zero, causing us to be called on that
// other thread, notice that the refcount is indeed zero, and destroy
// the object, all on that other thread. So if we are not the cached
// instance, then don't destroy ourselves since that other thread did
// it already.
//
// And even more, somebody might call CNetwork_CreateInstance again
// and create a brand new object, which might COINCIDENTALLY happen
// to have the same address as the old object we are trying to destroy
// here. But in that case, it's okay to destroy the new object because
// it is indeed the case that the object's reference count is zero and
// deserves to be destroyed.
if (this == _spThis && 0 == *pcRef)
{
*pcRef = 1000; // protect against cached pointers bumping us up then down
delete this;
}
LEAVECRITICAL;
// return TRUE to indicate that we've implemented this function
// (regardless of whether or not this object was actually deleted)
return TRUE;
}
STDMETHODIMP CNetFolder::ParseDisplayName(HWND hwnd, LPBC pbc, WCHAR* pszName, ULONG* pchEaten,
LPITEMIDLIST* ppidl, DWORD* pdwAttributes)
{
return E_NOTIMPL;
}
// new for Win2K, this enables enuming the hidden admin shares
#ifndef RESOURCE_SHAREABLE
#define RESOURCE_SHAREABLE 0x00000006
#endif
//
// in:
// hwnd NULL indicates no UI.
// grfFlags IShellFolder::EnumObjects() SHCONTF_ flags
// pnr in/out params
//
//
DWORD CNetFolder::_OpenEnum(HWND hwnd, DWORD grfFlags, LPNETRESOURCE pnr, HANDLE *phEnum)
{
DWORD dwType = (grfFlags & SHCONTF_NETPRINTERSRCH) ? RESOURCETYPE_PRINT : RESOURCETYPE_ANY;
DWORD dwScope = pnr ? RESOURCE_GLOBALNET : RESOURCE_CONTEXT;
if ((_uDisplayType == RESOURCEDISPLAYTYPE_SERVER) &&
(grfFlags & SHCONTF_SHAREABLE))
{
dwScope = RESOURCE_SHAREABLE; // hidden admin shares for this server
}
DWORD err = WNetOpenEnum(dwScope, dwType, RESOURCEUSAGE_ALL, pnr, phEnum);
if ((err != WN_SUCCESS) && hwnd)
{
// If it failed because you are not authenticated yet,
// we need to let the user loggin to this network resource.
//
// REVIEW: Ask LenS to review this code.
if (err == WN_NOT_AUTHENTICATED ||
err == ERROR_LOGON_FAILURE ||
err == WN_BAD_PASSWORD ||
err == WN_ACCESS_DENIED)
{
// Retry with password dialog box.
err = WNetAddConnection3(hwnd, pnr, NULL, NULL, CONNECT_TEMPORARY | CONNECT_INTERACTIVE);
if (err == WN_SUCCESS)
err = WNetOpenEnum(dwScope, dwType, RESOURCEUSAGE_ALL, pnr, phEnum);
}
UINT idTemplate = pnr && pnr->lpRemoteName ? IDS_ENUMERR_NETTEMPLATE2 : IDS_ENUMERR_NETTEMPLATE1;
SHEnumErrorMessageBox(hwnd, idTemplate, err, pnr ? pnr->lpRemoteName : NULL, TRUE, MB_OK | MB_ICONHAND);
}
return err;
}
// find the share part of a UNC
// \\server\share
// return pointer to "share" or pointer to empty string if none
LPCTSTR PathFindShareName(LPCTSTR pszUNC)
{
LPCTSTR psz = SkipServerSlashes(pszUNC);
if (*psz)
{
psz = StrChr(psz + 1, TEXT('\\'));
if (psz)
psz++;
else
psz = TEXT("");
}
return psz;
}
// Flags for the dwRemote field
#define RMF_CONTEXT 0x00000001 // Entire network is being enumerated
#define RMF_SHOWREMOTE 0x00000002 // Return Remote Services for next enumeration
#define RMF_STOP_ENUM 0x00000004 // Stop enumeration
#define RMF_GETLINKENUM 0x00000008 // Hoodlinks enum needs to be fetched
#define RMF_SHOWLINKS 0x00000010 // Hoodlinks need to be shown
#define RMF_FAKENETROOT 0x00000020 // Don't enumerate the workgroup items
#define RMF_ENTIRENETSHOWN 0x40000000 // Entire network object shown
#define RMF_REMOTESHOWN 0x80000000 // Return Remote Services for next enumeration
class CNetFolderEnum : public CEnumIDListBase
{
public:
// IEnumIDList
STDMETHOD(Next)(ULONG celt, LPITEMIDLIST *rgelt, ULONG *pceltFetched);
private:
CNetFolderEnum(CNetFolder *pnf, DWORD grfFlags, DWORD dwRemote, HANDLE hEnum);
~CNetFolderEnum();
friend HRESULT Create_NetFolderEnum(CNetFolder* pnsf, DWORD grfFlags, DWORD dwRemote, HANDLE hEnum, IEnumIDList** ppenum);
CNetFolder *_pnsf; // CNetFolder object we're enumerating
HANDLE _hEnum;
DWORD _grfFlags;
LONG _cItems; // Count of items in buffer
LONG _iItem; // Current index of the item in the buffer
DWORD _dwRemote;
union {
NETRESOURCE _anr[0];
BYTE _szBuffer[8192];
};
IEnumIDList *_peunk; // used for enumerating file system items (links)
};
CNetFolderEnum::CNetFolderEnum(CNetFolder *pnsf, DWORD grfFlags, DWORD dwRemote, HANDLE hEnum) : CEnumIDListBase()
{
_pnsf = pnsf;
_pnsf->AddRef();
_grfFlags = grfFlags;
_dwRemote = dwRemote;
_hEnum = hEnum;
}
HRESULT Create_NetFolderEnum(CNetFolder* pnf, DWORD grfFlags, DWORD dwRemote, HANDLE hEnum, IEnumIDList** ppenum)
{
HRESULT hr;
CNetFolderEnum* p= new CNetFolderEnum(pnf, grfFlags, dwRemote, hEnum);
if (p)
{
hr = p->QueryInterface(IID_PPV_ARG(IEnumIDList, ppenum));
p->Release();
}
else
{
hr = E_OUTOFMEMORY;
*ppenum = NULL;
}
return hr;
}
CNetFolderEnum::~CNetFolderEnum()
{
_pnsf->Release(); // release the "this" ptr we have
if (_peunk)
_peunk->Release();
if (_hEnum)
WNetCloseEnum(_hEnum);
}
STDMETHODIMP CNetFolderEnum::Next(ULONG celt, LPITEMIDLIST *ppidl, ULONG *pceltFetched)
{
HRESULT hr;
*ppidl = NULL;
if (pceltFetched)
*pceltFetched = 0;
// Time to stop enumeration?
if (_dwRemote & RMF_STOP_ENUM)
return S_FALSE; // Yes
// should we try and get the links enumerator?
if (_dwRemote & RMF_GETLINKENUM)
{
IShellFolder2* psfNetHood;
if (SUCCEEDED(_pnsf->v_GetFileFolder(&psfNetHood)))
psfNetHood->EnumObjects(NULL, _grfFlags, &_peunk);
if (_peunk)
_dwRemote |= RMF_SHOWLINKS;
_dwRemote &= ~RMF_GETLINKENUM;
}
// should we be showing the links?
if (_dwRemote & RMF_SHOWLINKS)
{
if (_peunk)
{
ULONG celtFetched;
LPITEMIDLIST pidl;
hr = _peunk->Next(1, &pidl, &celtFetched);
if (hr == S_OK && celtFetched == 1)
{
*ppidl = pidl;
if (pceltFetched)
*pceltFetched = celtFetched;
return S_OK; // Added link
}
}
_dwRemote &= ~RMF_SHOWLINKS; // Done enumerating links
}
hr = S_OK;
// Do we add the remote folder?
// (Note: as a hack to ensure that the remote folder is added
// to the 'hood despite what MPR says, RMF_SHOWREMOTE can be
// set without RMF_CONTEXT set.)
if ((_dwRemote & RMF_SHOWREMOTE) && !(_dwRemote & RMF_REMOTESHOWN))
{
// Yes
// Only try to put the remote entry in once.
_dwRemote |= RMF_REMOTESHOWN;
// Is this not the Context container?
// (See note above as to why we are asking this question.)
if (!(_dwRemote & RMF_CONTEXT))
{
// Yes; stop after the next time
_dwRemote |= RMF_STOP_ENUM;
}
// We have fallen thru because the remote services is not
// installed.
// Is this not the Context container AND the remote folder
// is not installed?
if (!(_dwRemote & RMF_CONTEXT))
{
// Yes; nothing else to enumerate
return S_FALSE;
}
}
if (_dwRemote & RMF_FAKENETROOT)
{
if ((!(_dwRemote & RMF_ENTIRENETSHOWN)) &&
(S_FALSE != SHShouldShowWizards(_punkSite)))
{
_pnsf->_CreateEntireNetwork(ppidl); // fake entire net
_dwRemote |= RMF_ENTIRENETSHOWN;
}
else
{
return S_FALSE; // no more to enumerate
}
}
else
{
while (TRUE)
{
ULONG err = WN_SUCCESS;
LPNETRESOURCE pnr;
if (_iItem >= _cItems)
{
DWORD dwSize = sizeof(_szBuffer);
_cItems = -1; // its signed
_iItem = 0;
err = WNetEnumResource(_hEnum, (DWORD*)&_cItems, _szBuffer, &dwSize);
DebugMsg(DM_TRACE, TEXT("Net EnumCallback: err=%d Count=%d"), err, _cItems);
}
pnr = &_anr[_iItem++];
// Note: the <= below is correct as we already incremented the index...
if (err == WN_SUCCESS && (_iItem <= _cItems))
{
// decide if the thing is a folder or not
ULONG grfFlagsItem = ((pnr->dwUsage & RESOURCEUSAGE_CONTAINER) ||
(pnr->dwType == RESOURCETYPE_DISK) ||
(pnr->dwType == RESOURCETYPE_ANY)) ?
SHCONTF_FOLDERS : SHCONTF_NONFOLDERS;
// If this is the context enumeration, we want to insert the
// Remote Services after the first container.
//
// Remember that we need to return the Remote Services in the next iteration.
if ((pnr->dwUsage & RESOURCEUSAGE_CONTAINER) &&
(_dwRemote & RMF_CONTEXT))
{
_dwRemote |= RMF_SHOWREMOTE;
}
if ((_pnsf->_uDisplayType == RESOURCEDISPLAYTYPE_SERVER) &&
(_grfFlags & SHCONTF_SHAREABLE))
{
// filter out ADMIN$ and IPC$, based on str len
if (lstrlen(PathFindShareName(pnr->lpRemoteName)) > 2)
{
grfFlagsItem = 0;
}
}
// if this is a network object, work out if we should hide or note, so
// convert the provider to its type number and open the key under:
//
// HKEY_CLASSES_ROOT\Network\Type\<type string>
if ((pnr->dwDisplayType == RESOURCEDISPLAYTYPE_NETWORK) &&
!(_grfFlags & SHCONTF_INCLUDEHIDDEN))
{
DWORD dwType;
if (WNetGetProviderType(pnr->lpProvider, &dwType) == WN_SUCCESS)
{
TCHAR szRegValue[MAX_PATH];
wsprintf(szRegValue, TEXT("Network\\Type\\%d"), HIWORD(dwType));
BOOL fHide = FALSE;
DWORD cb = sizeof(fHide);
if ((ERROR_SUCCESS == SHGetValue(HKEY_CLASSES_ROOT, szRegValue, TEXT("HideProvider"), NULL, &fHide, &cb)) && fHide)
{
grfFlagsItem = 0;
}
}
}
// Check if we found requested type of net resource.
if (_grfFlags & grfFlagsItem)
{
if (SUCCEEDED(_pnsf->_NetResToIDList(pnr, FALSE, TRUE, (_grfFlags & SHCONTF_NONFOLDERS), ppidl)))
{
break;
}
}
}
else if (err == WN_NO_MORE_ENTRIES)
{
hr = S_FALSE; // no more element
break;
}
else
{
DebugMsg(DM_ERROR, TEXT("sh ER - WNetEnumResource failed (%lx)"), err);
hr = E_FAIL;
break;
}
}
}
if (pceltFetched)
*pceltFetched = (S_OK == hr) ? 1 : 0;
return hr;
}
STDMETHODIMP CNetFolder::EnumObjects(HWND hwnd, DWORD grfFlags, IEnumIDList** ppenum)
{
NETRESOURCE nr = {0};
TCHAR szProvider[MAX_PATH];
nr.lpProvider = (LPTSTR) _GetProvider(NULL, NULL, szProvider, ARRAYSIZE(szProvider));
if (_uDisplayType != RESOURCEDISPLAYTYPE_ROOT &&
_uDisplayType != RESOURCEDISPLAYTYPE_NETWORK)
{
nr.lpRemoteName = _pszResName;
}
HRESULT hr;
HANDLE hEnum;
DWORD err = _OpenEnum(hwnd, grfFlags, &nr, &hEnum);
if (err == WN_SUCCESS)
{
hr = Create_NetFolderEnum(this, grfFlags, 0, hEnum, ppenum);
if (FAILED(hr))
{
WNetCloseEnum(hEnum);
}
}
else
{
hr = HRESULT_FROM_WIN32(err);
}
return hr;
}
LPCIDNETRESOURCE NET_IsValidID(LPCITEMIDLIST pidl)
{
if (pidl && !ILIsEmpty(pidl) && ((pidl->mkid.abID[0] & SHID_GROUPMASK) == SHID_NET))
return (LPCIDNETRESOURCE)pidl;
return NULL;
}
STDMETHODIMP CNetFolder::BindToObject(LPCITEMIDLIST pidl, LPBC pbc, REFIID riid, void **ppv)
{
HRESULT hr;
LPCIDNETRESOURCE pidn;
*ppv = NULL;
pidn = NET_IsValidID(pidl);
if (pidn)
{
IShellFolder *psfJunction;
LPITEMIDLIST pidlInit = NULL;
LPITEMIDLIST pidlTarget = NULL;
LPCITEMIDLIST pidlRight = _ILNext(pidl);
BOOL fRightIsEmpty = ILIsEmpty(pidlRight);
LPCIDNETRESOURCE pidnProvider = NET_FHasProvider(pidn) ? pidn :_pidnForProvider;
hr = S_OK;
// lets get the IDLISTs we are going to use to initialize the shell folder
// if we are doing a single level bind then then ILCombine otherwise
// be more careful.
pidlInit = ILCombineParentAndFirst(_pidl, pidl, pidlRight);
if (_pidlTarget)
pidlTarget = ILCombineParentAndFirst(_pidlTarget, pidl, pidlRight);
if (!pidlInit || (!pidlTarget && _pidlTarget))
hr = E_OUTOFMEMORY;
// now create the folder object we are using, and either return that
// object to the caller, or continue the binding down.
if (SUCCEEDED(hr))
{
hr = _CreateFolderForItem(pbc, pidlInit, pidlTarget, pidnProvider,
fRightIsEmpty ? riid : IID_IShellFolder,
fRightIsEmpty ? ppv : (void **)&psfJunction);
if (!fRightIsEmpty && SUCCEEDED(hr))
{
hr = psfJunction->BindToObject(pidlRight, pbc, riid, ppv);
psfJunction->Release();
}
}
ILFree(pidlInit);
ILFree(pidlTarget);
}
else
{
hr = E_INVALIDARG;
}
return hr;
}
STDMETHODIMP CNetFolder::BindToStorage(LPCITEMIDLIST pidl, LPBC pbc, REFIID riid, void **ppv)
{
return BindToObject(pidl, pbc, riid, ppv);
}
STDMETHODIMP CNetFolder::CompareIDs(LPARAM iCol, LPCITEMIDLIST pidl1, LPCITEMIDLIST pidl2)
{
HRESULT hr = E_INVALIDARG;
LPCIDNETRESOURCE pidn1 = NET_IsValidID(pidl1);
LPCIDNETRESOURCE pidn2 = NET_IsValidID(pidl2);
if (pidn1 && pidn2)
{
TCHAR szBuff1[MAX_PATH], szBuff2[MAX_PATH];
switch (iCol & SHCIDS_COLUMNMASK)
{
case ICOL_COMMENT:
{
hr = ResultFromShort(StrCmpLogicalRestricted(NET_CopyComment(pidn1, szBuff1, ARRAYSIZE(szBuff1)),
NET_CopyComment(pidn2, szBuff2, ARRAYSIZE(szBuff2))));
if (hr != 0)
return hr;
// drop down into the name comparison
}
case ICOL_NAME:
{
// Compare by name. This is the one case where we need to handle
// simple ids in either place. We will try to resync the items
// if we find a case of this before do the compares.
// Check for relative IDs. In particular if one item is at
// a server and the other is at RestOfNet then try to resync
// the two
//
if (NET_IsFake(pidn1) || NET_IsFake(pidn2))
{
// if either pidn1 or pidn2 is fake then we assume they are identical,
// this allows us to compare a simple net ID to a real net ID. we
// assume that if this fails later then the world will be happy
hr = 0;
}
else
{
// otherwise lets look at the names and provider strings accordingly
NET_CopyResName(pidn1, szBuff1, ARRAYSIZE(szBuff1));
NET_CopyResName(pidn2, szBuff2, ARRAYSIZE(szBuff2));
hr = ResultFromShort(StrCmpLogicalRestricted(szBuff1, szBuff2));
// If they're still identical, compare provider names.
if ((hr == 0) && (iCol & SHCIDS_ALLFIELDS))
{
LPCTSTR pszProv1 = _GetProvider(pidn1, NULL, szBuff1, ARRAYSIZE(szBuff1));
LPCTSTR pszProv2 = _GetProvider(pidn2, NULL, szBuff2, ARRAYSIZE(szBuff2));
if (pszProv1 && pszProv2)
hr = ResultFromShort(lstrcmp(pszProv1, pszProv2));
else
{
if (pszProv1 || pszProv2)
hr = ResultFromShort(pszProv1 ? 1 : -1);
else
hr = ResultFromShort(0);
}
}
}
// If they identical, compare the rest of IDs.
if (hr == 0)
hr = ILCompareRelIDs((IShellFolder*)this, (LPCITEMIDLIST)pidn1, (LPCITEMIDLIST)pidn2, iCol);
}
}
}
return hr;
}
STDMETHODIMP CNetFolder::CreateViewObject(HWND hwnd, REFIID riid, void **ppv)
{
HRESULT hr;
*ppv = NULL;
if (IsEqualIID(riid, IID_IShellView))
{
SFV_CREATE sSFV;
sSFV.cbSize = sizeof(sSFV);
sSFV.psvOuter = NULL;
sSFV.psfvcb = new CNetFolderViewCB(this); // failure is OK, we just get generic support
QueryInterface(IID_PPV_ARG(IShellFolder, &sSFV.pshf)); // in case we are agregated
hr = SHCreateShellFolderView(&sSFV, (IShellView**) ppv);
if (sSFV.pshf)
sSFV.pshf->Release();
if (sSFV.psfvcb)
sSFV.psfvcb->Release();
}
else if (IsEqualIID(riid, IID_IContextMenu))
{
IShellFolder* psfOuter;
hr = QueryInterface(IID_PPV_ARG(IShellFolder, &psfOuter));
if (SUCCEEDED(hr))
{
hr = CDefFolderMenu_Create(_pidl, hwnd, 0, NULL, psfOuter,
CNetwork_DFMCallBackBG, NULL, NULL, (IContextMenu**) ppv);
psfOuter->Release();
}
}
else
{
hr = E_NOINTERFACE;
}
return hr;
}
typedef HRESULT (CALLBACK *PFNGAOCALLBACK)(IShellFolder2 *psf, LPCITEMIDLIST pidl, ULONG* prgfInOut);
STDAPI GetAttributesCallback(IShellFolder2 *psf, UINT cidl, LPCITEMIDLIST* apidl, ULONG *prgfInOut, PFNGAOCALLBACK pfnGAOCallback)
{
HRESULT hr = S_OK;
ULONG rgfOut = 0;
for (UINT i = 0; i < cidl; i++)
{
ULONG rgfT = *prgfInOut;
hr = pfnGAOCallback(psf, apidl[i], &rgfT);
if (FAILED(hr))
{
rgfOut = 0;
break;
}
rgfOut |= rgfT;
}
*prgfInOut &= rgfOut;
return hr;
}
STDMETHODIMP CNetFolder::GetAttributesOf(UINT cidl, LPCITEMIDLIST* apidl, ULONG* prgfInOut)
{
HRESULT hr;
if (IsSelf(cidl, apidl))
{
*prgfInOut &= (SFGAO_CANLINK | SFGAO_HASPROPSHEET | SFGAO_HASSUBFOLDER |
SFGAO_FOLDER | SFGAO_FILESYSANCESTOR);
hr = S_OK;
}
else
{
hr = GetAttributesCallback(SAFECAST(this, IShellFolder2*), cidl, apidl, prgfInOut, _AttributesCallback);
}
return hr;
}
STDMETHODIMP CNetFolder::GetDisplayNameOf(LPCITEMIDLIST pidl, DWORD dwFlags, STRRET* pStrRet)
{
HRESULT hr;
LPCIDNETRESOURCE pidn = NET_IsValidID(pidl);
if (pidn)
{
TCHAR szPath[MAX_PATH];
LPCITEMIDLIST pidlNext = _ILNext(pidl);
if (dwFlags & SHGDN_FORPARSING)
{
if ((dwFlags & SHGDN_INFOLDER) ||
((dwFlags & SHGDN_FORADDRESSBAR) && (NET_GetDisplayType(pidn) == RESOURCEDISPLAYTYPE_ROOT))) // the non-infolder name for the root is not good for the address bar
{
NET_CopyResName(pidn, szPath, ARRAYSIZE(szPath));
if (ILIsEmpty(pidlNext))
{
// we just need the last part of the display name (IN FOLDER)
LPTSTR pszT = StrRChr(szPath, NULL, TEXT('\\'));
if (!pszT)
pszT = szPath;
else
pszT++; // move past '\'
hr = StringToStrRet(pszT, pStrRet);
}
else
{
hr = ILGetRelDisplayName((IShellFolder*) this, pStrRet, pidl, szPath, MAKEINTRESOURCE(IDS_DSPTEMPLATE_WITH_BACKSLASH), dwFlags);
}
}
else
{
LPCITEMIDLIST pidlRight = _ILNext(pidl);
if (ILIsEmpty(pidlRight))
{
hr = _GetPathForItem(pidn, szPath);
if (SUCCEEDED(hr))
{
hr = StringToStrRet(szPath, pStrRet);
}
}
else
{
IShellFolder *psfJunction;
//Get the pidn which has network provider information.
LPCIDNETRESOURCE pidnProvider = NET_FHasProvider(pidn) ? pidn :_pidnForProvider;
LPITEMIDLIST pidlInit, pidlTarget = NULL;
pidlInit = ILCombineParentAndFirst(_pidl, pidl, pidlRight);
if (_pidlTarget)
pidlTarget = ILCombineParentAndFirst(_pidlTarget, pidl, pidlRight);
if (!pidlInit || (_pidlTarget && !pidlTarget))
return E_OUTOFMEMORY;
hr = _CreateFolderForItem(NULL, pidlInit, pidlTarget, pidnProvider, IID_PPV_ARG(IShellFolder, &psfJunction));
if (SUCCEEDED(hr))
{
hr = psfJunction->GetDisplayNameOf(pidlRight, dwFlags, pStrRet);
psfJunction->Release();
}
ILFree(pidlInit);
ILFree(pidlTarget);
}
}
}
else
{
hr = _GetFormatName(pidn, pStrRet);
if (SUCCEEDED(hr) && !(dwFlags & SHGDN_INFOLDER) && (NET_GetFlags(pidn) & SHID_JUNCTION))
{
TCHAR szServer[MAX_PATH];
SHGetNameAndFlags(_pidlTarget ? _pidlTarget:_pidl, SHGDN_FORPARSING, szServer, ARRAYSIZE(szServer), NULL);
TCHAR szDisplay[MAX_PATH];
hr = SHGetComputerDisplayName(szServer, 0x0, szDisplay, ARRAYSIZE(szDisplay));
if (SUCCEEDED(hr))
{
StrRetFormat(pStrRet, pidl, MAKEINTRESOURCE(IDS_DSPTEMPLATE_WITH_ON), szDisplay);
}
}
}
}
else
hr = E_INVALIDARG;
return hr;
}
STDMETHODIMP CNetFolder::SetNameOf(HWND hwnd, LPCITEMIDLIST pidl, LPCOLESTR lpszName, DWORD dwRes, LPITEMIDLIST* ppidl)
{
if (ppidl)
*ppidl = NULL;
return E_NOTIMPL; // not supported
}
STDMETHODIMP CNetFolder::GetUIObjectOf(HWND hwnd, UINT cidl, LPCITEMIDLIST* apidl,
REFIID riid, UINT* prgfInOut, void **ppv)
{
HRESULT hr = E_INVALIDARG;
LPCIDNETRESOURCE pidn = cidl ? NET_IsValidID(apidl[0]) : NULL;
*ppv = NULL;
if ((IsEqualIID(riid, IID_IExtractIconA) || IsEqualIID(riid, IID_IExtractIconW)) && pidn)
{
UINT iIndex;
if (_IsPrintShare(pidn))
iIndex = (UINT)EIRESID(IDI_PRINTER_NET);
else if (NET_IsRemoteFld(pidn))
iIndex = II_RNA;
else
iIndex = SILGetIconIndex(apidl[0], c_aicmpNet, ARRAYSIZE(c_aicmpNet));
hr = SHCreateDefExtIcon(NULL, iIndex, iIndex, GIL_PERCLASS, II_FOLDER, riid, ppv);
}
else if (IsEqualIID(riid, IID_IContextMenu) && pidn)
{
HKEY ahkeys[NKID_COUNT];
hr = _OpenKeys(pidn, ahkeys);
if (SUCCEEDED(hr))
{
IShellFolder* psfOuter;
hr = QueryInterface(IID_PPV_ARG(IShellFolder, &psfOuter));
if (SUCCEEDED(hr))
{
hr = CDefFolderMenu_Create2(_pidl, hwnd, cidl, apidl,
psfOuter, _GetCallbackType(pidn),
ARRAYSIZE(ahkeys), ahkeys, (IContextMenu**) ppv);
psfOuter->Release();
}
SHRegCloseKeys(ahkeys, ARRAYSIZE(ahkeys));
}
}
else if (cidl && IsEqualIID(riid, IID_IDataObject))
{
// Point & Print printer installation assumes that the
// netresources from CNetData_GetData and the
// pidls from CIDLData_GetData are in the same order.
// Keep it this way.
CNetData *pnd = new CNetData(_pidl, cidl, apidl);
if (pnd)
{
hr = pnd->QueryInterface(riid, ppv);
pnd->Release();
}
else
hr = E_OUTOFMEMORY;
}
else if (pidn && IsEqualIID(riid, IID_IDropTarget))
{
// special support because this is an item (not a folder)
if (_IsPrintShare(pidn))
{
LPITEMIDLIST pidl;
hr = SHILCombine(_pidl, apidl[0], &pidl);
if (SUCCEEDED(hr))
{
hr = CPrinterDropTarget_CreateInstance(hwnd, pidl, (IDropTarget**)ppv);
ILFree(pidl);
}
}
else
{
IShellFolder *psf;
hr = BindToObject(apidl[0], NULL, IID_PPV_ARG(IShellFolder, &psf));
if (SUCCEEDED(hr))
{
hr = psf->CreateViewObject(hwnd, riid, ppv);
psf->Release();
}
}
}
else if (pidn && IsEqualIID(riid, IID_IQueryInfo))
{
if (NET_GetDisplayType(pidn) == RESOURCEDISPLAYTYPE_ROOT)
{
hr = CreateInfoTipFromText(MAKEINTRESOURCE(IDS_RESTOFNETTIP), riid, ppv);
}
else
{
// Someday maybe have infotips for other things too
}
}
return hr;
}
STDMETHODIMP CNetFolder::GetDefaultSearchGUID(LPGUID pguid)
{
*pguid = SRCID_SFindComputer;
return S_OK;
}
void WINAPI CNetFolder::_CopyEnumElement(void* pDest, const void* pSource, DWORD dwSize)
{
if (pDest && pSource)
memcpy(pDest, pSource, dwSize);
}
STDMETHODIMP CNetFolder::EnumSearches(IEnumExtraSearch** ppenum)
{
HRESULT hr = E_NOTIMPL;
*ppenum = NULL;
// if the restriction is set then this item should be enumerated from the registry
// so we fail, else enumerate it
// only enumerate if we actually have a network to search against
if (!SHRestricted(REST_HASFINDCOMPUTERS) &&
(GetSystemMetrics(SM_NETWORK) & RNC_NETWORKS))
{
EXTRASEARCH *pxs = (EXTRASEARCH *)LocalAlloc(LPTR, sizeof(EXTRASEARCH));
if (pxs)
{
pxs->guidSearch = SRCID_SFindComputer;
if (LoadStringW(g_hinst, IDS_FC_NAME, pxs->wszFriendlyName, sizeof(pxs->wszFriendlyName)))
{
*ppenum = (IEnumExtraSearch*)CStandardEnum_CreateInstance(IID_IEnumExtraSearch, FALSE,
1, sizeof(EXTRASEARCH), pxs, _CopyEnumElement);
if (*ppenum == NULL)
{
LocalFree(pxs);
hr = E_OUTOFMEMORY;
}
else
hr = S_OK;
}
}
else
hr = E_OUTOFMEMORY;
}
return hr;
}
STDMETHODIMP CNetFolder::GetDefaultColumn(DWORD dwRes, ULONG* pSort, ULONG* pDisplay)
{
return E_NOTIMPL;
}
HRESULT CNetFolder::_GetDefaultColumnState(UINT cColumns, UINT iColumn, DWORD* pdwState)
{
*pdwState = 0;
HRESULT hr = S_OK;
if (iColumn < cColumns)
{
*pdwState = s_net_cols[iColumn].csFlags;
if (iColumn >= 1)
*pdwState |= SHCOLSTATE_SLOW; // comment is slow for net root
}
else
{
hr = E_INVALIDARG;
}
return hr;
}
STDMETHODIMP CNetFolder::GetDetailsEx(LPCITEMIDLIST pidl, const SHCOLUMNID* pscid, VARIANT* pv)
{
HRESULT hr = E_NOTIMPL;
LPCIDNETRESOURCE pidn = NET_IsValidID(pidl);
if (pidn)
{
if (IsEqualSCID(*pscid, SCID_NETRESOURCE))
{
// Office calls SHGetDataFromIDList() with a large buffer to hold all
// of the strings in the NETRESOURCE structure, so we need to make sure
// that our variant can hold enough data to pass back to it:
BYTE rgBuffer[sizeof(NETRESOURCEW) + (4 * MAX_PATH * sizeof(WCHAR))];
hr = _GetNetResource(pidn, (NETRESOURCEW*) rgBuffer, sizeof(rgBuffer));
if (SUCCEEDED(hr))
{
hr = InitVariantFromBuffer(pv, rgBuffer, sizeof(rgBuffer));
if (SUCCEEDED(hr))
{
// Fixup pointers in structure to point within the variant
// instead of our stack variable (rgBuffer):
ASSERT(pv->vt == (VT_ARRAY | VT_UI1));
NETRESOURCEW* pnrw = (NETRESOURCEW*) pv->parray->pvData;
if (pnrw->lpLocalName)
{
pnrw->lpLocalName = (LPWSTR) ((BYTE*) pnrw +
((BYTE*) pnrw->lpLocalName - rgBuffer));
}
if (pnrw->lpRemoteName)
{
pnrw->lpRemoteName = (LPWSTR) ((BYTE*) pnrw +
((BYTE*) pnrw->lpRemoteName - rgBuffer));
}
if (pnrw->lpComment)
{
pnrw->lpComment = (LPWSTR) ((BYTE*) pnrw +
((BYTE*) pnrw->lpComment - rgBuffer));
}
if (pnrw->lpProvider)
{
pnrw->lpProvider = (LPWSTR) ((BYTE*) pnrw +
((BYTE*) pnrw->lpProvider - rgBuffer));
}
}
}
}
else if (IsEqualSCID(*pscid, SCID_DESCRIPTIONID))
{
SHDESCRIPTIONID did;
switch(SIL_GetType(pidl) & SHID_TYPEMASK)
{
case SHID_NET_DOMAIN:
did.dwDescriptionId = SHDID_NET_DOMAIN;
break;
case SHID_NET_SERVER:
did.dwDescriptionId = SHDID_NET_SERVER;
break;
case SHID_NET_SHARE:
did.dwDescriptionId = SHDID_NET_SHARE;
break;
case SHID_NET_RESTOFNET:
did.dwDescriptionId = SHDID_NET_RESTOFNET;
break;
default:
did.dwDescriptionId = SHDID_NET_OTHER;
break;
}
did.clsid = CLSID_NULL;
hr = InitVariantFromBuffer(pv, &did, sizeof(did));
}
else if (IsEqualSCID(*pscid, SCID_Comment))
{
TCHAR szTemp[MAX_PATH];
hr = InitVariantFromStr(pv, NET_CopyComment(pidn, szTemp, ARRAYSIZE(szTemp)));
}
else if (IsEqualSCID(*pscid, SCID_NAME))
{
TCHAR szTemp[MAX_PATH];
hr = InitVariantFromStr(pv, NET_CopyResName(pidn, szTemp, ARRAYSIZE(szTemp)));
}
}
else
{
IShellFolder2* psfFiles;
hr = v_GetFileFolder(&psfFiles);
if (SUCCEEDED(hr))
hr = psfFiles->GetDetailsEx(pidl, pscid, pv);
}
return hr;
}
HRESULT CNetFolder::_GetDetailsOf(UINT cColumns, LPCITEMIDLIST pidl, UINT iColumn, SHELLDETAILS *pDetails)
{
HRESULT hr = S_OK;
pDetails->str.uType = STRRET_CSTR;
pDetails->str.cStr[0] = 0;
if (NULL == pidl)
{
hr = GetDetailsOfInfo(s_net_cols, cColumns, iColumn, pDetails);
}
else
{
SHCOLUMNID scid;
hr = MapColumnToSCID(iColumn, &scid);
if (SUCCEEDED(hr))
{
VARIANT var;
hr = GetDetailsEx(pidl, &scid, &var);
if (SUCCEEDED(hr))
{
TCHAR szTemp[MAX_PATH];
hr = SHFormatForDisplay(scid.fmtid,
scid.pid,
(PROPVARIANT*)&var,
PUIFFDF_DEFAULT,
szTemp,
ARRAYSIZE(szTemp));
if (SUCCEEDED(hr))
{
hr = StringToStrRet(szTemp, &pDetails->str);
}
VariantClear(&var);
}
}
}
return hr;
}
HRESULT CNetFolder::_MapColumnToSCID(UINT cColumns, UINT iColumn, SHCOLUMNID* pscid)
{
return MapColumnToSCIDImpl(s_net_cols, cColumns, iColumn, pscid);
}
// IPersist methods
STDMETHODIMP CNetFolder::GetClassID(CLSID* pCLSID)
{
switch (_uDisplayType)
{
case RESOURCEDISPLAYTYPE_ROOT:
*pCLSID = CLSID_NetworkRoot;
break;
case RESOURCEDISPLAYTYPE_SERVER:
*pCLSID = CLSID_NetworkServer;
break;
case RESOURCEDISPLAYTYPE_DOMAIN:
*pCLSID = CLSID_NetworkDomain;
break;
case RESOURCEDISPLAYTYPE_SHARE:
*pCLSID = CLSID_NetworkShare;
break;
default:
*pCLSID = CLSID_NULL;
break;
}
return S_OK;
}
// IPersistFolder method
STDMETHODIMP CNetFolder::Initialize(LPCITEMIDLIST pidl)
{
ILFree(_pidl);
ILFree(_pidlTarget);
_pidl = _pidlTarget = NULL;
return SHILClone(pidl, &_pidl);
}
// IPersistFolder2 method
STDMETHODIMP CNetFolder::GetCurFolder(LPITEMIDLIST* ppidl)
{
return GetCurFolderImpl(_pidl, ppidl);
}
// IPersistFolder3 methods
STDMETHODIMP CNetFolder::InitializeEx(IBindCtx *pbc, LPCITEMIDLIST pidlRoot, const PERSIST_FOLDER_TARGET_INFO *pfti)
{
ILFree(_pidl);
ILFree(_pidlTarget);
_pidl = _pidlTarget = NULL;
HRESULT hr = SHILClone(pidlRoot, &_pidl);
if (SUCCEEDED(hr) && pfti && pfti->pidlTargetFolder)
{
hr = SHILClone(pfti->pidlTargetFolder, &_pidlTarget);
}
return hr;
}
STDMETHODIMP CNetFolder::GetFolderTargetInfo(PERSIST_FOLDER_TARGET_INFO *pfti)
{
HRESULT hr = S_OK;
ZeroMemory(pfti, sizeof(*pfti));
if (_pidlTarget)
hr = SHILClone(_pidlTarget, &pfti->pidlTargetFolder);
pfti->dwAttributes = FILE_ATTRIBUTE_DIRECTORY; // maybe add system?
pfti->csidl = -1;
return hr;
}
// IShellIconOverlay
HRESULT CNetFolder::_GetIconOverlayInfo(LPCIDNETRESOURCE pidn, int *pIndex, DWORD dwFlags)
{
HRESULT hr = E_FAIL;
//
// For netshare objects we want to get the icon overlay.
// If the share is "pinned" to be available offline it will
// have the "Offline Files" overlay.
//
if (RESOURCEDISPLAYTYPE_SHARE == NET_GetDisplayType(pidn))
{
TCHAR szPath[MAX_PATH];
hr = _GetPathForItem(pidn, szPath);
if (SUCCEEDED(hr))
{
IShellIconOverlayManager *psiom;
hr = GetIconOverlayManager(&psiom);
if (SUCCEEDED(hr))
{
WCHAR szPathW[MAX_PATH];
SHTCharToUnicode(szPath, szPathW, ARRAYSIZE(szPathW));
hr = psiom->GetFileOverlayInfo(szPathW, 0, pIndex, dwFlags);
psiom->Release();
}
}
}
return hr;
}
STDMETHODIMP CNetFolder::GetOverlayIndex(LPCITEMIDLIST pidl, int *pIndex)
{
HRESULT hr = E_FAIL;
LPCIDNETRESOURCE pidn = NET_IsValidID(pidl);
if (NULL != pidn)
{
hr = _GetIconOverlayInfo(pidn, pIndex, SIOM_OVERLAYINDEX);
}
return hr;
}
STDMETHODIMP CNetFolder::GetOverlayIconIndex(LPCITEMIDLIST pidl, int *pIndex)
{
HRESULT hr = E_FAIL;
LPCIDNETRESOURCE pidn = NET_IsValidID(pidl);
if (NULL != pidn)
{
hr = _GetIconOverlayInfo(pidn, pIndex, SIOM_ICONINDEX);
}
return hr;
}
//
// Helper function to allow external callers to query information from a
// network pidl...
//
// NOTE NOTE - This function returns a NETRESOURCE structure whose string
// pointers are not valid. On Win95 they were pointers back into the pidl's
// strings (even though the strings were copied into the supplied pv buffer.)
// Now we make the pointers really point into the buffer.
//
HRESULT CNetFolder::_GetNetResource(LPCIDNETRESOURCE pidn, NETRESOURCEW* pnr, int cb)
{
TCHAR szStrings[3][MAX_PATH];
LPWSTR psz, lpsz[3] = {NULL, NULL, NULL};
int i, cchT;
if (cb < sizeof(*pnr))
return DISP_E_BUFFERTOOSMALL;
ZeroMemory(pnr, cb);
NET_CopyResName(pidn, szStrings[0], ARRAYSIZE(szStrings[0]));
NET_CopyComment(pidn, szStrings[1], ARRAYSIZE(szStrings[1]));
_GetProvider(pidn, NULL, szStrings[2], ARRAYSIZE(szStrings[2]));
// Fill in some of the stuff first.
// pnr->dwScope = 0;
pnr->dwType = NET_GetType(pidn);
pnr->dwDisplayType = NET_GetDisplayType(pidn);
pnr->dwUsage = NET_GetUsage(pidn);
// pnr->lpLocalName = NULL;
// Now lets copy the strings into the buffer and make the pointers
// relative to the buffer...
psz = (LPWSTR)(pnr + 1);
cb -= sizeof(*pnr);
for (i = 0; i < ARRAYSIZE(szStrings); i++)
{
if (*szStrings[i])
{
cchT = (lstrlen(szStrings[i]) + 1) * sizeof(TCHAR);
if (cchT <= cb)
{
SHTCharToUnicode(szStrings[i], psz, cb/sizeof(TCHAR));
lpsz[i] = psz;
psz += cchT;
cb -= cchT * sizeof(TCHAR);
}
else
{
// A hint that the structure is ok,
// but the strings are missing
SetLastError(ERROR_INSUFFICIENT_BUFFER);
}
}
}
pnr->lpRemoteName = lpsz[0];
pnr->lpComment = lpsz[1];
pnr->lpProvider = lpsz[2];
return S_OK;
}
//
// This function opens a reg. database key based on the "network provider".
//
// Returns: hkey
//
// The caller is responsibe to close the key by calling RegCloseKey().
//
HKEY CNetFolder::_OpenProviderKey(LPCIDNETRESOURCE pidn)
{
TCHAR szProvider[MAX_PATH];
if (_GetProvider(pidn, NULL, szProvider, ARRAYSIZE(szProvider)))
{
HKEY hkeyProgID = NULL;
ReplaceSpacesWithUnderscore(szProvider);
RegOpenKey(HKEY_CLASSES_ROOT, szProvider, &hkeyProgID);
return hkeyProgID;
}
return NULL;
}
//
// This function opens a reg. database key based on the network provider type.
// The type is a number that is not localized, as opposed to the provider name
// which may be localized.
//
// Arguments:
// pidlAbs -- Absolute IDList to a network resource object.
//
// Returns: hkey
//
// Notes:
// The caller is responsible to close the key by calling RegCloseKey().
//
HKEY CNetFolder::_OpenProviderTypeKey(LPCIDNETRESOURCE pidn)
{
HKEY hkeyProgID = NULL;
TCHAR szProvider[MAX_PATH];
if (_GetProvider(pidn, NULL, szProvider, ARRAYSIZE(szProvider)))
{
// Now that we've got the provider name, get the provider id.
DWORD dwType;
if (WNetGetProviderType(szProvider, &dwType) == WN_SUCCESS)
{
// convert nis.wNetType to a string, and then open the key
// HKEY_CLASSES_ROOT\Network\Type\<type string>
TCHAR szRegValue[MAX_PATH];
wsprintf(szRegValue, TEXT("Network\\Type\\%d"), HIWORD(dwType));
RegOpenKey(HKEY_CLASSES_ROOT, szRegValue, &hkeyProgID);
}
}
return hkeyProgID;
}
HRESULT CNetFolder::_OpenKeys(LPCIDNETRESOURCE pidn, HKEY ahkeys[NKID_COUNT])
{
// See if there is a key specific to the type of Network object...
COMPILETIME_ASSERT(6 == NKID_COUNT);
ahkeys[0] = ahkeys[1] = ahkeys[2] = ahkeys[3] = ahkeys[4] = ahkeys[5] = NULL;
ahkeys[NKID_PROVIDERTYPE] = _OpenProviderTypeKey(pidn);
ahkeys[NKID_PROVIDER] = _OpenProviderKey(pidn);
if (NET_GetDisplayType(pidn) == RESOURCEDISPLAYTYPE_SHARE)
RegOpenKey(HKEY_CLASSES_ROOT, TEXT("NetShare"), &ahkeys[NKID_NETCLASS]);
else if (NET_GetDisplayType(pidn) == RESOURCEDISPLAYTYPE_SERVER)
RegOpenKey(HKEY_CLASSES_ROOT, TEXT("NetServer"), &ahkeys[NKID_NETCLASS]);
RegOpenKey(HKEY_CLASSES_ROOT, TEXT("Network"), &ahkeys[NKID_NETWORK]);
// make sure it is not a printer before adding "Folder" or "directory"
if (!_IsPrintShare(pidn))
{
// Shares should also support directory stuff...
if (NET_GetDisplayType(pidn) == RESOURCEDISPLAYTYPE_SHARE)
RegOpenKey(HKEY_CLASSES_ROOT, TEXT("Directory"), &ahkeys[NKID_DIRECTORY]);
RegOpenKey(HKEY_CLASSES_ROOT, c_szFolderClass, &ahkeys[NKID_FOLDER]);
}
return S_OK;
}
#define WNFMT_PLATFORM WNFMT_ABBREVIATED | WNFMT_INENUM
//
// This function retrieves the formatted (display) name of the specified network object.
//
HRESULT CNetFolder::_GetFormatName(LPCIDNETRESOURCE pidn, STRRET* pStrRet)
{
HRESULT hr = E_FAIL;
TCHAR szName[MAX_PATH];
NET_CopyResName(pidn, szName, ARRAYSIZE(szName));
if (NET_GetDisplayType(pidn) == RESOURCEDISPLAYTYPE_SERVER)
{
TCHAR szMachineName[MAX_PATH];
TCHAR szComment[MAX_PATH];
NET_CopyResName(pidn, szMachineName, ARRAYSIZE(szMachineName));
NET_CopyComment(pidn, szComment, ARRAYSIZE(szComment));
hr = SHBuildDisplayMachineName(szMachineName, szComment, szName, ARRAYSIZE(szName));
}
if (FAILED(hr) &&
(NET_GetDisplayType(pidn) != RESOURCEDISPLAYTYPE_ROOT) &&
(NET_GetDisplayType(pidn) != RESOURCEDISPLAYTYPE_NETWORK))
{
TCHAR szDisplayName[MAX_PATH], szProvider[MAX_PATH];
DWORD dwSize = ARRAYSIZE(szDisplayName);
LPCTSTR pszProvider = _GetProvider(pidn, NULL, szProvider, ARRAYSIZE(szProvider));
if (pszProvider)
{
DWORD dwRes = WNetFormatNetworkName(pszProvider, szName, szDisplayName, &dwSize, WNFMT_PLATFORM, 8 + 1 + 3);
if (dwRes == WN_SUCCESS)
lstrcpy(szName, szDisplayName);
}
}
return StringToStrRet(szName, pStrRet);
}
//
// resolve non-UNC share names (novell) to UNC style names
//
// returns:
// TRUE translated the name
// FALSE didn't translate (maybe error case)
//
// WARNING: If we use too much stack space then we will cause
// faults by over flowing the stack. Millennium #94818
BOOL CNetFolder::_GetPathForShare(LPCIDNETRESOURCE pidn, LPTSTR pszPath)
{
BOOL fRet = FALSE;
*pszPath = TEXT('\0');
LPTSTR pszAccessName = (LPTSTR)LocalAlloc(LPTR, sizeof(TCHAR) * MAX_PATH * 3);
if (pszAccessName)
{
LPTSTR pszRemoteName = pszAccessName + MAX_PATH;
LPTSTR pszProviderName = pszRemoteName + MAX_PATH;
NET_CopyResName(pidn, pszRemoteName, MAX_PATH);
if (NULL != _pszResName)
{
//
// Combine the folder name with the share name
// to create a UNC path.
//
// Borrow the pszProviderName buffer for a bit.
//
PathCombine(pszProviderName, _pszResName, pszRemoteName);
//
// To be safe: UNC prefix implies that name is available using FS access
// Theoretically it also should be routed to MPR, but it is late to do this
//
if (PathIsUNC(pszProviderName))
{
lstrcpy(pszPath, pszProviderName);
fRet = FALSE;
}
else
{
pszProviderName[0] = TEXT('\0');
}
}
if (!*pszPath)
{
// Check cache
ENTERCRITICAL;
if (lstrcmpi(g_szLastAttemptedJunctionName, pszRemoteName) == 0)
{
// cache hit
lstrcpy(pszPath, g_szLastResolvedJunctionName);
fRet = TRUE;
}
LEAVECRITICAL;
}
if (!*pszPath)
{
NETRESOURCE nr = {0};
DWORD err, dwRedir, dwResult;
DWORD cchAccessName;
nr.lpRemoteName = pszRemoteName;
nr.lpProvider = (LPTSTR) _GetProvider(pidn, NULL, pszProviderName, MAX_PATH);
nr.dwType = NET_GetType(pidn);
nr.dwUsage = NET_GetUsage(pidn);
nr.dwDisplayType = NET_GetDisplayType(pidn);
dwRedir = CONNECT_TEMPORARY;
// Prepare access name buffer and net resource request buffer
//
cchAccessName = MAX_PATH;
pszAccessName[0] = 0;
err = WNetUseConnection(NULL, &nr, NULL, NULL, dwRedir, pszAccessName, &cchAccessName, &dwResult);
if ((WN_SUCCESS != err) || !pszAccessName[0])
{
// perf idea: might be good to cache the last failed junction bind
// and early out on the next attempt. One slight problem this
// might encounter: what if we cache a failure, the user changes
// state to fix the problem, but we hit our failure cache...
//
lstrcpy(pszPath, pszRemoteName);
fRet = FALSE;
}
else
{
// Get the return name
lstrcpy(pszPath, pszAccessName);
fRet = TRUE;
// Update success cache entry
ENTERCRITICAL;
lstrcpy(g_szLastAttemptedJunctionName, pszRemoteName);
lstrcpy(g_szLastResolvedJunctionName, pszAccessName);
LEAVECRITICAL;
}
}
LocalFree(pszAccessName);
}
return fRet;
}
// in:
// pidn may be multi-level net resource pidl like
// [entire net] [provider] [server] [share] [... file sys]
// or [server] [share] [... file sys]
HRESULT CNetFolder::_GetPathForItem(LPCIDNETRESOURCE pidn, LPTSTR pszPath)
{
*pszPath = 0;
// loop down
for (; !ILIsEmpty((LPCITEMIDLIST)pidn) ; pidn = (LPCIDNETRESOURCE)_ILNext((LPCITEMIDLIST)pidn))
{
if (NET_GetFlags(pidn) & SHID_JUNCTION) // \\server\share or strike/sys
{
_GetPathForShare(pidn, pszPath);
break; // below this we don't know about any of the PIDLs
}
else
{
// if this is entire network then return the canonical name for
// this object.
if (NET_GetDisplayType(pidn) == RESOURCEDISPLAYTYPE_ROOT)
StrCpyN(pszPath, TEXT("EntireNetwork"), MAX_PATH);
else
NET_CopyResName(pidn, pszPath, MAX_PATH);
}
}
return *pszPath ? S_OK : E_NOTIMPL;
}
HRESULT CNetFolder::_GetPathForItemW(LPCIDNETRESOURCE pidn, LPWSTR pszPath)
{
return _GetPathForItem(pidn, pszPath);
}
// in:
// pidl
//
// takes the last items and create a folder for it, assuming the first section is the
// used to initialze. the riid and ppv are used to return an object.
//
HRESULT CNetFolder::_CreateFolderForItem(LPBC pbc, LPCITEMIDLIST pidl, LPCITEMIDLIST pidlTarget, LPCIDNETRESOURCE pidnForProvider, REFIID riid, void **ppv)
{
LPCITEMIDLIST pidlLast = ILFindLastID(pidl);
LPCIDNETRESOURCE pidn = NET_IsValidID(pidlLast);
if (!pidn)
return E_INVALIDARG;
HRESULT hr;
if (NET_IsRemoteFld(pidn))
{
// note: I think this is dead functionality. it was used in NT4 but we can't find
// the impl of this CLSID_Remote anymore...
IPersistFolder * ppf;
hr = SHCoCreateInstance(NULL, &CLSID_Remote, NULL, IID_PPV_ARG(IPersistFolder, &ppf));
if (SUCCEEDED(hr))
{
hr= ppf->Initialize(pidl);
if (SUCCEEDED(hr))
hr = ppf->QueryInterface(riid, ppv);
ppf->Release();
}
}
else if (NET_GetFlags(pidn) & SHID_JUNCTION) // \\server\share or strike/sys
{
PERSIST_FOLDER_TARGET_INFO * ppfti = (PERSIST_FOLDER_TARGET_INFO *) LocalAlloc(LPTR, sizeof(PERSIST_FOLDER_TARGET_INFO));
if (ppfti)
{
ppfti->pidlTargetFolder = (LPITEMIDLIST)pidlTarget;
_GetPathForItemW(pidn, ppfti->szTargetParsingName);
ppfti->csidl = -1;
ppfti->dwAttributes = FILE_ATTRIBUTE_DIRECTORY; // maybe add system?
hr = CFSFolder_CreateFolder(NULL, pbc, pidl, ppfti, riid, ppv);
LocalFree(ppfti);
}
else
hr = E_OUTOFMEMORY;
}
else
{
TCHAR szPath[MAX_PATH];
NET_CopyResName(pidn, szPath, ARRAYSIZE(szPath));
hr = _CreateInstance(pidl, pidlTarget, NET_GetDisplayType(pidn), pidnForProvider, szPath, riid, ppv);
}
return hr;
}
// get the provider for an item or the folder itself. since some items don't have the
// provider stored we fall back to the folder to get the provider in that case
//
// in:
// pidn item to get provider for. if NULL get provider for the folder
// pbc IBindCtx to get provider for. if NULL get provider from pidn or folder.
//
// returns:
// NULL no provider in the item or the folder
// non NULL address of passed in buffer
LPCTSTR CNetFolder::_GetProvider(LPCIDNETRESOURCE pidn, IBindCtx *pbc, LPTSTR pszProvider, UINT cchProvider)
{
// attempt to get the provider from the property bag
IPropertyBag *ppb;
if (pbc && SUCCEEDED(pbc->GetObjectParam(STR_PARSE_NETFOLDER_INFO, (IUnknown**)&ppb)))
{
HRESULT hr = SHPropertyBag_ReadStr(ppb, STR_PARSE_NETFOLDER_PROVIDERNAME, pszProvider, cchProvider);
ppb->Release();
if (SUCCEEDED(hr) && *pszProvider)
{
return pszProvider;
}
}
// from the IDLIST
if (pidn && NET_CopyProviderName(pidn, pszProvider, cchProvider))
return pszProvider;
// from our state
if (_pidnForProvider)
{
NET_CopyProviderName(_pidnForProvider, pszProvider, cchProvider);
return pszProvider;
}
*pszProvider = 0;
return NULL;
}
const NETPROVIDERS c_rgProviderMap[] =
{
{ TEXT("Microsoft Network"), HIWORD(WNNC_NET_LANMAN) },
{ TEXT("NetWare"), HIWORD(WNNC_NET_NETWARE) }
};
// construct a net idlist either copying the existing data from a pidl or
// from a NETRESOURCE structure
HRESULT CNetFolder::_CreateNetIDList(LPIDNETRESOURCE pidnIn,
LPCTSTR pszName, LPCTSTR pszProvider, LPCTSTR pszComment,
LPITEMIDLIST *ppidl)
{
LPBYTE pb;
UINT cbmkid = sizeof(IDNETRESOURCE) - sizeof(CHAR);
UINT cchName, cchProvider, cchComment, cbProviderType = 0;
LPIDNETRESOURCE pidn;
WORD wNetType = 0;
BOOL fUnicode = FALSE;
UINT cchAnsiName, cchAnsiProvider, cchAnsiComment;
CHAR szAnsiName[MAX_PATH], szAnsiProvider[MAX_PATH], szAnsiComment[MAX_PATH];
ASSERT(ppidl != NULL);
*ppidl = NULL;
if (!pszName)
pszName = c_szNULL; // For now put in an empty string...
if (pszProvider)
cbProviderType += sizeof(WORD);
// Win9x shipped with one set of provider name which are
// different on NT. Therefore lets convert the NT one to
// something that Win9x can understand.
if (pszProvider)
{
cbProviderType = sizeof(WORD);
DWORD dwType, dwRes = WNetGetProviderType(pszProvider, &dwType);
if (dwRes == WN_SUCCESS)
{
wNetType = HIWORD(dwType);
for (int i = 0; i < ARRAYSIZE(c_rgProviderMap); i++)
{
if (c_rgProviderMap[i].wNetType == wNetType)
{
pszProvider = c_rgProviderMap[i].lpName;
break;
}
}
}
}
// compute the string lengths ready to build an IDLIST
cchName = lstrlen(pszName)+1;
cchProvider = pszProvider ? lstrlen(pszProvider)+1 : 0;
cchComment = pszComment ? lstrlen(pszComment)+1 : 0;
cchAnsiName = 0;
cchAnsiProvider = 0;
cchAnsiComment = 0;
fUnicode = !DoesStringRoundTrip(pszName, szAnsiName, ARRAYSIZE(szAnsiProvider));
cchAnsiName = lstrlenA(szAnsiName)+1;
if (pszProvider)
{
fUnicode |= !DoesStringRoundTrip(pszProvider, szAnsiProvider, ARRAYSIZE(szAnsiProvider));
cchAnsiProvider = lstrlenA(szAnsiProvider)+1;
}
if (pszComment)
{
fUnicode |= !DoesStringRoundTrip(pszComment, szAnsiComment, ARRAYSIZE(szAnsiComment));
cchAnsiComment = lstrlenA(szAnsiComment)+1;
}
// allocate and fill the IDLIST header
cbmkid += cbProviderType+cchAnsiName + cchAnsiProvider + cchAnsiComment;
if (fUnicode)
cbmkid += (sizeof(WCHAR)*(cchName+cchProvider+cchComment));
pidn = (LPIDNETRESOURCE)_ILCreate(cbmkid + sizeof(USHORT));
if (!pidn)
return E_OUTOFMEMORY;
pidn->cb = (WORD)cbmkid;
pidn->bFlags = pidnIn->bFlags;
pidn->uType = pidnIn->uType;
pidn->uUsage = pidnIn->uUsage;
if (pszProvider)
pidn->uUsage |= NET_HASPROVIDER;
if (pszComment)
pidn->uUsage |= NET_HASCOMMENT;
pb = (LPBYTE) pidn->szNetResName;
//
// write the ANSI strings into the IDLIST
//
StrCpyA((PSTR) pb, szAnsiName);
pb += cchAnsiName;
if (pszProvider)
{
StrCpyA((PSTR) pb, szAnsiProvider);
pb += cchAnsiProvider;
}
if (pszComment)
{
StrCpyA((PSTR) pb, szAnsiComment);
pb += cchAnsiComment;
}
// if we are going to be UNICODE then lets write those strings also.
// Note that we must use unaligned string copies since the is no
// promse that the ANSI strings will have an even number of characters
// in them.
if (fUnicode)
{
pidn->uUsage |= NET_UNICODE;
ualstrcpyW((UNALIGNED WCHAR *)pb, pszName);
pb += cchName*sizeof(WCHAR);
if (pszProvider)
{
ualstrcpyW((UNALIGNED WCHAR *)pb, pszProvider);
pb += cchProvider*sizeof(WCHAR);
}
if (pszComment)
{
ualstrcpyW((UNALIGNED WCHAR *)pb, pszComment);
pb += cchComment*sizeof(WCHAR);
}
}
//
// and the trailing provider type
//
if (cbProviderType)
{
// Store the provider type
pb = (LPBYTE)pidn + pidn->cb - sizeof(WORD);
*((UNALIGNED WORD *)pb) = wNetType;
}
*ppidl = (LPITEMIDLIST)pidn;
return S_OK;
}
// wrapper for converting a NETRESOURCE into an IDLIST via _CreateNetPidl
HRESULT CNetFolder::_NetResToIDList(NETRESOURCE *pnr,
BOOL fKeepNullRemoteName,
BOOL fKeepProviderName,
BOOL fKeepComment,
LPITEMIDLIST *ppidl)
{
NETRESOURCE nr = *pnr;
LPITEMIDLIST pidl;
LPTSTR pszName, pszProvider, pszComment;
IDNETRESOURCE idn;
LPTSTR psz;
if (ppidl)
*ppidl = NULL;
switch (pnr->dwDisplayType)
{
case RESOURCEDISPLAYTYPE_NETWORK:
pszName = pnr->lpProvider;
break;
case RESOURCEDISPLAYTYPE_ROOT:
pszName =pnr->lpComment;
break;
default:
{
pszName = pnr->lpRemoteName;
if (!fKeepNullRemoteName && (!pszName || !*pszName))
return E_FAIL;
if (pszName && *pszName)
{
psz = (LPTSTR)SkipServerSlashes(pnr->lpRemoteName);
if ( *psz )
PathMakePretty(psz);
}
}
break;
}
pszProvider = fKeepProviderName ? nr.lpProvider:NULL;
pszComment = fKeepComment ? nr.lpComment:NULL;
idn.bFlags = (BYTE)(SHID_NET | (pnr->dwDisplayType & 0x0f));
idn.uType = (BYTE)(pnr->dwType & 0x0f);
idn.uUsage = (BYTE)(pnr->dwUsage & 0x0f);
// Is the current resource a share of some kind and not a container
if ((pnr->dwDisplayType == RESOURCEDISPLAYTYPE_SHARE || pnr->dwDisplayType == RESOURCEDISPLAYTYPE_SHAREADMIN) &&
!(pnr->dwUsage & RESOURCEUSAGE_CONTAINER))
{
// If so, remember to delegate children of this folder to FSFolder
idn.bFlags |= (BYTE)SHID_JUNCTION; // \\server\share type thing
}
HRESULT hr = _CreateNetIDList(&idn, pszName, pszProvider, pszComment, &pidl);
if (SUCCEEDED(hr))
{
if (ppidl)
*ppidl = pidl;
}
return hr;
}
HRESULT CNetFolder::_CreateEntireNetwork(LPITEMIDLIST *ppidl)
{
TCHAR szPath[MAX_PATH];
NETRESOURCE nr = {0};
// We need to add the Rest of network entry. This is psuedo
// bogus, as we should either always do it ourself or have
// MPR always do it, but here it goes...
LoadString(HINST_THISDLL, IDS_RESTOFNET, szPath, ARRAYSIZE(szPath));
nr.dwDisplayType = RESOURCEDISPLAYTYPE_ROOT;
nr.dwType = RESOURCETYPE_ANY;
nr.dwUsage = RESOURCEUSAGE_CONTAINER;
nr.lpComment = szPath;
return _NetResToIDList(&nr, FALSE, FALSE, FALSE, ppidl);
}
HRESULT CNetFolder::_CreateEntireNetworkFullIDList(LPITEMIDLIST *ppidl)
{
// CLSID_NetworkPlaces\EntireNetwork
return SHILCreateFromPath(TEXT("::{208D2C60-3AEA-1069-A2D7-08002B30309D}\\EntireNetwork"), ppidl, NULL);
}
//
// To be called back from within CDefFolderMenu
//
STDAPI CNetwork_DFMCallBackBG(IShellFolder *psf, HWND hwnd,
IDataObject *pdtobj, UINT uMsg,
WPARAM wParam, LPARAM lParam)
{
HRESULT hr = S_OK;
CNetFolder *pThis = FolderToNetFolder(psf);
if (NULL == pThis)
return E_UNEXPECTED;
switch(uMsg)
{
case DFM_MERGECONTEXTMENU_BOTTOM:
if (!(wParam & (CMF_VERBSONLY | CMF_DVFILE)))
{
CDefFolderMenu_MergeMenu(HINST_THISDLL, POPUP_PROPERTIES_BG, 0, (LPQCMINFO)lParam);
}
break;
case DFM_GETHELPTEXT:
LoadStringA(HINST_THISDLL, LOWORD(wParam) + IDS_MH_FSIDM_FIRST, (LPSTR)lParam, HIWORD(wParam));
break;
case DFM_GETHELPTEXTW:
LoadStringW(HINST_THISDLL, LOWORD(wParam) + IDS_MH_FSIDM_FIRST, (LPWSTR)lParam, HIWORD(wParam));
break;
case DFM_INVOKECOMMAND:
switch (wParam)
{
case FSIDM_PROPERTIESBG:
hr = SHPropertiesForPidl(hwnd, pThis->_pidl, (LPCTSTR)lParam);
break;
default:
// This is one of view menu items, use the default code.
hr = S_FALSE;
break;
}
break;
default:
hr = E_NOTIMPL;
break;
}
return hr;
}
//
// To be called back from within CDefFolderMenu
//
STDAPI CNetFolder::DFMCallBack(IShellFolder* psf, HWND hwnd,
IDataObject* pdtobj, UINT uMsg,
WPARAM wParam, LPARAM lParam)
{
HRESULT hr = S_OK;
switch(uMsg)
{
case DFM_MERGECONTEXTMENU:
if (pdtobj)
{
STGMEDIUM medium;
LPIDA pida;
LPQCMINFO pqcm = (LPQCMINFO)lParam;
UINT idCmdBase = pqcm->idCmdFirst; // must be called before merge
CDefFolderMenu_MergeMenu(HINST_THISDLL, POPUP_NETWORK_ITEM, 0, pqcm);
pida = DataObj_GetHIDA(pdtobj, &medium);
if (pida)
{
if (pida->cidl > 0)
{
LPIDNETRESOURCE pidn = (LPIDNETRESOURCE)IDA_GetIDListPtr(pida, 0);
// Only enable "connect" command if the first one is a share.
if (pidn)
{
ULONG rgf = 0;
if(NET_GetFlags(pidn) & SHID_JUNCTION &&
!SHRestricted(REST_NONETCONNECTDISCONNECT))
{
EnableMenuItem(pqcm->hmenu, idCmdBase + FSIDM_CONNECT,
MF_CHECKED | MF_BYCOMMAND);
}
}
}
HIDA_ReleaseStgMedium(pida, &medium);
}
}
break;
case DFM_GETHELPTEXT:
LoadStringA(HINST_THISDLL, LOWORD(wParam) + IDS_MH_FSIDM_FIRST, (LPSTR)lParam, HIWORD(wParam));
break;
case DFM_GETHELPTEXTW:
LoadStringW(HINST_THISDLL, LOWORD(wParam) + IDS_MH_FSIDM_FIRST, (LPWSTR)lParam, HIWORD(wParam));
break;
case DFM_INVOKECOMMAND:
switch(wParam)
{
case DFM_CMD_PROPERTIES:
hr = SHLaunchPropSheet(_PropertiesThreadProc, pdtobj, (LPCTSTR)lParam, psf, NULL);
break;
case DFM_CMD_LINK:
{
hr = S_FALSE; // do the default shortcut stuff
CNetFolder *pThis = FolderToNetFolder(psf);
if (pThis)
{
// net hood special case. in this case we want to create the shortuct
// in the net hood, not offer to put this on the desktop
IShellFolder2* psfFiles;
if (SUCCEEDED(pThis->v_GetFileFolder(&psfFiles)))
{
CFSFolder_CreateLinks(hwnd, psfFiles, pdtobj, (LPCTSTR)lParam, CMIC_MASK_FLAG_NO_UI);
hr = S_OK; // we created the links
}
}
}
break;
case FSIDM_CONNECT:
if (pdtobj)
{
STGMEDIUM medium;
LPIDA pida = DataObj_GetHIDA(pdtobj, &medium);
if (pida)
{
for (UINT i = 0; i < pida->cidl; i++)
{
LPIDNETRESOURCE pidn = (LPIDNETRESOURCE)IDA_GetIDListPtr(pida, i);
// Only execute "connect" on shares.
if (NET_GetFlags(pidn) & SHID_JUNCTION)
{
TCHAR szName[MAX_PATH];
LPTSTR pszName = NET_CopyResName(pidn, szName, ARRAYSIZE(szName));
DWORD err = SHStartNetConnectionDialog(hwnd, pszName, RESOURCETYPE_DISK);
DebugMsg(DM_TRACE, TEXT("CNet FSIDM_CONNECT (%s, %x)"), szName, err);
// events will get generated automatically
}
}
HIDA_ReleaseStgMedium(pida, &medium);
}
}
break;
default:
// This is one of view menu items, use the default code.
hr = S_FALSE;
break;
}
break;
default:
hr = E_NOTIMPL;
break;
}
return hr;
}
STDAPI CNetFolder::PrinterDFMCallBack(IShellFolder* psf, HWND hwnd,
IDataObject* pdtobj, UINT uMsg,
WPARAM wParam, LPARAM lParam)
{
HRESULT hr = S_OK;
switch(uMsg)
{
case DFM_MERGECONTEXTMENU:
//
// Returning S_FALSE indicates no need to get verbs from
// extensions.
//
hr = S_FALSE;
break;
// if anyone hooks our context menu, we want to be on top (Open)
case DFM_MERGECONTEXTMENU_TOP:
if (pdtobj)
{
LPQCMINFO pqcm = (LPQCMINFO)lParam;
// insert verbs
CDefFolderMenu_MergeMenu(HINST_THISDLL, POPUP_NETWORK_PRINTER, 0, pqcm);
SetMenuDefaultItem(pqcm->hmenu, 0, MF_BYPOSITION);
}
break;
case DFM_GETHELPTEXT:
LoadStringA(HINST_THISDLL, LOWORD(wParam) + IDS_MH_FSIDM_FIRST, (LPSTR)lParam, HIWORD(wParam));
break;
case DFM_GETHELPTEXTW:
LoadStringW(HINST_THISDLL, LOWORD(wParam) + IDS_MH_FSIDM_FIRST, (LPWSTR)lParam, HIWORD(wParam));
break;
case DFM_INVOKECOMMAND:
switch (wParam)
{
case DFM_CMD_PROPERTIES:
hr = SHLaunchPropSheet(_PropertiesThreadProc, pdtobj, (LPCTSTR)lParam, psf, NULL);
break;
case DFM_CMD_LINK:
// do the default create shortcut crap
return S_FALSE;
case FSIDM_OPENPRN:
case FSIDM_NETPRN_INSTALL:
{
STGMEDIUM medium;
LPIDA pida = DataObj_GetHIDA(pdtobj, &medium);
if (pida)
{
UINT action;
// set up the operation we are going to perform
switch (wParam)
{
case FSIDM_OPENPRN:
action = PRINTACTION_OPENNETPRN;
break;
case FSIDM_NETPRN_INSTALL:
action = PRINTACTION_NETINSTALL;
break;
default: // FSIDM_CONNECT_PRN
action = (UINT)-1;
break;
}
for (UINT i = 0; i < pida->cidl; i++)
{
LPIDNETRESOURCE pidn = (LPIDNETRESOURCE)IDA_GetIDListPtr(pida, i);
// Only execute command for a net print share
if (_IsPrintShare(pidn))
{
TCHAR szName[MAX_PATH];
NET_CopyResName(pidn,szName,ARRAYSIZE(szName));
SHInvokePrinterCommand(hwnd, action, szName, NULL, FALSE);
}
} // for (i...
HIDA_ReleaseStgMedium(pida, &medium);
} // if (medium.hGlobal)
break;
} // case ID_NETWORK_PRINTER_INSTALL, FSIDM_CONNECT_PRN
default:
hr = E_FAIL;
break;
} // switch(wparam)
break;
default:
hr = E_NOTIMPL;
break;
}
return hr;
}
//
// REVIEW: Almost identical code in fstreex.c
//
DWORD CALLBACK CNetFolder::_PropertiesThreadProc(void *pv)
{
PROPSTUFF* pps = (PROPSTUFF *)pv;
ULONG_PTR dwCookie = 0;
ActivateActCtx(NULL, &dwCookie);
CNetFolder *pThis = FolderToNetFolder(pps->psf);
if (pThis)
{
STGMEDIUM medium;
LPIDA pida = DataObj_GetHIDA(pps->pdtobj, &medium);
if (pida)
{
// Yes, do context menu.
HKEY ahkeys[NKID_COUNT];
LPCIDNETRESOURCE pnid = (LPCIDNETRESOURCE)IDA_GetIDListPtr(pida, 0);
if (pnid)
{
HRESULT hr = pThis->_OpenKeys(pnid, ahkeys);
if (SUCCEEDED(hr))
{
LPTSTR pszCaption = SHGetCaption(medium.hGlobal);
SHOpenPropSheet(pszCaption, ahkeys, ARRAYSIZE(ahkeys),
&CLSID_ShellNetDefExt,
pps->pdtobj, NULL, pps->pStartPage);
if (pszCaption)
SHFree(pszCaption);
SHRegCloseKeys(ahkeys, ARRAYSIZE(ahkeys));
}
}
HIDA_ReleaseStgMedium(pida, &medium);
}
}
return S_OK;
}
STDAPI CNetFolder::_AttributesCallback(IShellFolder2* psf, LPCITEMIDLIST pidl, ULONG* prgfInOut)
{
LPCIDNETRESOURCE pidn = (LPCIDNETRESOURCE)pidl;
ULONG rgfOut = SFGAO_CANLINK | SFGAO_HASPROPSHEET | SFGAO_HASSUBFOLDER |
SFGAO_FOLDER | SFGAO_FILESYSANCESTOR | SFGAO_STORAGEANCESTOR;
if (NET_GetFlags(pidn) & SHID_JUNCTION)
{
if ((NET_GetType(pidn) == RESOURCETYPE_DISK) ||
(NET_GetType(pidn) == RESOURCETYPE_ANY))
rgfOut |= SFGAO_FILESYSTEM | SFGAO_DROPTARGET | SFGAO_CANCOPY | SFGAO_CANMOVE | SFGAO_STORAGE;
else
rgfOut &= ~(SFGAO_FILESYSANCESTOR | SFGAO_STORAGEANCESTOR);
}
if (_IsPrintShare(pidn))
{
rgfOut |= SFGAO_DROPTARGET; // for drag and drop printing
rgfOut &= ~(SFGAO_FILESYSANCESTOR | SFGAO_STORAGEANCESTOR | SFGAO_FILESYSTEM | SFGAO_FOLDER | SFGAO_HASSUBFOLDER);
}
if (NET_IsRemoteFld(pidn))
{
rgfOut &= ~(SFGAO_FILESYSANCESTOR | SFGAO_STORAGEANCESTOR | SFGAO_FILESYSTEM);
}
*prgfInOut = rgfOut;
return S_OK;
}
// This is only used by the CNetRootFolder subclass, but because we can only QI for
// CLSID_NetFldr, and we can't access protected members of any CNetFolder instance
// from a member function of CNetRootFolder, we'll make it belong to CNetFolder
HRESULT CALLBACK CNetFolder::_AttributesCallbackRoot(IShellFolder2* psf, LPCITEMIDLIST pidl, ULONG* prgfInOut)
{
HRESULT hr;
CNetFolder* pNetF = FolderToNetFolder(psf);
if (pNetF)
{
if (NET_IsValidID(pidl))
{
hr = pNetF->CNetFolder::GetAttributesOf(1, &pidl, prgfInOut);
}
else
{
IShellFolder2* psfFiles;
hr = pNetF->v_GetFileFolder(&psfFiles);
if (SUCCEEDED(hr))
hr = psfFiles->GetAttributesOf(1, &pidl, prgfInOut);
}
}
else
hr = E_FAIL;
return hr;
}
// this is called by netfind.c
STDAPI CNetwork_EnumSearches(IShellFolder2* psf2, IEnumExtraSearch **ppenum)
{
*ppenum = NULL;
CNetFolder* pNetF = FolderToNetFolder(psf2);
return pNetF ? pNetF->EnumSearches(ppenum) : E_INVALIDARG;
}
// given the resulting ppidl and a pszRest continue to parse through and add in the remainder
// of the file system path.
HRESULT CNetFolder::_ParseRest(LPBC pbc, LPCWSTR pszRest, LPITEMIDLIST* ppidl, DWORD* pdwAttributes)
{
HRESULT hr = S_OK;
// skip leading \ if there is one present
if (pszRest && pszRest[0] == L'\\')
pszRest++;
if (pszRest && pszRest[0])
{
// need to QI to get the agregated case
IShellFolder* psfBind;
hr = QueryInterface(IID_PPV_ARG(IShellFolder, &psfBind));
if (SUCCEEDED(hr))
{
// pass down to pick off stuff below including regitems and file names
IShellFolder* psfSub;
hr = psfBind->BindToObject(*ppidl, NULL, IID_PPV_ARG(IShellFolder, &psfSub));
if (SUCCEEDED(hr))
{
LPITEMIDLIST pidlSubDir;
hr = psfSub->ParseDisplayName(NULL, pbc, (LPWSTR)pszRest, NULL, &pidlSubDir, pdwAttributes);
if (SUCCEEDED(hr))
{
hr = SHILAppend(pidlSubDir, ppidl);
}
psfSub->Release();
}
psfBind->Release();
}
}
else
{
if (pdwAttributes)
{
LPCITEMIDLIST apidlLast[1] = { ILFindLastID(*ppidl) };
hr = GetAttributesOf(1, apidlLast, pdwAttributes);
}
}
return hr;
}
// generate an IDLIST from the NETRESOURCESTRUCTURE we have by
// walking up its parents trying to determine where we
// are in the namespace
BOOL _GetParentResource(NETRESOURCE *pnr, DWORD *pdwbuf)
{
if ((pnr->dwDisplayType == RESOURCEDISPLAYTYPE_ROOT) ||
(WN_SUCCESS != WNetGetResourceParent(pnr, pnr, pdwbuf)))
{
return FALSE;
}
return TRUE;
}
HRESULT CNetFolder::_NetResToIDLists(NETRESOURCE *pnr, DWORD dwbuf, LPITEMIDLIST *ppidl)
{
HRESULT hr = S_OK;
do
{
LPITEMIDLIST pidlT;
hr = _NetResToIDList(pnr, TRUE, TRUE, TRUE, &pidlT);
if (SUCCEEDED(hr))
{
hr = SHILPrepend(pidlT, ppidl); // NOTE: SHILPrepend frees on failure
}
}
while (SUCCEEDED(hr) && _GetParentResource(pnr, &dwbuf));
return hr;
}
// get the parsable network name from the object
LPTSTR CNetFolder::_GetNameForParsing(LPCWSTR pwszName, LPTSTR pszBuffer, INT cchBuffer, LPTSTR *ppszRegItem)
{
LPTSTR pszRegItem = NULL;
INT cSlashes = 0;
*ppszRegItem = NULL;
SHUnicodeToTChar(pwszName, pszBuffer, cchBuffer);
// remove the trailing \ if there is one, NTLanMan barfs if we pass a string containing it
INT cchPath = lstrlen(pszBuffer)-1;
if (cchPath > 2)
{
// We don't need to call CharPrev if cchPath <= 2.
// Calling CharPrev is expensive.
LPTSTR lpTmp = CharPrev(pszBuffer, pszBuffer + cchPath + 1);
if (*lpTmp == TEXT('\\'))
*lpTmp = TEXT('\0');
}
// lets walk the name, look for \:: squence to signify the start of a regitem name,
// and if the number of slashes is > 2 then we should bail
LPTSTR pszUNC = pszBuffer+2;
while (pszUNC && *pszUNC && (cSlashes < 2))
{
if ((pszUNC[0] == TEXT('\\')) &&
(pszUNC[1] == TEXT(':')) && (pszUNC[2] == TEXT(':')))
{
*ppszRegItem = pszUNC;
break;
}
pszUNC = StrChr(pszUNC+1, TEXT('\\'));
cSlashes++;
}
return pszUNC;
}
HRESULT CNetFolder::_ParseNetName(HWND hwnd, LPBC pbc,
LPCWSTR pwszName, ULONG* pchEaten,
LPITEMIDLIST *ppidl, DWORD *pdwAttrib)
{
HRESULT hr;
struct _NRTEMP
{
NETRESOURCE nr;
TCHAR szBuffer[1024];
} nrOut = { 0 };
TCHAR szPath[MAX_PATH];
DWORD dwres, dwbuf = sizeof(nrOut.szBuffer);
LPTSTR pszServerShare = NULL;
LPTSTR pszRestOfName = NULL;
LPTSTR pszFakeRestOfName = NULL;
LPTSTR pszRegItem = NULL;
// validate the name before we start cracking it...
pszFakeRestOfName = _GetNameForParsing(pwszName, szPath, ARRAYSIZE(szPath), &pszRegItem);
NETRESOURCE nr = { 0 };
nr.lpRemoteName = szPath;
nr.dwType = RESOURCETYPE_ANY;
TCHAR szProviderTemp[256];
nr.lpProvider = (LPTSTR)_GetProvider(NULL, pbc, szProviderTemp, ARRAYSIZE(szProviderTemp));
dwres = WNetGetResourceInformation(&nr, &nrOut.nr, &dwbuf, &pszRestOfName);
if (WN_SUCCESS != dwres)
{
TCHAR cT;
LPTSTR pszTemp;
// truncate the string at the \\server\share to try and parse the name,
// note at this point if MPR resolves the alias on a Novel server this could
// get very confusing (eg. \\strike\foo\bah may resolve to \\string\bla,
// yet our concept of what pszRestOfName will be wrong!
if (pszFakeRestOfName)
{
cT = *pszFakeRestOfName;
*pszFakeRestOfName = TEXT('\0');
}
dwres = WNetGetResourceInformation(&nr, &nrOut.nr, &dwbuf, &pszTemp);
if (dwres != WN_SUCCESS)
{
// we failed to get a net connection using the truncated string,
// so lets try and use a new connect (eg. prompt for creds)
// NOTE: shouldn't we only be doing this if its an access denied type error?
dwres = WNetUseConnection(hwnd, &nr, NULL, NULL, hwnd ? CONNECT_INTERACTIVE:0, NULL, NULL, NULL);
if (dwres == WN_SUCCESS)
{
dwres = WNetGetResourceInformation(&nr, &nrOut, &dwbuf, &pszTemp);
}
}
if (pszFakeRestOfName)
*pszFakeRestOfName = cT;
pszRestOfName = pszFakeRestOfName;
}
if (WN_SUCCESS == dwres)
{
WCHAR wszRestOfName[MAX_PATH] = { 0 };
if (pszRestOfName)
SHTCharToUnicode(pszRestOfName, wszRestOfName, ARRAYSIZE(wszRestOfName));
// assume we are truncating at the regitem and parsing through
if (pszRegItem)
pszRestOfName = pszRegItem;
// attempt to convert the NETRESOURCE to a string to IDLISTS by walking the
// parents, then add in Entire Network
hr = _NetResToIDLists(&nrOut.nr, dwbuf, ppidl);
if (SUCCEEDED(hr))
{
LPITEMIDLIST pidlT;
hr = _CreateEntireNetwork(&pidlT);
if (SUCCEEDED(hr))
{
hr = SHILPrepend(pidlT, ppidl); // NOTE: SHILPrepend frees on failure
}
}
// if we have a local string then lets continue to parse it by binding to
// its parent folder, otherwise we just want to return the attributes
if (SUCCEEDED(hr))
{
if (SUCCEEDED(DisplayNameOf(this, *ppidl, SHGDN_FORPARSING, szPath, ARRAYSIZE(szPath))))
{
NPTRegisterNameToPidlTranslation(szPath, *ppidl); // no _ILNext b/c this is relative to the Net Places folder
}
hr = _ParseRest(pbc, wszRestOfName, ppidl, pdwAttrib);
}
}
else
{
hr = HRESULT_FROM_WIN32(dwres);
}
return hr;
}
//
// simple name parsing for the network paths. this makes big assumptions about the
// \\server\share format we are given, and the type of IDLISTs to return.
//
HRESULT CNetFolder::_AddUnknownIDList(DWORD dwDisplayType, LPITEMIDLIST *ppidl)
{
NETRESOURCE nr = { 0 };
nr.dwScope = RESOURCE_GLOBALNET;
nr.dwDisplayType = dwDisplayType;
nr.dwUsage = RESOURCEUSAGE_CONTAINER;
nr.lpRemoteName = TEXT("\0"); // null name means fake item
LPITEMIDLIST pidlT;
HRESULT hr = _NetResToIDList(&nr, TRUE, FALSE, FALSE, &pidlT);
if (SUCCEEDED(hr))
{
hr = SHILAppend(pidlT, ppidl);
if (FAILED(hr))
ILFree(pidlT);
}
return hr;
}
HRESULT CNetFolder::_ParseSimple(LPBC pbc, LPWSTR pszName, LPITEMIDLIST* ppidl, DWORD* pdwAttributes)
{
HRESULT hr = S_OK;
NETRESOURCE nr = {0};
LPWSTR pszSlash;
LPITEMIDLIST pidlT;
USES_CONVERSION;
*ppidl = NULL;
// create the entire network IDLIST, provider and domain elements
hr = _CreateEntireNetwork(ppidl);
if (SUCCEEDED(hr))
hr = _AddUnknownIDList(RESOURCEDISPLAYTYPE_NETWORK, ppidl);
if (SUCCEEDED(hr))
hr = _AddUnknownIDList(RESOURCEDISPLAYTYPE_DOMAIN, ppidl);
// create the server IDLIST
if (SUCCEEDED(hr))
{
pszSlash = StrChrW(pszName+2, L'\\');
if (pszSlash)
*pszSlash = L'\0';
nr.dwScope = RESOURCE_GLOBALNET;
nr.dwDisplayType = RESOURCEDISPLAYTYPE_SERVER;
nr.dwType = RESOURCETYPE_DISK;
nr.dwUsage = RESOURCEUSAGE_CONTAINER;
nr.lpRemoteName = W2T(pszName);
hr = _NetResToIDList(&nr, FALSE, FALSE, FALSE, &pidlT);
if (SUCCEEDED(hr))
hr = SHILAppend(pidlT, ppidl);
if (pszSlash)
*pszSlash = L'\\';
// if we have a trailing \ then lets add in the share part of the IDLIST
if (SUCCEEDED(hr) && pszSlash)
{
pszSlash = StrChrW(pszSlash+1, L'\\');
if (pszSlash)
*pszSlash = L'\0';
nr.dwDisplayType = RESOURCEDISPLAYTYPE_SHARE;
nr.dwUsage = RESOURCEUSAGE_CONNECTABLE;
nr.lpRemoteName = W2T(pszName);
hr = _NetResToIDList(&nr, FALSE, FALSE, FALSE, &pidlT);
if (SUCCEEDED(hr))
hr = SHILAppend(pidlT, ppidl);
if (pszSlash)
*pszSlash = L'\\';
}
}
if (SUCCEEDED(hr))
{
hr = _ParseRest(pbc, pszSlash, ppidl, pdwAttributes);
}
return hr;
}
// try parsing out the EntireNet or localised version. if we find that object then try and
// parse through that to the regitems or other objects which live below. this inturn
// will cause an instance of CNetFolder to be created to generate the other parsing names.
//
// returns:
// S_FALSE - not rest of net, try something else
// S_OK - was rest of net, use this
// FAILED(hr) - error result, return
HRESULT CNetRootFolder::_TryParseEntireNet(HWND hwnd, LPBC pbc, WCHAR *pwszName, LPITEMIDLIST *ppidl, DWORD *pdwAttributes)
{
HRESULT hr = S_FALSE; // skip, not rest of net
*ppidl = NULL;
if (!PathIsUNCW(pwszName))
{
const WCHAR szEntireNetwork[] = L"EntireNetwork";
WCHAR szRestOfNet[128];
INT cchRestOfNet = LoadStringW(HINST_THISDLL, IDS_RESTOFNET, szRestOfNet, ARRAYSIZE(szRestOfNet));
BOOL fRestOfNet = !StrCmpNIW(szRestOfNet, pwszName, cchRestOfNet);
if (!fRestOfNet && !StrCmpNIW(szEntireNetwork, pwszName, ARRAYSIZE(szEntireNetwork)-1))
{
fRestOfNet = TRUE;
cchRestOfNet = ARRAYSIZE(szEntireNetwork)-1;
}
if (fRestOfNet)
{
hr = _CreateEntireNetwork(ppidl);
if (SUCCEEDED(hr))
{
if (pdwAttributes)
{
GetAttributesOf(1, (LPCITEMIDLIST *)ppidl, pdwAttributes);
}
hr = S_OK;
}
//
// if we find extra stuff after the name then lets bind and continue the parsing
// from there on. this is needed so the we can access regitems burried inside
// entire net.
//
// eg: EntireNetwork\\::{clsid}
//
if (SUCCEEDED(hr) &&
(pwszName[cchRestOfNet] == L'\\') && pwszName[cchRestOfNet+1])
{
IShellFolder *psfRestOfNet;
hr = BindToObject(*ppidl, NULL, IID_PPV_ARG(IShellFolder, &psfRestOfNet));
if (SUCCEEDED(hr))
{
LPITEMIDLIST pidl;
hr = psfRestOfNet->ParseDisplayName(hwnd, pbc, pwszName+cchRestOfNet+1, NULL, &pidl, pdwAttributes);
if (SUCCEEDED(hr))
{
hr = SHILAppend(pidl, ppidl);
}
psfRestOfNet->Release();
}
}
}
}
return hr;
}
// CNetRootFolder::ParseDisplayname
// - swtich based on the file system context to see if we need to do a simple parse or not,
// - check for "EntireNet" and delegate parsing as required.
STDMETHODIMP CNetRootFolder::ParseDisplayName(HWND hwnd, LPBC pbc, WCHAR* pszName, ULONG* pchEaten, LPITEMIDLIST* ppidl, DWORD* pdwAttributes)
{
if (!ppidl)
return E_INVALIDARG;
*ppidl = NULL;
if (!pszName)
return E_INVALIDARG;
HRESULT hr = _TryParseEntireNet(hwnd, pbc, pszName, ppidl, pdwAttributes);
if (hr == S_FALSE)
{
if (PathIsUNCW(pszName))
{
LPCITEMIDLIST pidlMapped;
LPTSTR pszRest = NPTMapNameToPidl(pszName, &pidlMapped);
if (pidlMapped)
{
hr = SHILClone(pidlMapped, ppidl);
if (SUCCEEDED(hr))
{
hr = _ParseRest(pbc, pszRest, ppidl, pdwAttributes);
}
}
else
{
if (S_OK == SHIsFileSysBindCtx(pbc, NULL))
{
hr = _ParseSimple(pbc, pszName, ppidl, pdwAttributes);
}
else
{
hr = _ParseNetName(hwnd, pbc, pszName, pchEaten, ppidl, pdwAttributes);
}
}
}
else
{
hr = HRESULT_FROM_WIN32(ERROR_BAD_NET_NAME);
}
if ((HRESULT_FROM_WIN32(ERROR_BAD_NET_NAME) == hr))
{
IShellFolder2 *psfFiles;
if (SUCCEEDED(v_GetFileFolder(&psfFiles)))
{
hr = psfFiles->ParseDisplayName(hwnd, pbc, pszName, pchEaten, ppidl, pdwAttributes);
}
}
}
if (FAILED(hr))
{
ILFree(*ppidl);
*ppidl = NULL;
}
return hr;
}
STDMETHODIMP CNetRootFolder::EnumObjects(HWND hwnd, DWORD grfFlags, IEnumIDList** ppenum)
{
DWORD dwRemote = RMF_GETLINKENUM;
HANDLE hEnum = NULL;
// Do we enumerate the workgroup?
if (!SHRestricted(REST_ENUMWORKGROUP))
{
// Don't enumerate the workgroup, if the restriction says so
dwRemote |= RMF_FAKENETROOT;
// Check the WNet policy to see if we should be showing the
// entire net object. If not, mark it as shown so that the
// enumerator doesn't return it.
if (SHRestricted(REST_NOENTIRENETWORK))
dwRemote |= RMF_ENTIRENETSHOWN;
}
// if we are not faking the net root then lets call _OpenEnum, otherwise lets ignore
if (!(dwRemote & RMF_FAKENETROOT))
{
DWORD err = _OpenEnum(hwnd, grfFlags, NULL, &hEnum);
// Always add the remote folder to the 'hood
if (WN_SUCCESS != err)
{
// Yes; still show remote anyway (only)
dwRemote |= RMF_SHOWREMOTE;
}
else
{
// No; allow everything to be enumerated in the 'hood.
dwRemote |= RMF_CONTEXT;
}
}
HRESULT hr = Create_NetFolderEnum(this, grfFlags, dwRemote, hEnum, ppenum);
if (FAILED(hr) && hEnum)
{
WNetCloseEnum(hEnum);
}
return hr;
}
STDMETHODIMP CNetRootFolder::BindToObject(LPCITEMIDLIST pidl, LPBC pbc, REFIID riid, void **ppv)
{
HRESULT hr;
if (NET_IsValidID(pidl))
hr = CNetFolder::BindToObject(pidl, pbc, riid, ppv);
else
{
IShellFolder2* psfFiles;
hr = v_GetFileFolder(&psfFiles);
if (SUCCEEDED(hr))
hr = psfFiles->BindToObject(pidl, pbc, riid, ppv);
}
return hr;
}
STDMETHODIMP CNetRootFolder::CompareIDs(LPARAM iCol, LPCITEMIDLIST pidl1, LPCITEMIDLIST pidl2)
{
HRESULT hr = E_INVALIDARG;
// First obtain the collate type of the pidls and their respective
// collate order.
LONG iColateType1 = _GetFilePIDLType(pidl1);
LONG iColateType2 = _GetFilePIDLType(pidl2);
if (iColateType1 == iColateType2)
{
// pidls are of same type.
if (iColateType1 == _HOOD_COL_FILE) // two file system pidls
{
IShellFolder2* psfFiles;
if (SUCCEEDED(v_GetFileFolder(&psfFiles)))
{
if (0 == (iCol & SHCIDS_COLUMNMASK))
{
// special case this for perf, this is the name compare
hr = psfFiles->CompareIDs(iCol, pidl1, pidl2);
}
else
{
SHCOLUMNID scid;
MapColumnToSCID((UINT)iCol & SHCIDS_COLUMNMASK, &scid);
int iRet = CompareBySCID(psfFiles, &scid, pidl1, pidl2);
hr = ResultFromShort(iRet);
}
}
}
else
{
// pidls same and are not of type file,
// so both must be a type understood
// by the CNetwork class - pass on to compare.
hr = CNetFolder::CompareIDs(iCol, pidl1, pidl2);
}
}
else
{
// ensure that entire network ends up at the head of the list
LPCIDNETRESOURCE pidn1 = NET_IsValidID(pidl1);
LPCIDNETRESOURCE pidn2 = NET_IsValidID(pidl2);
if ((pidn1 && (NET_GetDisplayType(pidn1) == RESOURCEDISPLAYTYPE_ROOT)) ||
(pidn2 && (NET_GetDisplayType(pidn2) == RESOURCEDISPLAYTYPE_ROOT)))
{
if (iColateType1 == _HOOD_COL_FILE)
return ResultFromShort(1);
else
return ResultFromShort(-1);
}
// pidls are not of same type, so have already been correctly
// collated (consequently, sorting is first by type and
// then by subfield).
hr = ResultFromShort(((iColateType2 - iColateType1) > 0) ? 1 : -1);
}
return hr;
}
STDMETHODIMP CNetRootFolder::CreateViewObject(HWND hwnd, REFIID riid, void **ppv)
{
ASSERT(ILIsEqual(_pidl, (LPCITEMIDLIST)&c_idlNet));
if (IsEqualIID(riid, IID_IDropTarget))
{
return CNetRootDropTarget_CreateInstance(hwnd, _pidl, (IDropTarget**) ppv);
}
return CNetFolder::CreateViewObject(hwnd, riid, ppv);
}
STDMETHODIMP CNetRootFolder::GetAttributesOf(UINT cidl, LPCITEMIDLIST *apidl, ULONG* prgfInOut)
{
HRESULT hr;
if (IsSelf(cidl, apidl))
{
// The user can rename links in the hood.
hr = CNetFolder::GetAttributesOf(cidl, apidl, prgfInOut);
*prgfInOut |= SFGAO_CANRENAME;
}
else
{
hr = GetAttributesCallback(SAFECAST(this, IShellFolder2*), cidl, apidl, prgfInOut, _AttributesCallbackRoot);
}
return hr;
}
STDMETHODIMP CNetRootFolder::GetDisplayNameOf(LPCITEMIDLIST pidl, DWORD dwFlags, STRRET* pStrRet)
{
HRESULT hr;
if (NET_IsValidID(pidl) || IsSelf(1, &pidl))
{
hr = CNetFolder::GetDisplayNameOf(pidl, dwFlags, pStrRet);
}
else
{
IShellFolder2* psfFiles;
hr = v_GetFileFolder(&psfFiles);
if (SUCCEEDED(hr))
hr = psfFiles->GetDisplayNameOf(pidl, dwFlags, pStrRet);
}
return hr;
}
STDMETHODIMP CNetRootFolder::SetNameOf(HWND hwnd, LPCITEMIDLIST pidl, LPCOLESTR lpszName,
DWORD dwRes, LPITEMIDLIST* ppidl)
{
HRESULT hr;
if (NET_IsValidID(pidl))
{
hr = CNetFolder::SetNameOf(hwnd, pidl, lpszName, dwRes, ppidl);
}
else
{
IShellFolder2* psfFiles;
hr = v_GetFileFolder(&psfFiles);
if (SUCCEEDED(hr))
hr = psfFiles->SetNameOf(hwnd, pidl, lpszName, dwRes, ppidl);
}
return hr;
}
STDMETHODIMP CNetRootFolder::GetUIObjectOf(HWND hwnd, UINT cidl, LPCITEMIDLIST* apidl,
REFIID riid, UINT* prgfInOut, void **ppv)
{
HRESULT hr = E_INVALIDARG;
LPCIDNETRESOURCE pidn = cidl ? NET_IsValidID(apidl[0]) : NULL;
BOOL fStriped = FALSE;
*ppv = NULL;
if (pidn)
{
fStriped = _MakeStripToLikeKinds(&cidl, &apidl, TRUE);
if (IsEqualIID(riid, IID_IContextMenu))
{
HKEY ahkeys[NKID_COUNT];
hr = _OpenKeys(pidn, ahkeys);
if (SUCCEEDED(hr))
{
IShellFolder* psfOuter;
hr = QueryInterface(IID_PPV_ARG(IShellFolder, &psfOuter));
if (SUCCEEDED(hr))
{
hr = CDefFolderMenu_Create2(_pidl, hwnd, cidl, apidl,
psfOuter, _GetCallbackType(pidn),
ARRAYSIZE(ahkeys), ahkeys, (IContextMenu**) ppv);
psfOuter->Release();
}
SHRegCloseKeys(ahkeys, ARRAYSIZE(ahkeys));
}
}
else
hr = CNetFolder::GetUIObjectOf(hwnd, cidl, apidl, riid, prgfInOut, ppv);
}
else
{
fStriped = _MakeStripToLikeKinds(&cidl, &apidl, FALSE);
IShellFolder2* psfFiles;
hr = v_GetFileFolder(&psfFiles);
if (SUCCEEDED(hr))
hr = psfFiles->GetUIObjectOf(hwnd, cidl, apidl, riid, prgfInOut, ppv);
}
if (fStriped)
LocalFree((HLOCAL)apidl);
return hr;
}
STDMETHODIMP CNetRootFolder::GetClassID(CLSID* pCLSID)
{
*pCLSID = CLSID_NetworkPlaces;
return S_OK;
}
STDMETHODIMP CNetRootFolder::Initialize(LPCITEMIDLIST pidl)
{
ASSERT(ILIsEqual(pidl, (LPCITEMIDLIST)&c_idlNet));
ASSERT(AssertIsIDListInNameSpace(pidl, &CLSID_NetworkPlaces) && ILIsEmpty(_ILNext(pidl)));
// Only allow the Net root on the desktop
// Don't initialize more than once; we are a singleton object.
// This is theoretically redundant with the InterlockedCompareExchange
// below, but redundant reinitialization is by far the common case
// so we'll optimize it.
if (_pidl)
return S_OK;
LPITEMIDLIST pidlNew;
HRESULT hr = SHILClone(pidl, &pidlNew);
if (SUCCEEDED(hr))
{
if (SHInterlockedCompareExchange((void**)&_pidl, pidlNew, 0))
{
// Some other thread raced with us, throw away our copy
ILFree(pidlNew);
}
}
return hr;
}
LONG CNetFolder::_GetFilePIDLType(LPCITEMIDLIST pidl)
{
if (NET_IsValidID(pidl))
{
if (NET_IsRemoteFld((LPIDNETRESOURCE)pidl))
{
return _HOOD_COL_REMOTE;
}
if (NET_GetDisplayType((LPIDNETRESOURCE)pidl) == RESOURCEDISPLAYTYPE_ROOT)
{
return _HOOD_COL_RON;
}
return _HOOD_COL_NET;
}
return _HOOD_COL_FILE;
}
/* This function adds a provider name to an IDLIST that doesn't already have one. */
/* A new IDLIST pointer is returned; the old pointer is no longer valid. */
LPITEMIDLIST CNetFolder::_AddProviderToPidl(LPITEMIDLIST pidl, LPCTSTR lpProvider)
{
LPIDNETRESOURCE pidn = (LPIDNETRESOURCE)pidl;
if (!NET_FHasProvider(pidn))
{
LPITEMIDLIST pidlres;
TCHAR szName[MAX_PATH], szComment[MAX_PATH];
// construct a new IDLIST preserving the name, comment and other information
NET_CopyResName(pidn, szName, ARRAYSIZE(szName));
NET_CopyComment(pidn, szComment, ARRAYSIZE(szComment));
HRESULT hr = _CreateNetIDList(pidn, szName, lpProvider, szComment[0] ? szComment:NULL, &pidlres);
if (SUCCEEDED(hr) && !ILIsEmpty(_ILNext(pidl)))
{
LPITEMIDLIST pidlT;
hr = SHILCombine(pidlres, _ILNext(pidl), &pidlT);
if (SUCCEEDED(hr))
{
ILFree(pidlres);
pidlres = pidlT;
}
}
// if we have a result, free the old PIDL and return the new
if (SUCCEEDED(hr))
{
ILFree(pidl);
pidl = pidlres;
}
}
return pidl;
}
BOOL CNetFolder::_MakeStripToLikeKinds(UINT *pcidl, LPCITEMIDLIST **papidl, BOOL fNetObjects)
{
BOOL bRet = FALSE;
LPITEMIDLIST *apidl = (LPITEMIDLIST*)*papidl;
int cidl = *pcidl;
for (int i = 0; i < cidl; i++)
{
if ((NET_IsValidID(apidl[i]) != NULL) != fNetObjects)
{
LPCITEMIDLIST *apidlHomo = (LPCITEMIDLIST *)LocalAlloc(LPTR, sizeof(*apidlHomo) * cidl);
if (!apidlHomo)
return FALSE;
int cpidlHomo = 0;
for (i = 0; i < cidl; i++)
{
if ((NET_IsValidID(apidl[i]) != NULL) == fNetObjects)
apidlHomo[cpidlHomo++] = apidl[i];
}
// Setup to use the stripped version of the pidl array...
*pcidl = cpidlHomo;
*papidl = apidlHomo;
bRet = TRUE;
}
}
return bRet;
}
HRESULT CNetRootFolder::v_GetFileFolder(IShellFolder2 **psf)
{
HRESULT hr = SHCacheTrackingFolder((LPCITEMIDLIST)&c_idlNet, CSIDL_NETHOOD, &_psfFiles);
*psf = _psfFiles;
return hr;
}
//
// pmedium and pformatetcIn == NULL if we are handling QueryGetData
//
HRESULT CNetData::GetHDrop(FORMATETC *pformatetcIn, STGMEDIUM *pmedium)
{
HRESULT hr = E_INVALIDARG; // assume error
STGMEDIUM medium;
LPIDA pida = DataObj_GetHIDA(this, &medium);
if (pida)
{
// Get the first one to see the type.
LPCIDNETRESOURCE pidn = (LPCIDNETRESOURCE)IDA_GetIDListPtr(pida, 0);
if (NULL == pidn)
hr = E_FAIL;
if (pidn && (NET_GetFlags(pidn) & SHID_JUNCTION) &&
(NET_GetType(pidn) == RESOURCETYPE_DISK))
{
// Get HDrop only if we are handling IDataObject::GetData (pmedium != NULL)
if (pmedium)
{
// We have non-null FORMATETC and STGMEDIUM - get the HDrop
hr = CFSIDLData::GetHDrop(pformatetcIn, pmedium);
}
else
{
hr = S_OK; // We were handling QueryGetData
}
}
HIDA_ReleaseStgMedium(pida, &medium);
}
return hr;
}
LPTSTR NET_GetProviderFromRes(LPCIDNETRESOURCE pidn, LPTSTR pszBuff, UINT cchBuff);
// By the way...Win95 shipped with the below provider
// names. Since the name can be changed and be localized,
// we have to try and map these correctly for net pidl
// interop.
//
// get the network resource name from an item. this is not a file system path!
//
// example:
// server \\server or strike/sys
// share \\server\share or strike/sys
// printer \\server\printer
// provider "provider name"
// entire net "Entire Network"
//
// in:
// pidn the item
// cchBuff size of buffer in chars.
//
// out:
// pszBuff return buffer
//
// returns:
// address of the input buffer (pszBuff)
//
LPTSTR NET_CopyResName(LPCIDNETRESOURCE pidn, LPTSTR pszBuff, UINT cchBuff)
{
if (NET_IsUnicode(pidn))
{
LPBYTE pb = (LPBYTE)pidn->szNetResName;
pb += lstrlenA((LPSTR)pb) + 1; // Skip over ansi net name
if (NET_FHasProvider(pidn))
pb += lstrlenA((LPSTR)pb) + 1; // Skip over ansi provider
if (NET_FHasComment(pidn))
pb += lstrlenA((LPSTR)pb) + 1; // Skip over comment
ualstrcpyn(pszBuff, (LPNWSTR)pb, cchBuff);
}
else
{
SHAnsiToTChar(pidn->szNetResName, pszBuff, cchBuff);
}
return pszBuff;
}
//
// get the provider name from an item. some items do not have providers stored
// in them. for example the "*" indicates where the provider is stored in the
// two different forms of network pidls.
// [entire net] [provider *] [server] [share] [... file system]
// [server *] [share] [... file system]
// in:
// pidn item (single item PIDL) to try to get the provider name from
// cchBuff size in chars.
// out:
// pszBuff output
//
LPTSTR NET_CopyProviderName(LPCIDNETRESOURCE pidn, LPTSTR pszBuff, UINT cchBuff)
{
*pszBuff = 0;
if (!NET_FHasProvider(pidn))
return NULL;
// try the wNetType at the end of the pidl
const BYTE *pb = (LPBYTE)pidn + pidn->cb - sizeof(WORD);
DWORD dwNetType = *((UNALIGNED WORD *)pb) << 16;
if (dwNetType && (dwNetType <= WNNC_NET_LARGEST) &&
(WNetGetProviderName(dwNetType, pszBuff, (ULONG*)&cchBuff) == WN_SUCCESS))
{
return pszBuff;
}
// Try the old way...
pb = (LPBYTE)pidn->szNetResName + lstrlenA(pidn->szNetResName) + 1; // Skip over ansi net name
if (NET_IsUnicode(pidn))
{
pb += lstrlenA((LPSTR)pb) + 1; // Skip over ansi provider
if (NET_FHasComment(pidn))
pb += lstrlenA((LPSTR)pb) + 1; // Skip over comment
pb += (ualstrlen((LPNWSTR)pb) + 1) * sizeof(WCHAR); // skip over unicode net name
ualstrcpyn(pszBuff, (LPNWSTR)pb, cchBuff);
}
else
{
SHAnsiToTChar((LPSTR)pb, pszBuff, cchBuff);
}
// Map from Win95 net provider name if possible...
for (int i = 0; i < ARRAYSIZE(c_rgProviderMap); i++)
{
if (lstrcmp(pszBuff, c_rgProviderMap[i].lpName) == 0)
{
DWORD dwNetType = c_rgProviderMap[i].wNetType << 16;
if (dwNetType && (dwNetType <= WNNC_NET_LARGEST))
{
*pszBuff = 0;
WNetGetProviderName(dwNetType, pszBuff, (LPDWORD)&cchBuff);
}
break;
}
}
return pszBuff;
}
//
// get the comment if there is one from the net item
//
LPTSTR NET_CopyComment(LPCIDNETRESOURCE pidn, LPTSTR pszBuff, UINT cchBuff)
{
*pszBuff = 0;
LPCSTR pszT = pidn->szNetResName + lstrlenA(pidn->szNetResName) + 1;
if (NET_FHasComment(pidn))
{
if (NET_FHasProvider(pidn))
pszT += lstrlenA(pszT) + 1;
if (NET_IsUnicode(pidn))
{
pszT += lstrlenA(pszT) + 1; // Skip Ansi comment
LPNCWSTR pszW = (LPNCWSTR)pszT; // We're at the unicode portion of the pidl
pszW += ualstrlen(pszW) + 1; // Skip Unicode Name
if (NET_FHasProvider(pidn))
pszW += ualstrlen(pszW) + 1; // Skip Unicode Provider
ualstrcpyn(pszBuff, pszW, cchBuff);
}
else
{
SHAnsiToUnicode(pszT, pszBuff, cchBuff);
}
}
return pszBuff;
}
// pidlRemainder will be filled in (only in the TRUE return case) with a
// pointer to the part of the IDL (if any) past the remote regitem.
// This value may be used, for example, to differentiate between a remote
// printer folder and a printer under a remote printer folder
BOOL NET_IsRemoteRegItem(LPCITEMIDLIST pidl, REFCLSID rclsid, LPCITEMIDLIST* ppidlRemainder)
{
BOOL bRet = FALSE;
// in "My Network Places"
if (pidl && IsIDListInNameSpace(pidl, &CLSID_NetworkPlaces))
{
LPCITEMIDLIST pidlStart = pidl; // save this
// Now, search for a server item. HACKHACK: this assume everything from
// the NetHood to the server item is a shell pidl with a bFlags field!!
for (pidl = _ILNext(pidl); !ILIsEmpty(pidl); pidl = _ILNext(pidl))
{
if ((SIL_GetType(pidl) & SHID_TYPEMASK) == SHID_NET_SERVER)
{
LPITEMIDLIST pidlToTest;
// Found a server. Is the thing after it a remote registry item?
pidl = _ILNext(pidl);
*ppidlRemainder = _ILNext(pidl);
pidlToTest = ILCloneUpTo(pidlStart, *ppidlRemainder);
if (pidlToTest)
{
CLSID clsid;
bRet = SUCCEEDED(GetCLSIDFromIDList(pidlToTest, &clsid)) && IsEqualCLSID(rclsid, clsid);
ILFree(pidlToTest);
}
break; // done
}
}
}
return bRet;
}
//
// Get the provider name from an absolute IDLIST.
// Parameters:
// pidlAbs -- Specifies the Absolute IDList to the file system object
//
LPTSTR NET_GetProviderFromIDList(LPCITEMIDLIST pidlAbs, LPTSTR pszBuff, UINT cchBuff)
{
return NET_GetProviderFromRes((LPCIDNETRESOURCE)_ILNext(pidlAbs), pszBuff, cchBuff);
}
//
// Get the provider name from a relative IDLIST.
// in:
// pidn potentially multi level item to try to get the resource from
//
LPTSTR NET_GetProviderFromRes(LPCIDNETRESOURCE pidn, LPTSTR pszBuffer, UINT cchBuffer)
{
// If this guy is the REST of network item, we increment to the
// next IDLIST - If at root return NULL
if (pidn->cb == 0)
return NULL;
//
// If the IDLIST starts with a ROOT_REGITEM, then skip to the
// next item in the list...
if (pidn->bFlags == SHID_ROOT_REGITEM)
{
pidn = (LPIDNETRESOURCE)_ILNext((LPITEMIDLIST)pidn);
if (pidn->cb == 0)
return NULL;
}
// If the IDLIST includes Entire Network, the provider will be
// part of the next component.
if (NET_GetDisplayType(pidn) == RESOURCEDISPLAYTYPE_ROOT)
{
pidn = (LPIDNETRESOURCE)_ILNext((LPITEMIDLIST)pidn);
if (pidn->cb == 0)
return NULL;
}
// If the next component after the 'hood or Entire Network is
// a network object, its name is the provider name, else the
// provider name comes after the remote name.
if (NET_GetDisplayType(pidn) == RESOURCEDISPLAYTYPE_NETWORK)
{
// Simply return the name field back for the item.
return NET_CopyResName(pidn, pszBuffer, cchBuffer);
}
else
{
// Nope one of the items in the neighborhood view was selected
// The Provider name is stored after ther resource name
return NET_CopyProviderName(pidn, pszBuffer, cchBuffer);
}
}
#define PTROFFSET(pBase, p) ((int) ((LPBYTE)(p) - (LPBYTE)(pBase)))
//
// fill in pmedium with a NRESARRAY
//
// pmedium == NULL if we are handling QueryGetData
//
STDAPI CNetData_GetNetResource(IDataObject *pdtobj, STGMEDIUM *pmedium)
{
HRESULT hr = E_OUTOFMEMORY;
LPITEMIDLIST pidl;
STGMEDIUM medium;
LPIDA pida = DataObj_GetHIDA(pdtobj, &medium);
ASSERT(pida && pida->cidl);
// First, get the provider name from the first one (assuming they are common).
pidl = IDA_ILClone(pida, 0);
if (pidl)
{
TCHAR szProvider[MAX_PATH];
LPCTSTR pszProvider = NET_GetProviderFromIDList(pidl, szProvider, ARRAYSIZE(szProvider));
if (pmedium)
{
TCHAR szName[MAX_PATH];
UINT cbHeader = sizeof(NRESARRAY) + (sizeof(NETRESOURCE) * (pida->cidl - 1));
UINT cbRequired, iItem;
// Calculate required size
cbRequired = cbHeader;
if (pszProvider)
cbRequired += (lstrlen(pszProvider) + 1) * sizeof(TCHAR);
for (iItem = 0; iItem < pida->cidl; iItem++)
{
LPCIDNETRESOURCE pidn = (LPCIDNETRESOURCE)IDA_GetIDListPtr(pida, iItem);
NET_CopyResName(pidn, szName, ARRAYSIZE(szName));
cbRequired += (lstrlen(szName) + 1) * sizeof(TCHAR);
}
//
// Indicate that the caller should release hmem.
//
pmedium->pUnkForRelease = NULL;
pmedium->tymed = TYMED_HGLOBAL;
pmedium->hGlobal = GlobalAlloc(GPTR, cbRequired);
if (pmedium->hGlobal)
{
LPNRESARRAY panr = (LPNRESARRAY)pmedium->hGlobal;
LPTSTR pszT = (LPTSTR)((LPBYTE)panr + cbHeader);
LPTSTR pszEnd = (LPTSTR)((LPBYTE)panr + cbRequired);
UINT offProvider = 0;
panr->cItems = pida->cidl;
// Copy the provider name. This is not necessary,
// if we are dragging providers.
if (pszProvider)
{
lstrcpy(pszT, pszProvider);
offProvider = PTROFFSET(panr, pszT);
pszT += lstrlen(pszT) + 1;
}
//
// For each item, fill each NETRESOURCE and append resource
// name at the end. Note that we should put offsets in
// lpProvider and lpRemoteName.
//
for (iItem = 0; iItem < pida->cidl; iItem++)
{
LPNETRESOURCE pnr = &panr->nr[iItem];
LPCIDNETRESOURCE pidn = (LPCIDNETRESOURCE)IDA_GetIDListPtr(pida, iItem);
ASSERT(pnr->dwScope == 0);
ASSERT(pnr->lpLocalName==NULL);
ASSERT(pnr->lpComment==NULL);
pnr->dwType = NET_GetType(pidn);
pnr->dwDisplayType = NET_GetDisplayType(pidn);
pnr->dwUsage = NET_GetUsage(pidn);
NET_CopyResName(pidn, pszT, (UINT)(pszEnd-pszT));
if (pnr->dwDisplayType == RESOURCEDISPLAYTYPE_ROOT)
{
pnr->lpProvider = NULL;
pnr->lpRemoteName = NULL;
}
else if (pnr->dwDisplayType == RESOURCEDISPLAYTYPE_NETWORK)
{
*((UINT *) &pnr->lpProvider) = PTROFFSET(panr, pszT);
ASSERT(pnr->lpRemoteName == NULL);
}
else
{
*((UINT *) &pnr->lpProvider) = offProvider;
*((UINT *) &pnr->lpRemoteName) = PTROFFSET(panr, pszT);
}
pszT += lstrlen(pszT) + 1;
}
ASSERT(pszEnd == pszT);
hr = S_OK;
}
}
else
{
hr = S_OK; // handing QueryGetData, yes, we have it
}
ILFree(pidl);
}
HIDA_ReleaseStgMedium(pida, &medium);
return hr;
}
// fill in pmedium with an HGLOBAL version of a NRESARRAY
STDAPI CNetData_GetNetResourceForFS(IDataObject *pdtobj, STGMEDIUM *pmedium)
{
HRESULT hr = E_OUTOFMEMORY;
LPITEMIDLIST pidlAbs;
STGMEDIUM medium;
LPIDA pida = DataObj_GetHIDA(pdtobj, &medium);
ASSERT(pida && medium.hGlobal); // we created this...
//
// NOTES: Even though we may have multiple FS objects in this HIDA,
// we know that they share the root. Therefore, getting the pidl for
// the first item is always sufficient.
//
pidlAbs = IDA_ILClone(pida, 0);
if (pidlAbs)
{
LPITEMIDLIST pidl;
ASSERT(AssertIsIDListInNameSpace(pidlAbs, &CLSID_NetworkPlaces));
//
// Look for the JUNCTION point (starting from the second ID)
//
for (pidl = _ILNext(pidlAbs); !ILIsEmpty(pidl); pidl = _ILNext(pidl))
{
LPIDNETRESOURCE pidn = (LPIDNETRESOURCE)pidl;
if (NET_GetFlags(pidn) & SHID_JUNCTION)
{
//
// We found the JUNCTION point (which is s share).
// Return the HNRES to it.
//
TCHAR szProvider[MAX_PATH];
TCHAR szRemote[MAX_PATH];
UINT cbRequired;
LPCTSTR pszProvider = NET_GetProviderFromIDList(pidlAbs, szProvider, ARRAYSIZE(szProvider));
LPCTSTR pszRemoteName = NET_CopyResName(pidn, szRemote, ARRAYSIZE(szRemote));
UINT cbProvider = lstrlen(pszProvider) * sizeof(TCHAR) + sizeof(TCHAR);
//
// This should not be a provider node.
// This should not be the last ID in pidlAbs.
//
ASSERT(pszProvider != pszRemoteName);
ASSERT(!ILIsEmpty(_ILNext(pidl)));
cbRequired = sizeof(NRESARRAY) + cbProvider + lstrlen(pszRemoteName) * sizeof(TCHAR) + sizeof(TCHAR);
pmedium->pUnkForRelease = NULL;
pmedium->tymed = TYMED_HGLOBAL;
pmedium->hGlobal = GlobalAlloc(GPTR, cbRequired);
if (pmedium->hGlobal)
{
LPNRESARRAY panr = (LPNRESARRAY)pmedium->hGlobal;
LPNETRESOURCE pnr = &panr->nr[0];
LPTSTR pszT = (LPTSTR)(panr + 1);
ASSERT(pnr->dwScope == 0);
ASSERT(pnr->lpLocalName == NULL);
ASSERT(pnr->lpComment == NULL);
panr->cItems = 1;
pnr->dwType = NET_GetType(pidn);
pnr->dwDisplayType = NET_GetDisplayType(pidn);
pnr->dwUsage = NET_GetUsage(pidn);
*((UINT *) &pnr->lpProvider) = sizeof(NRESARRAY);
lstrcpy(pszT, pszProvider);
ASSERT(PTROFFSET(panr, pszT) == sizeof(NRESARRAY));
pszT += cbProvider / sizeof(TCHAR);
*((UINT *) &pnr->lpRemoteName) = sizeof(NRESARRAY) + cbProvider;
ASSERT(PTROFFSET(panr, pszT) == (int)sizeof(NRESARRAY) + (int)cbProvider);
lstrcpy(pszT, pszRemoteName);
ASSERT(((LPBYTE)panr) + cbRequired == (LPBYTE)pszT + (lstrlen(pszT) + 1) * sizeof(TCHAR));
hr = S_OK;
}
else
{
hr = E_OUTOFMEMORY;
}
break;
}
}
ASSERT(!ILIsEmpty(pidl)); // We should have found the junction point.
ILFree(pidlAbs);
}
HIDA_ReleaseStgMedium(pida, &medium);
return hr;
}
STDMETHODIMP CNetData::QueryGetData(FORMATETC *pformatetc)
{
if (pformatetc->tymed & TYMED_HGLOBAL)
{
if (pformatetc->cfFormat == g_cfNetResource)
return CNetData_GetNetResource(this, NULL);
if (pformatetc->cfFormat == CF_HDROP)
return GetHDrop(NULL, NULL);
}
return CFSIDLData::QueryGetData(pformatetc);
}
STDMETHODIMP CNetData::GetData(FORMATETC *pformatetc, STGMEDIUM *pmedium)
{
if (pformatetc->tymed & TYMED_HGLOBAL)
{
if (pformatetc->cfFormat == g_cfNetResource)
return CNetData_GetNetResource(this, pmedium);
if (pformatetc->cfFormat == CF_HDROP)
return GetHDrop(pformatetc, pmedium);
}
return CFSIDLData::GetData(pformatetc, pmedium);
}
BOOL GetPathFromDataObject(IDataObject *pdtobj, DWORD dwData, LPTSTR pszFileName)
{
BOOL bRet = FALSE;
BOOL fUnicode = FALSE;
HRESULT hr;
if (dwData & (DTID_FDESCW | DTID_FDESCA))
{
FORMATETC fmteW = {g_cfFileGroupDescriptorW, NULL, DVASPECT_CONTENT, -1, TYMED_HGLOBAL};
STGMEDIUM medium = {0};
hr = pdtobj->GetData(&fmteW, &medium);
if (SUCCEEDED(hr))
{
fUnicode = TRUE;
}
else
{
FORMATETC fmteA = {g_cfFileGroupDescriptorA, NULL, DVASPECT_CONTENT, -1, TYMED_HGLOBAL};
hr = pdtobj->GetData(&fmteA, &medium);
}
if (SUCCEEDED(hr))
{
if (fUnicode)
{
FILEGROUPDESCRIPTORW *pfgdW = (FILEGROUPDESCRIPTORW *)GlobalLock(medium.hGlobal);
if (pfgdW)
{
if (pfgdW->cItems == 1)
{
SHUnicodeToTChar(pfgdW->fgd[0].cFileName, pszFileName, MAX_PATH);
}
bRet = TRUE;
GlobalUnlock(medium.hGlobal);
}
}
else
{
FILEGROUPDESCRIPTORA *pfgdA = (FILEGROUPDESCRIPTORA*)GlobalLock(medium.hGlobal);
if (pfgdA)
{
if (pfgdA->cItems == 1)
{
SHAnsiToTChar(pfgdA->fgd[0].cFileName, pszFileName, MAX_PATH);
}
bRet = TRUE;
GlobalUnlock(medium.hGlobal);
}
}
ReleaseStgMedium(&medium);
}
}
return bRet;
}
class CNetRootDropTarget : public CIDLDropTarget
{
friend HRESULT CNetRootDropTarget_CreateInstance(HWND hwnd, LPCITEMIDLIST pidl, IDropTarget **ppdropt);
public:
CNetRootDropTarget(HWND hwnd) : CIDLDropTarget(hwnd) { };
// IDropTarget methods overwirte
STDMETHODIMP DragEnter(IDataObject *pdtobj, DWORD grfKeyState, POINTL pt, DWORD *pdwEffect);
STDMETHODIMP Drop(IDataObject *pdtobj, DWORD grfKeyState, POINTL pt, DWORD *pdwEffect);
private:
~CNetRootDropTarget();
IDropTarget *_pdtgHood; // file system drop target
};
CNetRootDropTarget::~CNetRootDropTarget()
{
if (_pdtgHood)
_pdtgHood->Release();
}
STDMETHODIMP CNetRootDropTarget::DragEnter(IDataObject *pdtobj, DWORD grfKeyState, POINTL pt, DWORD *pdwEffect)
{
CIDLDropTarget::DragEnter(pdtobj, grfKeyState, pt, pdwEffect);
if ((m_dwData & (DTID_NETRES | DTID_HIDA)) == (DTID_NETRES | DTID_HIDA))
{
// NETRESOURCE (DTID_NETRES) allow link
*pdwEffect &= DROPEFFECT_LINK;
}
else if (((m_dwData & (DTID_FDESCW | DTID_CONTENTS)) == (DTID_FDESCW | DTID_CONTENTS)) ||
((m_dwData & (DTID_FDESCA | DTID_CONTENTS)) == (DTID_FDESCA | DTID_CONTENTS)) )
{
// dragging an URL from the web browser gives a FILECONTENTS version
// of a .URL file. accept that here for Internet Shortcut (.url)
TCHAR szFileName[MAX_PATH];
if (GetPathFromDataObject(pdtobj, m_dwData, szFileName) &&
(0 == lstrcmpi(PathFindExtension(szFileName), TEXT(".url"))))
{
*pdwEffect &= DROPEFFECT_LINK;
}
else
{
*pdwEffect = DROPEFFECT_NONE;
}
}
else
{
*pdwEffect = DROPEFFECT_NONE;
}
m_dwEffectLastReturned = *pdwEffect;
return S_OK;
}
STDMETHODIMP CNetRootDropTarget::Drop(IDataObject *pdtobj, DWORD grfKeyState, POINTL pt, DWORD *pdwEffect)
{
*pdwEffect &= DROPEFFECT_LINK;
HRESULT hr = CIDLDropTarget::DragDropMenu(DROPEFFECT_LINK, pdtobj,
pt, pdwEffect, NULL, NULL, POPUP_NONDEFAULTDD, grfKeyState);
if (*pdwEffect)
{
if (!_pdtgHood)
{
LPITEMIDLIST pidl = SHCloneSpecialIDList(NULL, CSIDL_NETHOOD, FALSE);
if (pidl)
{
IShellFolder *psf;
if (SUCCEEDED(SHBindToObject(NULL, IID_X_PPV_ARG(IShellFolder, pidl, &psf))))
{
psf->CreateViewObject(_GetWindow(), IID_PPV_ARG(IDropTarget, &_pdtgHood));
psf->Release();
}
ILFree(pidl);
}
}
if (_pdtgHood)
{
// force link through the dwEffect and keyboard
*pdwEffect &= DROPEFFECT_LINK;
grfKeyState = MK_LBUTTON | MK_CONTROL | MK_SHIFT | MK_FAKEDROP;
hr = SHSimulateDrop(_pdtgHood, pdtobj, grfKeyState, NULL, pdwEffect);
}
else
*pdwEffect = 0;
}
CIDLDropTarget::DragLeave();
return hr;
}
HRESULT CNetRootDropTarget_CreateInstance(HWND hwnd, LPCITEMIDLIST pidl, IDropTarget **ppdropt)
{
*ppdropt = NULL;
HRESULT hr;
CNetRootDropTarget *pnrdt = new CNetRootDropTarget(hwnd);
if (pnrdt)
{
hr = pnrdt->_Init(pidl);
if (SUCCEEDED(hr))
pnrdt->QueryInterface(IID_PPV_ARG(IDropTarget, ppdropt));
pnrdt->Release();
}
else
hr = E_OUTOFMEMORY;
return hr;
}
// This part is psuedo bogus. Basically we have problems at times doing a
// translation from things like \\pyrex\user to the appropriate PIDL,
// especially if you want to avoid the overhead of hitting the network and
// also problems of knowing if the server is in the "HOOD"
//
// We must maintain the mapping table in UNICODE internally, because
// IShellFolder::ParseDisplayName uses UNICODE, and we don't want to have
// to deal with lstrlen(dbcs) != lstrlen(sbcs) problems.
//
typedef struct _NPT_ITEM
{
struct _NPT_ITEM *pnptNext; // Pointer to next item;
LPCITEMIDLIST pidl; // The pidl
USHORT cchName; // size of the name in characters.
WCHAR szName[1]; // The name to translate from
} NPT_ITEM;
// Each process will maintain their own list.
NPT_ITEM *g_pnptHead = NULL;
//
// Function to register translations from Path to IDList translations.
//
void NPTRegisterNameToPidlTranslation(LPCTSTR pszPath, LPCITEMIDLIST pidl)
{
NPT_ITEM *pnpt;
int cItemsRemoved = 0;
WCHAR szPath[MAX_PATH];
// We currently are only interested in UNC Roots
// If the table becomes large we can reduce this to only servers...
if (!PathIsUNC(pszPath))
return; // Not interested.
//
// If this item is not a root we need to count how many items to remove
//
SHTCharToUnicode(pszPath, szPath, ARRAYSIZE(szPath));
while (!PathIsUNCServerShare(szPath))
{
cItemsRemoved++;
if (!PathRemoveFileSpecW(szPath))
return; // Did not get back to a valid root
}
ENTERCRITICAL;
// We don't want to add duplicates
for (pnpt = g_pnptHead; pnpt != NULL ; pnpt = pnpt->pnptNext)
{
if (StrCmpIW(szPath, pnpt->szName) == 0)
break;
}
if (pnpt == NULL)
{
UINT cch = lstrlenW(szPath);
pnpt = (NPT_ITEM *)LocalAlloc(LPTR, sizeof(NPT_ITEM) + cch * sizeof(WCHAR));
if (pnpt)
{
pnpt->pidl = ILClone(pidl);
if (pnpt->pidl)
{
while (cItemsRemoved--)
{
ILRemoveLastID((LPITEMIDLIST)pnpt->pidl);
}
pnpt->pnptNext = g_pnptHead;
g_pnptHead = pnpt;
pnpt->cchName = (USHORT) cch;
lstrcpyW(pnpt->szName, szPath);
}
else
{
LocalFree((HLOCAL)pnpt);
}
}
}
LEAVECRITICAL;
}
// The main function to attemp to map a portion of the name into an idlist
// Right now limit it to UNC roots
//
LPWSTR NPTMapNameToPidl(LPCWSTR pszPath, LPCITEMIDLIST *ppidl)
{
NPT_ITEM *pnpt;
*ppidl = NULL;
ENTERCRITICAL;
// See if we can find the item in the list.
for (pnpt = g_pnptHead; pnpt != NULL ; pnpt = pnpt->pnptNext)
{
if (IntlStrEqNIW(pszPath, pnpt->szName, pnpt->cchName) &&
((pszPath[pnpt->cchName] == TEXT('\\')) || (pszPath[pnpt->cchName] == TEXT('\0'))))
{
break;
}
}
LEAVECRITICAL;
// See if we found a match
if (pnpt == NULL)
return NULL;
// Found a match
*ppidl = pnpt->pidl;
return (LPWSTR)pszPath + pnpt->cchName; // points to slash
}