Intravision/reactos
1
0
Fork 0
mirror of https://github.com/reactos/reactos.git synced 2025-08-05 21:12:59 +00:00
Code Issues Projects Releases Packages Wiki Activity

Git conversion: Make reactos the root directory, move rosapps, rostests, wallpapers into modules, and delete rossubsys.

Browse source
This commit is contained in:
Colin Finck 2017-10-03 07:45:34 +00:00
parent b94e2d8ca0
commit c2c66aff7d
24198 changed files with 0 additions and 37285 deletions
Download patch file Download diff file Expand all files Collapse all files

298
base/applications/fltmc/fltmc.cpp Normal file
View file

@ -0,0 +1,298 @@
/*
* 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;
}