windows-nt/Source/XPSP1/NT/ds/nw/nwscript/nds.c
2020-09-26 16:20:57 +08:00

1644 lines
38 KiB
C

/*************************************************************************
*
* NDS.C
*
* NT NetWare NDS routines
*
* Copyright (c) 1995 Microsoft Corporation
*
*************************************************************************/
#include <common.h>
DWORD GUserObjectID;
HANDLE GhRdrForUser;
HANDLE GhRdr;
WCHAR * NDSTREE_w = NULL;
UNICODE_STRING NDSTREE_u;
/********************************************************************
ExpandRelativeName
Routine Description:
If the name is a relative NDS name append the proper context
to the end. A relative name has periods on the end. Each
period represents one level up the NDS tree.
Arguments:
Return Value:
*******************************************************************/
void
ExpandRelativeName( LPSTR RelativeName, LPSTR AbsoluteName, unsigned int Len,
LPSTR Context )
{
PBYTE ptr;
unsigned int i;
unsigned int count = 0;
strncpy( AbsoluteName, RelativeName, Len );
if ( ( AbsoluteName[0] == '.' ) &&
( AbsoluteName[ strlen( AbsoluteName ) - 1 ] != '.' ) )
return;
if ( ( strlen( AbsoluteName ) + strlen( Context ) ) > Len )
{
DisplayMessage( IDR_NOT_ENOUGH_MEMORY );
return;
}
if ( AbsoluteName[0] == '\0' )
{
return;
}
ptr = &AbsoluteName[ strlen( AbsoluteName ) - 1 ];
// Count the number of periods and back up over them.
if ( *ptr != '.' )
{
//
// No periods at the end
// Assume this is a relative name and append the context
//
strcat( AbsoluteName, "." );
strcat( AbsoluteName + strlen( AbsoluteName ), Context );
return;
}
while ( *ptr == '.' )
{
ptr--;
count++;
}
ptr++;
*ptr = '\0';
// ptr now points to where the copy of the rest of the context should start
// skip the first "count" entries in the context
ptr = Context;
for ( i = 0; i < count; i++ )
{
ptr = strchr( ptr, '.' );
if ( ptr == NULL )
{
return;
}
ptr++;
}
ptr--;
// Now append
strcat( AbsoluteName, ptr );
}
/********************************************************************
NDSGetNameContext
Routine Description:
Get the current context
Arguments:
none
Return Value:
none
*******************************************************************/
NTSTATUS
NDSGetNameContext( LPSTR Context, BOOLEAN flag )
{
//
// For NdsResolveName.
//
UNICODE_STRING ReferredServer;
WCHAR ServerStr[MAX_NAME_LEN];
HANDLE hReferredServer;
DWORD dwHandleType;
NTSTATUS Status;
OEM_STRING oemStr;
UNICODE_STRING defaultcontext;
DWORD ThisObjectID;
BYTE Buffer[2048];
WCHAR NdsStr[1024];
PBYTE ptr;
defaultcontext.Length = 0;
defaultcontext.MaximumLength = sizeof( NdsStr );
defaultcontext.Buffer = NdsStr;
Status = NwNdsGetTreeContext( GhRdr, &NDSTREE_u, &defaultcontext );
if ( !NT_SUCCESS( Status ) ) {
return Status;
}
ReferredServer.Buffer = ServerStr;
ReferredServer.Length = 0;
ReferredServer.MaximumLength = sizeof( ServerStr );
Status = NwNdsResolveName ( GhRdr,
&defaultcontext,
&ThisObjectID,
&ReferredServer,
NULL,
0 );
if ( !NT_SUCCESS( Status ) ) {
return Status;
}
if ( ReferredServer.Length > 0 ) {
//
// We've been referred to another server, so we
// should change the global handle.
//
Status = NwNdsOpenGenericHandle( &ReferredServer,
&dwHandleType,
&hReferredServer );
if ( !NT_SUCCESS( Status ) ) {
DisplayMessage(IDR_NDS_USERNAME_FAILED);
return Status;
}
if( GhRdr != GhRdrForUser ) {
CloseHandle( GhRdr );
}
GhRdr = hReferredServer;
}
Status = NwNdsReadObjectInfo( GhRdr, ThisObjectID, Buffer, 2048 );
if ( !NT_SUCCESS( Status ) ) {
return Status;
}
ptr = Buffer + sizeof( NDS_RESPONSE_GET_OBJECT_INFO );
ptr += ROUNDUP4(*(DWORD *)ptr);
ptr += sizeof(DWORD);
ptr += sizeof(DWORD);
defaultcontext.Length = wcslen( (WCHAR *)ptr ) * 2;
defaultcontext.MaximumLength = defaultcontext.Length;
defaultcontext.Buffer = (WCHAR *)ptr;
oemStr.Length = 0;
oemStr.MaximumLength = NDS_NAME_CHARS;
oemStr.Buffer = Context;
RtlUnicodeStringToOemString( &oemStr, &defaultcontext, FALSE );
return 0;
}
/********************************************************************
NDSTypeless
Routine Description:
Change name to typelese
Arguments:
none
Return Value:
none
*******************************************************************/
unsigned int
NDSTypeless( LPSTR OrigName , LPSTR TypelessName )
{
int i,j;
PBYTE p;
i = 0;
j = 0;
if ( !_strnicmp( "CN=", OrigName, 3 ) ||
!_strnicmp( "OU=", OrigName, 3 ) )
{
i += 3;
}
else if ( !_strnicmp( "C=", OrigName, 2 ) ||
!_strnicmp( "O=", OrigName, 2 ) )
{
i += 2;
}
for ( ; (( i < NDS_NAME_CHARS ) && ( OrigName[i] ) ); i++ )
{
if ( !_strnicmp( ".CN=", &OrigName[i], 4 ) ||
!_strnicmp( ".OU=", &OrigName[i], 4 ) )
{
TypelessName[j++]= '.';
i += 3;
continue;
}
if ( !_strnicmp( ".C=", &OrigName[i], 3 ) ||
!_strnicmp( ".O=", &OrigName[i], 3 ) )
{
TypelessName[j++]= '.';
i += 2;
continue;
}
/*
* Strip out multiple blanks
*/
if ( !_strnicmp( " ", &OrigName[i], 2 ) )
{
continue;
}
TypelessName[j++] = OrigName[i];
}
TypelessName[j] = '\0';
return 0;
}
/********************************************************************
NDSAbbreviateName
Routine Description:
Abbreviate name
Arguments:
none
Return Value:
none
*******************************************************************/
unsigned int
NDSAbbreviateName( DWORD Flags, LPSTR OrigName , LPSTR AbbrevName )
{
BYTE Buffer[NDS_NAME_CHARS];
BYTE CurrentContext[NDS_NAME_CHARS];
PBYTE p;
PBYTE c;
NTSTATUS Status;
if ( OrigName[0] == '.' )
NDSTypeless( OrigName + 1, Buffer );
else
NDSTypeless( OrigName, Buffer );
/*
* We want a relative name
*/
if ( Flags & FLAGS_LOCAL_CONTEXT )
{
p = &Buffer[strlen(Buffer)-strlen(REQUESTER_CONTEXT)];
if ( !_strcmpi( REQUESTER_CONTEXT, p ) )
{
// The name is below us
if ( ( *(p-1) == '.' ) && ( p > Buffer ) )
p--;
*p = '\0';
strcpy( AbbrevName, Buffer );
}
else
{
//
// Going from back to front for each section of context
// in common with AbbrevName
// truncate both
// Going from back to front for each section of context
// left over
// concatonate a period to AbbrevName
//
// Example
//
// Name: w.x.y.z Context: a.b.z => w.x.y..
//
strcpy( CurrentContext, REQUESTER_CONTEXT );
strcpy( AbbrevName, Buffer );
if ( CurrentContext[0] && AbbrevName[0] )
{
c = &CurrentContext[ strlen( CurrentContext ) ] - 1;
p = &AbbrevName[ strlen( AbbrevName ) ] - 1;
//
// Strip off the matching names from end to front
//
for ( ;; )
{
if ( ( c == CurrentContext ) && ( *p == '.' ) )
{
*c = '\0';
*p = '\0';
break;
}
if ( *c != *p )
break;
if ( ( *c == '.' ) && ( *p == '.' ) )
{
*c = '\0';
*p = '\0';
}
if ( ( c == CurrentContext ) || ( p == AbbrevName ) )
{
break;
}
c--; p--;
}
//
// Count the remaining sections of the context and
// add that number of periods to the end of the buffer.
// That is how far we need to back up before getting
// to a matching branch of the tree.
//
if ( CurrentContext[0] ) {
strcat( AbbrevName, "." );
for ( c = CurrentContext; *c; c++ ) {
if ( *c == '.' )
strcat( AbbrevName, "." );
}
}
}
}
}
else
strcpy( AbbrevName, Buffer );
return 0;
}
/********************************************************************
NDSInitUserProperty
Routine Description:
none
Arguments:
none
Return Value:
0 = no error
*******************************************************************/
unsigned int
NDSInitUserProperty( )
{
NTSTATUS Status;
UNICODE_STRING ObjectName;
PWCHAR lpT;
UNICODE_STRING defaultcontext;
//
// For NdsResolveName.
//
UNICODE_STRING ReferredServer;
WCHAR ServerStr[MAX_NAME_LEN];
HANDLE hReferredServer;
DWORD dwHandleType;
//
// Get a handle to the redirector.
//
Status = NwNdsOpenTreeHandle( &NDSTREE_u, &GhRdr );
if ( !NT_SUCCESS( Status ) ) {
DisplayMessage(IDR_TREE_OPEN_FAILED);
return 1;
}
//
// Resolve the name that we have to an object id.
//
RtlInitUnicodeString( &ObjectName, TYPED_USER_NAME_w );
ReferredServer.Buffer = ServerStr;
ReferredServer.Length = 0;
ReferredServer.MaximumLength = sizeof( ServerStr );
Status = NwNdsResolveName ( GhRdr,
&ObjectName,
&GUserObjectID,
&ReferredServer,
NULL,
0 );
if ( !NT_SUCCESS( Status ) ) {
DisplayMessage(IDR_NDS_USERNAME_FAILED);
return 1;
}
if ( ReferredServer.Length > 0 ) {
//
// We've been referred to another server, so we
// should change the global handle.
//
Status = NwNdsOpenGenericHandle( &ReferredServer,
&dwHandleType,
&hReferredServer );
if ( !NT_SUCCESS( Status ) ) {
DisplayMessage(IDR_NDS_USERNAME_FAILED);
return 1;
}
CloseHandle( GhRdr );
GhRdr = hReferredServer;
}
//
// Save off this handle for the user so that we can use it to
// get information about the user.
//
GhRdrForUser = GhRdr;
//
// Set the current context to what we think it should be
// (At the user's location.)
//
lpT = wcschr( TYPED_USER_NAME_w, L'.' );
while (lpT) // handle usernames with embedded/escaped dots
{
if (*(lpT-1) == L'\\')
{
lpT = wcschr (lpT+1, L'.');
}
else
break;
}
if ( lpT )
{
RtlInitUnicodeString( &defaultcontext, lpT+1 );
}
else
{
RtlInitUnicodeString( &defaultcontext, L"" );
}
Status = NwNdsSetTreeContext( GhRdr, &NDSTREE_u, &defaultcontext );
if ( !NT_SUCCESS( Status ) ) {
DisplayMessage(IDR_NDS_CONTEXT_INVALID);
return 1;
}
return 0;
}
/********************************************************************
NDSCanonicalizeName
Routine Description:
return a canonicalized version of a name
Arguments:
Name - original name
CanonName - Canonicalized name
Len - length of CanonName
fCurrentContext - TRUE => use current contex, FALSE use
requester context
Return Value:
status error
*******************************************************************/
unsigned int
NDSCanonicalizeName( PBYTE Name, PBYTE CanonName, int Len, int fCurrentContext )
{
NTSTATUS Status;
int ccode = -1;
DWORD ThisObjectID;
OEM_STRING oemStr;
UNICODE_STRING ObjectName;
BYTE Buffer[2048];
BYTE FullName[NDS_NAME_CHARS];
PBYTE ptr;
UNICODE_STRING ReferredServer;
WCHAR ServerStr[MAX_NAME_LEN];
DWORD dwHandleType;
HANDLE hReferredServer;
unsigned char CurrentContext[NDS_NAME_CHARS];
//
// Cope with relative names
//
if ( fCurrentContext )
{
Status = NDSGetNameContext( CurrentContext, TRUE );
if ( !NT_SUCCESS( Status ) )
return Status;
ExpandRelativeName( Name, FullName, NDS_NAME_CHARS, CurrentContext );
}
else
ExpandRelativeName( Name, FullName, NDS_NAME_CHARS, REQUESTER_CONTEXT );
//
// Fill it in in case we have an error
//
strncpy( CanonName, FullName, Len);
//
// Resolve the name that we have to an object id.
//
// Unfortuneately, the name resolver doesn't understand periods at the
// front or end (absolute or relative names)
//
if ( FullName[0] == '.' )
{
oemStr.Length = (USHORT)strlen( FullName + 1 );
oemStr.MaximumLength = oemStr.Length;
oemStr.Buffer = FullName + 1;
}
else
{
oemStr.Length = (USHORT)strlen( FullName );
oemStr.MaximumLength = oemStr.Length;
oemStr.Buffer = FullName;
}
ObjectName.Length = 0;
ObjectName.MaximumLength = sizeof(Buffer);
ObjectName.Buffer = (WCHAR *)Buffer;
RtlOemStringToUnicodeString( &ObjectName, &oemStr, FALSE );
ReferredServer.Buffer = ServerStr;
ReferredServer.Length = 0;
ReferredServer.MaximumLength = sizeof( ServerStr );
Status = NwNdsResolveName ( GhRdr,
&ObjectName,
&ThisObjectID,
&ReferredServer,
NULL,
0 );
if ( !NT_SUCCESS( Status ) ) {
return Status;
}
if ( ReferredServer.Length > 0 ) {
//
// We've been referred to another server, so we
// should change the global handle.
//
Status = NwNdsOpenGenericHandle( &ReferredServer,
&dwHandleType,
&hReferredServer );
if ( !NT_SUCCESS( Status ) ) {
return Status;
}
if( GhRdr != GhRdrForUser ) {
CloseHandle( GhRdr );
}
GhRdr = hReferredServer;
}
Status = NwNdsReadObjectInfo( GhRdr, ThisObjectID, Buffer, 2048 );
if ( !NT_SUCCESS( Status ) ) {
return Status;
}
ptr = Buffer + sizeof( NDS_RESPONSE_GET_OBJECT_INFO );
ptr += ROUNDUP4(*(DWORD *)ptr);
ptr += sizeof(DWORD);
ptr += sizeof(DWORD);
RtlInitUnicodeString( &ObjectName, (PWCHAR)ptr );
oemStr.Length = 0;
oemStr.MaximumLength = (USHORT) Len;
oemStr.Buffer = CanonName;
RtlUnicodeStringToOemString( &oemStr, &ObjectName, FALSE );
return 0;
}
/********************************************************************
NDSGetUserProperty
Routine Description:
Return the NDS property for the object
Arguments:
Property - property name
Data - data buffer
Size - size of data buffer
Return Value:
0 no error
*******************************************************************/
unsigned int
NDSGetUserProperty( PBYTE Property,
PBYTE Data,
unsigned int Size,
SYNTAX * pSyntaxID,
unsigned int * pActualSize )
{
NTSTATUS Status = STATUS_SUCCESS;
int ccode = -1;
OEM_STRING oemStr;
UNICODE_STRING PropertyName;
WCHAR NdsStr[1024];
DWORD iterhandle = INITIAL_ITERATION;
PBYTE szBuffer;
DWORD dwBufferSize = 2048;
PNDS_RESPONSE_READ_ATTRIBUTE pReadAttribute;
PNDS_ATTRIBUTE pAttribute;
PBYTE pAttribValue;
BOOL fContinue = TRUE;
//
// Read the User property
//
szBuffer = (PBYTE)malloc(dwBufferSize);
if ( !szBuffer ) {
DisplayMessage(IDR_NOT_ENOUGH_MEMORY);
Status = STATUS_NO_MEMORY;
return Status;
}
memset( szBuffer, 0, dwBufferSize );
oemStr.Length = (USHORT) strlen( Property );
oemStr.MaximumLength = oemStr.Length;
oemStr.Buffer = Property;
PropertyName.Length = 0;
PropertyName.MaximumLength = sizeof(NdsStr);
PropertyName.Buffer = NdsStr;
RtlOemStringToUnicodeString( &PropertyName, &oemStr, FALSE );
while ( fContinue )
{
Status = NwNdsReadAttribute ( GhRdrForUser,
GUserObjectID,
&iterhandle,
&PropertyName,
szBuffer,
dwBufferSize );
if ( NT_SUCCESS(Status) && iterhandle != INITIAL_ITERATION )
{
dwBufferSize *= 2;
free( szBuffer );
szBuffer = (PBYTE)malloc(dwBufferSize);
if ( !szBuffer ) {
DisplayMessage(IDR_NOT_ENOUGH_MEMORY);
Status = STATUS_NO_MEMORY;
return Status;
}
memset( szBuffer, 0, dwBufferSize );
iterhandle = INITIAL_ITERATION;
}
else
{
fContinue = FALSE;
}
}
if ( !NT_SUCCESS(Status) )
{
if ( szBuffer )
free( szBuffer );
return Status;
}
if ( NT_SUCCESS(Status) )
{
int i;
pReadAttribute = (PNDS_RESPONSE_READ_ATTRIBUTE)szBuffer;
pAttribute = (PNDS_ATTRIBUTE)(szBuffer
+ sizeof(NDS_RESPONSE_READ_ATTRIBUTE));
if ( pSyntaxID )
{
*pSyntaxID = pAttribute->SyntaxID;
}
pAttribValue = (PBYTE)(pAttribute->AttribName) +
ROUNDUP4(pAttribute->AttribNameLength) +
sizeof(DWORD);
if ( pActualSize )
{
*pActualSize = *(DWORD *)pAttribValue;
}
memcpy( Data, pAttribValue + sizeof(DWORD),
min(*(DWORD *)pAttribValue, Size) );
}
return Status;
}
/********************************************************************
NDSGetVar
Routine Description:
Return value of user property
Get the syntax type of the property
Retrieve the data
Do any data conversion
Arguments:
Name - of NDS property IN
Value - value buffer OUT
Size - size of value buffer IN
Return Value:
none
*******************************************************************/
void
NDSGetVar ( PBYTE Name, PBYTE Value, unsigned int Size)
{
unsigned int err;
SYNTAX Syntax;
BYTE Buffer[ATTRBUFSIZE];
DWORD ActualSize;
Value[0] = 0;
err = NDSGetUserProperty( Name, Buffer, ATTRBUFSIZE, &Syntax, &ActualSize );
if ( err )
{
return;
}
switch ( Syntax )
{
case NDSI_BOOLEAN:
if ( *(PBYTE)Buffer )
{
strcpy( Value, "Y" );
}
else
{
strcpy( Value, "N" );
}
break;
case NDSI_DIST_NAME:
case NDSI_CE_STRING:
case NDSI_CI_STRING:
case NDSI_OCTET_STRING:
case NDSI_PR_STRING:
case NDSI_NU_STRING:
case NDSI_TEL_NUMBER:
case NDSI_CLASS_NAME:
ConvertUnicodeToAscii( Buffer );
if ( Syntax == NDSI_DIST_NAME )
NDSAbbreviateName(FLAGS_LOCAL_CONTEXT, Buffer, Buffer);
strncpy( Value, Buffer, Size );
break;
case NDSI_CI_LIST:
ConvertUnicodeToAscii( Buffer+8 );
strncpy( Value, Buffer+8, Size );
break;
break;
case NDSI_INTEGER:
case NDSI_COUNTER:
case NDSI_TIME:
case NDSI_INTERVAL:
case NDSI_TIMESTAMP:
sprintf( Value, "%d", *(int *)Buffer );
break;
case NDSI_PO_ADDRESS:
{
// 6 null terminated lines
int line,len;
PBYTE ptr = Buffer + 4;
// Stop if not 6 lines
if ( *(int *)Buffer != 6 )
break;
for (line = 0; line <= 5; line++) {
len = ROUNDUP4(*(int *)ptr);
ptr += 4;
if ( !len )
break;
ConvertUnicodeToAscii( ptr );
strcat( Value, ptr );
strcat( Value, "\n" );
ptr += len;
}
}
break;
case NDSI_FAX_NUMBER:
if ( *(int *)Buffer == 0 )
return;
ConvertUnicodeToAscii( Buffer+4 );
strncpy( Value, Buffer+4, Size );
break;
case NDSI_EMAIL_ADDRESS:
if ( *(int *)(Buffer+4) == 0 )
return;
ConvertUnicodeToAscii( Buffer+8 );
strncpy( Value, Buffer+8, Size );
break;
case NDSI_PATH:
{
int len;
len = *(int *)(Buffer+4);
if ( len == 0 )
break;
len = ROUNDUP4( len );
ConvertUnicodeToAscii( Buffer+8 );
strcpy( Value, Buffer+8 );
NDSAbbreviateName(FLAGS_LOCAL_CONTEXT, Value, Value);
strcat( Value, ":" );
if ( *(int *)(Buffer + 8 + len) == 0 )
break;
ConvertUnicodeToAscii( Buffer+8+len+4 );
strcat( Value, Buffer+8+len+4 );
break;
}
case NDSI_NET_ADDRESS:
case NDSI_OCTET_LIST:
case NDSI_OBJECT_ACL:
case NDSI_STREAM:
case NDSI_UNKNOWN:
case NDSI_REPLICA_POINTER:
case NDSI_BACK_LINK:
case NDSI_TYPED_NAME:
case NDSI_HOLD:
case NDSI_TAX_COUNT:
default:
Value[0] = '\0';
Value[1] = '\0';
break;
}
}
/********************************************************************
NDSChangeContext
Routine Description:
Change the current context
Arguments:
Context - context string IN
Return Value:
error number
*******************************************************************/
unsigned int
NDSChangeContext( PBYTE Context )
{
NTSTATUS Status;
OEM_STRING oemStr;
UNICODE_STRING defaultcontext;
WCHAR NdsStr[1024];
oemStr.Length = (USHORT)strlen( Context );
oemStr.MaximumLength = oemStr.Length;
oemStr.Buffer = Context;
defaultcontext.Length = 0;
defaultcontext.MaximumLength = sizeof(NdsStr);
defaultcontext.Buffer = NdsStr;
RtlOemStringToUnicodeString( &defaultcontext, &oemStr, FALSE );
Status = NwNdsSetTreeContext( GhRdr, &NDSTREE_u, &defaultcontext );
return Status;
}
/********************************************************************
NDSGetContext
Routine Description:
Retrieve the current context
Arguments:
Buffer - data buffer for context string OUT
len - length of data buffer IN
Return Value:
error number
*******************************************************************/
unsigned int
NDSGetContext( PBYTE Buffer,
unsigned int len )
{
NTSTATUS Status;
Status = NDSGetNameContext( Buffer, TRUE );
if ( !NT_SUCCESS( Status ) )
return Status;
NDSAbbreviateName(FLAGS_NO_CONTEXT, Buffer, Buffer);
return 0;
}
/********************************************************************
NDSfopenStream
Routine Description:
Open a file handle to an NDS stream property
Arguments:
Object - name of object IN
Property - name of property IN
pStream - pointer to file handle OUT
pFileSize - pointer to file size OUT
Return Value:
error
*******************************************************************/
unsigned int
NDSfopenStream ( PBYTE Object,
PBYTE Property,
PHANDLE pStream,
unsigned int * pFileSize )
{
//
// Status variables.
//
NTSTATUS Status;
int ccode = -1;
//
// For NwNdsOpenTreeHandle.
//
HANDLE hRdr;
OEM_STRING oemStr;
UNICODE_STRING ObjectName;
WCHAR NdsStr[1024];
//
// For NwNdsResolveName.
//
DWORD dwOid;
UNICODE_STRING ReferredServer;
WCHAR ServerStr[MAX_NAME_LEN];
DWORD dwHandleType;
HANDLE hReferredServer;
//
// Get a handle to the redirector.
//
Status = NwNdsOpenTreeHandle( &NDSTREE_u, &hRdr );
if ( !NT_SUCCESS( Status ) ) {
DisplayMessage(IDR_TREE_OPEN_FAILED);
return ccode;
}
//
// Resolve the name that we have to an object id.
//
if ( !Object )
{
return 1;
}
oemStr.Length = (USHORT)strlen( Object );
oemStr.MaximumLength = oemStr.Length;
oemStr.Buffer = Object;
ObjectName.Length = 0;
ObjectName.MaximumLength = sizeof(NdsStr);
ObjectName.Buffer = NdsStr;
RtlOemStringToUnicodeString( &ObjectName, &oemStr, FALSE );
ReferredServer.Buffer = ServerStr;
ReferredServer.Length = 0;
ReferredServer.MaximumLength = sizeof( ServerStr );
Status = NwNdsResolveName ( hRdr,
&ObjectName,
&dwOid,
&ReferredServer,
NULL,
0 );
if ( !NT_SUCCESS( Status ) ) {
return 0xffffffff;
}
if ( ReferredServer.Length > 0 ) {
//
// We've been referred to another server, so we
// must jump to that server before continuing.
//
Status = NwNdsOpenGenericHandle( &ReferredServer,
&dwHandleType,
&hReferredServer );
if ( !NT_SUCCESS( Status ) ) {
return 0xffffffff;
}
CloseHandle( hRdr );
hRdr = hReferredServer;
}
//
// Open the file stream.
//
oemStr.Length = (USHORT)strlen( Property );
oemStr.MaximumLength = oemStr.Length;
oemStr.Buffer = Property;
ObjectName.Length = 0;
ObjectName.MaximumLength = sizeof(NdsStr);
ObjectName.Buffer = NdsStr;
RtlOemStringToUnicodeString( &ObjectName, &oemStr, FALSE );
//
// Try to open a file stream for read access.
//
Status = NwNdsOpenStream( hRdr,
dwOid,
&ObjectName,
1, // Read access.
pFileSize );
if ( !NT_SUCCESS( Status ) ) {
return 0xffffffff;
}
*pStream = hRdr;
return 0;
}
/*
* IsMemberOfNDSGroup
* ------------------
*
* Returns true if currently logged in user object is member of group with given name
*
*/
unsigned int
IsMemberOfNDSGroup(
PBYTE nwGroup
)
{
NTSTATUS Status = STATUS_SUCCESS;
UINT nwRet;
BYTE szCanonTargetGroupName[NDS_NAME_CHARS+1];
UINT syntaxid;
UINT actualsize;
PBYTE szBuffer;
LPSTR pProp;
UINT i;
DWORD iterhandle = INITIAL_ITERATION;
DWORD dwBufferSize = ATTRBUFSIZE;
UINT fFoundGroup = FALSE;
PNDS_RESPONSE_READ_ATTRIBUTE pReadAttribute;
PNDS_ATTRIBUTE pAttribute;
PBYTE pAttribValue;
UNICODE_STRING PropertyName;
UINT numvalues = 0;
BOOL fContinue = TRUE;
szBuffer = (PBYTE)malloc(dwBufferSize);
if ( !szBuffer ) {
DisplayMessage(IDR_NOT_ENOUGH_MEMORY);
return FALSE;
}
memset( szBuffer, 0, dwBufferSize );
// Canonicalize name according to current context
strcpy( szCanonTargetGroupName, nwGroup );
nwRet = NDSCanonicalizeName( szCanonTargetGroupName,
szCanonTargetGroupName,
NDS_NAME_CHARS,
TRUE );
if (nwRet) {
if ( nwGroup[0] != '.' ) {
// Try an absolute name
strcpy( szCanonTargetGroupName, "." );
strcat( szCanonTargetGroupName, nwGroup );
nwRet = NDSCanonicalizeName( szCanonTargetGroupName,
szCanonTargetGroupName,
NDS_NAME_CHARS,
TRUE );
}
if ( nwRet )
goto CleanRet;
}
// Should check class name of object
RtlInitUnicodeString( &PropertyName, L"Group Membership" );
while ( fContinue )
{
Status = NwNdsReadAttribute ( GhRdrForUser,
GUserObjectID,
&iterhandle,
&PropertyName,
szBuffer,
dwBufferSize );
if ( NT_SUCCESS(Status) && iterhandle != INITIAL_ITERATION )
{
dwBufferSize *= 2;
free( szBuffer );
szBuffer = (PBYTE)malloc(dwBufferSize);
if ( !szBuffer ) {
DisplayMessage(IDR_NOT_ENOUGH_MEMORY);
return FALSE;
}
memset( szBuffer, 0, dwBufferSize );
iterhandle = INITIAL_ITERATION;
}
else
{
fContinue = FALSE;
}
}
if ( !NT_SUCCESS(Status) )
{
goto CleanRet;
}
pReadAttribute = (PNDS_RESPONSE_READ_ATTRIBUTE)szBuffer;
pAttribute = (PNDS_ATTRIBUTE)(szBuffer
+ sizeof(NDS_RESPONSE_READ_ATTRIBUTE));
pAttribute->SyntaxID;
pAttribValue = (PBYTE)(pAttribute->AttribName) +
ROUNDUP4(pAttribute->AttribNameLength) +
sizeof(DWORD);
numvalues = *(PUINT)((PBYTE)(pAttribute->AttribName) +
ROUNDUP4(pAttribute->AttribNameLength));
if ( *(DWORD *)pAttribValue == 0 )
{
goto CleanRet;
}
for ( i = 0; i < numvalues; i++ ) {
ConvertUnicodeToAscii( pAttribValue+sizeof(DWORD) );
if (!_stricmp(pAttribValue+sizeof(DWORD),szCanonTargetGroupName)) {
fFoundGroup = TRUE;
break;
}
pAttribValue += ROUNDUP4(*(PUINT)pAttribValue) + sizeof(DWORD);
}
CleanRet:
if (szBuffer ) {
free (szBuffer);
}
return fFoundGroup;
}
/********************************************************************
NDSGetProperty
Routine Description:
Return the NDS property for the object
Arguments:
Object - name of object IN
Property - property name IN
Data - data buffer OUT
Size - size of data buffer IN
pActualSize - real data size OUT
Return Value:
error
*******************************************************************/
unsigned int
NDSGetProperty ( PBYTE Object,
PBYTE Property,
PBYTE Data,
unsigned int Size,
unsigned int * pActualSize )
{
//
// Status variables.
//
NTSTATUS Status = STATUS_SUCCESS;
int ccode = -1;
//
// For NwNdsOpenTreeHandle.
//
HANDLE hRdr;
OEM_STRING oemStr;
UNICODE_STRING ObjectName;
WCHAR NdsStr[1024];
//
// For NwNdsResolveName.
//
DWORD dwOid;
UNICODE_STRING ReferredServer;
WCHAR ServerStr[MAX_NAME_LEN];
DWORD dwHandleType;
HANDLE hReferredServer;
//
// For NwNdsReadAttribute
//
PBYTE szBuffer;
DWORD dwBufferSize = 2048;
DWORD iterhandle = INITIAL_ITERATION;
PNDS_RESPONSE_READ_ATTRIBUTE pReadAttribute;
PNDS_ATTRIBUTE pAttribute;
PBYTE pAttribValue;
BOOL fContinue = TRUE;
//
// Allocate a buffer for the NDS request.
//
szBuffer = (PBYTE)malloc(dwBufferSize);
if ( !szBuffer ) {
DisplayMessage(IDR_NOT_ENOUGH_MEMORY);
Status = STATUS_NO_MEMORY;
return Status;
}
memset( szBuffer, 0, dwBufferSize );
//
// Get a handle to the redirector.
//
Status = NwNdsOpenTreeHandle( &NDSTREE_u, &hRdr );
if ( !NT_SUCCESS( Status ) ) {
DisplayMessage(IDR_TREE_OPEN_FAILED);
return ccode;
}
//
// Resolve the name that we have to an object id.
//
if ( !Object )
{
return 1;
}
oemStr.Length = (USHORT)strlen( Object );
oemStr.MaximumLength = oemStr.Length;
oemStr.Buffer = Object;
ObjectName.Length = 0;
ObjectName.MaximumLength = sizeof(NdsStr);
ObjectName.Buffer = NdsStr;
RtlOemStringToUnicodeString( &ObjectName, &oemStr, FALSE );
ReferredServer.Buffer = ServerStr;
ReferredServer.Length = 0;
ReferredServer.MaximumLength = sizeof( ServerStr );
Status = NwNdsResolveName ( hRdr,
&ObjectName,
&dwOid,
&ReferredServer,
NULL,
0 );
if ( !NT_SUCCESS( Status ) ) {
return 0xffffffff;
}
if ( ReferredServer.Length > 0 ) {
//
// We've been referred to another server, so we
// must jump to that server before continuing.
//
Status = NwNdsOpenGenericHandle( &ReferredServer,
&dwHandleType,
&hReferredServer );
if ( !NT_SUCCESS( Status ) ) {
return 0xffffffff;
}
CloseHandle( hRdr );
hRdr = hReferredServer;
}
//
// Get the attribute
//
oemStr.Length = (USHORT)strlen( Property );
oemStr.MaximumLength = oemStr.Length;
oemStr.Buffer = Property;
ObjectName.Length = 0;
ObjectName.MaximumLength = sizeof(NdsStr);
ObjectName.Buffer = NdsStr;
RtlOemStringToUnicodeString( &ObjectName, &oemStr, FALSE );
while ( fContinue )
{
Status = NwNdsReadAttribute ( hRdr,
dwOid,
&iterhandle,
&ObjectName,
szBuffer,
dwBufferSize );
if ( NT_SUCCESS(Status) && iterhandle != INITIAL_ITERATION )
{
dwBufferSize *= 2;
free( szBuffer );
szBuffer = (PBYTE)malloc(dwBufferSize);
if ( !szBuffer ) {
DisplayMessage(IDR_NOT_ENOUGH_MEMORY);
Status = STATUS_NO_MEMORY;
return Status;
}
memset( szBuffer, 0, dwBufferSize );
iterhandle = INITIAL_ITERATION;
}
else
{
fContinue = FALSE;
}
}
if ( !NT_SUCCESS(Status) )
{
NtClose( hRdr );
free( szBuffer );
return Status;
}
if ( NT_SUCCESS(Status) )
{
int i;
pReadAttribute = (PNDS_RESPONSE_READ_ATTRIBUTE)szBuffer;
pAttribute = (PNDS_ATTRIBUTE)(szBuffer
+ sizeof(NDS_RESPONSE_READ_ATTRIBUTE));
pAttribValue = (PBYTE)(pAttribute->AttribName) +
ROUNDUP4(pAttribute->AttribNameLength) +
sizeof(DWORD);
if ( pActualSize )
{
*pActualSize = *(DWORD *)pAttribValue;
}
memcpy( Data, pAttribValue + sizeof(DWORD),
min(*(DWORD *)pAttribValue, Size) );
}
NtClose( hRdr );
return Status;
}
/********************************************************************
NDSCleanup
Routine Description:
Does any NDS cleanup
Arguments:
none
Return Value:
none
*******************************************************************/
void
NDSCleanup ( void )
{
NtClose( GhRdr );
if( GhRdr != GhRdrForUser ) {
NtClose( GhRdrForUser );
}
}
/********************************************************************
NDSGetClassName
Routine Description:
return a class name for an object
Arguments:
szObjectName
ClassName
Return Value:
none
*******************************************************************/
unsigned int
NDSGetClassName( LPSTR szObjectName, LPSTR ClassName )
{
NTSTATUS Status;
int ccode = -1;
DWORD ThisObjectID;
OEM_STRING oemStr;
UNICODE_STRING ObjectName;
BYTE Buffer[2048];
BYTE FullName[NDS_NAME_CHARS];
PBYTE ptr;
UNICODE_STRING ReferredServer;
WCHAR ServerStr[MAX_NAME_LEN];
DWORD dwHandleType;
HANDLE hReferredServer;
DWORD Length;
//
// Resolve the name that we have to an object id.
//
oemStr.Length = (USHORT)strlen( szObjectName );
oemStr.MaximumLength = oemStr.Length;
oemStr.Buffer = szObjectName;
ObjectName.Length = 0;
ObjectName.MaximumLength = sizeof(Buffer);
ObjectName.Buffer = (WCHAR *)Buffer;
RtlOemStringToUnicodeString( &ObjectName, &oemStr, FALSE );
ReferredServer.Buffer = ServerStr;
ReferredServer.Length = 0;
ReferredServer.MaximumLength = sizeof( ServerStr );
Status = NwNdsResolveName ( GhRdr,
&ObjectName,
&ThisObjectID,
&ReferredServer,
NULL,
0 );
if ( !NT_SUCCESS( Status ) ) {
return Status;
}
if ( ReferredServer.Length > 0 ) {
//
// We've been referred to another server, so we
// should change the global handle.
//
Status = NwNdsOpenGenericHandle( &ReferredServer,
&dwHandleType,
&hReferredServer );
if ( !NT_SUCCESS( Status ) ) {
return Status;
}
if( GhRdr != GhRdrForUser ) {
CloseHandle( GhRdr );
}
GhRdr = hReferredServer;
}
Status = NwNdsReadObjectInfo( GhRdr, ThisObjectID, Buffer, 2048 );
if ( !NT_SUCCESS( Status ) ) {
return Status;
}
ptr = Buffer + sizeof( NDS_RESPONSE_GET_OBJECT_INFO ) + sizeof( DWORD );
RtlInitUnicodeString( &ObjectName, (PWCHAR)ptr );
oemStr.Length = 0;
oemStr.MaximumLength = NDS_NAME_CHARS;
oemStr.Buffer = ClassName;
RtlUnicodeStringToOemString( &oemStr, &ObjectName, FALSE );
return 0;
}