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

Browse Source
This commit is contained in:
Bill Zissimopoulos 2019-03-17 13:08:12 -07:00
parent 92dfb0be96
commit 1dfbb0d9bf
No known key found for this signature in database
GPG Key ID: 3D4F95D52C7B3EA3
1 changed files with 70 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -117,6 +117,36 @@ static void ea_init_bad_ea(
FspFileSystemAddEa(0, Ea, EaLength, PBytesTransferred); FspFileSystemAddEa(0, Ea, EaLength, PBytesTransferred);
} }
BOOLEAN AddGetEa(PFILE_GET_EA_INFORMATION SingleEa,
PFILE_GET_EA_INFORMATION Ea, ULONG Length, PULONG PBytesTransferred)
{
if (0 != SingleEa)
{
PUINT8 EaPtr = (PUINT8)Ea + FSP_FSCTL_ALIGN_UP(*PBytesTransferred, sizeof(ULONG));
PUINT8 EaEnd = (PUINT8)Ea + Length;
ULONG EaLen = FIELD_OFFSET(FILE_GET_EA_INFORMATION, EaName) +
SingleEa->EaNameLength + 1;
if (EaEnd < EaPtr + EaLen)
return FALSE;
memcpy(EaPtr, SingleEa, EaLen);
((PFILE_GET_EA_INFORMATION)EaPtr)->NextEntryOffset = FSP_FSCTL_ALIGN_UP(EaLen, sizeof(ULONG));
*PBytesTransferred = (ULONG)(EaPtr + EaLen - (PUINT8)Ea);
}
else if ((ULONG)FIELD_OFFSET(FILE_GET_EA_INFORMATION, EaName) <= *PBytesTransferred)
{
PUINT8 EaEnd = (PUINT8)Ea + *PBytesTransferred;
while (EaEnd > (PUINT8)Ea + Ea->NextEntryOffset)
Ea = (PVOID)((PUINT8)Ea + Ea->NextEntryOffset);
Ea->NextEntryOffset = 0;
}
return TRUE;
}
struct ea_check_ea_context struct ea_check_ea_context
{ {
ULONG Count; ULONG Count;
@ -177,16 +207,23 @@ static void ea_check_ea(HANDLE Handle)
union union
{ {
FILE_GET_EA_INFORMATION V; FILE_GET_EA_INFORMATION V;
UINT8 B[128]; UINT8 B[512];
} GetEa; } GetEa;
union
{
FILE_GET_EA_INFORMATION V;
UINT8 B[128];
} SingleGetEa;
ULONG EaLength = 0; ULONG EaLength = 0;
ULONG EaIndex; ULONG EaIndex;
struct ea_check_ea_context Context; struct ea_check_ea_context Context;
memset(&GetEa, 0, sizeof GetEa); EaLength = 0;
GetEa.V.EaNameLength = (UCHAR)strlen("bnameTwo"); memset(&SingleGetEa, 0, sizeof SingleGetEa);
lstrcpyA(GetEa.V.EaName, "bnameTwo"); SingleGetEa.V.EaNameLength = (UCHAR)strlen("bnameTwo");
EaLength = FIELD_OFFSET(FILE_GET_EA_INFORMATION, EaName) + GetEa.V.EaNameLength + 1; lstrcpyA(SingleGetEa.V.EaName, "bnameTwo");
AddGetEa(&SingleGetEa.V, &GetEa.V, sizeof GetEa, &EaLength);
AddGetEa(0, &GetEa.V, sizeof GetEa, &EaLength);
memset(&Context, 0, sizeof Context); memset(&Context, 0, sizeof Context);
Result = NtQueryEaFile(Handle, &Iosb, &Ea, sizeof Ea, FALSE, &GetEa.V, EaLength, 0, FALSE); Result = NtQueryEaFile(Handle, &Iosb, &Ea, sizeof Ea, FALSE, &GetEa.V, EaLength, 0, FALSE);
@ -206,10 +243,12 @@ static void ea_check_ea(HANDLE Handle)
ASSERT(STATUS_BUFFER_OVERFLOW == Result); ASSERT(STATUS_BUFFER_OVERFLOW == Result);
ASSERT(0 == Iosb.Information); ASSERT(0 == Iosb.Information);
memset(&GetEa, 0, sizeof GetEa); EaLength = 0;
GetEa.V.EaNameLength = (UCHAR)strlen("nonexistent"); memset(&SingleGetEa, 0, sizeof SingleGetEa);
lstrcpyA(GetEa.V.EaName, "nonexistent"); SingleGetEa.V.EaNameLength = (UCHAR)strlen("nonexistent");
EaLength = FIELD_OFFSET(FILE_GET_EA_INFORMATION, EaName) + GetEa.V.EaNameLength + 1; lstrcpyA(SingleGetEa.V.EaName, "nonexistent");
AddGetEa(&SingleGetEa.V, &GetEa.V, sizeof GetEa, &EaLength);
AddGetEa(0, &GetEa.V, sizeof GetEa, &EaLength);
memset(&Context, 0, sizeof Context); memset(&Context, 0, sizeof Context);
Result = NtQueryEaFile(Handle, &Iosb, &Ea, sizeof Ea, FALSE, &GetEa.V, EaLength, 0, FALSE); Result = NtQueryEaFile(Handle, &Iosb, &Ea, sizeof Ea, FALSE, &GetEa.V, EaLength, 0, FALSE);
@ -222,6 +261,28 @@ static void ea_check_ea(HANDLE Handle)
ASSERT(0 == Context.EaCount[2]); ASSERT(0 == Context.EaCount[2]);
ASSERT(1 == Context.EaCount[3]); ASSERT(1 == Context.EaCount[3]);
EaLength = 0;
memset(&SingleGetEa, 0, sizeof SingleGetEa);
SingleGetEa.V.EaNameLength = (UCHAR)strlen("bnameTwo");
lstrcpyA(SingleGetEa.V.EaName, "bnameTwo");
AddGetEa(&SingleGetEa.V, &GetEa.V, sizeof GetEa, &EaLength);
memset(&SingleGetEa, 0, sizeof SingleGetEa);
SingleGetEa.V.EaNameLength = (UCHAR)strlen("nonexistent");
lstrcpyA(SingleGetEa.V.EaName, "nonexistent");
AddGetEa(&SingleGetEa.V, &GetEa.V, sizeof GetEa, &EaLength);
AddGetEa(0, &GetEa.V, sizeof GetEa, &EaLength);
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(2 == Context.Count);
ASSERT(0 == Context.EaCount[0]);
ASSERT(1 == Context.EaCount[1]);
ASSERT(0 == Context.EaCount[2]);
ASSERT(1 == Context.EaCount[3]);
memset(&Context, 0, sizeof Context); memset(&Context, 0, sizeof Context);
Result = NtQueryEaFile(Handle, &Iosb, &Ea, sizeof Ea, FALSE, 0, 0, 0, FALSE); Result = NtQueryEaFile(Handle, &Iosb, &Ea, sizeof Ea, FALSE, 0, 0, 0, FALSE);
ASSERT(STATUS_SUCCESS == Result); ASSERT(STATUS_SUCCESS == Result);