mirror of https://github.com/lianthony/NT4.0
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.
502 lines
15 KiB
502 lines
15 KiB
/*++
|
|
|
|
Copyright (c) 1993 Microsoft Corporation
|
|
|
|
Module Name:
|
|
|
|
adapter.c
|
|
|
|
Abstract:
|
|
|
|
Adapter APIs.
|
|
|
|
Author:
|
|
|
|
Vladimir Z. Vulovic (vladimv) 19 - November - 1993
|
|
|
|
Revision History:
|
|
|
|
--*/
|
|
|
|
#include "local.h"
|
|
#include "rpldb.h"
|
|
#include "db.h"
|
|
#include "dblib.h"
|
|
#define RPLADAPTER_ALLOCATE
|
|
#include "adapter.h"
|
|
#undef RPLADAPTER_ALLOCATE
|
|
#include "database.h"
|
|
|
|
|
|
|
|
DWORD AdapterGetField(
|
|
IN PRPL_SESSION pSession,
|
|
IN DWORD FieldIndex,
|
|
OUT LPVOID * pData,
|
|
IN OUT LPINT pSpaceLeft
|
|
)
|
|
{
|
|
BYTE LocalBuffer[ 300];
|
|
PBYTE Buffer;
|
|
DWORD DataSize;
|
|
DWORD BufferSize;
|
|
JET_ERR JetError;
|
|
|
|
switch( FieldIndex) {
|
|
case ADAPTER_Flags:
|
|
Buffer = (PBYTE)pData;
|
|
BufferSize = sizeof( DWORD);
|
|
break;
|
|
default:
|
|
Buffer = LocalBuffer;
|
|
BufferSize = sizeof( LocalBuffer);
|
|
break;
|
|
}
|
|
JetError = JetRetrieveColumn( pSession->SesId, pSession->AdapterTableId,
|
|
AdapterTable[ FieldIndex].ColumnId, LocalBuffer,
|
|
sizeof( LocalBuffer), &DataSize, 0, NULL);
|
|
if ( JetError < 0) {
|
|
RplDump( ++RG_Assert, ("JetError=%d", JetError));
|
|
return( NERR_RplAdapterInfoCorrupted);
|
|
}
|
|
if ( Buffer != LocalBuffer) {
|
|
if ( BufferSize == DataSize) {
|
|
return( NO_ERROR);
|
|
} else {
|
|
RplDump( ++RG_Assert, ("Bad DataSize=0x%x", DataSize));
|
|
return( NERR_RplAdapterInfoCorrupted);
|
|
}
|
|
}
|
|
//
|
|
// We have done with fixed data. From here on we deal with unicode
|
|
// strings only.
|
|
//
|
|
if ( DataSize > sizeof( LocalBuffer)) {
|
|
RplDump( ++RG_Assert, ( "Too big DataSize=0x%x", DataSize));
|
|
return( NERR_RplAdapterInfoCorrupted);
|
|
}
|
|
if ( DataSize == 0) {
|
|
if ( JetError != JET_wrnColumnNull) {
|
|
RplDump( ++RG_Assert, ( "JetError=%d", JetError));
|
|
return( NERR_RplAdapterInfoCorrupted);
|
|
} else {
|
|
*pData = NULL; // so RPC rpcrt4!_tree_size_ndr() does not bomb here
|
|
return( NO_ERROR);
|
|
}
|
|
}
|
|
if ( DataSize & 1 != 0 || wcslen((PWCHAR)LocalBuffer) + 1 != DataSize/2) {
|
|
RplDump( ++RG_Assert, ("LocalBuffer=0x%x, DataSize=0x%x", LocalBuffer, DataSize));
|
|
return( NERR_RplAdapterInfoCorrupted);
|
|
}
|
|
*pData = MIDL_user_allocate( DataSize);
|
|
if ( *pData == NULL) {
|
|
RplDump( ++RG_Assert, ( "Error=%d", GetLastError()));
|
|
return( ERROR_NOT_ENOUGH_MEMORY);
|
|
}
|
|
memcpy( *pData, LocalBuffer, DataSize);
|
|
*pSpaceLeft -= DataSize;
|
|
return( NO_ERROR);
|
|
}
|
|
|
|
|
|
DWORD AdapterGetInfo(
|
|
IN PRPL_SESSION pSession,
|
|
IN LPWSTR AdapterName,
|
|
IN DWORD Level,
|
|
OUT LPVOID Buffer,
|
|
OUT PINT pSpaceLeft
|
|
)
|
|
{
|
|
DWORD Error;
|
|
LPRPL_ADAPTER_INFO_1 Info = Buffer;
|
|
|
|
switch( Level) {
|
|
case 1:
|
|
Error = AdapterGetField( pSession, ADAPTER_Flags, (LPVOID *)&Info->Flags, pSpaceLeft);
|
|
if ( Error != NO_ERROR) {
|
|
return( Error);
|
|
}
|
|
NOTHING; // fall through
|
|
case 0:
|
|
Error = AdapterGetField( pSession, ADAPTER_AdapterComment, &Info->AdapterComment, pSpaceLeft);
|
|
if ( Error != NO_ERROR) {
|
|
return( Error);
|
|
}
|
|
if ( AdapterName == NULL) {
|
|
Error = AdapterGetField( pSession, ADAPTER_AdapterName, &Info->AdapterName, pSpaceLeft);
|
|
if ( Error != NO_ERROR) {
|
|
return( Error);
|
|
}
|
|
} else {
|
|
DWORD DataSize = (wcslen( AdapterName) + 1) * sizeof(WCHAR);
|
|
Info->AdapterName = MIDL_user_allocate( DataSize);
|
|
if ( Info->AdapterName == NULL) {
|
|
return( ERROR_NOT_ENOUGH_MEMORY);
|
|
}
|
|
RplDump( RG_DebugLevel & RPL_DEBUG_ADAPTER, ( "AdapterName=0x%x", Info->AdapterName));
|
|
memcpy( Info->AdapterName, AdapterName, DataSize);
|
|
*pSpaceLeft -= DataSize;
|
|
}
|
|
break;
|
|
default:
|
|
return( ERROR_INVALID_LEVEL);
|
|
break;
|
|
}
|
|
return( NO_ERROR);
|
|
}
|
|
|
|
|
|
|
|
VOID AdapterGetInfoCleanup(
|
|
IN DWORD Level,
|
|
IN OUT LPVOID Buffer
|
|
)
|
|
{
|
|
switch( Level) {
|
|
case 0: {
|
|
LPRPL_ADAPTER_INFO_0 Info = Buffer;
|
|
if ( Info->AdapterName != NULL) {
|
|
MIDL_user_free( Info->AdapterName);
|
|
}
|
|
if ( Info->AdapterComment != NULL) {
|
|
MIDL_user_free( Info->AdapterComment);
|
|
}
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
DWORD AdapterSetInfo(
|
|
IN PRPL_SESSION pSession,
|
|
IN DWORD Level,
|
|
OUT LPVOID Buffer,
|
|
OUT LPDWORD pErrorParameter
|
|
)
|
|
{
|
|
LPRPL_ADAPTER_INFO_1 Info = Buffer;
|
|
switch( Level) {
|
|
case 1:
|
|
//
|
|
// Must initialize Flags - or will trap in GetField.
|
|
//
|
|
{
|
|
*pErrorParameter = ADAPTER_Flags;
|
|
CallM( JetSetColumn( pSession->SesId, pSession->AdapterTableId,
|
|
AdapterTable[ ADAPTER_Flags].ColumnId,
|
|
&Info->Flags, sizeof( Info->Flags), 0, NULL));
|
|
}
|
|
NOTHING; // fall through
|
|
case 0:
|
|
if ( Info->AdapterComment != NULL) {
|
|
*pErrorParameter = ADAPTER_AdapterComment;
|
|
CallM( JetSetColumn( pSession->SesId, pSession->AdapterTableId,
|
|
AdapterTable[ ADAPTER_AdapterComment].ColumnId,
|
|
Info->AdapterComment,
|
|
( wcslen( Info->AdapterComment) + 1) * sizeof(WCHAR),
|
|
0, NULL));
|
|
}
|
|
if ( Info->AdapterName != NULL) {
|
|
*pErrorParameter = ADAPTER_AdapterName;
|
|
CallM( JetSetColumn( pSession->SesId, pSession->AdapterTableId,
|
|
AdapterTable[ ADAPTER_AdapterName].ColumnId,
|
|
Info->AdapterName,
|
|
( wcslen( Info->AdapterName) + 1) * sizeof(WCHAR),
|
|
0, NULL));
|
|
}
|
|
break;
|
|
}
|
|
return( NO_ERROR);
|
|
}
|
|
|
|
|
|
NET_API_STATUS NET_API_FUNCTION
|
|
NetrRplAdapterAdd(
|
|
IN RPL_HANDLE ServerHandle,
|
|
IN DWORD Level,
|
|
IN LPRPL_ADAPTER_INFO_STRUCT AdapterInfoStruct,
|
|
OUT LPDWORD pErrorParameter OPTIONAL
|
|
)
|
|
{
|
|
LPRPL_ADAPTER_INFO_1 Info;
|
|
LPVOID Buffer;
|
|
DWORD Error;
|
|
DWORD ErrorParameter;
|
|
PRPL_SESSION pSession = &RG_ApiSession;
|
|
|
|
ErrorParameter = INVALID_ERROR_PARAMETER;
|
|
|
|
Buffer = Info = AdapterInfoStruct->AdapterInfo1;
|
|
switch( Level) {
|
|
case 1:
|
|
if ( Info->Flags != 0) {
|
|
ErrorParameter = ADAPTER_Flags;
|
|
break;
|
|
}
|
|
if ( RPL_STRING_TOO_LONG( Info->AdapterComment)) {
|
|
ErrorParameter = ADAPTER_AdapterComment;
|
|
break;
|
|
}
|
|
if ( !ValidHexName( Info->AdapterName, RPL_ADAPTER_NAME_LENGTH, TRUE)) {
|
|
ErrorParameter = ADAPTER_AdapterName;
|
|
break;
|
|
}
|
|
_wcsupr( Info->AdapterName);
|
|
break;
|
|
default:
|
|
return( ERROR_INVALID_LEVEL);
|
|
break;
|
|
}
|
|
|
|
if ( ErrorParameter != INVALID_ERROR_PARAMETER) {
|
|
if ( ARGUMENT_PRESENT( pErrorParameter)) {
|
|
*pErrorParameter = ErrorParameter;
|
|
}
|
|
return( ERROR_INVALID_PARAMETER);
|
|
}
|
|
|
|
EnterCriticalSection( &RG_ProtectDatabase);
|
|
Call( JetBeginTransaction( pSession->SesId));
|
|
|
|
//
|
|
// Verify that AdapterName is available in the database.
|
|
//
|
|
if ( RplFind( pSession, ADAPTER_TABLE_TAG, Info->AdapterName)) {
|
|
Error = NERR_RplAdapterNameUnavailable;
|
|
goto cleanup;
|
|
}
|
|
|
|
CallJ( JetPrepareUpdate( pSession->SesId, pSession->AdapterTableId, JET_prepInsert));
|
|
|
|
Error = AdapterSetInfo( pSession, Level, Buffer, &ErrorParameter);
|
|
if ( Error == ERROR_SUCCESS) {
|
|
ErrorParameter = 0;
|
|
CallJ( JetUpdate( pSession->SesId, pSession->AdapterTableId, NULL, 0, NULL));
|
|
}
|
|
|
|
cleanup:
|
|
if ( Error == NO_ERROR) {
|
|
Call( JetCommitTransaction( pSession->SesId, JET_bitCommitFlush));
|
|
} else {
|
|
Call( JetRollback( pSession->SesId, JET_bitRollbackAll));
|
|
}
|
|
LeaveCriticalSection( &RG_ProtectDatabase);
|
|
|
|
if ( Error != ERROR_SUCCESS) {
|
|
if ( ARGUMENT_PRESENT( pErrorParameter)) {
|
|
*pErrorParameter = ErrorParameter;
|
|
}
|
|
}
|
|
return( Error);
|
|
}
|
|
|
|
|
|
|
|
NET_API_STATUS NET_API_FUNCTION
|
|
NetrRplAdapterDel(
|
|
IN RPL_HANDLE ServerHandle,
|
|
IN LPWSTR AdapterName
|
|
)
|
|
/*++
|
|
If AdapterName is provided then delete matching record, else delete all
|
|
adapter records.
|
|
--*/
|
|
{
|
|
JET_ERR JetError;
|
|
DWORD Error = NO_ERROR;
|
|
PRPL_SESSION pSession = &RG_ApiSession;
|
|
|
|
if ( !ValidHexName( AdapterName, RPL_ADAPTER_NAME_LENGTH, FALSE)) {
|
|
return( ERROR_INVALID_PARAMETER);
|
|
}
|
|
if ( AdapterName != NULL) {
|
|
_wcsupr( AdapterName);
|
|
}
|
|
|
|
EnterCriticalSection( &RG_ProtectDatabase);
|
|
Call( JetBeginTransaction( pSession->SesId));
|
|
|
|
if ( AdapterName != NULL) {
|
|
if ( !RplFind( pSession, ADAPTER_TABLE_TAG, AdapterName)) {
|
|
Error = NERR_RplAdapterNotFound;
|
|
goto cleanup;
|
|
}
|
|
CallJ( JetDelete( pSession->SesId, pSession->AdapterTableId));
|
|
} else {
|
|
CallJ( JetSetCurrentIndex( pSession->SesId, pSession->AdapterTableId, ADAPTER_INDEX_AdapterName));
|
|
//
|
|
// The call to move to the beginning of the table is not redundant.
|
|
// E.g. JET_errNoCurrentRecord will be returned in case of empty table.
|
|
//
|
|
JetError = JetMove( pSession->SesId, pSession->AdapterTableId, JET_MoveFirst, 0);
|
|
if ( JetError < 0) {
|
|
if ( JetError != JET_errRecordNotFound
|
|
&& JetError != JET_errNoCurrentRecord) {
|
|
RplDump( ++RG_Assert, ("JetError=%d", JetError));
|
|
Error = NERR_RplInternal;
|
|
}
|
|
goto cleanup;
|
|
}
|
|
do {
|
|
CallJ( JetDelete( pSession->SesId, pSession->AdapterTableId));
|
|
JetError = JetMove( pSession->SesId, pSession->AdapterTableId, JET_MoveNext, 0);
|
|
} while ( JetError == JET_errSuccess);
|
|
}
|
|
|
|
cleanup:
|
|
if ( Error == NO_ERROR) {
|
|
Call( JetCommitTransaction( pSession->SesId, JET_bitCommitFlush));
|
|
} else {
|
|
Call( JetRollback( pSession->SesId, JET_bitRollbackAll));
|
|
}
|
|
LeaveCriticalSection( &RG_ProtectDatabase);
|
|
return( Error);
|
|
}
|
|
|
|
|
|
NET_API_STATUS NET_API_FUNCTION
|
|
NetrRplAdapterEnum(
|
|
IN RPL_HANDLE ServerHandle,
|
|
IN OUT LPRPL_ADAPTER_ENUM AdapterEnum,
|
|
IN DWORD PrefMaxLength,
|
|
OUT LPDWORD TotalEntries,
|
|
OUT LPDWORD pResumeHandle OPTIONAL
|
|
)
|
|
/*++
|
|
For more extensive comments see related code in NetrAdapterEnum.
|
|
--*/
|
|
{
|
|
LPBYTE Buffer;
|
|
DWORD TypicalSize;
|
|
DWORD CoreSize;
|
|
DWORD Error;
|
|
INT SpaceLeft;
|
|
DWORD ArrayLength;
|
|
DWORD EntriesRead;
|
|
JET_ERR JetError;
|
|
BOOL InfoError;
|
|
BOOL TableEnd;
|
|
PRPL_SESSION pSession = &RG_ApiSession;
|
|
|
|
//
|
|
// For now we here we disallow all levels but 0 level. All other code
|
|
// would work for levels 0 and 1.
|
|
//
|
|
switch( AdapterEnum->Level) {
|
|
case 1:
|
|
TypicalSize = CoreSize = sizeof( RPL_ADAPTER_INFO_1);
|
|
NOTHING; // fall through
|
|
case 0:
|
|
if ( AdapterEnum->Level == 0) {
|
|
TypicalSize = CoreSize = sizeof( RPL_ADAPTER_INFO_0);
|
|
}
|
|
TypicalSize += 8 * sizeof( WCHAR); // typical size of AdapterName
|
|
TypicalSize += 20 * sizeof( WCHAR); // typical size of AdapterComment
|
|
break;
|
|
default:
|
|
return( ERROR_INVALID_LEVEL);
|
|
break;
|
|
}
|
|
|
|
if ( PrefMaxLength == -1) {
|
|
SpaceLeft = DEFAULT_BUFFER_SIZE;
|
|
} else {
|
|
SpaceLeft = PrefMaxLength;
|
|
}
|
|
|
|
ArrayLength = SpaceLeft / TypicalSize;
|
|
if ( ArrayLength == 0) {
|
|
ArrayLength = 1; // try to return at least one element
|
|
}
|
|
|
|
Buffer = MIDL_user_allocate( ArrayLength * CoreSize);
|
|
if ( Buffer == NULL) {
|
|
return( ERROR_NOT_ENOUGH_MEMORY);
|
|
}
|
|
RplDump( RG_DebugLevel & RPL_DEBUG_ADAPTER, (
|
|
"AdapterEnum: Buffer=0x%x, ArrayLength=0x%x", Buffer, ArrayLength));
|
|
|
|
AdapterEnum->AdapterInfo.Level0->Buffer = (LPRPL_ADAPTER_INFO_0)Buffer;
|
|
|
|
EntriesRead = 0;
|
|
InfoError = FALSE;
|
|
Error = NO_ERROR;
|
|
|
|
EnterCriticalSection( &RG_ProtectDatabase);
|
|
Call( JetBeginTransaction( pSession->SesId));
|
|
|
|
if ( !RplFilterFirst( pSession, ADAPTER_TABLE_TAG, NULL, pResumeHandle, &TableEnd)) {
|
|
Error = NERR_RplCannotEnum;
|
|
goto cleanup;
|
|
}
|
|
if ( TableEnd == TRUE) {
|
|
goto cleanup;
|
|
}
|
|
for ( ; ; ) {
|
|
memset( Buffer, 0, CoreSize); // for cleanup to work properly
|
|
Error = AdapterGetInfo( pSession, NULL, AdapterEnum->Level, Buffer, &SpaceLeft);
|
|
if ( Error != NO_ERROR) {
|
|
InfoError = TRUE; // clean things up without holding crit sec
|
|
break;
|
|
}
|
|
EntriesRead++;
|
|
Buffer += CoreSize;
|
|
SpaceLeft -= CoreSize;
|
|
JetError = JetMove( pSession->SesId, pSession->AdapterTableId, JET_MoveNext, 0);
|
|
if ( JetError != JET_errSuccess) {
|
|
break; // assume end of table
|
|
}
|
|
if ( SpaceLeft <= 0) {
|
|
Error = ERROR_MORE_DATA;
|
|
break;
|
|
}
|
|
if ( EntriesRead >= ArrayLength) {
|
|
Error = ERROR_MORE_DATA;
|
|
break;
|
|
}
|
|
}
|
|
cleanup:
|
|
Call( JetCommitTransaction( pSession->SesId, 0));
|
|
LeaveCriticalSection( &RG_ProtectDatabase);
|
|
if ( InfoError == TRUE) {
|
|
AdapterGetInfoCleanup( AdapterEnum->Level, Buffer);
|
|
}
|
|
if ( Error == NO_ERROR) {
|
|
*TotalEntries = EntriesRead;
|
|
} else if ( Error == ERROR_MORE_DATA) {
|
|
*TotalEntries = EntriesRead * 2; // we cheat here
|
|
} else {
|
|
//
|
|
// Cleanup in case of "bad" errors.
|
|
//
|
|
while ( EntriesRead > 0) {
|
|
EntriesRead--;
|
|
Buffer -= CoreSize;
|
|
AdapterGetInfoCleanup( AdapterEnum->Level, Buffer);
|
|
}
|
|
MIDL_user_free( Buffer);
|
|
}
|
|
|
|
RplDump( RG_DebugLevel & RPL_DEBUG_ADAPTER, ("AdapterEnum: EntriesRead = 0x%x", EntriesRead));
|
|
|
|
AdapterEnum->AdapterInfo.Level0->EntriesRead = EntriesRead;
|
|
if ( EntriesRead == 0) {
|
|
AdapterEnum->AdapterInfo.Level0->Buffer = NULL;
|
|
}
|
|
|
|
if ( ARGUMENT_PRESENT( pResumeHandle)) {
|
|
if ( Error == ERROR_MORE_DATA && EntriesRead > 0) {
|
|
EnterCriticalSection( &RG_ProtectDatabase);
|
|
Call( JetBeginTransaction( pSession->SesId));
|
|
RplFilterSave( pSession, (DWORD)ServerHandle, NULL,
|
|
((LPRPL_ADAPTER_INFO_0)(Buffer-CoreSize))->AdapterName,
|
|
pResumeHandle);
|
|
Call( JetCommitTransaction( pSession->SesId, JET_bitCommitFlush));
|
|
LeaveCriticalSection( &RG_ProtectDatabase);
|
|
} else {
|
|
*pResumeHandle = 0; // resume from beginning
|
|
}
|
|
}
|
|
return( Error);
|
|
}
|
|
|