Public/winfsp
1
0
Fork 0
mirror of https://github.com/winfsp/winfsp.git synced 2025-04-23 00:43:00 -05:00
Code Issues Packages Projects Releases Wiki Activity

tst: winfsp-tests: ea_create_test

Browse Source
This commit is contained in:
Bill Zissimopoulos 2019-03-15 12:43:14 -07:00
parent 9fd491fa3d
commit 5aa06358fc
No known key found for this signature in database
GPG Key ID: 3D4F95D52C7B3EA3
1 changed files with 143 additions and 28 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

171
tst/winfsp-tests/ea-test.c
View File

@ -26,6 +26,147 @@
#include "winfsp-tests.h" #include "winfsp-tests.h"
typedef struct _FILE_GET_EA_INFORMATION
{
ULONG NextEntryOffset;
UCHAR EaNameLength;
CHAR EaName[1];
} FILE_GET_EA_INFORMATION, *PFILE_GET_EA_INFORMATION;
NTSYSAPI NTSTATUS NTAPI NtQueryEaFile(
IN HANDLE FileHandle,
OUT PIO_STATUS_BLOCK IoStatusBlock,
OUT PVOID Buffer,
IN ULONG Length,
IN BOOLEAN ReturnSingleEntry,
IN PVOID EaList OPTIONAL,
IN ULONG EaListLength,
IN PULONG EaIndex OPTIONAL,
IN BOOLEAN RestartScan);
NTSYSAPI NTSTATUS NTAPI NtSetEaFile(
IN HANDLE FileHandle,
OUT PIO_STATUS_BLOCK IoStatusBlock,
IN PVOID EaBuffer,
IN ULONG EaBufferSize);
static void ea_init_ea(
PFILE_FULL_EA_INFORMATION Ea, ULONG EaLength, PULONG PBytesTransferred)
{
union
{
FILE_FULL_EA_INFORMATION V;
UINT8 B[128];
} SingleEa;
memset(&SingleEa, 0, sizeof SingleEa);
SingleEa.V.EaNameLength = (UCHAR)strlen("name1");
SingleEa.V.EaValueLength = (USHORT)strlen("first");
lstrcpyA(SingleEa.V.EaName, "name1");
memcpy(SingleEa.V.EaName + SingleEa.V.EaNameLength + 1, "first", SingleEa.V.EaValueLength);
FspFileSystemAddEa(&SingleEa.V, Ea, EaLength, PBytesTransferred);
memset(&SingleEa, 0, sizeof SingleEa);
SingleEa.V.Flags = FILE_NEED_EA;
SingleEa.V.EaNameLength = (UCHAR)strlen("nameTwo");
SingleEa.V.EaValueLength = (USHORT)strlen("second");
lstrcpyA(SingleEa.V.EaName, "nameTwo");
memcpy(SingleEa.V.EaName + SingleEa.V.EaNameLength + 1, "second", SingleEa.V.EaValueLength);
FspFileSystemAddEa(&SingleEa.V, Ea, EaLength, PBytesTransferred);
memset(&SingleEa, 0, sizeof SingleEa);
SingleEa.V.EaNameLength = (UCHAR)strlen("n3");
SingleEa.V.EaValueLength = (USHORT)strlen("third");
lstrcpyA(SingleEa.V.EaName, "n3");
memcpy(SingleEa.V.EaName + SingleEa.V.EaNameLength + 1, "third", SingleEa.V.EaValueLength);
FspFileSystemAddEa(&SingleEa.V, Ea, EaLength, PBytesTransferred);
FspFileSystemAddEa(0, Ea, EaLength, PBytesTransferred);
}
struct ea_check_ea_context
{
ULONG Count;
ULONG EaCount[4];
};
static NTSTATUS ea_check_ea_enumerate(
FSP_FILE_SYSTEM *FileSystem, PVOID Context0,
PFILE_FULL_EA_INFORMATION SingleEa)
{
struct ea_check_ea_context *Context = Context0;
if (0 == strcmp(SingleEa->EaName, "NAME1"))
{
ASSERT(SingleEa->EaNameLength == (UCHAR)strlen("NAME1"));
ASSERT(SingleEa->EaValueLength == (UCHAR)strlen("first"));
ASSERT(0 == memcmp(SingleEa->EaName + SingleEa->EaNameLength + 1, "first", SingleEa->EaValueLength));
Context->EaCount[0]++;
}
if (0 == strcmp(SingleEa->EaName, "NAMETWO"))
{
ASSERT(SingleEa->EaNameLength == (UCHAR)strlen("NAMETWO"));
ASSERT(SingleEa->EaValueLength == (UCHAR)strlen("second"));
ASSERT(0 == memcmp(SingleEa->EaName + SingleEa->EaNameLength + 1, "second", SingleEa->EaValueLength));
Context->EaCount[1]++;
}
if (0 == strcmp(SingleEa->EaName, "N3"))
{
ASSERT(SingleEa->EaNameLength == (UCHAR)strlen("N3"));
ASSERT(SingleEa->EaValueLength == (UCHAR)strlen("third"));
ASSERT(0 == memcmp(SingleEa->EaName + SingleEa->EaNameLength + 1, "third", SingleEa->EaValueLength));
Context->EaCount[2]++;
}
Context->Count++;
return STATUS_SUCCESS;
}
static void ea_check_ea(HANDLE Handle)
{
NTSTATUS Result;
IO_STATUS_BLOCK Iosb;
union
{
FILE_FULL_EA_INFORMATION V;
UINT8 B[512];
} Ea;
union
{
FILE_GET_EA_INFORMATION V;
UINT8 B[128];
} GetEa;
ULONG EaLength = 0;
struct ea_check_ea_context Context;
memset(&Context, 0, sizeof Context);
Result = NtQueryEaFile(Handle, &Iosb, &Ea, sizeof Ea, FALSE, 0, 0, 0, FALSE);
ASSERT(STATUS_SUCCESS == Result);
Result = FspFileSystemEnumerateEa(0, ea_check_ea_enumerate, &Context, &Ea.V, (ULONG)Iosb.Information);
ASSERT(STATUS_SUCCESS == Result);
ASSERT(3 == Context.Count);
ASSERT(1 == Context.EaCount[0]);
ASSERT(1 == Context.EaCount[1]);
ASSERT(1 == Context.EaCount[2]);
memset(&GetEa, 0, sizeof GetEa);
GetEa.V.EaNameLength = (UCHAR)strlen("nameTwo");
lstrcpyA(GetEa.V.EaName, "nameTwo");
EaLength = FIELD_OFFSET(FILE_GET_EA_INFORMATION, EaName) + GetEa.V.EaNameLength + 1;
memset(&Context, 0, sizeof Context);
Result = NtQueryEaFile(Handle, &Iosb, &Ea, sizeof Ea, FALSE, &GetEa.V, EaLength, 0, FALSE);
ASSERT(STATUS_SUCCESS == Result);
Result = FspFileSystemEnumerateEa(0, ea_check_ea_enumerate, &Context, &Ea.V, (ULONG)Iosb.Information);
ASSERT(STATUS_SUCCESS == Result);
ASSERT(1 == Context.Count);
ASSERT(0 == Context.EaCount[0]);
ASSERT(1 == Context.EaCount[1]);
ASSERT(0 == Context.EaCount[2]);
}
static void ea_create_dotest(ULONG Flags, PWSTR Prefix) static void ea_create_dotest(ULONG Flags, PWSTR Prefix)
{ {
void *memfs = memfs_start(Flags); void *memfs = memfs_start(Flags);
@ -40,11 +181,6 @@ static void ea_create_dotest(ULONG Flags, PWSTR Prefix)
IO_STATUS_BLOCK Iosb; IO_STATUS_BLOCK Iosb;
LARGE_INTEGER LargeZero = { 0 }; LARGE_INTEGER LargeZero = { 0 };
union union
{
FILE_FULL_EA_INFORMATION V;
UINT8 B[128];
} SingleEa;
union
{ {
FILE_FULL_EA_INFORMATION V; FILE_FULL_EA_INFORMATION V;
UINT8 B[512]; UINT8 B[512];
@ -62,29 +198,7 @@ static void ea_create_dotest(ULONG Flags, PWSTR Prefix)
FILE_FLAG_BACKUP_SEMANTICS | FILE_FLAG_DELETE_ON_CLOSE, 0); FILE_FLAG_BACKUP_SEMANTICS | FILE_FLAG_DELETE_ON_CLOSE, 0);
ASSERT(INVALID_HANDLE_VALUE != DirHandle); ASSERT(INVALID_HANDLE_VALUE != DirHandle);
memset(&SingleEa, 0, sizeof SingleEa); ea_init_ea(&Ea.V, sizeof Ea, &EaLength);
SingleEa.V.EaNameLength = (UCHAR)strlen("name1");
SingleEa.V.EaValueLength = (USHORT)strlen("first");
lstrcpyA(SingleEa.V.EaName, "name1");
memcpy(SingleEa.V.EaName + SingleEa.V.EaNameLength + 1, "first", SingleEa.V.EaValueLength);
FspFileSystemAddEa(&SingleEa.V, &Ea.V, sizeof Ea, &EaLength);
memset(&SingleEa, 0, sizeof SingleEa);
SingleEa.V.Flags = FILE_NEED_EA;
SingleEa.V.EaNameLength = (UCHAR)strlen("nameTwo");
SingleEa.V.EaValueLength = (USHORT)strlen("second");
lstrcpyA(SingleEa.V.EaName, "nameTwo");
memcpy(SingleEa.V.EaName + SingleEa.V.EaNameLength + 1, "second", SingleEa.V.EaValueLength);
FspFileSystemAddEa(&SingleEa.V, &Ea.V, sizeof Ea, &EaLength);
memset(&SingleEa, 0, sizeof SingleEa);
SingleEa.V.EaNameLength = (UCHAR)strlen("n3");
SingleEa.V.EaValueLength = (USHORT)strlen("third");
lstrcpyA(SingleEa.V.EaName, "n3");
memcpy(SingleEa.V.EaName + SingleEa.V.EaNameLength + 1, "third", SingleEa.V.EaValueLength);
FspFileSystemAddEa(&SingleEa.V, &Ea.V, sizeof Ea, &EaLength);
FspFileSystemAddEa(0, &Ea.V, sizeof Ea, &EaLength);
UnicodePath.Length = (USHORT)wcslen(UnicodePathBuf) * sizeof(WCHAR); UnicodePath.Length = (USHORT)wcslen(UnicodePathBuf) * sizeof(WCHAR);
UnicodePath.MaximumLength = sizeof UnicodePathBuf; UnicodePath.MaximumLength = sizeof UnicodePathBuf;
@ -119,6 +233,7 @@ static void ea_create_dotest(ULONG Flags, PWSTR Prefix)
FILE_OPEN, FILE_DELETE_ON_CLOSE, FILE_OPEN, FILE_DELETE_ON_CLOSE,
0, 0); 0, 0);
ASSERT(STATUS_SUCCESS == Result); ASSERT(STATUS_SUCCESS == Result);
ea_check_ea(FileHandle);
CloseHandle(FileHandle); CloseHandle(FileHandle);
CloseHandle(DirHandle); CloseHandle(DirHandle);