Diff for /parser3/src/classes/string.C between versions 1.239 and 1.243

version 1.239, 2019/09/06 10:17:07 version 1.243, 2019/11/20 20:48:25
Line 7 Line 7
   
 #include "classes.h"  #include "classes.h"
 #include "pa_vmethod_frame.h"  #include "pa_vmethod_frame.h"
   #include "pa_base64.h"
   
 #include "pa_request.h"  #include "pa_request.h"
 #include "pa_vdouble.h"  #include "pa_vdouble.h"
Line 443  static void _lower(Request& r, MethodPar Line 444  static void _lower(Request& r, MethodPar
   
 #ifndef DOXYGEN  #ifndef DOXYGEN
 class String_sql_event_handlers: public SQL_Driver_query_event_handlers {  class String_sql_event_handlers: public SQL_Driver_query_event_handlers {
         const String& statement_string; const char* statement_cstr;  
         bool got_column;          bool got_column;
 public:  public:
         bool got_cell;          bool got_cell;
         const String* result;          const String* result;
 public:  public:
         String_sql_event_handlers(          String_sql_event_handlers():
                 const String& astatement_string, const char* astatement_cstr):  
                 statement_string(astatement_string), statement_cstr(astatement_cstr),  
                 got_column(false),                  got_column(false),
                 got_cell(false),                  got_cell(false),
                 result(&String::Empty) {}                  result(&String::Empty) {}
   
         bool add_column(SQL_Error& error, const char* /*str*/, size_t /*length*/) {          bool add_column(SQL_Error& error, const char* /*str*/, size_t /*length*/) {
                 if(got_column) {                  if(got_column) {
                         error=SQL_Error(PARSER_RUNTIME, /*statement_string,*/ "result must contain exactly one column");                          error=SQL_Error("result must contain exactly one column");
                         return true;                          return true;
                 }                  }
                 got_column=true;                  got_column=true;
Line 468  public: Line 466  public:
         bool add_row(SQL_Error& /*error*/) { /* ignore */ return false; }          bool add_row(SQL_Error& /*error*/) { /* ignore */ return false; }
         bool add_row_cell(SQL_Error& error, const char* str, size_t) {          bool add_row_cell(SQL_Error& error, const char* str, size_t) {
                 if(got_cell) {                  if(got_cell) {
                         error=SQL_Error(PARSER_RUNTIME, /*statement_string,*/ "result must not contain more then one row");                          error=SQL_Error("result must not contain more then one row");
                         return true;                          return true;
                 }                  }
   
                 try {                  try {
                         got_cell=true;                          got_cell=true;
                         result=new String(str, String::L_TAINTED /* no length as 0x00 can be inside */ );                          result=new String(str, String::L_TAINTED /* no length as 0x00 can be inside */ );
Line 524  const String* sql_result_string(Request& Line 521  const String* sql_result_string(Request&
         const String& statement_string=r.process_to_string(statement);          const String& statement_string=r.process_to_string(statement);
         const char* statement_cstr=statement_string.untaint_cstr(String::L_SQL, r.connection());          const char* statement_cstr=statement_string.untaint_cstr(String::L_SQL, r.connection());
   
         String_sql_event_handlers handlers(statement_string, statement_cstr);          String_sql_event_handlers handlers;
   
         r.connection()->query(          r.connection()->query(statement_cstr, placeholders_count, placeholders, offset, limit, handlers, statement_string);
                 statement_cstr,   
                 placeholders_count, placeholders,  
                 offset, limit,   
                 handlers,  
                 statement_string);  
                   
         if(bind)          if(bind)
                 unmarshal_bind_updates(*bind, placeholders_count, placeholders);                  unmarshal_bind_updates(*bind, placeholders_count, placeholders);
Line 650  static void _trim(Request& r, MethodPara Line 642  static void _trim(Request& r, MethodPara
         r.write(src.trim(kind, chars, &r.charsets.source()));          r.write(src.trim(kind, chars, &r.charsets.source()));
 }  }
   
   Base64Options base64_encode_options(Request& r, HashStringValue* options) {
           Base64Options result;
           if(options) {
                   int valid_options=0;
                   for(HashStringValue::Iterator i(*options); i; i.next()) {
                           String::Body key=i.key();
                           Value* value=i.value();
                           if(key == "pad") {
                                   result.pad=r.process(*value).as_bool();
                                   valid_options++;
                           } else if(key == "wrap") {
                                   result.wrap=r.process(*value).as_bool();
                                   valid_options++;
                           } else if(key == "url-safe") {
                                   if(r.process(*value).as_bool())
                                           result.set_url_safe_abc();
                                   valid_options++;
                           }
                   }
   
                   if(valid_options != options->count())
                           throw Exception(PARSER_RUNTIME, 0, CALLED_WITH_INVALID_OPTION);
           }
           return result;
   }
   
   Base64Options base64_decode_options(Request& r, HashStringValue* options) {
           Base64Options result;
           if(options) {
                   int valid_options=0;
                   for(HashStringValue::Iterator i(*options); i; i.next() ) {
                           String::Body key=i.key();
                           Value* value=i.value();
                           if(key == "pad") {
                                   result.pad=r.process(*value).as_bool();
                                   valid_options++;
                           } else if(key == "strict") {
                                   result.strict=r.process(*value).as_bool();
                                   valid_options++;
                           } else if(key == "url-safe") {
                                   if(r.process(*value).as_bool())
                                           result.set_url_safe_abc();
                                   valid_options++;
                           }
                   }
   
                   if(valid_options != options->count())
                           throw Exception(PARSER_RUNTIME, 0, CALLED_WITH_INVALID_OPTION);
           }
           return result;
   }
   
 static void _base64(Request& r, MethodParams& params) {  static void _base64(Request& r, MethodParams& params) {
         if(&r.get_self() == string_class) {          if(&r.get_self() == string_class) {
                 // decode: ^string:base64[encoded[;$.strict(true|false)]]                  // decode: ^string:base64[encoded[;options]]
                 const char* cstr=params.count() ? params.as_string(0, PARAMETER_MUST_BE_STRING).cstr() : "";                  const char* cstr=params.count() ? params.as_string(0, PARAMETER_MUST_BE_STRING).cstr() : "";
                 char* decoded=0;                  Base64Options options = base64_decode_options(r, params.count() > 1 ? params.as_hash(1) : NULL);
                 size_t length=0;  
   
                 bool strict=false;                  char* decoded=0;
                 if(params.count() > 1)                  size_t length=pa_base64_decode(cstr, strlen(cstr), decoded, options);
                         if(HashStringValue* options=params.as_hash(1)) {  
                                 int valid_options=0;  
                                 if(Value* vstrict=options->get(BASE64_STRICT_OPTION_NAME)) {  
                                         strict=r.process(*vstrict).as_bool();  
                                         valid_options++;  
                                 }  
                                 if(valid_options!=options->count())  
                                         throw Exception(PARSER_RUNTIME, 0, CALLED_WITH_INVALID_OPTION);  
                         }  
   
                 pa_base64_decode(cstr, strlen(cstr), decoded, length, strict);  
                 if(decoded && length){                  if(decoded && length){
                         if(memchr((const char*)decoded, 0, length))                          if(memchr(decoded, 0, length))
                                 throw Exception(PARSER_RUNTIME, 0, "Invalid \\x00 character found while decode to string. Decode it to file instead.");                                  throw Exception(PARSER_RUNTIME, 0, "Invalid \\x00 character found while decoding to string. Decode to file instead.");
   
                         fix_line_breaks(decoded, length);                          fix_line_breaks(decoded, length);
   
                         if(length)                          if(length)
                                 r.write(*new String(decoded, String::L_TAINTED));                                  r.write(*new String(decoded, String::L_TAINTED));
                 }                  }
         } else {          } else {
                 // encode: ^str.base64[]                  // encode: ^str.base64[options]
                 VString& self=GET_SELF(r, VString);                  VString& self=GET_SELF(r, VString);
                 const char* cstr=self.string().cstr();                  const char* cstr=self.string().cstr();
                 const char* encoded=pa_base64_encode(cstr, strlen(cstr));                  Base64Options options = base64_encode_options(r, params.count() ? params.as_hash(0) : NULL);
                 r.write(*new String(encoded, String::L_TAINTED/*once ?param=base64(something) was needed*/));                  const char* encoded=pa_base64_encode(cstr, strlen(cstr), options);
                   r.write(*new String(encoded, String::L_TAINTED /*once ?param=base64(something) was needed*/ ));
         }          }
 }  }
   

Removed from v.1.239  
changed lines
  Added in v.1.243


E-mail: