BlockStorage/siadrive
Archived
1
0
Fork 0
Code Activity

Changes

Browse Source
This commit is contained in:
Scott E. Graves
2017-03-26 03:19:50 -05:00
parent a2f41eed1c
commit dfc5ed91dd
1 changed files with 105 additions and 93 deletions
Download Patch File Download Diff File Expand all files Collapse all files

122
src/siadrive_dokan_api/siadokandrive.cpp
View File

@@ -121,7 +121,7 @@ private:
if (!_fileListStopRequested) if (!_fileListStopRequested)
{ {
// TODO Change to WaitForSingleObject() for immediate termination // TODO Change to WaitForSingleObject() for immediate termination
Sleep(5000); Sleep(1000);
} }
} }
})); }));
@@ -432,6 +432,8 @@ private:
CEventSystem::EventSystem.NotifyEvent(CreateSystemEvent(DokanFindFiles(cachePath, rootPath, siaQuery))); CEventSystem::EventSystem.NotifyEvent(CreateSystemEvent(DokanFindFiles(cachePath, rootPath, siaQuery)));
// TODO Grab live filesystem items
auto dirList = siaFileTree->QueryDirectories(rootPath); auto dirList = siaFileTree->QueryDirectories(rootPath);
for (auto& dir : dirList) for (auto& dir : dirList)
{ {
@@ -476,8 +478,8 @@ private:
HANDLE handle = reinterpret_cast<HANDLE>(id); HANDLE handle = reinterpret_cast<HANDLE>(id);
if (!DokanFileInfo->IsDirectory) if (!DokanFileInfo->IsDirectory)
{ {
LARGE_INTEGER li; LARGE_INTEGER li = { 0 };
li.LowPart = ::GetFileSize(handle, reinterpret_cast<LPDWORD>(&li.HighPart)); ::GetFileSizeEx(handle, &li);
HandleFileClose(filePath, id, li.QuadPart); HandleFileClose(filePath, id, li.QuadPart);
} }
::CloseHandle(reinterpret_cast<HANDLE>(DokanFileInfo->Context)); ::CloseHandle(reinterpret_cast<HANDLE>(DokanFileInfo->Context));
@@ -620,6 +622,7 @@ private:
if (!handle || (handle == INVALID_HANDLE_VALUE)) if (!handle || (handle == INVALID_HANDLE_VALUE))
{ {
// TODO Need to add to cache if missing
handle = ::CreateFile(&filePath[0], GENERIC_READ, FILE_SHARE_READ, nullptr, OPEN_EXISTING, 0, nullptr); handle = ::CreateFile(&filePath[0], GENERIC_READ, FILE_SHARE_READ, nullptr, OPEN_EXISTING, 0, nullptr);
if (handle == INVALID_HANDLE_VALUE) if (handle == INVALID_HANDLE_VALUE)
{ {
@@ -679,9 +682,8 @@ private:
opened = TRUE; opened = TRUE;
} }
DWORD fileSizeHigh = 0; LARGE_INTEGER li = { 0 };
DWORD fileSizeLow = ::GetFileSize(handle, &fileSizeHigh); if (!::GetFileSizeEx(handle, &li))
if (fileSizeLow == INVALID_FILE_SIZE)
{ {
DWORD error = GetLastError(); DWORD error = GetLastError();
if (opened) if (opened)
@@ -689,8 +691,6 @@ private:
return DokanNtStatusFromWin32(error); return DokanNtStatusFromWin32(error);
} }
UINT64 fileSize = (static_cast<UINT64>(fileSizeHigh) << 32) | fileSizeLow;
LARGE_INTEGER distanceToMove; LARGE_INTEGER distanceToMove;
if (DokanFileInfo->WriteToEndOfFile) if (DokanFileInfo->WriteToEndOfFile)
{ {
@@ -709,7 +709,7 @@ private:
// Paging IO cannot write after allocate file size. // Paging IO cannot write after allocate file size.
if (DokanFileInfo->PagingIo) if (DokanFileInfo->PagingIo)
{ {
if (static_cast<UINT64>(Offset) >= fileSize) if (static_cast<UINT64>(Offset) >= li.QuadPart)
{ {
*NumberOfBytesWritten = 0; *NumberOfBytesWritten = 0;
if (opened) if (opened)
@@ -717,9 +717,9 @@ private:
return STATUS_SUCCESS; return STATUS_SUCCESS;
} }
if ((static_cast<UINT64>(Offset) + NumberOfBytesToWrite) > fileSize) if ((static_cast<UINT64>(Offset) + NumberOfBytesToWrite) > li.QuadPart)
{ {
UINT64 bytes = fileSize - Offset; UINT64 bytes = li.QuadPart - Offset;
if (bytes >> 32) if (bytes >> 32)
{ {
NumberOfBytesToWrite = static_cast<DWORD>(bytes & 0xFFFFFFFFUL); NumberOfBytesToWrite = static_cast<DWORD>(bytes & 0xFFFFFFFFUL);
@@ -731,7 +731,7 @@ private:
} }
} }
if (static_cast<UINT64>(Offset) > fileSize) if (static_cast<UINT64>(Offset) > li.QuadPart)
{ {
// In the mirror sample helperZeroFileData is not necessary. NTFS will // In the mirror sample helperZeroFileData is not necessary. NTFS will
// zero a hole. // zero a hole.
@@ -749,15 +749,16 @@ private:
} }
} }
if (!::WriteFile(handle, Buffer, NumberOfBytesToWrite, NumberOfBytesWritten, nullptr)) if (::WriteFile(handle, Buffer, NumberOfBytesToWrite, NumberOfBytesWritten, nullptr))
{
// TODO Set status to changed
}
else
{ {
DWORD error = GetLastError(); DWORD error = GetLastError();
if (opened) if (opened)
CloseHandle(handle); CloseHandle(handle);
return DokanNtStatusFromWin32(error); return DokanNtStatusFromWin32(error);
}
else {
} }
// close the file when it is reopene bxkjuoqoa'qq d // close the file when it is reopene bxkjuoqoa'qq d
@@ -769,29 +770,31 @@ private:
static NTSTATUS DOKAN_CALLBACK Sia_SetEndOfFile(LPCWSTR FileName, LONGLONG ByteOffset, PDOKAN_FILE_INFO DokanFileInfo) static NTSTATUS DOKAN_CALLBACK Sia_SetEndOfFile(LPCWSTR FileName, LONGLONG ByteOffset, PDOKAN_FILE_INFO DokanFileInfo)
{ {
HANDLE handle; NTSTATUS ret = STATUS_SUCCESS;
LARGE_INTEGER offset;
handle = reinterpret_cast<HANDLE>(DokanFileInfo->Context); HANDLE handle = reinterpret_cast<HANDLE>(DokanFileInfo->Context);
if (!handle || handle == INVALID_HANDLE_VALUE) if (!handle || (handle == INVALID_HANDLE_VALUE))
{ {
return STATUS_INVALID_HANDLE; ret = STATUS_INVALID_HANDLE;
} }
else
{
// TODO Set status to changed
LARGE_INTEGER offset;
offset.QuadPart = ByteOffset; offset.QuadPart = ByteOffset;
if (!::SetFilePointerEx(handle, offset, nullptr, FILE_BEGIN)) if (!::SetFilePointerEx(handle, offset, nullptr, FILE_BEGIN))
{ {
DWORD error = GetLastError(); DWORD error = GetLastError();
return DokanNtStatusFromWin32(error); ret = DokanNtStatusFromWin32(error);
} }
else if (!::SetEndOfFile(handle))
if (!::SetEndOfFile(handle))
{ {
DWORD error = GetLastError(); DWORD error = GetLastError();
return DokanNtStatusFromWin32(error); ret = DokanNtStatusFromWin32(error);
}
} }
return STATUS_SUCCESS; return ret;
} }
static void DOKAN_CALLBACK Sia_Cleanup(LPCWSTR FileName, PDOKAN_FILE_INFO DokanFileInfo) static void DOKAN_CALLBACK Sia_Cleanup(LPCWSTR FileName, PDOKAN_FILE_INFO DokanFileInfo)
@@ -802,10 +805,11 @@ private:
HANDLE handle = reinterpret_cast<HANDLE>(DokanFileInfo->Context); HANDLE handle = reinterpret_cast<HANDLE>(DokanFileInfo->Context);
if (!DokanFileInfo->IsDirectory) if (!DokanFileInfo->IsDirectory)
{ {
LARGE_INTEGER li; LARGE_INTEGER li = { 0 };
li.LowPart = ::GetFileSize(handle, reinterpret_cast<LPDWORD>(&li.HighPart)); ::GetFileSizeEx(handle, &li);
HandleFileClose(filePath, DokanFileInfo->Context, li.QuadPart); HandleFileClose(filePath, DokanFileInfo->Context, li.QuadPart);
} }
::CloseHandle(handle); ::CloseHandle(handle);
DokanFileInfo->Context = 0; DokanFileInfo->Context = 0;
} }
@@ -898,6 +902,7 @@ private:
static NTSTATUS DOKAN_CALLBACK Sia_DeleteFileW(LPCWSTR FileName, PDOKAN_FILE_INFO DokanFileInfo) static NTSTATUS DOKAN_CALLBACK Sia_DeleteFileW(LPCWSTR FileName, PDOKAN_FILE_INFO DokanFileInfo)
{ {
// TODO Handle files that aren't cached // TODO Handle files that aren't cached
NTSTATUS ret = STATUS_SUCCESS;
HANDLE handle = reinterpret_cast<HANDLE>(DokanFileInfo->Context); HANDLE handle = reinterpret_cast<HANDLE>(DokanFileInfo->Context);
FilePath filePath(GetCacheLocation(), FileName); FilePath filePath(GetCacheLocation(), FileName);
@@ -905,42 +910,42 @@ private:
DWORD dwAttrib = ::GetFileAttributes(&filePath[0]); DWORD dwAttrib = ::GetFileAttributes(&filePath[0]);
if ((dwAttrib != INVALID_FILE_ATTRIBUTES) && (dwAttrib & FILE_ATTRIBUTE_DIRECTORY)) if ((dwAttrib != INVALID_FILE_ATTRIBUTES) && (dwAttrib & FILE_ATTRIBUTE_DIRECTORY))
{ {
return STATUS_ACCESS_DENIED; ret = STATUS_ACCESS_DENIED;
} }
else if (handle && (handle != INVALID_HANDLE_VALUE))
if (handle && (handle != INVALID_HANDLE_VALUE))
{ {
FILE_DISPOSITION_INFO fdi; FILE_DISPOSITION_INFO fdi;
fdi.DeleteFile = DokanFileInfo->DeleteOnClose; fdi.DeleteFile = DokanFileInfo->DeleteOnClose;
if (!::SetFileInformationByHandle(handle, FileDispositionInfo, &fdi, sizeof(FILE_DISPOSITION_INFO))) if (!::SetFileInformationByHandle(handle, FileDispositionInfo, &fdi, sizeof(FILE_DISPOSITION_INFO)))
{ {
return DokanNtStatusFromWin32(GetLastError()); ret = DokanNtStatusFromWin32(GetLastError());
} }
} }
return STATUS_SUCCESS; return ret;
} }
static NTSTATUS DOKAN_CALLBACK Sia_MoveFileW(LPCWSTR FileName, LPCWSTR NewFileName, BOOL ReplaceIfExisting, PDOKAN_FILE_INFO DokanFileInfo) static NTSTATUS DOKAN_CALLBACK Sia_MoveFileW(LPCWSTR FileName, LPCWSTR NewFileName, BOOL ReplaceIfExisting, PDOKAN_FILE_INFO DokanFileInfo)
{ {
// TODO Handle files that aren't cached // TODO Handle files that aren't cached
NTSTATUS ret = STATUS_SUCCESS;
FilePath filePath(GetCacheLocation(), FileName); FilePath filePath(GetCacheLocation(), FileName);
FilePath newFilePath(GetCacheLocation(), NewFileName); FilePath newFilePath(GetCacheLocation(), NewFileName);
HANDLE handle = reinterpret_cast<HANDLE>(DokanFileInfo->Context); HANDLE handle = reinterpret_cast<HANDLE>(DokanFileInfo->Context);
if (!handle || (handle == INVALID_HANDLE_VALUE)) if (!handle || (handle == INVALID_HANDLE_VALUE))
{ {
return STATUS_INVALID_HANDLE; ret = STATUS_INVALID_HANDLE;
} }
else
{
size_t len = wcslen(&newFilePath[0]); size_t len = wcslen(&newFilePath[0]);
DWORD bufferSize = static_cast<DWORD>(sizeof(FILE_RENAME_INFO) + (len * sizeof(newFilePath[0]))); DWORD bufferSize = static_cast<DWORD>(sizeof(FILE_RENAME_INFO) + (len * sizeof(newFilePath[0])));
PFILE_RENAME_INFO renameInfo = static_cast<PFILE_RENAME_INFO>(malloc(bufferSize)); PFILE_RENAME_INFO renameInfo = static_cast<PFILE_RENAME_INFO>(malloc(bufferSize));
if (!renameInfo) if (renameInfo)
{ {
return STATUS_BUFFER_OVERFLOW;
}
::ZeroMemory(renameInfo, bufferSize); ::ZeroMemory(renameInfo, bufferSize);
renameInfo->ReplaceIfExists = ReplaceIfExisting ? TRUE : FALSE; // some warning about converting BOOL to BOOLEAN renameInfo->ReplaceIfExists = ReplaceIfExisting ? TRUE : FALSE; // some warning about converting BOOL to BOOLEAN
@@ -952,29 +957,34 @@ private:
BOOL result = ::SetFileInformationByHandle(handle, FileRenameInfo, renameInfo, bufferSize); BOOL result = ::SetFileInformationByHandle(handle, FileRenameInfo, renameInfo, bufferSize);
free(renameInfo); free(renameInfo);
if (result) if (!result)
{ {
return STATUS_SUCCESS; DWORD error = GetLastError();
ret = DokanNtStatusFromWin32(error);
}
} }
else else
{ {
DWORD error = GetLastError(); ret = STATUS_BUFFER_OVERFLOW;
return DokanNtStatusFromWin32(error);
} }
} }
return ret;
}
static NTSTATUS DOKAN_CALLBACK Sia_SetFileAttributesW(LPCWSTR FileName, DWORD FileAttributes, PDOKAN_FILE_INFO DokanFileInfo) static NTSTATUS DOKAN_CALLBACK Sia_SetFileAttributesW(LPCWSTR FileName, DWORD FileAttributes, PDOKAN_FILE_INFO DokanFileInfo)
{ {
UNREFERENCED_PARAMETER(DokanFileInfo); UNREFERENCED_PARAMETER(DokanFileInfo);
NTSTATUS ret = STATUS_SUCCESS;
FilePath filePath(GetCacheLocation(), FileName); FilePath filePath(GetCacheLocation(), FileName);
if (!::SetFileAttributes(&filePath[0], FileAttributes)) if (!::SetFileAttributes(&filePath[0], FileAttributes))
{ {
DWORD error = GetLastError(); DWORD error = GetLastError();
return DokanNtStatusFromWin32(error); ret = DokanNtStatusFromWin32(error);
} }
return STATUS_SUCCESS; return ret;
} }
static NTSTATUS DOKAN_CALLBACK Sia_GetFileSecurityW( static NTSTATUS DOKAN_CALLBACK Sia_GetFileSecurityW(
@@ -1057,7 +1067,6 @@ private:
FilePath filePath(GetCacheLocation(), FileName); FilePath filePath(GetCacheLocation(), FileName);
HANDLE handle = reinterpret_cast<HANDLE>(DokanFileInfo->Context); HANDLE handle = reinterpret_cast<HANDLE>(DokanFileInfo->Context);
if (!handle || (handle == INVALID_HANDLE_VALUE)) if (!handle || (handle == INVALID_HANDLE_VALUE))
{ {
return STATUS_INVALID_HANDLE; return STATUS_INVALID_HANDLE;
@@ -1074,15 +1083,17 @@ private:
static NTSTATUS DOKAN_CALLBACK Sia_SetAllocationSize(LPCWSTR FileName, LONGLONG AllocSize, PDOKAN_FILE_INFO DokanFileInfo) static NTSTATUS DOKAN_CALLBACK Sia_SetAllocationSize(LPCWSTR FileName, LONGLONG AllocSize, PDOKAN_FILE_INFO DokanFileInfo)
{ {
NTSTATUS ret = STATUS_SUCCESS;
FilePath filePath(GetCacheLocation(), FileName); FilePath filePath(GetCacheLocation(), FileName);
HANDLE handle = reinterpret_cast<HANDLE>(DokanFileInfo->Context); HANDLE handle = reinterpret_cast<HANDLE>(DokanFileInfo->Context);
if (!handle || (handle == INVALID_HANDLE_VALUE)) if (!handle || (handle == INVALID_HANDLE_VALUE))
{ {
return STATUS_INVALID_HANDLE; ret = STATUS_INVALID_HANDLE;
} }
else
LARGE_INTEGER fileSize; {
LARGE_INTEGER fileSize = { 0 };
if (::GetFileSizeEx(handle, &fileSize)) if (::GetFileSizeEx(handle, &fileSize))
{ {
if (AllocSize < fileSize.QuadPart) if (AllocSize < fileSize.QuadPart)
@@ -1091,22 +1102,23 @@ private:
if (!::SetFilePointerEx(handle, fileSize, nullptr, FILE_BEGIN)) if (!::SetFilePointerEx(handle, fileSize, nullptr, FILE_BEGIN))
{ {
DWORD error = GetLastError(); DWORD error = GetLastError();
return DokanNtStatusFromWin32(error); ret = DokanNtStatusFromWin32(error);
} }
else if (!::SetEndOfFile(handle))
if (!::SetEndOfFile(handle))
{ {
DWORD error = GetLastError(); DWORD error = GetLastError();
return DokanNtStatusFromWin32(error); ret = DokanNtStatusFromWin32(error);
} }
} }
} }
else else
{ {
DWORD error = GetLastError(); DWORD error = GetLastError();
return DokanNtStatusFromWin32(error); ret = DokanNtStatusFromWin32(error);
} }
return STATUS_SUCCESS; }
return ret;
} }
public: public: