|
|
| version 1.80, 2001/04/26 11:37:40 | version 1.172.2.21.2.8, 2003/03/20 13:09:54 |
|---|---|
| Line 1 | Line 1 |
| /** @file | /** @file |
| Parser: string class. @see untasize_t.C. | Parser: string class. @see untasize_t.C. |
| Copyright (c) 2001 ArtLebedev Group (http://www.artlebedev.com) | Copyright (c) 2001-2003 ArtLebedev Group (http://www.artlebedev.com) |
| Author: Alexandr Petrosian <paf@design.ru> (http://paf.design.ru) | |
| Author: Alexander Petrosyan <paf@design.ru> (http://design.ru/paf) | |
| $Id$ | |
| */ | */ |
| #include "pa_config_includes.h" | static const char* IDENT_STRING_C="$Date$"; |
| #include "pcre.h" | #include "pcre.h" |
| #include "pa_pool.h" | |
| #include "pa_string.h" | #include "pa_string.h" |
| #include "pa_hash.h" | |
| #include "pa_exception.h" | #include "pa_exception.h" |
| #include "pa_common.h" | |
| #include "pa_array.h" | |
| #include "pa_globals.h" | |
| #include "pa_table.h" | #include "pa_table.h" |
| #include "pa_threads.h" | #include "pa_dictionary.h" |
| #include "pa_charset.h" | |
| //#include "pa_sapi.h" | |
| // String | |
| String::String(Pool& apool, const char *src, size_t src_size, bool tainted) : | // helpers |
| Pooled(apool) { | |
| last_chunk=&head; | |
| head.count=CR_PREALLOCATED_COUNT; | |
| append_here=head.rows; | |
| head.preallocated_link=0; | |
| link_row=&head.rows[head.count]; | |
| fused_rows=fsize=0; | |
| if(src) | /// String::match uses this as replace & global search table columns |
| if(tainted) | |
| APPEND_TAINTED(src, src_size, 0, 0); | |
| else | |
| APPEND_CLEAN(src, src_size, 0, 0); | |
| } | |
| void String::expand() { | const int MAX_MATCH_GROUPS=100; |
| size_t new_chunk_count=last_chunk->count+last_chunk->count*CR_GROW_PERCENT/100; | |
| last_chunk=static_cast<Chunk *>( | |
| 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::String(const String& src) : Pooled(src.pool()) { | class String_match_table_template_columns: public ArrayString { |
| head.count=CR_PREALLOCATED_COUNT; | public: |
| String_match_table_template_columns() { | |
| size_t src_used_rows=src.fused_rows; | *this+=new String("prematch"); |
| if(src_used_rows<=head.count) { | *this+=new String("match"); |
| // all new rows fit size_to preallocated area | *this+=new String("postmatch"); |
| size_t curr_chunk_rows=head.count; | for(int i=0; i<MAX_MATCH_GROUPS; i++) { |
| memcpy(head.rows, src.head.rows, sizeof(Chunk::Row)*src_used_rows); | char *cname=new(PointerFreeGC) char[3/*strlen("100")*/+1/*terminating 0*/]; |
| append_here=&head.rows[src_used_rows]; | *this+=new String(cname, sprintf(cname, "%d", 1+i)); |
| 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<Chunk *>( | |
| 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; | Table string_match_table_template(new String_match_table_template_columns); |
| } | |
| // 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; | |
| String& String::append(const String& src, Untaint_lang lang, bool forced) { | size_t fragment_start=0; |
| const Chunk *chunk=&src.head; | size_t fragment_end; |
| do { | for(Array_iterator<element_type> i(src); ; fragment_start=fragment_end) { |
| const Chunk::Row *row=chunk->rows; | const element_type& fragment=i.next(); |
| for(size_t i=0; i<chunk->count; i++, row++) { | fragment_end=fragment_start+fragment.size; |
| if(row==src.append_here) | |
| if(substr_start<fragment_start) // not reached fragments which may include 'substr'? | |
| continue; | |
| // found first fragment including piece of 'substr' | |
| size_t piece_end=min(substr_end, fragment_end); | |
| *this+=Fragment(fragment.lang, piece_end-substr_start); | |
| while(substr_end>fragment_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; | goto break2; |
| } | |
| APPEND(row->item.ptr, row->item.size, | |
| (lang!=UL_PASS_APPENDED && (row->item.lang==UL_TAINTED || forced))?lang:row->item.lang, | |
| row->item.origin.file, row->item.origin.line); | |
| } | } |
| chunk=row->link; | |
| } while(chunk); | break; |
| } | |
| break2: | break2: |
| return *this; | return *this; |
| } | } |
| /* | |
| 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); | |
| else | |
| APPEND_CLEAN(src, src_size); | |
| } | |
| String::String(const String& src): body(src.body), fragments(src.fragments) {} | |
| String& String::real_append(STRING_APPEND_PARAMS) { | String& String::append(const char* src, size_t size, Untaint_lang lang) { |
| if(!src) | if(!src) |
| return *this; | return *this; |
| if(!size) | if(!size) |
| Line 136 String& String::real_append(STRING_APPEN | Line 99 String& String::real_append(STRING_APPEN |
| if(!size) | if(!size) |
| return *this; | return *this; |
| if(chunk_is_full()) | body=CORD_cat_char_star(body, src, size); |
| expand(); | fragments+=Fragment(lang, size); |
| 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++; | |
| return *this; | return *this; |
| } | } |
| 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::hash_code() const { | uint String::hash_code() const { |
| uint result=0; | uint result=0; |
| CORD_iter5(body, 0, 0, CORD_batched_iter_fn_generic_hash_code, &result); | |
| const Chunk *chunk=&head; | |
| do { | |
| const Chunk::Row *row=chunk->rows; | |
| for(size_t i=0; i<chunk->count; 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: | |
| return result; | return result; |
| } | } |
| /// @todo move 'lang' skipping to pos | const String& String::mid(size_t substr_start, size_t substr_end) const { |
| int String::cmp(int& partial, const String& src, | String& result=*new String; |
| size_t this_offset, Untaint_lang lang) 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; | |
| size_t pos=0; | |
| while(true) { | |
| a_break=a_row==a_end; | |
| b_break=b_row==b_end; | |
| if(a_break || b_break) | |
| break; | |
| if(pos+a_row->item.size > this_offset) { | |
| if(lang!=UL_UNSPECIFIED && a_row->item.lang!=lang) | |
| return -1; // wrong lang -- bail out | |
| 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; | |
| pos+=a_row->item.size; | |
| 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; | |
| pos+=a_row->item.size; | |
| a_row++; a_countdown--; a_offset=0; | |
| } | |
| if(!b_countdown) { | |
| b_chunk=b_row->link; | |
| b_row=b_chunk->rows; | |
| b_countdown=b_chunk->count; | |
| } | |
| } else { | |
| a_offset-=a_row->item.size; | |
| pos+=a_row->item.size; | |
| a_row++; a_countdown--; | |
| } | |
| if(!a_countdown) { | |
| a_chunk=a_row->link; | |
| a_row=a_chunk->rows; | |
| a_countdown=a_chunk->count; | |
| } | |
| } | |
| 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; | |
| } else { | |
| partial=2; return +1; | |
| } | |
| } | |
| /// @todo move 'lang' skipping to pos | |
| int String::cmp(int& partial, const char* b_ptr, size_t src_size, | |
| size_t this_offset, Untaint_lang lang) const { | |
| partial=-1; | |
| size_t b_size=src_size?src_size:b_ptr?strlen(b_ptr):0; | |
| this_offset=min(this_offset, size()-1); | |
| const Chunk *a_chunk=&head; | |
| const Chunk::Row *a_row=a_chunk->rows; | |
| 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; | |
| size_t pos=0; | |
| while(true) { | |
| a_break=a_row==a_end; | |
| if(a_break || b_break) | |
| break; | |
| if(pos+a_row->item.size > this_offset) { | substr_start=min(substr_start, size()); |
| if(lang!=UL_UNSPECIFIED && a_row->item.lang!=lang) | substr_end=max(substr_start, substr_end); |
| return -1; // wrong lang -- bail out | if(substr_start==substr_end) |
| return result; | |
| 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; | |
| pos+=a_row->item.size; | |
| 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; | |
| pos+=a_row->item.size; | |
| a_row++; a_countdown--; a_offset=0; | |
| } | |
| } else { | |
| a_offset-=a_row->item.size; | |
| pos+=a_row->item.size; | |
| a_row++; a_countdown--; | |
| } | |
| if(!a_countdown) { | // first: letters themselves |
| a_chunk=a_row->link; | result.body=CORD_substr(body, substr_start, substr_end-substr_start); |
| a_row=a_chunk->rows; | |
| a_countdown=a_chunk->count; | |
| } | |
| } | |
| 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; | |
| } else { | |
| partial=2; return +1; | |
| } | |
| } | |
| #ifndef NO_STRING_ORIGIN | // next: their langs |
| const Origin& String::origin() const { | result.fragments.append(fragments, substr_start, substr_end); |
| 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" | |
| // when last peice is constant, | |
| // ex: parser_root_auto_path{dynamic} / auto.p{const} | |
| // using first piece | |
| Origin& last_origin=append_here[-1].item.origin; | |
| return last_origin.file ? last_origin : head.rows[0].item.origin; | |
| } | |
| #endif | |
| String& String::mid(size_t start, size_t finish) const { | |
| start=max(0, start); | |
| finish=min(size(), finish); | |
| if(start==finish) | |
| return *empty_string; | |
| 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; i<chunk->count; pos+=row->item.size, i++, row++) { | |
| if(row==append_here) | |
| goto break2; | |
| size_t item_finish=pos+row->item.size; | |
| if(item_finish > start) { // started now or already? | |
| bool started=result.size()==0; // 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; | |
| } | |
| } | |
| chunk=row->link; | |
| } while(chunk); | |
| break2: | |
| // SAPI::log(pool(), "piece of '%s' from %d to %d is '%s'", | // SAPI::log(pool(), "piece of '%s' from %d to %d is '%s'", |
| //cstr(), start, finish, result.cstr()); | //cstr(), substr_start, substr_end, result.cstr()); |
| return result; | return result; |
| } | } |
| int String::pos(const String& substr, | size_t String::pos(CORD substr, |
| size_t result, Untaint_lang lang) const { | size_t this_offset, Untaint_lang lang) const { |
| for(; result<size(); result++) { | // first: letters themselves |
| int partial; cmp(partial, substr, result, lang); | size_t substr_start=CORD_str(body, this_offset, substr); |
| if( | if(substr_start==CORD_NOT_FOUND) |
| partial==0 || // full match | return STRING_NOT_FOUND; |
| partial==2) // 'substr' starts 'this'+'result' | |
| return result; | // next: check the lang when specified |
| } | |
| if(lang==UL_UNSPECIFIED) // ignore lang? | |
| return -1; | 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<ArrayFragment::element_type> i(fragments); i.has_next(); fragment_start=fragment_end) { | |
| const Fragment& fragment=i.next(); | |
| fragment_end=fragment_start+fragment.size; | |
| if(substr_start<fragment_start) // not reached fragments which may include 'result'? | |
| continue; | |
| if(substr_end>fragment_end) // end of substr OUT of current fragment? | |
| break; | |
| return fragment.lang==lang?substr_end:STRING_NOT_FOUND; | |
| } | |
| return STRING_NOT_FOUND; | |
| } | } |
| int String::pos(const char *substr, size_t substr_size, | size_t String::pos(const String& substr, |
| size_t result, Untaint_lang lang) const { | size_t this_offset, Untaint_lang lang) const { |
| for(; result<size(); result++) { | return pos(substr.body, this_offset, lang); |
| int partial; cmp(partial, substr, substr_size, result, lang); | |
| if( | |
| partial==0 || // full match | |
| partial==2) // 'substr' starts 'this'+'result' | |
| return result; | |
| } | |
| return -1; | |
| } | } |
| void String::split(Array& result, | void String::split(ArrayString& result, |
| size_t* pos_after_ref, | size_t& pos_after, |
| const char *delim, size_t delim_size, | const char* delim, |
| Untaint_lang lang, int limit) const { | Untaint_lang lang, int limit) { |
| if(delim_size) { | size_t self_size=size(); |
| size_t pos_after=pos_after_ref?*pos_after_ref:0; | if(size_t delim_size=strlen(delim)) { |
| int pos_before; | int pos_before; |
| // while we have 'delim'... | // while we have 'delim'... |
| for(; (pos_before=pos(delim, delim_size, pos_after, lang))>=0 && limit; limit--) { | for(; (pos_before=pos(delim, pos_after, lang))>=0 && limit; limit--) { |
| result+=&mid(pos_after, pos_before); | result+=&mid(pos_after, pos_before); |
| pos_after=pos_before+delim_size; | pos_after=pos_before+delim_size; |
| } | } |
| // last piece | // last piece |
| if(pos_after<size() && limit) { | if(pos_after<self_size && limit) { |
| result+=&mid(pos_after, size()); | result+=&mid(pos_after, self_size); |
| pos_after=size(); | pos_after=self_size; |
| } | } |
| if(pos_after_ref) | |
| *pos_after_ref=pos_after; | |
| } else { // empty delim | } else { // empty delim |
| result+=this; | result+=this; |
| if(pos_after_ref) | pos_after+=self_size; |
| *pos_after_ref+=size(); | |
| } | } |
| } | } |
| void String::split(Array& result, | void String::split(ArrayString& result, |
| size_t* pos_after_ref, | size_t& pos_after, |
| const String& delim, Untaint_lang lang, | const String& delim, Untaint_lang lang, |
| int limit) const { | int limit) const { |
| if(delim.size()) { | if(delim) { |
| size_t pos_after=pos_after_ref?*pos_after_ref:0; | |
| int pos_before; | int pos_before; |
| // while we have 'delim'... | // while we have 'delim'... |
| for(; (pos_before=pos(delim, pos_after, lang))>=0 && limit; limit--) { | for(; (pos_before=pos(delim, pos_after, lang))>=0 && limit; limit--) { |
| Line 451 void String::split(Array& result, | Line 211 void String::split(Array& result, |
| result+=&mid(pos_after, size()); | result+=&mid(pos_after, size()); |
| pos_after=size(); | pos_after=size(); |
| } | } |
| if(pos_after_ref) | |
| *pos_after_ref=pos_after; | |
| } else { // empty delim | } else { // empty delim |
| result+=this; | result+=this; |
| if(pos_after_ref) | pos_after+=size(); |
| *pos_after_ref+=size(); | |
| } | } |
| } | } |
| static void regex_options(char *options, int *result){ | static void regex_options(const String& options, int *result, bool& need_pre_post_match){ |
| struct Regex_option { | struct Regex_option { |
| char key; | const char* keyL; |
| const char* keyU; | |
| int clear, set; | int clear, set; |
| int *result; | int *result; |
| bool *flag; | |
| } regex_option[]={ | } regex_option[]={ |
| {'i', 0, PCRE_CASELESS, result}, // a=A | {"i", "I", 0, PCRE_CASELESS, result}, // a=A |
| {'s', 0, PCRE_DOTALL, result}, // \n\n$ [default] | {"s", "S", 0, PCRE_DOTALL, result}, // \n\n$ [default] |
| {'x', 0, PCRE_EXTENDED, result}, // whitespace in regex ignored | {"x", "U", 0, PCRE_EXTENDED, result}, // whitespace in regex ignored |
| {'m', PCRE_DOTALL, PCRE_MULTILINE, result}, // ^aaa\n$^bbb\n$ | {"m", "M", PCRE_DOTALL, PCRE_MULTILINE, result}, // ^aaa\n$^bbb\n$ |
| {'g', 0, true, result+1}, // many rows | {"g", "G", 0, true, result+1}, // many rows |
| {0}, | {"'", 0, 0, 0, 0, &need_pre_post_match}, |
| {0} | |
| }; | }; |
| result[0]=PCRE_EXTRA | PCRE_DOTALL; | result[0]=PCRE_EXTRA | PCRE_DOTALL | PCRE_DOLLAR_ENDONLY; |
| result[1]=0; | result[1]=0; |
| if(options) | if(options) |
| for(Regex_option *o=regex_option; o->key; o++) | for(Regex_option *o=regex_option; o->keyL; o++) |
| if( | if(options.pos(o->keyL)>=0 |
| strchr(options, o->key) || | || (o->keyU && options.pos(o->keyU)>=0)) { |
| strchr(options, toupper(o->key))) { | if(o->flag) |
| *(o->result)&=~o->clear; | *o->flag=true; |
| *(o->result)|=o->set; | else { // result |
| *o->result &= ~o->clear; | |
| *o->result |= o->set; | |
| } | |
| } | } |
| } | } |
| /** | Table* String::match(Charset& source_charset, |
| returns true if fills table. | const String& regexp, |
| table format is defined and fixed[can be used by others]: | const String& options, |
| @verbatim | Row_action row_action, void *info, |
| pre-match/match/post-match/1/2/3/... | bool& just_matched) const { |
| @endverbatim | if(!regexp) |
| */ | throw Exception(0, |
| bool String::match(const unsigned char *pcre_tables, | 0, |
| const String *aorigin, | |
| const String& regexp, | |
| const String *options, | |
| Table **table, | |
| Row_action row_action, void *info) const { | |
| if(!regexp.size()) | |
| THROW(0, 0, | |
| aorigin, | |
| "regexp is empty"); | "regexp is empty"); |
| const char *pattern=regexp.cstr(UL_AS_IS); | |
| const char *errptr; | const char* pattern=regexp.cstr(); |
| const char* errptr; | |
| int erroffset; | int erroffset; |
| int option_bits[2]; regex_options(options?options->cstr():0, option_bits); | 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], | pcre *code=pcre_compile(pattern, option_bits[0], |
| &errptr, &erroffset, | &errptr, &erroffset, |
| pcre_tables); | source_charset.pcre_tables); |
| if(!code) | if(!code) |
| THROW(0, 0, | throw Exception(0, |
| ®exp.mid(erroffset, regexp.size()), | ®exp.mid(erroffset, regexp.size()), |
| "regular expression syntax error - %s", errptr); | "regular expression syntax error - %s", errptr); |
| int info_substrings=pcre_info(code, 0, 0); | int subpatterns=pcre_info(code, 0, 0); |
| if(info_substrings<0) { | if(subpatterns<0) { |
| (*pcre_free)(code); | pcre_free(code); |
| THROW(0, 0, | throw Exception(0, |
| aorigin, | ®exp, |
| "pcre_info error (%d)", | "pcre_info error (%d)", |
| info_substrings); | subpatterns); |
| } | } |
| int startoffset=0; | const char* subject=cstr(); |
| const char *subject=cstr(UL_AS_IS); | size_t subject_size=strlen(subject); |
| int length=strlen(subject); | const int ovecsize=(1/*match*/+MAX_MATCH_GROUPS)*3; |
| int ovecsize; | int ovector[ovecsize]; |
| int *ovector=(int *)malloc(sizeof(int)* | |
| (ovecsize=(1/*match*/+info_substrings)*3)); | // create table |
| Table* table=new Table(string_match_table_template); | |
| { // create table | |
| Array& columns=*NEW Array(pool()); | |
| columns+=string_pre_match_name; | |
| columns+=string_match_name; | |
| columns+=string_post_match_name; | |
| for(int i=1; i<=info_substrings; i++) { | |
| char *column=(char *)malloc(MAX_NUMBER); | |
| snprintf(column, MAX_NUMBER, "%d", i); | |
| columns+=NEW String(pool(), column); // .i column name | |
| } | |
| *table=NEW Table(pool(), aorigin, &columns); | |
| } | |
| int exec_option_bits=0; | int exec_option_bits=0; |
| int prestart=0; | |
| int poststart=0; | |
| int postfinish=size(); | |
| while(true) { | while(true) { |
| int exec_substrings=pcre_exec(code, 0, | int exec_substrings=pcre_exec(code, 0, |
| subject, length, startoffset, | subject, subject_size, prestart, |
| exec_option_bits, ovector, ovecsize); | exec_option_bits, ovector, ovecsize); |
| if(exec_substrings==PCRE_ERROR_NOMATCH) { | if(exec_substrings==PCRE_ERROR_NOMATCH) { |
| (*pcre_free)(code); | pcre_free(code); |
| (*row_action)(**table, 0/*last time, no row*/, 0, 0, info); | row_action(table, 0/*last time, no raw*/, 0, 0, poststart, postfinish, info); |
| return option_bits[1]!=0; // global=true+table, not global=false | 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) { | if(exec_substrings<0) { |
| (*pcre_free)(code); | pcre_free(code); |
| THROW(0, 0, | throw Exception(0, |
| aorigin, | ®exp, |
| "regular expression execute error (%d)", | "regular expression execute error (%d)", |
| exec_substrings); | exec_substrings); |
| } | } |
| Array& row=*NEW Array(pool()); | int prefinish=ovector[0]; |
| row+=&mid(0, ovector[0]); // .pre-match column value | poststart=ovector[1]; |
| row+=&mid(ovector[0], ovector[1]); // .match | ArrayString* row=new ArrayString; |
| row+=&mid(ovector[1], size()); // .post-match | 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++) { | for(int i=1; i<exec_substrings; i++) { |
| // -1:-1 case handled peacefully by mid() itself | // -1:-1 case handled peacefully by mid() itself |
| row+=&mid(ovector[i*2+0], ovector[i*2+1]); // .i column value | *row+=&mid(ovector[i*2+0], ovector[i*2+1]); // .i column value |
| } | } |
| (*row_action)(**table, &row, startoffset, ovector[0], info); | row_action(table, row, prestart, prefinish, poststart, postfinish, info); |
| if(!option_bits[1] || !(startoffset=ovector[1])) { // not global | going to hang | if(!global || prestart==poststart) { // not global | going to hang |
| (*pcre_free)(code); | pcre_free(code); |
| (*row_action)(**table, 0/*last time, no row*/, 0, 0, info); | row_action(table, 0/*last time, no row*/, 0, 0, poststart, postfinish, info); |
| return true; | return table; |
| } | } |
| prestart=poststart; | |
| /* | /* |
| if(option_bits[0] & PCRE_MULTILINE) | if(option_bits[0] & PCRE_MULTILINE) |
| Line 590 bool String::match(const unsigned char * | Line 351 bool String::match(const unsigned char * |
| */ | */ |
| } | } |
| } | } |
| String& String::change_case(Charset& source_charset, Change_case_kind kind) { | |
| String& result=*new String(); | |
| 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: | |
| throw Exception(0, | |
| this, | |
| "unknown change case kind #%d", | |
| static_cast<int>(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]; | |
| *dest++=(char)c; | |
| } | |
| result.body=new_cstr; | |
| result.fragments.append(fragments, 0, fragments.count()); | |
| return result; | |
| } | |
| 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; | |
| } | |
| 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; | |
| } | |
| 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<char**>(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<ArrayFragment::element_type> 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); | |
| } | |
| // 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; | |
| // 1: prolog | |
| const char* cur=(const char* )buf+prolog_size; | |
| // 2: fragments.count | |
| if(buf_size<sizeof(size_t)) // fragments.count don't fit? | |
| return false; | |
| size_t fragments_count=*reinterpret_cast<const size_t*>(cur); cur+=sizeof(size_t); | |
| buf_size-=sizeof(size_t); | |
| // 3: lang info | |
| size_t total_size=0; | |
| for(size_t f=0; f<fragments_count; f++) { | |
| size_t piece_size=sizeof(Untaint_lang)+sizeof(size_t); | |
| if(buf_size<piece_size) // lang+size | |
| return false; | |
| Untaint_lang lang=*reinterpret_cast<const Untaint_lang *>(cur); cur+=sizeof(Untaint_lang); | |
| size_t fragment_size=*reinterpret_cast<const size_t*>(cur); cur+=sizeof(size_t); | |
| fragments+=Fragment(lang, fragment_size); | |
| total_size+=fragment_size; | |
| buf_size-=piece_size; | |
| } | |
| // 4: letters | |
| if(buf_size!=total_size) | |
| return false; | |
| body=CORD_cat_char_star(CORD_EMPTY, cur, buf_size); | |
| return true; | |
| } |