Something went wrong. Try again.
Reactos
Something went wrong. Try again.
5.9 kB · 216 lines
C++
at master
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217/* * 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>#define RtlUTF8ToUnicodeN RtlUTF8ToUnicodeN_#include <ndk/rtlfuncs.h>
#ifdef __cplusplusextern "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[MaximumFileInfoByHandleClass] ={ FileBasicInformation, FileStandardInformation, FileNameInformation, FileRenameInformation, FileDispositionInformation, FileAllocationInformation, FileEndOfFileInformation, FileStreamInformation, FileCompressionInformation, FileAttributeTagInformation, FileIdBothDirectoryInformation, (FILE_INFORMATION_CLASS)-1, FileIoPriorityHintInformation, FileRemoteProtocolInformation};
/* Taken from kernel32 */DWORDBaseSetLastNTError(IN NTSTATUS Status){ DWORD dwErrCode; dwErrCode = RtlNtStatusToDosError(Status); SetLastError(dwErrCode); return dwErrCode;}
/* Quick implementation, still going farther than Wine implementation */BOOLWINAPISetFileInformationByHandle(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 < MaximumFileInfoByHandleClass) { 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