|
|
| version 1.2, 2001/01/26 15:28:39 | version 1.184, 2003/10/03 09:40:57 |
|---|---|
| Line 1 | Line 1 |
| #include <string.h> | /** @file |
| Parser: string class. @see untalength_t.C. | |
| #include "pa_pool.h" | Copyright (c) 2001-2003 ArtLebedev Group (http://www.artlebedev.com) |
| Author: Alexandr Petrosian <paf@design.ru> (http://paf.design.ru) | |
| */ | |
| static const char* IDENT_STRING_C="$Date$"; | |
| #include "pcre.h" | |
| #include "pa_string.h" | |
| #include "pa_exception.h" | |
| #include "pa_table.h" | |
| #include "pa_dictionary.h" | |
| #include "pa_charset.h" | |
| // 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'*/); | |
| } | |
| void *String::operator new(size_t size, Pool *apool) { | static int CORD_block_count_proc(char c, size_t size, void* client_data) |
| return apool->alloc(size); | { |
| 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; | |
| } | } |
| void String::construct(Pool *apool) { | // helpers |
| pool=apool; | |
| head.count=curr_chunk_rows=CR_PREALLOCATED_COUNT; | /// String::match uses this as replace & global search table columns |
| append_here=head.first; | |
| head.preallocated_link=0; | const int MAX_MATCH_GROUPS=100; |
| link_row=&head.first[curr_chunk_rows]; | |
| 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<MAX_MATCH_GROUPS; i++) { | |
| *this+=new String(String::Body::Format(1+i), String::L_CLEAN); | |
| } | |
| } | |
| }; | |
| Table string_match_table_template(new String_match_table_template_columns); | |
| // String::Body methods | |
| 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); | |
| } | } |
| void String::expand() { | static int CORD_batched_iter_fn_generic_hash_code(char c, void * client_data) { |
| curr_chunk_rows=curr_chunk_rows*100/CR_GROW_PERCENT; | uint& result=*static_cast<uint*>(client_data); |
| Chunk *chunk=static_cast<Chunk *>( | generic_hash_code(result, c); |
| pool->calloc(sizeof(Chunk::Row)*curr_chunk_rows+sizeof(Chunk *))); | return 0; |
| chunk->count=curr_chunk_rows; | |
| link_row->link=chunk; | |
| append_here=chunk->first; | |
| link_row=&chunk->first[curr_chunk_rows]; | |
| } | } |
| static int CORD_batched_iter_fn_generic_hash_code(const char* s, void * client_data) { | |
| uint& result=*static_cast<uint*>(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 methods | |
| String& String::operator += (char *src) { | String::String(const char* cstr, size_t helper_length, bool tainted): body(CORD_EMPTY) { |
| if(chunk_is_full()) | append_help_length(cstr, helper_length, tainted?L_TAINTED:L_CLEAN); |
| expand(); | } |
| String::String(const String::C cstr, bool tainted): body(CORD_EMPTY) { | |
| append_know_length(cstr.str, cstr.length, tainted?L_TAINTED:L_CLEAN); | |
| } | |
| 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::append_help_length(const char* str, size_t helper_length, Language lang) { | |
| if(!str) | |
| return *this; | |
| size_t known_length=helper_length?helper_length:strlen(str); | |
| if(!known_length) | |
| return *this; | |
| append_here->item.ptr=src; | return append_know_length(str, known_length, lang); |
| append_here->item.size=strlen(src); | } |
| append_here++; | 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; | |
| // 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; | return *this; |
| } | } |
| size_t String::size() { | /// @todo check in doc: whether it documents NOW bad situation "abc".mid(-1, 3) =were?="ab" |
| int result=0; | String& String::mid(size_t substr_begin, size_t substr_end) const { |
| Chunk *chunk=&head; | String& result=*new String; |
| do { | |
| Chunk::Row *row=chunk->first; | size_t self_length=length(); |
| for(int i=0; i<chunk->count; i++) { | substr_begin=min(substr_begin, self_length); |
| if(row==append_here) | substr_end=min(max(substr_end, substr_begin), self_length); |
| goto break2; | size_t substr_length=substr_end-substr_begin; |
| if(!substr_length) | |
| result+=row->item.size; | return result; |
| row++; | |
| } | // first: their langs |
| chunk=row->link; | result.langs.append(result.body, langs, substr_begin, substr_length); |
| } while(chunk); | // next: letters themselves |
| break2: | 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; | |
| } | |
| 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(langs.check_lang(lang, substr_begin, substr_length)) | |
| return substr_begin; | |
| this_offset=substr_begin+substr_length; | |
| } | |
| } | |
| 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_after<self_length && limit) { | |
| result+=&mid(pos_after, self_length); | |
| pos_after=self_length; | |
| } | |
| } else { // empty delim | |
| result+=this; | |
| pos_after+=self_length; | |
| } | |
| } | |
| void String::split(ArrayString& result, | |
| size_t& pos_after, | |
| const String& delim, Language lang, | |
| int limit) const { | |
| if(!delim.is_empty()) { | |
| 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_after<length() && limit) { | |
| result+=&mid(pos_after, length()); | |
| pos_after=length(); | |
| } | |
| } else { // empty delim | |
| result+=this; | |
| pos_after+=length(); | |
| } | |
| } | |
| 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"); | |
| 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 { | |
| *row+=0; // .prematch column value | |
| *row+=0; // .match | |
| *row+=0; // .postmatch | |
| } | |
| for(int i=1; i<exec_substrings; i++) { | |
| // -1:-1 case handled peacefully by mid() itself | |
| *row+=&mid(ovector[i*2+0], ovector[i*2+1]); // .i column value | |
| } | |
| row_action(table, row, prestart, prefinish, poststart, postfinish, info); | |
| 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 | |
| */ | |
| } | |
| } | |
| 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 { | |
| 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; | |
| return result; | |
| } | |
| 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); | |
| } | |
| // 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; | |
| } | |
| 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)"); | |
| 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)"); | |
| return result; | return result; |
| } | } |
| char *String::c_str() { | static int serialize_body_char(char c, char** cur) { |
| char *result=static_cast<char *>(pool->alloc(size()+1)); | *((*cur)++)=c; |
| return 0; // 0=continue | |
| char *copy_here=result; | }; |
| Chunk *chunk=&head; | static int serialize_body_piece(const char* s, char** cur) { |
| do { | size_t length=strlen(s); |
| Chunk::Row *row=chunk->first; | memcpy(*cur, s, length); *cur+=length; |
| for(int i=0; i<chunk->count; i++) { | return 0; // 0=continue |
| if(row==append_here) | }; |
| goto break2; | static int serialize_lang_piece(char alang, size_t asize, char** cur) { |
| // lang | |
| memcpy(copy_here, row->item.ptr, row->item.size); | memcpy(*cur, &alang, sizeof(alang)); *cur+=sizeof(alang); |
| copy_here+=row->item.size; | // length |
| row++; | memcpy(*cur, &asize, sizeof(asize)); *cur+=sizeof(asize); |
| } | |
| chunk=row->link; | return 0; // 0=continue |
| } while(chunk); | } |
| break2: | String::Cm String::serialize(size_t prolog_length) const { |
| *copy_here=0; | size_t fragments_count=langs.count(); |
| size_t buf_length= | |
| prolog_length //1 | |
| +sizeof(size_t) //2 | |
| +fragments_count*(sizeof(char)+sizeof(size_t)) //3 | |
| +body.length() //4 | |
| +1; // for zero terminator used in deserialize | |
| String::Cm result(new(PointerFreeGC) char[buf_length], buf_length); | |
| // 1: prolog | |
| char *cur=result.str+prolog_length; | |
| // 2: langs.count | |
| memcpy(cur, &fragments_count, sizeof(fragments_count)); cur+=sizeof(fragments_count); | |
| // 3: lang info | |
| langs.for_each(body, serialize_lang_piece, &cur); | |
| // 4: letters | |
| body.for_each(serialize_body_char, serialize_body_piece, &cur); | |
| // 5: zero terminator | |
| *cur=0; | |
| return result; | return result; |
| } | } |
| bool String::deserialize(size_t prolog_length, void *buf, size_t buf_length) { | |
| if(buf_length<=prolog_length) | |
| return false; | |
| buf_length-=prolog_length; | |
| buf_length-=1; // 5: zero terminator | |
| // 1: prolog | |
| const char* cur=(const char* )buf+prolog_length; | |
| // 2: langs.count | |
| if(buf_length<sizeof(size_t)) // langs.count don't fit? | |
| return false; | |
| size_t fragments_count=*reinterpret_cast<const size_t*>(cur); cur+=sizeof(size_t); | |
| buf_length-=sizeof(size_t); | |
| if(fragments_count) { | |
| // 3: lang info | |
| size_t total_length=0; | |
| for(size_t f=0; f<fragments_count; f++) { | |
| size_t piece_length=sizeof(char)+sizeof(size_t); | |
| if(buf_length<piece_length) // lang+length | |
| return false; | |
| Language lang=*reinterpret_cast<const Language *>(cur); cur+=sizeof(char); | |
| size_t fragment_length=*reinterpret_cast<const size_t*>(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 | |
| } |