|
|
| 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*/ )); | |
| } | } |
| } | } |