reactos操作系统实现(140)

在创建文件成功后，然后就往文件里写入数据，而写入数据说明就需要文件系统给文件分配储存空间，以便保存这些数据到磁盘里，下面就来分析分配文件大小的函数，如下：

#001 NTSTATUS
#002 VfatSetAllocationSizeInformation(PFILE_OBJECT
FileObject,
#003 PVFATFCB Fcb,
#004 PDEVICE_EXTENSION DeviceExt,
#005 PLARGE_INTEGER
AllocationSize)
#006 {
#007 ULONG OldSize;
#008 ULONG Cluster,
FirstCluster;
#009 NTSTATUS Status;
#010

获取每簇分配多少字节。
#011 ULONG ClusterSize =
DeviceExt->FatInfo.BytesPerCluster;

获取需要分配的大小字节。
#012 ULONG NewSize =
AllocationSize->u.LowPart;
#013 ULONG NCluster;
#014 BOOLEAN AllocSizeChanged =
FALSE;
#015
#016
DPRINT("VfatSetAllocationSizeInformation(File <%wZ>,
AllocationSize %d %d)/n", &Fcb->PathNameU,
#017
AllocationSize->HighPart, AllocationSize->LowPart);
#018

获取原来文件的大小。
#019 if (Fcb->Flags &
FCB_IS_FATX_ENTRY)
#020 OldSize =
Fcb->entry.FatX.FileSize;
#021 else
#022 OldSize = Fcb->entry.Fat.FileSize;

如果请求分配超过32位大小，就返回出错，不支持这样的大小。
#023 if
(AllocationSize->u.HighPart > 0)
#024 {
#025 return
STATUS_INVALID_PARAMETER;
#026 }

如果文件的大小与新需求的大小一样，说明它已经分配成功了。
#027 if (OldSize == NewSize)
#028 {
#029 return(STATUS_SUCCESS);
#030 }
#031

从目录项里获取文件的首簇开始位置。
#032 FirstCluster =
vfatDirEntryGetFirstCluster (DeviceExt, &Fcb->entry);
#033

如果请求的大小大于已经分配的大小，说明需要重新分配更大的空间。
#034 if (NewSize >
Fcb->RFCB.AllocationSize.u.LowPart)
#035 {

标志分配文件大小发生了改变。
#036 AllocSizeChanged = TRUE;

如果首簇为0，说明从第一项开始。
#037 if (FirstCluster == 0)
#038 {
#039 Fcb->LastCluster =
Fcb->LastOffset = 0;

获取下一簇的标识来判断磁盘是否满了。
#040 Status = NextCluster
(DeviceExt, FirstCluster, &FirstCluster, TRUE);
#041 if
(!NT_SUCCESS(Status))
#042 {
#043 DPRINT1("NextCluster failed. Status = %x/n", Status);
#044 return Status;
#045 }

如果下一簇就已经标记为已满标志，说明磁盘已经写满，不能再写入任何数据，直接返回出错。
#046 if (FirstCluster ==
0xffffffff)
#047 {
#048 return
STATUS_DISK_FULL;
#049 }

计算文件需要分配大小的最后簇号。
#050 Status =
OffsetToCluster(DeviceExt, FirstCluster,
#051
ROUND_DOWN(NewSize - 1, ClusterSize),
#052
&NCluster, TRUE);

如果尝试分配不成功，说明磁盘已经满了，不能再保存数据。
#053 if (NCluster ==
0xffffffff || !NT_SUCCESS(Status))
#054 {
#055 /* disk is full */
#056 NCluster = Cluster
= FirstCluster;
#057 Status =
STATUS_SUCCESS;

清空所有分配的簇。
#058 while (NT_SUCCESS(Status) &&
Cluster != 0xffffffff && Cluster > 1)
#059 {
#060 Status =
NextCluster (DeviceExt, FirstCluster, &NCluster, FALSE);
#061 WriteCluster
(DeviceExt, Cluster, 0);
#062 Cluster =
NCluster;
#063 }
#064 return
STATUS_DISK_FULL;
#065 }

到这里已经分配文件需要的大小成功，设置文件控制块指向分配的首簇号。
#066 if (Fcb->Flags
& FCB_IS_FATX_ENTRY)
#067 {
#068
Fcb->entry.FatX.FirstCluster = FirstCluster;
#069 }
#070 else
#071 {
#072 if
(DeviceExt->FatInfo.FatType == FAT32)
#073 {
#074
Fcb->entry.Fat.FirstCluster = (unsigned short)(FirstCluster &
0x0000FFFF);
#075
Fcb->entry.Fat.FirstClusterHigh = FirstCluster >> 16;
#076 }
#077 else
#078 {
#079 ASSERT((FirstCluster
>> 16) == 0);
#080
Fcb->entry.Fat.FirstCluster = (unsigned short)(FirstCluster &
0x0000FFFF);
#081 }
#082 }
#083 }
#084 else
#085 {

