Finish large file download attack prevention

This handles the no Content-Length header problem as stated in the
comments of FS#23413. We add a quick check to the callback that will
force an abort if the downloaded data exceeds the payload size, and then
check for this error in the post-download cleanup code.

Signed-off-by: Dan McGee <dan@archlinux.org>
This commit is contained in:
Dan McGee 2011-08-24 19:58:14 -05:00
parent 2e7d002315
commit f7a3c4c8df

View file

@ -83,12 +83,18 @@ static int curl_progress(void *file, double dltotal, double dlnow,
return 1; return 1;
} }
current_size = payload->initial_size + (off_t)dlnow;
/* is our filesize still under any set limit? */
if(payload->max_size && current_size > payload->max_size) {
return 1;
}
/* none of what follows matters if the front end has no callback */ /* none of what follows matters if the front end has no callback */
if(payload->handle->dlcb == NULL) { if(payload->handle->dlcb == NULL) {
return 0; return 0;
} }
current_size = payload->initial_size + (off_t)dlnow;
total_size = payload->initial_size + (off_t)dltotal; total_size = payload->initial_size + (off_t)dltotal;
if(DOUBLE_EQ(dltotal, 0.0) || prevprogress == total_size) { if(DOUBLE_EQ(dltotal, 0.0) || prevprogress == total_size) {
@ -341,6 +347,15 @@ static int curl_download_internal(struct dload_payload *payload,
} }
break; break;
case CURLE_ABORTED_BY_CALLBACK: case CURLE_ABORTED_BY_CALLBACK:
/* two cases here- interrupted by user, or we exceeded max file size. */
if(!dload_interrupted) {
handle->curlerr = CURLE_FILESIZE_EXCEEDED;
handle->pm_errno = ALPM_ERR_LIBCURL;
/* the hardcoded 'size exceeded' message is same as libcurl's normal */
_alpm_log(handle, ALPM_LOG_ERROR,
_("failed retrieving file '%s' from %s : %s\n"),
payload->remote_name, hostname, "Maximum file size exceeded");
}
goto cleanup; goto cleanup;
default: default:
/* delete zero length downloads */ /* delete zero length downloads */
@ -349,10 +364,12 @@ static int curl_download_internal(struct dload_payload *payload,
} }
if(!payload->errors_ok) { if(!payload->errors_ok) {
handle->pm_errno = ALPM_ERR_LIBCURL; handle->pm_errno = ALPM_ERR_LIBCURL;
_alpm_log(handle, ALPM_LOG_ERROR, _("failed retrieving file '%s' from %s : %s\n"), _alpm_log(handle, ALPM_LOG_ERROR,
_("failed retrieving file '%s' from %s : %s\n"),
payload->remote_name, hostname, error_buffer); payload->remote_name, hostname, error_buffer);
} else { } else {
_alpm_log(handle, ALPM_LOG_DEBUG, "failed retrieving file '%s' from %s : %s\n", _alpm_log(handle, ALPM_LOG_DEBUG,
"failed retrieving file '%s' from %s : %s\n",
payload->remote_name, hostname, error_buffer); payload->remote_name, hostname, error_buffer);
} }
goto cleanup; goto cleanup;