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
progress_callback(void * clientp,double dltotal,double dlnow,double ultotal,double ulnow)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
read_callback(void * ptr,size_t size,size_t nmemb,void * userp)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
test(char * URL)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 if((curl = curl_easy_init()) == NULL) {
101 fprintf(stderr, "curl_easy_init() failed\n");
102 curl_global_cleanup();
103 return TEST_ERR_MAJOR_BAD;
104 }
105
106 slist = curl_slist_append(slist, "Transfer-Encoding: chunked");
107 if(slist == NULL) {
108 fprintf(stderr, "curl_slist_append() failed\n");
109 curl_easy_cleanup(curl);
110 curl_global_cleanup();
111 return TEST_ERR_MAJOR_BAD;
112 }
113
114 /* First set the URL that is about to receive our POST. */
115 test_setopt(curl, CURLOPT_URL, URL);
116
117 /* Now specify we want to POST data */
118 test_setopt(curl, CURLOPT_POST, 1L);
119
120 #ifdef CURL_DOES_CONVERSIONS
121 /* Convert the POST data to ASCII */
122 test_setopt(curl, CURLOPT_TRANSFERTEXT, 1L);
123 #endif
124
125 /* we want to use our own read function */
126 test_setopt(curl, CURLOPT_READFUNCTION, read_callback);
127
128 /* pointer to pass to our read function */
129 test_setopt(curl, CURLOPT_READDATA, &pooh);
130
131 /* get verbose debug output please */
132 test_setopt(curl, CURLOPT_VERBOSE, 1L);
133
134 /* include headers in the output */
135 test_setopt(curl, CURLOPT_HEADER, 1L);
136
137 /* enforce chunked transfer by setting the header */
138 test_setopt(curl, CURLOPT_HTTPHEADER, slist);
139
140 test_setopt(curl, CURLOPT_HTTPAUTH, (long)CURLAUTH_DIGEST);
141 test_setopt(curl, CURLOPT_USERPWD, "foo:bar");
142
143 /* we want to use our own progress function */
144 test_setopt(curl, CURLOPT_NOPROGRESS, 0L);
145 test_setopt(curl, CURLOPT_PROGRESSFUNCTION, progress_callback);
146
147 /* Perform the request, res will get the return code */
148 res = curl_easy_perform(curl);
149
150 test_cleanup:
151
152 /* clean up the headers list */
153 if(slist)
154 curl_slist_free_all(slist);
155
156 /* always cleanup */
157 curl_easy_cleanup(curl);
158 curl_global_cleanup();
159
160 return res;
161 }
162