Diff for /parser3/src/include/pa_string.h between versions 1.16 and 1.124

version 1.16, 2001/02/12 13:26:54 version 1.124, 2002/02/08 08:30:14
Line 1 Line 1
 /*  /** @file
   $Id$          Parser: string class decl.
 */  
   
 /*  
   
         String                          Chunk0          Copyright (c) 2001, 2002 ArtLebedev Group (http://www.artlebedev.com)
         ======                          ========          Author: Alexandr Petrosian <paf@design.ru> (http://paf.design.ru)
         head--------------->[ptr, size, ...]  
         append_here-------->[ptr, size, ...]  
                                                 .  
                                                 .  
                                                 [ptr, size, ...]  
         link_row----------->[link to the next chunk]  
   
           $Id$
 */  */
   
 #ifndef PA_STRING_H  #ifndef PA_STRING_H
 #define PA_STRING_H  #define PA_STRING_H
   
 #ifdef HAVE_CONFIG_H  
 #include "pa_config.h"  
 #endif  
   
 #include <stddef.h>  
   
 #include "pa_pool.h"  #include "pa_pool.h"
 #include "pa_types.h"  #include "pa_types.h"
   
   class Table;
   
 #ifndef NO_STRING_ORIGIN  #ifndef NO_STRING_ORIGIN
 #       define STRING_APPEND_PARAMS const char *src, char *file, uint line  #       define STRING_APPEND_PARAMS \
 #       define APPEND(src, file, line) real_append(src, file, line)                  const char *src, size_t size,  \
                   uchar lang, \
                   const char *file, uint line
   /// appends piece to String  @see String::real_append
   #       define APPEND(src, size, lang, file, line) \
                   real_append(src, size, lang, file, line)
 #else  #else
 #       define STRING_APPEND_PARAMS const char *src  #       define STRING_APPEND_PARAMS \
 #       define APPEND(src, file, line) real_append(src)                  const char *src, \
                   size_t size, \
                   uchar lang
   /// appends piece to String  @see String::real_append
   #       define APPEND(src, size, lang, file, line) \
                   real_append(src, size, lang)
 #endif  #endif
   /// appends clean piece to String @see String::real_append
 class String_iterator;  #define APPEND_CLEAN(src, size, file, line) \
           APPEND(src, size, String::UL_CLEAN, file, line)
   /// appends piece to String as-is @see String::real_append
   #define APPEND_AS_IS(src, size, file, line) \
           APPEND(src, size, String::UL_AS_IS, file, line)
   /// appends tainted piece to String  @see String::real_append
   #define APPEND_TAINTED(src, size, file, line) \
           APPEND(src, size, String::UL_TAINTED, file, line)
   /// handy: appends const char* piece to String  @see String::real_append
   #define APPEND_CONST(src) APPEND_AS_IS(src, 0, 0, 0)
   
   class Array;
   class SQL_Connection;
   class Dictionary;
   
   /** 
           Pooled string.
   
           Internal structure:
           @verbatim
                   String                          Chunk0
                   ======                          ========
                   head--------------->[ptr, size, ...]
                   append_here-------->[ptr, size, ...]
                                                           .
                                                           .
                                                           [ptr, size, ...]
                   link_row----------->[link to the next chunk]
           @endverbatim
   
           All pieces remember 
           - the file and its line they are from [can be turned off by NO_STRING_ORIGIN]
           - whether they are tainted or not, 
             and the language which should be used to detaint them
   */
   #include "pa_pragma_pack_begin.h"
 class String : public Pooled {  class String : public Pooled {
         friend String_iterator;  
 public:  public:
   
         enum {          enum {
                 CR_PREALLOCATED_COUNT=5,                  CR_PREALLOCATED_COUNT=2, ///< default preallocated item count
                 CR_GROW_PERCENT=60                  CR_GROW_COUNT=2 ///< each time the String chunk_is_full() string expanded()
           };
   
           /** piece is tainted or not. the language to use when detaint
                   remember to change String_Untaint_lang_name @ untaint.C along
           */
           enum Untaint_lang {
                   UL_UNSPECIFIED=0, ///< zero value handy for hash lookup @see untaint_lang_name2enum
                   // these two must go before others, there are checks for >UL_AS_IS
                   UL_CLEAN, ///< clean
                   UL_AS_IS,     ///< leave all characters intact
   
                   UL_PASS_APPENDED,
                           /**<
                                   leave language built into string being appended.
                                   just a flag, that value not stored
                           */
                   UL_TAINTED,  ///< tainted, untaint language as assigned later 
                   // untaint languages. assigned by ^untaint[lang]{...}
                   UL_FILE_SPEC, ///< file specification
                   UL_HTTP_HEADER,    ///< text in HTTP response header
                   UL_MAIL_HEADER,    ///< text in mail header
                   UL_URI,       ///< text in uri
                   UL_TABLE,     ///< ^table:set body
                   UL_SQL,       ///< ^table:sql body
                   UL_JS,        ///< JavaScript code
                   UL_XML,         ///< ^dom:set xml
                   UL_HTML,      ///< HTML code (for editing)
                   UL_OPTIMIZE_BIT = 0x80  ///< flag, requiring cstr whitespace optimization
         };          };
   
 public:  public:
   
         String(Pool& apool);          String(Pool& apool, const char *src=0, size_t src_size=0, bool tainted=false);
         String(const String& src);          String(const String& src);
         size_t size() const { return fsize; }          size_t size() const;
         int used_rows() const { return fused_rows; }          /// convert to C string. if 'lang' known, forcing 'lang' to it
         char *cstr() const;          char *cstr(Untaint_lang lang=UL_AS_IS, 
                   SQL_Connection *connection=0,
                   Charset *cstr_charset=0) const {
   
                   char *result=(char *)malloc(cstr_bufsize(lang, connection, cstr_charset));
                   char *eol=store_to(result, lang, connection, cstr_charset);
                   *eol=0;
                   return result;
           }
           char *cstr_debug_origins() const;
           /// puts pieces to buf
           void serialize(size_t prolog_size,  void *& buf, size_t& buf_size) const;
           /// appends pieces from buf to self
           void deserialize(size_t prolog_size, void *buf, size_t buf_size, const char *file);
           /** append fragment
                   @see APPEND_AS_IS, APPEND_CLEAN, APPEND_TAINTED, APPEND_CONST
           */
         String& real_append(STRING_APPEND_PARAMS);          String& real_append(STRING_APPEND_PARAMS);
         bool operator == (const String& src) const;          /// @return <0 ==0 or >0 depending on comparison result
         String& append(const String_iterator& begin, const String_iterator& end);          int cmp (int& partial, const String& src, 
                   size_t this_offset=0, Untaint_lang lang=UL_UNSPECIFIED) const;
           bool operator < (const String& src) const {     int p; return cmp(p, src)<0; }
           bool operator > (const String& src) const {     int p; return cmp(p, src)>0; }
           bool operator <= (const String& src) const { int p; return cmp(p, src)<=0; }
           bool operator >= (const String& src) const { int p; return cmp(p, src)>=0; }
           bool operator == (const String& src) const {
                   if(size()!=src.size()) // can speed up in trivial case
                           return false;
                   int p; return cmp(p, src)==0;
           }
           bool operator != (const String& src) const { int p; return cmp(p, src)!=0; }
   
           /**
                    @param partial 
                           returns partial match status. 
                           - -1: strings too different
                           -  0: full match
                           -  1: means @c this starts @c src
                           -  2: means @src starts @this
           */
           int cmp(int& partial, const char* src_ptr, size_t src_size=0, 
                   size_t this_offset=0, Untaint_lang lang=UL_UNSPECIFIED) const;
           bool operator == (const char* src_ptr) const { 
                   size_t src_size=src_ptr?strlen(src_ptr):0;
                   if(size() != src_size)
                           return false;
                   int partial; // unused
                   return cmp(partial, src_ptr, src_size)==0; 
           }
           bool operator != (const char* src_ptr) const { 
                   int partial; // unused
                   return cmp(partial, src_ptr, 0)!=0; 
           }
   
           /** 
                   appends other String.
   
                   marking all tainted pieces of it with @a lang.
                   or marking ALL pieces of it with a @a lang when @a forced to.
           */
           String& append(const String& src, uchar lang, bool forced=false);
           String& operator << (const String& src) { return append(src, UL_PASS_APPENDED); }
           String& operator << (const char *src) { return APPEND_CONST(src); }
   
           /// simple hash code of string. used by Hash
         uint hash_code() const;          uint hash_code() const;
   
         const Origin& origin() const { return head.rows[0].item.origin; }          /// extracts first char of a string
           char first_char() const;
   
           /// extracts [start, finish) piece of string
           String& mid(size_t start, size_t finish) const;
   
           /// @return position of substr in string, -1 means "not found" [String version]
           int pos(const String& substr, 
                   int this_offset=0, Untaint_lang lang=UL_UNSPECIFIED) const;
           /// @return position of substr in string, -1 means "not found" [const char* version]
           int pos(const char *substr, size_t substr_size=0, 
                   int this_offset=0, Untaint_lang lang=UL_UNSPECIFIED) const;
   
           void split(Array& result, 
                   size_t *pos_after_ref, 
                   const char *delim, size_t delim_size, 
                   Untaint_lang lang=UL_UNSPECIFIED, int limit=-1) const;
           void split(Array& result, 
                   size_t *pos_after_ref, 
                   const String& delim, 
                   Untaint_lang lang=UL_UNSPECIFIED, int limit=-1) const;
   
           typedef void (*Row_action)(Table& table, Array *row, int start, int finish, 
                   void *info);
           /**
                   @return true if fills table.
                   table format is defined and fixed[can be used by others]: 
                   @verbatim
                           prematch/match/postmatch/1/2/3/...
                   @endverbatim
           */
           bool match(
                   const String *aorigin,          
                   const String& regexp, 
                   const String *options,
                   Table **table,
                   Row_action row_action, void *info,
                   bool *was_global=0) const;
           enum Change_case_kind {
                   CC_UPPER,
                   CC_LOWER
           };
           String& change_case(Pool& pool,
                   Change_case_kind kind) const;
           String& replace(Pool& pool, Dictionary& dict) const;
           double as_double() const;
           int as_int() const;
   
   #ifndef NO_STRING_ORIGIN
           /// origin of string. calculated by first row
           const Origin& origin() const;
   #endif
   
 private:  private:
   
           /// several String fragments
         struct Chunk {          struct Chunk {
                 // the number of rows in chunk                  ushort count; ///< the number of rows in chunk
                 int count;                  /// string fragment or a link to next chunk union
                 union Row {                  union Row {
                         // chunk item                          /// fragment
                         struct {                          struct { 
                                 const char *ptr;  // pointer to the start of string fragment                                  const char *ptr;  ///< pointer to the start
                                 size_t size;  // length of the fragment                                  ushort size;  ///< length
                                 Origin origin;  // origin of this fragment                                  unsigned char/*Untaint_lang*/ lang; ///< untaint flag, later untaint language
   #ifndef NO_STRING_ORIGIN
                                   Origin origin;  ///< origin
   #endif
                         } item;                          } item;
                         Chunk *link;  // link to the next chunk in chain                          Chunk *link;  ///< link to the next chunk in chain
                 } rows[CR_PREALLOCATED_COUNT];                  } rows[CR_PREALLOCATED_COUNT];
                 // next rows are here  
                 Chunk *preallocated_link;  
         }          }
                 head;  // the head chunk of the chunk chain                  head;  ///< the head chunk of the chunk chain
           Chunk *head_link; ///< next rows are here
   
         // next append would write to this record          /// next append would write to this record
         Chunk::Row *append_here;          Chunk::Row *append_here;
                   
         // the address of place where lies address           /** the address of place where lies address 
         // of the link to the next chunk to allocate                  of the link to the next chunk to allocate
           */
         Chunk::Row *link_row;          Chunk::Row *link_row;
   
 private:  private:
         // last chank allocated count          /// last chunk
         int curr_chunk_rows;          Chunk *last_chunk;
   
         // string size  
         size_t fsize;  
   
         // used rows in all chunks  
         int fused_rows;  
   
 private:  private:
   
         bool chunk_is_full() {          bool chunk_is_full() {
                 return append_here == link_row;                  return append_here == link_row;
         }          }
           uint used_rows() const;
         void expand();          void expand();
   
           size_t cstr_bufsize(Untaint_lang lang,
                   SQL_Connection *connection,
                   Charset *buf_charset) const;
           /// convert to C string, store to 'dest' which must be big enough for proper untaint
           char *store_to(char *dest, Untaint_lang lang=UL_UNSPECIFIED, 
                   SQL_Connection *connection=0,
                   Charset *store_to_charset=0) const;
   
           String& reconstruct(Pool& pool) const;
           void join_chain(Pool& pool, 
                                              uint& ai, const Chunk*& achunk, const Chunk::Row*& arow,
                                              uchar& joined_lang, const char *& joined_ptr, size_t& joined_size) const;
           String& replace_in_reconstructed(Pool& pool, Dictionary& dict) const;
   
 private: //disabled  private: //disabled
   
         String& operator = (const String&) { return *this; }          String& operator = (const String&) { return *this; }
   
 };  };
   #include "pa_pragma_pack_end.h"
   
   #define STRING_PREFIX_FOREACH_ROW(src, body) { \
           const Chunk *chunk=&(src).head;  \
           do { \
                   const Chunk::Row *row=chunk->rows; \
                   for(uint i=0; i<chunk->count; i++, row++) { \
                           if(row==(src).append_here) \
                                   goto break2; \
                           \
                           body \
                   } \
                   chunk=row->link; \
           } while(chunk); \
   }
   
 class Char_set {  #define STRING_FOREACH_ROW(body) STRING_PREFIX_FOREACH_ROW(*this, body)
 public:  #define STRING_SRC_FOREACH_ROW(body) STRING_PREFIX_FOREACH_ROW(src, body)
         Char_set();  
         void include(char c) { bools[static_cast<unsigned int>(c)]=true; }  
         bool in(char c) { return bools[static_cast<unsigned int>(c)]; }  
 private:          
         bool bools[0x100];  
 };  
   
 class String_iterator {  
 public:  
         String_iterator(String& astring);  
   
         void operator ++() { skip(); }  
         void operator ++(int) { skip(); }  
   
         void skip_to(Char_set& set);  
         void skip_to(char c);  
   
         bool eof() { return feof; }  
   
         // current char  
         char operator() const;  
   
 protected:  
         // home string  
         String& string;  
         // the row in which we are  
         Chunk::Row *read_here;  
         // position in that row's string fragment  
         int offset;  
         // when read_here reaches this row, move to the next chunk  
         Chunk::Row *link_row;  
   
         bool feof;  
   
 protected:  
   
         // advances position on one char  
         void skip() {  
   
         }  
 };  
   
 #endif  #endif

Removed from v.1.16  
changed lines
  Added in v.1.124


E-mail: