zydronium/SMBLibrary
1
0
Fork 0
mirror of https://github.com/TalAloni/SMBLibrary.git synced 2025-07-20 16:15:53 +02:00
Code Issues Projects Releases Packages Wiki Activity

NTDirectoryFileSystem: Pass-through LockFile / UnlockFile requests

Browse source
This commit is contained in:
Tal Aloni 2017-08-31 15:58:43 +03:00
parent e7da671877
commit dfbda03f6c
2 changed files with 14 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

2
SMBLibrary/Enums/NTStatus.cs
View file

@ -32,6 +32,7 @@ namespace SMBLibrary
STATUS_DATA_ERROR = 0xC000003E, // IO error STATUS_DATA_ERROR = 0xC000003E, // IO error
STATUS_SHARING_VIOLATION = 0xC0000043, STATUS_SHARING_VIOLATION = 0xC0000043,
STATUS_FILE_LOCK_CONFLICT = 0xC0000054, STATUS_FILE_LOCK_CONFLICT = 0xC0000054,
STATUS_LOCK_NOT_GRANTED = 0xC0000055,
STATUS_PRIVILEGE_NOT_HELD = 0xC0000061, STATUS_PRIVILEGE_NOT_HELD = 0xC0000061,
STATUS_LOGON_FAILURE = 0xC000006D, // Authentication failure. STATUS_LOGON_FAILURE = 0xC000006D, // Authentication failure.
STATUS_ACCOUNT_RESTRICTION = 0xC000006E, // The user has an empty password, which is not allowed STATUS_ACCOUNT_RESTRICTION = 0xC000006E, // The user has an empty password, which is not allowed
@ -39,6 +40,7 @@ namespace SMBLibrary
STATUS_INVALID_WORKSTATION = 0xC0000070, STATUS_INVALID_WORKSTATION = 0xC0000070,
STATUS_PASSWORD_EXPIRED = 0xC0000071, STATUS_PASSWORD_EXPIRED = 0xC0000071,
STATUS_ACCOUNT_DISABLED = 0xC0000072, STATUS_ACCOUNT_DISABLED = 0xC0000072,
STATUS_RANGE_NOT_LOCKED = 0xC000007E,
STATUS_DISK_FULL = 0xC000007F, STATUS_DISK_FULL = 0xC000007F,
STATUS_INSUFFICIENT_RESOURCES = 0xC000009A, STATUS_INSUFFICIENT_RESOURCES = 0xC000009A,
STATUS_MEDIA_WRITE_PROTECTED = 0xC00000A2, STATUS_MEDIA_WRITE_PROTECTED = 0xC00000A2,

12
SMBLibrary/Win32/NTFileStore/NTDirectoryFileSystem.cs
View file

@ -83,6 +83,12 @@ namespace SMBLibrary.Win32
[DllImport("ntdll.dll", ExactSpelling = true, SetLastError = false)] [DllImport("ntdll.dll", ExactSpelling = true, SetLastError = false)]
private static extern NTStatus NtFlushBuffersFile(IntPtr handle, out IO_STATUS_BLOCK ioStatusBlock); private static extern NTStatus NtFlushBuffersFile(IntPtr handle, out IO_STATUS_BLOCK ioStatusBlock);
[DllImport("ntdll.dll", ExactSpelling = true, SetLastError = false)]
private static extern NTStatus NtLockFile(IntPtr handle, IntPtr evt, IntPtr apcRoutine, IntPtr apcContext, out IO_STATUS_BLOCK ioStatusBlock, ref long byteOffset, ref long length, uint key, bool failImmediately, bool exclusiveLock);
[DllImport("ntdll.dll", ExactSpelling = true, SetLastError = false)]
private static extern NTStatus NtUnlockFile(IntPtr handle, out IO_STATUS_BLOCK ioStatusBlock, ref long byteOffset, ref long length, uint key);
[DllImport("ntdll.dll", ExactSpelling = true, SetLastError = false)] [DllImport("ntdll.dll", ExactSpelling = true, SetLastError = false)]
private static extern NTStatus NtQueryDirectoryFile(IntPtr handle, IntPtr evt, IntPtr apcRoutine, IntPtr apcContext, out IO_STATUS_BLOCK ioStatusBlock, byte[] fileInformation, uint length, uint fileInformationClass, bool returnSingleEntry, ref UNICODE_STRING fileName, bool restartScan); private static extern NTStatus NtQueryDirectoryFile(IntPtr handle, IntPtr evt, IntPtr apcRoutine, IntPtr apcContext, out IO_STATUS_BLOCK ioStatusBlock, byte[] fileInformation, uint length, uint fileInformationClass, bool returnSingleEntry, ref UNICODE_STRING fileName, bool restartScan);
@ -227,12 +233,14 @@ namespace SMBLibrary.Win32
public NTStatus LockFile(object handle, long byteOffset, long length, bool exclusiveLock) public NTStatus LockFile(object handle, long byteOffset, long length, bool exclusiveLock)
{ {
return NTStatus.STATUS_NOT_SUPPORTED; IO_STATUS_BLOCK ioStatusBlock;
return NtLockFile((IntPtr)handle, IntPtr.Zero, IntPtr.Zero, IntPtr.Zero, out ioStatusBlock, ref byteOffset, ref length, 0, true, exclusiveLock);
} }
public NTStatus UnlockFile(object handle, long byteOffset, long length) public NTStatus UnlockFile(object handle, long byteOffset, long length)
{ {
return NTStatus.STATUS_NOT_SUPPORTED; IO_STATUS_BLOCK ioStatusBlock;
return NtUnlockFile((IntPtr)handle, out ioStatusBlock, ref byteOffset, ref length, 0);
} }
public NTStatus QueryDirectory(out List<QueryDirectoryFileInformation> result, object handle, string fileName, FileInformationClass informationClass) public NTStatus QueryDirectory(out List<QueryDirectoryFileInformation> result, object handle, string fileName, FileInformationClass informationClass)