|
|
| version 1.2, 2001/01/26 15:28:39 | version 1.35, 2001/03/10 12:12:51 |
|---|---|
| Line 1 | Line 1 |
| /* | |
| $Id$ | |
| */ | |
| #include <string.h> | #include <string.h> |
| #include "pa_pool.h" | #include "pa_pool.h" |
| #include "pa_string.h" | |
| #include "pa_hash.h" | |
| #include "pa_exception.h" | |
| // String | |
| String::String(Pool& apool) : | |
| 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; | |
| } | |
| void *String::operator new(size_t size, Pool *apool) { | void String::expand() { |
| return apool->alloc(size); | int new_chunk_count=last_chunk->count+last_chunk->count*CR_GROW_PERCENT/100; |
| last_chunk=static_cast<Chunk *>( | |
| malloc(sizeof(int)+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; | |
| } | } |
| void String::construct(Pool *apool) { | String::String(const String& src) : |
| pool=apool; | Pooled(src.pool()) { |
| head.count=curr_chunk_rows=CR_PREALLOCATED_COUNT; | head.count=CR_PREALLOCATED_COUNT; |
| append_here=head.first; | |
| head.preallocated_link=0; | int src_used_rows=src.used_rows(); |
| link_row=&head.first[curr_chunk_rows]; | if(src_used_rows<=head.count) { |
| // all new rows fit into preallocated area | |
| int 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 | |
| int curr_chunk_rows=src_used_rows-head.count; | |
| Chunk *new_chunk=static_cast<Chunk *>( | |
| 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[new_chunk->count]; | |
| Chunk *old_chunk=src.head.preallocated_link; | |
| Chunk::Row *new_rows=new_chunk->rows; | |
| int rows_left_to_copy=new_chunk->count; | |
| 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; | |
| } | |
| } | |
| } | |
| link_row->link=0; | |
| fused_rows=src_used_rows; | |
| fsize=src.fsize; | |
| } | } |
| void String::expand() { | String& String::append(const String& src, Untaint_lang lang) { |
| curr_chunk_rows=curr_chunk_rows*100/CR_GROW_PERCENT; | int src_used_rows=src.used_rows(); |
| Chunk *chunk=static_cast<Chunk *>( | int dst_free_rows=link_row-append_here; |
| pool->calloc(sizeof(Chunk::Row)*curr_chunk_rows+sizeof(Chunk *))); | |
| chunk->count=curr_chunk_rows; | if(src_used_rows<=dst_free_rows) { |
| link_row->link=chunk; | // all new rows fit into last chunk |
| append_here=chunk->first; | memcpy(append_here, src.head.rows, sizeof(Chunk::Row)*src_used_rows); |
| link_row=&chunk->first[curr_chunk_rows]; | set_lang(append_here, lang, src_used_rows); |
| append_here+=src_used_rows; | |
| } else { | |
| // not all new rows fit into last chunk: shrinking it to used part, | |
| int used_rows=last_chunk->count-dst_free_rows; | |
| //int *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<Chunk *>( | |
| malloc(sizeof(int)+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; | |
| int rows_left_to_copy=src_used_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); | |
| set_lang(new_rows, lang, 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, rows_left_to_copy); | |
| break; | |
| } | |
| } | |
| link_row->link=0; | |
| } | |
| fused_rows+=src_used_rows; | |
| fsize+=src.fsize; | |
| return *this; | |
| } | } |
| void String::set_lang(Chunk::Row *row, Untaint_lang lang, size_t size) { | |
| if(lang==PASS_APPENDED) | |
| return; | |
| while(size--) { | |
| Untaint_lang& item_lang=(row++)->item.lang; | |
| if(item_lang==YES) // tainted? need untaint language assignment | |
| item_lang=lang; // assign untaint language | |
| } | |
| } | |
| String& String::real_append(STRING_APPEND_PARAMS) { | |
| if(!src) | |
| return *this; | |
| if(!size) | |
| size=strlen(src); | |
| if(!size) | |
| return *this; | |
| String& String::operator += (char *src) { | |
| if(chunk_is_full()) | if(chunk_is_full()) |
| expand(); | expand(); |
| append_here->item.ptr=src; | append_here->item.ptr=src; |
| append_here->item.size=strlen(src); | fsize+=append_here->item.size=size; |
| append_here++; | append_here->item.lang=tainted?/*Untaint_lang::*/YES:Untaint_lang::NO; |
| #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; |
| } | } |
| size_t String::size() { | char *String::cstr() const { |
| int result=0; | char *result=static_cast<char *>(malloc(size()+1)); |
| Chunk *chunk=&head; | |
| char *copy_here=result; | |
| const Chunk *chunk=&head; | |
| do { | do { |
| Chunk::Row *row=chunk->first; | const Chunk::Row *row=chunk->rows; |
| for(int i=0; i<chunk->count; i++) { | for(int i=0; i<chunk->count; i++) { |
| if(row==append_here) | if(row==append_here) |
| goto break2; | goto break2; |
| result+=row->item.size; | switch(row->item.lang) { |
| case NO: | |
| case YES: // for VString.get_double of tainted values | |
| case AS_IS: | |
| memcpy(copy_here, row->item.ptr, row->item.size); | |
| break; | |
| case HTML_TYPO: | |
| memset(copy_here, '?', row->item.size); | |
| break; | |
| default: | |
| THROW(0,0, | |
| this, | |
| "unknown untaint language #%d of %d piece", | |
| static_cast<int>(row->item.lang), | |
| i); | |
| } | |
| copy_here+=row->item.size; | |
| row++; | row++; |
| } | } |
| chunk=row->link; | chunk=row->link; |
| } while(chunk); | } while(chunk); |
| break2: | break2: |
| *copy_here=0; | |
| return result; | return result; |
| } | } |
| char *String::c_str() { | uint String::hash_code() const { |
| char *result=static_cast<char *>(pool->alloc(size()+1)); | uint result=0; |
| char *copy_here=result; | const Chunk *chunk=&head; |
| Chunk *chunk=&head; | |
| do { | do { |
| Chunk::Row *row=chunk->first; | const Chunk::Row *row=chunk->rows; |
| for(int i=0; i<chunk->count; i++) { | for(int i=0; i<chunk->count; i++) { |
| if(row==append_here) | if(row==append_here) |
| goto break2; | goto break2; |
| memcpy(copy_here, row->item.ptr, row->item.size); | result=Hash::generic_code(result, row->item.ptr, row->item.size); |
| copy_here+=row->item.size; | |
| row++; | row++; |
| } | } |
| chunk=row->link; | chunk=row->link; |
| } while(chunk); | } while(chunk); |
| break2: | break2: |
| *copy_here=0; | |
| return result; | return result; |
| } | } |
| int String::cmp(const String& src) const { | |
| 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; | |
| int result; | |
| while(true) { | |
| a_break=a_row==a_end; | |
| b_break=b_row==b_end; | |
| if(a_break || b_break) | |
| break; | |
| 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(!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; | |
| } | |
| } | |
| if(a_break==b_break) // ended simultaneously | |
| result=0; | |
| else if(a_break) // first bytes equal, but a ended before b | |
| result=-1; | |
| else | |
| result=+1; | |
| return result; | |
| } | |
| bool String::operator == (char* b_ptr) const { | |
| size_t b_size=b_ptr?strlen(b_ptr):0; | |
| if(size() != b_size) | |
| return false; | |
| 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; | |
| 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; | |
| } | |
| 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; | |
| } | |
| } | |
| return a_break==b_break; | |
| } |