mirror of
https://github.com/reactos/reactos.git
synced 2024-08-30 07:08:19 +00:00
216 lines
5.7 KiB
C++
216 lines
5.7 KiB
C++
/*
|
|
* COPYRIGHT: See COPYING in the top level directory
|
|
* PROJECT: BtrFS FSD for ReactOS
|
|
* FILE: dll/shellext/shellbtrfs/reactos.cpp
|
|
* PURPOSE: ReactOS glue for Win8.1
|
|
* PROGRAMMERS: Pierre Schweitzer <pierre@reactos.org>
|
|
*/
|
|
|
|
#include "shellext.h"
|
|
#include <initguid.h>
|
|
#include <ntddstor.h>
|
|
#include <ndk/rtlfuncs.h>
|
|
|
|
#ifdef __cplusplus
|
|
extern "C" {
|
|
#endif
|
|
/* So that we can link */
|
|
DEFINE_GUID(CLSID_WICImagingFactory, 0xcacaf262,0x9370,0x4615,0xa1,0x3b,0x9f,0x55,0x39,0xda,0x4c,0x0a);
|
|
|
|
/* Copied from ntoskrnl_vista */
|
|
NTSTATUS WINAPI RtlUTF8ToUnicodeN(PWSTR uni_dest, ULONG uni_bytes_max,
|
|
PULONG uni_bytes_written,
|
|
PCCH utf8_src, ULONG utf8_bytes)
|
|
{
|
|
NTSTATUS status;
|
|
ULONG i, j;
|
|
ULONG written;
|
|
ULONG ch;
|
|
ULONG utf8_trail_bytes;
|
|
WCHAR utf16_ch[3];
|
|
ULONG utf16_ch_len;
|
|
|
|
if (!utf8_src)
|
|
return STATUS_INVALID_PARAMETER_4;
|
|
if (!uni_bytes_written)
|
|
return STATUS_INVALID_PARAMETER;
|
|
|
|
written = 0;
|
|
status = STATUS_SUCCESS;
|
|
|
|
for (i = 0; i < utf8_bytes; i++)
|
|
{
|
|
/* read UTF-8 lead byte */
|
|
ch = (BYTE)utf8_src[i];
|
|
utf8_trail_bytes = 0;
|
|
if (ch >= 0xf5)
|
|
{
|
|
ch = 0xfffd;
|
|
status = STATUS_SOME_NOT_MAPPED;
|
|
}
|
|
else if (ch >= 0xf0)
|
|
{
|
|
ch &= 0x07;
|
|
utf8_trail_bytes = 3;
|
|
}
|
|
else if (ch >= 0xe0)
|
|
{
|
|
ch &= 0x0f;
|
|
utf8_trail_bytes = 2;
|
|
}
|
|
else if (ch >= 0xc2)
|
|
{
|
|
ch &= 0x1f;
|
|
utf8_trail_bytes = 1;
|
|
}
|
|
else if (ch >= 0x80)
|
|
{
|
|
/* overlong or trail byte */
|
|
ch = 0xfffd;
|
|
status = STATUS_SOME_NOT_MAPPED;
|
|
}
|
|
|
|
/* read UTF-8 trail bytes */
|
|
if (i + utf8_trail_bytes < utf8_bytes)
|
|
{
|
|
for (j = 0; j < utf8_trail_bytes; j++)
|
|
{
|
|
if ((utf8_src[i + 1] & 0xc0) == 0x80)
|
|
{
|
|
ch <<= 6;
|
|
ch |= utf8_src[i + 1] & 0x3f;
|
|
i++;
|
|
}
|
|
else
|
|
{
|
|
ch = 0xfffd;
|
|
utf8_trail_bytes = 0;
|
|
status = STATUS_SOME_NOT_MAPPED;
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
else
|
|
{
|
|
ch = 0xfffd;
|
|
utf8_trail_bytes = 0;
|
|
status = STATUS_SOME_NOT_MAPPED;
|
|
i = utf8_bytes;
|
|
}
|
|
|
|
/* encode ch as UTF-16 */
|
|
if ((ch > 0x10ffff) ||
|
|
(ch >= 0xd800 && ch <= 0xdfff) ||
|
|
(utf8_trail_bytes == 2 && ch < 0x00800) ||
|
|
(utf8_trail_bytes == 3 && ch < 0x10000))
|
|
{
|
|
/* invalid codepoint or overlong encoding */
|
|
utf16_ch[0] = 0xfffd;
|
|
utf16_ch[1] = 0xfffd;
|
|
utf16_ch[2] = 0xfffd;
|
|
utf16_ch_len = utf8_trail_bytes;
|
|
status = STATUS_SOME_NOT_MAPPED;
|
|
}
|
|
else if (ch >= 0x10000)
|
|
{
|
|
/* surrogate pair */
|
|
ch -= 0x010000;
|
|
utf16_ch[0] = 0xd800 + (ch >> 10 & 0x3ff);
|
|
utf16_ch[1] = 0xdc00 + (ch >> 0 & 0x3ff);
|
|
utf16_ch_len = 2;
|
|
}
|
|
else
|
|
{
|
|
/* single unit */
|
|
utf16_ch[0] = ch;
|
|
utf16_ch_len = 1;
|
|
}
|
|
|
|
if (!uni_dest)
|
|
{
|
|
written += utf16_ch_len;
|
|
continue;
|
|
}
|
|
|
|
for (j = 0; j < utf16_ch_len; j++)
|
|
{
|
|
if (uni_bytes_max >= sizeof(WCHAR))
|
|
{
|
|
*uni_dest++ = utf16_ch[j];
|
|
uni_bytes_max -= sizeof(WCHAR);
|
|
written++;
|
|
}
|
|
else
|
|
{
|
|
uni_bytes_max = 0;
|
|
status = STATUS_BUFFER_TOO_SMALL;
|
|
}
|
|
}
|
|
}
|
|
|
|
*uni_bytes_written = written * sizeof(WCHAR);
|
|
return status;
|
|
}
|
|
|
|
/* Quick and dirty table for conversion */
|
|
FILE_INFORMATION_CLASS ConvertToFileInfo[MaximumFileInfoByHandlesClass] =
|
|
{
|
|
FileBasicInformation, FileStandardInformation, FileNameInformation, FileRenameInformation,
|
|
FileDispositionInformation, FileAllocationInformation, FileEndOfFileInformation, FileStreamInformation,
|
|
FileCompressionInformation, FileAttributeTagInformation, FileIdBothDirectoryInformation, (FILE_INFORMATION_CLASS)-1,
|
|
FileIoPriorityHintInformation, FileRemoteProtocolInformation
|
|
};
|
|
|
|
/* Taken from kernel32 */
|
|
DWORD
|
|
BaseSetLastNTError(IN NTSTATUS Status)
|
|
{
|
|
DWORD dwErrCode;
|
|
dwErrCode = RtlNtStatusToDosError(Status);
|
|
SetLastError(dwErrCode);
|
|
return dwErrCode;
|
|
}
|
|
|
|
/* Quick implementation, still going farther than Wine implementation */
|
|
BOOL
|
|
WINAPI
|
|
SetFileInformationByHandle(HANDLE hFile,
|
|
FILE_INFO_BY_HANDLE_CLASS FileInformationClass,
|
|
LPVOID lpFileInformation,
|
|
DWORD dwBufferSize)
|
|
{
|
|
NTSTATUS Status;
|
|
IO_STATUS_BLOCK IoStatusBlock;
|
|
FILE_INFORMATION_CLASS FileInfoClass;
|
|
|
|
FileInfoClass = (FILE_INFORMATION_CLASS)-1;
|
|
|
|
/* Attempt to convert the class */
|
|
if (FileInformationClass < MaximumFileInfoByHandlesClass)
|
|
{
|
|
FileInfoClass = ConvertToFileInfo[FileInformationClass];
|
|
}
|
|
|
|
/* If wrong, bail out */
|
|
if (FileInfoClass == -1)
|
|
{
|
|
SetLastError(ERROR_INVALID_PARAMETER);
|
|
return FALSE;
|
|
}
|
|
|
|
/* And set the information */
|
|
Status = NtSetInformationFile(hFile, &IoStatusBlock, lpFileInformation,
|
|
dwBufferSize, FileInfoClass);
|
|
|
|
if (!NT_SUCCESS(Status))
|
|
{
|
|
BaseSetLastNTError(Status);
|
|
return FALSE;
|
|
}
|
|
|
|
return TRUE;
|
|
}
|
|
#ifdef __cplusplus
|
|
}
|
|
#endif
|