mars_nwe/mars-flaim
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

FTK. Changes to import SetFileValidData on Windows (if available).

Browse Source 
git-svn-id: https://svn.code.sf.net/p/flaim/code/trunk@1000 0109f412-320b-0410-ab79-c3e0c5ffbbe6
This commit is contained in:
ahodgkinson
2006-10-18 16:30:01 +00:00
parent 7d2fb4a5e8
commit dea37b9618
4 changed files with 58 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

43
ftk/src/ftkwin.cpp
View File

@@ -27,7 +27,8 @@
#if defined( FLM_WIN)
extern FLMATOMIC gv_openFiles;
extern FLMATOMIC gv_openFiles;
extern SET_FILE_VALID_DATA_FUNC gv_SetFileValidDataFunc;
/****************************************************************************
Desc:
@@ -193,23 +194,26 @@ RCODE F_FileHdl::openOrCreate(
// opened/created. This will allow subsequent file extend operations to
// be done via calls to SetFileValidData().
if( OpenProcessToken( GetCurrentProcess(),
TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY, &hToken))
if( gv_SetFileValidDataFunc)
{
bClosePrivToken = TRUE;
if( LookupPrivilegeValue( NULL, SE_MANAGE_VOLUME_NAME, &luid))
if( OpenProcessToken( GetCurrentProcess(),
TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY, &hToken))
{
ZeroMemory ( &tp, sizeof( tp));
bClosePrivToken = TRUE;
tp.PrivilegeCount = 1;
tp.Privileges[0].Luid = luid;
tp.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED;
if( AdjustTokenPrivileges( hToken, FALSE, &tp,
sizeof( TOKEN_PRIVILEGES), &oldtp, &udTokenPrivSize))
if( LookupPrivilegeValue( NULL, SE_MANAGE_VOLUME_NAME, &luid))
{
bRestoreTokenPrivileges = TRUE;
ZeroMemory ( &tp, sizeof( tp));
tp.PrivilegeCount = 1;
tp.Privileges[0].Luid = luid;
tp.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED;
if( AdjustTokenPrivileges( hToken, FALSE, &tp,
sizeof( TOKEN_PRIVILEGES), &oldtp, &udTokenPrivSize))
{
bRestoreTokenPrivileges = TRUE;
}
}
}
}
@@ -864,13 +868,14 @@ RCODE F_FileHdl::extendFile(
goto Exit;
}
if( SetFileValidData( m_hFile, ui64NewFileSize))
if( gv_SetFileValidDataFunc)
{
// The call was successful. We are done.
goto Exit;
if( (gv_SetFileValidDataFunc)( m_hFile, ui64NewFileSize))
{
goto Exit;
}
}
// Determine the number of bytes to extend
ui64TotalBytesToExtend = ui64NewFileSize - ui64FileSize;