--- parser3/src/main/pa_string.C 2001/02/21 12:44:02 1.27 +++ parser3/src/main/pa_string.C 2003/03/20 11:11:55 1.172.2.21.2.7 @@ -1,99 +1,96 @@ -/* - $Id: pa_string.C,v 1.27 2001/02/21 12:44:02 paf Exp $ +/** @file + Parser: string class. @see untasize_t.C. + + Copyright (c) 2001-2003 ArtLebedev Group (http://www.artlebedev.com) + Author: Alexandr Petrosian (http://paf.design.ru) */ -#include +static const char* IDENT_STRING_C="$Date: 2003/03/20 11:11:55 $"; + +#include "pcre.h" -#include "pa_pool.h" #include "pa_string.h" -#include "pa_hash.h" #include "pa_exception.h" +#include "pa_table.h" +#include "pa_dictionary.h" +#include "pa_charset.h" + +// 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; i( - pool().malloc(sizeof(int)+sizeof(Chunk::Row)*curr_chunk_rows+sizeof(Chunk *))); - chunk->count=curr_chunk_rows; - link_row->link=chunk; - append_here=chunk->rows; - link_row=&chunk->rows[curr_chunk_rows]; - link_row->link=0; -} - -String::String(const String& src) : - Pooled(src.pool()) { - head.count=CR_PREALLOCATED_COUNT; - - int src_used_rows=src.used_rows(); - if(src_used_rows<=head.count) { - // all new rows fit into preallocated area - 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 into preallocated area: splitting into two chunks - // preallocated chunk src to constructing head - memcpy(head.rows, src.head.rows, sizeof(Chunk::Row)*head.count); - // remaining rows into new_chunk - curr_chunk_rows=src_used_rows-head.count; - Chunk *new_chunk=static_cast( - pool().malloc(sizeof(int)+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[curr_chunk_rows]; - - Chunk *old_chunk=src.head.preallocated_link; - Chunk::Row *new_rows=new_chunk->rows; - int rows_left_to_copy=curr_chunk_rows; - while(true) { - int 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; +// String::ArrayFragment methods + +String::ArrayFragment& String::ArrayFragment::append(const ArrayFragment& src, + size_t substr_start, size_t substr_end) { + if(substr_start==substr_end) + return *this; + + size_t fragment_start=0; + size_t fragment_end; + for(Array_iterator i(src); ; fragment_start=fragment_end) { + const element_type& fragment=i.next(); + fragment_end=fragment_start+fragment.size; + + if(substr_startfragment_end) { // are there more fragments including pieces of 'substr'? + const element_type& fragment=i.next(); + fragment_end=fragment_start+fragment.size; + + if(substr_end>fragment_end) // are there still more? + simple_append(Fragment(fragment.lang, fragment.size)); // appending whole fragment + else { // no, it was last + simple_append(Fragment(fragment.lang, substr_end-fragment_start)); + goto break2; } } - } - link_row->link=0; - fused_rows=src_used_rows; - fsize=src.fsize; + + break; + } +break2: + + return *this; } /* -String(const String_iterator& begin, const String_iterator& end) { - ;//TODO +void String::ArrayFragment::mid(ArrayFragment& result, size_t substr_start, size_t substr_end) { } */ +// String methods + +String::String(const char* src, size_t src_size, bool tainted): body(CORD_EMPTY) { + if(src) + if(tainted) + APPEND_TAINTED(src, src_size, 0, 0); + else + APPEND_CLEAN(src, src_size, 0, 0); +} + +String::String(const String& src): body(src.body), fragments(src.fragments) {} + String& String::real_append(STRING_APPEND_PARAMS) { if(!src) return *this; @@ -102,283 +99,477 @@ String& String::real_append(STRING_APPEN if(!size) return *this; - if(chunk_is_full()) - expand(); - - append_here->item.ptr=src; - fsize+=append_here->item.size=size; -#ifndef NO_STRING_ORIGIN - append_here->item.origin.file=file; - append_here->item.origin.line=line; -#endif - append_here++; fused_rows++; + body=CORD_cat_char_star(body, src, size); + fragments+=Fragment(lang, size); return *this; } -char *String::cstr() const { - char *result=static_cast(pool().malloc(size()+1)); - - char *copy_here=result; - const Chunk *chunk=&head; - do { - const Chunk::Row *row=chunk->rows; - for(int i=0; icount; i++) { - if(row==append_here) - goto break2; - - memcpy(copy_here, row->item.ptr, row->item.size); - copy_here+=row->item.size; - row++; - } - chunk=row->link; - } while(chunk); -break2: - *copy_here=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::hash_code() const { + uint result=0; + CORD_iter5(body, 0, 0, CORD_batched_iter_fn_generic_hash_code, &result); return result; } -uint String::hash_code() const { - uint result=0; +const String& String::mid(size_t substr_start, size_t substr_end) const { + String& result=*new String; - const Chunk *chunk=&head; - do { - const Chunk::Row *row=chunk->rows; - for(int i=0; icount; i++) { - if(row==append_here) - goto break2; + substr_start=min(substr_start, size()); + substr_end=max(substr_start, substr_end); + if(substr_start==substr_end) + return result; - result=Hash::generic_code(result, row->item.ptr, row->item.size); - row++; - } - chunk=row->link; - } while(chunk); -break2: + // first: letters themselves + result.body=CORD_substr(body, substr_start, substr_end-substr_start); + + // next: their langs + result.fragments.append(fragments, substr_start, substr_end); + +// SAPI::log(pool(), "piece of '%s' from %d to %d is '%s'", + //cstr(), substr_start, substr_end, result.cstr()); return result; } -bool String::operator == (const String& src) const { - if(size() != src.size()) - return false; - - 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; - int a_offset=0; - int b_offset=0; - Chunk::Row *a_end=append_here; - Chunk::Row *b_end=src.append_here; - int a_countdown=a_chunk->count; - int b_countdown=b_chunk->count; - bool a_break=false; - bool b_break=false; - while(true) { - 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 - if(memcmp(a_row->item.ptr+a_offset, b_row->item.ptr+b_offset, a_row->item.size-a_offset)!=0) - return false; - a_row++; a_countdown--; a_offset=0; - b_row++; b_countdown--; b_offset=0; - } else if (size_diff>0) { // a longer - if(memcmp(a_row->item.ptr+a_offset, b_row->item.ptr+b_offset, b_row->item.size-b_offset)!=0) - return false; - a_offset+=b_row->item.size-b_offset; - b_row++; b_countdown--; b_offset=0; - } else { // b longer - if(memcmp(a_row->item.ptr+a_offset, b_row->item.ptr+b_offset, a_row->item.size-a_offset)!=0) - return false; - b_offset+=a_row->item.size-a_offset; - a_row++; a_countdown--; a_offset=0; - } - - a_break=a_row==a_end; - b_break=b_row==b_end; - if(a_break || b_break) +size_t String::pos(CORD substr, + size_t this_offset, Untaint_lang lang) const { + // first: letters themselves + size_t substr_start=CORD_str(body, this_offset, substr); + if(substr_start==CORD_NOT_FOUND) + return STRING_NOT_FOUND; + + // next: check the lang when specified + + if(lang==UL_UNSPECIFIED) // ignore lang? + return substr_start; + + // substr must be in one fragment, and fragments' lang must = lang + size_t substr_end=substr_start+CORD_len(substr); + size_t fragment_start=0; + size_t fragment_end; + for(Array_iterator i(fragments); i.has_next(); fragment_start=fragment_end) { + const Fragment& fragment=i.next(); + fragment_end=fragment_start+fragment.size; + + if(substr_startfragment_end) // end of substr OUT of current fragment? break; - if(!a_countdown) { - a_chunk=a_row->link; - a_row=a_chunk->rows; - a_countdown=a_chunk->count; - } - if(!b_countdown) { - b_chunk=b_row->link; - b_row=b_chunk->rows; - b_countdown=b_chunk->count; - } + return fragment.lang==lang?substr_end:STRING_NOT_FOUND; + } + + return STRING_NOT_FOUND; +} + +size_t String::pos(const String& substr, + size_t this_offset, Untaint_lang lang) const { + return pos(substr.body, this_offset, lang); +} + +void String::split(ArrayString& result, + size_t& pos_after, + const char* delim, + Untaint_lang lang, int limit) { + size_t self_size=size(); + if(size_t delim_size=strlen(delim)) { + int pos_before; + // while we have 'delim'... + for(; (pos_before=pos(delim, pos_after, lang))>=0 && limit; limit--) { + result+=&mid(pos_after, pos_before); + pos_after=pos_before+delim_size; + } + // last piece + if(pos_after=0 && limit; limit--) { + result+=&mid(pos_after, pos_before); + pos_after=pos_before+delim.size(); + } + // last piece + if(pos_afterkeyL; o++) + if(options.pos(o->keyL)>=0 + || (o->keyU && options.pos(o->keyU)>=0)) { + if(o->flag) + *o->flag=true; + else { // result + *o->result &= ~o->clear; + *o->result |= o->set; + } + } +} - const Chunk *a_chunk=&head; - const Chunk::Row *a_row=a_chunk->rows; - int a_offset=0; - int b_offset=0; - Chunk::Row *a_end=append_here; - int a_countdown=a_chunk->count; - bool a_break=false; - bool b_break=false; +Table* String::match(Charset& source_charset, + const String& regexp, + const String& options, + Row_action row_action, void *info, + bool& just_matched) const { + if(!regexp) + throw Exception(0, + 0, + "regexp is empty"); + + const char* pattern=regexp.cstr(); + const char* errptr; + int erroffset; + bool need_pre_post_match=false; + int option_bits[2]; 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.size()), + "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_size=strlen(subject); + const int ovecsize=(1/*match*/+MAX_MATCH_GROUPS)*3; + int ovector[ovecsize]; + + // create table + Table* table=new Table(string_match_table_template); + + int exec_option_bits=0; + int prestart=0; + int poststart=0; + int postfinish=size(); while(true) { - int size_diff= - (a_row->item.size-a_offset)- - (b_size-b_offset); - - if(size_diff==0) { // a has same size as b - if(memcmp(a_row->item.ptr+a_offset, b_ptr+b_offset, a_row->item.size-a_offset)!=0) - return false; - a_row++; a_countdown--; a_offset=0; - b_break=true; - } else if (size_diff>0) { // a longer - if(memcmp(a_row->item.ptr+a_offset, b_ptr+b_offset, b_size-b_offset)!=0) - return false; - a_offset+=b_size-b_offset; - b_break=true; - } else { // b longer - if(memcmp(a_row->item.ptr+a_offset, b_ptr+b_offset, a_row->item.size-a_offset)!=0) - return false; - b_offset+=a_row->item.size-a_offset; - a_row++; a_countdown--; a_offset=0; + int exec_substrings=pcre_exec(code, 0, + subject, subject_size, prestart, + exec_option_bits, ovector, ovecsize); + + 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 + } } - a_break=a_row==a_end; - if(a_break || b_break) - break; - - if(!a_countdown) { - a_chunk=a_row->link; - a_row=a_chunk->rows; - a_countdown=a_chunk->count; + 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 { + *row+=0; // .prematch column value + *row+=0; // .match + *row+=0; // .postmatch + } + + for(int i=1; i(from)], type, to-from+1); -} + 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: + throw Exception(0, + this, + "unknown change case kind #%d", + static_cast(kind)); // never + a=b=0; // calm, compiler + break; // never + } + + char* new_cstr=cstrm(); + char *dest=new_cstr; + unsigned char index; + for(const char* current=new_cstr; index=(unsigned char)*current; current++) { + unsigned char c=a[index]; + if(b) + c=b[c]; -// String_iterator + *dest++=(char)c; + } + result.body=new_cstr; + result.fragments.append(fragments, 0, fragments.count()); -String_iterator::String_iterator(String& astring) : string(astring) { - read_here=string.head.rows; - position=string.size()==0?0:read_here->item.ptr; - link_row=reinterpret_cast(string.head.preallocated_link); -} -/* -String_iterator::String_iterator(String_iterator& asi) { - //TODO + return result; } -char String_iterator::operator()() const { - return position?*position:0; +const String& String::replace(const Dictionary& dict) const { + String& result=*new String(); + const char* old_cstr=cstr(); + const char* prematch_start=old_cstr; + + for(const char* current=old_cstr; *current; ) { + if(Table::element_type row=dict.first_that_starts(current)) { + // prematch + if(size_t prematch_size=current-prematch_start) { + result.body=CORD_cat_char_star(result.body, prematch_start, prematch_size); + result.fragments.append(fragments, prematch_start-old_cstr, prematch_size); + } + + // match + + const String* a=row->get(0); + // skip 'a' in 'current'; move prematch_start + current+=a->size(); prematch_start=current; + + if(row->count()>1) { // are there any b? + const String* b=row->get(1); + if(size_t b_size=b->size()) { + result.body=CORD_cat(result.body, b->cstr_to_cord()); + result.fragments.append(b->fragments, 0, b->fragments.count()); + } + } + } else // simply advance + current++; + } + + // postmatch + if(size_t prematch_size=current-prematch_start) { + result.body=CORD_cat_char_star(result.body, prematch_start, prematch_size); + result.fragments.append(fragments, prematch_start-old_cstr, prematch_size); + } + + return result; } -void String_iterator::skip() { - if(!position) - return; +double String::as_double() const { + double result; + const char *str=cstr(); + + while(*str && isspace(*str)) + str++; + if(!*str) + return 0; + + char *error_pos; + // 0xABC + if(str[0]=='0') + if(str[1]=='x' || str[1]=='X') + result=(double)(unsigned long)strtol(str, &error_pos, 0); + else + result=(double)strtod(str+1/*skip leading 0*/, &error_pos); + else + result=(double)strtod(str, &error_pos); + + while(char c=*error_pos++) + if(!isspace(c)) + throw Exception("number.format", + this, + "invalid number (double)"); - if(++position== - read_here->item.ptr+ - read_here->item.size) { + return result; +} +int String::as_int() const { + int result; + const char *str=cstr(); + + while(*str && isspace(*str)) + str++; + if(!*str) + return 0; + + char *error_pos; + // 0xABC + if(str[0]=='0') + if(str[1]=='x' || str[1]=='X') + result=(int)(unsigned long)strtol(str, &error_pos, 0); + else + result=(int)strtol(str+1/*skip leading 0*/, &error_pos, 0); + else + result=(int)strtol(str, &error_pos, 0); + + while(char c=*error_pos++) + if(!isspace(c)) + throw Exception("number.format", + this, + "invalid number (int)"); - // next row - if(++read_here==string.append_here) { - position=0; - return; - } - if(read_here==link_row) { - String::Chunk *chunk=link_row->link; - if(!chunk) - string.pool().exception().raise(0, 0, - &string, - "String_iterator::skip() missed " - "read_here==string.append_here check"); + return result; +} - read_here=chunk->rows; - link_row=&chunk->rows[chunk->count]; - } - position=read_here->item.ptr; +inline void uint2uchars(uint word, uchar *bytes) { + bytes[0]=word&0xFF; + bytes[1]=(word>>8)&0xFF; + bytes[2]=(word>>16)&0xFF; + bytes[3]=(word>>24)&0xFF; +} +inline uint uchars2uint(uchar *bytes) { + return bytes[3]<<24 + | bytes[2]<<16 + | bytes[1]<<8 + | bytes[0]; +} + +static int CORD_batched_iter_fn_append(const char* s, void* client_data) { + char*& cur=*static_cast(client_data); + + size_t size=strlen(s); + memcpy(cur, s, size); cur+=size; + return 0; +}; +String::C String::serialize(size_t prolog_size) const { + size_t buf_size= + prolog_size + +fragments.count()*(sizeof(Untaint_lang)+sizeof(size_t)) + +size(); + C result(new(PointerFreeGC) char[buf_size], buf_size); + + // 1: prolog + char *cur=result.str+prolog_size; + + + // 2: fragments.count + size_t fragments_count=fragments.count(); + memcpy(cur, &fragments_count, sizeof(fragments_count)); cur+=sizeof(fragments_count); + + // 3: lang info + for(Array_iterator i(fragments); i.has_next(); ) { + const Fragment& fragment=i.next(); + // lang + memcpy(cur, &fragment.lang, sizeof(fragment.lang)); cur+=sizeof(fragment.lang); + // size + memcpy(cur, &fragment.size, sizeof(fragment.size)); cur+=sizeof(fragment.size); } -} -bool String_iterator::skip_to(char c) { - if(!position) + // 4: letters + CORD_iter5(body, 0, 0, CORD_batched_iter_fn_append, &cur); + + return result; +} +bool String::deserialize(size_t prolog_size, void *buf, size_t buf_size, const char* file) { + if(buf_size<=prolog_size) return false; + buf_size-=prolog_size; - while(true) { - if(char *found=static_cast( - memchr(position, c, read_here->item.size-(position-read_here->item.ptr)))) { - position=found; - return true; - } + // 1: prolog + const char* cur=(const char* )buf+prolog_size; + + // 2: fragments.count + if(buf_size(cur); cur+=sizeof(size_t); + buf_size-=sizeof(size_t); - // next row - if(++read_here==string.append_here) { - position=0; + // 3: lang info + size_t total_size=0; + for(size_t f=0; flink; - if(!chunk) - string.pool().exception().raise(0, 0, - &string, - "String_iterator::skip_to(char) missed " - "read_here==string.append_here check"); - read_here=chunk->rows; - link_row=&chunk->rows[chunk->count]; - } - position=read_here->item.ptr; + Untaint_lang lang=*reinterpret_cast(cur); cur+=sizeof(Untaint_lang); + size_t fragment_size=*reinterpret_cast(cur); cur+=sizeof(size_t); + fragments+=Fragment(lang, fragment_size); + total_size+=fragment_size; + + buf_size-=piece_size; } -} -int String_iterator::skip_to(Char_types& types) { - if(!position) + // 4: letters + if(buf_size!=total_size) return false; - while(true) { - int countdown=read_here->item.size-(position-read_here->item.ptr); - for(; countdown--; position++) - if(int type=types.get(*position)) - return type; - - // next row - if(++read_here==string.append_here) { - position=0; - return -1; - } - if(read_here==link_row) { - String::Chunk *chunk=link_row->link; - if(!chunk) - string.pool().exception().raise(0, 0, - &string, - "String_iterator::skip_to(Char_type) missed " - "read_here==string.append_here check"); + body=CORD_cat_char_star(CORD_EMPTY, cur, buf_size); - read_here=chunk->rows; - link_row=&chunk->rows[chunk->count]; - } - position=read_here->item.ptr; - } + return true; } -*/ \ No newline at end of file