blob: 8134f413206a34e0849981e1cf9fb8e130ddf68c [file] [log] [blame]
/****************************************************************************
**
** Copyright (C) 2011 Nokia Corporation and/or its subsidiary(-ies).
** All rights reserved.
** Contact: Nokia Corporation (qt-info@nokia.com)
**
** This file is part of the QtCore module of the Qt Toolkit.
**
** $QT_BEGIN_LICENSE:LGPL$
** GNU Lesser General Public License Usage
** This file may be used under the terms of the GNU Lesser General Public
** License version 2.1 as published by the Free Software Foundation and
** appearing in the file LICENSE.LGPL included in the packaging of this
** file. Please review the following information to ensure the GNU Lesser
** General Public License version 2.1 requirements will be met:
** http://www.gnu.org/licenses/old-licenses/lgpl-2.1.html.
**
** In addition, as a special exception, Nokia gives you certain additional
** rights. These rights are described in the Nokia Qt LGPL Exception
** version 1.1, included in the file LGPL_EXCEPTION.txt in this package.
**
** GNU General Public License Usage
** Alternatively, this file may be used under the terms of the GNU General
** Public License version 3.0 as published by the Free Software Foundation
** and appearing in the file LICENSE.GPL included in the packaging of this
** file. Please review the following information to ensure the GNU General
** Public License version 3.0 requirements will be met:
** http://www.gnu.org/copyleft/gpl.html.
**
** Other Usage
** Alternatively, this file may be used in accordance with the terms and
** conditions contained in a signed written agreement between you and Nokia.
**
**
**
**
**
** $QT_END_LICENSE$
**
****************************************************************************/
#include "qiconvcodec_p.h"
#include "qtextcodec_p.h"
#include <qlibrary.h>
#include <qdebug.h>
#include <qthreadstorage.h>
#include <errno.h>
#include <locale.h>
#include <stdio.h>
#include <dlfcn.h>
// unistd.h is needed for the _XOPEN_UNIX macro
#include <unistd.h>
#if defined(_XOPEN_UNIX) && !defined(Q_OS_QNX) && !defined(Q_OS_OSF)
# include <langinfo.h>
#endif
#if defined(Q_OS_HPUX)
# define NO_BOM
# define UTF16 "ucs2"
#elif defined(Q_OS_AIX)
# define NO_BOM
# define UTF16 "UCS-2"
#elif defined(Q_OS_FREEBSD) || defined(Q_OS_MAC)
# define NO_BOM
# if Q_BYTE_ORDER == Q_BIG_ENDIAN
# define UTF16 "UTF-16BE"
# else
# define UTF16 "UTF-16LE"
# endif
#else
# define UTF16 "UTF-16"
#endif
#if defined(Q_OS_MAC)
#ifndef GNU_LIBICONV
#define GNU_LIBICONV
#endif
typedef iconv_t (*Ptr_iconv_open) (const char*, const char*);
typedef size_t (*Ptr_iconv) (iconv_t, const char **, size_t *, char **, size_t *);
typedef int (*Ptr_iconv_close) (iconv_t);
static Ptr_iconv_open ptr_iconv_open = 0;
static Ptr_iconv ptr_iconv = 0;
static Ptr_iconv_close ptr_iconv_close = 0;
#endif
QT_BEGIN_NAMESPACE
extern bool qt_locale_initialized;
QIconvCodec::QIconvCodec()
: utf16Codec(0)
{
utf16Codec = QTextCodec::codecForMib(1015);
Q_ASSERT_X(utf16Codec != 0,
"QIconvCodec::convertToUnicode",
"internal error, UTF-16 codec not found");
if (!utf16Codec) {
fprintf(stderr, "QIconvCodec::convertToUnicode: internal error, UTF-16 codec not found\n");
utf16Codec = reinterpret_cast<QTextCodec *>(~0);
}
#if defined(Q_OS_MAC)
if (ptr_iconv_open == 0) {
QLibrary libiconv(QLatin1String("/usr/lib/libiconv"));
libiconv.setLoadHints(QLibrary::ExportExternalSymbolsHint);
ptr_iconv_open = reinterpret_cast<Ptr_iconv_open>(libiconv.resolve("libiconv_open"));
if (!ptr_iconv_open)
ptr_iconv_open = reinterpret_cast<Ptr_iconv_open>(libiconv.resolve("iconv_open"));
ptr_iconv = reinterpret_cast<Ptr_iconv>(libiconv.resolve("libiconv"));
if (!ptr_iconv)
ptr_iconv = reinterpret_cast<Ptr_iconv>(libiconv.resolve("iconv"));
ptr_iconv_close = reinterpret_cast<Ptr_iconv_close>(libiconv.resolve("libiconv_close"));
if (!ptr_iconv_close)
ptr_iconv_close = reinterpret_cast<Ptr_iconv_close>(libiconv.resolve("iconv_close"));
Q_ASSERT_X(ptr_iconv_open && ptr_iconv && ptr_iconv_close,
"QIconvCodec::QIconvCodec()",
"internal error, could not resolve the iconv functions");
# undef iconv_open
# define iconv_open ptr_iconv_open
# undef iconv
# define iconv ptr_iconv
# undef iconv_close
# define iconv_close ptr_iconv_close
}
#endif
}
QIconvCodec::~QIconvCodec()
{
}
QIconvCodec::IconvState::IconvState(iconv_t x)
: buffer(array), bufferLen(sizeof array), cd(x)
{
}
QIconvCodec::IconvState::~IconvState()
{
if (cd != reinterpret_cast<iconv_t>(-1))
iconv_close(cd);
if (buffer != array)
delete[] buffer;
}
void QIconvCodec::IconvState::saveChars(const char *c, int count)
{
if (count > bufferLen) {
if (buffer != array)
delete[] buffer;
buffer = new char[bufferLen = count];
}
memcpy(buffer, c, count);
}
static void qIconvCodecStateFree(QTextCodec::ConverterState *state)
{
delete reinterpret_cast<QIconvCodec::IconvState *>(state->d);
}
Q_GLOBAL_STATIC(QThreadStorage<QIconvCodec::IconvState *>, toUnicodeState)
QString QIconvCodec::convertToUnicode(const char* chars, int len, ConverterState *convState) const
{
if (utf16Codec == reinterpret_cast<QTextCodec *>(~0))
return QString::fromLatin1(chars, len);
int invalidCount = 0;
int remainingCount = 0;
char *remainingBuffer = 0;
IconvState **pstate;
if (convState) {
// stateful conversion
pstate = reinterpret_cast<IconvState **>(&convState->d);
if (convState->d) {
// restore state
remainingCount = convState->remainingChars;
remainingBuffer = (*pstate)->buffer;
} else {
// first time
convState->flags |= FreeFunction;
QTextCodecUnalignedPointer::encode(convState->state_data, qIconvCodecStateFree);
}
} else {
QThreadStorage<QIconvCodec::IconvState *> *ts = toUnicodeState();
if (!qt_locale_initialized || !ts) {
// we're running after the Q_GLOBAL_STATIC has been deleted
// or before the QCoreApplication initialization
// bad programmer, no cookie for you
return QString::fromLatin1(chars, len);
}
// stateless conversion -- use thread-local data
pstate = &toUnicodeState()->localData();
}
if (!*pstate) {
// first time, create the state
iconv_t cd = QIconvCodec::createIconv_t(UTF16, 0);
if (cd == reinterpret_cast<iconv_t>(-1)) {
static int reported = 0;
if (!reported++) {
fprintf(stderr,
"QIconvCodec::convertToUnicode: using Latin-1 for conversion, iconv_open failed\n");
}
return QString::fromLatin1(chars, len);
}
*pstate = new IconvState(cd);
}
IconvState *state = *pstate;
size_t inBytesLeft = len;
// best case assumption, each byte is converted into one UTF-16 character, plus 2 bytes for the BOM
#ifdef GNU_LIBICONV
// GNU doesn't disagree with POSIX :/
const char *inBytes = chars;
#else
char *inBytes = const_cast<char *>(chars);
#endif
QByteArray in;
if (remainingCount) {
// we have to prepend the remaining bytes from the previous conversion
inBytesLeft += remainingCount;
in.resize(inBytesLeft);
inBytes = in.data();
memcpy(in.data(), remainingBuffer, remainingCount);
memcpy(in.data() + remainingCount, chars, len);
remainingCount = 0;
}
size_t outBytesLeft = len * 2 + 2;
QByteArray ba(outBytesLeft, Qt::Uninitialized);
char *outBytes = ba.data();
do {
size_t ret = iconv(state->cd, &inBytes, &inBytesLeft, &outBytes, &outBytesLeft);
if (ret == (size_t) -1) {
if (errno == E2BIG) {
int offset = ba.size() - outBytesLeft;
ba.resize(ba.size() * 2);
outBytes = ba.data() + offset;
outBytesLeft = ba.size() - offset;
continue;
}
if (errno == EILSEQ) {
// conversion stopped because of an invalid character in the sequence
++invalidCount;
} else if (errno == EINVAL && convState) {
// conversion stopped because the remaining inBytesLeft make up
// an incomplete multi-byte sequence; save them for later
state->saveChars(inBytes, inBytesLeft);
remainingCount = inBytesLeft;
break;
}
if (errno == EILSEQ || errno == EINVAL) {
// skip the next character
++inBytes;
--inBytesLeft;
continue;
}
// some other error
// note, cannot use qWarning() since we are implementing the codecForLocale :)
perror("QIconvCodec::convertToUnicode: using Latin-1 for conversion, iconv failed");
if (!convState) {
// reset state
iconv(state->cd, 0, &inBytesLeft, 0, &outBytesLeft);
}
return QString::fromLatin1(chars, len);
}
} while (inBytesLeft != 0);
QString s = utf16Codec->toUnicode(ba.constData(), ba.size() - outBytesLeft);
if (convState) {
convState->invalidChars = invalidCount;
convState->remainingChars = remainingCount;
} else {
// reset state
iconv(state->cd, 0, &inBytesLeft, 0, &outBytesLeft);
}
return s;
}
Q_GLOBAL_STATIC(QThreadStorage<QIconvCodec::IconvState *>, fromUnicodeState)
static bool setByteOrder(iconv_t cd)
{
#if !defined(NO_BOM)
// give iconv() a BOM
char buf[4];
ushort bom[] = { QChar::ByteOrderMark };
char *outBytes = buf;
char *inBytes = reinterpret_cast<char *>(bom);
size_t outBytesLeft = sizeof buf;
size_t inBytesLeft = sizeof bom;
#if defined(GNU_LIBICONV)
const char **inBytesPtr = const_cast<const char **>(&inBytes);
#else
char **inBytesPtr = &inBytes;
#endif
if (iconv(cd, inBytesPtr, &inBytesLeft, &outBytes, &outBytesLeft) == (size_t) -1) {
return false;
}
#endif // NO_BOM
return true;
}
QByteArray QIconvCodec::convertFromUnicode(const QChar *uc, int len, ConverterState *convState) const
{
char *inBytes;
char *outBytes;
size_t inBytesLeft;
#if defined(GNU_LIBICONV)
const char **inBytesPtr = const_cast<const char **>(&inBytes);
#else
char **inBytesPtr = &inBytes;
#endif
QThreadStorage<QIconvCodec::IconvState *> *ts = fromUnicodeState();
if (!qt_locale_initialized || !ts) {
// we're running after the Q_GLOBAL_STATIC has been deleted
// or before the QCoreApplication initialization
// bad programmer, no cookie for you
if (!len)
// this is a special case - zero-sized string should be
// translated to empty but not-null QByteArray.
return QByteArray("");
return QString::fromRawData(uc, len).toLatin1();
}
IconvState *&state = ts->localData();
if (!state) {
state = new IconvState(QIconvCodec::createIconv_t(0, UTF16));
if (state->cd == reinterpret_cast<iconv_t>(-1)) {
if (!setByteOrder(state->cd)) {
perror("QIconvCodec::convertFromUnicode: using Latin-1 for conversion, iconv failed for BOM");
iconv_close(state->cd);
state->cd = reinterpret_cast<iconv_t>(-1);
return QString(uc, len).toLatin1();
}
}
}
if (state->cd == reinterpret_cast<iconv_t>(-1)) {
static int reported = 0;
if (!reported++) {
fprintf(stderr,
"QIconvCodec::convertFromUnicode: using Latin-1 for conversion, iconv_open failed\n");
}
return QString(uc, len).toLatin1();
}
size_t outBytesLeft = len;
QByteArray ba(outBytesLeft, Qt::Uninitialized);
outBytes = ba.data();
// now feed iconv() the real data
inBytes = const_cast<char *>(reinterpret_cast<const char *>(uc));
inBytesLeft = len * sizeof(QChar);
QByteArray in;
if (convState && convState->remainingChars) {
// we have one surrogate char to be prepended
in.resize(sizeof(QChar) + len);
inBytes = in.data();
QChar remaining = convState->state_data[0];
memcpy(in.data(), &remaining, sizeof(QChar));
memcpy(in.data() + sizeof(QChar), uc, inBytesLeft);
inBytesLeft += sizeof(QChar);
convState->remainingChars = 0;
}
int invalidCount = 0;
while (inBytesLeft != 0) {
if (iconv(state->cd, inBytesPtr, &inBytesLeft, &outBytes, &outBytesLeft) == (size_t) -1) {
if (errno == EINVAL && convState) {
// buffer ends in a surrogate
Q_ASSERT(inBytesLeft == 2);
convState->remainingChars = 1;
convState->state_data[0] = uc[len - 1].unicode();
break;
}
switch (errno) {
case EILSEQ:
++invalidCount;
// fall through
case EINVAL:
{
inBytes += sizeof(QChar);
inBytesLeft -= sizeof(QChar);
break;
}
case E2BIG:
{
int offset = ba.size() - outBytesLeft;
ba.resize(ba.size() * 2);
outBytes = ba.data() + offset;
outBytesLeft = ba.size() - offset;
break;
}
default:
{
// note, cannot use qWarning() since we are implementing the codecForLocale :)
perror("QIconvCodec::convertFromUnicode: using Latin-1 for conversion, iconv failed");
// reset to initial state
iconv(state->cd, 0, &inBytesLeft, 0, &outBytesLeft);
return QString(uc, len).toLatin1();
}
}
}
}
// reset to initial state
iconv(state->cd, 0, &inBytesLeft, 0, &outBytesLeft);
setByteOrder(state->cd);
ba.resize(ba.size() - outBytesLeft);
if (convState)
convState->invalidChars = invalidCount;
return ba;
}
QByteArray QIconvCodec::name() const
{
return "System";
}
int QIconvCodec::mibEnum() const
{
return 0;
}
iconv_t QIconvCodec::createIconv_t(const char *to, const char *from)
{
Q_ASSERT((to == 0 && from != 0) || (to != 0 && from == 0));
iconv_t cd = (iconv_t) -1;
#if defined(__GLIBC__) || defined(GNU_LIBICONV)
// both GLIBC and libgnuiconv will use the locale's encoding if from or to is an empty string
static const char empty_codeset[] = "";
const char *codeset = empty_codeset;
cd = iconv_open(to ? to : codeset, from ? from : codeset);
#else
char *codeset = 0;
#endif
#if defined(_XOPEN_UNIX) && !defined(Q_OS_QNX) && !defined(Q_OS_OSF)
if (cd == (iconv_t) -1) {
codeset = nl_langinfo(CODESET);
if (codeset)
cd = iconv_open(to ? to : codeset, from ? from : codeset);
}
#endif
if (cd == (iconv_t) -1) {
// Very poorly defined and followed standards causes lots of
// code to try to get all the cases... This logic is
// duplicated in QTextCodec, so if you change it here, change
// it there too.
// Try to determine locale codeset from locale name assigned to
// LC_CTYPE category.
// First part is getting that locale name. First try setlocale() which
// definitely knows it, but since we cannot fully trust it, get ready
// to fall back to environment variables.
char * ctype = qstrdup(setlocale(LC_CTYPE, 0));
// Get the first nonempty value from $LC_ALL, $LC_CTYPE, and $LANG
// environment variables.
char * lang = qstrdup(qgetenv("LC_ALL").constData());
if (!lang || lang[0] == 0 || strcmp(lang, "C") == 0) {
if (lang) delete [] lang;
lang = qstrdup(qgetenv("LC_CTYPE").constData());
}
if (!lang || lang[0] == 0 || strcmp(lang, "C") == 0) {
if (lang) delete [] lang;
lang = qstrdup(qgetenv("LANG").constData());
}
// Now try these in order:
// 1. CODESET from ctype if it contains a .CODESET part (e.g. en_US.ISO8859-15)
// 2. CODESET from lang if it contains a .CODESET part
// 3. ctype (maybe the locale is named "ISO-8859-1" or something)
// 4. locale (ditto)
// 5. check for "@euro"
// 1. CODESET from ctype if it contains a .CODESET part (e.g. en_US.ISO8859-15)
codeset = ctype ? strchr(ctype, '.') : 0;
if (codeset && *codeset == '.') {
++codeset;
cd = iconv_open(to ? to : codeset, from ? from : codeset);
}
// 2. CODESET from lang if it contains a .CODESET part
codeset = lang ? strchr(lang, '.') : 0;
if (cd == (iconv_t) -1 && codeset && *codeset == '.') {
++codeset;
cd = iconv_open(to ? to : codeset, from ? from : codeset);
}
// 3. ctype (maybe the locale is named "ISO-8859-1" or something)
if (cd == (iconv_t) -1 && ctype && *ctype != 0 && strcmp (ctype, "C") != 0)
cd = iconv_open(to ? to : ctype, from ? from : ctype);
// 4. locale (ditto)
if (cd == (iconv_t) -1 && lang && *lang != 0)
cd = iconv_open(to ? to : lang, from ? from : lang);
// 5. "@euro"
if ((cd == (iconv_t) -1 && ctype && strstr(ctype, "@euro")) || (lang && strstr(lang, "@euro")))
cd = iconv_open(to ? to : "ISO8859-15", from ? from : "ISO8859-15");
delete [] ctype;
delete [] lang;
}
return cd;
}
QT_END_NAMESPACE