--- parser3/src/include/pa_string.h 2001/04/05 13:19:42 1.71 +++ parser3/src/include/pa_string.h 2001/04/09 14:31:40 1.78 @@ -5,7 +5,7 @@ Author: Alexander Petrosyan (http://design.ru/paf) - $Id: pa_string.h,v 1.71 2001/04/05 13:19:42 paf Exp $ + $Id: pa_string.h,v 1.78 2001/04/09 14:31:40 paf Exp $ */ #ifndef PA_STRING_H @@ -13,6 +13,7 @@ #include "pa_config_includes.h" +#include #include #include "pa_pool.h" @@ -87,7 +88,7 @@ public: /// piece is tainted or not. the language to use when detaint enum Untaint_lang { - UL_UNKNOWN=0, ///< zero value handy for hash lookup @see untaint_lang_name2enum + UL_UNSPECIFIED=0, ///< zero value handy for hash lookup @see untaint_lang_name2enum UL_CLEAN, ///< clean UL_TAINTED, ///< tainted, untaint language as assigned later // untaint languages. assigned by ^untaint[lang]{...} @@ -98,7 +99,8 @@ public: */ UL_AS_IS, ///< leave all characters intact UL_FILE_NAME, ///< filename - UL_HEADER, ///< text in response header + UL_HTTP_HEADER, ///< text in HTTP response header + UL_MAIL_HEADER, ///< text in mail header UL_URI, ///< text in uri UL_TABLE, ///< ^table:set body UL_SQL, ///< ^table:sql body @@ -109,11 +111,14 @@ public: public: - String(Pool& apool, const char *src=0, bool tainted=false); + String(Pool& apool, const char *src=0, size_t src_size=0, bool tainted=false); String(const String& src); size_t size() const { return fsize; } + /// convert to C string, store to 'dest' which must be big enough for proper untaint + char *store_to(char *dest, + Untaint_lang lang=UL_UNSPECIFIED, SQL_Connection *connection=0) const; /// convert to C string. if 'lang' known, forcing 'lang' to it - char *cstr(Untaint_lang lang=UL_UNKNOWN, SQL_Connection *connection=0) const { + char *cstr(Untaint_lang lang=UL_UNSPECIFIED, SQL_Connection *connection=0) const { char *result=(char *)malloc(size()*UNTAINT_TIMES_BIGGER+1); char *eol=store_to(result, lang, connection); *eol=0; @@ -125,7 +130,7 @@ public: String& real_append(STRING_APPEND_PARAMS); /// @return <0 ==0 or >0 depending on comparison result int cmp (int& partial, const String& src, - size_t this_offset=0, Untaint_lang lang=UL_UNKNOWN) const; + size_t this_offset=0, Untaint_lang lang=UL_UNSPECIFIED) const; 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 { int p; return cmp(p, src)<=0; } @@ -146,7 +151,7 @@ public: - 2: means @src starts @this */ int cmp(int& partial, const char* src_ptr, size_t src_size=0, - size_t this_offset=0, Untaint_lang lang=UL_UNKNOWN) const; + size_t this_offset=0, Untaint_lang lang=UL_UNSPECIFIED) const; bool operator == (const char* src_ptr) const { size_t src_size=src_ptr?strlen(src_ptr):0; if(size() != src_size) @@ -162,6 +167,8 @@ public: 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); + String& operator << (const String& src) { return append(src, UL_PASS_APPENDED); } + String& operator << (const char *src) { return APPEND_CONST(src); } /// simple hash code of string. used by Hash uint hash_code() const; @@ -171,10 +178,10 @@ public: /// @return position of substr in string, -1 means "not found" [String version] int pos(const String& substr, - size_t this_offset=0, Untaint_lang lang=UL_UNKNOWN) const; + size_t this_offset=0, Untaint_lang lang=UL_UNSPECIFIED) const; /// @return position of substr in string, -1 means "not found" [const char* version] int pos(const char *substr, size_t substr_size, - size_t this_offset=0, Untaint_lang lang=UL_UNKNOWN) const; + size_t this_offset=0, Untaint_lang lang=UL_UNSPECIFIED) const; void split(Array& result, size_t *pos_after_ref, @@ -243,8 +250,6 @@ private: return append_here == link_row; } void expand(); - char *String::store_to(char *dest, - Untaint_lang lang=UL_UNKNOWN, SQL_Connection *connection=0) const; private: //disabled