windows-nt/Source/XPSP1/NT/inetsrv/iis/svcs/smtp/webhlpr/webhlpr.cpp

746 lines
17 KiB
C++
Raw Normal View History

2020-09-26 03:20:57 -05:00
#include <windows.h>
#include "admcmn.h"
// For Net* calls:
#include <lmaccess.h>
#include <lmwksta.h>
#include <lmapibuf.h>
#include <lmerr.h>
// For CAddr:: calls:
#include "cpool.h"
#include "abook.h"
#include "abtype.h"
#include "address.hxx"
#include "webhlpr.h"
static HRESULT GetDCName (
LPCWSTR wszServer,
LPCWSTR wszDomain,
LPWSTR * pwszDC
);
#if 0
// Taken from smtp/shash.hxx
#define TABLE_SIZE 241
static unsigned long ElfHash (const unsigned char * UserName);
static HRESULT RecursiveDeleteDirectory (
LPCWSTR wszPath
);
#endif
//
// Exported functions:
//
HRESULT
EnumerateTrustedDomains (
LPCWSTR wszComputer,
LPWSTR * pmszDomains
)
{
_ASSERT ( wszComputer && *wszComputer );
_ASSERT ( pmszDomains );
HRESULT hr = NOERROR;
DWORD dwErr = NOERROR;
LPWSTR mszDomains = NULL;
DWORD cchDomains = 0;
DWORD cchLocalDomain = 0;
LPWSTR wszPrimaryDomain = NULL;
DWORD cchPrimaryDomain = 0;
LPWSTR wszCurrent;
LPWSTR mszResult = NULL;
*pmszDomains = NULL;
//
// Get a list of the trusted 1st tier domains:
//
dwErr = NetEnumerateTrustedDomains (
const_cast <LPWSTR> (wszComputer),
&mszDomains
);
if ( dwErr != NOERROR ) {
//
// This didn't work, but we should always add the primary
// domain & computer name. Make an empty multisz to signify
// no trusted domains.
//
dwErr = NetApiBufferAllocate (
2 * sizeof ( WCHAR ),
(LPVOID *) &mszDomains
);
if ( dwErr != NOERROR ) {
BAIL_WITH_FAILURE ( hr, HRESULT_FROM_WIN32 ( dwErr ) );
}
mszDomains[0] = NULL;
mszDomains[1] = NULL;
}
_ASSERT ( mszDomains );
if ( mszDomains == NULL ) {
BAIL_WITH_FAILURE ( hr, E_UNEXPECTED );
}
//
// Check for an empty domain list:
//
if ( mszDomains[0] == NULL && mszDomains[1] == NULL ) {
cchDomains = 2;
}
else {
wszCurrent = mszDomains;
if ( mszDomains[0] == _T('\0') &&
mszDomains[1] == _T('\0') ) {
}
while ( wszCurrent && *wszCurrent ) {
DWORD cchCurrent;
cchCurrent = lstrlen ( wszCurrent ) + 1;
cchDomains += cchCurrent;
wszCurrent += cchCurrent;
}
cchDomains += 1; // Terminating NULL
}
//
// Add the local machine and primary domain:
//
hr = GetPrimaryDomain ( wszComputer, &wszPrimaryDomain );
BAIL_ON_FAILURE ( hr );
//
// PREFIX flagged this as something we need to fix. Easier to
// add check than to argue.
//
if ( !wszPrimaryDomain ) {
hr = E_FAIL;
goto Exit;
}
_ASSERT ( wszPrimaryDomain );
cchPrimaryDomain = lstrlen ( wszPrimaryDomain ) + 1;
if ( wszPrimaryDomain[0] != _T('\\') ||
wszPrimaryDomain[1] != _T('\\') ||
lstrcmpi ( wszPrimaryDomain + 2, wszComputer )
) {
//
// The primary domain isn't the local machine, so add the local
// machine to the list.
//
cchLocalDomain = lstrlen ( _T("\\\\") ) + lstrlen ( wszComputer ) + 1;
}
else {
//
// The primary domain is the local machine - no need to add it twice.
//
cchLocalDomain = 0;
}
mszResult = new WCHAR [ cchDomains + cchLocalDomain + cchPrimaryDomain];
if ( !mszResult ) {
BAIL_WITH_FAILURE ( hr, E_OUTOFMEMORY );
}
if ( cchLocalDomain ) {
wsprintf ( mszResult, _T("\\\\%s"), wszComputer );
StringToUpper ( mszResult );
}
wsprintf ( mszResult + cchLocalDomain, _T("%s"), wszPrimaryDomain );
StringToUpper ( mszResult + cchLocalDomain );
//
// Copy the rest of the domains in the domain list:
//
CopyMemory (
mszResult + cchLocalDomain + cchPrimaryDomain,
mszDomains,
cchDomains * sizeof (WCHAR)
);
*pmszDomains = mszResult;
Exit:
if ( wszPrimaryDomain ) {
delete [] wszPrimaryDomain;
}
if ( mszDomains ) {
NetApiBufferFree ( mszDomains );
}
return hr;
}
HRESULT
GetPrimaryDomain (
LPCWSTR wszComputer,
LPWSTR * pwszPrimaryDomain
)
{
HRESULT hr = NOERROR;
DWORD dwErr = NOERROR;
WKSTA_INFO_100 * pwkstaInfo = NULL;
LPWSTR wszLangroup = NULL;
*pwszPrimaryDomain = NULL;
//
// Get the workstation info to get the primary domain:
//
dwErr = NetWkstaGetInfo (
const_cast <LPWSTR> (wszComputer),
100,
(LPBYTE *)&pwkstaInfo
);
if ( dwErr != NOERROR ) {
BAIL_WITH_FAILURE ( hr, HRESULT_FROM_WIN32 ( dwErr ) );
}
_ASSERT ( pwkstaInfo );
wszLangroup = pwkstaInfo->wki100_langroup;
if ( wszLangroup && *wszLangroup ) {
*pwszPrimaryDomain = new WCHAR [ lstrlen ( wszLangroup ) + 1 ];
if ( !*pwszPrimaryDomain ) {
BAIL_WITH_FAILURE(hr, E_OUTOFMEMORY);
}
lstrcpy ( *pwszPrimaryDomain, wszLangroup );
}
else {
//
// If there is no primary domain, use the computer name prefixed
// by backslashes.
//
*pwszPrimaryDomain = new WCHAR [ 2 + lstrlen ( wszComputer ) + 1 ];
if ( !*pwszPrimaryDomain ) {
BAIL_WITH_FAILURE(hr, E_OUTOFMEMORY);
}
wsprintf ( *pwszPrimaryDomain, _T("\\\\%s"), wszComputer );
}
StringToUpper ( *pwszPrimaryDomain );
Exit:
if ( pwkstaInfo ) {
NetApiBufferFree ( pwkstaInfo );
}
return hr;
}
HRESULT CheckNTAccount (
LPCWSTR wszComputer,
LPCWSTR wszUsername,
BOOL * pfExists
)
{
HRESULT hr = NOERROR;
DWORD dwErr = NOERROR;
BOOL fFound;
BYTE sidUser [ MAX_PATH ];
DWORD cbSid = ARRAY_SIZE ( sidUser );
WCHAR wszDomain [ 512 ];
DWORD cchDomain = ARRAY_SIZE ( wszDomain );
SID_NAME_USE nuUser;
*pfExists = FALSE;
fFound = LookupAccountName (
wszComputer,
wszUsername,
&sidUser,
&cbSid,
wszDomain,
&cchDomain,
&nuUser
);
if ( !fFound ) {
dwErr = GetLastError ( );
//
// The not found error is ERROR_NONE_MAPPED:
//
if ( dwErr != ERROR_NONE_MAPPED ) {
BAIL_WITH_FAILURE ( hr, HRESULT_FROM_WIN32 ( dwErr ) );
}
}
*pfExists = fFound;
Exit:
return hr;
}
HRESULT CreateNTAccount (
LPCWSTR wszComputer,
LPCWSTR wszDomain,
LPCWSTR wszUsername,
LPCWSTR wszPassword // = ""
)
{
HRESULT hr = NOERROR;
DWORD dwErr = NOERROR;
DWORD dwParmErr = 0;
LPWSTR wszDC = NULL;
USER_INFO_3 UserInfo;
WCHAR wszNull[] = { 0 };
hr = GetDCName ( wszComputer, wszDomain, &wszDC );
BAIL_ON_FAILURE(hr);
ZeroMemory ( &UserInfo, sizeof ( UserInfo ) );
UserInfo.usri3_name = const_cast <LPWSTR> (wszUsername);
UserInfo.usri3_full_name = const_cast <LPWSTR> (wszUsername);
UserInfo.usri3_password = const_cast <LPWSTR> (wszPassword);
// UserInfo.usri3_password_age
UserInfo.usri3_priv = USER_PRIV_USER;
UserInfo.usri3_home_dir = wszNull;
UserInfo.usri3_comment = wszNull;
UserInfo.usri3_flags = UF_SCRIPT | UF_NORMAL_ACCOUNT;
UserInfo.usri3_script_path = wszNull;
UserInfo.usri3_auth_flags = 0;
UserInfo.usri3_usr_comment = wszNull;
UserInfo.usri3_parms = wszNull;
UserInfo.usri3_workstations = wszNull;
// UserInfo.usri3_last_logon =;
// UserInfo.usri3_last_logoff =;
UserInfo.usri3_acct_expires = TIMEQ_FOREVER;
UserInfo.usri3_max_storage = USER_MAXSTORAGE_UNLIMITED;
// UserInfo.usri3_units_per_week =;
// UserInfo.usri3_logon_hours = NULL;
// UserInfo.usri3_bad_pw_count =;
// UserInfo.usri3_num_logons =;
UserInfo.usri3_logon_server = wszNull;
UserInfo.usri3_country_code = 0;
UserInfo.usri3_code_page = CP_ACP;
// UserInfo.usri3_user_id =;
UserInfo.usri3_primary_group_id = DOMAIN_GROUP_RID_USERS;
UserInfo.usri3_profile = wszNull;
UserInfo.usri3_home_dir_drive = wszNull;
UserInfo.usri3_password_expired = FALSE;
//
// Add the user:
//
dwErr = NetUserAdd (
wszDC,
3,
(LPBYTE) &UserInfo,
&dwParmErr
);
if ( dwErr != NOERROR ) {
// TRACE ( _T("Failed to add user: %d, parmerr = %d\r\n"), dwErr, dwParmErr );
BAIL_WITH_FAILURE ( hr, HRESULT_FROM_WIN32 ( dwErr ) );
}
#if 0
//
// Attempt to add the user to the "USERS" group:
//
dwErr = NetGroupAddUser (
wszDC,
// (LPWSTR) (LPCWSTR) CString ( strDC + L"\\Users" ),
(LPWSTR) (LPCWSTR) strUsername
);
if ( dwErr != NOERROR ) {
TRACE ( _T("Couldn't add user to USERS group: %d\r\n"), dwErr );
}
#endif
Exit:
delete wszDC;
return hr;
}
LPCWSTR StripBackslashesFromComputerName ( LPCWSTR wsz )
{
if ( wsz[0] == _T('\\') &&
wsz[1] == _T('\\') ) {
return wsz + 2;
}
else {
return wsz;
}
}
static LPWSTR DuplicateString ( LPCWSTR wsz )
{
LPWSTR wszResult;
_ASSERT ( wsz );
wszResult = new WCHAR [ lstrlen ( wsz ) + 1 ];
if ( wszResult ) {
lstrcpy ( wszResult, wsz );
}
return wszResult;
}
void StringToUpper ( LPWSTR wsz )
{
while ( *wsz ) {
*wsz = towupper ( *wsz );
wsz++;
}
}
HRESULT GetDCName (
LPCWSTR wszServer,
LPCWSTR wszDomain,
LPWSTR * pwszDC
)
{
HRESULT hr = NOERROR;
DWORD dwError = NOERROR;
LPWSTR wszServerCopy = NULL;
LPWSTR wszDomainCopy = NULL;
LPCWSTR wszPlainServer = NULL;
LPCWSTR wszPlainDomain = NULL;
LPWSTR wszDC = NULL;
if ( wszServer == NULL ) {
wszServer = _T("");
}
wszServerCopy = DuplicateString ( wszServer );
wszDomainCopy = DuplicateString ( wszDomain );
if ( !wszServerCopy || !wszDomainCopy ) {
BAIL_WITH_FAILURE ( hr, E_OUTOFMEMORY );
}
StringToUpper ( wszServerCopy );
StringToUpper ( wszDomainCopy );
wszPlainServer = StripBackslashesFromComputerName ( wszServerCopy );
wszPlainDomain = StripBackslashesFromComputerName ( wszDomainCopy );
if ( ! lstrcmp ( wszPlainServer, wszPlainDomain ) ) {
*pwszDC = new WCHAR [ lstrlen (_T("\\\\")) + lstrlen (wszPlainServer) + 1 ];
if ( *pwszDC == NULL ) {
BAIL_WITH_FAILURE(hr, E_OUTOFMEMORY);
}
wsprintf ( *pwszDC, _T("\\\\%s"), wszPlainServer );
goto Exit;
}
dwError = NetGetDCName ( wszServer, wszDomain, (LPBYTE *) &wszDC );
if ( dwError != NOERROR ) {
//
// Error, try to get any DC name:
//
_ASSERT ( wszDC == NULL );
hr = HRESULT_FROM_WIN32 ( dwError );
dwError = NetGetAnyDCName ( wszServer, wszDomain, (LPBYTE *) &wszDC );
if ( dwError != NOERROR ) {
goto Exit;
}
}
_ASSERT ( dwError == NOERROR );
_ASSERT ( wszDC[0] == _T('\\') );
_ASSERT ( wszDC[1] == _T('\\') );
*pwszDC = new WCHAR [ lstrlen ( wszDC ) + 1 ];
if ( *pwszDC == NULL ) {
BAIL_WITH_FAILURE(hr, E_OUTOFMEMORY);
}
lstrcpy ( *pwszDC, wszDC );
Exit:
delete wszServerCopy;
delete wszDomainCopy;
if ( wszDC ) {
NetApiBufferFree ( wszDC );
}
return hr;
}
BOOL
IsValidEmailAddress (
LPCWSTR wszEmailAddress
)
{
char szBuf[512];
int cchCopied;
cchCopied = WideCharToMultiByte ( CP_ACP, 0, wszEmailAddress, -1, szBuf, sizeof (szBuf), NULL, NULL );
if ( cchCopied == 0 ) {
return FALSE;
}
return CAddr::ValidateEmailName( szBuf, FALSE );
}
#if 0
HRESULT
DeleteMailbox (
LPCWSTR wszServer,
LPCWSTR wszAlias,
LPCWSTR wszVDirPath,
LPCWSTR wszUsername, // OPTIONAL
LPCWSTR wszPassword // OPTIONAL
)
{
TraceFunctEnter ( "DeleteMailbox" );
HRESULT hr = NOERROR;
UCHAR szAlias[512];
ULONG lHashValue;
DWORD cchMailboxPath;
LPWSTR wszMailboxPath;
LPCWSTR wszFormat;
if ( !wszServer || !wszAlias || !wszVDirPath ) {
BAIL_WITH_FAILURE ( hr, E_INVALIDARG );
}
if ( !*wszServer || !*wszAlias || !*wszVDirPath ) {
BAIL_WITH_FAILURE ( hr, E_INVALIDARG );
}
//
// Compute the mailbox hash value:
//
WideCharToMultiByte ( CP_ACP, 0, wszAlias, -1, (char *) szAlias, ARRAY_SIZE ( szAlias ), NULL, NULL );
DebugTrace ( 0, "Deleting mailbox for %s", szAlias );
lHashValue = ElfHash ( szAlias ) % TABLE_SIZE;
DebugTrace ( 0, "Hash value = %u", lHashValue );
//
// Compute the mailbox directory path,
// this is: <vdir path>\<alias hash>\<alias>
//
cchMailboxPath =
lstrlen ( wszVDirPath ) + // <Virtual directory path>
1 + // '\'
20 + // <hash value>
1 + // '\'
lstrlen ( wszAlias ) + // <alias>
1; // null terminator
wszMailboxPath = new WCHAR [ cchMailboxPath ];
if ( !wszMailboxPath ) {
BAIL_WITH_FAILURE ( hr, E_OUTOFMEMORY );
}
// Avoid too many '\'s
if ( wszVDirPath [ lstrlen ( wszVDirPath ) - 1 ] == _T('\\') ) {
wszFormat = _T("%s%u\\%s");
}
else {
wszFormat = _T("%s\\%u\\%s");
}
wsprintf (
wszMailboxPath,
wszFormat,
wszVDirPath,
lHashValue,
wszAlias
);
RecursiveDeleteDirectory ( wszMailboxPath );
Exit:
TraceFunctLeave ();
return hr;
}
//---[ ElfHash() ]------------------------------------------------------------
//
// Pulled this function from the stacks project. Calculates the hash value
// given a username. The mailbox of a user is stored in a subdirectory of
// the form <virtual directory root>\<hash value%TABLE_SIZE>\<username>.
// The username is the lowercase of the name part of the proxy address for
// e.g. username@x.com.
//
// Params:
// UserName The username to be hashed.
//
// Return: (HRESULT)
// NOERROR if success, ADSI error code if failed.
//
//----------------------------------------------------------------------------
unsigned long ElfHash (const unsigned char * UserName)
{
unsigned long HashValue = 0, g;
while (*UserName)
{
HashValue = (HashValue << 4) + *UserName++;
if( g = HashValue & 0xF0000000)
HashValue ^= g >> 24;
HashValue &= ~g;
}
return HashValue;
}
HRESULT RecursiveDeleteDirectory (
LPCWSTR wszPath
)
{
TraceFunctEnter ( "DeleteDirectory" );
HRESULT hr = NOERROR;
DWORD cchOldCurrentDirectory;
LPWSTR wszOldCurrentDirectory = NULL;
DWORD cchPath = lstrlen ( wszPath );
HANDLE hSearch = INVALID_HANDLE_VALUE;
WIN32_FIND_DATA find;
BOOL fOk = TRUE;
HANDLE hFile;
//
// Save the current directory:
//
cchOldCurrentDirectory = GetCurrentDirectory ( 0, NULL );
_ASSERT ( cchOldCurrentDirectory > 0 );
wszOldCurrentDirectory = new WCHAR [ cchOldCurrentDirectory ];
if ( !wszOldCurrentDirectory ) {
BAIL_WITH_FAILURE(hr, E_OUTOFMEMORY);
}
GetCurrentDirectory ( cchOldCurrentDirectory, wszOldCurrentDirectory );
//
// Set the current directory to the path:
//
SetCurrentDirectory ( wszPath );
//
// Create the search string:
//
hSearch = FindFirstFile ( _T("*.*"), &find );
while ( hSearch != INVALID_HANDLE_VALUE && fOk ) {
//
// Skip the . and .. directories:
//
if (
lstrcmp ( find.cFileName, _T(".") ) &&
lstrcmp ( find.cFileName, _T("..") )
) {
if ( find.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY ) {
hr = RecursiveDeleteDirectory ( find.cFileName );
BAIL_ON_FAILURE(hr);
}
else {
//
// Delete this file:
//
hFile = CreateFile (
find.cFileName,
GENERIC_WRITE,
FILE_SHARE_DELETE,
NULL,
OPEN_EXISTING,
FILE_FLAG_DELETE_ON_CLOSE,
NULL
);
if ( hFile != INVALID_HANDLE_VALUE ) {
CloseHandle ( hFile );
}
// Otherwise, press on...
}
}
// Get the next file:
fOk = FindNextFile ( hSearch, &find );
}
//
// Delete the directory:
//
if ( hSearch != INVALID_HANDLE_VALUE ) {
FindClose ( hSearch );
hSearch = INVALID_HANDLE_VALUE;
}
hFile = CreateFile (
wszPath,
GENERIC_WRITE,
FILE_SHARE_DELETE,
NULL,
OPEN_EXISTING,
FILE_FLAG_DELETE_ON_CLOSE,
NULL
);
if ( hFile != INVALID_HANDLE_VALUE ) {
CloseHandle ( hFile );
}
Exit:
if ( wszOldCurrentDirectory ) {
//
// Restore the current directory:
//
SetCurrentDirectory ( wszOldCurrentDirectory );
delete wszOldCurrentDirectory;
}
if ( hSearch != INVALID_HANDLE_VALUE ) {
FindClose ( hSearch );
hSearch = INVALID_HANDLE_VALUE;
}
TraceFunctLeave ( );
return hr;
}
#endif