|
|
/*++
Copyright (c) 1996 Microsoft Corporation
Module Name:
rpcex.cxx
Abstract:
This module defines K2 rpc support.
Author:
Johnson Apacible (JohnsonA) June-19-1996
--*/
#include "ftpdp.hxx"
#include "time.h"
#include "timer.h"
# define ASSUMED_AVERAGE_USER_NAME_LEN ( 40)
# define CONN_LEEWAY ( 3)
//
// Private functions.
//
BOOLGenDoubleNullStringFromMultiLine( IN LPCWSTR lpsz, IN OUT LPWSTR * ppszz, IN OUT LPDWORD pcchLen){ DWORD cchLen; DWORD nLines; LPWSTR pszNext; LPCWSTR pszSrc;
DBG_ASSERT( lpsz != NULL && ppszz != NULL && pcchLen != NULL);
// Initialize
*ppszz = NULL; *pcchLen = 0;
//
// 1. Find the length of the the complete message including new lines
// For each new line we may potentially need an extra blank char
// So allocate space = nLines + length + 2 terminating null chars.
//
cchLen = lstrlenW( lpsz);
for ( pszSrc = lpsz, nLines = 0; *pszSrc != L'\0'; pszSrc++) {
if ( *pszSrc == L'\n') { nLines++; } } // for
// Allocate sufficient space for the string.
*ppszz = (LPWSTR ) TCP_ALLOC( (cchLen + nLines + 3) * sizeof(WCHAR)); if ( *ppszz == NULL) {
SetLastError(ERROR_NOT_ENOUGH_MEMORY); return (FALSE); }
//
// walk down the local copy and convert all the line feed characters to
// be null char
//
//
// Since the MULTI_SZ string cannot contain empty strings
// we convert empty lines (ones with just \n into " \0"
// i.e. with a blank character.
//
pszSrc = lpsz; LPWSTR pszDst = *ppszz;
if ( *pszSrc == L'\n') {
// first line is a linefeed. insert a blank and proceed to next line.
*pszDst = L' '; *(pszDst + 1) = L'\0';
// move forward
pszDst += 2; pszSrc++; }
for( ; *pszSrc != L'\0'; pszSrc++, pszDst++) {
if ( *pszSrc == L'\n') {
// we are at boundary of new line.
if ( pszSrc > lpsz && *(pszSrc - 1) == L'\n') {
// we detected an empty line. Store an additional blank.
*pszDst++ = L' '; }
*pszDst = L'\0'; // put null char in place of line feed.
} else {
*pszDst = *pszSrc; } } // for
*pszDst++ = L'\0'; // terminate with 1st null chars.
*pszDst++ = L'\0'; // terminate with 2nd null chars.
*pcchLen = DIFF(pszDst - *ppszz);
DBG_ASSERT( *pcchLen <= cchLen + nLines + 3);
return ( TRUE);} // GenDoubleNullStringFromMultiline()
BOOLFTP_SERVER_INSTANCE::WriteParamsToRegistry( LPFTP_CONFIG_INFO pConfig )/*++
This function writes parameters to the registry
Arguments: hkey HKEY for registry entry of parameters of FTP server. pConfig pointer to configuration information.
Returns: TRUE on success and FALSE if there is any failure.--*/{ DWORD err = NO_ERROR; BOOL fRet = TRUE;
HKEY hkey = NULL;
err = RegOpenKeyEx( HKEY_LOCAL_MACHINE, QueryRegParamKey(), 0, KEY_ALL_ACCESS, &hkey );
if( hkey == NULL ) { err = ERROR_INVALID_HANDLE; SetLastError( err); DBGPRINTF(( DBG_CONTEXT, "Invalid Registry key given. error %lu\n", err ));
return FALSE; }
//
// Write the registry data.
//
if( !err && IsFieldSet( pConfig->FieldControl, FC_FTP_ALLOW_ANONYMOUS ) ) { err = WriteRegistryDword( hkey, FTPD_ALLOW_ANONYMOUS, pConfig->fAllowAnonymous ); }
if( !err && IsFieldSet( pConfig->FieldControl, FC_FTP_ALLOW_GUEST_ACCESS ) ) { err = WriteRegistryDword( hkey, FTPD_ALLOW_GUEST_ACCESS, pConfig->fAllowGuestAccess ); }
if( !err && IsFieldSet( pConfig->FieldControl, FC_FTP_ANNOTATE_DIRECTORIES ) ) { err = WriteRegistryDword( hkey, FTPD_ANNOTATE_DIRS, pConfig->fAnnotateDirectories ); }
if( !err && IsFieldSet( pConfig->FieldControl, FC_FTP_ANONYMOUS_ONLY ) ) { err = WriteRegistryDword( hkey, FTPD_ANONYMOUS_ONLY, pConfig->fAnonymousOnly ); }
if( !err && IsFieldSet( pConfig->FieldControl, FC_FTP_LISTEN_BACKLOG ) ) { err = WriteRegistryDword( hkey, FTPD_LISTEN_BACKLOG, pConfig->dwListenBacklog ); }
if( !err && IsFieldSet( pConfig->FieldControl, FC_FTP_LOWERCASE_FILES ) ) { err = WriteRegistryDword( hkey, FTPD_LOWERCASE_FILES, pConfig->fLowercaseFiles ); }
if( !err && IsFieldSet( pConfig->FieldControl, FC_FTP_MSDOS_DIR_OUTPUT ) ) { err = WriteRegistryDword( hkey, FTPD_MSDOS_DIR_OUTPUT, pConfig->fMsdosDirOutput ); }
if( !err && IsFieldSet( pConfig->FieldControl, FC_FTP_SHOW_4_DIGIT_YEAR ) ) { err = WriteRegistryDword( hkey, FTPD_SHOW_4_DIGIT_YEAR, pConfig->fFourDigitYear ); }
if( !err && IsFieldSet( pConfig->FieldControl, FC_FTP_EXIT_MESSAGE ) ) { err = RegSetValueExW( hkey, FTPD_EXIT_MESSAGE_W, 0, REG_SZ, (BYTE *)pConfig->lpszExitMessage, ( wcslen( pConfig->lpszExitMessage ) + 1 ) * sizeof(WCHAR) ); }
if( !err && IsFieldSet( pConfig->FieldControl, FC_FTP_GREETING_MESSAGE ) ) {
LPWSTR pszzGreetingMessage = NULL; DWORD cchLen = 0;
if (GenDoubleNullStringFromMultiLine( pConfig->lpszGreetingMessage, &pszzGreetingMessage, &cchLen) ) {
DBG_ASSERT( pszzGreetingMessage != NULL);
err = RegSetValueExW( hkey, FTPD_GREETING_MESSAGE_W, 0, REG_MULTI_SZ, (BYTE *) pszzGreetingMessage, cchLen * sizeof(WCHAR));
TCP_FREE( pszzGreetingMessage); } else {
err = ERROR_NOT_ENOUGH_MEMORY; } }
if( !err && IsFieldSet( pConfig->FieldControl, FC_FTP_BANNER_MESSAGE ) ) {
LPWSTR pszzBannerMessage = NULL; DWORD cchLen = 0;
if (GenDoubleNullStringFromMultiLine( pConfig->lpszBannerMessage, &pszzBannerMessage, &cchLen) ) {
DBG_ASSERT( pszzBannerMessage != NULL);
err = RegSetValueExW( hkey, FTPD_BANNER_MESSAGE_W, 0, REG_MULTI_SZ, (BYTE *) pszzBannerMessage, cchLen * sizeof(WCHAR));
TCP_FREE( pszzBannerMessage); } else {
err = ERROR_NOT_ENOUGH_MEMORY; } }
if( !err && IsFieldSet( pConfig->FieldControl, FC_FTP_MAX_CLIENTS_MESSAGE ) ) { err = RegSetValueExW( hkey, FTPD_MAX_CLIENTS_MSG_W, 0, REG_SZ, (BYTE *)pConfig->lpszMaxClientsMessage, ( wcslen( pConfig->lpszMaxClientsMessage ) + 1 ) * sizeof(WCHAR) ); }
if( err ) { SetLastError( err ); return FALSE; }
return TRUE;
} // WriteParamsToRegistry
DWORDFTP_IIS_SERVICE::GetServiceConfigInfoSize( IN DWORD dwLevel ){ switch (dwLevel) { case 1: return sizeof(FTP_CONFIG_INFO); }
return 0;
} // FTP_IIS_SERVICE::GetServerConfigInfoSize
BOOLFTP_SERVER_INSTANCE::SetServiceConfig( IN PCHAR pBuffer )/*++
Description
Sets the common service admin information for the servers specified in dwServerMask.
Arguments:
pConfig - Admin information to set
Note:
--*/{ LPFTP_CONFIG_INFO pConfig = (LPFTP_CONFIG_INFO)pBuffer; DWORD err = NO_ERROR;
HKEY hKey;
err = RegOpenKeyEx( HKEY_LOCAL_MACHINE, QueryRegParamKey(), 0, KEY_ALL_ACCESS, &hKey );
if ( err != NO_ERROR ) { return(TRUE); }
//
// If success, then Write the new info to the registry, then read it back.
//
if( WriteParamsToRegistry( pConfig )) {
err = InitFromRegistry( pConfig->FieldControl); } else {
err = GetLastError(); }
IF_DEBUG( RPC ) { DBGPRINTF(( DBG_CONTEXT, "FtpSetServiceConfig returns with %d %lu\n", err )); }
return(err == NO_ERROR);
} // FTP_SERVER_INSTANCE::SetServiceConfig
BOOLFTP_SERVER_INSTANCE::GetServiceConfig( IN PCHAR pBuffer, IN DWORD dwLevel )/*++
Description
Retrieves the admin information
Arguments:
pBuffer - Buffer to fill up. dwLevel - info level of information to return.
Note:
--*/{
LPFTP_CONFIG_INFO pConfig = (LPFTP_CONFIG_INFO)pBuffer; DWORD err = NO_ERROR;
ZeroMemory( pConfig, sizeof(FTP_CONFIG_INFO) );
//
// Obtain and Return the admin information.
//
err = GetConfigInformation( pConfig);
IF_DEBUG( RPC) {
DBGPRINTF(( DBG_CONTEXT, "FtprGetAdminInformation() returns Error=%u\n", err)); }
return (err == NO_ERROR);
} // FTP_SERVER_INSTANCE::GetServiceConfig
BOOLFTP_SERVER_INSTANCE::EnumerateUsers( OUT PCHAR * pBuffer, OUT PDWORD nRead )/*++
Description
Enumerates the connected users.
Arguments:
pBuffer - Buffer to fill up. nRead - number of entries filled
--*/{ DWORD err; DWORD cbBuffer; LPIIS_USER_INFO_1 pInfo;
IF_DEBUG( RPC ) { DBGPRINTF(( DBG_CONTEXT,"Entering FtpEnumerateUsers\n")); }
//
// Determine the necessary buffer size.
//
cbBuffer = (GetCurrentConnectionsCount() + CONN_LEEWAY) * sizeof( IIS_USER_INFO_1 );
*nRead = 0;
pInfo = (LPIIS_USER_INFO_1)MIDL_user_allocate( cbBuffer);
if (pInfo == NULL) {
err = ERROR_NOT_ENOUGH_MEMORY; } else {
//
// Make a first attempt at enumerating the user info
//
err = NO_ERROR; if ( !::EnumerateUsers( (PCHAR)pInfo, &cbBuffer, nRead, this )) {
//
// Free up old buffer and allocate big one now.
// We will try once more to get the data again
// with a larger buffer.
//
if ( cbBuffer > 0) {
MIDL_user_free( pInfo ); pInfo = (LPIIS_USER_INFO_1)MIDL_user_allocate(cbBuffer);
if( pInfo == NULL ) {
err = ERROR_NOT_ENOUGH_MEMORY; } else {
//
// Since we do not lock the active connections list
// it is possible some one came in now and hence the
// buffer is insufficient to hold all people.
// Ignore this case, as we are never
// going to be accurate
//
::EnumerateUsers( (PCHAR)pInfo, &cbBuffer, nRead, this ); if ( *nRead == 0 ) { MIDL_user_free(pInfo); pInfo = NULL; } }
} // cbBuffer > 0
} // if unsuccessful at first attempt
}
if( err != NO_ERROR ) {
IF_DEBUG( RPC ) { DBGPRINTF(( DBG_CONTEXT, "I_FtprEnumerateUsers failed. Error = %lu\n", err )); } SetLastError(err); return(FALSE); }
*pBuffer = (PCHAR)pInfo;
IF_DEBUG( RPC ) { DBGPRINTF(( DBG_CONTEXT, "FtpEnumerateUsers returns %d entries, buffer [%x]\n", *nRead, *pBuffer )); } return TRUE;
} // EnumerateUsers
BOOLFTP_SERVER_INSTANCE::DisconnectUser( IN DWORD dwIdUser )/*++
Description
Disconnect the user
Arguments:
dwIdUser - Identifies the user to disconnect. If 0, then disconnect ALL users.
--*/{ IF_DEBUG( RPC ) { DBGPRINTF(( DBG_CONTEXT, "Entering FtpDisconnectUsers with id[%d]\n", dwIdUser)); }
if ( !::DisconnectUser( dwIdUser, this ) ) {
IF_DEBUG( RPC ) { DBGPRINTF(( DBG_CONTEXT, "DisconnectUser failed with %d\n", GetLastError())); } SetLastError(NERR_UserNotFound); return(FALSE); }
return(TRUE);} // DisconnectUser
BOOLFTP_SERVER_INSTANCE::GetStatistics( IN DWORD dwLevel, OUT PCHAR* pBuffer )/*++
Description
Disconnect Queries the server statistics
Arguments:
dwLevel - Info level. Currently only level 0 is supported.
pBuffer - Will receive a pointer to the statistics structure.
--*/{ APIERR err = NO_ERROR;
IF_DEBUG( RPC ) { DBGPRINTF(( DBG_CONTEXT, "FtpQueryStatistics2, level %lu\n", dwLevel )); }
//
// Return the proper statistics based on the infolevel.
//
switch( dwLevel ) { case 0 : {
LPFTP_STATISTICS_0 pstats0;
pstats0 = (LPFTP_STATISTICS_0) MIDL_user_allocate(sizeof(FTP_STATISTICS_0));
if( pstats0 == NULL ) {
err = ERROR_NOT_ENOUGH_MEMORY; } else { ATQ_STATISTICS atqStat;
ZeroMemory( pstats0, sizeof( FTP_STATISTICS_0 ) );
QueryStatsObj()->CopyToStatsBuffer( pstats0 );
//
// Get instance's bandwidth throttling statistics
//
if ( QueryBandwidthInfo() ) { if ( AtqBandwidthGetInfo( QueryBandwidthInfo(), ATQ_BW_STATISTICS, (ULONG_PTR *) &atqStat ) ) { pstats0->TotalBlockedRequests = atqStat.cBlockedRequests; pstats0->TotalRejectedRequests = atqStat.cRejectedRequests; pstats0->TotalAllowedRequests = atqStat.cAllowedRequests; pstats0->CurrentBlockedRequests = atqStat.cCurrentBlockedRequests; pstats0->MeasuredBandwidth = atqStat.MeasuredBandwidth; } }
pstats0->TimeOfLastClear = GetCurrentTimeInSeconds() - pstats0->TimeOfLastClear;
//
// Copy Global statistics counter values
//
pstats0->ConnectionAttempts = g_pFTPStats->QueryStatsObj()->ConnectionAttempts;
*pBuffer = (PCHAR)pstats0; }
}
break;
default : err = ERROR_INVALID_LEVEL; break; }
IF_DEBUG( RPC ) { DBGPRINTF(( DBG_CONTEXT, "FtpQueryStatistics2 returns Error = %lu\n", err )); }
SetLastError(err); return(err == NO_ERROR);
} // QueryStatistics
BOOLFTP_SERVER_INSTANCE::ClearStatistics( VOID )/*++
Description
Clears the server statistics
Arguments:
None.
--*/{
IF_DEBUG( RPC ) { DBGPRINTF(( DBG_CONTEXT, "Entering FtpClearStatistics2\n")); }
QueryStatsObj()->ClearStatistics(); return TRUE;
} // ClearStatistics
|
