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

dll: FspFileSystemLoop()

Browse Source
This commit is contained in:
Bill Zissimopoulos 2015-12-30 17:31:24 -08:00
parent db28be473f
commit 422d4ccc79
4 changed files with 45 additions and 62 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
build/VStudio/winfsp_dll.vcxproj
View File

@ -27,6 +27,7 @@
<ClCompile Include="..\..\src\dll\debug.c" /> <ClCompile Include="..\..\src\dll\debug.c" />
<ClCompile Include="..\..\src\dll\fsctl.c" /> <ClCompile Include="..\..\src\dll\fsctl.c" />
<ClCompile Include="..\..\src\dll\library.c" /> <ClCompile Include="..\..\src\dll\library.c" />
<ClCompile Include="..\..\src\dll\loop.c" />
<ClCompile Include="..\..\src\dll\ntstatus.c" /> <ClCompile Include="..\..\src\dll\ntstatus.c" />
</ItemGroup> </ItemGroup>
<ItemGroup> <ItemGroup>

3
build/VStudio/winfsp_dll.vcxproj.filters
View File

@ -37,6 +37,9 @@
<ClCompile Include="..\..\src\dll\fsctl.c"> <ClCompile Include="..\..\src\dll\fsctl.c">
<Filter>Source</Filter> <Filter>Source</Filter>
</ClCompile> </ClCompile>
<ClCompile Include="..\..\src\dll\loop.c">
<Filter>Source</Filter>
</ClCompile>
</ItemGroup> </ItemGroup>
<ItemGroup> <ItemGroup>
<None Include="..\..\src\dll\ntstatus.i"> <None Include="..\..\src\dll\ntstatus.i">

24
inc/winfsp/winfsp.h
View File

@ -21,36 +21,34 @@
#include <winfsp/fsctl.h> #include <winfsp/fsctl.h>
#if 0
typedef struct _FSP_FILE_SYSTEM FSP_FILE_SYSTEM; typedef struct _FSP_FILE_SYSTEM FSP_FILE_SYSTEM;
typedef NTSTATUS FSP_FILE_SYSTEM_PROCESSREQ(FSP_FILE_SYSTEM *, FSP_FSCTL_TRANSACT_REQ *); typedef NTSTATUS FSP_FILE_SYSTEM_DISPATCHER(FSP_FILE_SYSTEM *, FSP_FSCTL_TRANSACT_REQ *);
typedef VOID FSP_FILE_SYSTEM_OPERATION(FSP_FILE_SYSTEM *, FSP_FSCTL_TRANSACT_REQ *); typedef VOID FSP_FILE_SYSTEM_OPERATION(FSP_FILE_SYSTEM *, FSP_FSCTL_TRANSACT_REQ *);
typedef struct _FSP_FILE_SYSTEM typedef struct _FSP_FILE_SYSTEM
{ {
/* private */
UINT16 Version; UINT16 Version;
WCHAR VolumePath[FSP_FSCTL_VOLUME_NAME_SIZEMAX / sizeof(WCHAR)];
HANDLE VolumeHandle; HANDLE VolumeHandle;
FSP_FILE_SYSTEM_PROCESSREQ *ProcessRequest;
/* public */
PVOID UserContext; PVOID UserContext;
FSP_FILE_SYSTEM_DISPATCHER *Dispatcher;
FSP_FILE_SYSTEM_OPERATION *Operations[FspFsctlTransactKindCount]; FSP_FILE_SYSTEM_OPERATION *Operations[FspFsctlTransactKindCount];
} FSP_FILE_SYSTEM; } FSP_FILE_SYSTEM;
FSP_API NTSTATUS FspFileSystemCreate(PWSTR DevicePath, FSP_API NTSTATUS FspFileSystemCreate(PWSTR DevicePath,
const FSP_FSCTL_VOLUME_PARAMS *Params, FSP_FILE_SYSTEM_PROCESSREQ *ProcessRequest, const FSP_FSCTL_VOLUME_PARAMS *VolumeParams,
FSP_FILE_SYSTEM_DISPATCHER *Dispatcher,
FSP_FILE_SYSTEM **PFileSystem); FSP_FILE_SYSTEM **PFileSystem);
FSP_API VOID FspFileSystemDelete(FSP_FILE_SYSTEM *FileSystem); FSP_API VOID FspFileSystemDelete(FSP_FILE_SYSTEM *FileSystem);
FSP_API NTSTATUS FspFileSystemLoop(FSP_FILE_SYSTEM *FileSystem); FSP_API NTSTATUS FspFileSystemLoop(FSP_FILE_SYSTEM *FileSystem);
FSP_API NTSTATUS FspFileSystemDirectDispatcher(FSP_FILE_SYSTEM *FileSystem,
FSP_FSCTL_TRANSACT_REQ *Request);
FSP_API NTSTATUS FspFileSystemPoolDispatcher(FSP_FILE_SYSTEM *FileSystem,
FSP_FSCTL_TRANSACT_REQ *Request);
FSP_API NTSTATUS FspProcessRequestDirect(FSP_FILE_SYSTEM *FileSystem, FSP_API NTSTATUS FspSendResponse(FSP_FILE_SYSTEM *FileSystem,
FSP_FSCTL_TRANSACT_REQ *Request);
FSP_API NTSTATUS FspProcessRequestInPool(FSP_FILE_SYSTEM *FileSystem,
FSP_FSCTL_TRANSACT_REQ *Request);
FSP_API NTSTATUS FspProduceResponse(FSP_FILE_SYSTEM *FileSystem,
FSP_FSCTL_TRANSACT_RSP *Response); FSP_FSCTL_TRANSACT_RSP *Response);
FSP_API NTSTATUS FspProduceResponseWithStatus(FSP_FILE_SYSTEM *FileSystem, FSP_API NTSTATUS FspSendResponseWithStatus(FSP_FILE_SYSTEM *FileSystem,
FSP_FSCTL_TRANSACT_REQ *Request, NTSTATUS Result); FSP_FSCTL_TRANSACT_REQ *Request, NTSTATUS Result);
#endif
FSP_API NTSTATUS FspNtStatusFromWin32(DWORD Error); FSP_API NTSTATUS FspNtStatusFromWin32(DWORD Error);
FSP_API VOID FspDebugLog(const char *format, ...); FSP_API VOID FspDebugLog(const char *format, ...);

