[BACK]Return to untaint.C CVS log [TXT] [DIR] Up to [parser3project] / parser3 / src / main

Diff for /parser3/src/main/untaint.C between versions 1.115.2.12.2.8 and 1.115.2.12.2.26

version 1.115.2.12.2.8, 2003/03/20 16:57:43 version 1.115.2.12.2.26, 2003/04/08 13:58:30
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 144  inline bool need_quote_http_header(const Line 174  inline bool need_quote_http_header(const
         or marking ALL pieces of it with a @a lang when @a forced to,          or marking ALL pieces of it with a @a lang when @a forced to,
         and propagating OPTIMIZE language bit.          and propagating OPTIMIZE language bit.
 */  */
 String& String::append_to(String& dest, Untaint_lang lang, bool forced) const {  String& String::append_to(String& dest, Language lang, bool forced) const {
         if(!*this)          if(is_empty())
                 return dest;                  return dest;
   
         // first: letters          // first: letters
         dest.body=CORD_cat(dest.body, body);          dest.body<<body;
   
         // next: fragment infos          // next: fragment infos
                   
         if(lang==UL_PASS_APPENDED) // without language-chage string?          if(lang==L_PASS_APPENDED) // without language-change?
                 dest.fragments.append(fragments);                  dest.fragments.append(fragments);
         else if(forced) //forcing passed lang?          else if(forced) //forcing passed lang?
                 dest.fragments+=Fragment(lang, length());                  dest.fragments+=Fragment(lang, length());
         else if(lang&UL_OPTIMIZE_BIT) {          else if(lang&L_OPTIMIZE_BIT) {
                 Array_iterator<ArrayFragment::element_type> i(fragments);                  for(Array_iterator<ArrayFragment::element_type> i(fragments); i.has_next(); ) {
                 while(i.has_next()) {                          const Fragment fragment=i.next();
                         const Fragment& fragment=i.next();  
                         // main idea here:                          // main idea here:
                         // tainted piece would get OPTIMIZED bit from 'lang'                          // tainted piece would get OPTIMIZED bit from 'lang'
                         // clean piece would be marked OPTIMIZED manually                          // clean piece would be marked OPTIMIZED manually
                         // pieces with determined languages [not tainted|clean] would retain theirs langs                          // pieces with determined languages [not tainted|clean] would retain theirs langs
                         dest.fragments+=Fragment(static_cast<Untaint_lang>(                          dest.fragments+=Fragment(static_cast<Language>(
                                 fragment.lang==UL_TAINTED?lang:(                                  fragment.lang==L_TAINTED?lang:(
                                         fragment.lang==UL_CLEAN?UL_CLEAN|UL_OPTIMIZE_BIT: // ORing with OPTIMIZED flag                                          fragment.lang==L_CLEAN?L_CLEAN|L_OPTIMIZE_BIT: // ORing with OPTIMIZED flag
                                                 fragment.lang)),                                                  fragment.lang)),
                                 fragment.length);                                  fragment.length);
                 }                  }
         } else { // The core idea: tainted pieces got marked with context's lang          } else { // The core idea: tainted pieces got marked with context's lang
                 Array_iterator<ArrayFragment::element_type> i(fragments);                  for(Array_iterator<ArrayFragment::element_type> i(fragments); i.has_next(); ) {
                 while(i.has_next()) {                          const Fragment fragment=i.next();
                         const Fragment& fragment=i.next();  
                         dest.fragments+=Fragment(                          dest.fragments+=Fragment(
                                 fragment.lang==UL_TAINTED?lang:fragment.lang,                                  fragment.lang==L_TAINTED?lang:fragment.lang,
                                 fragment.length);                                  fragment.length);
                 }                  }
         }          }
   
           ASSERT_STRING_INVARIANT(dest);
         return dest;          return dest;
 }  }
   
Line 217  inline bool mail_header_char_valid_withi Line 246  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) {  /**
           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);
         }          }
 }  }
 CORD String::cstr_to_cord(Untaint_lang lang,   inline void pa_CORD_pos_advance(CORD_pos pos, size_t n) {
                   SQL_Connection *connection,          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, 
                     SQL_Connection* connection,
                   const Request_charsets *charsets) const {                    const Request_charsets *charsets) const {
         CORD_ec result; CORD_ec_init(result);          CORD_ec result; CORD_ec_init(result);
         CORD_pos pos; CORD_set_pos(pos, body, 0);          CORD_pos pos; body.set_pos(pos, 0);
   
         bool whitespace=true;          bool whitespace=true;
   
         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);
   
                 Untaint_lang to_lang=lang==UL_UNSPECIFIED?fragment.lang:lang;                  Language to_lang=lang==L_UNSPECIFIED?fragment.lang:lang;
                   bool optimize=(to_lang & L_OPTIMIZE_BIT)!=0;
                 switch(to_lang & ~UL_OPTIMIZE_BIT) {                  if(!optimize)
                 case UL_CLEAN:                          whitespace=false;
                 case UL_TAINTED:                          
                 case UL_AS_IS:                  switch(to_lang & ~L_OPTIMIZE_BIT) {
                   case L_CLEAN:
                   case L_TAINTED:
                   case L_AS_IS:
                         // clean piece                          // clean piece
   
                         // tainted piece, but undefined untaint language                          // tainted piece, but undefined untaint language
Line 251  CORD String::cstr_to_cord(Untaint_lang l Line 309  CORD String::cstr_to_cord(Untaint_lang 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 UL_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'
Line 261  CORD String::cstr_to_cord(Untaint_lang l Line 319  CORD String::cstr_to_cord(Untaint_lang l
                                         // 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 UL_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 UL_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 UL_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==',' || c=='<')) {
                                                 email=c=='<';                                                  email=c=='<';
                                                 dest+=sprintf(dest, "?=");                                                  to_string("?=");
                                                 to_quoted_printable=false;                                                  to_quoted_printable=false;
                                         }                                          }
                                         if(!email && (                                          if(!email && (
Line 320  CORD String::cstr_to_cord(Untaint_lang l Line 374  CORD String::cstr_to_cord(Untaint_lang 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 UL_TABLE:   
                         // tainted, untaint language: table                          // tainted, untaint language: table
                         escape(switch(c) {                          escape(switch(c) {
                                 case '\t': to_char(' ');  break;                                  case '\t': to_char(' ');  break;
Line 351  CORD String::cstr_to_cord(Untaint_lang l Line 400  CORD String::cstr_to_cord(Untaint_lang l
                         });                          });
                         break;                          break;
 #ifdef PA_SQL  #ifdef PA_SQL
                 case UL_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");
                         break;                          break;
 #endif  #endif
                 case UL_JS:                  case L_JS:
                         escape(switch(c) {                          escape(switch(c) {
                                 case '"': to_string("\\\"");  break;                                  case '"': to_string("\\\"");  break;
                                 case '\'': to_string("\\'");  break;                                  case '\'': to_string("\\'");  break;
Line 376  CORD String::cstr_to_cord(Untaint_lang l Line 424  CORD String::cstr_to_cord(Untaint_lang l
                                 _default;                                  _default;
                         });                          });
                         break;                          break;
                 case UL_XML:                  case L_XML:
                         escape(switch(c) {                          escape(switch(c) {
                                 case '&': to_string("&amp;");  break;                                  case '&': to_string("&amp;");  break;
                                 case '>': to_string("&gt;");  break;                                  case '>': to_string("&gt;");  break;
Line 386  CORD String::cstr_to_cord(Untaint_lang l Line 434  CORD String::cstr_to_cord(Untaint_lang l
                                 _default;                                  _default;
                         });                          });
                         break;                          break;
                 case UL_HTML:                  case L_HTML:
                         escape(switch(c) {                          escape(switch(c) {
                                 case '&': to_string("&amp;");  break;                                  case '&': to_string("&amp;");  break;
                                 case '>': to_string("&gt;");  break;                                  case '>': to_string("&gt;");  break;
Line 396  CORD String::cstr_to_cord(Untaint_lang l Line 444  CORD String::cstr_to_cord(Untaint_lang 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 & UL_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 CORD_ec_to_cord(result);          return StringBody(CORD_ec_to_cord(result));
 }  }
Removed from v.1.115.2.12.2.8  
changed lines
  Added in v.1.115.2.12.2.26

E-mail: