able/windows-nt
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity
windows-nt/Source/XPSP1/NT/net/sfm/afp/server/chgntfy.c
CryptoAlgo Inc b3cac27891 Add source files
2020-09-26 16:20:57 +08:00

4809 lines
136 KiB
C
Raw Blame History

This file contains invisible Unicode characters

This file contains invisible Unicode characters that are indistinguishable to humans but may be processed differently by a computer. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

/*
Copyright (c) 1992 Microsoft Corporation
Module Name:
chgntfy.c
Abstract:
This module contains the code for processing change notifies.
Author:
Jameel Hyder (microsoft!jameelh)
Revision History:
15 Jun 1995 JameelH Seperated the change notify code from idindex.c
Notes: Tab stop: 4
--*/
#define IDINDEX_LOCALS
#define FILENUM FILE_CHGNTFY
#include <afp.h>
#include <scavengr.h>
#include <fdparm.h>
#include <pathmap.h>
#include <afpinfo.h>
#include <access.h> // for AfpWorldId
#ifdef ALLOC_PRAGMA
#pragma alloc_text(PAGE, afpVerifyDFE)
#pragma alloc_text(PAGE, afpAddDfEntryAndCacheInfo)
#pragma alloc_text(PAGE, afpReadIdDb)
#pragma alloc_text(PAGE, AfpProcessChangeNotify)
#pragma alloc_text(PAGE, afpProcessPrivateNotify)
#pragma alloc_text(PAGE, AfpQueuePrivateChangeNotify)
#pragma alloc_text(PAGE, AfpCacheDirectoryTree)
#pragma alloc_text(PAGE, AfpQueueOurChange)
#endif
/*** afpVerifyDFE
*
* Check if our view of this item in our data-base matches whats on disk. If not
* update our view with whats on disk.
*/
VOID
afpVerifyDFE(
IN struct _VolDesc * pVolDesc,
IN PDFENTRY pDfeParent,
IN PUNICODE_STRING pUName, // munged unicode name
IN PFILESYSHANDLE pfshParentDir, // open handle to parent directory
IN PFILE_BOTH_DIR_INFORMATION pFBDInfo, // from enumerate
IN PUNICODE_STRING pNotifyPath, // to filter out our own AFP_AfpInfo change notifies
IN PDFENTRY * ppDfEntry
)
{
PDFENTRY pDfEntry = *ppDfEntry;
if (pFBDInfo->LastWriteTime.QuadPart > pDfEntry->dfe_LastModTime.QuadPart)
{
FILESYSHANDLE fshAfpInfo, fshData;
AFPINFO AfpInfo;
DWORD crinfo, openoptions = 0;
BOOLEAN SeenComment, WriteBackROAttr = False;
PSTREAM_INFO pStreams = NULL, pCurStream;
NTSTATUS Status = STATUS_SUCCESS;
BOOLEAN IsDir;
// Our view is stale, update it
DBGPRINT(DBG_COMP_IDINDEX, DBG_LEVEL_INFO,
("afpVerifyDFE: Updating stale database with fresh info\n\t%Z\n", pNotifyPath));
IsDir = (pFBDInfo->FileAttributes & FILE_ATTRIBUTE_DIRECTORY) ? True : False;
ASSERT (IS_VOLUME_NTFS(pVolDesc));
ASSERT (!(DFE_IS_DIRECTORY(pDfEntry) ^ IsDir));
// Update DFE from FBDInfo first
pDfEntry->dfe_CreateTime = AfpConvertTimeToMacFormat(&pFBDInfo->CreationTime);
pDfEntry->dfe_LastModTime = pFBDInfo->LastWriteTime;
pDfEntry->dfe_NtAttr = (USHORT)(pFBDInfo->FileAttributes & FILE_ATTRIBUTE_VALID_FLAGS);
if (!IsDir)
{
pDfEntry->dfe_DataLen = pFBDInfo->EndOfFile.LowPart;
}
// Open/Create the AfpInfo stream
fshAfpInfo.fsh_FileHandle = NULL;
fshData.fsh_FileHandle = NULL;
do
{
// open or create the AfpInfo stream
if (!NT_SUCCESS(AfpCreateAfpInfoWithNodeName(pfshParentDir,
pUName,
pNotifyPath,
pVolDesc,
&fshAfpInfo,
&crinfo)))
{
if (!(pFBDInfo->FileAttributes & FILE_ATTRIBUTE_READONLY))
{
// What other reason is there that we could not open
// this stream except that this file/dir is readonly?
Status = STATUS_UNSUCCESSFUL;
break;
}
openoptions = IsDir ? FILEIO_OPEN_DIR : FILEIO_OPEN_FILE;
Status = STATUS_UNSUCCESSFUL; // Assume failure
if (NT_SUCCESS(AfpIoOpen(pfshParentDir,
AFP_STREAM_DATA,
openoptions,
pUName,
FILEIO_ACCESS_NONE,
FILEIO_DENY_NONE,
False,
&fshData)))
{
if (NT_SUCCESS(AfpExamineAndClearROAttr(&fshData,
&WriteBackROAttr,
pVolDesc,
pNotifyPath)))
{
if (NT_SUCCESS(AfpCreateAfpInfo(&fshData,
&fshAfpInfo,
&crinfo)))
{
Status = STATUS_SUCCESS;
}
}
}
if (!NT_SUCCESS(Status))
{
// Skip this entry if you cannot get to the AfpInfo, cannot
// clear the RO attribute or whatever.
break;
}
}
// We successfully opened or created the AfpInfo stream. If
// it existed, then validate the ID, otherwise create all new
// Afpinfo for this file/dir.
if ((crinfo == FILE_OPENED) &&
(NT_SUCCESS(AfpReadAfpInfo(&fshAfpInfo, &AfpInfo))))
{
BOOLEAN fSuccess;
if ((AfpInfo.afpi_Id != pDfEntry->dfe_AfpId) &&
(pDfEntry->dfe_AfpId != AFP_ID_ROOT))
{
PDFENTRY pDFE;
DBGPRINT(DBG_COMP_IDINDEX, DBG_LEVEL_INFO,
("afpVerifyDFE: IdDb Id does not match the AfpInfo Id!!!\n"));
// Unlink it from the hash-bucket since we have the wrong id.
AfpUnlinkDouble(pDfEntry, dfe_NextOverflow, dfe_PrevOverflow);
// If some other entity has this id, then assign a new one. Else
// use the one from the AfpInfo stream.
pDFE = AfpFindDfEntryById(pVolDesc, AfpInfo.afpi_Id, DFE_ANY);
if (pDFE != NULL)
{
pDfEntry->dfe_AfpId = afpGetNextId(pVolDesc);
}
else
{
pDfEntry->dfe_AfpId = AfpInfo.afpi_Id;
}
// Re-insert it with the new id
afpInsertDFEInHashBucket(pVolDesc, pDfEntry, IsDir, &fSuccess);
}
// NOTE: should we set the finder invisible bit if the
// hidden attribute is set so system 6 will obey the
// hiddenness in finder?
pDfEntry->dfe_FinderInfo = AfpInfo.afpi_FinderInfo;
pDfEntry->dfe_BackupTime = AfpInfo.afpi_BackupTime;
pDfEntry->dfe_AfpAttr = AfpInfo.afpi_Attributes;
}
else
{
// AfpInfo stream was newly created, or we could not read
// the existing one because it was corrupt. Create new
// info for this file/dir. Trust the version from the IdDb
AfpInitAfpInfo(&AfpInfo, pDfEntry->dfe_AfpId, IsDir, pDfEntry->dfe_BackupTime);
AfpInfo.afpi_FinderInfo = pDfEntry->dfe_FinderInfo;
AfpInfo.afpi_Attributes = pDfEntry->dfe_AfpAttr;
if (IsDir)
{
// Keep track of see files vs. see folders
AfpInfo.afpi_AccessOwner = DFE_OWNER_ACCESS(pDfEntry);
AfpInfo.afpi_AccessGroup = DFE_GROUP_ACCESS(pDfEntry);
AfpInfo.afpi_AccessWorld = DFE_WORLD_ACCESS(pDfEntry);
}
else
{
AfpProDosInfoFromFinderInfo(&AfpInfo.afpi_FinderInfo,
&AfpInfo.afpi_ProDosInfo);
pDfEntry->dfe_RescLen = 0; // Assume no resource fork
// if this is a Mac application, make sure there is an APPL
// mapping for it.
if (AfpInfo.afpi_FinderInfo.fd_TypeD == *(PDWORD)"APPL")
{
AfpAddAppl(pVolDesc,
pDfEntry->dfe_FinderInfo.fd_CreatorD,
0,
pDfEntry->dfe_AfpId,
True,
pDfEntry->dfe_Parent->dfe_AfpId);
}
}
Status = AfpWriteAfpInfo(&fshAfpInfo, &AfpInfo);
if (!NT_SUCCESS(Status))
{
// We failed to write the AfpInfo stream;
Status = STATUS_UNSUCCESSFUL;
break;
}
}
// Check for comment and resource stream
pStreams = AfpIoQueryStreams(&fshAfpInfo);
if (pStreams == NULL)
{
Status = STATUS_NO_MEMORY;
break;
}
for (pCurStream = pStreams, SeenComment = False;
pCurStream->si_StreamName.Buffer != NULL;
pCurStream++)
{
if (IS_COMMENT_STREAM(&pCurStream->si_StreamName))
{
DFE_SET_COMMENT(pDfEntry);
SeenComment = True;
if (IsDir)
break; // Scan no further for directories
}
else if (!IsDir && IS_RESOURCE_STREAM(&pCurStream->si_StreamName))
{
pDfEntry->dfe_RescLen = pCurStream->si_StreamSize.LowPart;
if (SeenComment)
break; // We have all we need to
}
}
} while (False);
if (fshData.fsh_FileHandle != NULL)
{
AfpPutBackROAttr(&fshData, WriteBackROAttr);
AfpIoClose(&fshData);
}
if (fshAfpInfo.fsh_FileHandle != NULL)
AfpIoClose(&fshAfpInfo);
if (pStreams != NULL)
AfpFreeMemory(pStreams);
if (!NT_SUCCESS(Status))
{
*ppDfEntry = NULL;
}
else
{
// If this is the root directory, make sure we do not blow away the
// AFP_VOLUME_HAS_CUSTOM_ICON bit for NTFS Volume.
if (DFE_IS_ROOT(pDfEntry) &&
(pVolDesc->vds_Flags & AFP_VOLUME_HAS_CUSTOM_ICON))
{
// Don't bother writing back to disk since we do not
// try to keep this in sync in the permanent afpinfo
// stream with the actual existence of the icon<0d> file.
pDfEntry->dfe_FinderInfo.fd_Attr1 |= FINDER_FLAG_HAS_CUSTOM_ICON;
}
}
}
}
/*** afpAddDfEntryAndCacheInfo
*
* During the initial sync with disk on volume startup, add each entry
* we see during enumerate to the id index database.
*/
VOID
afpAddDfEntryAndCacheInfo(
IN PVOLDESC pVolDesc,
IN PDFENTRY pParentDfe,
IN PUNICODE_STRING pUName, // munged unicode name
IN PFILESYSHANDLE pfshEnumDir, // open handle to parent directory
IN PFILE_BOTH_DIR_INFORMATION pFBDInfo, // from enumerate
IN PUNICODE_STRING pNotifyPath, // For Afpinfo Stream
IN PDFENTRY * ppDfEntry,
IN BOOLEAN CheckDuplicate
)
{
BOOLEAN IsDir, SeenComment, WriteBackROAttr = False;
NTSTATUS Status = STATUS_SUCCESS;
FILESYSHANDLE fshAfpInfo, fshData;
DWORD crinfo, openoptions = 0;
PDFENTRY pDfEntry;
AFPINFO AfpInfo;
FINDERINFO FinderInfo;
PSTREAM_INFO pStreams = NULL, pCurStream;
UNICODE_STRING EmptyString;
DWORD NTAttr = 0;
TIME ModTime;
DWORD ModMacTime;
PAGED_CODE();
fshAfpInfo.fsh_FileHandle = NULL;
fshData.fsh_FileHandle = NULL;
IsDir = (pFBDInfo->FileAttributes & FILE_ATTRIBUTE_DIRECTORY) ? True : False;
do
{
if (IS_VOLUME_NTFS(pVolDesc))
{
if (IsDir || CheckDuplicate)
{
// Make sure we don't already have this item in our database.
// Multiple notifies for the same item could possibly occur if
// the PC is renaming or moving items around on the disk, or
// copying trees while we are trying to cache it, since we are
// also queueing up private notifies for directories.
afpFindDFEByUnicodeNameInSiblingList_CS(pVolDesc,
pParentDfe,
pUName,
&pDfEntry,
IsDir ? DFE_DIR : DFE_FILE);
if (pDfEntry != NULL)
{
Status = AFP_ERR_OBJECT_EXISTS;
DBGPRINT(DBG_COMP_CHGNOTIFY, DBG_LEVEL_WARN,
("afpAddDfEntryAndCacheInfo: Attempt to add a duplicate entry: %Z\n", pUName));
break;
}
}
openoptions = IsDir ? FILEIO_OPEN_DIR : FILEIO_OPEN_FILE;
if (NT_SUCCESS(AfpIoOpen(pfshEnumDir,
AFP_STREAM_DATA,
openoptions,
pUName,
FILEIO_ACCESS_NONE,
FILEIO_DENY_NONE,
False,
&fshData)))
{
// save the LastModify time on the data stream of the file: we need to restore it
AfpIoQueryTimesnAttr(&fshData,
NULL,
&ModTime,
&NTAttr);
ModMacTime = AfpConvertTimeToMacFormat(&ModTime);
}
else
{
// if we can't open the data file, just skip this entry!
Status = STATUS_UNSUCCESSFUL;
DBGPRINT(DBG_COMP_IDINDEX, DBG_LEVEL_ERR,
("Couldn't open data stream for %Z\n", pUName));
break;
}
AfpSetEmptyUnicodeString(&EmptyString, 0, NULL);
// open or create the AfpInfo stream
if (!NT_SUCCESS(AfpCreateAfpInfoWithNodeName(&fshData,
&EmptyString,
pNotifyPath,
pVolDesc,
&fshAfpInfo,
&crinfo)))
{
if (!(pFBDInfo->FileAttributes & FILE_ATTRIBUTE_READONLY))
{
// What other reason is there that we could not open
// this stream except that this file/dir is readonly?
Status = STATUS_UNSUCCESSFUL;
break;
}
Status = STATUS_UNSUCCESSFUL; // Assume failure
if (NT_SUCCESS(AfpExamineAndClearROAttr(&fshData,
&WriteBackROAttr,
pVolDesc,
pNotifyPath)))
{
if (NT_SUCCESS(AfpCreateAfpInfo(&fshData,
&fshAfpInfo,
&crinfo)))
{
Status = STATUS_SUCCESS;
}
}
if (!NT_SUCCESS(Status))
{
// Skip this entry if you cannot get to the AfpInfo, cannot
// clear the RO attribute or whatever.
break;
}
}
// We successfully opened or created the AfpInfo stream. If
// it existed, then validate the ID, otherwise create all new
// Afpinfo for this file/dir.
if ((crinfo == FILE_OPENED) &&
(NT_SUCCESS(AfpReadAfpInfo(&fshAfpInfo, &AfpInfo))))
{
// the file/dir had an AfpInfo stream on it
afpCheckDfEntry(pVolDesc,
AfpInfo.afpi_Id,
pUName,
IsDir,
pParentDfe,
&pDfEntry);
if (pDfEntry == NULL)
{
Status = STATUS_UNSUCCESSFUL;
break;
}
if (pDfEntry->dfe_AfpId != AfpInfo.afpi_Id)
{
// Write out the AFP_AfpInfo with the new AfpId
// and uninitialized icon coordinates
AfpInfo.afpi_Id = pDfEntry->dfe_AfpId;
AfpInfo.afpi_FinderInfo.fd_Location[0] =
AfpInfo.afpi_FinderInfo.fd_Location[1] =
AfpInfo.afpi_FinderInfo.fd_Location[2] =
AfpInfo.afpi_FinderInfo.fd_Location[3] = 0xFF;
AfpInfo.afpi_FinderInfo.fd_Attr1 &= ~FINDER_FLAG_SET;
if (!NT_SUCCESS(AfpWriteAfpInfo(&fshAfpInfo, &AfpInfo)))
{
// We failed to write the AfpInfo stream;
// delete the thing from the database
AfpDeleteDfEntry(pVolDesc, pDfEntry);
Status = STATUS_UNSUCCESSFUL;
break;
}
}
// NOTE: should we set the finder invisible bit if the
// hidden attribute is set so system 6 will obey the
// hiddenness in finder?
pDfEntry->dfe_FinderInfo = AfpInfo.afpi_FinderInfo;
pDfEntry->dfe_BackupTime = AfpInfo.afpi_BackupTime;
pDfEntry->dfe_AfpAttr = AfpInfo.afpi_Attributes;
}
else
{
// AfpInfo stream was newly created, or we could not read
// the existing one because it was corrupt. Create new
// info for this file/dir
pDfEntry = AfpAddDfEntry(pVolDesc,
pParentDfe,
pUName,
IsDir,
0);
if (pDfEntry == NULL)
{
Status = STATUS_UNSUCCESSFUL;
break;
}
if (NT_SUCCESS(AfpSlapOnAfpInfoStream(pVolDesc,
pNotifyPath,
NULL,
&fshAfpInfo,
pDfEntry->dfe_AfpId,
IsDir,
pUName,
&AfpInfo)))
{
// NOTE: should we set the finder invisible bit if the
// hidden attribute is set so system 6 will obey the
// hiddenness in finder?
pDfEntry->dfe_FinderInfo = AfpInfo.afpi_FinderInfo;
pDfEntry->dfe_BackupTime = AfpInfo.afpi_BackupTime;
pDfEntry->dfe_AfpAttr = AfpInfo.afpi_Attributes;
}
else
{
// We failed to write the AfpInfo stream;
// delete the thing from the database
AfpDeleteDfEntry(pVolDesc, pDfEntry);
Status = STATUS_UNSUCCESSFUL;
break;
}
}
ASSERT(pDfEntry != NULL);
if (IsDir)
{
// Keep track of see files vs. see folders
DFE_OWNER_ACCESS(pDfEntry) = AfpInfo.afpi_AccessOwner;
DFE_GROUP_ACCESS(pDfEntry) = AfpInfo.afpi_AccessGroup;
DFE_WORLD_ACCESS(pDfEntry) = AfpInfo.afpi_AccessWorld;
}
else
{
// it's a file
pDfEntry->dfe_RescLen = 0; // Assume no resource fork
// if this is a Mac application, make sure there is an APPL
// mapping for it.
if (AfpInfo.afpi_FinderInfo.fd_TypeD == *(PDWORD)"APPL")
{
AfpAddAppl(pVolDesc,
AfpInfo.afpi_FinderInfo.fd_CreatorD,
0,
pDfEntry->dfe_AfpId,
True,
pDfEntry->dfe_Parent->dfe_AfpId);
}
}
// Check for comment and resource stream
pStreams = AfpIoQueryStreams(&fshAfpInfo);
if (pStreams == NULL)
{
Status = STATUS_NO_MEMORY;
break;
}
for (pCurStream = pStreams, SeenComment = False;
pCurStream->si_StreamName.Buffer != NULL;
pCurStream++)
{
if (IS_COMMENT_STREAM(&pCurStream->si_StreamName))
{
DFE_SET_COMMENT(pDfEntry);
SeenComment = True;
if (IsDir)
break; // Scan no further for directories
}
else if (!IsDir && IS_RESOURCE_STREAM(&pCurStream->si_StreamName))
{
pDfEntry->dfe_RescLen = pCurStream->si_StreamSize.LowPart;
if (SeenComment)
break; // We have all we need to
}
}
AfpFreeMemory(pStreams);
}
else // CDFS
{
pDfEntry = AfpAddDfEntry(pVolDesc,
pParentDfe,
pUName,
IsDir,
0);
if (pDfEntry == NULL)
{
Status = STATUS_UNSUCCESSFUL;
break;
}
RtlZeroMemory(&FinderInfo, sizeof(FINDERINFO));
RtlZeroMemory(&pDfEntry->dfe_FinderInfo, sizeof(FINDERINFO));
pDfEntry->dfe_BackupTime = BEGINNING_OF_TIME;
pDfEntry->dfe_AfpAttr = 0;
if (IsDir)
{
DFE_OWNER_ACCESS(pDfEntry) = (DIR_ACCESS_SEARCH | DIR_ACCESS_READ);
DFE_GROUP_ACCESS(pDfEntry) = (DIR_ACCESS_SEARCH | DIR_ACCESS_READ);
DFE_WORLD_ACCESS(pDfEntry) = (DIR_ACCESS_SEARCH | DIR_ACCESS_READ);
}
if (IS_VOLUME_CD_HFS(pVolDesc))
{
Status = AfpIoOpen(pfshEnumDir,
AFP_STREAM_DATA,
openoptions,
pUName,
FILEIO_ACCESS_NONE,
FILEIO_DENY_NONE,
False,
&fshData);
if (!NT_SUCCESS(Status))
{
DBGPRINT(DBG_COMP_IDINDEX, DBG_LEVEL_ERR,
("afpAddDfEntryAndCacheInfo: AfpIoOpen on %Z failed (%lx) for CD_HFS\n", pUName,Status));
break;
}
pDfEntry->dfe_RescLen = 0;
// if it's a file...
if (!IsDir)
{
// if this is a Mac application, make sure there is an APPL
// mapping for it.
if (FinderInfo.fd_TypeD == *(PDWORD)"APPL")
{
AfpAddAppl( pVolDesc,
FinderInfo.fd_CreatorD,
0,
pDfEntry->dfe_AfpId,
True,
pDfEntry->dfe_Parent->dfe_AfpId);
}
// Check for resource stream
pStreams = AfpIoQueryStreams(&fshData);
if (pStreams == NULL)
{
Status = STATUS_NO_MEMORY;
break;
}
for (pCurStream = pStreams;
pCurStream->si_StreamName.Buffer != NULL;
pCurStream++)
{
if (IS_RESOURCE_STREAM(&pCurStream->si_StreamName))
{
pDfEntry->dfe_RescLen = pCurStream->si_StreamSize.LowPart;
break;
}
}
AfpFreeMemory(pStreams);
}
}
// NOTE: if CdHfs doesn't have finder info, should we check for
// that and set it ourselves using AfpSetFinderInfoByExtension?
else
{
AfpSetFinderInfoByExtension(pUName, &pDfEntry->dfe_FinderInfo);
pDfEntry->dfe_RescLen = 0;
}
}
// Record common NTFS & CDFS information
pDfEntry->dfe_CreateTime = AfpConvertTimeToMacFormat(&pFBDInfo->CreationTime);
pDfEntry->dfe_LastModTime = pFBDInfo->LastWriteTime;
pDfEntry->dfe_NtAttr = (USHORT)pFBDInfo->FileAttributes & FILE_ATTRIBUTE_VALID_FLAGS;
// if this is an HFS volume, check if Finder says file should be invisible
if (IS_VOLUME_CD_HFS(pVolDesc))
{
if (pDfEntry->dfe_FinderInfo.fd_Attr1 & FINDER_FLAG_INVISIBLE)
{
pDfEntry->dfe_NtAttr |= FILE_ATTRIBUTE_HIDDEN;
}
}
if (!IsDir)
{
pDfEntry->dfe_DataLen = pFBDInfo->EndOfFile.LowPart;
}
ASSERT(pDfEntry != NULL);
} while (False); // error handling loop
if (fshAfpInfo.fsh_FileHandle != NULL)
{
AfpIoClose(&fshAfpInfo);
// NTFS will set the LastModify time to current time on file since we modified the
// AfpInfo stream: restore the original time
if (!IsDir)
{
// force ntfs to "flush" any pending time updates before we reset the time
AfpIoChangeNTModTime(&fshData,
&ModTime);
AfpIoSetTimesnAttr(&fshData,
NULL,
&ModMacTime,
0,
0,
NULL,
NULL);
}
}
if (fshData.fsh_FileHandle != NULL)
{
if (IS_VOLUME_NTFS(pVolDesc))
{
AfpPutBackROAttr(&fshData, WriteBackROAttr);
}
AfpIoClose(&fshData);
}
if (!NT_SUCCESS(Status))
{
pDfEntry = NULL;
}
*ppDfEntry = pDfEntry;
}
/*** afpReadIdDb
*
* This is called when a volume is added, if an existing Afp_IdIndex stream
* is found on the root of the volume. The stream is is read in, the
* VolDesc is initialized with the header image on disk, and the
* IdDb sibling tree/hash tables are created from the data on disk.
*
**/
LOCAL NTSTATUS FASTCALL
afpReadIdDb(
IN PVOLDESC pVolDesc,
IN PFILESYSHANDLE pfshIdDb,
OUT BOOLEAN * pfVerifyIndex
)
{
PBYTE pReadBuf;
PIDDBHDR pIdDbHdr;
NTSTATUS Status;
LONG SizeRead = 0, Count;
FORKOFFST ForkOffst;
UNICODE_STRING uName;
UNALIGNED DISKENTRY * pCurDiskEnt = NULL;
DWORD NameLen, CurEntSize, SizeLeft;
LONG i, NumRead = 0;
PDFENTRY pParentDfe = NULL, pCurDfe = NULL;
struct _DirFileEntry ** DfeDirBucketStart;
BOOLEAN LastBuf = False, ReadDb = False;
PAGED_CODE( );
DBGPRINT(DBG_COMP_IDINDEX, DBG_LEVEL_ERR,
("afpReadIdDb: Reading existing Id Database header stream...\n") );
do
{
if ((pReadBuf = AfpAllocPANonPagedMemory(IDDB_UPDATE_BUFLEN)) == NULL)
{
Status=STATUS_INSUFFICIENT_RESOURCES;
break;
}
pIdDbHdr = (PIDDBHDR)pReadBuf;
// read in the header
Status = AfpIoRead(pfshIdDb,
&LIZero,
IDDB_UPDATE_BUFLEN,
&SizeRead,
(PBYTE)pIdDbHdr);
if (!NT_SUCCESS(Status) ||
(SizeRead < sizeof(IDDBHDR)) ||
(pIdDbHdr->idh_Signature == AFP_SERVER_SIGNATURE_INITIDDB) ||
(pIdDbHdr->idh_LastId < AFP_ID_NETWORK_TRASH))
{
DBGPRINT(DBG_COMP_IDINDEX, DBG_LEVEL_ERR,
("Index database corrupted for volume %Z, rebuilding database\n",
&pVolDesc->vds_Name) );
DBGPRINT(DBG_COMP_IDINDEX, DBG_LEVEL_ERR,
("Read sign = %lx, True sign = %lx\n",
pIdDbHdr->idh_Signature, AFP_SERVER_SIGNATURE));
// just recreate the stream
Status = AFP_ERR_BAD_VERSION;
AfpIoSetSize(pfshIdDb, 0);
}
else
{
if (pIdDbHdr->idh_Version == AFP_IDDBHDR_VERSION)
{
if ((pIdDbHdr->idh_Signature != AFP_SERVER_SIGNATURE) &&
(pIdDbHdr->idh_Signature != AFP_SERVER_SIGNATURE_MANUALSTOP))
{
DBGPRINT(DBG_COMP_IDINDEX, DBG_LEVEL_ERR,
("afpreadiddb: Totally corrupt sign = (%lx), required sign = (%lx)\n",
pIdDbHdr->idh_Signature, AFP_SERVER_SIGNATURE));
// just recreate the stream
Status = AFP_ERR_BAD_VERSION;
AfpIoSetSize(pfshIdDb, 0);
}
else
{
if (pIdDbHdr->idh_Signature == AFP_SERVER_SIGNATURE_MANUALSTOP)
{
DBGPRINT(DBG_COMP_IDINDEX, DBG_LEVEL_ERR,
("afpreadIdDb: **** Need to verify index after reading it\n"));
*pfVerifyIndex = True;
}
AfpIdDbHdrToVolDesc(pIdDbHdr, pVolDesc);
if (SizeRead < (sizeof(IDDBHDR) + sizeof(DWORD)))
{
DBGPRINT(DBG_COMP_IDINDEX, DBG_LEVEL_ERR,
("afpreadIdDb: Size incorrect\n"));
Count = 0;
Status = STATUS_END_OF_FILE;
}
else
{
Count = *(PDWORD)(pReadBuf + sizeof(IDDBHDR));
if (Count != 0)
{
ReadDb = True;
}
else
{
Status = STATUS_END_OF_FILE;
DBGPRINT(DBG_COMP_IDINDEX, DBG_LEVEL_ERR,
("afpreadIdDb: Count incorrect\n"));
}
}
}
}
else if ((pIdDbHdr->idh_Version != AFP_IDDBHDR_VERSION1) &&
(pIdDbHdr->idh_Version != AFP_IDDBHDR_VERSION2) &&
(pIdDbHdr->idh_Version != AFP_IDDBHDR_VERSION3) &&
(pIdDbHdr->idh_Version != AFP_IDDBHDR_VERSION4))
{
DBGPRINT(DBG_COMP_IDINDEX, DBG_LEVEL_ERR,
("afpreadIdDb: Bad Version (expected %x, actual %x)\n",
AFP_IDDBHDR_VERSION,pIdDbHdr->idh_Version));
// just recreate the stream
AfpIoSetSize(pfshIdDb, 0);
Status = AFP_ERR_BAD_VERSION;
}
else
{
DBGPRINT(DBG_COMP_IDINDEX, DBG_LEVEL_ERR,
("afpreadIdDb: Downlevel Version, re-index volume\n"));
Status = STATUS_END_OF_FILE;
AfpIdDbHdrToVolDesc(pIdDbHdr, pVolDesc);
// Dont blow away the header
AfpIoSetSize(pfshIdDb, sizeof(IDDBHDR));
AFPLOG_INFO(AFPSRVMSG_UPDATE_INDEX_VERSION,
STATUS_SUCCESS,
NULL,
0,
&pVolDesc->vds_Name);
}
}
if (!NT_SUCCESS(Status) || !ReadDb || (Count == 0))
{
break;
}
// read the count of entries from the stream
ForkOffst.QuadPart = SizeRead;
SizeLeft = SizeRead - (sizeof(IDDBHDR) + sizeof(DWORD));
pCurDiskEnt = (UNALIGNED DISKENTRY *)(pReadBuf + (sizeof(IDDBHDR) + sizeof(DWORD)));
// Start the database with the PARENT_OF_ROOT
// At this point there is only the ParentOfRoot DFE in the volume.
// Just access it rather can calling AfpFindDfEntryById
DfeDirBucketStart = pVolDesc->vds_pDfeDirBucketStart;
pParentDfe = DfeDirBucketStart[AFP_ID_PARENT_OF_ROOT];
ASSERT (pParentDfe != NULL);
ASSERT (pParentDfe->dfe_AfpId == AFP_ID_PARENT_OF_ROOT);
DBGPRINT(DBG_COMP_IDINDEX, DBG_LEVEL_INFO,
("afpreadIdDb: Number of entries %d\n", Count));
while ((NumRead < Count) &&
((pVolDesc->vds_Flags & (VOLUME_STOPPED | VOLUME_DELETED)) == 0))
{
//
// get the next entry
//
// We ensure that there are no partial entries. If an entry does not fit at
// the end, we write an invalid entry (AfpId == 0) and skip to the next
// buffer.
if ((SizeLeft < (sizeof(DISKENTRY))) || (pCurDiskEnt->dsk_AfpId == 0))
{
if (LastBuf) // we have already read to the end of file
{
DBGPRINT(DBG_COMP_IDINDEX, DBG_LEVEL_ERR,
("afpreadIdDb: Reached EOF\n"));
Status = STATUS_UNSUCCESSFUL;
break;
}
// Skip to the next page and continue with the next entry
Status = AfpIoRead(pfshIdDb,
&ForkOffst,
IDDB_UPDATE_BUFLEN,
&SizeRead,
pReadBuf);
if (!NT_SUCCESS(Status))
{
DBGPRINT(DBG_COMP_IDINDEX, DBG_LEVEL_ERR,
("afpreadIdDb: Read Eror %lx\n", Status));
break;
}
ForkOffst.QuadPart += SizeRead;
// if we read less than we asked for, then we reached EOF
LastBuf = (SizeRead < IDDB_UPDATE_BUFLEN) ? True : False;
SizeLeft = SizeRead;
pCurDiskEnt = (UNALIGNED DISKENTRY *)pReadBuf;
continue;
}
//
// check dsk_Signature for signature, just to be sure you are
// still aligned on a structure and not off in la-la land
//
if (pCurDiskEnt->dsk_Signature != AFP_DISKENTRY_SIGNATURE)
{
DBGPRINT(DBG_COMP_IDINDEX, DBG_LEVEL_ERR,
("afpreadIdDb: Signature mismatch\n"));
Status = STATUS_DATATYPE_MISALIGNMENT;
break;
}
ASSERT(pCurDiskEnt->dsk_AfpId != AFP_ID_NETWORK_TRASH);
// add current entry to database
if (pCurDiskEnt->dsk_AfpId != AFP_ID_ROOT)
{
AfpInitUnicodeStringWithNonNullTerm(&uName,
(pCurDiskEnt->dsk_Flags & DFE_FLAGS_NAMELENBITS) * sizeof(WCHAR),
(PWSTR)(pCurDiskEnt->dsk_Name));
}
else
{
// In case someone has reused a directory with an existing
// AFP_IdIndex stream for a different volume name
uName = pVolDesc->vds_Name;
}
DBGPRINT(DBG_COMP_IDINDEX, DBG_LEVEL_INFO,
("afpReadIdDb: Read %d: %Z\n",
pCurDiskEnt->dsk_AfpId,
&uName));
if ((pCurDfe = AfpAddDfEntry(pVolDesc,
pParentDfe,
&uName,
(BOOLEAN)((pCurDiskEnt->dsk_Flags & DFE_FLAGS_DIR) == DFE_FLAGS_DIR),
pCurDiskEnt->dsk_AfpId)) == NULL)
{
DBGPRINT(DBG_COMP_IDINDEX, DBG_LEVEL_ERR,
("afpreadIdDb: AfpAddDfEntry failed for %Z (ParentId %x)\n",
&uName, pParentDfe->dfe_AfpId));
Status = STATUS_UNSUCCESSFUL;
break;
}
// Initialize pointer to root.
if (DFE_IS_ROOT(pCurDfe))
{
pVolDesc->vds_pDfeRoot = pCurDfe;
}
afpUpdateDfeWithSavedData(pCurDfe, pCurDiskEnt);
NameLen = (DWORD)((pCurDiskEnt->dsk_Flags & DFE_FLAGS_NAMELENBITS)*sizeof(WCHAR));
CurEntSize = DWLEN(FIELD_OFFSET(DISKENTRY, dsk_Name) + NameLen);
NumRead ++;
SizeLeft -= CurEntSize;
pCurDiskEnt = (UNALIGNED DISKENTRY *)((PBYTE)pCurDiskEnt + CurEntSize);
// figure out who the next parent should be
if (pCurDfe->dfe_Flags & DFE_FLAGS_HAS_CHILD)
{
DBGPRINT(DBG_COMP_IDINDEX, DBG_LEVEL_INFO,
("afpreadIdDb: Moving down %Z\n", &pCurDfe->dfe_UnicodeName));
pParentDfe = pCurDfe;
}
else if (!(pCurDfe->dfe_Flags & DFE_FLAGS_HAS_SIBLING))
{
if (DFE_IS_PARENT_OF_ROOT(pParentDfe))
{
DBGPRINT(DBG_COMP_IDINDEX, DBG_LEVEL_INFO,
("afpreadIdDb: Done, NumRead %d, Count %d\n", NumRead, Count));
ASSERT(NumRead == Count);
break;
}
while (!(pParentDfe->dfe_Flags & DFE_FLAGS_HAS_SIBLING))
{
if (DFE_IS_ROOT(pParentDfe))
{
break;
}
pParentDfe = pParentDfe->dfe_Parent;
DBGPRINT(DBG_COMP_IDINDEX, DBG_LEVEL_INFO,
("afpreadIdDb: Moving up\n"));
}
pParentDfe = pParentDfe->dfe_Parent;
if (DFE_IS_PARENT_OF_ROOT(pParentDfe))
{
DBGPRINT(DBG_COMP_IDINDEX, DBG_LEVEL_INFO,
("afpreadIdDb: Reached Id 1\n"));
ASSERT(NumRead == Count);
break;
}
}
} // while
DBGPRINT(DBG_COMP_IDINDEX, DBG_LEVEL_INFO,
("Indexing: Read %ld entries (%lx)\n", NumRead,Status) );
if (!NT_SUCCESS(Status))
{
DBGPRINT(DBG_COMP_IDINDEX, DBG_LEVEL_ERR,
("afpReadIdDb: Indexing: Starting all over\n"));
// Free everything and start over
AfpFreeIdIndexTables(pVolDesc);
afpInitializeIdDb(pVolDesc);
}
} while (False);
if (pReadBuf != NULL)
AfpFreePANonPagedMemory(pReadBuf, IDDB_UPDATE_BUFLEN);
return Status;
}
VOID
AfpGetDirFileHashSizes(
IN PVOLDESC pVolDesc,
OUT PDWORD pdwDirHashSz,
OUT PDWORD pdwFileHashSz
)
{
FILESYSHANDLE fshIdDb;
ULONG CreateInfo;
PBYTE pReadBuf=NULL;
PIDDBHDR pIdDbHdr;
DWORD dwDirFileCount=0;
DWORD dwTemp;
DWORD dwDirHshTblLen, dwFileHshTblLen;
LONG SizeRead=0;
NTSTATUS Status;
BOOLEAN fFileOpened=FALSE;
// in case we bail out..
*pdwDirHashSz = IDINDEX_BUCKETS_DIR_INIT;
*pdwFileHashSz = IDINDEX_BUCKETS_FILE_INIT;
if (!IS_VOLUME_NTFS(pVolDesc))
{
goto AfpGetDirFileCount_Exit;
}
if ((pReadBuf = AfpAllocNonPagedMemory(1024)) == NULL)
{
goto AfpGetDirFileCount_Exit;
}
Status = AfpIoCreate(&pVolDesc->vds_hRootDir,
AFP_STREAM_IDDB,
&UNullString,
FILEIO_ACCESS_READWRITE,
FILEIO_DENY_WRITE,
FILEIO_OPEN_FILE_SEQ,
FILEIO_CREATE_INTERNAL,
FILE_ATTRIBUTE_NORMAL,
False,
NULL,
&fshIdDb,
&CreateInfo,
NULL,
NULL,
NULL);
if (!NT_SUCCESS(Status))
{
DBGPRINT(DBG_COMP_IDINDEX, DBG_LEVEL_ERR,
("AfpGetDirFileHashSizes: AfpIoCreate failed %lx for %Z\n",Status,&pVolDesc->vds_Name));
goto AfpGetDirFileCount_Exit;
}
fFileOpened = TRUE;
// if the file just got created, it didn't exist before!
if (CreateInfo != FILE_OPENED)
{
DBGPRINT(DBG_COMP_IDINDEX, DBG_LEVEL_ERR,
("AfpGetDirFileHashSizes: file created, not opened %d for %Z\n",CreateInfo,&pVolDesc->vds_Name));
goto AfpGetDirFileCount_Exit;
}
//
// if we reached here, it means that the file existed before, which means
// we made an effort earlier to index this volume. We are going to read the
// dwDirFileCount and calcualte the hash table sizes, but for now let's
// initiliaze this to a high value
//
*pdwDirHashSz = IDINDEX_BUCKETS_32K;
*pdwFileHashSz = IDINDEX_BUCKETS_32K;
// read in the header
Status = AfpIoRead(&fshIdDb,
&LIZero,
1024,
&SizeRead,
pReadBuf);
pIdDbHdr = (PIDDBHDR)pReadBuf;
if (!NT_SUCCESS(Status) ||
(SizeRead < (sizeof(IDDBHDR) + sizeof(DWORD))) ||
((pIdDbHdr->idh_Signature != AFP_SERVER_SIGNATURE) &&
(pIdDbHdr->idh_Signature != AFP_SERVER_SIGNATURE_MANUALSTOP)))
{
DBGPRINT(DBG_COMP_IDINDEX, DBG_LEVEL_ERR,
("Reading DirFileCount: database corrupted for volume %Z!\n",
&pVolDesc->vds_Name) );
goto AfpGetDirFileCount_Exit;
}
else if (pIdDbHdr->idh_Version == AFP_IDDBHDR_VERSION)
{
dwDirFileCount = *(PDWORD)(pReadBuf + sizeof(IDDBHDR));
DBGPRINT(DBG_COMP_IDINDEX, DBG_LEVEL_INFO,
("Reading DirFileCount = %d!\n",dwDirFileCount));
}
AfpGetDirFileCount_Exit:
if (fFileOpened)
{
AfpIoClose(&fshIdDb);
}
if (pReadBuf)
{
AfpFreeMemory(pReadBuf);
}
if (dwDirFileCount == 0)
{
return;
}
//
// first calculate the size of hashtable for Dirs
// We assume here that 10% of the (files+directories) are directories
// We want to try and keep no more than 20 nodes per hash bucket
//
dwDirHshTblLen = (dwDirFileCount / 10);
dwDirHshTblLen = (dwDirHshTblLen / 20);
if (dwDirHshTblLen <= IDINDEX_BUCKETS_DIR_MIN)
{
dwDirHshTblLen = IDINDEX_BUCKETS_DIR_MIN;
}
else if (dwDirHshTblLen >= IDINDEX_BUCKETS_MAX)
{
dwDirHshTblLen = IDINDEX_BUCKETS_MAX;
}
else
{
// find the smallest power of 2 that's bigger than dwDirHshTblLen
dwTemp = IDINDEX_BUCKETS_MAX;
while (dwDirHshTblLen < dwTemp)
{
dwTemp /= 2;
}
dwDirHshTblLen = 2*dwTemp;
}
//
// now, calculate the size of hashtable for Files
// (even though we should take 90% as number of files, we keep it that way!)
// We want to try and keep no more than 20 nodes per hash bucket
//
dwFileHshTblLen = (dwDirFileCount / 20);
if (dwFileHshTblLen <= IDINDEX_BUCKETS_FILE_MIN)
{
dwFileHshTblLen = IDINDEX_BUCKETS_FILE_MIN;
}
else if (dwFileHshTblLen >= IDINDEX_BUCKETS_MAX)
{
dwFileHshTblLen = IDINDEX_BUCKETS_MAX;
}
else
{
// find the smallest power of 2 that's bigger than dwFileHshTblLen
dwTemp = IDINDEX_BUCKETS_MAX;
while (dwFileHshTblLen < dwTemp)
{
dwTemp /= 2;
}
dwFileHshTblLen = 2*dwTemp;
}
*pdwDirHashSz = dwDirHshTblLen;
*pdwFileHashSz = dwFileHshTblLen;
}
/*** AfpFlushIdDb
*
* Initiated by the scavenger thread. If it is determined that the Afp_IdIndex
* stream for this volume is to be flushed to disk, then this is called to
* do the job. The swmr access is locked for read while the update is in
* progress. The vds_cScvgrIdDb and vds_cChangesIdDb are reset if there is
* no error writing the entire database. An initial count of zero is written
* to the database before it is updated, just in case an error occurs that
* prevents us from writing the entire database. When the entire thing is
* written, then the actual count of entries is written to disk. The
* parent-of-root entry is never saved on disk, nor is the network trash
* subtree of the database. The iddb header is written whether it is dirty
* or not.
*
* LOCKS: vds_VolLock (SPIN)
* LOCKS_ASSUMED: vds_idDbAccessLock (SWMR, Shared)
* LOCK_ORDER: vds_VolLock after vds_IdDbAccessLock
*/
VOID FASTCALL
AfpFlushIdDb(
IN PVOLDESC pVolDesc,
IN PFILESYSHANDLE phIdDb
)
{
PBYTE pWriteBuf;
NTSTATUS Status;
BOOLEAN WriteEntireHdr = False;
PDFENTRY pCurDfe;
DWORD SizeLeft; // the number of free bytes left in the writebuffer
DWORD CurEntSize, NumWritten = 0;
FORKOFFST ForkOffst;
UNALIGNED DISKENTRY * pCurDiskEnt = NULL;
PIDDBHDR pIdDbHdr;
BOOLEAN HdrDirty = False;
KIRQL OldIrql;
DWORD fbi = 0, CreateInfo;
#ifdef PROFILING
TIME TimeS, TimeE, TimeD;
INTERLOCKED_INCREMENT_LONG(&AfpServerProfile->perf_IdIndexUpdCount);
AfpGetPerfCounter(&TimeS);
#endif
DBGPRINT(DBG_COMP_IDINDEX, DBG_LEVEL_ERR,("\tWriting out IdDb...\n") );
ASSERT(VALID_VOLDESC(pVolDesc));
// Take the Swmr so that nobody can initiate changes to the IdDb
AfpSwmrAcquireShared(&pVolDesc->vds_IdDbAccessLock);
do
{
if ((pWriteBuf = AfpAllocPANonPagedMemory(IDDB_UPDATE_BUFLEN)) == NULL)
{
Status = STATUS_INSUFFICIENT_RESOURCES;
break;
}
pIdDbHdr = (PIDDBHDR)pWriteBuf;
// Snapshot the IdDbHdr and dirty bit
ACQUIRE_SPIN_LOCK(&pVolDesc->vds_VolLock, &OldIrql);
AfpVolDescToIdDbHdr(pVolDesc, pIdDbHdr);
if (!fAfpServerShutdownEvent)
{
DBGPRINT(DBG_COMP_VOLUME, DBG_LEVEL_ERR,
("AfpFlushIdDb: Corrupting signature during scavenger run\n"));
// If server going down due to admin stop, indicate it
// with a unique signature
if (fAfpAdminStop)
{
pIdDbHdr->idh_Signature = AFP_SERVER_SIGNATURE_MANUALSTOP;
}
else
{
pIdDbHdr->idh_Signature = AFP_SERVER_SIGNATURE_INITIDDB;
}
}
if (pVolDesc->vds_Flags & VOLUME_IDDBHDR_DIRTY)
{
HdrDirty = True;
// Clear the dirty bit
pVolDesc->vds_Flags &= ~VOLUME_IDDBHDR_DIRTY;
}
RELEASE_SPIN_LOCK(&pVolDesc->vds_VolLock, OldIrql);
// Set the count of entries to zero. Once we are done, we'll then
// overwrite it with the right value
*(PDWORD)(pWriteBuf + sizeof(IDDBHDR)) = 0;
// Set the pointer to the first entry we'll write past the header and
// the count of entries. Adjust space remaining.
pCurDiskEnt = (UNALIGNED DISKENTRY *)(pWriteBuf + sizeof(IDDBHDR) + sizeof(DWORD));
SizeLeft = IDDB_UPDATE_BUFLEN - (sizeof(IDDBHDR) + sizeof(DWORD));
ForkOffst.QuadPart = 0;
// start with the root (don't write out the parent of root)
pCurDfe = pVolDesc->vds_pDfeRoot;
ASSERT(pCurDfe != NULL);
Status = STATUS_SUCCESS;
DBGPRINT(DBG_COMP_VOLUME, DBG_LEVEL_ERR,
("AfpFlushIdDb: writing index file for vol %Z started..\n",&pVolDesc->vds_Name));
do
{
ASSERT(!DFE_IS_NWTRASH(pCurDfe));
// The size of the current entry is:
// DISKENTRY structure + namelen padded to 4*N
CurEntSize = DWLEN(FIELD_OFFSET(DISKENTRY, dsk_Name) + pCurDfe->dfe_UnicodeName.Length);
if (CurEntSize > SizeLeft)
{
// If there is atleast a DWORD left, write a ZERO there to
// indicate that the rest of the buffer is to be skipped.
// ZERO is an invalid AfpId.
// NOTE: dsk_AfpId NEEDS TO BE THE FIRST FIELD IN THE STRUCT
ASSERT(FIELD_OFFSET(DISKENTRY, dsk_AfpId) == 0);
if (SizeLeft > 0)
{
RtlZeroMemory(pWriteBuf + IDDB_UPDATE_BUFLEN - SizeLeft,
SizeLeft);
}
// write out the buffer and start at the beginning of buffer
Status = AfpIoWrite(phIdDb,
&ForkOffst,
IDDB_UPDATE_BUFLEN,
pWriteBuf);
if (!NT_SUCCESS(Status))
{
// Reset the dirty bit if need be
if (HdrDirty && (ForkOffst.QuadPart == 0))
{
AfpInterlockedSetDword(&pVolDesc->vds_Flags,
VOLUME_IDDBHDR_DIRTY,
&pVolDesc->vds_VolLock);
}
break;
}
ForkOffst.QuadPart += IDDB_UPDATE_BUFLEN;
SizeLeft = IDDB_UPDATE_BUFLEN;
pCurDiskEnt = (UNALIGNED DISKENTRY *)pWriteBuf;
}
NumWritten ++;
afpSaveDfeData(pCurDfe, pCurDiskEnt);
if (DFE_IS_DIRECTORY(pCurDfe))
{
PDIRENTRY pDirEntry = pCurDfe->dfe_pDirEntry;
pCurDiskEnt->dsk_RescLen = 0;
pCurDiskEnt->dsk_Access = *(PDWORD)(&pDirEntry->de_Access);
ASSERT (pCurDfe->dfe_pDirEntry != NULL);
if ((pCurDfe->dfe_FileOffspring > 0) ||
(pCurDfe->dfe_DirOffspring > 1) ||
((pCurDfe->dfe_DirOffspring > 0) &&
!DFE_IS_NWTRASH(pDirEntry->de_ChildDir)))
{
pCurDiskEnt->dsk_Flags |= DFE_FLAGS_HAS_CHILD;
}
if (pCurDfe->dfe_NextSibling != NULL)
{
// Make sure we ignore the nwtrash folder
if (!DFE_IS_NWTRASH(pCurDfe->dfe_NextSibling) ||
(pCurDfe->dfe_NextSibling->dfe_NextSibling != NULL))
{
pCurDiskEnt->dsk_Flags |= DFE_FLAGS_HAS_SIBLING;
}
}
}
else
{
BOOLEAN fHasSibling;
pCurDiskEnt->dsk_RescLen = pCurDfe->dfe_RescLen;
pCurDiskEnt->dsk_DataLen = pCurDfe->dfe_DataLen;
DFE_FILE_HAS_SIBLING(pCurDfe, fbi, &fHasSibling);
if (fHasSibling)
{
pCurDiskEnt->dsk_Flags |= DFE_FLAGS_HAS_SIBLING;
}
}
// stick the current entry into the write buffer
DBGPRINT(DBG_COMP_IDINDEX, DBG_LEVEL_INFO,
("AfpFlushIdDb: Writing %s %d: %Z, flags %x\n",
DFE_IS_DIRECTORY(pCurDfe) ? "Dir" : "File", pCurDfe->dfe_AfpId,
&pCurDfe->dfe_UnicodeName, pCurDiskEnt->dsk_Flags));
SizeLeft -= CurEntSize;
pCurDiskEnt = (UNALIGNED DISKENTRY *)((PBYTE)pCurDiskEnt + CurEntSize);
/*
* Get to the next DFE. The entire tree under the network-trash
* folder is ignored. The Next DFE is:
*
* if (the current DFE is a file)
* {
* if (there is another file sibling)
* {
* CurDfe = file Sibling;
* }
* else if (there is dir sibling)
* {
* CurDfe = dir Sibling;
* }
* else
* {
* CurDfe = Parent's next sibling directory;
* }
* }
* else
* {
* if (dir has any file children)
* {
* CurDfe = First file child;
* }
* else if (dir has any dir children)
* {
* CurDfe = First dir child;
* }
* if (there is a sibling)
* {
* CurDfe = dir Sibling;
* }
* else
* {
* CurDfe = Parent's next sibling directory;
* }
* }
* When we hit the root, we are done
*
*/
if (DFE_IS_FILE(pCurDfe))
{
if (pCurDfe->dfe_NextSibling != NULL)
{
pCurDfe = pCurDfe->dfe_NextSibling;
}
else
{
PDIRENTRY pDirEntry;
pDirEntry = pCurDfe->dfe_Parent->dfe_pDirEntry;
fbi++;
while ((fbi < MAX_CHILD_HASH_BUCKETS) &&
(pDirEntry->de_ChildFile[fbi] == NULL))
{
fbi++;
}
if (fbi < MAX_CHILD_HASH_BUCKETS)
{
pCurDfe = pDirEntry->de_ChildFile[fbi];
continue;
}
// There are no more files from this parent. Try its children
// next. Ignore the NWTRASH folder.
fbi = 0;
if (pDirEntry->de_ChildDir != NULL)
{
if (!DFE_IS_NWTRASH(pDirEntry->de_ChildDir))
{
pCurDfe = pDirEntry->de_ChildDir;
continue;
}
if (pDirEntry->de_ChildDir->dfe_NextSibling != NULL)
{
pCurDfe = pDirEntry->de_ChildDir->dfe_NextSibling;
continue;
}
}
// No more 'dir' siblings, move on up.
if (!DFE_IS_ROOT(pCurDfe))
{
goto move_up;
}
else
{
// we've reached the root: we're done: get out of the loop
pCurDfe = NULL;
}
}
}
else
{
PDIRENTRY pDirEntry;
// Reset ChildFileBucket index. First check for and handle
// all file children of this directory and then move on to
// its children or siblings
fbi = 0;
pDirEntry = pCurDfe->dfe_pDirEntry;
while ((fbi < MAX_CHILD_HASH_BUCKETS) &&
(pDirEntry->de_ChildFile[fbi] == NULL))
{
fbi++;
}
if (fbi < MAX_CHILD_HASH_BUCKETS)
{
pCurDfe = pDirEntry->de_ChildFile[fbi];
continue;
}
if (pDirEntry->de_ChildDir != NULL)
{
pCurDfe = pDirEntry->de_ChildDir;
// don't bother writing out the network trash tree
if (DFE_IS_NWTRASH(pCurDfe))
{
// could be null, if so, we're done
pCurDfe = pCurDfe->dfe_NextSibling;
}
}
else if (pCurDfe->dfe_NextSibling != NULL)
{
pCurDfe = pCurDfe->dfe_NextSibling;
// don't bother writing out the network trash tree
if (DFE_IS_NWTRASH(pCurDfe))
{
// could be null, if so, we're done
pCurDfe = pCurDfe->dfe_NextSibling;
}
}
else if (!DFE_IS_ROOT(pCurDfe))
{
move_up:
while (pCurDfe->dfe_Parent->dfe_NextSibling == NULL)
{
if (DFE_IS_ROOT(pCurDfe->dfe_Parent))
{
break;
}
pCurDfe = pCurDfe->dfe_Parent;
}
// if ROOT then you get NULL
pCurDfe = pCurDfe->dfe_Parent->dfe_NextSibling; // if ROOT then you get NULL
// Make sure we ignore the nwtrash folder
if ((pCurDfe != NULL) && DFE_IS_NWTRASH(pCurDfe))
{
pCurDfe = pCurDfe->dfe_NextSibling; // Could be null
}
}
else break;
}
} while ((pCurDfe != NULL) && NT_SUCCESS(Status));
if (NT_SUCCESS(Status))
{
DWORD LastWriteSize, SizeRead;
LastWriteSize = (DWORD)ROUND_TO_PAGES(IDDB_UPDATE_BUFLEN - SizeLeft);
if (SizeLeft != IDDB_UPDATE_BUFLEN)
{
// write out the last bunch of the entries. Zero out unused space
RtlZeroMemory(pWriteBuf + IDDB_UPDATE_BUFLEN - SizeLeft,
LastWriteSize - (IDDB_UPDATE_BUFLEN - SizeLeft));
Status = AfpIoWrite(phIdDb,
&ForkOffst,
LastWriteSize,
pWriteBuf);
if (!NT_SUCCESS(Status))
{
// Reset the dirty bit if need be
if (HdrDirty && (ForkOffst.QuadPart == 0))
{
AfpInterlockedSetDword(&pVolDesc->vds_Flags,
VOLUME_IDDBHDR_DIRTY,
&pVolDesc->vds_VolLock);
}
break;
}
}
// set the file length to IdDb plus count plus header
Status = AfpIoSetSize(phIdDb,
ForkOffst.LowPart + LastWriteSize);
ASSERT(Status == AFP_ERR_NONE);
// write out the count of entries written to the file. Do a read-modify-write
// of the first page
ForkOffst.QuadPart = 0;
Status = AfpIoRead(phIdDb,
&ForkOffst,
PAGE_SIZE,
&SizeRead,
pWriteBuf);
ASSERT (NT_SUCCESS(Status) && (SizeRead == PAGE_SIZE));
if (!NT_SUCCESS(Status))
{
// set the file length to header plus count of zero entries
AfpIoSetSize(phIdDb,
sizeof(IDDBHDR)+sizeof(DWORD));
break;
}
*(PDWORD)(pWriteBuf + sizeof(IDDBHDR)) = NumWritten;
Status = AfpIoWrite(phIdDb,
&ForkOffst,
PAGE_SIZE,
pWriteBuf);
if (!NT_SUCCESS(Status))
{
// set the file length to header plus count of zero entries
AfpIoSetSize(phIdDb,
sizeof(IDDBHDR)+sizeof(NumWritten));
break;
}
// not protected since scavenger is the only consumer of this field
pVolDesc->vds_cScvgrIdDb = 0;
DBGPRINT(DBG_COMP_IDINDEX, DBG_LEVEL_INFO,
("Wrote %ld entries on volume %Z\n",NumWritten,&pVolDesc->vds_Name) );
ASSERT(Status == AFP_ERR_NONE);
}
} while (False);
AfpSwmrRelease(&pVolDesc->vds_IdDbAccessLock);
if (!NT_SUCCESS(Status))
{
DBGPRINT(DBG_COMP_VOLUME, DBG_LEVEL_ERR,
("AfpFlushIdDb: writing index file for vol %Z failed (%lx)\n",
&pVolDesc->vds_Name,Status));
AFPLOG_ERROR(AFPSRVMSG_WRITE_IDDB,
Status,
NULL,
0,
&pVolDesc->vds_Name);
}
else
{
DBGPRINT(DBG_COMP_VOLUME, DBG_LEVEL_INFO,
("AfpFlushIdDb: writing index file for vol %Z finished.\n",&pVolDesc->vds_Name));
}
if (pWriteBuf != NULL)
{
AfpFreePANonPagedMemory(pWriteBuf, IDDB_UPDATE_BUFLEN);
}
#ifdef PROFILING
AfpGetPerfCounter(&TimeE);
TimeD.QuadPart = TimeE.QuadPart - TimeS.QuadPart;
INTERLOCKED_ADD_LARGE_INTGR(&AfpServerProfile->perf_IdIndexUpdTime,
TimeD,
&AfpStatisticsLock);
#endif
}
/*** AfpChangeNotifyThread
*
* Handle change notify requests queued by the notify completion routine.
*/
VOID
AfpChangeNotifyThread(
IN PVOID pContext
)
{
PKQUEUE NotifyQueue;
PLIST_ENTRY pTimerList, pList, pNotifyList, pNext, pVirtualNotifyList;
LIST_ENTRY TransitionList;
PVOL_NOTIFY pVolNotify;
PVOLDESC pVolDesc;
PVOLDESC pCurrVolDesc=NULL;
ULONG BasePriority;
PLONG pNotifyQueueCount;
KIRQL OldIrql;
NTSTATUS Status;
DWORD ThisVolItems=0;
BOOLEAN fSwmrLocked=False;
#ifdef PROFILING
TIME TimeS, TimeE, TimeD;
#endif
LONG ShutdownPriority = LOW_REALTIME_PRIORITY; // Priority higher than file server
DBGPRINT(DBG_COMP_INIT, DBG_LEVEL_INFO,
("AfpChangeNotifyThread: Starting %ld\n", pContext));
IoSetThreadHardErrorMode( FALSE );
// Boost our priority to just below low realtime.
// The idea is get the work done fast and get out of the way.
BasePriority = LOW_REALTIME_PRIORITY;
Status = NtSetInformationThread(NtCurrentThread(),
ThreadBasePriority,
&BasePriority,
sizeof(BasePriority));
ASSERT(NT_SUCCESS(Status));
pNotifyList = &AfpVolumeNotifyList[(LONG_PTR)pContext];
pVirtualNotifyList = &AfpVirtualMemVolumeNotifyList[(LONG_PTR)pContext];
NotifyQueue = &AfpVolumeNotifyQueue[(LONG_PTR)pContext];
pNotifyQueueCount = &AfpNotifyQueueCount[(LONG_PTR)pContext];
AfpThreadPtrsW[(LONG_PTR)pContext] = PsGetCurrentThread();
do
{
AFPTIME CurrentTime;
ASSERT (KeGetCurrentIrql() < DISPATCH_LEVEL);
if (!IsListEmpty(pVirtualNotifyList))
{
AfpSwmrAcquireExclusive(&AfpVolumeListSwmr);
pList = RemoveHeadList(pVirtualNotifyList);
AfpSwmrRelease(&AfpVolumeListSwmr);
}
else
{
// Wait for a change notify request to process or timeout
//pList = KeRemoveQueue(NotifyQueue, KernelMode, &TwoSecTimeOut);
pList = KeRemoveQueue(NotifyQueue, KernelMode, &OneSecTimeOut);
}
// We either have something to process or we timed out. In the latter case
// see if it is time to move some stuff from the list to the queue.
if ((NTSTATUS)((ULONG_PTR)pList) != STATUS_TIMEOUT)
{
pVolNotify = CONTAINING_RECORD(pList, VOL_NOTIFY, vn_List);
if (pVolNotify == &AfpTerminateNotifyThread)
{
DBGPRINT(DBG_COMP_VOLUME, DBG_LEVEL_ERR,
("ChangeNotifyThread Got AfpTerminateNotifyEvent Thread = %ld\n", (LONG_PTR)pContext));
if (pCurrVolDesc)
{
AfpSwmrRelease(&pCurrVolDesc->vds_IdDbAccessLock);
// remove that ref we put before grabbing the swmr lock
AfpVolumeDereference(pCurrVolDesc);
pCurrVolDesc = NULL;
}
// If these assertions fail, then there will be extra ref
// counts on some volumes due to unprocessed notifies,
// and the volumes will never go away.
ASSERT((*pNotifyQueueCount == 0) &&
(AfpNotifyListCount[(LONG_PTR)pContext] == 0) &&
IsListEmpty(pNotifyList));
break; // Asked to quit, so do so.
}
#ifdef PROFILING
AfpGetPerfCounter(&TimeS);
#endif
pVolDesc = pVolNotify->vn_pVolDesc;
//
// if we just moved to the next volume, release the lock for the
// previous volume, and also, grab the lock for the this volume
//
if (pVolDesc != pCurrVolDesc)
{
if ((pCurrVolDesc) && (fSwmrLocked))
{
AfpSwmrRelease(&pCurrVolDesc->vds_IdDbAccessLock);
// remove that ref we put before grabbing the swmr lock
AfpVolumeDereference(pCurrVolDesc);
}
pCurrVolDesc = pVolDesc;
ThisVolItems = 0;
// reference the volume, and deref when we release this swmr lock
if (AfpVolumeReference(pVolDesc))
{
AfpSwmrAcquireExclusive(&pVolDesc->vds_IdDbAccessLock);
fSwmrLocked = True;
}
else
{
DBGPRINT(DBG_COMP_VOLUME, DBG_LEVEL_ERR,
("AfpChangeNotifyThread: couldn't reference volume %Z!!\n",
&pVolDesc->vds_Name));
fSwmrLocked = False;
// Remove all private notifies which follow this one
AfpVolumeStopIndexing(pVolDesc, pVolNotify);
}
}
else
{
//
// if someone's waiting for the lock, release it and reacuire it
// to give him a chance
//
if ( (SWMR_SOMEONE_WAITING(&pVolDesc->vds_IdDbAccessLock)) &&
(ThisVolItems % 50 == 0) )
{
ASSERT(fSwmrLocked);
AfpSwmrRelease(&pVolDesc->vds_IdDbAccessLock);
AfpSwmrAcquireExclusive(&pVolDesc->vds_IdDbAccessLock);
}
}
ThisVolItems++;
if (!(((PFILE_NOTIFY_INFORMATION)(pVolNotify + 1))->Action & AFP_ACTION_PRIVATE))
{
(*pNotifyQueueCount) --;
}
ASSERT(VALID_VOLDESC(pVolDesc));
// The volume is already referenced for the notification processing.
// Dereference after finishing processing.
if ((pVolDesc->vds_Flags & (VOLUME_DELETED | VOLUME_STOPPED)) == 0)
{
#ifndef BLOCK_MACS_DURING_NOTIFYPROC
AfpSwmrAcquireExclusive(&pVolDesc->vds_IdDbAccessLock);
#endif
(*pVolNotify->vn_Processor)(pVolNotify);
#ifndef BLOCK_MACS_DURING_NOTIFYPROC
AfpSwmrRelease(&pVolDesc->vds_IdDbAccessLock);
#endif
}
AfpVolumeDereference(pVolDesc);
// was this a private notify?
if (((PFILE_NOTIFY_INFORMATION)(pVolNotify + 1))->Action & AFP_ACTION_PRIVATE)
{
// Free virtual memory
afpFreeNotify(pVolNotify);
}
else
{
// Free non-paged memory
AfpFreeMemory(pVolNotify);
}
#ifdef PROFILING
AfpGetPerfCounter(&TimeE);
TimeD.QuadPart = TimeE.QuadPart - TimeS.QuadPart;
INTERLOCKED_ADD_LARGE_INTGR(&AfpServerProfile->perf_ChangeNotifyTime,
TimeD,
&AfpStatisticsLock);
INTERLOCKED_INCREMENT_LONG(&AfpServerProfile->perf_ChangeNotifyCount);
#endif
if (*pNotifyQueueCount > 0)
continue;
}
//
// release the lock for the volume we just finished
//
if ((pCurrVolDesc) && (fSwmrLocked))
{
AfpSwmrRelease(&pCurrVolDesc->vds_IdDbAccessLock);
// remove that ref we put before grabbing the swmr lock
AfpVolumeDereference(pCurrVolDesc);
}
// If we timed out because there was nothing in the queue, or we
// just processed the last thing in the queue, then see if there are
// more things that can be moved into the queue.
InitializeListHead(&TransitionList);
// Look at the list to see if some stuff should move to the
// Queue now i.e. if the delta has elapsed since we were notified of this change
AfpGetCurrentTimeInMacFormat(&CurrentTime);
ACQUIRE_SPIN_LOCK(&AfpVolumeListLock, &OldIrql);
while (!IsListEmpty(pNotifyList))
{
pList = RemoveHeadList(pNotifyList);
pVolNotify = CONTAINING_RECORD(pList, VOL_NOTIFY, vn_List);
pVolDesc = pVolNotify->vn_pVolDesc;
if ((pVolNotify->vn_TimeStamp == AFP_QUEUE_NOTIFY_IMMEDIATELY) ||
((CurrentTime - pVolNotify->vn_TimeStamp) >= VOLUME_NTFY_DELAY) ||
(pVolDesc->vds_Flags & (VOLUME_STOPPED | VOLUME_DELETED)))
{
AfpNotifyListCount[(LONG_PTR)pContext] --;
(*pNotifyQueueCount) ++;
// Build up a list of items to queue up outside the spinlock
// since we will want to take the IdDb swmr for any volume
// which has notifies that we are about to process.
// Make sure you add things so they are processed in the
// order they came in with!!
InsertTailList(&TransitionList, pList);
}
else
{
// Put it back where we we took it from - its time has not come yet
// And we are done now since the list is ordered in time.
InsertHeadList(pNotifyList, pList);
break;
}
}
RELEASE_SPIN_LOCK(&AfpVolumeListLock, OldIrql);
while (!IsListEmpty(&TransitionList))
{
pList = TransitionList.Flink;
pVolNotify = CONTAINING_RECORD(pList, VOL_NOTIFY, vn_List);
pCurrVolDesc = pVolNotify->vn_pVolDesc;
//
// walk the entire list and collect all the items that belong to the
// same VolDesc and put them on the list. This way we take the swmr
// lock only for the volume we are working on
//
while (pList != &TransitionList)
{
pVolNotify = CONTAINING_RECORD(pList, VOL_NOTIFY, vn_List);
pVolDesc = pVolNotify->vn_pVolDesc;
pNext = pList->Flink;
if (pVolDesc == pCurrVolDesc)
{
RemoveEntryList(pList);
AfpVolumeQueueChangeNotify (pVolNotify, NotifyQueue);
//InsertTailList(pVirtualNotifyList, pList);
}
pList = pNext;
}
}
pCurrVolDesc = NULL;
} while (True);
DBGPRINT(DBG_COMP_INIT, DBG_LEVEL_INFO,
("AfpChangeNotifyThread: Quitting %ld\n", pContext));
// Raise the priority of the current thread, so that this thread
// completes before any other thread interrupts it
NtSetInformationThread (
NtCurrentThread( ),
ThreadBasePriority,
&ShutdownPriority,
sizeof(ShutdownPriority)
);
AfpThreadPtrsW[(LONG_PTR)pContext] = NULL;
KeSetEvent(&AfpStopConfirmEvent, IO_NETWORK_INCREMENT, False);
Status = PsTerminateSystemThread (STATUS_SUCCESS);
ASSERT (NT_SUCCESS(Status));
}
/*** AfpProcessChangeNotify
*
* A change item has been dequeued by one of the notify processing threads.
*
* LOCKS_ASSUMED: vds_IdDbAccessLock (SWMR, Exclusive)
*/
VOID FASTCALL
AfpProcessChangeNotify(
IN PVOL_NOTIFY pVolNotify
)
{
PVOLDESC pVolDesc;
UNICODE_STRING UName, UParent, UTail;
PFILE_NOTIFY_INFORMATION pFNInfo;
BOOLEAN CleanupHandle;
PLIST_ENTRY pList;
NTSTATUS status;
PDFENTRY pDfEntry;
FILESYSHANDLE handle;
DWORD afpChangeAction;
DWORD StreamId;
PFILE_BOTH_DIR_INFORMATION pFBDInfo = NULL;
LONG infobuflen;
LONGLONG infobuf[(sizeof(FILE_BOTH_DIR_INFORMATION) + (AFP_LONGNAME_LEN + 1) * sizeof(WCHAR))/sizeof(LONGLONG) + 1];
#if DBG
static PBYTE Action[] = { "",
"ADDED",
"REMOVED",
"MODIFIED",
"RENAMED OLD",
"RENAMED NEW",
"STREAM ADDED",
"STREAM REMOVED",
"STREAM MODIFIED"};
#endif
PAGED_CODE( );
DBGPRINT(DBG_COMP_CHGNOTIFY, DBG_LEVEL_INFO,
("AfpProcessChangeNotify: entered...\n"));
pFNInfo = (PFILE_NOTIFY_INFORMATION)(pVolNotify + 1);
ASSERT((pFNInfo->Action & AFP_ACTION_PRIVATE) == 0);
pVolDesc = pVolNotify->vn_pVolDesc;
ASSERT (VALID_VOLDESC(pVolDesc));
INTERLOCKED_DECREMENT_LONG(&pVolDesc->vds_cOutstandingNotifies);
StreamId = pVolNotify->vn_StreamId;
if ( (pFNInfo->Action == FILE_ACTION_REMOVED) ||
(pFNInfo->Action == FILE_ACTION_RENAMED_OLD_NAME) )
{
ASSERT(!IsListEmpty(&pVolDesc->vds_ChangeNotifyLookAhead));
DBGPRINT(DBG_COMP_CHGNOTIFY, DBG_LEVEL_INFO,
("AfpProcessChangeNotify: removing %s LookAhead change\n",
Action[pFNInfo->Action]));
pList = ExInterlockedRemoveHeadList(&pVolDesc->vds_ChangeNotifyLookAhead,
&(pVolDesc->vds_VolLock.SpinLock));
ASSERT(pList == &(pVolNotify->vn_DelRenLink));
}
// Process each of the entries in the list for this volume
while (True)
{
CleanupHandle = False;
status = STATUS_SUCCESS;
AfpInitUnicodeStringWithNonNullTerm(&UName,
(USHORT)pFNInfo->FileNameLength,
pFNInfo->FileName);
UName.MaximumLength += (AFP_LONGNAME_LEN+1)*sizeof(WCHAR);
ASSERT(IS_VOLUME_NTFS(pVolDesc) && !EXCLUSIVE_VOLUME(pVolDesc));
DBGPRINT(DBG_COMP_CHGNOTIFY, DBG_LEVEL_INFO,
("AfpProcessChangeNotify: Action: %s Name: %Z\n",
Action[pFNInfo->Action], &UName));
do
{
DBGPRINT(DBG_COMP_CHGNOTIFY, DBG_LEVEL_INFO,
("AfpProcessChangeNotify: !!!!Processing Change!!!!\n"));
// We have the idindex write lock, look up the entry by path,
// open a handle to the item, query the appropriate info,
// cache the info in the DFE. Where necessary, open a handle
// to the parent directory and update its cached ModTime.
Lookup_Entry:
pDfEntry = afpFindEntryByNtPath(pVolDesc,
pFNInfo->Action,
&UName,
&UParent,
&UTail);
if (pDfEntry != NULL)
{
// Open a handle to parent or entity relative to the volume root handle
CleanupHandle = True;
status = AfpIoOpen(&pVolDesc->vds_hRootDir,
StreamId,
FILEIO_OPEN_EITHER,
((pFNInfo->Action == FILE_ACTION_ADDED) ||
(pFNInfo->Action == FILE_ACTION_REMOVED) ||
(pFNInfo->Action == FILE_ACTION_RENAMED_OLD_NAME)) ?
&UParent : &UName,
FILEIO_ACCESS_NONE,
FILEIO_DENY_NONE,
False,
&handle);
if (!NT_SUCCESS(status))
{
DBGPRINT(DBG_COMP_CHGNOTIFY, DBG_LEVEL_ERR,
("AfpProcessChangeNotify: Failed to open (0x%lx) for %Z\n", status, &UName));
if (pFNInfo->Action == FILE_ACTION_ADDED) {
// Add the full-pathname of file relative to volume root
// to the DelayedNotifyList for the current VolumeDesc
// We assume here that the filename relative to the
// volume path will not be greater than 512
// characters (unicode)
status = AddToDelayedNotifyList(pVolDesc, &UName);
if (!NT_SUCCESS(status)) {
DBGPRINT(DBG_COMP_CHGNOTIFY, DBG_LEVEL_ERR,
("AfpProcessChangeNotify: Error in addToDelayedList (0x%lx)\n", status));
break;
}
}
CleanupHandle = False;
break;
}
switch (pFNInfo->Action)
{
case FILE_ACTION_ADDED:
{
UNICODE_STRING UEntity;
UNICODE_STRING UTemp;
PDFENTRY pDfeNew;
FILESYSHANDLE fshEnumDir;
BOOLEAN firstEnum = True;
ULONG fileLength;
// Update timestamp of parent dir.
AfpIoQueryTimesnAttr(&handle,
NULL,
&pDfEntry->dfe_LastModTime,
NULL);
// enumerate parent handle for this entity to get
// the file/dir info, then add entry to the IDDB
/*
if ((UTail.Length/sizeof(WCHAR)) <= AFP_LONGNAME_LEN)
*/
// Bug 311023
fileLength = RtlUnicodeStringToAnsiSize(&UTail)-1;
if (fileLength <= AFP_LONGNAME_LEN)
{
pFBDInfo = (PFILE_BOTH_DIR_INFORMATION)infobuf;
infobuflen = sizeof(infobuf);
}
else
{
infobuflen = sizeof(FILE_BOTH_DIR_INFORMATION) +
(MAXIMUM_FILENAME_LENGTH * sizeof(WCHAR));
if ((pFBDInfo = (PFILE_BOTH_DIR_INFORMATION)
AfpAllocNonPagedMemory(infobuflen)) == NULL)
{
status = STATUS_NO_MEMORY;
break; // out of case FILE_ACTION_ADDED
}
}
do
{
status = AfpIoQueryDirectoryFile(&handle,
pFBDInfo,
infobuflen,
FileBothDirectoryInformation,
True,
True,
firstEnum ? &UTail : NULL);
if ((status == STATUS_BUFFER_TOO_SMALL) ||
(status == STATUS_BUFFER_OVERFLOW))
{
// Since we queue our own ACTION_ADDED for directories when
// caching a tree, we may have the case where we queued it
// by shortname because it actually had a name > 31 chars.
// Note that a 2nd call to QueryDirectoryFile after a buffer
// Overflow must send a null filename, since if the name is
// not null, it will override the restartscan parameter
// which means the scan will not restart from the beginning
// and we will not find the file name we are looking for.
ASSERT((PBYTE)pFBDInfo == (PBYTE)infobuf);
// This should never happen, but if it does...
if ((PBYTE)pFBDInfo != (PBYTE)infobuf)
{
status = STATUS_UNSUCCESSFUL;
break;
}
firstEnum = False;
infobuflen = sizeof(FILE_BOTH_DIR_INFORMATION) +
(MAXIMUM_FILENAME_LENGTH * sizeof(WCHAR));
if ((pFBDInfo = (PFILE_BOTH_DIR_INFORMATION)
AfpAllocNonPagedMemory(infobuflen)) == NULL)
status = STATUS_NO_MEMORY;
}
} while ((status == STATUS_BUFFER_TOO_SMALL) ||
(status == STATUS_BUFFER_OVERFLOW));
if (status == STATUS_SUCCESS)
{
// If this file was created in a directory that has
// not been looked at by a mac, just ignore it.
// If this was not a FILE_ACTION_ADDED we would not
// have found it in the database at all if it was
// a file and the parent has not had its file
// children cached in, so we will ignore those
// notifies by default anyway since the DFE would
// come back as NULL from afpFindEntryByNtPath.
// We *do* want to process changes for directories
// even if the parent has not had its
// file children brought in since directories
// are only cached in once at volume startup.
if (((pFBDInfo->FileAttributes & FILE_ATTRIBUTE_DIRECTORY) == 0) &&
!DFE_CHILDREN_ARE_PRESENT(pDfEntry))
{
break;
}
// figure out which name to use
// If NT name > AFP_LONGNAME_LEN, use the NT shortname for
// Mac longname on NTFS, any other file system the shortname
// will be null, so ignore the file
// Bug 311023
AfpInitUnicodeStringWithNonNullTerm(&UTemp,
(USHORT)pFBDInfo->FileNameLength,
pFBDInfo->FileName);
fileLength=RtlUnicodeStringToAnsiSize(&UTemp)-1;
if (fileLength <= AFP_LONGNAME_LEN)
{
AfpInitUnicodeStringWithNonNullTerm(&UEntity,
(USHORT)pFBDInfo->FileNameLength,
pFBDInfo->FileName);
}
else if (pFBDInfo->ShortNameLength > 0)
{
AfpInitUnicodeStringWithNonNullTerm(&UEntity,
(USHORT)pFBDInfo->ShortNameLength,
pFBDInfo->ShortName);
}
else
{
DBGPRINT(DBG_COMP_CHGNOTIFY, DBG_LEVEL_ERR,
("AfpProcessChangeNotify: Name is > 31 with no short name ?\n") );
status = STATUS_UNSUCCESSFUL;
}
if (NT_SUCCESS(status))
{
// add the entry
afpAddDfEntryAndCacheInfo(pVolDesc,
pDfEntry,
&UEntity,
&handle,
pFBDInfo,
&UName,
&pDfeNew,
True);
if (pDfeNew == NULL)
{
DBGPRINT(DBG_COMP_CHGNOTIFY, DBG_LEVEL_WARN,
("AfpProcessChangeNotify: Could not add DFE for %Z\n", &UName));
}
else if (DFE_IS_DIRECTORY(pDfeNew))
{
// if a directory was added, we must see if it has
// children that we must cache as well
if (NT_SUCCESS(status = AfpIoOpen(&pVolDesc->vds_hRootDir,
AFP_STREAM_DATA,
FILEIO_OPEN_DIR,
&UName,
FILEIO_ACCESS_NONE,
FILEIO_DENY_NONE,
False,
&fshEnumDir)))
{
status = AfpCacheDirectoryTree(pVolDesc,
pDfeNew,
GETENTIRETREE | REENUMERATE,
&fshEnumDir,
&UName);
AfpIoClose(&fshEnumDir);
if (!NT_SUCCESS(status))
{
DBGPRINT(DBG_COMP_CHGNOTIFY, DBG_LEVEL_ERR,
("AfpProcessChangeNotify: Could not cache dir tree for %Z\n",
&UName) );
}
}
else
{
DBGPRINT(DBG_COMP_CHGNOTIFY, DBG_LEVEL_ERR,
("AfpProcessChangeNotify: Added dir %Z but couldn't open it for enumerating\n", &UName) );
}
}
}
}
if (((PBYTE)pFBDInfo != NULL) &&
((PBYTE)pFBDInfo != (PBYTE)infobuf))
{
AfpFreeMemory(pFBDInfo);
pFBDInfo = NULL;
}
}
break;
case FILE_ACTION_REMOVED:
{
// Remove entries from DelayedNotifyList for the
// volume belonging to the directory which is
// being removed since they need not be added to
// the IDDB after this
status = RemoveFromDelayedNotifyList(
pVolDesc,
&UName,
pFNInfo
);
if (!NT_SUCCESS(status)) {
DBGPRINT(DBG_COMP_CHGNOTIFY, DBG_LEVEL_ERR,
("Error in RemoveFromDelayedNotifyList 0x%lx\n", status));
}
// update timestamp of parent dir.
AfpIoQueryTimesnAttr(&handle,
NULL,
&pDfEntry->dfe_Parent->dfe_LastModTime,
NULL);
// remove entry from IDDb.
AfpDeleteDfEntry(pVolDesc, pDfEntry);
}
break;
case FILE_ACTION_MODIFIED:
{
FORKSIZE forklen;
DWORD NtAttr;
// NOTE: if a file is SUPERSEDED or OVERWRITTEN,
// you will only get a MODIFIED notification. Is
// there a way to check the creation date against
// what we have cached in order to figure out if
// this is what happened?
// query for times and attributes. If its a file,
// also query for the data fork length.
if (NT_SUCCESS(AfpIoQueryTimesnAttr(&handle,
&pDfEntry->dfe_CreateTime,
&pDfEntry->dfe_LastModTime,
&NtAttr)))
{
pDfEntry->dfe_NtAttr = (USHORT)NtAttr &
FILE_ATTRIBUTE_VALID_FLAGS;
}
if (DFE_IS_FILE(pDfEntry) &&
NT_SUCCESS(AfpIoQuerySize(&handle, &forklen)))
{
pDfEntry->dfe_DataLen = forklen.LowPart;
}
}
break;
case FILE_ACTION_RENAMED_OLD_NAME:
{
UNICODE_STRING UNewname;
PFILE_NOTIFY_INFORMATION pFNInfo2;
ULONG fileLength;
BOOLEAN checkIfDirectory=False;
status = STATUS_SUCCESS;
// The next item in the change buffer better be the
// new name -- consume this entry so we don't find
// it next time thru the loop. If there is none,
// then throw the whole thing out and assume the
// rename aborted in NTFS
if (pFNInfo->NextEntryOffset == 0)
break; // from switch
// If the next change in the buffer is not the
// new name, we don't want to advance over it,
// we want it to be processed next time thru
// the loop. Note we are assuming it is valid.
(PBYTE)pFNInfo2 = (PBYTE)pFNInfo + pFNInfo->NextEntryOffset;
ASSERT(pFNInfo2->Action == FILE_ACTION_RENAMED_NEW_NAME);
if (pFNInfo2->Action != FILE_ACTION_RENAMED_NEW_NAME)
{
DBGPRINT(DBG_COMP_CHGNOTIFY, DBG_LEVEL_ERR,
("AfpProcessChangeNotify: Rename did not come with new name!!!\n"));
break; // from switch
}
pFNInfo = pFNInfo2;
// update timestamp of parent dir.
AfpIoQueryTimesnAttr(&handle,
NULL,
&pDfEntry->dfe_Parent->dfe_LastModTime,
NULL);
// get the entity name from the path (subtract the
// parent path length from total length), if it is
// > 31 chars, we have to get the shortname by
// enumerating the parent for this item, since we
// already have a handle to the parent
AfpInitUnicodeStringWithNonNullTerm(&UNewname,
(USHORT)pFNInfo->FileNameLength,
pFNInfo->FileName);
if (DFE_IS_DIRECTORY(pDfEntry))
{
checkIfDirectory = True;
}
if (UParent.Length > 0)
{
// if the rename is not in the volume root,
// get rid of the path separator before the name
UNewname.Length -= UParent.Length + sizeof(WCHAR);
UNewname.Buffer = (PWCHAR)((PBYTE)UNewname.Buffer + UParent.Length + sizeof(WCHAR));
}
// Bug 311023
fileLength = RtlUnicodeStringToAnsiSize(&UNewname)-1;
if (fileLength > AFP_LONGNAME_LEN)
{
infobuflen = sizeof(FILE_BOTH_DIR_INFORMATION) +
(MAXIMUM_FILENAME_LENGTH * sizeof(WCHAR));
if ((pFBDInfo = (PFILE_BOTH_DIR_INFORMATION)
AfpAllocNonPagedMemory(infobuflen)) == NULL)
{
status = STATUS_NO_MEMORY;
break; // out of case FILE_ACTION_RENAMED
}
status = AfpIoQueryDirectoryFile(&handle,
pFBDInfo,
infobuflen,
FileBothDirectoryInformation,
True,
True,
&UNewname);
if (status == STATUS_SUCCESS)
{
// figure out which name to use
// If NT name > AFP_LONGNAME_LEN, use the NT shortname for
// Mac longname on NTFS, any other file system the shortname
// will be null, so ignore the file
// Bug 311023
AfpInitUnicodeStringWithNonNullTerm(
&UNewname,
(USHORT)
pFBDInfo->FileNameLength,
pFBDInfo->FileName);
fileLength=RtlUnicodeStringToAnsiSize(&UNewname)-1;
if (fileLength <= AFP_LONGNAME_LEN)
{
AfpInitUnicodeStringWithNonNullTerm(&UNewname,
(USHORT)pFBDInfo->FileNameLength,
pFBDInfo->FileName);
}
else if (pFBDInfo->ShortNameLength > 0)
{
AfpInitUnicodeStringWithNonNullTerm(&UNewname,
(USHORT)pFBDInfo->ShortNameLength,
pFBDInfo->ShortName);
}
else
{
DBGPRINT(DBG_COMP_CHGNOTIFY, DBG_LEVEL_ERR,
("AfpProcessChangeNotify: Name is > 31 with no short name ?\n") );
status = STATUS_UNSUCCESSFUL;
}
}
}
// rename the entry
if (NT_SUCCESS(status))
{
AfpRenameDfEntry(pVolDesc, pDfEntry, &UNewname);
}
// Check if a directory is being renamed
// If yes, check if there are any files/directories
// which were not added to the IDDB due to ChangeNotify
// requests getting delayed
if (checkIfDirectory) {
status = CheckAndProcessDelayedNotify (
pVolDesc,
&UName,
&UNewname,
&UParent
);
if (!NT_SUCCESS(status))
{
DBGPRINT(DBG_COMP_CHGNOTIFY, DBG_LEVEL_ERR,
("Error in CheckAndProcessDelayedNotify 0x%lx\n", status));
}
}
if ((PBYTE)pFBDInfo != NULL)
{
AfpFreeMemory(pFBDInfo);
pFBDInfo = NULL;
}
}
break;
case FILE_ACTION_MODIFIED_STREAM:
{
FORKSIZE forklen;
// If it is the AFP_Resource stream on a file,
// cache the resource fork length.
if ((StreamId == AFP_STREAM_RESC) &&
DFE_IS_FILE(pDfEntry) &&
NT_SUCCESS(AfpIoQuerySize(&handle, &forklen)))
{
pDfEntry->dfe_RescLen = forklen.LowPart;
}
else if (StreamId == AFP_STREAM_INFO)
{
AFPINFO afpinfo;
FILEDIRPARM fdparms;
// Read the afpinfo stream. If the file ID in
// the DfEntry does not match the one in the
// stream, write back the ID we know it by.
// Update our cached FinderInfo.
if (NT_SUCCESS(AfpReadAfpInfo(&handle, &afpinfo)))
{
pDfEntry->dfe_FinderInfo = afpinfo.afpi_FinderInfo;
pDfEntry->dfe_BackupTime = afpinfo.afpi_BackupTime;
if (pDfEntry->dfe_AfpId != afpinfo.afpi_Id)
{
// munge up a fake FILEDIRPARMS structure
AfpInitializeFDParms(&fdparms);
fdparms._fdp_Flags = pDfEntry->dfe_Flags;
fdparms._fdp_AfpId = pDfEntry->dfe_AfpId;
AfpConvertMungedUnicodeToAnsi(&pDfEntry->dfe_UnicodeName,
&fdparms._fdp_LongName);
// NOTE: can we open a handle to afpinfo
// relative to a afpinfo handle??
AfpSetAfpInfo(&handle,
FILE_BITMAP_FILENUM,
&fdparms,
NULL,
NULL);
}
}
}
}
break; // from switch
default:
ASSERTMSG("AfpProcessChangeNotify: Unexpected Action\n", False);
break;
} // switch
}
else
{
PFILE_NOTIFY_INFORMATION pFNInfo2;
DBGPRINT(DBG_COMP_CHGNOTIFY, DBG_LEVEL_WARN,
("AfpProcessChangeNotify: Could not find DFE for %Z\n", &UName));
// This item may have been deleted, renamed or moved
// by someone else in the interim, ignore this change
// if it was not a rename. If it was a rename, then
// try to add the item using the new name.
if ((pFNInfo->Action == FILE_ACTION_RENAMED_OLD_NAME) &&
(pFNInfo->NextEntryOffset != 0))
{
// If the next change in the buffer is not the
// new name, we don't want to advance over it,
// we want it to be processed next time thru
// the loop. Note we are assuming it is valid.
(PBYTE)pFNInfo2 = (PBYTE)pFNInfo + pFNInfo->NextEntryOffset;
ASSERT(pFNInfo2->Action == FILE_ACTION_RENAMED_NEW_NAME);
if (pFNInfo2->Action != FILE_ACTION_RENAMED_NEW_NAME)
{
DBGPRINT(DBG_COMP_CHGNOTIFY, DBG_LEVEL_ERR,
("AfpProcessChangeNotify: Rename did not come with new name!!! (no-DFE case)\n"));
break; // from error loop
}
pFNInfo = pFNInfo2;
pFNInfo->Action = FILE_ACTION_ADDED;
AfpInitUnicodeStringWithNonNullTerm(&UName,
(USHORT)pFNInfo->FileNameLength,
pFNInfo->FileName);
DBGPRINT(DBG_COMP_CHGNOTIFY, DBG_LEVEL_INFO,
("AfpProcessChangeNotify: Converting RENAME into Action: %s for Name: %Z\n",
Action[pFNInfo->Action], &UName));
goto Lookup_Entry;
}
}
} while (False);
if (CleanupHandle)
AfpIoClose(&handle);
// Advance to the next entry in the change buffer
if (pFNInfo->NextEntryOffset == 0)
{
break;
}
(PBYTE)pFNInfo += pFNInfo->NextEntryOffset;
DBGPRINT(DBG_COMP_CHGNOTIFY, DBG_LEVEL_WARN,
("More than one entry in notify ?\n"));
}
// Get new values for Free space on disk and update volume time
AfpUpdateVolFreeSpaceAndModTime(pVolDesc, TRUE);
}
/*** AddToDelayedNotifyList
*
* Add Pathname of delayed notify for added file to DelayedNotifyList of the
* corresponding volume
*
* LOCKS_ASSUMED: none
*/
NTSTATUS FASTCALL
AddToDelayedNotifyList(
IN PVOLDESC pVolDesc,
IN PUNICODE_STRING pUName
)
{
KIRQL OldIrql;
PDELAYED_NOTIFY pDelayedNotify;
NTSTATUS status = STATUS_SUCCESS;
pDelayedNotify = (PDELAYED_NOTIFY)AfpAllocNonPagedMemory (sizeof(DELAYED_NOTIFY));
if (pDelayedNotify == NULL)
{
status = STATUS_INSUFFICIENT_RESOURCES;
return status;
}
pDelayedNotify->filename.Length = 0;
pDelayedNotify->filename.MaximumLength = pUName->MaximumLength;
pDelayedNotify->filename.Buffer = (PWSTR)AfpAllocNonPagedMemory(pUName->MaximumLength);
if (pDelayedNotify->filename.Buffer == NULL)
{
status = STATUS_INSUFFICIENT_RESOURCES;
AfpFreeMemory(pDelayedNotify);
return status;
}
RtlCopyUnicodeString(&pDelayedNotify->filename, pUName);
ACQUIRE_SPIN_LOCK(&pVolDesc->vds_VolLock, &OldIrql);
InsertHeadList (&pVolDesc->vds_DelayedNotifyList, &pDelayedNotify->dn_List);
RELEASE_SPIN_LOCK(&pVolDesc->vds_VolLock, OldIrql);
return status;
}
/*** RemoveFromDelayedNotifyList
*
* Remove entry from DelayedNotifyList for files which are within the
* directory that was deleted
*
* LOCKS_ASSUMED: none
*/
NTSTATUS
RemoveFromDelayedNotifyList (
IN PVOLDESC pVolDesc,
IN PUNICODE_STRING pUName,
IN PFILE_NOTIFY_INFORMATION pFNInfo
)
{
PLIST_ENTRY pList, pNext;
PDELAYED_NOTIFY pDelayedNotify;
KIRQL OldIrql;
NTSTATUS status=STATUS_SUCCESS;
AfpInitUnicodeStringWithNonNullTerm(pUName,
(USHORT)pFNInfo->FileNameLength,
pFNInfo->FileName);
DBGPRINT(DBG_COMP_CHGNOTIFY, DBG_LEVEL_INFO,
("AfpProcessChangeNotify: Going to remove %Z \n", pUName));
ACQUIRE_SPIN_LOCK(&pVolDesc->vds_VolLock, &OldIrql);
if (!IsListEmpty(&pVolDesc->vds_DelayedNotifyList)) {
pList = pVolDesc->vds_DelayedNotifyList.Flink;
while (pList != &pVolDesc->vds_DelayedNotifyList)
{
pDelayedNotify = CONTAINING_RECORD (pList, DELAYED_NOTIFY, dn_List);
pNext = pList->Flink;
if (pDelayedNotify->filename.Length >= pUName->Length)
{
if (RtlCompareMemory ((PVOID)pUName->Buffer, (PVOID)pDelayedNotify->filename.Buffer, pUName->Length) == pUName->Length) {
DBGPRINT(DBG_COMP_CHGNOTIFY, DBG_LEVEL_ERR, ("AfpProcessChangeNotify: Correction no longer to be done for %Z\n", &pDelayedNotify->filename));
RemoveEntryList(pList);
AfpFreeMemory(pDelayedNotify->filename.Buffer);
AfpFreeMemory(pDelayedNotify);
}
}
pList = pNext;
}
}
RELEASE_SPIN_LOCK(&pVolDesc->vds_VolLock, OldIrql);
return status;
}
/*** CheckAndProcessDelayedNotify
*
* Check if the directory being renamed had entries that were not added to
* the IDDB. Rename the entries and issue new FILE_ACTION_ADDED ChangeNotify
* requests as if the NTFS filesystem has made the requests
*
* LOCKS_ASSUMED: none
*/
NTSTATUS
CheckAndProcessDelayedNotify (
IN PVOLDESC pVolDesc,
IN PUNICODE_STRING pUName,
IN PUNICODE_STRING pUNewname,
IN PUNICODE_STRING pUParent
)
{
PLIST_ENTRY pList;
PDELAYED_NOTIFY pDelayedNotify;
KIRQL OldIrql;
NTSTATUS status = STATUS_SUCCESS;
PFILE_NOTIFY_INFORMATION pFNInfo;
PDFENTRY pParentDfEntry;
UNICODE_STRING newNotifyName;
PVOL_NOTIFY pVolNotify;
ACQUIRE_SPIN_LOCK(&pVolDesc->vds_VolLock, &OldIrql);
pList = pVolDesc->vds_DelayedNotifyList.Flink;
while (1)
{
// finished the list?
if (pList == &pVolDesc->vds_DelayedNotifyList)
{
RELEASE_SPIN_LOCK(&pVolDesc->vds_VolLock, OldIrql);
break;
}
pDelayedNotify = CONTAINING_RECORD (pList, DELAYED_NOTIFY, dn_List);
pList = pList->Flink;
if (pDelayedNotify->filename.Length < pUName->Length)
{
continue;
}
if (RtlCompareMemory ((PVOID)pUName->Buffer,
(PVOID)pDelayedNotify->filename.Buffer,
pUName->Length) == pUName->Length)
{
DBGPRINT(DBG_COMP_CHGNOTIFY, DBG_LEVEL_ERR,
("AfpProcessChangeNotify: Correction required to be done for %Z\n", &pDelayedNotify->filename));
RemoveEntryList(&pDelayedNotify->dn_List);
RELEASE_SPIN_LOCK(&pVolDesc->vds_VolLock, OldIrql);
if ((newNotifyName.Buffer = (PWSTR)AfpAllocNonPagedMemory(1024)) == NULL)
{
status = STATUS_INSUFFICIENT_RESOURCES;
AfpFreeMemory(pDelayedNotify->filename.Buffer);
AfpFreeMemory(pDelayedNotify);
return status;
};
newNotifyName.Length = 0;
newNotifyName.MaximumLength = 1024;
if (pUParent->Length > 0)
{
RtlCopyUnicodeString (&newNotifyName, pUName);
// Copy the Parent name and the "/" separator and then the new name
RtlCopyMemory (newNotifyName.Buffer + pUParent->Length/2 + 1,
pUNewname->Buffer, pUNewname->Length);
newNotifyName.Length = pUParent->Length + pUNewname->Length + 2;
}
else
{
RtlCopyUnicodeString (&newNotifyName, pUNewname);
}
RtlAppendUnicodeToString (&newNotifyName,
pDelayedNotify->filename.Buffer + pUName->Length/2);
DBGPRINT(DBG_COMP_CHGNOTIFY, DBG_LEVEL_ERR,
("AfpProcessChangeNotify: After correction, name = %Z, Old=%Z, New=%Z, chgto=%Z\n",
&pDelayedNotify->filename, pUName, pUNewname, &newNotifyName));
pVolNotify = (PVOL_NOTIFY)AfpAllocNonPagedMemory(
sizeof(VOL_NOTIFY) +
(ULONG)(newNotifyName.Length) +
sizeof(FILE_NOTIFY_INFORMATION) +
(AFP_LONGNAME_LEN + 1)*sizeof(WCHAR));
if (pVolNotify != NULL)
{
AfpGetCurrentTimeInMacFormat(&pVolNotify->vn_TimeStamp);
pVolNotify->vn_pVolDesc = pVolDesc;
pVolNotify->vn_Processor = AfpProcessChangeNotify;
pVolNotify->vn_StreamId = AFP_STREAM_DATA;
pVolNotify->vn_TailLength = newNotifyName.Length;
pFNInfo = (PFILE_NOTIFY_INFORMATION)(pVolNotify + 1);
pFNInfo->Action = FILE_ACTION_ADDED;
pFNInfo->NextEntryOffset = 0;
pFNInfo->FileNameLength = newNotifyName.Length;
RtlCopyMemory((PVOID)&pFNInfo->FileName,
(PVOID)newNotifyName.Buffer,
newNotifyName.Length);
if (AfpVolumeReference(pVolDesc))
{
AfpVolumeInsertChangeNotifyList(pVolNotify, pVolDesc);
}
}
else
{
DBGPRINT(DBG_COMP_CHGNOTIFY, DBG_LEVEL_ERR,("Out of memory!!\n"));
ASSERT(0);
status = STATUS_INSUFFICIENT_RESOURCES;
AfpFreeMemory(newNotifyName.Buffer);
AfpFreeMemory(pDelayedNotify->filename.Buffer);
AfpFreeMemory(pDelayedNotify);
return status;
}
AfpFreeMemory(pDelayedNotify->filename.Buffer);
AfpFreeMemory(pDelayedNotify);
AfpFreeMemory(newNotifyName.Buffer);
ACQUIRE_SPIN_LOCK(&pVolDesc->vds_VolLock, &OldIrql);
pList = pVolDesc->vds_DelayedNotifyList.Flink; // start from the head again
}
}
return status;
}
/*** afpProcessPrivateNotify
*
* Similar to AfpProcessChangeNotify but optimized/special-cased for private notifies.
*
* LOCKS_ASSUMED: vds_idDbAccessLock (SWMR, Exclusive)
*/
VOID FASTCALL
afpProcessPrivateNotify(
IN PVOL_NOTIFY pVolNotify
)
{
PVOLDESC pVolDesc;
UNICODE_STRING UName;
PFILE_NOTIFY_INFORMATION pFNInfo;
NTSTATUS status;
PDFENTRY pParentDFE;
BOOLEAN CloseParentHandle = False, Verify = True;
BOOLEAN DirModified=TRUE;
BOOLEAN fNewVolume=FALSE;
PFILE_BOTH_DIR_INFORMATION pFBDInfo;
LONGLONG infobuf[(sizeof(FILE_BOTH_DIR_INFORMATION) +
(AFP_LONGNAME_LEN + 1) * sizeof(WCHAR))/sizeof(LONGLONG) + 1];
LONG infobuflen;
BOOLEAN fMemAlloced=FALSE;
PAGED_CODE( );
pFNInfo = (PFILE_NOTIFY_INFORMATION)(pVolNotify + 1);
ASSERT (pFNInfo->Action & AFP_ACTION_PRIVATE);
pVolDesc = pVolNotify->vn_pVolDesc;
ASSERT (VALID_VOLDESC(pVolDesc));
INTERLOCKED_DECREMENT_LONG(&pVolDesc->vds_cPrivateNotifies);
if (pVolDesc->vds_Flags & VOLUME_NEW_FIRST_PASS)
{
fNewVolume = TRUE;
}
pFBDInfo = (PFILE_BOTH_DIR_INFORMATION)infobuf;
AfpInitUnicodeStringWithNonNullTerm(&UName,
(USHORT)pFNInfo->FileNameLength,
pFNInfo->FileName);
// We always allocate extra space for notifies
UName.MaximumLength += (AFP_LONGNAME_LEN+1)*sizeof(WCHAR);
DBGPRINT(DBG_COMP_CHGNOTIFY, DBG_LEVEL_INFO,
("ParentId %d, Path: %Z\n",pVolNotify->vn_ParentId, &UName));
// Lookup the parent DFE using the AfpId stored in the notify
// structure, and setup the UParent and UTail appropriately
pParentDFE = AfpFindDfEntryById(pVolDesc,
pVolNotify->vn_ParentId,
DFE_DIR);
if (pParentDFE != NULL)
{
FILESYSHANDLE ParentHandle, DirHandle;
status = STATUS_SUCCESS;
// Open a handle to the directory relative to the volume root handle
// Special case volume root
ASSERT((UName.Length != 0) || (pVolNotify->vn_ParentId == AFP_ID_ROOT));
do
{
PDFENTRY pDfeNew;
UNICODE_STRING UParent, UTail;
if (pVolNotify->vn_ParentId == AFP_ID_ROOT)
{
AfpSetEmptyUnicodeString(&UParent, 0, NULL);
UTail = UName;
}
else
{
UParent.MaximumLength =
UParent.Length = UName.Length - (USHORT)pVolNotify->vn_TailLength - sizeof(WCHAR);
UParent.Buffer = UName.Buffer;
UTail.MaximumLength =
UTail.Length = (USHORT)pVolNotify->vn_TailLength;
UTail.Buffer = (PWCHAR)((PBYTE)UName.Buffer + UParent.Length + sizeof(WCHAR));
}
if (UName.Length != 0)
{
// Open a handle to parent relative to the volume root handle
status = AfpIoOpen(&pVolDesc->vds_hRootDir,
AFP_STREAM_DATA,
FILEIO_OPEN_DIR,
&UParent,
FILEIO_ACCESS_NONE,
FILEIO_DENY_NONE,
False,
&ParentHandle);
if (!NT_SUCCESS(status))
{
DBGPRINT(DBG_COMP_CHGNOTIFY, DBG_LEVEL_ERR,
("afpProcessPrivateNotify: Failed to open parent: %Z (0x%lx)\n",
&UParent, status));
break;
}
CloseParentHandle = True;
status = AfpIoQueryDirectoryFile(&ParentHandle,
pFBDInfo,
sizeof(infobuf),
FileBothDirectoryInformation,
True,
True,
&UTail);
//
// dir name longer than 31 char? Then we must allocate a buffer
//
if ((status == STATUS_BUFFER_OVERFLOW) ||
(status == STATUS_BUFFER_TOO_SMALL))
{
infobuflen = sizeof(FILE_BOTH_DIR_INFORMATION) +
(MAXIMUM_FILENAME_LENGTH * sizeof(WCHAR));
pFBDInfo = (PFILE_BOTH_DIR_INFORMATION)
AfpAllocNonPagedMemory(infobuflen);
if (pFBDInfo == NULL)
{
status = STATUS_NO_MEMORY;
break;
}
fMemAlloced = TRUE;
status = AfpIoQueryDirectoryFile(&ParentHandle,
pFBDInfo,
infobuflen,
FileBothDirectoryInformation,
True,
True,
&UTail);
}
if (!NT_SUCCESS(status))
{
DBGPRINT(DBG_COMP_CHGNOTIFY, DBG_LEVEL_ERR,
("afpProcessPrivateNotify: AfpIoQueryDirectoryFile failed: %d %Z at %Z(0x%lx)\n",
sizeof(infobuf),&UTail, &UParent,status));
break;
}
// Lookup this entry in the data-base. If not there then we need to add
// it. If its there, we need to verify it.
// NOTE: Use DFE_ANY here and not DFE_DIR !!!
pDfeNew = AfpFindEntryByUnicodeName(pVolDesc,
&UTail,
AFP_LONGNAME,
pParentDFE,
DFE_ANY);
if (pDfeNew == NULL)
{
Verify = False;
}
}
else
{
FILE_BASIC_INFORMATION FBasInfo;
status = AfpIoQueryBasicInfo(&pVolDesc->vds_hRootDir,
&FBasInfo);
if (!NT_SUCCESS(status))
{
DBGPRINT(DBG_COMP_CHGNOTIFY, DBG_LEVEL_ERR,
("afpProcessPrivateNotify: Could not get basic information for root\n"));
status = STATUS_UNSUCCESSFUL;
break;
}
afpConvertBasicToBothDirInfo(&FBasInfo, pFBDInfo);
pDfeNew = pParentDFE;
ParentHandle = pVolDesc->vds_hRootDir;
// Root directory needs special casing. The reason here is that we have
// no parent directory. Also we need to handle the AFP_HAS_CUSTOM_ICON
// bit in the volume descriptor since the finderinfo on the root volume
// doesn't have this bit saved.
if (pVolDesc->vds_Flags & AFP_VOLUME_HAS_CUSTOM_ICON)
{
pDfeNew->dfe_FinderInfo.fd_Attr1 |= FINDER_FLAG_HAS_CUSTOM_ICON;
}
}
if (!Verify)
{
ASSERT(pDfeNew == NULL);
afpAddDfEntryAndCacheInfo(pVolDesc,
pParentDFE,
&UTail,
&ParentHandle,
pFBDInfo,
&UName,
&pDfeNew,
False);
}
else if (pFBDInfo->LastWriteTime.QuadPart > pDfeNew->dfe_LastModTime.QuadPart)
{
pDfeNew->dfe_Flags &= ~DFE_FLAGS_INIT_COMPLETED;
afpVerifyDFE(pVolDesc,
pParentDFE,
&UTail,
&ParentHandle,
pFBDInfo,
&UName,
&pDfeNew);
DirModified = TRUE;
}
else
{
DirModified = FALSE;
}
if (pDfeNew == NULL)
{
DBGPRINT(DBG_COMP_CHGNOTIFY, DBG_LEVEL_ERR,
("afpProcessPrivateNotify: Could not add DFE for %Z\n", &UName));
status = STATUS_UNSUCCESSFUL;
break;
}
pParentDFE = pDfeNew;
//
// Now open the directory itself so that it can be enumerated
// Open the directory relative to its parent since we already
// have the parent handle open.
//
if (Verify && !DirModified && (pVolNotify->vn_ParentId != AFP_ID_ROOT))
{
}
else
{
status = AfpIoOpen(&ParentHandle,
AFP_STREAM_DATA,
FILEIO_OPEN_DIR,
&UTail,
FILEIO_ACCESS_NONE,
FILEIO_DENY_NONE,
False,
&DirHandle);
if (!NT_SUCCESS(status))
{
DBGPRINT(DBG_COMP_CHGNOTIFY, DBG_LEVEL_ERR,
("afpProcessPrivateNotify: AfpIoOpen failed: %Z (0x%lx)\n",
&UTail, status));
break;
}
}
} while (False);
if (fMemAlloced)
{
ASSERT(pFBDInfo != ((PFILE_BOTH_DIR_INFORMATION)(infobuf)));
AfpFreeMemory(pFBDInfo);
}
if (CloseParentHandle)
{
AfpIoClose(&ParentHandle);
}
if (NT_SUCCESS(status))
{
DWORD Method;
// Always get the root level files
if (Verify && !DirModified && (pVolNotify->vn_ParentId != AFP_ID_ROOT))
{
Method = GETDIRSKELETON;
}
else
{
Method = (GETENTIRETREE | REENUMERATE);
}
status = AfpCacheDirectoryTree(pVolDesc,
pParentDFE,
Method,
&DirHandle,
&UName);
if (!NT_SUCCESS(status))
{
DBGPRINT(DBG_COMP_CHGNOTIFY, DBG_LEVEL_ERR,
("afpProcessPrivateNotify: CacheDir failed %lx tree for %Z\n", status,&UName));
}
if (Method != GETDIRSKELETON)
{
AfpIoClose(&DirHandle);
}
}
}
afpActivateVolume(pVolDesc);
}
/*** afpActivateVolume
*
* If we just finished caching in the directory structure, activate the volume now.
* This is keyed off the AFP_INITIAL_CACHE bit in the volume flags.
*/
VOID FASTCALL
afpActivateVolume(
IN PVOLDESC pVolDesc
)
{
BOOLEAN fCdfs;
KIRQL OldIrql;
NTSTATUS Status = STATUS_SUCCESS;
UNICODE_STRING RootName;
PVOLDESC pWalkVolDesc;
IDDBHDR IdDbHdr;
BOOLEAN fPostIrp;
BOOLEAN fRetry=TRUE;
LARGE_INTEGER ActivationTime;
ULONG HighPart;
ULONG LowPart;
ACQUIRE_SPIN_LOCK(&pVolDesc->vds_VolLock, &OldIrql);
// if we have more notifies queued up, we aren't done yet
if (pVolDesc->vds_cPrivateNotifies != 0)
{
RELEASE_SPIN_LOCK(&pVolDesc->vds_VolLock, OldIrql);
return;
}
//
// ok, we're here because the scan of the volume completed
//
//
// if this was a newly created volume and if this was its first pass, we must
// post the change-notify irp and restart the scan
//
if (pVolDesc->vds_Flags & VOLUME_NEW_FIRST_PASS)
{
pVolDesc->vds_Flags &= ~VOLUME_NEW_FIRST_PASS;
// we post a change notify irp if this volume is not an exclusive volume
fPostIrp = (!EXCLUSIVE_VOLUME(pVolDesc));
RELEASE_SPIN_LOCK(&pVolDesc->vds_VolLock, OldIrql);
if (fPostIrp)
{
// Begin monitoring changes to the tree. Even though we may
// start processing PC changes before we have finished
// enumerating the tree, if we get notified of part of the
// tree we have yet to cache (and therefore can't find it's
// path in our database its ok, since we will end up
// picking up the change when we enumerate that branch. Also,
// by posting this before starting to cache the tree instead
// of after, we will pick up any changes that are made to parts
// of the tree we have already seen, otherwise we would miss
// those.
// Explicitly reference this volume for ChangeNotifies and post it
ASSERT (KeGetCurrentIrql() < DISPATCH_LEVEL);
DBGPRINT(DBG_COMP_VOLUME, DBG_LEVEL_ERR,
("AfpAdmWVolumeAdd: posting chg-notify irp on volume %Z in second pass\n",
&pVolDesc->vds_Name));
if (AfpVolumeReference(pVolDesc))
{
pVolDesc->vds_RequiredNotifyBufLen = AFP_VOLUME_NOTIFY_STARTING_BUFSIZE;
Status = AfpVolumePostChangeNotify(pVolDesc);
if (NT_SUCCESS(Status))
{
// scan the entire directory tree and sync disk with iddb, now that
// we are in the second pass for this (not-so) newly created volume
AfpSetEmptyUnicodeString(&RootName, 0, NULL);
AfpQueuePrivateChangeNotify(pVolDesc,
&RootName,
&RootName,
AFP_ID_ROOT);
}
else
{
AFPLOG_ERROR(AFPSRVMSG_START_VOLUME,
Status,
NULL,
0,
&pVolDesc->vds_Name);
DBGPRINT(DBG_COMP_VOLUME, DBG_LEVEL_ERR,
("AfpAdmWVolumeAdd: posting chg-notify failed (%lx)!!\n",Status));
AfpVolumeDereference(pVolDesc);
ASSERT(0);
}
}
else
{
DBGPRINT(DBG_COMP_VOLUME, DBG_LEVEL_ERR,
("AfpAdmWVolumeAdd: couldn't reference volume %Z!!\n",&pVolDesc->vds_Name));
}
}
}
//
// ok, we are through with all the passes and the scan is successful:
// mark the volume as 'officially' available to the clients
//
else if (pVolDesc->vds_Flags & VOLUME_INITIAL_CACHE)
{
pVolDesc->vds_Flags |= VOLUME_SCAVENGER_RUNNING;
pVolDesc->vds_Flags &= ~VOLUME_INITIAL_CACHE;
RELEASE_SPIN_LOCK(&pVolDesc->vds_VolLock, OldIrql);
// Set the type of the ICON<0d> file to 0's so that
// mac apps will not list it in their file-open dialogs
fCdfs = !IS_VOLUME_NTFS(pVolDesc);
if (!fCdfs)
{
PDFENTRY pdfetmp;
UNICODE_STRING iconstr;
WCHAR iconname[5] = AFPSERVER_VOLUME_ICON_FILE_ANSI;
AfpInitUnicodeStringWithNonNullTerm(&iconstr,
sizeof(iconname),
iconname);
if ((pdfetmp = AfpFindEntryByUnicodeName(pVolDesc,
&iconstr,
AFP_LONGNAME,
pVolDesc->vds_pDfeRoot,
DFE_FILE)) != NULL)
{
pdfetmp->dfe_FinderInfo.fd_TypeD = 0;
}
// Kick off the OurChange scavenger scheduled routine
// Explicitly reference this for the scavenger routine
if (AfpVolumeReference(pVolDesc))
{
// Schedule the scavenger to run periodically. This scavenger
// is queued since it acquires a SWMR.
AfpScavengerScheduleEvent(AfpOurChangeScavenger,
pVolDesc,
VOLUME_OURCHANGE_AGE,
False);
}
else
{
DBGPRINT(DBG_COMP_VOLUME, DBG_LEVEL_ERR,
("afpActivateVolume: couldn't reference volume %Z!!\n",
&pVolDesc->vds_Name));
}
}
Status = AfpIoQueryTimesnAttr(&pVolDesc->vds_hRootDir,
&pVolDesc->vds_pDfeRoot->dfe_CreateTime,
&pVolDesc->vds_pDfeRoot->dfe_LastModTime,
NULL);
pVolDesc->vds_CreateTime = pVolDesc->vds_pDfeRoot->dfe_CreateTime;
if (NT_SUCCESS(Status))
{
pVolDesc->vds_ModifiedTime =
AfpConvertTimeToMacFormat(&pVolDesc->vds_pDfeRoot->dfe_LastModTime);
}
else
{
pVolDesc->vds_ModifiedTime = pVolDesc->vds_pDfeRoot->dfe_CreateTime;
}
// Kick off the scavenger thread scheduled routine
// Explicitly reference this for the scavenger routine
if (AfpVolumeReference(pVolDesc))
{
//
// let's save the index file the way we know it now
// (setting vds_cScvgrIdDb to 1 will trigger it via scavenger thread)
//
if (IS_VOLUME_NTFS(pVolDesc))
{
ACQUIRE_SPIN_LOCK(&pVolDesc->vds_VolLock, &OldIrql);
pVolDesc->vds_cScvgrIdDb = 1;
RELEASE_SPIN_LOCK(&pVolDesc->vds_VolLock, OldIrql);
}
// Schedule the scavenger to run periodically. Always make the
// scavenger use the worker thread for CD-ROM volumes since we
// 'nudge' it every invocation to see if the CD is in the drive
AfpScavengerScheduleEvent(AfpVolumeScavenger,
pVolDesc,
fCdfs ?
VOLUME_CDFS_SCAVENGER_INTERVAL :
VOLUME_NTFS_SCAVENGER_INTERVAL,
fCdfs);
//
// another workaround for an Apple bug. If creation date on two
// volumes is identical, the alias manager gets all confused and points
// alias for one guy to another!
// See if this volume's creation date is the same as any of the other
// volume's creation date: if so, add 1 second
//
while (fRetry)
{
fRetry = FALSE;
ACQUIRE_SPIN_LOCK(&AfpVolumeListLock, &OldIrql);
for (pWalkVolDesc = AfpVolumeList;
pWalkVolDesc != NULL;
pWalkVolDesc = pWalkVolDesc->vds_Next)
{
if (pWalkVolDesc == pVolDesc)
{
continue;
}
if (pWalkVolDesc->vds_CreateTime == pVolDesc->vds_CreateTime)
{
DBGPRINT(DBG_COMP_CHGNOTIFY, DBG_LEVEL_ERR,
("Vol creation date conflict: %Z and %Z. Hack-o-rama at work\n",
&pVolDesc->vds_Name,&pWalkVolDesc->vds_Name));
pVolDesc->vds_CreateTime += 1;
fRetry = TRUE;
break;
}
}
RELEASE_SPIN_LOCK(&AfpVolumeListLock,OldIrql);
}
KeQuerySystemTime (&ActivationTime);
DBGPRINT(DBG_COMP_CHGNOTIFY, DBG_LEVEL_ERR,
("afpActivateVolume: volume %Z activated %s at %8lx%08lx , max notifies=%ld\n",
&pVolDesc->vds_Name,
(pVolDesc->vds_Flags & AFP_VOLUME_SUPPORTS_CATSRCH) ?
" " : "(CatSearch disabled)",
0xffffffff*ActivationTime.HighPart,
0xffffffff*ActivationTime.LowPart,
pVolDesc->vds_maxPrivateNotifies
));
if ((int)(pVolDesc->vds_IndxStTime.LowPart-ActivationTime.LowPart) >= 0)
{
LowPart = pVolDesc->vds_IndxStTime.LowPart-ActivationTime.LowPart;
HighPart = pVolDesc->vds_IndxStTime.HighPart-ActivationTime.HighPart;
}
else
{
LowPart = 0xffffffff - ActivationTime.LowPart + 1 + pVolDesc->vds_IndxStTime.LowPart;
HighPart = pVolDesc->vds_IndxStTime.HighPart-ActivationTime.HighPart -1;
}
DBGPRINT(DBG_COMP_CHGNOTIFY, DBG_LEVEL_ERR,
("Time taken for indexing = %8lx%08lx\n",
0xffffffff*(pVolDesc->vds_IndxStTime.HighPart-ActivationTime.HighPart),
0xffffffff*(pVolDesc->vds_IndxStTime.LowPart-ActivationTime.LowPart)
));
}
else
{
DBGPRINT(DBG_COMP_CHGNOTIFY, DBG_LEVEL_ERR,
("afpActivateVolume: couldn't reference volume %Z\n",&pVolDesc->vds_Name));
}
}
else
{
RELEASE_SPIN_LOCK(&pVolDesc->vds_VolLock, OldIrql);
}
}
/*** AfpShouldWeIgnoreThisNotification
*
* Check to see if this notification should be ignored. The following events are
* ignored.
*
* (((Action == FILE_ACTION_MODIFIED_STREAM) &&
* (Stream != AFP_RESC_STREAM) &&
* (Stream != AFP_INFO_STREAM)) ||
* (Its one of our own changes))
*
* LOCKS: vds_VolLock (SPIN)
*/
BOOLEAN FASTCALL
AfpShouldWeIgnoreThisNotification(
IN PVOL_NOTIFY pVolNotify
)
{
PFILE_NOTIFY_INFORMATION pFNInfo;
PVOLDESC pVolDesc;
UNICODE_STRING UName;
BOOLEAN ignore = False;
pFNInfo = (PFILE_NOTIFY_INFORMATION)(pVolNotify + 1);
pVolDesc = pVolNotify->vn_pVolDesc;
AfpInitUnicodeStringWithNonNullTerm(&UName,
(USHORT)pFNInfo->FileNameLength,
pFNInfo->FileName);
DBGPRINT(DBG_COMP_CHGNOTIFY, DBG_LEVEL_INFO,
("AfpShouldWeIgnoreThisNotification: Action: %d Name: %Z\n", pFNInfo->Action, &UName));
pVolNotify->vn_StreamId = AFP_STREAM_DATA;
if (pFNInfo->Action == FILE_ACTION_MODIFIED_STREAM)
{
UNICODE_STRING UStream;
ignore = True;
UStream.Length = UStream.MaximumLength = AfpInfoStream.Length;
UStream.Buffer = (PWCHAR)((PBYTE)UName.Buffer +
UName.Length - AfpInfoStream.Length);
if (EQUAL_UNICODE_STRING(&UStream, &AfpInfoStream, False))
{
pVolNotify->vn_StreamId = AFP_STREAM_INFO;
pFNInfo->FileNameLength -= AfpInfoStream.Length;
UName.Length -= AfpInfoStream.Length;
ignore = False;
}
else
{
UStream.Length = UStream.MaximumLength = AfpResourceStream.Length;
UStream.Buffer = (PWCHAR)((PBYTE)UName.Buffer +
UName.Length - AfpResourceStream.Length);
if (EQUAL_UNICODE_STRING(&UStream, &AfpResourceStream, False))
{
pVolNotify->vn_StreamId = AFP_STREAM_RESC;
pFNInfo->FileNameLength -= AfpResourceStream.Length;
UName.Length -= AfpResourceStream.Length;
ignore = False;
}
}
}
if (!ignore)
{
PLIST_ENTRY pList, pListHead;
POUR_CHANGE pChange;
DWORD afpChangeAction;
KIRQL OldIrql;
afpChangeAction = AFP_CHANGE_ACTION(pFNInfo->Action);
ASSERT(afpChangeAction <= AFP_CHANGE_ACTION_MAX);
ACQUIRE_SPIN_LOCK(&pVolDesc->vds_VolLock, &OldIrql);
// point to the head of the appropriate change action list
pListHead = &pVolDesc->vds_OurChangeList[afpChangeAction];
for (pList = pListHead->Flink;
pList != pListHead;
pList = pList->Flink)
{
pChange = CONTAINING_RECORD(pList, OUR_CHANGE, oc_Link);
// do a case *sensitive* unicode string compare
if (EQUAL_UNICODE_STRING_CS(&UName, &pChange->oc_Path))
{
RemoveEntryList(&pChange->oc_Link);
AfpFreeMemory(pChange);
// We were notified of our own change
ignore = True;
if (pFNInfo->Action == FILE_ACTION_RENAMED_OLD_NAME)
{
// consume the RENAMED_NEW_NAME if it exists
if (pFNInfo->NextEntryOffset != 0)
{
PFILE_NOTIFY_INFORMATION pFNInfo2;
(PBYTE)pFNInfo2 = (PBYTE)pFNInfo + pFNInfo->NextEntryOffset;
if (pFNInfo2->Action == FILE_ACTION_RENAMED_NEW_NAME)
{
ASSERT(pFNInfo2->NextEntryOffset == 0);
}
else
{
DBGPRINT(DBG_COMP_CHGNOTIFY, DBG_LEVEL_ERR,
("AfpShouldWeIgnoreThisNotification: Our Rename did not come with new name!!!\n"));
}
}
else
{
DBGPRINT(DBG_COMP_CHGNOTIFY, DBG_LEVEL_ERR,
("AfpShouldWeIgnoreThisNotification: Our Rename did not come with new name!!!\n"));
}
} // if rename
else
{
// We are ignoring this notify. Make sure its not a multiple
ASSERT(pFNInfo->NextEntryOffset == 0);
}
break;
}
} // while there are more of our changes to look thru
RELEASE_SPIN_LOCK(&pVolDesc->vds_VolLock, OldIrql);
}
else
{
// We are ignoring this notify. Make sure its not a multiple
ASSERT(pFNInfo ->NextEntryOffset == 0);
}
if (!ignore)
{
INTERLOCKED_INCREMENT_LONG(&pVolDesc->vds_cOutstandingNotifies);
DBGPRINT(DBG_COMP_CHGNOTIFY, DBG_LEVEL_INFO,
("ShouldWe: Action: %d Name: \t\t\t\t\t\n%Z\n", pFNInfo->Action, &UName));
}
return ignore;
}
/*** AfpCacheDirectoryTree
*
* Scan a directory tree and build the idindex database from this information.
* Do a breadth-first search. On volume start, this will cache the tree
* beginning at the root directory. For Directory ADDED notifications, this
* will cache the subtree beginning at the added directory, since a PC user can
* potentially move an entire subtree into a mac volume, but we will only
* be notified of the one directory addition.
*
* Only the first level is actually handled here. Sub-directories are queued up
* as faked notifies and handled that way.
*
* Method:
* REENUMERATE: In case we need to reenumerate just
* this level in the tree in order to
* get rid of any dead wood that a PC
* removed by its 'other' name
*
* GETDIRSKELETON: When we want to bring in only the skeleton
* of the tree by adding directories only
*
* GETFILES: When we need to fill in the files for this
* level of the tree because a mac has accessed
* a dir for the first time.
*
* GetDirSkeletonAndFiles:
* A Combination of the above two.
* When we want to bring in both files and directories.
* This is used when adding a volume, and we want the
* files in the root directory cached in, but no others.
* Also this will be used if we are rebuilding the
* Desktop database APPLs while caching the disk tree.
* The private ChangeNotifies we queue up for ADDED dirs
* will also read in the files if the volume is marked
* for rebuilding of the desktop.
*
* GETENTIRETREE: When we want to cache in the entire tree
*
* GetEntireTreeAndReEnumerate:
* Combines GETENTIRETREE and REENUMERATE
*
* LOCKS_ASSUMED: vds_IdDbAccessLock (SWMR, Exclusive)
*/
NTSTATUS
AfpCacheDirectoryTree(
IN PVOLDESC pVolDesc,
IN PDFENTRY pDFETreeRoot, // DFE of the tree root directory
IN DWORD Method, // See explanation in procedure comment
IN PFILESYSHANDLE phRootDir OPTIONAL, // open handle to tree root directory
IN PUNICODE_STRING pDirPath OPTIONAL
)
{
UNICODE_STRING UName, Path, ParentPath;
UNICODE_STRING RootName;
PDFENTRY pDFE;
PDFENTRY pChainDFE;
PDFENTRY pCurrDfe;
NTSTATUS Status = STATUS_SUCCESS;
PBYTE enumbuf = NULL;
PFILE_BOTH_DIR_INFORMATION pNextEntry;
FILESYSHANDLE fshEnumDir;
USHORT SavedPathLength;
BOOLEAN fQueueThisSubDir=FALSE;
BOOLEAN fOneSubDirAlreadyQueued=FALSE;
BOOLEAN fAllSubDirsVisited=TRUE;
BOOLEAN fExitLoop=FALSE;
#ifdef PROFILING
TIME TimeS, TimeE, TimeD;
DWORD NumScanned = 0;
AfpGetPerfCounter(&TimeS);
#endif
PAGED_CODE( );
ASSERT (VALID_DFE(pDFETreeRoot));
ASSERT (DFE_IS_DIRECTORY(pDFETreeRoot));
ASSERT((Method != GETFILES) || !DFE_CHILDREN_ARE_PRESENT(pDFETreeRoot));
// allocate the buffer that will hold enumerated files and dirs
if ((pVolDesc->vds_EnumBuffer == NULL) &&
((pVolDesc->vds_EnumBuffer = (PBYTE)AfpAllocPANonPagedMemory(AFP_ENUMBUF_SIZE)) == NULL))
{
return STATUS_NO_MEMORY;
}
do
{
fshEnumDir.fsh_FileHandle = NULL;
enumbuf = pVolDesc->vds_EnumBuffer;
// Get the volume root relative path to the directory tree being scanned
// Get extra space for one more entry to tag on for queuing notifies.
// In case we already have the path corres. to directory we are attempting
// to cache, get it from there. Note that in this case we are always
// guaranteed that extra space is available
if (ARGUMENT_PRESENT(pDirPath))
{
Path = *pDirPath;
}
else
{
AfpSetEmptyUnicodeString(&Path, 0, NULL);
Status = AfpHostPathFromDFEntry(pDFETreeRoot,
(AFP_LONGNAME_LEN+1)*sizeof(WCHAR),
&Path);
if (!NT_SUCCESS(Status))
{
break;
}
}
DBGPRINT(DBG_COMP_IDINDEX, DBG_LEVEL_INFO,
("AfpCacheDirectoryTree: ParentId %d, Path %Z\n",
pDFETreeRoot->dfe_AfpId, &Path));
if (Method != GETDIRSKELETON)
{
if (!ARGUMENT_PRESENT(phRootDir))
{
// Need to open a handle to the directory in order to enumerate
if (NT_SUCCESS(Status = AfpIoOpen(&pVolDesc->vds_hRootDir,
AFP_STREAM_DATA,
FILEIO_OPEN_DIR,
&Path,
FILEIO_ACCESS_NONE,
FILEIO_DENY_NONE,
False,
&fshEnumDir)))
{
phRootDir = &fshEnumDir;
}
else
{
break;
}
}
}
SavedPathLength = Path.Length;
DBGPRINT(DBG_COMP_IDINDEX, DBG_LEVEL_INFO,
("AfpCacheDirectoryTree: Scanning Tree: %Z\n", &Path));
if (Method & REENUMERATE)
{
afpMarkAllChildrenUnseen(pDFETreeRoot);
}
if (Method != GETDIRSKELETON)
{
while (True)
{
// keep enumerating till we get all the entries
Status = AfpIoQueryDirectoryFile(phRootDir,
(PFILE_BOTH_DIR_INFORMATION)enumbuf,
AFP_ENUMBUF_SIZE,
FileBothDirectoryInformation,
False, // return multiple entries
False, // don't restart scan
NULL);
ASSERT(Status != STATUS_PENDING);
if (Status != STATUS_SUCCESS)
{
if ((Status == STATUS_NO_MORE_FILES) ||
(Status == STATUS_NO_SUCH_FILE))
{
Status = STATUS_SUCCESS;
break; // that's it, we've seen everything there is
}
else
{
AFPLOG_HERROR(AFPSRVMSG_ENUMERATE,
Status,
NULL,
0,
phRootDir->fsh_FileHandle);
DBGPRINT(DBG_COMP_IDINDEX, DBG_LEVEL_ERR,
("AfpCacheDirectoryTree: dir enum failed %lx\n", Status));
break; // enumerate failed, bail out
}
}
// process the enumerated files and dirs in the current enumbuf
pNextEntry = (PFILE_BOTH_DIR_INFORMATION)enumbuf;
while (True)
{
BOOLEAN IsDir, WriteBackROAttr, FixIt;
WCHAR wc;
PFILE_BOTH_DIR_INFORMATION pCurrEntry;
fQueueThisSubDir = FALSE;
if (pNextEntry == NULL)
{
Status = STATUS_NO_MORE_ENTRIES;
break;
}
WriteBackROAttr = False;
IsDir = False;
// Move the structure to the next entry or NULL if we hit the end
pCurrEntry = pNextEntry;
(PBYTE)pNextEntry += pCurrEntry->NextEntryOffset;
if (pCurrEntry->NextEntryOffset == 0)
{
pNextEntry = NULL;
}
if (pCurrEntry->FileAttributes & FILE_ATTRIBUTE_DIRECTORY)
{
// Ignore dirs if we are only getting files for this level
if (Method == GETFILES)
{
continue;
}
IsDir = True;
}
else if (Method == GETDIRSKELETON)
{
// Ignore files if we are only getting the dir skeleton
continue;
}
// If NT name > AFP_LONGNAME_LEN, use the NT shortname for
// Mac longname on NTFS, any other file system the shortname
// will be null, so ignore the file
//if (pCurrEntry->FileNameLength <= (AFP_LONGNAME_LEN*sizeof(WCHAR)))
AfpInitUnicodeStringWithNonNullTerm(&UName,
(USHORT)pCurrEntry->FileNameLength,
pCurrEntry->FileName);
if ((RtlUnicodeStringToAnsiSize(&UName)-1) <= AFP_LONGNAME_LEN)
{
AfpInitUnicodeStringWithNonNullTerm(&UName,
(USHORT)pCurrEntry->FileNameLength,
pCurrEntry->FileName);
}
else if (pCurrEntry->ShortNameLength > 0)
{
AfpInitUnicodeStringWithNonNullTerm(&UName,
(USHORT)pCurrEntry->ShortNameLength,
pCurrEntry->ShortName);
}
else
{
DBGPRINT(DBG_COMP_IDINDEX, DBG_LEVEL_ERR,
("AfpCacheDirectoryTree: Name is > 31 with no short name ?\n"));
continue;
}
if (IsDir &&
(EQUAL_UNICODE_STRING_CS(&Dot, &UName) ||
EQUAL_UNICODE_STRING_CS(&DotDot, &UName)))
{
continue;
}
// Check if this entry is an invalid win32 name i.e. it has either
// a period or a space at end, if so convert it to the new format.
// NOTE: can we construct a path to use to catch our own changes ?
wc = UName.Buffer[(UName.Length - 1)/sizeof(WCHAR)];
if ((wc == UNICODE_SPACE) || (wc == UNICODE_PERIOD))
{
// NOTE: MacCD driver should fix this??
if (IS_VOLUME_NTFS(pVolDesc))
{
afpRenameInvalidWin32Name(phRootDir, IsDir, &UName);
}
}
#ifdef PROFILING
NumScanned++;
#endif
pDFE = NULL;
FixIt = False;
if (Method & REENUMERATE)
{
// If we have this item in our DB, just mark it as seen.
// Use DFE_ANY here since a mismatch is fatal.
afpFindDFEByUnicodeNameInSiblingList(pVolDesc,
pDFETreeRoot,
&UName,
&pDFE,
DFE_ANY);
if (pDFE != NULL)
{
// If we have a wrong type, fix it.
if (IsDir ^ DFE_IS_DIRECTORY(pDFE))
{
AfpDeleteDfEntry(pVolDesc, pDFE);
pDFE = NULL;
FixIt = True;
}
else
{
DFE_MARK_AS_SEEN(pDFE);
}
}
}
if ((Method != REENUMERATE) || FixIt)
{
// add this entry to the idindex database, and cache all the required
// information, but only for files since the directories are queued
// back and added at that time.
if (!IsDir)
{
// Construct a full path to the file in order to filter our
// own changes to AFP_AfpInfo stream when adding the file
if (Path.Length > 0)
{
// Append a path separator
Path.Buffer[Path.Length / sizeof(WCHAR)] = L'\\';
Path.Length += sizeof(WCHAR);
}
ASSERT(Path.Length + UName.Length <= Path.MaximumLength);
RtlAppendUnicodeStringToString(&Path, &UName);
if (pDFE == NULL)
{
afpAddDfEntryAndCacheInfo(pVolDesc,
pDFETreeRoot,
&UName,
phRootDir,
pCurrEntry,
&Path,
&pDFE,
True);
}
else if (pCurrEntry->LastWriteTime.QuadPart > pDFE->dfe_LastModTime.QuadPart)
{
afpVerifyDFE(pVolDesc,
pDFETreeRoot,
&UName,
phRootDir,
pCurrEntry,
&Path,
&pDFE);
}
// Restore the original length of the path to enum dir
Path.Length = SavedPathLength;
if (pDFE == NULL)
{
// one reason this could fail is if we encounter pagefile.sys
// if our volume is rooted at the drive root
DBGPRINT(DBG_COMP_IDINDEX, DBG_LEVEL_WARN,
("AfpCacheDirectoryTree: AddDfEntry failed %Z at %Z\n",&UName,&Path));
continue;
}
else if (Method == (GETENTIRETREE | REENUMERATE))
{
DFE_MARK_AS_SEEN(pDFE);
}
}
else
{
ASSERT(IsDir);
ASSERT ((Method != GETFILES) &&
(Method != REENUMERATE));
// queue this directory as a simulated Notify of a directory add.
// If we have too much stuff on the queue already, then queue
// only one subdirectory so that we are guaranteed to eventually
// visit all the subdirs. Also, on a huge volume, we want to limit
// how many directories enque per level of the tree
//
fQueueThisSubDir = TRUE;
// We dont use this flag DFE_FLAGS_INIT_COMPLETED
// anymore. So, reducing one lookup
#if 0
pCurrDfe = AfpFindEntryByUnicodeName(pVolDesc,
&UName,
AFP_LONGNAME,
pDFETreeRoot,
DFE_ANY);
//
// if this subdir is already complete, skip it
//
if ((pCurrDfe != NULL) &&
(pCurrDfe->dfe_Flags & DFE_FLAGS_INIT_COMPLETED))
{
fQueueThisSubDir = FALSE;
}
#endif
if (fQueueThisSubDir)
{
AfpQueuePrivateChangeNotify(pVolDesc,
&UName,
&Path,
pDFETreeRoot->dfe_AfpId);
}
}
}
if (fExitLoop)
{
break;
}
} // while more entries in the enumbuf
if ((!NT_SUCCESS(Status) && (Status != STATUS_NO_MORE_ENTRIES)) ||
(fExitLoop))
{
break;
}
} // while there are more files to enumerate
if (NT_SUCCESS(Status))
{
if (Method & REENUMERATE)
{
afpPruneUnseenChildren(pVolDesc, pDFETreeRoot);
}
DFE_MARK_CHILDREN_PRESENT(pDFETreeRoot);
}
else
{
DBGPRINT(DBG_COMP_IDINDEX, DBG_LEVEL_ERR,
("AfpCacheDirectoryTree: Status %lx\n", Status));
}
} /* if Method != GETDIRSKELETON */
else if (Method == GETDIRSKELETON)
{
PDFENTRY pcurrDfEntry;
PDFENTRY pDfEntry;
pcurrDfEntry = (pDFETreeRoot)->dfe_pDirEntry->de_ChildDir;
do
{
for (NOTHING;
pcurrDfEntry != NULL;
pcurrDfEntry = pcurrDfEntry->dfe_NextSibling)
{
#if 0
if (((*(_ppDfEntry))->dfe_NameHash == NameHash) &&
EQUAL_UNICODE_STRING(&((*(_ppDfEntry))->dfe_UnicodeName),
_pName,
True))
{
afpUpdateDfeAccessTime(_pVolDesc, *(_ppDfEntry));
Found = True;
break;
}
#endif
AfpQueuePrivateChangeNotify(pVolDesc,
&(pcurrDfEntry->dfe_UnicodeName),
&Path,
pDFETreeRoot->dfe_AfpId);
}
Status = STATUS_SUCCESS;
} while (False);
} /* if Method == GETDIRSKELETON */
} while (False);
ASSERT (enumbuf != NULL);
if ((pVolDesc->vds_cPrivateNotifies == 0) &&
(pVolDesc->vds_cOutstandingNotifies == 0))
{
if (enumbuf != NULL)
{
AfpFreePANonPagedMemory(enumbuf, AFP_ENUMBUF_SIZE);
}
pVolDesc->vds_EnumBuffer = NULL;
}
ASSERT (Path.Buffer != NULL);
if (!ARGUMENT_PRESENT(pDirPath) && (Path.Buffer != NULL))
{
AfpFreeMemory(Path.Buffer);
}
if (fshEnumDir.fsh_FileHandle != NULL)
{
AfpIoClose(&fshEnumDir);
}
#ifdef PROFILING
AfpGetPerfCounter(&TimeE);
TimeD.QuadPart = TimeE.QuadPart - TimeS.QuadPart;
INTERLOCKED_ADD_ULONG(&AfpServerProfile->perf_ScanTreeCount,
NumScanned,
&AfpStatisticsLock);
INTERLOCKED_ADD_LARGE_INTGR(&AfpServerProfile->perf_ScanTreeTime,
TimeD,
&AfpStatisticsLock);
#endif
return Status;
}
BOOLEAN FASTCALL
AfpVolumeAbortIndexing(
IN PVOLDESC pVolDesc
)
{
KIRQL OldIrql;
PKQUEUE pNotifyQueue;
PLIST_ENTRY pNotifyList;
PLIST_ENTRY pPrivateNotifyList;
LIST_ENTRY TransitionList;
PLIST_ENTRY pList, pNext;
LARGE_INTEGER Immediate;
PVOL_NOTIFY pVolNotify;
LONG index;
DWORD DerefCount=0;
DWORD PvtNotifyCount=0;
BOOLEAN fResult=TRUE;
BOOLEAN fNewVolume=FALSE;
BOOLEAN fCancelNotify=FALSE;
DBGPRINT(DBG_COMP_CHGNOTIFY, DBG_LEVEL_ERR,
("AbortIndexing: Aborting Index for Volume\n"));
ACQUIRE_SPIN_LOCK(&pVolDesc->vds_VolLock, &OldIrql);
fNewVolume = (pVolDesc->vds_Flags & VOLUME_NEW_FIRST_PASS) ? TRUE : FALSE;
pVolDesc->vds_Flags |= VOLUME_DELETED;
// set this so we don't reset the Indexing global flag again!
pVolDesc->vds_Flags |= VOLUME_INTRANSITION;
if (pVolDesc->vds_Flags & VOLUME_NOTIFY_POSTED)
{
ASSERT(pVolDesc->vds_pIrp != NULL);
fCancelNotify = TRUE;
}
RELEASE_SPIN_LOCK(&pVolDesc->vds_VolLock, OldIrql);
if (fCancelNotify)
{
IoCancelIrp(pVolDesc->vds_pIrp);
}
InitializeListHead(&TransitionList);
index = pVolDesc->vds_VolId % NUM_NOTIFY_QUEUES;
pNotifyQueue = &AfpVolumeNotifyQueue[index];
pNotifyList = &AfpVolumeNotifyList[index];
pPrivateNotifyList = &AfpVirtualMemVolumeNotifyList[index];
Immediate.HighPart = Immediate.LowPart = 0;
while (1)
{
pList = KeRemoveQueue(pNotifyQueue, KernelMode, &Immediate);
//
// finished the list?
//
if ((NTSTATUS)((ULONG_PTR)pList) == STATUS_TIMEOUT)
{
break;
}
pVolNotify = CONTAINING_RECORD(pList, VOL_NOTIFY, vn_List);
//
// some other notifications? keep them on temp list for now
//
if ((pVolNotify->vn_pVolDesc != pVolDesc) ||
(pVolNotify == &AfpTerminateNotifyThread))
{
InsertTailList(&TransitionList, pList);
}
//
// notification for this volume: get rid of it
//
else
{
ASSERT(pVolNotify->vn_pVolDesc == pVolDesc);
ASSERT((pVolNotify->vn_TimeStamp == AFP_QUEUE_NOTIFY_IMMEDIATELY) ||
(!fNewVolume));
// was this a private notify?
if (((PFILE_NOTIFY_INFORMATION)(pVolNotify + 1))->Action & AFP_ACTION_PRIVATE)
{
INTERLOCKED_DECREMENT_LONG(&pVolDesc->vds_cPrivateNotifies);
}
AfpFreeMemory(pVolNotify);
AfpVolumeDereference(pVolDesc);
AfpNotifyQueueCount[index]--;
}
}
ACQUIRE_SPIN_LOCK(&AfpVolumeListLock, &OldIrql);
pList = pNotifyList->Flink;
while (pList != pNotifyList)
{
pVolNotify = CONTAINING_RECORD(pList, VOL_NOTIFY, vn_List);
pNext = pList->Flink;
if (pVolNotify->vn_pVolDesc == pVolDesc)
{
RemoveEntryList(pList);
// was this a private notify?
if (((PFILE_NOTIFY_INFORMATION)(pVolNotify + 1))->Action & AFP_ACTION_PRIVATE)
{
PvtNotifyCount++;
}
DerefCount++;
AfpFreeMemory(pVolNotify);
AfpNotifyListCount[index]--;
}
pList = pNext;
}
RELEASE_SPIN_LOCK(&AfpVolumeListLock, OldIrql);
ASSERT (KeGetCurrentIrql() < DISPATCH_LEVEL);
AfpSwmrAcquireExclusive(&AfpVolumeListSwmr);
pList = pPrivateNotifyList->Flink;
while (pList != pPrivateNotifyList)
{
pVolNotify = CONTAINING_RECORD(pList, VOL_NOTIFY, vn_List);
pNext = pList->Flink;
if (pVolNotify->vn_pVolDesc == pVolDesc)
{
RemoveEntryList(pList);
afpFreeNotify(pVolNotify);
AfpVolumeDereference(pVolDesc);
//AfpNotifyListCount[index]--;
}
pList = pNext;
}
AfpSwmrRelease(&AfpVolumeListSwmr);
if (DerefCount > 0)
{
ACQUIRE_SPIN_LOCK(&pVolDesc->vds_VolLock, &OldIrql);
ASSERT(pVolDesc->vds_RefCount >= DerefCount);
ASSERT(pVolDesc->vds_cPrivateNotifies >= (LONG)PvtNotifyCount);
pVolDesc->vds_RefCount -= (DerefCount - 1);
pVolDesc->vds_cPrivateNotifies -= PvtNotifyCount;
RELEASE_SPIN_LOCK(&pVolDesc->vds_VolLock, OldIrql);
AfpVolumeDereference(pVolDesc);
}
//
// if there were any other notifications, put them back on the queue
//
while (!IsListEmpty(&TransitionList))
{
pList = TransitionList.Flink;
pVolNotify = CONTAINING_RECORD(pList, VOL_NOTIFY, vn_List);
ASSERT(pVolNotify->vn_pVolDesc != pVolDesc);
RemoveEntryList(pList);
AfpVolumeQueueChangeNotify(pVolNotify, pNotifyQueue);
}
return(fResult);
}
BOOLEAN FASTCALL
AfpVolumeStopIndexing(
IN PVOLDESC pVolDesc,
IN PVOL_NOTIFY pInVolNotify
)
{
PKQUEUE pNotifyQueue;
PLIST_ENTRY pNotifyList;
PLIST_ENTRY pPrivateNotifyList;
PLIST_ENTRY pList, pNext;
LARGE_INTEGER Immediate;
PVOL_NOTIFY pVolNotify;
LONG index;
BOOLEAN fResult=TRUE;
DBGPRINT(DBG_COMP_CHGNOTIFY, DBG_LEVEL_ERR,
("StopIndexing: Stopping Index for Volume\n"));
index = pVolDesc->vds_VolId % NUM_NOTIFY_QUEUES;
pNotifyQueue = &AfpVolumeNotifyQueue[index];
pNotifyList = &AfpVolumeNotifyList[index];
pPrivateNotifyList = &AfpVirtualMemVolumeNotifyList[index];
Immediate.HighPart = Immediate.LowPart = 0;
AfpSwmrAcquireExclusive(&AfpVolumeListSwmr);
pList = pPrivateNotifyList->Flink;
while (pList != pPrivateNotifyList)
{
pVolNotify = CONTAINING_RECORD(pList, VOL_NOTIFY, vn_List);
pNext = pList->Flink;
if ((pVolNotify->vn_pVolDesc == pVolDesc) &&
(pVolNotify != pInVolNotify))
{
RemoveEntryList(pList);
afpFreeNotify(pVolNotify);
AfpVolumeDereference(pVolDesc);
//AfpNotifyListCount[index]--;
}
pList = pNext;
}
AfpSwmrRelease(&AfpVolumeListSwmr);
return(fResult);
}
/*** AfpQueuePrivateChangeNotify
*
* LOCKS_ASSUMED: vds_idDbAccessLock (SWMR, Exclusive)
*/
VOID
AfpQueuePrivateChangeNotify(
IN PVOLDESC pVolDesc,
IN PUNICODE_STRING pName,
IN PUNICODE_STRING pPath,
IN DWORD ParentId
)
{
DWORD dwSize;
LONG Index;
PLIST_ENTRY pVirtualNotifyList;
DBGPRINT(DBG_COMP_CHGNOTIFY, DBG_LEVEL_INFO,
("PvtNotify: ParentId %d, Path %Z, Name %Z\n",
ParentId, pPath, pName));
pVirtualNotifyList = &AfpVirtualMemVolumeNotifyList[pVolDesc->vds_VolId % NUM_NOTIFY_QUEUES];
// Reference the volume for Notify processing
if (AfpVolumeReference(pVolDesc))
{
PVOL_NOTIFY pVolNotify;
PFILE_NOTIFY_INFORMATION pNotifyInfo;
DBGPRINT(DBG_COMP_CHGNOTIFY, DBG_LEVEL_INFO,
("AfpQueuePrivateChangeNotify: Queuing directory %Z\\%Z\n", pPath, pName));
// Allocate an extra component worths
dwSize = sizeof(VOL_NOTIFY) +
sizeof(FILE_NOTIFY_INFORMATION) +
pPath->Length +
pName->Length +
(AFP_LONGNAME_LEN+1)*sizeof(WCHAR)+
sizeof(WCHAR);
Index = NOTIFY_USIZE_TO_INDEX(pPath->Length+pName->Length+sizeof(WCHAR));
pVolNotify = afpAllocNotify (Index, TRUE);
if (pVolNotify != NULL)
{
LONG Offset = 0;
INTERLOCKED_INCREMENT_LONG(&pVolDesc->vds_cPrivateNotifies);
if (pVolDesc->vds_cPrivateNotifies > pVolDesc->vds_maxPrivateNotifies)
{
pVolDesc->vds_maxPrivateNotifies = pVolDesc->vds_cPrivateNotifies;
}
pVolNotify->vn_VariableLength = pPath->Length+pName->Length+sizeof(WCHAR);
pVolNotify->vn_pVolDesc = pVolDesc;
pVolNotify->vn_Processor = afpProcessPrivateNotify;
pVolNotify->vn_TimeStamp = AFP_QUEUE_NOTIFY_IMMEDIATELY;
pVolNotify->vn_ParentId = ParentId;
pVolNotify->vn_TailLength = pName->Length;
pVolNotify->vn_StreamId = AFP_STREAM_DATA;
pNotifyInfo = (PFILE_NOTIFY_INFORMATION)((PBYTE)pVolNotify + sizeof(VOL_NOTIFY));
pNotifyInfo->NextEntryOffset = 0;
pNotifyInfo->Action = FILE_ACTION_ADDED | AFP_ACTION_PRIVATE;
pNotifyInfo->FileNameLength = pName->Length + pPath->Length;
if (pPath->Length > 0)
{
RtlCopyMemory(pNotifyInfo->FileName,
pPath->Buffer,
pPath->Length);
pNotifyInfo->FileName[pPath->Length/sizeof(WCHAR)] = L'\\';
pNotifyInfo->FileNameLength += sizeof(WCHAR);
Offset = pPath->Length + sizeof(WCHAR);
}
if (pName->Length > 0)
{
RtlCopyMemory((PBYTE)pNotifyInfo->FileName + Offset,
pName->Buffer,
pName->Length);
}
AfpSwmrAcquireExclusive(&AfpVolumeListSwmr);
InsertTailList(pVirtualNotifyList, &pVolNotify->vn_List);
AfpSwmrRelease(&AfpVolumeListSwmr);
}
else
{
DBGPRINT(DBG_COMP_CHGNOTIFY, DBG_LEVEL_ERR,
("AfpQueuePrivateChangeNotify: Queuing of notify for directory %Z\\%Z failed\n",
pPath, pName));
AFPLOG_ERROR(AFPSRVMSG_VOLUME_INIT_FAILED,
STATUS_INSUFFICIENT_RESOURCES,
NULL,
0,
&pVolDesc->vds_Name);
//
// this will remove all the entries that have been queued so far
//
AfpVolumeAbortIndexing(pVolDesc);
// remove the refcount put above when vol referenced
AfpVolumeDereference(pVolDesc);
// remove the creation refcount
AfpVolumeDereference(pVolDesc);
}
}
else
{
DBGPRINT(DBG_COMP_CHGNOTIFY, DBG_LEVEL_ERR,
("AfpQueuePrivateChangeNotify: Queuing of notify for directory %Z\\%Z failed as Reference not possible\n",
pPath, pName));
}
}
/*** AfpQueueOurChange
*
* LOCKS: vds_VolLock (SPIN)
*/
VOID
AfpQueueOurChange(
IN PVOLDESC pVolDesc,
IN DWORD Action, // NT FILE_ACTION_XXX (ntioapi.h)
IN PUNICODE_STRING pPath,
IN PUNICODE_STRING pParentPath OPTIONAL // queues a ACTION_MODIFIED
)
{
POUR_CHANGE pchange = NULL;
KIRQL OldIrql;
#if DBG
static PBYTE ActionStrings[] =
{ "",
"ADDED",
"REMOVED",
"MODIFIED",
"RENAMED OLD",
"RENAMED NEW",
"STREAM ADDED",
"STREAM REMOVED",
"STREAM MODIFIED"
};
#endif
PAGED_CODE( );
ASSERT(IS_VOLUME_NTFS(pVolDesc) && !EXCLUSIVE_VOLUME(pVolDesc));
//
// if the volume is being built, we don't have change-notify posted.
// Don't queue this change: we are never going to get a change-notify!
//
if (pVolDesc->vds_Flags & VOLUME_NEW_FIRST_PASS)
{
ASSERT(!(pVolDesc->vds_Flags & VOLUME_NOTIFY_POSTED));
return;
}
pchange = (POUR_CHANGE)AfpAllocNonPagedMemory(sizeof(OUR_CHANGE) + pPath->Length);
if (pchange != NULL)
{
DBGPRINT(DBG_COMP_CHGNOTIFY, DBG_LEVEL_INFO,
("AfpQueueOurChange: Queueing a %s for %Z\n", ActionStrings[Action], pPath));
AfpGetCurrentTimeInMacFormat(&pchange->oc_Time);
AfpInitUnicodeStringWithNonNullTerm(&pchange->oc_Path,
pPath->Length,
(PWCHAR)((PBYTE)pchange + sizeof(OUR_CHANGE)));
RtlCopyMemory(pchange->oc_Path.Buffer,
pPath->Buffer,
pPath->Length);
ExInterlockedInsertTailList(&pVolDesc->vds_OurChangeList[AFP_CHANGE_ACTION(Action)],
&pchange->oc_Link,
&(pVolDesc->vds_VolLock.SpinLock));
}
if (ARGUMENT_PRESENT(pParentPath))
{
pchange = (POUR_CHANGE)AfpAllocNonPagedMemory(sizeof(OUR_CHANGE) + pParentPath->Length);
if (pchange != NULL)
{
DBGPRINT(DBG_COMP_CHGNOTIFY, DBG_LEVEL_INFO,
("AfpQueueOurChange: Queueing (parent) %s for %Z\n",
ActionStrings[FILE_ACTION_MODIFIED], pParentPath));
AfpGetCurrentTimeInMacFormat(&pchange->oc_Time);
AfpInitUnicodeStringWithNonNullTerm(&pchange->oc_Path,
pParentPath->Length,
(PWCHAR)((PBYTE)pchange + sizeof(OUR_CHANGE)));
RtlCopyMemory(pchange->oc_Path.Buffer,
pParentPath->Buffer,
pParentPath->Length);
ExInterlockedInsertTailList(&pVolDesc->vds_OurChangeList[AFP_CHANGE_ACTION(FILE_ACTION_MODIFIED)],
&pchange->oc_Link,
&(pVolDesc->vds_VolLock.SpinLock));
}
}
}
/*** AfpDequeueOurChange
*
* LOCKS: LOCKS: vds_VolLock (SPIN)
*/
VOID
AfpDequeueOurChange(
IN PVOLDESC pVolDesc,
IN DWORD Action, // NT FILE_ACTION_XXX (ntioapi.h)
IN PUNICODE_STRING pPath,
IN PUNICODE_STRING pParentPath OPTIONAL// queues a ACTION_MODIFIED
)
{
POUR_CHANGE pChange;
PLIST_ENTRY pList, pListHead;
KIRQL OldIrql;
#if DBG
static PBYTE ActionStrings[] =
{ "",
"ADDED",
"REMOVED",
"MODIFIED",
"RENAMED OLD",
"RENAMED NEW",
"STREAM ADDED",
"STREAM REMOVED",
"STREAM MODIFIED"
};
#endif
ASSERT(IS_VOLUME_NTFS(pVolDesc) && !EXCLUSIVE_VOLUME(pVolDesc));
ACQUIRE_SPIN_LOCK(&pVolDesc->vds_VolLock, &OldIrql);
// point to the head of the appropriate change action list
pListHead = &pVolDesc->vds_OurChangeList[AFP_CHANGE_ACTION(Action)];
for (pList = pListHead->Flink;
pList != pListHead;
pList = pList->Flink)
{
pChange = CONTAINING_RECORD(pList, OUR_CHANGE, oc_Link);
// do a case *sensitive* unicode string compare
if (EQUAL_UNICODE_STRING_CS(pPath, &pChange->oc_Path))
{
DBGPRINT(DBG_COMP_CHGNOTIFY, DBG_LEVEL_INFO,
("AfpDequeueOurChange: Dequeueing a %s for %Z\n",
ActionStrings[Action], pPath));
RemoveEntryList(&pChange->oc_Link);
AfpFreeMemory(pChange);
break;
}
}
if (ARGUMENT_PRESENT(pParentPath))
{
// point to the head of the appropriate change action list
pListHead = &pVolDesc->vds_OurChangeList[FILE_ACTION_MODIFIED];
for (pList = pListHead->Flink;
pList != pListHead;
pList = pList->Flink)
{
pChange = CONTAINING_RECORD(pList, OUR_CHANGE, oc_Link);
// do a case *sensitive* unicode string compare
if (EQUAL_UNICODE_STRING_CS(pParentPath, &pChange->oc_Path))
{
DBGPRINT(DBG_COMP_CHGNOTIFY, DBG_LEVEL_INFO,
("AfpDequeueOurChange: Dequeueing (parent) %s for %Z\n",
ActionStrings[FILE_ACTION_MODIFIED], pParentPath));
RemoveEntryList(&pChange->oc_Link);
AfpFreeMemory(pChange);
break;
}
}
}
RELEASE_SPIN_LOCK(&pVolDesc->vds_VolLock, OldIrql);
}
/*** AfpOurChangeScavenger
*
* This runs in a worker thread context since it takes an swmr.
*
* LOCKS: vds_VolLock (SPIN)
*/
AFPSTATUS FASTCALL
AfpOurChangeScavenger(
IN PVOLDESC pVolDesc
)
{
AFPTIME Now;
KIRQL OldIrql;
int i;
BOOLEAN DerefVol = False;
#if DBG
static PBYTE Action[] = { "",
"ADDED",
"REMOVED",
"MODIFIED",
"RENAMED OLD",
"RENAMED NEW",
"STREAM ADDED",
"STREAM REMOVED",
"STREAM MODIFIED"};
#endif
DBGPRINT(DBG_COMP_CHGNOTIFY, DBG_LEVEL_INFO,
("AfpOurChangeScavenger: OurChange scavenger for volume %Z entered...\n",
&pVolDesc->vds_Name));
// If this volume is going away, do not requeue this scavenger routine
// We don't take the volume lock to check these flags since they are
// one-way, i.e. once set they are never cleared.
if (pVolDesc->vds_Flags & (VOLUME_DELETED | VOLUME_STOPPED))
{
DBGPRINT(DBG_COMP_CHGNOTIFY, DBG_LEVEL_INFO,
("AfpOurChangeScavenger: OurChange scavenger for volume %Z: Final run\n",
&pVolDesc->vds_Name));
DerefVol = True;
}
CleanTurds:
AfpGetCurrentTimeInMacFormat(&Now);
ACQUIRE_SPIN_LOCK(&pVolDesc->vds_VolLock, &OldIrql);
for (i = 0; i < NUM_AFP_CHANGE_ACTION_LISTS; i++)
{
PLIST_ENTRY pList, pHead;
POUR_CHANGE pChange;
pHead = &pVolDesc->vds_OurChangeList[i];
while (!IsListEmpty(pHead))
{
pList = pHead->Flink;
pChange = CONTAINING_RECORD(pList, OUR_CHANGE, oc_Link);
if (((Now - pChange->oc_Time) > OURCHANGE_AGE) || DerefVol)
{
RemoveHeadList(pHead);
DBGPRINT(DBG_COMP_CHGNOTIFY, DBG_LEVEL_WARN,
("AfpOurChangeScavenger: freeing %Z (%s)\n",
&pChange->oc_Path, &Action[i]));
AfpFreeMemory(pChange);
}
else
{
// All subsequent items in list will have later times so
// don't bother checking them
break;
}
}
}
RELEASE_SPIN_LOCK(&pVolDesc->vds_VolLock, OldIrql);
// Check again if this volume is going away and if so do not requeue this
// scavenger routine. Note that while we were running, the volume may
// have been deleted but this scavenger event could not be killed because
// it wasn't found on the list. We don't want to requeue this routine again
// because it will take AFP_OURCHANGE_AGE minutes for the volume to go
// away otherwise. This closes the window more, but does not totally
// eliminate it from happening.
if (!DerefVol)
{
if (pVolDesc->vds_Flags & (VOLUME_DELETED | VOLUME_STOPPED))
{
DBGPRINT(DBG_COMP_CHGNOTIFY, DBG_LEVEL_INFO,
("AfpOurChangeScavenger: OurChanges scavenger for volume %Z: Final Run\n",
&pVolDesc->vds_Name));
DerefVol = True;
goto CleanTurds;
}
}
else
{
AfpVolumeDereference(pVolDesc);
return AFP_ERR_NONE;
}
return AFP_ERR_REQUEUE;
}


Reference in a new issue View git blame Copy permalink