860 lines
25 KiB
C
860 lines
25 KiB
C
/**
|
|
* FreeRDP: A Remote Desktop Protocol Implementation
|
|
* File System Virtual Channel
|
|
*
|
|
* Copyright 2010-2012 Marc-Andre Moreau <marcandre.moreau@gmail.com>
|
|
* Copyright 2010-2011 Vic Lee
|
|
* Copyright 2012 Gerald Richter
|
|
* Copyright 2015 Thincast Technologies GmbH
|
|
* Copyright 2015 DI (FH) Martin Haimberger <martin.haimberger@thincast.com>
|
|
* Copyright 2016 Inuvika Inc.
|
|
* Copyright 2016 David PHAM-VAN <d.phamvan@inuvika.com>
|
|
* Copyright 2017 Armin Novak <armin.novak@thincast.com>
|
|
* Copyright 2017 Thincast Technologies GmbH
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
|
|
#ifdef HAVE_CONFIG_H
|
|
#include "config.h"
|
|
#endif
|
|
|
|
#include <errno.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <time.h>
|
|
|
|
#include <winpr/wtypes.h>
|
|
#include <winpr/crt.h>
|
|
#include <winpr/path.h>
|
|
#include <winpr/file.h>
|
|
#include <winpr/stream.h>
|
|
|
|
#include <freerdp/channels/rdpdr.h>
|
|
|
|
#include "drive_file.h"
|
|
|
|
#ifdef WITH_DEBUG_RDPDR
|
|
#define DEBUG_WSTR(msg, wstr) do { LPSTR lpstr; ConvertFromUnicode(CP_UTF8, 0, wstr, -1, &lpstr, 0, NULL, NULL); WLog_DBG(TAG, msg, lpstr); free(lpstr); } while (0)
|
|
#else
|
|
#define DEBUG_WSTR(msg, wstr) do { } while (0)
|
|
#endif
|
|
|
|
static void drive_file_fix_path(WCHAR* path)
|
|
{
|
|
size_t i;
|
|
size_t length;
|
|
length = (int) _wcslen(path);
|
|
|
|
for (i = 0; i < length; i++)
|
|
{
|
|
if (path[i] == L'\\')
|
|
path[i] = L'/';
|
|
}
|
|
|
|
#ifdef WIN32
|
|
|
|
if ((length == 3) && (path[1] == L':') && (path[2] == L'/'))
|
|
return;
|
|
|
|
#else
|
|
|
|
if ((length == 1) && (path[0] == L'/'))
|
|
return;
|
|
|
|
#endif
|
|
|
|
if ((length > 0) && (path[length - 1] == L'/'))
|
|
path[length - 1] = L'\0';
|
|
}
|
|
|
|
static WCHAR* drive_file_combine_fullpath(const WCHAR* base_path, const WCHAR* path,
|
|
UINT32 PathLength)
|
|
{
|
|
WCHAR* fullpath;
|
|
UINT32 base_path_length;
|
|
|
|
if (!base_path || !path)
|
|
return NULL;
|
|
|
|
base_path_length = _wcslen(base_path) * 2;
|
|
fullpath = (WCHAR*)calloc(1, base_path_length + PathLength + sizeof(WCHAR));
|
|
|
|
if (!fullpath)
|
|
{
|
|
WLog_ERR(TAG, "malloc failed!");
|
|
return NULL;
|
|
}
|
|
|
|
CopyMemory(fullpath, base_path, base_path_length);
|
|
CopyMemory((char*)fullpath + base_path_length, path, PathLength);
|
|
drive_file_fix_path(fullpath);
|
|
return fullpath;
|
|
}
|
|
|
|
static BOOL drive_file_remove_dir(const WCHAR* path)
|
|
{
|
|
WIN32_FIND_DATAW findFileData;
|
|
BOOL ret = TRUE;
|
|
INT len;
|
|
HANDLE dir;
|
|
WCHAR* fullpath;
|
|
WCHAR* path_slash;
|
|
UINT32 base_path_length;
|
|
|
|
if (!path)
|
|
return FALSE;
|
|
|
|
base_path_length = _wcslen(path) * 2;
|
|
path_slash = (WCHAR*)calloc(1, base_path_length + sizeof(WCHAR) * 3);
|
|
|
|
if (!path_slash)
|
|
{
|
|
WLog_ERR(TAG, "malloc failed!");
|
|
return FALSE;
|
|
}
|
|
|
|
CopyMemory(path_slash, path, base_path_length);
|
|
path_slash[base_path_length / 2] = L'/';
|
|
path_slash[base_path_length / 2 + 1] = L'*';
|
|
DEBUG_WSTR("Search in %s", path_slash);
|
|
dir = FindFirstFileW(path_slash, &findFileData);
|
|
path_slash[base_path_length / 2 + 1] = 0;
|
|
|
|
if (dir == INVALID_HANDLE_VALUE)
|
|
{
|
|
free(path_slash);
|
|
return FALSE;
|
|
}
|
|
|
|
do
|
|
{
|
|
len = _wcslen(findFileData.cFileName);
|
|
|
|
if ((len == 1 && findFileData.cFileName[0] == L'.') || (len == 2 &&
|
|
findFileData.cFileName[0] == L'.' && findFileData.cFileName[1] == L'.'))
|
|
{
|
|
continue;
|
|
}
|
|
|
|
fullpath = drive_file_combine_fullpath(path_slash, findFileData.cFileName, len * 2);
|
|
DEBUG_WSTR("Delete %s", fullpath);
|
|
|
|
if (findFileData.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)
|
|
{
|
|
ret = drive_file_remove_dir(fullpath);
|
|
}
|
|
else
|
|
{
|
|
ret = DeleteFileW(fullpath);
|
|
}
|
|
|
|
free(fullpath);
|
|
|
|
if (!ret)
|
|
break;
|
|
}
|
|
while (ret && FindNextFileW(dir, &findFileData) != 0);
|
|
|
|
FindClose(dir);
|
|
|
|
if (ret)
|
|
{
|
|
if (!RemoveDirectoryW(path))
|
|
{
|
|
ret = FALSE;
|
|
}
|
|
}
|
|
|
|
free(path_slash);
|
|
return ret;
|
|
}
|
|
|
|
static BOOL drive_file_set_fullpath(DRIVE_FILE* file, WCHAR* fullpath)
|
|
{
|
|
if (!file || !fullpath)
|
|
return FALSE;
|
|
|
|
free(file->fullpath);
|
|
file->fullpath = fullpath;
|
|
file->filename = _wcsrchr(file->fullpath, L'/');
|
|
|
|
if (file->filename == NULL)
|
|
file->filename = file->fullpath;
|
|
else
|
|
file->filename += 1;
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
static BOOL drive_file_init(DRIVE_FILE* file)
|
|
{
|
|
UINT CreateDisposition = 0;
|
|
DWORD dwAttr = GetFileAttributesW(file->fullpath);
|
|
|
|
if (dwAttr != INVALID_FILE_ATTRIBUTES)
|
|
{
|
|
/* The file exists */
|
|
file->is_dir = (dwAttr & FILE_ATTRIBUTE_DIRECTORY) != 0;
|
|
|
|
if (file->is_dir)
|
|
{
|
|
if (file->CreateDisposition == FILE_CREATE)
|
|
{
|
|
SetLastError(ERROR_ALREADY_EXISTS);
|
|
return FALSE;
|
|
}
|
|
|
|
if (file->CreateOptions & FILE_NON_DIRECTORY_FILE)
|
|
{
|
|
SetLastError(ERROR_ACCESS_DENIED);
|
|
return FALSE;
|
|
}
|
|
|
|
return TRUE;
|
|
}
|
|
else
|
|
{
|
|
if (file->CreateOptions & FILE_DIRECTORY_FILE)
|
|
{
|
|
SetLastError(ERROR_DIRECTORY);
|
|
return FALSE;
|
|
}
|
|
}
|
|
}
|
|
else
|
|
{
|
|
file->is_dir = ((file->CreateOptions & FILE_DIRECTORY_FILE) ? TRUE : FALSE);
|
|
|
|
if (file->is_dir)
|
|
{
|
|
/* Should only create the directory if the disposition allows for it */
|
|
if ((file->CreateDisposition == FILE_OPEN_IF) || (file->CreateDisposition == FILE_CREATE))
|
|
{
|
|
if (CreateDirectoryW(file->fullpath, NULL) != 0)
|
|
{
|
|
return TRUE;
|
|
}
|
|
}
|
|
|
|
SetLastError(ERROR_FILE_NOT_FOUND);
|
|
return FALSE;
|
|
}
|
|
}
|
|
|
|
if (file->file_handle == INVALID_HANDLE_VALUE)
|
|
{
|
|
switch (file->CreateDisposition)
|
|
{
|
|
case FILE_SUPERSEDE: /* If the file already exists, replace it with the given file. If it does not, create the given file. */
|
|
CreateDisposition = CREATE_ALWAYS;
|
|
break;
|
|
|
|
case FILE_OPEN: /* If the file already exists, open it instead of creating a new file. If it does not, fail the request and do not create a new file. */
|
|
CreateDisposition = OPEN_EXISTING;
|
|
break;
|
|
|
|
case FILE_CREATE: /* If the file already exists, fail the request and do not create or open the given file. If it does not, create the given file. */
|
|
CreateDisposition = CREATE_NEW;
|
|
break;
|
|
|
|
case FILE_OPEN_IF: /* If the file already exists, open it. If it does not, create the given file. */
|
|
CreateDisposition = OPEN_ALWAYS;
|
|
break;
|
|
|
|
case FILE_OVERWRITE: /* If the file already exists, open it and overwrite it. If it does not, fail the request. */
|
|
CreateDisposition = TRUNCATE_EXISTING;
|
|
break;
|
|
|
|
case FILE_OVERWRITE_IF: /* If the file already exists, open it and overwrite it. If it does not, create the given file. */
|
|
CreateDisposition = CREATE_ALWAYS;
|
|
break;
|
|
|
|
default:
|
|
break;
|
|
}
|
|
|
|
#ifndef WIN32
|
|
file->SharedAccess = 0;
|
|
#endif
|
|
file->file_handle = CreateFileW(file->fullpath, file->DesiredAccess,
|
|
file->SharedAccess, NULL, CreateDisposition,
|
|
file->FileAttributes, NULL);
|
|
}
|
|
|
|
if (file->file_handle == INVALID_HANDLE_VALUE)
|
|
{
|
|
/* Get the error message, if any. */
|
|
DWORD errorMessageID = GetLastError();
|
|
|
|
if (errorMessageID != 0)
|
|
{
|
|
#ifdef WIN32
|
|
LPSTR messageBuffer = NULL;
|
|
size_t size = FormatMessageA(FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM |
|
|
FORMAT_MESSAGE_IGNORE_INSERTS,
|
|
NULL, errorMessageID, MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT), (LPSTR)&messageBuffer, 0, NULL);
|
|
WLog_ERR(TAG, "Error in drive_file_init: %s %s", messageBuffer, file->fullpath);
|
|
/* Free the buffer. */
|
|
LocalFree(messageBuffer);
|
|
#endif
|
|
}
|
|
}
|
|
|
|
return file->file_handle != INVALID_HANDLE_VALUE;
|
|
}
|
|
|
|
DRIVE_FILE* drive_file_new(const WCHAR* base_path, const WCHAR* path, UINT32 PathLength, UINT32 id,
|
|
UINT32 DesiredAccess, UINT32 CreateDisposition,
|
|
UINT32 CreateOptions, UINT32 FileAttributes, UINT32 SharedAccess)
|
|
{
|
|
DRIVE_FILE* file;
|
|
|
|
if (!base_path || !path)
|
|
return NULL;
|
|
|
|
file = (DRIVE_FILE*) calloc(1, sizeof(DRIVE_FILE));
|
|
|
|
if (!file)
|
|
{
|
|
WLog_ERR(TAG, "calloc failed!");
|
|
return NULL;
|
|
}
|
|
|
|
file->file_handle = INVALID_HANDLE_VALUE;
|
|
file->find_handle = INVALID_HANDLE_VALUE;
|
|
file->id = id;
|
|
file->basepath = (WCHAR*) base_path;
|
|
file->FileAttributes = FileAttributes;
|
|
file->DesiredAccess = DesiredAccess;
|
|
file->CreateDisposition = CreateDisposition;
|
|
file->CreateOptions = CreateOptions;
|
|
file->SharedAccess = SharedAccess;
|
|
drive_file_set_fullpath(file, drive_file_combine_fullpath(base_path, path, PathLength));
|
|
|
|
if (!drive_file_init(file))
|
|
{
|
|
drive_file_free(file);
|
|
return NULL;
|
|
}
|
|
|
|
return file;
|
|
}
|
|
|
|
BOOL drive_file_free(DRIVE_FILE* file)
|
|
{
|
|
BOOL rc = FALSE;
|
|
|
|
if (!file)
|
|
return FALSE;
|
|
|
|
if (file->file_handle != INVALID_HANDLE_VALUE)
|
|
{
|
|
CloseHandle(file->file_handle);
|
|
file->file_handle = INVALID_HANDLE_VALUE;
|
|
}
|
|
|
|
if (file->find_handle != INVALID_HANDLE_VALUE)
|
|
{
|
|
FindClose(file->find_handle);
|
|
file->find_handle = INVALID_HANDLE_VALUE;
|
|
}
|
|
|
|
if (file->delete_pending)
|
|
{
|
|
if (file->is_dir)
|
|
{
|
|
if (!drive_file_remove_dir(file->fullpath))
|
|
goto fail;
|
|
}
|
|
else if (!DeleteFileW(file->fullpath))
|
|
goto fail;
|
|
}
|
|
|
|
rc = TRUE;
|
|
fail:
|
|
DEBUG_WSTR("Free %s", file->fullpath);
|
|
free(file->fullpath);
|
|
free(file);
|
|
return rc;
|
|
}
|
|
|
|
BOOL drive_file_seek(DRIVE_FILE* file, UINT64 Offset)
|
|
{
|
|
LARGE_INTEGER loffset;
|
|
|
|
if (!file)
|
|
return FALSE;
|
|
|
|
loffset.QuadPart = Offset;
|
|
return SetFilePointerEx(file->file_handle, loffset, NULL, FILE_BEGIN);
|
|
}
|
|
|
|
BOOL drive_file_read(DRIVE_FILE* file, BYTE* buffer, UINT32* Length)
|
|
{
|
|
UINT32 read;
|
|
|
|
if (!file || !buffer || !Length)
|
|
return FALSE;
|
|
|
|
DEBUG_WSTR("Read file %s", file->fullpath);
|
|
|
|
if (ReadFile(file->file_handle, buffer, *Length, &read, NULL))
|
|
{
|
|
*Length = read;
|
|
return TRUE;
|
|
}
|
|
|
|
return FALSE;
|
|
}
|
|
|
|
BOOL drive_file_write(DRIVE_FILE* file, BYTE* buffer, UINT32 Length)
|
|
{
|
|
UINT32 written;
|
|
|
|
if (!file || !buffer)
|
|
return FALSE;
|
|
|
|
DEBUG_WSTR("Write file %s", file->fullpath);
|
|
|
|
while (Length > 0)
|
|
{
|
|
if (!WriteFile(file->file_handle, buffer, Length, &written, NULL))
|
|
return FALSE;
|
|
|
|
Length -= written;
|
|
buffer += written;
|
|
}
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
BOOL drive_file_query_information(DRIVE_FILE* file, UINT32 FsInformationClass, wStream* output)
|
|
{
|
|
WIN32_FILE_ATTRIBUTE_DATA fileAttributes;
|
|
DEBUG_WSTR("FindFirstFile %s", file->fullpath);
|
|
|
|
if (!file || !output)
|
|
return FALSE;
|
|
|
|
if (!GetFileAttributesExW(file->fullpath, GetFileExInfoStandard, &fileAttributes))
|
|
goto out_fail;
|
|
|
|
switch (FsInformationClass)
|
|
{
|
|
case FileBasicInformation:
|
|
|
|
/* http://msdn.microsoft.com/en-us/library/cc232094.aspx */
|
|
if (!Stream_EnsureRemainingCapacity(output, 4 + 36))
|
|
goto out_fail;
|
|
|
|
Stream_Write_UINT32(output, 36); /* Length */
|
|
Stream_Write_UINT32(output, fileAttributes.ftCreationTime.dwLowDateTime); /* CreationTime */
|
|
Stream_Write_UINT32(output, fileAttributes.ftCreationTime.dwHighDateTime); /* CreationTime */
|
|
Stream_Write_UINT32(output, fileAttributes.ftLastAccessTime.dwLowDateTime); /* LastAccessTime */
|
|
Stream_Write_UINT32(output, fileAttributes.ftLastAccessTime.dwHighDateTime); /* LastAccessTime */
|
|
Stream_Write_UINT32(output, fileAttributes.ftLastWriteTime.dwLowDateTime); /* LastWriteTime */
|
|
Stream_Write_UINT32(output, fileAttributes.ftLastWriteTime.dwHighDateTime); /* LastWriteTime */
|
|
Stream_Write_UINT32(output, fileAttributes.ftLastWriteTime.dwLowDateTime); /* ChangeTime */
|
|
Stream_Write_UINT32(output, fileAttributes.ftLastWriteTime.dwHighDateTime); /* ChangeTime */
|
|
Stream_Write_UINT32(output, fileAttributes.dwFileAttributes); /* FileAttributes */
|
|
/* Reserved(4), MUST NOT be added! */
|
|
break;
|
|
|
|
case FileStandardInformation:
|
|
|
|
/* http://msdn.microsoft.com/en-us/library/cc232088.aspx */
|
|
if (!Stream_EnsureRemainingCapacity(output, 4 + 22))
|
|
goto out_fail;
|
|
|
|
Stream_Write_UINT32(output, 22); /* Length */
|
|
Stream_Write_UINT32(output, fileAttributes.nFileSizeLow); /* AllocationSize */
|
|
Stream_Write_UINT32(output, fileAttributes.nFileSizeHigh); /* AllocationSize */
|
|
Stream_Write_UINT32(output, fileAttributes.nFileSizeLow); /* EndOfFile */
|
|
Stream_Write_UINT32(output, fileAttributes.nFileSizeHigh); /* EndOfFile */
|
|
Stream_Write_UINT32(output, 0); /* NumberOfLinks */
|
|
Stream_Write_UINT8(output, file->delete_pending ? 1 : 0); /* DeletePending */
|
|
Stream_Write_UINT8(output, fileAttributes.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY ? TRUE :
|
|
FALSE); /* Directory */
|
|
/* Reserved(2), MUST NOT be added! */
|
|
break;
|
|
|
|
case FileAttributeTagInformation:
|
|
|
|
/* http://msdn.microsoft.com/en-us/library/cc232093.aspx */
|
|
if (!Stream_EnsureRemainingCapacity(output, 4 + 8))
|
|
goto out_fail;
|
|
|
|
Stream_Write_UINT32(output, 8); /* Length */
|
|
Stream_Write_UINT32(output, fileAttributes.dwFileAttributes); /* FileAttributes */
|
|
Stream_Write_UINT32(output, 0); /* ReparseTag */
|
|
break;
|
|
|
|
default:
|
|
/* Unhandled FsInformationClass */
|
|
goto out_fail;
|
|
}
|
|
|
|
return TRUE;
|
|
out_fail:
|
|
Stream_Write_UINT32(output, 0); /* Length */
|
|
return FALSE;
|
|
}
|
|
|
|
BOOL drive_file_set_information(DRIVE_FILE* file, UINT32 FsInformationClass, UINT32 Length,
|
|
wStream* input)
|
|
{
|
|
INT64 size;
|
|
WCHAR* fullpath;
|
|
ULARGE_INTEGER liCreationTime;
|
|
ULARGE_INTEGER liLastAccessTime;
|
|
ULARGE_INTEGER liLastWriteTime;
|
|
ULARGE_INTEGER liChangeTime;
|
|
FILETIME ftCreationTime;
|
|
FILETIME ftLastAccessTime;
|
|
FILETIME ftLastWriteTime;
|
|
FILETIME* pftCreationTime = NULL;
|
|
FILETIME* pftLastAccessTime = NULL;
|
|
FILETIME* pftLastWriteTime = NULL;
|
|
UINT32 FileAttributes;
|
|
UINT32 FileNameLength;
|
|
LARGE_INTEGER liSize;
|
|
UINT8 delete_pending;
|
|
UINT8 ReplaceIfExists;
|
|
DWORD attr;
|
|
|
|
if (!file || !input)
|
|
return FALSE;
|
|
|
|
switch (FsInformationClass)
|
|
{
|
|
case FileBasicInformation:
|
|
if (Stream_GetRemainingLength(input) < 36)
|
|
return FALSE;
|
|
|
|
/* http://msdn.microsoft.com/en-us/library/cc232094.aspx */
|
|
Stream_Read_UINT64(input, liCreationTime.QuadPart);
|
|
Stream_Read_UINT64(input, liLastAccessTime.QuadPart);
|
|
Stream_Read_UINT64(input, liLastWriteTime.QuadPart);
|
|
Stream_Read_UINT64(input, liChangeTime.QuadPart);
|
|
Stream_Read_UINT32(input, FileAttributes);
|
|
|
|
if (!PathFileExistsW(file->fullpath))
|
|
return FALSE;
|
|
|
|
if (file->file_handle == INVALID_HANDLE_VALUE)
|
|
{
|
|
WLog_ERR(TAG, "Unable to set file time %s (%"PRId32")", file->fullpath, GetLastError());
|
|
return FALSE;
|
|
}
|
|
|
|
if (liCreationTime.QuadPart != 0)
|
|
{
|
|
ftCreationTime.dwHighDateTime = liCreationTime.HighPart;
|
|
ftCreationTime.dwLowDateTime = liCreationTime.LowPart;
|
|
pftCreationTime = &ftCreationTime;
|
|
}
|
|
|
|
if (liLastAccessTime.QuadPart != 0)
|
|
{
|
|
ftLastAccessTime.dwHighDateTime = liLastAccessTime.HighPart;
|
|
ftLastAccessTime.dwLowDateTime = liLastAccessTime.LowPart;
|
|
pftLastAccessTime = &ftLastAccessTime;
|
|
}
|
|
|
|
if (liLastWriteTime.QuadPart != 0)
|
|
{
|
|
ftLastWriteTime.dwHighDateTime = liLastWriteTime.HighPart;
|
|
ftLastWriteTime.dwLowDateTime = liLastWriteTime.LowPart;
|
|
pftLastWriteTime = &ftLastWriteTime;
|
|
}
|
|
|
|
if (liChangeTime.QuadPart != 0 && liChangeTime.QuadPart > liLastWriteTime.QuadPart)
|
|
{
|
|
ftLastWriteTime.dwHighDateTime = liChangeTime.HighPart;
|
|
ftLastWriteTime.dwLowDateTime = liChangeTime.LowPart;
|
|
pftLastWriteTime = &ftLastWriteTime;
|
|
}
|
|
|
|
DEBUG_WSTR("SetFileTime %s", file->fullpath);
|
|
|
|
if (!SetFileTime(file->file_handle, pftCreationTime, pftLastAccessTime, pftLastWriteTime))
|
|
{
|
|
WLog_ERR(TAG, "Unable to set file time to %s", file->fullpath);
|
|
return FALSE;
|
|
}
|
|
|
|
SetFileAttributesW(file->fullpath, FileAttributes);
|
|
break;
|
|
|
|
case FileEndOfFileInformation:
|
|
|
|
/* http://msdn.microsoft.com/en-us/library/cc232067.aspx */
|
|
case FileAllocationInformation:
|
|
if (Stream_GetRemainingLength(input) < 8)
|
|
return FALSE;
|
|
|
|
/* http://msdn.microsoft.com/en-us/library/cc232076.aspx */
|
|
Stream_Read_INT64(input, size);
|
|
|
|
if (file->file_handle == INVALID_HANDLE_VALUE)
|
|
{
|
|
WLog_ERR(TAG, "Unable to truncate %s to %"PRId64" (%"PRId32")", file->fullpath, size,
|
|
GetLastError());
|
|
return FALSE;
|
|
}
|
|
|
|
if (!SetFilePointerEx(file->file_handle, liSize, NULL, FILE_BEGIN))
|
|
{
|
|
WLog_ERR(TAG, "Unable to truncate %s to %d (%"PRId32")", file->fullpath, size, GetLastError());
|
|
return FALSE;
|
|
}
|
|
|
|
DEBUG_WSTR("Truncate %s", file->fullpath);
|
|
|
|
if (SetEndOfFile(file->file_handle) == 0)
|
|
{
|
|
WLog_ERR(TAG, "Unable to truncate %s to %d (%"PRId32")", file->fullpath, size, GetLastError());
|
|
return FALSE;
|
|
}
|
|
|
|
break;
|
|
|
|
case FileDispositionInformation:
|
|
|
|
/* http://msdn.microsoft.com/en-us/library/cc232098.aspx */
|
|
/* http://msdn.microsoft.com/en-us/library/cc241371.aspx */
|
|
if (file->is_dir && !PathIsDirectoryEmptyW(file->fullpath))
|
|
break; /* TODO: SetLastError ??? */
|
|
|
|
if (Length)
|
|
{
|
|
if (Stream_GetRemainingLength(input) < 1)
|
|
return FALSE;
|
|
|
|
Stream_Read_UINT8(input, delete_pending);
|
|
}
|
|
else
|
|
delete_pending = 1;
|
|
|
|
if (delete_pending)
|
|
{
|
|
DEBUG_WSTR("SetDeletePending %s", file->fullpath);
|
|
attr = GetFileAttributesW(file->fullpath);
|
|
|
|
if (attr & FILE_ATTRIBUTE_READONLY)
|
|
{
|
|
SetLastError(ERROR_ACCESS_DENIED);
|
|
return FALSE;
|
|
}
|
|
}
|
|
|
|
file->delete_pending = delete_pending;
|
|
break;
|
|
|
|
case FileRenameInformation:
|
|
if (Stream_GetRemainingLength(input) < 6)
|
|
return FALSE;
|
|
|
|
/* http://msdn.microsoft.com/en-us/library/cc232085.aspx */
|
|
Stream_Read_UINT8(input, ReplaceIfExists);
|
|
Stream_Seek_UINT8(input); /* RootDirectory */
|
|
Stream_Read_UINT32(input, FileNameLength);
|
|
|
|
if (Stream_GetRemainingLength(input) < FileNameLength)
|
|
return FALSE;
|
|
|
|
fullpath = drive_file_combine_fullpath(file->basepath, (WCHAR*)Stream_Pointer(input),
|
|
FileNameLength);
|
|
|
|
if (!fullpath)
|
|
{
|
|
WLog_ERR(TAG, "drive_file_combine_fullpath failed!");
|
|
return FALSE;
|
|
}
|
|
|
|
#ifdef _WIN32
|
|
|
|
if (file->file_handle != INVALID_HANDLE_VALUE)
|
|
{
|
|
CloseHandle(file->file_handle);
|
|
file->file_handle = INVALID_HANDLE_VALUE;
|
|
}
|
|
|
|
#endif
|
|
DEBUG_WSTR("MoveFileExW %s", file->fullpath);
|
|
|
|
if (MoveFileExW(file->fullpath, fullpath,
|
|
MOVEFILE_COPY_ALLOWED | (ReplaceIfExists ? MOVEFILE_REPLACE_EXISTING : 0)))
|
|
{
|
|
if (!drive_file_set_fullpath(file, fullpath))
|
|
return FALSE;
|
|
}
|
|
else
|
|
{
|
|
free(fullpath);
|
|
return FALSE;
|
|
}
|
|
|
|
#ifdef _WIN32
|
|
drive_file_init(file);
|
|
#endif
|
|
break;
|
|
|
|
default:
|
|
return FALSE;
|
|
}
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
BOOL drive_file_query_directory(DRIVE_FILE* file, UINT32 FsInformationClass, BYTE InitialQuery,
|
|
const WCHAR* path, UINT32 PathLength, wStream* output)
|
|
{
|
|
size_t length;
|
|
WCHAR* ent_path;
|
|
|
|
if (!file || !path || !output)
|
|
return FALSE;
|
|
|
|
if (InitialQuery != 0)
|
|
{
|
|
/* release search handle */
|
|
if (file->find_handle != INVALID_HANDLE_VALUE)
|
|
FindClose(file->find_handle);
|
|
|
|
ent_path = drive_file_combine_fullpath(file->basepath, path, PathLength);
|
|
/* open new search handle and retrieve the first entry */
|
|
file->find_handle = FindFirstFileW(ent_path, &file->find_data);
|
|
free(ent_path);
|
|
|
|
if (file->find_handle == INVALID_HANDLE_VALUE)
|
|
goto out_fail;
|
|
}
|
|
else if (!FindNextFileW(file->find_handle, &file->find_data))
|
|
goto out_fail;
|
|
|
|
length = _wcslen(file->find_data.cFileName) * 2;
|
|
|
|
switch (FsInformationClass)
|
|
{
|
|
case FileDirectoryInformation:
|
|
|
|
/* http://msdn.microsoft.com/en-us/library/cc232097.aspx */
|
|
if (!Stream_EnsureRemainingCapacity(output, 4 + 64 + length))
|
|
goto out_fail;
|
|
|
|
Stream_Write_UINT32(output, 64 + length); /* Length */
|
|
Stream_Write_UINT32(output, 0); /* NextEntryOffset */
|
|
Stream_Write_UINT32(output, 0); /* FileIndex */
|
|
Stream_Write_UINT32(output, file->find_data.ftCreationTime.dwLowDateTime); /* CreationTime */
|
|
Stream_Write_UINT32(output, file->find_data.ftCreationTime.dwHighDateTime); /* CreationTime */
|
|
Stream_Write_UINT32(output, file->find_data.ftLastAccessTime.dwLowDateTime); /* LastAccessTime */
|
|
Stream_Write_UINT32(output, file->find_data.ftLastAccessTime.dwHighDateTime); /* LastAccessTime */
|
|
Stream_Write_UINT32(output, file->find_data.ftLastWriteTime.dwLowDateTime); /* LastWriteTime */
|
|
Stream_Write_UINT32(output, file->find_data.ftLastWriteTime.dwHighDateTime); /* LastWriteTime */
|
|
Stream_Write_UINT32(output, file->find_data.ftLastWriteTime.dwLowDateTime); /* ChangeTime */
|
|
Stream_Write_UINT32(output, file->find_data.ftLastWriteTime.dwHighDateTime); /* ChangeTime */
|
|
Stream_Write_UINT32(output, file->find_data.nFileSizeLow); /* EndOfFile */
|
|
Stream_Write_UINT32(output, file->find_data.nFileSizeHigh); /* EndOfFile */
|
|
Stream_Write_UINT32(output, file->find_data.nFileSizeLow); /* AllocationSize */
|
|
Stream_Write_UINT32(output, file->find_data.nFileSizeHigh); /* AllocationSize */
|
|
Stream_Write_UINT32(output, file->find_data.dwFileAttributes); /* FileAttributes */
|
|
Stream_Write_UINT32(output, length); /* FileNameLength */
|
|
Stream_Write(output, file->find_data.cFileName, length);
|
|
break;
|
|
|
|
case FileFullDirectoryInformation:
|
|
|
|
/* http://msdn.microsoft.com/en-us/library/cc232068.aspx */
|
|
if (!Stream_EnsureRemainingCapacity(output, 4 + 68 + length))
|
|
goto out_fail;
|
|
|
|
Stream_Write_UINT32(output, 68 + length); /* Length */
|
|
Stream_Write_UINT32(output, 0); /* NextEntryOffset */
|
|
Stream_Write_UINT32(output, 0); /* FileIndex */
|
|
Stream_Write_UINT32(output, file->find_data.ftCreationTime.dwLowDateTime); /* CreationTime */
|
|
Stream_Write_UINT32(output, file->find_data.ftCreationTime.dwHighDateTime); /* CreationTime */
|
|
Stream_Write_UINT32(output, file->find_data.ftLastAccessTime.dwLowDateTime); /* LastAccessTime */
|
|
Stream_Write_UINT32(output, file->find_data.ftLastAccessTime.dwHighDateTime); /* LastAccessTime */
|
|
Stream_Write_UINT32(output, file->find_data.ftLastWriteTime.dwLowDateTime); /* LastWriteTime */
|
|
Stream_Write_UINT32(output, file->find_data.ftLastWriteTime.dwHighDateTime); /* LastWriteTime */
|
|
Stream_Write_UINT32(output, file->find_data.ftLastWriteTime.dwLowDateTime); /* ChangeTime */
|
|
Stream_Write_UINT32(output, file->find_data.ftLastWriteTime.dwHighDateTime); /* ChangeTime */
|
|
Stream_Write_UINT32(output, file->find_data.nFileSizeLow); /* EndOfFile */
|
|
Stream_Write_UINT32(output, file->find_data.nFileSizeHigh); /* EndOfFile */
|
|
Stream_Write_UINT32(output, file->find_data.nFileSizeLow); /* AllocationSize */
|
|
Stream_Write_UINT32(output, file->find_data.nFileSizeHigh); /* AllocationSize */
|
|
Stream_Write_UINT32(output, file->find_data.dwFileAttributes); /* FileAttributes */
|
|
Stream_Write_UINT32(output, length); /* FileNameLength */
|
|
Stream_Write_UINT32(output, 0); /* EaSize */
|
|
Stream_Write(output, file->find_data.cFileName, length);
|
|
break;
|
|
|
|
case FileBothDirectoryInformation:
|
|
|
|
/* http://msdn.microsoft.com/en-us/library/cc232095.aspx */
|
|
if (!Stream_EnsureRemainingCapacity(output, 4 + 93 + length))
|
|
goto out_fail;
|
|
|
|
Stream_Write_UINT32(output, 93 + length); /* Length */
|
|
Stream_Write_UINT32(output, 0); /* NextEntryOffset */
|
|
Stream_Write_UINT32(output, 0); /* FileIndex */
|
|
Stream_Write_UINT32(output, file->find_data.ftCreationTime.dwLowDateTime); /* CreationTime */
|
|
Stream_Write_UINT32(output, file->find_data.ftCreationTime.dwHighDateTime); /* CreationTime */
|
|
Stream_Write_UINT32(output, file->find_data.ftLastAccessTime.dwLowDateTime); /* LastAccessTime */
|
|
Stream_Write_UINT32(output, file->find_data.ftLastAccessTime.dwHighDateTime); /* LastAccessTime */
|
|
Stream_Write_UINT32(output, file->find_data.ftLastWriteTime.dwLowDateTime); /* LastWriteTime */
|
|
Stream_Write_UINT32(output, file->find_data.ftLastWriteTime.dwHighDateTime); /* LastWriteTime */
|
|
Stream_Write_UINT32(output, file->find_data.ftLastWriteTime.dwLowDateTime); /* ChangeTime */
|
|
Stream_Write_UINT32(output, file->find_data.ftLastWriteTime.dwHighDateTime); /* ChangeTime */
|
|
Stream_Write_UINT32(output, file->find_data.nFileSizeLow); /* EndOfFile */
|
|
Stream_Write_UINT32(output, file->find_data.nFileSizeHigh); /* EndOfFile */
|
|
Stream_Write_UINT32(output, file->find_data.nFileSizeLow); /* AllocationSize */
|
|
Stream_Write_UINT32(output, file->find_data.nFileSizeHigh); /* AllocationSize */
|
|
Stream_Write_UINT32(output, file->find_data.dwFileAttributes); /* FileAttributes */
|
|
Stream_Write_UINT32(output, length); /* FileNameLength */
|
|
Stream_Write_UINT32(output, 0); /* EaSize */
|
|
Stream_Write_UINT8(output, 0); /* ShortNameLength */
|
|
/* Reserved(1), MUST NOT be added! */
|
|
Stream_Zero(output, 24); /* ShortName */
|
|
Stream_Write(output, file->find_data.cFileName, length);
|
|
break;
|
|
|
|
case FileNamesInformation:
|
|
|
|
/* http://msdn.microsoft.com/en-us/library/cc232077.aspx */
|
|
if (!Stream_EnsureRemainingCapacity(output, 4 + 12 + length))
|
|
goto out_fail;
|
|
|
|
Stream_Write_UINT32(output, 12 + length); /* Length */
|
|
Stream_Write_UINT32(output, 0); /* NextEntryOffset */
|
|
Stream_Write_UINT32(output, 0); /* FileIndex */
|
|
Stream_Write_UINT32(output, length); /* FileNameLength */
|
|
Stream_Write(output, file->find_data.cFileName, length);
|
|
break;
|
|
|
|
default:
|
|
WLog_ERR(TAG, "unhandled FsInformationClass %"PRIu32, FsInformationClass);
|
|
/* Unhandled FsInformationClass */
|
|
goto out_fail;
|
|
}
|
|
|
|
return TRUE;
|
|
out_fail:
|
|
Stream_Write_UINT32(output, 0); /* Length */
|
|
Stream_Write_UINT8(output, 0); /* Padding */
|
|
return FALSE;
|
|
}
|