1
0
mirror of https://github.com/ONLYOFFICE/core.git synced 2025-04-18 14:04:06 +03:00
core/UnicodeConverter/UnicodeConverter.cpp
2023-03-01 23:21:57 +03:00

499 lines
15 KiB
C++

/*
* (c) Copyright Ascensio System SIA 2010-2023
*
* This program is a free software product. You can redistribute it and/or
* modify it under the terms of the GNU Affero General Public License (AGPL)
* version 3 as published by the Free Software Foundation. In accordance with
* Section 7(a) of the GNU AGPL its Section 15 shall be amended to the effect
* that Ascensio System SIA expressly excludes the warranty of non-infringement
* of any third-party rights.
*
* This program is distributed WITHOUT ANY WARRANTY; without even the implied
* warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. For
* details, see the GNU AGPL at: http://www.gnu.org/licenses/agpl-3.0.html
*
* You can contact Ascensio System SIA at 20A-6 Ernesta Birznieka-Upish
* street, Riga, Latvia, EU, LV-1050.
*
* The interactive user interfaces in modified source and object code versions
* of the Program must display Appropriate Legal Notices, as required under
* Section 5 of the GNU AGPL version 3.
*
* Pursuant to Section 7(b) of the License you must retain the original Product
* logo when distributing the program. Pursuant to Section 7(e) we decline to
* grant you any rights under trademark law for use of our trademarks.
*
* All the Product's GUI elements, including illustrations and icon sets, as
* well as technical writing content are licensed under the terms of the
* Creative Commons Attribution-ShareAlike 4.0 International. See the License
* terms at http://creativecommons.org/licenses/by-sa/4.0/legalcode
*
*/
#include "stdint.h"
#include "./UnicodeConverter.h"
#include "unicode/utypes.h"
#include "unicode/ustring.h"
#include "unicode/ucnv.h" /* C Converter API */
#include "unicode/usprep.h"
//#include "../DesktopEditor/common/File.h"
#if !defined (_WIN32) && !defined (_WIN64)
#if defined (_LINUX) && !defined(__ANDROID__) && !defined(_IOS) && !defined(_MAC)
#define USE_ICONV
#include "iconv.h"
#endif
#else
#include <windows.h>
#endif
std::string g_overrideIcuDataPath = "";
namespace NSUnicodeConverter
{
class CUnicodeConverter_Private
{
public:
CUnicodeConverter_Private()
{
#if 0
std::wstring sDumpPath = NSFile::GetProcessDirectory() + L"/codepages.txt";
NSFile::CFileBinary oFile;
oFile.CreateFileW(sDumpPath);
int32_t count = ucnv_countAvailable();
for (int i = 0; i < count; ++i)
{
std::string sCodePage = ucnv_getAvailableName(i);
UErrorCode _error = U_ZERO_ERROR;
int nCountAliases = ucnv_countAliases(sCodePage.c_str(), &_error);
char** palices = new char*[nCountAliases];
ucnv_getAliases(sCodePage.c_str(), (const char**)palices, &_error);
oFile.WriteStringUTF8(L"-----------------------------\r\n");
oFile.WriteFile((BYTE*)sCodePage.c_str(), sCodePage.length());
oFile.WriteStringUTF8(L"\r\n");
for (int j = 0; j < nCountAliases; ++j)
{
oFile.WriteFile((BYTE*)palices[j], strlen(palices[j]));
oFile.WriteStringUTF8(L"\r\n");
}
}
oFile.CloseFile();
#endif
}
~CUnicodeConverter_Private()
{
}
std::string SASLprepToUtf8(const wchar_t* sInput, const unsigned int& nInputLen)
{
#ifndef DISABLE_ICU
std::string sRes;
UErrorCode status = U_ZERO_ERROR;
int32_t nUCharCapacity = (int32_t)nInputLen;// UTF-16 uses 2 code-points per char
UChar* pUChar = new UChar[(uint32_t)nUCharCapacity * sizeof(UChar)];
if (pUChar)
{
const UChar* pUCharStart = pUChar;
int32_t nUCharLength = 0;
u_strFromWCS(pUChar, nUCharCapacity, &nUCharLength, sInput, (int32_t)nInputLen, &status);
if (U_SUCCESS(status))
{
UStringPrepProfile *profile = usprep_openByType(USPREP_RFC4013_SASLPREP, &status);
UParseError parseError;
int32_t nOutputLen = nUCharLength * 2;
if (U_SUCCESS(status))
{
UChar* pOutput = new UChar[(uint32_t)nOutputLen * sizeof(UChar) * 3];
nOutputLen = usprep_prepare(profile, pUCharStart, nUCharLength, pOutput, nOutputLen, 0, &parseError, &status );
if (U_SUCCESS(status))
{
UConverter* conv = ucnv_open("UTF-8", &status);
if (U_SUCCESS(status))
{
const UChar* pOutputLimit = pOutput + nOutputLen;
const UChar* pOutputStart = pOutput;
char *sResStart = new char[nOutputLen * ucnv_getMaxCharSize(conv)];
char *sResCur = sResStart;
const char *sResLimit = sResCur + nOutputLen * ucnv_getMaxCharSize(conv);
ucnv_fromUnicode(conv, &sResCur, sResLimit, &pOutputStart, pOutputLimit, NULL, TRUE, &status);
if (U_SUCCESS(status))
{
sRes = std::string(sResStart, sResCur - sResStart);
}
delete []sResStart;
ucnv_close(conv);
}
}
delete []pOutput;
usprep_close(profile);
}
}
delete []pUChar;
}
return sRes;
#else
std::wstring strInput(sInput, nInputLen);
std::string sRes(strInput.begin(), strInput.end());
return sRes;
#endif
}
static std::string fromUnicode(const wchar_t* sInput, const unsigned int& nInputLen, const char* converterName)
{
std::string sRes = "";
UErrorCode status = U_ZERO_ERROR;
UConverter* conv = ucnv_open(converterName, &status);
if (U_SUCCESS(status))
{
int32_t nUCharCapacity = (int32_t)nInputLen;// UTF-16 uses 2 code-points per char
//UChar* pUChar = new UChar[nUCharCapacity];
UChar* pUChar = (UChar*)malloc((uint32_t)nUCharCapacity * sizeof(UChar));
if (pUChar)
{
const UChar* pUCharStart = pUChar;
int32_t nUCharLength = 0;
u_strFromWCS(pUChar, nUCharCapacity, &nUCharLength, sInput, (int32_t)nInputLen, &status);
if (U_SUCCESS(status))
{
const UChar* pUCharLimit = pUCharStart + nUCharLength;
sRes.resize((uint32_t)nUCharLength * (uint8_t)ucnv_getMaxCharSize(conv));// UTF-16 uses 2 code-points per char
char *sResStart = &sRes[0];
char *sResCur = sResStart;
const char *sResLimit = sResCur + sRes.size();
ucnv_fromUnicode(conv, &sResCur, sResLimit, &pUCharStart, pUCharLimit, NULL, TRUE, &status);
if (U_SUCCESS(status))
{
sRes.resize(sResCur - sResStart);
}
else
{
sRes.clear();
}
}
//delete []pUCharStart;
}
ucnv_close(conv);
}
if (sRes.empty() && nInputLen > 0)
{
std::wstring ws(sInput, nInputLen);
sRes = std::string(ws.begin(), ws.end());
}
return sRes;
}
std::wstring toUnicode(const char* sInput, const unsigned int& nInputLen, int nCodePage, bool isExact)
{
std::wstring sRes = L"";
UErrorCode status = U_ZERO_ERROR;
UConverter* conv = ucnv_openCCSID(nCodePage, UCNV_UNKNOWN, &status);
if (U_SUCCESS(status))
{
//UConverter* conv = ucnv_openCCSID(5347, UCNV_IBM, &status);
if (U_SUCCESS(status))
{
const char* source = sInput;
const char* sourceLimit = source + nInputLen;
unsigned int uBufSize = (nInputLen / (uint8_t)ucnv_getMinCharSize(conv));
UChar* targetStart = new UChar[uBufSize * sizeof(UChar)];
if (targetStart)
{
UChar* target = targetStart;
UChar* targetLimit = target + uBufSize;
ucnv_toUnicode(conv, &target, targetLimit, &source, sourceLimit, NULL, TRUE, &status);
if (U_SUCCESS(status))
{
size_t nTargetSize = target - targetStart;
sRes.resize(nTargetSize * 2);// UTF-16 uses 2 code-points per char
int32_t nResLen = 0;
u_strToWCS(&sRes[0], (int32_t)sRes.size(), &nResLen, targetStart, (int32_t)nTargetSize, &status);
if (U_SUCCESS(status))
{
sRes.resize((size_t)nResLen);
}
else
{
sRes.clear();
}
}
delete []targetStart;
}
ucnv_close(conv);
}
}
else
{//10008 & other
sRes = convert_string(sInput, nInputLen, nCodePage);
}
if (!isExact && sRes.empty() && nInputLen > 0)
{
std::string ws(sInput, nInputLen);
sRes = std::wstring(ws.begin(), ws.end());
}
return sRes;
}
std::wstring toUnicode(const char* sInput, const unsigned int& nInputLen, const char* converterName, bool isExact)
{
std::wstring sRes = L"";
UErrorCode status = U_ZERO_ERROR;
UConverter* conv = ucnv_open(converterName, &status);
if (U_SUCCESS(status))
{
//UConverter* conv = ucnv_openCCSID(5347, UCNV_IBM, &status);
if (U_SUCCESS(status))
{
const char* source = sInput;
const char* sourceLimit = source + nInputLen;
unsigned int uBufSize = (nInputLen / (uint8_t)ucnv_getMinCharSize(conv));
UChar* targetStart = new UChar[uBufSize * sizeof(UChar)];
if (targetStart)
{
UChar* target = targetStart;
UChar* targetLimit = target + uBufSize;
ucnv_toUnicode(conv, &target, targetLimit, &source, sourceLimit, NULL, TRUE, &status);
if (U_SUCCESS(status))
{
size_t nTargetSize = target - targetStart;
sRes.resize(nTargetSize * 2);// UTF-16 uses 2 code-points per char
int32_t nResLen = 0;
u_strToWCS(&sRes[0], (int32_t)sRes.size(), &nResLen, targetStart, (int32_t)nTargetSize, &status);
if (U_SUCCESS(status))
{
sRes.resize((size_t)nResLen);
}
else
{
sRes.clear();
}
}
delete []targetStart;
}
ucnv_close(conv);
}
}
if (isExact && sRes.empty() && nInputLen > 0)
{
std::string ws(sInput, nInputLen);
sRes = std::wstring(ws.begin(), ws.end());
}
return sRes;
}
std::wstring convert_string(const char* sInput, const unsigned int& nInputLen, int nCodepage = 0)
{
bool ansi = true;
std::wstring sResult;
size_t insize = (size_t)nInputLen;
char* inptr = (char*)sInput;
if (nCodepage > 0)
{
#if defined (_WIN32) || defined (_WIN64)
int outsize_with_0 = MultiByteToWideChar(nCodepage, 0, inptr, -1, NULL, NULL);
sResult.resize(outsize_with_0);
if (MultiByteToWideChar(nCodepage, 0, inptr, -1, (LPWSTR)sResult.c_str(), outsize_with_0) > 0)
{
sResult.erase(outsize_with_0 - 1);
ansi = false;
}
#elif defined(USE_ICONV)
std::string sCodepage = "CP" + std::to_string(nCodepage);
iconv_t ic= iconv_open("WCHAR_T", sCodepage.c_str());
if (ic != (iconv_t) -1)
{
sResult.resize(insize);
char* outptr = (char*)sResult.c_str();
size_t nconv = 0, avail = (insize) * sizeof(wchar_t), outsize = insize;
nconv = iconv (ic, &inptr, &insize, &outptr, &avail);
if (nconv == 0)
{
if (avail > 0)
{
outsize = outsize - avail/sizeof(wchar_t);
sResult.erase(sResult.begin() + outsize);
}
ansi = false;
}
iconv_close(ic);
}
#endif
}
if (ansi)
{
std::string inp(sInput, nInputLen);
sResult = std::wstring(inp.begin(), inp.end());
}
return sResult;
}
std::string GetUtf8StringFromUnicode(const wchar_t* pUnicodes, size_t lCount)
{
if (NULL == pUnicodes || 0 == lCount)
return "";
unsigned char* pData = new unsigned char[6 * lCount + 1];
size_t lOutputCount = 0;
unsigned char* pCodesCur = pData;
const wchar_t* pEnd = pUnicodes + lCount;
const wchar_t* pCur = pUnicodes;
bool bIsUtf16 = (2 == sizeof(wchar_t)) ? true : false;
while (pCur < pEnd)
{
unsigned int code = (unsigned int)*pCur++;
if (bIsUtf16)
{
if (code >= 0xD800 && code <= 0xDFFF && pCur < pEnd)
{
code = 0x10000 + (((code & 0x3FF) << 10) | (0x03FF & *pCur++));
}
}
if (code < 0x80)
{
*pCodesCur++ = (unsigned char)code;
}
else if (code < 0x0800)
{
*pCodesCur++ = 0xC0 | (code >> 6);
*pCodesCur++ = 0x80 | (code & 0x3F);
}
else if (code < 0x10000)
{
*pCodesCur++ = 0xE0 | (code >> 12);
*pCodesCur++ = 0x80 | (code >> 6 & 0x3F);
*pCodesCur++ = 0x80 | (code & 0x3F);
}
else if (code < 0x1FFFFF)
{
*pCodesCur++ = 0xF0 | (code >> 18);
*pCodesCur++ = 0x80 | (code >> 12 & 0x3F);
*pCodesCur++ = 0x80 | (code >> 6 & 0x3F);
*pCodesCur++ = 0x80 | (code & 0x3F);
}
else if (code < 0x3FFFFFF)
{
*pCodesCur++ = 0xF8 | (code >> 24);
*pCodesCur++ = 0x80 | (code >> 18 & 0x3F);
*pCodesCur++ = 0x80 | (code >> 12 & 0x3F);
*pCodesCur++ = 0x80 | (code >> 6 & 0x3F);
*pCodesCur++ = 0x80 | (code & 0x3F);
}
else if (code < 0x7FFFFFFF)
{
*pCodesCur++ = 0xFC | (code >> 30);
*pCodesCur++ = 0x80 | (code >> 24 & 0x3F);
*pCodesCur++ = 0x80 | (code >> 18 & 0x3F);
*pCodesCur++ = 0x80 | (code >> 12 & 0x3F);
*pCodesCur++ = 0x80 | (code >> 6 & 0x3F);
*pCodesCur++ = 0x80 | (code & 0x3F);
}
}
lOutputCount = (size_t)(pCodesCur - pData);
*pCodesCur++ = 0;
std::string s((char*)pData, lOutputCount);
if (pData) delete []pData;
pData = NULL;
return s;
}
};
}
namespace NSUnicodeConverter
{
CUnicodeConverter::CUnicodeConverter()
{
m_pInternal = new CUnicodeConverter_Private();
}
CUnicodeConverter::~CUnicodeConverter()
{
delete m_pInternal;
}
std::string CUnicodeConverter::fromUnicode(const wchar_t* sInput, const unsigned int& nInputLen, const char* converterName)
{
return m_pInternal->fromUnicode(sInput, nInputLen, converterName);
}
std::string CUnicodeConverter::fromUnicode(const std::wstring &sInput, const char *converterName)
{
return this->fromUnicode(sInput.c_str(), (unsigned int)sInput.size(), converterName);
}
std::wstring CUnicodeConverter::toUnicode(const char* sInput, const unsigned int& nInputLen, const char* converterName, bool isExact)
{
return m_pInternal->toUnicode(sInput, nInputLen, converterName, isExact);
}
std::wstring CUnicodeConverter::toUnicode(const std::string &sInput, const char *converterName, bool isExact)
{
return this->toUnicode(sInput.c_str(), (unsigned int)sInput.size(), converterName, isExact);
}
std::wstring CUnicodeConverter::toUnicode(const char* sInput, const unsigned int& nInputLen, int nCodePage, bool isExact)
{
if (nInputLen < 1) return std::wstring(L"");
std::wstring result;
result = m_pInternal->toUnicode(sInput, nInputLen, nCodePage, isExact);
if (result.empty() && nInputLen > 0)
{
result = m_pInternal->convert_string(sInput, nInputLen, nCodePage);
}
return result;
}
std::wstring CUnicodeConverter::toUnicode(const std::string &sInput, int nCodePage, bool isExact)
{
return this->toUnicode(sInput.c_str(), (unsigned int)sInput.size(), nCodePage, isExact);
}
std::string CUnicodeConverter::SASLprepToUtf8(const std::wstring &sSrc)
{
return m_pInternal->SASLprepToUtf8(sSrc.c_str(), sSrc.length());
}
void CUnicodeConverter::setIcuDataPath(const std::wstring& sDirectory)
{
if (!g_overrideIcuDataPath.empty())
return;
CUnicodeConverter_Private internal;
g_overrideIcuDataPath = internal.GetUtf8StringFromUnicode(sDirectory.c_str(), sDirectory.length());
u_setDataDirectory(g_overrideIcuDataPath.c_str());
}
}