--- parser3/src/include/pa_string.h 2002/02/20 12:40:24 1.129 +++ parser3/src/include/pa_string.h 2002/08/08 08:43:38 1.142 @@ -3,18 +3,16 @@ Copyright (c) 2001, 2002 ArtLebedev Group (http://www.artlebedev.com) Author: Alexandr Petrosian (http://paf.design.ru) - - $Id: pa_string.h,v 1.129 2002/02/20 12:40:24 paf Exp $ */ #ifndef PA_STRING_H #define PA_STRING_H +static const char* IDENT_STRING_H="$Date: 2002/08/08 08:43:38 $"; + #include "pa_pool.h" #include "pa_types.h" -class Table; - #ifndef NO_STRING_ORIGIN # define STRING_APPEND_PARAMS \ const char *src, size_t size, \ @@ -44,6 +42,7 @@ class Table; /// handy: appends const char* piece to String @see String::real_append #define APPEND_CONST(src) APPEND_AS_IS(src, 0, 0, 0) +class Table; class Array; class SQL_Connection; class Dictionary; @@ -107,17 +106,18 @@ public: public: + static String& OnPool(Pool& apool, const char *local_src=0, size_t src_size=0, bool tainted=false); String(Pool& apool, const char *src=0, size_t src_size=0, bool tainted=false); String(const String& src); - bool is_empty() const { return append_here==head.rows; } + bool is_empty() const { return append_here==head.chunk.rows; } size_t size() const; /// convert to C string. if 'lang' known, forcing 'lang' to it char *cstr(Untaint_lang lang=UL_AS_IS, SQL_Connection *connection=0, - Charset *cstr_charset=0) const { + Charset *cstr_charset=0, const char *cstr_charset_name=0) const { char *result=(char *)malloc(cstr_bufsize(lang, connection, cstr_charset)); - char *eol=store_to(result, lang, connection, cstr_charset); + char *eol=store_to(result, lang, connection, cstr_charset, cstr_charset_name); *eol=0; return result; } @@ -125,7 +125,7 @@ public: /// 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); + bool 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 */ @@ -154,6 +154,11 @@ public: */ int cmp(int& partial, const char* src_ptr, size_t src_size=0, size_t this_offset=0, Untaint_lang lang=UL_UNSPECIFIED) const; + /// this starts with src + bool starts_with(const char* src_ptr, size_t src_size=0) const { + int p; cmp(p, src_ptr, src_size); + return p==0 || p==2; + } bool operator == (const char* src_ptr) const { size_t src_size=src_ptr?strlen(src_ptr):0; if(size() != src_size) @@ -173,8 +178,10 @@ public: /// simple hash code of string. used by Hash uint hash_code() const; - /// extracts first char of a string - char first_char() const; + /// extracts first char of a string, if any + char first_char() const { + return is_empty()?*head.chunk.rows[0].item.ptr:0; + } /// extracts [start, finish) piece of string String& mid(size_t start, size_t finish) const; @@ -195,7 +202,9 @@ public: const String& delim, Untaint_lang lang=UL_UNSPECIFIED, int limit=-1) const; - typedef void (*Row_action)(Table& table, Array *row, int start, int finish, + typedef void (*Row_action)(Table& table, Array *row, + int prestart, int prefinish, + int poststart, int postfinish, void *info); /** @return true if fills table. @@ -221,6 +230,8 @@ public: double as_double() const; int as_int() const; + String& join_chains(Pool& pool, char** cstr) const; + #ifndef NO_STRING_ORIGIN /// origin of string. calculated by first row const Origin& origin() const; @@ -229,15 +240,13 @@ public: private: /** several String fragments - - 'mutable' because can write after it's end, after it was appended to somebody - @see String::append */ - mutable struct Chunk { + struct Chunk { typedef uchar count_type; count_type count; ///< the number of rows in chunk + // here could be some padding bytes /// string fragment or a link to next chunk union - union Row { + typedef union Row { typedef uchar item_size_type; /// fragment struct { @@ -248,11 +257,19 @@ private: Origin origin; ///< origin #endif } item; + // we are using the fact that there's no padding before this field! Chunk *link; ///< link to the next chunk in chain - } rows[CR_PREALLOCATED_COUNT]; - } - head; ///< the head chunk of the chunk chain - Chunk *initial_head_link; ///< next rows initially are here[can shrink! @see String::append]. must go right after 'head' + } rows_type[CR_PREALLOCATED_COUNT]; + rows_type rows; + }; + /** + 'mutable' because can write after it's end, after it was appended to somebody + @see String::append + */ + mutable struct { + Chunk chunk; + Chunk *link_storage; + } head; ///< the head chunk of the chunk chain /// next append would write to this record Chunk::Row *append_here; @@ -277,13 +294,12 @@ private: /// 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; + Charset *store_to_charset=0, + const char *store_to_charset_name=0) const; - String& reconstruct(Pool& pool) const; void join_chain(Pool& pool, - uint& ai, const Chunk*& achunk, const Chunk::Row*& arow, + const Chunk*& achunk, const Chunk::Row*& arow, uint& acountdown, uchar& joined_lang, const char *& joined_ptr, size_t& joined_size) const; - String& replace_in_reconstructed(Pool& pool, Dictionary& dict) const; private: //disabled @@ -292,18 +308,22 @@ private: //disabled }; #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; icount; i++, row++) { \ - if(row==(src).append_here) \ - goto break2; \ - \ - body \ - } \ - chunk=row->link; \ - } while(chunk); \ +#define STRING_PREPARED_FOREACH_ROW(self, body) \ + while(row!=(self).append_here) { \ + if(countdown==0) { \ + chunk=row->link; \ + row=chunk->rows; \ + countdown=chunk->count; \ + }; \ + { body } \ + row++; countdown--; \ + } + +#define STRING_PREFIX_FOREACH_ROW(self, body) { \ + const Chunk *chunk=&(self).head.chunk; \ + const Chunk::Row *row=chunk->rows; \ + uint countdown=chunk->count; \ + STRING_PREPARED_FOREACH_ROW(self, body) \ } #define STRING_FOREACH_ROW(body) STRING_PREFIX_FOREACH_ROW(*this, body)