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.
529 lines
16 KiB
529 lines
16 KiB
/*++
|
|
Copyright (c) 1987-1999 Microsoft Corporation
|
|
|
|
Module Name:
|
|
|
|
3connect.c
|
|
|
|
Abstract:
|
|
|
|
This module implements the tree connect SMB related routines. It also implements the
|
|
three flavours of this routine ( user level and share level non NT server tree connect
|
|
SMB construction and the tree connect SMB construction for SMB servers)
|
|
|
|
--*/
|
|
|
|
#include "precomp.h"
|
|
#pragma hdrstop
|
|
|
|
//
|
|
// The order of these names should match the order in which the enumerated type
|
|
// NET_ROOT_TYPE is defined. This facilitates easy access of share type names
|
|
//
|
|
|
|
#ifdef ALLOC_PRAGMA
|
|
#pragma alloc_text(PAGE, BuildCanonicalNetRootInformation)
|
|
#pragma alloc_text(PAGE, CoreBuildTreeConnectSmb)
|
|
#pragma alloc_text(PAGE, LmBuildTreeConnectSmb)
|
|
#pragma alloc_text(PAGE, NtBuildTreeConnectSmb)
|
|
#endif
|
|
|
|
PCHAR s_NetRootTypeName[] = {
|
|
SHARE_TYPE_NAME_DISK,
|
|
SHARE_TYPE_NAME_PIPE,
|
|
SHARE_TYPE_NAME_COMM,
|
|
SHARE_TYPE_NAME_PRINT,
|
|
SHARE_TYPE_NAME_WILD
|
|
};
|
|
|
|
extern NTSTATUS
|
|
BuildTreeConnectSecurityInformation(
|
|
PSMB_EXCHANGE pExchange,
|
|
PBYTE pBuffer,
|
|
PBYTE pPasswordLength,
|
|
PULONG pSmbBufferSize);
|
|
|
|
NTSTATUS
|
|
BuildCanonicalNetRootInformation(
|
|
PUNICODE_STRING pServerName,
|
|
PUNICODE_STRING pNetRootName,
|
|
NET_ROOT_TYPE NetRootType,
|
|
BOOLEAN fUnicode,
|
|
BOOLEAN fPostPendServiceString,
|
|
PBYTE *pBufferPointer,
|
|
PULONG pBufferSize)
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
This routine builds the desired net root information for a tree connect SMB
|
|
|
|
Arguments:
|
|
|
|
pServerName - the server name
|
|
|
|
pNetRootName - the net root name
|
|
|
|
NetRootType - the net root type ( print,pipe,disk etc.,)
|
|
|
|
fUnicode - TRUE if it is to be built in UNICODE
|
|
|
|
pBufferPointer - the SMB buffer
|
|
|
|
pBufferSize - the size on input. modified to the remaining size on output
|
|
|
|
Return Value:
|
|
|
|
RXSTATUS - The return status for the operation
|
|
|
|
Notes:
|
|
|
|
This routine relies upon the names being in certain formats to ensure that a
|
|
valid UNC name can be formulated.
|
|
1) The RDBSS netroot names start with a \ and also include the server name as
|
|
part of the net root name. This is mandated by the prefix table search requirements
|
|
in RDBSS.
|
|
|
|
--*/
|
|
{
|
|
NTSTATUS Status;
|
|
|
|
PAGED_CODE();
|
|
|
|
if (fUnicode) {
|
|
// Align the buffer and adjust the size accordingly.
|
|
PBYTE pBuffer = *pBufferPointer;
|
|
RxDbgTrace( 0, (DEBUG_TRACE_CREATE),
|
|
("BuildCanonicalNetRootInformation -- tcstring as unicode %wZ\n", pNetRootName));
|
|
pBuffer = ALIGN_SMB_WSTR(pBuffer);
|
|
|
|
if(sizeof(WCHAR) + (ULONG)(pBuffer - *pBufferPointer) > *pBufferSize) {
|
|
return STATUS_BUFFER_OVERFLOW;
|
|
}
|
|
*pBufferSize -= (ULONG)(pBuffer - *pBufferPointer) + sizeof(WCHAR);
|
|
|
|
*((PWCHAR)pBuffer) = L'\\';
|
|
pBuffer += sizeof(WCHAR);
|
|
*pBufferPointer = pBuffer;
|
|
Status = SmbPutUnicodeStringAndUpcase(pBufferPointer,pNetRootName,pBufferSize);
|
|
|
|
} else {
|
|
RxDbgTrace( 0, (DEBUG_TRACE_CREATE), ("BuildCanonicalNetRootInformation -- tcstring as ascii\n"));
|
|
if(sizeof(CHAR) > *pBufferSize) {
|
|
return STATUS_BUFFER_OVERFLOW;
|
|
}
|
|
*((PCHAR)*pBufferPointer) = '\\';
|
|
*pBufferPointer += sizeof(CHAR);
|
|
*pBufferSize -= sizeof(CHAR);
|
|
Status = SmbPutUnicodeStringAsOemStringAndUpcase(pBufferPointer,pNetRootName,pBufferSize);
|
|
}
|
|
|
|
if (NT_SUCCESS(Status) && fPostPendServiceString) {
|
|
// Put the desired service name in ASCII ( always )
|
|
ULONG Length = strlen(s_NetRootTypeName[NetRootType]) + 1;
|
|
if (*pBufferSize >= Length) {
|
|
RtlCopyMemory(*pBufferPointer,s_NetRootTypeName[NetRootType],Length);
|
|
*pBufferSize -= Length;
|
|
} else {
|
|
Status = STATUS_BUFFER_OVERFLOW;
|
|
}
|
|
}
|
|
|
|
return Status;
|
|
}
|
|
|
|
|
|
NTSTATUS
|
|
CoreBuildTreeConnectSmb(
|
|
PSMB_EXCHANGE pExchange,
|
|
PGENERIC_ANDX pAndXSmb,
|
|
PULONG pAndXSmbBufferSize)
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
This routine builds the tree connect SMB for a pre NT server
|
|
|
|
Arguments:
|
|
|
|
pExchange - the exchange instance
|
|
|
|
pAndXSmb - the tree connect to be filled in...it's not really a andX
|
|
|
|
pAndXSmbBufferSize - the SMB buffer size on input modified to remaining size on
|
|
output.
|
|
|
|
Return Value:
|
|
|
|
RXSTATUS - The return status for the operation
|
|
|
|
--*/
|
|
{
|
|
NTSTATUS Status;
|
|
USHORT PasswordLength;
|
|
|
|
PMRX_NET_ROOT NetRoot;
|
|
|
|
UNICODE_STRING ServerName;
|
|
UNICODE_STRING NetRootName;
|
|
|
|
PSMBCE_SERVER pServer;
|
|
|
|
PREQ_TREE_CONNECT pTreeConnect = (PREQ_TREE_CONNECT)pAndXSmb;
|
|
|
|
ULONG OriginalBufferSize = *pAndXSmbBufferSize;
|
|
|
|
BOOLEAN AppendServiceString;
|
|
PBYTE pBuffer;
|
|
PCHAR ServiceName;
|
|
ULONG Length;
|
|
|
|
PAGED_CODE();
|
|
|
|
NetRoot = pExchange->SmbCeContext.pVNetRoot->pNetRoot;
|
|
|
|
RxDbgTrace( 0, (DEBUG_TRACE_ALWAYS),
|
|
("CoreBuildTreeConnectSmb buffer,remptr %08lx %08lx, nrt=%08lx\n",
|
|
pAndXSmb,
|
|
pAndXSmbBufferSize,
|
|
NetRoot->Type));
|
|
|
|
pServer = SmbCeGetExchangeServer(pExchange);
|
|
SmbCeGetServerName(NetRoot->pSrvCall,&ServerName);
|
|
SmbCeGetNetRootName(NetRoot,&NetRootName);
|
|
ServiceName = s_NetRootTypeName[NetRoot->Type];
|
|
Length = strlen(ServiceName) + 1;
|
|
|
|
pTreeConnect->WordCount = 0;
|
|
AppendServiceString = FALSE;
|
|
|
|
if(*pAndXSmbBufferSize < FIELD_OFFSET(REQ_TREE_CONNECT,Buffer)+1) {
|
|
return STATUS_BUFFER_OVERFLOW;
|
|
}
|
|
|
|
pBuffer = (PBYTE)pTreeConnect + FIELD_OFFSET(REQ_TREE_CONNECT,Buffer);
|
|
*pBuffer = 0x04;
|
|
pBuffer++;
|
|
*pAndXSmbBufferSize -= (FIELD_OFFSET(REQ_TREE_CONNECT,Buffer)+1);
|
|
|
|
// put in the netname
|
|
|
|
//RxDbgTrace( 0, (DEBUG_TRACE_ALWAYS), ("CoreBuildTreeConnectSmb before bcnri buffer,rem %08lx %08lx\n",pBuffer,*pAndXSmbBufferSize));
|
|
Status = BuildCanonicalNetRootInformation(
|
|
&ServerName,
|
|
&NetRootName,
|
|
pExchange->SmbCeContext.pVNetRoot->pNetRoot->Type,
|
|
(BOOLEAN)(pServer->Dialect >= NTLANMAN_DIALECT),
|
|
AppendServiceString,
|
|
&pBuffer,
|
|
pAndXSmbBufferSize);
|
|
|
|
if (!NT_SUCCESS(Status))
|
|
return Status;
|
|
|
|
if(*pAndXSmbBufferSize < 1) {
|
|
return STATUS_BUFFER_OVERFLOW;
|
|
}
|
|
|
|
// put in the password
|
|
pBuffer = (PBYTE)pTreeConnect + OriginalBufferSize - *pAndXSmbBufferSize;
|
|
//RxDbgTrace( 0, (DEBUG_TRACE_ALWAYS), ("CoreBuildTreeConnectSmb88 buffer,rem %08lx %08lx\n",pBuffer,*pAndXSmbBufferSize));
|
|
|
|
*pBuffer = 0x04;
|
|
pBuffer++;
|
|
*pAndXSmbBufferSize -= 1;
|
|
|
|
if (pServer->SecurityMode == SECURITY_MODE_SHARE_LEVEL) {
|
|
// The password information needs to be sent as part of the tree connect
|
|
// SMB for share level servers.
|
|
|
|
//RxDbgTrace( 0, (DEBUG_TRACE_ALWAYS), ("CoreBuildTreeConnectSmb before btcsi buffer,rem %08lx %08lx\n",pBuffer,*pAndXSmbBufferSize));
|
|
Status = BuildTreeConnectSecurityInformation(
|
|
pExchange,
|
|
pBuffer,
|
|
(PBYTE)&PasswordLength,
|
|
pAndXSmbBufferSize);
|
|
}
|
|
|
|
if (!NT_SUCCESS(Status))
|
|
return Status;
|
|
|
|
if(*pAndXSmbBufferSize < 1) {
|
|
return STATUS_BUFFER_OVERFLOW;
|
|
}
|
|
// string in the service string based on the netroot type
|
|
|
|
//RxDbgTrace( 0, (DEBUG_TRACE_ALWAYS), ("CoreBuildTreeConnectSmb beforesscopy buffer,rem %08lx %08lx\n",pBuffer,*pAndXSmbBufferSize));
|
|
|
|
|
|
pBuffer = (PBYTE)pTreeConnect + OriginalBufferSize - *pAndXSmbBufferSize;
|
|
*pBuffer = 0x04;
|
|
pBuffer++;
|
|
*pAndXSmbBufferSize -= 1;
|
|
|
|
if (*pAndXSmbBufferSize >= Length) {
|
|
RtlCopyMemory(pBuffer,ServiceName,Length);
|
|
*pAndXSmbBufferSize -= Length;
|
|
} else {
|
|
Status = STATUS_BUFFER_OVERFLOW;
|
|
}
|
|
|
|
//RxDbgTrace( 0, (DEBUG_TRACE_ALWAYS), ("CoreBuildTreeConnectSmb beforesscopy buffer,rem %08lx %08lx\n",pBuffer,*pAndXSmbBufferSize));
|
|
SmbPutUshort(
|
|
&pTreeConnect->ByteCount,
|
|
(USHORT)(OriginalBufferSize
|
|
- *pAndXSmbBufferSize
|
|
- FIELD_OFFSET(REQ_TREE_CONNECT,Buffer)
|
|
)
|
|
);
|
|
|
|
RxDbgTrace( 0, (DEBUG_TRACE_ALWAYS), ("CoreBuildTreeConnectSmb end buffer,rem %08lx %08lx\n",pBuffer,*pAndXSmbBufferSize));
|
|
return Status;
|
|
}
|
|
|
|
|
|
NTSTATUS
|
|
LmBuildTreeConnectSmb(
|
|
PSMB_EXCHANGE pExchange,
|
|
PGENERIC_ANDX pAndXSmb,
|
|
PULONG pAndXSmbBufferSize)
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
This routine builds the tree connect SMB for a pre NT server
|
|
|
|
Arguments:
|
|
|
|
pExchange - the exchange instance
|
|
|
|
pAndXSmb - the tree connect to be filled in...it's not really a andX
|
|
|
|
pAndXSmbBufferSize - the SMB buffer size on input modified to remaining size on
|
|
output.
|
|
|
|
Return Value:
|
|
|
|
RXSTATUS - The return status for the operation
|
|
|
|
--*/
|
|
{
|
|
NTSTATUS Status;
|
|
USHORT PasswordLength;
|
|
|
|
PMRX_NET_ROOT NetRoot;
|
|
|
|
UNICODE_STRING ServerName;
|
|
UNICODE_STRING NetRootName;
|
|
|
|
PSMBCE_SERVER pServer;
|
|
|
|
PREQ_TREE_CONNECT_ANDX pTreeConnectAndX = (PREQ_TREE_CONNECT_ANDX)pAndXSmb;
|
|
|
|
ULONG OriginalBufferSize = *pAndXSmbBufferSize;
|
|
|
|
BOOLEAN AppendServiceString;
|
|
PBYTE pBuffer;
|
|
PCHAR ServiceName;
|
|
ULONG Length;
|
|
|
|
PAGED_CODE();
|
|
|
|
NetRoot = pExchange->SmbCeContext.pVNetRoot->pNetRoot;
|
|
RxDbgTrace( 0, (DEBUG_TRACE_ALWAYS),
|
|
("LmBuildTreeConnectSmb buffer,remptr %08lx %08lx, nrt=%08lx\n",
|
|
pAndXSmb,
|
|
pAndXSmbBufferSize,
|
|
NetRoot->Type));
|
|
|
|
if(OriginalBufferSize < FIELD_OFFSET(REQ_TREE_CONNECT_ANDX,Buffer)) {
|
|
return STATUS_BUFFER_OVERFLOW;
|
|
}
|
|
|
|
pServer = SmbCeGetExchangeServer(pExchange);
|
|
SmbCeGetServerName(NetRoot->pSrvCall,&ServerName);
|
|
SmbCeGetNetRootName(NetRoot,&NetRootName);
|
|
ServiceName = s_NetRootTypeName[NetRoot->Type];
|
|
Length = strlen(ServiceName) + 1;
|
|
|
|
AppendServiceString = TRUE;
|
|
pTreeConnectAndX->WordCount = 4;
|
|
SmbPutUshort(&pTreeConnectAndX->AndXReserved,0);
|
|
SmbPutUshort(
|
|
&pTreeConnectAndX->Flags,0);
|
|
pBuffer = (PBYTE)pTreeConnectAndX + FIELD_OFFSET(REQ_TREE_CONNECT_ANDX,Buffer);
|
|
*pAndXSmbBufferSize -= (FIELD_OFFSET(REQ_TREE_CONNECT_ANDX,Buffer));
|
|
|
|
if (pServer->SecurityMode == SECURITY_MODE_SHARE_LEVEL) {
|
|
// The password information needs to be sent as part of the tree connect
|
|
// SMB for share level servers.
|
|
|
|
//RxDbgTrace( 0, (DEBUG_TRACE_ALWAYS), ("LmBuildTreeConnectSmb before btcsi buffer,rem %08lx %08lx\n",pBuffer,*pAndXSmbBufferSize));
|
|
Status = BuildTreeConnectSecurityInformation(
|
|
pExchange,
|
|
pBuffer,
|
|
(PBYTE)&PasswordLength,
|
|
pAndXSmbBufferSize);
|
|
|
|
if (Status == STATUS_SUCCESS) {
|
|
pBuffer += PasswordLength;
|
|
SmbPutUshort(&pTreeConnectAndX->PasswordLength,PasswordLength);
|
|
}
|
|
} else {
|
|
|
|
if(*pAndXSmbBufferSize < sizeof(CHAR)) {
|
|
return STATUS_BUFFER_OVERFLOW;
|
|
}
|
|
|
|
// No password is required for user level security servers as part of tree
|
|
// connect
|
|
SmbPutUshort(&pTreeConnectAndX->PasswordLength,0x1);
|
|
*((PCHAR)pBuffer) = '\0';
|
|
pBuffer += sizeof(CHAR);
|
|
*pAndXSmbBufferSize -= sizeof(CHAR);
|
|
Status = STATUS_SUCCESS;
|
|
}
|
|
|
|
if (Status == STATUS_SUCCESS) {
|
|
Status = BuildCanonicalNetRootInformation(
|
|
&ServerName,
|
|
&NetRootName,
|
|
pExchange->SmbCeContext.pVNetRoot->pNetRoot->Type,
|
|
(BOOLEAN)(pServer->Dialect >= NTLANMAN_DIALECT),
|
|
AppendServiceString,
|
|
&pBuffer,
|
|
pAndXSmbBufferSize);
|
|
|
|
//RxDbgTrace( 0, (DEBUG_TRACE_ALWAYS), ("LmBuildTreeConnectSmb beforesscopy buffer,rem %08lx %08lx\n",pBuffer,*pAndXSmbBufferSize));
|
|
|
|
if (Status == STATUS_SUCCESS) {
|
|
SmbPutUshort(
|
|
&pTreeConnectAndX->ByteCount,
|
|
(USHORT)(OriginalBufferSize
|
|
- *pAndXSmbBufferSize
|
|
- FIELD_OFFSET(REQ_TREE_CONNECT_ANDX,Buffer)
|
|
)
|
|
);
|
|
}
|
|
|
|
RxDbgTrace( 0, (DEBUG_TRACE_ALWAYS),
|
|
("LmBuildTreeConnectSmb end buffer,rem %08lx %08lx\n",
|
|
pBuffer,
|
|
*pAndXSmbBufferSize));
|
|
}
|
|
|
|
return Status;
|
|
}
|
|
|
|
NTSTATUS
|
|
NtBuildTreeConnectSmb(
|
|
PSMB_EXCHANGE pExchange,
|
|
PGENERIC_ANDX pAndXSmb,
|
|
PULONG pAndXSmbBufferSize)
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
This routine builds the tree connect SMB for a pre NT server
|
|
|
|
Arguments:
|
|
|
|
pExchange - the exchange instance
|
|
|
|
pAndXSmb - the session setup to be filled in
|
|
|
|
pAndXSmbBufferSize - the SMB buffer size on input modified to remaining size on
|
|
output.
|
|
|
|
Return Value:
|
|
|
|
RXSTATUS - The return status for the operation
|
|
|
|
--*/
|
|
{
|
|
NTSTATUS Status = STATUS_SUCCESS;
|
|
|
|
UNICODE_STRING ServerName;
|
|
UNICODE_STRING NetRootName;
|
|
|
|
PSMBCE_SERVER pServer;
|
|
|
|
PREQ_TREE_CONNECT_ANDX pTreeConnect = (PREQ_TREE_CONNECT_ANDX)pAndXSmb;
|
|
|
|
ULONG OriginalBufferSize = *pAndXSmbBufferSize;
|
|
PBYTE pBuffer;
|
|
ULONG BufferSize;
|
|
|
|
PAGED_CODE();
|
|
|
|
BufferSize = OriginalBufferSize;
|
|
|
|
if(BufferSize < FIELD_OFFSET(REQ_TREE_CONNECT_ANDX,Buffer) + 1) {
|
|
return STATUS_BUFFER_OVERFLOW;
|
|
}
|
|
|
|
pServer = SmbCeGetExchangeServer(pExchange);
|
|
|
|
SmbCeGetServerName(pExchange->SmbCeContext.pVNetRoot->pNetRoot->pSrvCall,&ServerName);
|
|
SmbCeGetNetRootName(pExchange->SmbCeContext.pVNetRoot->pNetRoot,&NetRootName);
|
|
|
|
pTreeConnect->AndXCommand = 0xff; // No ANDX
|
|
pTreeConnect->AndXReserved = 0x00; // Reserved (MBZ)
|
|
|
|
SmbPutUshort(&pTreeConnect->AndXOffset, 0x0000); // No AndX as of yet.
|
|
|
|
pTreeConnect->WordCount = 4;
|
|
|
|
SmbPutUshort( &pTreeConnect->Flags, 0 );
|
|
|
|
pBuffer = (PBYTE)pTreeConnect + FIELD_OFFSET(REQ_TREE_CONNECT_ANDX,Buffer);
|
|
BufferSize -= FIELD_OFFSET(REQ_TREE_CONNECT_ANDX,Buffer);
|
|
|
|
if(pServer->SecurityMode == SECURITY_MODE_USER_LEVEL){
|
|
// No password information is required as part of tree connect for user level
|
|
// security servers. Therefore send a null string as the password.
|
|
SmbPutUshort(&pTreeConnect->PasswordLength,0x1);
|
|
|
|
*((PCHAR)pBuffer) = '\0';
|
|
pBuffer += sizeof(CHAR);
|
|
BufferSize -= sizeof(CHAR);
|
|
} else {
|
|
USHORT PasswordLength;
|
|
//plug in the password for this server.....qweee
|
|
Status = BuildTreeConnectSecurityInformation(
|
|
pExchange,
|
|
pBuffer,
|
|
(PBYTE)&PasswordLength,
|
|
&BufferSize);
|
|
|
|
if (Status == STATUS_SUCCESS ) {
|
|
pBuffer += PasswordLength;
|
|
SmbPutUshort(&pTreeConnect->PasswordLength,PasswordLength);
|
|
}
|
|
}
|
|
|
|
if (NT_SUCCESS(Status)) {
|
|
Status = BuildCanonicalNetRootInformation(
|
|
&ServerName,
|
|
&NetRootName,
|
|
NET_ROOT_WILD, //let the server tell us! pNetRoot->Type,
|
|
BooleanFlagOn(pServer->DialectFlags,DF_UNICODE),
|
|
TRUE, //postpend the service string
|
|
&pBuffer,
|
|
&BufferSize);
|
|
}
|
|
|
|
if (NT_SUCCESS(Status)) {
|
|
SmbPutUshort(
|
|
&pTreeConnect->ByteCount,
|
|
(USHORT)(OriginalBufferSize -
|
|
FIELD_OFFSET(REQ_TREE_CONNECT_ANDX,Buffer) -
|
|
BufferSize));
|
|
}
|
|
|
|
// update the buffer size to reflect the amount consumed.
|
|
*pAndXSmbBufferSize = BufferSize;
|
|
|
|
return Status;
|
|
}
|
|
|