以下进行不首簇的文件分配。
#086 if
(Fcb->LastCluster > 0)
#087 {
#088 if
(Fcb->RFCB.AllocationSize.u.LowPart - ClusterSize == Fcb->LastOffset)
#089 {
#090 Cluster =
Fcb->LastCluster;
#091 Status = STATUS_SUCCESS;
#092 }
#093 else
#094 {
#095 Status =
OffsetToCluster(DeviceExt, Fcb->LastCluster,
#096
Fcb->RFCB.AllocationSize.u.LowPart - ClusterSize -
Fcb->LastOffset,
#097
&Cluster, FALSE);
#098 }
#099 }
#100 else
#101 {
#102 Status =
OffsetToCluster(DeviceExt, FirstCluster,
#103
Fcb->RFCB.AllocationSize.u.LowPart - ClusterSize,
#104
&Cluster, FALSE);
#105 }
#106 if
(!NT_SUCCESS(Status))
#107 {
#108 return Status;
#109 }
#110
#111 Fcb->LastCluster =
Cluster;
#112 Fcb->LastOffset =
Fcb->RFCB.AllocationSize.u.LowPart - ClusterSize;
#113
#114 /* FIXME: Check
status */
#115 /* Cluster points now
to the last cluster within the chain */
#116 Status =
OffsetToCluster(DeviceExt, Cluster,
#117 ROUND_DOWN(NewSize - 1,
ClusterSize) - Fcb->LastOffset,
#118
&NCluster, TRUE);
#119 if (NCluster ==
0xffffffff || !NT_SUCCESS(Status))
#120 {
#121 /* disk is full */
#122 NCluster = Cluster;
#123 Status =
NextCluster (DeviceExt, FirstCluster, &NCluster, FALSE);
#124 WriteCluster(DeviceExt,
Cluster, 0xffffffff);
#125 Cluster = NCluster;
#126 while
(NT_SUCCESS(Status) && Cluster != 0xffffffff && Cluster > 1)
#127 {
#128 Status = NextCluster
(DeviceExt, FirstCluster, &NCluster, FALSE);
#129 WriteCluster
(DeviceExt, Cluster, 0);
#130 Cluster = NCluster;
#131 }
#132 return STATUS_DISK_FULL;
#133 }
#134 }
#135
UpdateFileSize(FileObject, Fcb, NewSize, ClusterSize);
#136 }
#137 else if (NewSize +
ClusterSize <= Fcb->RFCB.AllocationSize.u.LowPart)
#138 {

下面进行文件减小的分配。
#139 AllocSizeChanged = TRUE;
#140 /* FIXME: Use the cached
cluster/offset better way. */
#141 Fcb->LastCluster =
Fcb->LastOffset = 0;
#142
UpdateFileSize(FileObject, Fcb, NewSize, ClusterSize);
#143 if (NewSize > 0)
#144 {
#145 Status =
OffsetToCluster(DeviceExt, FirstCluster,
#146 ROUND_DOWN(NewSize
- 1, ClusterSize),
#147 &Cluster, FALSE);
#148
#149 NCluster = Cluster;
#150 Status = NextCluster
(DeviceExt, FirstCluster, &NCluster, FALSE);
#151
WriteCluster(DeviceExt, Cluster, 0xffffffff);
#152 Cluster = NCluster;
#153 }
#154 else
#155 {
#156 if (Fcb->Flags
& FCB_IS_FATX_ENTRY)
#157 {
#158
Fcb->entry.FatX.FirstCluster = 0;
#159 }
#160 else
#161 {
#162 if
(DeviceExt->FatInfo.FatType == FAT32)
#163 {
#164
Fcb->entry.Fat.FirstCluster = 0;
#165
Fcb->entry.Fat.FirstClusterHigh = 0;
#166 }
#167 else
#168 {
#169
Fcb->entry.Fat.FirstCluster = 0;
#170 }
#171 }
#172
#173 NCluster = Cluster =
FirstCluster;
#174 Status =
STATUS_SUCCESS;
#175 }
#176 while
(NT_SUCCESS(Status) && 0xffffffff != Cluster && Cluster > 1)
#177 {
#178 Status = NextCluster
(DeviceExt, FirstCluster, &NCluster, FALSE);
#179 WriteCluster
(DeviceExt, Cluster, 0);
#180 Cluster = NCluster;
#181 }
#182 }
#183 else
#184 {

更新文件的大小。
#185 UpdateFileSize(FileObject, Fcb, NewSize,
ClusterSize);
#186 }

更新文件的大小到目录项。
#187 /* Update the on-disk
directory entry */
#188 Fcb->Flags |=
FCB_IS_DIRTY;
#189 if (AllocSizeChanged)
#190 {
#191 VfatUpdateEntry(Fcb);
#192 }
#193 return STATUS_SUCCESS;
#194 }