150 lines
3.3 KiB
C++
150 lines
3.3 KiB
C++
#include <nt.h>
|
|
#include <ntrtl.h>
|
|
#include <nturtl.h>
|
|
#include <windows.h>
|
|
#include "restmap.h"
|
|
#include "shellapi.h"
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
|
|
struct _EVENT_STR_MAP
|
|
{
|
|
DWORD EventId;
|
|
LPWSTR pEventStr;
|
|
} EventMap[ 12 ] =
|
|
{
|
|
{SrEventInvalid , L"INVALID " },
|
|
{SrEventStreamChange, L"FILE-MODIFY" },
|
|
{SrEventAclChange, L"ACL-CHANGE " },
|
|
{SrEventAttribChange, L"ATTR-CHANGE" },
|
|
{SrEventStreamOverwrite,L"FILE-MODIFY" },
|
|
{SrEventFileDelete, L"FILE-DELETE" },
|
|
{SrEventFileCreate, L"FILE-CREATE" },
|
|
{SrEventFileRename, L"FILE-RENAME" },
|
|
{SrEventDirectoryCreate,L"DIR-CREATE " },
|
|
{SrEventDirectoryRename,L"DIR-RENAME " },
|
|
{SrEventDirectoryDelete,L"DIR-DELETE " },
|
|
{SrEventMaximum, L"INVALID-MAX" }
|
|
};
|
|
|
|
LPWSTR
|
|
GetEventString(
|
|
DWORD EventId
|
|
)
|
|
{
|
|
LPWSTR pStr = L"NOT-FOUND";
|
|
|
|
for( int i=0; i<sizeof(EventMap)/sizeof(_EVENT_STR_MAP);i++)
|
|
{
|
|
if ( EventMap[i].EventId == EventId )
|
|
{
|
|
pStr = EventMap[i].pEventStr;
|
|
}
|
|
}
|
|
|
|
return pStr;
|
|
}
|
|
|
|
void
|
|
PrintUsage()
|
|
{
|
|
printf("Usage: restmap <option>");
|
|
printf("\n 1 = dumpmap [filename]");
|
|
printf("\n 2 = createmap <filename> <drive> <RPNum>");
|
|
}
|
|
|
|
|
|
|
|
void
|
|
PrintRestoreMap(LPWSTR pszFileName)
|
|
{
|
|
RestoreMapEntry *prme = NULL;
|
|
HANDLE hFile;
|
|
LPWSTR pszSrc, pszDest, pszTemp;
|
|
LPBYTE pbAcl = NULL;
|
|
|
|
hFile = CreateFile(pszFileName, GENERIC_READ, 0, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
|
|
if (hFile == INVALID_HANDLE_VALUE)
|
|
goto Err;
|
|
|
|
while (ERROR_SUCCESS == ReadRestoreMapEntry(hFile, &prme))
|
|
{
|
|
GetPaths(prme, &pszSrc, &pszTemp, &pszDest, &pbAcl);
|
|
printf("\n%S\tAttr=%08d\tAcl=%S\tSrc=%S\tTmp=%S\tDest=%S\tAcl=%S",
|
|
GetEventString(prme->m_dwOperation),
|
|
prme->m_dwAttribute,
|
|
(prme->m_cbAcl > 0) ? L"Yes" : L"No",
|
|
pszSrc,
|
|
pszTemp ? pszTemp : L"",
|
|
pszDest ? pszDest : L"",
|
|
pbAcl ? (prme->m_fAclInline ? L"inline" : (LPWSTR) pbAcl) : L"");
|
|
}
|
|
|
|
FreeRestoreMapEntry(prme);
|
|
|
|
CloseHandle(hFile);
|
|
|
|
Err:
|
|
return;
|
|
}
|
|
|
|
|
|
|
|
void __cdecl
|
|
main()
|
|
{
|
|
LPWSTR * argv = NULL;
|
|
int argc;
|
|
HGLOBAL hMem = NULL;
|
|
HANDLE hFile = NULL;
|
|
int option;
|
|
|
|
argv = CommandLineToArgvW(GetCommandLine(), &argc);
|
|
if (! argv)
|
|
{
|
|
printf("Error parsing arguments");
|
|
goto done;
|
|
}
|
|
if (argc < 2)
|
|
{
|
|
PrintUsage();
|
|
goto done;
|
|
}
|
|
|
|
option = _wtoi(argv[1]);
|
|
switch (option)
|
|
{
|
|
case 1:
|
|
PrintRestoreMap(argv[2]);
|
|
break;
|
|
|
|
case 2:
|
|
if (argc != 5)
|
|
{
|
|
PrintUsage();
|
|
goto done;
|
|
}
|
|
|
|
hFile = CreateFile(argv[2], GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
|
|
if (hFile != INVALID_HANDLE_VALUE)
|
|
{
|
|
printf("CreateRestoreMap...DWORD %ld", CreateRestoreMap(argv[3], _wtoi(argv[4]), hFile));
|
|
CloseHandle(hFile);
|
|
}
|
|
else
|
|
printf("Error creating file");
|
|
break;
|
|
|
|
default:
|
|
PrintUsage();
|
|
goto done;
|
|
break;
|
|
}
|
|
|
|
done:
|
|
if (argv) hMem = GlobalHandle(argv);
|
|
if (hMem) GlobalFree(hMem);
|
|
}
|
|
|
|
|