--- parser3/src/main/untaint.C 2001/04/20 09:05:24 1.42 +++ parser3/src/main/untaint.C 2001/04/23 09:46:17 1.45 @@ -5,7 +5,7 @@ Author: Alexander Petrosyan (http://design.ru/paf) - $Id: untaint.C,v 1.42 2001/04/20 09:05:24 paf Exp $ + $Id: untaint.C,v 1.45 2001/04/23 09:46:17 paf Exp $ */ #include "pa_config_includes.h" @@ -120,15 +120,16 @@ inline bool need_quote_http_header(const } /** - @test optimize whitespaces for all but 'html' @todo fix theoretical \n mem overrun in TYPO replacements - @test mail-header */ -char *String::store_to(char *dest, Untaint_lang lang, SQL_Connection *connection) const { +char *String::store_to(char *dest, Untaint_lang lang, + SQL_Connection *connection, + const char *charset) const { // $MAIN:html-typo table Table *user_typo_table=static_cast(pool().tag()); Table *typo_table=user_typo_table?user_typo_table:default_typo_table; + bool whitespace=true; const Chunk *chunk=&head; do { const Chunk::Row *row=chunk->rows; @@ -141,6 +142,23 @@ char *String::store_to(char *dest, Untai switch(lang==UL_UNSPECIFIED?row->item.lang:lang) { case UL_CLEAN: // clean piece + { // optimizing whitespace + const char *src=row->item.ptr; + for(int size=row->item.size; size--; src++) + switch(*src) { + case ' ': case '\n': case '\r': case '\t': + if(!whitespace) { + *dest++=*src; + whitespace=true; + } + break; + default: + whitespace=false; + *dest++=*src; + break; + } + } + break; case UL_TAINTED: // tainted piece, but undefined untaint language // for VString.as_double of tainted values @@ -180,8 +198,24 @@ char *String::store_to(char *dest, Untai break; case UL_MAIL_HEADER: // tainted, untaint language: mail-header - memcpy(dest, row->item.ptr, row->item.size); - dest+=row->item.size; + { + // Subject: Re: parser3: =?koi8-r?Q?=D3=C5=CD=C9=CE=C1=D2?= + const char *src=row->item.ptr; + bool to_base_64=false; + for(int size=row->item.size; size--; src++) { + if(*src & 0x80) { + if(!to_base_64) { + dest+=sprintf(dest, "=?%.15s?Q?", charset); + to_base_64=true; + } + dest+=sprintf(dest, "=%02X", *src & 0xFF); + } else { + *dest++=*src; + } + } + if(to_base_64) // close + dest+=sprintf(dest, "?="); + } break; case UL_TABLE: // tainted, untaint language: table @@ -297,6 +331,9 @@ char *String::store_to(char *dest, Untai static_cast(row->item.lang), i); // never } + + if((lang==UL_UNSPECIFIED?row->item.lang:lang)!=UL_CLEAN) + whitespace=false; } chunk=row->link; } while(chunk);