|
|
| version 1.115.2.12.2.12, 2003/03/24 16:08:35 | version 1.118.2.1, 2003/09/23 09:54:28 |
|---|---|
| Line 17 static const char* IDENT_UNTAINT_C="$Dat | Line 17 static const char* IDENT_UNTAINT_C="$Dat |
| #include "pa_common.h" | #include "pa_common.h" |
| #include "pa_charset.h" | #include "pa_charset.h" |
| #include "pa_request_charsets.h" | #include "pa_request_charsets.h" |
| #include "pa_sapi.h" | |
| extern "C" { // author forgot to do that | extern "C" { // author forgot to do that |
| #include "ec.h" | #include "ec.h" |
| } | } |
| //#define PA_SQL | #define PA_SQL |
| #ifdef PA_SQL | #ifdef PA_SQL |
| #include "pa_sql_connection.h" | #include "pa_sql_connection.h" |
| #endif | #endif |
| // defines | |
| #undef CORD_ec_append | |
| // redefining to intercept flushes and implement whitespace optimization | |
| // of all consequent white space chars leaving only first one | |
| #define CORD_ec_append(x, c) \ | |
| { \ | |
| bool skip=false; \ | |
| if(optimize) switch(c) { \ | |
| case ' ': case '\r': case '\n': case '\t': \ | |
| if(whitespace) \ | |
| skip=true; /*skipping subsequent*/ \ | |
| else \ | |
| whitespace=true; \ | |
| break; \ | |
| default: \ | |
| whitespace=false; \ | |
| break; \ | |
| } \ | |
| if(!skip) { \ | |
| if ((x)[0].ec_bufptr == (x)[0].ec_buf + CORD_BUFSZ) { \ | |
| CORD_ec_flush_buf(x); \ | |
| } \ | |
| *((x)[0].ec_bufptr)++ = (c); \ | |
| } \ | |
| } | |
| #define escape(action) \ | #define escape(action) \ |
| for(; fragment_length--; CORD_next(pos)) { \ | for(; fragment_length--; CORD_next(pos)) { \ |
| char c=CORD_pos_fetch(pos); \ | char c=CORD_pos_fetch(pos); \ |
| action \ | action \ |
| } | } |
| #define _default default: CORD_ec_append(result, c); break | #define _default default: CORD_ec_append(result, c); break |
| #define encode(need_encode_func, prefix) \ | #define encode(need_encode_func, prefix, otherwise) \ |
| if(need_encode_func(c)) { \ | if(need_encode_func(c)) { \ |
| static const char* hex="0123456789ABCDEF"; \ | static const char* hex="0123456789ABCDEF"; \ |
| CORD_ec_append(result, prefix); \ | CORD_ec_append(result, prefix); \ |
| CORD_ec_append(result, hex[((unsigned char)c)/0x10]); \ | CORD_ec_append(result, hex[((unsigned char)c)/0x10]); \ |
| CORD_ec_append(result, hex[((unsigned char)c)%0x10]); \ | CORD_ec_append(result, hex[((unsigned char)c)%0x10]); \ |
| } else \ | } else \ |
| CORD_ec_append(result, c); | CORD_ec_append(result, otherwise); |
| #define to_char(c) CORD_ec_append(result, c) | #define to_char(c) CORD_ec_append(result, c) |
| #define to_string(s) CORD_ec_append_cord(result, s) | #define to_string(s) CORD_ec_append_cord(result, s) |
| Line 51 inline bool need_file_encode(unsigned ch | Line 81 inline bool need_file_encode(unsigned ch |
| return strchr( | return strchr( |
| "*?'\"<>|" | "*?'\"<>|" |
| #ifndef WIN32 | #ifndef WIN32 |
| ":\\~" | ":\\" |
| #endif | #endif |
| , c)!=0; | , c)!=0; |
| } | } |
| Line 137 inline bool need_quote_http_header(const | Line 167 inline bool need_quote_http_header(const |
| return false; | return false; |
| } | } |
| #ifndef DOXYGEN | |
| struct Append_fragment_info { | |
| String::Language lang; | |
| const StringBody dest_body; | |
| String::Languages* dest_languages; | |
| }; | |
| #endif | |
| int append_fragment_optimizing(char alang, size_t asize, Append_fragment_info& info) { | |
| String::Language lang=(String::Language)alang; | |
| // main idea here: | |
| // tainted piece would get OPTIMIZED bit from 'lang' | |
| // clean piece would be marked OPTIMIZED manually | |
| // pieces with determined languages [not tainted|clean] would retain theirs langs | |
| info.dest_languages->append(info.dest_body, | |
| lang==String::Language::L_TAINTED? | |
| info.lang | |
| :lang==String::Language::L_CLEAN? | |
| (String::Language)(String::Language::L_CLEAN|String::Language::L_OPTIMIZE_BIT) // ORing with OPTIMIZED flag | |
| :lang, | |
| asize); | |
| return 0; // 0=continue | |
| } | |
| int append_fragment_nonoptimizing(char alang, size_t asize, Append_fragment_info& info) { | |
| String::Language lang=(String::Language)alang; | |
| // The core idea: tainted pieces got marked with context's lang | |
| info.dest_languages->append(info.dest_body, | |
| lang==String::Language::L_TAINTED? | |
| info.lang | |
| :lang, | |
| asize); | |
| return 0; // 0=continue | |
| } | |
| /** | /** |
| appends to other String, | appends to other String, |
| Line 148 String& String::append_to(String& dest, | Line 212 String& String::append_to(String& dest, |
| if(is_empty()) | if(is_empty()) |
| return dest; | return dest; |
| // first: letters | // first: fragment infos |
| dest.body.append(body); | |
| // next: fragment infos | |
| if(lang==L_PASS_APPENDED) // without language-chage string? | if(lang==L_PASS_APPENDED) // without language-change? |
| dest.fragments.append(fragments); | dest.langs.append(dest.body, body, langs); |
| else if(forced) //forcing passed lang? | else if(forced) //forcing passed lang? |
| dest.fragments+=Fragment(lang, length()); | dest.langs.append(dest.body, lang, length()); |
| else if(lang&L_OPTIMIZE_BIT) { | else { |
| Array_iterator<ArrayFragment::element_type> i(fragments); | Append_fragment_info info={lang, dest.body, &dest.langs}; |
| while(i.has_next()) { | langs.for_each(body, lang&L_OPTIMIZE_BIT? |
| const Fragment& fragment=i.next(); | append_fragment_optimizing |
| // main idea here: | :append_fragment_nonoptimizing, info); |
| // tainted piece would get OPTIMIZED bit from 'lang' | |
| // clean piece would be marked OPTIMIZED manually | |
| // pieces with determined languages [not tainted|clean] would retain theirs langs | |
| dest.fragments+=Fragment(static_cast<Language>( | |
| fragment.lang==L_TAINTED?lang:( | |
| fragment.lang==L_CLEAN?L_CLEAN|L_OPTIMIZE_BIT: // ORing with OPTIMIZED flag | |
| fragment.lang)), | |
| fragment.length); | |
| } | |
| } else { // The core idea: tainted pieces got marked with context's lang | |
| Array_iterator<ArrayFragment::element_type> i(fragments); | |
| while(i.has_next()) { | |
| const Fragment& fragment=i.next(); | |
| dest.fragments+=Fragment( | |
| fragment.lang==L_TAINTED?lang:fragment.lang, | |
| fragment.length); | |
| } | |
| } | } |
| // next: letters | |
| dest.body<<body; | |
| return dest; | return dest; |
| } | } |
| Line 217 inline bool mail_header_char_valid_withi | Line 264 inline bool mail_header_char_valid_withi |
| || c>='0' && c<='9' | || c>='0' && c<='9' |
| || strchr("!*+-/", c); | || strchr("!*+-/", c); |
| } | } |
| inline void ec_append(CORD_ec& result, CORD_pos& pos, size_t size) { | inline bool addr_spec_soon(const char *src) { |
| for(char c; c=*src; src++) | |
| if(c=='<') | |
| return true; | |
| else if(!(c==' ' || c=='\t')) | |
| return false; | |
| return false; | |
| } | |
| /** | |
| RFC | |
| Upper case should be used for hexadecimal digits "A" through "F" | |
| The 8-bit hexadecimal value 20 (e.g., ISO-8859-1 SPACE) | |
| may be represented as "_" | |
| */ | |
| inline bool mail_header_nonspace_char(char c) { | |
| return c != 0x20; | |
| } | |
| inline void ec_append(CORD_ec& result, bool& optimize, bool& whitespace, CORD_pos pos, size_t size) { | |
| while(size--) { | while(size--) { |
| CORD_ec_append(result, CORD_pos_fetch(pos)); | CORD_ec_append(result, CORD_pos_fetch(pos)); |
| CORD_next(pos); | CORD_next(pos); |
| } | } |
| } | } |
| inline void pa_CORD_pos_advance(CORD_pos pos, size_t n) { | |
| while(true) { | |
| size_t avail=CORD_pos_chars_left(pos); | |
| if(avail==0) { | |
| CORD_next(pos); | |
| if(!--n) | |
| break; | |
| } else if(avail<n) { | |
| CORD_pos_advance(pos, avail); | |
| n-=avail; | |
| } else { // avail>=n | |
| CORD_pos_advance(pos, n); | |
| break; | |
| } | |
| } | |
| } | |
| StringBody String::cstr_to_string_body(Language lang, | StringBody String::cstr_to_string_body(Language lang, |
| SQL_Connection* connection, | SQL_Connection* connection, |
| const Request_charsets *charsets) const { | const Request_charsets *charsets) const { |
| Line 234 StringBody String::cstr_to_string_body(L | Line 314 StringBody String::cstr_to_string_body(L |
| size_t fragment_begin=0; | size_t fragment_begin=0; |
| size_t fragment_end; | size_t fragment_end; |
| for(Array_iterator<Fragment> i(fragments); i.has_next(); fragment_begin=fragment_end) { | for(Array_iterator<Fragment> i(fragments); i.has_next(); fragment_begin=fragment_end) { |
| const Fragment& fragment=i.next(); | const Fragment fragment=i.next(); |
| size_t fragment_length=fragment.length; | size_t fragment_length=fragment.length; |
| fragment_end=fragment_begin+fragment_length; | fragment_end=fragment_begin+fragment_length; |
| //fprintf(stderr, "%d, %d\n", fragment.lang, fragment.length); | //fprintf(stderr, "%d, %d\n", fragment.lang, fragment.length); |
| Language to_lang=lang==L_UNSPECIFIED?fragment.lang:lang; | Language to_lang=lang==L_UNSPECIFIED?fragment.lang:lang; |
| bool optimize=(to_lang & L_OPTIMIZE_BIT)!=0; | |
| if(!optimize) | |
| whitespace=false; | |
| switch(to_lang & ~L_OPTIMIZE_BIT) { | switch(to_lang & ~L_OPTIMIZE_BIT) { |
| case L_CLEAN: | case L_CLEAN: |
| case L_TAINTED: | case L_TAINTED: |
| Line 252 StringBody String::cstr_to_string_body(L | Line 335 StringBody String::cstr_to_string_body(L |
| // for ^process{body} evaluation | // for ^process{body} evaluation |
| // tainted, untaint language: as-is | // tainted, untaint language: as-is |
| ec_append(result, pos, fragment_length); | ec_append(result, optimize, whitespace, pos, fragment_length); |
| break; | break; |
| case L_FILE_SPEC: | case L_FILE_SPEC: |
| // tainted, untaint language: file [name] | // tainted, untaint language: file [name] |
| escape( | escape( |
| // Macintosh has problems with small Russian letter 'r' | // Macintosh has problems with small Russian letter 'r' |
| if( c=='\xF0' && charsets && charsets->source().name()=="windows-1251" ) { | if( c=='\xF0' && charsets && charsets->source().NAME()=="WINDOWS-1251" ) { |
| // fixing that letter for most common charset | // fixing that letter for most common charset |
| to_char('p'); | to_char('p'); |
| } else // fallback to default | } else // fallback to default |
| encode(need_file_encode, '_'); | encode(need_file_encode, '_', c); |
| ); | ); |
| break; | break; |
| #ifdef LATER | |
| case L_URI: | case L_URI: |
| // tainted, untaint language: uri | // tainted, untaint language: uri |
| { | { |
| const char *substr= | const char *fragment_str=body.mid(fragment_begin, fragment_length).cstr(); |
| CORD_to_const_char_star( | // skip source [we use recoded version] |
| CORD_substr(body, fragment_begin, fragment_length)); | pa_CORD_pos_advance(pos, fragment_length); |
| const void *client_ptr; | String::C output(fragment_str, fragment_length); |
| size_t client_size; | |
| if(charsets) | if(charsets) |
| Charset::transcode( | output=Charset::transcode(output, |
| charsets->source(), substr, fragment_length, | charsets->source(), |
| charsets->client(), client_ptr, client_size); | charsets->client()); |
| else { | |
| client_ptr=substr; | |
| client_size=fragment_length; | |
| } | |
| char c; | char c; |
| for(const char* src=(const char* )client_ptr; c=*src++; ) | for(const char* src=output.str; c=*src++; ) |
| encode(need_uri_encode, '%'); | encode(need_uri_encode, '%', c); |
| } | } |
| break; | break; |
| #endif | |
| case L_HTTP_HEADER: | case L_HTTP_HEADER: |
| // tainted, untaint language: http-field-content-text | // tainted, untaint language: http-field-content-text |
| escape( | escape( |
| encode(need_uri_encode, '%'); | encode(need_uri_encode, '%', c); |
| ); | ); |
| break; | break; |
| #ifdef LATER | |
| case L_MAIL_HEADER: | case L_MAIL_HEADER: |
| // tainted, untaint language: mail-header | // tainted, untaint language: mail-header |
| // http://www.ietf.org/rfc/rfc2047.txt | // http://www.ietf.org/rfc/rfc2047.txt |
| if(charsets) { | if(charsets) { |
| const void *mail_ptr=fragment.ptr; | size_t mail_size; |
| size_t mail_size=fragment_length; | const char *mail_ptr= |
| const char* charset_name=charsets->mail().name()->cstr(); | body.mid(fragment_begin, mail_size=fragment_length).cstr(); |
| // skip source [we use recoded version] | |
| pa_CORD_pos_advance(pos, mail_size); | |
| const char* charset_name=charsets->mail().NAME().cstr(); | |
| // Subject: Re: parser3: =?koi8-r?Q?=D3=C5=CD=C9=CE=C1=D2?= | // Subject: Re: parser3: =?koi8-r?Q?=D3=C5=CD=C9=CE=C1=D2?= |
| const char* src=(const char* )mail_ptr; | |
| bool to_quoted_printable=false; | bool to_quoted_printable=false; |
| bool email=false; | bool email=false; |
| for(const char* end=src+mail_size; src<end; src++) { | uchar c; |
| for(const char* src=mail_ptr; c=(uchar)*src++; ) { | |
| //RFC + An 'encoded-word' MUST NOT appear in any portion of an 'addr-spec'. | //RFC + An 'encoded-word' MUST NOT appear in any portion of an 'addr-spec'. |
| if(to_quoted_printable && (c==',' || c=='<')) { | if(to_quoted_printable && (c==',' || addr_spec_soon(src) || c == '"')) { |
| email=c=='<'; | email=c=='<'; |
| dest+=sprintf(dest, "?="); | to_string("?="); |
| to_quoted_printable=false; | to_quoted_printable=false; |
| } | } |
| if(!email && ( | if(!email && ( |
| Line 321 StringBody String::cstr_to_string_body(L | Line 400 StringBody String::cstr_to_string_body(L |
| || to_quoted_printable && !mail_header_char_valid_within_Qencoded(c) | || to_quoted_printable && !mail_header_char_valid_within_Qencoded(c) |
| )) { | )) { |
| if(!to_quoted_printable) { | if(!to_quoted_printable) { |
| dest+=sprintf(dest, "=?%s?Q?", charset_name.get()); | to_string("=?"); |
| to_string(charset_name); | |
| to_string("?Q?"); | |
| to_quoted_printable=true; | to_quoted_printable=true; |
| } | } |
| //RFC Upper case should be used for hexadecimal digits "A" through "F" | encode(mail_header_nonspace_char, '=', '_'); |
| if(c == 0x20) // RFC The 8-bit hexadecimal value 20 (e.g., ISO-8859-1 SPACE) | |
| *dest++='_'; // RFC may be represented as "_" | |
| else | |
| dest+=sprintf(dest, "=%02X", c & 0xFF); | |
| } else | } else |
| *dest++=c; | to_char(c); |
| if(c=='>') | if(c=='>') |
| email=false; | email=false; |
| } | } |
| if(to_quoted_printable) // close | if(to_quoted_printable) // close |
| dest+=sprintf(dest, "?="); | to_string("?="); |
| } else { | } else |
| memcpy(dest, fragment.ptr, fragment_length); | ec_append(result, optimize, whitespace, pos, fragment_length); |
| dest+=fragment_length; | |
| } | |
| break; | break; |
| #endif | |
| case L_TABLE: | case L_TABLE: |
| // tainted, untaint language: table | // tainted, untaint language: table |
| escape(switch(c) { | escape(switch(c) { |
| Line 355 StringBody String::cstr_to_string_body(L | Line 429 StringBody String::cstr_to_string_body(L |
| case L_SQL: | case L_SQL: |
| // tainted, untaint language: sql | // tainted, untaint language: sql |
| if(connection) { | if(connection) { |
| dest+=connection->quote(0, | const char *fragment_str=body.mid(fragment_begin, fragment_length).cstr(); |
| CORD_substr(body, fragment_begin, fragment_length), | // skip source [we use recoded version] |
| fragment_length); | pa_CORD_pos_advance(pos, fragment_length); |
| ...^^^ | |
| dest+=connection->quote(dest, fragment.ptr, fragment_length); | to_string(connection->quote(fragment_str, fragment_length)); |
| else | } else |
| throw Exception(0, | throw Exception(0, |
| 0, | 0, |
| "untaint in SQL language failed - no connection specified"); | "untaint in SQL language failed - no connection specified"); |
| Line 397 StringBody String::cstr_to_string_body(L | Line 470 StringBody String::cstr_to_string_body(L |
| }); | }); |
| break; | break; |
| default: | default: |
| throw Exception(0, | SAPI::die("unknown untaint language #%d", |
| 0, | static_cast<int>(to_lang)); // should never |
| "unknown untaint language #%d", | |
| static_cast<int>(to_lang)); // should never | |
| break; // never | break; // never |
| } | } |
| #ifdef LATER | |
| if(to_lang & L_OPTIMIZE_BIT) { | |
| // optimizing whitespace | |
| char *stop=dest; dest=start; | |
| for(char c=start; src<stop; src++) | |
| switch(c) { | |
| // of all consequent white space chars leaving only first one | |
| case ' ': case '\r': case '\n': case '\t': | |
| if(!whitespace) { | |
| *dest++=c; | |
| whitespace=true; | |
| } | |
| break; | |
| default: | |
| whitespace=false; | |
| *dest++=c; | |
| break; | |
| }; | |
| } else // piece without optimization | |
| whitespace=false; | |
| #endif | |
| } | } |
| return StringBody(CORD_ec_to_cord(result)); | return StringBody(CORD_ec_to_cord(result)); |