lh | 9ed821d | 2023-04-07 01:36:19 -0700 | [diff] [blame] | 1 | /*************************************************************************** |
| 2 | * _ _ ____ _ |
| 3 | * Project ___| | | | _ \| | |
| 4 | * / __| | | | |_) | | |
| 5 | * | (__| |_| | _ <| |___ |
| 6 | * \___|\___/|_| \_\_____| |
| 7 | * |
| 8 | * Copyright (C) 1998 - 2016, 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 | #include "test.h" |
| 23 | |
| 24 | #include "memdebug.h" |
| 25 | |
| 26 | static const char * const post[]={ |
| 27 | "one", |
| 28 | "two", |
| 29 | "three", |
| 30 | "and a final longer crap: four", |
| 31 | NULL |
| 32 | }; |
| 33 | |
| 34 | |
| 35 | struct WriteThis { |
| 36 | int counter; |
| 37 | }; |
| 38 | |
| 39 | static int progress_callback(void *clientp, double dltotal, double dlnow, |
| 40 | double ultotal, double ulnow) |
| 41 | { |
| 42 | FILE *moo; |
| 43 | static int prev_ultotal = -1; |
| 44 | static int prev_ulnow = -1; |
| 45 | (void)clientp; /* UNUSED */ |
| 46 | (void)dltotal; /* UNUSED */ |
| 47 | (void)dlnow; /* UNUSED */ |
| 48 | |
| 49 | /* to avoid depending on timing, which will cause this progress function to |
| 50 | get called a different number of times depending on circumstances, we |
| 51 | only log these lines if the numbers are different from the previous |
| 52 | invoke */ |
| 53 | if((prev_ultotal != (int)ultotal) || |
| 54 | (prev_ulnow != (int)ulnow)) { |
| 55 | |
| 56 | moo = fopen(libtest_arg2, "ab"); |
| 57 | if(moo) { |
| 58 | fprintf(moo, "Progress callback called with UL %d out of %d\n", |
| 59 | (int)ulnow, (int)ultotal); |
| 60 | fclose(moo); |
| 61 | } |
| 62 | prev_ulnow = (int) ulnow; |
| 63 | prev_ultotal = (int) ultotal; |
| 64 | } |
| 65 | return 0; |
| 66 | } |
| 67 | |
| 68 | static size_t read_callback(void *ptr, size_t size, size_t nmemb, void *userp) |
| 69 | { |
| 70 | struct WriteThis *pooh = (struct WriteThis *)userp; |
| 71 | const char *data; |
| 72 | |
| 73 | if(size*nmemb < 1) |
| 74 | return 0; |
| 75 | |
| 76 | data = post[pooh->counter]; |
| 77 | |
| 78 | if(data) { |
| 79 | size_t len = strlen(data); |
| 80 | memcpy(ptr, data, len); |
| 81 | pooh->counter++; /* advance pointer */ |
| 82 | return len; |
| 83 | } |
| 84 | return 0; /* no more data left to deliver */ |
| 85 | } |
| 86 | |
| 87 | int test(char *URL) |
| 88 | { |
| 89 | CURL *curl; |
| 90 | CURLcode res=CURLE_OK; |
| 91 | struct curl_slist *slist = NULL; |
| 92 | struct WriteThis pooh; |
| 93 | pooh.counter = 0; |
| 94 | |
| 95 | if(curl_global_init(CURL_GLOBAL_ALL) != CURLE_OK) { |
| 96 | fprintf(stderr, "curl_global_init() failed\n"); |
| 97 | return TEST_ERR_MAJOR_BAD; |
| 98 | } |
| 99 | |
| 100 | curl = curl_easy_init(); |
| 101 | if(!curl) { |
| 102 | fprintf(stderr, "curl_easy_init() failed\n"); |
| 103 | curl_global_cleanup(); |
| 104 | return TEST_ERR_MAJOR_BAD; |
| 105 | } |
| 106 | |
| 107 | slist = curl_slist_append(slist, "Transfer-Encoding: chunked"); |
| 108 | if(slist == NULL) { |
| 109 | fprintf(stderr, "curl_slist_append() failed\n"); |
| 110 | curl_easy_cleanup(curl); |
| 111 | curl_global_cleanup(); |
| 112 | return TEST_ERR_MAJOR_BAD; |
| 113 | } |
| 114 | |
| 115 | /* First set the URL that is about to receive our POST. */ |
| 116 | test_setopt(curl, CURLOPT_URL, URL); |
| 117 | |
| 118 | /* Now specify we want to POST data */ |
| 119 | test_setopt(curl, CURLOPT_POST, 1L); |
| 120 | |
| 121 | #ifdef CURL_DOES_CONVERSIONS |
| 122 | /* Convert the POST data to ASCII */ |
| 123 | test_setopt(curl, CURLOPT_TRANSFERTEXT, 1L); |
| 124 | #endif |
| 125 | |
| 126 | /* we want to use our own read function */ |
| 127 | test_setopt(curl, CURLOPT_READFUNCTION, read_callback); |
| 128 | |
| 129 | /* pointer to pass to our read function */ |
| 130 | test_setopt(curl, CURLOPT_READDATA, &pooh); |
| 131 | |
| 132 | /* get verbose debug output please */ |
| 133 | test_setopt(curl, CURLOPT_VERBOSE, 1L); |
| 134 | |
| 135 | /* include headers in the output */ |
| 136 | test_setopt(curl, CURLOPT_HEADER, 1L); |
| 137 | |
| 138 | /* enforce chunked transfer by setting the header */ |
| 139 | test_setopt(curl, CURLOPT_HTTPHEADER, slist); |
| 140 | |
| 141 | test_setopt(curl, CURLOPT_HTTPAUTH, (long)CURLAUTH_DIGEST); |
| 142 | test_setopt(curl, CURLOPT_USERPWD, "foo:bar"); |
| 143 | |
| 144 | /* we want to use our own progress function */ |
| 145 | test_setopt(curl, CURLOPT_NOPROGRESS, 0L); |
| 146 | test_setopt(curl, CURLOPT_PROGRESSFUNCTION, progress_callback); |
| 147 | |
| 148 | /* Perform the request, res will get the return code */ |
| 149 | res = curl_easy_perform(curl); |
| 150 | |
| 151 | test_cleanup: |
| 152 | |
| 153 | /* clean up the headers list */ |
| 154 | if(slist) |
| 155 | curl_slist_free_all(slist); |
| 156 | |
| 157 | /* always cleanup */ |
| 158 | curl_easy_cleanup(curl); |
| 159 | curl_global_cleanup(); |
| 160 | |
| 161 | return res; |
| 162 | } |