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.
2483 lines
69 KiB
2483 lines
69 KiB
// smb.cpp : SMB shares, sessions and open resources
|
|
|
|
#include "stdafx.h"
|
|
#include "cmponent.h"
|
|
#include "safetemp.h"
|
|
#include "FileSvc.h"
|
|
#include "DynamLnk.h" // DynamicDLL
|
|
#include "smb.h"
|
|
#include "ShrPgSMB.h" // Share Properties Pages
|
|
#include "permpage.h" // CSecurityInformation
|
|
#include "compdata.h"
|
|
#include "shrpub.h"
|
|
#include <activeds.h>
|
|
#include <dsrole.h>
|
|
#include <dsgetdc.h>
|
|
#include <lmwksta.h>
|
|
#include <winsock2.h>
|
|
|
|
#define DONT_WANT_SHELLDEBUG
|
|
#include "shlobjp.h" // LPITEMIDLIST
|
|
#include "wraps.h" // Wrap_ILCreateFromPath
|
|
|
|
#include "macros.h"
|
|
USE_HANDLE_MACROS("FILEMGMT(smb.cpp)")
|
|
|
|
#ifdef _DEBUG
|
|
#define new DEBUG_NEW
|
|
#undef THIS_FILE
|
|
static char THIS_FILE[] = __FILE__;
|
|
#endif
|
|
|
|
//
|
|
// no Publish page for these system shares
|
|
//
|
|
LPCTSTR g_pszSystemShares[] = { _T("SYSVOL"), _T("NETLOGON"), _T("DEBUG") };
|
|
|
|
class CSMBSecurityInformation : public CShareSecurityInformation
|
|
{
|
|
STDMETHOD(GetSecurity) (SECURITY_INFORMATION RequestedInformation,
|
|
PSECURITY_DESCRIPTOR *ppSecurityDescriptor,
|
|
BOOL fDefault );
|
|
STDMETHOD(SetSecurity) (SECURITY_INFORMATION SecurityInformation,
|
|
PSECURITY_DESCRIPTOR pSecurityDescriptor );
|
|
public:
|
|
SHARE_INFO_502* m_pvolumeinfo;
|
|
PSECURITY_DESCRIPTOR m_pDefaultDescriptor;
|
|
CSMBSecurityInformation();
|
|
~CSMBSecurityInformation();
|
|
};
|
|
|
|
typedef enum _SmbApiIndex
|
|
{
|
|
SMB_SHARE_ENUM = 0,
|
|
SMB_SESSION_ENUM,
|
|
SMB_FILE_ENUM,
|
|
SMB_API_BUFFER_FREE,
|
|
SMB_SHARE_DEL,
|
|
SMB_SESSION_DEL,
|
|
SMB_FILE_CLOSE,
|
|
SMB_SHARE_GET_INFO,
|
|
SMB_SHARE_SET_INFO,
|
|
SMB_CONNECTION_ENUM
|
|
};
|
|
|
|
// not subject to localization
|
|
static LPCSTR g_apchFunctionNames[] = {
|
|
"NetShareEnum",
|
|
"NetSessionEnum",
|
|
"NetFileEnum",
|
|
"NetApiBufferFree",
|
|
"NetShareDel",
|
|
"NetSessionDel",
|
|
"NetFileClose",
|
|
"NetShareGetInfo",
|
|
"NetShareSetInfo",
|
|
"NetConnectionEnum",
|
|
NULL
|
|
};
|
|
|
|
// not subject to localization
|
|
DynamicDLL g_SmbDLL( _T("NETAPI32.DLL"), g_apchFunctionNames );
|
|
|
|
typedef DWORD (*APIBUFFERFREEPROC) (LPVOID);
|
|
|
|
VOID SMBFreeData(PVOID* ppv)
|
|
{
|
|
if (*ppv != NULL)
|
|
{
|
|
ASSERT( NULL != g_SmbDLL[SMB_API_BUFFER_FREE] );
|
|
(void) ((APIBUFFERFREEPROC)g_SmbDLL[SMB_API_BUFFER_FREE])( *ppv );
|
|
*ppv = NULL;
|
|
}
|
|
}
|
|
|
|
|
|
SmbFileServiceProvider::SmbFileServiceProvider( CFileMgmtComponentData* pFileMgmtData )
|
|
// not subject to localization
|
|
: FileServiceProvider( pFileMgmtData )
|
|
{
|
|
VERIFY( m_strTransportSMB.LoadString( IDS_TRANSPORT_SMB ) );
|
|
}
|
|
|
|
/*
|
|
NET_API_STATUS NET_API_FUNCTION
|
|
NetShareEnum (
|
|
IN LPTSTR servername,
|
|
IN DWORD level,
|
|
OUT LPBYTE *bufptr,
|
|
IN DWORD prefmaxlen,
|
|
OUT LPDWORD entriesread,
|
|
OUT LPDWORD totalentries,
|
|
IN OUT LPDWORD resume_handle
|
|
);
|
|
*/
|
|
|
|
typedef DWORD (*SHAREENUMPROC) (LPTSTR,DWORD,LPBYTE*,DWORD,LPDWORD,LPDWORD,LPDWORD);
|
|
|
|
HRESULT SmbFileServiceProvider::PopulateShares(
|
|
IResultData* pResultData,
|
|
CFileMgmtCookie* pcookie)
|
|
{
|
|
TEST_NONNULL_PTR_PARAM(pcookie);
|
|
|
|
if ( !g_SmbDLL.LoadFunctionPointers() )
|
|
{
|
|
ASSERT(FALSE); // NETAPI32 isn't installed?
|
|
return S_OK;
|
|
}
|
|
|
|
SHARE_INFO_2* psi2 = NULL;
|
|
DWORD dwEntriesRead = 0;
|
|
DWORD dwTotalEntries = 0;
|
|
DWORD hEnumHandle = 0;
|
|
NET_API_STATUS retval = NERR_Success;
|
|
do {
|
|
retval = ((SHAREENUMPROC)g_SmbDLL[SMB_SHARE_ENUM])(
|
|
const_cast<LPTSTR>(pcookie->QueryTargetServer()),
|
|
2,
|
|
(PBYTE*)&psi2,
|
|
(DWORD)-1L,
|
|
&dwEntriesRead,
|
|
&dwTotalEntries,
|
|
&hEnumHandle );
|
|
if (NERR_Success == retval)
|
|
{
|
|
AddSMBShareItems( pResultData, pcookie, psi2, dwEntriesRead );
|
|
psi2 = NULL;
|
|
break;
|
|
} else if (ERROR_MORE_DATA == retval) {
|
|
ASSERT( NULL != hEnumHandle );
|
|
AddSMBShareItems( pResultData, pcookie, psi2, dwEntriesRead );
|
|
psi2 = NULL;
|
|
continue;
|
|
/*
|
|
} else if (RPC_S_SERVER_UNAVAILABLE == retval && 0 == hEnumHandle) {
|
|
// SMB just isn't installed, don't worry about it
|
|
break;
|
|
*/
|
|
} else {
|
|
if (ERROR_ACCESS_DENIED == retval)
|
|
{
|
|
(void) DoErrMsgBox(
|
|
GetActiveWindow(),
|
|
MB_OK | MB_ICONSTOP,
|
|
0,
|
|
IDS_POPUP_SMB_SHARES_NOACCESS
|
|
);
|
|
} else
|
|
{
|
|
(void) DoErrMsgBox(
|
|
GetActiveWindow(),
|
|
MB_OK | MB_ICONSTOP,
|
|
retval,
|
|
IDS_POPUP_SMB_SHARES
|
|
);
|
|
}
|
|
break;
|
|
}
|
|
} while (TRUE);
|
|
|
|
return HRESULT_FROM_WIN32(retval);
|
|
}
|
|
|
|
//
|
|
// skip sharenames that contain leading/trailing spaces
|
|
//
|
|
BOOL IsInvalidSharename(LPCTSTR psz)
|
|
{
|
|
return (!psz || !*psz || _istspace(psz[0]) || _istspace(psz[lstrlen(psz) - 1]));
|
|
}
|
|
|
|
/*
|
|
typedef struct _SHARE_INFO_2 {
|
|
LPTSTR shi2_netname;
|
|
DWORD shi2_type;
|
|
LPTSTR shi2_remark;
|
|
DWORD shi2_permissions;
|
|
DWORD shi2_max_uses;
|
|
DWORD shi2_current_uses;
|
|
LPTSTR shi2_path;
|
|
LPTSTR shi2_passwd;
|
|
} SHARE_INFO_2, *PSHARE_INFO_2, *LPSHARE_INFO_2;
|
|
*/
|
|
HRESULT SmbFileServiceProvider::AddSMBShareItems(
|
|
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) );
|
|
// CODEWORK should use MMC_ICON_CALLBACK
|
|
tRDItem.nCol = COLNUM_SHARES_SHARED_FOLDER;
|
|
tRDItem.mask = RDI_STR | RDI_IMAGE | RDI_PARAM;
|
|
tRDItem.str = MMC_CALLBACK;
|
|
|
|
SHARE_INFO_2* psi2 = (SHARE_INFO_2*)pinfo;
|
|
|
|
DWORD nItemsToAdd = 0;
|
|
for (DWORD i = 0; i < nItems; i++ )
|
|
{
|
|
switch ((psi2[i]).shi2_type)
|
|
{
|
|
case STYPE_PRINTQ: // Do not show print shares
|
|
case STYPE_DEVICE: // Do not show device shares
|
|
break;
|
|
|
|
default:
|
|
if (!IsInvalidSharename(psi2[i].shi2_netname))
|
|
nItemsToAdd++;
|
|
break;
|
|
}
|
|
}
|
|
|
|
CSmbShareCookie* pcookiearray = new CSmbShareCookie[nItemsToAdd];
|
|
CSmbCookieBlock* pCookieBlock = new CSmbCookieBlock(
|
|
pcookiearray,
|
|
nItemsToAdd,
|
|
pParentCookie->QueryNonNULLMachineName(),
|
|
pinfo );
|
|
pParentCookie->m_listResultCookieBlocks.AddHead( pCookieBlock );
|
|
|
|
CString str;
|
|
|
|
for ( ; nItems > 0; nItems--, psi2++ )
|
|
{
|
|
switch (psi2->shi2_type)
|
|
{
|
|
case STYPE_PRINTQ: // Do not show print shares
|
|
case STYPE_DEVICE: // Do not show device shares
|
|
continue;
|
|
|
|
default:
|
|
if (!IsInvalidSharename(psi2->shi2_netname))
|
|
{
|
|
pcookiearray->m_pobject = psi2;
|
|
// WARNING cookie cast
|
|
tRDItem.lParam = reinterpret_cast<LPARAM>((CCookie*)pcookiearray);
|
|
|
|
if (psi2->shi2_path &&
|
|
lstrlen(psi2->shi2_path) == 3 &&
|
|
*(psi2->shi2_path + 1) == _T(':') &&
|
|
*(psi2->shi2_path + 2) == _T('\\') &&
|
|
(*psi2->shi2_path >= _T('a') && *psi2->shi2_path <= _T('z') ||
|
|
*psi2->shi2_path >= _T('A') && *psi2->shi2_path <= _T('Z')))
|
|
{
|
|
tRDItem.nImage = iIconSMBShare;
|
|
} else
|
|
{
|
|
tRDItem.nImage = iIconSharesFolder;
|
|
}
|
|
|
|
HRESULT hr = pResultData->InsertItem(&tRDItem);
|
|
ASSERT(SUCCEEDED(hr));
|
|
pcookiearray++;
|
|
}
|
|
break;
|
|
}
|
|
}
|
|
ASSERT( pcookiearray ==
|
|
((CSmbShareCookie*)(pCookieBlock->QueryBaseCookie(0)))+nItemsToAdd );
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
|
|
/*
|
|
NET_API_STATUS NET_API_FUNCTION
|
|
NetSessionEnum (
|
|
IN LPTSTR servername OPTIONAL,
|
|
IN LPTSTR UncClientName OPTIONAL,
|
|
IN LPTSTR username OPTIONAL,
|
|
IN DWORD level,
|
|
OUT LPBYTE *bufptr,
|
|
IN DWORD prefmaxlen,
|
|
OUT LPDWORD entriesread,
|
|
OUT LPDWORD totalentries,
|
|
IN OUT LPDWORD resume_handle OPTIONAL
|
|
);
|
|
*/
|
|
|
|
typedef DWORD (*SESSIONENUMPROC) (LPTSTR,LPTSTR,LPTSTR,DWORD,LPBYTE*,DWORD,LPDWORD,LPDWORD,LPDWORD);
|
|
|
|
// 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 SmbFileServiceProvider::EnumerateSessions(
|
|
IResultData* pResultData,
|
|
CFileMgmtCookie* pcookie,
|
|
bool bAddToResultPane)
|
|
{
|
|
TEST_NONNULL_PTR_PARAM(pcookie);
|
|
|
|
if ( !g_SmbDLL.LoadFunctionPointers() )
|
|
return S_OK;
|
|
|
|
SESSION_INFO_1* psi1 = NULL;
|
|
DWORD dwEntriesRead = 0;
|
|
DWORD dwTotalEntries = 0;
|
|
DWORD hEnumHandle = 0;
|
|
HRESULT hr = S_OK;
|
|
NET_API_STATUS retval = NERR_Success;
|
|
do {
|
|
retval = ((SESSIONENUMPROC)g_SmbDLL[SMB_SESSION_ENUM])(
|
|
const_cast<LPTSTR>(pcookie->QueryTargetServer()),
|
|
NULL,
|
|
NULL,
|
|
1,
|
|
(PBYTE*)&psi1,
|
|
(DWORD)-1L,
|
|
&dwEntriesRead,
|
|
&dwTotalEntries,
|
|
&hEnumHandle );
|
|
if (NERR_Success == retval)
|
|
{
|
|
hr = HandleSMBSessionItems( pResultData, pcookie, psi1, dwEntriesRead,
|
|
bAddToResultPane );
|
|
psi1 = NULL;
|
|
break;
|
|
} else if (ERROR_MORE_DATA == retval) {
|
|
ASSERT( NULL != hEnumHandle );
|
|
hr = HandleSMBSessionItems( pResultData, pcookie, psi1, dwEntriesRead,
|
|
bAddToResultPane );
|
|
psi1 = NULL;
|
|
continue;
|
|
} else {
|
|
if (ERROR_ACCESS_DENIED == retval)
|
|
{
|
|
(void) DoErrMsgBox(
|
|
GetActiveWindow(),
|
|
MB_OK | MB_ICONSTOP,
|
|
0,
|
|
IDS_POPUP_SMB_SESSIONS_NOACCESS
|
|
);
|
|
} else
|
|
{
|
|
(void) DoErrMsgBox(
|
|
GetActiveWindow(),
|
|
MB_OK | MB_ICONSTOP,
|
|
retval,
|
|
IDS_POPUP_SMB_SESSIONS
|
|
);
|
|
}
|
|
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 _SESSION_INFO_1 {
|
|
LPTSTR sesi1_cname; // client name (no backslashes)
|
|
LPTSTR sesi1_username;
|
|
DWORD sesi1_num_opens;
|
|
DWORD sesi1_time;
|
|
DWORD sesi1_idle_time;
|
|
DWORD sesi1_user_flags;
|
|
} SESSION_INFO_1, *PSESSION_INFO_1, *LPSESSION_INFO_1;
|
|
|
|
*/
|
|
|
|
|
|
// 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 SmbFileServiceProvider::HandleSMBSessionItems(
|
|
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) );
|
|
// CODEWORK should use MMC_ICON_CALLBACK
|
|
tRDItem.nImage = iIconSMBSession;
|
|
tRDItem.nCol = COLNUM_SESSIONS_USERNAME;
|
|
tRDItem.mask = RDI_STR | RDI_IMAGE | RDI_PARAM;
|
|
tRDItem.str = MMC_CALLBACK;
|
|
|
|
SESSION_INFO_1* psi1 = (SESSION_INFO_1*)pinfo;
|
|
|
|
CSmbSessionCookie* pcookiearray = new CSmbSessionCookie[nItems];
|
|
CSmbCookieBlock* pCookieBlock = new CSmbCookieBlock(
|
|
pcookiearray,nItems,pParentCookie->QueryNonNULLMachineName(),pinfo );
|
|
bool bAdded = false;
|
|
if ( !fDeleteAllItems || !bAddToResultPane )
|
|
{
|
|
pParentCookie->m_listResultCookieBlocks.AddHead( pCookieBlock );
|
|
bAdded = true;
|
|
}
|
|
|
|
for ( ; nItems > 0; nItems--, psi1++, pcookiearray++ )
|
|
{
|
|
pcookiearray->m_pobject = psi1;
|
|
|
|
if ( bAddToResultPane )
|
|
{
|
|
if (fDeleteAllItems)
|
|
{
|
|
DWORD dwApiResult = CloseSession( pcookiearray );
|
|
if (0L != dwApiResult)
|
|
{
|
|
CString strName;
|
|
TranslateIPToComputerName(psi1->sesi1_cname, strName);
|
|
(void) DoErrMsgBox(GetActiveWindow(), MB_OK | MB_ICONSTOP, dwApiResult,
|
|
IDS_POPUP_SMB_DISCONNECTALLSESSION_ERROR,
|
|
strName );
|
|
//return S_FALSE;
|
|
}
|
|
continue;
|
|
}
|
|
|
|
// WARNING cookie cast
|
|
if (psi1->sesi1_username && *(psi1->sesi1_username)) // bug#3903: exclude NULL session
|
|
{
|
|
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;
|
|
}
|
|
|
|
|
|
/*
|
|
NET_API_STATUS NET_API_FUNCTION
|
|
NetFileEnum (
|
|
IN LPTSTR servername OPTIONAL,
|
|
IN LPTSTR basepath OPTIONAL,
|
|
IN LPTSTR username OPTIONAL,
|
|
IN DWORD level,
|
|
OUT LPBYTE *bufptr,
|
|
IN DWORD prefmaxlen,
|
|
OUT LPDWORD entriesread,
|
|
OUT LPDWORD totalentries,
|
|
IN OUT LPDWORD resume_handle OPTIONAL
|
|
);
|
|
|
|
*/
|
|
|
|
typedef DWORD (*FILEENUMPROC) (LPTSTR,LPTSTR,LPTSTR,DWORD,LPBYTE*,DWORD,LPDWORD,LPDWORD,LPDWORD);
|
|
|
|
// 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 SmbFileServiceProvider::EnumerateResources(
|
|
IResultData* pResultData,
|
|
CFileMgmtCookie* pcookie)
|
|
{
|
|
TEST_NONNULL_PTR_PARAM(pcookie);
|
|
|
|
if ( !g_SmbDLL.LoadFunctionPointers() )
|
|
return S_OK;
|
|
|
|
FILE_INFO_3* pfi3 = NULL;
|
|
DWORD dwEntriesRead = 0;
|
|
DWORD dwTotalEntries = 0;
|
|
DWORD hEnumHandle = 0;
|
|
HRESULT hr = S_OK;
|
|
NET_API_STATUS retval = NERR_Success;
|
|
do {
|
|
retval = ((FILEENUMPROC)g_SmbDLL[SMB_FILE_ENUM])(
|
|
const_cast<LPTSTR>(pcookie->QueryTargetServer()),
|
|
NULL,
|
|
NULL,
|
|
3,
|
|
(PBYTE*)&pfi3,
|
|
(DWORD)-1L,
|
|
&dwEntriesRead,
|
|
&dwTotalEntries,
|
|
&hEnumHandle );
|
|
if (NERR_Success == retval)
|
|
{
|
|
hr = HandleSMBResourceItems( pResultData, pcookie, pfi3, dwEntriesRead );
|
|
pfi3 = NULL;
|
|
break;
|
|
} else if (ERROR_MORE_DATA == retval) {
|
|
ASSERT( NULL != hEnumHandle );
|
|
hr = HandleSMBResourceItems( pResultData, pcookie, pfi3, dwEntriesRead );
|
|
pfi3 = NULL;
|
|
continue;
|
|
} else {
|
|
if (ERROR_ACCESS_DENIED == retval)
|
|
{
|
|
(void) DoErrMsgBox(
|
|
GetActiveWindow(),
|
|
MB_OK | MB_ICONSTOP,
|
|
0,
|
|
IDS_POPUP_SMB_RESOURCES_NOACCESS
|
|
);
|
|
} else
|
|
{
|
|
(void) DoErrMsgBox(
|
|
GetActiveWindow(),
|
|
MB_OK | MB_ICONSTOP,
|
|
retval,
|
|
IDS_POPUP_SMB_RESOURCES
|
|
);
|
|
}
|
|
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 SMB has this information
|
|
COLNUM_RESOURCES_OPEN_MODE
|
|
} COLNUM_RESOURCES;
|
|
|
|
typedef struct _FILE_INFO_3 {
|
|
DWORD fi3_id;
|
|
DWORD fi3_permissions;
|
|
DWORD fi3_num_locks;
|
|
LPTSTR fi3_pathname;
|
|
LPTSTR fi3_username;
|
|
} FILE_INFO_3, *PFILE_INFO_3, *LPFILE_INFO_3;
|
|
*/
|
|
|
|
// 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 SmbFileServiceProvider::HandleSMBResourceItems(
|
|
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) );
|
|
// CODEWORK should use MMC_ICON_CALLBACK
|
|
tRDItem.nImage = iIconSMBResource;
|
|
tRDItem.nCol = COLNUM_RESOURCES_FILENAME;
|
|
tRDItem.mask = RDI_STR | RDI_IMAGE | RDI_PARAM;
|
|
tRDItem.str = MMC_CALLBACK;
|
|
|
|
FILE_INFO_3* pfi3 = (FILE_INFO_3*)pinfo;
|
|
|
|
CSmbResourceCookie* pcookiearray = new CSmbResourceCookie[nItems];
|
|
CSmbCookieBlock* pCookieBlock = new CSmbCookieBlock(
|
|
pcookiearray,nItems,pParentCookie->QueryNonNULLMachineName(),pinfo );
|
|
if (!fDeleteAllItems)
|
|
{
|
|
pParentCookie->m_listResultCookieBlocks.AddHead( pCookieBlock );
|
|
}
|
|
|
|
CString str;
|
|
for ( ; nItems > 0; nItems--, pfi3++, pcookiearray++ )
|
|
{
|
|
pcookiearray->m_pobject = pfi3;
|
|
|
|
if (fDeleteAllItems)
|
|
{
|
|
DWORD dwApiResult = CloseResource( pcookiearray );
|
|
if (0L != dwApiResult)
|
|
{
|
|
(void) DoErrMsgBox(GetActiveWindow(), MB_OK | MB_ICONSTOP, dwApiResult,
|
|
IDS_POPUP_SMB_DISCONNECTALLRESOURCE_ERROR,
|
|
pfi3->fi3_pathname );
|
|
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;
|
|
}
|
|
|
|
|
|
/*
|
|
NET_API_STATUS NET_API_FUNCTION
|
|
NetShareDel (
|
|
IN LPTSTR servername,
|
|
IN LPTSTR netname,
|
|
IN DWORD reserved
|
|
);
|
|
*/
|
|
|
|
typedef DWORD (*SHAREGETINFOPROC) (LPTSTR,LPTSTR,DWORD,LPBYTE*);
|
|
typedef DWORD (*SHAREDELPROC) (LPTSTR,LPTSTR,DWORD);
|
|
|
|
DWORD SmbFileServiceProvider::OpenShare( LPCTSTR lpcszServerName, LPCTSTR lpcszShareName )
|
|
{
|
|
if ( !g_SmbDLL.LoadFunctionPointers() )
|
|
return NERR_Success;
|
|
|
|
BOOL bLocal = IsLocalComputername(lpcszServerName);
|
|
|
|
PTSTR pszPath = NULL;
|
|
SHARE_INFO_2 *pshi2 = NULL;
|
|
if (bLocal)
|
|
{
|
|
NET_API_STATUS dwRet = ((SHAREGETINFOPROC)g_SmbDLL[SMB_SHARE_GET_INFO])(
|
|
const_cast<LPTSTR>(lpcszServerName),
|
|
const_cast<LPTSTR>(lpcszShareName),
|
|
2,
|
|
(LPBYTE*)&pshi2);
|
|
|
|
if (NERR_Success != dwRet)
|
|
return dwRet;
|
|
|
|
pszPath = pshi2->shi2_path;
|
|
} else
|
|
{
|
|
pszPath = (PTSTR)calloc(lstrlen(lpcszServerName) + lstrlen(lpcszShareName) + 4, sizeof(TCHAR));
|
|
if (pszPath)
|
|
{
|
|
TCHAR *p = pszPath;
|
|
|
|
if (_T('\\') != *lpcszServerName)
|
|
{
|
|
*p++ = _T('\\');
|
|
*p++ = _T('\\');
|
|
}
|
|
lstrcpy(p, lpcszServerName);
|
|
p += lstrlen(lpcszServerName);
|
|
*p++ = _T('\\');
|
|
lstrcpy(p, lpcszShareName);
|
|
}
|
|
}
|
|
|
|
if (!pszPath || !*pszPath)
|
|
{
|
|
(void) DoErrMsgBox(GetActiveWindow(), MB_OK, 0, IDS_CANNOT_OPENSHARE, lpcszShareName);
|
|
} else if (32 >= (INT_PTR) ShellExecute(
|
|
NULL, // Handle to window
|
|
_T("explore"), // Action to take
|
|
pszPath, // Folder to explore
|
|
NULL, // Parameters
|
|
NULL, // Default directory
|
|
SW_SHOWNORMAL // Show command
|
|
))
|
|
{
|
|
(void) DoErrMsgBox(GetActiveWindow(), MB_OK | MB_ICONSTOP, 0, IDS_MSG_EXPLORE_FAILURE, pszPath);
|
|
|
|
}
|
|
|
|
if (!bLocal && pszPath)
|
|
free(pszPath);
|
|
|
|
if (pshi2)
|
|
FreeData(pshi2);
|
|
|
|
return NERR_Success;
|
|
}
|
|
|
|
/*
|
|
NET_API_STATUS NET_API_FUNCTION
|
|
NetConnectionEnum (
|
|
IN LMSTR servername OPTIONAL,
|
|
IN LMSTR qualifier,
|
|
IN DWORD level,
|
|
OUT LPBYTE *bufptr,
|
|
IN DWORD prefmaxlen,
|
|
OUT LPDWORD entriesread,
|
|
OUT LPDWORD totalentries,
|
|
IN OUT LPDWORD resume_handle OPTIONAL
|
|
);
|
|
*/
|
|
|
|
typedef DWORD (*CONNECTIONENUMPROC) (LPTSTR,LPTSTR,DWORD,LPBYTE *,DWORD,LPDWORD,LPDWORD,LPDWORD);
|
|
|
|
// S_OK: user wants to continue
|
|
// S_FALSE: user wants to cancel the operation and keep the share
|
|
// E_FAIL: share doesn't exist, needs to refresh
|
|
HRESULT SmbFileServiceProvider::ConfirmDeleteShare( LPCTSTR lpcszServerName, LPCTSTR lpcszShareName )
|
|
{
|
|
if ( !g_SmbDLL.LoadFunctionPointers() )
|
|
return E_FAIL;
|
|
|
|
CONNECTION_INFO_1* pBuf = NULL;
|
|
DWORD dwEntry = 0;
|
|
DWORD dwTotal = 0;
|
|
NET_API_STATUS dwRet = ((CONNECTIONENUMPROC)g_SmbDLL[SMB_CONNECTION_ENUM])(
|
|
const_cast<LPTSTR>(lpcszServerName),
|
|
const_cast<LPTSTR>(lpcszShareName),
|
|
1,
|
|
(LPBYTE*)&pBuf,
|
|
(DWORD)-1,
|
|
&dwEntry,
|
|
&dwTotal,
|
|
NULL);
|
|
|
|
if (NERR_NetNameNotFound == dwRet)
|
|
return E_FAIL;
|
|
|
|
if (NERR_Success != dwRet && ERROR_MORE_DATA != dwRet)
|
|
return E_FAIL;
|
|
|
|
UINT cConns = dwTotal;
|
|
UINT cOpens = 0;
|
|
for (DWORD i = 0; i < dwEntry; i++)
|
|
{
|
|
cOpens += pBuf[i].coni1_num_opens;
|
|
}
|
|
|
|
NetApiBufferFree(pBuf);
|
|
|
|
if (cConns > 0)
|
|
{
|
|
if (cOpens > 0)
|
|
{
|
|
if (IDYES != DoErrMsgBox(
|
|
GetActiveWindow(),
|
|
MB_YESNO,
|
|
0,
|
|
IDS_s_CONFIRM_DELETESHARE_FILE,
|
|
cOpens,
|
|
cConns,
|
|
lpcszShareName))
|
|
{
|
|
return S_FALSE;
|
|
}
|
|
} else // (cConns > 0)
|
|
{
|
|
if (IDYES != DoErrMsgBox(
|
|
GetActiveWindow(),
|
|
MB_YESNO,
|
|
0,
|
|
IDS_s_CONFIRM_DELETESHARE_CONN,
|
|
cConns,
|
|
lpcszShareName))
|
|
{
|
|
return S_FALSE;
|
|
}
|
|
}
|
|
} else
|
|
{
|
|
if (IDYES != DoErrMsgBox(
|
|
GetActiveWindow(),
|
|
MB_YESNO,
|
|
0,
|
|
IDS_s_CONFIRM_DELETESHARE,
|
|
lpcszShareName))
|
|
{
|
|
return S_FALSE;
|
|
}
|
|
}
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
DWORD SmbFileServiceProvider::DeleteShare( LPCTSTR lpcszServerName, LPCTSTR lpcszShareName )
|
|
{
|
|
if ( !g_SmbDLL.LoadFunctionPointers() )
|
|
return S_OK;
|
|
|
|
SHARE_INFO_2 *pshi2 = NULL;
|
|
NET_API_STATUS dwRet = NERR_Success;
|
|
BOOL bLocal = TRUE;
|
|
|
|
bLocal = IsLocalComputername(lpcszServerName);
|
|
|
|
if (bLocal)
|
|
{
|
|
dwRet = ((SHAREGETINFOPROC)g_SmbDLL[SMB_SHARE_GET_INFO])(
|
|
const_cast<LPTSTR>(lpcszServerName),
|
|
const_cast<LPTSTR>(lpcszShareName),
|
|
2,
|
|
(LPBYTE*)&pshi2);
|
|
|
|
if (NERR_NetNameNotFound == dwRet)
|
|
return NERR_Success;
|
|
|
|
if (NERR_Success != dwRet)
|
|
return dwRet;
|
|
}
|
|
|
|
dwRet = ((SHAREDELPROC)g_SmbDLL[SMB_SHARE_DEL])(
|
|
const_cast<LPTSTR>(lpcszServerName),
|
|
const_cast<LPTSTR>(lpcszShareName),
|
|
0L );
|
|
|
|
if (NERR_NetNameNotFound == dwRet)
|
|
dwRet = NERR_Success;
|
|
|
|
if (NERR_Success == dwRet)
|
|
{
|
|
IADsContainer *piADsContainer = m_pFileMgmtData->GetIADsContainer(); // no need to AddRef
|
|
if (piADsContainer)
|
|
{
|
|
CString strCNName = _T("CN=");
|
|
strCNName += lpcszShareName;
|
|
(void)piADsContainer->Delete(_T("volume"), (LPTSTR)(LPCTSTR)strCNName);
|
|
}
|
|
}
|
|
|
|
if (bLocal)
|
|
{
|
|
if (dwRet == NERR_Success)
|
|
{
|
|
SHChangeNotify(
|
|
SHCNE_NETUNSHARE,
|
|
SHCNF_PATH | SHCNF_FLUSHNOWAIT ,
|
|
pshi2->shi2_path,
|
|
0);
|
|
}
|
|
|
|
FreeData( pshi2 );
|
|
}
|
|
|
|
return dwRet;
|
|
}
|
|
|
|
/*
|
|
NET_API_STATUS NET_API_FUNCTION
|
|
NetSessionDel (
|
|
IN LPTSTR servername OPTIONAL,
|
|
IN LPTSTR UncClientName,
|
|
IN LPTSTR username
|
|
);
|
|
*/
|
|
|
|
typedef DWORD (*SESSIONDELPROC) (LPTSTR,LPTSTR,LPTSTR);
|
|
|
|
BOOL BlockRemoteAdminSession(
|
|
IN PCTSTR i_pszTargetServer,
|
|
IN PCTSTR i_pszClientName,
|
|
IN PCTSTR i_pszUserName,
|
|
IN DWORD i_dwNumOpenSessions
|
|
);
|
|
|
|
BOOL BlockRemoteAdminFile(
|
|
IN PCTSTR i_pszTargetServer,
|
|
IN PCTSTR i_pszPathName,
|
|
IN PCTSTR i_pszUserName
|
|
);
|
|
|
|
DWORD SmbFileServiceProvider::CloseSession(CFileMgmtResultCookie* pcookie)
|
|
{
|
|
if ( !g_SmbDLL.LoadFunctionPointers() )
|
|
return S_OK;
|
|
|
|
ASSERT( FILEMGMT_SESSION == pcookie->QueryObjectType() );
|
|
SESSION_INFO_1* psi1 = (SESSION_INFO_1*)pcookie->m_pobject;
|
|
ASSERT( NULL != psi1 &&
|
|
NULL != psi1->sesi1_cname &&
|
|
TEXT('\0') != *(psi1->sesi1_cname) );
|
|
|
|
PCTSTR pszTargetServer = pcookie->QueryTargetServer();
|
|
if (BlockRemoteAdminSession(pszTargetServer, psi1->sesi1_cname, psi1->sesi1_username, psi1->sesi1_num_opens))
|
|
return NERR_Success;
|
|
|
|
CString strCName = _T("\\\\");
|
|
strCName += psi1->sesi1_cname;
|
|
DWORD dwRetval = ((SESSIONDELPROC)g_SmbDLL[SMB_SESSION_DEL])(
|
|
const_cast<LPTSTR>(pszTargetServer),
|
|
const_cast<LPTSTR>((LPCTSTR)strCName),
|
|
psi1->sesi1_username );
|
|
return (NERR_NoSuchSession == dwRetval) ? NERR_Success : dwRetval;
|
|
}
|
|
|
|
/*
|
|
NET_API_STATUS NET_API_FUNCTION
|
|
NetFileClose (
|
|
IN LPTSTR servername OPTIONAL,
|
|
IN DWORD fileid
|
|
);
|
|
*/
|
|
|
|
typedef DWORD (*FILECLOSEPROC) (LPTSTR,DWORD);
|
|
|
|
DWORD SmbFileServiceProvider::CloseResource(CFileMgmtResultCookie* pcookie)
|
|
{
|
|
if ( !g_SmbDLL.LoadFunctionPointers() )
|
|
return S_OK;
|
|
|
|
ASSERT( FILEMGMT_RESOURCE == pcookie->QueryObjectType() );
|
|
FILE_INFO_3* pfileinfo = (FILE_INFO_3*)pcookie->m_pobject;
|
|
ASSERT( NULL != pfileinfo );
|
|
|
|
PCTSTR pszTargetServer = pcookie->QueryTargetServer();
|
|
if (BlockRemoteAdminFile(pszTargetServer, pfileinfo->fi3_pathname, pfileinfo->fi3_username))
|
|
return NERR_Success;
|
|
|
|
DWORD dwRetval = ((FILECLOSEPROC)g_SmbDLL[SMB_FILE_CLOSE])(
|
|
const_cast<LPTSTR>(pszTargetServer),
|
|
pfileinfo->fi3_id );
|
|
return (NERR_FileIdNotFound == dwRetval) ? NERR_Success : dwRetval;
|
|
}
|
|
|
|
/*
|
|
NET_API_STATUS NET_API_FUNCTION
|
|
NetShareGetInfo (
|
|
IN LPTSTR servername,
|
|
IN LPTSTR netname,
|
|
IN DWORD level,
|
|
OUT LPBYTE *bufptr
|
|
);
|
|
|
|
NET_API_STATUS NET_API_FUNCTION
|
|
NetShareSetInfo (
|
|
IN LPTSTR servername,
|
|
IN LPTSTR netname,
|
|
IN DWORD level,
|
|
IN LPBYTE buf,
|
|
OUT LPDWORD parm_err
|
|
);
|
|
|
|
typedef struct _SHARE_INFO_2 {
|
|
LPTSTR shi2_netname;
|
|
DWORD shi2_type;
|
|
LPTSTR shi2_remark;
|
|
DWORD shi2_permissions;
|
|
DWORD shi2_max_uses;
|
|
DWORD shi2_current_uses;
|
|
LPTSTR shi2_path;
|
|
LPTSTR shi2_passwd;
|
|
} SHARE_INFO_2, *PSHARE_INFO_2, *LPSHARE_INFO_2;
|
|
*/
|
|
|
|
typedef DWORD (*SHARESETINFOPROC) (LPTSTR,LPTSTR,DWORD,LPBYTE,LPDWORD);
|
|
|
|
VOID SmbFileServiceProvider::DisplayShareProperties(
|
|
LPPROPERTYSHEETCALLBACK pCallBack,
|
|
LPDATAOBJECT pDataObject,
|
|
LONG_PTR handle)
|
|
{
|
|
HRESULT hr = S_OK;
|
|
CSharePageGeneralSMB * pPage = new CSharePageGeneralSMB();
|
|
if ( !pPage->Load( m_pFileMgmtData, pDataObject ) )
|
|
return;
|
|
|
|
//
|
|
// enforce the following calling order to work around MMC bug#464475:
|
|
// m_psp.pfnCallback ==> my PropSheetPageProc ==> MMC callback ==> MFC callback
|
|
//
|
|
MMCPropPageCallback(INOUT &pPage->m_psp);
|
|
|
|
// 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 = &CSharePageGeneralSMB::PropSheetPageProc;
|
|
pPage->m_handle = handle;
|
|
|
|
HPROPSHEETPAGE hPage=MyCreatePropertySheetPage(&pPage->m_psp);
|
|
pCallBack->AddPage(hPage);
|
|
|
|
if (pPage->m_dwShareType & (STYPE_IPC | STYPE_SPECIAL))
|
|
{
|
|
(void) DoErrMsgBox(GetActiveWindow(), MB_OK | MB_ICONINFORMATION, 0, IDS_s_POPUP_ADMIN_SHARE);
|
|
return;
|
|
}
|
|
|
|
//
|
|
// display the "Publish" page
|
|
//
|
|
if (m_pFileMgmtData->GetSchemaSupportSharePublishing() && CheckPolicyOnSharePublish(pPage->m_strShareName))
|
|
{
|
|
CSharePagePublish * pPagePublish = new CSharePagePublish();
|
|
if ( !pPagePublish->Load( m_pFileMgmtData, pDataObject ) )
|
|
return;
|
|
|
|
//
|
|
// enforce the following calling order to work around MMC bug#464475:
|
|
// m_psp.pfnCallback ==> my PropSheetPageProc ==> MMC callback ==> MFC callback
|
|
//
|
|
MMCPropPageCallback(INOUT &pPagePublish->m_psp);
|
|
|
|
// This mechanism deletes the pPagePublish when the property sheet is finished
|
|
pPagePublish->m_pfnOriginalPropSheetPageProc = pPagePublish->m_psp.pfnCallback;
|
|
pPagePublish->m_psp.lParam = reinterpret_cast<LPARAM>(pPagePublish);
|
|
pPagePublish->m_psp.pfnCallback = &CSharePagePublish::PropSheetPageProc;
|
|
pPagePublish->m_handle = handle;
|
|
|
|
HPROPSHEETPAGE hPagePublish=MyCreatePropertySheetPage(&pPagePublish->m_psp);
|
|
pCallBack->AddPage(hPagePublish);
|
|
}
|
|
|
|
//
|
|
// display the "Share Security" page
|
|
//
|
|
CComObject<CSMBSecurityInformation>* psecinfo = NULL;
|
|
hr = CComObject<CSMBSecurityInformation>::CreateInstance(&psecinfo);
|
|
if ( SUCCEEDED(hr) )
|
|
MyCreateShareSecurityPage(
|
|
pCallBack,
|
|
psecinfo,
|
|
pPage->m_strMachineName,
|
|
pPage->m_strShareName );
|
|
|
|
CreateFolderSecurityPropPage(pCallBack, pDataObject);
|
|
|
|
}
|
|
|
|
DWORD SmbFileServiceProvider::ReadShareType(
|
|
LPCTSTR ptchServerName,
|
|
LPCTSTR ptchShareName,
|
|
OUT DWORD* pdwShareType)
|
|
{
|
|
if ( !g_SmbDLL.LoadFunctionPointers() )
|
|
{
|
|
ASSERT(FALSE);
|
|
return S_OK;
|
|
}
|
|
*pdwShareType = 0;
|
|
|
|
SHARE_INFO_2* psi2 = NULL;
|
|
NET_API_STATUS retval = ((SHAREGETINFOPROC)g_SmbDLL[SMB_SHARE_GET_INFO])(
|
|
const_cast<LPTSTR>(ptchServerName),
|
|
const_cast<LPTSTR>(ptchShareName),
|
|
2,
|
|
(LPBYTE*)&psi2);
|
|
if (NERR_Success != retval)
|
|
return retval;
|
|
|
|
ASSERT( NULL != psi2 );
|
|
*pdwShareType = psi2->shi2_type;
|
|
|
|
FreeData(psi2);
|
|
|
|
return NERR_Success;
|
|
}
|
|
|
|
DWORD SmbFileServiceProvider::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_SmbDLL.LoadFunctionPointers() )
|
|
{
|
|
ASSERT(FALSE);
|
|
return S_OK;
|
|
}
|
|
|
|
if (ppvPropertyBlock) *ppvPropertyBlock = NULL;
|
|
if (pdwShareType) *pdwShareType = 0;
|
|
if (pfEditDescription) *pfEditDescription = TRUE;
|
|
if (pfEditPath) *pfEditPath = FALSE;
|
|
|
|
SHARE_INFO_2* psi2 = NULL;
|
|
NET_API_STATUS retval = ((SHAREGETINFOPROC)g_SmbDLL[SMB_SHARE_GET_INFO])(
|
|
const_cast<LPTSTR>(ptchServerName),
|
|
const_cast<LPTSTR>(ptchShareName),
|
|
2,
|
|
(LPBYTE*)&psi2);
|
|
if (NERR_Success != retval)
|
|
return retval;
|
|
|
|
strDescription = psi2->shi2_remark;
|
|
strPath = psi2->shi2_path;
|
|
|
|
if (pdwShareType)
|
|
*pdwShareType = psi2->shi2_type;
|
|
|
|
if (ppvPropertyBlock)
|
|
{
|
|
*ppvPropertyBlock = psi2; // will be freed by the caller
|
|
} else
|
|
{
|
|
FreeData((LPVOID)psi2);
|
|
}
|
|
|
|
return NERR_Success;
|
|
}
|
|
|
|
DWORD SmbFileServiceProvider::WriteShareProperties(
|
|
OUT LPCTSTR ptchServerName,
|
|
OUT LPCTSTR ptchShareName,
|
|
OUT PVOID pvPropertyBlock,
|
|
OUT LPCTSTR ptchDescription,
|
|
OUT LPCTSTR ptchPath)
|
|
{
|
|
ASSERT( NULL != pvPropertyBlock );
|
|
if ( !g_SmbDLL.LoadFunctionPointers() )
|
|
return S_OK;
|
|
|
|
SHARE_INFO_2* psi2 = (SHARE_INFO_2*)pvPropertyBlock;
|
|
//
|
|
// CODEWORK Note that this leaves psi2 invalid after the call, but that any subsequent
|
|
// use will replace these pointers.
|
|
//
|
|
psi2->shi2_remark = const_cast<LPTSTR>(ptchDescription);
|
|
psi2->shi2_path = const_cast<LPTSTR>(ptchPath);
|
|
DWORD dwDummy;
|
|
DWORD retval = ((SHARESETINFOPROC)g_SmbDLL[SMB_SHARE_SET_INFO])(
|
|
const_cast<LPTSTR>(ptchServerName),
|
|
const_cast<LPTSTR>(ptchShareName),
|
|
2,
|
|
(LPBYTE)psi2,
|
|
&dwDummy);
|
|
psi2->shi2_remark = NULL;
|
|
psi2->shi2_path = NULL;
|
|
return retval;
|
|
}
|
|
|
|
HRESULT TranslateManagedBy(
|
|
IN PCTSTR i_pszDCName,
|
|
IN PCTSTR i_pszIn,
|
|
OUT CString& o_strOut,
|
|
IN ADS_NAME_TYPE_ENUM i_formatIn,
|
|
IN ADS_NAME_TYPE_ENUM i_formatOut
|
|
)
|
|
{
|
|
o_strOut.Empty();
|
|
|
|
HRESULT hr = S_OK;
|
|
if (!i_pszIn || !*i_pszIn)
|
|
return hr;
|
|
|
|
CComPtr<IADsNameTranslate> spiADsNameTranslate;
|
|
hr = CoCreateInstance(CLSID_NameTranslate, NULL, CLSCTX_INPROC_SERVER, IID_IADsNameTranslate, (void **)&spiADsNameTranslate);
|
|
if (FAILED(hr)) return hr;
|
|
|
|
hr = spiADsNameTranslate->Init(ADS_NAME_INITTYPE_SERVER, (LPTSTR)i_pszDCName);
|
|
if (FAILED(hr)) return hr;
|
|
|
|
hr = spiADsNameTranslate->Set(i_formatIn, (LPTSTR)i_pszIn);
|
|
if (FAILED(hr)) return hr;
|
|
|
|
CComBSTR sbstr;
|
|
hr = spiADsNameTranslate->Get(i_formatOut, &sbstr);
|
|
|
|
if (SUCCEEDED(hr))
|
|
o_strOut = (BSTR)sbstr;
|
|
|
|
return hr;
|
|
}
|
|
|
|
HRESULT SmbFileServiceProvider::ReadSharePublishInfo(
|
|
LPCTSTR ptchServerName,
|
|
LPCTSTR ptchShareName,
|
|
OUT BOOL* pbPublish,
|
|
OUT CString& strUNCPath,
|
|
OUT CString& strDescription,
|
|
OUT CString& strKeywords,
|
|
OUT CString& strManagedBy)
|
|
{
|
|
HRESULT hr = S_OK;
|
|
|
|
do {
|
|
CString strADsPath, strDCName;
|
|
hr = GetADsPathOfComputerObject(ptchServerName, strADsPath, strDCName);
|
|
if (S_OK != hr) break;
|
|
|
|
CComPtr<IADsContainer> spiADsContainer;
|
|
hr = ADsGetObject(strADsPath, IID_IADsContainer, (void**)&spiADsContainer);
|
|
if (FAILED(hr)) break;
|
|
|
|
CString strCNName = _T("CN=");
|
|
strCNName += ptchShareName;
|
|
|
|
CComPtr<IDispatch> spiDispatch;
|
|
hr = spiADsContainer->GetObject(_T("volume"), (LPTSTR)(LPCTSTR)strCNName, &spiDispatch);
|
|
if (HRESULT_FROM_WIN32(ERROR_DS_NO_SUCH_OBJECT) == hr)
|
|
{
|
|
hr = S_OK;
|
|
*pbPublish = FALSE;
|
|
break;
|
|
}
|
|
if (FAILED(hr)) break;
|
|
|
|
*pbPublish = TRUE;
|
|
|
|
CComPtr<IADs> spiADs;
|
|
hr = spiDispatch->QueryInterface(IID_IADs, (void**)&spiADs);
|
|
if (FAILED(hr)) break;
|
|
|
|
VARIANT var;
|
|
VariantInit(&var);
|
|
|
|
hr = spiADs->Get(_T("uNCName"), &var);
|
|
if (FAILED(hr)) break;
|
|
strUNCPath = V_BSTR(&var);
|
|
VariantClear(&var);
|
|
|
|
hr = spiADs->Get(_T("description"), &var);
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
hr = GetSingleOrMultiValuesFromVarArray(&var, strDescription);
|
|
VariantClear(&var);
|
|
} else if (E_ADS_PROPERTY_NOT_FOUND == hr)
|
|
hr = S_OK;
|
|
else
|
|
break;
|
|
|
|
hr = spiADs->Get(_T("keywords"), &var);
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
hr = GetSingleOrMultiValuesFromVarArray(&var, strKeywords);
|
|
VariantClear(&var);
|
|
} else if (E_ADS_PROPERTY_NOT_FOUND == hr)
|
|
hr = S_OK;
|
|
else
|
|
break;
|
|
|
|
hr = spiADs->Get(_T("managedBy"), &var);
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
// 1st, try map to a UPN [email protected]
|
|
hr = TranslateManagedBy(strDCName,
|
|
V_BSTR(&var),
|
|
strManagedBy,
|
|
ADS_NAME_TYPE_1779,
|
|
ADS_NAME_TYPE_USER_PRINCIPAL_NAME);
|
|
|
|
// in case no UPN, map to NT4 style domain\user
|
|
if (FAILED(hr))
|
|
hr = TranslateManagedBy(strDCName,
|
|
V_BSTR(&var),
|
|
strManagedBy,
|
|
ADS_NAME_TYPE_1779,
|
|
ADS_NAME_TYPE_NT4);
|
|
VariantClear(&var);
|
|
} else if (E_ADS_PROPERTY_NOT_FOUND == hr)
|
|
hr = S_OK;
|
|
else
|
|
break;
|
|
|
|
} while (0);
|
|
|
|
return hr;
|
|
}
|
|
|
|
HRESULT SmbFileServiceProvider::WriteSharePublishInfo(
|
|
LPCTSTR ptchServerName,
|
|
LPCTSTR ptchShareName,
|
|
IN BOOL bPublish,
|
|
LPCTSTR ptchDescription,
|
|
LPCTSTR ptchKeywords,
|
|
LPCTSTR ptchManagedBy)
|
|
{
|
|
HRESULT hr = S_OK;
|
|
|
|
do {
|
|
CString strADsPath, strDCName;
|
|
hr = GetADsPathOfComputerObject(ptchServerName, strADsPath, strDCName);
|
|
if (S_OK != hr) break;
|
|
|
|
CComPtr<IADsContainer> spiADsContainer;
|
|
hr = ADsGetObject(strADsPath, IID_IADsContainer, (void**)&spiADsContainer);
|
|
if (FAILED(hr)) break;
|
|
|
|
CString strCNName = _T("CN=");
|
|
strCNName += ptchShareName;
|
|
|
|
if (!bPublish)
|
|
{
|
|
hr = spiADsContainer->Delete(_T("volume"), (LPTSTR)(LPCTSTR)strCNName);
|
|
if (HRESULT_FROM_WIN32(ERROR_DS_NO_SUCH_OBJECT) == hr)
|
|
hr = S_OK;
|
|
break;
|
|
}
|
|
|
|
BOOL bNewObject = FALSE;
|
|
CComPtr<IDispatch> spiDispatch;
|
|
hr = spiADsContainer->GetObject(_T("volume"), (LPTSTR)(LPCTSTR)strCNName, &spiDispatch);
|
|
if (HRESULT_FROM_WIN32(ERROR_DS_NO_SUCH_OBJECT) == hr)
|
|
{
|
|
hr = spiADsContainer->Create(_T("volume"), (LPTSTR)(LPCTSTR)strCNName, &spiDispatch);
|
|
bNewObject = TRUE;
|
|
}
|
|
if (FAILED(hr)) break;
|
|
|
|
CComPtr<IADs> spiADs;
|
|
hr = spiDispatch->QueryInterface(IID_IADs, (void**)&spiADs);
|
|
if (FAILED(hr)) break;
|
|
|
|
VARIANT var;
|
|
VariantInit(&var);
|
|
|
|
if (bNewObject)
|
|
{
|
|
CString strUNCName;
|
|
if (!_tcsncmp(ptchServerName, _T("\\\\"), 2))
|
|
{
|
|
strUNCName = ptchServerName;
|
|
} else
|
|
{
|
|
strUNCName = _T("\\\\");
|
|
strUNCName += ptchServerName;
|
|
}
|
|
strUNCName += _T("\\");
|
|
strUNCName += ptchShareName;
|
|
|
|
V_BSTR(&var) = SysAllocString(strUNCName);
|
|
V_VT(&var) = VT_BSTR;
|
|
|
|
hr = spiADs->Put(_T("uNCName"), var);
|
|
VariantClear(&var);
|
|
if (FAILED(hr)) break;
|
|
}
|
|
|
|
// according to schema, description is multi valued.
|
|
// but we're treating it as single value
|
|
if (ptchDescription && *ptchDescription)
|
|
{
|
|
V_BSTR(&var) = SysAllocString(ptchDescription);
|
|
V_VT(&var) = VT_BSTR;
|
|
hr = spiADs->Put(_T("description"), var);
|
|
VariantClear(&var);
|
|
} else if (!bNewObject)
|
|
{
|
|
V_VT(&var)=VT_NULL;
|
|
hr = spiADs->PutEx(ADS_PROPERTY_CLEAR, _T("description"), var);
|
|
VariantClear(&var);
|
|
}
|
|
|
|
if (FAILED(hr)) break;
|
|
|
|
if (ptchKeywords && *ptchKeywords)
|
|
{
|
|
hr = PutMultiValuesIntoVarArray(ptchKeywords, &var);
|
|
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
hr = spiADs->Put(_T("keywords"), var);
|
|
VariantClear(&var);
|
|
}
|
|
} else if (!bNewObject)
|
|
{
|
|
V_VT(&var)=VT_NULL;
|
|
hr = spiADs->PutEx(ADS_PROPERTY_CLEAR, _T("keywords"), var);
|
|
VariantClear(&var);
|
|
}
|
|
|
|
if (FAILED(hr)) break;
|
|
|
|
if (ptchManagedBy && *ptchManagedBy)
|
|
{
|
|
CString strManagedByFQDN;
|
|
hr = TranslateManagedBy(strDCName,
|
|
ptchManagedBy,
|
|
strManagedByFQDN,
|
|
(_tcschr(ptchManagedBy, _T('@')) ? ADS_NAME_TYPE_USER_PRINCIPAL_NAME : ADS_NAME_TYPE_NT4),
|
|
ADS_NAME_TYPE_1779);
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
V_BSTR(&var) = SysAllocString(strManagedByFQDN);
|
|
V_VT(&var) = VT_BSTR;
|
|
hr = spiADs->Put(_T("managedBy"), var);
|
|
VariantClear(&var);
|
|
}
|
|
} else if (!bNewObject)
|
|
{
|
|
V_VT(&var)=VT_NULL;
|
|
hr = spiADs->PutEx(ADS_PROPERTY_CLEAR, _T("managedBy"), var);
|
|
VariantClear(&var);
|
|
}
|
|
|
|
if (FAILED(hr)) break;
|
|
|
|
hr = spiADs->SetInfo(); // commit
|
|
|
|
} while (0);
|
|
|
|
return hr;
|
|
}
|
|
|
|
//
|
|
// These methods cover the seperate API to determine whether IntelliMirror
|
|
// caching is enabled. By default, SFM they are disabled.
|
|
//
|
|
// We read this data at level 501 in order to determine whether the target
|
|
// server is NT4. NetShareGetInfo[1005] actually succeeds on an NT4 server,
|
|
// whereas NetShareGetInfo[501] fails with ERROR_INVALID_LEVEL. We want this
|
|
// to fail so that we can disable the checkbox where the underlying
|
|
// functionality is not supported.
|
|
//
|
|
DWORD SmbFileServiceProvider::ReadShareFlags(
|
|
LPCTSTR ptchServerName,
|
|
LPCTSTR ptchShareName,
|
|
DWORD* pdwFlags )
|
|
{
|
|
ASSERT( NULL != pdwFlags );
|
|
|
|
if ( !g_SmbDLL.LoadFunctionPointers() )
|
|
{
|
|
ASSERT(FALSE);
|
|
return S_OK;
|
|
}
|
|
|
|
*pdwFlags = 0;
|
|
SHARE_INFO_501* pshi501 = NULL;
|
|
NET_API_STATUS retval = ((SHAREGETINFOPROC)g_SmbDLL[SMB_SHARE_GET_INFO])(
|
|
const_cast<LPTSTR>(ptchServerName),
|
|
const_cast<LPTSTR>(ptchShareName),
|
|
501,
|
|
(LPBYTE*)&pshi501);
|
|
if (NERR_Success != retval)
|
|
return retval;
|
|
ASSERT( NULL != pshi501 );
|
|
*pdwFlags = pshi501->shi501_flags;
|
|
FreeData( pshi501 );
|
|
|
|
return NERR_Success;
|
|
}
|
|
|
|
DWORD SmbFileServiceProvider::WriteShareFlags(
|
|
LPCTSTR ptchServerName,
|
|
LPCTSTR ptchShareName,
|
|
DWORD dwFlags )
|
|
{
|
|
if ( !g_SmbDLL.LoadFunctionPointers() )
|
|
return S_OK;
|
|
|
|
SHARE_INFO_1005 shi1005;
|
|
ZeroMemory( &shi1005, sizeof(shi1005) );
|
|
shi1005.shi1005_flags = dwFlags;
|
|
DWORD dwDummy;
|
|
DWORD retval = ((SHARESETINFOPROC)g_SmbDLL[SMB_SHARE_SET_INFO])(
|
|
const_cast<LPTSTR>(ptchServerName),
|
|
const_cast<LPTSTR>(ptchShareName),
|
|
1005,
|
|
(LPBYTE)&shi1005,
|
|
&dwDummy);
|
|
return retval;
|
|
}
|
|
|
|
BOOL SmbFileServiceProvider::GetCachedFlag( DWORD dwFlags, DWORD dwFlagToCheck )
|
|
{
|
|
return (dwFlags & CSC_MASK) == dwFlagToCheck;
|
|
}
|
|
|
|
VOID SmbFileServiceProvider::SetCachedFlag( DWORD* pdwFlags, DWORD dwNewFlag )
|
|
{
|
|
*pdwFlags &= ~CSC_MASK;
|
|
|
|
*pdwFlags |= dwNewFlag;
|
|
}
|
|
|
|
VOID SmbFileServiceProvider::FreeShareProperties(PVOID pvPropertyBlock)
|
|
{
|
|
FreeData( pvPropertyBlock );
|
|
}
|
|
|
|
DWORD SmbFileServiceProvider::QueryMaxUsers(PVOID pvPropertyBlock)
|
|
{
|
|
SHARE_INFO_2* psi2 = (SHARE_INFO_2*)pvPropertyBlock;
|
|
ASSERT( NULL != psi2 );
|
|
return psi2->shi2_max_uses;
|
|
}
|
|
|
|
VOID SmbFileServiceProvider::SetMaxUsers(PVOID pvPropertyBlock, DWORD dwMaxUsers)
|
|
{
|
|
SHARE_INFO_2* psi2 = (SHARE_INFO_2*)pvPropertyBlock;
|
|
ASSERT( NULL != psi2 );
|
|
psi2->shi2_max_uses = dwMaxUsers;
|
|
}
|
|
|
|
VOID SmbFileServiceProvider::FreeData(PVOID pv)
|
|
{
|
|
SMBFreeData( &pv );
|
|
}
|
|
|
|
LPCTSTR SmbFileServiceProvider::QueryTransportString()
|
|
{
|
|
return m_strTransportSMB;
|
|
}
|
|
|
|
CSmbCookieBlock::~CSmbCookieBlock()
|
|
{
|
|
SMBFreeData( &m_pvCookieData );
|
|
}
|
|
|
|
DEFINE_COOKIE_BLOCK(CSmbCookie)
|
|
DEFINE_FORWARDS_MACHINE_NAME( CSmbCookie, m_pCookieBlock )
|
|
|
|
void CSmbCookie::AddRefCookie() { m_pCookieBlock->AddRef(); }
|
|
void CSmbCookie::ReleaseCookie() { m_pCookieBlock->Release(); }
|
|
|
|
HRESULT CSmbCookie::GetTransport( FILEMGMT_TRANSPORT* pTransport )
|
|
{
|
|
*pTransport = FILEMGMT_SMB;
|
|
return S_OK;
|
|
}
|
|
|
|
HRESULT CSmbShareCookie::GetShareName( CString& strShareName )
|
|
{
|
|
SHARE_INFO_2* psi2 = (SHARE_INFO_2*)m_pobject;
|
|
ASSERT( NULL != psi2 );
|
|
strShareName = psi2->shi2_netname;
|
|
return S_OK;
|
|
}
|
|
|
|
HRESULT CSmbShareCookie::GetExplorerViewDescription(
|
|
OUT CString& strExplorerViewDescription )
|
|
{
|
|
strExplorerViewDescription = GetShareInfo()->shi2_remark;
|
|
return S_OK;
|
|
}
|
|
|
|
HRESULT
|
|
CSmbShareCookie::GetSharePIDList( OUT LPITEMIDLIST *ppidl )
|
|
{
|
|
ASSERT(ppidl);
|
|
ASSERT(NULL == *ppidl); // prevent memory leak
|
|
*ppidl = NULL;
|
|
|
|
SHARE_INFO_2* psi2 = (SHARE_INFO_2*)m_pobject;
|
|
ASSERT( NULL != psi2 );
|
|
|
|
PCTSTR pszTargetServer = m_pCookieBlock->QueryTargetServer();
|
|
CString csPath;
|
|
|
|
if (pszTargetServer)
|
|
{
|
|
if ( _tcslen(pszTargetServer) >= 2 &&
|
|
_T('\\') == *pszTargetServer &&
|
|
_T('\\') == *(pszTargetServer + 1) )
|
|
{
|
|
csPath = pszTargetServer;
|
|
} else
|
|
{
|
|
csPath = _T("\\\\");
|
|
csPath += pszTargetServer;
|
|
}
|
|
csPath += _T("\\");
|
|
csPath += psi2->shi2_netname;
|
|
} else
|
|
{
|
|
csPath = psi2->shi2_path;
|
|
}
|
|
|
|
if (FALSE == csPath.IsEmpty())
|
|
*ppidl = ILCreateFromPath(csPath);
|
|
|
|
return ((*ppidl) ? S_OK : E_FAIL);
|
|
}
|
|
|
|
HRESULT CSmbSessionCookie::GetSessionClientName( CString& strName )
|
|
{
|
|
SESSION_INFO_1* psi1 = (SESSION_INFO_1*)m_pobject;
|
|
ASSERT( NULL != psi1 );
|
|
TranslateIPToComputerName(psi1->sesi1_cname, strName);
|
|
return S_OK;
|
|
}
|
|
|
|
HRESULT CSmbSessionCookie::GetSessionUserName( CString& strShareName )
|
|
{
|
|
SESSION_INFO_1* psi1 = (SESSION_INFO_1*)m_pobject;
|
|
ASSERT( NULL != psi1 );
|
|
strShareName = psi1->sesi1_username;
|
|
return S_OK;
|
|
}
|
|
|
|
HRESULT CSmbResourceCookie::GetFileID( DWORD* pdwFileID )
|
|
{
|
|
FILE_INFO_3* pfileinfo = (FILE_INFO_3*)m_pobject;
|
|
ASSERT( NULL != pdwFileID && NULL != pfileinfo );
|
|
*pdwFileID = pfileinfo->fi3_id;
|
|
return S_OK;
|
|
}
|
|
|
|
BSTR CSmbShareCookie::GetColumnText( int nCol )
|
|
{
|
|
switch (nCol)
|
|
{
|
|
case COLNUM_SHARES_SHARED_FOLDER:
|
|
return GetShareInfo()->shi2_netname;
|
|
case COLNUM_SHARES_SHARED_PATH:
|
|
return GetShareInfo()->shi2_path;
|
|
case COLNUM_SHARES_TRANSPORT:
|
|
return const_cast<BSTR>((LPCTSTR)g_strTransportSMB);
|
|
case COLNUM_SHARES_COMMENT:
|
|
return GetShareInfo()->shi2_remark;
|
|
default:
|
|
ASSERT(FALSE);
|
|
break;
|
|
}
|
|
return L"";
|
|
}
|
|
|
|
BSTR CSmbShareCookie::QueryResultColumnText( int nCol, CFileMgmtComponentData& /*refcdata*/ )
|
|
{
|
|
if (COLNUM_SHARES_NUM_SESSIONS == nCol)
|
|
return MakeDwordResult( GetNumOfCurrentUses() );
|
|
|
|
return GetColumnText(nCol);
|
|
}
|
|
|
|
extern CString g_cstrClientName;
|
|
extern CString g_cstrGuest;
|
|
extern CString g_cstrYes;
|
|
extern CString g_cstrNo;
|
|
|
|
BSTR CSmbSessionCookie::GetColumnText( int nCol )
|
|
{
|
|
switch (nCol)
|
|
{
|
|
case COLNUM_SESSIONS_USERNAME:
|
|
if ( (GetSessionInfo()->sesi1_user_flags & SESS_GUEST) &&
|
|
( !(GetSessionInfo()->sesi1_username) ||
|
|
_T('\0') == *(GetSessionInfo()->sesi1_username) ) )
|
|
{
|
|
return const_cast<BSTR>(((LPCTSTR)g_cstrGuest));
|
|
} else
|
|
{
|
|
return GetSessionInfo()->sesi1_username;
|
|
}
|
|
case COLNUM_SESSIONS_COMPUTERNAME:
|
|
{
|
|
TranslateIPToComputerName(GetSessionInfo()->sesi1_cname, g_cstrClientName);
|
|
return const_cast<BSTR>(((LPCTSTR)g_cstrClientName));
|
|
}
|
|
case COLNUM_SESSIONS_TRANSPORT:
|
|
return const_cast<BSTR>((LPCTSTR)g_strTransportSMB);
|
|
case COLNUM_SESSIONS_IS_GUEST:
|
|
if (GetSessionInfo()->sesi1_user_flags & SESS_GUEST)
|
|
return const_cast<BSTR>(((LPCTSTR)g_cstrYes));
|
|
else
|
|
return const_cast<BSTR>(((LPCTSTR)g_cstrNo));
|
|
|
|
default:
|
|
ASSERT(FALSE);
|
|
break;
|
|
}
|
|
return L"";
|
|
}
|
|
|
|
BSTR CSmbSessionCookie::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 MakeElapsedTimeResult( GetIdleTime() );
|
|
default:
|
|
break;
|
|
}
|
|
|
|
return GetColumnText(nCol);
|
|
}
|
|
|
|
BSTR CSmbResourceCookie::GetColumnText( int nCol )
|
|
{
|
|
switch (nCol)
|
|
{
|
|
case COLNUM_RESOURCES_FILENAME:
|
|
return GetFileInfo()->fi3_pathname;
|
|
case COLNUM_RESOURCES_USERNAME:
|
|
return GetFileInfo()->fi3_username;
|
|
case COLNUM_RESOURCES_TRANSPORT:
|
|
return const_cast<BSTR>((LPCTSTR)g_strTransportSMB);
|
|
case COLNUM_RESOURCES_OPEN_MODE:
|
|
return MakePermissionsResult( GetFileInfo()->fi3_permissions );
|
|
default:
|
|
ASSERT(FALSE);
|
|
break;
|
|
}
|
|
|
|
return L"";
|
|
}
|
|
|
|
BSTR CSmbResourceCookie::QueryResultColumnText( int nCol, CFileMgmtComponentData& /*refcdata*/ )
|
|
{
|
|
if (COLNUM_RESOURCES_NUM_LOCKS == nCol)
|
|
return MakeDwordResult( GetNumOfLocks() );
|
|
|
|
return GetColumnText(nCol);
|
|
}
|
|
|
|
CSMBSecurityInformation::CSMBSecurityInformation()
|
|
: m_pvolumeinfo( NULL ),
|
|
m_pDefaultDescriptor( NULL )
|
|
{
|
|
}
|
|
|
|
CSMBSecurityInformation::~CSMBSecurityInformation()
|
|
{
|
|
if (NULL != m_pDefaultDescriptor)
|
|
{
|
|
LocalFree(m_pDefaultDescriptor);
|
|
m_pvolumeinfo->shi502_security_descriptor = NULL;
|
|
}
|
|
SMBFreeData( (PVOID*)&m_pvolumeinfo );
|
|
}
|
|
|
|
STDMETHODIMP CSMBSecurityInformation::GetSecurity (
|
|
SECURITY_INFORMATION RequestedInformation,
|
|
PSECURITY_DESCRIPTOR *ppSecurityDescriptor,
|
|
BOOL fDefault )
|
|
{
|
|
MFC_TRY;
|
|
|
|
// NOTE: we allow NULL == ppSecurityDescriptor, see SetSecurity
|
|
if (0 == RequestedInformation )
|
|
{
|
|
ASSERT(FALSE);
|
|
return E_INVALIDARG;
|
|
}
|
|
|
|
if (fDefault)
|
|
return E_NOTIMPL;
|
|
|
|
if ( NULL != ppSecurityDescriptor )
|
|
*ppSecurityDescriptor = NULL;
|
|
|
|
if ( !g_SmbDLL.LoadFunctionPointers() )
|
|
{
|
|
ASSERT(FALSE); // NETAPI32 isn't installed?
|
|
return S_OK;
|
|
}
|
|
|
|
SMBFreeData( (PVOID*)&m_pvolumeinfo );
|
|
NET_API_STATUS dwErr = ((SHAREGETINFOPROC)g_SmbDLL[SMB_SHARE_GET_INFO])(
|
|
QueryMachineName(),
|
|
QueryShareName(),
|
|
502,
|
|
(LPBYTE*)&m_pvolumeinfo );
|
|
if (NERR_Success != dwErr)
|
|
{
|
|
return HRESULT_FROM_WIN32(dwErr);
|
|
}
|
|
ASSERT(NULL != m_pvolumeinfo);
|
|
|
|
if ( NULL == ppSecurityDescriptor )
|
|
return S_OK;
|
|
|
|
if (NULL == m_pvolumeinfo->shi502_security_descriptor)
|
|
{
|
|
if (NULL == m_pDefaultDescriptor)
|
|
{
|
|
HRESULT hr = NewDefaultDescriptor(
|
|
&m_pDefaultDescriptor,
|
|
RequestedInformation );
|
|
if ( !SUCCEEDED(hr) )
|
|
return hr;
|
|
}
|
|
m_pvolumeinfo->shi502_security_descriptor = m_pDefaultDescriptor;
|
|
}
|
|
ASSERT( NULL != m_pvolumeinfo->shi502_security_descriptor );
|
|
|
|
// We have to pass back a LocalAlloc'ed copy of the SD
|
|
return MakeSelfRelativeCopy(
|
|
m_pvolumeinfo->shi502_security_descriptor,
|
|
ppSecurityDescriptor );
|
|
|
|
MFC_CATCH;
|
|
}
|
|
|
|
STDMETHODIMP CSMBSecurityInformation::SetSecurity (
|
|
SECURITY_INFORMATION SecurityInformation,
|
|
PSECURITY_DESCRIPTOR pSecurityDescriptor )
|
|
{
|
|
MFC_TRY;
|
|
|
|
if ( !g_SmbDLL.LoadFunctionPointers() )
|
|
{
|
|
ASSERT(FALSE); // NETAPI32 isn't installed?
|
|
return S_OK;
|
|
}
|
|
|
|
// First get the current settings
|
|
// We call GetSecurity with NULL == ppSecurityDescriptor, this indicates to
|
|
// GetSecurity that it should refresh the shi502 structure but not return
|
|
// a copy of an actual security descriptor.
|
|
HRESULT hr = GetSecurity( SecurityInformation, NULL, FALSE );
|
|
if ( FAILED(hr) )
|
|
return hr;
|
|
|
|
// Now set the new values
|
|
m_pvolumeinfo->shi502_security_descriptor = pSecurityDescriptor;
|
|
NET_API_STATUS dwErr = ((SHARESETINFOPROC)g_SmbDLL[SMB_SHARE_SET_INFO])(
|
|
QueryMachineName(),
|
|
QueryShareName(),
|
|
502,
|
|
(LPBYTE)m_pvolumeinfo,
|
|
NULL );
|
|
if (NERR_Success != dwErr)
|
|
{
|
|
return HRESULT_FROM_WIN32(dwErr);
|
|
}
|
|
|
|
return S_OK;
|
|
|
|
MFC_CATCH;
|
|
}
|
|
|
|
|
|
//
|
|
// helper functions
|
|
//
|
|
HRESULT GetDCInfo(
|
|
IN LPCTSTR ptchServerName,
|
|
OUT CString& strDCName
|
|
)
|
|
/*
|
|
Function: retrieve DC name of the domain the server belongs to.
|
|
|
|
Return:
|
|
S_OK: if succeeded
|
|
others: if server does not belong to a domain, or error occurred
|
|
*/
|
|
{
|
|
strDCName.Empty();
|
|
|
|
//
|
|
// get domain name of the server
|
|
//
|
|
PDSROLE_PRIMARY_DOMAIN_INFO_BASIC pBuffer = NULL;
|
|
DWORD dwErr = DsRoleGetPrimaryDomainInformation(
|
|
ptchServerName,
|
|
DsRolePrimaryDomainInfoBasic,
|
|
(PBYTE *)&pBuffer);
|
|
if (ERROR_SUCCESS != dwErr)
|
|
return HRESULT_FROM_WIN32(dwErr);
|
|
|
|
if (pBuffer->MachineRole == DsRole_RoleStandaloneWorkstation ||
|
|
pBuffer->MachineRole == DsRole_RoleStandaloneServer)
|
|
{
|
|
DsRoleFreeMemory(pBuffer);
|
|
return S_FALSE; // server does not belong to a domain
|
|
}
|
|
|
|
CString strDomainName = (pBuffer->DomainNameDns ? pBuffer->DomainNameDns : pBuffer->DomainNameFlat);
|
|
|
|
DsRoleFreeMemory(pBuffer);
|
|
|
|
if (!strDomainName)
|
|
return E_OUTOFMEMORY;
|
|
|
|
if (strDomainName.IsEmpty())
|
|
return S_FALSE; // something is wrong, treat it as server does not belong to a domain
|
|
|
|
//
|
|
// In case the DNS name is in absolute form, remove the ending dot
|
|
//
|
|
int nlen = strDomainName.GetLength();
|
|
if ( _T('.') == strDomainName[nlen - 1] )
|
|
strDomainName.SetAt(nlen - 1, _T('\0'));
|
|
|
|
//
|
|
// get DC name of that domain
|
|
//
|
|
PDOMAIN_CONTROLLER_INFO pDCInfo = NULL;
|
|
dwErr = DsGetDcName(
|
|
NULL, // Run on Current Server.
|
|
strDomainName,
|
|
NULL,
|
|
NULL,
|
|
DS_DIRECTORY_SERVICE_REQUIRED | DS_RETURN_DNS_NAME,
|
|
&pDCInfo
|
|
);
|
|
if (ERROR_SUCCESS != dwErr)
|
|
return HRESULT_FROM_WIN32(dwErr);
|
|
|
|
if ( _T('\\') == *(pDCInfo->DomainControllerName) )
|
|
strDCName = pDCInfo->DomainControllerName + 2;
|
|
else
|
|
strDCName = pDCInfo->DomainControllerName;
|
|
|
|
NetApiBufferFree(pDCInfo);
|
|
|
|
if (!strDCName)
|
|
return E_OUTOFMEMORY;
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
HRESULT GetADsPathOfComputerObject(
|
|
IN LPCTSTR ptchServerName,
|
|
OUT CString& strADsPath,
|
|
OUT CString& strDCName
|
|
)
|
|
/*
|
|
Function: retrieve LDAP://<DC>/<FQDN of computer object>.
|
|
|
|
Return:
|
|
S_OK: if succeeded
|
|
others: if server does not belong to a domain, or error occurred
|
|
*/
|
|
{
|
|
//
|
|
// Get NT4 account name for this server
|
|
//
|
|
PWKSTA_INFO_100 wki100 = NULL;
|
|
NET_API_STATUS NetStatus = NetWkstaGetInfo((LPTSTR)ptchServerName, 100, (LPBYTE *)&wki100 );
|
|
if (ERROR_SUCCESS != NetStatus)
|
|
return HRESULT_FROM_WIN32(NetStatus);
|
|
|
|
HRESULT hr = S_FALSE;
|
|
CString strNT4Name;
|
|
if (wki100->wki100_langroup)
|
|
{
|
|
strNT4Name = wki100->wki100_langroup;
|
|
strNT4Name += _T("\\");
|
|
strNT4Name += wki100->wki100_computername;
|
|
strNT4Name += _T("$");
|
|
hr = S_OK;
|
|
}
|
|
|
|
NetApiBufferFree((LPBYTE)wki100);
|
|
|
|
if (S_OK != hr)
|
|
return hr;
|
|
|
|
//
|
|
// get DC name of the server's domain
|
|
//
|
|
hr = GetDCInfo(ptchServerName, strDCName);
|
|
if (S_OK != hr)
|
|
return hr;
|
|
|
|
//
|
|
// get computerDN
|
|
//
|
|
CComPtr<IADsNameTranslate> spiADsNameTranslate;
|
|
hr = CoCreateInstance(CLSID_NameTranslate, NULL, CLSCTX_INPROC_SERVER, IID_IADsNameTranslate, (void **)&spiADsNameTranslate);
|
|
if (FAILED(hr)) return hr;
|
|
|
|
hr = spiADsNameTranslate->Init(ADS_NAME_INITTYPE_SERVER, (LPTSTR)(LPCTSTR)strDCName);
|
|
if (FAILED(hr)) return hr;
|
|
|
|
hr = spiADsNameTranslate->Set(ADS_NAME_TYPE_NT4, (LPTSTR)(LPCTSTR)strNT4Name);
|
|
if (FAILED(hr)) return hr;
|
|
|
|
CComBSTR sbstrComputerDN;
|
|
hr = spiADsNameTranslate->Get(ADS_NAME_TYPE_1779, &sbstrComputerDN);
|
|
if (FAILED(hr)) return hr;
|
|
|
|
strADsPath = _T("LDAP://");
|
|
strADsPath += strDCName;
|
|
strADsPath += _T("/");
|
|
strADsPath += sbstrComputerDN;
|
|
|
|
return hr;
|
|
}
|
|
|
|
HRESULT CheckSchemaVersion(IN LPCTSTR ptchServerName)
|
|
/*
|
|
Function: check if the schema allows volume object be created as child of computer object.
|
|
|
|
Return:
|
|
S_OK: yes, it's the new schema, it allows
|
|
S_FALSE: it doesn't allow, or the server doesn't belong to a domain at all
|
|
others: error occurred
|
|
*/
|
|
{
|
|
HRESULT hr = S_OK;
|
|
|
|
//
|
|
// get DC name of the server's domain
|
|
//
|
|
CString strDCName;
|
|
hr = GetDCInfo(ptchServerName, strDCName);
|
|
if (S_OK != hr)
|
|
return hr;
|
|
|
|
//
|
|
// Get schema naming context from the rootDSE.
|
|
//
|
|
CComPtr<IADs> spiRootADs;
|
|
hr = ADsGetObject(_T("LDAP://rootDSE"),
|
|
IID_IADs,
|
|
(void**)&spiRootADs);
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
VARIANT var;
|
|
VariantInit(&var);
|
|
hr = spiRootADs->Get(_T("schemaNamingContext"), &var);
|
|
if (FAILED(hr)) return hr;
|
|
|
|
//
|
|
// get LDAP path to the schema of Connection-Point class
|
|
//
|
|
CString strADsPath = _T("LDAP://");
|
|
strADsPath += strDCName;
|
|
strADsPath += _T("/CN=Connection-Point,");
|
|
strADsPath += V_BSTR(&var);
|
|
|
|
VariantClear(&var);
|
|
|
|
CComPtr<IADs> spiADs;
|
|
hr = ADsGetObject(strADsPath, IID_IADs, (void**)&spiADs);
|
|
if (FAILED(hr)) return hr;
|
|
|
|
VariantInit(&var);
|
|
hr = spiADs->Get(_T("systemPossSuperiors"), &var);
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
hr = IsValueInVarArray(&var, _T("computer"));
|
|
VariantClear(&var);
|
|
} else if (E_ADS_PROPERTY_NOT_FOUND == hr)
|
|
hr = S_FALSE;
|
|
}
|
|
|
|
return hr;
|
|
}
|
|
|
|
BOOL CheckPolicyOnSharePublish(IN LPCTSTR ptchShareName)
|
|
{
|
|
if (!ptchShareName && !*ptchShareName)
|
|
return FALSE; // invalid share name
|
|
|
|
//
|
|
// no publish page on hidden shares
|
|
//
|
|
int len = lstrlen(ptchShareName);
|
|
if (_T('$') == *(ptchShareName + len - 1))
|
|
return FALSE;
|
|
|
|
//
|
|
// no publish page on system shares
|
|
//
|
|
int n = sizeof(g_pszSystemShares) / sizeof(LPCTSTR);
|
|
for (int i = 0; i < n; i++)
|
|
{
|
|
if (!lstrcmpi(ptchShareName, g_pszSystemShares[i]))
|
|
return FALSE;
|
|
}
|
|
|
|
//
|
|
// check group policy
|
|
//
|
|
BOOL bAddPublishPage = TRUE; // by default, we display the share publish page
|
|
|
|
HKEY hKey = NULL;
|
|
DWORD dwType = 0;
|
|
DWORD dwData = 0;
|
|
DWORD cbData = sizeof(dwData);
|
|
LONG lErr = RegOpenKeyEx(
|
|
HKEY_CURRENT_USER,
|
|
_T("Software\\Policies\\Microsoft\\Windows NT\\SharedFolders"),
|
|
0,
|
|
KEY_QUERY_VALUE,
|
|
&hKey);
|
|
if (ERROR_SUCCESS == lErr)
|
|
{
|
|
lErr = RegQueryValueEx(hKey, _T("PublishSharedFolders"), 0, &dwType, (LPBYTE)&dwData, &cbData);
|
|
|
|
if (ERROR_SUCCESS == lErr &&
|
|
REG_DWORD == dwType &&
|
|
0 == dwData) // policy is disabled
|
|
bAddPublishPage = FALSE;
|
|
|
|
RegCloseKey(hKey);
|
|
}
|
|
|
|
return bAddPublishPage;
|
|
}
|
|
|
|
void mystrtok(
|
|
IN LPCTSTR pszString,
|
|
IN OUT int* pnIndex, // start from 0
|
|
IN LPCTSTR pszCharSet,
|
|
OUT CString& strToken
|
|
)
|
|
{
|
|
strToken.Empty();
|
|
|
|
if (!pszString || !*pszString ||
|
|
!pszCharSet || !pnIndex ||
|
|
*pnIndex >= lstrlen(pszString))
|
|
{
|
|
return;
|
|
}
|
|
|
|
TCHAR *ptchStart = (PTSTR)pszString + *pnIndex;
|
|
if (!*pszCharSet)
|
|
{
|
|
strToken = ptchStart;
|
|
return;
|
|
}
|
|
|
|
//
|
|
// move p to the 1st char of the token
|
|
//
|
|
TCHAR *p = ptchStart;
|
|
while (*p)
|
|
{
|
|
if (_tcschr(pszCharSet, *p))
|
|
p++;
|
|
else
|
|
break;
|
|
}
|
|
|
|
ptchStart = p; // adjust ptchStart to point at the 1st char of the token
|
|
|
|
//
|
|
// move p to the char after the last char of the token
|
|
//
|
|
while (*p)
|
|
{
|
|
if (_tcschr(pszCharSet, *p))
|
|
break;
|
|
else
|
|
p++;
|
|
}
|
|
|
|
//
|
|
// ptchStart: points at the 1st char of the token
|
|
// p: points at the char after the last char of the token
|
|
//
|
|
if (ptchStart != p)
|
|
{
|
|
strToken = CString(ptchStart, (int)(p - ptchStart));
|
|
*pnIndex = (int)(p - pszString);
|
|
}
|
|
|
|
// return
|
|
}
|
|
|
|
//
|
|
// write a semi-colon separated string into a VARIANT
|
|
//
|
|
HRESULT PutMultiValuesIntoVarArray(
|
|
IN LPCTSTR pszValues,
|
|
OUT VARIANT* pVar
|
|
)
|
|
{
|
|
if (!pVar || !pszValues || !*pszValues)
|
|
return E_INVALIDARG;
|
|
|
|
//
|
|
// get count of items
|
|
//
|
|
int nCount = 0;
|
|
int nIndex = 0;
|
|
CString strToken;
|
|
mystrtok(pszValues, &nIndex, _T(";"), strToken);
|
|
while (!strToken.IsEmpty())
|
|
{
|
|
nCount++;
|
|
mystrtok(pszValues, &nIndex, _T(";"), strToken);
|
|
}
|
|
|
|
if (!nCount)
|
|
return E_INVALIDARG;
|
|
|
|
//
|
|
// create an array of variants to hold all the data
|
|
//
|
|
SAFEARRAYBOUND bounds = {nCount, 0};
|
|
SAFEARRAY* psa = SafeArrayCreate(VT_VARIANT, 1, &bounds);
|
|
VARIANT* varArray;
|
|
|
|
SafeArrayAccessData(psa, (void**)&varArray);
|
|
|
|
nCount = 0;
|
|
nIndex = 0;
|
|
mystrtok(pszValues, &nIndex, _T(";"), strToken);
|
|
while (!strToken.IsEmpty())
|
|
{
|
|
strToken.TrimLeft();
|
|
strToken.TrimRight();
|
|
|
|
VariantInit(&(varArray[nCount]));
|
|
varArray[nCount].vt = VT_BSTR;
|
|
varArray[nCount].bstrVal = SysAllocString(strToken);
|
|
|
|
nCount++;
|
|
|
|
mystrtok(pszValues, &nIndex, _T(";"), strToken);
|
|
}
|
|
|
|
SafeArrayUnaccessData(psa);
|
|
|
|
//
|
|
// return the variant array
|
|
//
|
|
VariantInit(pVar);
|
|
pVar->vt = VT_ARRAY | VT_VARIANT;
|
|
pVar->parray = psa;
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
//
|
|
// read a multi-valued VARIANT into a semicolon separated string
|
|
//
|
|
HRESULT GetSingleOrMultiValuesFromVarArray(
|
|
IN VARIANT* pVar,
|
|
OUT CString& strValues
|
|
)
|
|
{
|
|
strValues.Empty();
|
|
|
|
if (V_VT(pVar) != VT_BSTR &&
|
|
V_VT(pVar) != (VT_ARRAY | VT_VARIANT))
|
|
return E_INVALIDARG;
|
|
|
|
if (V_VT(pVar) == VT_BSTR)
|
|
{
|
|
strValues = V_BSTR(pVar);
|
|
} else
|
|
{
|
|
LONG lstart, lend;
|
|
SAFEARRAY *sa = V_ARRAY(pVar);
|
|
VARIANT varItem;
|
|
|
|
SafeArrayGetLBound(sa, 1, &lstart);
|
|
SafeArrayGetUBound(sa, 1, &lend);
|
|
|
|
VariantInit(&varItem);
|
|
for (LONG i = lstart; i <= lend; i++)
|
|
{
|
|
SafeArrayGetElement(sa, &i, &varItem);
|
|
|
|
if (!strValues.IsEmpty())
|
|
strValues += _T(";");
|
|
strValues += V_BSTR(&varItem);
|
|
|
|
VariantClear(&varItem);
|
|
}
|
|
}
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
HRESULT IsValueInVarArray(
|
|
IN VARIANT* pVar,
|
|
IN LPCTSTR ptchValue
|
|
)
|
|
{
|
|
if (V_VT(pVar) != VT_BSTR &&
|
|
V_VT(pVar) != (VT_ARRAY | VT_VARIANT))
|
|
return E_INVALIDARG;
|
|
|
|
HRESULT hr = S_FALSE;
|
|
|
|
if (V_VT(pVar) == VT_BSTR)
|
|
{
|
|
if (!lstrcmpi(ptchValue, V_BSTR(pVar)))
|
|
hr = S_OK;
|
|
} else
|
|
{
|
|
LONG lstart, lend;
|
|
SAFEARRAY *sa = V_ARRAY(pVar);
|
|
VARIANT varItem;
|
|
|
|
SafeArrayGetLBound(sa, 1, &lstart);
|
|
SafeArrayGetUBound(sa, 1, &lend);
|
|
|
|
VariantInit(&varItem);
|
|
for (LONG i = lstart; (i <= lend) && (S_FALSE == hr); i++)
|
|
{
|
|
SafeArrayGetElement(sa, &i, &varItem);
|
|
|
|
if (!lstrcmpi(ptchValue, V_BSTR(&varItem)))
|
|
hr = S_OK;
|
|
|
|
VariantClear(&varItem);
|
|
}
|
|
}
|
|
|
|
return hr;
|
|
}
|
|
|
|
/*
|
|
We block the operation if:
|
|
A) You are monitoring a remote computer, not the local one
|
|
B) The session username matches the currently logged on user
|
|
C) The number of files opened on the server is greater than 0
|
|
D) The client name is either one of our IP addresses or our ComputerName.
|
|
*/
|
|
BOOL BlockRemoteAdminSession(
|
|
IN PCTSTR i_pszTargetServer,
|
|
IN PCTSTR i_pszClientName,
|
|
IN PCTSTR i_pszUserName,
|
|
IN DWORD i_dwNumOpenSessions
|
|
)
|
|
{
|
|
if (!i_pszClientName || !i_pszUserName)
|
|
return FALSE;
|
|
|
|
// if we're monitoring the local machine
|
|
if (!i_pszTargetServer || !*i_pszTargetServer)
|
|
return FALSE;
|
|
|
|
// if number of sessions is not greater than 0
|
|
if (0 == i_dwNumOpenSessions)
|
|
return FALSE;
|
|
|
|
// get username, return if user name doesn't match
|
|
TCHAR szUser[UNLEN+1] = _T("");
|
|
DWORD dwSize = UNLEN+1;
|
|
GetUserName(szUser, &dwSize);
|
|
if (lstrcmpi(szUser, i_pszUserName))
|
|
return FALSE;
|
|
|
|
// compare with local computer's name
|
|
TCHAR tszComputer[MAX_COMPUTERNAME_LENGTH + 1] = _T("");
|
|
dwSize = MAX_COMPUTERNAME_LENGTH + 1;
|
|
GetComputerName(tszComputer, &dwSize);
|
|
BOOL bMatch = (0 == lstrcmpi(tszComputer, i_pszClientName));
|
|
|
|
// compare with each IP address of the local computer
|
|
if (!bMatch)
|
|
{
|
|
// convert WCHAR to char
|
|
int iBytes = 0;
|
|
char szComputer[MAX_COMPUTERNAME_LENGTH + 1] = "";
|
|
char szClientName[MAX_COMPUTERNAME_LENGTH + 1] = "";
|
|
|
|
#if defined UNICODE
|
|
iBytes = MAX_COMPUTERNAME_LENGTH + 1;
|
|
WideCharToMultiByte(CP_ACP, 0, tszComputer, -1, szComputer, iBytes, NULL, NULL);
|
|
|
|
iBytes = MAX_COMPUTERNAME_LENGTH + 1;
|
|
WideCharToMultiByte(CP_ACP, 0, i_pszClientName, -1, szClientName, iBytes, NULL, NULL);
|
|
#else
|
|
lstrcpy(szComputer, tszComputer);
|
|
lstrcpy(szClientName, i_pszClientName);
|
|
#endif
|
|
|
|
WORD wVersionRequested = MAKEWORD( 2, 2 );
|
|
WSADATA wsaData = {0};
|
|
|
|
int err = WSAStartup( wVersionRequested, &wsaData );
|
|
if (err == 0)
|
|
{
|
|
struct hostent *pHostEnt = gethostbyname(szComputer);
|
|
if (pHostEnt)
|
|
{
|
|
struct in_addr IPAddr = {0};
|
|
PSTR pszIP = NULL;
|
|
int i = 0;
|
|
while (pHostEnt->h_addr_list[i])
|
|
{
|
|
memcpy(&IPAddr, pHostEnt->h_addr_list[i++], pHostEnt->h_length);
|
|
pszIP = inet_ntoa(IPAddr);
|
|
if (!_stricmp(pszIP, szClientName))
|
|
{
|
|
bMatch = TRUE;
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
WSACleanup( );
|
|
}
|
|
}
|
|
|
|
// keep this session for remote admin purpose
|
|
if (bMatch)
|
|
DoErrMsgBox(GetActiveWindow(), MB_OK, 0, IDS_POPUP_REMOTEADMINSESSION);
|
|
|
|
return bMatch;
|
|
}
|
|
|
|
/*
|
|
We block the operation if:
|
|
A) You are monitoring a remote computer, not the local one
|
|
B) The openfile pathname matches \PIPE\srvsvc or \PIPE\MacFile
|
|
C) The openfile username matches the currently logged on user
|
|
*/
|
|
BOOL BlockRemoteAdminFile(
|
|
IN PCTSTR i_pszTargetServer,
|
|
IN PCTSTR i_pszPathName,
|
|
IN PCTSTR i_pszUserName
|
|
)
|
|
{
|
|
if (!i_pszPathName || !i_pszUserName)
|
|
return FALSE;
|
|
|
|
// if we're monitoring the local machine
|
|
if (!i_pszTargetServer || !*i_pszTargetServer)
|
|
return FALSE;
|
|
|
|
if (lstrcmpi(_T("\\PIPE\\srvsvc"), i_pszPathName) && lstrcmpi(_T("\\PIPE\\MacFile"), i_pszPathName))
|
|
return FALSE;
|
|
|
|
// get username, return if user name doesn't match
|
|
TCHAR szUser[UNLEN+1] = _T("");
|
|
DWORD dwSize = UNLEN+1;
|
|
GetUserName(szUser, &dwSize);
|
|
if (lstrcmpi(szUser, i_pszUserName))
|
|
return FALSE;
|
|
|
|
// this is the admin named pipe, keep it for remote admin purpose
|
|
DoErrMsgBox(GetActiveWindow(), MB_OK, 0, IDS_POPUP_REMOTEADMINFILE, i_pszPathName);
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
void TranslateIPToComputerName(LPCTSTR ptszIP, CString& strName)
|
|
{
|
|
int iBytes = 0;
|
|
char szIP[MAX_PATH] = "";
|
|
|
|
#if defined UNICODE
|
|
iBytes = MAX_PATH;
|
|
WideCharToMultiByte(CP_ACP, 0, ptszIP, -1, szIP, iBytes, NULL, NULL);
|
|
#else
|
|
lstrcpy(szIP, ptszIP);
|
|
#endif
|
|
|
|
strName = ptszIP;
|
|
|
|
ULONG inaddr = inet_addr(szIP);
|
|
if (INADDR_NONE != inaddr)
|
|
{
|
|
struct hostent *pHostEnt = gethostbyaddr((char *)&inaddr, sizeof(inaddr), AF_INET);
|
|
if (pHostEnt)
|
|
{
|
|
#if defined UNICODE
|
|
DWORD cb = MultiByteToWideChar(CP_ACP, 0, pHostEnt->h_name, -1, NULL, 0);
|
|
strName = CString(_T('0'), cb);
|
|
LPTSTR ptszName = strName.GetBuffer(cb);
|
|
MultiByteToWideChar(CP_ACP, 0, pHostEnt->h_name, -1, ptszName, cb);
|
|
strName.ReleaseBuffer();
|
|
|
|
#else
|
|
strName = pHostEnt->h_name;
|
|
#endif
|
|
}
|
|
}
|
|
}
|