[BACK]Return to pa_string.C CVS log [TXT] [DIR] Up to [parser3project] / parser3 / src / main

Diff for /parser3/src/main/pa_string.C between versions 1.51 and 1.61

version 1.51, 2001/03/24 19:30:07 version 1.61, 2001/04/03 08:23:08
Line 1 Line 1
 /** @file  /** @file
         Parser: string class. @see untaint.C.          Parser: string class. @see untasize_t.C.
   
         Copyright (c) 2001 ArtLebedev Group (http://www.artlebedev.com)          Copyright (c) 2001 ArtLebedev Group (http://www.artlebedev.com)
   
Line 14 Line 14
 #include "pa_string.h"  #include "pa_string.h"
 #include "pa_hash.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_sapi.h"
   
 // String  // String
   
 String::String(Pool& apool, const char *src, bool tainted) :  String::String(Pool& apool, const char *src, bool tasize_ted) :
         Pooled(apool) {          Pooled(apool) {
         last_chunk=&head;          last_chunk=&head;
         head.count=CR_PREALLOCATED_COUNT;          head.count=CR_PREALLOCATED_COUNT;
Line 27  String::String(Pool& apool, const char * Line 33  String::String(Pool& apool, const char *
         fused_rows=fsize=0;          fused_rows=fsize=0;
   
         if(src)          if(src)
                 if(tainted)                  if(tasize_ted)
                         APPEND_TAINTED(src, 0, 0, 0);                          APPEND_TAINTED(src, 0, 0, 0);
                 else                  else
                         APPEND(src, 0, 0, 0);                          APPEND_CONST(src);
 }  }
   
 void String::expand() {  void String::expand() {
         int new_chunk_count=last_chunk->count+last_chunk->count*CR_GROW_PERCENT/100;          size_t new_chunk_count=last_chunk->count+last_chunk->count*CR_GROW_PERCENT/100;
         last_chunk=static_cast<Chunk *>(          last_chunk=static_cast<Chunk *>(
                 malloc(sizeof(int)+sizeof(Chunk::Row)*new_chunk_count+sizeof(Chunk *)));                  malloc(sizeof(size_t)+sizeof(Chunk::Row)*new_chunk_count+sizeof(Chunk *)));
         last_chunk->count=new_chunk_count;          last_chunk->count=new_chunk_count;
         link_row->link=last_chunk;          link_row->link=last_chunk;
         append_here=last_chunk->rows;          append_here=last_chunk->rows;
Line 47  void String::expand() { Line 53  void String::expand() {
 String::String(const String& src) :     Pooled(src.pool()) {  String::String(const String& src) :     Pooled(src.pool()) {
         head.count=CR_PREALLOCATED_COUNT;          head.count=CR_PREALLOCATED_COUNT;
                   
         int src_used_rows=src.fused_rows;          size_t src_used_rows=src.fused_rows;
         if(src_used_rows<=head.count) {          if(src_used_rows<=head.count) {
                 // all new rows fit into preallocated area                  // all new rows fit size_to preallocated area
                 int curr_chunk_rows=head.count;                  size_t curr_chunk_rows=head.count;
                 memcpy(head.rows, src.head.rows, sizeof(Chunk::Row)*src_used_rows);                  memcpy(head.rows, src.head.rows, sizeof(Chunk::Row)*src_used_rows);
                 append_here=&head.rows[src_used_rows];                  append_here=&head.rows[src_used_rows];
                 link_row=&head.rows[curr_chunk_rows];                  link_row=&head.rows[curr_chunk_rows];
Line 63  String::String(const String& src) : Pool Line 69  String::String(const String& src) : Pool
                 //   allocating only enough mem to fit src string rows                  //   allocating only enough mem to fit src string rows
                 //   next append would allocate a new chunk                  //   next append would allocate a new chunk
                 //                  //
                 // new rows don't fit into preallocated area: splitting into two chunks                  // new rows don't fit size_to preallocated area: splitting size_to two chunks
                 // preallocated chunk src to constructing head                  // preallocated chunk src to constructing head
                 memcpy(head.rows, src.head.rows, sizeof(Chunk::Row)*head.count);                  memcpy(head.rows, src.head.rows, sizeof(Chunk::Row)*head.count);
                 // remaining rows into new_chunk                  // remaining rows size_to new_chunk
                 int curr_chunk_rows=src_used_rows-head.count;                  size_t curr_chunk_rows=src_used_rows-head.count;
                 Chunk *new_chunk=static_cast<Chunk *>(                  Chunk *new_chunk=static_cast<Chunk *>(
                         malloc(sizeof(int)+sizeof(Chunk::Row)*curr_chunk_rows+sizeof(Chunk *)));                          malloc(sizeof(size_t)+sizeof(Chunk::Row)*curr_chunk_rows+sizeof(Chunk *)));
                 new_chunk->count=curr_chunk_rows;                  new_chunk->count=curr_chunk_rows;
                 head.preallocated_link=new_chunk;                  head.preallocated_link=new_chunk;
                 append_here=link_row=&new_chunk->rows[new_chunk->count];                  append_here=link_row=&new_chunk->rows[new_chunk->count];
   
                 Chunk *old_chunk=src.head.preallocated_link;                   Chunk *old_chunk=src.head.preallocated_link; 
                 Chunk::Row *new_rows=new_chunk->rows;                  Chunk::Row *new_rows=new_chunk->rows;
                 int rows_left_to_copy=new_chunk->count;                  size_t rows_left_to_copy=new_chunk->count;
                 while(true) {                  while(true) {
                         int old_count=old_chunk->count;                          size_t old_count=old_chunk->count;
                         Chunk *next_chunk=old_chunk->rows[old_count].link;                          Chunk *next_chunk=old_chunk->rows[old_count].link;
                         if(next_chunk) {                          if(next_chunk) {
                                 // not last source chunk                                  // not last source chunk
Line 102  String::String(const String& src) : Pool Line 108  String::String(const String& src) : Pool
 }  }
   
 String& String::append(const String& src, Untaint_lang lang, bool forced) {  String& String::append(const String& src, Untaint_lang lang, bool forced) {
         int src_used_rows=src.fused_rows;          const Chunk *chunk=&src.head; 
         int dst_free_rows=link_row-append_here;  
           
         if(src_used_rows<=dst_free_rows) {  
                 // all new rows fit into 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 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, 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;  
   
         return *this;  
 }  
 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) // tainted? need untaint language assignment  
                         item_lang=lang;  // assign untaint language  
         }  
 }  
   
 /*void String::change_lang(Untaint_lang lang) {  
         Chunk *chunk=&head;   
         do {          do {
                 Chunk::Row *row=chunk->rows;                  const Chunk::Row *row=chunk->rows;
                 for(int i=0; i<chunk->count; i++) {                  for(size_t i=0; i<chunk->count; i++, row++) {
                         if(row==append_here)                          if(row==src.append_here)
                                 goto break2;                                  goto break2;
                           
                         row->item.lang=lang;                          APPEND(row->item.ptr, row->item.size, 
                         row++;                                  (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;                  chunk=row->link;
         } while(chunk);          } while(chunk);
 break2:  break2:
         return;          return *this;
 }  }
 */  
 String& String::real_append(STRING_APPEND_PARAMS) {  String& String::real_append(STRING_APPEND_PARAMS) {
         if(!src)          if(!src)
                 return *this;                  return *this;
Line 194  String& String::real_append(STRING_APPEN Line 138  String& String::real_append(STRING_APPEN
   
         append_here->item.ptr=src;          append_here->item.ptr=src;
         fsize+=append_here->item.size=size;          fsize+=append_here->item.size=size;
         append_here->item.lang=tainted?UL_YES:UL_NO;          append_here->item.lang=lang;
 #ifndef NO_STRING_ORIGIN  #ifndef NO_STRING_ORIGIN
         append_here->item.origin.file=file;          append_here->item.origin.file=file;
         append_here->item.origin.line=line;          append_here->item.origin.line=line;
Line 210  uint String::hash_code() const { Line 154  uint String::hash_code() const {
         const Chunk *chunk=&head;           const Chunk *chunk=&head; 
         do {          do {
                 const Chunk::Row *row=chunk->rows;                  const Chunk::Row *row=chunk->rows;
                 for(int i=0; i<chunk->count; i++) {                  for(size_t i=0; i<chunk->count; i++) {
                         if(row==append_here)                          if(row==append_here)
                                 goto break2;                                  goto break2;
   
Line 223  break2: Line 167  break2:
         return result;          return result;
 }  }
   
 int String::cmp(const String& src) const {  /// @todo move 'lang' skipping to pos
   int String::cmp(int& partial, const String& src, 
                                   size_t this_offset, Untaint_lang lang) const {
           partial=-1;
           this_offset=min(this_offset, size()-1);
   
         const Chunk *a_chunk=&head;          const Chunk *a_chunk=&head;
         const Chunk *b_chunk=&src.head;          const Chunk *b_chunk=&src.head;
         const Chunk::Row *a_row=a_chunk->rows;          const Chunk::Row *a_row=a_chunk->rows;
         const Chunk::Row *b_row=b_chunk->rows;          const Chunk::Row *b_row=b_chunk->rows;
         int a_offset=0;          size_t a_offset=this_offset;
         int b_offset=0;          size_t b_offset=0;
         Chunk::Row *a_end=append_here;          Chunk::Row *a_end=append_here;
         Chunk::Row *b_end=src.append_here;          Chunk::Row *b_end=src.append_here;
         int a_countdown=a_chunk->count;          size_t a_countdown=a_chunk->count;
         int b_countdown=b_chunk->count;          size_t b_countdown=b_chunk->count;
         bool a_break=false;          bool a_break=false;
         bool b_break=false;          bool b_break=false;
         int result;          size_t result;
           size_t pos=0; 
         while(true) {          while(true) {
                 a_break=a_row==a_end;                  a_break=a_row==a_end;
                 b_break=b_row==b_end;                  b_break=b_row==b_end;
                 if(a_break || b_break)                  if(a_break || b_break)
                         break;                          break;
   
                 int size_diff=                  if(pos+a_row->item.size > this_offset) {
                         (a_row->item.size-a_offset)-                          if(lang!=UL_UNKNOWN && a_row->item.lang!=lang) 
                         (b_row->item.size-b_offset);                                  return -1; // wrong lang -- bail out
   
                 if(size_diff==0) { // a has same size as b                          int size_diff=
                         result=memcmp(a_row->item.ptr+a_offset, b_row->item.ptr+b_offset, a_row->item.size-a_offset);                                  (a_row->item.size-a_offset)-
                         if(result)                                  (b_row->item.size-b_offset);
                                 return result;                          
                         a_row++; a_countdown--; a_offset=0;                          if(size_diff==0) { // a has same size as b
                         b_row++; b_countdown--; b_offset=0;                                  result=memcmp(a_row->item.ptr+a_offset, b_row->item.ptr+b_offset, 
                 } else if (size_diff>0) { // a longer                                          a_row->item.size-a_offset);
                         result=memcmp(a_row->item.ptr+a_offset, b_row->item.ptr+b_offset, b_row->item.size-b_offset);                                  if(result)
                         if(result)                                          return result;
                                 return result;                                  pos+=a_row->item.size;
                         a_offset+=b_row->item.size-b_offset;                                  a_row++; a_countdown--; a_offset=0;
                         b_row++; b_countdown--; b_offset=0;                                  b_row++; b_countdown--; b_offset=0;
                 } else { // b longer                          } else if (size_diff>0) { // a longer
                         result=memcmp(a_row->item.ptr+a_offset, b_row->item.ptr+b_offset, a_row->item.size-a_offset);                                  result=memcmp(a_row->item.ptr+a_offset, b_row->item.ptr+b_offset, 
                         if(result)                                          b_row->item.size-b_offset);
                                 return result;                                  if(result)
                         b_offset+=a_row->item.size-a_offset;                                          return result;
                         a_row++; a_countdown--; a_offset=0;                                  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) {                  if(!a_countdown) {
Line 272  int String::cmp(const String& src) const Line 242  int String::cmp(const String& src) const
                         a_row=a_chunk->rows;                          a_row=a_chunk->rows;
                         a_countdown=a_chunk->count;                          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          if(a_break==b_break) { // ended simultaneously
                 result=0;                  partial=0; return 0;
         else if(a_break) // first bytes equal, but a ended before b          } else if(a_break) { // first bytes equal, but a ended before b
                 result=-1;                  partial=1; return -1;
         else          } else {
                 result=+1;                  partial=2; return +1;
         return result;          }
 }  }
   
 int String::cmp(const char* b_ptr, int& partial, size_t src_size) const {  /// @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;          size_t b_size=src_size?src_size:b_ptr?strlen(b_ptr):0;
           this_offset=min(this_offset, size()-1);
   
         partial=-1;  
         const Chunk *a_chunk=&head;          const Chunk *a_chunk=&head;
         const Chunk::Row *a_row=a_chunk->rows;          const Chunk::Row *a_row=a_chunk->rows;
         int a_offset=0;          size_t a_offset=this_offset;
         int b_offset=0;          size_t b_offset=0;
         Chunk::Row *a_end=append_here;          Chunk::Row *a_end=append_here;
         int a_countdown=a_chunk->count;          size_t a_countdown=a_chunk->count;
         bool a_break=false;          bool a_break=false;
         bool b_break=false;          bool b_break=false;
           size_t pos=0;
         while(true) {          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(int 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(int 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(int 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;  
                 }  
   
                 a_break=a_row==a_end;                  a_break=a_row==a_end;
                 if(a_break || b_break)                  if(a_break || b_break)
                         break;                          break;
   
                   if(pos+a_row->item.size > this_offset) {
                           if(lang!=UL_UNKNOWN && a_row->item.lang!=lang) 
                                   return -1; // wrong lang -- bail out
   
                           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) {                  if(!a_countdown) {
                         a_chunk=a_row->link;                          a_chunk=a_row->link;
                         a_row=a_chunk->rows;                          a_row=a_chunk->rows;
                         a_countdown=a_chunk->count;                          a_countdown=a_chunk->count;
                 }                  }
         }          }
         if(a_break==b_break) // ended simultaneously          if(a_break==b_break) { // ended simultaneously
                 return partial=0;                  partial=0; return 0;
         else if(a_break) // first bytes equal, but a ended before b          } else if(a_break) { // first bytes equal, but a ended before b
                 return partial=1;                  partial=1; return -1;
         else          } else {
                 return partial=2;                  partial=2; return +1;
           }
 }  }
   
 #ifndef NO_STRING_ORIGIN  #ifndef NO_STRING_ORIGIN
Line 352  const Origin& String::origin() const { Line 336  const Origin& String::origin() const {
         return append_here[-1].item.origin;           return append_here[-1].item.origin; 
 }  }
 #endif  #endif
   
   String& String::piece(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'",
                   //cstr(), start, finish, result.cstr());
           return result;
   }
   
   int String::pos(const String& substr, 
                                   size_t result, Untaint_lang lang) const {
           for(; result<size(); result++) {
                   int partial; cmp(partial, substr, result, lang);
                   if(
                           partial==0 || // full match
                           partial==2) // 'substr' starts 'this'+'result'
                           return result;
           }
           
           return -1;
   }
   
   int String::pos(const char *substr, size_t substr_size, 
                                   size_t result, Untaint_lang lang) const {
           for(; result<size(); result++) {
                   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, 
                                      size_t* pos_after_ref, 
                                      const char *delim, size_t delim_size, 
                                      Untaint_lang lang, int limit) const {
           if(delim_size) {
                   size_t pos_after=pos_after_ref?*pos_after_ref:0;
                   int pos_before;
                   // while we have 'delim'...
                   for(; (pos_before=pos(delim, delim_size, pos_after, lang))>=0 && limit; limit--) {
                           result+=&piece(pos_after, pos_before);
                           pos_after=pos_before+delim_size;
                   }
                   // last piece
                   if(pos_after<size() && limit) {
                           result+=&piece(pos_after, size());
                           pos_after=size();
                   }
                   if(pos_after_ref)
                           *pos_after_ref=pos_after;
           } else { // empty delim
                   result+=this;
                   if(pos_after_ref)
                           *pos_after_ref+=size();
           }
   }
   
   void String::split(Array& result, 
                                      size_t* pos_after_ref, 
                                      const String& delim, Untaint_lang lang, 
                                      int limit) const {
           if(delim.size()) {
                   size_t pos_after=pos_after_ref?*pos_after_ref:0;
                   int pos_before;
                   // while we have 'delim'...
                   for(; (pos_before=pos(delim, pos_after, lang))>=0 && limit; limit--) {
                           result+=&piece(pos_after, pos_before);
                           pos_after=pos_before+delim.size();
                   }
                   // last piece
                   if(pos_after<size() && limit) {
                           result+=&piece(pos_after, size());
                           pos_after=size();
                   }
                   if(pos_after_ref)
                           *pos_after_ref=pos_after;
           } else { // empty delim
                   result+=this;
                   if(pos_after_ref)
                           *pos_after_ref+=size();
           }
   }
   
   Table& String::match(const String *aorigin,
                                            const String& regexp, const String& options) const {
           Array *columns=0;
           Table& result=*NEW Table(pool(), aorigin, columns);
           return result;
   }
Removed from v.1.51  
changed lines
  Added in v.1.61

E-mail: