|
|
| version 1.27, 2001/03/10 11:03:48 | version 1.58, 2001/03/29 17:23:21 |
|---|---|
| Line 1 | Line 1 |
| /* | /** @file |
| $Id$ | Parser: string class decl. |
| */ | |
| /* | Copyright (c) 2001 ArtLebedev Group (http://www.artlebedev.com) |
| String Chunk0 | Author: Alexander Petrosyan <paf@design.ru> (http://design.ru/paf) |
| ====== ======== | |
| head--------------->[ptr, size, ...] | |
| append_here-------->[ptr, size, ...] | |
| . | |
| . | |
| [ptr, size, ...] | |
| link_row----------->[link to the next chunk] | |
| $Id$ | |
| */ | */ |
| #ifndef PA_STRING_H | #ifndef PA_STRING_H |
| #define PA_STRING_H | #define PA_STRING_H |
| #ifdef HAVE_CONFIG_H | #include "pa_config_includes.h" |
| #include "pa_config.h" | |
| #endif | |
| #include <stddef.h> | #include <stddef.h> |
| #include "pa_pool.h" | #include "pa_pool.h" |
| #include "pa_types.h" | #include "pa_types.h" |
| /** | |
| $MAIN:html-typo table elements must enlarge string not more that that | |
| that's a tradeoff - otherwise we'd have to scan string twice: | |
| - first for buffer length | |
| - second for replacements themselves | |
| */ | |
| #define UNTAINT_TIMES_BIGGER 10 | |
| #ifndef NO_STRING_ORIGIN | #ifndef NO_STRING_ORIGIN |
| # define STRING_APPEND_PARAMS const char *src, size_t size, bool tainted, char *file, uint line | # define STRING_APPEND_PARAMS \ |
| # define APPEND(src, size, file, line) real_append(src, size, false, file, line) | const char *src, size_t size, \ |
| # define APPEND_TAINTED(src, size, file, line) real_append(src, size, true, file, line) | String::Untaint_lang lang, \ |
| const char *file, uint line | |
| /// appends piece to String @see String::real_append | |
| # define APPEND(src, size, lang, file, line) \ | |
| real_append(src, size, lang, file, line) | |
| #else | #else |
| # define STRING_APPEND_PARAMS const char *src, size_t size, bool tainted | # define STRING_APPEND_PARAMS \ |
| # define APPEND(src, size, file, line) real_append(src, size, false) | const char *src, \ |
| # define APPEND_TAINTED(src, size, file, line) real_append(src, size, true) | size_t size, \ |
| String::Untaint_lang lang | |
| /// appends piece to String @see String::real_append | |
| # define APPEND(src, size, lang, file, line) \ | |
| real_append(src, size, lang) | |
| #endif | #endif |
| #define APPEND_CONST(src) APPEND(src, 0, 0, 0) | /// appends clean piece to String @see String::real_append |
| #define APPEND_CLEAN(src, size, file, line) \ | |
| APPEND(src, size, String::UL_NO, file, line) | |
| /// appends tainted piece to String @see String::real_append | |
| #define APPEND_TAINTED(src, size, file, line) \ | |
| APPEND(src, size, String::UL_YES, file, line) | |
| /// handy: appends const char* piece to String @see String::real_append | |
| #define APPEND_CONST(src) APPEND_CLEAN(src, 0, 0, 0) | |
| /** | |
| Pooled string. | |
| Internal structure: | |
| @verbatim | |
| String Chunk0 | |
| ====== ======== | |
| head--------------->[ptr, size, ...] | |
| append_here-------->[ptr, size, ...] | |
| . | |
| . | |
| [ptr, size, ...] | |
| link_row----------->[link to the next chunk] | |
| @endverbatim | |
| All pieces remember | |
| - the file and its line they are from [can be turned off by NO_STRING_ORIGIN] | |
| - whether they are tainted or not, | |
| and the language which should be used to detaint them | |
| */ | |
| class String : public Pooled { | class String : public Pooled { |
| public: | public: |
| enum { | enum { |
| CR_PREALLOCATED_COUNT=5, | CR_PREALLOCATED_COUNT=5, ///< default preallocated item count |
| CR_GROW_PERCENT=60 | CR_GROW_PERCENT=60 ///< each time the Array chunk_is_full() array expanded() |
| }; | }; |
| /// piece is tainted or not. the language to use when detaint | |
| enum Untaint_lang { | enum Untaint_lang { |
| NO, // clean | UL_UNKNOWN=0, ///< when get by name fails |
| YES, // tainted, untaint language as assigned later | UL_NO, ///< clean |
| UL_YES, ///< tainted, untaint language as assigned later | |
| // untaint languages. assigned by ^untaint[lang]{...} | // untaint languages. assigned by ^untaint[lang]{...} |
| APPENDED, // leave langage build into string being appended | UL_PASS_APPENDED, |
| AS_IS, | /**< |
| SQL, | leave language built into string being appended. |
| JS, | just a flag, that value not stored |
| TABLE, | */ |
| HTML, | UL_AS_IS, ///< leave all characters intact |
| HTML_TYPO | UL_FILE_NAME, ///< filename |
| UL_HEADER, ///< text in response header | |
| UL_URI, ///< text in uri | |
| UL_TABLE, ///< ^table:set body | |
| UL_SQL, ///< ^table:sql body | |
| UL_JS, ///< JavaScript code | |
| UL_HTML, ///< HTML code (for editing) | |
| UL_HTML_TYPO ///< HTML code with TYPOgraphic replacements (for showing) | |
| }; | }; |
| public: | public: |
| String(Pool& apool); | String(Pool& apool, const char *src=0, bool tainted=false); |
| String(const String& src); | String(const String& src); |
| size_t size() const { return fsize; } | size_t size() const { return fsize; } |
| int used_rows() const { return fused_rows; } | /// convert to C string |
| char *cstr() const; | char *cstr() const { |
| char *result=(char *)malloc(size()*UNTAINT_TIMES_BIGGER+1); | |
| char *eol=store_to(result); | |
| *eol=0; | |
| return result; | |
| } | |
| /** append fragment | |
| @see APPEND_CLEAN, APPEND_TAINTED, APPEND_CONST | |
| */ | |
| String& real_append(STRING_APPEND_PARAMS); | String& real_append(STRING_APPEND_PARAMS); |
| int cmp (const String& src) const; | /// @return <0 ==0 or >0 depending on comparison result |
| bool operator < (const String& src) const { return cmp(src)<0; } | int cmp (int& partial, const String& src, size_t this_offset=0) const; |
| bool operator > (const String& src) const { return cmp(src)>0; } | bool operator < (const String& src) const { int p; return cmp(p, src)<0; } |
| bool operator <= (const String& src) const { return cmp(src)<=0; } | bool operator > (const String& src) const { int p; return cmp(p, src)>0; } |
| bool operator >= (const String& src) const { return cmp(src)>=0; } | bool operator <= (const String& src) const { int p; return cmp(p, src)<=0; } |
| bool operator >= (const String& src) const { int p; return cmp(p, src)>=0; } | |
| bool operator == (const String& src) const { | bool operator == (const String& src) const { |
| if(size()!=src.size()) // can speed up in trivial case | if(size()!=src.size()) // can speed up in trivial case |
| return false; | return false; |
| return cmp(src)==0; | int p; return cmp(p, src)==0; |
| } | } |
| bool operator != (const String& src) const { return cmp(src)!=0; } | bool operator != (const String& src) const { int p; return cmp(p, src)!=0; } |
| bool operator == (char* src) const; | /** |
| String& append(const String& src, Untaint_lang lang); | @param partial |
| returns partial match status. | |
| - -1: strings too different | |
| - 0: full match | |
| - 1: means @c this starts @c src | |
| - 2: means @src starts @this | |
| */ | |
| int cmp(int& partial, const char* src_ptr, size_t src_size=0) const; | |
| bool operator == (const char* src_ptr) const { | |
| size_t src_size=src_ptr?strlen(src_ptr):0; | |
| if(size() != src_size) | |
| return false; | |
| int partial; // unused | |
| return cmp(partial, src_ptr, src_size)==0; | |
| } | |
| /** | |
| appends other String. | |
| marking all tainted pieces of it with @a lang. | |
| or marking ALL pieces of it with a @a lang when @a forced to. | |
| */ | |
| String& append(const String& src, Untaint_lang lang, bool forced=false); | |
| /// simple hash code of string. used by Hash | |
| uint hash_code() const; | uint hash_code() const; |
| const Origin& origin() const { return head.rows[0].item.origin; } | /// extracts [start, finish) piece of string |
| String& piece(size_t start, size_t finish) const; | |
| /// @return position of substr in string, -1 means "not found" | |
| int pos(const String& substr, size_t this_offset=0) const; | |
| #ifndef NO_STRING_ORIGIN | |
| /// origin of string. calculated by first row | |
| const Origin& origin() const; | |
| #endif | |
| private: | private: |
| struct Chunk { | struct Chunk { |
| // the number of rows in chunk | // the number of rows in chunk |
| int count; | size_t count; |
| union Row { | union Row { |
| // fragment | // fragment |
| struct { | struct { |
| Line 130 private: | Line 215 private: |
| return append_here == link_row; | return append_here == link_row; |
| } | } |
| void expand(); | void expand(); |
| void set_lang(Chunk::Row *row, Untaint_lang lang, size_t size); | void set_lang(Chunk::Row *row, Untaint_lang lang, bool forced, size_t size); |
| char *String::store_to(char *dest) const; | |
| private: //disabled | private: //disabled |