mirror of https://github.com/tongzx/nt5src
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.
1044 lines
28 KiB
1044 lines
28 KiB
// fpnw.cpp : FPNW shares, sessions and open resources
|
|
|
|
#include "stdafx.h"
|
|
#include "cmponent.h"
|
|
#include "safetemp.h"
|
|
#include "FileSvc.h"
|
|
#include "DynamLnk.h" // DynamicDLL
|
|
#include "fpnw.h"
|
|
#include "ShrProp.h" // Share Properties Pages
|
|
#include "permpage.h" // CSecurityInformation
|
|
#include "compdata.h"
|
|
|
|
#define DONT_WANT_SHELLDEBUG
|
|
#include "shlobjp.h" // LPITEMIDLIST
|
|
#include "wraps.h" // Wrap_ILCreateFromPath
|
|
|
|
extern "C"
|
|
{
|
|
#include <ntseapi.h> // PSECURITY_DESCRIPTOR
|
|
}
|
|
|
|
// CODEWORK We should be more sophisticated about
|
|
// determining whether FPNW is installed.
|
|
// This API should appear in fpnwapi:
|
|
// BOOL IsNetWareInstalled( VOID );
|
|
|
|
#include "macros.h"
|
|
USE_HANDLE_MACROS("FILEMGMT(fpnw.cpp)")
|
|
|
|
#ifdef _DEBUG
|
|
#define new DEBUG_NEW
|
|
#undef THIS_FILE
|
|
static char THIS_FILE[] = __FILE__;
|
|
#endif
|
|
|
|
class CFPNWSecurityInformation : public CShareSecurityInformation
|
|
{
|
|
STDMETHOD(GetSecurity) (SECURITY_INFORMATION RequestedInformation,
|
|
PSECURITY_DESCRIPTOR *ppSecurityDescriptor,
|
|
BOOL fDefault );
|
|
STDMETHOD(SetSecurity) (SECURITY_INFORMATION SecurityInformation,
|
|
PSECURITY_DESCRIPTOR pSecurityDescriptor );
|
|
public:
|
|
FPNWVOLUMEINFO_2* m_pvolumeinfo;
|
|
PSECURITY_DESCRIPTOR m_pDefaultDescriptor;
|
|
CFPNWSecurityInformation();
|
|
~CFPNWSecurityInformation();
|
|
};
|
|
|
|
typedef enum _FpnwApiIndex
|
|
{
|
|
FPNW_VOLUME_ENUM = 0,
|
|
FPNW_CONNECTION_ENUM,
|
|
FPNW_FILE_ENUM,
|
|
FPNW_API_BUFFER_FREE,
|
|
FPNW_VOLUME_DEL,
|
|
FPNW_CONNECTION_DEL,
|
|
FPNW_FILE_CLOSE,
|
|
FPNW_VOLUME_GET_INFO,
|
|
FPNW_VOLUME_SET_INFO
|
|
};
|
|
|
|
// not subject to localization
|
|
static LPCSTR g_apchFunctionNames[] = {
|
|
"FpnwVolumeEnum",
|
|
"FpnwConnectionEnum",
|
|
"FpnwFileEnum",
|
|
"FpnwApiBufferFree",
|
|
"FpnwVolumeDel",
|
|
"FpnwConnectionDel",
|
|
"FpnwFileClose",
|
|
"FpnwVolumeGetInfo",
|
|
"FpnwVolumeSetInfo",
|
|
NULL
|
|
};
|
|
|
|
// not subject to localization
|
|
DynamicDLL g_FpnwDLL( _T("FPNWCLNT.DLL"), g_apchFunctionNames );
|
|
|
|
typedef DWORD (*APIBUFFERFREEPROC) (LPVOID);
|
|
|
|
VOID FPNWFreeData(PVOID* ppv)
|
|
{
|
|
if (*ppv != NULL)
|
|
{
|
|
ASSERT( NULL != g_FpnwDLL[FPNW_API_BUFFER_FREE] );
|
|
(void) ((APIBUFFERFREEPROC)g_FpnwDLL[FPNW_API_BUFFER_FREE])( *ppv );
|
|
*ppv = NULL;
|
|
}
|
|
}
|
|
|
|
FpnwFileServiceProvider::FpnwFileServiceProvider( CFileMgmtComponentData* pFileMgmtData )
|
|
: FileServiceProvider( pFileMgmtData )
|
|
{
|
|
VERIFY( m_strTransportFPNW.LoadString( IDS_TRANSPORT_FPNW ) );
|
|
}
|
|
|
|
|
|
/*
|
|
DWORD
|
|
FpnwVolumeEnum(
|
|
IN LPWSTR pServerName OPTIONAL,
|
|
IN DWORD dwLevel,
|
|
OUT LPBYTE *ppVolumeInfo,
|
|
OUT PDWORD pEntriesRead,
|
|
IN OUT PDWORD resumeHandle OPTIONAL
|
|
);
|
|
*/
|
|
|
|
typedef DWORD (*VOLUMEENUMPROC) (LPWSTR,DWORD,LPBYTE*,PDWORD,PDWORD);
|
|
|
|
HRESULT FpnwFileServiceProvider::PopulateShares(
|
|
IResultData* pResultData,
|
|
CFileMgmtCookie* pcookie)
|
|
{
|
|
TEST_NONNULL_PTR_PARAM(pcookie);
|
|
|
|
if ( !g_FpnwDLL.LoadFunctionPointers() )
|
|
return S_OK;
|
|
|
|
FPNWVOLUMEINFO* pvolumeinfo = NULL;
|
|
DWORD dwEntriesRead = 0;
|
|
DWORD hEnumHandle = 0;
|
|
NET_API_STATUS retval = NERR_Success;
|
|
do {
|
|
retval = ((VOLUMEENUMPROC)g_FpnwDLL[FPNW_VOLUME_ENUM])(
|
|
const_cast<LPTSTR>(pcookie->QueryTargetServer()),
|
|
1,
|
|
(PBYTE*)&pvolumeinfo,
|
|
&dwEntriesRead,
|
|
&hEnumHandle );
|
|
if (NERR_Success == retval)
|
|
{
|
|
AddFPNWShareItems( pResultData, pcookie, pvolumeinfo, dwEntriesRead );
|
|
pvolumeinfo = NULL;
|
|
break;
|
|
} else if (ERROR_MORE_DATA == retval) {
|
|
ASSERT( NULL != hEnumHandle );
|
|
AddFPNWShareItems( pResultData, pcookie, pvolumeinfo, dwEntriesRead );
|
|
pvolumeinfo = NULL;
|
|
continue;
|
|
} else if (RPC_S_SERVER_UNAVAILABLE == retval && 0 == hEnumHandle) {
|
|
// FPNW just isn't installed, don't worry about it
|
|
retval = NERR_Success;
|
|
break;
|
|
} else {
|
|
(void) DoErrMsgBox(GetActiveWindow(), MB_OK | MB_ICONSTOP, retval, IDS_POPUP_FPNW_SHARES, pcookie->QueryNonNULLMachineName() );
|
|
break;
|
|
}
|
|
} while (TRUE);
|
|
|
|
return HRESULT_FROM_WIN32(retval);
|
|
}
|
|
|
|
/*
|
|
typedef struct _FPNWVolumeInfo
|
|
{
|
|
LPWSTR lpVolumeName;
|
|
DWORD dwType;
|
|
DWORD dwMaxUses;
|
|
|
|
DWORD dwCurrentUses;
|
|
LPWSTR lpPath;
|
|
|
|
} FPNWVOLUMEINFO, *PFPNWVOLUMEINFO;
|
|
*/
|
|
|
|
HRESULT FpnwFileServiceProvider::AddFPNWShareItems(
|
|
IResultData* pResultData,
|
|
CFileMgmtCookie* pParentCookie,
|
|
PVOID pinfo,
|
|
DWORD nItems)
|
|
{
|
|
TEST_NONNULL_PTR_PARAM(pParentCookie);
|
|
TEST_NONNULL_PTR_PARAM(pinfo);
|
|
|
|
if (0 >= nItems)
|
|
return S_OK;
|
|
|
|
RESULTDATAITEM tRDItem;
|
|
::ZeroMemory( &tRDItem, sizeof(tRDItem) );
|
|
tRDItem.nCol = COLNUM_SHARES_SHARED_FOLDER;
|
|
// CODEWORK should use MMC_ICON_CALLBACK here
|
|
tRDItem.nImage = iIconFPNWShare;
|
|
tRDItem.mask = RDI_STR | RDI_IMAGE | RDI_PARAM;
|
|
tRDItem.str = MMC_CALLBACK;
|
|
|
|
FPNWVOLUMEINFO* pvolumeinfo = (FPNWVOLUMEINFO*)pinfo;
|
|
|
|
DWORD nItemsToAdd = 0;
|
|
for (DWORD i = 0; i < nItems; i++ )
|
|
{
|
|
if (!IsInvalidSharename(pvolumeinfo[i].lpVolumeName))
|
|
nItemsToAdd++;
|
|
}
|
|
|
|
CFpnwShareCookie* pcookiearray = new CFpnwShareCookie[nItemsToAdd];
|
|
CFpnwCookieBlock* pCookieBlock = new CFpnwCookieBlock(
|
|
pcookiearray,nItemsToAdd,pParentCookie->QueryNonNULLMachineName(),pinfo );
|
|
pParentCookie->m_listResultCookieBlocks.AddHead( pCookieBlock );
|
|
|
|
for ( ; nItems > 0; nItems--, pvolumeinfo++, pcookiearray++ )
|
|
{
|
|
if (IsInvalidSharename(pvolumeinfo->lpVolumeName))
|
|
continue;
|
|
|
|
pcookiearray->m_pobject = pvolumeinfo;
|
|
// WARNING cookie cast
|
|
tRDItem.lParam = reinterpret_cast<LPARAM>((CCookie*)pcookiearray);
|
|
HRESULT hr = pResultData->InsertItem(&tRDItem);
|
|
ASSERT(SUCCEEDED(hr));
|
|
}
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
|
|
typedef DWORD (*CONNECTIONENUMPROC) (LPWSTR,DWORD,LPBYTE*,PDWORD,PDWORD);
|
|
|
|
// if pResultData is not NULL, add sessions/resources to the listbox
|
|
// if pResultData is NULL, delete all sessions/resources
|
|
// if pResultData is NULL, return SUCCEEDED(hr) to continue or
|
|
// FAILED(hr) to abort
|
|
HRESULT FpnwFileServiceProvider::EnumerateSessions(
|
|
IResultData* pResultData,
|
|
CFileMgmtCookie* pcookie,
|
|
bool bAddToResultPane)
|
|
{
|
|
TEST_NONNULL_PTR_PARAM(pcookie);
|
|
|
|
if ( !g_FpnwDLL.LoadFunctionPointers() )
|
|
return S_OK;
|
|
|
|
FPNWCONNECTIONINFO* pconninfo = NULL;
|
|
DWORD dwEntriesRead = 0;
|
|
DWORD hEnumHandle = 0;
|
|
HRESULT hr = S_OK;
|
|
NET_API_STATUS retval = NERR_Success;
|
|
do {
|
|
retval = ((CONNECTIONENUMPROC)g_FpnwDLL[FPNW_CONNECTION_ENUM])(
|
|
const_cast<LPTSTR>(pcookie->QueryTargetServer()),
|
|
1,
|
|
(PBYTE*)&pconninfo,
|
|
&dwEntriesRead,
|
|
&hEnumHandle );
|
|
if (NERR_Success == retval)
|
|
{
|
|
hr = HandleFPNWSessionItems( pResultData, pcookie, pconninfo, dwEntriesRead,
|
|
bAddToResultPane);
|
|
pconninfo = NULL;
|
|
break;
|
|
} else if (ERROR_MORE_DATA == retval) {
|
|
ASSERT( NULL != hEnumHandle );
|
|
hr = HandleFPNWSessionItems( pResultData, pcookie, pconninfo, dwEntriesRead,
|
|
bAddToResultPane);
|
|
pconninfo = NULL;
|
|
continue;
|
|
} else if (RPC_S_SERVER_UNAVAILABLE == retval && 0 == hEnumHandle) {
|
|
// FPNW just isn't installed, don't worry about it
|
|
retval = NERR_Success;
|
|
break;
|
|
} else {
|
|
(void) DoErrMsgBox(GetActiveWindow(), MB_OK | MB_ICONSTOP, retval, IDS_POPUP_FPNW_SESSIONS, pcookie->QueryNonNULLMachineName() );
|
|
break;
|
|
}
|
|
} while (S_OK == hr);
|
|
|
|
return HRESULT_FROM_WIN32(retval);
|
|
}
|
|
|
|
|
|
/*
|
|
typedef enum _COLNUM_SESSIONS {
|
|
COLNUM_SESSIONS_USERNAME = 0,
|
|
COLNUM_SESSIONS_COMPUTERNAME,
|
|
COLNUM_SESSIONS_NUM_FILES,
|
|
COLNUM_SESSIONS_CONNECTED_TIME,
|
|
COLNUM_SESSIONS_IDLE_TIME,
|
|
COLNUM_SESSIONS_IS_GUEST
|
|
} COLNUM_SESSIONS;
|
|
|
|
typedef struct _FPNWConnectionInfo
|
|
{
|
|
DWORD dwConnectionId;
|
|
FPNWSERVERADDR WkstaAddress;
|
|
|
|
DWORD dwAddressType;
|
|
LPWSTR lpUserName;
|
|
|
|
DWORD dwOpens;
|
|
DWORD dwLogonTime;
|
|
BOOL fLoggedOn;
|
|
DWORD dwForcedLogoffTime;
|
|
BOOL fAdministrator;
|
|
} FPNWCONNECTIONINFO;
|
|
*/
|
|
|
|
|
|
CString g_strFpnwNotLoggedIn;
|
|
BOOL g_fFpnwStringsLoaded = FALSE;
|
|
|
|
|
|
// returns S_FALSE if enumeration should stop
|
|
HRESULT FpnwFileServiceProvider::HandleFPNWSessionItems(
|
|
IResultData* pResultData,
|
|
CFileMgmtCookie* pParentCookie,
|
|
PVOID pinfo,
|
|
DWORD nItems,
|
|
BOOL bAddToResultPane)
|
|
{
|
|
TEST_NONNULL_PTR_PARAM(pParentCookie);
|
|
TEST_NONNULL_PTR_PARAM(pinfo);
|
|
|
|
if (0 >= nItems)
|
|
return S_OK;
|
|
|
|
BOOL fDeleteAllItems = (NULL == pResultData);
|
|
|
|
RESULTDATAITEM tRDItem;
|
|
::ZeroMemory( &tRDItem, sizeof(tRDItem) );
|
|
tRDItem.nCol = COLNUM_SESSIONS_USERNAME;
|
|
tRDItem.mask = RDI_STR | RDI_IMAGE | RDI_PARAM;
|
|
tRDItem.str = MMC_CALLBACK;
|
|
tRDItem.nImage = iIconFPNWSession;
|
|
// CODEWORK should use MMC_ICON_CALLBACK here
|
|
#ifndef UNICODE
|
|
#error
|
|
#endif
|
|
|
|
FPNWCONNECTIONINFO* pconninfo = (FPNWCONNECTIONINFO*)pinfo;
|
|
|
|
CFpnwSessionCookie* pcookiearray = new CFpnwSessionCookie[nItems];
|
|
CFpnwCookieBlock* pCookieBlock = new CFpnwCookieBlock(
|
|
pcookiearray,nItems,pParentCookie->QueryNonNULLMachineName(),pinfo );
|
|
bool bAdded = false;
|
|
if ( !fDeleteAllItems || !bAddToResultPane )
|
|
{
|
|
pParentCookie->m_listResultCookieBlocks.AddHead( pCookieBlock );
|
|
bAdded = true;
|
|
}
|
|
|
|
|
|
for ( ; nItems > 0; nItems--, pconninfo++, pcookiearray++ )
|
|
{
|
|
pcookiearray->m_pobject = pconninfo;
|
|
|
|
if ( bAddToResultPane )
|
|
{
|
|
if (fDeleteAllItems)
|
|
{
|
|
DWORD dwApiResult = CloseSession( pcookiearray );
|
|
if (0L != dwApiResult)
|
|
{
|
|
(void) DoErrMsgBox(GetActiveWindow(), MB_OK | MB_ICONSTOP, dwApiResult,
|
|
IDS_POPUP_FPNW_DISCONNECTALLSESSION_ERROR);
|
|
//return S_FALSE;
|
|
}
|
|
continue;
|
|
}
|
|
|
|
// WARNING cookie cast
|
|
tRDItem.lParam = reinterpret_cast<LPARAM>((CCookie*)pcookiearray);
|
|
HRESULT hr = pResultData->InsertItem(&tRDItem);
|
|
ASSERT(SUCCEEDED(hr));
|
|
}
|
|
}
|
|
|
|
if ( !bAdded ) // they were not added to the parent cookie's list
|
|
delete pCookieBlock;
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
|
|
/*
|
|
DWORD
|
|
FpnwFileEnum(
|
|
IN LPWSTR pServerName OPTIONAL,
|
|
IN DWORD dwLevel,
|
|
IN LPWSTR pPathName OPTIONAL,
|
|
OUT LPBYTE *ppFileInfo,
|
|
OUT PDWORD pEntriesRead,
|
|
IN OUT PDWORD resumeHandle OPTIONAL
|
|
);
|
|
*/
|
|
|
|
typedef DWORD (*FILEENUMPROC) (LPWSTR,DWORD,LPWSTR,LPBYTE*,PDWORD,PDWORD);
|
|
|
|
// if pResultData is not NULL, add sessions/resources to the listbox
|
|
// if pResultData is NULL, delete all sessions/resources
|
|
// if pResultData is NULL, return SUCCEEDED(hr) to continue or
|
|
// FAILED(hr) to abort
|
|
HRESULT FpnwFileServiceProvider::EnumerateResources(
|
|
IResultData* pResultData,
|
|
CFileMgmtCookie* pcookie)
|
|
{
|
|
TEST_NONNULL_PTR_PARAM(pcookie);
|
|
|
|
if ( !g_FpnwDLL.LoadFunctionPointers() )
|
|
return S_OK;
|
|
|
|
FPNWFILEINFO* pfileinfo = NULL;
|
|
DWORD dwEntriesRead = 0;
|
|
DWORD hEnumHandle = 0;
|
|
HRESULT hr = S_OK;
|
|
NET_API_STATUS retval = NERR_Success;
|
|
do {
|
|
retval = ((FILEENUMPROC)g_FpnwDLL[FPNW_FILE_ENUM])(
|
|
const_cast<LPTSTR>(pcookie->QueryTargetServer()),
|
|
1,
|
|
NULL, // basepath
|
|
(PBYTE*)&pfileinfo,
|
|
&dwEntriesRead,
|
|
&hEnumHandle );
|
|
if (NERR_Success == retval)
|
|
{
|
|
hr = HandleFPNWResourceItems( pResultData, pcookie, pfileinfo, dwEntriesRead );
|
|
pfileinfo = NULL;
|
|
break;
|
|
} else if (ERROR_MORE_DATA == retval) {
|
|
ASSERT( NULL != hEnumHandle );
|
|
hr = HandleFPNWResourceItems( pResultData, pcookie, pfileinfo, dwEntriesRead );
|
|
pfileinfo = NULL;
|
|
continue;
|
|
} else if (RPC_S_SERVER_UNAVAILABLE == retval && 0 == hEnumHandle) {
|
|
// FPNW just isn't installed, don't worry about it
|
|
retval = NERR_Success;
|
|
break;
|
|
} else {
|
|
(void) DoErrMsgBox(GetActiveWindow(), MB_OK | MB_ICONSTOP, retval, IDS_POPUP_FPNW_RESOURCES, pcookie->QueryNonNULLMachineName() );
|
|
break;
|
|
}
|
|
} while (S_OK == hr);
|
|
|
|
return HRESULT_FROM_WIN32(retval);
|
|
}
|
|
|
|
|
|
/*
|
|
typedef enum _COLNUM_RESOURCES {
|
|
COLNUM_RESOURCES_FILENAME = 0,
|
|
COLNUM_RESOURCES_USERNAME,
|
|
COLNUM_RESOURCES_NUM_LOCKS, // we don't try to display sharename for now, since
|
|
// only FPNW has this information
|
|
COLNUM_RESOURCES_OPEN_MODE
|
|
} COLNUM_RESOURCES;
|
|
|
|
typedef struct _FPNWFileInfo
|
|
{
|
|
DWORD dwFileId;
|
|
LPWSTR lpPathName;
|
|
LPWSTR lpVolumeName;
|
|
DWORD dwPermissions;
|
|
|
|
|
|
DWORD dwLocks;
|
|
LPWSTR lpUserName;
|
|
|
|
FPNWSERVERADDR WkstaAddress;
|
|
DWORD dwAddressType;
|
|
|
|
} FPNWFILEINFO, *PFPNWFILEINFO;
|
|
*/
|
|
|
|
HRESULT FpnwFileServiceProvider::HandleFPNWResourceItems(
|
|
IResultData* pResultData,
|
|
CFileMgmtCookie* pParentCookie,
|
|
PVOID pinfo,
|
|
DWORD nItems)
|
|
{
|
|
TEST_NONNULL_PTR_PARAM(pParentCookie);
|
|
TEST_NONNULL_PTR_PARAM(pinfo);
|
|
|
|
if (0 >= nItems)
|
|
return S_OK;
|
|
|
|
BOOL fDeleteAllItems = (NULL == pResultData);
|
|
|
|
RESULTDATAITEM tRDItem;
|
|
::ZeroMemory( &tRDItem, sizeof(tRDItem) );
|
|
tRDItem.nCol = COLNUM_RESOURCES_FILENAME;
|
|
tRDItem.mask = RDI_STR | RDI_IMAGE | RDI_PARAM;
|
|
tRDItem.str = MMC_CALLBACK;
|
|
tRDItem.nImage = iIconFPNWResource;
|
|
// CODEWORK should use MMC_ICON_CALLBACK here
|
|
|
|
FPNWFILEINFO* pfileinfo = (FPNWFILEINFO*)pinfo;
|
|
|
|
CFpnwResourceCookie* pcookiearray = new CFpnwResourceCookie[nItems];
|
|
CFpnwCookieBlock* pCookieBlock = new CFpnwCookieBlock(
|
|
pcookiearray,nItems,pParentCookie->QueryNonNULLMachineName(),pinfo );
|
|
if (!fDeleteAllItems)
|
|
{
|
|
pParentCookie->m_listResultCookieBlocks.AddHead( pCookieBlock );
|
|
}
|
|
|
|
CString str;
|
|
for ( ; nItems > 0; nItems--, pfileinfo++, pcookiearray++ )
|
|
{
|
|
pcookiearray->m_pobject = pfileinfo;
|
|
|
|
if (fDeleteAllItems)
|
|
{
|
|
DWORD dwApiResult = CloseResource( pcookiearray );
|
|
if (0L != dwApiResult)
|
|
{
|
|
(void) DoErrMsgBox(GetActiveWindow(), MB_OK | MB_ICONSTOP, dwApiResult,
|
|
IDS_POPUP_FPNW_DISCONNECTALLRESOURCE_ERROR,
|
|
pfileinfo->lpPathName );
|
|
return S_FALSE;
|
|
}
|
|
continue;
|
|
}
|
|
|
|
// WARNING cookie cast
|
|
tRDItem.lParam = reinterpret_cast<LPARAM>((CCookie*)pcookiearray);
|
|
HRESULT hr = pResultData->InsertItem(&tRDItem);
|
|
ASSERT(SUCCEEDED(hr));
|
|
}
|
|
|
|
if (fDeleteAllItems) // they were not added to the parent cookie's list
|
|
delete pCookieBlock;
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
|
|
/*
|
|
DWORD
|
|
FpnwVolumeDel(
|
|
IN LPWSTR pServerName OPTIONAL,
|
|
IN LPWSTR pVolumeName
|
|
);
|
|
*/
|
|
|
|
typedef DWORD (*VOLUMEDELPROC) (LPTSTR,LPTSTR);
|
|
|
|
NET_API_STATUS FpnwFileServiceProvider::DeleteShare( LPCTSTR lpcszServerName, LPCTSTR lpcszShareName )
|
|
{
|
|
if ( !g_FpnwDLL.LoadFunctionPointers() )
|
|
return S_OK;
|
|
|
|
NET_API_STATUS dwRet = ((VOLUMEDELPROC)g_FpnwDLL[FPNW_VOLUME_DEL])(
|
|
const_cast<LPTSTR>(lpcszServerName),
|
|
const_cast<LPTSTR>(lpcszShareName) );
|
|
|
|
return (NERR_NetNameNotFound == dwRet ? NERR_Success : dwRet);
|
|
}
|
|
|
|
/*
|
|
DWORD FpnwConnectionDel(
|
|
IN LPWSTR pServerName OPTIONAL,
|
|
IN DWORD dwConnectionId
|
|
);
|
|
*/
|
|
|
|
typedef DWORD (*CONNECTIONDELPROC) (LPWSTR,DWORD);
|
|
|
|
DWORD FpnwFileServiceProvider::CloseSession(CFileMgmtResultCookie* pcookie)
|
|
{
|
|
if ( !g_FpnwDLL.LoadFunctionPointers() )
|
|
return S_OK;
|
|
|
|
ASSERT( FILEMGMT_SESSION == pcookie->QueryObjectType() );
|
|
FPNWCONNECTIONINFO* pconninfo = (FPNWCONNECTIONINFO*)pcookie->m_pobject;
|
|
ASSERT( NULL != pconninfo );
|
|
|
|
DWORD dwRet = ((CONNECTIONDELPROC)g_FpnwDLL[FPNW_CONNECTION_DEL])(
|
|
const_cast<LPWSTR>(pcookie->QueryTargetServer()),
|
|
pconninfo->dwConnectionId );
|
|
|
|
return (NERR_ClientNameNotFound == dwRet ? NERR_Success : dwRet);
|
|
}
|
|
|
|
/*
|
|
DWORD
|
|
FpnwFileClose(
|
|
IN LPWSTR pServerName OPTIONAL,
|
|
IN DWORD nFileId
|
|
);
|
|
*/
|
|
|
|
typedef DWORD (*FILECLOSEPROC) (LPWSTR,DWORD);
|
|
|
|
DWORD FpnwFileServiceProvider::CloseResource(CFileMgmtResultCookie* pcookie)
|
|
{
|
|
if ( !g_FpnwDLL.LoadFunctionPointers() )
|
|
return S_OK;
|
|
|
|
ASSERT( FILEMGMT_RESOURCE == pcookie->QueryObjectType() );
|
|
FPNWFILEINFO* pfileinfo = (FPNWFILEINFO*)pcookie->m_pobject;
|
|
ASSERT( NULL != pfileinfo );
|
|
|
|
DWORD dwRet = ((FILECLOSEPROC)g_FpnwDLL[FPNW_FILE_CLOSE])(
|
|
const_cast<LPWSTR>(pcookie->QueryTargetServer()),
|
|
pfileinfo->dwFileId );
|
|
|
|
return (NERR_FileIdNotFound == dwRet ? NERR_Success : dwRet);
|
|
}
|
|
|
|
/*
|
|
DWORD
|
|
FpnwVolumeGetInfo(
|
|
IN LPWSTR pServerName OPTIONAL,
|
|
IN LPWSTR pVolumeName,
|
|
IN DWORD dwLevel,
|
|
OUT LPBYTE *ppVolumeInfo
|
|
);
|
|
|
|
|
|
//
|
|
// The following fields are modified by a call to FpnwVolumeSetInfo :
|
|
//
|
|
// DWORD dwMaxUses; // Maximum number of connections that are
|
|
// PSECURITY_DESCRIPTOR FileSecurityDescriptor;
|
|
//
|
|
// All other fields in FPNWVOLUMEINFO structure are ignored. You may send
|
|
// in a pointer to an FPNWVOLUMEINFO_2 structure instead of FPNWVOLUMEINFO.
|
|
//
|
|
|
|
DWORD
|
|
FpnwVolumeSetInfo(
|
|
IN LPWSTR pServerName OPTIONAL,
|
|
IN LPWSTR pVolumeName,
|
|
IN DWORD dwLevel,
|
|
IN LPBYTE pVolumeInfo
|
|
);
|
|
|
|
typedef struct _FPNWVolumeInfo
|
|
{
|
|
LPWSTR lpVolumeName; // Name of the volume
|
|
DWORD dwType; // Specifics of the volume. FPNWVOL_TYPE_???
|
|
DWORD dwMaxUses; // Maximum number of connections that are
|
|
// allowed to the volume
|
|
DWORD dwCurrentUses; // Current number of connections to the volume
|
|
LPWSTR lpPath; // Path of the volume
|
|
|
|
} FPNWVOLUMEINFO, *PFPNWVOLUMEINFO;
|
|
*/
|
|
|
|
typedef DWORD (*VOLUMEGETINFOPROC) (LPWSTR,LPWSTR,DWORD,LPBYTE*);
|
|
typedef DWORD (*VOLUMESETINFOPROC) (LPWSTR,LPWSTR,DWORD,LPBYTE);
|
|
|
|
VOID FpnwFileServiceProvider::DisplayShareProperties(
|
|
LPPROPERTYSHEETCALLBACK pCallBack,
|
|
LPDATAOBJECT pDataObject,
|
|
LONG_PTR handle)
|
|
{
|
|
CSharePageGeneral * pPage = new CSharePageGeneral();
|
|
if ( !pPage->Load( m_pFileMgmtData, pDataObject ) )
|
|
return;
|
|
|
|
// This mechanism deletes the CFileMgmtGeneral when the property sheet is finished
|
|
pPage->m_pfnOriginalPropSheetPageProc = pPage->m_psp.pfnCallback;
|
|
pPage->m_psp.lParam = reinterpret_cast<LPARAM>(pPage);
|
|
pPage->m_psp.pfnCallback = &CSharePageGeneral::PropSheetPageProc;
|
|
pPage->m_handle = handle;
|
|
|
|
MMCPropPageCallback(INOUT &pPage->m_psp);
|
|
HPROPSHEETPAGE hPage=MyCreatePropertySheetPage(&pPage->m_psp);
|
|
pCallBack->AddPage(hPage);
|
|
|
|
CComObject<CFPNWSecurityInformation>* psecinfo = NULL;
|
|
HRESULT hRes = CComObject<CFPNWSecurityInformation>::CreateInstance(&psecinfo);
|
|
if ( SUCCEEDED(hRes) )
|
|
MyCreateShareSecurityPage(
|
|
pCallBack,
|
|
psecinfo,
|
|
pPage->m_strMachineName,
|
|
pPage->m_strShareName );
|
|
|
|
CreateFolderSecurityPropPage(pCallBack, pDataObject);
|
|
|
|
}
|
|
|
|
DWORD FpnwFileServiceProvider::ReadShareProperties(
|
|
LPCTSTR ptchServerName,
|
|
LPCTSTR ptchShareName,
|
|
OUT PVOID* ppvPropertyBlock,
|
|
OUT CString& /*strDescription*/,
|
|
OUT CString& strPath,
|
|
OUT BOOL* pfEditDescription,
|
|
OUT BOOL* pfEditPath,
|
|
OUT DWORD* pdwShareType)
|
|
{
|
|
if ( !g_FpnwDLL.LoadFunctionPointers() )
|
|
{
|
|
ASSERT(FALSE);
|
|
return S_OK;
|
|
}
|
|
|
|
if (ppvPropertyBlock) *ppvPropertyBlock = NULL;
|
|
if (pdwShareType) *pdwShareType = 0;
|
|
if (pfEditDescription) *pfEditDescription = FALSE;
|
|
if (pfEditPath) *pfEditPath = FALSE;
|
|
|
|
USES_CONVERSION; // CODEWORK should store in native WCHAR
|
|
FPNWVOLUMEINFO* pvolumeinfo = NULL;
|
|
NET_API_STATUS retval = ((VOLUMEGETINFOPROC)g_FpnwDLL[FPNW_VOLUME_GET_INFO])(
|
|
T2OLE(const_cast<LPTSTR>(ptchServerName)),
|
|
T2OLE(const_cast<LPTSTR>(ptchShareName)),
|
|
1,
|
|
(LPBYTE*)&pvolumeinfo);
|
|
if (NERR_Success == retval)
|
|
{
|
|
strPath = pvolumeinfo->lpPath;
|
|
if (ppvPropertyBlock)
|
|
{
|
|
*ppvPropertyBlock = pvolumeinfo; // will be freed by the caller
|
|
} else
|
|
{
|
|
FreeData((LPVOID)pvolumeinfo);
|
|
}
|
|
}
|
|
|
|
return retval;
|
|
}
|
|
|
|
DWORD FpnwFileServiceProvider::WriteShareProperties(LPCTSTR ptchServerName, LPCTSTR ptchShareName,
|
|
PVOID pvPropertyBlock, LPCTSTR /*ptchDescription*/, LPCTSTR ptchPath)
|
|
{
|
|
ASSERT( NULL != pvPropertyBlock );
|
|
if ( !g_FpnwDLL.LoadFunctionPointers() )
|
|
return S_OK;
|
|
|
|
FPNWVOLUMEINFO* pvolumeinfo = (FPNWVOLUMEINFO*)pvPropertyBlock;
|
|
pvolumeinfo->lpPath = const_cast<LPTSTR>(ptchPath);
|
|
DWORD retval = ((VOLUMESETINFOPROC)g_FpnwDLL[FPNW_VOLUME_SET_INFO])(
|
|
const_cast<LPTSTR>(ptchServerName),
|
|
const_cast<LPTSTR>(ptchShareName),
|
|
1,
|
|
(LPBYTE)pvolumeinfo);
|
|
pvolumeinfo->lpPath = NULL;
|
|
return retval;
|
|
}
|
|
|
|
VOID FpnwFileServiceProvider::FreeShareProperties(PVOID pvPropertyBlock)
|
|
{
|
|
FreeData( pvPropertyBlock );
|
|
}
|
|
|
|
|
|
DWORD FpnwFileServiceProvider::QueryMaxUsers(PVOID pvPropertyBlock)
|
|
{
|
|
FPNWVOLUMEINFO* pvolumeinfo = (FPNWVOLUMEINFO*)pvPropertyBlock;
|
|
ASSERT( NULL != pvolumeinfo );
|
|
return pvolumeinfo->dwMaxUses;
|
|
}
|
|
|
|
VOID FpnwFileServiceProvider::SetMaxUsers(PVOID pvPropertyBlock, DWORD dwMaxUsers)
|
|
{
|
|
FPNWVOLUMEINFO* pvolumeinfo = (FPNWVOLUMEINFO*)pvPropertyBlock;
|
|
ASSERT( NULL != pvolumeinfo );
|
|
pvolumeinfo->dwMaxUses = dwMaxUsers;
|
|
}
|
|
VOID FpnwFileServiceProvider::FreeData(PVOID pv)
|
|
{
|
|
FPNWFreeData( &pv );
|
|
}
|
|
|
|
CFpnwCookieBlock::~CFpnwCookieBlock()
|
|
{
|
|
FPNWFreeData( &m_pvCookieData );
|
|
}
|
|
|
|
DEFINE_COOKIE_BLOCK(CFpnwCookie)
|
|
DEFINE_FORWARDS_MACHINE_NAME( CFpnwCookie, m_pCookieBlock )
|
|
|
|
void CFpnwCookie::AddRefCookie() { m_pCookieBlock->AddRef(); }
|
|
void CFpnwCookie::ReleaseCookie() { m_pCookieBlock->Release(); }
|
|
|
|
LPCTSTR FpnwFileServiceProvider::QueryTransportString()
|
|
{
|
|
return m_strTransportFPNW;
|
|
}
|
|
|
|
HRESULT CFpnwCookie::GetTransport( FILEMGMT_TRANSPORT* pTransport )
|
|
{
|
|
*pTransport = FILEMGMT_FPNW;
|
|
return S_OK;
|
|
}
|
|
|
|
HRESULT CFpnwShareCookie::GetShareName( CString& strShareName )
|
|
{
|
|
FPNWVOLUMEINFO* pvolumeinfo = (FPNWVOLUMEINFO*)m_pobject;
|
|
ASSERT( NULL != pvolumeinfo );
|
|
USES_CONVERSION;
|
|
strShareName = OLE2T(pvolumeinfo->lpVolumeName);
|
|
return S_OK;
|
|
}
|
|
|
|
HRESULT CFpnwShareCookie::GetSharePIDList( OUT LPITEMIDLIST *ppidl )
|
|
{
|
|
ASSERT(ppidl);
|
|
ASSERT(NULL == *ppidl); // prevent memory leak
|
|
*ppidl = NULL;
|
|
|
|
FPNWVOLUMEINFO* pvolumeinfo = (FPNWVOLUMEINFO*)m_pobject;
|
|
ASSERT( NULL != pvolumeinfo );
|
|
ASSERT( _tcslen(pvolumeinfo->lpPath) >= 3 &&
|
|
_T(':') == *(pvolumeinfo->lpPath + 1) );
|
|
USES_CONVERSION;
|
|
|
|
PCTSTR pszTargetServer = m_pCookieBlock->QueryTargetServer();
|
|
CString csPath;
|
|
|
|
if (pszTargetServer)
|
|
{
|
|
//
|
|
// since MS Windows user cannot see shares created for MAC or Netware users,
|
|
// we have to use $ share to retrieve the pidl here.
|
|
//
|
|
CString csTemp = OLE2T(pvolumeinfo->lpPath);
|
|
csTemp.SetAt(1, _T('$'));
|
|
if ( _tcslen(pszTargetServer) >= 2 &&
|
|
_T('\\') == *pszTargetServer &&
|
|
_T('\\') == *(pszTargetServer + 1) )
|
|
{
|
|
csPath = pszTargetServer;
|
|
} else
|
|
{
|
|
csPath = _T("\\\\");
|
|
csPath += pszTargetServer;
|
|
}
|
|
csPath += _T("\\");
|
|
csPath += csTemp;
|
|
} else
|
|
{
|
|
csPath = OLE2T(pvolumeinfo->lpPath);
|
|
}
|
|
|
|
if (FALSE == csPath.IsEmpty())
|
|
*ppidl = ILCreateFromPath(csPath);
|
|
|
|
return ((*ppidl) ? S_OK : E_FAIL);
|
|
}
|
|
|
|
HRESULT CFpnwSessionCookie::GetSessionID( DWORD* pdwSessionID )
|
|
{
|
|
FPNWCONNECTIONINFO* pconninfo = (FPNWCONNECTIONINFO*)m_pobject;
|
|
ASSERT( NULL != pdwSessionID && NULL != pconninfo );
|
|
*pdwSessionID = pconninfo->dwConnectionId;
|
|
return S_OK;
|
|
}
|
|
|
|
HRESULT CFpnwResourceCookie::GetFileID( DWORD* pdwFileID )
|
|
{
|
|
FPNWFILEINFO* pfileinfo = (FPNWFILEINFO*)m_pobject;
|
|
ASSERT( NULL != pdwFileID && NULL != pfileinfo );
|
|
*pdwFileID = pfileinfo->dwFileId;
|
|
return S_OK;
|
|
}
|
|
|
|
BSTR CFpnwShareCookie::GetColumnText( int nCol )
|
|
{
|
|
switch (nCol)
|
|
{
|
|
case COLNUM_SHARES_SHARED_FOLDER:
|
|
return GetShareInfo()->lpVolumeName;
|
|
case COLNUM_SHARES_SHARED_PATH:
|
|
return GetShareInfo()->lpPath;
|
|
case COLNUM_SHARES_TRANSPORT:
|
|
return const_cast<BSTR>((LPCTSTR)g_strTransportFPNW);
|
|
case COLNUM_SHARES_COMMENT:
|
|
break; // not known for FPNW
|
|
default:
|
|
ASSERT(FALSE);
|
|
break;
|
|
}
|
|
return L"";
|
|
}
|
|
|
|
BSTR CFpnwShareCookie::QueryResultColumnText( int nCol, CFileMgmtComponentData& /*refcdata*/ )
|
|
{
|
|
if (COLNUM_SHARES_NUM_SESSIONS == nCol)
|
|
return MakeDwordResult( GetNumOfCurrentUses() );
|
|
|
|
return GetColumnText(nCol);
|
|
}
|
|
|
|
BSTR CFpnwSessionCookie::GetColumnText( int nCol )
|
|
{
|
|
switch (nCol)
|
|
{
|
|
case COLNUM_SESSIONS_USERNAME:
|
|
return GetSessionInfo()->lpUserName;
|
|
case COLNUM_SESSIONS_COMPUTERNAME:
|
|
break; // not known for FPNW
|
|
case COLNUM_SESSIONS_TRANSPORT:
|
|
return const_cast<BSTR>((LPCTSTR)g_strTransportFPNW);
|
|
case COLNUM_SESSIONS_IS_GUEST:
|
|
break; // not known for FPNW
|
|
default:
|
|
ASSERT(FALSE);
|
|
break;
|
|
}
|
|
return L"";
|
|
}
|
|
|
|
BSTR CFpnwSessionCookie::QueryResultColumnText( int nCol, CFileMgmtComponentData& /*refcdata*/ )
|
|
{
|
|
switch (nCol)
|
|
{
|
|
case COLNUM_SESSIONS_NUM_FILES:
|
|
return MakeDwordResult( GetNumOfOpenFiles() );
|
|
case COLNUM_SESSIONS_CONNECTED_TIME:
|
|
return MakeElapsedTimeResult( GetConnectedTime() );
|
|
case COLNUM_SESSIONS_IDLE_TIME:
|
|
return L""; // not known for FPNW
|
|
default:
|
|
break;
|
|
}
|
|
return GetColumnText(nCol);
|
|
}
|
|
|
|
BSTR CFpnwResourceCookie::GetColumnText( int nCol )
|
|
{
|
|
switch (nCol)
|
|
{
|
|
case COLNUM_RESOURCES_FILENAME:
|
|
return GetFileInfo()->lpPathName;
|
|
case COLNUM_RESOURCES_USERNAME:
|
|
return GetFileInfo()->lpUserName;
|
|
case COLNUM_RESOURCES_TRANSPORT:
|
|
return const_cast<BSTR>((LPCTSTR)g_strTransportFPNW);
|
|
case COLNUM_RESOURCES_OPEN_MODE:
|
|
return MakePermissionsResult(GetFileInfo()->dwPermissions);
|
|
default:
|
|
ASSERT(FALSE);
|
|
break;
|
|
}
|
|
return L"";
|
|
}
|
|
|
|
BSTR CFpnwResourceCookie::QueryResultColumnText( int nCol, CFileMgmtComponentData& /*refcdata*/ )
|
|
{
|
|
if (COLNUM_RESOURCES_NUM_LOCKS == nCol)
|
|
return MakeDwordResult( GetNumOfLocks() );
|
|
|
|
return GetColumnText(nCol);
|
|
}
|
|
|
|
CFPNWSecurityInformation::CFPNWSecurityInformation()
|
|
: m_pvolumeinfo( NULL ),
|
|
m_pDefaultDescriptor( NULL )
|
|
{
|
|
}
|
|
|
|
CFPNWSecurityInformation::~CFPNWSecurityInformation()
|
|
{
|
|
if (NULL != m_pDefaultDescriptor)
|
|
{
|
|
delete m_pDefaultDescriptor;
|
|
m_pvolumeinfo->FileSecurityDescriptor = NULL;
|
|
}
|
|
FPNWFreeData( (PVOID*)&m_pvolumeinfo );
|
|
}
|
|
|
|
STDMETHODIMP CFPNWSecurityInformation::GetSecurity (
|
|
SECURITY_INFORMATION RequestedInformation,
|
|
PSECURITY_DESCRIPTOR *ppSecurityDescriptor,
|
|
BOOL fDefault )
|
|
{
|
|
MFC_TRY;
|
|
|
|
if (0 == RequestedInformation || NULL == ppSecurityDescriptor)
|
|
{
|
|
ASSERT(FALSE);
|
|
return E_INVALIDARG;
|
|
}
|
|
|
|
if (fDefault)
|
|
return E_NOTIMPL;
|
|
|
|
*ppSecurityDescriptor = NULL;
|
|
|
|
if ( !g_FpnwDLL.LoadFunctionPointers() )
|
|
return S_OK;
|
|
|
|
FPNWFreeData( (PVOID*)&m_pvolumeinfo );
|
|
NET_API_STATUS dwErr = ((VOLUMEGETINFOPROC)g_FpnwDLL[FPNW_VOLUME_GET_INFO])(
|
|
QueryMachineName(),
|
|
QueryShareName(),
|
|
2,
|
|
(LPBYTE*)&m_pvolumeinfo );
|
|
if (NERR_Success != dwErr)
|
|
{
|
|
// AndyHe confirms errors are in WIN32 error namespace
|
|
return HRESULT_FROM_WIN32(dwErr);
|
|
}
|
|
ASSERT( NULL != m_pvolumeinfo );
|
|
if (NULL == m_pvolumeinfo->FileSecurityDescriptor)
|
|
{
|
|
if (NULL == m_pDefaultDescriptor)
|
|
{
|
|
HRESULT hr = NewDefaultDescriptor(
|
|
&m_pDefaultDescriptor,
|
|
RequestedInformation );
|
|
if ( !SUCCEEDED(hr) )
|
|
return hr;
|
|
}
|
|
m_pvolumeinfo->FileSecurityDescriptor = m_pDefaultDescriptor;
|
|
}
|
|
ASSERT( NULL != m_pvolumeinfo->FileSecurityDescriptor );
|
|
|
|
// We have to pass back a LocalAlloc'ed copy of the SD
|
|
return MakeSelfRelativeCopy(
|
|
m_pvolumeinfo->FileSecurityDescriptor,
|
|
ppSecurityDescriptor );
|
|
|
|
MFC_CATCH;
|
|
}
|
|
|
|
STDMETHODIMP CFPNWSecurityInformation::SetSecurity (
|
|
SECURITY_INFORMATION SecurityInformation,
|
|
PSECURITY_DESCRIPTOR pSecurityDescriptor )
|
|
{
|
|
MFC_TRY;
|
|
|
|
if ( !g_FpnwDLL.LoadFunctionPointers() )
|
|
return S_OK;
|
|
|
|
// First get the current settings
|
|
PSECURITY_DESCRIPTOR dummy;
|
|
HRESULT hr = GetSecurity( SecurityInformation, &dummy, FALSE );
|
|
if ( FAILED(hr) )
|
|
return hr;
|
|
|
|
// Now set the new values
|
|
m_pvolumeinfo->FileSecurityDescriptor = pSecurityDescriptor;
|
|
NET_API_STATUS dwErr = ((VOLUMESETINFOPROC)g_FpnwDLL[FPNW_VOLUME_SET_INFO])(
|
|
QueryMachineName(),
|
|
QueryShareName(),
|
|
2,
|
|
(LPBYTE)m_pvolumeinfo );
|
|
if (NERR_Success != dwErr)
|
|
{
|
|
return HRESULT_FROM_WIN32(dwErr);
|
|
}
|
|
|
|
return S_OK;
|
|
|
|
MFC_CATCH;
|
|
}
|