mirror of
https://github.com/reactos/reactos.git
synced 2024-09-12 13:54:41 +00:00
299 lines
7.5 KiB
C++
299 lines
7.5 KiB
C++
/*
|
|
* PROJECT: ReactOS fltmc utility
|
|
* LICENSE: GPL - See COPYING in the top level directory
|
|
* FILE: base/applications/fltmc/fltmc.c
|
|
* PURPOSE: Control utility for file system filter drivers
|
|
* PROGRAMMERS: Copyright 2016 Ged Murphy (gedmurphy@gmail.com)
|
|
*/
|
|
|
|
// Please leave this temporary hack in place
|
|
// it's used to keep VS2015 happy for development.
|
|
#ifdef __REACTOS__
|
|
#include <stdarg.h>
|
|
#include <windef.h>
|
|
#include <winbase.h>
|
|
#include <wchar.h>
|
|
#else
|
|
#include <Windows.h>
|
|
#endif
|
|
#include <fltuser.h>
|
|
#include <atlstr.h>
|
|
#include "resource.h"
|
|
|
|
EXTERN_C int wmain(int argc, WCHAR *argv[]);
|
|
|
|
void
|
|
LoadAndPrintString(ULONG MessageId, ...)
|
|
{
|
|
va_list args;
|
|
|
|
CAtlStringW Message;
|
|
if (Message.LoadStringW(MessageId))
|
|
{
|
|
va_start(args, MessageId);
|
|
vwprintf(Message.GetBuffer(), args);
|
|
va_end(args);
|
|
}
|
|
}
|
|
|
|
void
|
|
PrintErrorText(_In_ ULONG ErrorCode)
|
|
{
|
|
WCHAR Buffer[256];
|
|
if (FormatMessageW(FORMAT_MESSAGE_FROM_SYSTEM,
|
|
0,
|
|
ErrorCode,
|
|
0,
|
|
Buffer,
|
|
256,
|
|
0))
|
|
{
|
|
wprintf(L"%s\n", Buffer);
|
|
}
|
|
}
|
|
|
|
DWORD
|
|
SetDriverLoadPrivilege()
|
|
{
|
|
TOKEN_PRIVILEGES TokenPrivileges;
|
|
HANDLE hToken;
|
|
LUID luid;
|
|
BOOL bSuccess;
|
|
DWORD dwError = ERROR_SUCCESS;
|
|
|
|
bSuccess = OpenProcessToken(GetCurrentProcess(),
|
|
TOKEN_ADJUST_PRIVILEGES,
|
|
&hToken);
|
|
if (bSuccess == FALSE)
|
|
return GetLastError();
|
|
|
|
if (!LookupPrivilegeValueW(NULL, SE_LOAD_DRIVER_NAME, &luid))
|
|
return GetLastError();
|
|
|
|
TokenPrivileges.PrivilegeCount = 1;
|
|
TokenPrivileges.Privileges[0].Luid = luid;
|
|
TokenPrivileges.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED;
|
|
|
|
bSuccess = AdjustTokenPrivileges(hToken,
|
|
FALSE,
|
|
&TokenPrivileges,
|
|
sizeof(TOKEN_PRIVILEGES),
|
|
NULL,
|
|
NULL);
|
|
if (bSuccess == FALSE)
|
|
dwError = GetLastError();
|
|
|
|
CloseHandle(hToken);
|
|
|
|
return dwError;
|
|
}
|
|
|
|
void
|
|
LoadFilter(_In_ LPWSTR FilterName)
|
|
{
|
|
DWORD dwError;
|
|
dwError = SetDriverLoadPrivilege();
|
|
if (dwError != ERROR_SUCCESS)
|
|
{
|
|
LoadAndPrintString(IDS_ERROR_PRIV, HRESULT_FROM_WIN32(dwError));
|
|
return;
|
|
}
|
|
|
|
HRESULT hr = FilterLoad(FilterName);
|
|
if (hr != S_OK)
|
|
{
|
|
LoadAndPrintString(IDS_ERROR_LOAD, hr);
|
|
PrintErrorText(hr);
|
|
}
|
|
}
|
|
|
|
void
|
|
UnloadFilter(_In_ LPWSTR FilterName)
|
|
{
|
|
DWORD dwError;
|
|
dwError = SetDriverLoadPrivilege();
|
|
if (dwError != ERROR_SUCCESS)
|
|
{
|
|
LoadAndPrintString(IDS_ERROR_PRIV, HRESULT_FROM_WIN32(dwError));
|
|
return;
|
|
}
|
|
|
|
HRESULT hr = FilterUnload(FilterName);
|
|
if (hr != S_OK)
|
|
{
|
|
LoadAndPrintString(IDS_ERROR_UNLOAD, hr);
|
|
PrintErrorText(hr);
|
|
}
|
|
}
|
|
|
|
void
|
|
PrintFilterInfo(_In_ PVOID Buffer,
|
|
_In_ BOOL IsNewStyle)
|
|
{
|
|
WCHAR FilterName[128] = { 0 };
|
|
WCHAR Altitude[64] = { 0 };
|
|
|
|
if (IsNewStyle)
|
|
{
|
|
PFILTER_AGGREGATE_STANDARD_INFORMATION FilterAggInfo;
|
|
FilterAggInfo = (PFILTER_AGGREGATE_STANDARD_INFORMATION)Buffer;
|
|
|
|
if (FilterAggInfo->Type.MiniFilter.FilterNameLength < 128)
|
|
{
|
|
CopyMemory(FilterName,
|
|
(PCHAR)FilterAggInfo + FilterAggInfo->Type.MiniFilter.FilterNameBufferOffset,
|
|
FilterAggInfo->Type.MiniFilter.FilterNameLength);
|
|
FilterName[FilterAggInfo->Type.MiniFilter.FilterNameLength] = UNICODE_NULL;
|
|
}
|
|
|
|
if (FilterAggInfo->Type.MiniFilter.FilterNameLength < 64)
|
|
{
|
|
CopyMemory(Altitude,
|
|
(PCHAR)FilterAggInfo + FilterAggInfo->Type.MiniFilter.FilterAltitudeBufferOffset,
|
|
FilterAggInfo->Type.MiniFilter.FilterAltitudeLength);
|
|
FilterName[FilterAggInfo->Type.MiniFilter.FilterAltitudeLength] = UNICODE_NULL;
|
|
}
|
|
|
|
wprintf(L"%-38s %-10lu %-10s %-10lu\n",
|
|
FilterName,
|
|
FilterAggInfo->Type.MiniFilter.NumberOfInstances,
|
|
Altitude,
|
|
FilterAggInfo->Type.MiniFilter.FrameID);
|
|
}
|
|
else
|
|
{
|
|
PFILTER_FULL_INFORMATION FilterInfo;
|
|
FilterInfo = (PFILTER_FULL_INFORMATION)Buffer;
|
|
|
|
if (FilterInfo->FilterNameLength < 128)
|
|
{
|
|
CopyMemory(FilterName,
|
|
FilterInfo->FilterNameBuffer,
|
|
FilterInfo->FilterNameLength);
|
|
FilterName[FilterInfo->FilterNameLength] = UNICODE_NULL;
|
|
}
|
|
|
|
wprintf(L"%-38s %-10lu %-10lu\n",
|
|
FilterName,
|
|
FilterInfo->NumberOfInstances,
|
|
FilterInfo->FrameID);
|
|
}
|
|
}
|
|
|
|
void
|
|
ListFilters()
|
|
{
|
|
HANDLE FindHandle;
|
|
BYTE Buffer[1024];
|
|
ULONG BytesReturned;
|
|
BOOL IsNewStyle = TRUE;
|
|
HRESULT hr;
|
|
|
|
hr = FilterFindFirst(FilterAggregateStandardInformation,
|
|
Buffer,
|
|
1024,
|
|
&BytesReturned,
|
|
&FindHandle);
|
|
if (!SUCCEEDED(hr))
|
|
{
|
|
IsNewStyle = FALSE;
|
|
hr = FilterFindFirst(FilterFullInformation,
|
|
Buffer,
|
|
1024,
|
|
&BytesReturned,
|
|
&FindHandle);
|
|
}
|
|
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
if (IsNewStyle)
|
|
{
|
|
LoadAndPrintString(IDS_DISPLAY_FILTERS1);
|
|
wprintf(L"------------------------------ ------------- ------------ -----\n");
|
|
}
|
|
else
|
|
{
|
|
LoadAndPrintString(IDS_DISPLAY_FILTERS2);
|
|
wprintf(L"------------------------------ ------------- -----\n");
|
|
}
|
|
|
|
PrintFilterInfo(Buffer, IsNewStyle);
|
|
|
|
do
|
|
{
|
|
hr = FilterFindNext(FindHandle,
|
|
IsNewStyle ? FilterAggregateStandardInformation : FilterFullInformation,
|
|
Buffer,
|
|
1024,
|
|
&BytesReturned);
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
PrintFilterInfo(Buffer, IsNewStyle);
|
|
}
|
|
|
|
} while (SUCCEEDED(hr));
|
|
|
|
FilterFindClose(FindHandle);
|
|
}
|
|
|
|
if (!SUCCEEDED(hr) && hr != HRESULT_FROM_WIN32(ERROR_NO_MORE_ITEMS))
|
|
{
|
|
LoadAndPrintString(IDS_ERROR_PRIV, hr);
|
|
PrintErrorText(hr);
|
|
}
|
|
}
|
|
|
|
int wmain(int argc, WCHAR *argv[])
|
|
{
|
|
if (argc < 2)
|
|
{
|
|
LoadAndPrintString(IDS_USAGE);
|
|
return 0;
|
|
}
|
|
|
|
wprintf(L"\n");
|
|
|
|
if (!_wcsicmp(argv[1], L"help"))
|
|
{
|
|
LoadAndPrintString(IDS_USAGE);
|
|
}
|
|
else if (!_wcsicmp(argv[1], L"load"))
|
|
{
|
|
if (argc == 3)
|
|
{
|
|
LoadFilter(argv[2]);
|
|
}
|
|
else
|
|
{
|
|
LoadAndPrintString(IDS_USAGE_LOAD);
|
|
wprintf(L"fltmc.exe load [name]\n\n");
|
|
}
|
|
}
|
|
else if (!_wcsicmp(argv[1], L"unload"))
|
|
{
|
|
if (argc == 3)
|
|
{
|
|
UnloadFilter(argv[2]);
|
|
}
|
|
else
|
|
{
|
|
LoadAndPrintString(IDS_USAGE_UNLOAD);
|
|
wprintf(L"fltmc.exe unload [name]\n\n");
|
|
}
|
|
}
|
|
else if (!_wcsicmp(argv[1], L"filters"))
|
|
{
|
|
if (argc == 2)
|
|
{
|
|
ListFilters();
|
|
}
|
|
else
|
|
{
|
|
LoadAndPrintString(IDS_USAGE_FILTERS);
|
|
wprintf(L"fltmc.exe filters\n\n");
|
|
}
|
|
}
|
|
|
|
return 0;
|
|
}
|