Elliott Hughes | cac3980 | 2018-04-27 16:19:43 -0700 | [diff] [blame] | 1 | /*************************************************************************** |
| 2 | * _ _ ____ _ |
| 3 | * Project ___| | | | _ \| | |
| 4 | * / __| | | | |_) | | |
| 5 | * | (__| |_| | _ <| |___ |
| 6 | * \___|\___/|_| \_\_____| |
| 7 | * |
| 8 | * Copyright (C) 1998 - 2018, Daniel Stenberg, <daniel@haxx.se>, et al. |
| 9 | * |
| 10 | * This software is licensed as described in the file COPYING, which |
| 11 | * you should have received as part of this distribution. The terms |
| 12 | * are also available at https://curl.haxx.se/docs/copyright.html. |
| 13 | * |
| 14 | * You may opt to use, copy, modify, merge, publish, distribute and/or sell |
| 15 | * copies of the Software, and permit persons to whom the Software is |
| 16 | * furnished to do so, under the terms of the COPYING file. |
| 17 | * |
| 18 | * This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY |
| 19 | * KIND, either express or implied. |
| 20 | * |
| 21 | ***************************************************************************/ |
| 22 | |
| 23 | #include "curl_setup.h" |
| 24 | #include <curl/curl.h> |
| 25 | #include "curl_range.h" |
| 26 | #include "sendf.h" |
| 27 | #include "strtoofft.h" |
| 28 | |
| 29 | /* Only include this function if one or more of FTP, FILE are enabled. */ |
| 30 | #if !defined(CURL_DISABLE_FTP) || !defined(CURL_DISABLE_FILE) |
| 31 | |
| 32 | /* |
| 33 | Check if this is a range download, and if so, set the internal variables |
| 34 | properly. |
| 35 | */ |
| 36 | CURLcode Curl_range(struct connectdata *conn) |
| 37 | { |
| 38 | curl_off_t from, to; |
| 39 | char *ptr; |
| 40 | char *ptr2; |
| 41 | struct Curl_easy *data = conn->data; |
| 42 | |
| 43 | if(data->state.use_range && data->state.range) { |
| 44 | CURLofft from_t; |
| 45 | CURLofft to_t; |
| 46 | from_t = curlx_strtoofft(data->state.range, &ptr, 0, &from); |
| 47 | if(from_t == CURL_OFFT_FLOW) |
| 48 | return CURLE_RANGE_ERROR; |
| 49 | while(*ptr && (ISSPACE(*ptr) || (*ptr == '-'))) |
| 50 | ptr++; |
| 51 | to_t = curlx_strtoofft(ptr, &ptr2, 0, &to); |
| 52 | if(to_t == CURL_OFFT_FLOW) |
| 53 | return CURLE_RANGE_ERROR; |
| 54 | if((to_t == CURL_OFFT_INVAL) && !from_t) { |
| 55 | /* X - */ |
| 56 | data->state.resume_from = from; |
| 57 | DEBUGF(infof(data, "RANGE %" CURL_FORMAT_CURL_OFF_T " to end of file\n", |
| 58 | from)); |
| 59 | } |
| 60 | else if((from_t == CURL_OFFT_INVAL) && !to_t) { |
| 61 | /* -Y */ |
| 62 | data->req.maxdownload = to; |
| 63 | data->state.resume_from = -to; |
| 64 | DEBUGF(infof(data, "RANGE the last %" CURL_FORMAT_CURL_OFF_T " bytes\n", |
| 65 | to)); |
| 66 | } |
| 67 | else { |
| 68 | /* X-Y */ |
| 69 | curl_off_t totalsize; |
| 70 | |
| 71 | /* Ensure the range is sensible - to should follow from. */ |
| 72 | if(from > to) |
| 73 | return CURLE_RANGE_ERROR; |
| 74 | |
| 75 | totalsize = to - from; |
| 76 | if(totalsize == CURL_OFF_T_MAX) |
| 77 | return CURLE_RANGE_ERROR; |
| 78 | |
| 79 | data->req.maxdownload = totalsize + 1; /* include last byte */ |
| 80 | data->state.resume_from = from; |
| 81 | DEBUGF(infof(data, "RANGE from %" CURL_FORMAT_CURL_OFF_T |
| 82 | " getting %" CURL_FORMAT_CURL_OFF_T " bytes\n", |
| 83 | from, data->req.maxdownload)); |
| 84 | } |
| 85 | DEBUGF(infof(data, "range-download from %" CURL_FORMAT_CURL_OFF_T |
| 86 | " to %" CURL_FORMAT_CURL_OFF_T ", totally %" |
| 87 | CURL_FORMAT_CURL_OFF_T " bytes\n", |
| 88 | from, to, data->req.maxdownload)); |
| 89 | } |
| 90 | else |
| 91 | data->req.maxdownload = -1; |
| 92 | return CURLE_OK; |
| 93 | } |
| 94 | |
| 95 | #endif |