|
|
/*++
Copyright (c) 1989 Microsoft Corporation
Module Name:
openclos.c
Abstract:
This module implements the name cache for file basic and standard information.
Author:
Yun Lin [YunLin] 2-Octorber-1998
Revision History:
--*/
#include "precomp.h"
#include "smbce.h"
#pragma hdrstop
#ifdef ALLOC_PRAGMA
#pragma alloc_text(PAGE, MRxSmbIsStreamFile)
#pragma alloc_text(PAGE, MRxSmbCacheFileNotFound)
#pragma alloc_text(PAGE, MRxSmbCreateFileInfoCache)
#pragma alloc_text(PAGE, MRxSmbIsFileInfoCacheFound)
#pragma alloc_text(PAGE, MRxSmbInvalidateFileInfoCache)
#pragma alloc_text(PAGE, MRxSmbUpdateBasicFileInfoCache)
#pragma alloc_text(PAGE, MRxSmbCreateBasicFileInfoCache)
#pragma alloc_text(PAGE, MRxSmbUpdateFileInfoCacheStatus)
#pragma alloc_text(PAGE, MRxSmbCreateStandardFileInfoCache)
#pragma alloc_text(PAGE, MRxSmbUpdateFileInfoCacheFileSize)
#pragma alloc_text(PAGE, MRxSmbUpdateStandardFileInfoCache)
#pragma alloc_text(PAGE, MRxSmbInvalidateFileNotFoundCache)
#pragma alloc_text(PAGE, MRxSmbUpdateBasicFileInfoCacheAll)
#pragma alloc_text(PAGE, MRxSmbInvalidateBasicFileInfoCache)
#pragma alloc_text(PAGE, MRxSmbUpdateBasicFileInfoCacheStatus)
#pragma alloc_text(PAGE, MRxSmbInvalidateStandardFileInfoCache)
#pragma alloc_text(PAGE, MRxSmbInvalidateInternalFileInfoCache)
#pragma alloc_text(PAGE, MRxSmbUpdateStandardFileInfoCacheStatus)
#endif
extern FAST_MUTEX MRxSmbFileInfoCacheLock;
VOID MRxSmbCreateFileInfoCache( PRX_CONTEXT RxContext, PSMBPSE_FILEINFO_BUNDLE FileInfo, PSMBCEDB_SERVER_ENTRY pServerEntry, NTSTATUS Status ) /*++
Routine Description:
This routine creates name cache entry for both file basic and standard information.
Arguments:
RxContext - the RDBSS context FileInfo - the file information package including basic and standard information Status - the status returned from server response of query file information
Return Value:
none
--*/ { PAGED_CODE();
if (!pServerEntry->Server.IsLoopBack) { MRxSmbCreateBasicFileInfoCache(RxContext,&FileInfo->Basic,pServerEntry,Status); MRxSmbCreateStandardFileInfoCache(RxContext,&FileInfo->Standard,pServerEntry,Status); } }
VOID MRxSmbCreateBasicFileInfoCache( PRX_CONTEXT RxContext, PFILE_BASIC_INFORMATION Basic, PSMBCEDB_SERVER_ENTRY pServerEntry, NTSTATUS Status ) /*++
Routine Description:
This routine creates name cache entry for the file basic information.
Arguments:
RxContext - the RDBSS context Basic - the file basic information package Status - the status returned from server response of query file information
Return Value:
none
--*/ { RxCaptureFcb; PUNICODE_STRING OriginalFileName = GET_ALREADY_PREFIXED_NAME_FROM_CONTEXT(RxContext);
PNAME_CACHE NameCache = NULL; PMRX_NET_ROOT NetRoot; PSMBCEDB_NET_ROOT_ENTRY pNetRootEntry; PNAME_CACHE_CONTROL NameCacheCtl; RX_NC_CHECK_STATUS NameCacheStatus; PFILE_BASIC_INFORMATION FileInfoCache = NULL;
PAGED_CODE();
if (pServerEntry->Server.IsLoopBack || (MRxSmbIsLongFileName(RxContext) && pServerEntry->Server.Dialect != LANMAN21_DIALECT)) { return; }
if (RxContext->MajorFunction == IRP_MJ_CREATE) { NetRoot = RxContext->Create.pNetRoot; } else { ASSERT(capFcb != NULL); NetRoot = capFcb->pNetRoot; }
pNetRootEntry = SmbCeGetAssociatedNetRootEntry(NetRoot); NameCacheCtl = &pNetRootEntry->NameCacheCtlGFABasic;
ExAcquireFastMutex(&MRxSmbFileInfoCacheLock);
NameCache = RxNameCacheFetchEntry(NameCacheCtl,OriginalFileName);
if (NameCache == NULL) { NameCache = RxNameCacheCreateEntry ( NameCacheCtl, OriginalFileName, TRUE); // case insensitive match
}
if (NameCache != NULL) { FileInfoCache = (PFILE_BASIC_INFORMATION)NameCache->ContextExtension;
*FileInfoCache = *Basic;
NameCache->PriorStatus = Status;
RxNameCacheActivateEntry( NameCacheCtl, NameCache, NAME_CACHE_OBJ_GET_FILE_ATTRIB_LIFETIME, MRxSmbStatistics.SmbsReceived.LowPart);
//DbgPrint(" Create File Attrib cache : %x %wZ\n",Basic->FileAttributes,OriginalFileName);
//DbgPrint(" Create File Attrib cache : %I64X %I64X %wZ\n",Basic->CreationTime,Basic->LastAccessTime,OriginalFileName);
}
ExReleaseFastMutex(&MRxSmbFileInfoCacheLock); }
VOID MRxSmbCreateStandardFileInfoCache( PRX_CONTEXT RxContext, PFILE_STANDARD_INFORMATION Standard, PSMBCEDB_SERVER_ENTRY pServerEntry, NTSTATUS Status ) /*++
Routine Description:
This routine creates name cache entry for the file standard information.
Arguments:
RxContext - the RDBSS context Standard - the file standard information package Status - the status returned from server response of query file information
Return Value:
none
--*/ { RxCaptureFcb; PUNICODE_STRING OriginalFileName = GET_ALREADY_PREFIXED_NAME_FROM_CONTEXT(RxContext);
PNAME_CACHE NameCache = NULL; PMRX_NET_ROOT NetRoot = capFcb->pNetRoot; PSMBCEDB_NET_ROOT_ENTRY pNetRootEntry = SmbCeGetAssociatedNetRootEntry(NetRoot); PNAME_CACHE_CONTROL NameCacheCtl = &pNetRootEntry->NameCacheCtlGFAStandard; PFILE_STANDARD_INFORMATION FileInfoCache = NULL;
PAGED_CODE();
if (pServerEntry->Server.IsLoopBack || (MRxSmbIsLongFileName(RxContext) && pServerEntry->Server.Dialect != LANMAN21_DIALECT)) { return; }
ExAcquireFastMutex(&MRxSmbFileInfoCacheLock);
NameCache = RxNameCacheFetchEntry(NameCacheCtl,OriginalFileName);
if (NameCache == NULL) { NameCache = RxNameCacheCreateEntry ( NameCacheCtl, OriginalFileName, TRUE); // case insensitive match
}
if (NameCache != NULL) { FileInfoCache = (PFILE_STANDARD_INFORMATION)NameCache->ContextExtension;
*FileInfoCache = *Standard;
NameCache->PriorStatus = Status;
RxNameCacheActivateEntry( NameCacheCtl, NameCache, NAME_CACHE_OBJ_GET_FILE_ATTRIB_LIFETIME, MRxSmbStatistics.SmbsReceived.LowPart);
//DbgPrint(" Create Standard cache : %I64x %wZ\n",((PFILE_STANDARD_INFORMATION)NameCache->ContextExtension)->EndOfFile,OriginalFileName);
}
ExReleaseFastMutex(&MRxSmbFileInfoCacheLock); }
VOID MRxSmbCreateInternalFileInfoCache( PRX_CONTEXT RxContext, PFILE_INTERNAL_INFORMATION Internal, PSMBCEDB_SERVER_ENTRY pServerEntry, NTSTATUS Status ) /*++
Routine Description:
This routine creates name cache entry for the file internal information.
Arguments:
RxContext - the RDBSS context Standard - the file standard information package Status - the status returned from server response of query file information
Return Value:
none
--*/ { RxCaptureFcb; PUNICODE_STRING OriginalFileName = GET_ALREADY_PREFIXED_NAME_FROM_CONTEXT(RxContext);
PNAME_CACHE NameCache = NULL; PMRX_NET_ROOT NetRoot = capFcb->pNetRoot; PSMBCEDB_NET_ROOT_ENTRY pNetRootEntry = SmbCeGetAssociatedNetRootEntry(NetRoot); PNAME_CACHE_CONTROL NameCacheCtl = &pNetRootEntry->NameCacheCtlGFAInternal; PFILE_INTERNAL_INFORMATION FileInfoCache = NULL;
PAGED_CODE();
if (pServerEntry->Server.IsLoopBack) { return; }
ExAcquireFastMutex(&MRxSmbFileInfoCacheLock);
NameCache = RxNameCacheFetchEntry(NameCacheCtl,OriginalFileName);
if (NameCache == NULL) { NameCache = RxNameCacheCreateEntry ( NameCacheCtl, OriginalFileName, TRUE); // case insensitive match
}
if (NameCache != NULL) { FileInfoCache = (PFILE_INTERNAL_INFORMATION)NameCache->ContextExtension;
*FileInfoCache = *Internal;
NameCache->PriorStatus = Status;
RxNameCacheActivateEntry( NameCacheCtl, NameCache, NAME_CACHE_OBJ_GET_FILE_ATTRIB_LIFETIME, MRxSmbStatistics.SmbsReceived.LowPart);
//DbgPrint(" Create Internal cache : %I64x %wZ\n",((PFILE_INTERNAL_INFORMATION)NameCache->ContextExtension)->IndexNumber,OriginalFileName);
}
ExReleaseFastMutex(&MRxSmbFileInfoCacheLock); }
VOID MRxSmbUpdateFileInfoCacheFromDelete( PRX_CONTEXT RxContext ) /*++
Routine Description:
This routine updates the status of the name cache entry as STATUS_OBJECT_NAME_NOT_FOUND for both file basic and standard information.
Arguments:
RxContext - the RDBSS context
Return Value:
none
--*/ { MRxSmbUpdateBasicFileInfoCacheStatus(RxContext,STATUS_OBJECT_NAME_NOT_FOUND); MRxSmbUpdateStandardFileInfoCacheStatus(RxContext,STATUS_OBJECT_NAME_NOT_FOUND);
// Trounce FullDirCache
MRxSmbInvalidateFullDirectoryCacheParent(RxContext, FALSE); }
VOID MRxSmbUpdateFileInfoCacheStatus( PRX_CONTEXT RxContext, NTSTATUS Status ) /*++
Routine Description:
This routine updates the status of the name cache entry for both file basic and standard information.
Arguments:
RxContext - the RDBSS context Status - the status needs to be put on the cache
Return Value:
none
--*/ { MRxSmbUpdateBasicFileInfoCacheStatus(RxContext,Status); MRxSmbUpdateStandardFileInfoCacheStatus(RxContext,Status); }
VOID MRxSmbUpdateBasicFileInfoCacheStatus( PRX_CONTEXT RxContext, NTSTATUS Status ) /*++
Routine Description:
This routine updates the status of the name cache entry for the file basic information.
Arguments:
RxContext - the RDBSS context Status - the status needs to be put on the cache
Return Value:
none
--*/ { RxCaptureFcb; PUNICODE_STRING OriginalFileName = GET_ALREADY_PREFIXED_NAME_FROM_CONTEXT(RxContext); UNICODE_STRING FileName;
PNAME_CACHE NameCache = NULL; PMRX_NET_ROOT NetRoot = capFcb->pNetRoot; PSMBCEDB_NET_ROOT_ENTRY pNetRootEntry = SmbCeGetAssociatedNetRootEntry(NetRoot); PNAME_CACHE_CONTROL NameCacheCtl = &pNetRootEntry->NameCacheCtlGFABasic;
PAGED_CODE();
ExAcquireFastMutex(&MRxSmbFileInfoCacheLock);
NameCache = RxNameCacheFetchEntry(NameCacheCtl,OriginalFileName);
if (NameCache != NULL) { NameCache->PriorStatus = Status; RxNameCacheActivateEntry(NameCacheCtl, NameCache, 0, 0);
//DbgPrint("Update status basic : %x %wZ\n",Status,OriginalFileName);
RxLog(("Update status basic : %x %wZ\n",Status,OriginalFileName)); } else { RxLog(("Update status basic fails: %x %wZ\n",Status,OriginalFileName)); }
if (MRxSmbIsStreamFile(OriginalFileName,&FileName)) {
// if it is a stream file, we invalid the root file name cache needs since we are not
// sure what could happen to the root file on the server
NameCache = RxNameCacheFetchEntry(NameCacheCtl,&FileName);
if (NameCache != NULL) { RxNameCacheExpireEntry(NameCacheCtl, NameCache); }
//DbgPrint("Update status basic : %x %wZ\n",Status,&FileName);
}
ExReleaseFastMutex(&MRxSmbFileInfoCacheLock); }
VOID MRxSmbUpdateStandardFileInfoCacheStatus( PRX_CONTEXT RxContext, NTSTATUS Status ) /*++
Routine Description:
This routine updates the status of the name cache entry for the file standard information.
Arguments:
RxContext - the RDBSS context Status - the status needs to be put on the cache
Return Value:
none
--*/ { RxCaptureFcb; PUNICODE_STRING OriginalFileName = GET_ALREADY_PREFIXED_NAME_FROM_CONTEXT(RxContext); UNICODE_STRING FileName;
PNAME_CACHE NameCache = NULL; PMRX_NET_ROOT NetRoot = capFcb->pNetRoot; PSMBCEDB_NET_ROOT_ENTRY pNetRootEntry = SmbCeGetAssociatedNetRootEntry(NetRoot); PNAME_CACHE_CONTROL NameCacheCtl = &pNetRootEntry->NameCacheCtlGFAStandard;
PAGED_CODE();
ExAcquireFastMutex(&MRxSmbFileInfoCacheLock);
NameCache = RxNameCacheFetchEntry(NameCacheCtl,OriginalFileName);
if (NameCache != NULL) { NameCache->PriorStatus = Status; RxNameCacheActivateEntry(NameCacheCtl, NameCache, 0, 0); } /*
if (MRxSmbIsStreamFile(OriginalFileName,&FileName)) { NameCache = RxNameCacheFetchEntry(NameCacheCtl,&FileName);
if (NameCache != NULL) { RxNameCacheExpireEntry(NameCacheCtl, NameCache); } } */
ExReleaseFastMutex(&MRxSmbFileInfoCacheLock); }
VOID MRxSmbInvalidateFileInfoCache( PRX_CONTEXT RxContext ) /*++
Routine Description:
This routine invalidates the name cache entry for both file basic and standard information.
Arguments:
RxContext - the RDBSS context
Return Value:
none
--*/ { PAGED_CODE();
MRxSmbInvalidateBasicFileInfoCache(RxContext); MRxSmbInvalidateStandardFileInfoCache(RxContext);
MRxSmbInvalidateFullDirectoryCache(RxContext);
// Don't do it here, since this invalidates Full Dir Cache
// when the file is not there.
// MRxSmbInvalidateFullDirectoryCacheParent(RxContext);
}
VOID MRxSmbInvalidateBasicFileInfoCache( PRX_CONTEXT RxContext ) /*++
Routine Description:
This routine invalidates the name cache entry for file basic information.
Arguments:
RxContext - the RDBSS context
Return Value:
none
--*/ { RxCaptureFcb; PUNICODE_STRING OriginalFileName = GET_ALREADY_PREFIXED_NAME_FROM_CONTEXT(RxContext); UNICODE_STRING FileName;
PNAME_CACHE NameCache = NULL; PMRX_NET_ROOT NetRoot = capFcb->pNetRoot; PSMBCEDB_NET_ROOT_ENTRY pNetRootEntry = SmbCeGetAssociatedNetRootEntry(NetRoot); PNAME_CACHE_CONTROL NameCacheCtl = &pNetRootEntry->NameCacheCtlGFABasic;
PAGED_CODE();
ExAcquireFastMutex(&MRxSmbFileInfoCacheLock);
NameCache = RxNameCacheFetchEntry(NameCacheCtl,OriginalFileName);
if (NameCache != NULL) { RxNameCacheExpireEntry(NameCacheCtl, NameCache); }
if (MRxSmbIsStreamFile(OriginalFileName,&FileName)) { NameCache = RxNameCacheFetchEntry(NameCacheCtl,&FileName);
if (NameCache != NULL) { RxNameCacheExpireEntry(NameCacheCtl, NameCache); } }
ExReleaseFastMutex(&MRxSmbFileInfoCacheLock); }
VOID MRxSmbInvalidateStandardFileInfoCache( PRX_CONTEXT RxContext ) /*++
Routine Description:
This routine invalidates the name cache entry for the file standard information.
Arguments:
RxContext - the RDBSS context
Return Value:
none
--*/ { RxCaptureFcb; PUNICODE_STRING OriginalFileName = GET_ALREADY_PREFIXED_NAME_FROM_CONTEXT(RxContext); UNICODE_STRING FileName;
PNAME_CACHE NameCache = NULL; PMRX_NET_ROOT NetRoot = capFcb->pNetRoot; PSMBCEDB_NET_ROOT_ENTRY pNetRootEntry = SmbCeGetAssociatedNetRootEntry(NetRoot); PNAME_CACHE_CONTROL NameCacheCtl = &pNetRootEntry->NameCacheCtlGFAStandard;
PAGED_CODE();
ExAcquireFastMutex(&MRxSmbFileInfoCacheLock);
NameCache = RxNameCacheFetchEntry(NameCacheCtl,OriginalFileName);
if (NameCache != NULL) { RxNameCacheExpireEntry(NameCacheCtl, NameCache); //DbgPrint("Invalid Standard cache : %I64x %wZ\n",((PFILE_STANDARD_INFORMATION)NameCache->ContextExtension)->EndOfFile,OriginalFileName);
}
if (MRxSmbIsStreamFile(OriginalFileName,&FileName)) { NameCache = RxNameCacheFetchEntry(NameCacheCtl,&FileName);
if (NameCache != NULL) { RxNameCacheExpireEntry(NameCacheCtl, NameCache); } }
ExReleaseFastMutex(&MRxSmbFileInfoCacheLock); }
VOID MRxSmbInvalidateInternalFileInfoCache( PRX_CONTEXT RxContext ) /*++
Routine Description:
This routine invalidates the name cache entry for file internal information.
Arguments:
RxContext - the RDBSS context
Return Value:
none
--*/ { RxCaptureFcb; PUNICODE_STRING OriginalFileName = GET_ALREADY_PREFIXED_NAME_FROM_CONTEXT(RxContext); UNICODE_STRING FileName;
PNAME_CACHE NameCache = NULL; PMRX_NET_ROOT NetRoot = capFcb->pNetRoot; PSMBCEDB_NET_ROOT_ENTRY pNetRootEntry = SmbCeGetAssociatedNetRootEntry(NetRoot); PNAME_CACHE_CONTROL NameCacheCtl = &pNetRootEntry->NameCacheCtlGFAInternal;
PAGED_CODE();
ExAcquireFastMutex(&MRxSmbFileInfoCacheLock);
NameCache = RxNameCacheFetchEntry(NameCacheCtl,OriginalFileName);
if (NameCache != NULL) { RxNameCacheExpireEntry(NameCacheCtl, NameCache);
//DbgPrint("Invalid Internal cache : %wZ\n",OriginalFileName);
RxLog(("Invalid Internal cache : %wZ\n",OriginalFileName)); }
if (MRxSmbIsStreamFile(OriginalFileName,&FileName)) { NameCache = RxNameCacheFetchEntry(NameCacheCtl,&FileName);
if (NameCache != NULL) { RxNameCacheExpireEntry(NameCacheCtl, NameCache); } }
ExReleaseFastMutex(&MRxSmbFileInfoCacheLock); }
VOID MRxSmbUpdateFileInfoCacheFileSize( PRX_CONTEXT RxContext, PLARGE_INTEGER FileSize ) /*++
Routine Description:
This routine updates file size on the name cache entry for the file standard information.
Arguments:
RxContext - the RDBSS context
Return Value:
none
--*/ { RxCaptureFcb; PUNICODE_STRING OriginalFileName = GET_ALREADY_PREFIXED_NAME_FROM_CONTEXT(RxContext); UNICODE_STRING FileName;
PNAME_CACHE NameCache = NULL; PMRX_NET_ROOT NetRoot = capFcb->pNetRoot; PSMBCEDB_NET_ROOT_ENTRY pNetRootEntry = SmbCeGetAssociatedNetRootEntry(NetRoot); PNAME_CACHE_CONTROL NameCacheCtl = &pNetRootEntry->NameCacheCtlGFAStandard; PFILE_STANDARD_INFORMATION FileInfoCache = NULL;
PAGED_CODE();
ExAcquireFastMutex(&MRxSmbFileInfoCacheLock);
NameCache = RxNameCacheFetchEntry(NameCacheCtl,OriginalFileName);
if (NameCache != NULL) { FileInfoCache = (PFILE_STANDARD_INFORMATION)NameCache->ContextExtension;
FileInfoCache->AllocationSize.QuadPart = FileSize->QuadPart; FileInfoCache->EndOfFile.QuadPart = FileSize->QuadPart;
RxNameCacheActivateEntry(NameCacheCtl, NameCache, 0, 0);
//DbgPrint("Update File size cache : %I64x %wZ\n",((PFILE_STANDARD_INFORMATION)NameCache->ContextExtension)->EndOfFile,OriginalFileName);
} /*
if (MRxSmbIsStreamFile(OriginalFileName,&FileName)) { NameCache = RxNameCacheFetchEntry(NameCacheCtl,&FileName);
if (NameCache != NULL) { RxNameCacheExpireEntry(NameCacheCtl, NameCache); } }*/
ExReleaseFastMutex(&MRxSmbFileInfoCacheLock);
// Mark FullDir Cache, weak for bdi : Current Invalidate for correctness
MRxSmbInvalidateFullDirectoryCacheParent(RxContext, TRUE); }
VOID MRxSmbUpdateBasicFileInfoCache( PRX_CONTEXT RxContext, ULONG FileAttributes, PLARGE_INTEGER pLastWriteTime ) /*++
Routine Description:
This routine updates file attributs and last write time on the name cache entry for the file basic information.
Arguments:
RxContext - the RDBSS context FileAttributes - new file attributes pLastWriteTime - address of file last write time
Return Value:
none
--*/ { FILE_BASIC_INFORMATION Basic;
Basic.ChangeTime.QuadPart = 0; Basic.CreationTime.QuadPart = 0; Basic.LastWriteTime.QuadPart = 0; Basic.LastAccessTime.QuadPart = 0;
if (pLastWriteTime != NULL && pLastWriteTime->QuadPart != 0) { Basic.LastWriteTime = *pLastWriteTime; }
Basic.FileAttributes = FileAttributes;
MRxSmbUpdateBasicFileInfoCacheAll(RxContext,&Basic);
// Mark FullDir Cache, weak for bdi : Current Invalidate for correctness
MRxSmbInvalidateFullDirectoryCacheParent(RxContext, TRUE); }
VOID MRxSmbUpdateBasicFileInfoCacheAll( PRX_CONTEXT RxContext, PFILE_BASIC_INFORMATION Basic ) /*++
Routine Description:
This routine updates the name cache entry for the file basic information.
Arguments:
RxContext - the RDBSS context Basic - file basic information
Return Value:
none
--*/ { RxCaptureFcb; PUNICODE_STRING OriginalFileName = GET_ALREADY_PREFIXED_NAME_FROM_CONTEXT(RxContext); UNICODE_STRING FileName;
PNAME_CACHE NameCache = NULL; PMRX_NET_ROOT NetRoot = capFcb->pNetRoot; PSMBCEDB_NET_ROOT_ENTRY pNetRootEntry = SmbCeGetAssociatedNetRootEntry(NetRoot); PNAME_CACHE_CONTROL NameCacheCtl = &pNetRootEntry->NameCacheCtlGFABasic; PFILE_BASIC_INFORMATION BasicFileInfoCache = NULL;
PAGED_CODE();
ExAcquireFastMutex(&MRxSmbFileInfoCacheLock);
NameCache = RxNameCacheFetchEntry(NameCacheCtl,OriginalFileName);
if (NameCache != NULL) { ULONG SavedAttributes = 0;
BasicFileInfoCache = (PFILE_BASIC_INFORMATION)NameCache->ContextExtension;
if (Basic->CreationTime.QuadPart != 0) { BasicFileInfoCache->CreationTime = Basic->CreationTime; }
if (Basic->LastAccessTime.QuadPart != 0) { BasicFileInfoCache->LastAccessTime = Basic->LastAccessTime; }
if (Basic->LastWriteTime.QuadPart != 0) { BasicFileInfoCache->LastWriteTime = Basic->LastWriteTime; }
SavedAttributes = BasicFileInfoCache->FileAttributes & ( FILE_ATTRIBUTE_DIRECTORY | FILE_ATTRIBUTE_ENCRYPTED | FILE_ATTRIBUTE_COMPRESSED | FILE_ATTRIBUTE_SPARSE_FILE | FILE_ATTRIBUTE_REPARSE_POINT );
//DbgPrint("Update File Attrib cache 1: %x %wZ\n",FileAttributes,OriginalFileName);
//DbgPrint("Update File Attrib cache 2: %x %wZ\n",BasicFileInfoCache->FileAttributes,OriginalFileName);
BasicFileInfoCache->FileAttributes = Basic->FileAttributes;
BasicFileInfoCache->FileAttributes |= SavedAttributes;
if (BasicFileInfoCache->FileAttributes & ~FILE_ATTRIBUTE_NORMAL) { BasicFileInfoCache->FileAttributes &= ~FILE_ATTRIBUTE_NORMAL; }
RxNameCacheActivateEntry(NameCacheCtl, NameCache, 0, 0);
//DbgPrint("Update File Attrib cache 3: %x %wZ\n",BasicFileInfoCache->FileAttributes,OriginalFileName);
//DbgPrint("Update File Attrib cache : %I64X %I64X %wZ\n",BasicFileInfoCache->CreationTime,BasicFileInfoCache->LastAccessTime,OriginalFileName);
}
if (MRxSmbIsStreamFile(OriginalFileName,&FileName)) { // if it is a stream file, we need to invalid the root file since we are not sure how this
// could affect the root file.
NameCache = RxNameCacheFetchEntry(NameCacheCtl,&FileName);
if (NameCache != NULL) { // expiring in XX routine
RxNameCacheExpireEntry(NameCacheCtl, NameCache); } }
ExReleaseFastMutex(&MRxSmbFileInfoCacheLock);
// Mark FullDir Cache, weak for bdi : Current Invalidate for correctness
MRxSmbInvalidateFullDirectoryCacheParent(RxContext, TRUE); }
VOID MRxSmbUpdateStandardFileInfoCache( PRX_CONTEXT RxContext, PFILE_STANDARD_INFORMATION Standard, BOOLEAN IsDirectory ) /*++
Routine Description:
This routine updates the name cache entry for the file standard information.
Arguments:
RxContext - the RDBSS context Standard - file standard information IsDirectory - file is a directory
Return Value:
none
--*/ { RxCaptureFcb; PUNICODE_STRING OriginalFileName = GET_ALREADY_PREFIXED_NAME_FROM_CONTEXT(RxContext); UNICODE_STRING FileName;
PNAME_CACHE NameCache = NULL; PMRX_NET_ROOT NetRoot = capFcb->pNetRoot; PSMBCEDB_NET_ROOT_ENTRY pNetRootEntry = SmbCeGetAssociatedNetRootEntry(NetRoot); PNAME_CACHE_CONTROL NameCacheCtl = &pNetRootEntry->NameCacheCtlGFAStandard; PFILE_STANDARD_INFORMATION StandardFileInfoCache = NULL;
PAGED_CODE();
ExAcquireFastMutex(&MRxSmbFileInfoCacheLock);
NameCache = RxNameCacheFetchEntry(NameCacheCtl,OriginalFileName);
if (NameCache != NULL) { StandardFileInfoCache = (PFILE_STANDARD_INFORMATION)NameCache->ContextExtension;
if (Standard != NULL) { *StandardFileInfoCache = *Standard; } else { StandardFileInfoCache->Directory = IsDirectory; }
RxNameCacheActivateEntry(NameCacheCtl, NameCache, 0, 0);
//DbgPrint(" Update Standard cache : %I64x %wZ\n",((PFILE_STANDARD_INFORMATION)NameCache->ContextExtension)->EndOfFile,OriginalFileName);
} /*
if (MRxSmbIsStreamFile(OriginalFileName,&FileName)) { NameCache = RxNameCacheFetchEntry(NameCacheCtl,&FileName);
if (NameCache != NULL) { RxNameCacheExpireEntry(NameCacheCtl, NameCache); } } */
ExReleaseFastMutex(&MRxSmbFileInfoCacheLock);
// Mark FullDir Cache, weak for bdi : Current Invalidate for correctness
MRxSmbInvalidateFullDirectoryCacheParent(RxContext, TRUE); }
BOOLEAN MRxSmbIsFileInfoCacheFound( PRX_CONTEXT RxContext, PSMBPSE_FILEINFO_BUNDLE FileInfo, NTSTATUS *Status, PUNICODE_STRING OriginalFileName ) /*++
Routine Description:
This routine looks for the name cache entry of both file basic and standard information.
Arguments:
RxContext - the RDBSS context FileInfo - buffer to return file basic and standard information Status - status retured on the last reponse from server
Return Value:
BOOLEAN - name cache found
--*/ { PFILE_BASIC_INFORMATION Basic; PFILE_STANDARD_INFORMATION Standard;
BOOLEAN CacheFound = FALSE;
if (MRxSmbIsBasicFileInfoCacheFound(RxContext,&FileInfo->Basic,Status,OriginalFileName)) { if (*Status == STATUS_SUCCESS) { if (MRxSmbIsStandardFileInfoCacheFound(RxContext,&FileInfo->Standard,Status,OriginalFileName)) { CacheFound = TRUE; } } else {
// if an error stored on the file basic information cache, return cache found
CacheFound = TRUE; } }
return CacheFound; }
// these file attributes may be different between streams on a file
ULONG StreamAttributes = FILE_ATTRIBUTE_COMPRESSED | FILE_ATTRIBUTE_DIRECTORY | FILE_ATTRIBUTE_SPARSE_FILE;
BOOLEAN MRxSmbIsBasicFileInfoCacheFound( PRX_CONTEXT RxContext, PFILE_BASIC_INFORMATION Basic, NTSTATUS *Status, PUNICODE_STRING OriginalFileName ) /*++
Routine Description:
This routine looks for the name cache entry of the file basic information.
Arguments:
RxContext - the RDBSS context Basic - buffer to return file basic information Status - status retured on the last reponse from server
Return Value:
BOOLEAN - name cache found
--*/ { RxCaptureFcb; UNICODE_STRING FileName;
PNAME_CACHE NameCache = NULL; PMRX_NET_ROOT NetRoot; PSMBCEDB_NET_ROOT_ENTRY pNetRootEntry; PNAME_CACHE_CONTROL NameCacheCtl; RX_NC_CHECK_STATUS NameCacheStatus;
BOOLEAN CacheFound = FALSE; BOOLEAN RootFound = FALSE; ULONG RootAttributes = 0; NTSTATUS RootStatus = STATUS_SUCCESS;
PAGED_CODE();
if (OriginalFileName == NULL) { OriginalFileName = GET_ALREADY_PREFIXED_NAME_FROM_CONTEXT(RxContext); }
if (RxContext->MajorFunction == IRP_MJ_CREATE) { NetRoot = RxContext->Create.pNetRoot; } else { ASSERT(capFcb != NULL); NetRoot = capFcb->pNetRoot; }
pNetRootEntry = SmbCeGetAssociatedNetRootEntry(NetRoot); NameCacheCtl = &pNetRootEntry->NameCacheCtlGFABasic;
ExAcquireFastMutex(&MRxSmbFileInfoCacheLock);
if (MRxSmbIsStreamFile(OriginalFileName,&FileName)) {
// check for stream file attribute changes
NameCache = RxNameCacheFetchEntry(NameCacheCtl,&FileName);
if (NameCache != NULL) { NameCacheStatus = RxNameCacheCheckEntry( NameCache, NameCache->Context);
if (NameCacheStatus == RX_NC_SUCCESS) { RootFound = TRUE; RootStatus = NameCache->PriorStatus; RootAttributes = ((PFILE_BASIC_INFORMATION)NameCache->ContextExtension)->FileAttributes & ~StreamAttributes; RxNameCacheActivateEntry(NameCacheCtl, NameCache, 0, 0); } else { RxNameCacheExpireEntry(NameCacheCtl, NameCache); } } }
NameCache = RxNameCacheFetchEntry(NameCacheCtl,OriginalFileName);
if (NameCache != NULL) { //
// Found it. Now check entry for not expired.
// Note - The NameCache entry has been pulled off the active list.
//
NameCacheStatus = RxNameCacheCheckEntry( NameCache, NameCache->Context);
if (NameCacheStatus == RX_NC_SUCCESS && (!RootFound || (*Status == RootStatus && (Basic->FileAttributes & ~StreamAttributes) == RootAttributes))) {
// The name cache matches if it is not expired and the attributes matches the one of
// the root file if it is a stream file. If this is a match, return the old status,
// file info and reactivate the entry but leave expiration time unchanged.
*Status = NameCache->PriorStatus; RxNameCacheOpSaved(NameCacheCtl);
*Basic = *((PFILE_BASIC_INFORMATION)NameCache->ContextExtension);
CacheFound = TRUE;
// put the entry back to the active list without changing the expire time
RxNameCacheActivateEntry(NameCacheCtl, NameCache, 0, 0);
//DbgPrint(" Found Basic cache : %x %wZ\n",Basic->FileAttributes,OriginalFileName);
//DbgPrint(" Get File Attrib cache : %I64X %I64X %wZ\n",Basic->CreationTime,Basic->LastAccessTime,OriginalFileName);
} else { // put the entry back to the expire list
RxNameCacheExpireEntry(NameCacheCtl, NameCache); } } else { //DbgPrint(" No Basic cache : %wZ\n",OriginalFileName);
}
ExReleaseFastMutex(&MRxSmbFileInfoCacheLock);
return CacheFound; }
BOOLEAN MRxSmbIsStandardFileInfoCacheFound( PRX_CONTEXT RxContext, PFILE_STANDARD_INFORMATION Standard, NTSTATUS *Status, PUNICODE_STRING OriginalFileName ) /*++
Routine Description:
This routine looks for the name cache entry of the file standard information.
Arguments:
RxContext - the RDBSS context Standard - buffer to return file standard information Status - status retured on the last reponse from server
Return Value:
BOOLEAN - name cache found
--*/ { RxCaptureFcb; RxCaptureFobx; UNICODE_STRING FileName;
PNAME_CACHE NameCache = NULL; PMRX_NET_ROOT NetRoot = capFcb->pNetRoot; PSMBCEDB_NET_ROOT_ENTRY pNetRootEntry = SmbCeGetAssociatedNetRootEntry(NetRoot); PNAME_CACHE_CONTROL NameCacheCtl = &pNetRootEntry->NameCacheCtlGFAStandard; RX_NC_CHECK_STATUS NameCacheStatus;
BOOLEAN CacheFound = FALSE; BOOLEAN RootFound = FALSE; NTSTATUS RootStatus = STATUS_SUCCESS;
PMRX_SMB_SRV_OPEN smbSrvOpen;
PAGED_CODE();
if (OriginalFileName == NULL) { OriginalFileName = GET_ALREADY_PREFIXED_NAME_FROM_CONTEXT(RxContext); }
if (RxContext->MajorFunction == IRP_MJ_CREATE) { smbSrvOpen = MRxSmbGetSrvOpenExtension(RxContext->pRelevantSrvOpen); } else { smbSrvOpen = MRxSmbGetSrvOpenExtension(capFobx->pSrvOpen); }
ExAcquireFastMutex(&MRxSmbFileInfoCacheLock);
if (MRxSmbIsStreamFile(OriginalFileName,&FileName)) {
// check for stream file attribute changes
NameCache = RxNameCacheFetchEntry(NameCacheCtl,&FileName);
if (NameCache != NULL) { NameCacheStatus = RxNameCacheCheckEntry( NameCache, NameCache->Context);
if (NameCacheStatus == RX_NC_SUCCESS) { RootFound = TRUE; RootStatus = NameCache->PriorStatus; RxNameCacheActivateEntry(NameCacheCtl, NameCache, 0, 0); } else { RxNameCacheExpireEntry(NameCacheCtl, NameCache); } } }
NameCache = RxNameCacheFetchEntry(NameCacheCtl,OriginalFileName);
if (NameCache != NULL) { //
// Found it. Now check entry for not expired.
// Note - The NameCache entry has been pulled off the active list.
//
NameCacheStatus = RxNameCacheCheckEntry( NameCache, NameCache->Context);
if (NameCacheStatus == RX_NC_SUCCESS && (!RootFound || *Status == RootStatus)) {
// The name cache matches if it is not expired and the status matches the one of
// the root file if it is a stream file. If this is a match, return the old status,
// file info and reactivate the entry but leave expiration time unchanged.
*Status = NameCache->PriorStatus; RxNameCacheOpSaved(NameCacheCtl);
*Standard = *((PFILE_STANDARD_INFORMATION)NameCache->ContextExtension);
if (FlagOn(smbSrvOpen->Flags,SMB_SRVOPEN_FLAG_SUCCESSFUL_OPEN) && !FlagOn(smbSrvOpen->Flags,SMB_SRVOPEN_FLAG_NOT_REALLY_OPEN)) { RxGetFileSizeWithLock((PFCB)capFcb,&Standard->EndOfFile.QuadPart); }
CacheFound = TRUE;
// put the entry back to the active list without changing the expire time
RxNameCacheActivateEntry(NameCacheCtl, NameCache, 0, 0);
//DbgPrint(" Get Standard cache : %I64x %wZ\n",((PFILE_STANDARD_INFORMATION)NameCache->ContextExtension)->EndOfFile,OriginalFileName);
} else { // put the entry back to the expire list
RxNameCacheExpireEntry(NameCacheCtl, NameCache); } }
ExReleaseFastMutex(&MRxSmbFileInfoCacheLock);
return CacheFound; }
BOOLEAN MRxSmbIsInternalFileInfoCacheFound( PRX_CONTEXT RxContext, PFILE_INTERNAL_INFORMATION Internal, NTSTATUS *Status, PUNICODE_STRING OriginalFileName ) /*++
Routine Description:
This routine looks for the name cache entry of the file basic information.
Arguments:
RxContext - the RDBSS context Basic - buffer to return file basic information Status - status retured on the last reponse from server
Return Value:
BOOLEAN - name cache found
--*/ { RxCaptureFcb; UNICODE_STRING FileName;
PNAME_CACHE NameCache = NULL; PMRX_NET_ROOT NetRoot; PSMBCEDB_NET_ROOT_ENTRY pNetRootEntry; PNAME_CACHE_CONTROL NameCacheCtl; RX_NC_CHECK_STATUS NameCacheStatus;
BOOLEAN CacheFound = FALSE;
PAGED_CODE();
if (OriginalFileName == NULL) { OriginalFileName = GET_ALREADY_PREFIXED_NAME_FROM_CONTEXT(RxContext); }
//DbgPrint(" Query Internal cache : %wZ\n",OriginalFileName);
if (RxContext->MajorFunction == IRP_MJ_CREATE) { NetRoot = RxContext->Create.pNetRoot; } else { ASSERT(capFcb != NULL); NetRoot = capFcb->pNetRoot; }
pNetRootEntry = SmbCeGetAssociatedNetRootEntry(NetRoot); NameCacheCtl = &pNetRootEntry->NameCacheCtlGFAInternal;
ExAcquireFastMutex(&MRxSmbFileInfoCacheLock);
NameCache = RxNameCacheFetchEntry(NameCacheCtl,OriginalFileName);
if (NameCache != NULL) { //
// Found it. Now check entry for not expired.
// Note - The NameCache entry has been pulled off the active list.
//
NameCacheStatus = RxNameCacheCheckEntry( NameCache, NameCache->Context);
if (NameCacheStatus == RX_NC_SUCCESS) {
// The name cache matches if it is not expired and the attributes matches the one of
// the root file if it is a stream file. If this is a match, return the old status,
// file info and reactivate the entry but leave expiration time unchanged.
*Status = NameCache->PriorStatus; RxNameCacheOpSaved(NameCacheCtl);
*Internal = *((PFILE_INTERNAL_INFORMATION)NameCache->ContextExtension);
CacheFound = TRUE;
// put the entry back to the active list without changing the expire time
RxNameCacheActivateEntry(NameCacheCtl, NameCache, 0, 0);
//DbgPrint(" Found Internal cache : %I64x %wZ\n",Internal->IndexNumber,OriginalFileName);
} else { // put the entry back to the expire list
RxNameCacheExpireEntry(NameCacheCtl, NameCache); //DbgPrint("Internal cache expired : %wZ\n",OriginalFileName);
} } else { //DbgPrint(" No Internal cache : %wZ\n",OriginalFileName);
}
ExReleaseFastMutex(&MRxSmbFileInfoCacheLock);
return CacheFound; }
NTSTATUS MRxSmbGetFileInfoCacheStatus( PRX_CONTEXT RxContext ) /*++
Routine Description:
This routine looks for the status of the name cache entry of either file basic or standard information.
Arguments:
RxContext - the RDBSS context
Return Value:
NTSTATUS - statu of the name cache if found, otherwise, STATUS_SUCCESS
--*/ { RxCaptureFcb; PUNICODE_STRING OriginalFileName = GET_ALREADY_PREFIXED_NAME_FROM_CONTEXT(RxContext);
PNAME_CACHE NameCache = NULL; PMRX_NET_ROOT NetRoot = capFcb->pNetRoot; PSMBCEDB_NET_ROOT_ENTRY pNetRootEntry = SmbCeGetAssociatedNetRootEntry(NetRoot); PNAME_CACHE_CONTROL NameCacheCtlBasic = &pNetRootEntry->NameCacheCtlGFABasic; PNAME_CACHE_CONTROL NameCacheCtlStandard = &pNetRootEntry->NameCacheCtlGFAStandard; NTSTATUS Status = STATUS_MORE_PROCESSING_REQUIRED;
PAGED_CODE();
ExAcquireFastMutex(&MRxSmbFileInfoCacheLock);
NameCache = RxNameCacheFetchEntry(NameCacheCtlBasic,OriginalFileName);
if (NameCache != NULL) { RX_NC_CHECK_STATUS NameCacheStatus; //
// Found it. Now check entry for not expired
//
NameCacheStatus = RxNameCacheCheckEntry(NameCache,NameCache->Context);
if (NameCacheStatus == RX_NC_SUCCESS) { //
// If the cache has not expired, return the previous status.
//
Status = NameCache->PriorStatus; RxNameCacheOpSaved(NameCacheCtlBasic);
// put the entry back to the active list without changing the expire time
RxNameCacheActivateEntry(NameCacheCtlBasic, NameCache, 0, 0);
//DbgPrint(" Get Basic Status : %x %wZ\n",Status,OriginalFileName);
RxLog((" Get Basic Status : %x %wZ\n",Status,OriginalFileName)); } else { // put the entry back to the expire list
RxNameCacheExpireEntry(NameCacheCtlBasic, NameCache); } } else { NameCache = RxNameCacheFetchEntry(NameCacheCtlStandard,OriginalFileName);
if (NameCache != NULL) { RX_NC_CHECK_STATUS NameCacheStatus; //
// Found it. Now check entry for not expired
//
NameCacheStatus = RxNameCacheCheckEntry(NameCache,NameCache->Context);
if (NameCacheStatus == RX_NC_SUCCESS) { //
// If the cache has not expired, return the previous status.
//
Status = NameCache->PriorStatus; RxNameCacheOpSaved(NameCacheCtlStandard);
// put the entry back to the active list without changing the expire time
RxNameCacheActivateEntry(NameCacheCtlStandard, NameCache, 0, 0); } else { // put the entry back to the expire list
RxNameCacheExpireEntry(NameCacheCtlStandard, NameCache); } } }
ExReleaseFastMutex(&MRxSmbFileInfoCacheLock);
return Status; }
BOOLEAN MRxSmbIsFileNotFoundCached( PRX_CONTEXT RxContext ) /*++
Routine Description:
This routine checks if the name cache entry exists as File Not Found.
Arguments:
RxContext - the RDBSS context
Return Value:
BOOLEAN - name cache found
--*/ { RxCaptureFcb; PUNICODE_STRING OriginalFileName = GET_ALREADY_PREFIXED_NAME_FROM_CONTEXT(RxContext); UNICODE_STRING StreamlessName;
PNAME_CACHE NameCache = NULL; PMRX_NET_ROOT NetRoot = capFcb->pNetRoot; PSMBCEDB_NET_ROOT_ENTRY pNetRootEntry = SmbCeGetAssociatedNetRootEntry(NetRoot); PNAME_CACHE_CONTROL NameCacheCtl = &pNetRootEntry->NameCacheCtlFNF;
BOOLEAN CacheFound = FALSE;
PAGED_CODE();
// If a file does not exist, its stream doesn't either
MRxSmbIsStreamFile( OriginalFileName, &StreamlessName );
ExAcquireFastMutex(&MRxSmbFileInfoCacheLock);
NameCache = RxNameCacheFetchEntry(NameCacheCtl,&StreamlessName);
if (NameCache != NULL) { RX_NC_CHECK_STATUS NameCacheStatus; //
// Found it. Now check entry for not expired.
// Note - The NameCache entry has been pulled off the active list.
//
NameCacheStatus = RxNameCacheCheckEntry( NameCache, //MRxSmbStatistics.SmbsReceived.LowPart
NameCache->Context);
if ((NameCacheStatus == RX_NC_SUCCESS) && (NameCache->PriorStatus == STATUS_OBJECT_NAME_NOT_FOUND)) { //
// This is a match. Return the old status, file info and
// reactivate the entry but leave expiration time unchanged.
//
CacheFound = TRUE;
// put the entry back to the active list without changing the expire time
RxNameCacheActivateEntry(NameCacheCtl, NameCache, 0, 0); } else { // put the entry back to the expire list
RxNameCacheExpireEntry(NameCacheCtl, NameCache); } }
ExReleaseFastMutex(&MRxSmbFileInfoCacheLock);
return CacheFound; }
VOID MRxSmbCacheFileNotFound( PRX_CONTEXT RxContext ) /*++
Routine Description:
This routine creates the name cache entry for File Not Found.
Arguments:
RxContext - the RDBSS context
Return Value:
BOOLEAN - name cache found
--*/ { RxCaptureFcb; PUNICODE_STRING OriginalFileName = GET_ALREADY_PREFIXED_NAME_FROM_CONTEXT(RxContext);
PNAME_CACHE NameCache = NULL; PMRX_NET_ROOT NetRoot = capFcb->pNetRoot; PSMBCEDB_NET_ROOT_ENTRY pNetRootEntry = SmbCeGetAssociatedNetRootEntry(NetRoot); PNAME_CACHE_CONTROL NameCacheCtl = &pNetRootEntry->NameCacheCtlFNF;
PAGED_CODE();
// Never cache stream file opens
if( MRxSmbIsStreamFile( OriginalFileName, NULL ) ) { return; }
ExAcquireFastMutex(&MRxSmbFileInfoCacheLock);
NameCache = RxNameCacheFetchEntry(NameCacheCtl,OriginalFileName);
if (NameCache != NULL) { NameCache->PriorStatus = STATUS_OBJECT_NAME_NOT_FOUND; RxNameCacheActivateEntry( NameCacheCtl, NameCache, NAME_CACHE_OBJ_NAME_NOT_FOUND_LIFETIME, MRxSmbStatistics.SmbsReceived.LowPart); } else { if (FlagOn(NetRoot->Flags,NETROOT_FLAG_UNIQUE_FILE_NAME)) { NameCache = RxNameCacheCreateEntry ( NameCacheCtl, OriginalFileName, TRUE); // case insensitive match
if (NameCache != NULL) { NameCache->PriorStatus = STATUS_OBJECT_NAME_NOT_FOUND; RxNameCacheActivateEntry( NameCacheCtl, NameCache, NAME_CACHE_OBJ_NAME_NOT_FOUND_LIFETIME, MRxSmbStatistics.SmbsReceived.LowPart); } } }
ExReleaseFastMutex(&MRxSmbFileInfoCacheLock); }
#if 0
VOID MRxSmbCacheFileNotFoundFromQueryDirectory( PRX_CONTEXT RxContext ) /*++
Routine Description:
This routine creates the name cache entry for File Not Found.
Arguments:
RxContext - the RDBSS context
Return Value:
BOOLEAN - name cache found
--*/ { RxCaptureFcb; RxCaptureFobx; PUNICODE_STRING OriginalFileName = GET_ALREADY_PREFIXED_NAME_FROM_CONTEXT(RxContext); PUNICODE_STRING Template = &capFobx->UnicodeQueryTemplate; UNICODE_STRING FileName;
PNAME_CACHE NameCache = NULL; PMRX_NET_ROOT NetRoot = capFcb->pNetRoot; PSMBCEDB_NET_ROOT_ENTRY pNetRootEntry = SmbCeGetAssociatedNetRootEntry(NetRoot); PNAME_CACHE_CONTROL NameCacheCtl = &pNetRootEntry->NameCacheCtlFNF;
PAGED_CODE();
ExAcquireFastMutex(&MRxSmbFileInfoCacheLock);
NameCache = RxNameCacheFetchEntry(NameCacheCtl,&FileName);
if (NameCache != NULL) { if ((NameCache == NULL) && (OriginalFileName->Length > sizeof(WCHAR))) { //
// Do lookup now since we may have skipped it at entry.
//
NameCache = RxNameCacheFetchEntry(NameCacheCtl,&FileName); if (NameCache == NULL) { NameCache = RxNameCacheCreateEntry ( NameCacheCtl, OriginalFileName, TRUE); // case insensitive match
} } if (NameCache != NULL) { NameCache->PriorStatus = STATUS_OBJECT_NAME_NOT_FOUND; RxNameCacheActivateEntry( NameCacheCtl, NameCache, NAME_CACHE_OBJ_NAME_NOT_FOUND_LIFETIME, MRxSmbStatistics.SmbsReceived.LowPart); } }
ExReleaseFastMutex(&MRxSmbFileInfoCacheLock); } #endif
VOID MRxSmbInvalidateFileNotFoundCache( PRX_CONTEXT RxContext ) /*++
Routine Description:
This routine invalidates the name cache entry as File Not Found.
Arguments:
RxContext - the RDBSS context
Return Value:
BOOLEAN - name cache found
--*/ { RxCaptureFcb; PUNICODE_STRING OriginalFileName = GET_ALREADY_PREFIXED_NAME_FROM_CONTEXT(RxContext); UNICODE_STRING StreamlessName;
PNAME_CACHE NameCache = NULL; PMRX_NET_ROOT NetRoot = capFcb->pNetRoot; PSMBCEDB_NET_ROOT_ENTRY pNetRootEntry = SmbCeGetAssociatedNetRootEntry(NetRoot); PNAME_CACHE_CONTROL NameCacheCtl = &pNetRootEntry->NameCacheCtlFNF;
PAGED_CODE();
// If we invalidate a stream, this invalidates the associated file entry
MRxSmbIsStreamFile( OriginalFileName, &StreamlessName );
ExAcquireFastMutex(&MRxSmbFileInfoCacheLock);
NameCache = RxNameCacheFetchEntry(NameCacheCtl, &StreamlessName);
if (NameCache != NULL) { RxNameCacheExpireEntry(NameCacheCtl, NameCache); }
ExReleaseFastMutex(&MRxSmbFileInfoCacheLock); }
VOID MRxSmbInvalidateFileNotFoundCacheForRename( PRX_CONTEXT RxContext ) /*++
Routine Description:
This routine invalidates the name cache entry as File Not Found.
Arguments:
RxContext - the RDBSS context
Return Value:
BOOLEAN - name cache found
--*/ { RxCaptureFcb; UNICODE_STRING RenameName; UNICODE_STRING StreamlessName;
PNAME_CACHE NameCache = NULL; PMRX_NET_ROOT NetRoot = capFcb->pNetRoot; PSMBCEDB_NET_ROOT_ENTRY pNetRootEntry = SmbCeGetAssociatedNetRootEntry(NetRoot); PNAME_CACHE_CONTROL NameCacheCtl = &pNetRootEntry->NameCacheCtlFNF;
PFILE_RENAME_INFORMATION RenameInformation = RxContext->Info.Buffer;
RenameName.Buffer = &RenameInformation->FileName[0]; RenameName.Length = (USHORT)RenameInformation->FileNameLength;
//DbgPrint("Invalidate FNF cache %wZ\n", &RenameName);
PAGED_CODE();
// If we rename a stream, invalidate the name without the stream
MRxSmbIsStreamFile( &RenameName, &StreamlessName );
ExAcquireFastMutex(&MRxSmbFileInfoCacheLock);
NameCache = RxNameCacheFetchEntry(NameCacheCtl,&StreamlessName);
if (NameCache != NULL) { RxNameCacheExpireEntry(NameCacheCtl, NameCache); }
ExReleaseFastMutex(&MRxSmbFileInfoCacheLock); }
BOOLEAN MRxSmbIsStreamFile( PUNICODE_STRING FileName, PUNICODE_STRING AdjustFileName ) /*++
Routine Description:
This routine checks if it is a stream file and return the root file name if true.
Arguments:
FileName - the file name needs to be parsed AdjustFileName - the file name contains only root name of the stream
Return Value:
BOOLEAN - stream file
--*/ { USHORT i; BOOLEAN IsStream = FALSE; NTSTATUS Status = STATUS_SUCCESS;
for (i=0;i<FileName->Length/sizeof(WCHAR);i++) { if (FileName->Buffer[i] == L':') { IsStream = TRUE; break; } }
if (AdjustFileName != NULL) { if (IsStream) { AdjustFileName->Length = AdjustFileName->MaximumLength = i * sizeof(WCHAR); AdjustFileName->Buffer = FileName->Buffer; } else { AdjustFileName->Length = AdjustFileName->MaximumLength = FileName->Length; AdjustFileName->Buffer = FileName->Buffer; } }
return IsStream; }
BOOLEAN EnableInfoCache = TRUE;
BOOLEAN MRxSmbIsLongFileName( PRX_CONTEXT RxContext ) /*++
Routine Description:
This routine checks if it is a short file name and return the first part of short name if true.
Arguments:
FileName - the file name needs to be parsed AdjustFileName - the file name contains only root name of the stream
Return Value:
BOOLEAN - stream file
--*/ { RxCaptureFcb; RxCaptureFobx; PMRX_SMB_FCB smbFcb = MRxSmbGetFcbExtension(capFcb); BOOLEAN IsLongName = FALSE;
if (!EnableInfoCache) { return TRUE; }
if (FlagOn(smbFcb->MFlags, SMB_FCB_FLAG_LONG_FILE_NAME)) { IsLongName = TRUE; } else { USHORT i; USHORT Left = 0; USHORT Right = 0; OEM_STRING OemString; BOOLEAN RightPart = FALSE; WCHAR LastChar = 0; WCHAR CurrentChar = 0; PUNICODE_STRING FileName = GET_ALREADY_PREFIXED_NAME_FROM_CONTEXT(RxContext); PSMBCE_V_NET_ROOT_CONTEXT pVNetRootContext = NULL; PSMBCE_NET_ROOT pSmbNetRoot = NULL;
if (RxContext->MajorFunction == IRP_MJ_CREATE) { pVNetRootContext = RxContext->Create.pVNetRoot->Context; pSmbNetRoot = &pVNetRootContext->pNetRootEntry->NetRoot; } else { ASSERT(capFobx != NULL); pVNetRootContext = (PSMBCE_V_NET_ROOT_CONTEXT)capFobx->pSrvOpen->pVNetRoot->Context; pSmbNetRoot = &pVNetRootContext->pNetRootEntry->NetRoot; }
for (i=0;i<FileName->Length/sizeof(WCHAR);i++) { LastChar = CurrentChar; CurrentChar = FileName->Buffer[i];
if (CurrentChar == L'\\') { RightPart = FALSE; Left = 0; Right = 0; continue; }
if (CurrentChar == L'.') { if (RightPart) { IsLongName = TRUE; break; } else { RightPart = TRUE; Right = 0; continue; } }
if (CurrentChar >= L'0' && CurrentChar <= L'9' || CurrentChar >= L'a' && CurrentChar <= L'z' || CurrentChar >= L'A' && CurrentChar <= L'Z' || CurrentChar == L'~' || CurrentChar == L'_' || CurrentChar == L'$' || CurrentChar == L'@') { if (RightPart) { if (++Right > 3) { IsLongName = TRUE; break; } } else { if (++Left > 8) { IsLongName = TRUE; break; } }
if (pSmbNetRoot->NetRootFileSystem != NET_ROOT_FILESYSTEM_NTFS) { if (CurrentChar >= L'A' && CurrentChar <= L'Z' && LastChar >= L'a' && LastChar <= L'z' || CurrentChar >= L'a' && CurrentChar <= L'z' && LastChar >= L'A' && LastChar <= L'Z') { // On FAT volume, name with mixture of cases will be treated as long name
IsLongName = TRUE; break; } } } else { // if not, an alternate name may be created by the server which will
// be different from this name.
IsLongName = TRUE; break; } } }
if (IsLongName) { SetFlag(smbFcb->MFlags, SMB_FCB_FLAG_LONG_FILE_NAME); }
return IsLongName;
}
VOID MRxSmbCreateSuffix(PUNICODE_STRING Source, PUNICODE_STRING Target) /*++
Routine Description:
This routine creates 'cat.dog' on input of 'a\b\cat.dog'
***Note that Target and Source share a buffer after completion, so ***changing one will change the other
Arguments:
Source, Target are UNICODE Strings
Return Value:
VOID
--*/ { ULONG i; PWCH BaseFileName = Source->Buffer;
PAGED_CODE( );
for ( i = 0; i < Source->Length / sizeof(WCHAR); i++ ) {
//
// If s points to a directory separator, set BaseFileName to
// the character after the separator.
//
if ( Source->Buffer[i] == ((WCHAR)L'\\') ) { BaseFileName = &Source->Buffer[i]; } }
Target->Length = Source->Length - ((BaseFileName - Source->Buffer + 1) * sizeof(WCHAR)); Target->MaximumLength = Target->Length;
if (Target->Length) { Target->Buffer = BaseFileName + 1; } else { Target->Buffer = NULL; }
return; }
VOID MRxSmbCreateParentDirPrefix(PUNICODE_STRING Source, PUNICODE_STRING Target) /*++
Routine Description:
This routine creates 'a\b' on input of 'a\b\cat.dog' ***Note that Target and Source share a buffer after completion, so ***changing one will change the other
Arguments:
Source, Target are UNICODE Strings
Return Value:
VOID
--*/ { ULONG i; PWCH BaseFileName = Source->Buffer;
PAGED_CODE();
for ( i = 0; i < Source->Length / sizeof(WCHAR); i++ ) {
//
// If s points to a directory separator, set BaseFileName to
// the character after the separator.
//
if ( Source->Buffer[i] == ((WCHAR)L'\\') ) { BaseFileName = &Source->Buffer[i]; } }
Target->Length = ((BaseFileName - Source->Buffer) * sizeof(WCHAR)); Target->MaximumLength = Target->Length;
Target->Buffer = Source->Buffer;
return; }
VOID MRxSmbCacheFullDirectory( PRX_CONTEXT RxContext, PVOID Contents, ULONG Length, PMRX_SMB_FOBX smbFobx ) /*++
Routine Description:
This routine creates the name cache entry for Partial Directory.
Arguments:
RxContext - the RDBSS context
Return Value:
VOID
--*/ { RxCaptureFcb; PUNICODE_STRING OriginalFileName = GET_ALREADY_PREFIXED_NAME_FROM_CONTEXT(RxContext); PNAME_CACHE NameCache = NULL; RX_NC_CHECK_STATUS NameCacheStatus; PMRX_NET_ROOT NetRoot = capFcb->pNetRoot; PSMBCEDB_NET_ROOT_ENTRY pNetRootEntry = SmbCeGetAssociatedNetRootEntry(NetRoot); PNAME_CACHE_CONTROL NameCacheCtl = &pNetRootEntry->NameCacheCtlPartialDir; PMRX_SMB_FCB smbFcb = MRxSmbGetFcbExtension(capFcb); #if DBG
UNICODE_STRING smbtemplate = {0,0,NULL}; #endif
PAGED_CODE();
if (Length > NAME_CACHE_PARTIAL_DIR_BUFFER_SIZE) {
return;
}
ExAcquireFastMutex(&MRxSmbFileInfoCacheLock);
NameCache = RxNameCacheFetchEntry(NameCacheCtl,OriginalFileName);
// DbgPrint("NameCacheCtl %x\n", (ULONG) NameCacheCtl);
if (NameCache == NULL) {
if (FlagOn(NetRoot->Flags,NETROOT_FLAG_UNIQUE_FILE_NAME)) { NameCache = RxNameCacheCreateEntry (NameCacheCtl, OriginalFileName, TRUE); // case insensitive match
} } else {
RxDbgTrace( 0, Dbg, ("Cache Found, Reactivating... :%wZ: size %ld\n",OriginalFileName,Length)); SmbLog(LOG,MRxSmbReactivatingCache, LOGUSTR(*OriginalFileName) LOGULONG(Length));
//
// Found it. Now check entry for not expired.
// Note - The NameCache entry has been pulled off the active list.
//
NameCacheStatus = RxNameCacheCheckEntry(NameCache, NameCache->Context);
// The assert below could be false if there are multiple threads trying to cache this directory.
// ASSERT( NameCacheStatus != RX_NC_SUCCESS);
// Clean out the Buffers
RtlZeroMemory(NameCache->ContextExtension, sizeof(FULL_DIR_CACHE)); }
if (NameCache != NULL) {
PFULL_DIR_CACHE Cache = (PFULL_DIR_CACHE)NameCache->ContextExtension; ULONG SidLength;
NameCache->PriorStatus = STATUS_SUCCESS;
Cache->CharInvalidates = 0; Cache->Flags = 0;
Cache->CharFlags = 0;
Cache->NiBufferLength = NAME_CACHE_PARTIAL_DIR_BUFFER_SIZE;
RxDbgTrace( 0, Dbg, ("Cached :%wZ: StrLen %d, BufA %x, Buf0 %x, Contents %x, Length %d\n",OriginalFileName, OriginalFileName->Length, &(Cache->Buffer[0]), Cache->Buffer[0], Contents, Length)); SmbLog(LOG,MRxSmbCached, LOGUSTR(*OriginalFileName) LOGXSHORT(OriginalFileName->Length) //LOGPTR(&(Cache->Buffer[0])));
//LOGPTR(Cache->Buffer[0])
LOGPTR(Contents) LOGULONG(Length));
RtlCopyMemory(&(Cache->smbFobx), // dest
smbFobx, // source
sizeof (MRX_SMB_FOBX));
// Now setup our smbFobx's UnalignedBuffer to Buffer
// which has the entire server response
// Assert the following.
Cache->smbFobx.Enumeration.UnalignedDirEntrySideBuffer = NULL;
//
// Put the SID in the Cache entry
//
SidLength = SeLengthSid(&smbFcb->Sid); RtlCopySid(SidLength,&Cache->Sid,&smbFcb->Sid);
// Note: All we are saving is the Srv Response bytes.
// The Cached smbFobx is never visible to anybody else
// So, we cache using the built in buffer.
// Also, note that NameCache doesn't support registering
// free function for Expiration, which forces us to use
// the static buffer. The following will leak paged pool.
RtlCopyMemory( &(Cache->Buffer[0]), //dst
smbFobx->Enumeration.UnalignedDirEntrySideBuffer, //Src
Length);
// DbgPrint("Name :%wZ: StrLen %d, CI %d, Buf0 %ld Hash %d\n", (PUNICODE_STRING) &(NameCache->Name), NameCache->Name.Length, (ULONG) NameCache->CaseInsensitive, (ULONG) *((PBYTE)(NameCache->ContextExtension) + sizeof (MRX_SMB_FOBX)), NameCache->HashValue);
{ BOOLEAN ReturnSingleEntry = FALSE; ULONG FileIndex = 0; NTSTATUS Status;
// Now it is time to get Names-Information for
// PartialDir searches. Use the same SmbFobx, but make
// sure we get the entire directory.
if (RxContext->QueryDirectory.ReturnSingleEntry) { ReturnSingleEntry = TRUE; RxContext->QueryDirectory.ReturnSingleEntry = FALSE; }
if (RxContext->QueryDirectory.FileIndex != 0) { FileIndex = RxContext->QueryDirectory.FileIndex; RxContext->QueryDirectory.FileIndex = 0; }
ASSERT ((Cache->smbFobx).Enumeration.UnalignedDirEntrySideBuffer == NULL);
MRxSmbAllocateSideBuffer(RxContext, &(Cache->smbFobx), TRANS2_FIND_FIRST2, &smbtemplate);
if (Cache->smbFobx.Enumeration.UnalignedDirEntrySideBuffer == NULL) { RxNameCacheExpireEntry(NameCacheCtl, NameCache); goto FINALLY; }
RtlCopyMemory( (Cache->smbFobx).Enumeration.UnalignedDirEntrySideBuffer,//dst
smbFobx->Enumeration.UnalignedDirEntrySideBuffer, //Src
Length);
// Setup the rest of our smbFobx
ClearFlag(Cache->smbFobx.Enumeration.Flags,SMBFOBX_ENUMFLAG_SEARCH_HANDLE_OPEN); Cache->smbFobx.Enumeration.ResumeInfo = NULL;
Cache->smbFobx.Enumeration.EndOfSearchReached = TRUE; Cache->smbFobx.Enumeration.IsUnicode = TRUE; Cache->smbFobx.Enumeration.IsNonNtT2Find = FALSE;
Cache->smbFobx.Enumeration.FilesReturned = smbFobx->Enumeration.FilesReturned;
Cache->smbFobx.Enumeration.EntryOffset = 0;
Cache->smbFobx.Enumeration.TotalDataBytesReturned = smbFobx->Enumeration.TotalDataBytesReturned;
Cache->smbFobx.Enumeration.Flags &= ~SMBFOBX_ENUMFLAG_SEARCH_NOT_THE_FIRST;
Cache->smbFobx.Enumeration.ErrorStatus = RX_MAP_STATUS(SUCCESS);
// Setup for Names Information
// Cache->smbFobx.Enumeration.FileNameOffset =
// (USHORT)FIELD_OFFSET(FILE_NAMES_INFORMATION,FileName[0]);
// Cache->smbFobx.Enumeration.FileNameLengthOffset =
// (USHORT)FIELD_OFFSET(FILE_NAMES_INFORMATION,FileNameLength);
Cache->smbFobx.Enumeration.WildCardsFound = smbFobx->Enumeration.WildCardsFound;
Status = MrxSmbUnalignedDirEntryCopyTail( RxContext, FileBothDirectoryInformation, &(Cache->NiBuffer[0]), &(Cache->NiBufferLength), &(Cache->smbFobx) );
ASSERT (Status == STATUS_SUCCESS); ASSERT ((Cache->smbFobx).Enumeration.UnalignedDirEntrySideBuffer == NULL);
// The FindFirst must have gotten everything.
// Now, we reset the smbFobx to its original values
RtlCopyMemory(&(Cache->smbFobx), // dest
smbFobx, // source
sizeof (MRX_SMB_FOBX));
// Reset the RxContext to the User's spec
if (ReturnSingleEntry) { RxContext->QueryDirectory.ReturnSingleEntry = TRUE; }
if (FileIndex != 0) { RxContext->QueryDirectory.FileIndex = FileIndex; }
// Since we copied over the users's smbFobx again, reset this
// again. We won't use the side buffer to copy the srv's ff response buffer.
Cache->smbFobx.Enumeration.UnalignedDirEntrySideBuffer = NULL; }
RxNameCacheActivateEntry( NameCacheCtl, NameCache, DIR_CACHE_LIFE_TIME, MRxSmbStatistics.SmbsReceived.LowPart);
RxDbgTrace( 0, Dbg, ("Cached Full Dir :%wZ: size %ld\n",OriginalFileName,Length)); SmbLog(LOG,MRxSmbCachedFullDir, LOGUSTR(*OriginalFileName) LOGULONG(Length));
} FINALLY:
ExReleaseFastMutex(&MRxSmbFileInfoCacheLock);
return; }
VOID MRxSmbInvalidateFullDirectoryCache( PRX_CONTEXT RxContext ) /*++
Routine Description:
This routine invalidates Partial Directory Cache
Arguments:
RxContext - the RDBSS context
Return Value:
VOID
--*/ { RxCaptureFcb; PUNICODE_STRING OriginalFileName = GET_ALREADY_PREFIXED_NAME_FROM_CONTEXT(RxContext); PNAME_CACHE NameCache = NULL; PMRX_NET_ROOT NetRoot = capFcb->pNetRoot; PSMBCEDB_NET_ROOT_ENTRY pNetRootEntry = SmbCeGetAssociatedNetRootEntry(NetRoot); PNAME_CACHE_CONTROL NameCacheCtl = &pNetRootEntry->NameCacheCtlPartialDir;
PAGED_CODE();
ExAcquireFastMutex(&MRxSmbFileInfoCacheLock);
NameCache = RxNameCacheFetchEntry(NameCacheCtl,OriginalFileName);
if (NameCache != NULL) { //
// put the entry back to the expire list
//
RxDbgTrace( 0, Dbg, ( "Invalidate Full Dir :%wZ: \n", OriginalFileName)); SmbLog(LOG,MRxSmbInvalidateFullDir, LOGUSTR(*OriginalFileName));
RxNameCacheExpireEntry(NameCacheCtl, NameCache); }
ExReleaseFastMutex(&MRxSmbFileInfoCacheLock);
return; }
BOOLEAN MRxSmbIsFullDirectoryCached( PRX_CONTEXT RxContext, PVOID Buffer, PULONG Length, PMRX_SMB_FOBX smbFobx, NTSTATUS *Status ) /*++
Routine Description:
This routine checks if the name cache entry exists as Partial Directory
Arguments:
RxContext - the RDBSS context
Return Value:
BOOLEAN - name cache found
--*/ { RxCaptureFcb; PUNICODE_STRING OriginalFileName = GET_ALREADY_PREFIXED_NAME_FROM_CONTEXT(RxContext);
PNAME_CACHE NameCache = NULL; RX_NC_CHECK_STATUS NameCacheStatus;
PMRX_NET_ROOT NetRoot = capFcb->pNetRoot; PSMBCEDB_NET_ROOT_ENTRY pNetRootEntry = SmbCeGetAssociatedNetRootEntry(NetRoot); PNAME_CACHE_CONTROL NameCacheCtl = &pNetRootEntry->NameCacheCtlPartialDir;
BOOLEAN CacheFound = FALSE; BOOLEAN Expired = FALSE; PLIST_ENTRY pListEntry; PMRX_SMB_FCB smbFcb = MRxSmbGetFcbExtension(capFcb);
#if DBG
UNICODE_STRING smbtemplate = {0,0,NULL}; #endif
PAGED_CODE();
ExAcquireFastMutex(&MRxSmbFileInfoCacheLock);
NameCache = RxNameCacheFetchEntry(NameCacheCtl,OriginalFileName);
if (NameCache != NULL) {
//
// Found it. Now check entry for not expired.
// Note - The NameCache entry has been pulled off the active list.
//
NameCacheStatus = RxNameCacheCheckEntry(NameCache, NameCache->Context);
if (NameCacheStatus == RX_NC_SUCCESS) {
PFULL_DIR_CACHE Cache = (PFULL_DIR_CACHE)NameCache->ContextExtension; //
// Verify matching SIDs
//
if(RtlEqualSid(&smbFcb->Sid,&Cache->Sid)) { if (!(Cache->Flags & FLAG_FDC_NAMES_INFO_ONLY)) {
//
// This is a match. Return the old status, file info and
// reactivate the entry but leave expiration time unchanged.
//
RxDbgTrace( 0, Dbg, ("Found :%wZ: in FullDirCache\n",OriginalFileName)); SmbLog(LOG,MRxSmbFoundInFDC, LOGUSTR(*OriginalFileName));
CacheFound = TRUE;
// Setup the Fobx correctly (to look like just after
// SmbCeTransact
// Mark the SmbFobx as satisfied from Full Dir Cache
smbFobx->Enumeration.Flags |= SMBFOBX_ENUMFLAG_FULL_DIR_CACHE;
// Allocate SideBuffer
// Now setup our smbFobx's UnalignedBuffer to Buffer
// which has the entire response
ASSERT (smbFobx->Enumeration.UnalignedDirEntrySideBuffer == NULL);
MRxSmbAllocateSideBuffer(RxContext, smbFobx, TRANS2_FIND_FIRST2, &smbtemplate); ASSERT (smbFobx->Enumeration.UnalignedDirEntrySideBuffer != NULL); if (smbFobx->Enumeration.UnalignedDirEntrySideBuffer == NULL) { RxNameCacheExpireEntry(NameCacheCtl, NameCache); NameCache = NULL; CacheFound = FALSE; goto FINALLY; } RtlCopyMemory( smbFobx->Enumeration.UnalignedDirEntrySideBuffer, //Dst
&(Cache->Buffer[0]), //Src
Cache->smbFobx.Enumeration.TotalDataBytesReturned);
// This handle is bogus, but the handle has been closed
// since we hit EndOfSearchReached
smbFobx->Enumeration.SearchHandle = Cache->smbFobx.Enumeration.SearchHandle;
// Doesn't make sense, since EndOfSearchReached is TRUE
// and Handle has been closed.
smbFobx->Enumeration.Version = Cache->smbFobx.Enumeration.Version;
ClearFlag(smbFobx->Enumeration.Flags,SMBFOBX_ENUMFLAG_SEARCH_HANDLE_OPEN); smbFobx->Enumeration.ResumeInfo = NULL;
smbFobx->Enumeration.EndOfSearchReached = TRUE; smbFobx->Enumeration.IsUnicode = TRUE; smbFobx->Enumeration.IsNonNtT2Find = FALSE;
smbFobx->Enumeration.FilesReturned = Cache->smbFobx.Enumeration.FilesReturned; smbFobx->Enumeration.EntryOffset = Cache->smbFobx.Enumeration.EntryOffset;
ASSERT (smbFobx->Enumeration.EntryOffset == 0);
smbFobx->Enumeration.TotalDataBytesReturned = Cache->smbFobx.Enumeration.TotalDataBytesReturned;
smbFobx->Enumeration.Flags |= SMBFOBX_ENUMFLAG_SEARCH_NOT_THE_FIRST;
smbFobx->Enumeration.ErrorStatus = RX_MAP_STATUS(SUCCESS); // FileNameOffset and FileNameLengthOffset depends on
// the FileInformationClass, that must come with the
// smbFobx already. Don't touch.
smbFobx->Enumeration.WildCardsFound = Cache->smbFobx.Enumeration.WildCardsFound;
RxNameCacheActivateEntry(NameCacheCtl, NameCache, 0, 0);
// *Length -= Cache->smbFobx.Enumeration.TotalDataBytesReturned;
} else {
// Expire this entry and let the user go to the server.
// On the way back, we'll cache the fresh dir bdi anyways.
CacheFound = FALSE; Expired = TRUE;
RxNameCacheExpireEntry(NameCacheCtl, NameCache); }
} else { // SIDs did not match so expire the entry
Expired = TRUE; RxNameCacheExpireEntry(NameCacheCtl, NameCache); } } else { // Expire it!
Expired = TRUE; RxNameCacheExpireEntry(NameCacheCtl, NameCache); } }
FINALLY:
ExReleaseFastMutex(&MRxSmbFileInfoCacheLock);
if ((NameCache != NULL) && (NameCacheStatus == RX_NC_SUCCESS) && !(Expired)) { *Status = MrxSmbUnalignedDirEntryCopyTail(RxContext, FileBothDirectoryInformation, Buffer, Length, smbFobx); if (smbFobx->Enumeration.UnalignedDirEntrySideBuffer == NULL) {
// The FindFirst got everything.
// Need to fail the FindNext. Mark satisfied out of cache.
SetFlag(smbFobx->Enumeration.Flags,SMBFOBX_ENUMFLAG_READ_FROM_CACHE); } } return CacheFound; }
VOID MRxSmbInvalidateFullDirectoryCacheParent(PRX_CONTEXT RxContext, BOOLEAN Benign) /*++
Routine Description:
This routine invalidates Partial Directory Cache
Arguments:
RxContext - the RDBSS context
Return Value:
VOID
--*/ { RxCaptureFcb; PUNICODE_STRING OriginalFileName = GET_ALREADY_PREFIXED_NAME_FROM_CONTEXT(RxContext); UNICODE_STRING ParentDir; UNICODE_STRING FileNameSuffix; WCHAR InhibitChar; ULONG InhibitMask;
PNAME_CACHE NameCache = NULL; RX_NC_CHECK_STATUS NameCacheStatus; PMRX_NET_ROOT NetRoot = capFcb->pNetRoot; PSMBCEDB_NET_ROOT_ENTRY pNetRootEntry = SmbCeGetAssociatedNetRootEntry(NetRoot); PNAME_CACHE_CONTROL NameCacheCtl = &pNetRootEntry->NameCacheCtlPartialDir; BOOLEAN ExpireEntry;
PAGED_CODE(); MRxSmbCreateParentDirPrefix(OriginalFileName, &ParentDir);
if(ParentDir.Length == 0) { return; }
ExAcquireFastMutex(&MRxSmbFileInfoCacheLock);
NameCache = RxNameCacheFetchEntry(NameCacheCtl, &ParentDir);
if (NameCache != NULL) { //
// put the entry back to the expire list
//
// Found it. Now check entry for not expired.
// Note - The NameCache entry has been pulled off the active list.
//
NameCacheStatus = RxNameCacheCheckEntry(NameCache, NameCache->Context);
if (NameCacheStatus == RX_NC_SUCCESS) {
BOOLEAN CacheValid = TRUE; PFULL_DIR_CACHE Cache = NameCache->ContextExtension;
// Mark as non-bdi
Cache->Flags |= FLAG_FDC_NAMES_INFO_ONLY;
if (!(Benign) && !(MRxSmbIsFileInPartialDirectoryCache(NameCache,OriginalFileName,&CacheValid, NULL))) {
// Search for that one character etc. and disect cache
MRxSmbCreateSuffix(OriginalFileName, &FileNameSuffix);
InhibitChar = RtlUpcaseUnicodeChar(FileNameSuffix.Buffer[0]);
InhibitMask = 0;
if ((InhibitChar >= L'A') && (InhibitChar <= L'Z')) {
InhibitMask = (1 << ((USHORT) InhibitChar - (USHORT)(L'A'))); } else { if ((InhibitChar >= L'0') && (InhibitChar <= L'9')) { InhibitMask = 1 << 30; } else { switch (InhibitChar) { case L'~': InhibitMask = 1 << 26; break; case L'_': InhibitMask = 1 << 27; break; case L'$': InhibitMask = 1 << 28; break; case L'@': InhibitMask = 1 << 29; break; default : InhibitMask = 1 << 31; break; // We didn't find a suitable character to invalidate,
// We use the 'rest' bit.
// We HAVE TO remember that a modification has been made:
// like a file-create or delete that we couldn't record in the
// first 31 bits. This is so that, if a query comes along for
// 'test.dat, starting with an apostrophe char (or some Japanese
// Unicode Char, for that matter), and that file was created
// just before, we don't say STATUS_OBJECT_NOT_FOUND.
// We will lookup the 'rest' bit and pass the query on to the server}
} } }
if (!(Cache->CharFlags & InhibitMask)) { Cache->CharFlags |= InhibitMask; Cache->CharInvalidates++;
RxDbgTrace( 0, Dbg, ( "Inv Parent Cache :%wZ:%wZ: %x %x\n", OriginalFileName, &ParentDir,Cache->CharFlags,InhibitMask )); SmbLog(LOG,MRxSmbInvParentCache, LOGUSTR(*OriginalFileName) LOGUSTR(ParentDir) LOGULONG(Cache->CharFlags) LOGULONG(InhibitMask));
}
}
if (Cache->CharInvalidates > MAX_CHAR_INVALIDATES_FULL_DIR) {
RxDbgTrace( 0, Dbg, ("Expire Cache %x Num Inv %d\n", Cache->CharFlags, Cache->CharInvalidates)); SmbLog(LOG,MRxSmbExpireCache, LOGULONG(Cache->CharFlags) LOGXSHORT(Cache->CharInvalidates));
ExpireEntry = TRUE; } else { ExpireEntry = FALSE; }
} else { ExpireEntry = TRUE; }
if(ExpireEntry) { RxNameCacheExpireEntry(NameCacheCtl,NameCache); } else { RxNameCacheActivateEntry(NameCacheCtl,NameCache,0,0); }
}
ExReleaseFastMutex(&MRxSmbFileInfoCacheLock);
return; }
VOID MRxSmbInvalidateFullDirectoryCacheParentForRename( PRX_CONTEXT RxContext, BOOLEAN Benign ) /*++
Routine Description:
This routine invalidates Partial Directory Cache
Arguments:
RxContext - the RDBSS context
Return Value:
VOID
--*/ { RxCaptureFcb; PUNICODE_STRING OriginalFileName = GET_ALREADY_PREFIXED_NAME_FROM_CONTEXT(RxContext); UNICODE_STRING ParentDir; UNICODE_STRING FileNameSuffix; UNICODE_STRING RenameName = {0,0,NULL}; WCHAR InhibitChar; ULONG InhibitMask;
PNAME_CACHE NameCache = NULL; RX_NC_CHECK_STATUS NameCacheStatus; PMRX_NET_ROOT NetRoot = capFcb->pNetRoot; PSMBCEDB_NET_ROOT_ENTRY pNetRootEntry = SmbCeGetAssociatedNetRootEntry(NetRoot); PNAME_CACHE_CONTROL NameCacheCtl = &pNetRootEntry->NameCacheCtlPartialDir; BOOLEAN ExpireEntry;
PFILE_RENAME_INFORMATION RenameInformation = RxContext->Info.Buffer;
PAGED_CODE();
RenameName.Buffer = &RenameInformation->FileName[0]; RenameName.Length = (USHORT)RenameInformation->FileNameLength;
MRxSmbCreateParentDirPrefix(&RenameName, &ParentDir);
if(ParentDir.Length == 0) { return; }
ExAcquireFastMutex(&MRxSmbFileInfoCacheLock);
NameCache = RxNameCacheFetchEntry(NameCacheCtl, &ParentDir);
if (NameCache != NULL) { //
// put the entry back to the expire list
//
// Found it. Now check entry for not expired.
// Note - The NameCache entry has been pulled off the active list.
//
NameCacheStatus = RxNameCacheCheckEntry(NameCache, NameCache->Context);
if (NameCacheStatus == RX_NC_SUCCESS) {
BOOLEAN CacheValid = TRUE; PFULL_DIR_CACHE Cache = NameCache->ContextExtension;
// Mark as non-bdi
Cache->Flags |= FLAG_FDC_NAMES_INFO_ONLY;
if (!(Benign) && !(MRxSmbIsFileInPartialDirectoryCache(NameCache,&RenameName,&CacheValid, NULL))) {
// Search for that one character etc. and disect cache
MRxSmbCreateSuffix(&RenameName, &FileNameSuffix);
InhibitChar = RtlUpcaseUnicodeChar(FileNameSuffix.Buffer[0]);
InhibitMask = 0;
if ((InhibitChar >= L'A') && (InhibitChar <= L'Z')) {
InhibitMask = (1 << ((USHORT) InhibitChar - (USHORT)(L'A'))); } else { if ((InhibitChar >= L'0') && (InhibitChar <= L'9')) { InhibitMask = 1 << 30; } else { switch (InhibitChar) { case L'~': InhibitMask = 1 << 26; break; case L'_': InhibitMask = 1 << 27; break; case L'$': InhibitMask = 1 << 28; break; case L'@': InhibitMask = 1 << 29; break; default : InhibitMask = 1 << 31; break; // We didn't find a suitable character to invalidate,
// We use the 'rest' bit.
// We HAVE TO remember that a modification has been made:
// like a file-create or delete that we couldn't record in the
// first 31 bits. This is so that, if a query comes along for
// 'test.dat, starting with an apostrophe char (or some Japanese
// Unicode Char, for that matter), and that file was created
// just before, we don't say STATUS_OBJECT_NOT_FOUND.
// We will lookup the 'rest' bit and pass the query on to the server}
} } }
if (!(Cache->CharFlags & InhibitMask)) {
Cache->CharFlags |= InhibitMask; Cache->CharInvalidates++;
RxDbgTrace( 0, Dbg, ( "Inv Rename Parent Cache :%wZ:%wZ: %x %x\n", &RenameName, &ParentDir, Cache->CharFlags, InhibitMask )); SmbLog(LOG,MRxSmbInvalidateRenameParentCache, LOGUSTR(RenameName) LOGUSTR(ParentDir) LOGULONG(Cache->CharFlags) LOGULONG(InhibitMask));
}
}
if (Cache->CharInvalidates > MAX_CHAR_INVALIDATES_FULL_DIR) {
RxDbgTrace( 0, Dbg, ("Expire Cache %x Num Inv %d\n", Cache->CharFlags, Cache->CharInvalidates)); SmbLog(LOG,MRxSmbExpireCache, LOGULONG(Cache->CharFlags) LOGXSHORT(Cache->CharInvalidates));
ExpireEntry = TRUE; } else { ExpireEntry = FALSE; } } else { ExpireEntry = TRUE; }
if(ExpireEntry) { RxNameCacheExpireEntry(NameCacheCtl, NameCache); } else { RxNameCacheActivateEntry(NameCacheCtl, NameCache, 0, 0); } }
ExReleaseFastMutex(&MRxSmbFileInfoCacheLock);
return; }
BOOLEAN MRxSmbIsFileInFullDirectoryCache( PRX_CONTEXT RxContext, BOOLEAN *FileFound, PFILE_BASIC_INFORMATION pBuffer ) /*++
Routine Description:
This routine checks if the name cache entry exists as Partial Directory
Arguments:
RxContext - the RDBSS context
Return Value:
BOOLEAN - name cache found
--*/ { RxCaptureFcb; PUNICODE_STRING OriginalFileName = GET_ALREADY_PREFIXED_NAME_FROM_CONTEXT(RxContext); UNICODE_STRING TargetDirPrefix;
PNAME_CACHE NameCache = NULL; PMRX_NET_ROOT NetRoot = capFcb->pNetRoot; PSMBCEDB_NET_ROOT_ENTRY pNetRootEntry = SmbCeGetAssociatedNetRootEntry(NetRoot); PNAME_CACHE_CONTROL NameCacheCtl = &pNetRootEntry->NameCacheCtlPartialDir;
BOOLEAN CacheFound = FALSE;
PAGED_CODE();
*FileFound = FALSE;
MRxSmbCreateParentDirPrefix(OriginalFileName, &TargetDirPrefix);
if(TargetDirPrefix.Length == 0) { return CacheFound; }
ExAcquireFastMutex(&MRxSmbFileInfoCacheLock);
NameCache = RxNameCacheFetchEntry(NameCacheCtl,&TargetDirPrefix);
if (NameCache != NULL) {
RX_NC_CHECK_STATUS NameCacheStatus;
//
// Found it. Now check entry for not expired.
// Note - The NameCache entry has been pulled off the active list.
//
NameCacheStatus = RxNameCacheCheckEntry(NameCache, NameCache->Context);
if (NameCacheStatus == RX_NC_SUCCESS) { //
// This is a match. Return the old status, file info and
// reactivate the entry but leave expiration time unchanged.
//
CacheFound = TRUE; *FileFound = MRxSmbIsFileInPartialDirectoryCache(NameCache, OriginalFileName, &CacheFound, pBuffer);
if (*FileFound) { RxDbgTrace( 0, Dbg, ( " Found in FullDirCache :%wZ: :%wZ:\n", OriginalFileName, &TargetDirPrefix)); SmbLog(LOG,MRxSmbFoundInFDC2, LOGUSTR(*OriginalFileName) LOGUSTR(TargetDirPrefix)); }
RxNameCacheActivateEntry(NameCacheCtl, NameCache, 0, 0);
} else { RxNameCacheExpireEntry(NameCacheCtl, NameCache); } }
ExReleaseFastMutex(&MRxSmbFileInfoCacheLock);
return CacheFound; }
BOOLEAN MRxSmbIsFileInPartialDirectoryCache (PNAME_CACHE NameCache, PUNICODE_STRING OriginalFileName, PBOOLEAN CacheValid, PFILE_BASIC_INFORMATION pBuffer) /*++
Routine Description:
This routine checks if the FileName exists in name cache entry
Arguments:
NameCache - Partial Directory Cache
OriginalFileName - PUNICODE_STRING for searching
Return Value:
BOOLEAN - File exists in NameCache
--*/ { PFULL_DIR_CACHE Cache = (PFULL_DIR_CACHE) NameCache->ContextExtension;
FILE_BOTH_DIR_INFORMATION *pDirInfo = (PFILE_BOTH_DIR_INFORMATION) &(Cache->NiBuffer[0]);
UNICODE_STRING FileNameSuffix;
WCHAR InhibitChar; ULONG InhibitMask = 0;
MRxSmbCreateSuffix(OriginalFileName, &FileNameSuffix);
InhibitChar = RtlUpcaseUnicodeChar(FileNameSuffix.Buffer[0]);
InhibitMask = 0;
if ((InhibitChar >= L'A') && (InhibitChar <= L'Z')) {
InhibitMask = (1 << ((USHORT) InhibitChar - (USHORT)(L'A'))); } else { if ((InhibitChar >= L'0') && (InhibitChar <= L'9')) { InhibitMask = 1 << 30; } else { switch (InhibitChar) { case L'~': InhibitMask = 1 << 26; break; case L'_': InhibitMask = 1 << 27; break; case L'$': InhibitMask = 1 << 28; break; case L'@': InhibitMask = 1 << 29; break; default : InhibitMask = 1 << 31; break; // We didn't find a suitable character to invalidate,
// We use the 'rest' bit.
// We HAVE TO remember that a modification has been made:
// like a file-create or delete that we couldn't record in the
// first 31 bits. This is so that, if a query comes along for
// 'test.dat, starting with an apostrophe char (or some Japanese
// Unicode Char, for that matter), and that file was created
// just before, we don't say STATUS_OBJECT_NOT_FOUND.
// We will lookup the 'rest' bit and pass the query on to the server}
} } }
if (Cache->CharFlags & InhibitMask) { *CacheValid = FALSE;
RxDbgTrace( 0, Dbg, ("--------- Cache blown :%wZ: %x Inb Mask %x\n", OriginalFileName, Cache->CharFlags, InhibitMask)); SmbLog(LOG,MRxSmbCacheBlown, LOGUSTR(*OriginalFileName) LOGULONG(Cache->CharFlags) LOGULONG(InhibitMask));
return (FALSE); }
while( TRUE ) {
//DbgPrint( "Checking FileNameSuffix :%wZ: with Entry :%wZ: ",&FileNameSuffix,pDirInfo->FileName);
if ((pDirInfo->FileNameLength == FileNameSuffix.Length) && RtlEqualMemory( FileNameSuffix.Buffer, pDirInfo->FileName, FileNameSuffix.Length)) {
//DbgPrint (" TRUE\n");
// Pass on the Basic Attributes, latest or not.
// Two cases:
// Buffer exists: This is from a query
// In this case, if Creation Time is zero, then the Basic Attributes
// is invalid. So, we mark CacheValid as FALSE, which is returned to
// the caller's caller (in fileinfo.c).
// Note: This case DOES NOT assassinate character.
// Buffer was NULL: This is from Invalidate
// Simply set CreationTime to zero to mark File BasicInfo as invalid.
// However, return TRUE, so that Character assassination doesn't happen.
if (pBuffer != NULL) { if (0 != pDirInfo->CreationTime.QuadPart) { pBuffer->CreationTime = pDirInfo->CreationTime; pBuffer->LastAccessTime = pDirInfo->LastAccessTime; pBuffer->LastWriteTime = pDirInfo->LastWriteTime; pBuffer->ChangeTime = pDirInfo->ChangeTime; pBuffer->FileAttributes = pDirInfo->FileAttributes; } else { *CacheValid = FALSE; } } else { pDirInfo->CreationTime.QuadPart = 0; SmbLog(LOG,MRxSmbCacheBlown, LOGUSTR(*OriginalFileName) LOGULONG(Cache->CharFlags) LOGULONG(InhibitMask)); }
return (TRUE); }
if (0 == pDirInfo->NextEntryOffset) { break ; }
pDirInfo = (FILE_BOTH_DIR_INFORMATION *)((PBYTE)pDirInfo + pDirInfo->NextEntryOffset) ; //DbgPrint (" FALSE\n");
} return (FALSE); }
BOOLEAN MRxSmbNonTrivialFileName ( PRX_CONTEXT RxContext ) /*++
Routine Description:
This routine checks if the file name is non-trivial
Arguments:
RxContext
Return Value:
returns boolean
--*/ { PUNICODE_STRING OriginalFileName = GET_ALREADY_PREFIXED_NAME_FROM_CONTEXT(RxContext);
PAGED_CODE( );
return (OriginalFileName->Length > 0); }
|