--- parser3/src/main/pa_charset.C 2002/07/30 14:09:23 1.25 +++ parser3/src/main/pa_charset.C 2009/10/04 22:02:57 1.81 @@ -1,24 +1,39 @@ /** @file Parser: Charset connection implementation. - Copyright(c) 2001, 2002 ArtLebedev Group (http://www.artlebedev.com) + Copyright(c) 2001-2009 ArtLebedev Group (http://www.artlebedev.com) Author: Alexander Petrosyan(http://paf.design.ru) - - $Id: pa_charset.C,v 1.25 2002/07/30 14:09:23 paf Exp $ */ +static const char * const IDENT_CHARSET_C="$Date: 2009/10/04 22:02:57 $"; + #include "pa_charset.h" +#include "pa_charsets.h" #ifdef XML #include "libxml/encoding.h" #endif -// globals +//#define PA_PATCHED_LIBXML_BACKWARD + +// reduce memory usage by pre-calculation utf-8 string length +#define PRECALCULATE_DEST_LENGTH +// globals -// consts +Charset::UTF8CaseTable::Rec UTF8CaseToUpperRecords[]={ +#include "utf8-to-upper.inc" +}; +Charset::UTF8CaseTable UTF8CaseToUpper={ + sizeof(UTF8CaseToUpperRecords)/sizeof(Charset::UTF8CaseTable::Rec), + UTF8CaseToUpperRecords}; -#define MAX_CHARSET_UNI_CODES 500 +Charset::UTF8CaseTable::Rec UTF8CaseToLowerRecords[]={ +#include "utf8-to-lower.inc" +}; +Charset::UTF8CaseTable UTF8CaseToLower={ + sizeof(UTF8CaseToLowerRecords)/sizeof(Charset::UTF8CaseTable::Rec), + UTF8CaseToLowerRecords}; // helpers @@ -26,7 +41,7 @@ inline void prepare_case_tables(unsigned unsigned char *lcc_table=tables+lcc_offset; unsigned char *fcc_table=tables+fcc_offset; for(int i=0; i<0x100; i++) - lcc_table[i]=fcc_table[i]=i; + lcc_table[i]=fcc_table[i]=(unsigned char)i; } inline void cstr2ctypes(unsigned char *tables, const unsigned char *cstr, unsigned char bit) { @@ -37,7 +52,7 @@ inline void cstr2ctypes(unsigned char *t ctypes_table[c]|=bit; } } -inline unsigned int to_wchar_code(const char *cstr) { +inline unsigned int to_wchar_code(const char* cstr) { if(!cstr || !*cstr) return 0; if(cstr[1]==0) @@ -46,7 +61,7 @@ inline unsigned int to_wchar_code(const char *error_pos; return(unsigned int)strtol(cstr, &error_pos, 0); } -inline bool to_bool(const char *cstr) { +inline bool to_bool(const char* cstr) { return cstr && *cstr!=0; } static void element2ctypes(unsigned char c, bool belongs, @@ -73,38 +88,28 @@ static void element2case(unsigned char f // methods -extern "C" unsigned char pcre_default_tables[]; // pcre/chartables.c -Charset::Charset(Pool& apool, const String& aname, const String *request_file_spec): Pooled(apool), - fname(aname) { - - char *name_cstr=fname.cstr(); - for(char *c=name_cstr; *c; c++) - *c = toupper(*c); +Charset::Charset(Request_charsets* charsets, const String::Body ANAME, const String* afile_spec): + FNAME(ANAME), + FNAME_CSTR(ANAME.cstrm()) { - if(request_file_spec) { + if(afile_spec) { fisUTF8=false; - loadDefinition(*request_file_spec); + load_definition(*charsets, *afile_spec); #ifdef XML - addEncoding(name_cstr); + addEncoding(FNAME_CSTR); #endif } else { fisUTF8=true; // grab default onces [for UTF-8 so to be able to make a-z =>A-Z - memcpy(pcre_tables, pcre_default_tables, sizeof(pcre_tables)); + memcpy(pcre_tables, _pcre_default_tables, sizeof(pcre_tables)); } #ifdef XML - initTranscoder(&aname, name_cstr); + initTranscoder(FNAME, FNAME_CSTR); #endif } -Charset::~Charset() { -#ifdef XML - // not deleting transcoder, that's not our business -#endif -} - -void Charset::loadDefinition(const String& request_file_spec) { +void Charset::load_definition(Request_charsets& charsets, const String& afile_spec) { // pcre_tables // lowcase, flipcase, bits digit+word+whitespace, masks @@ -115,46 +120,40 @@ void Charset::loadDefinition(const Strin cstr2ctypes(pcre_tables,(const unsigned char *)"*+?{^.$|()[", ctype_meta); // charset - memset(tables.fromTable, 0, sizeof(tables.fromTable)); - tables.toTable=(Charset_TransRec *)calloc(sizeof(Charset_TransRec)*MAX_CHARSET_UNI_CODES); - tables.toTableSize=0; - // strangly vital - tables.toTable[tables.toTableSize].intCh=0; - tables.toTable[tables.toTableSize].extCh=(XMLByte)0; - tables.toTableSize++; + memset(&tables, 0, sizeof(tables)); // loading text - char *data=file_read_text(pool(), request_file_spec); + char *data=file_read_text(charsets, afile_spec); // ignore header getrow(&data); // parse cells char *row; - while(row=getrow(&data)) { + while((row=getrow(&data))) { // remove empty&comment lines if(!*row || *row=='#') continue; // char white-space digit hex-digit letter word lowercase unicode1 unicode2 - unsigned int c=0; + unsigned char c=0; char *cell; - for(int column=0; cell=lsplit(&row, '\t'); column++) { + for(int column=0; (cell=lsplit(&row, '\t')); column++) { switch(column) { - case 0: c=to_wchar_code(cell); break; + case 0: c=(unsigned char)to_wchar_code(cell); break; // pcre_tables case 1: element2ctypes(c, to_bool(cell), pcre_tables, ctype_space, cbit_space); break; case 2: element2ctypes(c, to_bool(cell), pcre_tables, ctype_digit, cbit_digit); break; case 3: element2ctypes(c, to_bool(cell), pcre_tables, ctype_xdigit); break; case 4: element2ctypes(c, to_bool(cell), pcre_tables, ctype_letter); break; case 5: element2ctypes(c, to_bool(cell), pcre_tables, ctype_word, cbit_word); break; - case 6: element2case(c, to_wchar_code(cell), pcre_tables); break; + case 6: element2case(c, (unsigned char)to_wchar_code(cell), pcre_tables); break; case 7: case 8: // charset if(tables.toTableSize>MAX_CHARSET_UNI_CODES) - throw Exception("parser.runtime", - &request_file_spec, + throw Exception(PARSER_RUNTIME, + &afile_spec, "charset must contain not more then %d unicode values", MAX_CHARSET_UNI_CODES); XMLCh unicode=(XMLCh)to_wchar_code(cell); @@ -178,8 +177,8 @@ void Charset::loadDefinition(const Strin static int sort_cmp_Trans_rec_intCh(const void *a, const void *b) { return - static_cast(a)->intCh- - static_cast(b)->intCh; + static_cast(a)->intCh- + static_cast(b)->intCh; } void Charset::sort_ToTable() { @@ -190,60 +189,43 @@ void Charset::sort_ToTable() { //fclose(f); } +// @todo: precache for spedup searching static XMLByte xlatOneTo(const XMLCh toXlat, - const Charset::Tables& tables, - XMLByte not_found) { - unsigned int lowOfs = 0; - unsigned int hiOfs = tables.toTableSize - 1; - XMLByte curByte = 0; - do { - // Calc the mid point of the low and high offset. - const unsigned int midOfs =((hiOfs - lowOfs) / 2)+lowOfs; - - // If our test char is greater than the mid point char, then - // we move up to the upper half. Else we move to the lower - // half. If its equal, then its our guy. - if(toXlat>tables.toTable[midOfs].intCh) - lowOfs = midOfs; - else if(toXlatcur) + lo = i+1; else - return tables.toTable[midOfs].extCh; - } while(lowOfs+1 outEnd) - break; - - // We can do it, so update the source index - srcPtr++; - - // And spit out the bytes. We spit them out in reverse order - // here, so bump up the output pointer and work down as we go. - outPtr+= encodedBytes; - switch(encodedBytes) { - case 6: *--outPtr = XMLByte((curVal | 0x80UL) & 0xBFUL); - curVal>>= 6; - case 5: *--outPtr = XMLByte((curVal | 0x80UL) & 0xBFUL); - curVal>>= 6; - case 4: *--outPtr = XMLByte((curVal | 0x80UL) & 0xBFUL); - curVal>>= 6; - case 3: *--outPtr = XMLByte((curVal | 0x80UL) & 0xBFUL); - curVal>>= 6; - case 2: *--outPtr = XMLByte((curVal | 0x80UL) & 0xBFUL); - curVal>>= 6; - case 1: *--outPtr = XMLByte(curVal | gFirstByteMark[encodedBytes]); - } - - // Add the encoded bytes back in again to indicate we've eaten them - outPtr+= encodedBytes; - } - - // Update the bytes eaten - srcLen = srcPtr - srcData; - - // Return the characters read - toFillLen = outPtr - toFill; - - return srcPtr==srcEnd?(int)toFillLen:-1; -} -static size_t transcodeFromUTF8( - const XMLByte *srcData, size_t& srcLen, - XMLByte* toFill, size_t& toFillLen, - const Charset::Tables& tables) { + // use the replacement character + *outPtr++= '?'; + srcPtr++; + continue; + } + + // Figure out how many bytes we need + unsigned int encodedBytes; + if(curVal<0x80) + encodedBytes = 1; + else if(curVal<0x800) + encodedBytes = 2; + else if(curVal<0x10000) + encodedBytes = 3; + else if(curVal<0x200000) + encodedBytes = 4; + else if(curVal<0x4000000) + encodedBytes = 5; + else if(curVal<= 0x7FFFFFFF) + encodedBytes = 6; + else { + // use the replacement character + *outPtr++= '?'; + srcPtr++; + continue; + } + + // If we cannot fully get this char into the output buffer + if (outPtr + encodedBytes > outEnd) + break; + + // We can do it, so update the source index + srcPtr++; + + // And spit out the bytes. We spit them out in reverse order + // here, so bump up the output pointer and work down as we go. + outPtr+= encodedBytes; + switch(encodedBytes) { + case 6: *--outPtr = XMLByte((curVal | 0x80UL) & 0xBFUL); + curVal>>= 6; + case 5: *--outPtr = XMLByte((curVal | 0x80UL) & 0xBFUL); + curVal>>= 6; + case 4: *--outPtr = XMLByte((curVal | 0x80UL) & 0xBFUL); + curVal>>= 6; + case 3: *--outPtr = XMLByte((curVal | 0x80UL) & 0xBFUL); + curVal>>= 6; + case 2: *--outPtr = XMLByte((curVal | 0x80UL) & 0xBFUL); + curVal>>= 6; + case 1: *--outPtr = XMLByte(curVal | gFirstByteMark[encodedBytes]); + } + + // Add the encoded bytes back in again to indicate we've eaten them + outPtr+= encodedBytes; + } + + // Update the bytes eaten + srcLen = srcPtr - srcData; + + // Return the characters read + toFillLen = outPtr - toFill; + + //return srcPtr==srcEnd?(int)toFillLen:-1; +/* +xmlCharEncodingInputFunc +Returns : +the number of byte written, or -1 by lack of space, or -2 if the transcoding failed. The value of inlen after return is the +number of octets consumed as the return value is positive, else unpredictiable. The value of outlen after return is the number +of ocetes consumed. +*/ + return 0; +} +/// @todo digital entites only when xml/html output [at output in html/xml mode, in html part of a letter] +static int transcodeFromUTF8(const XMLByte* srcData, int& srcLen, + XMLByte* toFill, int& toFillLen, + const Charset::Tables& tables) { const XMLByte* srcPtr=srcData; const XMLByte* srcEnd=srcData+srcLen; XMLByte* outPtr=toFill; XMLByte* outEnd=toFill+toFillLen; - // We now loop until we either run out of input data, or room to store - while ((srcPtr < srcEnd) && (outPtr < outEnd)) { - // Get the next leading byte out - const XMLByte firstByte = *srcPtr; - - // Special-case ASCII, which is a leading byte value of<= 127 - if(firstByte<= 127) { - *outPtr++= firstByte; - srcPtr++; - continue; - } - - // See how many trailing src bytes this sequence is going to require - const unsigned int trailingBytes = gUTFBytes[firstByte]; - - // If there are not enough source bytes to do this one, then we - // are done. Note that we done>= here because we are implicitly - // counting the 1 byte we get no matter what. - if(srcPtr+trailingBytes>= srcEnd) - break; - - // Looks ok, so lets build up the value - uint tmpVal=0; - switch(trailingBytes) { - case 5: tmpVal+=*srcPtr++; tmpVal<<=6; - case 4: tmpVal+=*srcPtr++; tmpVal<<=6; - case 3: tmpVal+=*srcPtr++; tmpVal<<=6; - case 2: tmpVal+=*srcPtr++; tmpVal<<=6; - case 1: tmpVal+=*srcPtr++; tmpVal<<=6; - case 0: tmpVal+=*srcPtr++; - break; - - default: - throw Exception(0, - 0, - "transcodeFromUTF8 error: wrong trailingBytes value(%d)", trailingBytes); - } - tmpVal-=gUTFOffsets[trailingBytes]; - - // If it will fit into a single char, then put it in. Otherwise - // fail [*encode it as a surrogate pair. If its not valid, use the - // replacement char.*] - if(!(tmpVal & 0xFFFF0000)) { + // We now loop until we either run out of input data, or room to store + while ((srcPtr < srcEnd) && (outPtr < outEnd)) { + // Get the next leading byte out + const XMLByte firstByte =* srcPtr; + + // Special-case ASCII, which is a leading byte value of<= 127 + if(firstByte<=127) { + *outPtr++= firstByte; + srcPtr++; + continue; + } + + // See how many trailing src bytes this sequence is going to require + const unsigned int trailingBytes = gUTFBytes[firstByte]; + + // If there are not enough source bytes to do this one, then we + // are done. Note that we done>= here because we are implicitly + // counting the 1 byte we get no matter what. + if(srcPtr+trailingBytes>= srcEnd) + break; + + // Looks ok, so lets build up the value + uint tmpVal=0; + switch(trailingBytes) { + case 5: tmpVal+=*srcPtr++; tmpVal<<=6; + case 4: tmpVal+=*srcPtr++; tmpVal<<=6; + case 3: tmpVal+=*srcPtr++; tmpVal<<=6; + case 2: tmpVal+=*srcPtr++; tmpVal<<=6; + case 1: tmpVal+=*srcPtr++; tmpVal<<=6; + case 0: tmpVal+=*srcPtr++; + break; + + default: + throw Exception(0, + 0, + "transcodeFromUTF8 error: wrong trailingBytes value(%d)", trailingBytes); // never + } + tmpVal-=gUTFOffsets[trailingBytes]; + + // If it will fit into a single char, then put it in. Otherwise + // fail [*encode it as a surrogate pair. If its not valid, use the + // replacement char.*] + if(!(tmpVal & 0xFFFF0000)) { if(XMLByte xlat=xlatOneTo(tmpVal, tables, 0)) *outPtr++=xlat; - else - outPtr+=sprintf((char *)outPtr, "&#%d;", tmpVal); // &#decimal; + else { + outPtr+=sprintf((char *)outPtr, "&#%u;", tmpVal); // &#decimal; + } + } else { + const XMLByte* recoverPtr=srcPtr-trailingBytes-1; + for(uint i=0; i<=trailingBytes; i++) + outPtr+=sprintf((char*)outPtr, "%%%02X", *recoverPtr++); + } + } + + // Update the bytes eaten + srcLen = srcPtr - srcData; + + // Return the characters read + toFillLen = outPtr - toFill; + + //return srcPtr==srcEnd?(int)toFillLen:-1; +/* +xmlCharEncodingOutputFunc +Returns : +the number of byte written, or -1 by lack of space, or -2 if the transcoding failed. The value of inlen after return is the +number of octets consumed as the return value is positive, else unpredictiable. The value of outlen after return is the number +of ocetes consumed. +*/ + return 0; +} + +static bool isEscaped(XMLByte c){ + return + !( + (c<=127) + && ( + ((c>='0') && (c<='9')) + || ((c>='A') && (c<='Z')) + || ((c>='a') && (c<='z')) + || strchr("*@-_+./", c)!=0 + ) + ); +} + +// read one UTF8 char and return length of this char (in bytes) +static unsigned int readUTF8Char(const XMLByte*& srcPtr, const XMLByte* srcEnd, XMLByte& firstByte, XMLCh& UTF8Char){ + if(!srcPtr || !*srcPtr || srcPtr>=srcEnd) + return 0; + + firstByte=*srcPtr; + + if(firstByte<=127){ + UTF8Char=firstByte; + srcPtr++; + return 1; + } + + unsigned int trailingBytes=gUTFBytes[firstByte]; + + if(srcPtr+trailingBytes>=srcEnd){ + return 0; // not enough bytes in source string for reading + } + + uint tmpVal=0; + switch(trailingBytes){ + case 5: tmpVal+=*srcPtr++; tmpVal<<=6; + case 4: tmpVal+=*srcPtr++; tmpVal<<=6; + case 3: tmpVal+=*srcPtr++; tmpVal<<=6; + case 2: tmpVal+=*srcPtr++; tmpVal<<=6; + case 1: tmpVal+=*srcPtr++; tmpVal<<=6; + case 0: tmpVal+=*srcPtr++; + } + + tmpVal-=gUTFOffsets[trailingBytes]; + UTF8Char=tmpVal; + + return trailingBytes+1; +} + +// skip UTF8 char and return length of this char (in bytes) +static unsigned int skipUTF8Char(const XMLByte*& srcPtr, const XMLByte* srcEnd){ + if(!srcPtr || !*srcPtr || srcPtr>=srcEnd) + return 0; + + unsigned int trailingBytes=gUTFBytes[*srcPtr]+1; + srcPtr+=trailingBytes; + + return trailingBytes; +} + +// read non-UTF8 char, and return number of bytes needed for store this char in UTF8 +static unsigned int readChar(const XMLByte*& srcPtr, const XMLByte* srcEnd, XMLByte& firstByte, XMLCh& UTF8Char, const Charset::Tables& tables){ + if(!srcPtr || !*srcPtr || srcPtr>=srcEnd) + return 0; + + firstByte=*srcPtr++; + UTF8Char=tables.fromTable[firstByte]; + + if(UTF8Char<0x80) + return 1; + else if(UTF8Char<0x800) + return 2; + else if(UTF8Char<0x10000) + return 3; + else if(UTF8Char<0x200000) + return 4; + else if(UTF8Char<0x4000000) + return 5; + else if(UTF8Char<= 0x7FFFFFFF) + return 6; + + // will use the replacement character '?' + firstByte=0; + return 1; +} + +static int escapeUTF8(const XMLByte* srcData, size_t& srcLen, + XMLByte* toFill, size_t& toFillLen) { + const XMLByte* srcPtr=srcData; + XMLByte* outPtr=toFill; + XMLByte* outEnd=toFill+toFillLen; + // loop until we either run out of input data, or room to store + for(UTF8_string_iterator i((XMLByte *)srcPtr, srcLen); (outPtr < outEnd) && i.has_next(); ){ + if(i.getCharSize()==1){ + if(isEscaped(i.getFirstByte())) // %XX + outPtr+=sprintf((char*)outPtr, "%%%02X", i.getFirstByte()); + else + *outPtr++=i.getFirstByte(); } else - throw Exception(0, - 0, - "transcodeFromUTF8 error: too big tmpVal(0x%08X)", tmpVal); + outPtr+=sprintf((char*)outPtr, "%%u%04X", i.next()); // %uXXXX + } + // Update the bytes eaten + srcLen=srcPtr-srcData; + + // Return the characters read + toFillLen=outPtr-toFill; + + return 0; +} + +static int escape(const XMLByte* srcData, size_t& srcLen, + XMLByte *toFill, size_t& toFillLen, + const Charset::Tables& tables) { + const XMLByte* srcPtr=srcData; + const XMLByte* srcEnd=srcData+srcLen; + XMLByte* outPtr=toFill; + XMLByte firstByte; + XMLCh UTF8Char; + uint charSize; + + while(charSize=readChar(srcPtr, srcEnd, firstByte, UTF8Char, tables)){ + if(charSize==1){ + if(firstByte){ + if(isEscaped(firstByte)) // %XX + outPtr+=sprintf((char*)outPtr, "%%%02X", firstByte); + else + *outPtr++=firstByte; + } else // add replacement char '?' + *outPtr++='?'; + } else + outPtr+=sprintf((char*)outPtr, "%%u%04X", UTF8Char); // %uXXXX + } + + // Update the bytes eaten + srcLen = srcPtr - srcData; + + // Return the characters read + toFillLen = outPtr - toFill; + + return 0; +} + + +String::C Charset::escape(const String::C src, const Charset& source_charset){ + size_t src_length=src.length; + if(!src_length) + return String::C("", 0); + +#ifdef PRECALCULATE_DEST_LENGTH + size_t dest_length=0; + + if(source_charset.isUTF8()){ + for(UTF8_string_iterator i((XMLByte *)src.str, src_length); i.has_next(); ){ + if(i.getCharSize()==1) + dest_length+=!isEscaped(i.getFirstByte())?1/*as-is*/:3/*%XX*/; + else + dest_length+=6/*%uXXXX*/; + } + } else { + const XMLByte* srcPtr=(XMLByte*)src.str; + const XMLByte* srcEnd=srcPtr+src_length; + XMLByte firstByte; + XMLCh UTF8Char; + while(uint charSize=readChar(srcPtr, srcEnd, firstByte, UTF8Char, source_charset.tables)){ + if(charSize==1) + dest_length+=(!firstByte/*replacement char '?'*/ || !isEscaped(firstByte))?1:3/*'%XX'*/; + else + dest_length+=6; // '%uXXXX' + } } +#else + size_t dest_length=src_length*6; // enough for %uXXXX but too memory-hungry +#endif + + //throw Exception(0,0,"%u",dest_length); + +#ifndef NDEBUG + size_t saved_dest_length=dest_length; +#endif + XMLByte *dest_body=new(PointerFreeGC) XMLByte[dest_length+1/*for terminator*/]; - // Update the bytes eaten - srcLen = srcPtr - srcData; + int status; + if(source_charset.isUTF8()){ + status=::escapeUTF8((XMLByte *)src.str, src_length, dest_body, dest_length); + } else { + status=::escape((XMLByte *)src.str, src_length, dest_body, dest_length, source_charset.tables); + } - // Return the characters read - toFillLen = outPtr - toFill; + if(status<0) + throw Exception(0, + 0, + "Charset::escape buffer overflow"); - return srcPtr==srcEnd?(int)toFillLen:-1; + assert(dest_length<=saved_dest_length); + dest_body[dest_length]=0; // terminator + return String::C((char*)dest_body, dest_length); } + +String::Body Charset::escape(const String::Body src, const Charset& source_charset) { + const char *src_ptr=src.cstr(); + size_t src_size=strlen(src_ptr); + + String::C dest=Charset::escape(String::C(src_ptr, src_size), source_charset); -/// @todo not so memory-hungry with prescan -void Charset::transcodeToUTF8(Pool& pool, - const void *source_body, size_t source_content_length, - const void *& adest_body, size_t& dest_content_length) const { - dest_content_length=source_content_length*6/*so that surly enough, max utf8 seq len=6*/; - XMLByte *dest_body=(XMLByte*)pool.malloc(dest_content_length); + return String::Body(dest.length ? dest.str:0); +} + +String& Charset::escape(const String& src, const Charset& source_charset) { + if(src.is_empty()) + return *new String(); + + return *new String(escape((String::Body)src, source_charset), String::L_CLEAN); +} + +const String::C Charset::transcodeToUTF8(const String::C src) const { + int src_length=src.length; + +#ifdef PRECALCULATE_DEST_LENGTH + int dest_length=0; + const XMLByte* srcPtr=(XMLByte*)src.str; + const XMLByte* srcEnd=srcPtr+src_length; + XMLByte firstByte; + XMLCh UTF8Char; + while(uint charSize=readChar(srcPtr, srcEnd, firstByte, UTF8Char, tables)) + dest_length+=charSize; +#else + int dest_length=src_length*6; // so that surly enough (max utf8 seq len=6) but too memory-hyngry +#endif + + //throw Exception(0,0,"%u",dest_length); + +#ifndef NDEBUG + int saved_dest_length=dest_length; +#endif + XMLByte *dest_body=new(PointerFreeGC) XMLByte[dest_length+1/*for terminator*/]; if(::transcodeToUTF8( - (XMLByte *)source_body, source_content_length, - dest_body, dest_content_length, + (XMLByte *)src.str, src_length, + dest_body, dest_length, tables)<0) - throw(0, 0, + throw Exception(0, 0, "Charset::transcodeToUTF8 buffer overflow"); - // return - adest_body=dest_body; + assert(dest_length<=saved_dest_length); + dest_body[dest_length]=0; // terminator + return String::C((char*)dest_body, dest_length); +} + +static XMLCh change_case_UTF8(const XMLCh src, const Charset::UTF8CaseTable& table) { + int lo = 0; + int hi = table.size - 1; + while(lo<=hi) { + // Calc the mid point of the low and high offset. + const unsigned int i = (lo + hi) / 2; + + XMLCh cur=table.records[i].from; + if(src==cur) + return table.records[i].to; + if(src>cur) + lo = i+1; + else + hi = i-1; + } + + // not found + return src; +} + +static void store_UTF8(XMLCh src, XMLByte*& outPtr){ + if(!src) { + // use the replacement character + *outPtr++= '?'; + return; + } + + // Figure out how many bytes we need + unsigned int encodedBytes; + if(src<0x80) + encodedBytes = 1; + else if(src<0x800) + encodedBytes = 2; + else if(src<0x10000) + encodedBytes = 3; + else if(src<0x200000) + encodedBytes = 4; + else if(src<0x4000000) + encodedBytes = 5; + else if(src<= 0x7FFFFFFF) + encodedBytes = 6; + else { + // use the replacement character + *outPtr++= '?'; + return; + } + + // And spit out the bytes. We spit them out in reverse order + // here, so bump up the output pointer and work down as we go. + outPtr+= encodedBytes; + switch(encodedBytes) { + case 6: *--outPtr = XMLByte((src | 0x80UL) & 0xBFUL); + src>>= 6; + case 5: *--outPtr = XMLByte((src | 0x80UL) & 0xBFUL); + src>>= 6; + case 4: *--outPtr = XMLByte((src | 0x80UL) & 0xBFUL); + src>>= 6; + case 3: *--outPtr = XMLByte((src | 0x80UL) & 0xBFUL); + src>>= 6; + case 2: *--outPtr = XMLByte((src | 0x80UL) & 0xBFUL); + src>>= 6; + case 1: *--outPtr = XMLByte(src | gFirstByteMark[encodedBytes]); + } + + // Add the encoded bytes back in again to indicate we've eaten them + outPtr+= encodedBytes; +} + +static void change_case_UTF8(XMLCh src, XMLByte*& outPtr, + const Charset::UTF8CaseTable& table) { + store_UTF8(change_case_UTF8(src, table), outPtr); +}; +void change_case_UTF8(const XMLByte* srcData, size_t srcLen, + XMLByte* toFill, size_t toFillLen, + const Charset::UTF8CaseTable& table) { + const XMLByte* srcPtr=srcData; + const XMLByte* srcEnd=srcData+srcLen; + XMLByte* outPtr=toFill; + XMLByte* outEnd=toFill+toFillLen; + + // We now loop until we either run out of input data, or room to store + while ((srcPtr < srcEnd) && (outPtr < outEnd)) { + // Get the next leading byte out + const XMLByte firstByte =* srcPtr; + + if(firstByte<=127) { + change_case_UTF8(firstByte, outPtr, table); + srcPtr++; + continue; + } + + // See how many trailing src bytes this sequence is going to require + const unsigned int trailingBytes = gUTFBytes[firstByte]; + + // Looks ok, so lets build up the value + uint tmpVal=0; + switch(trailingBytes) { + case 5: tmpVal+=*srcPtr++; tmpVal<<=6; + case 4: tmpVal+=*srcPtr++; tmpVal<<=6; + case 3: tmpVal+=*srcPtr++; tmpVal<<=6; + case 2: tmpVal+=*srcPtr++; tmpVal<<=6; + case 1: tmpVal+=*srcPtr++; tmpVal<<=6; + case 0: tmpVal+=*srcPtr++; + break; + + default: + throw Exception(0, + 0, + "change_case_UTF8 error: wrong trailingBytes value(%d)", trailingBytes); + } + tmpVal-=gUTFOffsets[trailingBytes]; + + // If it will fit into a single char, then put it in. Otherwise + // fail [*encode it as a surrogate pair. If its not valid, use the + // replacement char.*] + if(!(tmpVal & 0xFFFF0000)) + change_case_UTF8(tmpVal, outPtr, table); + else + throw Exception(0, + 0, + "change_case_UTF8 error: too big tmpVal(0x%08X)", tmpVal); + } + + if(srcPtr!=outPtr) + throw Exception(0, + 0, + "change_case_UTF8 error: end pointers do not match"); } -void Charset::transcodeFromUTF8(Pool& pool, - const void *source_body, size_t source_content_length, - const void *& adest_body, size_t& dest_content_length) const { - dest_content_length=source_content_length*6/*so that surly enough, "ÿ" has max ratio */; - XMLByte *dest_body=(XMLByte*)pool.malloc(dest_content_length); + +static size_t getDecNumLength(XMLCh UTF8Char){ + return + (UTF8Char < 100) + ?2 + :(UTF8Char < 1000) + ?3 + :(UTF8Char < 10000) + ?4 + :5; +} + +const String::C Charset::transcodeFromUTF8(const String::C src) const { +#ifdef PRECALCULATE_DEST_LENGTH + int dest_length=0; + for(UTF8_string_iterator i((XMLByte *)src.str, src.length); i.has_next(); ){ + if(i.getCharSize()==1) + dest_length++; + else + dest_length+=(i.next() & 0xFFFF0000) + ?i.getCharSize()*3 // '%XX' for each byte + :(xlatOneTo(i.next(), tables, 0)!=0) + ?1 // can convert it to single char + :getDecNumLength(i.next())+3; // &#XX; - &#XXXXX; + } +#else + // so that surly enough, "&#XXX;" has max ratio (huh? 8 bytes needed for '&#XXXXX;') + int dest_length=src.length*6; +#endif + +#ifndef NDEBUG + int saved_dest_length=dest_length; +#endif + XMLByte *dest_body=new(PointerFreeGC) XMLByte[dest_length+1/*for terminator*/]; if(::transcodeFromUTF8( - (XMLByte *)source_body, source_content_length, - dest_body, dest_content_length, + (XMLByte *)src.str, (int&)src.length, + dest_body, dest_length, tables)<0) - throw(0, 0, + throw Exception(0, 0, - "Charset::transcodeToUTF8 buffer overflow"); + "Charset::transcodeFromUTF8 buffer overflow"); - // return - adest_body=dest_body; + assert(dest_length<=saved_dest_length); + dest_body[dest_length]=0; // terminator + return String::C((char*)dest_body, dest_length); } /// transcode using both charsets -void Charset::transcodeToCharset(Pool& pool, - const Charset& dest_charset, - const void *source_body, size_t source_content_length, - const void *& adest_body, size_t& adest_content_length) const { - if(&dest_charset==this) { - adest_body=source_body; - adest_content_length=source_content_length; - } else { - size_t dest_content_length=source_content_length; - unsigned char *dest_body=(unsigned char *)pool.malloc(dest_content_length); - - const XMLByte* srcPtr=(XMLByte *)source_body; - const XMLByte* srcEnd=(XMLByte *)source_body+source_content_length; - - for(XMLByte* outPtr=dest_body; srcPtrname=name_cstr; - handler->input=xml256CharEncodingInputFunc; handler->inputInfo=&tables; - handler->output=xml256CharEncodingOutputFunc; handler->outputInfo=&tables; + if(handlers_count==MAX_CHARSETS) + throw Exception(0, + 0, + "already allocated %d handlers, no space for new encoding '%s'", + MAX_CHARSETS, name_cstr); + + xmlCharEncodingHandler* handler=new(UseGC) xmlCharEncodingHandler; + { + handler->name=name_cstr; + handler->input=inputFuncs[handlers_count]; + handler->output=outputFuncs[handlers_count]; + ::tables[handlers_count]=&tables; + handlers_count++; + } xmlRegisterCharEncodingHandler(handler); + } -void Charset::initTranscoder(const String *source, const char *name_cstr) { +void Charset::initTranscoder(const String::Body NAME, const char* name_cstr) { ftranscoder=xmlFindCharEncodingHandler(name_cstr); - transcoder(source); // check right way + transcoder(NAME); // check right way } -xmlCharEncodingHandler *Charset::transcoder(const String *source) { +xmlCharEncodingHandler& Charset::transcoder(const String::Body NAME) { if(!ftranscoder) - throw Exception("parser.runtime", - source, + throw Exception(PARSER_RUNTIME, + new String(NAME, String::L_TAINTED), "unsupported encoding"); - return ftranscoder; + return *ftranscoder; } -const char *Charset::transcode_cstr(xmlChar *s) { +String::C Charset::transcode_cstr(const xmlChar* s) { if(!s) - return ""; + return String::C("", 0); - int inlen=strlen((const char *)s); - int outlen=inlen+1; // max - char *out=(char *)malloc(outlen*sizeof(char)); - - int size; - if(xmlCharEncodingOutputFunc output=transcoder(0)->output) { - size=output( + int inlen=strlen((const char*)s); + int outlen=inlen*6/*strlen("ÿ")*/; // max +#ifndef NDEBUG + int saved_outlen=outlen; +#endif + char *out=new(PointerFreeGC) char[outlen+1]; + + int error; + if(xmlCharEncodingOutputFunc output=transcoder(FNAME).output) { + error=output( (unsigned char*)out, &outlen, - (const unsigned char*)s, &inlen, - transcoder(0)->outputInfo); - } else - memcpy(out, s, size=inlen); - if(size<0) + (const unsigned char*)s, &inlen +#ifdef PA_PATCHED_LIBXML_BACKWARD + ,0 +#endif + ); + } else { + memcpy(out, s, outlen=inlen); + error=0; + } + if(error<0) throw Exception(0, 0, - "transcode_cstr failed (%d)", size); + "transcode_cstr failed (%d)", error); - out[size]=0; - return out; -} -String& Charset::transcode(xmlChar *s) { - return *NEW String(pool(), transcode_cstr(s)); + assert(outlen<=saved_outlen); out[outlen]=0; + return String::C(out, outlen); } -const char *Charset::transcode_cstr(GdomeDOMString *s) { - return s?transcode_cstr(BAD_CAST s->str):""; -} -String& Charset::transcode(GdomeDOMString *s) { - return *NEW String(pool(), transcode_cstr(s)); +const String& Charset::transcode(const xmlChar* s) { + String::C cstr=transcode_cstr(s); + return *new String(cstr.str, String::L_TAINTED); } /// @test less memory using -maybe- xmlParserInputBufferCreateMem -xmlChar *Charset::transcode_buf2xchar(const char *buf, size_t buf_size) { - int outlen=buf_size*6/*max*/+1; - unsigned char *out=(unsigned char*)malloc(outlen*sizeof(unsigned char)); - - int size; - if(xmlCharEncodingInputFunc input=transcoder(0)->input) { - size=input( +xmlChar* Charset::transcode_buf2xchar(const char* buf, size_t buf_size) { + xmlChar* out; + int outlen; + int error; +#ifndef NDEBUG + int saved_outlen; +#endif + if(xmlCharEncodingInputFunc input=transcoder(FNAME).input) { + outlen=buf_size*6/*max UTF8 bytes per char*/; +#ifndef NDEBUG + saved_outlen=outlen; +#endif + out=(xmlChar*)xmlMalloc(outlen+1); + error=input( out, &outlen, - (const unsigned char *)buf, (int *)&buf_size, - transcoder(0)->inputInfo); - } else - memcpy(out, buf, size=buf_size); + (const unsigned char*)buf, (int*)&buf_size +#ifdef PA_PATCHED_LIBXML_BACKWARD + ,0 +#endif + ); + } else { + outlen=buf_size; +#ifndef NDEBUG + saved_outlen=outlen; +#endif + out=(xmlChar*)xmlMalloc(outlen+1); + memcpy(out, buf, outlen); + error=0; + } - if(size<0) + if(error<0) throw Exception(0, 0, - "transcode_buf failed (%d)", size); + "transcode_buf failed (%d)", error); + + assert(outlen<=saved_outlen); out[outlen]=0; + return out; +} + +xmlChar* Charset::transcode(const String& s) { + String::Body sbody=s.cstr_to_string_body_untaint(String::L_AS_IS); + return transcode_buf2xchar(sbody.cstr(), sbody.length()); +} + +xmlChar* Charset::transcode(const String::Body s) { + return transcode_buf2xchar(s.cstr(), s.length()); +} +#endif + +String::Body Charset::transcode(const String::Body src, + const Charset& source_transcoder, + const Charset& dest_transcoder) { + + const char *src_ptr=src.cstr(); + size_t src_size=strlen(src_ptr); - out[size]=0; - return (xmlChar *)out; + String::C dest=Charset::transcode(String::C(src_ptr, src_size), source_transcoder, dest_transcoder); + + return String::Body(dest.length ? dest.str:0); } -GdomeDOMString_auto_ptr Charset::transcode_buf2dom(const char *buf, size_t buf_size) { - return GdomeDOMString_auto_ptr((gchar*)transcode_buf2xchar(buf, buf_size)); + +String& Charset::transcode(const String& src, + const Charset& source_transcoder, + const Charset& dest_transcoder) { + if(src.is_empty()) + return *new String(); + + return *new String(transcode((String::Body)src, source_transcoder, dest_transcoder), String::L_CLEAN); } -GdomeDOMString_auto_ptr Charset::transcode(const String& s) { - const char *cstr=s.cstr(String::UL_UNSPECIFIED); - return transcode_buf2dom(cstr, strlen(cstr)); +void Charset::transcode(ArrayString& src, + const Charset& source_transcoder, + const Charset& dest_transcoder) { + for(size_t i=0; isource_transcoder, + *info->dest_transcoder); +} + +void Charset::transcode(HashStringString& src, + const Charset& source_transcoder, + const Charset& dest_transcoder) { + Transcode_pair_info info={&source_transcoder, &dest_transcoder}; + src.for_each_ref(transcode_pair, &info); +} + +size_t getUTF8BytePos(const XMLByte* srcBegin, const XMLByte* srcEnd, size_t charPos){ + const XMLByte* ptr=srcBegin; + while(charPos-- && skipUTF8Char(ptr, srcEnd)); + + return ptr-srcBegin; +} + +size_t getUTF8CharPos(const XMLByte* srcBegin, const XMLByte* srcEnd, size_t bytePos){ + size_t charPos=0; + const XMLByte* ptr=srcBegin; + const XMLByte* ptrEnd=srcBegin+bytePos; + while(skipUTF8Char(ptr, srcEnd)){ + if(ptr>ptrEnd) + return charPos; + charPos++; + } + + // scan till end but position in bytes still too low + throw Exception(0, + 0, + "Error convertion byte pos to char pos"); +} + +size_t lengthUTF8(const XMLByte* srcBegin, const XMLByte* srcEnd){ + size_t size=0; + while(skipUTF8Char(srcBegin, srcEnd)) + size++; + + return size; +} + +bool UTF8_string_iterator::has_next(){ + fcharSize=readUTF8Char(fsrcPtr, fsrcEnd, ffirstByte, fUTF8Char); + return fcharSize!=0; +}