77
src0/dll/loop.c → src/dll/loop.c
View File

@ -6,62 +6,45 @@
#include <dll/library.h> #include <dll/library.h>
typedef struct _FSP_WORK_ITEM typedef struct _FSP_DISPATCHER_WORK_ITEM
{ {
FSP_FILE_SYSTEM *FileSystem; FSP_FILE_SYSTEM *FileSystem;
__declspec(align(MEMORY_ALLOCATION_ALIGNMENT)) UINT8 RequestBuf[]; __declspec(align(MEMORY_ALLOCATION_ALIGNMENT)) UINT8 RequestBuf[];
} FSP_WORK_ITEM; } FSP_DISPATCHER_WORK_ITEM;
FSP_API NTSTATUS FspFileSystemCreate(PWSTR DevicePath, FSP_API NTSTATUS FspFileSystemCreate(PWSTR DevicePath,
const FSP_FSCTL_VOLUME_PARAMS *Params, FSP_FILE_SYSTEM_PROCESSREQ *ProcessRequest, const FSP_FSCTL_VOLUME_PARAMS *VolumeParams,
FSP_FILE_SYSTEM_DISPATCHER *Dispatcher,
FSP_FILE_SYSTEM **PFileSystem) FSP_FILE_SYSTEM **PFileSystem)
{ {
NTSTATUS Result; NTSTATUS Result;
WCHAR VolumePathBuf[MAX_PATH]; FSP_FILE_SYSTEM *FileSystem;
HANDLE VolumeHandle = INVALID_HANDLE_VALUE;
FSP_FILE_SYSTEM *FileSystem = 0;
*PFileSystem = 0; *PFileSystem = 0;
if (0 == ProcessRequest)
ProcessRequest = FspProcessRequestDirect;
FileSystem = MemAlloc(sizeof *FileSystem); FileSystem = MemAlloc(sizeof *FileSystem);
if (0 == FileSystem) if (0 == FileSystem)
return STATUS_INSUFFICIENT_RESOURCES;
memset(FileSystem, 0, sizeof *FileSystem);
Result = FspFsctlCreateVolume(DevicePath, VolumeParams,
FileSystem->VolumePath, sizeof FileSystem->VolumePath,
&FileSystem->VolumeHandle);
if (!NT_SUCCESS(Result))
{ {
Result = STATUS_INSUFFICIENT_RESOURCES; MemFree(FileSystem);
goto exit; return Result;
} }
Result = FspFsctlCreateVolume(DevicePath, Params, 0, VolumePathBuf, sizeof VolumePathBuf); FileSystem->Dispatcher = 0 != Dispatcher ? Dispatcher : FspFileSystemDirectDispatcher;
if (!NT_SUCCESS(Result))
goto exit;
Result = FspFsctlOpenVolume(VolumePathBuf, &VolumeHandle);
if (!NT_SUCCESS(Result))
goto exit;
memset(FileSystem, 0, sizeof *FileSystem);
FileSystem->VolumeHandle = VolumeHandle;
FileSystem->ProcessRequest = ProcessRequest;
*PFileSystem = FileSystem; *PFileSystem = FileSystem;
Result = STATUS_SUCCESS; return STATUS_SUCCESS;
exit:
if (!NT_SUCCESS(Result))
{
if (INVALID_HANDLE_VALUE != VolumeHandle)
CloseHandle(VolumeHandle);
MemFree(FileSystem);
}
return Result;
} }
FSP_API VOID FspFileSystemDelete(FSP_FILE_SYSTEM *FileSystem) FSP_API VOID FspFileSystemDelete(FSP_FILE_SYSTEM *FileSystem)
{ {
if (INVALID_HANDLE_VALUE != FileSystem->VolumeHandle)
CloseHandle(FileSystem->VolumeHandle); CloseHandle(FileSystem->VolumeHandle);
MemFree(FileSystem); MemFree(FileSystem);
} }
@ -92,7 +75,7 @@ FSP_API NTSTATUS FspFileSystemLoop(FSP_FILE_SYSTEM *FileSystem)
if (0 == NextRequest) if (0 == NextRequest)
break; break;
Result = FileSystem->ProcessRequest(FileSystem, Request); Result = FileSystem->Dispatcher(FileSystem, Request);
if (!NT_SUCCESS(Result)) if (!NT_SUCCESS(Result))
goto exit; goto exit;
@ -106,19 +89,19 @@ exit:
return Result; return Result;
} }
FSP_API NTSTATUS FspProcessRequestDirect(FSP_FILE_SYSTEM *FileSystem, FSP_API NTSTATUS FspFileSystemDirectDispatcher(FSP_FILE_SYSTEM *FileSystem,
FSP_FSCTL_TRANSACT_REQ *Request) FSP_FSCTL_TRANSACT_REQ *Request)
{ {
if (FspFsctlTransactKindCount <= Request->Kind || 0 == FileSystem->Operations[Request->Kind]) if (FspFsctlTransactKindCount <= Request->Kind || 0 == FileSystem->Operations[Request->Kind])
return FspProduceResponseWithStatus(FileSystem, Request, STATUS_INVALID_DEVICE_REQUEST); return FspSendResponseWithStatus(FileSystem, Request, STATUS_INVALID_DEVICE_REQUEST);
FileSystem->Operations[Request->Kind](FileSystem, Request); FileSystem->Operations[Request->Kind](FileSystem, Request);
return STATUS_SUCCESS; return STATUS_SUCCESS;
} }
static DWORD WINAPI FspProcessRequestInPoolWorker(PVOID Param) static DWORD WINAPI FspFileSystemPoolDispatcherWorker(PVOID Param)
{ {
FSP_WORK_ITEM *WorkItem = Param; FSP_DISPATCHER_WORK_ITEM *WorkItem = Param;
FSP_FSCTL_TRANSACT_REQ *Request = (PVOID)WorkItem->RequestBuf; FSP_FSCTL_TRANSACT_REQ *Request = (PVOID)WorkItem->RequestBuf;
WorkItem->FileSystem->Operations[Request->Kind](WorkItem->FileSystem, Request); WorkItem->FileSystem->Operations[Request->Kind](WorkItem->FileSystem, Request);
@ -127,23 +110,21 @@ static DWORD WINAPI FspProcessRequestInPoolWorker(PVOID Param)
return 0; return 0;
} }
FSP_API NTSTATUS FspProcessRequestInPool(FSP_FILE_SYSTEM *FileSystem, FSP_API NTSTATUS FspFileSystemPoolDispatcher(FSP_FILE_SYSTEM *FileSystem,
FSP_FSCTL_TRANSACT_REQ *Request) FSP_FSCTL_TRANSACT_REQ *Request)
{ {
if (FspFsctlTransactKindCount <= Request->Kind || 0 == FileSystem->Operations[Request->Kind]) if (FspFsctlTransactKindCount <= Request->Kind || 0 == FileSystem->Operations[Request->Kind])
return FspProduceResponseWithStatus(FileSystem, Request, STATUS_INVALID_DEVICE_REQUEST); return FspSendResponseWithStatus(FileSystem, Request, STATUS_INVALID_DEVICE_REQUEST);
FSP_WORK_ITEM *WorkItem; FSP_DISPATCHER_WORK_ITEM *WorkItem = MemAlloc(sizeof *WorkItem + Request->Size);
BOOLEAN Success;
WorkItem = MemAlloc(sizeof *WorkItem + Request->Size);
if (0 == WorkItem) if (0 == WorkItem)
return STATUS_INSUFFICIENT_RESOURCES; return STATUS_INSUFFICIENT_RESOURCES;
WorkItem->FileSystem = FileSystem; WorkItem->FileSystem = FileSystem;
memcpy(WorkItem->RequestBuf, Request, Request->Size); memcpy(WorkItem->RequestBuf, Request, Request->Size);
Success = QueueUserWorkItem(FspProcessRequestInPoolWorker, WorkItem, WT_EXECUTEDEFAULT); BOOLEAN Success = QueueUserWorkItem(
FspFileSystemPoolDispatcherWorker, WorkItem, WT_EXECUTEDEFAULT);
if (!Success) if (!Success)
{ {
NTSTATUS Result0 = FspNtStatusFromWin32(GetLastError()); NTSTATUS Result0 = FspNtStatusFromWin32(GetLastError());
@ -154,13 +135,13 @@ FSP_API NTSTATUS FspProcessRequestInPool(FSP_FILE_SYSTEM *FileSystem,
return STATUS_SUCCESS; return STATUS_SUCCESS;
} }
FSP_API NTSTATUS FspProduceResponse(FSP_FILE_SYSTEM *FileSystem, FSP_API NTSTATUS FspSendResponse(FSP_FILE_SYSTEM *FileSystem,
FSP_FSCTL_TRANSACT_RSP *Response) FSP_FSCTL_TRANSACT_RSP *Response)
{ {
return FspFsctlTransact(FileSystem->VolumeHandle, Response, Response->Size, 0, 0); return FspFsctlTransact(FileSystem->VolumeHandle, Response, Response->Size, 0, 0);
} }
FSP_API NTSTATUS FspProduceResponseWithStatus(FSP_FILE_SYSTEM *FileSystem, FSP_API NTSTATUS FspSendResponseWithStatus(FSP_FILE_SYSTEM *FileSystem,
FSP_FSCTL_TRANSACT_REQ *Request, NTSTATUS Result) FSP_FSCTL_TRANSACT_REQ *Request, NTSTATUS Result)
{ {
FSP_FSCTL_TRANSACT_RSP Response; FSP_FSCTL_TRANSACT_RSP Response;
@ -169,5 +150,5 @@ FSP_API NTSTATUS FspProduceResponseWithStatus(FSP_FILE_SYSTEM *FileSystem,
Response.Kind = Request->Kind; Response.Kind = Request->Kind;
Response.Hint = Request->Hint; Response.Hint = Request->Hint;
Response.IoStatus.Status = Result; Response.IoStatus.Status = Result;
return FspProduceResponse(FileSystem, &Response); return FspSendResponse(FileSystem, &Response);
} }