xref: /third_party/curl/tests/libtest/lib579.c (revision 13498266) 
1/***************************************************************************
2 *                                  _   _ ____  _
3 *  Project                     ___| | | |  _ \| |
4 *                             / __| | | | |_) | |
5 *                            | (__| |_| |  _ <| |___
6 *                             \___|\___/|_| \_\_____|
7 *
8 * Copyright (C) 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.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 * SPDX-License-Identifier: curl
22 *
23 ***************************************************************************/
24#include "test.h"
25
26#include "memdebug.h"
27
28static const char * const post[]={
29  "one",
30  "two",
31  "three",
32  "and a final longer crap: four",
33  NULL
34};
35
36
37struct WriteThis {
38  int counter;
39};
40
41static int progress_callback(void *clientp, double dltotal, double dlnow,
42                             double ultotal, double ulnow)
43{
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    FILE *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(char *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  curl = curl_easy_init();
102  if(!curl) {
103    fprintf(stderr, "curl_easy_init() failed\n");
104    curl_global_cleanup();
105    return TEST_ERR_MAJOR_BAD;
106  }
107
108  slist = curl_slist_append(slist, "Transfer-Encoding: chunked");
109  if(!slist) {
110    fprintf(stderr, "curl_slist_append() failed\n");
111    curl_easy_cleanup(curl);
112    curl_global_cleanup();
113    return TEST_ERR_MAJOR_BAD;
114  }
115
116  /* First set the URL that is about to receive our POST. */
117  test_setopt(curl, CURLOPT_URL, URL);
118
119  /* Now specify we want to POST data */
120  test_setopt(curl, CURLOPT_POST, 1L);
121
122  /* we want to use our own read function */
123  test_setopt(curl, CURLOPT_READFUNCTION, read_callback);
124
125  /* pointer to pass to our read function */
126  test_setopt(curl, CURLOPT_READDATA, &pooh);
127
128  /* get verbose debug output please */
129  test_setopt(curl, CURLOPT_VERBOSE, 1L);
130
131  /* include headers in the output */
132  test_setopt(curl, CURLOPT_HEADER, 1L);
133
134  /* enforce chunked transfer by setting the header */
135  test_setopt(curl, CURLOPT_HTTPHEADER, slist);
136
137  test_setopt(curl, CURLOPT_HTTPAUTH, (long)CURLAUTH_DIGEST);
138  test_setopt(curl, CURLOPT_USERPWD, "foo:bar");
139
140  /* we want to use our own progress function */
141  test_setopt(curl, CURLOPT_NOPROGRESS, 0L);
142  CURL_IGNORE_DEPRECATION(
143    test_setopt(curl, CURLOPT_PROGRESSFUNCTION, progress_callback);
144  )
145
146  /* Perform the request, res will get the return code */
147  res = curl_easy_perform(curl);
148
149test_cleanup:
150
151  /* clean up the headers list */
152  if(slist)
153    curl_slist_free_all(slist);
154
155  /* always cleanup */
156  curl_easy_cleanup(curl);
157  curl_global_cleanup();
158
159  return res;
160}
161