|
|
| version 1.77, 2025/05/26 01:56:54 | version 1.82, 2026/07/19 20:25:15 |
|---|---|
| Line 1 | Line 1 |
| /** @file | /** @file |
| Parser: @b curl parser class. | Parser: @b curl parser class. |
| Copyright (c) 2001-2024 Art. Lebedev Studio (http://www.artlebedev.com) | Copyright (c) 2001-2026 Art. Lebedev Studio (https://www.artlebedev.com) |
| Authors: Konstantin Morshnev <moko@design.ru> | Authors: Konstantin Morshnev <moko@design.ru> |
| */ | */ |
| Line 15 | Line 15 |
| #include "pa_vdate.h" | #include "pa_vdate.h" |
| #include "pa_vtable.h" | #include "pa_vtable.h" |
| #include "pa_common.h" | #include "pa_common.h" |
| #include "pa_http.h" | #include "pa_http.h" |
| #include "ltdl.h" | #include "ltdl.h" |
| #ifdef _MSC_VER | |
| #include "winsock2.h" // AF_INET & co | |
| #endif | |
| volatile const char * IDENT_CURL_C="$Id$"; | volatile const char * IDENT_CURL_C="$Id$"; |
| class MCurl: public Methoded { | class MCurl: public Methoded { |
| Line 44 typedef void (*t_curl_formfree)(struct c | Line 48 typedef void (*t_curl_formfree)(struct c |
| #define GLINK(name) f_##name=(t_##name)lt_dlsym(handle, #name); | #define GLINK(name) f_##name=(t_##name)lt_dlsym(handle, #name); |
| #define DLINK(name) GLINK(name) if(!f_##name) return "function " #name " was not found"; | #define DLINK(name) GLINK(name) if(!f_##name) return "function " #name " was not found"; |
| static const char *dlink(char *dlopen_file_spec) { | static const char *dlink(char *dlopen_file_spec) { |
| pa_dlinit(); | pa_dlinit(); |
| Line 74 static const char *dlink(char *dlopen_fi | Line 78 static const char *dlink(char *dlopen_fi |
| return 0; | return 0; |
| } | } |
| #ifdef WIN32 | |
| #define CURL_LIBRARY "libcurl" LT_MODULE_EXT | |
| #else | |
| #define CURL_LIBRARY "libcurl" LT_MODULE_EXT ",libcurl" LT_MODULE_EXT ".4" | |
| #endif | |
| // 'parser' options can be set outside of 'session' operator | |
| const char *curl_library=CURL_LIBRARY; | |
| bool curl_parser_resolve = false; | |
| struct ParserOptions : public PA_Allocated { | struct ParserOptions : public PA_Allocated { |
| // real options | // real options |
| Line 87 struct ParserOptions : public PA_Allocat | Line 101 struct ParserOptions : public PA_Allocat |
| struct curl_httppost *f_post; | struct curl_httppost *f_post; |
| FILE *f_stderr; | FILE *f_stderr; |
| // resolving | |
| struct curl_slist *resolve; | |
| bool parser_resolve; | |
| long resolve_family; | |
| long port; | |
| // if response content-length check required | // if response content-length check required |
| bool no_body; | bool no_body; |
| // stuff to walkaround curl request content-length bugs | // stuff to walkaround curl request content-length bugs |
| bool is_post; | bool is_post; |
| bool has_content_length; | bool has_content_length; |
| ParserOptions() : filename(0), content_type(0), is_text(true), charset(0), response_charset(0), url(0), f_post(0), f_stderr(0), no_body(false), is_post(false), has_content_length(false){} | ParserOptions() : filename(0), content_type(0), is_text(true), charset(0), response_charset(0), url(0), f_post(0), f_stderr(0), resolve(0), |
| parser_resolve(curl_parser_resolve), resolve_family(CURL_IPRESOLVE_V4), port(0), no_body(false), is_post(false), has_content_length(false){} | |
| ~ParserOptions() { | ~ParserOptions() { |
| f_curl_formfree(f_post); | f_curl_formfree(f_post); |
| if(f_stderr) | if(f_stderr) |
| Line 141 public: | Line 162 public: |
| } | } |
| }; | }; |
| #ifdef WIN32 | |
| #define CURL_LIBRARY "libcurl" LT_MODULE_EXT | |
| #else | |
| #define CURL_LIBRARY "libcurl" LT_MODULE_EXT ",libcurl" LT_MODULE_EXT ".4" | |
| #endif | |
| bool curl_linked = false; | bool curl_linked = false; |
| const char *curl_status = 0; | const char *curl_status = 0; |
| const char *curl_library=CURL_LIBRARY; | |
| static void temp_curl(void (*action)(Request&, MethodParams&), Request& r, MethodParams& params){ | static void temp_curl(void (*action)(Request&, MethodParams&), Request& r, MethodParams& params){ |
| if(!curl_linked) | if(!curl_linked) |
| curl_status=dlink(pa_strdup(curl_library)); | curl_status=dlink(pa_strdup(curl_library)); |
| Line 195 struct CurlOption : public PA_Allocated{ | Line 208 struct CurlOption : public PA_Allocated{ |
| CURL_POSTFIELDS, | CURL_POSTFIELDS, |
| CURL_FORM, | CURL_FORM, |
| CURL_HEADERS, | CURL_HEADERS, |
| CURL_RESOLVE, | |
| CURL_FILE, | CURL_FILE, |
| CURL_STDERR, | CURL_STDERR, |
| CURL_HTTP_VERSION, | CURL_HTTP_VERSION, |
| Line 241 public: | Line 255 public: |
| CURL_OPT(CURL_INT, FOLLOWLOCATION); | CURL_OPT(CURL_INT, FOLLOWLOCATION); |
| CURL_OPT(CURL_INT, UNRESTRICTED_AUTH); | CURL_OPT(CURL_INT, UNRESTRICTED_AUTH); |
| CURL_OPT(CURL_INT, IPRESOLVE); | CURL_OPT(CURL_INT, IPRESOLVE); |
| CURL_OPT(CURL_RESOLVE, RESOLVE); | |
| CURL_OPT(CURL_POST, POST); | CURL_OPT(CURL_POST, POST); |
| CURL_OPT(CURL_INT, HTTPGET); | CURL_OPT(CURL_INT, HTTPGET); |
| Line 389 static struct curl_slist *curl_headers(H | Line 404 static struct curl_slist *curl_headers(H |
| return slist; | return slist; |
| } | } |
| static struct curl_slist *curl_resolve(HashStringValue *value_hash) { | |
| struct curl_slist *slist=NULL; | |
| for(HashStringValue::Iterator i(*value_hash); i; i.next() ){ | |
| const char *key=i.key().cstr(); | |
| const String &address=i.value()->as_string(); | |
| if(address.is_empty() || strchr(key, ':')){ | |
| slist=f_curl_slist_append(slist, address.is_empty() ? key : pa_strcat(key, ":", address.cstr()) ); | |
| } else { | |
| // port-less specification covers both http and https ports by default | |
| slist=f_curl_slist_append(slist, pa_strcat(key, ":80:", address.cstr()) ); | |
| slist=f_curl_slist_append(slist, pa_strcat(key, ":443:", address.cstr()) ); | |
| } | |
| } | |
| return slist; | |
| } | |
| static const char* curl_transcode(const String &s, Request& r){ | static const char* curl_transcode(const String &s, Request& r){ |
| return options().charset ? Charset::transcode(s.cstr(), r.charsets.source(), *options().charset).cstr() : s.cstr(); | return options().charset ? Charset::transcode(s.cstr(), r.charsets.source(), *options().charset).cstr() : s.cstr(); |
| } | } |
| Line 491 static void curl_setopt(HashStringValue: | Line 524 static void curl_setopt(HashStringValue: |
| // integer curl option | // integer curl option |
| long value_int=(long)v.as_double(); | long value_int=(long)v.as_double(); |
| res=f_curl_easy_setopt(curl(), opt->id, value_int); | res=f_curl_easy_setopt(curl(), opt->id, value_int); |
| if(opt->id==CURLOPT_PORT) | |
| options().port=value_int; // can be used instead of :port in url, saving for 'parser' resolve mode | |
| if(opt->id==CURLOPT_IPRESOLVE) | |
| options().resolve_family=value_int; | |
| break; | break; |
| } | } |
| case CurlOption::CURL_NO_BODY:{ | case CurlOption::CURL_NO_BODY:{ |
| Line 539 static void curl_setopt(HashStringValue: | Line 576 static void curl_setopt(HashStringValue: |
| res=f_curl_easy_setopt(curl(), opt->id, value_hash ? curl_headers(value_hash, r) : 0); | res=f_curl_easy_setopt(curl(), opt->id, value_hash ? curl_headers(value_hash, r) : 0); |
| break; | break; |
| } | } |
| case CurlOption::CURL_RESOLVE:{ | |
| // 'curl|parser' resolver mode | |
| if(const String* mode=v.get_string()){ | |
| bool parser_resolve; | |
| if(*mode == "parser") | |
| parser_resolve=true; | |
| else if(*mode == "curl" || mode->is_empty()) | |
| parser_resolve=false; | |
| else | |
| throw Exception("curl", mode, "invalid resolve option value, must be 'parser' or 'curl'"); | |
| if(foptions) | |
| foptions->parser_resolve=parser_resolve; | |
| else | |
| curl_parser_resolve=parser_resolve; | |
| break; | |
| } | |
| // hash of $.[host:port][address[,address]] entries to populate DNS cache with, bypassing the resolver | |
| HashStringValue *value_hash=v.as_hash("failed to set option 'resolve': value"); | |
| res=f_curl_easy_setopt(curl(), opt->id, options().resolve=value_hash ? curl_resolve(value_hash) : 0); | |
| break; | |
| } | |
| case CurlOption::CURL_FILE:{ | case CurlOption::CURL_FILE:{ |
| // file-spec curl option | // file-spec curl option |
| const char *file_spec_cstr=curl_check_file(r.full_disk_path(v.as_string())); | const char *file_spec_cstr=curl_check_file(r.full_disk_path(v.as_string())); |
| Line 724 static int curl_header(char *data, size_ | Line 784 static int curl_header(char *data, size_ |
| throw Exception("curl", 0, "failed to set " message ": %s", f_curl_easy_strerror(res)); \ | throw Exception("curl", 0, "failed to set " message ": %s", f_curl_easy_strerror(res)); \ |
| } | } |
| // defined in inet.C | |
| const char *pa_name2ip(const char *name, int family, Table *table, const char *exception_type); | |
| static void pa_curl_parser_resolve(){ | |
| const char *url=options().url; | |
| if(!url) | |
| return; | |
| // url scheme was validated to be http:// or https:// before | |
| bool is_http=!pa_strncasecmp(url, "http://"); | |
| char *host=pa_strdup(url+(is_http ? 7 : 8)); | |
| if(char *uri=strpbrk(host, "/?#")) | |
| *uri=0; | |
| char *name=host; | |
| if(char *userinfo_host=lsplit(name, '@')) | |
| name=userinfo_host; | |
| if(*name=='[') // no resolving for IPv6 literal | |
| return; | |
| long port=options().port; | |
| if(char *port_cstr=lsplit(name, ':')) | |
| if(!port) | |
| port=atol(port_cstr); | |
| if(!port) | |
| port=is_http ? 80 : 443; | |
| if(!*name) | |
| return; | |
| const char *prefix=pa_strcat(name, ":", pa_itoa(port)); | |
| size_t prefix_length=strlen(prefix); | |
| // explicit entries have priority over auto-resolving + last duplicate wins in curl, so we should check | |
| for(struct curl_slist *e=options().resolve; e; e=e->next){ | |
| const char *data=e->data; | |
| // support optional '+host:port:address' curl 7.75+ syntax, '-host:port' should not match | |
| if(*data=='+') | |
| data++; | |
| if(!pa_strncasecmp(data, prefix, prefix_length) && data[prefix_length]==':') | |
| return; | |
| } | |
| int family=options().resolve_family==CURL_IPRESOLVE_V4 ? AF_INET : options().resolve_family==CURL_IPRESOLVE_V6 ? AF_INET6 : AF_UNSPEC; | |
| const char *address=pa_name2ip(name, family, 0, "curl.host"); | |
| if(!address) | |
| return; | |
| options().resolve=f_curl_slist_append(options().resolve, pa_strcat(prefix, ":", address)); | |
| CURLcode res; | |
| CURL_SETOPT(CURLOPT_RESOLVE, options().resolve, "curl resolve list"); | |
| } | |
| static void _curl_load_action(Request& r, MethodParams& params){ | static void _curl_load_action(Request& r, MethodParams& params){ |
| if(params.count()==1) | if(params.count()==1) |
| _curl_options(r, params); | _curl_options(r, params); |
| Line 739 static void _curl_load_action(Request& r | Line 854 static void _curl_load_action(Request& r |
| CURL_SETOPT(CURLOPT_WRITEFUNCTION, curl_writer, "curl writer function"); | CURL_SETOPT(CURLOPT_WRITEFUNCTION, curl_writer, "curl writer function"); |
| CURL_SETOPT(CURLOPT_WRITEDATA, &body, "curl write buffer"); | CURL_SETOPT(CURLOPT_WRITEDATA, &body, "curl write buffer"); |
| char error[CURL_ERROR_SIZE+1]=""; | |
| CURL_SETOPT(CURLOPT_ERRORBUFFER, error, "curl error buffer"); | |
| if(options().is_post && !options().has_content_length){ | if(options().is_post && !options().has_content_length){ |
| // libcurl bug walkaround. Prior to 7.38 (Debian Jessie) curl passed Content-length: -1 | // libcurl bug walkaround. Prior to 7.38 (Debian Jessie) curl passed Content-length: -1 |
| // after that no Content-length header is passed, that hangs request to nginx. | // after that no Content-length header is passed, that hangs request to nginx. |
| CURL_SETOPT(CURLOPT_POSTFIELDSIZE, 0, "post content-length"); | CURL_SETOPT(CURLOPT_POSTFIELDSIZE, 0, "post content-length"); |
| } | } |
| if(options().parser_resolve) | |
| pa_curl_parser_resolve(); | |
| ALTER_EXCEPTION_SOURCE(res=f_curl_easy_perform(curl()), new String(options().url)); | ALTER_EXCEPTION_SOURCE(res=f_curl_easy_perform(curl()), new String(options().url)); |
| if(res != CURLE_OK){ | if(res != CURLE_OK){ |
| const char *ex_type = 0; | const char *ex_type = 0; |
| Line 767 static void _curl_load_action(Request& r | Line 888 static void _curl_load_action(Request& r |
| check_file_size(response.content_length, new String(options().url)); break; | check_file_size(response.content_length, new String(options().url)); break; |
| default: break; | default: break; |
| } | } |
| throw Exception( PA_DEFAULT(ex_type, "curl.fail"), new String(options().url), "%s", f_curl_easy_strerror(res)); | throw Exception( PA_DEFAULT(ex_type, "curl.fail"), new String(options().url), "%s", error[0] ? error : f_curl_easy_strerror(res)); |
| } | } |
| // assure trailing zero | // assure trailing zero |