diff options
Diffstat (limited to 'src')
| -rw-r--r-- | src/tool_cb_hdr.c | 46 | ||||
| -rw-r--r-- | src/tool_cb_hdr.h | 25 | ||||
| -rw-r--r-- | src/tool_operate.c | 21 |
3 files changed, 55 insertions, 37 deletions
diff --git a/src/tool_cb_hdr.c b/src/tool_cb_hdr.c index ddc4c6638..ef340f798 100644 --- a/src/tool_cb_hdr.c +++ b/src/tool_cb_hdr.c @@ -41,10 +41,9 @@ static char *parse_filename(const char *ptr, size_t len); size_t tool_header_cb(void *ptr, size_t size, size_t nmemb, void *userdata) { - HeaderData *hdrdata = userdata; - struct getout *urlnode = hdrdata->urlnode; - struct OutStruct *outs = hdrdata->outs; - struct OutStruct *heads = hdrdata->heads; + struct HdrCbData *hdrcbdata = userdata; + struct OutStruct *outs = hdrcbdata->outs; + struct OutStruct *heads = hdrcbdata->heads; const char *str = ptr; const size_t cb = size * nmemb; const char *end = (char*)ptr + cb; @@ -57,39 +56,35 @@ size_t tool_header_cb(void *ptr, size_t size, size_t nmemb, void *userdata) */ size_t failure = (size * nmemb) ? 0 : 1; - if(!outs->config) + if(!heads->config) return failure; #ifdef DEBUGBUILD if(size * nmemb > (size_t)CURL_MAX_HTTP_HEADER) { - warnf(outs->config, "Header data exceeds single call write limit!\n"); + warnf(heads->config, "Header data exceeds single call write limit!\n"); return failure; } #endif - /* --dump-header option */ - if(outs->config->headerfile) { - fwrite(ptr, size, nmemb, heads->stream); - } /* - ** This callback callback MIGHT set the filename upon appropriate - ** conditions and server specifying filename in Content-Disposition. - */ - - if(!outs->config->content_disposition) - return cb; - - if(!urlnode) - return failure; + * Write header data when curl option --dump-header (-D) is given. + */ - if(!checkprefix("http://", urlnode->url) && - !checkprefix("https://", urlnode->url)) - return cb; + if(heads->config->headerfile && heads->stream) { + size_t rc = fwrite(ptr, size, nmemb, heads->stream); + if(rc != cb) + return rc; + } - if(!(urlnode->flags & GETOUT_USEREMOTE)) - return cb; + /* + * This callback sets the filename where output shall be written when + * curl options --remote-name (-O) and --remote-header-name (-J) have + * been simultaneously given and additionally server returns an HTTP + * Content-Disposition header specifying a filename property. + */ - if((cb > 20) && checkprefix("Content-disposition:", str)) { + if(hdrcbdata->honor_cd_filename && + (cb > 20) && checkprefix("Content-disposition:", str)) { const char *p = str + 20; /* look for the 'filename=' parameter @@ -123,6 +118,7 @@ size_t tool_header_cb(void *ptr, size_t size, size_t nmemb, void *userdata) outs->s_isreg = TRUE; outs->fopened = FALSE; outs->stream = NULL; + hdrcbdata->honor_cd_filename = FALSE; break; } else diff --git a/src/tool_cb_hdr.h b/src/tool_cb_hdr.h index 0300c0068..bd5043139 100644 --- a/src/tool_cb_hdr.h +++ b/src/tool_cb_hdr.h @@ -23,15 +23,26 @@ ***************************************************************************/ #include "tool_setup.h" -/* Structure to pass as userdata in tool_header_cb */ -typedef struct { - /* getout object pointer currently processing */ - struct getout *urlnode; - /* output stream */ +/* + * curl operates using a single HdrCbData struct variable, a + * pointer to this is passed as userdata pointer to tool_header_cb. + * + * 'outs' member is a pointer to the OutStruct variable used to keep + * track of information relative to curl's output writing. + * + * 'heads' member is a pointer to the OutStruct variable used to keep + * track of information relative to header response writing. + * + * 'honor_cd_filename' member is TRUE when tool_header_cb is allowed + * to honor Content-Disposition filename property and accordingly + * set 'outs' filename, otherwise FALSE; + */ + +struct HdrCbData { struct OutStruct *outs; - /* header output stream */ struct OutStruct *heads; -} HeaderData; + bool honor_cd_filename; +}; /* ** callback for CURLOPT_HEADERFUNCTION diff --git a/src/tool_operate.c b/src/tool_operate.c index d01e38326..8a5569bd8 100644 --- a/src/tool_operate.c +++ b/src/tool_operate.c @@ -128,6 +128,7 @@ int operate(struct Configurable *config, int argc, argv_item_t argv[]) struct ProgressData progressbar; struct getout *urlnode; + struct HdrCbData hdrcbdata; struct OutStruct heads; metalinkfile *mlfile_last = NULL; @@ -141,6 +142,7 @@ int operate(struct Configurable *config, int argc, argv_item_t argv[]) errorbuffer[0] = '\0'; /* default headers output stream is stdout */ + memset(&hdrcbdata, 0, sizeof(struct HdrCbData)); memset(&heads, 0, sizeof(struct OutStruct)); heads.stream = stdout; heads.config = config; @@ -531,7 +533,6 @@ int operate(struct Configurable *config, int argc, argv_item_t argv[]) long retry_sleep_default; long retry_sleep; char *this_url; - HeaderData hdrdata; int metalink_next_res = 0; outfile = NULL; @@ -1256,12 +1257,19 @@ int operate(struct Configurable *config, int argc, argv_item_t argv[]) if(config->proto_redir_present) my_setopt_flags(curl, CURLOPT_REDIR_PROTOCOLS, config->proto_redir); - hdrdata.urlnode = urlnode; - hdrdata.outs = &outs; - hdrdata.heads = &heads; + if(config->content_disposition + && (urlnode->flags & GETOUT_USEREMOTE) + && (checkprefix("http://", this_url) || + checkprefix("https://", this_url))) + hdrcbdata.honor_cd_filename = TRUE; + else + hdrcbdata.honor_cd_filename = FALSE; + + hdrcbdata.outs = &outs; + hdrcbdata.heads = &heads; my_setopt(curl, CURLOPT_HEADERFUNCTION, tool_header_cb); - my_setopt(curl, CURLOPT_HEADERDATA, &hdrdata); + my_setopt(curl, CURLOPT_HEADERDATA, &hdrcbdata); if(config->resolve) /* new in 7.21.3 */ @@ -1590,6 +1598,7 @@ int operate(struct Configurable *config, int argc, argv_item_t argv[]) if(outs.alloc_filename) Curl_safefree(outs.filename); memset(&outs, 0, sizeof(struct OutStruct)); + hdrcbdata.outs = NULL; /* Free loop-local allocated memory and close loop-local opened fd */ @@ -1696,6 +1705,8 @@ int operate(struct Configurable *config, int argc, argv_item_t argv[]) easysrc_cleanup(); #endif + hdrcbdata.heads = NULL; + /* Close function-local opened file descriptors */ if(heads.fopened && heads.stream) |