建站
咨詢
售后
建站咨詢
新聞中心
文件讀寫程序分為4個函數(shù)，分別是DriverUnload()、DriverEntry()、CreateFileTest()和OpenFileTest()。這4個函數(shù)的功能非常明確，DriverUnload()是一個卸載例程，DriverEntry()是驅動程序的入口，CreateFileTest()是用來新建文件的，OpenFileTest()是用來打開已建立文件并進行讀寫的函數(shù)。
創(chuàng)新互聯(lián)建站是由多位在大型網(wǎng)絡公司、廣告設計公司的優(yōu)秀設計人員和策劃人員組成的一個具有豐富經(jīng)驗的團隊，其中包括網(wǎng)站策劃、網(wǎng)頁美工、網(wǎng)站程序員、網(wǎng)頁設計師、平面廣告設計師、網(wǎng)絡營銷人員及形象策劃。承接：成都網(wǎng)站建設、成都做網(wǎng)站、網(wǎng)站改版、網(wǎng)頁設計制作、網(wǎng)站建設與維護、網(wǎng)絡推廣、數(shù)據(jù)庫開發(fā),以高性價比制作企業(yè)網(wǎng)站、行業(yè)門戶平臺等全方位的服務。
1. 文件的創(chuàng)建、打開與關閉
對于文件的創(chuàng)建或打開，在內核驅動中都是通過內核函數(shù)ZwCreateFile()進行操作的。和Win32 API類似，會通過參數(shù)接收返回的文件句柄。它的返回值是一個操作是否成功的狀態(tài)碼。ZwCreateFile()函數(shù)的定義如下：
- NTSTATUS ZwCreateFile(
- __out PHANDLE FileHandle,
- __in ACCESS_MASK DesiredAccess,
- __in POBJECT_ATTRIBUTES ObjectAttributes,
- __out PIO_STATUS_BLOCK IoStatusBlock,
- __in_opt PLARGE_INTEGER AllocationSize,
- __in ULONG FileAttributes,
- __in ULONG ShareAccess,
- __in ULONG CreateDisposition,
- __in ULONG CreateOptions,
- __in_opt PVOID EaBuffer,
- __in ULONG EaLength
- );
參數(shù)介紹如下。
FileHandle：用來接收創(chuàng)建文件后的文件句柄。
DesiredAccess：打開文件操作的描述，讀或寫，一般指定為 GENERIC_READ 或 GENERIC_WRITE；該參數(shù)和 CreateFile()函數(shù)中的參數(shù)相同。
ObjectAttributes：指向 OBJECT_ATTRIBUTES 結構體的指針，該結構體包含要創(chuàng)建或打開的文件名。
IoStatusBlock：指向 IO_STATUS_BLOCK 結構體的指針，該結構體用于接收操作結果的狀態(tài)。
AllocationSize：該參數(shù)指向一個 64 位的整數(shù)，用于文件初始化分配時的大小。
FileAttributes：通常為 FILE_ATTRIBUTE_NORMAL，該參數(shù)和 CreateFile()函數(shù)中的參數(shù)相同。
ShareAccess：指定文件的共享方式，可以指定為 FILE_SHARE_READ、FILE_SHARE_WRITE 或 FILE_SHARE_DELETE，該參數(shù)和 CreateFile()函數(shù)中的參數(shù)相同。
CreateDisposition：描述本次調用 ZwCreateFile()函數(shù)的意圖，可以指定為 FILE_CREATE、FILE_OPEN、FILE_OPEN_IF 等。
CreateOptions：通常指定為 FILE_SYNCHRONOUS_IO_NONALERT，表示文件是同步操作，比如在寫入文件時，調用 ZwWriteFile()函數(shù)，在 ZwWriteFile()調用返回時，文件寫操作已經(jīng)完成。
EaBuffer：該參數(shù)表示一個指針，指向可選的擴展屬性區(qū)，一般為 NULL。
EaLength：該參數(shù)表示擴展屬性區(qū)的長度，一般為 0。
ZwCreateFile()函數(shù)的第 3 個參數(shù)是一個指向 OBJECT_ATTRIBUTES 的結構體，該結構體的定義如下：
- typedef struct _OBJECT_ATTRIBUTES {
- ULONG Length;
- HANDLE RootDirectory;
- PUNICODE_STRING ObjectName;
- ULONG Attributes;
- PVOID SecurityDescriptor;
- PVOID SecurityQualityOfService;
- } OBJECT_ATTRIBUTES, *POBJECT_ATTRIBUTES;
- typedef CONST OBJECT_ATTRIBUTES *PCOBJECT_ATTRIBUTES;
該結構體通常不需要用戶逐個進行初始化，而是使用InitializeObjectAttributes()函數(shù)進行初始化，該函數(shù)的定義如下：
- VOID InitializeObjecttAttributes(
- OUT POBJECT_ATTRIBUTES InitializedAttributes,
- IN PUNICODE_STRING ObjectName,
- IN ULONG Attributes,
- IN HANDLE RootDirectory,
- IN PSECURITY_DESCRIPTOR SecurityDescriptor
- );
從InitializeObjectAttributes()函數(shù)的定義可以看出，其參數(shù)與OBJECT_ATTRIBUTES結構體的成員變量相同。InitializeObjectAttributes()函數(shù)的參數(shù)說明如下。
InitializeAttributes：指向 OBJECT_ATTRIBUTES 結構體的指針。
ObjectName：對象名稱，用 UNICODE_STRING 描述，對于 ZwCreateFile()函數(shù)而言，該處指定為文件名。
Attributes：一般設置為 OBJ_CASE_INSENSITIVE，意味著名字字符串不區(qū)分大小寫。
RootDirectory：一般設置為 NULL。
SecurityDescriptor：用于設置安全描述符，一般設置為 NULL。
ObjectName 必須使用 UNICODE_STRING 類型進行描述，UNICODE_STRING 是內核對寬字符串封裝的一種數(shù)據(jù)結構，該結構體的定義如下：
- typedef struct _UNICODE_STRING {
- USHORT Length;
- USHORT MaximumLength;
- PWSTR Buffer;
- } UNICODE_STRING, *PUNICODE_STRING;
結構體成員說明如下。
Length：字符串的參數(shù)，單位是字節(jié)，如果是 N 個字符，那幺 Length 的值為 N 個字符的 2 倍。
MaximumLength：整個字符緩沖區(qū)的最大長度，單位是字節(jié)。
Buffer：緩沖區(qū)的指針。
對于 UNICODE_STRING 類型的字符串，通過 KdPrint()也可以進行調試輸出，輸出的方式類似如下：
- UNICODE_STRING uniString;
- KdPrint(("%wZ", &uniString));
UNICODE_STRING類型的字符串在使用前需要進行初始化，初始化的方法有兩種：一種是使用內核函數(shù)RtlInitUnicodeString()進行初始化，另一種方式是自行申請內存空間來進行初始化。通常情況下都是用第1種方法。RtlInitUnicodeString()函數(shù)的定義如下：
- VOID RtlInitUnicodeString(
- IN OUT PUNICODE_STRING DestinationString,
- IN PCWSTR SourceString
- );
參數(shù)說明如下。
DestinationString：要初始化的 UNICODE_STRING 字符串的指針。
SourceString：字符串的內容。
在為InitializeObjectAttributes()函數(shù)傳遞第2個參數(shù)時，需要指定的文件名是一個符號鏈接。在應用層下，描述一個文件的完整路徑是“c:\a.txt”；而在內核下，描述的方式為“\??\c:\a.txt”。符號鏈接在內核模式下以“\??\”（或者是“\DosDevices\”）開頭；在用戶模式下使用符號鏈接，則以“\\.\”開頭。
上面介紹的ZwCreateFile()函數(shù)不但可以創(chuàng)建文件，還可以打開文件。但是由于它的參數(shù)過于繁多，因此內核函數(shù)中專門提供了一個用于進行文件打開的函數(shù)ZwOpenFile()，其定義如下：
- NTSTATUS ZwOpenFile(
- OUT PHANDLE FileHandle,
- IN ACCESS_MASK DesiredAccess,
- IN POBJECT_ATTRIBUTES ObjectAttributes,
- OUT PIO_STATUS_BLOCK IoStatusBlock,
- IN ULONG ShareAccess,
- IN ULONG OpenOptions
- );
ZwOpenFile()函數(shù)相當于一個只用來打開文件的精簡版的ZwCreateFile()函數(shù)，其各參數(shù)使用方法與ZwCreateFile()函數(shù)相同，這里不重復介紹。
文件句柄的關閉使用內核函數(shù)ZwClose()，其定義如下：
- NTSTATUS ZwClose(IN HANDLE Handle);
該函數(shù)只包含一個參數(shù)，即被打開文件的句柄。該函數(shù)除了可以關閉文件句柄以外，還可以關閉其它類型資源的句柄，比如注冊表句柄等。
2. 文件的相關操作
文件相關的操作主要介紹4個內核函數(shù)，分別是ZwReadFile()、ZwWriteFile()、ZwQueryInformationFile()和ZwSetInformationFile()。例子代碼中實現(xiàn)了對文件的讀寫操作，判斷打開的文件是否為目錄，獲取文件的長度和設置文件的指針。
首先來看ZwQueryInformationFile()和ZwSetInformationFile()兩個函數(shù)的定義。ZwQueryIn formationFile()函數(shù)的定義如下：
- NTSTATUS ZwQueryInformationFile(
- IN HANDLE FileHandle,
- OUT PIO_STATUS_BLOCK IoStatusBlock,
- OUT PVOID FileInformation,
- IN ULONG Length,
- IN FILE_INFORMATION_CLASS FileInformationClass
- );
參數(shù)說明如下。
FileHandle：被打開的文件句柄。
IoStatusBlock：返回設置的狀態(tài)。
FileInformation：依據(jù) FileInformationClass 的不同而不同。
Length：FileInformation 數(shù)據(jù)的長度。
FileInformationClass：描述需獲取的屬性類型。
ZwSetInformationFile()函數(shù)的定義如下：
- NTSTATUS ZwSetInformationFile(
- IN HANDLE FileHandle,
- OUT PIO_STATUS_BLOCK IoStatusBlock,
- IN PVOID FileInformation,
- IN ULONG Length,
- IN FILE_INFORMATION_CLASS FileInformationClass
- );
ZwSetInformationFile()函數(shù)的參數(shù)與 ZwQueryInformationFile()函數(shù)的參數(shù)幾乎相同，但是兩個函數(shù)的第 3 個參數(shù)稍有差別，差別在于對 ZwQueryInformationFile()來說是一個輸出參數(shù)，對于 ZwSetInformationFile()來說是一個輸入?yún)?shù)。這里一定要注意。
對于ZwQueryInformationFile()和ZwSetInformationFile()這兩個函數(shù)來說，第5個參數(shù)決定了要讀取或設置的屬性的類型，第3個參數(shù)根據(jù)第5個參數(shù)來接受或傳遞相應的值。
兩個函數(shù)的第5個參數(shù)的常用值有3種類型，分別是FileStandardInformation、FileBasic Information和FilePositionInformation。每種類型分別又對應不同的結構體，這些結構體則是被ZwQueryInformationFile()和ZwSetInformationFile()函數(shù)的第3個參數(shù)所用。
FileStandardInformation對應的結構體定義如下：
- typedef struct FILE_STANDARD_INFORMATION {
- LARGE_INTEGER AllocationSize; // 為文件分配的大?。ㄕ加么厮璐笮。?nbsp;
- LARGE_INTEGER EndOfFile; // 距離文件結尾的字節(jié)數(shù)
- ULONG NumberOfLinks; // 有多少個鏈接文件
- BOOLEAN DeletePending; // 是否準備刪除
- BOOLEAN Directory; // 是否為目錄
- } FILE_STANDARD_INFORMATION, *PFILE_STANDARD_INFORMATION;
FileBasicInformation對應的結構體定義如下：
- typedef struct FILE_BASIC_INFORMATION {
- LARGE_INTEGER CreationTime; // 文件創(chuàng)建時間
- LARGE_INTEGER LastAccessTime; // 最后訪問時間
- LARGE_INTEGER LastWriteTime; // 最后寫入時間
- LARGE_INTEGER ChangeTime; // 修改時間
- ULONG FileAttributes; // 文件屬性
- } FILE_BASIC_INFORMATION, *PFILE_BASIC_INFORMATION;
FilePositionInformation對應的結構體定義如下：
- typedef struct FILE_POSITION_INFORMATION {
- LARGE_INTEGER CurrentByteOffset; // 當前文件指針位置
- } FILE_POSITION_INFORMATION, *PFILE_POSITION_INFORMATION;
明白了第3個參數(shù)和第5個參數(shù)以后，就可以清楚第4個參數(shù)的取值了，該取值是第3個參數(shù)的大小。
上面的結構體中大量使用了LARGE_INTEGER的數(shù)據(jù)類型，它其實是一個聯(lián)合體。LARGE_INTEGER的定義如下：
- typedef union _LARGE_INTEGER {
- struct {
- ULONG LowPart;
- LONG HighPart;
- };
- struct {
- ULONG LowPart;
- LONG HighPart;
- } u;
- LONGLONG QuadPart;
- } LARGE_INTEGER;
該結構體主要是用來表示64位的整數(shù)類型，通常使用其QuadPart成員。
ZwReadFile()函數(shù)的定義如下：
- NTSTATUS ZwReadFile(
- IN HANDLE FileHandle,
- IN HANDLE Event OPTIONAL,
- IN PIO_APC_ROUTINE ApcRoutine OPTIONAL,
- IN PVOID ApcContext OPTIONAL,
- OUT PIO_STATUS_BLOCK IoStatusBlock,
- OUT PVOID Buffer,
- IN ULONG Length,
- IN PLARGE_INTEGER ByteOffset OPTIONAL,
- IN PULONG Key OPTIONAL
- );
參數(shù)說明如下。
FileHandle：打開文件的句柄。
Event：用于異步完成讀取時，一般設置為 NULL。
ApcRoutine：回調例程，用于異步完成讀取時，一般設置為 NULL。
ApcContext：一般設置為 NULL。
IoStatusBlock：指向 IO_STATUS_BLOCK 的指針，記錄讀取操作的狀態(tài)，IoStatusBlock.Information 用于記錄讀取的字節(jié)數(shù)。
Buffer：保存讀取文件內容的緩沖區(qū)。
Length：準備讀取文件內容的字節(jié)數(shù)。
ByteOffset：指定讀取內容的偏移地址。
Key：讀取文件時的附加信息，一般設置為 NULL。
ZwWriteFile()函數(shù)的定義如下：
- NTSTATUS ZwWriteFile(
- IN HANDLE FileHandle,
- IN HANDLE Event OPTIONAL,
- IN PIO_APC_ROUTINE ApcRoutine OPTIONAL,
- IN PVOID ApcContext OPTIONAL,
- OUT PIO_STATUS_BLOCK IoStatusBlock,
- IN PVOID Buffer,
- IN ULONG Length,
- IN PLARGE_INTEGER ByteOffset OPTIONAL,
- IN PULONG Key OPTIONAL
- );
該函數(shù)的參數(shù)類似于ZwReadFile()函數(shù)，Buffer中保存的是欲寫入文件內容的緩沖區(qū)。
3. 內存管理函數(shù)
文件讀寫代碼中用到了3個內存相關的內核函數(shù)，分別是ExAllocatePool()、RtlFillMem ory()和ExFreePool()。
ExAllocatePool()函數(shù)用于申請一塊內存空間，其定義如下：
- PVOID ExAllocatePool(
- IN POOL_TYPE PoolType,
- IN SIZE_T NumberOfBytes
- );
參數(shù)說明如下。
PoolType：該參數(shù)是一個枚舉值，常用的值有兩個，分別是 NonPagedPool 和 PagedPool；前者表示非分頁內存，而后者表示分頁內存；永遠不會被交換到文件中的虛擬內存稱為非分頁內存，可以被交換到文件中的虛擬內存稱為分頁內存。
NumberOfBytes：表示需要分配的內存大小。
該函數(shù)的返回值是一個內存地址。
RtlFillMemory()函數(shù)用于填充內存，其定義如下：
- VOID RtlFillMemory(
- IN VOID UNALIGNED *Destination,
- IN SIZE_T Length,
- IN UCHAR Fill
- );
參數(shù)說明如下。
Desination：填充內存地址的起始位置。
Length：填充的長度。
Fill：需要填充的字節(jié)。
ExFreePool()函數(shù)用于回收 ExAllocatePool()申請的內存空間，其定義如下：
- VOID ExFreePool(IN PVOID P);
該函數(shù)只有一個參數(shù)，是指向ExAllocatePool()函數(shù)分配內存空間的指針。
分享標題：網(wǎng)絡安全編程：內核下文件讀寫函數(shù)
鏈接地址：http://m.5511xx.com/article/dhgdecj.html
