Blob Blame History Raw
/********************************************************/
/* ntapi: Native API core library */
/* Copyright (C) 2013--2017 Z. Gilboa */
/* Released under GPLv2 and GPLv3; see COPYING.NTAPI. */
/********************************************************/
#include <psxtypes/psxtypes.h>
#include <ntapi/nt_status.h>
#include <ntapi/nt_unicode.h>
#include "ntapi_impl.h"
static int32_t __fastcall __utf16_to_utf8_handler_1byte_or_null_termination(nt_utf16_callback_args * args)
{
/*******************************************/
/* from: 00000000 0xxxxxxx (little endian) */
/* to: 0xxxxxxx (utf-8) */
/*******************************************/
uint8_t * dst;
if (args->dst >= args->dst_cap)
return NT_STATUS_BUFFER_TOO_SMALL;
dst = (uint8_t *)args->dst;
*dst = *(uint8_t *)(args->src);
/* advance source and destination buffer */
args->src++;
args->dst = (void *)((uintptr_t)(args->dst) + 1);
/* bytes_written */
args->bytes_written++;
return NT_STATUS_SUCCESS;
}
static int32_t __fastcall __utf16_to_utf8_handler_2bytes(nt_utf16_callback_args * args)
{
/*******************************************/
/* from: 00000yyy yyxxxxxx (little endian) */
/* to: 110yyyyy 10xxxxxx (utf-8) */
/*******************************************/
const wchar16_t * src;
uint8_t * dst;
wchar16_t wx;
wchar16_t wy;
if ((uintptr_t)(args->dst) + 1 >= (uintptr_t)(args->dst_cap))
return NT_STATUS_BUFFER_TOO_SMALL;
src = args->src;
dst = (uint8_t *)args->dst;
wy = *src;
wy >>= 6;
wx = *src;
wx <<= 10;
wx >>= 10;
/* write the y part */
*dst = (char)(0xC0 | wy);
dst++;
/* write the x part */
*dst = (char)(0x80 | wx);
/* advance source and destination buffer */
args->src++;
args->dst = (void *)((uintptr_t)(args->dst) + 2);
/* bytes_written */
args->bytes_written += 2;
return NT_STATUS_SUCCESS;
}
static int32_t __fastcall __utf16_to_utf8_handler_3bytes(nt_utf16_callback_args * args)
{
/********************************************/
/* from: zzzzyyyy yyxxxxxx (little endian) */
/* to: 1110zzzz 10yyyyyy 10xxxxxx (utf-8) */
/********************************************/
const wchar16_t * src;
uint8_t * dst;
wchar16_t wx;
wchar16_t wy;
wchar16_t wz;
if ((uintptr_t)(args->dst) + 2 >= (uintptr_t)(args->dst_cap))
return NT_STATUS_BUFFER_TOO_SMALL;
src = args->src;
dst = (uint8_t *)args->dst;
wz = *src;
wz >>= 12;
wy = *src;
wy <<= 4;
wy >>= 10;
wx = *src;
wx <<= 10;
wx >>= 10;
/* write the z part */
*dst = (char)(0xE0 | wz);
dst++;
/* write the y part */
*dst = (char)(0x80 | wy);
dst++;
/* write the x part */
*dst = (char)(0x80 | wx);
/* advance source and destination buffer */
args->src++;
args->dst = (void *)((uintptr_t)(args->dst) + 3);
/* bytes_written */
args->bytes_written += 3;
return NT_STATUS_SUCCESS;
}
static int32_t __fastcall __utf16_to_utf8_handler_4bytes(nt_utf16_callback_args * args)
{
/****************************************************************/
/* from: 110110ww wwzzzzyy 110111yy yyxxxxxx (little endian) */
/* to: 11110uuu 10uuzzzz 10yyyyyy 10xxxxxx (utf-8) */
/****************************************************************/
const wchar16_t * src;
uint8_t * dst;
wchar16_t wx;
wchar16_t wz;
wchar16_t wy_low;
wchar16_t wy_high;
wchar16_t ww;
wchar16_t uuuuu;
wchar16_t u_low;
wchar16_t u_high;
if ((uintptr_t)(args->dst) + 3 >= (uintptr_t)(args->dst_cap))
return NT_STATUS_BUFFER_TOO_SMALL;
src = args->src;
dst = (uint8_t *)args->dst;
/* low two bytes */
wy_high = *src;
wy_high <<= 14;
wy_high >>= 10;
wz = *src;
wz <<= 10;
wz >>= 12;
ww = *src;
ww <<= 6;
ww >>= 12;
/* (surrogate pair) */
src++;
/* high two bytes */
wx = *src;
wx <<= 10;
wx >>= 10;
wy_low = *src;
wy_low <<= 6;
wy_low >>= 12;
/* uuuuu */
uuuuu = ww + 1;
u_low = uuuuu;
u_low >>= 2;
u_high = uuuuu;
u_high <<= 14;
u_high >>= 10;
/* 1st byte: 11110uuu */
*dst = (char)(0xF0 | u_low);
dst++;
/* 2nd byte: 10uuzzzz */
*dst = (char)(0x80 | u_high | wz);
dst++;
/* 3rd byte: 10yyyyyy */
*dst = (char)(0x80 | wy_low | wy_high);
dst++;
/* 4th byte: 10xxxxxx */
*dst = (char)(0x80 | wx);
/* advance source and destination buffer */
args->src += 2;
args->dst = (void *)((uintptr_t)(args->dst) + 4);
/* bytes_written */
args->bytes_written += 4;
return NT_STATUS_SUCCESS;
}
static int32_t __fastcall __update_stream_leftover_info_utf16(
__in_out nt_unicode_conversion_params_utf16_to_utf8 * params)
{
int32_t status;
ptrdiff_t offset;
wchar16_t * wlead;
offset = (uintptr_t)params->src + (uintptr_t)params->src_size_in_bytes - (uintptr_t)params->addr_failed;
wlead = (wchar16_t *)params->addr_failed;
if ((offset == 2) && (*wlead >= 0xD800) && (*wlead < 0xDC00)) {
/* possibly the lead of a surrogate pair lead */
params->leftover_count = 2;
params->leftover_bytes = *wlead;
params->leftover_bytes <<= 16;
status = NT_STATUS_SUCCESS;
} else {
params->leftover_count = 0;
params->leftover_bytes = 0;
status = NT_STATUS_ILLEGAL_CHARACTER;
}
return status;
}
int32_t __stdcall __ntapi_uc_convert_unicode_stream_utf16_to_utf8(
__in_out nt_unicode_conversion_params_utf16_to_utf8 * params)
{
int32_t status;
nt_utf16_callback_args args;
ntapi_uc_utf16_callback_fn * callback_fn[5];
callback_fn[0] = (ntapi_uc_utf16_callback_fn *)__utf16_to_utf8_handler_1byte_or_null_termination;
callback_fn[1] = (ntapi_uc_utf16_callback_fn *)__utf16_to_utf8_handler_1byte_or_null_termination;
callback_fn[2] = (ntapi_uc_utf16_callback_fn *)__utf16_to_utf8_handler_2bytes;
callback_fn[3] = (ntapi_uc_utf16_callback_fn *)__utf16_to_utf8_handler_3bytes;
callback_fn[4] = (ntapi_uc_utf16_callback_fn *)__utf16_to_utf8_handler_4bytes;
args.src = params->src;
args.dst = params->dst;
args.dst_cap = (void *)((uintptr_t)(params->dst) + (params->dst_size_in_bytes));
args.bytes_written = params->bytes_written;
status = __ntapi_uc_validate_unicode_stream_utf16(
params->src,
params->src_size_in_bytes,
&params->code_points,
&params->addr_failed,
callback_fn,
&args);
params->bytes_written = args.bytes_written;
switch (status) {
case NT_STATUS_SUCCESS:
params->addr_failed = 0;
params->leftover_bytes = 0;
params->leftover_count = 0;
return status;
case NT_STATUS_BUFFER_TOO_SMALL:
params->addr_failed = args.src;
params->leftover_bytes = 0;
params->leftover_count = 0;
return status;
default:
status = __update_stream_leftover_info_utf16(params);
}
/* (optimized out on 32-bit architectures) */
params->leftover_bytes <<= (8 * (sizeof(uintptr_t) - sizeof(uint32_t)));
return status;
}
int32_t __stdcall __ntapi_uc_convert_unicode_stream_utf16_to_utf32(
__in_out nt_unicode_conversion_params_utf16_to_utf32 * params)
{
(void)params;
return NT_STATUS_SUCCESS;
}