--- parser3/src/main/pa_string.C 2001/04/02 15:59:56 1.59 +++ parser3/src/main/pa_string.C 2003/10/21 05:11:00 1.185 @@ -1,431 +1,622 @@ /** @file - Parser: string class. @see untasize_t.C. + Parser: string class. @see untalength_t.C. - Copyright (c) 2001 ArtLebedev Group (http://www.artlebedev.com) - - Author: Alexander Petrosyan (http://design.ru/paf) - - $Id: pa_string.C,v 1.59 2001/04/02 15:59:56 paf Exp $ + Copyright (c) 2001-2003 ArtLebedev Group (http://www.artlebedev.com) + Author: Alexandr Petrosian (http://paf.design.ru) */ -#include "pa_config_includes.h" +static const char* IDENT_STRING_C="$Date: 2003/10/21 05:11:00 $"; + +#include "pcre.h" -#include "pa_pool.h" #include "pa_string.h" -#include "pa_hash.h" #include "pa_exception.h" -#include "pa_common.h" +#include "pa_table.h" +#include "pa_dictionary.h" +#include "pa_charset.h" + +const String String::Empty; + +// cord lib extension + +#ifndef DOXYGEN +typedef struct { + ssize_t countdown; + char target; /* Character we're looking for */ +} chr_data; +#endif +static int CORD_range_contains_chr_greater_then_proc(char c, size_t size, void* client_data) +{ + register chr_data * d = (chr_data *)client_data; + + if (d -> countdown<=0) return(2); + d -> countdown -= size; + if (c > d -> target) return(1); + return(0); +} +int CORD_range_contains_chr_greater_then(CORD x, size_t i, size_t n, int c) +{ + chr_data d; + + d.countdown = n; + d.target = c; + return(CORD_block_iter(x, i, CORD_range_contains_chr_greater_then_proc, &d) == 1/*alternatives: 0 normally ended, 2=struck 'n'*/); +} + +static int CORD_block_count_proc(char c, size_t size, void* client_data) +{ + int* result=(int*)client_data; + (*result)++; + return(0); // 0=continue +} +size_t CORD_block_count(CORD x) +{ + size_t result=0; + CORD_block_iter(x, 0, CORD_block_count_proc, &result); + return result; +} + +// helpers + +/// String::match uses this as replace & global search table columns + +const int MAX_MATCH_GROUPS=100; + +class String_match_table_template_columns: public ArrayString { +public: + String_match_table_template_columns() { + *this+=new String("prematch"); + *this+=new String("match"); + *this+=new String("postmatch"); + for(int i=0; icount+last_chunk->count*CR_GROW_PERCENT/100; - last_chunk=static_cast( - malloc(sizeof(size_t)+sizeof(Chunk::Row)*new_chunk_count+sizeof(Chunk *))); - last_chunk->count=new_chunk_count; - link_row->link=last_chunk; - append_here=last_chunk->rows; - link_row=&last_chunk->rows[last_chunk->count]; - link_row->link=0; +String::Body String::Body::Format(int value) { + char local[MAX_NUMBER]; + size_t length=snprintf(local, MAX_NUMBER, "%d", value); + return String::Body(pa_strdup(local, length), length); } -String::String(const String& src) : Pooled(src.pool()) { - head.count=CR_PREALLOCATED_COUNT; - - size_t src_used_rows=src.fused_rows; - if(src_used_rows<=head.count) { - // all new rows fit size_to preallocated area - size_t curr_chunk_rows=head.count; - memcpy(head.rows, src.head.rows, sizeof(Chunk::Row)*src_used_rows); - append_here=&head.rows[src_used_rows]; - link_row=&head.rows[curr_chunk_rows]; - } else { - // warning: - // heavily relies on the fact - // "preallocated area is the same for all strings" - // - // info: - // allocating only enough mem to fit src string rows - // next append would allocate a new chunk - // - // new rows don't fit size_to preallocated area: splitting size_to two chunks - // preallocated chunk src to constructing head - memcpy(head.rows, src.head.rows, sizeof(Chunk::Row)*head.count); - // remaining rows size_to new_chunk - size_t curr_chunk_rows=src_used_rows-head.count; - Chunk *new_chunk=static_cast( - malloc(sizeof(size_t)+sizeof(Chunk::Row)*curr_chunk_rows+sizeof(Chunk *))); - new_chunk->count=curr_chunk_rows; - head.preallocated_link=new_chunk; - append_here=link_row=&new_chunk->rows[new_chunk->count]; - - Chunk *old_chunk=src.head.preallocated_link; - Chunk::Row *new_rows=new_chunk->rows; - size_t rows_left_to_copy=new_chunk->count; - while(true) { - size_t old_count=old_chunk->count; - Chunk *next_chunk=old_chunk->rows[old_count].link; - if(next_chunk) { - // not last source chunk - // taking it all - memcpy(new_rows, old_chunk->rows, sizeof(Chunk::Row)*old_count); - new_rows+=old_count; - rows_left_to_copy-=old_count; - - old_chunk=next_chunk; - } else { - // the last source chunk - // taking only those rows of chunk that _left_to_copy - memcpy(new_rows, old_chunk->rows, sizeof(Chunk::Row)*rows_left_to_copy); - break; - } - } - } - link_row->link=0; - fused_rows=src_used_rows; - fsize=src.fsize; +static int CORD_batched_iter_fn_generic_hash_code(char c, void * client_data) { + uint& result=*static_cast(client_data); + generic_hash_code(result, c); + return 0; +} +static int CORD_batched_iter_fn_generic_hash_code(const char* s, void * client_data) { + uint& result=*static_cast(client_data); + generic_hash_code(result, s); + return 0; +}; +uint String::Body::hash_code() const { + uint result=0; + CORD_iter5(body, 0, + CORD_batched_iter_fn_generic_hash_code, + CORD_batched_iter_fn_generic_hash_code, &result); + return result; } -String& String::append(const String& src, Untaint_lang lang, bool forced) { - size_t src_used_rows=src.fused_rows; - size_t dst_free_rows=link_row-append_here; - - if(src_used_rows<=dst_free_rows) { - // all new rows fit size_to last chunk - memcpy(append_here, src.head.rows, sizeof(Chunk::Row)*src_used_rows); - set_lang(append_here, lang, forced, src_used_rows); - append_here+=src_used_rows; - } else { - // not all new rows fit size_to last chunk: shrinking it to used part, - size_t used_rows=last_chunk->count-dst_free_rows; - //size_t *countp=append_here - link_row=&last_chunk->rows[last_chunk->count=used_rows]; - // allocating only enough mem to fit src string rows - // next append would allocate a new chunk - last_chunk=static_cast( - malloc(sizeof(size_t)+sizeof(Chunk::Row)*src_used_rows+sizeof(Chunk *))); - last_chunk->count=src_used_rows; - link_row->link=last_chunk; - append_here=link_row=&last_chunk->rows[src_used_rows]; - - const Chunk *old_chunk=&src.head; - Chunk::Row *new_rows=last_chunk->rows; - size_t rows_left_to_copy=src_used_rows; - while(true) { - size_t old_count=old_chunk->count; - Chunk *next_chunk=old_chunk->rows[old_count].link; - if(next_chunk) { - // not last source chunk - // taking it all - memcpy(new_rows, old_chunk->rows, sizeof(Chunk::Row)*old_count); - set_lang(new_rows, lang, forced, old_count); - new_rows+=old_count; - rows_left_to_copy-=old_count; - - old_chunk=next_chunk; - } else { - // the last source chunk - // taking only those rows of chunk that _left_to_copy - memcpy(new_rows, old_chunk->rows, sizeof(Chunk::Row)*rows_left_to_copy); - set_lang(new_rows, lang, forced, rows_left_to_copy); - break; - } - } - link_row->link=0; - } - fused_rows+=src_used_rows; - fsize+=src.fsize; +// String methods - return *this; +String::String(const char* cstr, size_t helper_length, bool tainted): body(CORD_EMPTY) { + append_help_length(cstr, helper_length, tainted?L_TAINTED:L_CLEAN); } -void String::set_lang(Chunk::Row *row, Untaint_lang lang, bool forced, size_t size) { - if(lang==UL_PASS_APPENDED) - return; - - while(size--) { - Untaint_lang& item_lang=(row++)->item.lang; - if(item_lang==UL_YES || forced) // tasize_ted? need untasize_t language assignment - item_lang=lang; // assign untasize_t language - } +String::String(const String::C cstr, bool tainted): body(CORD_EMPTY) { + append_know_length(cstr.str, cstr.length, tainted?L_TAINTED:L_CLEAN); } -/*void String::change_lang(Untaint_lang lang) { - Chunk *chunk=&head; - do { - Chunk::Row *row=chunk->rows; - for(size_t i=0; icount; i++) { - if(row==append_here) - goto break2; - - row->item.lang=lang; - row++; - } - chunk=row->link; - } while(chunk); -break2: - return; +String& String::append_know_length(const char* str, size_t known_length, Language lang) { + if(!known_length) + return *this; + + // first: langs + langs.append(body, lang, known_length); + // next: letters themselves + body.append_know_length(str, known_length); + + ASSERT_STRING_INVARIANT(*this); + return *this; } -*/ -String& String::real_append(STRING_APPEND_PARAMS) { - if(!src) +String& String::append_help_length(const char* str, size_t helper_length, Language lang) { + if(!str) return *this; - if(!size) - size=strlen(src); - if(!size) + size_t known_length=helper_length?helper_length:strlen(str); + if(!known_length) return *this; - if(chunk_is_full()) - expand(); + return append_know_length(str, known_length, lang); +} +String& String::append_strdup(const char* str, size_t helper_length, Language lang) { + size_t known_length=helper_length?helper_length:strlen(str); + if(!known_length) + return *this; - append_here->item.ptr=src; - fsize+=append_here->item.size=size; - append_here->item.lang=lang; -#ifndef NO_STRING_ORIGIN - append_here->item.origin.file=file; - append_here->item.origin.line=line; -#endif - append_here++; fused_rows++; + // first: langs + langs.append(body, lang, known_length); + // next: letters themselves + body.append_strdup_know_length(str, known_length); + ASSERT_STRING_INVARIANT(*this); return *this; } -uint String::hash_code() const { - uint result=0; - - const Chunk *chunk=&head; - do { - const Chunk::Row *row=chunk->rows; - for(size_t i=0; icount; i++) { - if(row==append_here) - goto break2; - - result=Hash::generic_code(result, row->item.ptr, row->item.size); - row++; - } - chunk=row->link; - } while(chunk); -break2: +/// @todo check in doc: whether it documents NOW bad situation "abc".mid(-1, 3) =were?="ab" +String& String::mid(size_t substr_begin, size_t substr_end) const { + String& result=*new String; + + size_t self_length=length(); + substr_begin=min(substr_begin, self_length); + substr_end=min(max(substr_end, substr_begin), self_length); + size_t substr_length=substr_end-substr_begin; + if(!substr_length) + return result; + + // first: their langs + result.langs.append(result.body, langs, substr_begin, substr_length); + // next: letters themselves + result.body=body.mid(substr_begin, substr_length); + +// SAPI::log("piece of '%s' from %d to %d is '%s'", + //cstr(), substr_begin, substr_end, result.cstr()); + ASSERT_STRING_INVARIANT(result); return result; } -int String::cmp(int& partial, const String& src, size_t this_offset) const { - partial=-1; - this_offset=min(this_offset, size()-1); - - const Chunk *a_chunk=&head; - const Chunk *b_chunk=&src.head; - const Chunk::Row *a_row=a_chunk->rows; - const Chunk::Row *b_row=b_chunk->rows; - size_t a_offset=this_offset; - size_t b_offset=0; - Chunk::Row *a_end=append_here; - Chunk::Row *b_end=src.append_here; - size_t a_countdown=a_chunk->count; - size_t b_countdown=b_chunk->count; - bool a_break=false; - bool b_break=false; - size_t result; - for(size_t pos=0; true; pos+=a_row->item.size) { - a_break=a_row==a_end; - b_break=b_row==b_end; - if(a_break || b_break) - break; +size_t String::pos(const String::Body substr, size_t this_offset, Language lang) const { + size_t substr_length=substr.length(); + while(true) { + size_t substr_begin=body.pos(substr, this_offset); + + if(substr_begin==CORD_NOT_FOUND) + return STRING_NOT_FOUND; - if(pos+a_row->item.size > this_offset) { - int size_diff= - (a_row->item.size-a_offset)- - (b_row->item.size-b_offset); - - if(size_diff==0) { // a has same size as b - result=memcmp(a_row->item.ptr+a_offset, b_row->item.ptr+b_offset, a_row->item.size-a_offset); - if(result) - return result; - a_row++; a_countdown--; a_offset=0; - b_row++; b_countdown--; b_offset=0; - } else if (size_diff>0) { // a longer - result=memcmp(a_row->item.ptr+a_offset, b_row->item.ptr+b_offset, b_row->item.size-b_offset); - if(result) - return result; - a_offset+=b_row->item.size-b_offset; - b_row++; b_countdown--; b_offset=0; - } else { // b longer - result=memcmp(a_row->item.ptr+a_offset, b_row->item.ptr+b_offset, a_row->item.size-a_offset); - if(result) - return result; - b_offset+=a_row->item.size-a_offset; - a_row++; a_countdown--; a_offset=0; - } + if(langs.check_lang(lang, substr_begin, substr_length)) + return substr_begin; - if(!b_countdown) { - b_chunk=b_row->link; - b_row=b_chunk->rows; - b_countdown=b_chunk->count; - } - } else { - a_row++; a_countdown--; a_offset-=a_row->item.size; - } + this_offset=substr_begin+substr_length; + } +} - if(!a_countdown) { - a_chunk=a_row->link; - a_row=a_chunk->rows; - a_countdown=a_chunk->count; +size_t String::pos(const String& substr, + size_t this_offset, Language lang) const { + return pos(substr.body, this_offset, lang); +} + +void String::split(ArrayString& result, + size_t& pos_after, + const char* delim, + Language lang, int limit) const { + size_t self_length=length(); + if(size_t delim_length=strlen(delim)) { + int pos_before; + // while we have 'delim'... + for(; (pos_before=pos(delim, pos_after, lang))!=STRING_NOT_FOUND && limit; limit--) { + result+=&mid(pos_after, pos_before); + pos_after=pos_before+delim_length; } + // last piece + if(pos_afterrows; - size_t a_offset=this_offset; - size_t b_offset=0; - Chunk::Row *a_end=append_here; - size_t a_countdown=a_chunk->count; - bool a_break=false; - bool b_break=false; - for(size_t pos=0; true; pos+=a_row->item.size) { - a_break=a_row==a_end; - if(a_break || b_break) - break; +static void regex_options(const String* options, int *result, bool& need_pre_post_match){ + struct Regex_option { + const char* keyL; + const char* keyU; + int clear, set; + int *result; + bool *flag; + } regex_option[]={ + {"i", "I", 0, PCRE_CASELESS, result}, // a=A + {"s", "S", 0, PCRE_DOTALL, result}, // \n\n$ [default] + {"x", "U", 0, PCRE_EXTENDED, result}, // whitespace in regex ignored + {"m", "M", PCRE_DOTALL, PCRE_MULTILINE, result}, // ^aaa\n$^bbb\n$ + {"g", "G", 0, true, result+1}, // many rows + {"'", 0, 0, 0, 0, &need_pre_post_match}, + {0} + }; + result[0]=PCRE_EXTRA | PCRE_DOTALL | PCRE_DOLLAR_ENDONLY; + result[1]=0; + + if(options && !options->is_empty()) + for(Regex_option *o=regex_option; o->keyL; o++) + if(options->pos(o->keyL)!=STRING_NOT_FOUND + || (o->keyU && options->pos(o->keyU)!=STRING_NOT_FOUND)) { + if(o->flag) + *o->flag=true; + else { // result + *o->result &= ~o->clear; + *o->result |= o->set; + } + } +} + +Table* String::match(Charset& source_charset, + const String& regexp, + const String* options, + Row_action row_action, void *info, + bool& just_matched) const { + if(regexp.is_empty()) + throw Exception(0, + 0, + "regexp is empty"); - if(pos+a_row->item.size > this_offset) { - int size_diff= - (a_row->item.size-a_offset)- - (b_size-b_offset); - - if(size_diff==0) { // a has same size as b - if(size_t result=memcmp(a_row->item.ptr+a_offset, b_ptr+b_offset, - a_row->item.size-a_offset)!=0) - return result; - a_row++; a_countdown--; a_offset=0; - b_break=true; - } else if (size_diff>0) { // a longer - if(size_t result=memcmp(a_row->item.ptr+a_offset, b_ptr+b_offset, - b_size-b_offset)!=0) - return result; - a_offset+=b_size-b_offset; - b_break=true; - } else { // b longer - if(size_t result=memcmp(a_row->item.ptr+a_offset, b_ptr+b_offset, - a_row->item.size-a_offset)!=0) - return result; - b_offset+=a_row->item.size-a_offset; - a_row++; a_countdown--; a_offset=0; + const char* pattern=regexp.cstr(); + const char* errptr; + int erroffset; + bool need_pre_post_match=false; + int option_bits[2]={0}; regex_options(options, option_bits, need_pre_post_match); + bool global=option_bits[1]!=0; + pcre *code=pcre_compile(pattern, option_bits[0], + &errptr, &erroffset, + source_charset.pcre_tables); + + if(!code) + throw Exception(0, + ®exp.mid(erroffset, regexp.length()), + "regular expression syntax error - %s", errptr); + + int subpatterns=pcre_info(code, 0, 0); + if(subpatterns<0) { + pcre_free(code); + throw Exception(0, + ®exp, + "pcre_info error (%d)", + subpatterns); + } + + const char* subject=cstr(); + size_t subject_length=strlen(subject); + const int oveclength=(1/*match*/+MAX_MATCH_GROUPS)*3; + int ovector[oveclength]; + + // create table + Table::Action_options table_options; + Table& table=*new Table(string_match_table_template, table_options); + + int exec_option_bits=0; + int prestart=0; + int poststart=0; + int postfinish=length(); + while(true) { + int exec_substrings=pcre_exec(code, 0, + subject, subject_length, prestart, + exec_option_bits, ovector, oveclength); + + if(exec_substrings==PCRE_ERROR_NOMATCH) { + pcre_free(code); + row_action(table, 0/*last time, no raw*/, 0, 0, poststart, postfinish, info); + if(global || subpatterns) + return &table; // global or with subpatterns=true+result + else { + just_matched=false; return 0; // not global=no result } + } + + if(exec_substrings<0) { + pcre_free(code); + throw Exception(0, + ®exp, + "regular expression execute error (%d)", + exec_substrings); + } + + int prefinish=ovector[0]; + poststart=ovector[1]; + ArrayString* row=new ArrayString; + if(need_pre_post_match) { + *row+=&mid(0, prefinish); // .prematch column value + *row+=&mid(prefinish, poststart); // .match + *row+=&mid(poststart, postfinish); // .postmatch } else { - a_row++; a_countdown--; a_offset-=a_row->item.size; + *row+=&Empty; // .prematch column value + *row+=&Empty; // .match + *row+=&Empty; // .postmatch + } + + for(int i=1; ilink; - a_row=a_chunk->rows; - a_countdown=a_chunk->count; + if(!global || prestart==poststart) { // not global | going to hang + pcre_free(code); + row_action(table, 0/*last time, no row*/, 0, 0, poststart, postfinish, info); + return &table; } + prestart=poststart; + +/* + if(option_bits[0] & PCRE_MULTILINE) + exec_option_bits|=PCRE_NOTBOL; // start of subject+startoffset not BOL +*/ } - if(a_break==b_break) { // ended simultaneously - partial=0; return 0; - } else if(a_break) { // first bytes equal, but a ended before b - partial=1; return -1; +} + +String& String::change_case(Charset& source_charset, Change_case_kind kind) const { + String& result=*new String(); + if(is_empty()) + return result; + + char* new_cstr=cstrm(); + char *dest=new_cstr; + if(source_charset.isUTF8()) { + switch(kind) { + case CC_UPPER: + change_case_UTF8((const XMLByte*)new_cstr, (XMLByte*)new_cstr, UTF8CaseToUpper); + break; + case CC_LOWER: + change_case_UTF8((const XMLByte*)new_cstr, (XMLByte*)new_cstr, UTF8CaseToLower); + break; + default: + assert(!"unknown change case kind"); + break; // never + } + } else { - partial=2; return +1; + const unsigned char *tables=source_charset.pcre_tables; + + const unsigned char *a; + const unsigned char *b; + switch(kind) { + case CC_UPPER: + a=tables+lcc_offset; + b=tables+fcc_offset; + break; + case CC_LOWER: + a=tables+lcc_offset; + b=0; + break; + default: + assert(!"unknown change case kind"); + a=b=0; // calm, compiler + break; // never + } + + unsigned char index; + for(const char* current=new_cstr; index=(unsigned char)*current; current++) { + unsigned char c=a[index]; + if(b) + c=b[c]; + + *dest++=(char)c; + } } -} + result.langs=langs; + result.body=new_cstr; -#ifndef NO_STRING_ORIGIN -const Origin& String::origin() const { - if(!fused_rows) - THROW(0, 0, - 0, - "String::origin() of empty string called"); - - // determining origin by last appended piece - // because first one frequently constant. - // ex: ^load[/file] "document_root" + "/file" - return append_here[-1].item.origin; + return result; } -#endif -String& String::piece(size_t start, size_t finish) const { - start=max(0, start); - finish=min(size(), finish); - - String& result=*NEW String(pool()); - - size_t pos=0; - const Chunk *chunk=&head; - do { - const Chunk::Row *row=chunk->rows; - for(size_t i=0; icount; pos+=row->item.size, i++, row++) { - if(row==append_here) - goto break2; - - if(start>=pos) { // started now or already? - size_t item_finish=pos+row->item.size; - bool started=start < item_finish; // started now? - bool finished=finish < item_finish; // finished now? - size_t offset=started?start-pos:0; - size_t size=finished?finish-pos:row->item.size; - result.APPEND( - row->item.ptr+offset, size-offset, - row->item.lang, - row->item.origin.file, row->item.origin.line); - if(finished) - goto break2; +const String& String::replace(const Dictionary& dict) const { + String& result=*new String(); + const char* old_cstr=cstr(); + const char* prematch_begin=old_cstr; + + const char* current=old_cstr; + while(*current) { + if(Dictionary::Subst subst=dict.first_that_begins(current)) { + // prematch + if(size_t prematch_length=current-prematch_begin) { + result.langs.append(result.body, langs, prematch_begin-old_cstr, prematch_length); + result.body.append_strdup_know_length(prematch_begin, prematch_length); } - } - chunk=row->link; - } while(chunk); -break2: + + // match + // skip 'a' in 'current'; move prematch_begin + current+=subst.from_length; prematch_begin=current; + + if(const String* b=subst.to) // are there any b? + result<<*b; + } else // simply advance + current++; + } + + // postmatch + if(size_t postmatch_length=current-prematch_begin) { + result.langs.append(result.body, langs, prematch_begin-old_cstr, postmatch_length); + result.body.append_strdup_know_length(prematch_begin, postmatch_length); + } + + ASSERT_STRING_INVARIANT(result); return result; } -int String::pos(const String& substr, size_t result) const { - for(; result(cur); cur+=sizeof(size_t); + buf_length-=sizeof(size_t); - return -1; -} \ No newline at end of file + if(fragments_count) { + // 3: lang info + size_t total_length=0; + for(size_t f=0; f(cur); cur+=sizeof(char); + size_t fragment_length=*reinterpret_cast(cur); cur+=sizeof(size_t); + langs.append(total_length, lang, fragment_length); + total_length+=fragment_length; + + buf_length-=piece_length; + } + + // 4: letters + if(buf_length!=total_length) + return false; + + // serialize wrote extra zero byte there, we can rely on that + body=String::Body(cur, buf_length); + } + + ASSERT_STRING_INVARIANT(*this); + return true; +} + +const char* String::Body::v() const { + return CORD_to_const_char_star(body); +} +const char* String::Languages::v() const { + if(opt.is_not_just_lang) + return CORD_to_const_char_star(langs); + else + return (const char*)&langs; +} +const char* String::v() const { +#define LIMIT_VIEW 20 + char* buf=(char*)malloc(MAX_STRING); + const char*body_view=body.v(); + const char*langs_view=langs.v(); + snprintf(buf, MAX_STRING, + "%d:%.*s%s} " + "{%d:%s", + langs.count(), LIMIT_VIEW, langs_view, strlen(langs_view)>LIMIT_VIEW?"...":"", + strlen(body_view), body_view + ); + + return buf; +#undef LIMIT_VIEW +}