1/*************************************************************************** 2 * _ _ ____ _ 3 * Project ___| | | | _ \| | 4 * / __| | | | |_) | | 5 * | (__| |_| | _ <| |___ 6 * \___|\___/|_| \_\_____| 7 * 8 * Copyright (C) 1998 - 2011, 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 http://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 "test.h" 24 25#include "memdebug.h" 26 27static const char * const post[]={ 28 "one", 29 "two", 30 "three", 31 "and a final longer crap: four", 32 NULL 33}; 34 35 36struct WriteThis { 37 int counter; 38}; 39 40static int progress_callback(void *clientp, double dltotal, double dlnow, 41 double ultotal, double ulnow) 42{ 43 FILE *moo; 44 static int prev_ultotal = -1; 45 static int prev_ulnow = -1; 46 (void)clientp; /* UNUSED */ 47 (void)dltotal; /* UNUSED */ 48 (void)dlnow; /* UNUSED */ 49 50 /* to avoid depending on timing, which will cause this progress function to 51 get called a different number of times depending on circumstances, we 52 only log these lines if the numbers are different from the previous 53 invoke */ 54 if((prev_ultotal != (int)ultotal) || 55 (prev_ulnow != (int)ulnow)) { 56 57 moo = fopen(libtest_arg2, "ab"); 58 if(moo) { 59 fprintf(moo, "Progress callback called with UL %d out of %d\n", 60 (int)ulnow, (int)ultotal); 61 fclose(moo); 62 } 63 prev_ulnow = (int) ulnow; 64 prev_ultotal = (int) ultotal; 65 } 66 return 0; 67} 68 69static size_t read_callback(void *ptr, size_t size, size_t nmemb, void *userp) 70{ 71 struct WriteThis *pooh = (struct WriteThis *)userp; 72 const char *data; 73 74 if(size*nmemb < 1) 75 return 0; 76 77 data = post[pooh->counter]; 78 79 if(data) { 80 size_t len = strlen(data); 81 memcpy(ptr, data, len); 82 pooh->counter++; /* advance pointer */ 83 return len; 84 } 85 return 0; /* no more data left to deliver */ 86} 87 88int test(char *URL) 89{ 90 CURL *curl; 91 CURLcode res=CURLE_OK; 92 struct curl_slist *slist = NULL; 93 struct WriteThis pooh; 94 pooh.counter = 0; 95 96 if (curl_global_init(CURL_GLOBAL_ALL) != CURLE_OK) { 97 fprintf(stderr, "curl_global_init() failed\n"); 98 return TEST_ERR_MAJOR_BAD; 99 } 100 101 if ((curl = curl_easy_init()) == NULL) { 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_INFILE, &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, 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 151test_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} 163